Merge pull request #24 from Merck/treatment-group

allow user to specify treatment group (#15)

R/meta_forestly.R

@@ -21,11 +21,12 @@
 #' @param dataset_adsl ADSL source dataset.
 #' @param dataset_adae ADAE source dataset.
 #' @param population_term A character value of population term name.
+#' @param observation_term A character value of observation term name.
 #' @param population_subset An unquoted condition for selecting the
 #'   populations from ADSL dataset.
-#' @param observation_term A character value of observation term name.
 #' @param observation_subset An unquoted condition for selecting the
 #'   observations from ADAE dataset.
+#' @param treatment_group A character value of treatment group name.
 #' @param parameter_term A character value of parameter term name.
 #'
 #' @return A metalite object.
@@ -43,9 +44,10 @@ meta_forestly <- function(
     dataset_adsl,
     dataset_adae,
     population_term,
-    population_subset = SAFFL == "Y",
     observation_term,
+    population_subset = SAFFL == "Y",
     observation_subset = SAFFL == "Y",
+    treatment_group = "TRTA",
     parameter_term = "any;rel;ser") {
   meta <- metalite::meta_adam(
     population = as.data.frame(dataset_adsl),
@@ -59,13 +61,13 @@ meta_forestly <- function(
     )) |>
     metalite::define_population(
       name = population_term,
-      group = "TRTA",
+      group = treatment_group,
       subset = !!rlang::enquo(population_subset),
       label = ""
     ) |>
     metalite::define_observation(
       name = observation_term,
-      group = "TRTA",
+      group = treatment_group,
       subset = !!rlang::enquo(observation_subset),
       label = ""
     )