R/get_bar_labels_resp.R

In cabaceo/shinymod: Reusable Shiny Modules and Helper Functions

#' @title Helper function for \code{mk_barplot_resp} and \code{mk_barploth_resp}.
#' Not for external use.
#'
#' @description \code{get_bar_labels_resp} takes in the original data frame, and
#' returns a data frame of labels and their postions for \code{mk_barplot_resp}
#' and \code{mk_barploth_resp}.
#'
#' @param df Data frame of the original data where bar charts are made from.
#' @param gp String. Name of a categorical variable to group the response var.
#' @param resp String. Name of a continuous variable as the response var.
#' @param fillby String. Name of a second categorical variable for sub-dividing
#'               and coloring the bars.
#' @return A data frame of bar labels and their positions.
#' @seealso \code{\link{mk_barplot_resp}} and \code{\link{mk_barploth_resp}}.
get_bar_labels_resp = function(df, gp, resp, fillby) {
        gp_symb = as.name(gp)
        resp_symb = as.name(resp)
        fillby_symb = as.name(fillby)

        if (fillby == gp) {
                stop("fillby variable can't be the same as grouping var!")
        } else if (fillby == "1") {
                df %>% group_by(!!gp_symb) %>%
                        summarise(!!resp_symb := sum(!!resp_symb, na.rm = T)) %>%
                        mutate(mid_pos = !!resp_symb / 2)
        } else {
                df %>% group_by(!!fillby_symb, !!gp_symb) %>%
                        summarise(!!resp_symb := sum(!!resp_symb, na.rm = T)) %>%
                        mutate(mid_pos = !!resp_symb / 2)
        }
}