R/ggplot-coord_date.R

In mdancho84/tidyquant: Tidy Quantitative Financial Analysis

#' Zoom in on plot regions using date ranges or date-time ranges
#'
#' @details
#' The `coord_` functions prevent loss of data during zooming, which is
#' necessary when zooming in on plots that calculate `stats` using data
#' outside of the zoom range (e.g. when plotting moving averages
#' with [geom_ma()]). Setting limits using `scale_x_date`
#' changes the underlying data which causes moving averages to fail.
#'
#' `coord_x_date` is a wrapper for `coord_cartesian`
#' that enables quickly zooming in on plot regions using a date range.
#'
#' `coord_x_datetime` is a wrapper for `coord_cartesian`
#' that enables quickly zooming in on plot regions using a date-time range.
#'
#' @param xlim Limits for the x axis, entered as character dates in "YYYY-MM-DD"
#' format for date or "YYYY-MM-DD HH:MM:SS" for date-time.
#' @param ylim Limits for the y axis, entered as values
#' @inheritParams ggplot2::coord_cartesian
#'
#' @seealso
#' [ggplot2::coord_cartesian()]
#'
#' @name coord_x_date
#'
#' @export
#'
#' @examples
#' # Load libraries
#' library(dplyr)
#' library(ggplot2)
#'
#' # coord_x_date
#' AAPL <- tq_get("AAPL", from = "2013-01-01", to = "2016-12-31")
#' AAPL %>%
#'     ggplot(aes(x = date, y = adjusted)) +
#'     geom_line() +                         # Plot stock price
#'     geom_ma(n = 50) +                     # Plot 50-day Moving Average
#'     geom_ma(n = 200, color = "red") +     # Plot 200-day Moving Average
#'     # Zoom in
#'     coord_x_date(xlim = c("2016-01-01", "2016-12-31"))
#'
#'
#' # coord_x_datetime
#' time_index <- seq(from = as.POSIXct("2012-05-15 07:00"),
#'                   to   = as.POSIXct("2012-05-17 18:00"),
#'                   by   = "hour")
#' set.seed(1)
#' value <- rnorm(n = length(time_index))
#' hourly_data <- tibble(time.index = time_index,
#'                       value      = value)
#' hourly_data %>%
#'     ggplot(aes(x = time.index, y = value)) +
#'     geom_point() +
#'     coord_x_datetime(xlim = c("2012-05-15 07:00:00", "2012-05-15 16:00:00"))

#' @rdname coord_x_date
#' @export
coord_x_date <- function(xlim = NULL, ylim = NULL, expand = TRUE) {

    if (!is.null(xlim)) {
        xlim <- lubridate::as_date(xlim)
    }

    ggplot2::coord_cartesian(xlim = xlim, ylim = ylim, expand = expand)

}

#' @rdname coord_x_date
#' @export
coord_x_datetime <- function(xlim = NULL, ylim = NULL, expand = TRUE) {

    if (!is.null(xlim)) {
        xlim <- lubridate::as_datetime(xlim)
    }

    ggplot2::coord_cartesian(xlim = xlim, ylim = ylim, expand = expand)

}