feat(model): modify trajectory to send ASU patients to rehab (plus minor docstring change in rehab traj)

Author: amyheather

NAMESPACE

@@ -14,6 +14,7 @@ export(model)
 export(sample_routing)
 export(transform_to_lnorm)
 importFrom(simmer,add_generator)
+importFrom(simmer,branch)
 importFrom(simmer,get_attribute)
 importFrom(simmer,get_mon_arrivals)
 importFrom(simmer,get_mon_resources)

R/create_asu_trajectory.R

@@ -13,7 +13,7 @@
 #' of routing to each destination (e.g.
 #' \code{param$asu_routing$stroke$rehab = 0.24}).
 #'
-#' @importFrom simmer trajectory
+#' @importFrom simmer branch trajectory
 #' @importFrom stats rlnorm
 #'
 #' @return Simmer trajectory object. Defines patient journey logic through the
@@ -57,5 +57,19 @@ create_asu_trajectory <- function(env, patient_type, param) {
         meanlog = los_params[["meanlog"]],
         sdlog = los_params[["sdlog"]]
       )
-    })
+    }) |>
+
+    # If that patient's destination is rehab, then start on that trajectory
+    branch(
+      option = function() {
+        # Retrieve attribute, and use to get post-ASU destination as a string
+        dest_index <- get_attribute(env, "post_asu_destination")
+        dest_names <- names(param[["asu_routing"]][[patient_type]])
+        dest <- dest_names[dest_index]
+        # Return 1 for rehab and 0 otherwise
+        if (dest == "rehab") 1L else 0L
+      },
+      continue = FALSE,  # Do not continue main trajectory after branch
+      create_rehab_trajectory(env, patient_type, param)
+    )
 }

R/create_rehab_trajectory.R

@@ -1,7 +1,7 @@
 #' Create rehab patient trajectory.
 #'
-#'  Represents patient stay on the rehabilitation unit - samples their (1)
-#'  destination after rehab, and (2) length of stay (LOS) on the unit.
+#' Represents patient stay on the rehabilitation unit - samples their (1)
+#' destination after rehab, and (2) length of stay (LOS) on the unit.
 #'
 #' @param env Simmer environment object. The simulation environment where
 #' generators will be added.