Wrap and unwrap rasters

R/disp.R

@@ -69,7 +69,7 @@
 #' # disp
 #' disp_output <- disp(
 #'   N_t = sim_data$n1_map,
-#'   id = sim_data$id,
+#'   id = unwrap(sim_data$id),
 #'   data_table = sim_data$data_table,
 #'   kernel = sim_data$kernel,
 #'   dens_dep = sim_data$dens_dep,

R/initialise.R

@@ -298,11 +298,11 @@ initialise <- function(
   # output list
   out <- list(
     n1_map = as.matrix(n1_map, wide = TRUE),
-    id = id,
+    id = wrap(id),
     resolution = resolution,
     r = r,
     r_sd = r_sd,
-    K_map = K_map,
+    K_map = wrap(K_map),
     K_sd = K_sd,
     growth = growth,
     A = A,

R/print.sim_data.R

@@ -32,8 +32,8 @@
 
   # prepare output
   n1_sm <- summary(as.numeric(x$n1_map))
-  K_sm <- x$K_map
+  K_sm <- unwrap(x$K_map)
 
   names <- names(x)
   names <- names[(names %in% c(
     "resolution", "r", "r_sd", "K_sd", "growth", "A",

R/sim.R

@@ -176,14 +176,14 @@ sim <- function(
   options(warn = -1)
 
   # Extract data from the sim_data object
-  K_map <- obj$K_map # carrying capacity
+  K_map <- unwrap(obj$K_map) # carrying capacity
   K_sd <- obj$K_sd # sd of carrying capacity (additional cell specific variation) #nolint
   dynamics <- obj$dynamics # population growth function
   n1_map <- obj$n1_map # population numbers at the first time step
   r <- obj$r # intrinsic population growth rate
   r_sd <- obj$r_sd # sd of intrinsic growth rate (time specific variation)
   A <- obj$A # Allee effect coefficient
-  id <- obj$id # square identifiers
+  id <- unwrap(obj$id) # square identifiers
   ncells <- obj$ncells # number of cells in the study area
   data_table <- obj$data_table
   changing_env <- obj$changing_env

R/summary.sim_data.R

@@ -33,7 +33,7 @@ summary.sim_data <- function(object, ...) {
 
   # prepare summaries
   n1_sm <- summary(as.numeric(object$n1_map))
-  K_sm <- summary(as.numeric(as.matrix(object$K_map)))
+  K_sm <- summary(as.numeric(as.matrix(unwrap(object$K_map))))
 
   names <- c(
     "growth", "r", "A", "kernel_fun", "dens_dep", "border",

R/update.sim_data.R

@@ -96,11 +96,15 @@ update.sim_data <- function(object, ..., evaluate = TRUE) {
       }
     }
 
+    # unwrap K_map if not updated
+    if(!"K_map" %in% names(extras)) {
+      call$K_map <- unwrap(object$K_map)
+    }
+
     # transform call to call object
     call <- as.call(call)
 
 
-
     # evaluate or return the call
     if (evaluate) {
       eval(call, parent.frame())

tests/testthat/test-initialise.R

@@ -531,6 +531,6 @@ test_that("K_add_stochasticity works", {
   )
 
 
-  expect_s4_class(test_sim_data_1$K_map, "SpatRaster")
-  expect_s4_class(test_sim_data_2$K_map, "SpatRaster")
+  expect_s4_class(test_sim_data_1$K_map, "PackedSpatRaster")
+  expect_s4_class(test_sim_data_2$K_map, "PackedSpatRaster")
 })