From d8e18ebd6ce8cf2f23d04b85836d408946718115 Mon Sep 17 00:00:00 2001
From: Jaswant Panchumarti <jaswant.panchumarti@kitware.com>
Date: Fri, 15 Nov 2024 13:32:56 -0500
Subject: [PATCH] docs(examples): Add VTK polydata file viewer

---
 examples/vtk/file_viewer.py | 119 ++++++++++++++++++++++++++++++++++++
 1 file changed, 119 insertions(+)
 create mode 100644 examples/vtk/file_viewer.py

diff --git a/examples/vtk/file_viewer.py b/examples/vtk/file_viewer.py
new file mode 100644
index 0000000..fe6b26e
--- /dev/null
+++ b/examples/vtk/file_viewer.py
@@ -0,0 +1,119 @@
+from trame.app import get_server
+from trame.ui.html import DivLayout
+from trame.widgets import html, client
+from trame_vtklocal.widgets import vtklocal
+from trame.decorators import TrameApp, trigger
+
+from vtkmodules.vtkIOXML import vtkXMLPolyDataReader
+from vtkmodules.vtkRenderingCore import (
+    vtkActor,
+    vtkPolyDataMapper,
+    vtkRenderer,
+    vtkRenderWindow,
+    vtkRenderWindowInteractor,
+)
+
+# Required for vtk factory
+import vtkmodules.vtkRenderingOpenGL2  # noqa
+from vtkmodules.vtkInteractionStyle import vtkInteractorStyleSwitch  # noqa
+
+CLIENT_TYPE = "vue3"
+
+# -----------------------------------------------------------------------------
+# VTK pipeline
+# -----------------------------------------------------------------------------
+
+
+def create_vtk_pipeline(file_name):
+    renderer = vtkRenderer()
+    renderWindow = vtkRenderWindow()
+    renderWindow.AddRenderer(renderer)
+
+    renderWindowInteractor = vtkRenderWindowInteractor()
+    renderWindowInteractor.SetRenderWindow(renderWindow)
+    renderWindowInteractor.GetInteractorStyle().SetCurrentStyleToTrackballCamera()
+
+    reader = vtkXMLPolyDataReader(file_name=file_name)
+
+    mapper = vtkPolyDataMapper(scalar_visibility=False)
+    mapper.SetInputConnection(reader.GetOutputPort())
+
+    actor = vtkActor()
+    actor.SetMapper(mapper)
+
+    renderer.AddActor(actor)
+    renderer.SetBackground(0.1, 0.2, 0.4)
+    renderer.ResetCamera()
+
+    return renderWindow
+
+
+# -----------------------------------------------------------------------------
+# GUI
+# -----------------------------------------------------------------------------
+
+
+@TrameApp()
+class DemoApp:
+    def __init__(self, server=None):
+        self.server = get_server(server, client_type=CLIENT_TYPE)
+        self.server.cli.add_argument("-f", "--filename", required=True)
+        args, _ = self.server.cli.parse_known_args()
+        file_name = args.filename
+
+        self.render_window = create_vtk_pipeline(file_name)
+        self.server.state.update(dict(mem_blob=0, mem_vtk=0))
+        self.html_view = None
+        self.ui = self._ui()
+        # print(self.ui)
+
+    @trigger("export")
+    def export(self, format):
+        return self.html_view.export(format)
+
+    def reset_camera(self):
+        self.html_view.reset_camera()
+
+    def _ui(self):
+        with DivLayout(self.server) as layout:
+            client.Style("body { margin: 0; }")
+            with html.Div(
+                style="position: absolute; left: 0; top: 0; width: 100vw; height: 100vh;"
+            ):
+                self.html_view = vtklocal.LocalView(
+                    self.render_window,
+                    throttle_rate=20,
+                    cache_size=("cache", 0),
+                    memory_vtk="mem_vtk = $event",
+                    memory_arrays="mem_blob = $event",
+                )
+            html.Div(
+                "Scene: {{ (mem_vtk / 1024).toFixed(1) }}KB - Arrays: {{ (mem_blob / 1024).toFixed(1) }}KB - cache: {{ (cache/1024).toFixed(1) }}KB ",
+                style="position: absolute; top: 1rem; left: 1rem; z-index: 10; background: white; padding: 1rem; border-radius: 1rem;",
+            )
+            html.Button(
+                "Export json",
+                click="utils.download('scene-wasm.json', trigger('export', ['json']), 'application/octet-stream')",
+                style="position: absolute; top: 3rem; right: 1rem; z-index: 10;",
+            )
+            html.Button(
+                "Export zip",
+                click="utils.download('scene-wasm.zip', trigger('export', ['zip']), 'application/octet-stream')",
+                style="position: absolute; top: 3rem; right: 7rem; z-index: 10;",
+            )
+            html.Button(
+                "Reset Camera",
+                click=self.reset_camera,
+                style="position: absolute; top: 3rem; right: 14rem; z-index: 10;",
+            )
+
+        return layout
+
+
+# -----------------------------------------------------------------------------
+# Main
+# -----------------------------------------------------------------------------
+
+if __name__ == "__main__":
+    app = DemoApp()
+    app.server.start()