cardio 2025.10.1__py3-none-any.whl → 2025.12.0__py3-none-any.whl

cardio/ui.py

@@ -1,9 +1,13 @@
 import functools as ft
+import time
 
+import numpy as np
 from trame.ui.vuetify3 import SinglePageWithDrawerLayout
+from trame.widgets import html
 from trame.widgets import vtk as vtk_widgets
 from trame.widgets import vuetify3 as vuetify
 
+from .orientation import EulerAxis, euler_angle_to_rotation_matrix
 from .scene import Scene
 from .volume_property_presets import list_volume_property_presets
 from .window_level import presets
@@ -37,10 +41,46 @@ class UI:
 
         match event["type"]:
             case "KeyPress":
-                if event["key"].isdigit():
-                    key_num = int(event["key"])
+                key = event["key"]
+                current_time = time.time() * 1000
+
+                if key in self.last_keypress_time:
+                    time_since_last = current_time - self.last_keypress_time[key]
+                    if time_since_last < self.keypress_debounce_ms:
+                        return
+
+                self.last_keypress_time[key] = current_time
+
+                if key.isdigit():
+                    key_num = int(key)
                     if key_num in presets.keys():
                         self.server.state.mpr_window_level_preset = key_num
+                elif key == "l":
+                    current = getattr(self.server.state, "mpr_crosshairs_enabled", True)
+                    self.server.state.mpr_crosshairs_enabled = not current
+                elif key == "h":
+                    current = getattr(self.server.state, "help_overlay_visible", False)
+                    self.server.state.help_overlay_visible = not current
+                elif key == "v":
+                    current = getattr(self.server.state, "maximized_view", "")
+                    self.server.state.maximized_view = (
+                        "" if current == "volume" else "volume"
+                    )
+                elif key == "a":
+                    current = getattr(self.server.state, "maximized_view", "")
+                    self.server.state.maximized_view = (
+                        "" if current == "axial" else "axial"
+                    )
+                elif key == "c":
+                    current = getattr(self.server.state, "maximized_view", "")
+                    self.server.state.maximized_view = (
+                        "" if current == "coronal" else "coronal"
+                    )
+                elif key == "s":
+                    current = getattr(self.server.state, "maximized_view", "")
+                    self.server.state.maximized_view = (
+                        "" if current == "sagittal" else "sagittal"
+                    )
 
             case "LeftButtonPress":
                 self.left_dragging = True
@@ -99,32 +139,52 @@ class UI:
                 event["position"]["y"],
             ]
 
-    def _handle_slice_scroll(self, view_name, base_slice_delta):
-        """Handle slice scrolling for a specific view."""
-        match view_name:
-            case "axial":
-                current_slice = getattr(self.server.state, "axial_slice", 0.0)
-                bounds = getattr(self.server.state, "axial_slice_bounds", [0.0, 100.0])
-                slice_attr = "axial_slice"
-            case "sagittal":
-                current_slice = getattr(self.server.state, "sagittal_slice", 0.0)
-                bounds = getattr(
-                    self.server.state, "sagittal_slice_bounds", [0.0, 100.0]
+    def _get_scroll_vector(self, view_name):
+        """Get the current normal vector for a view after rotation."""
+        base_normals = {
+            "axial": np.array([0.0, 0.0, 1.0]),
+            "sagittal": np.array([1.0, 0.0, 0.0]),
+            "coronal": np.array([0.0, 1.0, 0.0]),
+        }
+
+        if view_name not in base_normals:
+            return np.array([0.0, 0.0, 1.0])
+
+        # Build cumulative rotation from visible rotations
+        rotation_sequence = getattr(self.server.state, "mpr_rotation_sequence", [])
+        cumulative_rotation = np.eye(3)
+
+        for i in range(len(rotation_sequence)):
+            is_visible = getattr(self.server.state, f"mpr_rotation_visible_{i}", True)
+            if is_visible:
+                angle = getattr(self.server.state, f"mpr_rotation_angle_{i}", 0)
+                rotation = rotation_sequence[i]
+                rotation_matrix = euler_angle_to_rotation_matrix(
+                    EulerAxis(rotation["axis"]), angle
                 )
-                slice_attr = "sagittal_slice"
-            case "coronal":
-                current_slice = getattr(self.server.state, "coronal_slice", 0.0)
-                bounds = getattr(
-                    self.server.state, "coronal_slice_bounds", [0.0, 100.0]
-                )
-                slice_attr = "coronal_slice"
-            case _:
-                return
+                cumulative_rotation = cumulative_rotation @ rotation_matrix
+
+        return cumulative_rotation @ base_normals[view_name]
 
-        min_bound, max_bound = min(bounds), max(bounds)
-        slice_delta = base_slice_delta if bounds[1] > bounds[0] else -base_slice_delta
-        new_slice = max(min_bound, min(max_bound, current_slice + slice_delta))
-        setattr(self.server.state, slice_attr, new_slice)
+    def _handle_slice_scroll(self, view_name, base_slice_delta):
+        """Handle slice scrolling for a specific view using rotated scroll vectors."""
+        if view_name not in {"axial", "sagittal", "coronal"}:
+            return
+
+        # Get current origin
+        origin = getattr(self.server.state, "mpr_origin", [0.0, 0.0, 0.0])
+
+        # Get scroll vector for this view (considering rotation)
+        scroll_vector = self._get_scroll_vector(view_name)
+
+        # Update origin along the rotated scroll direction
+        new_origin = [
+            origin[0] + base_slice_delta * scroll_vector[0],
+            origin[1] + base_slice_delta * scroll_vector[1],
+            origin[2] + base_slice_delta * scroll_vector[2],
+        ]
+
+        self.server.state.mpr_origin = new_origin
 
     def __init__(self, server, scene: Scene):
         self.server = server
@@ -135,6 +195,10 @@ class UI:
         self.window_sensitivity = 5.0
         self.level_sensitivity = 2.0
         self.slice_sensitivity = 1.0
+        self.last_keypress_time = {}
+        self.keypress_debounce_ms = 100
+        self.server.state.help_overlay_visible = False
+        self.server.state.maximized_view = ""
 
         self.setup()
 
@@ -181,9 +245,9 @@ class UI:
                 )
 
             with layout.content:
