cardio 2025.8.1__py3-none-any.whl → 2025.9.0__py3-none-any.whl

cardio/scene.py

@@ -93,7 +93,7 @@ class Scene(ps.BaseSettings):
     project_name: str = "Cardio"
     current_frame: int = 0
     screenshot_directory: pl.Path = pl.Path("./data/screenshots")
-    screenshot_subdirectory_format: pl.Path = pl.Path("%Y-%m-%d-%H-%M-%S")
+    screenshot_subdirectory_format: str = "%Y-%m-%d-%H-%M-%S"
     rotation_factor: float = 3.0
     background: Background = Field(
         default_factory=Background,
@@ -102,6 +102,36 @@ class Scene(ps.BaseSettings):
     meshes: MeshList = Field(default_factory=list)
     volumes: VolumeList = Field(default_factory=list)
     segmentations: SegmentationList = Field(default_factory=list)
+    mpr_enabled: bool = Field(
+        default=False,
+        description="Enable multi-planar reconstruction (MPR) mode with quad-view layout",
+    )
+    active_volume_label: str = Field(
+        default="",
+        description="Label of the volume to use for multi-planar reconstruction",
+    )
+    axial_slice: float = Field(
+        default=0.5, description="Axial slice position as fraction (0.0 to 1.0)"
+    )
+    sagittal_slice: float = Field(
+        default=0.5, description="Sagittal slice position as fraction (0.0 to 1.0)"
+    )
+    coronal_slice: float = Field(
+        default=0.5, description="Coronal slice position as fraction (0.0 to 1.0)"
+    )
+    mpr_window: float = Field(
+        default=800.0, description="Window width for MPR image display"
+    )
+    mpr_level: float = Field(
+        default=200.0, description="Window level for MPR image display"
+    )
+    mpr_window_level_preset: int = Field(
+        default=7, description="Window/level preset key for MPR views"
+    )
+    mpr_rotation_sequence: list = Field(
+        default_factory=list,
+        description="Dynamic rotation sequence for MPR views - list of rotation steps",
+    )
 
     # Field validators for JSON string inputs
     @field_validator("meshes", mode="before")
@@ -134,6 +164,11 @@ class Scene(ps.BaseSettings):
         default_factory=vtk.vtkRenderWindowInteractor
     )
 
+    # MPR render windows
+    _axial_renderWindow: vtk.vtkRenderWindow = PrivateAttr(default=None)
+    _coronal_renderWindow: vtk.vtkRenderWindow = PrivateAttr(default=None)
+    _sagittal_renderWindow: vtk.vtkRenderWindow = PrivateAttr(default=None)
+
     @property
     def renderer(self) -> vtk.vtkRenderer:
         return self._renderer
@@ -146,8 +181,26 @@ class Scene(ps.BaseSettings):
     def renderWindowInteractor(self) -> vtk.vtkRenderWindowInteractor:
         return self._renderWindowInteractor
 
+    @property
+    def axial_renderWindow(self) -> vtk.vtkRenderWindow:
+        return self._axial_renderWindow
+
+    @property
+    def coronal_renderWindow(self) -> vtk.vtkRenderWindow:
+        return self._coronal_renderWindow
+
+    @property
+    def sagittal_renderWindow(self) -> vtk.vtkRenderWindow:
+        return self._sagittal_renderWindow
+
     @model_validator(mode="after")
     def setup_scene(self):
