wsba-hockey 1.1.9__py3-none-any.whl → 1.2.0__py3-none-any.whl

wsba_hockey/evidence/weakside-breakout/wsba_nhl_apps/wsba_nhl_apps/heatmaps/rink_plot.py

@@ -1,245 +0,0 @@
-
-import numpy as np
-import plotly.graph_objects as go
-import io
-import base64
-import requests as rs
-from PIL import Image
-
-def rink(setting = "full", vertical = False):
-    '''
-    Function to plot rink in Plotly. Takes 2 arguments :
-
-    setting : full (default) for full ice, offense positive half of the ice, ozone positive quarter of ice, defense for negative half of the ice, dzone for negative quarter of the ice, and neutral for the neutral zone
-    vertical : True if you want a vertical rink, False (default) is for an horizontal rink
-
-    '''
-
-    def faceoff_circle(x, y, outer=True):
-        segments = []
-        theta = np.linspace(0, 2*np.pi, 300)
-        if outer:
-            # Outer circle
-            x_outer = x + 15*np.cos(theta)
-            y_outer = y + 15*np.sin(theta)
-            outer_circle = go.Scatter(x=x_outer, y=y_outer, mode='lines', line=dict(width=2, color='red'), showlegend=False, hoverinfo='skip')
-
-            segments.append(outer_circle)
-
-        # Inner circle
-        x_inner = x + np.cos(theta)
-        y_inner = y + np.sin(theta)
-        inner_circle = go.Scatter(x=x_inner, y=y_inner, mode='lines', fill='toself', fillcolor='rgba(255, 0, 0, 0.43)', line=dict(color='rgba(255, 0, 0, 1)', width=2), showlegend=False, hoverinfo='skip')
-        
-        segments.append(inner_circle)
-        
-        return segments  #segments
-    
-    fig = go.Figure()
-
-    if vertical :
-        setting_dict = {
-            "full" : [-101, 101],
-            "offense" : [0, 101],
-            "ozone" : [25, 101],
-            "defense" : [-101, 0],
-            "dzone" : [-101, -25],
-            "neutral" : [-25,25]
-        }
-        fig.update_layout(xaxis=dict(range=[-42.6, 42.6], showgrid=False, zeroline=False, showticklabels=False, constrain="domain"), yaxis=dict(range=setting_dict[setting], showgrid=False, zeroline=False, showticklabels=False, constrain="domain"),
-                        showlegend=False, autosize=True, template="plotly_white")
-        fig.update_yaxes(
-            scaleanchor="x",
-            scaleratio=1,
-        )
-        def goal_crease(flip=1):
-            x_seq = np.linspace(-4, 4, 100)
-            x_goal = np.concatenate(([-4], x_seq, [4]))
-            y_goal = flip * np.concatenate(([89], 83 + x_seq**2/4**2*1.5, [89]))
-            goal_crease = go.Scatter(x=x_goal, y=y_goal, fill='toself', fillcolor='rgba(173, 216, 230, 0.3)', line=dict(color='red'))
-            return goal_crease
-
-        # Outer circle
-        theta = np.linspace(0, 2*np.pi, 300)
-        x_outer = 15 * np.cos(theta)
-        y_outer = 15 * np.sin(theta)
-        fig.add_trace(go.Scatter(x=x_outer, y=y_outer, mode='lines', line=dict(color='royalblue', width=2), showlegend=False, hoverinfo='skip'))
-        # Inner circle
-        theta2 = np.linspace(np.pi/2, 3*np.pi/2, 300)
-        x_inner = 42.5 + 10 * np.cos(theta2)
-        y_inner = 10 * np.sin(theta2)
-        fig.add_trace(go.Scatter(x=x_inner, y=y_inner, mode='lines', line=dict(color='red', width=2), showlegend=False, hoverinfo='skip'))
-        # Rink boundaries
-        fig.add_shape(type='rect', xref='x', yref='y', x0=-42.5, y0=25, x1=42.5, y1=26, line=dict(color='royalblue', width=1), fillcolor='royalblue', opacity=1)
-        fig.add_shape(type='rect', xref='x', yref='y', x0=-42.5, y0=-25, x1=42.5, y1=-26, line=dict(color='royalblue', width=1), fillcolor='royalblue', opacity=1)
-        fig.add_shape(type='rect', xref='x', yref='y', x0=-42.5, y0=-0.5, x1=42.5, y1=0.5, line=dict(color='red', width=2), fillcolor='red')
-        
-        # Goal crease
-        fig.add_trace(goal_crease())
-        fig.add_trace(goal_crease(-1))
-        # Goal lines
-        goal_line_extreme = 42.5 - 28 + np.sqrt(28**2 - (28-11)**2)
-        fig.add_shape(type='line', xref='x', yref='y', x0=-goal_line_extreme, y0=89, x1=goal_line_extreme, y1=89, line=dict(color='red', width=2))
-        fig.add_shape(type='line', xref='x', yref='y', x0=-goal_line_extreme, y0=-89, x1=goal_line_extreme, y1=-89, line=dict(color='red', width=2))
-
-        # Faceoff circles
-        fig.add_traces(faceoff_circle(-22, 69))
-        fig.add_traces(faceoff_circle(22, 69))
-        fig.add_traces(faceoff_circle(-22, -69))
-        fig.add_traces(faceoff_circle(22, -69))
-        fig.add_traces(faceoff_circle(-22, -20, False))
-        fig.add_traces(faceoff_circle(22, -20, False))
-        fig.add_traces(faceoff_circle(-22, 20, False))
-        fig.add_traces(faceoff_circle(22, 20, False))
-        
-        # Sidelines
-        theta_lines = np.linspace(0, np.pi/2, 20)
-        x_lines1 = np.concatenate(([-42.5], -42.5 + 28 - 28*np.cos(theta_lines), 42.5 - 28 + 28*np.cos(np.flip(theta_lines))))
-        y_lines1 = np.concatenate(([15], 72 + 28*np.sin(theta_lines), 72 + 28*np.sin(np.flip(theta_lines))))
-        x_lines2 = np.concatenate(([-42.5], -42.5 + 28 - 28*np.cos(theta_lines), 42.5 - 28 + 28*np.cos(np.flip(theta_lines))))
-        y_lines2 = np.concatenate(([15], -72 - 28*np.sin(theta_lines), -72 - 28*np.sin(np.flip(theta_lines))))
-        fig.add_trace(go.Scatter(x=x_lines1, y=y_lines1, mode='lines', line=dict(color='white', width=2), showlegend=False, hoverinfo='skip'))
-        fig.add_trace(go.Scatter(x=x_lines2, y=y_lines2, mode='lines', line=dict(color='white', width=2), showlegend=False, hoverinfo='skip'))
-        fig.add_shape(type='line', xref='x', yref='y', x0=42.5, y0=-72.5, x1=42.5, y1=72.5, line=dict(color='white', width=2))
-        fig.add_shape(type='line', xref='x', yref='y', x0=-42.5, y0=-72.5, x1=-42.5, y1=72.5, line=dict(color='white', width=2))
-        
-        # Add goals
-        goal_width = 6     # feet
-        goal_depth = 4     # feet
-
-        # Top goal
-        fig.add_shape(
-            type="rect",
-            xref="x",
-            yref="y",
-            x0=-goal_width / 2,
-            y0=89,
-            x1=goal_width / 2,
-            y1=89 + goal_depth,
-            line=dict(color="red", width=2),
-        )
-        # Bottom goal
-        fig.add_shape(
-            type="rect",
-            xref="x",
-            yref="y",
-            x0=-goal_width / 2,
-            y0=-89 - goal_depth,
-            x1=goal_width / 2,
-            y1=-89,
-            line=dict(color="red", width=2),
-        )
-
-    else : 
-        setting_dict = {
-            "full" : [-101, 101],
-            "offense" : [0, 101],
-            "ozone" : [25, 101],
-            "defense" : [-101, 0],
-            "dzone" : [-101, -25]
-        }
-        fig.update_layout(xaxis=dict(range=setting_dict[setting], showgrid=False, zeroline=False, showticklabels=False), yaxis=dict(range=[-42.6, 42.6], showgrid=False, zeroline=False, showticklabels=False, constrain="domain"),
-                        showlegend=True, autosize =True, template="plotly_white")
-        fig.update_yaxes(
-            scaleanchor="x",
-            scaleratio=1,
-        )
-        def goal_crease(flip=1):
-            y_seq = np.linspace(-4, 4, 100)
-            y_goal = np.concatenate(([-4], y_seq, [4]))
-            x_goal = flip * np.concatenate(([89], 83 + y_seq**2/4**2*1.5, [89]))
-            goal_crease = go.Scatter(x=x_goal, y=y_goal, fill='toself', fillcolor='rgba(173, 216, 230, 0.3)', line=dict(color='red'), showlegend=False, hoverinfo='skip')
-            return goal_crease
-        
-        # Outer circle
-        theta = np.linspace(0, 2 * np.pi, 300)
-        x_outer = 15 * np.sin(theta)
-        y_outer = 15 * np.cos(theta)
-        fig.add_trace(go.Scatter(x=x_outer, y=y_outer, mode='lines', line=dict(color='royalblue', width=2), showlegend=False, hoverinfo='skip'))
-        # Inner circle
-        theta2 = np.linspace(3 * np.pi / 2, np.pi / 2, 300)  # Update theta2 to rotate the plot by 180 degrees
-        x_inner = 10 * np.sin(theta2)  # Update x_inner to rotate the plot by 180 degrees
-        y_inner = -42.5 - 10 * np.cos(theta2)  # Update y_inner to rotate the plot by 180 degrees
-        fig.add_trace(go.Scatter(x=x_inner, y=y_inner, mode='lines', line=dict(color='red', width=2), showlegend=False, hoverinfo='skip'))
-        
-        # Rink boundaries
-        fig.add_shape(type='rect', xref='x', yref='y', x0=25, y0=-42.5, x1=26, y1=42.5, line=dict(color='royalblue', width=1), fillcolor='royalblue', opacity=1)
-        fig.add_shape(type='rect', xref='x', yref='y', x0=-25, y0=-42.5, x1=-26, y1=42.5, line=dict(color='royalblue', width=1), fillcolor='royalblue', opacity=1)
-        fig.add_shape(type='rect', xref='x', yref='y', x0=-0.5, y0=-42.5, x1=0.5, y1=42.5, line=dict(color='red', width=2), fillcolor='red')
-        # Goal crease
-        fig.add_trace(goal_crease())
-        fig.add_trace(goal_crease(-1))
-        # Goal lines
-        goal_line_extreme = 42.5 - 28 + np.sqrt(28 ** 2 - (28 - 11) ** 2)
-        fig.add_shape(type='line', xref='x', yref='y', x0=89, y0=-goal_line_extreme, x1=89, y1=goal_line_extreme, line=dict(color='red', width=2))
-        fig.add_shape(type='line', xref='x', yref='y', x0=-89, y0=-goal_line_extreme, x1=-89, y1=goal_line_extreme, line=dict(color='red', width=2))
-        # Faceoff circles
-        fig.add_traces(faceoff_circle(-69, -22))
-        fig.add_traces(faceoff_circle(-69, 22))
-        fig.add_traces(faceoff_circle(69, -22))
-        fig.add_traces(faceoff_circle(69, 22))
-        fig.add_traces(faceoff_circle(-20, -22, False))
-        fig.add_traces(faceoff_circle(-20, 22, False))
-        fig.add_traces(faceoff_circle(20, -22, False))
-        fig.add_traces(faceoff_circle(20, 22, False))
-
-        # Sidelines
-        theta_lines = np.linspace(0, np.pi / 2, 20)
-        x_lines1 = np.concatenate(([15], 72 + 28 * np.sin(theta_lines), 72 + 28 * np.sin(np.flip(theta_lines))))
-        y_lines1 = np.concatenate(([-42.5], -42.5 + 28 - 28 * np.cos(theta_lines), 42.5 - 28 + 28 * np.cos(np.flip(theta_lines))))
-        x_lines2 = np.concatenate(([15], -72 - 28 * np.sin(theta_lines), -72 - 28 * np.sin(np.flip(theta_lines))))
-        y_lines2 = np.concatenate(([-42.5], -42.5 + 28 - 28 * np.cos(theta_lines), 42.5 - 28 + 28 * np.cos(np.flip(theta_lines))))
-        fig.add_trace(go.Scatter(x=x_lines1, y=y_lines1, mode='lines', line=dict(color='white', width=2), showlegend=False, hoverinfo='skip'))
-        fig.add_trace(go.Scatter(x=x_lines2, y=y_lines2, mode='lines', line=dict(color='white', width=2), showlegend=False, hoverinfo='skip'))
-        fig.add_shape(type='line', xref='x', yref='y', x0=-72.5, y0=-42.5, x1=72.5, y1=-42.5, line=dict(color='white', width=2))
-        fig.add_shape(type='line', xref='x', yref='y', x0=-72.5, y0=42.5, x1=72.5, y1=42.5, line=dict(color='white', width=2))
-
-        # Add goals
-        goal_width = 6     # feet
-        goal_depth = 4     # feet
-
-        # Right goal
-        fig.add_shape(
-            type="rect",
-            xref="x",
-            yref="y",
-            x0=89,
-            y0=-goal_width / 2,
-            x1=89 + goal_depth,
-            y1=goal_width / 2,
-            line=dict(color="red", width=2),
-        )
-        # Left goal
-        fig.add_shape(
-            type="rect",
-            xref="x",
-            yref="y",
-            x0=-89 - goal_depth,
-            y0=-goal_width / 2,
-            x1=-89,
-            y1=goal_width / 2,
-            line=dict(color="red", width=2),
-        )
-
-    # Add logo
-    logo = Image.open(rs.get('https://weakside-breakout.s3.us-east-2.amazonaws.com/utils/wsba.png',stream=True).raw)
-
-    fig.add_layout_image(
-        dict(
-            source=logo,
-            xref="x",
-            yref="y",
-            x=-12,
-            y=12,
-            sizex=24,
-            sizey=24,
-            sizing="stretch",
-            opacity=1)
-        )
-    
-    #Set background to transparent
-    fig.update_layout(
-        paper_bgcolor="rgba(0,0,0,0)", 
-        plot_bgcolor="rgba(0,0,0,0)"
-    )
-    return fig

