visualize: added dynamic cell grid

Author: BitterB0NG0

files/inp/todo_41.inp

@@ -0,0 +1,7 @@
+Bad Geometry INP File!
+1 0 3 4 imp:@=1
+
+1 so 3
+
+sdef
+nps 10

src/pymcnp/Visualize.py

@@ -19,12 +19,6 @@ class Visualize(_doer.Doer):
         inpt: Files to visualize.
     """
 
-    GRID = pyvista.ImageData(
-        dimensions=(_show.pyvista.RESOLUTION, _show.pyvista.RESOLUTION, _show.pyvista.RESOLUTION),
-        spacing=(_show.pyvista.BOUND / _show.pyvista.RESOLUTION, _show.pyvista.BOUND / _show.pyvista.RESOLUTION, _show.pyvista.BOUND / _show.pyvista.RESOLUTION),
-        origin=(-_show.pyvista.BOUND / 2, -_show.pyvista.BOUND / 2, -_show.pyvista.BOUND / 2),
-    )
-
     def __init__(self, inpt: Inp):
         """
         Initializes `Visualize`.
@@ -41,6 +35,20 @@ def __init__(self, inpt: Inp):
 
         self.inpt = inpt
 
+    @property
+    def _grid(self):
+        surfaces = self.to_show_surfaces()
+
+        return pyvista.ImageData(
+            dimensions=(_show.pyvista.RESOLUTION, _show.pyvista.RESOLUTION, _show.pyvista.RESOLUTION),
+            spacing=(
+                (surfaces.bounds[1] - surfaces.bounds[0]) / _show.pyvista.RESOLUTION,
+                (surfaces.bounds[3] - surfaces.bounds[2]) / _show.pyvista.RESOLUTION,
+                (surfaces.bounds[5] - surfaces.bounds[4]) / _show.pyvista.RESOLUTION,
+            ),
+            origin=(-(surfaces.bounds[1] - surfaces.bounds[0]) / 2, -(surfaces.bounds[3] - surfaces.bounds[2]) / 2, -(surfaces.bounds[5] - surfaces.bounds[4]) / 2),
+        )
+
     def to_show_cells(self) -> pyvista.Plotter:
         """
         Visualizes INP cells.
@@ -62,8 +70,9 @@ def to_show_cells(self) -> pyvista.Plotter:
 
             cells[str(cell.number)] = shape
 
-            self.GRID['cell'] = shape.cell(self.GRID.points).astype(numpy.float32)
-            plot.add_volume(self.GRID, scalars='cell', opacity=[0, 0, 0.01, 0.01])
+            grid = self._grid
+            grid['cell'] = shape.cell(grid.points).astype(numpy.float32)
+            plot.add_volume(grid, scalars='cell', opacity=[0, 0, 0.01, 0.01])
             plot.add_mesh(shape.surface, opacity=0.9)
 
         return plot
@@ -113,8 +122,9 @@ def to_show_cell(self, number: str) -> pyvista.Plotter:
             if str(cell.number) != number:
                 continue
 
-            self.GRID['cell'] = shape.cell(self.GRID.points).astype(numpy.float32)
-            plot.add_volume(self.GRID, scalars='cell', opacity=[0, 0, 0.01, 0.01])
+            grid = self._grid
+            grid['cell'] = shape.cell(grid.points).astype(numpy.float32)
+            plot.add_volume(grid, scalars='cell', opacity=[0, 0, 0.01, 0.01])
             plot.add_mesh(shape.surface, opacity=0.9)
 
             break

src/pymcnp/_cli/visualize.py

@@ -93,5 +93,8 @@ def main() -> None:
     except errors.CliError as err:
         _io.error(str(err))
         exit(3)
+    except errors.TypesError as err:
+        _io.error(str(err))
+        exit(3)
 
     _io.done()

tests/test_pymcnp/test__cli/test_visualize.py

@@ -20,5 +20,6 @@ def test_invalid(self):
             os.system(f'pymcnp visualize {pathlib.Path(__file__).parent.parent.parent.parent / "files" / "inp" / "invalid_00.inp"}')
             os.system(f'pymcnp visualize {pathlib.Path(__file__).parent.parent.parent.parent / "files" / "inp" / "invalid_01.inp"}')
             os.system(f'pymcnp visualize {pathlib.Path(__file__).parent.parent.parent.parent / "files" / "inp" / "invalid_02.inp"}')
+            os.system(f'pymcnp visualize {pathlib.Path(__file__).parent.parent.parent.parent / "files" / "inp" / "todo_41.inp"}')
             os.system(f'pymcnp visualize {pathlib.Path(__file__).parent.parent.parent.parent / "files" / "inp" / "valid_00.inp"} --cell=934587439')
             os.system('pymcnp visualize hello')