+        # Validate unique labels
+        self._validate_unique_labels()
+
+        # Validate active volume label
+        self._validate_active_volume_label()
+
         # Configure VTK objects
         self._renderer.SetBackground(
             *self.background.light,
@@ -173,6 +226,44 @@ class Scene(ps.BaseSettings):
 
         return self
 
+    def _validate_unique_labels(self):
+        mesh_labels = [mesh.label for mesh in self.meshes]
+        volume_labels = [volume.label for volume in self.volumes]
+        segmentation_labels = [seg.label for seg in self.segmentations]
+
+        if len(mesh_labels) != len(set(mesh_labels)):
+            duplicates = [
+                label for label in set(mesh_labels) if mesh_labels.count(label) > 1
+            ]
+            raise ValueError(f"Duplicate mesh labels found: {duplicates}")
+
+        if len(volume_labels) != len(set(volume_labels)):
+            duplicates = [
+                label for label in set(volume_labels) if volume_labels.count(label) > 1
+            ]
+            raise ValueError(f"Duplicate volume labels found: {duplicates}")
+
+        if len(segmentation_labels) != len(set(segmentation_labels)):
+            duplicates = [
+                label
+                for label in set(segmentation_labels)
+                if segmentation_labels.count(label) > 1
+            ]
+            raise ValueError(f"Duplicate segmentation labels found: {duplicates}")
+
+    def _validate_active_volume_label(self):
+        """Validate that active_volume_label refers to an existing volume."""
+        if self.active_volume_label and self.volumes:
+            volume_labels = [volume.label for volume in self.volumes]
+            if self.active_volume_label not in volume_labels:
+                raise ValueError(
+                    f"Active volume label '{self.active_volume_label}' not found in available volumes: {volume_labels}"
+                )
+        elif self.active_volume_label and not self.volumes:
+            raise ValueError(
+                "Active volume label specified but no volumes are available"
+            )
+
     @property
     def nframes(self) -> int:
         ns = []
@@ -209,6 +300,46 @@ class Scene(ps.BaseSettings):
         self.show_frame(self.current_frame)
         self.renderer.ResetCamera()
 
+        # Set default camera elevation to -90 degrees
+        camera = self.renderer.GetActiveCamera()
+        camera.Elevation(-90)
+
+    def setup_mpr_render_windows(self):
+        """Initialize MPR render windows when MPR mode is enabled."""
+        if self._axial_renderWindow is None:
+            # Create axial render window
+            self._axial_renderWindow = vtk.vtkRenderWindow()
+            axial_renderer = vtk.vtkRenderer()
+            axial_renderer.SetBackground(0.0, 0.0, 0.0)  # Black background
+            self._axial_renderWindow.AddRenderer(axial_renderer)
+            self._axial_renderWindow.SetOffScreenRendering(True)
+
+            # Create and set interactor for axial
+            axial_interactor = vtk.vtkRenderWindowInteractor()
+            self._axial_renderWindow.SetInteractor(axial_interactor)
+
+            # Create coronal render window
+            self._coronal_renderWindow = vtk.vtkRenderWindow()
+            coronal_renderer = vtk.vtkRenderer()
+            coronal_renderer.SetBackground(0.0, 0.0, 0.0)  # Black background
+            self._coronal_renderWindow.AddRenderer(coronal_renderer)
+            self._coronal_renderWindow.SetOffScreenRendering(True)
+
+            # Create and set interactor for coronal
+            coronal_interactor = vtk.vtkRenderWindowInteractor()
+            self._coronal_renderWindow.SetInteractor(coronal_interactor)
+
+            # Create sagittal render window
+            self._sagittal_renderWindow = vtk.vtkRenderWindow()
+            sagittal_renderer = vtk.vtkRenderer()
+            sagittal_renderer.SetBackground(0.0, 0.0, 0.0)  # Black background
+            self._sagittal_renderWindow.AddRenderer(sagittal_renderer)
+            self._sagittal_renderWindow.SetOffScreenRendering(True)
+
+            # Create and set interactor for sagittal
+            sagittal_interactor = vtk.vtkRenderWindowInteractor()
+            self._sagittal_renderWindow.SetInteractor(sagittal_interactor)
+
     def hide_all_frames(self):
         for a in self.renderer.GetActors():
             a.SetVisibility(False)

cardio/transfer_function_pair.py

@@ -0,0 +1,25 @@
+# Third Party
+import pydantic as pc
+import vtk
+
+# Internal
+from .color_transfer_function import ColorTransferFunctionConfig
+from .piecewise_function import PiecewiseFunctionConfig
+
+
+class TransferFunctionPairConfig(pc.BaseModel):
+    """Configuration for a pair of opacity and color transfer functions."""
+
+    opacity: PiecewiseFunctionConfig = pc.Field(
+        description="Opacity transfer function configuration"
+    )
+    color: ColorTransferFunctionConfig = pc.Field(
+        description="Color transfer function configuration"
+    )
+
+    @property
+    def vtk_functions(
+        self,
+    ) -> tuple[vtk.vtkPiecewiseFunction, vtk.vtkColorTransferFunction]:
+        """Create VTK transfer functions from this pair configuration."""
+        return self.opacity.vtk_function, self.color.vtk_function