mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-28 23:10:14 +08:00
Lindsey Heagy
61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
from SimPEG import Utils, np, sp
|
|
from SimPEG.Data import BaseData
|
|
from FieldsFDEM import FieldsFDEM
|
|
|
|
class DataFDEM(BaseData):
|
|
"""
|
|
docstring for DataFDEM
|
|
"""
|
|
|
|
txLoc = None #: txLoc
|
|
txType = None #: txType
|
|
nTx = 1 #: Number of transmitters
|
|
rxLoc = None #: rxLoc
|
|
rxType = None #: rxType
|
|
freq = None #: freq
|
|
|
|
|
|
@property
|
|
def omega(self):
|
|
return 2*np.pi*self.freq
|
|
|
|
@property
|
|
def nFreq(self):
|
|
"""Number of frequencies"""
|
|
return self.freq.size
|
|
|
|
def __init__(self, **kwargs):
|
|
BaseData.__init__(self, **kwargs)
|
|
Utils.setKwargs(self, **kwargs)
|
|
|
|
@property
|
|
def nRx(self):
|
|
return self.rxLoc.shape[0]
|
|
|
|
def projectFields(self, u):
|
|
P = sp.identity(self.prob.mesh.nE)
|
|
Pes = range(self.nFreq)
|
|
#TODO: this is hardcoded to 1Tx
|
|
for i, freqInd in enumerate(range(self.nFreq)):
|
|
e = u.get_e(freqInd)
|
|
Pes[i] = P*e
|
|
Pe = np.concatenate(Pes)
|
|
return Pe
|
|
|
|
def projectFieldsDeriv(self, u):
|
|
# TODO : more general
|
|
return sp.identity(self.prob.mesh.nE)
|
|
|
|
####################################################
|
|
# Interpolation Matrices
|
|
####################################################
|
|
|
|
@property
|
|
def Qrx(self):
|
|
if self._Qrx is None:
|
|
if self.rxType == 'bz':
|
|
locType = 'Fz'
|
|
self._Qrx = self.prob.mesh.getInterpolationMat(self.rxLoc, locType=locType)
|
|
return self._Qrx
|
|
_Qrx = None
|