inp/surface: updated to_pyvista

BitterB0NG0 · BitterB0NG0 · commit 200cf17a5c13 · 2025-02-18T14:35:36.000-07:00
diff --git a/src/pymcnp/_help.py b/src/pymcnp/_help.py
@@ -8,6 +8,8 @@
     # CELL
     'geometry',
     # SURFACE
+    'p_0',
+    'p_1',
     'px',
     'py',
     'pz',
diff --git a/src/pymcnp/inp/surface/option_arb.py b/src/pymcnp/inp/surface/option_arb.py
@@ -402,6 +402,6 @@ def to_pyvista(self):
             else:
                 faces.append([n // 1000 % 10, n // 100 % 10, n // 10 % 10, n // 1 % 10])
 
-        vis = _visualization.PyMcnpVisualization.get_polyhedron(vectices, faces)
+        vis = _visualization.McnpVisualization.get_polyhedron(vectices, faces)
 
         return vis.data
diff --git a/src/pymcnp/inp/surface/option_box.py b/src/pymcnp/inp/surface/option_box.py
@@ -173,7 +173,7 @@ def to_pyvista(self):
         cross = _visualization.Vector(1, 0, 0) * a1
         angle = _visualization.Vector(1, 0, 0) & a1
 
-        vis = _visualization.PyMcnpVisualization.get_box(a1.norm(), a2.norm(), a3.norm())
+        vis = _visualization.McnpVisualization.get_box(a1.norm(), a2.norm(), a3.norm())
         vis = vis.add_rotation(cross, angle, (0, 0, 0))
         vis = vis.add_translation(v)
 
diff --git a/src/pymcnp/inp/surface/option_c_x.py b/src/pymcnp/inp/surface/option_c_x.py
@@ -85,7 +85,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_C_x``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cylinder_unbounded(self.r.value)
+        vis = _visualization.McnpVisualization.get_cylinder_unbounded(self.r.value)
         vis = vis.add_rotation(_visualization.Vector(0, 1, 0), 90, (0, 0, 0))
         vis = vis.add_translation(_visualization.Vector(0, self.y.value, self.z.value))
 
diff --git a/src/pymcnp/inp/surface/option_c_y.py b/src/pymcnp/inp/surface/option_c_y.py
@@ -85,7 +85,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_C_y``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cylinder_unbounded(self.r.value)
+        vis = _visualization.McnpVisualization.get_cylinder_unbounded(self.r.value)
         vis = vis.add_rotation(_visualization.Vector(1, 0, 0), 90, (0, 0, 0))
         vis = vis.add_translation(_visualization.Vector(self.x.value, 0, self.z.value))
 
diff --git a/src/pymcnp/inp/surface/option_c_z.py b/src/pymcnp/inp/surface/option_c_z.py
@@ -84,7 +84,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_C_z``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cylinder_unbounded(self.r.value)
+        vis = _visualization.McnpVisualization.get_cylinder_unbounded(self.r.value)
         vis = vis.add_rotation(_visualization.Vector(0, 1, 0), 90, (0, 0, 0))
         vis = vis.add_translation(_visualization.Vector(self.x.value, self.y.value, 0))
 
diff --git a/src/pymcnp/inp/surface/option_cx.py b/src/pymcnp/inp/surface/option_cx.py
@@ -71,7 +71,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_Cx``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cylinder_unbounded(self.r.value)
+        vis = _visualization.McnpVisualization.get_cylinder_unbounded(self.r.value)
         vis = vis.add_rotation(_visualization.Vector(0, 1, 0), 90, (0, 0, 0))
 
         return vis.data
diff --git a/src/pymcnp/inp/surface/option_cy.py b/src/pymcnp/inp/surface/option_cy.py
@@ -71,7 +71,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_Cy``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cylinder_unbounded(self.r.value)
+        vis = _visualization.McnpVisualization.get_cylinder_unbounded(self.r.value)
         vis = vis.add_rotation(_visualization.Vector(1, 0, 0), 90, (0, 0, 0))
 
         return vis.data
diff --git a/src/pymcnp/inp/surface/option_cz.py b/src/pymcnp/inp/surface/option_cz.py
@@ -71,6 +71,6 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_Cz``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cylinder_unbounded(self.r.value)
+        vis = _visualization.McnpVisualization.get_cylinder_unbounded(self.r.value)
 
         return vis.data
diff --git a/src/pymcnp/inp/surface/option_ell.py b/src/pymcnp/inp/surface/option_ell.py
@@ -140,7 +140,7 @@ def to_pyvista(self):
             cross = v2 * _visualization.Vector(1, 0, 0)
             angle = v2 & _visualization.Vector(1, 0, 0)
 
-        vis = _visualization.PyMcnpVisualization.get_ellipsoid(major_length, minor_length)
+        vis = _visualization.McnpVisualization.get_ellipsoid(major_length, minor_length)
         vis = vis.add_rotation(cross, angle, (0, 0, 0))
         vis = vis.add_translation(center)
 
diff --git a/src/pymcnp/inp/surface/option_k_x.py b/src/pymcnp/inp/surface/option_k_x.py
@@ -106,7 +106,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_K_x``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cone_quadratic(
+        vis = _visualization.McnpVisualization.get_cone_quadratic(
             self.t_squared.value ** (1 / 2), self.plusminus_1.value
         )
         vis = vis.add_rotation(_visualization.Vector(0, 1, 0), 90, (0, 0, 0))
diff --git a/src/pymcnp/inp/surface/option_k_y.py b/src/pymcnp/inp/surface/option_k_y.py
@@ -106,7 +106,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_K_y``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cone_quadratic(
+        vis = _visualization.McnpVisualization.get_cone_quadratic(
             self.t_squared.value ** (1 / 2), self.plusminus_1.value
         )
         vis = vis.add_rotation(_visualization.Vector(1, 0, 0), 90, (0, 0, 0))
diff --git a/src/pymcnp/inp/surface/option_k_z.py b/src/pymcnp/inp/surface/option_k_z.py
@@ -106,7 +106,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_K_z``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cone_quadratic(
+        vis = _visualization.McnpVisualization.get_cone_quadratic(
             self.t_squared.value ** (1 / 2), self.plusminus_1.value
         )
         vis = vis.add_translation(_visualization.Vector(self.x.value, self.y.value, self.z.value))
diff --git a/src/pymcnp/inp/surface/option_kx.py b/src/pymcnp/inp/surface/option_kx.py
@@ -85,7 +85,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_Kx``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cone_quadratic(
+        vis = _visualization.McnpVisualization.get_cone_quadratic(
             self.t_squared.value ** (1 / 2), self.plusminus_1.value
         )
         vis = vis.add_rotation(_visualization.Vector(0, 1, 0), 90, (0, 0, 0))
diff --git a/src/pymcnp/inp/surface/option_ky.py b/src/pymcnp/inp/surface/option_ky.py
@@ -85,7 +85,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_Ky``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cone_quadratic(
+        vis = _visualization.McnpVisualization.get_cone_quadratic(
             self.t_squared.value ** (1 / 2), self.plusminus_1.value
         )
         vis = vis.add_rotation(_visualization.Vector(1, 0, 0), 90, (0, 0, 0))
diff --git a/src/pymcnp/inp/surface/option_kz.py b/src/pymcnp/inp/surface/option_kz.py
@@ -85,7 +85,7 @@ def to_pyvista(self):
             ``pyvista.PolyData`` for ``SurfaceOption_Kz``.
         """
 
-        vis = _visualization.PyMcnpVisualization.get_cone_quadratic(
+        vis = _visualization.McnpVisualization.get_cone_quadratic(
             self.t_squared.value ** (1 / 2), self.plusminus_1.value
         )
         vis = vis.add_translation(_visualization.Vector(0, 0, self.z.value))
diff --git a/src/pymcnp/inp/surface/option_p_0.py b/src/pymcnp/inp/surface/option_p_0.py
@@ -5,6 +5,7 @@
 from ...utils import types
 from ...utils import errors
 from ...utils import _parser
+from ...utils import _visualization
 
 
 class SurfaceOption_P0(_option.SurfaceOption_, keyword='p'):
@@ -82,3 +83,17 @@ def from_mcnp(source: str):
         d = types.Real.from_mcnp(tokens[4])
 
         return SurfaceOption_P0(a, b, c, d)
+
+    def to_pyvista(self):
+        """
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_Px``.
+
+        Returns:
+            ``pyvista.PolyData`` for ``SurfaceOption_Px``
+        """
+
+        vis = _visualization.McnpVisualization.get_plane(
+            self.a.value, self.b.value, self.c.value, self.d.value
+        )
+
+        return vis.data
diff --git a/src/pymcnp/inp/surface/option_p_1.py b/src/pymcnp/inp/surface/option_p_1.py
@@ -5,6 +5,7 @@
 from ...utils import types
 from ...utils import errors
 from ...utils import _parser
+from ...utils import _visualization
 
 
 class SurfaceOption_P1(_option.SurfaceOption_, keyword='p'):
@@ -128,3 +129,21 @@ def from_mcnp(source: str):
         z3 = types.Real.from_mcnp(tokens[9])
 
         return SurfaceOption_P1(x1, y1, z1, x2, y2, z2, x3, y3, z3)
+
+    def to_pyvista(self):
+        """
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_P0``.
+
+        Returns:
+            ``pyvista.PolyData`` for ``SurfaceOption_P1``
+        """
+
+        a = _visualization.Vector(self.x2 - self.x1, self.y2 - self.y1, self.z2 - self.z1)
+        b = _visualization.Vector(self.x3 - self.x1, self.y3 - self.y1, self.z3 - self.z1)
+        n = a * b
+
+        vis = _visualization.McnpVisualization.get_plane(
+            n.x, n.y, n.z, n.x * self.x1 + n.y * self.y1 + n.z * self.z1
+        )
+
+        return vis.data
diff --git a/src/pymcnp/inp/surface/option_px.py b/src/pymcnp/inp/surface/option_px.py
@@ -65,13 +65,13 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_Px``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_Px``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_Px``
+            ``pyvista.PolyData`` for ``SurfaceOption_Px``
         """
 
-        vis = _visualization.PyMcnpVisualization.get_plane(1, 0, 0, self.d.value)
+        vis = _visualization.McnpVisualization.get_plane(1, 0, 0, self.d.value)
         vis = vis.add_rotation(_visualization.Vector(0, 1, 0), 90, (0, 0, 0))
 
         return vis.data
diff --git a/src/pymcnp/inp/surface/option_py.py b/src/pymcnp/inp/surface/option_py.py
@@ -65,13 +65,13 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_Py``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_Py``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_Py``
+            ``pyvista.PolyData`` for ``SurfaceOption_Py``
         """
 
-        vis = _visualization.PyMcnpVisualization.get_plane(0, 1, 0, self.d.value)
+        vis = _visualization.McnpVisualization.get_plane(0, 1, 0, self.d.value)
         vis = vis.add_rotation(_visualization.Vector(1, 0, 0), 90, (0, 0, 0))
 
         return vis.data
diff --git a/src/pymcnp/inp/surface/option_pz.py b/src/pymcnp/inp/surface/option_pz.py
@@ -65,12 +65,12 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_Pz``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_Pz``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_Pz``
+            ``pyvista.PolyData`` for ``SurfaceOption_Pz``
         """
 
-        vis = _visualization.PyMcnpVisualization.get_plane(0, 0, 1, self.d.value)
+        vis = _visualization.McnpVisualization.get_plane(0, 0, 1, self.d.value)
 
         return vis.data
diff --git a/src/pymcnp/inp/surface/option_rcc.py b/src/pymcnp/inp/surface/option_rcc.py
@@ -116,10 +116,10 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_Rcc``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_Rcc``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_Rcc``
+            ``pyvista.PolyData`` for ``SurfaceOption_Rcc``
         """
 
         v = _visualization.Vector(self.vx.value, self.vy.value, self.vz.value)
@@ -128,7 +128,7 @@ def to_pyvista(self):
         cross = v * _visualization.Vector(0, 0, 1)
         angle = v & _visualization.Vector(0, 0, 1)
 
-        vis = _visualization.PyMcnpVisualization.get_cylinder_circle(h.norm(), self.r.value)
+        vis = _visualization.McnpVisualization.get_cylinder_circle(h.norm(), self.r.value)
         vis = vis.add_rotation(cross, angle, (0, 0, 0))
         vis = vis.add_translation(v)
 
diff --git a/src/pymcnp/inp/surface/option_rec.py b/src/pymcnp/inp/surface/option_rec.py
@@ -160,10 +160,10 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_Rec``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_Rec``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_Rec``
+            ``pyvista.PolyData`` for ``SurfaceOption_Rec``
         """
 
         v = _visualization.Vector(self.vx.value, self.vy.value, self.vz.value)
@@ -174,9 +174,7 @@ def to_pyvista(self):
         cross = v * _visualization.Vector(0, 0, 1)
         angle = v & _visualization.Vector(0, 0, 1)
 
-        vis = _visualization.PyMcnpVisualization.get_cylinder_ellipse(
-            h.norm(), v1.norm(), v2.norm()
-        )
+        vis = _visualization.McnpVisualization.get_cylinder_ellipse(h.norm(), v1.norm(), v2.norm())
         vis = vis.add_rotation(cross, angle, (0, 0, 0))
         vis = vis.add_translation(v)
 
diff --git a/src/pymcnp/inp/surface/option_rhp.py b/src/pymcnp/inp/surface/option_rhp.py
@@ -184,10 +184,10 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_Rhp``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_Rhp``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_Rhp``
+            ``pyvista.PolyData`` for ``SurfaceOption_Rhp``
         """
 
         v = _visualization.Vector(self.vx.value, self.vy.value, self.vz.value)
@@ -199,7 +199,7 @@ def to_pyvista(self):
         cross = v * _visualization.Vector(0, 0, 1)
         angle = v & _visualization.Vector(0, 0, 1)
 
-        vis = _visualization.PyMcnpVisualization.get_cylinder_hexagon(
+        vis = _visualization.McnpVisualization.get_cylinder_hexagon(
             h.norm(), r.apothem(), s.apothem(), t.apothem()
         )
         vis = vis.add_rotation(cross, angle, (0, 0, 0))
diff --git a/src/pymcnp/inp/surface/option_rpp.py b/src/pymcnp/inp/surface/option_rpp.py
@@ -108,13 +108,13 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_Rpp``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_Rpp``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_Rpp``
+            ``pyvista.PolyData`` for ``SurfaceOption_Rpp``
         """
 
-        vis = _visualization.PyMcnpVisualization.get_parallelipiped(
+        vis = _visualization.McnpVisualization.get_parallelipiped(
             self.xmin.value,
             self.xmax.value,
             self.ymin.value,
diff --git a/src/pymcnp/inp/surface/option_s.py b/src/pymcnp/inp/surface/option_s.py
@@ -86,13 +86,13 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_S``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_S``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_S``
+            ``pyvista.PolyData`` for ``SurfaceOption_S``
         """
 
-        vis = _visualization.PyMcnpVisualization.get_sphere(self.r.value)
+        vis = _visualization.McnpVisualization.get_sphere(self.r.value)
         vis = vis.add_translation(_visualization.Vector(self.x.value, self.y.value, self.z.value))
 
         return vis.data
diff --git a/src/pymcnp/inp/surface/option_so.py b/src/pymcnp/inp/surface/option_so.py
@@ -65,12 +65,12 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_So``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_So``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_So``
+            ``pyvista.PolyData`` for ``SurfaceOption_So``
         """
 
-        vis = _visualization.PyMcnpVisualization.get_sphere(self.r.value)
+        vis = _visualization.McnpVisualization.get_sphere(self.r.value)
 
         return vis.data
diff --git a/src/pymcnp/inp/surface/option_sph.py b/src/pymcnp/inp/surface/option_sph.py
@@ -86,13 +86,13 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_Sph``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_Sph``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_Sph``
+            ``pyvista.PolyData`` for ``SurfaceOption_Sph``
         """
 
-        vis = _visualization.PyMcnpVisualization.get_sphere(self.r.value)
+        vis = _visualization.McnpVisualization.get_sphere(self.r.value)
         vis = vis.add_translation(
             _visualization.Vector(self.vx.value, self.vy.value, self.vz.value)
         )
diff --git a/src/pymcnp/inp/surface/option_sx.py b/src/pymcnp/inp/surface/option_sx.py
@@ -72,13 +72,13 @@ def from_mcnp(source: str):
 
     def to_pyvista(self):
         """
-        Generates ``pyvista[.]PolyData`` from ``SurfaceOption_Sx``.
+        Generates ``pyvista.PolyData`` from ``SurfaceOption_Sx``.
 
         Returns:
-            ``pyvista[.]PolyData`` for ``SurfaceOption_Sx``
+            ``pyvista.PolyData`` for ``SurfaceOption_Sx``
         """
 
-        vis = _visualization.PyMcnpVisualization.get_sphere(self.r.value)
+        vis = _visualization.McnpVisualization.get_sphere(self.r.value)
         vis = vis.add_translation(_visualization.Vector(self.x.value, 0, 0))
 
         return vis.data
diff --git a/src/pymcnp/inp/surface/option_sy.py b/src/pymcnp/inp/surface/option_sy.py
diff --git a/src/pymcnp/inp/surface/option_sz.py b/src/pymcnp/inp/surface/option_sz.py
diff --git a/src/pymcnp/inp/surface/option_trc.py b/src/pymcnp/inp/surface/option_trc.py
diff --git a/src/pymcnp/inp/surface/option_tx.py b/src/pymcnp/inp/surface/option_tx.py
diff --git a/src/pymcnp/inp/surface/option_ty.py b/src/pymcnp/inp/surface/option_ty.py
diff --git a/src/pymcnp/inp/surface/option_tz.py b/src/pymcnp/inp/surface/option_tz.py
diff --git a/src/pymcnp/inp/surface/option_wed.py b/src/pymcnp/inp/surface/option_wed.py
