Update utils.py

utils.py

@@ -1,52 +1,22 @@
-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}")
+import cv2
+
+def save_video(frames, output_path, fps=30):
+    if not frames:
+        return
+    height, width = frames[0].shape[:2]
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
+    for frame in frames:
+        out.write(frame)
+    out.release()
+
+def extract_frames(video_path):
+    cap = cv2.VideoCapture(video_path)
+    frames = []
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            break
+        frames.append(frame)
+    cap.release()
+    return frames