Update functions/pitch_summary_functions.py

functions/pitch_summary_functions.py

@@ -19,6 +19,8 @@ from matplotlib.offsetbox import OffsetImage, AnnotationBbox
 import matplotlib.pyplot as plt
 import matplotlib.gridspec as gridspec
 import PIL
+from stuff_model import calculate_arm_angles as caa
+
 
 
 ### PITCH COLOURS ###
@@ -394,6 +396,14 @@ def break_plot(df: pl.DataFrame, ax: plt.Axes):
     ax.set_xlim((-25, 25))
     ax.set_ylim((-25, 25))
 
+    df_aa = caa.calculate_arm_angles(df,df['pitcher_id'][0])['arm_angle']
+    # Plot average arm angle
+    mean_arm_angle = df_aa.mean()
+    x_end = 30
+    y_end = x_end * np.tan(np.radians(mean_arm_angle))
+    ax.plot([0, x_end], [0, y_end], color='grey', linestyle='--', linewidth=2,zorder=0,alpha=0.7)
+
+    
     # Add horizontal and vertical lines
     ax.hlines(y=0, xmin=-50, xmax=50, color=colour_palette[8], alpha=0.5, linestyles='--', zorder=1)
     ax.vlines(x=0, ymin=-50, ymax=50, color=colour_palette[8], alpha=0.5, linestyles='--', zorder=1)