R/evol.R

In etiennebacher/tennistravel: tennistravel

#' Compute the evolution of distance per year for a specific player
#'
#' @param player Player name
#'
#' @return A tibble
#' @export
#'
evol_dist <- function(player) {
  
  tennis_data %>% 
    dplyr::filter(grepl(paste(player, collapse="|"), player_name)) %>% 
    dplyr::select(player_name, tourney_year, contains(c("_d", "_a"))) %>% 
    mutate(
      dist = custom_dist(lat_d, long_d, lat_a, long_a),
      tourney_year = factor(tourney_year)
    ) %>% 
    dplyr::select(-contains(c("_a", "_d"))) %>% 
    group_by(player_name, tourney_year) %>% 
    summarise(dist = sum(dist, na.rm = T)) %>% 
    ungroup() %>% 
    arrange(player_name, tourney_year)
  
}

#' Compute the evolution of footprint per year for a specific player
#'
#' @param player Player name
#' @param flightClass Flight class of the individual. Default is 'Unknown', as in package footprint.
#' @param output Unit of footprint. Default is CO2, as in package footprint.
#'
#' @return A tibble
#' @export
#'
evol_footprint <- function(player, flightClass = "Unknown", output = "co2e") {
  
  tennis_data %>% 
    dplyr::filter(grepl(paste(player, collapse="|"), player_name)) %>% 
    dplyr::select(player_name, tourney_year, contains(c("_d", "_a"))) %>% 
    mutate(
      footprint = custom_footprint(lat_d, long_d, lat_a, long_a,
                                   flightClass = flightClass,
                                   output = output),
      tourney_year = factor(tourney_year)
    ) %>% 
    dplyr::select(-contains(c("_a", "_d"))) %>% 
    group_by(player_name, tourney_year) %>% 
    summarise(footprint = sum(footprint, na.rm = T)) %>% 
    ungroup() %>% 
    arrange(player_name, tourney_year)
  
}


#' Plot the footprint or the distance made by one or several players
#'
#' @param player Player(s) name(s)
#' @param indicator "dist" or "footprint"
#'
#' @return A graph
#' @export
#'

plot_evol <- function(player, indicator) {
  
  if (indicator == "dist") {
    data_type <- evol_dist(player)
    plot_title <- paste0("Distance (in km)")
  } else if (indicator == "footprint") {
    data_type <- evol_footprint(player)
    plot_title <- paste0("Carbon footprint (in kg of CO2)")
  }
  
  data_type %>%
    tidyr::complete(player_name, tourney_year) %>% 
    group_by(player_name) %>% 
    arrange(indicator) %>% 
    e_charts(tourney_year, color = "yellow") %>%
    e_line_(indicator, color = "yellow") %>% 
    e_legend(show = FALSE) %>%
    e_title(
      text =  plot_title,
      textStyle = list(color = "#fff", font = "normal")
    ) %>% 
    e_tooltip(
      trigger = "axis"
    ) %>% 
    e_x_axis(axisLabel = list(color = "#fff")) %>% 
    e_y_axis(axisLabel = list(color = "#fff"))
  
}