R/process_data.r

In Display-Lab/goals-of-care: VA Goals of Care Conversations Feedback

#' @title Process Data
#' @description Process input data frame of clc or hbpc data
#' @param df dataframe of CLC or HBPC data
#' @param envir Environment with constants for processing df.
#' @return list of dataframes: rate and category data frames.
#' @import dplyr
#' @importFrom lubridate make_date 
process_data <- function(df, envir){
  cat(paste("\n\n--- Processing", environmentName(envir), "Data\n"))
  # Check for valid inputs
  valid_input <- check_input(df, envir$COL_NAMES)
  
  if(!valid_input){
    print("Aborting: Problem with input data.\n")
    return(list())
  }
  
  # Report specific processing input
  if(environmentName(envir) == 'hbpc'){
    df_filtered <- filter_hbpc_data(df) 
  }
  
  if(environmentName(envir) == 'clc'){
    df_filtered <- filter_clc_data(df) %>% process_clc_data()
  }
  
  if(environmentName(envir) == 'dementia'){
    df_filtered <- filter_dementia_data(df) %>% process_dementia_data()
  }
  
  # Common processing
  rate_df <- calc_rate_sums(df_filtered, envir$ID_COLS, envir$NUMER_COLS, envir$DENOM_COLS, envir$GROUP_COLS)
  category_df <- calc_category_sums(df_filtered, envir$ID_COLS, envir$CATEGORIES, envir$GROUP_COLS)
  
  return(list(rate=rate_df, category=category_df))
}

#' @title Process CLC Data
#' @description Data processing specific to CLC input data
#' @import dplyr
process_clc_data <- function(.data){
  .data %>% 
      mutate(trtsp_1 = recode(trtsp_1, 
                       "Long-Term NH Care"="Long-Term Care Residents", 
                       "Short-Term NH Care"="Short-Stay Patients")
      )
}

#' @title Process Dementia Data
#' @description Data processing specific to Dementia input data
#' @import dplyr
process_dementia_data <- function(.data){
  .data %>% 
      mutate(trtsp_1 = recode(trtsp_1, 
                       "Long-Term NH Care"="Long-Term Care Residents", 
                       "Short-Term NH Care"="Short-Stay Patients")
      )
}