R/visualization_files.R

In fdrennan/test: What the Package Does (One Line, Title Case)

#' pre_modeling
#' @export
pre_modeling <- function(input_data, baseline) {
  analysis_data <- data_clean(input_data)
  times <- unique(analysis_data$Time)[
    order(as.numeric(gsub("[A-z]| ", "", unique(analysis_data$Time))))
  ]
  analysis_data <- analysis_data %>%
    mutate(
      Time = factor(Time, levels = times),
      TreatmentNew = factor(TreatmentNew,
        levels = c(
          "Wild Type", "Negative Control", "Positve Control",
          "Other Comparator", "Vehicle",
          grep("Dose", levels(TreatmentNew), value = TRUE)
        )
      ),
      TreatmentNew = droplevels(TreatmentNew)
    )
  ready_final_model <- transform_diagnostics(analysis_data, baseline)
  transformed_data <-
    variance_test_basic(
      transformed_data = ready_final_model$transformed_data,
      variable = ready_final_model$variable
    )
  ready_final_model$transformed_data <- transformed_data
  return(ready_final_model)
}

#' quickplot
#' @export
quickplot <- function(ready_final_model) {
  transformed_data <- ready_final_model$transformed_data
  if (all(is.na(transformed_data$Baseline))) {
    var <- "Response_Transformed"
    plots <- vizualization(
      transformed_data = transformed_data,
      power = power
    )
  } else {
    var <- "Response_Transformed_bc"
    transformed_data <- transformed_data %>%
      mutate(Response_Transformed_bc = Response_Transformed - Baseline)
    plots <- vizualization_cb(
      transformed_data = transformed_data,
      power = power
    )
  }
}