seavision-python 0.1.1__py3-none-any.whl

seavision/gui/shared/conversion.py

@@ -0,0 +1,55 @@
+"""Conversion utilities between OpenCV/numpy and Qt image types."""
+
+import numpy as np
+from PySide6.QtGui import QImage
+
+def numpy_bgr_to_qimage(frame: np.ndarray) -> QImage:
+    """
+    Convert a BGR or BGRA numpy array to a QImage.
+
+    Args:
+        frame: numpy array of shape (H, W, 3) for BGR or (H, W, 4) for BGRA.
+            Must be dtype uint8 and contiguous.
+
+    Returns:
+        A QImage that owns its own pixel data (safe to use after the numpy array
+            is freed).
+
+    Raises:
+        ValueError: If the input array is not 3D or does not have 3 channels
+    """
+
+    # Check the shape of the input array - images are 3D arrays
+    if frame.ndim != 3:
+        raise ValueError(
+            f"Expected 3D array (H, W, C), got shape {frame.shape}"
+        )
+    
+    height, width, channels = frame.shape
+
+    # Check there are three colour channels (BGR)
+    if channels != 3:
+        raise ValueError(
+            f"Expected 3 channels (BGR), got {channels}"
+        )
+    
+    # Make the array contiguous in memory (C order) to ensure QImage can use it 
+    # directly
+    frame = np.ascontiguousarray(frame)
+    bytes_per_line = channels * width
+
+    # Create the QImage
+    image = QImage(
+        frame.data,
+        width, 
+        height, 
+        bytes_per_line,
+        QImage.Format.Format_BGR888
+    )
+
+    # Use image.copy() to ensure the QImage owns its own data, so it remains 
+    # valid even if the original numpy array is freed
+
+    # TODO: This approach is performance intensive see note in Phase 1 guide.
+    # A more fficient approach may need to be implemented here.
+    return image.copy()

seavision/gui/shared/session_utils.py

@@ -0,0 +1,37 @@
+"""Utility functions for session management in the GUI."""
+
+from pathlib import Path
+
+
+def resolve_video_paths(
+    sources_in_file: list[str],
+    video_dir: str,
+) -> tuple[dict[str, str], list[str]]:
+    """
+    Resolve CSV source names to local file paths.
+
+    This is extracted from ValidationTab.open_session for testability.
+
+    Args:
+        sources_in_file: Source file values from the CSV.
+        video_dir: Local directory to search for matching videos.
+
+    Returns:
+        Tuple of (resolved_mapping, s3_sources) where resolved_mapping
+        maps source names to local paths and s3_sources lists any S3
+        URIs that couldn't be resolved.
+    """
+    video_dir_path = Path(video_dir)
+    resolved: dict[str, str] = {}
+    s3_sources: list[str] = []
+
+    for source in sources_in_file:
+        if source.startswith("s3://"):
+            s3_sources.append(source)
+            continue
+
+        candidate = video_dir_path / Path(source).name
+        if candidate.exists():
+            resolved[source] = str(candidate)
+
+    return resolved, s3_sources

seavision/gui/validation/button_styles.py

