Update ARM and ACTARM.

insightsengineering · Sep 28, 2024 · b079041 · b079041
1 parent 55f8550
commit b079041
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/R/radab.R b/R/radab.R
@@ -76,8 +76,8 @@ radab <- function(adsl,
   }
 
   # validate and initialize related variables
-  param_init_list <- relvar_init(param, paramcd)
-  unit_init_list <- relvar_init(param, avalu)
+  param_init_list <- random.cdisc.data:::relvar_init(param, paramcd)
+  unit_init_list <- random.cdisc.data:::relvar_init(param, avalu)
 
   adpc <- adpc %>% dplyr::filter(ASMED == "PLASMA")
   adab0 <- expand.grid(
@@ -386,6 +386,11 @@ radab <- function(adsl,
       last_ada
     ))
 
+  # Carry over ARM and ACTARM for all records.
+  arm <- adab %>% filter(!is.na(ARM), !is.na(ACTARM)) %>% select(USUBJID, ARM, ACTARM) %>% distinct(.)
+  adab$ARM <- arm$ARM[match(adab$USUBJID, arm$USUBJID)]
+  adab$ACTARM <- arm$ACTARM[match(adab$USUBJID, arm$USUBJID)]
+
   if (length(na_vars) > 0 && na_percentage > 0) {
     adab <- mutate_na(ds = adab, na_vars = na_vars, na_percentage = na_percentage)
   }