R/plot_ridge_tag.R

In utah-osa/hcctools2: OSA's Health Care Compare Tool package

#' plot_ridge_tag
#'
#' @param df  a dataframe of bundled procedures
#' @param tag a tag of interest
#' @param resp the response that we want to compare the tag against. the default is tp_med
#'
#' @return
#' @export
#'
#' @examples
#' #cscopy %>% plot_ridge_tag(tag="surg_bun_t_biopsy")
plot_ridge_tag <- function(df, tag, resp = "tp_med") {

  max_lim <- df %>% dplyr::pull(tp_med) %>% max()

  tag_type <-
    stringr::str_extract(tag, "_[[:alpha:]]+$") %>%
    stringr::str_replace("_", "")

  ylabel <- paste0("Bundle has '", tag_type, "' tag?")

  xlabel <- "Median Bundle Price"

  temp_title <-
    paste0("Ridge Plot: Bundle with '", tag_type, "' tag")

  # scale_fill_values <- c("TRUE" = "#003f5c",
  #                        "FALSE" = "#ffa600")


  df %>%
    #filter(tp_med < 35000) %>%
    ggplot2::ggplot(data = .,
                    ggplot2::aes_(x = as.name(resp), y = as.name(tag))) +
    ggridges::stat_density_ridges(
      ggplot2::aes_(fill = as.name(tag), alpha = .8),
      # stat = "binline",
      # bins = 20,
      # scale = 0.95,,
      # draw_baseline = FALSE
      quantile_lines = TRUE,
      quantiles = 2
    ) +
    ggplot2::geom_label(
      data = df %>%
        dplyr::group_by(!!sym(tag)) %>%
        dplyr::summarise(!!sym(resp) := median(!!sym(resp))),
      ggplot2::aes_(label = as.name(resp), #label = dollar(as.name(resp)),
                    color = as.name(tag)),
      size = 3.5,
      show.legend = FALSE
    ) +
    ggplot2::scale_x_continuous(labels = scales::dollar_format(prefix = "$")) +
    ggplot2::labs(title = temp_title,
                  fill = paste0("Bundle has '", tag_type, "'")) +
    ggplot2:: xlab(xlabel) +
    ggplot2:: ylab(ylabel) +
    #theme(legend.title = element_blank())+
    ggplot2::scale_y_discrete(labels = c("TRUE" = "Yes",
                                         "FALSE" = "No")) +
    ggplot2::scale_fill_manual(
      values = cust_color_pal2,
      labels = c("TRUE" = "Yes",
                 "FALSE" = "No"),
      guide = ggplot2::guide_legend(reverse = TRUE)
    ) +
    ggplot2::scale_color_manual(values = cust_color_pal2) +
    ggplot2::xlim(0, max_lim)

}