@@ -0,0 +1,144 @@
+"""
+Style sheet strings for validation GUI buttons.
+"""
+
+# --- Main review action buttons ---
+_BUTTON_STYLE_CONFIRM = """
+    QPushButton {
+        background-color: #2d7a4d;
+        color: white;
+        padding: 8px 16px;
+        font-weight: bold;
+        border: 1px solid #1e5a35;
+        border-radius: 4px;
+    }
+    QPushButton:hover {
+        background-color: #359a5d;
+    }
+    QPushButton:pressed {
+        background-color: #1e5a35;
+    }
+    QPushButton:disabled {
+        background-color: #3a3a3a;
+        color: #777;
+        border-color: #333;
+    }
+    QPushButton:focus {
+        border: 2px solid #5dba7d;
+    }
+"""
+
+_BUTTON_STYLE_REJECT = """
+    QPushButton {
+        background-color: #a03030;
+        color: white;
+        padding: 8px 16px;
+        font-weight: bold;
+        border: 1px solid #7a2020;
+        border-radius: 4px;
+    }
+    QPushButton:hover {
+        background-color: #c04040;
+    }
+    QPushButton:pressed {
+        background-color: #7a2020;
+    }
+    QPushButton:disabled {
+        background-color: #3a3a3a;
+        color: #777;
+        border-color: #333;
+    }
+    QPushButton:focus {
+        border: 2px solid #e06060;
+    }
+"""
+
+_BUTTON_STYLE_SKIP = """
+    QPushButton {
+        background-color: #4a4a4a;
+        color: #ddd;
+        padding: 8px 16px;
+        font-weight: bold;
+        border: 1px solid #3a3a3a;
+        border-radius: 4px;
+    }
+    QPushButton:hover {
+        background-color: #5a5a5a;
+    }
+    QPushButton:pressed {
+        background-color: #3a3a3a;
+    }
+    QPushButton:disabled {
+        background-color: #2a2a2a;
+        color: #555;
+        border-color: #222;
+    }
+    QPushButton:focus {
+        border: 2px solid #7a7a7a;
+    }
+"""
+
+_BUTTON_STYLE_ADD = """
+    QPushButton {
+        background-color: #2d6da8;
+        color: white;
+        padding: 8px 16px;
+        border: 1px solid #1e4a7a;
+        border-radius: 4px;
+    }
+    QPushButton:hover {
+        background-color: #3d8dc8;
+    }
+    QPushButton:pressed {
+        background-color: #1e4a7a;
+    }
+    QPushButton:disabled {
+        background-color: #3a3a3a;
+        color: #777;
+        border-color: #333;
+    }
+    QPushButton:checked {
+        background-color: #1a4a7a;
+        border: 2px solid #88bbee;
+    }
+    QPushButton:focus {
+        border: 2px solid #6daddd;
+    }
+"""
+
+
+class ReviewButtonStyles:
+    """Predefined styles for the main review action buttons."""
+
+    def __init__(self):
+        self.confirm = _BUTTON_STYLE_CONFIRM
+        self.reject = _BUTTON_STYLE_REJECT
+        self.skip = _BUTTON_STYLE_SKIP
+        self.add = _BUTTON_STYLE_ADD
+
+
+# --- Transport bar buttons ---
+_BUTTON_STYLE_TRANSPORT = """
+    QPushButton {
+        padding: 4px 8px;
+        border: 1px solid #555;
+        border-radius: 3px;
+        background-color: #3a3a3a;
+        color: #ddd;
+        font-weight: bold;
+    }
+    QPushButton:hover { background-color: #4a4a4a; }
+    QPushButton:pressed { background-color: #2a2a2a; }
+    QPushButton:disabled {
+        background-color: #2a2a2a;
+        color: #555;
+        border-color: #333;
+    }
+"""
+
+
+class TransportButtonStyles:
+    """Predefined style for transport bar buttons."""
+
+    def __init__(self):
+        self.transport = _BUTTON_STYLE_TRANSPORT

seavision/gui/validation/detection_detail.py

