creatumlibre.ui.mode.ui_input_mode

 1from enum import Enum
 2
 3from PyQt6.QtCore import QEvent, Qt
 4
 5
 6class UiMode:
 7    """
 8    Handles UI interaction modes like selecting regions, point clouds, etc.
 9    """
10
11    def __init__(self, name: str, init_key=None, key_action=None):
12        self.name = name
13        self.mode = InputMode.IDLE  # Use Enum for mode state
14        self.init_key = init_key  # Store required key for mode activation
15        self.key_action = key_action  # Store required key for action execution
16
17    def set_mode(self, mode):
18        """Set the UI mode."""
19        self.mode = mode
20        print(f"Active mode: {self.mode}")
21
22    def get_mode(self):
23        """Return the current UI mode."""
24        return self.mode
25
26    def key_interpretation(self):
27        """Interpret key presses based on the active mode."""
28        if self.mode == InputMode.IDLE:
29            return None
30        if self.mode == InputMode.SELECT_REGION:
31            return (
32                self.init_key or "Control"
33            )  #  Default to Control unless explicitly set
34        if self.mode == InputMode.SELECT_POINT_CLOUD:
35            return self.init_key or "Shift"  #  Default to Shift for point clouds
36        return None  #  Future modes can extend this logic
37
38    def reset_mode(self):
39        """Reset the mode to IDLE."""
40        self.mode = InputMode.IDLE
41
42    def key_action_handler(self, event):
43        """Handle key actions based on the current mode."""
44        if event.key() == Qt.Key_Escape:
45            self.reset_mode()
46
47        elif self.mode in (InputMode.SELECT_REGION, InputMode.MOVE_OBJECTS):
48            if event.mouseButton() == Qt.MouseButton.LeftButton:
49                if event.type() == QEvent.Type.MouseButtonPress:
50                    self.set_mode(MOUSE_ACTION.START)
51                elif event.type() == QEvent.Type.MouseMove:
52                    self.set_mode(MOUSE_ACTION.DRAG)
53                elif event.type() == QEvent.Type.MouseButtonRelease:
54                    self.set_mode(MOUSE_ACTION.STOP)
55        elif self.mode == InputMode.SELECT_POINT_CLOUD and event.key() == self.init_key:
56            self.set_mode(InputMode.IDLE)
57        elif self.key_action and event.key() == self.key_action:
58            # Execute the action associated with the key
59            print(f"Executing action for key: {self.key_action}")
60            # Here you would call the actual function that handles the action
61
62
63class InputMode(Enum):
64    IDLE = 0
65    SELECT_REGION = 1
66    SELECT_POINT_CLOUD = 2
67    MOVE_OBJECTS = 3
68
69
70class MOUSE_ACTION(Enum):
71    IDLE = 0  # no mode
72    START = 1  # mouse down
73    DRAG = 2  # mouse move
74    STOP = 3  # mouse up
75
76
77class TransformMode(Enum):
78    NONE = 0
79    SCALE = 1
80    ROTATE = 2
81    PERSPECTIVE = 3
82    TRANSLATE = 4