Update gully_drs_core/ball_detection.py

gully_drs_core/ball_detection.py

@@ -3,6 +3,12 @@ import cv2
 import numpy as np
 from .model_utils import load_model
 
+def find_bounce_point(path):
+    for i in range(1, len(path)-1):
+        if path[i-1][1] > path[i][1] < path[i+1][1]:  # y dips = bounce
+            return path[i]
+    return None
+
 def analyze_video(file_path):
     model = load_model()
     cap = cv2.VideoCapture(file_path)
@@ -21,7 +27,7 @@ def analyze_video(file_path):
         for r in results:
             for box in r.boxes:
                 cls = int(box.cls[0])
-                if cls == 32:  # Assuming class 32 = cricket ball
+                if cls == 32:  # cricket ball
                     x1, y1, x2, y2 = map(int, box.xyxy[0])
                     cx, cy = (x1 + x2) // 2, (y1 + y2) // 2
                     ball_path.append((cx, cy))
@@ -29,8 +35,22 @@ def analyze_video(file_path):
         frames.append(frame)
 
     cap.release()
+
+    bounce_point = find_bounce_point(ball_path)
+    impact_point = ball_path[-1] if ball_path else None
+
+    # Define stump zone (middle 1/3 width, bottom 100px)
+    stump_zone = (width // 2 - 30, height - 100, width // 2 + 30, height)
+
+    # Decision: is the last point in the stump zone?
+    decision = "OUT" if impact_point and stump_zone[0] <= impact_point[0] <= stump_zone[2] else "NOT OUT"
+
     return {
         "trajectory": ball_path,
         "fps": fps,
-        "frames": frames
+        "frames": frames,
+        "bounce_point": bounce_point,
+        "impact_point": impact_point,
+        "decision": decision,
+        "stump_zone": stump_zone
     }