R/collapse_tree_and_model.R

In richelbilderbeek/pirouette: Create a Bayesian Posterior from a Phylogeny

#' Internal function
#'
#' Internal function to relevel the \code{tree_and_model}, so that
#' \link{pir_plot} has the legend labels in the right order
#' @inheritParams default_params_doc
#' @return a releveled \code{tree_and_model}
#' @author Richèl J.C. Bilderbeek
#' @export
collapse_tree_and_model <- function(tree_and_model) {
  # This is the right order
  target_levels <- c(
    "true_generative",
    "twin_generative",
    "true_candidate",
    "twin_candidate"
  )
  # Remove levels that are not present
  target_levels <- target_levels[target_levels %in% tree_and_model]
  generative_levels <- stringr::str_subset(target_levels, "generative")
  candidate_levels <- stringr::str_subset(target_levels, "candidate")

  inference_model <- forcats::fct_collapse(
    tree_and_model,
    generative = generative_levels,
    candidate = candidate_levels
  )
  inference_model
}