datalab-platform 0.0.1.dev0__py3-none-any.whl → 1.0.1__py3-none-any.whl

datalab/adapters_plotpy/objects/scalar.py

@@ -0,0 +1,565 @@
+# Copyright (c) DataLab Platform Developers, BSD 3-Clause license, see LICENSE file.
+
+"""
+PlotPy Adapter Scalar Module
+----------------------------
+
+This module contains adapters for scalar results (GeometryResult, TableResult)
+to avoid circular imports with the base and factories modules.
+"""
+
+# pylint: disable=invalid-name  # Allows short reference names like x, y, ...
+
+from __future__ import annotations
+
+from collections.abc import Iterable
+from typing import TYPE_CHECKING, Literal
+
+import guidata.dataset as gds
+import numpy as np
+from guidata.configtools import get_font
+from plotpy.builder import make
+from plotpy.items import (
+    AnnotatedCircle,
+    AnnotatedEllipse,
+    AnnotatedPoint,
+    AnnotatedRectangle,
+    AnnotatedSegment,
+    AnnotatedShape,
+    LabelItem,
+    Marker,
+    PolygonShape,
+)
+from sigima.objects.base import BaseObj
+from sigima.objects.scalar import KindShape
+from sigima.objects.signal import SignalObj
+from sigima.tools import coordinates
+from sigima.tools.signal import pulse
+
+from datalab.adapters_metadata import (
+    GeometryAdapter,
+    TableAdapter,
+    resultadapter_to_html,
+)
+from datalab.adapters_plotpy.base import (
+    config_annotated_shape,
+    items_to_json,
+    json_to_items,
+    set_plot_item_editable,
+)
+from datalab.config import PLOTPY_CONF, Conf, _
+
+if TYPE_CHECKING:
+    from plotpy.styles import ShapeParam
+
+
+class ResultPlotPyAdapter:
+    """Adapter for converting `sigima` table or geometry adapters to PlotPy
+
+    Args:
+        result: Table or geometry adapter
+    """
+
+    def __init__(self, result_adapter: TableAdapter | GeometryAdapter) -> None:
+        self.result_adapter = result_adapter
+
+    def get_other_items(self, obj: BaseObj) -> list:  # pylint: disable=unused-argument
+        """Return other items associated to this result (excluding label item)
+
+        Those items are not serialized to JSON.
+
+        Args:
+            obj: object (signal/image)
+
+        Returns:
+            List of other items
+        """
+        return []
+
+
+class GeometryPlotPyAdapter(ResultPlotPyAdapter):
+    """Adapter for converting `sigima` geometry adapters to PlotPy
+
+    Args:
+        result: Geometry adapter
+
+    Raises:
+        AssertionError: invalid argument
+    """
+
+    def __init__(self, result_adapter: GeometryAdapter) -> None:
+        assert isinstance(result_adapter, GeometryAdapter)
+        super().__init__(result_adapter)
+
+    def iterate_shape_items(
+        self, fmt: str, lbl: bool, prefix: Literal["s", "i"]
+    ) -> Iterable:
+        """Iterate over metadata shape plot items.
+
+        Args:
+            fmt: numeric format (e.g. "%.3f")
+            lbl: if True, show shape labels
+            prefix: "s" for signal, "i" for image
+
+        Yields:
+            Plot item
+        """
+        max_shapes = Conf.view.max_shapes_to_draw.get(200)
+        total_coords = len(self.result_adapter.result.coords)
+
+        # Yield shapes up to the maximum limit
+        for idx, coords in enumerate(self.result_adapter.result.coords):
+            if idx >= max_shapes:
+                break
+            yield self.create_shape_item(coords, fmt, lbl, prefix)
+
+        # If shapes were truncated, create a warning label
+        if total_coords > max_shapes:
+            warning_text = "⚠ " + _("Only %d out of %d shapes are displayed") % (
+                max_shapes,
+                total_coords,
+            )
+            warning_label = make.label(warning_text, "BR", (0, 0), "BR")
+            warning_label.labelparam.font.bold = True
+            warning_label.labelparam.font.size = 10
+            warning_label.labelparam.bgalpha = 0.8
+            warning_label.labelparam.bgcolor = "#ff9800"  # Orange background
+            warning_label.labelparam.textcolor = "#000000"  # Black text
+            warning_label.labelparam.update_item(warning_label)
+            warning_label.set_readonly(True)
+            yield warning_label
+
+    def create_shape_item(
+        self, coords: np.ndarray, fmt: str, lbl: bool, prefix: Literal["s", "i"]
+    ) -> (
+        AnnotatedPoint
+        | Marker
+        | AnnotatedRectangle
+        | AnnotatedCircle
+        | AnnotatedSegment
+        | AnnotatedEllipse
+        | PolygonShape
+        | None
+    ):
+        """Create individual shape item from coordinates
+
+        Args:
+            coords: coordinate array
+            fmt: numeric format (e.g. "%.3f")
+            lbl: if True, show shape labels
+            prefix: "s" for signal, "i" for image
+
+        Returns:
+            Plot item
+        """
+        if self.result_adapter.result.kind == KindShape.POINT:
+            assert len(coords) == 2, "Coordinates must be a 2-element array"
+            x0, y0 = coords
+            item = AnnotatedPoint(x0, y0)
+            sparam: ShapeParam = item.shape.shapeparam
+            sparam.symbol.marker = "Ellipse"
+            sparam.symbol.size = 6
+            sparam.sel_symbol.marker = "Ellipse"
+            sparam.sel_symbol.size = 6
+            aparam = item.annotationparam
+            aparam.title = self.result_adapter.title
+            sparam.update_item(item.shape)
+            aparam.update_item(item)
+        elif self.result_adapter.result.kind == KindShape.MARKER:
+            assert len(coords) == 2, "Coordinates must be a 2-element array"
+            x0, y0 = coords
+            item = self.__make_marker_item(x0, y0, fmt)
+        elif self.result_adapter.result.kind == KindShape.RECTANGLE:
+            assert len(coords) == 4, "Coordinates must be a 4-element array"
+            x0, y0, dx, dy = coords
+            item = make.annotated_rectangle(
+                x0, y0, x0 + dx, y0 + dy, title=self.result_adapter.title
+            )
+        elif self.result_adapter.result.kind == KindShape.CIRCLE:
+            assert len(coords) == 3, "Coordinates must be a 3-element array"
+            xc, yc, r = coords
+            x0, y0, x1, y1 = coordinates.circle_to_diameter(xc, yc, r)
+            item = make.annotated_circle(
+                x0, y0, x1, y1, title=self.result_adapter.title
+            )
+        elif self.result_adapter.result.kind == KindShape.SEGMENT:
+            assert len(coords) == 4, "Coordinates must be a 4-element array"
+            x0, y0, x1, y1 = coords
+            item = make.annotated_segment(
+                x0, y0, x1, y1, title=self.result_adapter.title
+            )
+        elif self.result_adapter.result.kind == KindShape.ELLIPSE:
+            assert len(coords) == 5, "Coordinates must be a 5-element array"
+            xc, yc, a, b, t = coords
+            coords = coordinates.ellipse_to_diameters(xc, yc, a, b, t)
+            x0, y0, x1, y1, x2, y2, x3, y3 = coords
+            item = make.annotated_ellipse(
+                x0, y0, x1, y1, x2, y2, x3, y3, title=self.result_adapter.title
+            )
+        elif self.result_adapter.result.kind == KindShape.POLYGON:
+            assert len(coords) >= 6, "Coordinates must be at least 6-element array"
+            assert len(coords) % 2 == 0, "Coordinates must be even-length array"
+            x, y = coords[::2], coords[1::2]
+            # Filter out NaN coordinates to avoid performance issues
+            # Polygons are padded with NaNs to create regular arrays, but we only
+            # need to draw the valid coordinates
+            valid_mask = ~(np.isnan(x) | np.isnan(y))
+            x, y = x[valid_mask], y[valid_mask]
+            item = make.polygon(x, y, title=self.result_adapter.title, closed=False)
+        else:
+            raise NotImplementedError(
+                f"Unsupported shape kind: {self.result_adapter.result.kind}"
+            )
+
+        if isinstance(item, AnnotatedShape):
+            config_annotated_shape(item, fmt, lbl, "results", f"{prefix}/annotation")
+            # Apply settings for annotated shapes (except AnnotatedPoint)
+            if not isinstance(item, AnnotatedPoint):
+                if prefix == "s":
+                    config_param = Conf.view.sig_shape_param.get()
+                else:
+                    config_param = Conf.view.ima_shape_param.get()
+                shape_param: ShapeParam = item.shape.shapeparam
+                gds.update_dataset(shape_param, config_param)
+                shape_param.update_item(item.shape)
+
+        if isinstance(item, Marker):
+            item.set_style("results", f"{prefix}/marker")
+            # Apply cursor/marker settings from config
+            if prefix == "s":
+                config_param = Conf.view.sig_marker_param.get()
+            else:
+                config_param = Conf.view.ima_marker_param.get()
+            param = item.markerparam
+            gds.update_dataset(param, config_param)
+            param.update_item(item)
+
+        set_plot_item_editable(item, False)
+        return item
+
+    def __make_marker_item(self, x0: float, y0: float, fmt: str) -> Marker:
+        """Make marker item
+
+        Args:
+            x0: x coordinate
+            y0: y coordinate
+            fmt: numeric format (e.g. '%.3f')
+        """
+        if np.isnan(x0):
+            mstyle = "-"
+
+            def label(x, y):  # pylint: disable=unused-argument
+                return (self.result_adapter.title + ": " + fmt) % y
+
+        elif np.isnan(y0):
+            mstyle = "|"
+
+            def label(x, y):  # pylint: disable=unused-argument
+                return (self.result_adapter.title + ": " + fmt) % x
+
+        else:
+            mstyle = "+"
+            txt = self.result_adapter.title + ": (" + fmt + ", " + fmt + ")"
+
+            def label(x, y):
+                return txt % (x, y)
+
+        marker = make.marker(position=(x0, y0), markerstyle=mstyle, label_cb=label)
+        return marker
+
+
+def create_pulse_segment(
+    x0: float, y0: float, x1: float, y1: float, label: str
+) -> AnnotatedSegment:
+    """Create a signal segment item for pulse visualization.
+
+    Args:
+        x0: X-coordinate of the start point
+        y0: Y-coordinate of the start point
+        x1: X-coordinate of the end point
+        y1: Y-coordinate of the end point
+        label: Label for the segment
+
+    Returns:
+        Annotated segment item styled for pulse visualization
+    """
+    item = make.annotated_segment(x0, y0, x1, y1, label, show_computations=False)
+
+    # Configure label appearance similar to Sigima's vistools
+    item.label.labelparam.bgalpha = 0.5
+    item.label.labelparam.anchor = "T"
+    item.label.labelparam.yc = 10
+    item.label.labelparam.update_item(item.label)
+
+    # Configure segment appearance
+    param = item.shape.shapeparam
+    param.line.color = "#33ff00"  # Green color for baselines/plateaus
+    param.line.width = 5
+    param.symbol.facecolor = "#26be00"
+    param.symbol.edgecolor = "#33ff00"
+    param.symbol.marker = "Ellipse"
+    param.symbol.size = 11
+    param.update_item(item.shape)
+
+    # Make non-interactive
+    item.set_movable(False)
+    item.set_resizable(False)
+    item.set_selectable(False)
+
+    return item
+
+
+def create_pulse_crossing_marker(
+    orientation: Literal["h", "v"], position: float, label: str
+) -> Marker:
+    """Create a crossing marker for pulse visualization.
+
+    Args:
+        orientation: 'h' for horizontal, 'v' for vertical cursor
+        position: Position of the cursor along the relevant axis
+        label: Label for the cursor
+
+    Returns:
+        Marker item styled for crossing visualization
+    """
+    if orientation == "h":
+        cursor = make.hcursor(position, label=label)
+    elif orientation == "v":
+        cursor = make.vcursor(position, label=label)
+    else:
+        raise ValueError("Orientation must be 'h' or 'v'")
+
+    # Configure appearance similar to Sigima's vistools
+    cursor.set_movable(False)
+    cursor.set_selectable(False)
+    cursor.markerparam.line.color = "#a7ff33"  # Light green
+    cursor.markerparam.line.width = 3
+    cursor.markerparam.symbol.marker = "NoSymbol"
+    cursor.markerparam.text.textcolor = "#ffffff"
+    cursor.markerparam.text.background_color = "#000000"
+    cursor.markerparam.text.background_alpha = 0.5
+    cursor.markerparam.text.font.bold = True
+    cursor.markerparam.update_item(cursor)
+
+    return cursor
+
+
+def are_values_valid(values: list[float | None]) -> bool:
+    """Check if all values are valid (not None or nan)
+
+    Args:
+        values: list of values
+
+    Returns:
+        True if all values are valid, False otherwise
+    """
+    for v in values:
+        if v is None or (isinstance(v, float) and np.isnan(v)):
+            return False
+    return True
+
+
+class TablePlotPyAdapter(ResultPlotPyAdapter):
+    """Adapter for converting `sigima` table adapters to PlotPy
+
+    Args:
+        result: Table adapter
+
+    Raises:
+        AssertionError: invalid argument
+    """
+
+    def __init__(self, result_adapter: TableAdapter) -> None:
+        assert isinstance(result_adapter, TableAdapter)
+        super().__init__(result_adapter)
+
+    def get_other_items(self, obj: BaseObj) -> list:
+        """Return other items associated to this result (excluding label item)
+
+        Those items are not serialized to JSON.
+
+        Args:
+            obj: object (signal/image)
+
+        Returns:
+            List of other items
+        """
+        items = []
+        if self.result_adapter.result.is_pulse_features():
+            pulse_items = self.create_pulse_visualization_items(obj)
+            items.extend(pulse_items)
+        return items
+
+    def create_pulse_visualization_items(
+        self, obj: SignalObj
+    ) -> list[AnnotatedSegment | Marker]:
+        """Create pulse visualization items from table data.
+
+        Args:
+            obj: Signal object containing the pulse data
+
+        Returns:
+            List of PlotPy items for pulse visualization
+        """
+        items = []
+        df = self.result_adapter.to_dataframe()
+        for _index, row in df.iterrows():
+            # Start baseline
+            xs0, xs1 = row["xstartmin"], row["xstartmax"]
+            ys = pulse.get_range_mean_y(obj.x, obj.y, (xs0, xs1))
+            if are_values_valid([xs0, xs1, ys]):
+                items.append(create_pulse_segment(xs0, ys, xs1, ys, "Start baseline"))
+            # End baseline
+            xe0, xe1 = row["xendmin"], row["xendmax"]
+            ye = pulse.get_range_mean_y(obj.x, obj.y, (xe0, xe1))
+            if are_values_valid([xe0, xe1, ye]):
+                items.append(create_pulse_segment(xe0, ye, xe1, ye, "End baseline"))
+            if "xplateaumin" in row and "xplateaumax" in row:
+                xp0, xp1 = row["xplateaumin"], row["xplateaumax"]
+                yp = pulse.get_range_mean_y(obj.x, obj.y, (xp0, xp1))
+                if are_values_valid([xp0, xp1, yp]):
+                    items.append(create_pulse_segment(xp0, yp, xp1, yp, "Plateau"))
+            for metric in ("x0", "x50", "x100"):
+                if metric in row:
+                    x = row[metric]
+                    metric_str = metric.replace("x", "x|<sub>") + "%</sub>"
+                    if are_values_valid([x]):
+                        items.append(create_pulse_crossing_marker("v", x, metric_str))
+        return items
+
+
+class MergedResultPlotPyAdapter:
+    """Adapter for merging multiple result adapters into a single label.
+
+    This adapter manages a merged label that displays all results for a given object.
+    Instead of creating individual labels for each result (which causes overlapping),
+    it creates a single label with all results concatenated as HTML.
+
+    Args:
+        result_adapters: List of result adapters (GeometryAdapter or TableAdapter)
+        obj: Signal or image object associated with the results
+    """
+
+    def __init__(
+        self,
+        result_adapters: list[GeometryAdapter | TableAdapter],
+        obj: BaseObj,
+    ) -> None:
+        self.result_adapters = result_adapters
+        self.obj = obj
+        self._cached_label: LabelItem | None = None
+
+    def get_cached_label(self) -> LabelItem | None:
+        """Get the cached merged label item, if it exists.
+
+        Returns:
+            Merged label item, or None if not cached
+        """
+        return self._cached_label
+
+    def invalidate_cached_label(self) -> None:
+        """Invalidate the cached merged label item."""
+        self._cached_label = None
+
+    @property
+    def item_json(self) -> str | None:
+        """JSON representation of the merged label item.
+
+        The position is stored in all result adapters so they stay in sync.
+        """
+        if self.result_adapters:
+            return self.result_adapters[0].get_applicative_attr("item_json")
+        return None
+
+    @item_json.setter
+    def item_json(self, value: str | None) -> None:
+        """Set JSON representation of the merged label item.
+
+        The position is stored in all result adapters to keep them synchronized.
+        """
+        for result_adapter in self.result_adapters:
+            result_adapter.set_applicative_attr("item_json", value)
+
+    def create_merged_label(self) -> LabelItem | None:
+        """Create a single merged label from all result adapters.
+
+        Returns:
+            Merged label item, or None if no results
+        """
+        if not self.result_adapters:
+            return None
+
+        # Create the label with merged content
+        merged_html = resultadapter_to_html(self.result_adapters, self.obj)
+        item = make.label(merged_html, "TL", (0, 0), "TL", title="Results")
+        font = get_font(PLOTPY_CONF, "results", "label/font")
+        item.set_style("results", "label")
+        item.labelparam.font.update_param(font)
+        item.labelparam.update_item(item)
+
+        # Make label read-only (user cannot delete it to remove individual results)
+        item.set_readonly(True)
+
+        self._cached_label = item
+        return item
+
+    def get_merged_label(self) -> LabelItem | None:
+        """Get the merged label, creating it if necessary or updating if it exists.
+
+        Returns:
+            Merged label item, or None if no results
+        """
+        if not self.result_adapters:
+            self._cached_label = None
+            return None
+
+        # Try to restore existing label from stored JSON position
+        if self.item_json and self._cached_label is None:
+            stored_item = json_to_items(self.item_json)[0]
+            if isinstance(stored_item, LabelItem):
+                # Update the stored item with current merged content
+                merged_html = resultadapter_to_html(self.result_adapters, self.obj)
+                stored_item.set_text(merged_html)
+                stored_item.set_readonly(True)
+                self._cached_label = stored_item
+                return stored_item
+
+        # Update existing cached label if present
+        if self._cached_label is not None:
+            merged_html = resultadapter_to_html(self.result_adapters, self.obj)
+            self._cached_label.set_text(merged_html)
+            return self._cached_label
+
+        # Create new label
+        return self.create_merged_label()
+
+    def update_obj_metadata_from_item(self, item: LabelItem) -> None:
+        """Update all result adapters' metadata with the label item position.
+
+        Args:
+            item: Merged label item (after user moved it)
+        """
+        if item is not None:
+            self.item_json = items_to_json([item])
+            # Update all result adapters in the object's metadata
+            for result_adapter in self.result_adapters:
+                result_adapter.add_to(self.obj)
+
+    def get_other_items(self) -> list:
+        """Return other items from all result adapters (e.g., geometric shapes).
+
+        Returns:
+            List of all other items from all result adapters
+        """
+        items = []
+        for result_adapter in self.result_adapters:
+            if isinstance(result_adapter, GeometryAdapter):
+                plotpy_adapter = GeometryPlotPyAdapter(result_adapter)
+            elif isinstance(result_adapter, TableAdapter):
+                plotpy_adapter = TablePlotPyAdapter(result_adapter)
+            else:
+                raise NotImplementedError(
+                    f"Unsupported result adapter type: {type(result_adapter)}"
+                )
+            items.extend(plotpy_adapter.get_other_items(self.obj))
+        return items