R/comparison_plotter.R

In lubrunn/DSP_App_Abgabe: What the Package Does (One Line, Title Case)

#### here are the plot creations for covid and stocks from the comparison tabs



### this function plots the stock data according to user input
#'@export
#'@rdname comparison_plotter
stock_plotter <- function(df, input_metric, input_comp, input_roll){


    ### replace missing values in dow
 #
####### filter out relevant variable and companies
    df <- df %>% select(Dates, input_metric, name) %>%
      filter(
          grepl(paste(input_comp, collapse = "|"), name))
    ## convert to wide format for dygraph
    df <- df %>% tidyr::pivot_wider(names_from = name, values_from = input_metric)

    ### replace missing values in dow with first value they had (went public on 01/04/2019)
    if ("DOW" %in% names(df)){
      df[is.na(df[,"DOW"]), "DOW"] <- df[df$Dates == "2019-03-20", "DOW"]
    }



    ##### one df for values, one for dates
    df_values <- df %>% select(-Dates)
    df_dates <- as.Date(df$Dates)
    # set up time series
    don <- xts::xts(df_values, df_dates)

    ### plot
    dygraphs::dygraph(don,
                      ylab = input_metric,
                      group = "comp_plots",
                      main = glue::glue("{input_metric}")) %>%
      ### when 1 cimmpany selceted change label to company name
      {if (length(input_comp) == 1) dygraphs::dySeries(.,label = input_metric) else .} %>%
      #### when mutiple comapnies slecte and adj. close selcted scael the data
      {if (length(input_comp) > 1 & input_metric %in% c("Adj.Close", "log_Close")) dygraphs::dyRebase(.,value = 100) else . } %>%
      dygraphs::dyOptions(axisLineWidth = 2) %>%
      dygraphs::dyLegend() %>%
      dygraphs::dyShading(from = min(df_dates), to = max(df_dates), color = "white") %>%
      {if (input_roll == T) dygraphs::dyRoller(., rollPeriod = 7, showRoller = F) else .} #### smoothing



}




###### this function plots the covid data according to user input
#'@export
#'@rdname comparison_plotter
covid_plotter <- function(df, selected_metric, input_country, input_roll = F){

  ## select relevant variables
  df <- df %>% select(date, selected_metric, location)

  # widen df
  df <- df %>% pivot_wider(names_from = location, values_from = selected_metric)

  # covnert to date for xts
  df$date <- as.Date(df$date)
  # convert to xts
  don <- xts::xts(subset(df, select = input_country), df$date)


  ##### format selected metrics so it looks nicer
  selected_metric <- gsub("_", " ", selected_metric) %>% stringr::str_to_title()

  ###### set up dygraph
  dygraphs::dygraph(don,
                    ylab = selected_metric,
                    group = "comp_plots",
                    main = glue::glue("Control Variable")) %>%
    #### when mutiple countries selected change label shown on hover
   {if (length(input_country) > 1) dygraphs::dySeries(.) else dygraphs::dySeries(.,label = input_country)}  %>%
    dygraphs::dyOptions(axisLineWidth = 2) %>%
    dygraphs::dyLegend() %>%
    dygraphs::dyShading(from = min(df$date), to = max(df$date), color = "white") %>%
    #### when smoothin selected show moving averages
    {if (input_roll == T) dygraphs::dyRoller(., rollPeriod = 7, showRoller = F) else .}



  }


#'@rdname comparison_plotter
##### other controls plot
controls_plotter  <- function(df, input_controls_comp, input_ControlCountry, input_roll){

### put values into seperate df
  df_values <- df %>% select(-Date)


  ## convert to time series
  don <- xts::xts(df_values, as.Date(df$Date))

  ### plot
  dygraphs::dygraph(don,
                    ylab = input_controls_comp,
                    group = "comp_plots",
                    main = glue::glue("Control Variable")) %>%
    #### when mutiple countries selected change label shown on hover
    {if (length(input_ControlCountry) > 1) dygraphs::dySeries(.) else dygraphs::dySeries(.,label = input_ControlCountry)}  %>%
    dygraphs::dyOptions(axisLineWidth = 2) %>%
    dygraphs::dyLegend() %>%
    dygraphs::dyShading(from = min(df$Date), to = max(df$Date), color = "white") %>%
    #### when smoothin selected show moving averages
    {if (input_roll == T) dygraphs::dyRoller(., rollPeriod = 7, showRoller = F) else .}



}