wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-05 11:26:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

getSources --> getSource

Browse Source
This commit is contained in:
Lindsey
2015-05-04 11:29:17 -07:00
parent c844f7b36a
commit 47fca47b33
2 changed files with 21 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simpegEM/FDEM/FDEM.py
+19 -4
View File
@@ -28,7 +28,7 @@ class BaseFDEMProblem(BaseEMProblem):
rhs = RHS(freq)
Ainv = self.Solver(A, **self.solverOpts)
sol = Ainv * rhs
Srcs = self.survey.getSources(freq)
Srcs = self.survey.getSource(freq)
F[Srcs, self._fieldType] = sol
return F
@@ -45,7 +45,7 @@ class BaseFDEMProblem(BaseEMProblem):
A = self.getA(freq)
Ainv = self.Solver(A, **self.solverOpts)
for src in self.survey.getSources(freq):
for src in self.survey.getSource(freq):
u_src = u[src, self.solType]
w = self.getADeriv(freq, u_src, v)
Ainvw = Ainv * w
@@ -77,7 +77,7 @@ class BaseFDEMProblem(BaseEMProblem):
AT = self.getA(freq).T
ATinv = self.Solver(AT, **self.solverOpts)
for src in self.survey.getSources(freq):
for src in self.survey.getSource(freq):
u_src = u[src, self.solType]
for rx in src.rxList:
@@ -108,7 +108,7 @@ class BaseFDEMProblem(BaseEMProblem):
:rtype: numpy.ndarray (nE or nF, nSrc)
:return: RHS
"""
Srcs = self.survey.getSources(freq)
Srcs = self.survey.getSource(freq)
if self._eqLocs is 'FE':
S_m = np.zeros((self.mesh.nF,len(Srcs)), dtype=complex)
S_e = np.zeros((self.mesh.nE,len(Srcs)), dtype=complex)
@@ -126,6 +126,21 @@ class BaseFDEMProblem(BaseEMProblem):
return S_m, S_e
def getSourceDeriv(self,freq,adjoint=False):
Srcs = self.survey.getSource(freq)
if self._eqLocs is 'FE':
S_m = np.zeros((self.mesh.nF,len(Srcs)), dtype=complex)
S_e = np.zeros((self.mesh.nE,len(Srcs)), dtype=complex)
elif self._eqLocs is 'EF':
S_m = np.zeros((self.mesh.nE,len(Srcs)), dtype=complex)
S_e = np.zeros((self.mesh.nF,len(Srcs)), dtype=complex)
for i, src in enumerate(Srcs):
smi, sei = src.getSourceDeriv(self)
if smi is not None:
S_m[:,i] = smi
if sei is not None:
S_e[:,i] = sei
raise NotImplementedError('getSourceDeriv not implemented yet')
return None, None
simpegEM/FDEM/SurveyFDEM.py
+2 -2
View File
@@ -331,10 +331,10 @@ class SurveyFDEM(Survey.BaseSurvey):
if getattr(self, '_nSrcByFreq', None) is None:
self._nSrcByFreq = {}
for freq in self.freqs:
self._nSrcByFreq[freq] = len(self.getSources(freq))
self._nSrcByFreq[freq] = len(self.getSource(freq))
return self._nSrcByFreq
def getSources(self, freq):
def getSource(self, freq):
"""Returns the sources associated with a specific frequency."""
assert freq in self._freqDict, "The requested frequency is not in this survey."
return self._freqDict[freq]