add ansys visualization interface plotter

Author: dipinknair

pyproject.toml

@@ -80,7 +80,7 @@ doc = [
     "sphinxemoji==0.3.1",
 ]
 viz = [
-    "pyvista>=0.39.1",
+    "ansys-tools-visualization-interface==0.2.6",
     "usd-core==24.3",
 ]
 

src/ansys/mechanical/core/embedding/app.py

@@ -34,13 +34,13 @@
 from ansys.mechanical.core.embedding.warnings import connect_warnings, disconnect_warnings
 
 try:
-    import pyvista  # noqa: F401
+    import ansys.tools.visualization_interface  # noqa: F401
 
-    HAS_PYVISTA = True
+    HAS_ANSYS_VIZ = True
     """Whether or not PyVista exists."""
 except:
 
-    HAS_PYVISTA = False
+    HAS_ANSYS_VIZ = False
 
 
 def _get_default_addin_configuration() -> AddinConfiguration:
@@ -229,13 +229,9 @@ def execute_script(self, script: str) -> typing.Any:
         rets = None
         return self.script_engine.ExecuteCode(script, SCRIPT_SCOPE, light_mode, args, rets)
 
-    def plot(self) -> None:
-        """Visualize the model in 3d.
-
-        Requires installation using the viz option. E.g.
-        pip install ansys-mechanical-core[viz]
-        """
-        if not HAS_PYVISTA:
+    def plotter(self):
+        """Return ansys.tools.visualization_interface Plotter object."""
+        if not HAS_ANSYS_VIZ:
             warnings.warn(
                 "Installation of viz option required! Use pip install ansys-mechanical-core[viz]"
             )
@@ -245,9 +241,21 @@ def plot(self) -> None:
             warnings.warn("Plotting is only supported with version 2024R2 and later!")
             return
 
-        from ansys.mechanical.core.embedding.viz.pyvista_plotter import plot_model
+        # TODO Check if anything loaded inside app or else show warning and return
+
+        from ansys.mechanical.core.embedding.viz.pyvista_plotter import to_pyvista_plotter
+
+        return to_pyvista_plotter(self)
+
+    def plot(self) -> None:  # TODO : add clipping plane
+        """Visualize the model in 3d.
+
+        Requires installation using the viz option. E.g.
+        pip install ansys-mechanical-core[viz]
+        """
+        _plotter = self.plotter()
 
-        plot_model(self)
+        return _plotter.show()
 
     @property
     def poster(self) -> Poster:

src/ansys/mechanical/core/embedding/viz/pyvista_plotter.py

@@ -29,6 +29,7 @@
 
 import Ansys  # isort: skip
 
+from ansys.tools.visualization_interface import Plotter
 import numpy as np
 import pyvista as pv
 
@@ -57,7 +58,7 @@ def _get_nodes_and_coords(tri_tessellation: "Ansys.Mechanical.Scenegraph.TriTess
 
 def to_pyvista_plotter(app: "ansys.mechanical.core.embedding.App"):
     """Convert the app's geometry to a pyvista plotter instance."""
-    plotter = pv.Plotter()
+    plotter = Plotter()
     for body in app.DataModel.GetObjectsByType(
         Ansys.Mechanical.DataModel.Enums.DataModelObjectCategory.Body
     ):
@@ -68,9 +69,3 @@ def to_pyvista_plotter(app: "ansys.mechanical.core.embedding.App"):
         color = pv.Color(bgr_to_rgb_tuple(body.Color))
         plotter.add_mesh(polydata, color=color, smooth_shading=True)
     return plotter
-
-
-def plot_model(app: "ansys.mechanical.core.embedding.App"):
-    """Plot the model."""
-    plotter = to_pyvista_plotter(app)
-    plotter.show()