From b26412e7a9955e94a60d16cb34bc00b00c8915c8 Mon Sep 17 00:00:00 2001
From: GudniRos <gkrosen@gmail.com>
Date: Thu, 13 Aug 2015 12:26:16 -0700
Subject: [PATCH] Made changes to data utils

---
 simpegMT/Utils/dataUtils.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/simpegMT/Utils/dataUtils.py b/simpegMT/Utils/dataUtils.py
index 615a5432..efa32509 100644
--- a/simpegMT/Utils/dataUtils.py
+++ b/simpegMT/Utils/dataUtils.py
@@ -2,7 +2,7 @@
 import numpy as np, matplotlib.pyplot as plt, sys
 import SimPEG as simpeg
 import simpegMT as simpegmt
-
+import numpy.lib.recfunctions as recFunc
 def getAppRes(MTdata):
     # Make impedance
     zList = []
@@ -131,7 +131,16 @@ def printTime():
 def convert3Dto1Dobject(MTdata,rxType3D='zyx'):
     # Find the unique locations
     # Need to find the locations
-    recData = MTdata.toRecArray()
+    recDataTemp = MTdata.toRecArray()
+    # Calculte and add the DET of the tensor to the recArray
+    if 'det' in rxType3D:
+        Zon = (recDataTemp['zxxr']+1j*recDataTemp['zxxi'])*(recDataTemp['zyyr']+1j*recDataTemp['zyyi'])
+        Zoff = (recDataTemp['zxyr']+1j*recDataTemp['zxyi'])*(recDataTemp['zyxr']+1j*recDataTemp['zyxi'])
+        det = np.sqrt(Zon.data - Zoff.data)
+        recData = recFunc.append_fields(recDataTemp,['zdetr','zdeti'],[det.real,det.imag] )
+    else:
+        recData = recDataTemp
+
     uniLocs = rec2ndarr(np.unique(recData[['x','y','z']])).data
     mtData1DList = []
     if 'zxy' in rxType3D: