R/create_latlong_rasters.R

In elpidiofilho/mdsFuncs: Auxiliar functions to pre-process raster data and modelling with machine learning models using caret package

#' Create lat long rasters
#' @description Creates latitude and longitude rasters
#' @param r reference raster
#' @param todisk logical whether or not to write files to disk
#' @param filetype raster files extension .tif, .asc...
#' @param folder Folder where the output files will be saved
#' @importFrom terra crds crop mask writeRaster rast
#' @importFrom here here
#' @return A list with lat e long raster
#' @export
#' @author Elpidio Filho
#' @examples
#' # create_lat_long_rasters(r, todisk = FALSE)
create_lat_long_rasters <- function(r, todisk = FALSE,
                                    filetype = ".tif",
                                    folder = NULL) {
  if (class(r) %in% c("RasterStack", "RasterLayer")) {
    r <- rast(r)
  } else {
    if (class(r) != "SpatRaster") {
      stop("Error : file must be in RasterStack, RasterLayer or SpatRaster format")
    }
  }
  r1 <- r[[1]]
  lat <- long <- r1
  xy <- terra::crds(long, na.rm = FALSE) # obtém as coordenadas de cada célula
  long[] <- xy[, 1]
  lat[] <- xy[, 2]
  contorno <- extract_raster_contour(r1)
  long <- long |>
    terra::crop(contorno) |>
    terra::mask(contorno)
  lat <- lat |>
    terra::crop(contorno) |>
    terra::mask(contorno)
  names(lat) <- "lat"
  names(long) <- "long"
  if (todisk == TRUE) {
    terra::writeRaster(long, here(folder, paste0("long", filetype)),
      overwrite = TRUE
    )
    terra::writeRaster(lat, here(folder, paste0("lat", filetype)),
      overwrite = TRUE
    )
  }
  return(list(lat, long))
}