test_video.py

#!/usr/bin/env python3

"""
Test script to debug video stream issues with AR.Drone 2.0
"""

import time
import sys
import libardrone

def main():
    print("AR.Drone 2.0 Video Stream Test")
    print("================================")
    
    # Initialize drone
    print("1. Initializing drone connection...")
    try:
        drone = libardrone.ARDrone()
    except Exception as e:
        print(f"ERROR: Failed to initialize drone: {e}")
        return
    
    print("2. Drone initialized successfully")
    
    # Wait for initial connection
    print("3. Waiting for connection stabilization...")
    time.sleep(2)
    
    # Enable video stream explicitly
    print("4. Enabling video stream...")
    drone.enable_video_stream()
    
    # Check connection and video data
    print("5. Monitoring video stream for 10 seconds...")
    start_time = time.time()
    last_image_check = ""
    
    while time.time() - start_time < 10:
        # Check if we have navigation data
        if hasattr(drone, 'navdata') and drone.navdata:
            nav_keys = list(drone.navdata.keys())
            if nav_keys:
                print(f"   Navdata keys available: {nav_keys}")
                
                # Check drone state
                if 'drone_state' in drone.navdata:
                    state = drone.navdata['drone_state']
                    video_enabled = state.get('video_mask', 0)
                    video_thread = state.get('video_thread_on', 0)
                    print(f"   Video enabled: {video_enabled}, Video thread: {video_thread}")
        
        # Check if we have video data
        if hasattr(drone, 'image') and drone.image:
            if drone.image != last_image_check:
                print(f"   Video data received! Image size: {len(drone.image)} bytes")
                last_image_check = drone.image
            else:
                print("   No new video frames...")
        else:
            print("   No video data yet...")
        
        time.sleep(1)
    
    print("6. Test completed")
    
    # Clean shutdown
    print("7. Shutting down...")
    drone.halt()
    print("Done!")

if __name__ == "__main__":
    main()