Create utils.py

utils.py

@@ -0,0 +1,52 @@
+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}")