Spaces:
Sleeping
Sleeping
File size: 3,490 Bytes
40518b9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 | 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)
|