utils.py

import math

def euclidean_distance(p1, p2):
    """
    Returns the Euclidean distance between two (x, y) points.
    """
    return math.sqrt((p1[0] - p2[0]) ** 2 + (p1[1] - p2[1]) ** 2)

def estimate_ball_speed(ball_positions, fps=25, scale=0.01):
    """
    Estimates average ball speed in meters/second.

    Args:
        ball_positions: List of (frame_idx, x, y)
        fps: Frame rate of video
        scale: Pixel-to-meter conversion (assumed or calibrated)

    Returns:
        float: estimated speed (m/s)
    """
    if len(ball_positions) < 2:
        return 0.0

    distances = []
    for i in range(1, len(ball_positions)):
        _, x1, y1 = ball_positions[i - 1]
        _, x2, y2 = ball_positions[i]
        dist_px = euclidean_distance((x1, y1), (x2, y2))
        distances.append(dist_px)

    avg_px_per_frame = sum(distances) / len(distances)
    speed_m_per_s = avg_px_per_frame * scale * fps
    return round(speed_m_per_s, 2)

def frame_to_time(frame_idx, fps):
    """
    Converts frame index to timestamp (seconds).
    """
    return round(frame_idx / fps, 2)

def format_impact_info(impact_frame, impact_type, fps):
    """
    Creates a descriptive impact summary.
    """
    time_sec = frame_to_time(impact_frame, fps)
    return f"Impact at {time_sec}s on {impact_type.upper()}"

def log(message):
    """
    Simple print logger with prefix.
    """
    print(f"[LBW-DRS] {message}")