Update functions/pitch_summary_functions.py

functions/pitch_summary_functions.py

@@ -19,6 +19,7 @@ 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 +395,17 @@ 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)
+
+    
+
     # 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)
@@ -401,7 +413,7 @@ def break_plot(df: pl.DataFrame, ax: plt.Axes):
     # Set axis labels and title
     ax.set_xlabel('Horizontal Break (in)', fontdict=font_properties_axes)
     ax.set_ylabel('Induced Vertical Break (in)', fontdict=font_properties_axes)
-    ax.set_title("Pitch Breaks", fontdict=font_properties_titles)
+    ax.set_title(f"Pitch Breaks - Arm Angle: {mean_arm_angle:.0f}°", fontdict=font_properties_titles)
 
     # Remove legend
     ax.get_legend().remove()