import cv2
import numpy as np
import math
from utils.draw_text_with_background import draw_text_with_background

def display_counter(frame, counter, position=(40, 240), color=(0, 0, 0), background_color=(192, 192, 192)):
    """Display the repetition counter."""
    text = f"Count: {counter}"
    draw_text_with_background(frame, text, position, 
                             cv2.FONT_HERSHEY_SIMPLEX, 0.7, color, background_color, 1)

def display_stage(frame, stage, label="Stage", position=(40, 270), color=(0, 0, 0), background_color=(192, 192, 192)):
    """Display the current exercise stage."""
    text = f"{label}: {stage}"
    draw_text_with_background(frame, text, position, 
                             cv2.FONT_HERSHEY_SIMPLEX, 0.7, color, background_color, 1)

def draw_progress_bar(frame, exercise, value, position, size=(200, 20), color=(0, 255, 0), background_color=(255, 255, 255)):
    """Draw a progress bar for tracking exercise repetitions."""
    x, y = position
    width, height = size
    
    # Get max value for the exercise type
    max_value = 10  # Default
    if exercise == "squat":
        max_value = 15
    elif exercise == "push_up":
        max_value = 10
    elif exercise == "hammer_curl":
        max_value = 12
    
    # Calculate fill width
    fill_width = int((value / max_value) * width)
    fill_width = min(fill_width, width)  # Ensure it doesn't exceed max width
    
    # Draw background
    cv2.rectangle(frame, (x, y), (x + width, y + height), background_color, -1)
    cv2.rectangle(frame, (x, y), (x + width, y + height), (0, 0, 0), 1)
    
    # Draw fill
    if fill_width > 0:
        cv2.rectangle(frame, (x, y), (x + fill_width, y + height), color, -1)
    
    # Draw text
    text = f"{value}/{max_value}"
    text_size = cv2.getTextSize(text, cv2.FONT_HERSHEY_SIMPLEX, 0.5, 1)[0]
    text_x = x + (width - text_size[0]) // 2
    text_y = y + (height + text_size[1]) // 2
    cv2.putText(frame, text, (text_x, text_y), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 0), 1)
    
    # Draw label above the progress bar
    label = f"{exercise.replace('_', ' ').title()} Progress"
    draw_text_with_background(frame, label, (x, y - 10), 
                             cv2.FONT_HERSHEY_SIMPLEX, 0.6, (255, 255, 255), (118, 29, 14), 1)

def draw_gauge_meter(frame, angle, text, position, radius=50, color=(0, 0, 255)):
    """Draw a gauge meter visualization showing the angle."""
    x, y = position
    start_angle = 180
    end_angle = 0
    
    # Draw outer circle
    cv2.circle(frame, (x, y), radius, (200, 200, 200), 2)
    
    # Calculate the angle position on the gauge
    gauge_angle = start_angle - (angle * (start_angle - end_angle) / 180)
    gauge_angle = max(min(gauge_angle, start_angle), end_angle) # Constrain angle
    
    # Convert to radians
    gauge_angle_rad = math.radians(gauge_angle)
    
    # Calculate point on circle
    gauge_x = int(x + radius * math.cos(gauge_angle_rad))
    gauge_y = int(y - radius * math.sin(gauge_angle_rad))
    
    # Draw line from center to angle point
    cv2.line(frame, (x, y), (gauge_x, gauge_y), color, 2)
    
    # Draw center circle
    cv2.circle(frame, (x, y), 5, color, -1)
    
    # Draw text
    cv2.putText(frame, f"{int(angle)}°", (x - 20, y + radius + 20), 
                cv2.FONT_HERSHEY_SIMPLEX, 0.6, color, 2)
    
    # Draw title
    cv2.putText(frame, text, (x - radius, y - radius - 10),
                cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 0), 1)