Save vdj alignment sequences to a CSV

.gitignore

@@ -1,3 +1,3 @@
-*.txt
+*.csv
 *.fasta
 *.fastq

src/repertoire.r

@@ -19,21 +19,23 @@ amplify_rows <- function(data, column, factor) {
   return(rep(data, factor))
 }
 
-save_data <- function(data, name) {
-  if (name == "sequence") {
-    file_name <- paste("data/", name, ".fasta", sep = "")
-    Biostrings::writeXStringSet(data, file_name, format = "fasta")
-  } else {
-    file_name <- paste("data/", name, ".txt", sep = "")
-    cat(data, file = file_name, sep = "\n")
-  }
+save_data <- function(data) {
+  Biostrings::writeXStringSet(data$sequence, "data/sequence.fasta")
+  vdj_sequences <- data[-1]
+  write.csv(vdj_sequences, "data/vdj_alignment.csv", row.names = FALSE)
 }
 
 process_data <- function(repertoire, sequencing_runs) {
-  columns <- c("sequence", "v_call", "j_call")
+  columns <- c(
+    "sequence", "v_sequence_alignment",
+    "d_sequence_alignment", "j_sequence_alignment"
+  )
   data <- repertoire[, columns]
-  amplified_data <- mapply(data, names(data), sequencing_runs, FUN = amplify_rows)
-  invisible(mapply(amplified_data, names(amplified_data), FUN = save_data))
+  amplified_data <- mapply(data, names(data),
+    sequencing_runs,
+    FUN = amplify_rows
+  )
+  save_data(amplified_data)
 }
 
 parse_cli_arguments <- function(args) {
@@ -48,4 +50,4 @@ arguments <- parse_cli_arguments(args)
 number_of_sequences <- as.integer(arguments[1])
 sequencing_runs <- as.integer(arguments[2])
 repertoire <- generate_repertoire(number_of_sequences)
-process_data(repertoire, sequencing_runs)
+process_data(repertoire, sequencing_runs)