rocket-welder-sdk 1.1.42__py3-none-any.whl → 1.1.44__py3-none-any.whl

rocket_welder_sdk/varint.py

@@ -0,0 +1,213 @@
+"""
+Core varint and ZigZag encoding/decoding algorithms.
+Matches C# Varint static class from RocketWelder.SDK.Protocols.
+
+Single source of truth for all varint operations in the SDK.
+Compatible with Protocol Buffers varint encoding.
+
+Varint encoding uses 7 bits per byte with MSB as continuation flag.
+ZigZag encoding maps signed integers to unsigned for efficient varint encoding
+of values near zero (both positive and negative).
+"""
+
+from __future__ import annotations
+
+from typing import Tuple
+
+# Maximum bytes needed for a uint32 varint
+MAX_BYTES_UINT32 = 5
+
+
+def zigzag_encode(value: int) -> int:
+    """
+    ZigZag encode a signed integer to unsigned.
+
+    Maps negative numbers to odd positives: 0->0, -1->1, 1->2, -2->3, 2->4, etc.
+    This allows efficient varint encoding of signed values near zero.
+
+    Args:
+        value: Signed integer to encode
+
+    Returns:
+        Unsigned integer (ZigZag encoded)
+    """
+    return (value << 1) ^ (value >> 31)
+
+
+def zigzag_decode(value: int) -> int:
+    """
+    ZigZag decode an unsigned integer to signed.
+
+    Reverses the ZigZag encoding: 0->0, 1->-1, 2->1, 3->-2, 4->2, etc.
+
+    Args:
+        value: Unsigned integer (ZigZag encoded)
+
+    Returns:
+        Signed integer (decoded)
+    """
+    return (value >> 1) ^ -(value & 1)
+
+
+def get_byte_count(value: int) -> int:
+    """
+    Calculate the number of bytes needed to encode a value as varint.
+
+    Args:
+        value: Unsigned integer value
+
+    Returns:
+        Number of bytes needed (1-5)
+    """
+    if value < 0x80:
+        return 1
+    if value < 0x4000:
+        return 2
+    if value < 0x200000:
+        return 3
+    if value < 0x10000000:
+        return 4
+    return 5
+
+
+def write_varint(buffer: bytearray, offset: int, value: int) -> int:
+    """
+    Write a varint to a buffer at the given offset.
+
+    Args:
+        buffer: Destination buffer (must have at least 5 bytes from offset)
+        offset: Starting offset in buffer
+        value: Unsigned value to encode
+
+    Returns:
+        Number of bytes written (1-5)
+    """
+    i = 0
+    while value >= 0x80:
+        buffer[offset + i] = (value & 0x7F) | 0x80
+        value >>= 7
+        i += 1
+    buffer[offset + i] = value
+    return i + 1
+
+
+def write_zigzag(buffer: bytearray, offset: int, value: int) -> int:
+    """
+    Write a ZigZag-encoded signed integer as varint.
+
+    Args:
+        buffer: Destination buffer (must have at least 5 bytes from offset)
+        offset: Starting offset in buffer
+        value: Signed value to encode
+
+    Returns:
+        Number of bytes written (1-5)
+    """
+    return write_varint(buffer, offset, zigzag_encode(value))
+
+
+def read_varint(data: bytes, offset: int) -> Tuple[int, int]:
+    """
+    Read a varint from bytes at the given offset.
+
+    Args:
+        data: Source buffer
+        offset: Starting offset in buffer
+
+    Returns:
+        Tuple of (decoded_value, bytes_read)
+
+    Raises:
+        EOFError: If buffer ends before varint completes
+        ValueError: If varint is malformed (too long)
+    """
+    result = 0
+    shift = 0
+    i = 0
+
+    while True:
+        if offset + i >= len(data):
+            raise EOFError("Unexpected end of varint")
+
+        b = data[offset + i]
+        result |= (b & 0x7F) << shift
+        i += 1
+
+        if (b & 0x80) == 0:
+            break
+
+        shift += 7
+        if shift >= 35:
+            raise ValueError("Varint too long (corrupted data)")
+
+    return result, i
+
+
+def read_zigzag(data: bytes, offset: int) -> Tuple[int, int]:
+    """
+    Read a ZigZag-encoded signed integer.
+
+    Args:
+        data: Source buffer
+        offset: Starting offset in buffer
+
+    Returns:
+        Tuple of (decoded_signed_value, bytes_read)
+    """
+    encoded, bytes_read = read_varint(data, offset)
+    return zigzag_decode(encoded), bytes_read
+
+
+def try_read_varint(data: bytes, offset: int) -> Tuple[bool, int, int]:
+    """
+    Try to read a varint, returning False if not enough data.
+
+    Args:
+        data: Source buffer
+        offset: Starting offset in buffer
+
+    Returns:
+        Tuple of (success, value, bytes_read)
+        If success is False, value and bytes_read are 0
+    """
+    result = 0
+    shift = 0
+
+    for i in range(min(5, len(data) - offset)):
+        b = data[offset + i]
+        result |= (b & 0x7F) << shift
+
+        if (b & 0x80) == 0:
+            return True, result, i + 1
+
+        shift += 7
+
+    return False, 0, 0
+
+
+def encode_varint(value: int) -> bytes:
+    """
+    Convenience function to encode a value to varint bytes.
+
+    Args:
+        value: Unsigned value to encode
+
+    Returns:
+        Bytes containing the varint encoding
+    """
+    buffer = bytearray(MAX_BYTES_UINT32)
+    length = write_varint(buffer, 0, value)
+    return bytes(buffer[:length])
+
+
+def encode_zigzag(value: int) -> bytes:
+    """
+    Convenience function to encode a signed value to ZigZag varint bytes.
+
+    Args:
+        value: Signed value to encode
+
+    Returns:
+        Bytes containing the ZigZag varint encoding
+    """
+    return encode_varint(zigzag_encode(value))

