migrate add_coords function to package

matthewkling · Sep 11, 2019 · e82202c · e82202c
1 parent 2c9c338
commit e82202c
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/R/utils.R b/R/utils.R
@@ -38,6 +38,18 @@ add_lat <- function(x){
       stack(lat, x)
 }
 
+#' Augment a raster object, adding layers containing cell row-column indices.
+#'
+#' @param x raster layer or stack
+#' @return the input layer, with an additional layer of latitude values
+add_coords <- function(windrose){
+   rows <- cols <- windrose[[1]]
+   rows[] <- rep(1:nrow(rows), each=ncol(rows))
+   cols[] <- rep(1:ncol(rows), nrow(rows))
+   windrose <- stack(windrose, rows, cols)
+   names(windrose) <- c("SW", "W", "NW", "N", "NE", "E", "SE", "S", "row", "col")
+   return(windrose)
+}
 
 #' Get neighbor names for a windrose object
 #'

diff --git a/man/add_coords.Rd b/man/add_coords.Rd