camera_detect.py

#!/usr/bin/env python3
"""
Camera Detection Script - Find all available cameras
"""

import cv2
import sys

def find_cameras():
    """Test different camera indices to find available cameras"""
    available_cameras = []
    
    print("Searching for available cameras...")
    
    # Test camera indices 0-10 (usually sufficient)
    for i in range(11):
        cap = cv2.VideoCapture(i)
        if cap.isOpened():
            ret, frame = cap.read()
            if ret:
                height, width = frame.shape[:2]
                available_cameras.append({
                    'index': i,
                    'width': width,
                    'height': height
                })
                print(f"Camera {i}: Found - Resolution: {width}x{height}")
            else:
                print(f"Camera {i}: Opened but no frame received")
            cap.release()
        else:
            print(f"Camera {i}: Not available")
    
    return available_cameras

def test_camera(camera_index):
    """Test a specific camera with live preview"""
    cap = cv2.VideoCapture(camera_index)
    
    if not cap.isOpened():
        print(f"Cannot open camera {camera_index}")
        return False
    
    print(f"Testing camera {camera_index}. Press 'q' to quit, 's' to save frame")
    
    frame_count = 0
    try:
        while True:
            ret, frame = cap.read()
            if not ret:
                print("Failed to grab frame")
                break
            
            frame_count += 1
            
            # Add text overlay
            cv2.putText(frame, f"Camera {camera_index} - Frame {frame_count}", 
                       (10, 30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
            
            try:
                cv2.imshow(f'Camera {camera_index} Test', frame)
            except cv2.error:
                # If display fails, save frame instead
                if frame_count % 30 == 0:  # Save every 30th frame
                    filename = f'camera_{camera_index}_test_frame_{frame_count}.jpg'
                    cv2.imwrite(filename, frame)
                    print(f"Saved {filename}")
            
            key = cv2.waitKey(1) & 0xFF
            if key == ord('q'):
                break
            elif key == ord('s'):
                filename = f'camera_{camera_index}_saved_frame.jpg'
                cv2.imwrite(filename, frame)
                print(f"Frame saved as {filename}")
    
    except KeyboardInterrupt:
        print("\nInterrupted by user")
    
    finally:
        cap.release()
        cv2.destroyAllWindows()
    
    return True

if __name__ == "__main__":
    if len(sys.argv) > 1:
        try:
            camera_index = int(sys.argv[1])
            test_camera(camera_index)
        except ValueError:
            print("Please provide a valid camera index number")
    else:
        cameras = find_cameras()
        
        if cameras:
            print(f"\nFound {len(cameras)} available camera(s):")
            for cam in cameras:
                print(f"  Camera {cam['index']}: {cam['width']}x{cam['height']}")
            
            print(f"\nTo test a specific camera, run:")
            print(f"python3 {sys.argv[0]} <camera_index>")
        else:
            print("No cameras found!")