wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-02 16:14:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updated to src implementation of SimPEG.

Browse Source
This commit is contained in:
GudniRos
2015-05-01 13:01:13 -07:00
parent 25799f3680
commit 6eabd68a1d
2 changed files with 7 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simpegMT/ProblemMT.py
+2 -2
View File
@@ -114,7 +114,7 @@ class MTProblem(Problem.BaseProblem):
print 'Starting work for {:.3e}'.format(freq)
sys.stdout.flush()
A = self.getA(freq)
rhs = self.getRHS(freq,m_back)
rhs, ep = self.getRHS(freq,m_back)
Ainv = self.Solver(A, **self.solverOpts)
e = Ainv * rhs
@@ -198,7 +198,7 @@ class MTProblem(Problem.BaseProblem):
eBG_bp = homo1DModelSource(self.mesh,freq,backSigma)
Abg = self.getAbg(freq)
return -Abg*eBG_bp
return -Abg*eBG_bp, eBG_bp
##################################################################
# Inversion stuff
simpegMT/SurveyMT.py
+5 -15
View File
@@ -131,10 +131,10 @@ class RxMT(Survey.BaseRx):
# Call this Source or polarization or something...?
# Note: Might need to add tests to make sure that both polarization have the same rxList.
class srcMT(Survey.BaseTx):
class srcMT(Survey.BaseSrc):
'''
Sources for the MT problem.
Use the SimPEG BaseTx, since the source fields share properties with the transmitters.
Use the SimPEG BaseSrc, since the source fields share properties with the transmitters.
:param float freq: The frequency of the source
:param list rxList: A list of receivers associated with the source
@@ -145,12 +145,11 @@ class srcMT(Survey.BaseTx):
rxPair = RxMT
knownTxTypes = ['pol_xy','pol_x','pol_y'] # ORThogonal POLarization
knownSrcTypes = ['pol_xy','pol_x','pol_y'] # ORThogonal POLarization
def __init__(self, freq, rxList, srcPol = 'pol_xy'): # remove txType? hardcode to one thing. always polarizations
def __init__(self, freq, rxList, srcPol = 'pol_xy'): # remove rxType? hardcode to one thing. always polarizations
self.freq = float(freq)
Survey.BaseTx.__init__(self, None, srcPol, rxList)
# Survey.BaseTx.__init__(self, loc, 'polarization', rxList)
Survey.BaseSrc.__init__(self, None, srcPol, rxList)
@@ -173,7 +172,6 @@ class SurveyMT(Survey.BaseSurvey):
def __init__(self, srcList, **kwargs):
# Sort these by frequency
self.srcList = srcList
self.txList = self.srcList # Hack - make txList index srcList, since it is used in the backend.
Survey.BaseSurvey.__init__(self, **kwargs)
_freqDict = {}
@@ -194,14 +192,6 @@ class SurveyMT(Survey.BaseSurvey):
def nFreq(self):
"""Number of frequencies"""
return len(self._freqDict)
# Don't need this
# @property
# def nTxByFreq(self):
# if getattr(self, '_nTxByFreq', None) is None:
# self._nTxByFreq = {}
# for freq in self.freqs:
# self._nTxByFreq[freq] = len(self.getTransmitters(freq))
# return self._nTxByFreq
# TODO: Rename to getSources
def getSources(self, freq):