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
|
import pyvista as pv
from pyvista.trame.ui import plotter_ui
from vtkmodules.vtkFiltersSources import vtkSphereSource
from vtkmodules.vtkRenderingCore import (
vtkActor,
vtkPolyDataMapper,
vtkRenderer,
vtkRenderWindow,
)
from trame.app import get_server
from trame.widgets import vtk as vtk_widgets
CLIENT_TYPE = "vue3"
USE_PV = False
RADIUS = 0.1
if CLIENT_TYPE == "vue2":
from trame.ui.vuetify2 import SinglePageLayout
from trame.widgets import vuetify2 as vuetify
else:
from trame.ui.vuetify3 import SinglePageLayout
from trame.widgets import vuetify3 as vuetify
class App:
def __init__(self, server=None):
self.server = get_server(server, client_type=CLIENT_TYPE)
if USE_PV:
self.plotter = pv.Plotter()
else:
self.renderWindow = vtkRenderWindow()
self.renderer = vtkRenderer()
self.renderWindow.AddRenderer(self.renderer)
self.sources_counter = 0
self.ui()
@property
def ctrl(self):
return self.server.controller
@property
def state(self):
return self.server.state
def add_actor(self):
center = (self.sources_counter * RADIUS * 2.1, 0, 0)
self.sources_counter += 1
if USE_PV:
mapper = pv.DataSetMapper(pv.Sphere(radius=RADIUS, center=center))
actor = pv.Actor(mapper=mapper)
self.plotter.add_actor(actor)
else:
_source = vtkSphereSource()
_source.SetRadius(RADIUS)
_source.SetCenter(*center)
mapper = vtkPolyDataMapper()
actor = vtkActor()
mapper.SetInputConnection(_source.GetOutputPort())
actor.SetMapper(mapper)
self.renderer.AddActor(actor)
self.ctrl.view_update()
self.ctrl.view_reset_camera()
def ui(self):
with SinglePageLayout(self.server) as layout:
self.layout = layout
layout.icon.click = self.ctrl.view_reset_camera
with layout.content:
with vuetify.VCol():
vuetify.VBtn("AddActor", click=self.add_actor)
with vuetify.VContainer(
fluid=True,
classes="pa-0 fill-height",
):
view = (
plotter_ui(self.plotter, mode="client")
if USE_PV
else vtk_widgets.VtkLocalView(self.renderWindow)
)
self.ctrl.view_update = view.update
self.ctrl.view_reset_camera = view.reset_camera
if __name__ == "__main__":
app = App()
app.add_actor()
app.add_actor()
app.server.start()
|
