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

cardio/transfer_functions.py

@@ -1,272 +0,0 @@
-# System
-import pathlib as pl
-
-import numpy as np
-import pydantic as pc
-
-# Third Party
-import tomlkit as tk
-import vtk
-
-# Internal
-from .types import RGBColor, ScalarComponent
-
-
-def blend_transfer_functions(tfs, scalar_range=(-2000, 2000), num_samples=512):
-    """
-    Blend multiple transfer functions using volume rendering emission-absorption model.
-
-    Based on the volume rendering equation from:
-    - Levoy, M. "Display of Surfaces from Volume Data" IEEE Computer Graphics and Applications, 1988
-    - Kajiya, J.T. & Von Herzen, B.P. "Ray tracing volume densities" ACM SIGGRAPH Computer Graphics, 1984
-    - Engel, K. et al. "Real-time Volume Graphics" A K Peters, 2006, Chapter 2
-
-    The volume rendering integral: I = ∫ C(s) * μ(s) * T(s) ds
-    where C(s) = emission color, μ(s) = opacity, T(s) = transmission
-
-    For discrete transfer functions, this becomes:
-    - Total emission = Σ(color_i * opacity_i)
-    - Total absorption = Σ(opacity_i)
-    - Final color = total_emission / total_absorption
-    """
-    if len(tfs) == 1:
-        return tfs[0]
-
-    sample_points = np.linspace(
-        start=scalar_range[0],
-        stop=scalar_range[1],
-        num=num_samples,
-    )
-
-    # Initialize arrays to store blended values
-    blended_opacity = []
-    blended_color = []
-
-    for scalar_val in sample_points:
-        # Accumulate emission and absorption for volume rendering
-        total_emission = [0.0, 0.0, 0.0]
-        total_absorption = 0.0
-
-        for otf, ctf in tfs:
-            # Get opacity and color for this scalar value
-            layer_opacity = otf.GetValue(scalar_val)
-            layer_color = [0.0, 0.0, 0.0]
-            ctf.GetColor(scalar_val, layer_color)
-
-            # Volume rendering accumulation:
-            # Emission = color * opacity (additive)
-            # Absorption = opacity (multiplicative through transmission)
-            for i in range(3):
-                total_emission[i] += layer_color[i] * layer_opacity
-
-            total_absorption += layer_opacity
-
-        # Clamp values to reasonable ranges
-        total_absorption = min(total_absorption, 1.0)
-        for i in range(3):
-            total_emission[i] = min(total_emission[i], 1.0)
-
-        # For the final color, normalize emission by absorption if absorption > 0
-        if total_absorption > 0.001:  # Avoid division by zero
-            final_color = [total_emission[i] / total_absorption for i in range(3)]
-        else:
-            final_color = [0.0, 0.0, 0.0]
-
-        # Clamp final colors
-        final_color = [min(c, 1.0) for c in final_color]
-
-        blended_opacity.append(total_absorption)
-        blended_color.append(final_color)
-
-    # Create new VTK transfer functions with blended values
-    blended_otf = vtk.vtkPiecewiseFunction()
-    blended_ctf = vtk.vtkColorTransferFunction()
-
-    for i, scalar_val in enumerate(sample_points):
-        blended_otf.AddPoint(scalar_val, blended_opacity[i])
-        blended_ctf.AddRGBPoint(
-            scalar_val, blended_color[i][0], blended_color[i][1], blended_color[i][2]
-        )
-
-    return blended_otf, blended_ctf
-
-
-class PiecewiseFunctionPoint(pc.BaseModel):
-    """A single point in a piecewise function."""
-
-    x: float = pc.Field(description="Scalar value")
-    y: ScalarComponent
-
-
-class ColorTransferFunctionPoint(pc.BaseModel):
-    """A single point in a color transfer function."""
-
-    x: float = pc.Field(description="Scalar value")
-    color: RGBColor
-
-
-class PiecewiseFunctionConfig(pc.BaseModel):
-    """Configuration for a VTK piecewise function (opacity)."""
-
-    points: list[PiecewiseFunctionPoint] = pc.Field(
-        min_length=1, description="Points defining the piecewise function"
-    )
-
-    @property
-    def vtk_function(self) -> vtk.vtkPiecewiseFunction:
-        """Create VTK piecewise function from this configuration."""
-        otf = vtk.vtkPiecewiseFunction()
-        for point in self.points:
-            otf.AddPoint(point.x, point.y)
-        return otf
-
-
-class ColorTransferFunctionConfig(pc.BaseModel):
-    """Configuration for a VTK color transfer function."""
-
-    points: list[ColorTransferFunctionPoint] = pc.Field(
-        min_length=1, description="Points defining the color transfer function"
-    )
-
-    @property
-    def vtk_function(self) -> vtk.vtkColorTransferFunction:
-        """Create VTK color transfer function from this configuration."""
-        ctf = vtk.vtkColorTransferFunction()
-        for point in self.points:
-            ctf.AddRGBPoint(point.x, *point.color)
-        return ctf
-
-
-class TransferFunctionConfig(pc.BaseModel):
-    """Configuration for a single transfer function (legacy format for compatibility)."""
-
-    window: float = pc.Field(gt=0, description="Window width for transfer function")
-    level: float = pc.Field(description="Window level for transfer function")
-    locolor: RGBColor
-    hicolor: RGBColor
-    opacity: ScalarComponent
-
-    @property
-    def vtk_functions(
-        self,
-    ) -> tuple[vtk.vtkPiecewiseFunction, vtk.vtkColorTransferFunction]:
-        """Create VTK transfer functions from this configuration."""
-        # Create opacity transfer function
-        otf = vtk.vtkPiecewiseFunction()
-        otf.AddPoint(self.level - self.window * 0.50, 0.0)
-        otf.AddPoint(self.level + self.window * 0.14, self.opacity)
-        otf.AddPoint(self.level + self.window * 0.50, 0.0)
-
-        # Create color transfer function
-        ctf = vtk.vtkColorTransferFunction()
-        ctf.AddRGBPoint(
-            self.level - self.window / 2,
-            *self.locolor,
-        )
-        ctf.AddRGBPoint(
-            self.level + self.window / 2,
-            *self.hicolor,
-        )
-
-        return otf, ctf
-
-
-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
-
-
-class VolumePropertyConfig(pc.BaseModel):
-    """Configuration for volume rendering properties and transfer functions."""
-
-    name: str = pc.Field(description="Display name of the preset")
-    description: str = pc.Field(description="Description of the preset")
-
-    # Lighting parameters
-    ambient: ScalarComponent
-    diffuse: ScalarComponent
-    specular: ScalarComponent
-
-    # Transfer functions
-    transfer_functions: list[TransferFunctionPairConfig] = pc.Field(
-        min_length=1, description="List of transfer function pairs to blend"
-    )
-
-    @property
-    def vtk_property(self) -> vtk.vtkVolumeProperty:
-        """Create a fully configured VTK volume property from this configuration."""
-        # Get VTK transfer functions from each pair config
-        tfs = [pair.vtk_functions for pair in self.transfer_functions]
-
-        # Blend all transfer functions into a single composite
-        blended_otf, blended_ctf = blend_transfer_functions(tfs)
-
-        # Create and configure the volume property
-        _vtk_property = vtk.vtkVolumeProperty()
-        _vtk_property.SetScalarOpacity(blended_otf)
-        _vtk_property.SetColor(blended_ctf)
-        _vtk_property.ShadeOn()
-        _vtk_property.SetInterpolationTypeToLinear()
-        _vtk_property.SetAmbient(self.ambient)
-        _vtk_property.SetDiffuse(self.diffuse)
-        _vtk_property.SetSpecular(self.specular)
-
-        return _vtk_property
-
-
-def load_preset(preset_name: str) -> VolumePropertyConfig:
-    """Load a specific preset from its individual file."""
-    assets_dir = pl.Path(__file__).parent / "assets"
-    preset_file = assets_dir / f"{preset_name}.toml"
-
-    if not preset_file.exists():
-        available = list(list_available_presets().keys())
-        raise KeyError(
-            f"Transfer function preset '{preset_name}' not found. "
-            f"Available presets: {available}"
-        )
-
-    with preset_file.open("rt", encoding="utf-8") as fp:
-        raw_data = tk.load(fp)
-
-    try:
-        return VolumePropertyConfig.model_validate(raw_data)
-    except pc.ValidationError as e:
-        raise ValueError(f"Invalid preset file '{preset_name}.toml': {e}") from e
-
-
-def list_available_presets() -> dict[str, str]:
-    """
-    List all available transfer function presets.
-
-    Returns:
-        Dictionary mapping preset names to descriptions
-    """
-    assets_dir = pl.Path(__file__).parent / "assets"
-    preset_files = assets_dir.glob("*.toml")
-
-    presets = {}
-    for preset_file in preset_files:
-        preset_name = preset_file.stem
-        try:
-            with preset_file.open("rt", encoding="utf-8") as fp:
-                preset_data = tk.load(fp)
-                presets[preset_name] = preset_data["description"]
-        except (KeyError, OSError):
-            # Skip files that don't have the expected structure
-            continue
-
-    return presets

