1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
r"""
Version for trame 1.x - https://github.com/Kitware/trame/blob/release-v1/examples/VTK/ContourGeometry/ClientView.py
Delta v1..v2 - https://github.com/Kitware/trame/commit/3852cba56cd63f6efa684d5e5fb00881a6111e81
Installation requirements:
pip install trame trame-vuetify trame-vtk vtk
"""
from pathlib import Path
from vtkmodules.vtkFiltersCore import vtkContourFilter
from vtkmodules.vtkIOXML import vtkXMLImageDataReader
from trame.app import get_server
from trame.ui.vuetify import SinglePageLayout
from trame.widgets import vtk as vtk_widgets
from trame.widgets import vuetify
# -----------------------------------------------------------------------------
# Trame initialization
# -----------------------------------------------------------------------------
server = get_server(client_type="vue2")
state, ctrl = server.state, server.controller
state.trame__title = "VTK contour - Remote/Local rendering"
# -----------------------------------------------------------------------------
# VTK pipeline
# -----------------------------------------------------------------------------
data_directory = Path(__file__).parent.parent.with_name("data")
head_vti = data_directory / "head.vti"
reader = vtkXMLImageDataReader()
reader.SetFileName(head_vti)
reader.Update()
contour = vtkContourFilter()
contour.SetInputConnection(reader.GetOutputPort())
contour.SetComputeNormals(1)
contour.SetComputeScalars(0)
# Extract data range => Update store/state
data_range = reader.GetOutput().GetPointData().GetScalars().GetRange()
contour_value = 0.5 * (data_range[0] + data_range[1])
# Configure contour with valid values
contour.SetNumberOfContours(1)
contour.SetValue(0, contour_value)
# Share with client
state.data_range = data_range
# -----------------------------------------------------------------------------
# Callbacks
# -----------------------------------------------------------------------------
@state.change("contour_value")
def update_contour(contour_value, **kwargs):
contour.SetValue(0, contour_value)
ctrl.ds_update()
# -----------------------------------------------------------------------------
# GUI
# -----------------------------------------------------------------------------
with SinglePageLayout(server) as layout:
layout.title.set_text("Contour Application - Local rendering")
with layout.toolbar:
vuetify.VSpacer()
vuetify.VSlider(
value=("contour_value", contour_value),
min=("data_range[0]",),
max=("data_range[1]",),
hide_details=True,
dense=True,
style="max-width: 300px",
change="contour_value = Number($event)",
)
vuetify.VSwitch(
v_model="$vuetify.theme.dark",
hide_details=True,
)
with vuetify.VBtn(icon=True, click="$refs.view.resetCamera()"):
vuetify.VIcon("mdi-crop-free")
vuetify.VProgressLinear(
indeterminate=True,
absolute=True,
bottom=True,
active=("trame__busy",),
)
with layout.content:
with vuetify.VContainer(fluid=True, classes="pa-0 fill-height"):
with vtk_widgets.VtkView() as view:
layout.icon.click = view.reset_camera
with vtk_widgets.VtkGeometryRepresentation():
polydata = vtk_widgets.VtkPolyData("contour", dataset=contour)
ctrl.ds_update = polydata.update
# -----------------------------------------------------------------------------
# Main
# -----------------------------------------------------------------------------
if __name__ == "__main__":
server.start()
|
