Update utils.py

utils.py

@@ -4,15 +4,20 @@ import cv2
 import numpy as np
 
 def analyze_frame_sequence(frames):
+    # Simulate realistic trajectory
+    h, w = frames[0].shape[:2]
+    trajectory_curve = [(200+i*10, 300 - i*8) for i in range(6)]  # sloping downward
     return {
         "pitch": "in line",
         "impact": "in line",
         "trajectory": "hitting",
         "shot_offered": True,
-        "pitch_point": (200, 300),  # example pixel position
-        "impact_point": (220, 320),
-        "trajectory_curve": [(230, 330), (250, 310), (270, 290), (290, 270)],
-        "stump_zone": [(300, 200), (340, 400)]  # top-left and bottom-right of projected stumps
+        "pitch_point": trajectory_curve[0],
+        "impact_point": trajectory_curve[2],
+        "trajectory_curve": trajectory_curve,
+        "stump_zone": [(trajectory_curve[-1][0] - 20, trajectory_curve[-1][1] - 40),
+                        (trajectory_curve[-1][0] + 20, trajectory_curve[-1][1] + 40)],
+        "hit_point": trajectory_curve[-1]
     }
 
 def make_decision(analysis):
@@ -29,65 +34,61 @@ def make_decision(analysis):
 def overlay_text(frame, text, pos, size=1.0, color=(255,255,255)):
     cv2.putText(frame, text, pos, cv2.FONT_HERSHEY_SIMPLEX, size, color, 2, cv2.LINE_AA)
 
-def overlay_annotations(frame, analysis):
-    # pitch point
-    if 'pitch_point' in analysis:
-        cv2.circle(frame, analysis['pitch_point'], 10, (0, 255, 0) if analysis['pitch'] == 'in line' else (0, 0, 255), -1)
-        overlay_text(frame, f"Pitch: {analysis['pitch'].capitalize()}", (analysis['pitch_point'][0]+10, analysis['pitch_point'][1]), 0.7)
-
-    # impact point
-    if 'impact_point' in analysis:
-        cv2.circle(frame, analysis['impact_point'], 10, (0, 255, 0) if analysis['impact'] == 'in line' else (0, 0, 255), -1)
-        overlay_text(frame, f"Impact: {analysis['impact'].capitalize()}", (analysis['impact_point'][0]+10, analysis['impact_point'][1]), 0.7)
-
-    # shot icon
+def overlay_annotations_dynamic(frame, analysis, t):
+    if t < len(analysis['trajectory_curve']):
+        ball_pos = analysis['trajectory_curve'][t]
+        cv2.circle(frame, ball_pos, 8, (0, 255, 0) if analysis['trajectory'] == 'hitting' else (0, 0, 255), -1)
+
+    # Draw complete arc
+    for i in range(1, min(t + 1, len(analysis['trajectory_curve']))):
+        cv2.line(frame, analysis['trajectory_curve'][i - 1], analysis['trajectory_curve'][i],
+                 (0, 255, 0) if analysis['trajectory'] == 'hitting' else (0, 0, 255), 2)
+
+    # Highlight stumps zone
+    x1, y1 = analysis['stump_zone'][0]
+    x2, y2 = analysis['stump_zone'][1]
+    overlay = frame.copy()
+    cv2.rectangle(overlay, (x1, y1), (x2, y2), (255, 255, 255), -1)
+    alpha = 0.2
+    frame[:] = cv2.addWeighted(overlay, alpha, frame, 1 - alpha, 0)
+    cv2.rectangle(frame, (x1, y1), (x2, y2), (180, 180, 180), 2)
+
+    # Shot
     shot_icon = "✓" if analysis['shot_offered'] else "✗"
     overlay_text(frame, f"Shot Offered: {shot_icon}", (30, 170), 0.8, (0,255,0) if analysis['shot_offered'] else (0,0,255))
 
-    # trajectory arc
-    if 'trajectory_curve' in analysis:
-        for i in range(len(analysis['trajectory_curve'])-1):
-            cv2.line(frame, analysis['trajectory_curve'][i], analysis['trajectory_curve'][i+1], (0,255,0) if analysis['trajectory']=="hitting" else (0,0,255), 2, cv2.LINE_AA)
-
-    # transparent stump zone
-    if 'stump_zone' in analysis:
-        x1, y1 = analysis['stump_zone'][0]
-        x2, y2 = analysis['stump_zone'][1]
-        overlay = frame.copy()
-        cv2.rectangle(overlay, (x1, y1), (x2, y2), (255, 255, 255), -1)
-        alpha = 0.2
-        frame[:] = cv2.addWeighted(overlay, alpha, frame, 1 - alpha, 0)
-        cv2.rectangle(frame, (x1, y1), (x2, y2), (180, 180, 180), 2)
-
-    overlay_text(frame, f"Trajectory: {analysis['trajectory'].capitalize()}", (30, 130), 0.8, (0,255,0) if analysis['trajectory']=="hitting" else (0,0,255))
+    overlay_text(frame, f"Trajectory: {analysis['trajectory'].capitalize()}", (30, 130), 0.8,
+                 (0, 255, 0) if analysis['trajectory'] == 'hitting' else (0, 0, 255))
     overlay_text(frame, "ICC LBW Review • Third-Umpire Analysis", (30, frame.shape[0] - 30), 0.6, (180,180,180))
 
-
 def render_annotated_clip(frames, analysis, decision, reason, output_path):
     h, w = frames[0].shape[:2]
     out = cv2.VideoWriter(output_path, cv2.VideoWriter_fourcc(*'mp4v'), 20.0, (w, h))
 
-    # First 1s: Verdict card
+    # Verdict card (1s)
     verdict_card = np.zeros_like(frames[0])
     overlay_text(verdict_card, f"FINAL DECISION: {decision}", (50, 200), 2.0, (255,255,255))
     overlay_text(verdict_card, reason, (50, 250), 0.9, (200,200,200))
     for _ in range(20):
         out.write(verdict_card)
 
-    # Real-time with annotations
-    for frame in frames:
-        annotated = frame.copy()
-        overlay_annotations(annotated, analysis)
-        out.write(annotated)
-
-    # Slow-motion replay
-    for i in range(0, len(frames), 2):
-        annotated = frames[i].copy()
-        overlay_annotations(annotated, analysis)
-        out.write(annotated)
+    # Dynamic replay
+    for t, f in enumerate(frames):
+        annotated = f.copy()
+        overlay_annotations_dynamic(annotated, analysis, t)
         out.write(annotated)
 
-    # End frame 0.5s static
+    # Slow-motion: emphasize final 3 points
+    for pt in analysis['trajectory_curve'][-3:]:
+        for _ in range(2):
+            f = frames[-1].copy()
+            cv2.circle(f, pt, 10, (0,255,0) if analysis['trajectory']=="hitting" else (0,0,255), -1)
+            cv2.rectangle(f, analysis['stump_zone'][0], analysis['stump_zone'][1], (180,180,180), 2)
+            overlay_text(f, "Trajectory End", (pt[0] + 10, pt[1] - 10), 0.8, (255,255,255))
+            overlay_text(f, "ICC LBW Review • Third-Umpire Analysis", (30, h - 30), 0.6, (180,180,180))
+            out.write(f)
+
+    # End card (0.5s)
     for _ in range(10):
         out.write(verdict_card)