optimise .sim_network_bp by moving contact distribution density outsi…

…de loop
epiverse-trace · Feb 23, 2024 · 018f9dd · 018f9dd
1 parent bf89eb7
commit 018f9dd
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/R/sim_network_bp.R b/R/sim_network_bp.R
@@ -51,17 +51,17 @@
   ancestor_idx <- 1L
   prev_ancestors <- 1L
 
+  if (config$network == "adjusted") {
+    # sample contact distribution (excess degree distribution)
+    q <- contact_distribution(0:1e4 + 1) * (0:1e4 + 1)
+    q <- q / sum(q)
+  } else {
+    q <- contact_distribution(0:1e4)
+  }
+
   # run loop until no more individuals are sampled
   while (next_gen_size > 0) {
-    if (config$network == "adjusted") {
-      # sample contact distribution (excess degree distribution)
-      q <- contact_distribution(0:1e4 + 1) * (0:1e4 + 1)
-      q <- q / sum(q)
-    } else {
-      q <- contact_distribution(0:1e4)
-    }
     contacts <- sample(0:1e4, size = next_gen_size, replace = TRUE, prob = q)
-
     # add contacts if sampled
     if (sum(contacts) > 0L) {
       chain_size <- chain_size + sum(contacts)