R/likert.R

In brilstl/ggtheme.ois: helper function to add ois style to ggplot2

#' @title helper function to make ois likert scale in ggplot2
#' @import ggplot2 dplyr forcats
#' @importFrom rlang :=
#' @importFrom rlang quo_is_null
#' @param .data het dataframe wat wordt meegeven aan de plot
#' @param y_as de waarde die op de y-as worden getoond
#' @param fill de waarde waarmee de balken worden gevuld
#' @param facet de waarde die de groepen onderverdeelt in 'facets'
#' @export
likert_plot <- function(.data, y_as, fill, facet = NULL, ...){


  # check evaluation of vars ----


  fill <- dplyr::enquo(fill)
  y_as <- dplyr::enquo(y_as)
  facet <- dplyr::enquo(facet)



  # prepare data distribution ----


  if(rlang::quo_is_null(facet)){

  .data <- .data %>%
    dplyr::select({{y_as}}, {{fill}}) %>%
    dplyr::group_by({{y_as}}) %>%
    dplyr::mutate(n = n()) %>%
    dplyr::mutate(n = paste0(" (n = ", format(n, big.mark = ".", decimal.mark = ","), ")")) %>%
    dplyr::ungroup() %>%
    dplyr::mutate('{{y_as}}' := paste0({{y_as}}, " ", n),
                  '{{y_as}}' := as_factor({{y_as}})) %>%
    dplyr::ungroup() %>%
    dplyr::mutate('{{fill}}' := as_factor({{fill}}),
                  '{{fill}}' := fct_explicit_na({{fill}}, "niet ingevuld")) %>%
    dplyr::group_by({{y_as}}, {{fill}}) %>%
    dplyr::count() %>%
    dplyr::ungroup() %>%
    dplyr::group_by({{y_as}}) %>%
    dplyr::mutate(percent = n/sum(n) * 100) %>%
    dplyr::ungroup()


  }
  else{

    .data <- .data %>%
      dplyr::select({{y_as}}, {{fill}}, {{facet}}) %>%
      dplyr::group_by({{y_as}}, {{facet}}) %>%
      dplyr::mutate(n = n()) %>%
      dplyr::mutate(n = paste0(" (n = ", format(n, big.mark = ".", decimal.mark = ","), ")")) %>%
      dplyr::ungroup() %>%
      dplyr::mutate('{{y_as}}' := paste0({{y_as}}, " ", n),
                    '{{y_as}}' := as_factor({{y_as}})) %>%
      dplyr::ungroup() %>%
      dplyr::mutate('{{fill}}' := as_factor({{fill}}),
                    '{{fill}}' := fct_explicit_na({{fill}}, "niet ingevuld")) %>%
      dplyr::group_by({{y_as}}, {{fill}}, {{facet}}) %>%
      dplyr::count() %>%
      dplyr::ungroup() %>%
      dplyr::group_by({{y_as}}, {{facet}}) %>%
      dplyr::mutate(percent = n/sum(n) * 100) %>%
      dplyr::ungroup()


  }

  kleur <- ggtheme.ois::gray_shades(.data, {{fill}}, ...)

  # facet ----

  if(rlang::quo_is_null(facet)){

    facet_attach <- NULL

  }else{

    facet_attach <- ggplot2::facet_wrap(facets = {{facet}},
                                        scales = "free_y",
                                        ncol = 1)

  }

  # plot data ----

  .data %>%
    ggplot2::ggplot(aes(y = {{y_as}},
               x = percent, fill = {{fill}})) +
    ggplot2::scale_fill_manual(values = c(kleur)) +
    ggplot2::geom_bar(stat = "identity",
             position = ggplot2::position_stack(),
             width = 0.8) +
    ggplot2::guides(fill = ggplot2::guide_legend(byrow = TRUE)) +
    ggplot2::labs(x=NULL, y=NULL, fill = NULL) +
    ggplot2::scale_x_reverse(breaks = seq(0, 100, 20),
                             labels = c("100%","80", "60", "40", "20", "0"),
                             expand = c(0.051,0.01)) +
    ggplot2::theme_bw() +
    ggtheme.ois::theme_ois() +
    facet_attach


}