File: vtk-92.py

package info (click to toggle)
python-trame 3.12.0-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 101,620 kB
  • sloc: python: 13,515; sh: 183; javascript: 93; makefile: 7
file content (59 lines) | stat: -rw-r--r-- 1,698 bytes parent folder | download
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
import vtk

from trame.app import TrameApp
from trame.ui.vuetify3 import SinglePageLayout
from trame.widgets import vuetify3, vtk as vtk_widgets


class PushCamera(TrameApp):
    def __init__(self, server=None):
        super().__init__(server)
        self._setup_vtk()
        self._build_ui()

    def _setup_vtk(self):
        render_window = vtk.vtkRenderWindow()
        render_window.SetOffScreenRendering(True)

        renderer = vtk.vtkRenderer()
        render_window.AddRenderer(renderer)

        cone = vtk.vtkConeSource()
        mapper = vtk.vtkPolyDataMapper()
        actor = vtk.vtkActor()

        mapper.SetInputConnection(cone.GetOutputPort())
        actor.SetMapper(mapper)
        renderer.AddActor(actor)
        renderer.ResetCamera()

        self.render_window = render_window
        self.camera = renderer.GetActiveCamera()

    def _build_ui(self):
        with SinglePageLayout(self.server) as self.ui:
            self.ui.title.set_text("Client-side Camera Control")

            with self.ui.content:
                with vuetify3.VContainer(
                    fluid=True,
                    classes="pa-0 fill-height",
                ):
                    vtk_widgets.VtkLocalView(self.render_window, ctx_name="view")

            with self.ui.toolbar:
                vuetify3.VSpacer()
                vuetify3.VBtn("Set Top View", click=self.set_top_view)

    def set_top_view(self):
        self.camera.SetPosition(0, 5, 0)
        self.camera.SetFocalPoint(0, 0, 0)
        self.camera.SetViewUp(0, 0, -1)

        self.ctx.view.push_camera()
        # self.ctx.view.update()


if __name__ == "__main__":
    app = PushCamera()
    app.server.start()