feat: create cluster-mutation lookup tables

DESCRIPTION

@@ -9,26 +9,29 @@ Description: A pipeline for scanning a SARS-CoV-2 phylogeny for clades with outl
 Depends:
   R (>= 4.1.0)
 Imports: 
-  ape,
-  dplyr,
-  foreach,
-  ggiraph,
-  ggplot2,
-  ggtree,
-  glue,
-  htmlwidgets,
-  knitr,
-  lubridate,
-  mgcv,
-  mlesky,
-  phangorn (>= 2.9.0),
-  rlang,
-  Rmpi,
-  scales,
-  stats,
-  stringr, 
-  treedater,
-  utils
+    ape,
+    dplyr,
+    foreach,
+    ggiraph,
+    ggplot2,
+    ggtree,
+    glue,
+    htmlwidgets,
+    knitr,
+    lubridate,
+    mgcv,
+    mlesky,
+    phangorn (>= 2.9.0),
+    purrr,
+    readr,
+    rlang,
+    Rmpi,
+    scales,
+    stats,
+    stringr,
+    tibble,
+    treedater,
+    utils
 Suggests: 
     doMPI,
     svglite,

R/parse_sc0.R

@@ -22,3 +22,28 @@ get_mutation_list <- function(sc0) {
 
   cmuts
 }
+
+#' Extract data.frames containing the 'defining' and 'all' mutations for each cluster
+#'
+#' @inheritParams   get_mutation_list
+#'
+#' @return   List of two data.frames with names "all" and "defining". These contain all mutations-
+#'   and just the defining mutations for each cluster in the phylogeny. The data.frames have
+#'   identical structure with column names \code{cluster_id} and \code{mutation}.
+
+get_mutation_tables <- function(sc0) {
+  mutation_list <- get_mutation_list(sc0)
+  defining_mutations <- purrr::map_df(
+    mutation_list, ~ tibble::tibble(mutation = .x[["defining"]]),
+    .id = "cluster_id"
+  )
+  all_mutations <- purrr::map_df(
+    mutation_list, ~ tibble::tibble(mutation = .x[["all"]]),
+    .id = "cluster_id"
+  )
+
+  list(
+    defining = defining_mutations,
+    all = all_mutations
+  )
+}

R/treeview.R

@@ -52,6 +52,10 @@ treeview <- function(e0,
   # node (row) in sc0
   cmuts <- get_mutation_list(sc0)
 
+  cmut_tables <- get_mutation_tables(sc0)
+  readr::write_csv(cmut_tables[["defining"]], file.path(output_dir, "defining_mutations.csv"))
+  readr::write_csv(cmut_tables[["all"]], file.path(output_dir, "all_mutations.csv"))
+
   tr1 <- e0$tre
 
   stopifnot(all(branch_cols %in% colnames(e0$Y)))

inst/WORDLIST

@@ -40,11 +40,14 @@ phangorn
 phylo
 phylodynamics
 polymorphism
+purrr
+readr
 Reformats
 repo
 rlang
 Rmpi
 RoxygenNote
+sc
 scannint
 sina
 stringr
@@ -56,6 +59,7 @@ SystemRequirements
 testthat
 tfpscanner
 Thr
+tibble
 tis
 tooltips
 treedata