(Hopefully) Fixed a bug in HeatChargeSimulationData.plot_field() from tcad/data/sim_data.py. I don't know how to test this function yet.

Author: jewettaijfc

tidy3d/components/tcad/data/sim_data.py

@@ -157,6 +157,7 @@ def plot_field(
         vmin: Optional[float] = None,
         vmax: Optional[float] = None,
         ax: Ax = None,
+        swap_axes: bool = False,
         **sel_kwargs,
     ) -> Ax:
         """Plot the data for a monitor with simulation plot overlaid.
@@ -252,6 +253,7 @@ def plot_field(
                 vmax=vmax,
                 cbar_kwargs={"label": field_name},
                 grid=False,
+                swap_axes=swap_axes,
             )
 
             # compute parameters for structures overlay plot
@@ -260,10 +262,8 @@ def plot_field(
 
             # compute plot bounds
             field_data_bounds = field_data.bounds
-            min_bounds = list(field_data_bounds[0])
-            max_bounds = list(field_data_bounds[1])
-            min_bounds.pop(axis)
-            max_bounds.pop(axis)
+            _, min_bounds = self.pop_axis_and_swap(field_data_bounds[0], axis, swap_axes=swap_axes)
+            _, max_bounds = self.pop_axis_and_swap(field_data_bounds[1], axis, swap_axes=swap_axes)
 
         if isinstance(field_data, SpatialDataArray):
             # interp out any monitor.size==0 dimensions
@@ -315,8 +315,8 @@ def plot_field(
             axis = "xyz".index(planar_coord)
             position = float(field_data.coords[planar_coord])
 
-            xy_coord_labels = list("xyz")
-            xy_coord_labels.pop(axis)
+            _, xy_coord_labels = self.pop_axis_and_swap(list("xyz"), axis=axis, swap_axes=swap_axes)
+
             x_coord_label, y_coord_label = xy_coord_labels[0], xy_coord_labels[1]
             field_data.plot(
                 ax=ax,
@@ -327,6 +327,7 @@ def plot_field(
                 vmax=vmax,
                 robust=robust,
                 cbar_kwargs={"label": field_name},
+                swap_axes=swap_axes,
             )
 
             # compute plot bounds
@@ -344,6 +345,7 @@ def plot_field(
                 alpha=structures_alpha,
                 ax=ax,
                 property=property_to_plot,
+                swap_axes=swap_axes,
                 **interp_kwarg,
             )
 

tidy3d/components/tcad/simulation/heat_charge.py

@@ -1588,7 +1588,7 @@ def plot_sources(
         plane = Box(center=center, size=size)
 
         source_shapes = self.scene._filter_structures_plane(
-            structures=structures, plane=plane, property_list=source_list
+            structures=structures, plane=plane, property_list=source_list, swap_axes=swap_axes
         )
 
         source_min, source_max = self.source_bounds(property=property)