-
-
Notifications
You must be signed in to change notification settings - Fork 54
/
24_vtk_download_image.py
98 lines (80 loc) · 2.91 KB
/
24_vtk_download_image.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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
from trame.app import get_server
from trame.widgets import vuetify, vtk
from trame.ui.vuetify import SinglePageLayout
# -----------------------------------------------------------------------------
# VTK
# -----------------------------------------------------------------------------
from vtkmodules.vtkFiltersSources import vtkConeSource
from vtkmodules.vtkIOImage import vtkPNGWriter
from vtkmodules.vtkRenderingCore import (
vtkRenderer,
vtkRenderWindow,
vtkRenderWindowInteractor,
vtkPolyDataMapper,
vtkActor,
vtkWindowToImageFilter,
)
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()
cone_source = vtkConeSource()
mapper = vtkPolyDataMapper()
actor = vtkActor()
mapper.SetInputConnection(cone_source.GetOutputPort())
actor.SetMapper(mapper)
renderer.AddActor(actor)
renderer.ResetCamera()
renderWindow.Render()
# -----------------------------------------------------------------------------
# Web App setup
# -----------------------------------------------------------------------------
server = get_server(client_type="vue2")
state, ctrl = server.state, server.controller
@state.change("resolution")
def update_resolution(resolution, **kwargs):
cone_source.SetResolution(int(resolution))
ctrl.view_update()
@ctrl.trigger("download_screenshot")
def download_screenshot():
w2if = vtkWindowToImageFilter()
w2if.SetInput(renderWindow)
w2if.SetInputBufferTypeToRGB()
w2if.ReadFrontBufferOff()
#
writer1 = vtkPNGWriter()
writer1.SetInputConnection(w2if.GetOutputPort())
writer1.SetWriteToMemory(True)
writer1.Write()
#
return server.protocol.addAttachment(memoryview(writer1.GetResult()))
with SinglePageLayout(server) as layout:
with layout.toolbar:
vuetify.VSpacer()
vuetify.VSlider(
v_model=("resolution", DEFAULT_RESOLUTION),
min=3,
max=60,
style="max-width: 300px",
dense=True,
hide_details=True,
)
with vuetify.VBtn(
icon=True,
click="utils.download('vtk.png', trigger('download_screenshot'), 'image/png')",
):
vuetify.VIcon("mdi-camera")
with layout.content:
with vuetify.VContainer(classes="pa-0 fill-height", fluid=True):
view = vtk.VtkRemoteView(renderWindow)
ctrl.view_update = view.update
# -----------------------------------------------------------------------------
# start server
# -----------------------------------------------------------------------------
if __name__ == "__main__":
server.start()