import io import json import zipfile from enum import IntEnum from trame_client.widgets.core import AbstractElement from trame_vtk.modules import common from trame_vtk import reference_id try: import zlib # noqa ZIP_COMPRESSION = zipfile.ZIP_DEFLATED except ImportError: ZIP_COMPRESSION = zipfile.ZIP_STORED def activate_module_for(helper, server, vtk_or_paraview_obj): if helper is not None: return helper if vtk_or_paraview_obj.IsA("vtkSMRemoteObject"): from trame_vtk.modules import paraview server.enable_module(paraview) return paraview.get_helper(server) else: from trame_vtk.modules import vtk server.enable_module(vtk) return vtk.get_helper(server) class HtmlElement(AbstractElement): def __init__(self, _elem_name, children=None, **kwargs): super().__init__(_elem_name, children, **kwargs) self._helper = None if self.server: self.server.enable_module(common) @property def module(self): return self._helper class VtkPiecewiseEditor(HtmlElement): def __init__(self, children=None, **kwargs): """ VtkPiecewiseEditor will create a UI to view and edit opacity function """ super().__init__("vtk-piecewise-editor", children, **kwargs) self._attr_names += [ "value", ("max_zoom", "maxZoom"), "padding", ("content_rectangle_style", "contentRectangleStyle"), ("result_opacity_line_style", "resultOpacityLineStyle"), ("inside_zoom_style", "insideZoomStyle"), ("gaussian_opacity_style", "gaussianOpacityStyle"), ("active_gaussian_opacity_style", "activeGaussianOpacityStyle"), ("active_linear_opacity_style", "activeLinearOpacityStyle"), ("linear_opacity_style", "linearOpacityStyle"), ("linear_opacity_control_style", "linearOpacityControlStyle"), ("active_linear_opacity_control_style", "activeLinearOpacityControlStyle"), ("zoom_control_style", "zoomControlStyle"), ] self._event_names += [ "opacities", "input", ] class VtkAlgorithm(HtmlElement): def __init__(self, children=None, **kwargs): """ VtkAlgorithm will create a vtk.js object from its vtkClass and state :param vtk_class: Name of vtk.js class to instantiate from `here `_ :type vtk_class: str :param state: JS object listing all the value to override :type state: str/obj :param port: Port to bind into parent VtkAlgorithm or VtkRepresentation :type port: int """ super().__init__("vtk-algorithm", children, **kwargs) self._attr_names += [ ("vtk_class", "vtkClass"), "state", "port", ] class VtkCellData(HtmlElement): def __init__(self, children=None, **kwargs): super().__init__("vtk-cell-data", children, **kwargs) class VtkDataArray(HtmlElement): def __init__(self, **kwargs): """ VtkDataArray :param name: :type name: str :param registration: :type registration: str :param type: :type type: str :param values: :type values: Array/TypedArray :param number_of_components: :type number_of_components: int """ super().__init__("vtk-data-array", **kwargs) self._attr_names += [ "name", "registration", "type", "values", "number_of_components", ] class VtkFieldData(HtmlElement): def __init__(self, children=None, **kwargs): super().__init__("vtk-field-data", children, **kwargs) class VtkGeometryRepresentation(HtmlElement): def __init__(self, children=None, **kwargs): """ VtkGeometryRepresentation :param id: Identifier used in picking :type id: str :param color_map_preset: Name of a vtk.js color preset :type color_map_preset: str :param color_data_range: :type color_data_range: [min, max] :param actor: Properties to edit on actor :type values: {} :param mapper: Properties to edit on mapper :type mapper: {} :param property: Properties to edit on property :type property: {} """ super().__init__("vtk-geometry-representation", children, **kwargs) self._attr_names += [ "id", "color_map_preset", "color_data_range", "actor", "mapper", "property", ] class VtkGlyphRepresentation(HtmlElement): def __init__(self, children=None, **kwargs): """ VtkGlyphRepresentation :param color_map_preset: Name of a vtk.js color preset :type color_map_preset: str :param color_data_range: :type color_data_range: [min, max] :param actor: Properties to edit on actor :type values: {} :param mapper: Properties to edit on mapper :type mapper: {} :param property: Properties to edit on property :type property: {} """ super().__init__("vtk-glyph-representation", children, **kwargs) self._attr_names += [ "color_map_preset", "color_data_range", "actor", "mapper", "property", ] class VtkMesh(HtmlElement): def __init__( self, name, dataset=None, field_to_keep=None, point_arrays=None, cell_arrays=None, **kwargs, ): super().__init__("vtk-mesh", **kwargs) self.__name = name self.__dataset = dataset self.__field_to_keep = field_to_keep self.__point_arrays = point_arrays self.__cell_arrays = cell_arrays self._attr_names += ["port", "state"] self._helper = None if dataset: self._helper = activate_module_for(self._helper, self.server, dataset) self._attributes["state"] = f':state="{name}"' self.update() def set_dataset(self, dataset): """ Change this mesh's internal dataset and update shared state""" self.__dataset = dataset self._helper = activate_module_for(self._helper, self.server, dataset) self.update() def update(self, **kwargs): """ Propagate changes in internal data to shared state """ if self.__dataset: self.server.state[self.__name] = self._helper.mesh( self.__dataset, field_to_keep=kwargs.get("field_to_keep", self.__field_to_keep), point_arrays=kwargs.get("point_arrays", self.__point_arrays), cell_arrays=kwargs.get("cell_arrays", self.__cell_arrays), ) class VtkPointData(HtmlElement): def __init__(self, children=None, **kwargs): super().__init__("vtk-point-data", children, **kwargs) class VtkPolyData(HtmlElement): def __init__(self, name, children=None, dataset=None, **kwargs): """ VtkPolyData :param port: Port to bind into parent VtkAlgorithm or VtkRepresentation :type port: int :param points: Points (xyz array) :type points: Array :param verts: CellArray to use for verts :type verts: Array :param lines: CellArray to use for lines :type lines: Array :param polys: CellArray to use for polys :type polys: Array :param strips: CellArray to use for strips :type strips: Array :param connectivity: One of 'manual', 'points', 'triangles', 'strips'. When 'manual' is used we expect the user to provide cells. For the other options, the JS code will generate the cells automatically. :type connectivity: str """ super().__init__("vtk-polydata", children, **kwargs) self.__name = name self.__dataset = dataset self._attr_names += [ "port", "points", "verts", "lines", "polys", "strips", "connectivity", ] self._helper = None if dataset: self._helper = activate_module_for(self._helper, self.server, dataset) self._attributes["bind"] = f'v-bind="{name}.mesh"' self.update() def set_dataset(self, dataset): """ Change this polydata's internal dataset and update shared state """ self.__dataset = dataset self._helper = activate_module_for(self._helper, self.server, dataset) self.update() def update(self): """ Propagate changes in internal data to shared state """ if self.__dataset: self.server.state[self.__name] = self._helper.mesh(self.__dataset) class VtkReader(HtmlElement): def __init__(self, **kwargs): """ VtkReader :param port: Port to bind into parent VtkAlgorithm or VtkRepresentation :type port: int :param parse_as_array_buffer: base64 strings :type parse_as_array_buffer: str :param parse_as_text: text content to parse :type parse_as_text: str :param render_on_update: Once ready trigger a render :type render_on_update: bool :param reset_camera_on_update: Once ready trigger a resetCamera :type reset_camera_on_update: bool :param url: url to download data from :type url: str :param vtk_class: Class name to use as reader :type vtk_class: str """ super().__init__("vtk-reader", **kwargs) self._attr_names += [ ("parse_as_array_buffer", "parseAsArrayBuffer"), ("parse_as_text", "parseAsText"), "port", ("render_on_update", "renderOnUpdate"), ("reset_camera_on_update", "resetCameraOnUpdate"), "url", ("vtk_class", "vtkClass"), ] class VtkRemoteLocalView(HtmlElement): """ The VtkRemoteLocalView component is a blend of VtkLocalView and VtkRemoteView where the user can choose dynamically which mode they want to be in. When instantiating a VtkRemoteLocalView several variables and triggers will be created for you to more easily control your view. >>> rl_view = vtk.VtkRemoteLocalView( ... # Instance of the view (required) [VTK: vtkRenderWindow | Paraview: viewProxy] ... view=..., ... # Just VtkRemoteLocalView params ... namespace=..., # Prefix for variables and triggers. See below. (required) ... mode="local", # Decide between local or remote. See below. ... disable_auto_switch=True # Skip automatic remote rendering switch while local rendering is pending ... # VtkRemoteView params ... ** remote_view_params, ... # VtkLocalView params ... **local_view_params, ... ) """ _next_id = 0 def __init__(self, view, enable_rendering=True, widgets=[], **kwargs): super().__init__("vtk-remote-local-view", **kwargs) self._helper = None self._helper = activate_module_for(self._helper, self.server, view) self._helper.has_capabilities("web", "rendering") __ns = kwargs.get("namespace", "view") if __ns == "view": VtkRemoteLocalView._next_id += 1 if VtkRemoteLocalView._next_id > 1: __ns = f"view{VtkRemoteLocalView._next_id}" self.__mode_key = f"{__ns}Mode" self.__scene_id = f"{__ns}Scene" self.__view_key_id = f"{__ns}Id" self.__ref = kwargs.get("ref", __ns) self.__rendering = enable_rendering self.__namespace = __ns self._widgets = widgets # !!! HACK !!! # Allow user to configure view mode by providing (..., local/remote) and or "local/remote" __mode_expression = kwargs.get("mode", self.__mode_key) __mode_start = "remote" if isinstance(__mode_expression, (tuple, list)): if len(__mode_expression) == 2: __mode_expression, __mode_start = __mode_expression else: __mode_expression = __mode_expression[0] elif __mode_expression in ["local", "remote"]: __mode_start = __mode_expression __mode_expression = self.__mode_key self._attributes["mode"] = f':mode="{__mode_expression}"' # !!! HACK !!! self.server.state[self.__view_key_id] = self._helper.id(view) self.__view = view self.__wrapped_view = self._helper.view(view, name=__ns, mode=__mode_start) # Provide mandatory attributes self._attributes["ref"] = f'ref="{self.__ref}"' self._attributes["refprefix"] = f'refPrefix="{self.__ref}"' self._attributes["namespace"] = f'namespace="{__ns}"' # vue2/3 compatibility self._attributes["view_id"] = f':viewId="{self.__view_key_id}"' self._attributes["view_state"] = f':viewState="{self.__scene_id}"' self._attr_names += [ # "mode", # <--- Managed by hand above "context_name", ("enable_picking", "enablePicking"), ("interactive_quality", "interactiveQuality"), ("interactive_ratio", "interactiveRatio"), ("still_ratio", "stillRatio"), ("still_quality", "stillQuality"), ("interactor_events", "interactorEvents"), "interactor_settings", ("box_selection", "boxSelection"), ("disable_auto_switch", "disableAutoSwitch"), ("picking_modes", "pickingModes"), ] self._event_names += [ "click", "hover", "select", ("on_local_image_capture", "onLocalImageCapture"), ("on_remote_image_capture", "onRemoteImageCapture"), "resize", ("reset_camera", "resetCamera"), ("view_state_change", "viewStateChange"), ("before_scene_loaded", "beforeSceneLoaded"), ("after_scene_loaded", "afterSceneLoaded"), ("on_ready", "onReady"), ("box_selection_change", "BoxSelection"), "StartAnimation", "Animation", "EndAnimation", "MouseEnter", "MouseLeave", "StartMouseMove", "MouseMove", "EndMouseMove", "LeftButtonPress", "LeftButtonRelease", "MiddleButtonPress", "MiddleButtonRelease", "RightButtonPress", "RightButtonRelease", "KeyPress", "KeyDown", "KeyUp", "StartMouseWheel", "MouseWheel", "EndMouseWheel", "StartPinch", "Pinch", "EndPinch", "StartPan", "Pan", "EndPan", "StartRotate", "Rotate", "EndRotate", "Button3D", "Move3D", "StartPointerLock", "EndPointerLock", "StartInteraction", "Interaction", "EndInteraction", ] self._server.controller.on_server_ready.add(self.update_geometry) def push_remote_camera_on_end_interaction(self): # vtkCommand.EndInteractionEvent 45 # from vtkmodules.vtkCommonCore import vtkCommand # print("vtkCommand.EndInteractionEvent", vtkCommand.EndInteractionEvent) self.__view.GetInteractor().AddObserver(45, self._push_camera) def update_geometry( self, reset_camera=False, widgets=None, orientation_axis=0, **kwargs ): """ Force update to geometry """ if widgets is None: widgets = self._widgets if not self.server.protocol: return delta_state = self._helper.scene( self.__view, new_state=False, widgets=widgets, orientation_axis=orientation_axis, ) self.server.protocol.publish("trame.vtk.delta", delta_state) full_state = self._helper.scene( self.__view, new_state=True, widgets=widgets, orientation_axis=orientation_axis, ) self.server.state[self.__scene_id] = full_state def export_geometry(self, widgets=None, orientation_axis=0, format="zip", **kwargs): """Export standalone scene for OfflineViewer :param format: Can be either be "zip" or "json". """ encoded_data = None if widgets is None: widgets = self._widgets if not self.server.protocol: return encoded_data = self._helper.export( self.__view, widgets=widgets, orientation_axis=orientation_axis, ) if encoded_data: json_out = json.dumps(encoded_data) if format == "json": return json_out.encode(encoding="UTF-8", errors="strict") zip_buffer = io.BytesIO() with zipfile.ZipFile(zip_buffer, "a") as zfile: zfile.writestr("index.json", json_out, compress_type=ZIP_COMPRESSION) return zip_buffer.getvalue() def update_image(self, reset_camera=False): """ Force update to image """ if self.server.protocol: self._helper.push_image(self.__view, reset_camera) def set_local_rendering(self, local=True, **kwargs): self.server.state[self.__mode_key] = "local" if local else "remote" def set_remote_rendering(self, remote=True, **kwargs): self.server.state[self.__mode_key] = "remote" if remote else "local" def update(self, reset_camera=False, widgets=None, orientation_axis=0, **kwargs): # need to do both to keep things in sync if self.__rendering: self.update_image(reset_camera) self.update_geometry( reset_camera=reset_camera, widgets=widgets, orientation_axis=orientation_axis, ) if reset_camera: self.server.js_call( self.__ref, "setCamera", self._helper.camera(self.__view), ) def _push_camera(self, *args, **kwargs): self.push_camera() def push_camera(self, camera=None, center_of_rotation=None, **kwargs): if camera is None: if hasattr(self.__view, "GetRenderers"): # VTK camera = self.__view.GetRenderers().GetFirstRenderer().GetActiveCamera() elif hasattr(self.__view, "GetActiveCamera"): # ParaView camera = self.__view.GetActiveCamera() if camera is None: return camera_params = dict( position=camera.GetPosition(), focalPoint=camera.GetFocalPoint(), viewUp=camera.GetViewUp(), parallelProjection=camera.GetParallelProjection(), parallelScale=camera.GetParallelScale(), viewAngle=camera.GetViewAngle(), ) if center_of_rotation is not None: camera_params["centerOfRotation"] = center_of_rotation self.server.js_call( self.__ref, "setCamera", camera_params, ) def replace_view(self, new_view, **kwargs): self.server.state[self.__view_key_id] = self._helper.id(new_view) _mode = self.server.state[self.__mode_key] self.__view = new_view self.__wrapped_view = self._helper.view( new_view, name=self.__namespace, mode=_mode, force_replace=True ) self.update() self.resize() def reset_camera(self, **kwargs): self.server.js_call(ref=self.__ref, method="resetCamera") def resize(self, **kwargs): self.server.js_call(ref=self.__ref, method="resize") @property def view(self): """ Get linked vtkRenderWindow instance """ return self.__wrapped_view def capture_image(self, format="image/png", opts={}): self.server.js_call( self.__ref, "captureImage", format, opts, ) def get_widgets(self): return self._widgets def set_widgets(self, value): self._widgets = value self.update_geometry() def release_resources(self): self.__wrapped_view.release_resources() self.__view = None def get_scene_object_id(self, vtk_obj): return reference_id(vtk_obj) class VtkRemoteView(HtmlElement): """ The VtkRemoteView component relies on the server for rendering by sending images to the client by binding your vtkRenderWindow to it. This component gives you control over the image size and quality to reduce latency while interacting. >>> remote_view = vtk.vtkRemoteView( ... # Instance of the view (required) [VTK: vtkRenderWindow | Paraview: viewProxy] ... view=..., ... ref=..., # Identifier for this component ... interactive_quality=60, # [0, 100] 0 for fastest render, 100 for best quality ... interactive_ratio=..., # [0.1, 1] Image size scale factor while interacting ... interactor_events=( # Enable vtk.js interactor events for method binding ... "events", ... ["EndAnimation"], ... ), ... EndAnimation=end, # Bind method to the enabled event ... box_selection=True, # toggle selection box rendering ... box_selection_change=fn, # Bind method to get rect selection ... ) """ _next_id = 0 def __init__(self, view, ref=None, **kwargs): super().__init__("vtk-remote-view", **kwargs) self._is_animating = False self._img_quality = None self._helper = activate_module_for(None, self.server, view) self._helper.has_capabilities("web", "rendering") if ref is None: VtkRemoteView._next_id += 1 ref = f"trame__remote_view_{VtkRemoteView._next_id}" self.__view = view self.__ref = ref self.__view_key_id = f"{ref}Id" self.server.state[self.__view_key_id] = self._helper.id(view) self._attributes["ref"] = f'ref="{ref}"' self._attributes["view_id"] = f':viewId="{self.__view_key_id}"' self._attr_names += [ ("enable_picking", "enablePicking"), ("interactive_quality", "interactiveQuality"), ("interactive_ratio", "interactiveRatio"), ("still_ratio", "stillRatio"), ("still_quality", "stillQuality"), ("interactor_events", "interactorEvents"), ("box_selection", "boxSelection"), "visible", ("picking_modes", "pickingModes"), ] self._event_names += [ ("on_image_capture", "onImageCapture"), ("box_selection_change", "BoxSelection"), "click", "hover", "select", "StartAnimation", "Animation", "EndAnimation", "MouseEnter", "MouseLeave", "StartMouseMove", "MouseMove", "EndMouseMove", "LeftButtonPress", "LeftButtonRelease", "MiddleButtonPress", "MiddleButtonRelease", "RightButtonPress", "RightButtonRelease", "KeyPress", "KeyDown", "KeyUp", "StartMouseWheel", "MouseWheel", "EndMouseWheel", "StartPinch", "Pinch", "EndPinch", "StartPan", "Pan", "EndPan", "StartRotate", "Rotate", "EndRotate", "Button3D", "Move3D", "StartPointerLock", "EndPointerLock", "StartInteraction", "Interaction", "EndInteraction", ] def update(self, **kwargs): """ Force image to be pushed to client """ if self.server.protocol: self._helper.push_image(self.__view) def start_animation(self, fps=30, quality=100, ratio=1): if not self.server.protocol: return if self._is_animating: return self._is_animating = True self._img_quality = self._helper.get_current_image_quality(self.__view) self._helper.start_animation(self.__view, fps, quality, ratio) def stop_animation(self): if not self.server.protocol: return if not self._is_animating: return self._is_animating = False self._helper.set_image_quality(self.__view, **self._img_quality) self._helper.stop_animation(self.__view) self.update() def reset_camera(self, **kwargs): self.server.js_call(ref=self.__ref, method="resetCamera") def replace_view(self, new_view, **kwargs): self.__view = new_view self.server.state[self.__view_key_id] = self._helper.id(new_view) self.update() self.resize() def resize(self, **kwargs): self.server.js_call(ref=self.__ref, method="resize") def capture_image(self, format="image/png", opts={}): self.server.js_call( self.__ref, "captureImage", format, opts, ) def release_resources(self): self.__view = None @property def ref_name(self): return self.__ref class VtkShareDataset(HtmlElement): def __init__(self, children=None, **kwargs): super().__init__("vtk-share-dataset", children, **kwargs) self._attr_names += ["port", "name"] class VtkLocalView(HtmlElement): """ The VtkLocalView component relies on the server for defining the vtkRenderWindow but then only the geometry is exchanged with the client. The server does not need a GPU as no rendering is happening on the server. The vtkRenderWindow is only used to retrieve the scene data and parameters (coloring by, representations, ...). By relying on the same vtkRenderWindow, you can easily switch from a ``VtkRemoteView`` to a ``VtkLocalView`` or vice-versa. This component gives you controls on how you want to map mouse interaction with the camera. The default setting mimic default VTK interactor style so you will rarely have to override to the ``interactor_settings``. The widgets argument allow to properly capture behavior for those assuming they can be handled by the client side. >>> local_view = vtk.VtkLocalView( ... view=..., # Instance of the view (required) ... widgets=[], # List of vtkWidgets in view ... # - VTK: vtkRenderWindow ... # - Paraview: viewProxy ... ref=..., # Identifier for this component ... context_name=..., # Namespace for geometry cache ... interactor_settings=..., # Options for camera controls. See below. ... interactor_events=( # Enable vtk.js interactor events for method binding ... "events", ... ["EndAnimation"], ... ), ... EndAnimation=end, # Bind method to the enabled event ... box_selection=True, # toggle selection box rendering ... box_selection_change=fn, # Bind method to get rect selection ... ) """ _next_id = 0 def __init__(self, view, ref=None, widgets=[], **kwargs): super().__init__("vtk-local-view", **kwargs) self._helper = activate_module_for(None, self.server, view) self._helper.has_capabilities("web") if ref is None: VtkLocalView._next_id += 1 ref = f"trame__local_view_{VtkLocalView._next_id}" self.__scene_id = f"scene_{ref}" self.__view = view self.__ref = ref self._attributes["ref"] = f'ref="{ref}"' self._attributes["view_state"] = f':viewState="{self.__scene_id}"' self._widgets = widgets self._attr_names += [ ("interactor_events", "interactorEvents"), "interactor_settings", ("context_name", "contextName"), ("box_selection", "boxSelection"), ("picking_modes", "pickingModes"), ] self._event_names += [ ("on_image_capture", "onImageCapture"), "resize", ("reset_camera", "resetCamera"), ("view_state_change", "viewStateChange"), ("before_scene_loaded", "beforeSceneLoaded"), ("after_scene_loaded", "afterSceneLoaded"), ("on_ready", "onReady"), ("box_selection_change", "BoxSelection"), "select", "hover", "click", "StartAnimation", "Animation", "EndAnimation", "MouseEnter", "MouseLeave", "StartMouseMove", "MouseMove", "EndMouseMove", "LeftButtonPress", "LeftButtonRelease", "MiddleButtonPress", "MiddleButtonRelease", "RightButtonPress", "RightButtonRelease", "KeyPress", "KeyDown", "KeyUp", "StartMouseWheel", "MouseWheel", "EndMouseWheel", "StartPinch", "Pinch", "EndPinch", "StartPan", "Pan", "EndPan", "StartRotate", "Rotate", "EndRotate", "Button3D", "Move3D", "StartPointerLock", "EndPointerLock", "StartInteraction", "Interaction", "EndInteraction", ] self.update() self._server.controller.on_server_ready.add(self.update) def get_widgets(self): return self._widgets def set_widgets(self, value): self._widgets = value self.update() def update(self, widgets=None, orientation_axis=0, **kwargs): """ Force geometry to be pushed """ if widgets is None: widgets = self._widgets if not self.server.protocol: return delta_state = self._helper.scene( self.__view, new_state=False, widgets=widgets, orientation_axis=orientation_axis, ) self.server.protocol.publish("trame.vtk.delta", delta_state) full_state = self._helper.scene( self.__view, new_state=True, widgets=widgets, orientation_axis=orientation_axis, ) self.server.state[self.__scene_id] = full_state def export(self, widgets=None, orientation_axis=0, format="zip", **kwargs): """Export standalone scene for OfflineViewer :param format: Can be either be "zip" or "json". """ encoded_data = None if widgets is None: widgets = self._widgets if not self.server.protocol: return encoded_data = self._helper.export( self.__view, widgets=widgets, orientation_axis=orientation_axis, ) if encoded_data: json_out = json.dumps(encoded_data) if format == "json": return json_out.encode(encoding="UTF-8", errors="strict") zip_buffer = io.BytesIO() with zipfile.ZipFile(zip_buffer, "a") as zfile: zfile.writestr("index.json", json_out, compress_type=ZIP_COMPRESSION) return zip_buffer.getvalue() def reset_camera(self, **kwargs): """ Move camera to center actors within the frame """ self.server.js_call(ref=self.__ref, method="resetCamera") def replace_view(self, new_view, **kwargs): self.__view = new_view self.server.js_call( self.__ref, "setSynchronizedViewId", self._helper.id(new_view) ) self.update() def resize(self, **kwargs): self.server.js_call(ref=self.__ref, method="resize") def push_camera(self, camera=None, center_of_rotation=None, **kwargs): if camera is None: if hasattr(self.__view, "GetRenderers"): # VTK camera = self.__view.GetRenderers().GetFirstRenderer().GetActiveCamera() elif hasattr(self.__view, "GetActiveCamera"): # ParaView camera = self.__view.GetActiveCamera() if camera is None: return camera_params = dict( position=camera.GetPosition(), focalPoint=camera.GetFocalPoint(), viewUp=camera.GetViewUp(), parallelProjection=camera.GetParallelProjection(), parallelScale=camera.GetParallelScale(), viewAngle=camera.GetViewAngle(), ) if center_of_rotation is not None: camera_params["centerOfRotation"] = center_of_rotation self.server.js_call( self.__ref, "setCamera", camera_params, ) def capture_image(self, format="image/png", opts={}): self.server.js_call( self.__ref, "captureImage", format, opts, ) def release_resources(self): self._server.controller.on_server_ready.discard(self.update) self.__view = None self._widgets = None @property def ref_name(self): return self.__ref def get_scene_object_id(self, vtk_object): return reference_id(vtk_object) class VtkView(HtmlElement): _next_id = 0 def __init__(self, children=None, ref=None, **kwargs): super().__init__("vtk-view", children, **kwargs) if ref is None: VtkView._next_id += 1 ref = f"trame__client_view_{VtkView._next_id}" self._ref = ref self._attributes["ref"] = f'ref="{ref}"' self._attr_names += [ "background", "cube_axes_style", ("interactor_events", "interactorEvents"), "interactor_settings", "picking_modes", "show_cube_axes", ] self._event_names += [ "hover", "click", "select", "resize", "StartAnimation", "Animation", "EndAnimation", "MouseEnter", "MouseLeave", "StartMouseMove", "MouseMove", "EndMouseMove", "LeftButtonPress", "LeftButtonRelease", "MiddleButtonPress", "MiddleButtonRelease", "RightButtonPress", "RightButtonRelease", "KeyPress", "KeyDown", "KeyUp", "StartMouseWheel", "MouseWheel", "EndMouseWheel", "StartPinch", "Pinch", "EndPinch", "StartPan", "Pan", "EndPan", "StartRotate", "Rotate", "EndRotate", "Button3D", "Move3D", "StartPointerLock", "EndPointerLock", "StartInteraction", "Interaction", "EndInteraction", ] def reset_camera(self, **kwargs): """ Move camera to center actors within the frame """ self.server.js_call(ref=self._ref, method="resetCamera") def set_camera(self, camera): """ Update camera (position, focalPoint, viewUp) """ self.server.js_call(self._ref, "setCamera", camera) @property def ref_name(self): return self._ref class VtkWebXRHelper(HtmlElement): """ The VtkWebXRHelper component provides an easy way to add WebXR support to a VtkView or a VtkLocalView. Just add this widget in a VtkView or VtkLocalView widget and start an XR session with `start_xr()`. This widget will fire the `enterXR` and `exitXR` events when an XR session is effectively entered and exited. >>> webxr_helper = vtk.VtkWebXRHelper( ... ref=..., # Identifier for this component ... draw_controllers_ray=True, # Enable XR controllers rays ... ) """ _next_id = 0 class XrSessionTypes(IntEnum): """ WebXR Session Types supported by vtk.js """ HmdVR = 0 MobileAR = 1 LookingGlassVR = 2 HmdAR = 3 def __init__(self, ref=None, **kwargs): super().__init__( "vtk-webXR-helper", **kwargs, ) if ref is None: VtkWebXRHelper._next_id += 1 ref = f"trame__webxr_helper_{VtkWebXRHelper._next_id}" self.__ref = ref self._attributes["ref"] = f'ref="{ref}"' self._attr_names += [("draw_controllers_ray", "drawControllersRay")] self._event_names += [ ("enter_xr", "enterXR"), ("exit_xr", "exitXR"), ] def start_xr(self, session_type: XrSessionTypes): """ Start a WebXR session """ self.server.js_call(self.__ref, "startXR", session_type) def stop_xr(self): """ Stop the current WebXR session """ self.server.js_call(self.__ref, "stopXR")