rocket-welder-sdk 0.0.1__py3-none-any.whl → 1.0.0__py3-none-any.whl

rocket_welder_sdk/__init__.py

@@ -0,0 +1,8 @@
+"""
+Rocket Welder SDK - Python client library for RocketWelder video streaming services.
+"""
+
+from .client import Client
+
+__version__ = "1.0.0"
+__all__ = ["Client"]

rocket_welder_sdk/client.py

@@ -0,0 +1,183 @@
+"""
+RocketWelder client implementation.
+"""
+
+import os
+import sys
+import time
+import threading
+from typing import Optional, Callable, Dict, Any, Iterator
+import numpy as np
+import cv2
+
+
+class Client:
+    """Client for RocketWelder video streaming services."""
+    
+    def __init__(self, connection_string: str):
+        """
+        Initialize client with connection string.
+        
+        Args:
+            connection_string: Connection string (e.g., "shm://buffer_name")
+        """
+        self.connection_string = connection_string
+        self._callback: Optional[Callable[[np.ndarray], None]] = None
+        self._callback_with_metadata: Optional[Callable[[np.ndarray, Dict[str, Any]], None]] = None
+        self._running = False
+        self._thread: Optional[threading.Thread] = None
+        
+        # TODO: Parse connection string
+        # TODO: Initialize based on protocol (shm://, mjpeg+http://, etc.)
+    
+    @classmethod
+    def from_args(cls, argv: list) -> 'Client':
+        """
+        Create client from command line arguments and environment variables.
+        Environment variable CONNECTION_STRING is checked first, then overridden by args.
+        
+        Args:
+            argv: Command line arguments (typically sys.argv)
+        
+        Returns:
+            Client instance
+        """
+        # Check environment variable first
+        connection_string = os.environ.get('CONNECTION_STRING')
+        
+        # Override with command line args if present
+        if argv:
+            for arg in argv[1:]:  # Skip program name
+                if (arg.startswith('shm://') or 
+                    arg.startswith('mjpeg+http://') or 
+                    arg.startswith('mjpeg+tcp://')):
+                    connection_string = arg
+                    break
+        
+        return cls(connection_string or 'shm://default')
+    
+    @classmethod
+    def from_env(cls) -> 'Client':
+        """
+        Create client from environment variable CONNECTION_STRING.
+        
+        Returns:
+            Client instance
+        """
+        connection_string = os.environ.get('CONNECTION_STRING', 'shm://default')
+        return cls(connection_string)
+    
+    @classmethod
+    def from_connection_string(cls, connection_string: str) -> 'Client':
+        """
+        Create client from a specific connection string.
+        
+        Args:
+            connection_string: Connection string
+        
+        Returns:
+            Client instance
+        """
+        return cls(connection_string)
+    
+    def on_frame(self, callback: Callable) -> Callable:
+        """
+        Decorator/method to set frame processing callback.
+        
+        The callback can have one of these signatures:
+        - callback(frame: np.ndarray) -> None
+        - callback(frame: np.ndarray, metadata: dict) -> None
+        
+        Args:
+            callback: Function to process frames
+        
+        Returns:
+            The callback function (for decorator usage)
+        """
+        import inspect
+        sig = inspect.signature(callback)
+        param_count = len(sig.parameters)
+        
+        if param_count == 1:
+            self._callback = callback
+        elif param_count == 2:
+            self._callback_with_metadata = callback
+        else:
+            raise ValueError(f"Callback must have 1 or 2 parameters, got {param_count}")
+        
+        return callback
+    
+    def start(self):
+        """Start frame processing."""
+        if self._running:
+            return
+        
+        if not self._callback and not self._callback_with_metadata:
+            raise RuntimeError("Frame callback must be set before starting")
+        
+        self._running = True
+        self._thread = threading.Thread(target=self._process_frames)
+        self._thread.daemon = True
+        self._thread.start()
+    
+    def stop(self):
+        """Stop frame processing."""
+        if not self._running:
+            return
+        
+        self._running = False
+        if self._thread:
+            self._thread.join(timeout=5.0)
+            self._thread = None
+    
+    def is_running(self) -> bool:
+        """Check if client is running."""
+        return self._running
+    
+    def frames(self) -> Iterator[np.ndarray]:
+        """
+        Iterator interface for frame processing.
+        
+        Yields:
+            Frame as numpy array
+        """
+        # TODO: Implement actual frame reading
+        # For now, generate dummy frames for testing
+        while True:
+            # Create dummy frame
+            frame = np.zeros((480, 640, 3), dtype=np.uint8)
+            yield frame
+            time.sleep(0.033)  # ~30 FPS
+    
+    def _process_frames(self):
+        """Internal method to process frames in thread."""
+        # TODO: Implement actual frame processing
+        # For now, generate dummy frames for testing
+        while self._running:
+            # Create dummy frame
+            frame = np.zeros((480, 640, 3), dtype=np.uint8)
+            
+            # Create dummy metadata
+            metadata = {
+                'timestamp': time.time(),
+                'format': 'BGR',
+                'width': 640,
+                'height': 480
+            }
+            
+            # Call appropriate callback
+            if self._callback:
+                self._callback(frame)
+            elif self._callback_with_metadata:
+                self._callback_with_metadata(frame, metadata)
+            
+            time.sleep(0.033)  # ~30 FPS
+    
+    def __enter__(self):
+        """Context manager entry."""
+        self.start()
+        return self
+    
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """Context manager exit."""
+        self.stop()

