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
|
# SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
# SPDX-License-Identifier: BSD-3-Clause
""" Apply a series of events produced by vtk-js RenderWindowInteractor to a
vtkRenderwindow via the vtkRemoteInteractionAdapter class. The final image is
the expected scene after all interactions have been applied.
"""
from vtkmodules.vtkRenderingCore import (
vtkActor,
vtkPolyDataMapper,
vtkRenderer,
vtkRenderWindow,
vtkRenderWindowInteractor,
)
from vtkmodules.vtkWebCore import vtkRemoteInteractionAdapter
from vtkmodules.vtkFiltersSources import vtkConeSource
from vtkmodules.test import Testing
import os
import json
# Required for rendering initialization,
import vtkmodules.vtkRenderingOpenGL2 # noqa
# Required for interactor initialization
from vtkmodules.vtkInteractionStyle import vtkInteractorStyleSwitch # noqa
# The scene to test. In some platforms reusing the window & renderer across
# the two test cases causes segfault. We start all tests from a clean state by
# creating the scene from scratch each time.
class Scene:
def __init__(self):
self.dataFile = os.path.join(
Testing.VTK_DATA_ROOT, "Data", "remote_events.json"
)
self.imageFile = "TestRemoteInteractionAdapter.png"
self.adapter = vtkRemoteInteractionAdapter()
print("dataFile: {}".format(self.dataFile))
if not os.path.isfile(self.dataFile):
raise RuntimeError("Datafile is missing")
self.renderer = vtkRenderer()
self.renderWindow = vtkRenderWindow()
self.renderWindow.AddRenderer(self.renderer)
self.renderWindow.SetSize(300, 300)
self.renderWindowInteractor = vtkRenderWindowInteractor()
self.renderWindowInteractor.SetRenderWindow(self.renderWindow)
self.renderWindowInteractor.GetInteractorStyle().SetCurrentStyleToTrackballCamera()
self.cone_source = vtkConeSource()
self.mapper = vtkPolyDataMapper()
self.mapper.SetInputConnection(self.cone_source.GetOutputPort())
self.actor = vtkActor()
self.actor.SetMapper(self.mapper)
self.renderer.AddActor(self.actor)
self.renderer.ResetCamera()
self.renderWindowInteractor.Initialize()
class TestRemoteInteractorAdapter(Testing.vtkTest):
def test0(self):
"""Use class methods API for ProcessEvent"""
scene = Scene()
adapter = vtkRemoteInteractionAdapter()
adapter.SetInteractor(scene.renderWindowInteractor)
with open(scene.dataFile, "r") as f:
data = json.load(f)
for event in data["events"]:
event_str = json.dumps(event)
status = adapter.ProcessEvent(event_str)
assert status, f"Failed to process event\n {event_str}"
scene.renderWindowInteractor.Render()
self.assertImageMatch(scene.renderWindow, scene.imageFile)
def test1(self):
"""Use static method API for ProcessEvent"""
scene = Scene()
with open(scene.dataFile, "r") as f:
data = json.load(f)
for event in data["events"]:
event_str = json.dumps(event)
status = vtkRemoteInteractionAdapter.ProcessEvent(
scene.renderWindowInteractor, event_str
)
assert status, f"Failed to process event\n {event_str}"
scene.renderWindowInteractor.Render()
self.assertImageMatch(scene.renderWindow, scene.imageFile)
if __name__ == "__main__":
Testing.main([(TestRemoteInteractorAdapter, "test")])
|
