rocket-welder-sdk 1.1.33__py3-none-any.whl → 1.1.34a2__py3-none-any.whl

rocket_welder_sdk/__init__.py

@@ -15,6 +15,14 @@ from .gst_metadata import GstCaps, GstMetadata
 from .opencv_controller import OpenCvController
 from .periodic_timer import PeriodicTimer, PeriodicTimerSync
 from .rocket_welder_client import RocketWelderClient
+from .session_id import (
+    get_actions_url,
+    get_keypoints_url,
+    get_nng_urls,
+    get_segmentation_url,
+    get_session_id_from_env,
+    parse_session_id,
+)
 
 # Alias for backward compatibility and README examples
 Client = RocketWelderClient
@@ -58,4 +66,11 @@ __all__ = [
     "PeriodicTimerSync",
     "Protocol",
     "RocketWelderClient",
+    # SessionId utilities for NNG URL generation
+    "get_actions_url",
+    "get_keypoints_url",
+    "get_nng_urls",
+    "get_segmentation_url",
+    "get_session_id_from_env",
+    "parse_session_id",
 ]

rocket_welder_sdk/rocket_welder_client.py

@@ -16,6 +16,8 @@ from .connection_string import ConnectionMode, ConnectionString, Protocol
 from .controllers import DuplexShmController, IController, OneWayShmController
 from .frame_metadata import FrameMetadata  # noqa: TC001 - used at runtime in callbacks
 from .opencv_controller import OpenCvController
+from .session_id import get_nng_urls, get_session_id_from_env
+from .transport.nng_transport import NngFrameSink
 
 if TYPE_CHECKING:
     import numpy.typing as npt
@@ -53,6 +55,9 @@ class RocketWelderClient:
         self._controller: Optional[IController] = None
         self._lock = threading.Lock()
 
