R/scale_fill_wfa.R

In ciantudur/wfaplot: Create Graphics the WFA Way

#' WFA Fill Scale
#'
#' This function applies an WFA-themed fill scale to a ggplot object.
#' @param palette Select palette to use. Defaults to "main".
#' @param discrete Defaults to TRUE. Set to FALSE for continuous variable.
#' @param reverse Defaults to FALSE. Set to TRUE to reverse the order of the
#' fill scale.
#' @keywords
#' fill
#' scale
#' @import
#' ggplot2
#' @export
#' @examples
#' scale_fill_wfa("fuchsia")
#' scale_fill_wfa(palette = "main", discrete = "TRUE", reverse = "FALSE")
scale_fill_wfa <-
  function(palette = "main", discrete = TRUE,
           reverse = FALSE) {
    pal <- wfa_pal(palette = palette, reverse = reverse)

    if (discrete) {
      ggplot2::discrete_scale("fill", paste0("wfa_", palette), palette = pal)
    } else {
      ggplot2::scale_fill_gradientn(colours = pal(256))
    }
  }