fix pv example

Author: jourdain

05_paraview/StateLoader.py

@@ -1,34 +1,32 @@
-from paraview.web import venv  # Available in PV 5.10
-
-import os
+from paraview.web import venv  # Available in PV 5.10-RC2+
+from paraview import simple
 
+from pathlib import Path
 from trame.app import get_server
-from trame.widgets import vuetify, paraview
+from trame.widgets import vuetify, paraview, client
 from trame.ui.vuetify import SinglePageLayout
 
-from paraview import simple
-
 # -----------------------------------------------------------------------------
-# trame setup
+# Trame setup
 # -----------------------------------------------------------------------------
 
 server = get_server()
 state, ctrl = server.state, server.controller
 
+# Preload paraview modules onto server
+paraview.initialize(server)
+
 # -----------------------------------------------------------------------------
 # ParaView code
 # -----------------------------------------------------------------------------
 
-layout = None
-
 
-def load_data():
+def load_data(**kwargs):
     # CLI
-    server.cli.add_argument("--data", help="Path to state file", dest="data")
     args, _ = server.cli.parse_known_args()
 
-    full_path = os.path.abspath(args.data)
-    working_directory = os.path.dirname(full_path)
+    full_path = str(Path(args.data).resolve().absolute())
+    working_directory = str(Path(args.data).parent.resolve().absolute())
 
     # ParaView
     simple.LoadState(
@@ -38,28 +36,41 @@ def load_data():
     )
     view = simple.GetActiveView()
     view.MakeRenderWindowInteractor(True)
+    simple.Render(view)
 
     # HTML
-    with layout:
+    with SinglePageLayout(server) as layout:
+        layout.icon.click = ctrl.view_reset_camera
+        layout.title.set_text("ParaView State Viewer")
+
         with layout.content:
             with vuetify.VContainer(fluid=True, classes="pa-0 fill-height"):
                 html_view = paraview.VtkRemoteView(view)
                 ctrl.view_reset_camera = html_view.reset_camera
                 ctrl.view_update = html_view.update
 
 
+ctrl.on_server_ready.add(load_data)
+
 # -----------------------------------------------------------------------------
 # GUI
 # -----------------------------------------------------------------------------
+
 state.trame__title = "State Viewer"
 
 with SinglePageLayout(server) as layout:
     layout.icon.click = ctrl.view_reset_camera
     layout.title.set_text("ParaView State Viewer")
 
+    with layout.content:
+        with vuetify.VContainer(fluid=True, classes="pa-0 fill-height"):
+            client.Loading("Loading state")
+
+
 # -----------------------------------------------------------------------------
 # Main
 # -----------------------------------------------------------------------------
 
 if __name__ == "__main__":
+    server.cli.add_argument("--data", help="Path to state file", dest="data")
     server.start()