R/plot_box.R

Defines functions plot_box

Documented in plot_box

#' A function to create standardized CCMH box plot's
#'
#' @name plot_box
#' @description Creates standardized CCMH box plot's using ggplot2.
#' @param data A data file containing all variables that will be used in the plot
#' @param x.var A quoted string or unquoted characters to indicate the variable that will be plotted on the x-axis. The variable class must be a character or factor.
#' @param y.var A quoted string or unquoted characters to indicate the variable that will be plotted on the y axis. The variable class must be numeric.
#' @param group.var A quoted string to indicate the variable that will group the boxes on the x-axis and specify the color of the boxes. The x-axis variable could used to specify colors of each individual boxes. By default, `""`.
#' @param color A hex code or list of hex codes that indicates the color of the grouped boxes. See Note 1 for more details. By default, `c('#E6AB02', '#1B9E77', '#666666', '#D95F02', '#6db6ff', '#b66dff', '#ff6db6', '#920000')`.
#' @param x.order.manual A list of quoted strings to indicate the order of the x axis. By default, `NULL`.
#' @param save A logical statement indicates whether the plot should be saved as a file under a local folder. If false, the plot will be returned as an object. By default, `FALSE`.
#' @param path A quoted string to indicate the file's pathway and name if `save = TRUE`. By default, `"plot.png"`.
#' @param plot.width A numeric value to indicate the plot's width. By default, `15.24`.
#' @param plot.height A numeric value to indicate the plot's height. By default, `12.7`.
#' @param plot.units A quoted string to indicate the plot's width and height size units. Options include `in`, `cm`, `mm`, and `px`. By default, `cm`.
#' @param plot.dpi A numeric value to indicate the plot's image resolution. By default, `320`.
#' @param plot.device A quoted string or function to indicate the plot's device. By default, `NULL`.
#' @param mean.sym A logical statement specifies whether a point should be displayed to indicate the mean. By default, `FALSE`.
#' @param mean.sym.shape A numeric value to indicate the mean point shape. By default, `3` or Crosshair.
#' @param mean.sym.size A numeric value to indicate the mean point size. By default, `2`.
#' @param mean.sym.color A hex code to indicate the mean point color. By default, `"#000000"` or black.
#' @param mean.sym.position A numeric value to indicate the mean point position. By default, `0.75`.
#' @param plot.title A quoted string to indicate the title of the plot. By default, `NULL`.
#' @param x.title A quoted string to indicate the x axis title. By default, `NULL`.
#' @param y.title A quoted string to indicate the y axis title. By default, `NULL`.
#' @param plot.title.size A numeric value to indicate the plot title text size. By default, `20`.
#' @param axis.title.size A numeric value to indicate the axis title text size. By default, `16`.
#' @param axis.label.size A numeric value to indicate the axis labels text size. By default, `14`.
#' @param text.font A quoted string to indicate the plot's text font. By default, `Avenir`.
#' @param y.label.type A quoted string or list to customize the y axis labels. Options include `"numeric"`, `"percent"`, `"comma"`, `"sci`, or `"dollar"` . By default, `numeric`.
#' @param y.min A numeric value to indicate the minimum number presented on the y axis. By default, `NULL`.
#' @param y.max A numeric value to indicate the maximum number presented on the y axis. By default, `NULL`.
#' @param y.breaks Numeric values or a sequence that indicates breaks on the y axis. By default, `NULL`.
#' @param x.wrap A numeric value to indicate the number of characters to wrap the x axis text. By default, `15`.
#' @param x.remove A logical statement to indicate whether x-axis title, labels, and ticks should be removed. By default, `FALSE`.
#' @param caption A logical statement to indicate whether the CCMH caption should be included in the plot. By default, `FALSE`.
#' @param caption.size A numeric value to indicate the caption text size. By default, `12`.
#' @param legend.position A quoted string or numeric vector to indicate the location of the legend. Options include `"left"`,`"top"`, `"right"`, `"bottom"`, `"none"`, or numeric vector c(x,y). By default, `"none"`.
#' @param legend.order.manual A list of quoted strings to indicate the order of the legend. By default, `NULL`.
#' @param legend.title A quoted string to indicate the legend title. By default, `NULL`.
#' @param legend.title.size A numeric value to indicate the legend title text size. By default, `16`.
#' @param legend.label.size A numeric value to indicate the legend label text size. By default, `14`.
#' @param coord.flip A logical statement to indicate if the plot's x and y axis should be flipped. By default, `FALSE`.
#' @param y.grid.major A logical statement to indicate if the y axis major grid lines should be added to the plot. By default, `FALSE`,
#' @param y.grid.major.color A hex code that indicates the color of the y axis major grid lines. By default, `"#d3d3d3"` or a shade of light grey.
#' @param y.grid.major.size A numeric value to indicate the thickness of the y axis major grid lines. By default, `0.5`.
#' @param y.grid.major.linetype A numeric value to indicate the line type of the y axis major grid. By default, `1`.
#' @param y.axis.line A logical statement to indicate if the y axis line should be included. By default, `TRUE`.
#' @param x.axis.line A logical statement to indicate if the x axis line should be included. By default, `TRUE`.
#' @param remove.axis.ticks A logical statement to indicate if the axis ticks should be excluded. By default, `FALSE`.
#' @param plot.element1 A ggplot plot function and arguments needed for the plot and specified as an object. By default, `NULL`.
#' @param plot.element2 A ggplot plot function and arguments needed for the plot and specified as an object. By default, `NULL`.
#' @param plot.element3 A ggplot plot function and arguments needed for the plot and specified as an object. By default, `NULL`.
#' @param plot.element4 A ggplot plot function and arguments needed for the plot and specified as an object. By default, `NULL`.
#' @param plot.element5 A ggplot plot function and arguments needed for the plot and specified as an object. By default, `NULL`.
#' @param plot.element6 A ggplot plot function and arguments needed for the plot and specified as an object. By default, `NULL`.
#' @param plot.element7 A ggplot plot function and arguments needed for the plot and specified as an object. By default, `NULL`.
#' @param plot.element8 A ggplot plot function and arguments needed for the plot and specified as an object. By default, `NULL`.
#' @param plot.element9 A ggplot plot function and arguments needed for the plot and specified as an object. By default, `NULL`.
#'
#' @note Note 1. The argument `color` specifies the color of items within the grouped variable. For example, the group variable items could be the name of automobile companies (i.e., Ford, Dodge, BMV). You can specify color randomly for each company using this list: `c("#21501b", "#c51329", "#074e67")`. Or you can specify color directly to each specific company using this list: `c("Ford" = "#21501b", "Dodge" = "#c51329", "BMV" = "#074e67")`. Default colors are color blind safe.
#' @note The `plot.element` arguments specify additional ggplot2 graphical element(s) needed to complete a specific box plot, but are not specified as an argument in the `box_plot` function. For example, the axis label text will always be black unless specified in one of the `plot.element` arguments. To change the axis label text color in one of the `plot.element` arguments, you would first create an object to represent graphing element (e.g., green.axis.text <- ggplot2::theme(axis.text = ggplot2::element_text(color = "green"))). Then the object would be added to one of the `plot.element` arguments (e.g., `plot.element01 = green.axis.text`).
#' @return A box plot returned as a object or saved under a local file.
#' @export
#'

