wip interaction keybindings

Author: Milotrince

genesis/ext/pyrender/interaction/__init__.py

@@ -1,9 +1,12 @@
-from .plugins.mesh_selector import MeshPointSelectorPlugin
-from .plugins.mouse_interaction import MouseSpringViewerPlugin
-from .viewer_plugin_base import (
+from .base_interaction import (
     EVENT_HANDLE_STATE,
     EVENT_HANDLED,
     VIEWER_PLUGIN_MAP,
-    ViewerPluginBase,
+    BaseViewerInteraction,
     register_viewer_plugin,
 )
+from .plugins.mesh_selector import MeshPointSelectorPlugin
+from .plugins.mouse_interaction import MouseSpringViewerPlugin
+from .plugins.viewer_controls import ViewerControls
+from .ray import Plane, Ray, RayHit
+from .vec3 import Color, Pose, Quat, Vec3

…render/interaction/viewer_plugin_base.py …pyrender/interaction/base_interaction.pygenesis/ext/pyrender/interaction/viewer_plugin_base.py renamed to genesis/ext/pyrender/interaction/base_interaction.py genesis/ext/pyrender/interaction/viewer_plugin_base.py renamed to genesis/ext/pyrender/interaction/base_interaction.py

@@ -15,7 +15,7 @@
 EVENT_HANDLED: Literal[True] = True
 
 # Global map from options class to viewer plugin class
-VIEWER_PLUGIN_MAP: dict[Type["ViewerPluginOptions"], Type["ViewerPluginBase"]] = {}
+VIEWER_PLUGIN_MAP: dict[Type["ViewerPluginOptions"], Type["BaseViewerInteraction"]] = {}
 
 
 def register_viewer_plugin(options_cls: Type["ViewerPluginOptions"]):
@@ -38,25 +38,27 @@ def register_viewer_plugin(options_cls: Type["ViewerPluginOptions"]):
     class ViewerInteraction(ViewerInteractionBase):
         ...
     """
-    def _impl(plugin_cls: Type["ViewerPluginBase"]):
+    def _impl(plugin_cls: Type["BaseViewerInteraction"]):
         VIEWER_PLUGIN_MAP[options_cls] = plugin_cls
         return plugin_cls
     return _impl
 
 # Note: Viewer window is based on pyglet.window.Window, mouse events are defined in pyglet.window.BaseWindow
 
-class ViewerPluginBase():
+class BaseViewerInteraction():
     """
     Base class for handling pyglet.window.Window events.
     """
 
     def __init__(
         self,
+        viewer,
         options: "ViewerPluginOptions",
         camera: "Node",
         scene: "Scene",
         viewport_size: tuple[int, int],
     ):
+        self.viewer = viewer
         self.options: "ViewerPluginOptions" = options
         self.camera: 'Node' = camera
         self.scene: 'Scene' = scene

genesis/ext/pyrender/interaction/keybindings.py

@@ -0,0 +1,33 @@
+from pyglet.window.key import symbol_string
+
+
+class Keybindings:
+    def __init__(self, map: dict[str, int] = {}, **kwargs: dict[str, int]):
+        self._map: dict[str, int] = {**map, **kwargs}
+    
+    def __getattr__(self, name: str) -> int:
+        if name in self._map:
+            return self._map[name]
+        raise AttributeError(f"'Keybindings' object has no attribute '{name}'")
+
+    def as_instruction_texts(self, padding, exclude: tuple[str]) -> list[str]:
+        width = 4 + padding
+        return [
+            f"{'[' + symbol_string(self._map[action]).lower():>{width}}]: " +
+            action.replace('_', ' ') for action in self._map.keys() if action not in exclude
+        ]
+    
+    def apply_override_mapping(self, override_mapping: dict[str, int]) -> None:
+        current_keys = self._map.keys()
+        for action, key in override_mapping.items():
+            if action not in self._map:
+                raise KeyError(f"Action '{action}' not found. Available actions: {list(self._map.keys())}")
+            if key in current_keys:
+                raise ValueError(f"Key '{symbol_string(key)}' is already assigned to another action.")
+            self._map[action] = key
+    
+    def extend_mapping(self, additional_mapping: dict[str, int]) -> None:
+        for action, key in additional_mapping.items():
+            if action in self._map:
+                raise KeyError(f"Action '{action}' already exists in keyboard mapping.")
+            self._map[action] = key

genesis/ext/pyrender/interaction/mouse_spring.py

@@ -0,0 +1,9 @@
+from .mesh_selector import MeshPointSelectorPlugin
+from .mouse_interaction import MouseSpringViewerPlugin
+from .viewer_controls import ViewerControls
+
+__all__ = [
+    "ViewerControls",
+    "MeshPointSelectorPlugin",
+    "MouseSpringViewerPlugin",
+]