Merge pull request #419 from Liripo/patch-1

Update loadContigs.R
ncborcherding · Oct 15, 2024 · 3428d08 · 3428d08
2 parents e9d8f6d + 5ed7311
commit 3428d08
Showing 1 changed file with 14 additions and 13 deletions.
diff --git a/R/loadContigs.R b/R/loadContigs.R
@@ -43,18 +43,19 @@ loadContigs <- function(input,
                         format = "10X") {
   #Loading from directory, recursively
   if (inherits(x=input, what ="character")) {
-    format.list <- list("WAT3R" = "barcode_results.csv", 
-                        "10X" =  "filtered_contig_annotations.csv", 
-                        "AIRR" = "airr_rearrangement.tsv", 
-                        "Dandelion" = "all_contig_dandelion.tsv",
-                        "Immcantation" = "_data.tsv",
-                        "MiXCR" = "clones.tsv", 
-                        "JSON" = ".json",
-                        "TRUST4" = "barcode_report.tsv", 
-                        "BD" = "Contigs_AIRR.tsv",
-                        "Omniscope" =c("_OSB.csv", "_OST.csv"),
-                        "ParseBio" = "barcode_report.tsv")
-        file.pattern <- format.list[[format]]
+    file.pattern <- switch(format,
+                       "WAT3R" = "barcode_results.csv",
+                       "10X" = "filtered_contig_annotations.csv",
+                       "AIRR" = "airr_rearrangement.tsv",
+                       "Dandelion" = "all_contig_dandelion.tsv",
+                       "Immcantation" = "_data.tsv",
+                       "MiXCR" = "clones.tsv",
+                       "JSON" = ".json",
+                       "TRUST4" = "barcode_report.tsv",
+                       "BD" = "Contigs_AIRR.tsv",
+                       "Omniscope" = c("_OSB.csv", "_OST.csv"),
+                       "ParseBio" = "barcode_report.tsv",
+                       stop("Invalid format: ",format))
         contig.files <- list.files(input, paste0(file.pattern, collapse = "|"), recursive = TRUE, full.names = TRUE)
 
         if (format %in% c("10X", "WAT3R", "Omniscope")) {
@@ -285,4 +286,4 @@ loadContigs <- function(input,
     colnames(df[[i]]) <- c("barcode", "chain", "reads", "v_gene", "d_gene", "j_gene", "c_gene", "cdr3_nt", "cdr3", "productive")
   } 
   return(df)
-}
+}