R/gg_theme.R

In fkeck/editheme: Color Palettes and Themes for RStudio

#' Theme and color scales for \code{ggplot2}
#'
#' @inheritParams ggplot2::theme_grey
#' @param theme A string corresponding to one of the 31 themes available in RStudio (see \code{\link{list_pal}}).
#' If not specified, the function tries to retrieve the active theme using \pkg{rstudioapi}.
#'
#' @export
#' @rdname ggplot_theme
#'
#' @examples
#'
#' require(ggplot2)
#'
#'\dontrun{
#' ggplot(iris, aes(Sepal.Length, Petal.Length, color = Species)) +
#'   geom_point() +
#'   labs(title = "Edgar Anderson's Iris Data",
#'   subtitle = "This is a demo", caption = "This is a caption") +
#'   theme_editor() +
#'   scale_color_editor()
#'
#' ggplot(iris, aes(Sepal.Length, Petal.Length)) +
#'   geom_point(color = col_fg(fade = 0.2)) +
#'   geom_smooth(color = get_pal()[1], fill = get_pal()[2]) +
#'   labs(title = "Edgar Anderson's Iris Data",
#'   subtitle = "This is a demo", caption = "This is a caption") +
#'   theme_editor()
#'
#' ggplot(iris, aes(Sepal.Length, Petal.Length, color = Species)) +
#'   geom_point() +
#'   labs(title = "Facet wrap") +
#'   facet_wrap(~Species) +
#'   theme_editor() +
#'   scale_color_editor()
#'}
#'
#' for(i in list_pal()){
#'  p <- ggplot(iris, aes(Sepal.Length, Petal.Length)) +
#'   geom_point(color = col_fg(i, fade = 0.2)) +
#'   geom_smooth(color = get_pal(theme = i)[1], fill = get_pal(theme = i)[2]) +
#'   labs(title = i, caption = "A caption") +
#'   theme_editor(theme = i) +
#'   scale_fill_editor(theme = i)
#'   print(p)
#' }
#'
theme_editor <- function(theme = NA, base_size = 10, base_family = "sans"){

  pal <- get_pal(theme)
  col_bg <- col_bg(theme)
  col_fg <- col_fg(theme)

  theme_classic(base_family = base_family, base_size = base_size) +
    theme(plot.background = element_rect(fill = col_bg, colour = NA),
          panel.background = element_rect(fill = col_bg, colour = NA),
          axis.line = element_line(colour = col_fg),
          axis.ticks = element_line(colour = col_fg),
          axis.text = element_text(colour = col_fg),
          title = element_text(colour = col_fg),
          legend.background = element_rect(fill = col_bg, color = NA),
          legend.key = element_rect(fill = col_bg, colour = NA, linetype = 0),
          legend.box.background = element_rect(fill = col_bg, colour = NA),
          legend.text = element_text(colour = col_fg),
          strip.background = element_rect(fill = col_fg(theme, fade = 0.2),
                                          colour = col_fg(theme, fade = 0.1)),
          strip.text = element_text(colour = col_bg)
          )
}

#' @inheritParams ggplot2::scale_colour_hue
#' @export
#' @rdname ggplot_theme
scale_fill_editor <- function(theme = NA, ...) {
  discrete_scale("fill", "editor", scales::manual_pal(unname(get_pal(theme = theme))), ...)
}

#' @export
#' @rdname ggplot_theme
scale_colour_editor <- function(theme = NA, ...) {
  discrete_scale("colour", "editor", scales::manual_pal(unname(get_pal(theme = theme))), ...)
}

#' @export
#' @rdname ggplot_theme
scale_color_editor <- scale_colour_editor