R/observed_bias_order.R

In LucyMcGowan/tipr: Tipping Point Analyses

#' Order observed bias data frame for plotting
#'
#' @param d Observed bias data frame. Must have columns `dropped` and `type`
#' @param by Character. Variable in `d` to order by.
#'
#' @return Data frame in the correct order
#' @export
observed_bias_order <- function(d, by) {
  grps_ <- d[d$type == "group" & !grepl("Hypothetical", d$dropped), ]
  grps <- which(d$type == "group" & !grepl("Hypothetical", d$dropped))
  grps <- grps[order(grps_[[by]], decreasing = TRUE)]

  hypo_ <- d[d$type == "tip", ]
  hypo <- which(d$type == "tip")
  hypo <- hypo[order(hypo_[[by]])]

  d <- d[c(hypo, grps, order(d[[by]][d$type == "covariate"], decreasing = TRUE)), ]
  d$dropped <- factor(d$dropped,
                      levels = d$dropped)
  d
}