R/get_coordinates.R

In reconhub/epiflows: Predicting Disease Spread from Flow Data

#' @rdname add_coordinates
#' @export
get_coordinates <- function(x, ...) {
  UseMethod("get_coordinates", x)
}


#' @export
#' @rdname add_coordinates
#' @param location a character specifying a single location to return as a vector of
#'   coordinates. You cannot specify multiple locations with this parameter.
#'   Defaults to `NULL`, indicating all locations.
#' @param ... unused
get_coordinates.epiflows <- function(x, location = NULL, ...) {
  res <- try(get_vars(x, "coordinates", id = TRUE), silent = TRUE)
  if (inherits(res, "try-error")) {
    xprint <- deparse(substitute(x))
    return(NULL)
  }
  if (is.character(location) && length(location) == 1L) {
    res <- as.numeric(drop(res[res$id == location, 2:3]))
  }
  res
}