aspatial_siri functions can now handle one population

Previously, the aspatial_siri_functions broke if there was only one active population. Now they run just fine.

R/aspatial_siri_seasons.R

@@ -100,7 +100,15 @@ siri_model_summer <- function(inputs) {
     })
 
   # Set up initial vectors
-  population_list <- array_branch(segment_abundance[, occupied_indices], 2)
+  if (active_pops > 1) {
+    population_list <- array_branch(segment_abundance[, occupied_indices], 2)
+  }
+  if (active_pops == 1) {
+    population_list <- list(segment_abundance[, occupied_indices])
+  }
+  if (active_pops == 0) {
+    return(segment_abundance)
+  }
   carrying_capacity <- carrying_capacity[occupied_indices]
 
   population_new <- pmap(list(initial_pop = population_list,
@@ -219,7 +227,15 @@ siri_model_winter <- function(inputs) {
     })
 
   # Set up initial vectors
-  population_list <- array_branch(segment_abundance[, occupied_indices], 2)
+  if (active_pops > 1) {
+    population_list <- array_branch(segment_abundance[, occupied_indices], 2)
+  }
+  if (active_pops == 1) {
+    population_list <- list(segment_abundance[, occupied_indices])
+  }
+  if (active_pops == 0) {
+    return(segment_abundance)
+  }
   carrying_capacity <- carrying_capacity[occupied_indices]
 
   population_new <- pmap(list(initial_pop = population_list,

R/disease_simulator.R

@@ -403,7 +403,7 @@ disease_simulator <- function(inputs) {
           }
 
           ## Season functions ##
-          if (process == "season_functions" && is.list(season_function_list)) {
+          if (occupied_populations && process == "season_functions" && is.list(season_function_list)) {
             transformed <- season_function_list[[season]](r, tm,
                                                           carrying_capacity,
                                                           segment_abundance,