rocket_welder_sdk-1.0.0.dist-info/METADATA

@@ -0,0 +1,36 @@
+Metadata-Version: 2.4
+Name: rocket-welder-sdk
+Version: 1.0.0
+Summary: Client library for RocketWelder video streaming services
+Home-page: https://github.com/modelingevolution/rocket-welder-sdk
+Author: RocketWelder
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Multimedia :: Video
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: numpy>=1.20.0
+Requires-Dist: opencv-python>=4.5.0
+Requires-Dist: zerobuffer-ipc>=1.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: black>=22.0; extra == "dev"
+Requires-Dist: mypy>=1.0; extra == "dev"
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+Client library for RocketWelder video streaming services

rocket_welder_sdk-1.0.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+rocket_welder_sdk/__init__.py,sha256=LT2gnntM29r0IaokaJYOBOIccEcIYZm6FU_t9Eg1pzY,164
+rocket_welder_sdk/client.py,sha256=W4WZdzAJ5eHuDUxQZyR_2rI9i1w2Zxr-UfwcFzKOwDw,5686
+rocket_welder_sdk-1.0.0.dist-info/METADATA,sha256=_cgAcpZzxxEpx8LXVSEzby3YLS6YwohB40zcCkl0CYU,1264
+rocket_welder_sdk-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rocket_welder_sdk-1.0.0.dist-info/top_level.txt,sha256=2iZvBjnwVCUW-uDE23-eJld5PZ9-mlPI69QiXM5IrTA,18
+rocket_welder_sdk-1.0.0.dist-info/RECORD,,

