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

Better fields object.

Browse Source
This commit is contained in:
rowanc1
2014-05-15 12:20:56 -07:00
parent 205d2e6b5e
commit 4899346f6b
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simpegEM/TDEM/SurveyTDEM.py
+3 -6
View File
@@ -146,14 +146,11 @@ class SurveyTDEM(Survey.BaseSurvey):
for tx in self.txList:
for rx in tx.rxList:
Ptv = rx.projectFieldsDeriv(tx, self.mesh, self.prob.timeMesh, u, v, adjoint=True)
Ptv = Ptv.reshape((-1, self.prob.timeMesh.nN), order='F')
if rx.projField not in f: # first time we are projecting
Ptv = Ptv.reshape((-1, 1, self.prob.timeMesh.nN), order='F')
f[tx, rx.projField, :] = Ptv
else:
Ptv = Ptv.reshape((-1, self.prob.timeMesh.nN), order='F')
addedPtv = f[tx, rx.projField, :] + Ptv
addedPtv = addedPtv.reshape((-1, 1, self.prob.timeMesh.nN), order='F')
f[tx, rx.projField, :] = addedPtv
else: # there are already fields, so let's add to them!
f[tx, rx.projField, :] += Ptv
return f