R/mpc_tidy.R

In malcalakovalski/fim: Fiscal Impact Measure

#' Title
#'
#' @param path 
#' @param sheet 
#' @param range 
#'
#' @return
#' @export
#'
#' @examples
read_mpc_file <- function(path = 'data/forecast.xlsx', sheet = 'mpc', range = 'B2:N14'){
  readxl::read_excel(path = path, 
                     sheet = sheet, range = range) %>% 
    tidyr::pivot_longer(-variable,
                        names_to = 'lag') %>% 
    tidyr::pivot_wider(names_from = variable,
                       values_from = value) %>% 
    dplyr::select(-lag)
}



#' Title
#'
#' @param var 
#' @param data 
#'
#' @return
#' @export
#' @importFrom rlang enquo as_label 
#'
#' @examples
get_timing <- function(timing, var){
  
  var <- enquo(var)
  string <- as_label(var)
  
  args <- switch(as_label(var),
                 
                 # PURCHASES
                 purchases = 1,
                 real_federal_purchases = 1,
                 federal_purchases = 1,
                 federal_purchases_counterfactual = 1,
                 real_state_purchases = 1,
                 state_purchases = 1,
                 state_purchases_counterfactual = 1,
                 
                 
                 gross_consumption_grants = 1,
                 consumption_grants = 1,
                 consumption_grants_counterfactual = 1,
                 investment_grants = 1,
                 investment_grants_counterfactual = 1,
                 # TRANSFERS
                 
                 health_outlays = ,
                 federal_health_outlays = ,
                 state_health_outlays = ,
                 health_outlays_counterfactual = ,
                 federal_health_outlays_counterfactual = ,
                 state_health_outlays_counterfactual = timing$health_outlays,
                 
                 social_benefits = ,
                 federal_social_benefits = ,
                 state_social_benefits = ,
                 social_benefits_counterfactual =  ,
                 federal_social_benefits_counterfactual = ,
                 state_social_benefits_counterfactual =  timing$social_benefits,
                 
                 
                 subsidies = ,
                 federal_subsidies = ,
                 state_subsidies = ,
                 subsidies_counterfactual = ,
                 federal_subsidies_counterfactual = ,
                 state_subsidies_counterfactual = timing$subsidies,
                 
                 
                 ui = ,
                 federal_ui = ,
                 state_ui = ,
                 ui_counterfactual = ,
                 federal_ui_counterfactual = ,
                 state_ui_counterfactual = timing$ui,
                 
                 ui_arp = ,
                 federal_ui_arp = ,
                 state_ui_arp = ,
                 ui_arp_counterfactual = timing$ui_arp,
                 federal_ui_arp_counterfactual = ,
                 state_ui_arp_counterfactual = timing$ui_arp,
                 
                 
                 rebate_checks = timing$rebate_checks,
                 rebate_checks_arp = timing$rebate_checks_arp,
                 rebate_checks_counterfactual = timing$rebate_checks,
                 rebate_checks_arp_counterfactual = timing$rebate_checks_arp,
                 
                 federal_rebate_checks = timing$rebate_checks,
                 federal_rebate_checks_arp = timing$rebate_checks_arp,
                 federal_rebate_checks_counterfactual = timing$rebate_checks,
                 federal_rebate_checks_arp_counterfactual = timing$rebate_checks_arp,
                 
                 corporate_taxes = ,
                 federal_corporate_taxes = ,
                 state_corporate_taxes = ,
                 corporate_taxes_counterfactual = ,
                 federal_corporate_taxes_counterfactual = ,
                 state_corporate_taxes_counterfactual = timing$corporate_taxes,
                 
                 non_corporate_taxes = ,
                 federal_non_corporate_taxes = ,
                 state_non_corporate_taxes = ,
                 non_corporate_taxes_counterfactual = ,
                 federal_non_corporate_taxes_counterfactual = ,
                 state_non_corporate_taxes_counterfactual = timing$non_corporate_taxes,
                 
                 federal_aid_to_small_businesses_arp = timing$aid_to_small_businesses_arp,
                 federal_other_direct_aid_arp = timing$other_direct_aid_arp,
                 federal_other_vulnerable_arp = timing$other_vulnerable_arp,
                 federal_aid_to_small_businesses_arp_counterfactual = timing$aid_to_small_businesses_arp,
                 federal_other_direct_aid_arp_counterfactual = timing$other_direct_aid_arp,
                 federal_other_vulnerable_arp_counterfactual = timing$other_vulnerable_arp)
  
  return(args)
}

#' Title
#'
#' @param data 
#' @param timing_file 
#' @param vars 
#'
#' @return
#' @export
#'
#' @examples
mpc_tidy <-function(data, timing_file = NULL, vars){
  
  timing <- timing_file %||% read_mpc_file()
 
   
  mutate(data, 
         across({{vars}},
                ~ 400 * roll::roll_sum(.x - counterfactual(.x, deflator = consumption_deflator_growth), 
                                 weights = rev(get_timing(timing, .x)),
                                 width = length(get_timing(timing, .x)),
                                 online = FALSE,
                                 min_obs = 1) / lag(gdp),
                .names = '{.col}_consumption'))
  
  #rlang::eval_tidy(call, data = data)
}

mpc_alt <-function(timing_file = NULL, vars){
  
  timing <- timing_file %||% read_mpc_file()
  
  roll::roll_sum({{vars}}, weights = rev(get_timing(timing, {{vars}})),
                                       width = length(get_timing(timing, {{vars}})),
                                       online = FALSE,
                                       min_obs = 1)
}