From 4899346f6b4179aade813413b6ba7e1f5871431a Mon Sep 17 00:00:00 2001 From: rowanc1 Date: Thu, 15 May 2014 12:20:56 -0700 Subject: [PATCH] Better fields object. --- simpegEM/TDEM/SurveyTDEM.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/simpegEM/TDEM/SurveyTDEM.py b/simpegEM/TDEM/SurveyTDEM.py index 845cb515..c9a67856 100644 --- a/simpegEM/TDEM/SurveyTDEM.py +++ b/simpegEM/TDEM/SurveyTDEM.py @@ -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