add e,b,h,j Primary to kwargs for raw e src

This commit is contained in:
Lindsey Heagy
2015-10-18 11:15:23 -07:00
parent 1cb06297ec
commit 670511eb61
+17 -1
View File
@@ -138,14 +138,30 @@ class SrcFDEM_RawVec_e(SrcFDEM):
:param rxList: receiver list
"""
def __init__(self, rxList, freq, S_e):
def __init__(self, rxList, freq, S_e, ePrimary=None, bPrimary=None, hPrimary=None, jPrimary=None):
self._S_e = np.array(S_e,dtype=complex)
self._ePrimary = ePrimary
self._bPrimary = bPrimary
self._hPrimary = hPrimary
self._jPrimary = jPrimary
self.freq = float(freq)
SrcFDEM.__init__(self, rxList)
def S_e(self, prob):
return self._S_e
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_m(SrcFDEM):
"""