+        # NNG publishers for streaming results (auto-created if SessionId env var is set)
+        self._nng_publishers: dict[str, NngFrameSink] = {}
+
         # Preview support
         self._preview_enabled = (
             self._connection.parameters.get("preview", "false").lower() == "true"
@@ -72,6 +77,44 @@ class RocketWelderClient:
         with self._lock:
             return self._controller is not None and self._controller.is_running
 
+    @property
+    def nng_publishers(self) -> dict[str, NngFrameSink]:
+        """Get NNG publishers for streaming results.
+
+        Returns:
+            Dictionary with 'segmentation', 'keypoints', 'actions' keys.
+            Empty if SessionId env var was not set at startup.
+
+        Example:
+            client.nng_publishers["segmentation"].write_frame(seg_data)
+        """
+        return self._nng_publishers
+
+    def _create_nng_publishers(self, session_id: str) -> None:
+        """Create NNG publishers for result streaming.
+
+        Args:
+            session_id: SessionId string (e.g., "ps-{guid}")
+        """
+        try:
+            urls = get_nng_urls(session_id)
+
+            for name, url in urls.items():
+                sink = NngFrameSink.create_publisher(url)
+                self._nng_publishers[name] = sink
+                logger.info("NNG publisher ready: %s at %s", name, url)
+
+            logger.info(
+                "NNG publishers created for SessionId=%s: seg=%s, kp=%s, actions=%s",
+                session_id,
+                urls["segmentation"],
+                urls["keypoints"],
+                urls["actions"],
+            )
+        except Exception as ex:
+            logger.warning("Failed to create NNG publishers: %s", ex)
+            # Don't fail start() - NNG is optional for backwards compatibility
+
     def get_metadata(self) -> Optional[GstMetadata]:
         """
         Get the current GStreamer metadata.
@@ -119,6 +162,11 @@ class RocketWelderClient:
             else:
                 raise ValueError(f"Unsupported protocol: {self._connection.protocol}")
 
+            # Auto-create NNG publishers if SessionId env var is set
+            session_id = get_session_id_from_env()
+            if session_id:
+                self._create_nng_publishers(session_id)
+
             # If preview is enabled, wrap the callback to capture frames
             if self._preview_enabled:
                 self._original_callback = on_frame
@@ -189,6 +237,15 @@ class RocketWelderClient:
                 if self._preview_enabled:
                     self._preview_queue.put(None)  # Sentinel value
 
+                # Clean up NNG publishers
+                for name, sink in self._nng_publishers.items():
+                    try:
+                        sink.close()
+                        logger.debug("Closed NNG publisher: %s", name)
+                    except Exception as ex:
+                        logger.warning("Failed to close NNG publisher %s: %s", name, ex)
+                self._nng_publishers.clear()
+
                 logger.info("RocketWelder client stopped")
 
     def show(self, cancellation_token: Optional[threading.Event] = None) -> None:

rocket_welder_sdk/session_id.py

@@ -0,0 +1,115 @@
+"""SessionId parsing utilities for NNG URL generation.
+
+SessionId format: ps-{guid} (e.g., ps-a1b2c3d4-e5f6-7890-abcd-ef1234567890)
+Prefix "ps" = PipelineSession.
+
+This module provides utilities to:
+1. Parse SessionId from environment variable
+2. Extract the Guid portion
+3. Generate NNG IPC URLs for streaming results
+"""
+
+from __future__ import annotations
+
+import logging
+import os
+import uuid
+
+logger = logging.getLogger(__name__)
+
+SESSION_ID_PREFIX = "ps-"
+SESSION_ID_ENV_VAR = "SessionId"
+
+
+def parse_session_id(session_id: str) -> uuid.UUID:
+    """Parse SessionId (ps-{guid}) to extract Guid.
+
+    Args:
+        session_id: SessionId string (e.g., "ps-a1b2c3d4-...")
+
+    Returns:
+        UUID extracted from SessionId
+
+    Raises:
+        ValueError: If session_id format is invalid
+
+    Examples:
+        >>> parse_session_id("ps-a1b2c3d4-e5f6-7890-abcd-ef1234567890")
+        UUID('a1b2c3d4-e5f6-7890-abcd-ef1234567890')
+        >>> parse_session_id("a1b2c3d4-e5f6-7890-abcd-ef1234567890")  # backwards compat
+        UUID('a1b2c3d4-e5f6-7890-abcd-ef1234567890')
+    """
+    if session_id.startswith(SESSION_ID_PREFIX):
+        return uuid.UUID(session_id[len(SESSION_ID_PREFIX) :])
+    # Fallback: try parsing as raw guid for backwards compatibility
+    return uuid.UUID(session_id)
+
+
+def get_session_id_from_env() -> str | None:
+    """Get SessionId from environment variable.
+
+    Returns:
+        SessionId string or None if not set
+    """
+    return os.environ.get(SESSION_ID_ENV_VAR)
+
+
+def get_nng_urls(session_id: str) -> dict[str, str]:
+    """Generate NNG IPC URLs from SessionId.
+
+    Args:
+        session_id: SessionId string (e.g., "ps-a1b2c3d4-...")
+
+    Returns:
+        Dictionary with 'segmentation', 'keypoints', 'actions' URLs
+
+    Examples:
+        >>> urls = get_nng_urls("ps-a1b2c3d4-e5f6-7890-abcd-ef1234567890")
+        >>> urls["segmentation"]
+        'ipc:///tmp/rw-a1b2c3d4-e5f6-7890-abcd-ef1234567890-seg.sock'
+    """
+    guid = parse_session_id(session_id)
+    return {
+        "segmentation": f"ipc:///tmp/rw-{guid}-seg.sock",
+        "keypoints": f"ipc:///tmp/rw-{guid}-kp.sock",
+        "actions": f"ipc:///tmp/rw-{guid}-actions.sock",
+    }
+
+
+def get_segmentation_url(session_id: str) -> str:
+    """Get NNG URL for segmentation stream.
+
+    Args:
+        session_id: SessionId string (e.g., "ps-a1b2c3d4-...")
+
+    Returns:
+        IPC URL for segmentation stream
+    """
+    guid = parse_session_id(session_id)
+    return f"ipc:///tmp/rw-{guid}-seg.sock"
+
+
+def get_keypoints_url(session_id: str) -> str:
+    """Get NNG URL for keypoints stream.
+
+    Args:
+        session_id: SessionId string (e.g., "ps-a1b2c3d4-...")
+
+    Returns:
+        IPC URL for keypoints stream
+    """
+    guid = parse_session_id(session_id)
+    return f"ipc:///tmp/rw-{guid}-kp.sock"
+
+
+def get_actions_url(session_id: str) -> str:
+    """Get NNG URL for actions stream.
+
+    Args:
+        session_id: SessionId string (e.g., "ps-a1b2c3d4-...")
+
+    Returns:
+        IPC URL for actions stream
+    """
+    guid = parse_session_id(session_id)
+    return f"ipc:///tmp/rw-{guid}-actions.sock"

{rocket_welder_sdk-1.1.33.dist-info → rocket_welder_sdk-1.1.34a2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rocket-welder-sdk
-Version: 1.1.33
+Version: 1.1.34a2
 Summary: High-performance video streaming SDK for RocketWelder services using ZeroBuffer IPC
 Home-page: https://github.com/modelingevolution/rocket-welder-sdk
 Author: ModelingEvolution

{rocket_welder_sdk-1.1.33.dist-info → rocket_welder_sdk-1.1.34a2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-rocket_welder_sdk/__init__.py,sha256=nxaj88kgewtAbyTw7wLGvgjIkY9VOYUuBVtzw1MMW7c,2033
+rocket_welder_sdk/__init__.py,sha256=7jawqof-tt0S0G7CMMOmHRsVLbUf-o8sSREP1VxEeJ4,2400
 rocket_welder_sdk/bytes_size.py,sha256=Myl29-wyWCIYdbMmgaxXebT8Dz8_Fwcr3fnfaNW81P0,7463
 rocket_welder_sdk/connection_string.py,sha256=NIC6OiOXF-DeBFCWzgMFOWsenrSS45hY81j_HLMSpgo,9974
 rocket_welder_sdk/controllers.py,sha256=X7D7OyiuHhGTrmrSNG2Tljsy1_4PGRObytB4dLyvRrg,32600
@@ -8,8 +8,9 @@ rocket_welder_sdk/keypoints_protocol.py,sha256=NKiSPrevWG4_RrD6jtFxPjwftlaPWe1Cq
 rocket_welder_sdk/opencv_controller.py,sha256=MDM6_yFBB9BaMa5jnZRqw7xZZB-WuLr7EPrrfHQ2DK4,9905
 rocket_welder_sdk/periodic_timer.py,sha256=hnObybmrnf3J47QrNKJhYAytLKwria016123NvPRfQ0,9369
 rocket_welder_sdk/py.typed,sha256=0cXFZXmes4Y-vnl4lO3HtyyyWaFNw85B7tJdFeCtHDc,67
-rocket_welder_sdk/rocket_welder_client.py,sha256=mguSN8E4nV8JsNOJvhHFNv7yIz7Ppb25_-9m68d6iD4,15244
+rocket_welder_sdk/rocket_welder_client.py,sha256=2tGL98hLXU_Xl2LzjhdLpQkp9oxzb4D3WmPOcMZMyw4,17474
 rocket_welder_sdk/segmentation_result.py,sha256=q3n8m1sooPcattfXisL9em5gZKbEZ1ueGqeXfOrdaHY,13417
+rocket_welder_sdk/session_id.py,sha256=4RVNxuMzJsJ3Me1LveTe_RSDevEUsOozlamSRMXfXGA,3114
 rocket_welder_sdk/external_controls/__init__.py,sha256=ldOLGhLLS5BQL8m4VKFYV0SvsNNlV2tghlc7rkqadU8,699
 rocket_welder_sdk/external_controls/contracts.py,sha256=3DU6pdpteN50gF2fsS7C2279dGjDa0tZLrLntkBa2LM,2607
 rocket_welder_sdk/external_controls/contracts_old.py,sha256=XWriuXJZu5caTSS0bcTIOZcKnj-IRCm96voA4gqLBfU,2980
@@ -31,7 +32,7 @@ rocket_welder_sdk/ui/icons.py,sha256=DcDklZkPmiEzlOD4IR7VTJOtGPCuuh_OM_WN7ScghWE
 rocket_welder_sdk/ui/ui_events_projection.py,sha256=siiNhjLEBOPfTKw1ZhOPGkwIN5rLDH7V9VCZTNrhEtQ,7836
 rocket_welder_sdk/ui/ui_service.py,sha256=uRdpyJGoCQmtOli_HKSrxLwhZYG-XRuHIYdkmFz1zNk,12026
 rocket_welder_sdk/ui/value_types.py,sha256=f7OA_9zgXEDPoITc8v8SfAR23I4XeFhE3E2_GcAbR6k,1616
-rocket_welder_sdk-1.1.33.dist-info/METADATA,sha256=WK6edMkZs55WIe1AOJjECFfHB2SmTMS31IpschstdpE,24847
-rocket_welder_sdk-1.1.33.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-rocket_welder_sdk-1.1.33.dist-info/top_level.txt,sha256=2iZvBjnwVCUW-uDE23-eJld5PZ9-mlPI69QiXM5IrTA,18
-rocket_welder_sdk-1.1.33.dist-info/RECORD,,
+rocket_welder_sdk-1.1.34a2.dist-info/METADATA,sha256=XlxEOkkIEDYqUlxlPbBv3ZhWNeiqo9cXjnHm8yY_16A,24849
+rocket_welder_sdk-1.1.34a2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rocket_welder_sdk-1.1.34a2.dist-info/top_level.txt,sha256=2iZvBjnwVCUW-uDE23-eJld5PZ9-mlPI69QiXM5IrTA,18
+rocket_welder_sdk-1.1.34a2.dist-info/RECORD,,