plot_box <- function(data,
                     x.var,
                     y.var,
                     group.var = "",
                     color = c('#666666', '#1B9E77',
                               '#E6AB02', '#D95F02',
                               '#6db6ff', '#b66dff',
                               '#ff6db6', '#920000'),
                     x.order.manual = NULL,
                     save = FALSE,
                     path = "plot.png",
                     plot.width = 15.24,
                     plot.height = 12.7,
                     plot.units = "cm",
                     plot.dpi = 320,
                     plot.device = NULL,
                     mean.sym = FALSE,
                     mean.sym.shape = 3,
                     mean.sym.size = 2,
                     mean.sym.color = "#000000",
                     mean.sym.position = 0.75,
                     plot.title = "",
                     x.title = "",
                     y.title = "",
                     plot.title.size = 20,
                     axis.title.size = 16,
                     axis.label.size = 14,
                     text.font = "Avenir",
                     y.label.type = "numeric",
                     y.min = NULL,
                     y.max = NULL,
                     y.breaks = NULL,
                     x.wrap = 15,
                     x.remove = FALSE,
                     caption = FALSE,
                     caption.size = 12,
                     legend.position = "none",
                     legend.order.manual = NULL,
                     legend.title = NULL,
                     legend.title.size= 16,
                     legend.label.size= 14,
                     coord.flip = FALSE,
                     y.grid.major = FALSE,
                     y.grid.major.color = "#d3d3d3",
                     y.grid.major.size = 0.5,
                     y.grid.major.linetype = 1,
                     y.axis.line = TRUE,
                     x.axis.line = TRUE,
                     remove.axis.ticks = FALSE,
                     plot.element1 = NULL,
                     plot.element2 = NULL,
                     plot.element3 = NULL,
                     plot.element4 = NULL,
                     plot.element5 = NULL,
                     plot.element6 = NULL,
                     plot.element7 = NULL,
                     plot.element8 = NULL,
                     plot.element9 = NULL){

    # Specify the x-axis variable as a symbol
      x.var1 <- rlang::as_name(rlang::enquo(x.var))
      x.var1 <- rlang::sym(x.var1)

    # Specify the y-axis variable as a symbol
      y.var1 <- rlang::as_name(rlang::enquo(y.var))
      y.var1 <- rlang::sym(y.var1)

    # Specify grouping variable
    if(group.var != ""){

      # Specify the fill variable as a symbol
        group.var1 <- rlang::sym(group.var)

      # Rename grouping variable
        data$group77d8214 <-  data[[{{group.var1}}]]

     } else{

     }

  # Remove rownames
    rownames(data) <- NULL

  # Error message to indicate the x.order.manual items does not match the items in x-axis variable
    if(!is.null(x.order.manual)){

      x.var <- unique(data[[{{x.var1}}]])
      test1 <- ifelse(x.var %in% x.order.manual, "TRUE", "FALSE")
      test2 <- ifelse(x.order.manual %in% x.var, "TRUE", "FALSE")
      test.final <- c(test1, test2)
      test.final <- FALSE %in% test.final

      if(test.final == TRUE){

        stop("Unique characters listed x.order.manual must match the characters of x-axis variable")

      } else{

      }
    } else{

    }

  # Error message to indicate the legend.order.manual items does not match the items in the filler variable
    if(!is.null(legend.order.manual) &
       group.var != ""){

      group.varz <- unique(data$group77d8214)
      test1 <- ifelse(group.varz %in% legend.order.manual, "TRUE", "FALSE")
      test2 <- ifelse(legend.order.manual %in% group.varz, "TRUE", "FALSE")
      test.final <- c(test1, test2)
      test.final <- FALSE %in% test.final

      if(test.final == TRUE){

        stop("Unique characters listed legend.order.manual must match the characters of fill variable")

      } else{

      }
    } else{

    }

  # Specify custom x axis order
    if(!is.null(x.order.manual)){

      data[[{{x.var1}}]] <- factor(data[[{{x.var1}}]],
                                   levels = c(x.order.manual))

    } else{

    }

  # Specify if the legend should be reordered
    if(!is.null(legend.order.manual) &
       group.var != ""){

      data$group77d8214 <- factor(data$group77d8214,
                                  levels = legend.order.manual)

    } else{

    }

    # Specify Y Major Grid
      if(y.grid.major == TRUE){
        grid.maj <- ggplot2::element_line(color = y.grid.major.color,
                                          linewidth = y.grid.major.size,
                                          linetype = y.grid.major.linetype)
      } else {
        grid.maj <- ggplot2::element_blank()
      }

    # Specify Y breaks
      if(is.null(y.breaks)){
        y.breaks.c <- ggplot2::waiver()
      } else {
        y.breaks.c <- y.breaks
      }

    # Specify y axis lables
      if(y.label.type == "percent"){
        y.label.typea <- scales::percent

      } else if(y.label.type == "comma"){
        y.label.typea <- scales::comma

      } else if(y.label.type == "numeric"){
        y.label.typea <- ggplot2::waiver()

      } else if(y.label.type == "sci"){
        y.label.typea <- scales::scientific

      } else if(y.label.type == "dollar"){
        y.label.typea <- scales::dollar

      } else {
        y.label.typea <- NULL
      }

    # Specify y axis line
      if(y.axis.line == TRUE){
        y.axis.line.c <- ggplot2::element_line(colour = 'black',
                                               linewidth=0.5,
                                               linetype='solid')
      } else{
        y.axis.line.c <- ggplot2::element_blank()
      }

    # Specify x axis line
      if(x.axis.line == TRUE){
        x.axis.line.c <- ggplot2::element_line(colour = 'black',
                                               linewidth=0.5,
                                               linetype='solid')
      } else{
        x.axis.line.c <- ggplot2::element_blank()
      }

  # Specify color
    if(group.var == ""){
      color <- color[1]
    } else{

    }

  # Specify basic plot information and if grouping will occur
    plot.1 <- ggplot2::ggplot(data,
      {if(group.var == ""){

        ggplot2::aes(x = {{x.var1}},
                     y = {{y.var1}},
                     fill = color)

        } else{

          ggplot2::aes(x = {{x.var1}},
                       y = {{y.var1}},
                       fill = group77d8214)

        }})  +

  # Specify the plot as a box plot based on grouping
    {if(group.var == ""){

      ggplot2::geom_boxplot(fill = color)

     } else {

       ggplot2::geom_boxplot(ggplot2::aes(fill = group77d8214))

     }} +

    # Remove legend information based on alpha and specify fill color
      {if(group.var != "") {

        ggplot2::scale_fill_manual(values = color)

      } else{

      }
      } +

  # Specify if a mean point should be placed in the box plot
    {if(mean.sym == TRUE){

      ggplot2::stat_summary(fun = "mean",
                           position = ggplot2::position_dodge(width = mean.sym.position),
                           geom = "point",
                           shape = mean.sym.shape,
                           size = mean.sym.size,
                           color = mean.sym.color)}

      else {

    }}+

  # Specify legend information
    ggplot2::guides(color = ggplot2::guide_legend(byrow = TRUE,
                                                  override.aes = ggplot2::aes(label = ""))) +

  # Specify title information
    ggplot2::labs(title = plot.title,
                  y = y.title,
                  x = x.title,
                  fill = legend.title) +

  # Specify y-axis scale
    ggplot2::scale_y_continuous(labels = y.label.typea,
                                limits = c(y.min,
                                            y.max),
                                expand = c(0,0),
                                breaks = y.breaks.c)+

  # Specify the plots theme information
    ggplot2::theme(panel.grid = ggplot2::element_blank(),
                   panel.grid.major.y = grid.maj,
                   panel.background = ggplot2::element_rect(fill = 'white',
                                                            color = NA),
                   plot.margin = ggplot2::margin(t = 0.5,  # Top margin
                                                 r = 0.5,  # Right margin
                                                 b = 0.5,  # Bottom margin
                                                 l = 0.5,  # Left margin
                                                 unit = "cm"),
                   text = ggplot2::element_text(family = text.font),
                   plot.title = ggplot2::element_text(size = plot.title.size,
                                                      hjust = 0.5),
                   axis.title = ggplot2::element_text(size = axis.title.size),
                   axis.text = ggplot2::element_text(size = axis.label.size),
                   axis.title.y = ggplot2::element_text(margin = ggplot2::margin(t = 0,
                                                                                 r = 7,
                                                                                 b = 0,
                                                                                 l = 0),
                                                        angle = 0,
                                                        vjust = 0.5),
                   axis.title.x = ggplot2::element_text(margin = ggplot2::margin(t = 7,
                                                                                 r = 0,
                                                                                 b = 0,
                                                                                 l = 0)),
                   axis.line.x = x.axis.line.c,
                   axis.line.y = y.axis.line.c,
                   plot.caption = ggplot2::element_text(size = caption.size),
                   legend.position = legend.position,
                   legend.background = ggplot2::element_rect(fill = "white",
                                                             colour = "white",
                                                             inherit.blank = TRUE,
                                                             color = "white"),
                   legend.key = ggplot2::element_blank(),
                   legend.title = ggplot2::element_text(size= legend.title.size),
                   legend.text = ggplot2::element_text(size = legend.label.size)) +

  # Specify x axis label text wrap
    ggplot2::scale_x_discrete(labels = scales::wrap_format(x.wrap)) +

  # Specify if x axis should be removed
    {if(x.remove == TRUE){

      ggplot2::theme(axis.title.x = ggplot2::element_blank(),
                     axis.text.x = ggplot2::element_blank(),
                     axis.ticks.x = ggplot2::element_blank())

    } else{

    }} +

  # Specify to remove axis ticks
    {if(remove.axis.ticks == TRUE){

      ggplot2::theme(axis.ticks = ggplot2::element_blank())

    } else {

     }} +

  # Specify if the coords should be flipped
    {if(coord.flip == TRUE){

      ggplot2::coord_flip()

      } else{

      }
     } +

  # Specify plot elements
    plot.element1 +
    plot.element2 +
    plot.element3 +
    plot.element4 +
    plot.element5 +
    plot.element6 +
    plot.element7 +
    plot.element8 +
    plot.element9

  # Specify caption
    {if(caption == TRUE){

      plot.1 <- plot.1 + CCMHr::ccmh_caption()

    } else{

    }}

  # Specify if the graph should be saved as file or returned as an object
    if(save == TRUE){

      ggsave(paste0(path),
             plot = plot.1,
             width = plot.width,
             height = plot.height,
             units = plot.units,
             dpi = plot.dpi,
             device = plot.device)

    } else{

      return(plot.1)

    }

}

#' @rdname plot_box
#' @export
  box_plot <- plot_box
CCMH-PSU/CCMHr documentation built on Jan. 31, 2024, 1:08 a.m.