{rocket_welder_sdk-0.0.1.dist-info → rocket_welder_sdk-1.0.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (71.0.4)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

rocket_welder_sdk-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+rocket_welder_sdk

rocket_welder_camera/camera.py

@@ -1,76 +0,0 @@
-# rocket_welder_camera/camera.py
-import cv2
-import threading
-import socket
-import struct
-import numpy as np
-from urllib.parse import urlparse
-
-class RocketWelderCamera:
-    def __init__(self, url):
-        self.url = url
-        parsed_url = urlparse(url)
-        self.host = parsed_url.hostname
-        self.port = parsed_url.port
-        self.stream_name = parsed_url.path[1:]
-        self.buffer_size = 10
-        self.circular_buffer = [None] * self.buffer_size
-        self.current_frame = -1
-        self.lock = threading.Lock()
-        self.thread = threading.Thread(target=self.receive_frames)
-        self.thread.daemon = True
-        self.thread.start()
-
-    @staticmethod
-    def write_prefixed_ascii_string(socket, value):
-        # Encode the string to ASCII bytes
-        name = value.encode('ascii')
-        
-        # Send the length of the string as a single byte
-        length_byte = len(name).to_bytes(1, 'little')
-        socket.send(length_byte)
-        
-        # Send the actual ASCII bytes
-        socket.send(name)
-
-    def receive_frames(self):
-        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-        sock.connect((self.host, self.port))
-        self.write_prefixed_ascii_string(sock,self.stream_name)
-
-        while True:
-            header_data = sock.recv(32)
-            if len(header_data) != 32:
-                continue
-            
-            frame_number, frame_size, stream_position, xor = struct.unpack('<QQQQ', header_data)
-            xor_check = frame_number ^ frame_size ^ stream_position
-            if frame_size == 0 or xor_check != xor:
-                continue
-            
-            frame_size = int(frame_size)
-            frame_data = b''
-            
-            while len(frame_data) < frame_size:
-                packet = sock.recv(frame_size - len(frame_data))
-                if not packet:
-                    break
-                frame_data += packet
-            
-            if len(frame_data) != frame_size:
-                continue
-            
-            frame = np.frombuffer(frame_data, dtype=np.uint8)
-            image = cv2.imdecode(frame, cv2.IMREAD_COLOR)
-            
-            with self.lock:
-                self.current_frame = (self.current_frame + 1) % self.buffer_size
-                self.circular_buffer[self.current_frame] = image
-
-    def get_frame(self):
-        with self.lock:
-            frame = self.circular_buffer[self.current_frame]
-            if frame is not None:
-                return frame
-            else:
-                return None

rocket_welder_sdk-0.0.1.dist-info/METADATA

@@ -1,51 +0,0 @@
-Metadata-Version: 2.1
-Name: rocket-welder-sdk
-Version: 0.0.1
-Summary: Supporting sdk for RocketWelder
-Home-page: https://github.com/rocket-welder-sdk/rocket-welder-sdk
-Author: Rafal Maciag
-Author-email: rafal.maciag@modelingevolution.com
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
-Description-Content-Type: text/markdown
-Requires-Dist: numpy
-Requires-Dist: opencv-python
-Requires-Dist: requests
-
-# SDK for Rocket Welder
-
-Example
-
-```python
-# main.py
-import cv2
-from rocket_welder_camera.camera import RocketWelderCamera
-
-def main():
-    url = 'tcp://{YOUR-HOST}:{YOUR-PORT}/{STREAM-NAME}'
-    cam = RocketWelderCamera(url)
-
-    cv2.namedWindow('Frame', cv2.WINDOW_NORMAL)
-    
-    while True:
-        frame = cam.get_frame()
-        if frame is not None:
-            # Display the frame
-            cv2.imshow('Frame', frame)
-            
-            # Print the current frame number
-            # print(f'Frame number: {cam.current_frame}')
-        
-        # Exit loop if 'q' is pressed
-        if cv2.waitKey(1) & 0xFF == ord('q'):
-            break
-    
-    cv2.destroyAllWindows()
-
-if __name__ == '__main__':
-    main()
-```
-
-Usually, YOUR-PORT is 7000 and YOUR-STREAM is "default".

rocket_welder_sdk-0.0.1.dist-info/RECORD

@@ -1,8 +0,0 @@
-rocket_welder_camera/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rocket_welder_camera/camera.py,sha256=IGyz6g6arKkc9-_JyK9GWa_iT5tE_t6qGk4EEw1kZcI,2629
-tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/test_camera.py,sha256=8oHvTbQRhiCp8cYrEuTkLUpzpTROMd7m3paNLLAGg0c,1255
-rocket_welder_sdk-0.0.1.dist-info/METADATA,sha256=T2gcVRA0M91XbN7RzPpIc42-Fx69kJhYidvjE1mpFm0,1364
-rocket_welder_sdk-0.0.1.dist-info/WHEEL,sha256=rWxmBtp7hEUqVLOnTaDOPpR-cZpCDkzhhcBce-Zyd5k,91
-rocket_welder_sdk-0.0.1.dist-info/top_level.txt,sha256=qQCTu4qgoQUYLOGduATIONjktVBNsnWiYFNVB5_bEyY,27
-rocket_welder_sdk-0.0.1.dist-info/RECORD,,

rocket_welder_sdk-0.0.1.dist-info/top_level.txt

@@ -1,2 +0,0 @@
-rocket_welder_camera
-tests

tests/test_camera.py

@@ -1,36 +0,0 @@
-# tests/test_camera.py
-import unittest
-from unittest.mock import patch, MagicMock
-import numpy as np
-from rocket_welder_camera.camera import RocketWelderCamera
-import struct
-
-class TestRocketWelderCamera(unittest.TestCase):
-    @patch('rocket_welder_camera.camera.socket.socket')
-    @patch('rocket_welder_camera.camera.cv2.imdecode')
-    def test_receive_frames(self, mock_imdecode, mock_socket):
-        mock_sock_instance = MagicMock()
-        mock_socket.return_value = mock_sock_instance
-        mock_sock_instance.recv.side_effect = [
-            struct.pack('>QQQ', 1, 5, 1),  # Mock header
-            b'\x00\x01\x02\x03\x04'        # Mock frame data
-        ]
-        
-        mock_imdecode.return_value = np.zeros((480, 640, 3), dtype=np.uint8)
-        
-        camera = RocketWelderCamera('tcp://pi-51:8082/a')
-        camera.receive_frames()
-
-        
-        frame = camera.get_frame()
-        
-        self.assertIsNotNone(frame)
-        self.assertEqual(frame.shape, (480, 640, 3))
-    
-    def test_get_frame_empty_buffer(self):
-        camera = RocketWelderCamera('tcp://pi-51:8082/a')
-        frame = camera.get_frame()
-        self.assertIsNone(frame)
-
-if __name__ == '__main__':
-    unittest.main()