basic_image_transfer.py

#!/usr/bin/env python3
"""
Basic Image Transfer Example

This example demonstrates sending and receiving images between
PC and Device via UART using PyDIPLink.
"""

import serial
import time
from pydiplink import (
    send_image_data,
    capture_image_and_show,
    IMAGE_FORMAT_RGB888,
    IMAGE_FORMAT_GRAYSCALE,
)


def main():
    # Configure serial port
    PORT = "/dev/ttyUSB0"  # Change to your port (COM3 on Windows)
    BAUD = 500000

    print(f"Connecting to {PORT} at {BAUD} baud...")

    try:
        ser = serial.Serial(PORT, BAUD, timeout=5)
        print("Connected successfully!")

        # Example 1: Send RGB888 image to device
        print("\n=== Example 1: Send RGB888 Image ===")
        try:
            send_image_data(
                ser,
                image_path="test_image.jpg",  # Your image path
                width=320,
                height=240,
                image_format=IMAGE_FORMAT_RGB888,
            )
            print("Image sent successfully!")
        except FileNotFoundError as e:
            print(f"Error: {e}")
            print("Please provide a valid image path")

        time.sleep(1)

        # Example 2: Send grayscale image
        print("\n=== Example 2: Send Grayscale Image ===")
        try:
            send_image_data(
                ser,
                image_path="test_image.jpg",
                width=160,
                height=120,
                image_format=IMAGE_FORMAT_GRAYSCALE,
            )
            print("Grayscale image sent successfully!")
        except FileNotFoundError as e:
            print(f"Error: {e}")

        # Example 3: Receive image from device
        print("\n=== Example 3: Receive Image from Device ===")
        print("Waiting for device to send image...")
        print("(Device should send STW command)")

        try:
            capture_image_and_show(ser, save_dir="received_images")
            print("Image received and saved!")
        except TimeoutError as e:
            print(f"Timeout: {e}")
            print("Make sure your device is sending data")

    except serial.SerialException as e:
        print(f"Serial port error: {e}")
        print("\nTroubleshooting:")
        print("- Check that device is connected")
        print("- Verify correct port name")
        print("- On Linux, add user to dialout group:")
        print("  sudo usermod -a -G dialout $USER")

    except KeyboardInterrupt:
        print("\nInterrupted by user")

    finally:
        if 'ser' in locals():
            ser.close()
            print("\nSerial port closed")


if __name__ == "__main__":
    main()