diff --git a/src/repertoire.r b/src/repertoire.r index 0cb40b8..dcf2909 100644 --- a/src/repertoire.r +++ b/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) \ No newline at end of file