R/stacked_bar_chart.R

In GenderInfer: This is a Collection of Functions to Analyse Gender Differences

#' Create a stacked bar chart with significance bars to compare with the
#' female baseline for gender analysis.
#' @name stacked_bar_chart
#' @param data_df, is the output dataframe from \code{\link{percent_df}}
#' @param baseline_female, female baseline in percentage from \code{\link{baseline}}
#' @param x_label, label for x axis
#' @param y_label, label for y axis
#' @param baseline_label, label used to define the baseline name.
#' @return This function create a bar chart containing the percentage of
#' submission with the corresponding baseline.
#' @importFrom ggplot2 geom_point
#' @importFrom ggplot2 geom_line
#' @importFrom ggplot2 scale_y_continuous
#' @importFrom ggplot2 sec_axis
#' @importFrom ggplot2 geom_hline
#' @importFrom ggplot2 scale_color_manual
#' @export

stacked_bar_chart <- function(data_df, baseline_female, x_label,
                              y_label, baseline_label) {

  labels <- x_values <- y_values <- gender <- pos <- NULL
  lower_CI <- upper_CI <- NULL
  data_df$gender <- factor(gsub("_percentage", "", data_df$gender),
                           levels = c("male", "female"))
  data_df$pos <- ifelse(data_df$gender == "male", 90, data_df$y_values)
  data_df$labels <- paste(data_df$y_values, "%")
  data_df$labels <- ifelse(data_df$gender == "male", data_df$significance,
                            data_df$labels)
  plot <- ggplot() +
    geom_bar(aes(x = x_values, y = y_values, fill = gender),
             data = {{data_df}}, stat = "identity") +
    scale_fill_manual(values = c(alpha("#2A7886", 0.7), alpha("#512B58", .7)),
                    labels = c("Male", "Female"), name = "") +
    scale_x_discrete(limits = rev(levels(droplevels(data_df$x_values)))) +
    geom_text(data = {{data_df}}, aes(x = x_values, y = pos,
                                     label = labels),
              size = 18 / .pt, vjust = 0, nudge_y = 0.5) +
    theme(legend.position = "bottom", legend.direction = "horizontal") +
    geom_line() +
    geom_hline(aes(yintercept = {{baseline_female}},
                   color = paste({{baseline_label}},
                                 {{baseline_female}}, "%")))  +
    scale_color_manual(values = alpha("#D7191C", .7), name = "") +
    geom_errorbar(data = {{data_df}}, aes(x = x_values,
                                          ymin = lower_CI,
                                          ymax = upper_CI), width = 0.3) +
    xlab({{x_label}}) + ylab({{y_label}}) + theme_gd()
  plot
}