Create visualization_utils.py

gully_drs_core/visualization_utils.py

@@ -0,0 +1,58 @@
+import plotly.graph_objects as go
+import numpy as np
+
+def plot_3d_trajectory(ball_path_3d, stump_zone):
+    # Extract X, Y, Z coordinates
+    x = [point[0] for point in ball_path_3d]
+    y = [point[1] for point in ball_path_3d]
+    z = [point[2] for point in ball_path_3d]
+    
+    # Create the 3D trajectory line
+    trajectory = go.Scatter3d(
+        x=x, y=y, z=z,
+        mode='lines+markers',
+        line=dict(color='limegreen', width=4),
+        marker=dict(size=5, color='orange'),
+        name='Ball Trajectory'
+    )
+    
+    # Define the virtual pitch (a flat plane at z=0)
+    pitch_x = [0, 640, 640, 0]
+    pitch_y = [0, 0, 480, 480]
+    pitch_z = [0, 0, 0, 0]
+    pitch = go.Mesh3d(
+        x=pitch_x, y=pitch_y, z=pitch_z,
+        color='green', opacity=0.5,
+        name='Pitch'
+    )
+    
+    # Define the stumps (a simple vertical rectangle)
+    stump_x1, stump_y1, stump_x2, stump_y2 = stump_zone
+    stumps = go.Mesh3d(
+        x=[stump_x1, stump_x2, stump_x2, stump_x1],
+        y=[stump_y1, stump_y1, stump_y2, stump_y2],
+        z=[0, 0, 50, 50],  # Stumps height
+        color='white', opacity=0.8,
+        name='Stumps'
+    )
+    
+    # Combine all elements into a 3D plot
+    fig = go.Figure(data=[trajectory, pitch, stumps])
+    
+    # Customize the layout
+    fig.update_layout(
+        scene=dict(
+            xaxis_title='X (Width)',
+            yaxis_title='Y (Height)',
+            zaxis_title='Z (Bounce)',
+            xaxis=dict(range=[0, 640]),
+            yaxis=dict(range=[0, 480]),
+            zaxis=dict(range=[0, 60]),
+            aspectmode='manual',
+            aspectratio=dict(x=1, y=0.75, z=0.2)  # Mimic cricket pitch proportions
+        ),
+        title="3D Ball Trajectory Analysis",
+        showlegend=True
+    )
+    
+    return fig