From 8bd5f6e416f3f5b087efca1f9d5090f26accb799 Mon Sep 17 00:00:00 2001
From: Edgar Manukyan <edgar.manukyan@roche.com>
Date: Sun, 8 Dec 2024 01:13:24 +0000
Subject: [PATCH] #107 add multiple topics, vs templates

---
 R/generate_code.R          |  16 +++++-
 R/generate_code_tempates.R | 113 +++++++++++++++++++++++++++++++++++++
 2 files changed, 127 insertions(+), 2 deletions(-)

diff --git a/R/generate_code.R b/R/generate_code.R
index 509ecbb6..b5e258c3 100644
--- a/R/generate_code.R
+++ b/R/generate_code.R
@@ -44,6 +44,11 @@
 #'     !mapping_is_dataset %in% c("TRUE")
 #'   )
 #'
+#' old_width <- options(width = 20)
+#' generate_code(spec, domain)
+#' # Restore original width
+#' options(width = old_width$width)
+#'
 #' # CM domain ----
 #'
 #' spec <- read_spec("cm_sdtm_oak_spec_cdash.csv")
@@ -74,10 +79,10 @@ generate_code <- function(spec, domain, out_dir = ".") {
 
   topics <- unique(spec_domain$topic)
 
-  # Do by topic
   code_by_topics <- purrr::map(topics, \(topic) {
+
     spec_domain_topic <- spec_domain |>
-      dplyr::filter(topic %in% topic)
+      dplyr::filter(topic %in% {{topic}})
 
     domain_topic <- paste(domain, topic, sep = "_") |>
       tolower()
@@ -104,6 +109,13 @@ generate_code <- function(spec, domain, out_dir = ".") {
       append(cm_template_prefix, after = 0L) |>
       append(cm_template_suffix) |>
       styler::style_text()
+  } else {
+    code_by_topics |>
+      purrr::map(remove_last_pipe) |>
+      unlist() |>
+      append(vs_template_prefix, after = 0L) |>
+      append(vs_template_suffix) |>
+      styler::style_text()
   }
 
   # Save the code to a file
diff --git a/R/generate_code_tempates.R b/R/generate_code_tempates.R
index a6fdf937..00961250 100644
--- a/R/generate_code_tempates.R
+++ b/R/generate_code_tempates.R
@@ -60,3 +60,116 @@ dplyr::select("STUDYID", "DOMAIN", "USUBJID", "CMSEQ", "CMTRT", "CMCAT", "CMINDC
               "CMDOSE", "CMDOSTXT", "CMDOSU", "CMDOSFRM", "CMDOSFRQ", "CMROUTE",
               "CMSTDTC", "CMENDTC","CMSTDY", "CMENDY", "CMENRTPT", "CMENTPT")
 ')
+
+#' The template suffix for the vs code
+#'
+#' @noRd
+#' @keywords internal
+vs_template_prefix <- stringr::str_glue('
+library(sdtm.oak)
+library(dplyr)
+
+
+# Read Specification
+
+# Read CT Specification
+study_ct <- read.csv("./datasets/sdtm_ct.csv")
+
+# Read in raw data
+vitals_raw_data <- read.csv("./datasets/vitals_raw_data.csv",
+                   stringsAsFactors = FALSE)
+
+vitals_raw_data <- admiral::convert_blanks_to_na(vitals_raw_data)
+
+
+# derive oak_id_vars
+vitals_raw_data <- vitals_raw_data %>%
+  generate_oak_id_vars(
+    pat_var = "PATNUM",
+    raw_src = "vitals_raw_data"
+  )
+
+# Read in DM domain to derive study day
+dm <- read.csv("./datasets/dm.csv")
+
+dm <- admiral::convert_blanks_to_na(dm)
+')
+
+#' The template suffix for the vs code
+#'
+#' @noRd
+#' @keywords internal
+vs_template_suffix <- stringr::str_glue('
+# Combine all the topic variables into a single data frame. ----
+vs_combined <- dplyr::bind_rows(
+  vs_asmntdn, vs_sys_bp, vs_dia_bp, vs_pulse, vs_temp,
+  vs_resprt, vs_oxy_sat
+)
+
+# Map qualifiers common to all topic variables ----
+
+vs <- vs_combined %>%
+  # Map VSDTC using assign_ct algorithm
+  assign_datetime(
+    raw_dat = vitals_raw,
+    raw_var = c("VTLD", "VTLTM"),
+    tgt_var = "VSDTC",
+    raw_fmt = c(list(c("d-m-y", "dd-mmm-yyyy")), "H:M")
+  ) %>%
+  # Map VSTPT from TMPTC using assign_ct
+  assign_ct(
+    raw_dat = vitals_raw,
+    raw_var = "TMPTC",
+    tgt_var = "VSTPT",
+    ct_spec = study_ct,
+    ct_clst = "TPT",
+    id_vars = oak_id_vars()
+  ) %>%
+  # Map VSTPTNUM from TMPTC using assign_ct
+  assign_ct(
+    raw_dat = vitals_raw,
+    raw_var = "TMPTC",
+    tgt_var = "VSTPTNUM",
+    ct_spec = study_ct,
+    ct_clst = "TPTNUM",
+    id_vars = oak_id_vars()
+  ) %>%
+  # Map VISIT from VISIT_NAME using assign_ct
+  assign_ct(
+    raw_dat = vitals_raw,
+    raw_var = "VISIT_NAME",
+    tgt_var = "VISIT",
+    ct_spec = study_ct,
+    ct_clst = "VISIT",
+    id_vars = oak_id_vars()
+  ) %>%
+  # Map VISITNUM from VISIT_NAME using assign_ct
+  assign_ct(
+    raw_dat = vitals_raw,
+    raw_var = "VISIT_NAME",
+    tgt_var = "VISITNUM",
+    ct_spec = study_ct,
+    ct_clst = "VISITNUM",
+    id_vars = oak_id_vars()
+  ) %>%
+  dplyr::mutate(
+    STUDYID = "test_study",
+    DOMAIN = "VS",
+    VSCAT = "VITAL SIGNS",
+    USUBJID = paste0("test_study", "-", .data$patient_number)
+  ) %>%
+  derive_seq(tgt_var = "VSSEQ",
+             rec_vars= c("USUBJID", "VISITNUM", "VSTPTNUM", "VSTESTCD")) %>%
+  # A bug in derive_study_day V0.1 that clears the time values in VSDTC
+  derive_study_day(
+    sdtm_in = .,
+    dm_domain = dm,
+    tgdt = "VSDTC",
+    refdt = "RFXSTDTC",
+    study_day_var = "VSDY"
+  ) %>%
+  dplyr::select("STUDYID", "DOMAIN", "USUBJID", "VSSEQ",
+                "VSTESTCD", "VSTEST", "VSCAT", "VSPOS",
+                "VSORRES", "VSORRESU", "VSLOC", "VSLAT",
+                "VISIT", "VISITNUM", "VSDY", "VSTPT", "VSTPTNUM", "VSDTC" )
+')