R/scales.R

In lmgibson/cplthemes: Custom ggplot theme and table styling for the California Policy Lab

#' CPL [ggplot2] theme
#'
#' Applying CPL's color palettes as [ggplot2] defaults
#'
#' @export
#' @param palette Palette name.
func_cpl_palette <- function(palette = COLOR_SCHEMA) {
    palette_list <- cpl_palette

    types <- palette_list[[palette]]

    function(n) {
        types[[n]]
    }
}

#' Discrete color scale that aligns with CPL style
#'
#' @md
#' @param ... other arguments passed to \code{discrete_scale()}
#' @export
scale_color_discrete <- function(...) {
    ggplot2::discrete_scale("colour", "cpl", func_cpl_palette(COLOR_SCHEMA), ...)
}

#' Discrete color scale that aligns with CPL style
#'
#' @md
#' @param ... other arguments passed to \code{discrete_scale()}
#' @export
scale_colour_discrete <- function(...) {
    ggplot2::discrete_scale("colour", "cpl", func_cpl_palette(COLOR_SCHEMA), ...)
}

#' Discrete color scale that aligns with CPL style
#'
#' @md
#' @param ... other arguments passed to \code{discrete_scale()}
#' @export
scale_fill_discrete <- function(...) {
    ggplot2::discrete_scale("fill", "cpl", func_cpl_palette(COLOR_SCHEMA), ...)
}