From 059d318ab619723758ff95f75ad21ce5ac7b0fc6 Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Sun, 18 Oct 2015 14:45:37 -0700 Subject: [PATCH] Primary fields kwargs for S_m --- simpegEM/FDEM/SurveyFDEM.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/simpegEM/FDEM/SurveyFDEM.py b/simpegEM/FDEM/SurveyFDEM.py index 8c357e97..9fdf23d2 100644 --- a/simpegEM/FDEM/SurveyFDEM.py +++ b/simpegEM/FDEM/SurveyFDEM.py @@ -173,14 +173,31 @@ class SrcFDEM_RawVec_m(SrcFDEM): """ def __init__(self, rxList, freq, S_m, integrate = True): - self._S_m = np.array(S_m,dtype=complex) + self._S_m = np.array(S_m,dtype=complex, ePrimary=None, bPrimary=None, hPrimary=None, jPrimary=None) self.freq = float(freq) self.integrate = integrate + self._ePrimary = ePrimary + self._bPrimary = bPrimary + self._hPrimary = hPrimary + self._jPrimary = jPrimary + SrcFDEM.__init__(self, rxList) def S_m(self, prob): return self._S_m + def ePrimary(self, prob): + return self._ePrimary + + def bPrimary(self, prob): + return self._bPrimary + + def hPrimary(self, prob): + return self._hPrimary + + def jPrimary(self, prob): + return self._jPrimary + class SrcFDEM_RawVec(SrcFDEM): """