wsba_hockey/evidence/weakside-breakout/wsba_nhl_apps/wsba_nhl_apps/line-combos/app.py

@@ -1,245 +0,0 @@
-import pandas as pd
-import pyarrow.dataset as ds
-import plotly.express as px
-import plot as wsba_plt
-import numpy as np
-from urllib.parse import *
-from shiny import *
-from shinywidgets import output_widget, render_widget 
-
-app_ui = ui.page_fluid(
-    ui.tags.link(
-        rel='stylesheet',
-        href='https://fonts.googleapis.com/css2?family=Bebas+Neue&display=swap'
-    ),
-    ui.tags.style(
-        """
-        body {
-            background-color: #09090b;
-            color: white;
-            font-family: 'Bebas Neue', sans-serif;
-        }
-
-        .custom-input input.form-control,
-        .custom-input .selectize-control,
-        .custom-input .selectize-input {
-            background-color: #09090b !important;  /* black background */
-            color: white !important;               /* white font color */
-            border-radius: 4px;
-            border: 1px solid #444;
-        }
-
-        .custom-input .selectize-dropdown,
-        .custom-input .selectize-dropdown-content {
-            background-color: #09090b !important;
-            color: white !important;
-        }          
-
-        .custom-input .selectize-control.multi .item {
-            background-color: #09090b !important;
-            color: white !important;
-            border-radius: 4px;
-            padding: 2px 6px;
-            margin: 2px 4px 2px 0;
-        }
-
-        label.control-label {
-            color: white !important;
-        }   
-
-        .selectize-control.multi {
-            width: 300px !important;
-        }
-
-        .form-row {
-            display: flex;
-            gap: 12px;
-            flex-wrap: wrap;
-            justify-content: center;
-        }
-
-        .submit-button {
-            display: flex;
-            justify-content: center;
-        }
-
-        .hide {
-            display: none;
-        }
-
-        .table thead tr {
-            white-space: nowrap;
-            text-align: center;
-            color: white;
-            background-color: #09090b;
-        }
-
-        .table thead th {
-            white-space: nowrap;
-            text-align: center;
-            color: #09090b;
-        }
-
-        .table tbody tr {
-            --bs-table-bg: #09090b;
-            --bs-table-color-state: white;
-        }
-
-        .table tbody tr td {
-            white-space: nowrap;
-            text-align: center;
-            overflow: hidden;
-            text-overflow: ellipsis;
-            color: white;
-            background-color: #09090b;
-        }
-        """
-    ),
-    output_widget("line_combos"),
-    ui.output_data_frame("with_out")
-)
-
-def server(input, output, session):
-    queries = reactive.Value({})
-    team_data = reactive.Value(pd.DataFrame())
-    player_data = reactive.Value(pd.DataFrame())
-
-    col = [
-        'season','season_type','game_id','game_date',
-        'away_team_abbr','home_team_abbr','event_num','period',
-        'seconds_elapsed',"strength_state","strength_state_venue",
-        "event_type","description",
-        "penalty_duration",
-        "event_team_abbr","event_team_venue",
-        "x_adj","y_adj",
-        "event_distance","event_angle","event_length","seconds_since_last",
-        "away_on_1_id","away_on_2_id","away_on_3_id","away_on_4_id","away_on_5_id","away_on_6_id","away_goalie_id",
-        "home_on_1_id","home_on_2_id","home_on_3_id","home_on_4_id","home_on_5_id","home_on_6_id","home_goalie_id",
-        'rush','rebound','empty_net','xG'
-    ]
-
-    @output()
-    @render_widget
-    def line_combos():
-        #Retreive query parameters
-        search = session.input[".clientdata_url_search"]()
-        query = parse_qs(urlparse(search).query)
-        
-        print(query)
-        #If no input data is provided automatically provide a select skater and plot all 5v5 fenwick shots
-        defaults = {
-            'season':['20182019'],
-            'team':['BOS'],
-            'strength_state':['5v5'],
-            'season_type':['2'],
-            'skaters':['8473419,8470638']
-        }
-        
-        for key in defaults.keys():
-            if key not in query.keys():
-                query.update({key:defaults[key]})
-
-        #Iterate through query and parse params with multiple selections
-        for param in query.keys():
-            q_string = query[param][0]
-            query[param] = q_string.split(',')
-
-        print(query)
-        #Determine which season to load based on the input
-        season = query['season'][0]
-        queries.set(query)
-
-        #Load appropriate dataframe
-        dataset = ds.dataset(f's3://weakside-breakout/pbp/parquet/nhl_pbp_{season}.parquet', format='parquet')
-        filter_expr = ((ds.field('away_team_abbr') == query['team'][0]) | (ds.field('home_team_abbr') == query['team'][0])) & ((ds.field('season_type') == int(query['season_type'][0])))
-
-        table = dataset.to_table(columns=col,filter=filter_expr)
-        df = table.to_pandas()
-    
-        #Prepare dataframe for plotting based on URL parameters
-        team_data.set(df[(df['away_team_abbr']==query['team'][0]) | (df['home_team_abbr']==query['team'][0])])
-        player_data.set(wsba_plt.player_events(df,query['skaters']))
-
-        #Return empty rink if no data exists else continue
-        if df.empty:
-            return wsba_plt.wsba_rink()
-        else:
-            rink = wsba_plt.wsba_rink()
-
-            try:
-                for_plot = wsba_plt.heatmap(df,team=query['team'][0],skaters=query['skaters'],events=['missed-shot','shot-on-goal','goal'],strengths=query['strength_state'],onice='for')
-                against_plot = wsba_plt.heatmap(df,team=query['team'][0],skaters=query['skaters'],events=['missed-shot','shot-on-goal','goal'],strengths=query['strength_state'],onice='against')
-
-                for trace in for_plot[1].data:
-                    rink.add_trace(trace)
-                
-                for trace in against_plot[1].data:
-                    rink.add_trace(trace)
-
-                season = int(season[0:4])
-
-                return rink.add_annotation(
-                    text='Lower xG',
-                    xref="paper",
-                    yref="paper",
-                    xanchor='right',
-                    yanchor='top',
-                    font=dict(color='white'),
-                    x=0.3,
-                    y=0.04,
-                    showarrow=False
-                ).add_annotation(
-                    text='Higher xG',
-                    xref="paper",
-                    yref="paper",
-                    xanchor='right',
-                    yanchor='top',
-                    font=dict(color='white'),
-                    x=0.76,
-                    y=0.04,
-                    showarrow=False
-                )
-            except:
-                return wsba_plt.wsba_rink()
-    
-    @output()
-    @render.data_frame
-    def with_out():
-        query_load = queries.get()
-
-        ids = query_load['skaters']
-        team = query_load['team'][0]
-        team_pbp = team_data.get()
-        team_pbp['home_on_ice'] = team_pbp['home_on_1_id'].astype(str) + ";" + team_pbp['home_on_2_id'].astype(str) + ";" + team_pbp['home_on_3_id'].astype(str) + ";" + team_pbp['home_on_4_id'].astype(str) + ";" + team_pbp['home_on_5_id'].astype(str) + ";" + team_pbp['home_on_6_id'].astype(str)
-        team_pbp['away_on_ice'] = team_pbp['away_on_1_id'].astype(str) + ";" + team_pbp['away_on_2_id'].astype(str) + ";" + team_pbp['away_on_3_id'].astype(str) + ";" + team_pbp['away_on_4_id'].astype(str) + ";" + team_pbp['away_on_5_id'].astype(str) + ";" + team_pbp['away_on_6_id'].astype(str)
-
-        team_pbp['onice'] = team_pbp['away_on_ice']+';'+team_pbp['home_on_ice']
-
-        if len(ids)>2:
-            mask = ((team_pbp['onice'].str.contains(ids[0])) & (team_pbp['onice'].str.contains(ids[1])) & (team_pbp['onice'].str.contains(ids[2])))
-        else:
-            mask = ((team_pbp['onice'].str.contains(ids[0])) & (team_pbp['onice'].str.contains(ids[1])))
-    
-        team_pbp = team_pbp.loc[~mask]
-
-        if 'Other' in query_load['strength_state']:
-            strength_state = query_load['strength_state'] + team_pbp.loc[~(team_pbp['strength_state'].isin(['5v5','5v4','4v5'])),'strength_state'].drop_duplicates().to_list()
-        else:
-            strength_state = query_load['strength_state']
-        
-        with_stats = wsba_plt.calculate_stats(player_data.get(), query_load['team'][0], strength_state).replace({team:f'{team} With'})
-        without_stats = wsba_plt.calculate_stats(team_pbp, query_load['team'][0], strength_state).replace({team:f'{team} Without'})
-
-        total = pd.concat([with_stats, without_stats])[['Team',
-                                                       'TOI',
-                                                       'GF/60','GA/60',
-                                                       'SF/60','SA/60',
-                                                       'FF/60','FA/60',
-                                                       'xGF/60','xGA/60',
-                                                       'GF%','SF%',
-                                                       'FF%','xGF%',
-                                                       'GSAx']].round(2)
-        
-        return render.DataTable(total)
-
-app = App(app_ui, server)