RMHDR-251 New ECG Measures for 2024Q2 Release

scripts/process-data/fitbitecg.R

@@ -0,0 +1,212 @@
+# Functions
+
+#' Create i2b2 summaries
+#'
+#' `ecg_stat_summarize()` summarizes the ECG data in a data frame on 
+#' specific time scales (all data and weekly) for the following statistics: 
+#' Number of records.
+#'
+#' @param df A data frame with `participantidentifier`, `concept`, `value`, and 
+#' some combination of `startdate`, `enddate`, `date`, or `datetime` columns.
+#'
+#' @return A data frame that differs in size from `df`. The original `value` 
+#' column is dropped and replaced by a new `value` column containing the values 
+#' of each summary statistic computed for each group of data (group by 
+#' `participantidentifier` and `concept` for all data, and 
+#' `participantidentifier`, `concept`, `year`, `week` for weekly summaries).
+ecg_stat_summarize <- function(df) {
+  cat("Running ecg_stat_summarize()...\n")
+
+  if (!is.data.frame(df)) stop("df must be a data frame")
+  if (!all(c("participantidentifier", "concept", "value") %in% colnames(df))) stop("'participantidentifier', 'concept', and 'value' columns must be present in df")
+
+  summarize_alltime <- function(df) {
+    if ("startdate" %in% colnames(df) & "enddate" %in% colnames(df)) {
+      # Do nothing
+    } else if ("date" %in% colnames(df)) {
+      df <- 
+        df %>% 
+        dplyr::rename(startdate = date) %>% 
+        dplyr::mutate(enddate = NA)
+    } else if ("datetime" %in% colnames(df) & !"date" %in% colnames(df)) {
+      df <- 
+        df %>% 
+        dplyr::rename(startdate = datetime) %>% 
+        dplyr::mutate(enddate = NA)
+    } else {
+      stop("No 'startdate', 'enddate', 'date', or 'datetime' column found")
+    }
+
+    df %>%
+      dplyr::select(participantidentifier, startdate, enddate, concept, value) %>%
+      dplyr::group_by(participantidentifier, concept) %>%
+      dplyr::reframe(startdate = lubridate::as_date(min(startdate)),
+                     enddate = lubridate::as_date(max(enddate)),
+                     numrecords = dplyr::n()) %>%
+      dplyr::ungroup() %>% 
+      tidyr::pivot_longer(cols = c(numrecords),
+                          names_to = "stat",
+                          values_to = "value") %>%
+      dplyr::mutate(concept = paste0("mhp:summary:alltime:", stat, ":", concept)) %>%
+      dplyr::select(participantidentifier, startdate, enddate, concept, value) %>%
+      dplyr::distinct()
+  }
+
+  summarize_weekly <- function(df) {
+    if ("startdate" %in% colnames(df)) {
+      # Do nothing
+    } else if ("date" %in% colnames(df)) {
+      df <- 
+        df %>% 
+        dplyr::rename(startdate = date)
+    } else if ("datetime" %in% colnames(df) & !"date" %in% colnames(df)) {
+      df <- 
+        df %>% 
+        dplyr::rename(startdate = datetime)
+    } else {
+      stop("No 'startdate, enddate, date, or datetime' column found")
+    }
+
+    df %>%
+      dplyr::select(participantidentifier, 
+                    concept, 
+                    value, 
+                    startdate) %>% 
+      dplyr::mutate(startdate = lubridate::as_date(startdate)) %>% 
+      dplyr::filter(startdate >= lubridate::floor_date(min(startdate), 
+                                                       unit = "week", 
+                                                       week_start = 7)) %>% 
+      dplyr::mutate(startdate = lubridate::floor_date(startdate, unit = "week", week_start = 7), 
+                    enddate = startdate + lubridate::days(6)) %>% 
+      dplyr::group_by(participantidentifier, 
+                      concept, 
+                      startdate, 
+                      enddate) %>% 
+      dplyr::reframe(numrecords = dplyr::n()) %>% 
+      dplyr::ungroup() %>% 
+      tidyr::pivot_longer(cols = c(numrecords), 
+                          names_to = "stat", 
+                          values_to = "value") %>% 
+      dplyr::mutate(concept = paste0("mhp:summary:weekly:", stat, ":", concept)) %>% 
+      dplyr::select(participantidentifier, startdate, enddate, concept, value) %>% 
+      dplyr::distinct()
+  }
+
+  result <- 
+    dplyr::bind_rows(summarize_alltime(df), 
+                     summarize_weekly(df)) %>% 
+    dplyr::distinct()
+
+  return(result)
+}
+
+library(dplyr)
+
+dataset <- "fitbitecg"
+
+cat(glue::glue("Transforming data for {dataset}"),"\n")
+
+# Get variables for this dataset
+vars <- 
+  selected_vars %>% 
+  filter(grepl(paste0(dataset, "$"), Export, ignore.case = TRUE, perl = TRUE)) %>% 
+  pull(Variable)
+
+# Load the desired subset of this dataset in memory and do some feature engineering for derived variables
+df <- 
+  arrow::open_dataset(file.path(downloadLocation, glue::glue("dataset_{dataset}"))) %>% 
+  select(all_of(c(vars))) %>% 
+  filter(ResultClassification %in% c("Normal Sinus Rhythm", "Atrial Fibrillation")) %>% 
+  rename(StartDate = StartTime) %>% 
+  mutate(EndDate = base_s3_uri %>% stringr::str_extract("[0-9]{4}-[0-9]{2}-[0-9]{2}")) %>%
+  collect()
+
+colnames(df) <- tolower(colnames(df))
+
+# Create lists for ID variables and i2b2 concept variables
+excluded_concepts <- c("participantidentifier", "startdate", "enddate", "starttime")
+
+approved_concepts_summarized <- 
+  setdiff(
+    tolower(selected_vars$Variable[selected_vars$Export==dataset]),
+    excluded_concepts
+  )
+
+# Pivot data frame from long to wide
+df_melted_filtered <- 
+  df %>% 
+  mutate("SinusRhythm" = if_else(resultclassification == "Normal Sinus Rhythm", 1, NA),
+         "AtrialFibrillation" = if_else(resultclassification == "Atrial Fibrillation", 1, NA)) %>% 
+  select(-c(resultclassification)) %>% 
+  recoverSummarizeR::melt_df(excluded_concepts = excluded_concepts) %>% 
+  select(if("participantidentifier" %in% colnames(.)) "participantidentifier",
+         dplyr::matches("(?<!_)date(?!_)", perl = T),
+         if("concept" %in% colnames(.)) "concept",
+         if("value" %in% colnames(.)) "value") %>% 
+  tidyr::drop_na("value") %>% 
+  mutate(value = as.numeric(value))
+cat("recoverSummarizeR::melt_df() completed.\n")
+
+# Generate i2b2 summaries
+df_summarized <- 
+  df_melted_filtered %>% 
+  rename(startdate = dplyr::any_of(c("date", "datetime"))) %>% 
+  mutate(enddate = if (!("enddate" %in% names(.))) NA else enddate) %>% 
+  select(all_of(c("participantidentifier", "startdate", "enddate", "concept", "value"))) %>% 
+  ecg_stat_summarize() %>% 
+  mutate(value = as.numeric(value)) %>% 
+  distinct()
+cat("recoverSummarizeR::stat_summarize() completed.\n")
+
+# Add i2b2 columns from concept map (ontology file) and clean the output
+output_concepts <- 
+  process_df(df_summarized, 
+             concept_map, 
+             concept_replacements_reversed, 
+             concept_map_concepts = "CONCEPT_CD", 
+             concept_map_units = "UNITS_CD") %>% 
+  dplyr::mutate(nval_num = signif(nval_num, 9)) %>% 
+  dplyr::arrange(concept) %>% 
+  dplyr::mutate(dplyr::across(.cols = dplyr::everything(), .fns = as.character)) %>% 
+  replace(is.na(.), "<null>") %>% 
+  dplyr::filter(nval_num != "<null>" | tval_char != "<null>")
+cat("recoverSummarizeR::process_df() completed.\n")
+
+# Identify the participants who have output concepts derived from fitbit variables
+curr_fitbit_participants <- 
+  sort(unique(output_concepts$participantidentifier)) %>% 
+  as.data.frame() %>% 
+  dplyr::rename(participantidentifier = ".")
+
+prev_fitbit_participants <- 
+  read.csv(file.path(outputConceptsDir, "fitbit_participants.csv"))
+
+fitbit_participants <- 
+  dplyr::bind_rows(prev_fitbit_participants, 
+                   curr_fitbit_participants) %>% 
+  distinct()
+
+fitbit_participants %>% 
+  write.csv(file.path(outputConceptsDir, "fitbit_participants.csv"), 
+            row.names = F)
+
+# Write the output
+output_concepts %>% 
+  write.csv(file.path(outputConceptsDir, glue::glue("{dataset}.csv")), row.names = F)
+cat(glue::glue("output_concepts written to {file.path(outputConceptsDir, paste0(dataset, '.csv'))}"), "\n")
+
+cat(glue::glue("Finished transforming data for {dataset}"),"\n\n")
+
+# Remove objects created here from the global environment
+rm(ecg_stat_summarize,
+   dataset,
+   vars, 
+   df, 
+   excluded_concepts, 
+   approved_concepts_summarized, 
+   df_melted_filtered, 
+   df_summarized, 
+   output_concepts,
+   curr_fitbit_participants, 
+   prev_fitbit_participants, 
+   fitbit_participants)