wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-13 17:45:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adding fixes to code.

Browse Source
This commit is contained in:
GudniRos
2015-05-07 16:26:35 -07:00
parent 6eabd68a1d
commit 6eafbdca80
4 changed files with 3204 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simpegMT/ProblemMT.py
+59 -52
View File
@@ -165,11 +165,16 @@ class MTProblem(Problem.BaseProblem):
:rtype: scipy.sparse.csr_matrix
:return: A
"""
mui = self.MfMui
from SimPEG import Mesh
Mback = Mesh.TensorMesh(self.mesh.h,self.mesh.x0)
Mback.setCellGradBC('dirichlet')
mui = Mback.getFaceInnerProduct(1/mu_0)
sigmaBG = self.backModel
MsigBG = Mback.getEdgeInnerProduct(sigmaBG)
sigBG = self.MeSigmaBG
C = self.mesh.edgeCurl
C = Mback.edgeCurl
return C.T*mui*C - 1j*omega(freq)*sigBG
return C.T*mui*C - 1j*omega(freq)*MsigBG
def getADeriv(self, freq, u, v, adjoint=False):
sig = self.curTModel
@@ -198,7 +203,7 @@ class MTProblem(Problem.BaseProblem):
eBG_bp = homo1DModelSource(self.mesh,freq,backSigma)
Abg = self.getAbg(freq)
return -Abg*eBG_bp, eBG_bp
return Abg*eBG_bp, eBG_bp
##################################################################
# Inversion stuff
@@ -228,70 +233,72 @@ class MTProblem(Problem.BaseProblem):
def Jvec(self, m, v, u=None):
if u is None:
u = self.fields(m)
# if u is None:
# u = self.fields(m)
self.curModel = m
# self.curModel = m
Jv = self.dataPair(self.survey)
# Jv = self.dataPair(self.survey)
for freq in self.survey.freqs:
A = self.getA(freq)
solver = self.Solver(A, **self.solverOpts)
# for freq in self.survey.freqs:
# A = self.getA(freq)
# solver = self.Solver(A, **self.solverOpts)
for tx in self.survey.getTransmitters(freq):
u_tx = u[tx, self.solType]
w = self.getADeriv(freq, u_tx, v)
Ainvw = solver.solve(w)
for rx in tx.rxList:
fAinvw = self.calcFields(Ainvw, freq, rx.projField)
P = lambda v: rx.projectFieldsDeriv(tx, self.mesh, u, v)
# for tx in self.survey.getTransmitters(freq):
# u_tx = u[tx, self.solType]
# w = self.getADeriv(freq, u_tx, v)
# Ainvw = solver.solve(w)
# for rx in tx.rxList:
# fAinvw = self.calcFields(Ainvw, freq, rx.projField)
# P = lambda v: rx.projectFieldsDeriv(tx, self.mesh, u, v)
df_dm = self.calcFieldsDeriv(u_tx, freq, rx.projField, v)
if df_dm is None:
Jv[tx, rx] = - P(fAinvw)
else:
Jv[tx, rx] = - P(fAinvw) + P(df_dm)
# df_dm = self.calcFieldsDeriv(u_tx, freq, rx.projField, v)
# if df_dm is None:
# Jv[tx, rx] = - P(fAinvw)
# else:
# Jv[tx, rx] = - P(fAinvw) + P(df_dm)
return Utils.mkvc(Jv)
# return Utils.mkvc(Jv)
pass
def Jtvec(self, m, v, u=None):
if u is None:
u = self.fields(m)
# if u is None:
# u = self.fields(m)
self.curModel = m
# self.curModel = m
# Ensure v is a data object.
if not isinstance(v, self.dataPair):
v = self.dataPair(self.survey, v)
# # Ensure v is a data object.
# if not isinstance(v, self.dataPair):
# v = self.dataPair(self.survey, v)
Jtv = np.zeros(self.mapping.nP)
# Jtv = np.zeros(self.mapping.nP)
for freq in self.survey.freqs:
AT = self.getA(freq).T
solver = self.Solver(AT, **self.solverOpts)
# for freq in self.survey.freqs:
# AT = self.getA(freq).T
# solver = self.Solver(AT, **self.solverOpts)
for tx in self.survey.getTransmitters(freq):
u_tx = u[tx, self.solType]
# for tx in self.survey.getTransmitters(freq):
# u_tx = u[tx, self.solType]
for rx in tx.rxList:
PTv = rx.projectFieldsDeriv(tx, self.mesh, u, v[tx, rx], adjoint=True)
fPTv = self.calcFields(PTv, freq, rx.projField, adjoint=True)
# for rx in tx.rxList:
# PTv = rx.projectFieldsDeriv(tx, self.mesh, u, v[tx, rx], adjoint=True)
# fPTv = self.calcFields(PTv, freq, rx.projField, adjoint=True)
w = solver.solve( fPTv )
Jtv_rx = - self.getADeriv(freq, u_tx, w, adjoint=True)
# w = solver.solve( fPTv )
# Jtv_rx = - self.getADeriv(freq, u_tx, w, adjoint=True)
df_dm = self.calcFieldsDeriv(u_tx, freq, rx.projField, PTv, adjoint=True)
# df_dm = self.calcFieldsDeriv(u_tx, freq, rx.projField, PTv, adjoint=True)
if df_dm is not None:
Jtv_rx += df_dm
# if df_dm is not None:
# Jtv_rx += df_dm
real_or_imag = rx.projComp
if real_or_imag == 'real':
Jtv += Jtv_rx.real
elif real_or_imag == 'imag':
Jtv += - Jtv_rx.real
else:
raise Exception('Must be real or imag')
# real_or_imag = rx.projComp
# if real_or_imag == 'real':
# Jtv += Jtv_rx.real
# elif real_or_imag == 'imag':
# Jtv += - Jtv_rx.real
# else:
# raise Exception('Must be real or imag')
return Jtv
# return Jtv
pass