Fixed 1D test and current code to work, where the src in the 1D problem is partly implemented

simpegMT/Sources/backgroundModelSources.py

@@ -1,23 +1,27 @@
 import SimPEG as simpeg, numpy as np
 
-def homo1DModelSource(mesh,freq,m_back):
+def homo1DModelSource(mesh,freq,sigma_1d):
     '''
         Function that calculates and return background fields
 
         :param Simpeg mesh object mesh: Holds information on the discretization
         :param float freq: The frequency to solve at
-        :param np.array m_back: Background model of conductivity to base the calculations on.
+        :param np.array sigma_1d: Background model of conductivity to base the calculations on, 1d model.
         :rtype: numpy.ndarray (mesh.nE,2)
         :return: eBG_bp, E fields for the background model at both polarizations.
 
     '''
-
     # import
     from simpegMT.Utils import get1DEfields
     # Get a 1d solution for a halfspace background
-    mesh1d = simpeg.Mesh.TensorMesh([mesh.hz],np.array([mesh.x0[2]]))
-    # Note: Everything is using e^iwt
-    e0_1d = get1DEfields(mesh1d,mesh.r(m_back,'CC','CC','M')[0,0,:],freq)
+    if mesh.dim == 1:
+        mesh1d = mesh
+    elif mesh.dim == 2:
+        mesh1d = simpeg.Mesh.TensorMesh([mesh.hy],np.array([mesh.x0[1]]))
+    elif mesh.dim == 3:
+        mesh1d = simpeg.Mesh.TensorMesh([mesh.hz],np.array([mesh.x0[2]]))
+    # # Note: Everything is using e^iwt
+    e0_1d = get1DEfields(mesh1d,sigma_1d,freq)
     # Setup x (east) polarization (_x)
     ex_px = np.zeros(mesh.vnEx,dtype=complex)
     ey_px = np.zeros((mesh.nEy,1),dtype=complex)
@@ -32,7 +36,7 @@ def homo1DModelSource(mesh,freq,m_back):
     ey_py = np.zeros(mesh.vnEy, dtype='complex128')
     ez_py = np.zeros((mesh.nEz,1), dtype='complex128')
     # Assign the source to ey_py
-    
+
     for i in np.arange(mesh.vnEy[0]):
         for j in np.arange(mesh.vnEy[1]):
             ey_py[i,j,:] = e0_1d