puda-drivers 0.0.11__tar.gz → 0.0.12__tar.gz

{puda_drivers-0.0.11 → puda_drivers-0.0.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: puda-drivers
-Version: 0.0.11
+Version: 0.0.12
 Summary: Hardware drivers for the PUDA platform.
 Project-URL: Homepage, https://github.com/zhao-bears/puda-drivers
 Project-URL: Issues, https://github.com/zhao-bears/puda-drivers/issues
@@ -85,6 +85,7 @@ When file logging is enabled, logs are saved to timestamped files (unless a cust
 The `First` machine integrates motion control, deck management, liquid handling, and camera capabilities:
 
 ```python
+import time
 import logging
 from puda_drivers.machines import First
 from puda_drivers.core.logging import setup_logging
@@ -102,14 +103,8 @@ machine = First(
     camera_index=0,
 )
 
-# Connect all devices
-machine.connect()
-
-# Home the gantry
-machine.qubot.home()
-
-# Initialize the pipette
-machine.pipette.initialize()
+# Start up the machine (connects all controllers, homes gantry, and initializes pipette)
+machine.startup()
 
 # Load labware onto the deck
 machine.load_deck({
@@ -119,19 +114,19 @@ machine.load_deck({
 })
 
 # Start video recording
-machine.camera.start_video_recording()
+machine.start_video_recording()
 
 # Perform liquid handling operations
 machine.attach_tip(slot="A3", well="G8")
 machine.aspirate_from(slot="C2", well="A1", amount=100, height_from_bottom=10.0)
 machine.dispense_to(slot="C2", well="B4", amount=100, height_from_bottom=30.0)
-machine.drop_tip(slot="C1", well="A1")
+machine.drop_tip(slot="C1", well="A1", height_from_bottom=10)
 
 # Stop video recording
-machine.camera.stop_video_recording()
+machine.stop_video_recording()
 
-# Disconnect all devices
-machine.disconnect()
+# Shutdown the machine (gracefully disconnects all controllers)
+machine.shutdown()
 ```
 
 **Discovering Available Methods**: To explore what methods are available on any class instance, you can use Python's built-in `help()` function:

{puda_drivers-0.0.11 → puda_drivers-0.0.12}/README.md

@@ -64,6 +64,7 @@ When file logging is enabled, logs are saved to timestamped files (unless a cust
 The `First` machine integrates motion control, deck management, liquid handling, and camera capabilities:
 
 ```python
+import time
 import logging
 from puda_drivers.machines import First
 from puda_drivers.core.logging import setup_logging
@@ -81,14 +82,8 @@ machine = First(
     camera_index=0,
 )
 
-# Connect all devices
-machine.connect()
-
-# Home the gantry
-machine.qubot.home()
-
-# Initialize the pipette
-machine.pipette.initialize()
+# Start up the machine (connects all controllers, homes gantry, and initializes pipette)
+machine.startup()
 
 # Load labware onto the deck
 machine.load_deck({
@@ -98,19 +93,19 @@ machine.load_deck({
 })
 
 # Start video recording
-machine.camera.start_video_recording()
+machine.start_video_recording()
 
 # Perform liquid handling operations
 machine.attach_tip(slot="A3", well="G8")
 machine.aspirate_from(slot="C2", well="A1", amount=100, height_from_bottom=10.0)
 machine.dispense_to(slot="C2", well="B4", amount=100, height_from_bottom=30.0)
-machine.drop_tip(slot="C1", well="A1")
+machine.drop_tip(slot="C1", well="A1", height_from_bottom=10)
 
 # Stop video recording
-machine.camera.stop_video_recording()
+machine.stop_video_recording()
 
-# Disconnect all devices
-machine.disconnect()
+# Shutdown the machine (gracefully disconnects all controllers)
+machine.shutdown()
 ```
 
 **Discovering Available Methods**: To explore what methods are available on any class instance, you can use Python's built-in `help()` function:

{puda_drivers-0.0.11 → puda_drivers-0.0.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "puda-drivers"
-version = "0.0.11"
+version = "0.0.12"
 description = "Hardware drivers for the PUDA platform."
 readme = "README.md"
 authors = [

{puda_drivers-0.0.11 → puda_drivers-0.0.12}/src/puda_drivers/machines/first.py

@@ -9,7 +9,9 @@ This class demonstrates the integration of:
 
 import logging
 import time
+from pathlib import Path
 from typing import Optional, Dict, Tuple, Type, Union
+import numpy as np
 from puda_drivers.move import GCodeController, Deck
 from puda_drivers.core import Position
 from puda_drivers.transfer.liquid.sartorius import SartoriusController
@@ -118,21 +120,44 @@ class First:
             camera_index if camera_index is not None else self.DEFAULT_CAMERA_INDEX,
         )
         
-    def connect(self):
-        """Connect all controllers."""
-        self._logger.info("Connecting all controllers")
+    def startup(self):
+        """
+        Start up the machine by connecting all controllers and initializing subsystems.
+        
+        This method:
+        - Connects to all controllers (gantry, pipette, camera)
+        - Homes the gantry to establish a known position
+        - Initializes the pipette to reset it to a known state
+        
+        The machine is ready for operations after this method completes.
+        """
+        self._logger.info("Starting up machine and connecting all controllers")
         self.qubot.connect()
         self.pipette.connect()
         self.camera.connect()
         self._logger.info("All controllers connected successfully")
         
-    def disconnect(self):
-        """Disconnect all controllers."""
-        self._logger.info("Disconnecting all controllers")
+        # Home the gantry to establish known position
+        self._logger.info("Homing gantry...")
+        self.qubot.home()
+        
+        # Initialize the pipette (all pipette operations need to wait 5 seconds for completion)
+        self._logger.info("Initializing pipette...")
+        self.pipette.initialize()
+        time.sleep(5)
+        self._logger.info("Machine startup complete - ready for operations")
+        
+    def shutdown(self):
+        """
+        Gracefully shut down the machine by disconnecting all controllers.
+        
+        This method ensures all connections are properly closed and resources are released.
+        """
+        self._logger.info("Shutting down machine and disconnecting all controllers")
         self.qubot.disconnect()
         self.pipette.disconnect()
         self.camera.disconnect()
-        self._logger.info("All controllers disconnected successfully")
+        self._logger.info("Machine shutdown complete")
         
     def load_labware(self, slot: str, labware_name: str):
         """Load a labware object into a slot."""
@@ -350,3 +375,86 @@ class First:
         else:
             self._logger.debug("Absolute A position for slot '%s': %s", slot, pos)
         return pos
+    
+    def start_video_recording(
+        self,
+        filename: Optional[Union[str, Path]] = None,
+        fps: Optional[float] = None
+    ) -> Path:
+        """
+        Start recording a video.
+        
+        Args:
+            filename: Optional filename for the video. If not provided, a timestamped
+                    filename will be generated. If provided without extension, .mp4 will be added.
+            fps: Optional frames per second for the video. Defaults to 30.0 if not specified.
+        
+        Returns:
+            Path to the video file where recording is being saved
+            
+        Raises:
+            IOError: If camera is not connected or recording fails to start
+            ValueError: If already recording
+        """
+        return self.camera.start_video_recording(filename=filename, fps=fps)
+    
+    def stop_video_recording(self) -> Optional[Path]:
+        """
+        Stop recording a video.
+        
+        Returns:
+            Path to the saved video file, or None if no recording was in progress
+            
+        Raises:
+            IOError: If video writer fails to release
+        """
+        return self.camera.stop_video_recording()
+    
+    def record_video(
+        self,
+        duration_seconds: float,
+        filename: Optional[Union[str, Path]] = None,
+        fps: Optional[float] = None
+    ) -> Path:
+        """
+        Record a video for a specified duration.
+        
+        Args:
+            duration_seconds: Duration of the video in seconds
+            filename: Optional filename for the video. If not provided, a timestamped
+                    filename will be generated. If provided without extension, .mp4 will be added.
+            fps: Optional frames per second for the video. Defaults to 30.0 if not specified.
+        
+        Returns:
+            Path to the saved video file
+            
+        Raises:
+            IOError: If camera is not connected or recording fails
+        """
+        return self.camera.record_video(
+            duration_seconds=duration_seconds,
+            filename=filename,
+            fps=fps
+        )
+    
+    def capture_image(
+        self, 
+        save: bool = False, 
+        filename: Optional[Union[str, Path]] = None
+    ) -> np.ndarray:
+        """
+        Capture a single image from the camera.
+        
+        Args:
+            save: If True, save the image to the captures folder
+            filename: Optional filename for the saved image. If not provided and save=True,
+                     a timestamped filename will be generated. If provided without extension,
+                     .jpg will be added.
+        
+        Returns:
+            Captured image as a numpy array (BGR format)
+            
+        Raises:
+            IOError: If camera is not connected or capture fails
+        """
+        return self.camera.capture_image(save=save, filename=filename)

{puda_drivers-0.0.11 → puda_drivers-0.0.12}/tests/first.py

@@ -1,6 +1,7 @@
 """Test script for the First machine driver."""
 
 import random
+import time
 import logging
 from puda_drivers.machines import First
 from puda_drivers.labware import get_available_labware
@@ -8,7 +9,7 @@ from puda_drivers.core import setup_logging
 
 setup_logging(
     enable_file_logging=False,
-    log_level=logging.DEBUG,  # Use logging.DEBUG to see all (DEBUG, INFO, WARNING, ERROR, CRITICAL) logs
+    log_level=logging.INFO,  # Use logging.DEBUG to see all (DEBUG, INFO, WARNING, ERROR, CRITICAL) logs
 )
 
 if __name__ == "__main__":
@@ -34,15 +35,15 @@ if __name__ == "__main__":
     print(machine.deck)
     print(machine.deck["C2"])
 
-    machine.connect()
-    machine.qubot.home()
-    machine.pipette.initialize()
+    machine.startup()  # Connects all controllers, homes gantry, and initializes pipette
     
-    machine.camera.start_video_recording()
+    # machine.record_video(duration_seconds=10, filename="test.mp4")
+    machine.start_video_recording()
     machine.attach_tip(slot="A3", well="G8")
     machine.aspirate_from(slot="C2", well="A1", amount=100, height_from_bottom=10)
+    # machine.capture_image()
     machine.dispense_to(slot="C2", well="B4", amount=100, height_from_bottom=50)
-    machine.drop_tip(slot="C1", well="A1")
+    machine.drop_tip(slot="C1", well="A1", height_from_bottom=10)
     
     # tiprack_wells = machine.deck["A3"].wells
     # # get pick up pipette one by one and drop it in the trash bin
@@ -58,7 +59,7 @@ if __name__ == "__main__":
     #     machine.drop_tip("C1", "A1")
 
 
-    machine.camera.stop_video_recording()
+    machine.stop_video_recording()
     
-    # Disconnect machine
-    machine.disconnect()
+    # Shutdown machine
+    machine.shutdown()

{puda_drivers-0.0.11 → puda_drivers-0.0.12}/uv.lock

@@ -186,7 +186,7 @@ wheels = [
 
 [[package]]
 name = "puda-drivers"
-version = "0.0.11"
+version = "0.0.12"
 source = { editable = "." }
 dependencies = [
     { name = "opencv-python" },