{rocket_welder_sdk-1.1.42.dist-info → rocket_welder_sdk-1.1.44.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rocket-welder-sdk
-Version: 1.1.42
+Version: 1.1.44
 Summary: High-performance video streaming SDK for RocketWelder services using ZeroBuffer IPC
 Home-page: https://github.com/modelingevolution/rocket-welder-sdk
 Author: ModelingEvolution
@@ -32,8 +32,6 @@ Requires-Dist: zerobuffer-ipc>=1.1.17
 Requires-Dist: pydantic>=2.5.0
 Requires-Dist: py-micro-plumberd>=0.1.8
 Requires-Dist: typing-extensions>=4.0.0
-Provides-Extra: nng
-Requires-Dist: pynng>=0.7.2; extra == "nng"
 Provides-Extra: dev
 Requires-Dist: pytest>=7.0; extra == "dev"
 Requires-Dist: pytest-cov>=4.0; extra == "dev"
@@ -42,7 +40,6 @@ Requires-Dist: black>=22.0; extra == "dev"
 Requires-Dist: mypy>=1.0; extra == "dev"
 Requires-Dist: ruff>=0.1.0; extra == "dev"
 Requires-Dist: types-setuptools; extra == "dev"
-Requires-Dist: pynng>=0.7.2; extra == "dev"
 Dynamic: author
 Dynamic: home-page
 Dynamic: requires-python

{rocket_welder_sdk-1.1.42.dist-info → rocket_welder_sdk-1.1.44.dist-info}/RECORD

@@ -1,40 +1,45 @@
-rocket_welder_sdk/__init__.py,sha256=xAy90cHORXwtphmNrvWaNs-YP4MAx3GIOZX_jJoIWxs,3530
+rocket_welder_sdk/__init__.py,sha256=r2zUZzUMZuAPUy5Sat5EWaxeocIjRbdqBDoC1i0C89w,3413
+rocket_welder_sdk/binary_frame_reader.py,sha256=5IyYO7yv07gLpQ9Toyw1zLkTrnqHLjxIeMVRn8B40-E,6270
+rocket_welder_sdk/binary_frame_writer.py,sha256=2-VEJNHw0fL30TXsCBHyhHltvgx7KUahmbamwdSU8sw,6658
 rocket_welder_sdk/bytes_size.py,sha256=Myl29-wyWCIYdbMmgaxXebT8Dz8_Fwcr3fnfaNW81P0,7463
+rocket_welder_sdk/confidence.py,sha256=AlgVM7gNDff3UxcQHhA5sJlTb4VeOMFAraF3W59AeSI,6498
 rocket_welder_sdk/connection_string.py,sha256=NIC6OiOXF-DeBFCWzgMFOWsenrSS45hY81j_HLMSpgo,9974
 rocket_welder_sdk/controllers.py,sha256=uuVyKsNHtpaIzQt8pXCAu7SywoFa7cfww8VSa4LNi_E,33130
+rocket_welder_sdk/delta_frame.py,sha256=51RFWI2odKVak-d3qhv7S8DiXbk4M5f-bbuTT1unXfk,4518
 rocket_welder_sdk/frame_metadata.py,sha256=TMLIY47cIdIlxqk9xj7I3M8FZFmZ3GcVoLZht7prjQM,3929
 rocket_welder_sdk/gst_metadata.py,sha256=6Ov-DekGsKNuAo3UIc-g9fh4_SAPdHYQTnnPqCK3Ks0,16180
-rocket_welder_sdk/keypoints_protocol.py,sha256=NKiSPrevWG4_RrD6jtFxPjwftlaPWe1CqoFVKRMwp4k,21858
+rocket_welder_sdk/keypoints_protocol.py,sha256=kN8ok6Ptpgl4txoQE9DpKhMoI7zr73vTw9CTvCICtgE,36921
 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=dhjjak7F6xTlLZDVl7TPPiX0KpyWb0DQhrTP3GrkuPM,26655
-rocket_welder_sdk/segmentation_result.py,sha256=c2C42C9l8iaRyTWR6ZOL1kLO4jHP_yjeVMf5DaKkY8E,17525
-rocket_welder_sdk/session_id.py,sha256=sRhzQw90shqq_DJVtrsSggcGZ775kz7cRfbI-1LMeSA,7027
+rocket_welder_sdk/rocket_welder_client.py,sha256=VqxP-GkJKCZCjGZq2tVJ7v5yrh-ztXxlF3CT__818vg,23384
+rocket_welder_sdk/segmentation_result.py,sha256=b3xpv6AZyWzm924cd1RfXwCuEwdkPL5TUA0gzT3HpBA,29153
+rocket_welder_sdk/session_id.py,sha256=9GM4T6xmcNxR9gxZnDUlQGviNd3x3hT71q8y85XtH6g,1884
+rocket_welder_sdk/varint.py,sha256=SmffemQCXToRzs3lb7hWQWDY7NmKv4XG_Wb0oeMrb_I,5058
 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
-rocket_welder_sdk/high_level/__init__.py,sha256=jzLaAhmay_k03gaPgv9fdfBjn6b1xbiQoqSC-9Vk1aQ,1378
-rocket_welder_sdk/high_level/client.py,sha256=IFnbO9JI1KhnCljHiwFZbhx4qMW_z8BdTz4uAi0d7TA,8284
-rocket_welder_sdk/high_level/connection_strings.py,sha256=q1uZJQ7mt1RR-E8MJzIwG6vz3Ddruoc3pTCdTvg_pe4,10434
+rocket_welder_sdk/high_level/__init__.py,sha256=OKbI3l0PFo1Cb_v0kbv16Wr05urz8heVGps0dJcTcG4,1507
+rocket_welder_sdk/high_level/client.py,sha256=rVnnrn68PEvc68p5acuOBZfEjobWWed8XcssTRzxNr8,11502
+rocket_welder_sdk/high_level/connection_strings.py,sha256=3EpFafyrmX_QNhs2nPscC63EMg5ikscvthu-0C6xC1c,9861
 rocket_welder_sdk/high_level/data_context.py,sha256=SXJvDpDBFi8Lm4XqSRSHK7YUUHuugXGo4ZRCb6_z5l0,4833
-rocket_welder_sdk/high_level/frame_sink_factory.py,sha256=JeQXp7_OoAQzcAb6XOsBt_Ewen_KbvYQ82YLDyx71tQ,4000
+rocket_welder_sdk/high_level/frame_sink_factory.py,sha256=tLhMV_qnY-ZUtffIzhycM9znNjYguRumba4vphpNx7E,3493
 rocket_welder_sdk/high_level/schema.py,sha256=2Vv0rDwahtGswWB_ceaCdc7JDtmbkx4wE2jQePzeTpU,5367
-rocket_welder_sdk/high_level/transport_protocol.py,sha256=EFF0bgNn9hxRMj67FwU6MVu-UiEFINSGhd2VC8agrgc,7393
-rocket_welder_sdk/transport/__init__.py,sha256=iU5keW_VYHybvGBkDX1AxcCp9IhYvbqPBegu-w9IPX8,787
-rocket_welder_sdk/transport/frame_sink.py,sha256=YcsTcVGHSDWhhtt-PpWyjIZ9nImAHibgKoN_pcdr4VM,3334
-rocket_welder_sdk/transport/frame_source.py,sha256=G1rBAQS1AgOOdtASB0_CYon8g20hUGXpP2exCp5hlhk,2169
-rocket_welder_sdk/transport/nng_transport.py,sha256=o-qgcmHCGnwtdPe-mwwrC-a9H0rgS-VdH2QvU-6kFlI,5838
+rocket_welder_sdk/high_level/transport_protocol.py,sha256=pdl7l1Tu9b7IYb95q6kzDY7-g1fbhSnhdV_gGhLeYzk,2943
+rocket_welder_sdk/transport/__init__.py,sha256=J1iMvmIwC89PO8VZwd-qxKoAoB6BOxEY82aB9wLdB5c,950
+rocket_welder_sdk/transport/frame_sink.py,sha256=--uL0kqxoz10Qya3eKpH6vjbMuSZHP0C9q6uwmbEo38,3332
+rocket_welder_sdk/transport/frame_source.py,sha256=pgWFmqSLoIjGx8vW6Uv5uYCpcC49_iBSVU0oJZHDuPw,2171
 rocket_welder_sdk/transport/stream_transport.py,sha256=FhxFlZT-CTo6aPq6VclM1A_ecqfeHcwR4Ty1vmlt3W0,5886
 rocket_welder_sdk/transport/tcp_transport.py,sha256=Tui6nKgu50C1KV_UDANQKgVK3M52XN7zrkZTcWIpGmY,4745
 rocket_welder_sdk/transport/unix_socket_transport.py,sha256=t44Q2Fj6fMv53Ll561hN35-ZbHPdAQt0t9tUAXoL0rY,12245
+rocket_welder_sdk/transport/websocket_transport.py,sha256=ti-Vr2I90JvgJgRfjqo2PfTYk5B86SYZhNofuxpwdBo,9591
 rocket_welder_sdk/ui/__init__.py,sha256=5-fCkv3vG0VVVZkselrifkMmx8d51EJp3nNT9mt4JK4,886
 rocket_welder_sdk/ui/controls.py,sha256=cRBxdUDWMOYlCxtw1PV11fTlte6qLxSfCGpr9bQiLcE,11162
 rocket_welder_sdk/ui/icons.py,sha256=DcDklZkPmiEzlOD4IR7VTJOtGPCuuh_OM_WN7ScghWE,8934592
 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.42.dist-info/METADATA,sha256=l7nDxLzpcdnP9GgA5iV_shN1hxpw4hcUsrNrjKxe1nA,24847
-rocket_welder_sdk-1.1.42.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-rocket_welder_sdk-1.1.42.dist-info/top_level.txt,sha256=2iZvBjnwVCUW-uDE23-eJld5PZ9-mlPI69QiXM5IrTA,18
-rocket_welder_sdk-1.1.42.dist-info/RECORD,,
+rocket_welder_sdk-1.1.44.dist-info/METADATA,sha256=XszsJzlnvEQCiPGYl-QxGK6TCTEYmWvi0AzvlHOqYKA,24739
+rocket_welder_sdk-1.1.44.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rocket_welder_sdk-1.1.44.dist-info/top_level.txt,sha256=2iZvBjnwVCUW-uDE23-eJld5PZ9-mlPI69QiXM5IrTA,18
+rocket_welder_sdk-1.1.44.dist-info/RECORD,,

rocket_welder_sdk/transport/nng_transport.py

@@ -1,197 +0,0 @@
-"""NNG transport using pynng library.
-
-NNG (nanomsg next generation) provides high-performance, scalable messaging patterns.
-Supported patterns:
-- Pub/Sub: One publisher to many subscribers
-- Push/Pull: Load-balanced distribution to workers
-"""
-
-from typing import Any, Optional, cast
-
-import pynng
-
-from .frame_sink import IFrameSink
-from .frame_source import IFrameSource
-
-
-class NngFrameSink(IFrameSink):
-    """
-    Frame sink that publishes to NNG Pub/Sub or Push/Pull pattern.
-
-    Each frame is sent as a single NNG message (no framing needed - NNG handles message boundaries).
-    """
-
-    def __init__(self, socket: Any, leave_open: bool = False):
-        """
-        Create an NNG frame sink from a socket.
-
-        Args:
-            socket: pynng socket (Publisher or Pusher)
-            leave_open: If True, doesn't close socket on close
-        """
-        self._socket: Any = socket
-        self._leave_open = leave_open
-        self._closed = False
-
-    @classmethod
-    def create_publisher(cls, url: str) -> "NngFrameSink":
-        """
-        Create an NNG Publisher frame sink bound to the specified URL.
-
-        Args:
-            url: NNG URL (e.g., "tcp://127.0.0.1:5555", "ipc:///tmp/mysocket")
-
-        Returns:
-            Frame sink ready to publish messages
-        """
-        socket = pynng.Pub0()
-        socket.listen(url)
-        return cls(socket, leave_open=False)
-
-    @classmethod
-    def create_pusher(cls, url: str, bind_mode: bool = True) -> "NngFrameSink":
-        """
-        Create an NNG Pusher frame sink.
-
-        Args:
-            url: NNG URL (e.g., "tcp://127.0.0.1:5555", "ipc:///tmp/mysocket")
-            bind_mode: If True, listens (bind); if False, dials (connect)
-
-        Returns:
-            Frame sink ready to push messages
-        """
-        socket = pynng.Push0()
-        if bind_mode:
-            socket.listen(url)
-        else:
-            socket.dial(url)
-        return cls(socket, leave_open=False)
-
-    def write_frame(self, frame_data: bytes) -> None:
-        """Write frame to NNG socket (no length prefix - NNG handles message boundaries)."""
-        if self._closed:
-            raise ValueError("Cannot write to closed sink")
-
-        self._socket.send(frame_data)
-
-    async def write_frame_async(self, frame_data: bytes) -> None:
-        """Write frame asynchronously."""
-        if self._closed:
-            raise ValueError("Cannot write to closed sink")
-
-        await self._socket.asend(frame_data)
-
-    def flush(self) -> None:
-        """Flush is a no-op for NNG (data sent immediately)."""
-        pass
-
-    async def flush_async(self) -> None:
-        """Flush asynchronously is a no-op for NNG."""
-        pass
-
-    def close(self) -> None:
-        """Close the NNG sink."""
-        if self._closed:
-            return
-        self._closed = True
-        if not self._leave_open:
-            self._socket.close()
-
-    async def close_async(self) -> None:
-        """Close the NNG sink asynchronously."""
-        self.close()
-
-
-class NngFrameSource(IFrameSource):
-    """
-    Frame source that subscribes to NNG Pub/Sub or Pull pattern.
-
-    Each NNG message is treated as a complete frame (no framing needed - NNG handles message boundaries).
-    """
-
-    def __init__(self, socket: Any, leave_open: bool = False):
-        """
-        Create an NNG frame source from a socket.
-
-        Args:
-            socket: pynng socket (Subscriber or Puller)
-            leave_open: If True, doesn't close socket on close
-        """
-        self._socket: Any = socket
-        self._leave_open = leave_open
-        self._closed = False
-
-    @classmethod
-    def create_subscriber(cls, url: str, topic: bytes = b"") -> "NngFrameSource":
-        """
-        Create an NNG Subscriber frame source connected to the specified URL.
-
-        Args:
-            url: NNG URL (e.g., "tcp://127.0.0.1:5555", "ipc:///tmp/mysocket")
-            topic: Optional topic filter (empty for all messages)
-
-        Returns:
-            Frame source ready to receive messages
-        """
-        socket = pynng.Sub0()
-        socket.subscribe(topic)
-        socket.dial(url)
-        return cls(socket, leave_open=False)
-
-    @classmethod
-    def create_puller(cls, url: str, bind_mode: bool = True) -> "NngFrameSource":
-        """
-        Create an NNG Puller frame source.
-
-        Args:
-            url: NNG URL (e.g., "tcp://127.0.0.1:5555", "ipc:///tmp/mysocket")
-            bind_mode: If True, listens (bind); if False, dials (connect)
-
-        Returns:
-            Frame source ready to pull messages
-        """
-        socket = pynng.Pull0()
-        if bind_mode:
-            socket.listen(url)
-        else:
-            socket.dial(url)
-        return cls(socket, leave_open=False)
-
-    @property
-    def has_more_frames(self) -> bool:
-        """Check if more frames available (NNG blocks waiting for messages)."""
-        return not self._closed
-
-    def read_frame(self) -> Optional[bytes]:
-        """Read frame from NNG socket (blocking)."""
-        if self._closed:
-            return None
-
-        try:
-            return cast("bytes", self._socket.recv())
-        except pynng.Closed:
-            self._closed = True
-            return None
-
-    async def read_frame_async(self) -> Optional[bytes]:
-        """Read frame asynchronously."""
-        if self._closed:
-            return None
-
-        try:
-            return cast("bytes", await self._socket.arecv())
-        except pynng.Closed:
-            self._closed = True
-            return None
-
-    def close(self) -> None:
-        """Close the NNG source."""
-        if self._closed:
-            return
-        self._closed = True
-        if not self._leave_open:
-            self._socket.close()
-
-    async def close_async(self) -> None:
-        """Close the NNG source asynchronously."""
-        self.close()