fix multi mode handling in tidy3d

Author: mustafacc

gds_fdtd/lum_tools.py

@@ -1,12 +1,11 @@
 """
 gds_fdtd simulation toolbox.
 
-Lumerical tools module.
+Lumerical tools interface module.
 @author: Mustafa Hammood, 2025
 """
 
 from gds_fdtd.core import structure, component
-from .core import port
 import logging
 import lumapi
 import numpy as np

gds_fdtd/simprocessor.py

@@ -47,7 +47,6 @@ def make_source(
     Returns:
         td.ModeSource: Generated source.
     """
-    import tidy3d as td
 
     if port.direction == 0:
         x_buffer = -buffer
@@ -92,9 +91,6 @@ def make_structures(device, buffer: float=4.):
     Returns:
         list: list of structures generated from the device.
     """
-    import tidy3d as td
-    import numpy as np
-
     # TODO find a better way to handle material..
 
     # TODO fix box tox handling here
@@ -214,7 +210,6 @@ def make_field_monitor(device, freqs=2e14, axis="z", z_center=None):
     Returns:
         FieldMonitor: Generated Tidy3D field monitor object
     """
-    import numpy as np
 
     # identify a device field z_center if None
     if z_center is None:
@@ -491,27 +486,6 @@ def load_component_from_tech(cell, tech, z_span=4, z_center=None):
         bounds=bounds,
     )
 
-def build_sim_from_tech(tech: dict, cell, in_port=0, **kwargs):
-
-    z_span = kwargs.pop("z_span", 4)  # Default value 4 if z_span is not provided
-
-    device = load_component_from_tech(cell=cell, tech=tech, z_span=z_span)
-
-    if isinstance(in_port, int):
-        return make_t3d_sim(
-            device=device,
-            in_port=device.ports[in_port],
-            z_span=z_span,
-            **kwargs,
-        )
-    elif in_port == "all":
-        return make_t3d_sim(
-            device=device,
-            in_port=device.ports[:],
-            z_span=z_span,
-            **kwargs,
-        )
-
 
 def from_gdsfactory(c: 'gf.Component', tech: dict, z_span: float = 4.) -> 'component':
     """Convert gdsfactory Component to a component.

gds_fdtd/t3d_tools.py

@@ -67,7 +67,7 @@ def measure_transmission(in_port: port, in_mode_idx: int, out_mode_idx: int):
                     logging.warning(
                         "Multiple results handler is WIP, using first results entry"
                     )
-                    results = self.results[0]
+                    results = self.results[-1]
             else:
                 results = self.results
 
@@ -130,28 +130,25 @@ def measure_transmission(in_port: port, in_mode_idx: int, out_mode_idx: int):
         if isinstance(self.results, list) and len(self.results) == 1:
             self.results = self.results[0]
 
+    def _plot_any_axis(self, job_result, freq):
+        """Try x/y/z until one works, put axis name in title."""
+        for axis in ("x", "y", "z"):
+            try:
+                fig, ax = plt.subplots(1, 1, figsize=(16, 3))
+                job_result.plot_field(f"{axis}_field", "Ey", freq=freq, ax=ax)
+                ax.set_title(f"**{axis.upper()}‑field**")      # highlight chosen axis
+                fig.show()
+            except Exception:
+                plt.close(fig)
+        # nothing matched → silently skip
+
+
     def visualize_results(self):
         self.s_parameters.plot()
+        freq = td.C_0 / ((self.wavl_max + self.wavl_min) / 2)
 
-        try:
-            if isinstance(self.results, list):
-                for job_result in self.results:
-                    fig, ax = plt.subplots(1, 1, figsize=(16, 3))
-                    job_result.plot_field(
-                        "field",
-                        "Ey",
-                        freq=td.C_0 / ((self.wavl_max + self.wavl_min) / 2),
-                        ax=ax,
-                    )
-                    fig.show()
-            else:
-                fig, ax = plt.subplots(1, 1, figsize=(16, 3))
-                self.results.plot_field(
-                    "field",
-                    "Ey",
-                    freq=td.C_0 / ((self.wavl_max + self.wavl_min) / 2),
-                    ax=ax,
-                )
-                fig.show()
-        except:
-            return
+        if isinstance(self.results, list):
+            for job_result in self.results:
+                self._plot_any_axis(job_result, freq)
+        else:
+            self._plot_any_axis(self.results, freq)