simpeg · dccowan · May 7, 2025 · May 7, 2025 · May 8, 2025 · Jul 9, 2025
diff --git a/simpeg/electromagnetics/natural_source/sources.py b/simpeg/electromagnetics/natural_source/sources.py
@@ -3,7 +3,10 @@
 from ... import maps
 from ..frequency_domain.sources import BaseFDEMSrc
 from ..utils import omega
-from .utils.source_utils import homo1DModelSource
+from .utils.source_utils import (
+    primary_e_1d_solution,
+    project_1d_fields_to_mesh_edges,
+)
 import discretize
 from discretize.utils import volume_average
 
@@ -13,6 +16,7 @@
 #################
 
 
+# Rename to BasePlanewave
 class Planewave(BaseFDEMSrc):
     """
     Source class for the 1D and pseudo-3D problems.
@@ -30,7 +34,7 @@ class Planewave(BaseFDEMSrc):
 
 
 # Need to implement such that it works for all dims.
-# Rename to be more descriptive
+# Rename to be more descriptive (I suggest PlanewavePrimarySecondary)
 class PlanewaveXYPrimary(Planewave):
     """
     NSEM planewave source for both polarizations (x and y)
@@ -106,9 +110,14 @@ def ePrimary(self, simulation):
         """
         if self._ePrimary is None:
             sigma_1d, _ = self._get_sigmas(simulation)
-            self._ePrimary = homo1DModelSource(
-                simulation.mesh, self.frequency, sigma_1d
-            )
+            e_1d = primary_e_1d_solution(simulation.mesh, sigma_1d, self.frequency)
+            e_1d = project_1d_fields_to_mesh_edges(simulation.mesh, e_1d)
+
+            # To output X and Y polarization for 1D mesh
+            if simulation.mesh.dim == 1:
+                e_1d = np.c_[e_1d, -e_1d]
+
+            self._ePrimary = e_1d
         return self._ePrimary
 
     def bPrimary(self, simulation):

diff --git a/simpeg/electromagnetics/natural_source/utils/__init__.py b/simpeg/electromagnetics/natural_source/utils/__init__.py
@@ -7,6 +7,11 @@
 """
 
 from .solutions_1d import get1DEfields  # Add the names of the functions
+from .source_utils import (
+    primary_e_1d_solution,
+    primary_h_1d_solution,
+    project_1d_fields_to_mesh_edges,
+)
 from .analytic_1d import getEHfields, getImpedance
 from .data_utils import (
     appResPhs,