SimPEG.MT Mergify.

Merge branch 'move2Simpeg' of https://github.com/simpeg/simpegmt into mt/dev Conflicts: .gitignore .travis.yml LICENSE docs/conf.py docs/index.rst requirements.txt setup.py
2026-07-31 12:40:46 +08:00 · 2016-01-14 15:36:58 -08:00
parent e15913cf84 d77b393d42
commit 558e8879fe
28 changed files with 3595 additions and 0 deletions
@@ -0,0 +1,135 @@
+import unittest
+import SimPEG as simpeg
+from SimPEG import MT
+from SimPEG.Utils import meshTensor
+import numpy as np
+# Define the tolerances
+TOLr = 5e-2
+TOLp = 5e-2
+
+
+def setupSurvey(sigmaHalf,tD=True):
+
+    # Frequency
+    nFreq = 33
+    freqs = np.logspace(3,-3,nFreq)
+    # Make the mesh
+    ct = 5
+    air = meshTensor([(ct,25,1.3)])
+    # coreT0 = meshTensor([(ct,15,1.2)])
+    # coreT1 = np.kron(meshTensor([(coreT0[-1],15,1.3)]),np.ones((7,)))
+    core = np.concatenate( (  np.kron(meshTensor([(ct,15,-1.2)]),np.ones((10,))) , meshTensor([(ct,20)]) ) )
+    bot = meshTensor([(core[0],15,-1.3)])
+    x0 = -np.array([np.sum(np.concatenate((core,bot)))])
+    m1d = simpeg.Mesh.TensorMesh([np.concatenate((bot,core,air))], x0=x0)
+    # Make the model
+    sigma = np.zeros(m1d.nC) + sigmaHalf
+    sigma[m1d.gridCC > 0 ] = 1e-8
+    sigmaBack = sigma.copy()
+    # Add structure
+    shallow = (m1d.gridCC < -200) * (m1d.gridCC > -600)
+    deep = (m1d.gridCC < -3000) * (m1d.gridCC > -5000)
+    sigma[shallow] = 1
+    sigma[deep] = 0.1
+
+    rxList = []
+    for rxType in ['z1dr','z1di']:
+        rxList.append(MT.Rx(simpeg.mkvc(np.array([0.0]),2).T,rxType))
+    # Source list
+    srcList =[]
+    if tD:
+        for freq in freqs:
+            srcList.append(MT.SrcMT.polxy_1DhomotD(rxList,freq))
+    else:
+        for freq in freqs:
+            srcList.append(MT.SrcMT.polxy_1Dprimary(rxList,freq))
+
+    survey = MT.Survey(srcList)
+    return survey, sigma, m1d
+
+def getAppResPhs(MTdata):
+    # Make impedance
+    def appResPhs(freq,z):
+        app_res = ((1./(8e-7*np.pi**2))/freq)*np.abs(z)**2
+        app_phs = np.arctan2(z.imag,z.real)*(180/np.pi)
+        return app_res, app_phs
+    zList = []
+    for src in MTdata.survey.srcList:
+        zc = [src.freq]
+        for rx in src.rxList:
+            if 'i' in rx.rxType:
+                m=1j
+            else:
+                m = 1
+            zc.append(m*MTdata[src,rx])
+        zList.append(zc)
+    return [appResPhs(zList[i][0],np.sum(zList[i][1:3])) for i in np.arange(len(zList))]
+
+def calculateAnalyticSolution(srcList,mesh,model):
+    surveyAna = MT.Survey(srcList)
+    data1D = MT.Data(surveyAna)
+    for src in surveyAna.srcList:
+        elev = src.rxList[0].locs[0]
+        anaEd, anaEu, anaHd, anaHu = MT.Utils.MT1Danalytic.getEHfields(mesh,model,src.freq,elev)
+        anaE = anaEd+anaEu
+        anaH = anaHd+anaHu
+        # Scale the solution
+        # anaE = (anaEtemp/anaEtemp[-1])#.conj()
+        # anaH = (anaHtemp/anaEtemp[-1])#.conj()
+        anaZ = anaE/anaH
+        for rx in src.rxList:
+            data1D[src,rx] = getattr(anaZ, rx.projComp)
+    return data1D
+
+def dataMis_AnalyticTotalDomain(sigmaHalf):
+
+    # Make the survey
+
+    # Total domain solution
+    surveyTD, sigma, mesh = setupSurvey(sigmaHalf)
+    problemTD = MT.Problem1D.eForm_TotalField(mesh)
+    problemTD.pair(surveyTD)
+    # Analytic data
+    dataAnaObj = calculateAnalyticSolution(surveyTD.srcList,mesh,sigma)
+    # dataTDObj = MT.DataMT.DataMT(surveyTD, surveyTD.dpred(sigma))
+    dataTD = surveyTD.dpred(sigma)
+    dataAna = simpeg.mkvc(dataAnaObj)
+    return np.all((dataTD - dataAna)/dataAna < 2.)
+    # surveyTD.dtrue = -simpeg.mkvc(dataAna,2)
+    # surveyTD.dobs = -simpeg.mkvc(dataAna,2)
+    # surveyTD.Wd = np.ones(surveyTD.dtrue.shape) #/(np.abs(surveyTD.dtrue)*0.01)
+    # # Setup the data misfit
+    # dmis = simpeg.DataMisfit.l2_DataMisfit(surveyTD)
+    # dmis.Wd = surveyTD.Wd
+    # return dmis.eval(sigma)
+
+
+def dataMis_AnalyticPrimarySecondary(sigmaHalf):
+
+    # Make the survey
+    # Primary secondary
+    surveyPS, sigmaPS, mesh = setupSurvey(sigmaHalf,tD=False)
+    problemPS = MT.Problem1D.eForm_psField(mesh)
+    problemPS.sigmaPrimary = sigmaPS
+    problemPS.pair(surveyPS)
+    # Analytic data
+    dataAnaObj = calculateAnalyticSolution(surveyPS.srcList,mesh,sigmaPS)
+
+    dataPS = surveyPS.dpred(sigmaPS)
+    dataAna = simpeg.mkvc(dataAnaObj)
+    return np.all((dataPS - dataAna)/dataAna < 2.)
+
+
+
+class TestNumericVsAnalytics(unittest.TestCase):
+
+    def setUp(self):
+        pass
+    # Total Fields
+    # def test_appRes2en2(self):self.assertTrue(dataMis_AnalyticTotalDomain(2e-2))
+
+    # Primary/secondary
+    def test_appRes2en2_ps(self):self.assertTrue(dataMis_AnalyticPrimarySecondary(2e-2))
+
+if __name__ == '__main__':
+    unittest.main()