rocket-welder-sdk 0.0.1__tar.gz

rocket_welder_sdk-0.0.1/PKG-INFO

@@ -0,0 +1,51 @@
+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/README.md

@@ -0,0 +1,35 @@
+# 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/rocket_welder_camera/camera.py

@@ -0,0 +1,76 @@
+# 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/rocket_welder_sdk.egg-info/PKG-INFO

@@ -0,0 +1,51 @@
+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/rocket_welder_sdk.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+setup.py
+rocket_welder_camera/__init__.py
+rocket_welder_camera/camera.py
+rocket_welder_sdk.egg-info/PKG-INFO
+rocket_welder_sdk.egg-info/SOURCES.txt
+rocket_welder_sdk.egg-info/dependency_links.txt
+rocket_welder_sdk.egg-info/requires.txt
+rocket_welder_sdk.egg-info/top_level.txt
+tests/__init__.py
+tests/test_camera.py

rocket_welder_sdk-0.0.1/rocket_welder_sdk.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

rocket_welder_sdk-0.0.1/rocket_welder_sdk.egg-info/requires.txt

@@ -0,0 +1,3 @@
+numpy
+opencv-python
+requests

rocket_welder_sdk-0.0.1/rocket_welder_sdk.egg-info/top_level.txt

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

rocket_welder_sdk-0.0.1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

rocket_welder_sdk-0.0.1/setup.py

@@ -0,0 +1,33 @@
+from setuptools import setup, find_packages
+
+with open("README.md", "r") as fh:
+    long_description = fh.read()
+
+setup(
+    name="rocket-welder-sdk",
+    version="0.0.1",
+    author="Rafal Maciag",
+    author_email="rafal.maciag@modelingevolution.com",
+    description="Supporting sdk for RocketWelder",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/rocket-welder-sdk/rocket-welder-sdk",
+    packages=find_packages(),
+    install_requires=[
+        'numpy',
+        'opencv-python',
+        'requests',  # Add any other dependencies here
+    ],
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+    ],
+    python_requires='>=3.6',
+    package_data={
+        'rocket_welder_camera': ['*.py'],  # Include only Python files
+    },
+    exclude_package_data={
+        '': ['tests/*'],  # Exclude test files
+    },
+)

rocket_welder_sdk-0.0.1/tests/test_camera.py

@@ -0,0 +1,36 @@
+# 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()