R/plot_lake_contours.R

In WDNR-Water-Use/CSLSlevels: CSLS Lake Level Analysis

#' Draw contours for a given lake
#'
#' Given a) a single lake level or a vector of lake levels as well as b) the
#' name of the lake of interest, this loads the raster associated with the lake
#' and draws contours at the desired levels. Note that by default, each contour
#' is color coded. To turn this off, set color_values to the all the same color
#' (e.g., c("black", "black", "black")) and set legend_position to "none".
#'
#' @param lake name of lake to analyze, e.g., "Long"
#' @param lake_levels a single value or vector of lake elevations (mamsl) to
#'                    draw contors for.
#' @param color_legend_title legend title for colors.
#' @param color_labels a vector corresponding to lake_levels with the name to
#'                     use for each level in the legend. Defaults to
#'                     c(10, 50, 90) for the 10%, 50%, and 90% exceedance
#'                     probabilities.
#' @param color_values a vector corresponding to lake_levels with the color to
#'                     use for each level in the legend. Defaults to
#'                     c("#a80000", "#ffab00", "#004da8") for Tuscan Red
#'                     (#a80000), Electron Gold (#ffab00), and Ultra Blue
#'                     (#004da8). Note that the order of this vector does not
#'                     necessarily correspond with the order of lake_levels or
#'                     color_labels, it may take some trial and error to find
#'                     the right order for the visualization you have in mind.
#' @param legend_position position of legend, defaults to "top", set to "none"
#'                        to remove legend (note: still need to include
#'                        color_name, color_labels, and color_values even if
#'                        leave the legend off the returned plot).
#' @param title_name string to use for plot title
#' @param text_size font size for text in plot
#'
#' @return plot_obj, a plot with the contours of the lake.
#'
#' @importFrom rlang .data
#' @importFrom raster rasterToContour
#' @import ggplot2
#' @import ggspatial
#' @import extrafont
#'
#' @export

plot_lake_contours <- function(lake,
                               lake_levels,
                               color_legend_title = "Exceedance Probability (%)",
                               color_labels = c(10, 50, 90),
                               color_values = c("#a80000", "#ffab00", "#004da8"),
                               legend_position = "right",
                               title_name = NULL,
                               text_size = 10) {
  # Get raster and draw contours
  lake_raster <- CSLSlevels::lake_raster[[lake]]
  contours    <- rasterToContour(lake_raster, levels = lake_levels)

  # Plot contours
  plot_obj    <- ggplot() +
                 layer_spatial(data = contours,
                               aes(color = .data$level),
                               fill = NA,
                               size = 0.8) +
                 theme_bw() +
                 labs(title = title_name,
                      color = color_legend_title) +
                 scale_color_manual(breaks = lake_levels,
                                    labels = color_labels,
                                    values = color_values) +
                 theme_bw() +
                 theme(text = element_text(family = "Segoe UI Semilight",
                                           size = text_size),
                       panel.grid.major = element_blank(),
                       plot.title = element_text(hjust = 0.5),
                       legend.position = legend_position)
  return(plot_obj)
}