start of the FDEM

simpegEM/FDEM/DataFDEM.py

@@ -0,0 +1,50 @@
+from SimPEG import Utils, np
+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)
+
+    def projectFields(self, u):
+        #TODO: this is hardcoded to 1Tx
+        return self.Qrx.dot(u.b[:,:,0].T).T
+
+    def projectFieldsAdjoint(self, d):
+        # TODO: fix this
+        pass
+
+    ####################################################
+    # 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