Refactor shared stats and skeleton parsing

Move `ProcessorStats` from `dlc_processor` into `services.inference.base` so it can be imported as a shared inference type, and update dependent imports/tests accordingly. Add a typed `SkeletonPacket` protocol plus `resolve_packet_skeleton()` to consistently resolve packet-provided skeleton metadata into a `ResolvedSkeleton` only when required fields are present.

Author: C-Achard

dlclivegui/display/skeleton.py

@@ -4,6 +4,7 @@
 
 from dataclasses import dataclass
 from enum import Enum, auto
+from typing import Protocol
 
 import cv2
 import numpy as np
@@ -85,11 +86,6 @@ def resolve_skeleton(
     )
 
 
-# ###################### #
-#  Skeleton I/O          #
-# ###################### #
-
-
 def skeleton_definition_from_metadata(
     *,
     identifier: str,
@@ -126,6 +122,39 @@ def skeleton_definition_from_metadata(
     )
 
 
+# ###################### #
+#  Skeleton I/O          #
+# ###################### #
+
+
+class SkeletonPacket(Protocol):
+    keypoint_names: list[str] | None
+    skeleton_id: str | None
+    skeleton_edges: tuple[tuple[str, str], ...] | None
+
+
+def resolve_packet_skeleton(
+    packet: SkeletonPacket,
+) -> ResolvedSkeleton | None:
+    """Resolve skeleton metadata supplied by a pose packet."""
+    if not packet.keypoint_names:
+        return None
+
+    if not packet.skeleton_id or not packet.skeleton_edges:
+        return None
+
+    definition = skeleton_definition_from_metadata(
+        identifier=packet.skeleton_id,
+        display_name=packet.skeleton_id,
+        edges=packet.skeleton_edges,
+    )
+
+    return resolve_skeleton(
+        definition,
+        packet.keypoint_names,
+    )
+
+
 # ###################### #
 #  Rendering outcomes   #
 # ###################### #

dlclivegui/services/dlc_processor.py

@@ -9,7 +9,6 @@
 import time
 from collections import deque
 from contextlib import contextmanager
-from dataclasses import dataclass
 from typing import Any
 
 import numpy as np
@@ -27,7 +26,7 @@
 from dlclivegui.utils.stats import WorkerTimingStats
 from dlclivegui.utils.utils import format_thread_stack
 
-from .inference.base import PoseBackends, PosePacket, PoseResult, PoseSource, WorkerState
+from .inference.base import PoseBackends, PosePacket, PoseResult, PoseSource, ProcessorStats, WorkerState
 
 logger = logging.getLogger(__name__)
 STOP_WORKER_TIMEOUT = 10.0  # # seconds to wait in STOPPING state before scheduling background reaping
@@ -93,27 +92,6 @@ def validate_pose_array(
     return arr
 
 
-@dataclass
-class ProcessorStats:
-    """Statistics for DLC processor performance."""
-
-    frames_enqueued: int = 0
-    frames_processed: int = 0
-    frames_dropped: int = 0
-    queue_size: int = 0
-    processing_fps: float = 0.0
-    average_latency: float = 0.0
-    last_latency: float = 0.0
-    # Profiling metrics
-    avg_queue_wait: float = 0.0
-    avg_inference_time: float = 0.0
-    avg_signal_emit_time: float = 0.0
-    avg_total_process_time: float = 0.0
-    # Separated timing for GPU vs socket processor
-    avg_gpu_inference_time: float = 0.0  # Pure model inference
-    avg_processor_overhead: float = 0.0  # Socket processor overhead
-
-
 class DLCLiveProcessor(QObject):
     """Background pose estimation using DLCLive with queue-based threading."""
 

dlclivegui/services/inference/base.py

@@ -42,3 +42,24 @@ class PoseResult:
     pose: np.ndarray | None
     timestamp: float
     packet: PosePacket | None = None
+
+
+@dataclass
+class ProcessorStats:
+    """Statistics for DLC processor performance."""
+
+    frames_enqueued: int = 0
+    frames_processed: int = 0
+    frames_dropped: int = 0
+    queue_size: int = 0
+    processing_fps: float = 0.0
+    average_latency: float = 0.0
+    last_latency: float = 0.0
+    # Profiling metrics
+    avg_queue_wait: float = 0.0
+    avg_inference_time: float = 0.0
+    avg_signal_emit_time: float = 0.0
+    avg_total_process_time: float = 0.0
+    # Separated timing for GPU vs socket processor
+    avg_gpu_inference_time: float = 0.0  # Pure model inference
+    avg_processor_overhead: float = 0.0  # Socket processor overhead

dlclivegui/utils/stats.py

@@ -7,7 +7,7 @@
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from dlclivegui.services.dlc_processor import ProcessorStats
+    from dlclivegui.services.inference.base import ProcessorStats
 
 
 @dataclass

tests/services/test_dlc_processor.py

@@ -12,6 +12,8 @@
 from dlclivegui.processors.processor_utils import ProcessorSpec
 from dlclivegui.services.dlc_processor import (
     DLCLiveProcessor,
+)
+from dlclivegui.services.inference.base import (
     ProcessorStats,
     WorkerState,
 )