/
23_vtk_interactive_render_exit.py
81 lines (63 loc) · 2.35 KB
/
23_vtk_interactive_render_exit.py
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
from trame.app import get_server
from trame.widgets import html, vtk
from trame.ui.html import DivLayout
# -----------------------------------------------------------------------------
# VTK
# -----------------------------------------------------------------------------
from vtkmodules.vtkFiltersSources import vtkSphereSource, vtkConeSource
from vtkmodules.vtkRenderingCore import (
vtkRenderer,
vtkRenderWindow,
vtkRenderWindowInteractor,
vtkPolyDataMapper,
vtkActor,
)
import vtkmodules.vtkRenderingOpenGL2 # noqa
from vtkmodules.vtkInteractionStyle import vtkInteractorStyleSwitch # noqa
DEFAULT_RESOLUTION = 6
renderer = vtkRenderer()
renderWindow = vtkRenderWindow()
renderWindow.AddRenderer(renderer)
renderWindowInteractor = vtkRenderWindowInteractor()
renderWindowInteractor.SetRenderWindow(renderWindow)
renderWindowInteractor.GetInteractorStyle().SetCurrentStyleToTrackballCamera()
def show(source):
mapper = vtkPolyDataMapper()
actor = vtkActor()
mapper.SetInputConnection(source.GetOutputPort())
actor.SetMapper(mapper)
renderer.AddActor(actor)
return source
cone = show(vtkConeSource())
sphere = show(vtkSphereSource())
renderer.ResetCamera()
renderWindow.Render()
# -----------------------------------------------------------------------------
# Web App setup
# -----------------------------------------------------------------------------
server = get_server()
state, ctrl = server.state, server.controller
with DivLayout(server) as layout:
container = layout.root
container.style = "width: 100vw; height: 100vh;"
view = vtk.VtkRemoteView(renderWindow, interactive_ratio=1, interactive_quality=0)
ctrl.view_update = view.update
html.Input(
type="range",
min=8,
max=1000,
step=1,
v_model=("resolution", 8),
style="position: absolute; top: 20px; left: 20px; z-index: 1; width: 25%; min-width: 300px;",
)
@state.change("resolution")
def update_resolution(resolution, **kwargs):
cone.SetResolution(int(resolution))
sphere.SetPhiResolution(int(resolution))
sphere.SetThetaResolution(int(resolution))
ctrl.view_update()
# -----------------------------------------------------------------------------
# start server
# -----------------------------------------------------------------------------
if __name__ == "__main__":
server.start()