support 3D visualization for jupyter using trame

Author: zhangzhanqun

examples/ICFD/icfd_free_convection_flow.py

@@ -1,6 +1,6 @@
 """
-Thermal flow
-============
+Free convection flow
+====================
 
 This example shows a simple ICFD forced-convection input deck with a coarse mesh.
 The executable file for LS-DYNA is `` ls-dyna_smp_d_R13.1_138-g8429c8a10f_winx64_ifort190.exe``.

pyproject.toml

@@ -28,6 +28,7 @@ classifiers = [
 dependencies = ["ansys-dpf-core>=0.7.2",
                 "ansys-api-dyna==0.3.6",
                 "ansys-platform-instancemanagement~=1.0", 
+				"pyvista[jupyter]==0.42.3",
 ]
 
 [project.optional-dependencies]

src/ansys/dyna/core/pre/dynabase.py

@@ -274,6 +274,14 @@ def get_data(self) -> List:
         return None
 
 
+class ParameterType(Enum):
+    """Contains the parameter types."""
+
+    R = 1
+    I = 2
+    C = 3
+
+
 class DynaBase:
     """Contains methods for creating a general LS-DYNA keyword."""
 

src/ansys/dyna/core/pre/graphics/graphics.py

@@ -5,7 +5,7 @@
 
 import numpy as np
 import pyvista as pv
-from pyvista.plotting.plotting import Plotter
+from pyvista.plotting import Plotter
 import vtk
 
 import ansys.dyna.core.pre as pre
@@ -631,12 +631,13 @@ def __draw_parts(self, parts: List = [], update: bool = False, spline: bool = Fa
     def _show_selector(self):
         """Chooses between using Trame or Python visualizer."""
         if self._use_trame:  # pragma: no cover
+            pv.set_jupyter_backend("server")
             visualizer = TrameVisualizer()
             visualizer.set_scene(self._plotter)
             visualizer.show()
         else:
             self._plotter.camera_position = "xy"
-            self._plotter.show()
+            self._plotter.show(jupyter_backend="static")
 
     def __update_bt_icons(self):
         """Update the icons on display."""

src/ansys/dyna/core/pre/graphics/trame_gui.py

@@ -2,7 +2,8 @@
 try:
     from pyvista.trame.ui import plotter_ui
     from trame.app import get_server
-    from trame.ui.vuetify import SinglePageLayout
+    from trame.ui.vuetify3 import SinglePageLayout
+    from trame.widgets import vuetify3
 
     _HAS_TRAME = True
 
@@ -18,7 +19,7 @@ def __init__(self) -> None:
         if not _HAS_TRAME:  # pragma: no cover
             raise ModuleNotFoundError("The package 'pyvista[trame]' is required to use this function.")
 
-        self.server = get_server()
+        self.server = get_server(client_type="vue3")
         self.state, self.ctrl = self.server.state, self.server.controller
 
     def set_scene(self, plotter):
@@ -31,16 +32,17 @@ def set_scene(self, plotter):
         plotter : pv.Plotter
             PyVista plotter to render the mesh.
         """
-        self.state.trame__title = "PyPrime Viewer"
+        self.state.trame__title = "PyDYNA Viewer"
 
         with SinglePageLayout(self.server) as layout:
             layout.icon.click = self.ctrl.view_reset_camera
-            layout.title.set_text("PyPrime")
+            layout.title.set_text("PyDYNA")
 
             with layout.content:
-                # Use PyVista UI template for Plotters
-                view = plotter_ui(plotter)
-                self.ctrl.view_update = view.update
+                with vuetify3.VContainer(fluid=True, classes="pa-0 fill-height"):
+                    # Use PyVista UI template for Plotters
+                    view = plotter_ui(plotter)
+                    self.ctrl.view_update = view.update
 
             # hide footer with trame watermark
             layout.footer.hide()

src/ansys/dyna/core/pre/launcher.py

@@ -3,6 +3,7 @@
 import os
 import socket
 import subprocess
+import sys
 import threading
 from time import sleep
 from zipfile import ZipFile
@@ -74,6 +75,17 @@ def port_in_use(port, host=LOCALHOST):
             return True
 
 
+def get_virtualenv_path():
+    """Get the virtual environment path."""
+
+    python_executable_path = sys.executable
+    if hasattr(sys, "real_prefix"):
+        virtual_env_path = sys.prefix
+    else:
+        virtual_env_path = os.path.dirname(python_executable_path)
+    return virtual_env_path
+
+
 def launch_grpc(port=DYNAPRE_DEFAULT_PORT, ip=LOCALHOST, server_path="") -> tuple:  # pragma: no cover
     """
     Launch the pre service locally in gRPC mode.
@@ -123,6 +135,7 @@ def launch_grpc(port=DYNAPRE_DEFAULT_PORT, ip=LOCALHOST, server_path="") -> tupl
             # threadserver.run()
             # threadserver.setDaemon(True)
             # threadserver.start()
+            # env_path = get_virtualenv_path()
             process = subprocess.Popen("python kwserver.py", cwd=server_path, shell=True)
             waittime = 0
             while not DynaSolution.grpc_local_server_on():