test_controller.py

#!/usr/bin/env python3
"""
Test Xbox Controller with Pygame
"""

import pygame
import sys
import time

def main():
    print("🎮 Testing Xbox Controller...")
    
    # Initialize pygame
    pygame.init()
    pygame.joystick.init()
    
    # Check for joysticks
    joystick_count = pygame.joystick.get_count()
    print(f"Found {joystick_count} joystick(s)")
    
    if joystick_count == 0:
        print("❌ No joysticks found!")
        print("Make sure your Xbox controller is connected and try:")
        print("1. Unplug and replug the controller")
        print("2. Check: ls /dev/input/js*")
        print("3. Check: lsusb | grep -i xbox")
        return
    
    # Initialize the first joystick
    joystick = pygame.joystick.Joystick(0)
    joystick.init()
    
    print(f"✅ Controller: {joystick.get_name()}")
    print(f"   Axes: {joystick.get_numaxes()}")
    print(f"   Buttons: {joystick.get_numbuttons()}")
    print(f"   Hats: {joystick.get_numhats()}")
    
    print("\n🎮 Controller Test - Press buttons and move sticks!")
    print("   Press ESC or close window to exit")
    print("   Watch the values change as you use the controller")
    
    # Create a small window
    screen = pygame.display.set_mode((600, 400))
    pygame.display.set_caption("Xbox Controller Test")
    font = pygame.font.Font(None, 24)
    clock = pygame.time.Clock()
    
    running = True
    while running:
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                running = False
            elif event.type == pygame.KEYDOWN:
                if event.key == pygame.K_ESCAPE:
                    running = False
        
        # Clear screen
        screen.fill((0, 0, 0))
        
        y = 20
        line_height = 25
        
        # Display controller info
        info_text = f"Controller: {joystick.get_name()}"
        text = font.render(info_text, True, (255, 255, 255))
        screen.blit(text, (10, y))
        y += line_height
        
        # Display axes (analog sticks and triggers)
        for i in range(joystick.get_numaxes()):
            axis_value = joystick.get_axis(i)
            axis_text = f"Axis {i}: {axis_value:.3f}"
            color = (0, 255, 0) if abs(axis_value) > 0.1 else (128, 128, 128)
            text = font.render(axis_text, True, color)
            screen.blit(text, (10, y))
            y += line_height
        
        # Display buttons
        for i in range(joystick.get_numbuttons()):
            button_pressed = joystick.get_button(i)
            button_text = f"Button {i}: {'PRESSED' if button_pressed else 'released'}"
            color = (255, 255, 0) if button_pressed else (128, 128, 128)
            text = font.render(button_text, True, color)
            screen.blit(text, (10, y))
            y += line_height
        
        # Display hats (D-pad)
        for i in range(joystick.get_numhats()):
            hat_value = joystick.get_hat(i)
            hat_text = f"Hat {i}: {hat_value}"
            color = (255, 0, 255) if hat_value != (0, 0) else (128, 128, 128)
            text = font.render(hat_text, True, color)
            screen.blit(text, (10, y))
            y += line_height
        
        pygame.display.flip()
        clock.tick(60)
    
    pygame.quit()
    print("✅ Controller test completed!")

if __name__ == "__main__":
    main()