feat: Add StructuredGrid support

Author: AlejandroFernandezLuces

src/ansys/tools/visualization_interface/backends/pyvista/pyvista_interface.py

@@ -246,7 +246,7 @@ def plot(
             self._show_edges = plotting_options["show_edges"]
 
         # Check what kind of object we are dealing with
-        if isinstance(plottable_object, (pv.PolyData, pv.UnstructuredGrid)):
+        if isinstance(plottable_object, (pv.PolyData, pv.UnstructuredGrid, pv.StructuredGrid)):
             if "clipping_plane" in plotting_options:
                 mesh = self.clip(plottable_object, plotting_options["clipping_plane"])
                 plotting_options.pop("clipping_plane", None)

src/ansys/tools/visualization_interface/backends/pyvista/widgets/mesh_slider.py

@@ -82,6 +82,8 @@ def callback(self, state: bool) -> None:
                     mesh_id = "UnstructuredGrid(" + mesh.memory_address + ")"
                 elif isinstance(mesh, pv.MultiBlock):
                     mesh_id = "MultiBlock(" + mesh.memory_address + ")"
+                elif isinstance(mesh, pv.StructuredGrid):
+                    mesh_id = "StructuredGrid(" + mesh.memory_address + ")"
                 self._mesh_actor_list.append(self.plotter_helper._pl.scene.actors[mesh_id])
                 self.plotter_helper._pl.scene.remove_actor(mesh_id)
 

tests/test_generic_plotter.py

@@ -22,6 +22,7 @@
 """Test module for the generic plotter."""
 from pathlib import Path
 
+import numpy as np
 import pyvista as pv
 
 from ansys.tools.visualization_interface import ClipPlane, MeshObjectPlot, Plotter
@@ -52,6 +53,7 @@ def test_plotter_add_mb():
     pl.plot(mb)
     pl.show()
 
+
 def test_plotter_add_unstructured_grid():
     """Adds unstructured grid to the plotter."""
     pl = Plotter()
@@ -60,6 +62,18 @@ def test_plotter_add_unstructured_grid():
     pl.plot(ug)
     pl.show()
 
+
+def test_plotter_add_structured_grid():
+    """Adds unstructured grid to the plotter."""
+    pl = Plotter()
+    xrng = np.arange(-10, 10, 2, dtype=np.float32)
+    yrng = np.arange(-10, 10, 5, dtype=np.float32)
+    zrng = np.arange(-10, 10, 1, dtype=np.float32)
+    x, y, z = np.meshgrid(xrng, yrng, zrng, indexing='ij')
+    grid = pv.StructuredGrid(x, y, z)
+    pl.plot(grid)
+
+
 def test_plotter_add_custom():
     """Adds a MeshObjectPlot object to the plotter."""
     sphere = pv.Sphere()
@@ -90,6 +104,19 @@ def test_clipping_plane():
     pl.show()
 
 
+def test_clipping_plane_structured_grid():
+    """Test clipping plane usage with unstructured grid."""
+    pl = Plotter()
+    xrng = np.arange(-10, 10, 2, dtype=np.float32)
+    yrng = np.arange(-10, 10, 5, dtype=np.float32)
+    zrng = np.arange(-10, 10, 1, dtype=np.float32)
+    x, y, z = np.meshgrid(xrng, yrng, zrng, indexing='ij')
+    grid = pv.StructuredGrid(x, y, z)
+    clipping_plane = ClipPlane()
+    pl.plot(grid, clipping_plane=clipping_plane)
+    pl.show()
+
+
 def test_plotter_add_list():
     """Adds a list to the plotter."""
     pl = Plotter()