R/opp_shot_chart.R

In KdotHudash/HuddyTest: NCAA Men's Basketball Play-By-Play Functionality

#' Plot all shots allowed by a team
#'
#' Plots team shot locations from one game or multiple games.
#'
#' @param game_ids Vector of ESPN game_ids
#' @param team Team Name
#' @param heatmap Use a density-type heatmap (Default = FALSE)
#' @export
opp_shot_chart <- function(game_ids, team, heatmap = F) {
  if(any(is.na(game_ids))) {
    error("game_ids missing with no default")
  }
  if(any(is.na(team))) {
    error("team missing with no default")
  }
  df <- get_shot_locs(game_ids)
  
  if(!is.null(df)) {
    side_one <- court %>% filter(side == 1)
    team_shots <- df %>% filter(!team_name %in% c(team, dict$ESPN_PBP[dict$ESPN == team]))
    
    ### flip shots if they are on the wrong side
    team_shots[team_shots$y > 47, "x"] <- 50 - team_shots[team_shots$y > 47, "x"]
    team_shots[team_shots$y > 47, "y"] <- 94 - team_shots[team_shots$y > 47, "y"]
    
    ### only pick one color
    color <- as.character(unique(team_shots$color))[1]
    
    if(heatmap){
      p1 <-
        ggplot2::ggplot() +
        ggplot2::stat_density_2d(data = team_shots,
                                 aes(x = x, y = y, fill = stat(density / max(density))),
                                 geom = "raster", contour = FALSE, interpolate = TRUE, n = 200) +
        ggplot2::geom_polygon(data = side_one, aes(x = x, y = y, group = group), col = "gray") +
        ggplot2::geom_point(alpha = 0.2, size = 1.5) +
        ggplot2::coord_equal() +
        ggplot2::xlab("") +
        ggplot2::ylab("")   +
        ggplot2::scale_fill_viridis_c("Shot Frequency",
                                      limits = c(0, 1),
                                      breaks = c(0, 1),
                                      labels = c("Lower", "Higher"),
                                      option = "plasma") +
        ggplot2::theme_void() +
        ggplot2::theme(
          axis.text.x = element_blank(),
          axis.text.y = element_blank(),
          axis.ticks.x = element_blank(),
          axis.ticks.y = element_blank(),
          axis.title = element_blank(),
          plot.title = element_text(size = 16, hjust = 0.5),
          plot.subtitle = element_text(size = 12, hjust = 0.5),
          plot.caption = element_text(size = 8, hjust = 0)) +
        ggplot2::labs(
          title = paste(team, "Shots Against"),
          shape = "Shot Outcome",
          caption = "Meyappan Subbaiah (@msubbaiah1) Data Accessed via ncaahoopR")
      return(p1)
    }
    
    p1 <-
      suppressMessages(ggplot2::ggplot() +
                         ggplot2::geom_point(data = team_shots,
                                             aes(
                                               x = x,
                                               y = y,
                                               shape = outcome),
                                             color = color,
                                             size = 3) +
                         ggplot2::geom_polygon(data = side_one, aes(x = x, y = y, group = group), col = "gray") +
                         ggplot2::geom_point(alpha = 0.2, size = 1.5) +
                         ggplot2::scale_color_manual(values = color) +
                         ggplot2:: xlab("") +
                         ggplot2::ylab("")  +
                         ggplot2::theme_void() +
                         ggplot2::theme(
                           axis.text.x = element_blank(),
                           axis.text.y = element_blank(),
                           axis.ticks.x = element_blank(),
                           axis.ticks.y = element_blank(),
                           axis.title = element_blank(),
                           plot.title = element_text(size = 16, hjust = 0.5),
                           plot.subtitle = element_text(size = 12, hjust = 0.5),
                           plot.caption = element_text(size = 8, hjust = 0),
                           plot.background = element_rect(fill = 'cornsilk')) +
                         ggplot2::labs(
                           title = paste0(team, " Shots Against"),
                           shape = "Shot Outcome",
                           caption = "Meyappan Subbaiah (@msubbaiah1) Data Accessed via ncaahoopR"))
    return(p1)
  }
}