R/brookings_save.R

In BrookingsInstitution/ggbrookings: Grammar of Graphics in Brookings Style

#' Save ggplot2 plots in standard Brookings sizes
#'
#' This is a function to save standardized images for Brookings publications.
#' For more functionality like custom widths and heights, use ggsave() directly.
#'
#' @param filename File name to create on disk.
#' @param size Size of plot. Must be one of "small" (3.25 x 2 inches),
#' "medium" (6.5 x 4 inches), or "large" (9 x 6.5 inches).
#' @param dpi Plot resolution. Also accepts a string input: "retina" (320),
#' "print" (300), or "screen" (72). Applies only to raster output types.
#' @param height Adjust height manually
#' @param plot Plot to save, defaults to last plot displayed.
#' @param ... Pass additional arguments to ggplot2::ggsave()
#'
#' @examples
#'
#' \dontrun{
#' library(tidyverse)
#' library(ggbrookings)
#'
#' theme_set(theme_brookings())
#'
#' cars %>%
#'   ggplot(aes(speed, dist)) +
#'   geom_point()
#'
#' brookings_save("test.png")
#' }
#'
#' @export
brookings_save <- function(filename,

                      size = "medium",
                      dpi = 300,
                      height = NULL,
                      plot = ggplot2::last_plot(),
                      ...) {

  stopifnot(is.character(filename))

  if (!size %in% c("small", "medium", "large")) {

    stop("Error: size must be one of 'small', 'medium', or 'large'")

  }

  if (is.null(height)){
    sizes <- list(
      small = c(width = 3.12, height = 2),
      medium = c(width = 6.25, height = 4),
      large = c(width = 8.75, height = 6.5)
    )
  } else {
    sizes <- list(
      small = c(width = 3.12, height = height),
      medium = c(width = 6.25, height = height),
      large = c(width = 8.75, height = height)
    )
  }

  selected_size <- sizes[[size]]

  ggplot2::ggsave(
    filename = filename,
    plot = plot,
    width = selected_size["width"],
    height = selected_size["height"],
    ...
  )

}