R/plot_summary.R

In hs97/WilliamsGrad: Cultural Origin and Academic Achievement

#' This function offers users various ways to plot their results
#'
#'  @param x a dataframe to be plotted
#'  @param type indicates which type of summary is desired,
#'  currently there are "count" and "ratio"
#'
#'  @return a plot
#'  @import ggplot2
#'  @export

plot_summary <- function(x, type) {

  stopifnot(type %in% c("count", "ratio"))
  honor_names <- c(
    "cum" = "Cum Laude",
    "magna" = "Magna",
    "summa" = "Summa",
    "phi_beta_kappa" = "PBK",
    "sigma_xi" = "Sigma Xi"
  )

  if(type == "count") {
    x %>%
      filter(value != "None")  %>%
      filter(origin != "non-Asian") %>%
      ggplot(aes(x = year, group = origin, fill = origin)) +
      theme_bw(base_size = 16) +
      facet_grid(value~origin, scales = "free", labeller = labeller(value = honor_names)) +
      geom_bar() +
      theme(legend.position = "bottom",
            plot.margin = margin(5, 25, 0, 25))
  }else if (type == "ratio") {
    x %>%
      summarize_ratio %>%
      filter(value != "None")  %>%
      mutate(set = "Ratio Indicator") %>%
      ggplot(aes(x = year, y = ratio, group = origin, fill = origin)) +
      theme_bw(base_size = 16) +
      facet_grid(value~origin, scales = "free", labeller = labeller(value = honor_names)) +
      geom_bar(stat = "identity") +
      theme(legend.position = "bottom",
            plot.margin = margin(5, 25, 0, 25))
  }
 }