R/partial_residual_helper.R

In dustinfife/flexplot: Graphically Based Data Analysis Using 'flexplot'

### ggplot seems to be converting character to factors, then converting factors to numbers.
### so a 0/1 becomes a 1/2
# identify those variables that are ordered factors in ggplot, but numeric IRL
# then convert
convert_ordered_factors = function(plot_data, data) {
  ggplot_classes = lapply(plot_data, is.ordered) %>% unlist
  data_classes   = lapply(data[,names(plot_data)], is.ordered) %>% unlist
  ordered_mismatch = ggplot_classes & !data_classes
  if (!any(ordered_mismatch)) return(data)
  
  variables_to_convert = names(which(ordered_mismatch))
  data[[variables_to_convert]] = factor(data[[variables_to_convert]], ordered=T) %>% as.numeric
  return(data)
}

summary_function_depending_on_class = function(x) {
  if (is.numeric(x)) return(mean(x))
  return(x[1])
}

identify_residualized_variables = function(lm_formula, added_term) {
  all_variables_in_model =                  paste0(lm_formula)[3] %>% strsplit("+", fixed=T) %>% unlist %>% trimws
  added_term_variables   = find_predictors_in_formula(added_term) %>% strsplit("+", fixed=T) %>% unlist %>% trimws
  residualized_variables = all_variables_in_model[!(all_variables_in_model %in% added_term_variables)]
  y_label = paste0(paste0(lm_formula)[2], " | ", paste0(residualized_variables, collapse = " + "))
  return(y_label)
}

find_predictors_in_formula = function(formula) {
  string_of_formula = paste0(formula)
  if (length(string_of_formula)==3) return(string_of_formula[3])
  return(string_of_formula[2])
}