R/plot-methods.R

In ropensci/neotoma: Access to the Neotoma Paleoecological Database Through R

#' @importFrom graphics legend par plot points
#' @export
plot.site <- function(x, database = NULL, ...) {
  if(is.null(database)) {
    plot(x$long, x$lat, xlab = "Longitude", ylab = "Latitude", ...)
  }  else {
    plot(x$long, x$lat, xlab = "Longitude", ylab = "Latitude", type = 'n')
    points(x$long, x$lat, ...)
    
  }
}

#' @export
plot.dataset <- function(x, ...) {
    site <- get_site(x)
    plot(site, ...)
}

#' @export
plot.dataset_list <- function(x, ...) {
  site <- get_site(x)
  
  site <- site[!(is.na(site$long) | is.na(site$lat)),]
  
  types <- sapply(x, function(x)x$dataset.meta$dataset.type)
  
  types[is.na(types)] <- "Undefined"
  
  types <- factor(types)
  
  # Split the factor levels so they plot better in the legend:
  levels(types) <- gsub(' ', '\n', levels(types))
  
  old_par <- par()
  par(mar=c(5, 4, 4, 5))
  plot(site, pch=as.numeric(types)-1, ..., bty='L')
  legend(max(site$long) + 0.5, max(site$lat), levels(types), 
         pch = (1:length(types)) - 1, xpd = TRUE, cex = 0.75)
  par(mar = old_par$mar)
}

#' @export
plot.download <- function(x, ...) {
  site <- get_site(x)
  plot(site, ...)
}

#' @export
plot.download_list <- function(x, ...) {
  dataset <- get_dataset(x)

  plot(dataset, ...)
  
}