-                # Single VR view (default mode)
+                # Single VR view (volume maximized)
                 with vuetify.VContainer(
-                    v_if="!mpr_enabled",
+                    v_if="maximized_view === 'volume'",
                     fluid=True,
                     classes="pa-0 fill-height",
                 ):
@@ -197,9 +261,9 @@ class UI:
                     self.server.controller.view_reset_camera = view.reset_camera
                     self.server.controller.on_server_ready.add(view.update)
 
-                # Quad-view layout (MPR mode) - directly in content like app.py
+                # Quad-view layout (MPR mode - default)
                 with vuetify.VContainer(
-                    v_if="mpr_enabled",
+                    v_if="!maximized_view",
                     fluid=True,
                     classes="pa-0",
                     style="height: calc(100vh - 85px);",
@@ -263,6 +327,10 @@ class UI:
                     self.server.controller.on_server_ready.add(
                         self._update_all_mpr_views
                     )
+                    # Finalize MPR initialization after UI is ready to avoid race condition
+                    self.server.controller.on_server_ready.add(
+                        self.server.controller.finalize_mpr_initialization
+                    )
 
                     # Store individual view update functions
                     self.server.controller.axial_update = axial_view.update
@@ -270,21 +338,131 @@ class UI:
                     self.server.controller.sagittal_update = sagittal_view.update
                     self.server.controller.volume_update = volume_view.update
 
-            with layout.drawer:
-                # MPR Mode Toggle (only show when volumes are present)
-                if self.scene.volumes:
-                    vuetify.VListSubheader("View Mode")
-                    vuetify.VCheckbox(
-                        v_model=("mpr_enabled", False),
-                        label="Multi-Planar Reconstruction (MPR)",
-                        title="Toggle between single 3D view and quad-view with slice planes",
-                        dense=True,
-                        hide_details=True,
+                # Maximized axial view
+                with vuetify.VContainer(
+                    v_if="maximized_view === 'axial'",
+                    fluid=True,
+                    classes="pa-0 fill-height",
+                ):
+                    axial_maximized_view = vtk_widgets.VtkRemoteView(
+                        self.scene.axial_renderWindow,
+                        interactor_events=("event_types", self.handled_events),
+                        **self.event_listeners_for_view("axial"),
+                        interactive_ratio=1,
+                    )
+
+                # Maximized coronal view
+                with vuetify.VContainer(
+                    v_if="maximized_view === 'coronal'",
+                    fluid=True,
+                    classes="pa-0 fill-height",
+                ):
+                    coronal_maximized_view = vtk_widgets.VtkRemoteView(
+                        self.scene.coronal_renderWindow,
+                        interactor_events=("event_types", self.handled_events),
+                        **self.event_listeners_for_view("coronal"),
+                        interactive_ratio=1,
+                    )
+
+                # Maximized sagittal view
+                with vuetify.VContainer(
+                    v_if="maximized_view === 'sagittal'",
+                    fluid=True,
+                    classes="pa-0 fill-height",
+                ):
+                    sagittal_maximized_view = vtk_widgets.VtkRemoteView(
+                        self.scene.sagittal_renderWindow,
+                        interactor_events=("event_types", self.handled_events),
+                        **self.event_listeners_for_view("sagittal"),
+                        interactive_ratio=1,
                     )
 
-                    # Volume selection dropdown (only show when MPR is enabled)
+                with vuetify.VDialog(
+                    v_model=("help_overlay_visible", False),
+                    max_width="700px",
+                    scrim="rgba(0, 0, 0, 0.7)",
+                ):
+                    with vuetify.VCard(
+                        classes="pa-6",
+                        style="background: rgba(33, 33, 33, 0.95);",
+                    ):
+                        vuetify.VCardTitle(
+                            "Keyboard Shortcuts & Controls", classes="text-h5 mb-4"
+                        )
+
+                        with vuetify.VCardText():
+                            html.H3("Keyboard Shortcuts", classes="text-h6 mb-3")
+                            with vuetify.VTable(density="compact", classes="mb-4"):
+                                with html.Thead():
+                                    with html.Tr():
+                                        html.Th("Key")
+                                        html.Th("Action")
+                                with html.Tbody():
+                                    with html.Tr():
+                                        html.Td("h")
+                                        html.Td("Toggle this help window")
+                                    with html.Tr():
+                                        html.Td("v")
+                                        html.Td("Toggle 3D volume view")
+                                    with html.Tr():
+                                        html.Td("a")
+                                        html.Td("Toggle axial view")
+                                    with html.Tr():
+                                        html.Td("c")
+                                        html.Td("Toggle coronal view")
+                                    with html.Tr():
+                                        html.Td("s")
+                                        html.Td("Toggle sagittal view")
+                                    with html.Tr():
+                                        html.Td("l")
+                                        html.Td("Toggle crosshairs")
+
+                            html.H3("Window/Level Presets", classes="text-h6 mb-3")
+                            with vuetify.VTable(density="compact", classes="mb-4"):
+                                with html.Thead():
+                                    with html.Tr():
+                                        html.Th("Key")
+                                        html.Th("Preset")
+                                        html.Th("Window")
+                                        html.Th("Level")
+                                with html.Tbody():
+                                    for key, preset in presets.items():
+                                        with html.Tr():
+                                            html.Td(str(key))
+                                            html.Td(preset.name)
+                                            html.Td(str(preset.window))
+                                            html.Td(str(preset.level))
+
+                            html.H3("Mouse Controls (MPR Mode)", classes="text-h6 mb-3")
+                            with vuetify.VTable(density="compact"):
+                                with html.Thead():
+                                    with html.Tr():
+                                        html.Th("Action")
+                                        html.Th("Effect")
+                                with html.Tbody():
+                                    with html.Tr():
+                                        html.Td("Left Drag ←/→")
+                                        html.Td("Narrow/widen window")
+                                    with html.Tr():
+                                        html.Td("Left Drag ↑/↓")
+                                        html.Td("Increase/decrease level")
+                                    with html.Tr():
+                                        html.Td("Right Drag ↑/↓")
+                                        html.Td("Scroll through slices")
+
+                        with vuetify.VCardActions():
+                            vuetify.VSpacer()
+                            vuetify.VBtn(
+                                "Close",
+                                click="help_overlay_visible = false",
+                                variant="text",
+                            )
+
+            with layout.drawer:
+                # Volume selection dropdown
+                if self.scene.volumes:
                     vuetify.VSelect(
-                        v_if="mpr_enabled",
+                        v_if="!maximized_view || maximized_view === 'volume'",
                         v_model=("active_volume_label", ""),
                         items=("volume_items", []),
                         item_title="text",
@@ -296,11 +474,11 @@ class UI:
 
                     # Window/Level controls for MPR
                     vuetify.VListSubheader(
-                        "Window/Level", v_if="mpr_enabled && active_volume_label"
+                        "Window/Level", v_if="!maximized_view && active_volume_label"
                     )
 
                     vuetify.VSelect(
-                        v_if="mpr_enabled && active_volume_label",
+                        v_if="!maximized_view && active_volume_label",
                         v_model=("mpr_window_level_preset", 7),
                         items=("mpr_presets", []),
                         item_title="text",
@@ -311,7 +489,7 @@ class UI:
                     )
 
                     vuetify.VSlider(
-                        v_if="mpr_enabled && active_volume_label",
+                        v_if="!maximized_view && active_volume_label",
                         v_model="mpr_window",
                         min=1.0,
                         max=2000.0,
@@ -324,7 +502,7 @@ class UI:
                     )
 
                     vuetify.VSlider(
-                        v_if="mpr_enabled && active_volume_label",
+                        v_if="!maximized_view && active_volume_label",
                         v_model="mpr_level",
                         min=-1000.0,
                         max=1000.0,
@@ -336,55 +514,23 @@ class UI:
                         thumb_label=True,
                     )
 
-                    # Slice position controls (only show when MPR is enabled and volume is selected)
-                    vuetify.VListSubheader(
-                        "Slice Positions", v_if="mpr_enabled && active_volume_label"
-                    )
-
-                    vuetify.VSlider(
-                        v_if="mpr_enabled && active_volume_label",
-                        v_model=("axial_slice", 0.0),
-                        min=("axial_slice_bounds[0]", 0.0),
-                        max=("axial_slice_bounds[1]", 100.0),
-                        hint="A (I ↔ S)",
-                        persistent_hint=True,
-                        dense=True,
-                        hide_details=False,
-                        thumb_label=True,
-                    )
-
-                    vuetify.VSlider(
-                        v_if="mpr_enabled && active_volume_label",
-                        v_model=("sagittal_slice", 0.0),
-                        min=("sagittal_slice_bounds[0]", 0.0),
-                        max=("sagittal_slice_bounds[1]", 100.0),
-                        hint="S (R ↔ L)",
-                        persistent_hint=True,
-                        dense=True,
-                        hide_details=False,
-                        thumb_label=True,
-                    )
-
-                    vuetify.VSlider(
-                        v_if="mpr_enabled && active_volume_label",
-                        v_model=("coronal_slice", 0.0),
-                        min=("coronal_slice_bounds[0]", 0.0),
-                        max=("coronal_slice_bounds[1]", 100.0),
-                        hint="C (P ↔ A)",
-                        persistent_hint=True,
+                    vuetify.VCheckbox(
+                        v_if="!maximized_view && active_volume_label",
+                        v_model=("mpr_crosshairs_enabled", True),
+                        label="Show Crosshairs",
                         dense=True,
-                        hide_details=False,
-                        thumb_label=True,
+                        hide_details=True,
+                        classes="mt-2",
                     )
 
                     # MPR Rotation controls
                     vuetify.VListSubheader(
-                        "Rotations", v_if="mpr_enabled && active_volume_label"
+                        "Rotations", v_if="!maximized_view && active_volume_label"
                     )
 
                     # Rotation buttons
                     with vuetify.VRow(
-                        v_if="mpr_enabled && active_volume_label",
+                        v_if="!maximized_view && active_volume_label",
                         no_gutters=True,
                         classes="mb-2",
                     ):
@@ -419,7 +565,7 @@ class UI:
                     # Reset rotations button
                     vuetify.VBtn(
                         "Reset",
-                        v_if="mpr_enabled && active_volume_label && mpr_rotation_sequence && mpr_rotation_sequence.length > 0",
+                        v_if="!maximized_view && active_volume_label && mpr_rotation_sequence && mpr_rotation_sequence.length > 0",
                         click=self.server.controller.reset_rotations,
                         small=True,
                         dense=True,
@@ -433,7 +579,7 @@ class UI:
                     # Individual rotation sliders
                     for i in range(self.scene.max_mpr_rotations):
                         with vuetify.VRow(
-                            v_if=f"mpr_enabled && active_volume_label && mpr_rotation_sequence && mpr_rotation_sequence.length > {i}",
+                            v_if=f"!maximized_view && active_volume_label && mpr_rotation_sequence && mpr_rotation_sequence.length > {i}",
                             no_gutters=True,
                             classes="align-center mb-1",
                         ):
@@ -475,7 +621,7 @@ class UI:
 
                     # Angle units selector
                     with vuetify.VRow(
-                        v_if="mpr_enabled && active_volume_label",
+                        v_if="!maximized_view && active_volume_label",
                         no_gutters=True,
                         classes="align-center mb-2 mt-2",
                     ):
@@ -495,7 +641,7 @@ class UI:
                     # Save rotations button
                     vuetify.VBtn(
                         "Save Rotations",
-                        v_if="mpr_enabled && active_volume_label && mpr_rotation_sequence && mpr_rotation_sequence.length > 0",
+                        v_if="!maximized_view && active_volume_label && mpr_rotation_sequence && mpr_rotation_sequence.length > 0",
                         click=self.server.controller.save_rotation_angles,
                         small=True,
                         dense=True,