fix 1880 (#1884)

NanoComp · Jan 5, 2022 · efe3b4b · efe3b4b
1 parent 2e563e7
commit efe3b4b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/visualization.py b/python/visualization.py
@@ -477,20 +477,20 @@ def get_boundary_volumes(thickness, direction, side, cylindrical=False):
             else:
                 raise ValueError("Invalid simulation dimensions")
             for permutation in itertools.product(dims, [mp.Low, mp.High]):
-                if (permutation[0] == mp.X) and (permutation[1] == mp.Low):
+                if ((permutation[0] == mp.X) and (permutation[1] == mp.Low)) and (sim.dimensions == mp.CYLINDRICAL or sim.is_cylindrical):
                     continue
                 vol = get_boundary_volumes(boundary.thickness,*permutation)
                 ax = plot_volume(sim,ax,vol,output_plane,plotting_parameters=boundary_parameters)
         # two sides are the same
         elif boundary.side == mp.ALL:
             for side in [mp.Low, mp.High]:
-                if (boundary.direction == mp.X) and (side == mp.Low):
+                if ((boundary.direction == mp.X) and (side == mp.Low)) and (sim.dimensions == mp.CYLINDRICAL or sim.is_cylindrical):
                     continue
                 vol = get_boundary_volumes(boundary.thickness,boundary.direction,side)
                 ax = plot_volume(sim,ax,vol,output_plane,plotting_parameters=boundary_parameters)
         # only one side
         else:                
-            if (boundary.direction == mp.X) and (boundary.side == mp.Low):
+            if ((boundary.direction == mp.X) and (boundary.side == mp.Low)) and (sim.dimensions == mp.CYLINDRICAL or sim.is_cylindrical):
                 continue
             vol = get_boundary_volumes(boundary.thickness,boundary.direction,boundary.side)
             ax = plot_volume(sim,ax,vol,output_plane,plotting_parameters=boundary_parameters)