cardio-2025.8.0.dist-info/RECORD

@@ -1,22 +0,0 @@
-cardio/__init__.py,sha256=EgRG0mJMRZ4hUzkTXtNr6REwgAliqXoRa3FbkMJSTNI,505
-cardio/app.py,sha256=sJ2IbOrV2X-5XTkORUbxuvqvg3-Iaotkq_ofIjVzJCA,1791
-cardio/assets/bone.toml,sha256=vv8uVYSHIoKuHkNCoBOkGe2_qoEbXMvQO6ypm3mMOtA,675
-cardio/assets/vascular_closed.toml,sha256=XtaZS_Zd6NSAtY3ZlUfiog3T86u9Ii0oSutU2wBQy78,1267
-cardio/assets/vascular_open.toml,sha256=1M3sV1IGt3zh_3vviysKEk9quKfjF9xUBcIq3kxVHFM,879
-cardio/assets/xray.toml,sha256=siPem0OZ2OkWH0e5pizftpItJKGJgxKJ_S2K0316ubQ,693
-cardio/logic.py,sha256=Ir4W1pz_hNlQA3QgzIrh701LJY7qBF4O05qkNKfszJw,15126
-cardio/mesh.py,sha256=DUsjzYsTuUDsICqRK3pj49FtXjQ79CJ0HaEYupEE9R4,9392
-cardio/object.py,sha256=6YnEwXL5CUWoG-RAI8ETZG_DWmWi1URjVMiimCxD1zk,6448
-cardio/property_config.py,sha256=XJYcKeRcq8s9W9jqxzVer75r5jBLuvebv780FYdPV8U,1723
-cardio/scene.py,sha256=u9j2ZDobTvO8snTnNk-n3NmO7ZQ2cSC6maRCfkcMQaM,7401
-cardio/screenshot.py,sha256=l8bLgxnU5O0FlnmsyVAzKwM9Y0401IzcdnDP0WqFSTY,640
-cardio/segmentation.py,sha256=enj1V5Rw42dxjqSRUiylEQcdgvj8A-Do9Z638tf7nc8,6706
-cardio/transfer_functions.py,sha256=oOKShXBOBpDt8VTEZyc4IT9mCMxuYkWCKHE9kzGrs3o,9183
-cardio/types.py,sha256=DYDgA5QmYdU3QQrEgZMouEbMEIf40DJCeXo4V7cDXtg,356
-cardio/ui.py,sha256=ifbXHRSDY2Sm-zcriIk4wVySuF0NwHrZu8NCN098gIY,23180
-cardio/utils.py,sha256=zgyJ2PWTAWIaU8SVA9KF_XFB4251QxYAT6w3cqaIerA,3051
-cardio/volume.py,sha256=vCOUWKeN99Whdv0b9IEalFGUpKuNe4DILKWBL02nUUM,1892
-cardio-2025.8.0.dist-info/WHEEL,sha256=4n27za1eEkOnA7dNjN6C5-O2rUiw6iapszm14Uj-Qmk,79
-cardio-2025.8.0.dist-info/entry_points.txt,sha256=xRd1otqKtW9xmidJ4CFRX1V9KWmsBbtxrgMtDColq3w,40
-cardio-2025.8.0.dist-info/METADATA,sha256=CCzucRfA2xbbEqCI89K-52Nl16jEnzgXC8auFNkEWLo,3326
-cardio-2025.8.0.dist-info/RECORD,,