@@ -0,0 +1,164 @@
+"""
+Detection detail panel - shows properties of the selected detection.
+Will eventually be extended to allow editing of bounding box position.
+"""
+
+from PySide6.QtWidgets import QFormLayout, QLabel, QWidget
+
+from seavision.engine.detectors.base import Detection
+from seavision.gui.validation.validation_model import (
+    ValidatedDetection,
+    ValidationStatus
+)
+
+class DetectionDetailPanel(QWidget):
+    """
+    Read-only panel displaying all properties of a selected detection.
+    
+    Uses QFormLayout for a clean label-value grid. Updated whenever the table
+    selection changes. Shows placeholder dashes when no detection is selected.
+    """
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._build_layout()
+        self.clear()
+
+    def _build_layout(self) -> None:
+        """Create the label-value pairs."""
+        layout = QFormLayout(self)
+
+        self._status_value = QLabel()
+        self._status_value.setStyleSheet(
+            "font-size: 14px; font-weight: bold; padding: 4px 0;"
+        )
+        layout.addRow("Status:", self._status_value)
+        
+        self._frame_value = QLabel()
+        layout.addRow("Frame:", self._frame_value)
+
+        self._time_value = QLabel()
+        layout.addRow("Timestamp:", self._time_value)
+
+        self._confidence_value = QLabel()
+        layout.addRow("Confidence:", self._confidence_value)
+
+        self._label_value = QLabel()
+        layout.addRow("Label:", self._label_value)
+
+        self._track_value = QLabel()
+        layout.addRow("Track ID:", self._track_value)
+
+        self._position_value = QLabel()
+        layout.addRow("Centre coordinates (xc, yc):", self._position_value)
+
+        self._bbox_coords_value = QLabel()
+        layout.addRow("Bounding box (x1, y1, x2, y2):", self._bbox_coords_value)
+
+        self._size_value = QLabel()
+        layout.addRow("Size:", self._size_value)
+
+        self._area_value = QLabel()
+        layout.addRow("Area:", self._area_value)
+
+    def set_detection(
+        self, vd: ValidatedDetection, fps: float | None = None
+    ) -> None:
+        """
+        Update all fields to show the given detection's properties.
+
+        Args:
+            vd: A ValidatedDetection object.
+            fps: Video frame rate for timestamp calculation.
+        """
+        detection = vd.detection
+        
+        self._frame_value.setText(str(detection.frame_number))
+        
+        if fps is not None and fps > 0:
+            timestamp = detection.frame_number / fps
+            mins = int(timestamp // 60)
+            secs = timestamp % 60
+            self._time_value.setText(f"{mins}:{secs:04.1f}")
+        else:
+            self._time_value.setText("—")
+
+        if detection.confidence is not None:
+            self._confidence_value.setText(f"{detection.confidence:.3f}")
+        else:
+            self._confidence_value.setText("—")
+
+        self._label_value.setText(detection.label if detection.label else "—")
+
+        if detection.track_id is not None:
+            self._track_value.setText(str(detection.track_id))
+        else:
+            self._track_value.setText("—")
+
+        # Use corrected geometry if available
+        if vd.corrected_geometry is not None:
+            geom = vd.corrected_geometry
+            xc = geom["xc"]
+            yc = geom["yc"]
+            w = geom["width"]
+            h = geom["height"]
+        else:
+            xc = detection.xc
+            yc = detection.yc
+            w = detection.width
+            h = detection.height
+        
+        # Position: centre coordinates
+        self._position_value.setText(
+            f"({xc:.1f}, {yc:.1f})"
+        )
+
+        # Bounding box coordinates
+        x1 = xc - w / 2
+        y1 = yc - h / 2
+        x2 = xc + w / 2
+        y2 = yc + h / 2
+        self._bbox_coords_value.setText(
+            f"({x1:.1f}, {y1:.1f}, {x2:.1f}, {y2:.1f})"
+        )
+
+        # Size: width x height
+        self._size_value.setText(
+            f"{w:.1f} x {h:.1f}"
+        )
+
+        # Area: width * height
+        area = w * h
+        self._area_value.setText(f"{area:.0f} px²")
+
+        # Status — prominent display with colour coding
+        status_config = {
+            ValidationStatus.PENDING:   ("Pending",   "#cc8800"),
+            ValidationStatus.CONFIRMED: ("Confirmed", "#2d8a4e"),
+            ValidationStatus.REJECTED:  ("Rejected",  "#c0392b"),
+            ValidationStatus.SKIPPED:   ("Skipped",   "#888888"),
+            ValidationStatus.CORRECTED: ("Corrected", "#2d6da8"),
+        }
+        label, colour = status_config.get(vd.status, ("Unknown", "#000000"))
+        display_text = label
+        if vd.is_manual:
+            display_text += " (manual)"
+
+        self._status_value.setText(display_text)
+        self._status_value.setStyleSheet(
+            f"font-size: 14px; font-weight: bold; padding: 4px 0; "
+            f"color: {colour};"
+        )
+
+    def clear(self) -> None:
+        """Reset all fields to a placeholder state."""
+        self._frame_value.setText("—")
+        self._time_value.setText("—")
+        self._confidence_value.setText("—")
+        self._label_value.setText("—")
+        self._status_value.setText("—")
+        self._track_value.setText("—")
+        self._position_value.setText("—")
+        self._bbox_coords_value.setText("—")
+        self._size_value.setText("—")
+        self._area_value.setText("—")