test_rt_trigger.py

#!/usr/bin/env python3
"""
Test RT (Right Trigger) for Turbo Speed
"""

import pygame
import sys
import time

def main():
    print("🎮 Testing RT (Right Trigger) for Turbo Speed...")
    
    # 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!")
        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("\n🎮 RT Trigger Test - Hold RT to see turbo effect!")
    print("   Press ESC or close window to exit")
    print("   Watch the RT axis value and turbo multiplier")
    
    # Create a window
    screen = pygame.display.set_mode((600, 500))
    pygame.display.set_caption("RT Trigger Turbo Test")
    font = pygame.font.Font(None, 24)
    clock = pygame.time.Clock()
    
    # Settings
    base_speed = 1200.0
    boost_multiplier = 2.0
    
    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 = 30
        
        # Get RT trigger value
        try:
            rt_value = float(joystick.get_axis(5))  # RT axis
        except:
            rt_value = 0.0
        
        # Calculate turbo effect
        if rt_value > 0.1:
            boost_factor = 1.0 + (rt_value * (boost_multiplier - 1.0))
            current_speed = base_speed * boost_factor
            turbo_active = True
        else:
            boost_factor = 1.0
            current_speed = base_speed
            turbo_active = False
        
        # 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 RT trigger info
        rt_text = f"RT Trigger Value: {rt_value:.3f}"
        rt_color = (0, 255, 0) if rt_value > 0.1 else (128, 128, 128)
        text = font.render(rt_text, True, rt_color)
        screen.blit(text, (10, y))
        y += line_height
        
        # Display turbo status
        turbo_text = f"Turbo Active: {'YES' if turbo_active else 'NO'}"
        turbo_color = (255, 255, 0) if turbo_active else (128, 128, 128)
        text = font.render(turbo_text, True, turbo_color)
        screen.blit(text, (10, y))
        y += line_height
        
        # Display speed info
        speed_text = f"Base Speed: {base_speed:.0f} px/sec"
        text = font.render(speed_text, True, (200, 200, 200))
        screen.blit(text, (10, y))
        y += line_height
        
        boost_text = f"Boost Factor: {boost_factor:.2f}x"
        boost_color = (255, 100, 100) if turbo_active else (200, 200, 200)
        text = font.render(boost_text, True, boost_color)
        screen.blit(text, (10, y))
        y += line_height
        
        current_text = f"Current Speed: {current_speed:.0f} px/sec"
        current_color = (255, 255, 0) if turbo_active else (200, 200, 200)
        text = font.render(current_text, True, current_color)
        screen.blit(text, (10, y))
        y += line_height
        
        # Visual indicator
        y += 20
        bar_width = 400
        bar_height = 30
        bar_x = 10
        bar_y = y
        
        # Background bar
        pygame.draw.rect(screen, (50, 50, 50), (bar_x, bar_y, bar_width, bar_height))
        
        # Turbo bar
        if turbo_active:
            turbo_width = int(bar_width * rt_value)
            pygame.draw.rect(screen, (255, 100, 100), (bar_x, bar_y, turbo_width, bar_height))
        
        # Border
        pygame.draw.rect(screen, (200, 200, 200), (bar_x, bar_y, bar_width, bar_height), 2)
        
        # Instructions
        y += 60
        inst_text = "Hold RT (Right Trigger) to activate turbo speed!"
        text = font.render(inst_text, True, (255, 255, 255))
        screen.blit(text, (10, y))
        
        pygame.display.flip()
        clock.tick(60)
    
    pygame.quit()
    print("✅ RT trigger test completed!")

if __name__ == "__main__":
    main()