Mercurial > repos > prog > lcmsmatching
diff search-mz @ 0:e66bb061af06 draft
planemo upload for repository https://github.com/workflow4metabolomics/lcmsmatching.git commit 3529b25417f8e1a5836474c9adec4b696d35099d-dirty
author | prog |
---|---|
date | Tue, 12 Jul 2016 12:02:37 -0400 |
parents | |
children | 253d531a0193 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/search-mz Tue Jul 12 12:02:37 2016 -0400 @@ -0,0 +1,542 @@ +#!/usr/bin/env Rscript +# vi: ft=R +args <- commandArgs(trailingOnly = F) +script.path <- sub("--file=","",args[grep("--file=",args)]) +library(getopt) +source(file.path(dirname(script.path), 'msdb-common.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'MsFileDb.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'MsPeakForestDb.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'MsXlsDb.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'Ms4TabSqlDb.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'MsDbLogger.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'MsDbInputDataFrameStream.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'MsDbOutputDataFrameStream.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'htmlhlp.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'strhlp.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'fshlp.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'biodb-common.R'), chdir = TRUE) +source(file.path(dirname(script.path), 'nethlp.R'), chdir = TRUE) + +############# +# CONSTANTS # +############# + +PROG <- sub('^.*/([^/]+)$', '\\1', commandArgs()[4], perl = TRUE) + +# Authorized database types +MSDB.XLS <- 'xls' +MSDB.4TABSQL <- '4tabsql' +MSDB.FILE <- 'file' +MSDB.PEAKFOREST <- 'peakforest' +MSDB.VALS <- c(MSDB.XLS, MSDB.4TABSQL, MSDB.FILE, MSDB.PEAKFOREST) + +# Authorized mode values +POS_MODE <- 'pos' +NEG_MODE <- 'neg' +MSDB.MODE.VALS <- c(POS_MODE, NEG_MODE) + +# Default +MSDB.DFT <- list() +MSDB.DFT[['mzshift']] <- 0 # in ppm +MSDB.DFT[['mzprec']] <- 5 # in ppm +MSDB.DFT[['mztolunit']] <- MSDB.DFT.MZTOLUNIT +MSDB.DFT[['precursor-rt-tol']] <- 5 +MSDB.DFT[['molids-sep']] <- MSDB.DFT.MATCH.SEP +MSDB.DFT[['db-fields']] <- concat.kv.list(msdb.get.dft.db.fields()) +MSDB.DFT[['db-ms-modes']] <- concat.kv.list(MSDB.DFT.MODES) +MSDB.DFT[['input-col-names']] <- concat.kv.list(msdb.get.dft.input.fields()) +MSDB.DFT[['output-col-names']] <- concat.kv.list(msdb.get.dft.output.fields()) +MSDB.DFT[['pos-prec']] <- paste(MSDB.DFT.PREC[[MSDB.TAG.POS]], collapse = ',') +MSDB.DFT[['neg-prec']] <- paste(MSDB.DFT.PREC[[MSDB.TAG.NEG]], collapse = ',') + +############## +# PRINT HELP # +############## + +print.help <- function(spec, status = 0) { + cat(getopt(spec, usage = TRUE, command = PROG)) + q(status = status) +} + +############################### +# SET DEFAULT ARGUMENT VALUES # +############################### + +set.dft.arg.val <-function(opt) { + + for (f in names(MSDB.DFT)) + if (is.null(opt[[f]])) + opt[[f]] <- MSDB.DFT[[f]] + + # Set default values + if ( opt$database == MSDB.XLS && ! is.null(opt$url) && is.null(opt[['cache-dir']])) + opt[['cache-dir']] <- file.path(opt$url, 'cache') + + if ( ! is.null(opt$rtcol) && opt$rtcol == '') + opt$rtcol <- NULL + + return(opt) +} + +######################### +# PARSE ARGUMENT VALUES # +######################### + +parse.arg.val <- function(opt) { + + # Parse input column names + if ( ! is.null(opt[['db-fields']])) { + cust <- split.kv.list(opt[['db-fields']]) + opt[['db-fields']] <- split.kv.list(MSDB.DFT[['db-fields']]) + opt[['db-fields']][names(cust)] <- cust + } + + # Parse MS modes + if ( ! is.null(opt[['db-ms-modes']])) { + cust <- split.kv.list(opt[['db-ms-modes']]) + opt[['db-ms-modes']] <- split.kv.list(MSDB.DFT[['db-ms-modes']]) + opt[['db-ms-modes']][names(cust)] <- cust + } + + # Parse retention time columns + if ( ! is.null(opt$rtcol)) + opt$rtcol <- strsplit(opt$rtcol, ',')[[1]] + + # Parse input column names + if ( ! is.null(opt[['input-col-names']])) { + custcols <- split.kv.list(opt[['input-col-names']]) + dftcols <- split.kv.list(MSDB.DFT[['input-col-names']]) + opt[['input-col-names']] <- c(custcols, dftcols[ ! names(dftcols) %in% names(custcols)]) + } + + # Parse output column names + if ( ! is.null(opt[['output-col-names']])) { + custcols <- split.kv.list(opt[['output-col-names']]) + dftcols <- split.kv.list(MSDB.DFT[['output-col-names']]) + opt[['output-col-names']] <- c(custcols, dftcols[ ! names(dftcols) %in% names(custcols)]) + } + + # Parse lists of precursors + if ( ! is.null(opt[['pos-prec']])) + opt[['pos-prec']] <- split.str(opt[['pos-prec']], unlist = TRUE) + if ( ! is.null(opt[['neg-prec']])) + opt[['neg-prec']] <- split.str(opt[['neg-prec']], unlist = TRUE) + + return(opt) +} + +################################# +# PRINT DEFAULT ARGUMENT VALUES # +################################# + +print.dft.arg.val <- function(opt) { + + print.flags <- MSDB.DFT + names(print.flags) <- vapply(names(print.flags), function(x) paste0('print-', x), FUN.VALUE = '') + for (f in names(print.flags)) + if ( ! is.null(opt[[f]])) { + cat(print.flags[[f]]) + q(status = 0) + } +} + +make.getopt.spec.print.dft <- function() { + + spec <- character() + + for (f in names(MSDB.DFT)) + spec <- c(spec, paste0('print-', f), NA_character_, 0, 'logical', paste0('Print default value of --', f)) + + return(spec) +} + +############################## +# MAKE GETOPT SPECIFICATIONS # +############################## + +make.getopt.spec <- function() { + spec = c( + 'help', 'h', 0, 'logical', 'Print this help.', + 'mode', 'm', 1, 'character', paste0('MS mode. Possible values are:', paste(MSDB.MODE.VALS, collapse = ", "), '.'), + 'mzshift', 's', 1, 'numeric', paste0('Shift on m/z, in ppm. Default is ', MSDB.DFT$mzshift,'.'), + 'mzprec', 'p', 1, 'numeric', paste0('Tolerance on m/z, in ppm. Default is ', MSDB.DFT$mzprec,'.'), + 'mztolunit', NA_character_, 1, 'character', paste0('Tolerance on m/z, in ppm. Default is ', MSDB.DFT$mztolunit,'.'), + 'rttol', 'r', 1, 'numeric', paste0('Tolerance on retention times. Unset by default.'), + 'rttolx', 'x', 1, 'numeric', paste0('Tolerance on retention times. Unset by default.'), + 'rttoly', 'y', 1, 'numeric', paste0('Tolerance on retention times. Unset by default.'), + 'rtcol', 'c', 1, 'character', paste0('Chromatographic column to use. Unset by default. If set, use the corresponding column to filter on retention times, if retention times are provided.'), + 'all-cols', NA_character_, 0, 'logical', 'Use all available chromatographic columns to match retention times.', + 'check-cols', NA_character_, 0, 'logical', 'Check that the chromatographic column names specified with option -c really exist.', + 'list-cols', NA_character_, 0, 'logical', 'List all chromatographic columns present in the database. Write list inside the file specified by -o option.', + 'same-rows', 'a', 0, 'logical', 'If set, output exactly the same number of rows as the input. This means that in case of multiple matches for one mz, then only one line is output (i.e.: the mz value is not duplicated on several lines). In the main output file, an "ms.matching" column is output with inside, for each mz, a comma separated list of matched component/molecule IDs. If unset, then only the main output file is used, and one single is written to it with one line per peak match, and eventual mz line duplicated if there are multiple matches for this mz.', + 'same-cols', 'b', 0, 'logical', 'If set, output the same columns as inside the input. All input columns are copied to the output.', + 'input-file', 'i', 1, 'character', 'Set input file.', + 'output-file', 'o', 1, 'character', 'Set file to use for the main output.', + 'peak-output-file', NA_character_, 1, 'character', 'If set and if --same-rows is set, then output all matches inside the specified file, with one mz match per line. The output columns are: mz, rt, id, col, colrt, composition, attribution. This means that if an mz value is matched several times, then it will repeated on several lines, with one match description per line.', + 'html-output-file', NA_character_, 1, 'character', 'Set file to use for the HTML output.', + 'no-main-table-in-html-output', NA_character_, 0, 'logical', 'Do not display main table in HTML output.', + 'precursor-match', NA_character_, 0, 'logical', 'Remove peaks whose molecule precursor peak has not been matched. Unset by default.', + 'precursor-rt-tol', NA_character_, 1, 'numeric', paste0('Precursor retention time tolerance. Only used when precursor-match is enabled. Default is ', MSDB.DFT[['precursor-rt-tol']], '.'), + 'pos-prec', NA_character_, 1, 'character', paste0('Set the list of precursors to use in positive mode. Default is "', MSDB.DFT[['pos-prec']], '".'), + 'neg-prec', NA_character_, 1, 'character', paste0('Set the list of precursors to use in negative mode. Default is "', MSDB.DFT[['neg-prec']], '".'), + 'input-col-names', NA_character_, 1, 'character', paste0('Set the input column names. Default is "', MSDB.DFT[['input-col-names']], '".'), + 'output-col-names', NA_character_, 1, 'character', paste0('Set the output column names. Default is "', MSDB.DFT[['output-col-names']], '".'), + 'molids-sep', NA_character_, 1, 'character', paste0('Set character separator used to when concatenating molecule IDs in output. Default is "', MSDB.DFT[['molids-sep']] , '".'), + 'first-val', NA_character_, 0, 'logical', 'Keep only the first value in multi-value fields. Unset by default.', + 'excel2011comp', NA_character_, 0, 'logical', 'Excel 2011 compatiblity mode. Output ASCII text files instead of UTF-8 files, where greek letters are replaced with their latin names, plusminus sign is replaced with +- and apostrophe is replaced with \"prime\". All other non-ASCII characters are repladed with underscore.', + 'database', 'd', 1, 'character', paste0('Set database to use: "xls" for an Excel database, "file" for a single file database, "4tabsql" for a 4Tab SQL database, and "peakforest" for a connection to PeakForest database.'), + 'url', NA_character_, 1, 'character', 'URL of database. For "peakforest" database it is the HTTP URL, for the "xls" database it is the path to the directory containing the Excel files, for the "file" database it is the path to the file database and for the "4tabsql" database it is the IP address of the server.', + 'cache-dir', NA_character_, 1, 'character', 'Path to directory where to store cache files. Only used when database flag is set to "xls".', + 'useragent', NA_character_, 1, 'character', 'User agent. Used by the "Peakforest" database.', + 'db-name', NA_character_, 1, 'character', 'Name of the database. Used by the "4tabsql" database.', + 'db-user', NA_character_, 1, 'character', 'Name of the database. Used by the "4tabsql" database.', + 'db-password', NA_character_, 1, 'character', 'Name of the database. Used by the "4tabsql" database.', + 'db-fields', NA_character_, 1, 'character', paste0('Comma separated key/value list giving the field names to be used in the single file database (option --db-file). Default is "', MSDB.DFT[['db-fields']], '".'), + 'db-ms-modes', NA_character_, 1, 'character', paste0('Comma separated key/value list giving the MS modes to be used in the single file database (option --db-file). Default is "', MSDB.DFT[['db-ms-modes']], '".'), + 'debug', NA_character_, 0, 'logical', 'Set debug mode.' + ) + + spec <- c(spec, make.getopt.spec.print.dft()) + + if ( ! is.null(spec)) + spec <- matrix(spec, byrow = TRUE, ncol = 5) + + return(spec) +} + +############# +# READ ARGS # +############# + +read_args <- function() { + + # options + spec <- make.getopt.spec() + opt <- getopt(spec) + + # help + if ( ! is.null(opt$help)) + print.help(spec) + + print.dft.arg.val(opt) # Print default values + opt <- set.dft.arg.val(opt) # Set default values + opt <- parse.arg.val(opt) # Parse list values + + # Check values + error <- .check.db.conn.opts(opt) + if (is.null(opt[['output-file']])) { + warning("You must set a path for the output file.") + error <- TRUE + } + if (is.null(opt[['list-cols']])) { + if (is.null(opt[['input-file']])) { + warning("You must provide an input file.") + error <- TRUE + } + if (is.null(opt$mode) || ( ! opt$mode %in% MSDB.MODE.VALS)) { + warning("You must specify a mode through the --mode option.") + error <- TRUE + } + if (is.null(opt$mzprec)) { + warning("You must set a precision in MZ with the --mzprec option.") + error <- TRUE + } + if ( ( ! is.null(opt$rtcol) || ! is.null(opt[['all-cols']])) && (is.null(opt$rttolx) || is.null(opt$rttoly))) { + warning("When chromatographic columns are set, you must provide values for --rttolx and -rttoly.") + error <- TRUE + } + if (is.null(opt$mztolunit) || ( ! opt$mztolunit %in% MSDB.MZTOLUNIT.VALS)) { + warning("You must specify an M/Z tolerance unit through the --mztolunit option.") + error <- TRUE + } + } + + # help + if (error) + print.help(spec, status = 1) + + return(opt) +} + + ##################################### + # CHECK DATABASE CONNECTION OPTIONS # + ##################################### + + .check.db.conn.opts <- function(opt) { + + # Print default values + if ( ! is.null(opt[['print-db-fields']])) { + cat(MSDB.DFT[['db-fields']]) + q(status = 0) + } + if ( ! is.null(opt[['print-db-ms-modes']])) { + cat(MSDB.DFT[['db-ms-modes']]) + q(status = 0) + } + + # Check values + error <- FALSE + if (is.null(opt$database)) { + warning("You must provide a database type through --database option.") + error <- TRUE + } + if ( ! opt$database %in% MSDB.VALS) { + warning(paste0("Invalid value \"", opt$database, "\" for --database option.")) + error <- TRUE + } + if (opt$database == MSDB.FILE) { + if (is.null(opt$url)) { + warning("When using single file database, you must specify the location of the database file with option --url.") + error <- TRUE + } + if ( ! file.exists(opt$url)) { + warning(paste0("The file path \"", opt$url,"\" specified with --db-file option is not valid.")) + error <- TRUE + } + } + if (opt$database == MSDB.XLS) { + if (is.null(opt$url)) { + warning("When using Excel database, you must specify the location of the Excel files directory with option --url.") + error <- TRUE + } + if ( ! file.exists(opt$url)) { + warning(paste0("The directory path \"", opt$url,"\" specified with --xls-dir option is not valid.")) + error <- TRUE + } + } + if (opt$database == MSDB.4TABSQL) { + if (is.null(opt$url)) { + warning("When using 4Tab SQL database, you must specify the URL of the SQL server with option --url.") + error <- TRUE + } + if (is.null(opt[['db-name']])) { + warning("When using 4Tab SQL database, you must specify the database name through the --db-name option.") + error <- TRUE + } + if (is.null(opt[['db-user']])) { + warning("When using 4Tab SQL database, you must specify the database user through the --db-user option.") + error <- TRUE + } + if (is.null(opt[['db-password']])) { + warning("When using 4Tab SQL database, you must specify the database user password through the --db-password option.") + error <- TRUE + } + } + if (opt$database == MSDB.PEAKFOREST) { + if (is.null(opt$url)) { + warning("When using PeakForest database, you must specify the URL of the PeakForest server with option --url.") + error <- TRUE + } + if (is.null(opt$useragent)) { + warning("When using PeakForest database, you must specify a user agent with option --useragent.") + error <- TRUE + } + } + + return(error) + } + + ############################# + # DISPLAY COMMAND LINE HELP # + ############################# + + .disp.cmd.line.help <- function(optspec, opt, prog, error = FALSE) { + + if ( ! is.null(opt$help) || error ) { + cat(getopt(optspec, usage = TRUE, command = prog)) + q(status = 1) + } + } + + ################# + # LOAD DATABASE # + ################# + + .load.db <- function(opt) { + + if (is.null(opt[['pos-prec']]) && is.null(opt[['neg-prec']])) { + precursors <- NULL + } else { + precursors <- list() + precursors[[MSDB.TAG.POS]] <- opt[['pos-prec']] + precursors[[MSDB.TAG.NEG]] <- opt[['neg-prec']] + } + + db <- switch(opt$database, + peakforest = MsPeakForestDb$new(url = opt$url, useragent = opt$useragent), + xls = MsXlsDb(db_dir = opt$url, cache_dir = opt[['cache-dir']]), + '4tabsql' = Ms4TabSqlDb(host = extract.address(opt$url), port = extract.port(opt$url), dbname = opt[['db-name']], user = opt[['db-user']], password = opt[['db-password']]), + file = MsFileDb(file = opt$url), + NULL) + db$setPrecursors(precursors) + if (db$areDbFieldsSettable()) + db$setDbFields(opt[['db-fields']]) + if (db$areDbMsModesSettable()) + db$setDbMsModes(opt[['db-ms-modes']]) + db$addObservers(MsDbLogger$new()) + + return(db) + } + +############### +# OUTPUT HTML # +############### + +output.html <- function(db, main, peaks, file, opt, output.fields) { + + # Replace public database IDs by URLs + if ( ! is.null(peaks)) + for (extdb in c(MSDB.TAG.KEGG, MSDB.TAG.HMDB, MSDB.TAG.CHEBI, MSDB.TAG.PUBCHEM)) { + field <- output.fields[[extdb]] + if (field %in% colnames(peaks)) + peaks[[field]] <- vapply(peaks[[field]], function(id) paste0('<a href="', get.entry.url(class = extdb, accession = id, content.type = RBIODB.HTML), '">', id, '</a>'), FUN.VALUE = '') + } + + # Write HTML + html <- HtmlWriter(file = file) + html$writeBegTag('html') + html$writeBegTag('header') + html$writeTag('title', text = "LC/MS matching results") + html$writeBegTag('style') + html$write('table, th, td { border-collapse: collapse; }') + html$write('table, th { border: 1px solid black; }') + html$write('td { border-left: 1px solid black; border-right: 1px solid black; }') + html$write('th, td { padding: 5px; }') + html$write('th { background-color: LightBlue; }') + html$write('tr:nth-child(even) { background-color: LemonChiffon; }') + html$write('tr:nth-child(odd) { background-color: LightGreen; }') + html$writeEndTag('style') + html$writeEndTag('header') + html$writeBegTag('body') + html$writeTag('h1', text = "LC/MS matching") + + # Write parameters + html$writeTag('h2', text = "Parameters") + html$writeBegTag('ul') + html$writeTag('li', paste0("Mode = ", opt$mode, ".")) + html$writeTag('li', paste0("M/Z precision = ", opt$mzprec, ".")) + html$writeTag('li', paste0("M/Z shift = ", opt$mzshift, ".")) + html$writeTag('li', paste0("Precursor match = ", (if (is.null(opt[['precursor-match']])) "no" else "yes"), ".")) + if ( ! is.null(opt[['precursor-match']])) { + html$writeTag('li', paste0("Positive precursors = ", paste0(opt[['pos-prec']], collapse = ', '), ".")) + html$writeTag('li', paste0("Negative precursors = ", paste0(opt[['neg-prec']], collapse = ', '), ".")) + } + if ( ! is.null(opt$rtcol)) { + html$writeTag('li', paste0("Columns = ", paste(opt$rtcol, collapse = ", "), ".")) + html$writeTag('li', paste0("RTX = ", opt$rttolx, ".")) + html$writeTag('li', paste0("RTY = ", opt$rttoly, ".")) + if ( ! is.null(opt[['precursor-match']])) + html$writeTag('li', paste0("RTZ = ", opt[['precursor-rt-tol']], ".")) + } + html$writeEndTag('ul') + + # Write results + html$writeTag('h2', text = "Results") + results <- FALSE + if ( ! is.null(main) && nrow(main) > 0 && is.null(opt[['no-main-table-in-html-output']])) { + html$writeTag('h3', text = "Main output") + html$writeTable(main) + results <- TRUE + } + if ( ! is.null(peaks) && nrow(peaks) > 0) { + html$writeTag('h3', text = "Matched peaks") + html$writeTable(peaks) + results <- TRUE + } + if ( ! results) + html$writeTag('p', 'None.') + + html$writeEndTag('body') + html$writeEndTag('html') +} + +######## +# MAIN # +######## + +options(error = function() { traceback(2) ; quit(status = 1) }, warn = 2 ) + +# Read command line arguments +opt <- read_args() + +if (is.null(opt$debug)) { + options(error = function() { quit(status = 1) }, warn = 0 ) +} + +# Load database +db <- .load.db(opt) + +# Print columns +if ( ! is.null(opt[['list-cols']])) { + cols <- db$getChromCol() + df.write.tsv(cols, file = opt[['output-file']]) + q(status = 0) +} + +# Read input +if ( ! is.null(opt[['input-file']]) && ! file.exists(opt[['input-file']])) + stop(paste0("Input file \"", opt[['input-file']], "\" does not exist.")) +if (file.info(opt[['input-file']])$size > 0) { + + # Load file into data frame + input <- read.table(file = opt[['input-file']], header = TRUE, sep = "\t") + + # Convert each column that is identified by a number into a name + for (field in names(opt[['input-col-names']])) { + if ( ! opt[['input-col-names']][[field]] %in% colnames(input) && length(grep('^[0-9]+$', opt[['input-col-names']][[field]])) > 0) { + col.index <- as.integer(opt[['input-col-names']][[field]]) + if (col.index < 1 || col.index > length(colnames(input))) + stop(paste0("No column n°", col.index, " for input field ", field, ".")) + opt[['input-col-names']][[field]] <- colnames(input)[[col.index]] + } + } +} else { + input <- data.frame() + input[[opt[['input-col-names']][['mz']]]] <- double() + input[[opt[['input-col-names']][['rt']]]] <- double() +} + +# Check mz column +if ( ! opt[['input-col-names']][['mz']] %in% colnames(input)) + stop(paste0('No column named "', opt[['input-col-names']][['mz']], '" in input file.')) + +# Set columns 'all-cols' specified +if ( ! is.null(opt[['all-cols']])) + opt$rtcol <- db$getChromCol() + +# Check chrom columns +if ( ! is.null(opt[['check-cols']]) && ! is.null(opt$rtcol)) { + dbcols <- db$getChromCol() + unknown.cols <- opt$rtcol[ ! opt$rtcol %in% dbcols] + if (length(unknown.cols) > 0) { + stop(paste0("Unknown chromatographic column", (if (length(unknown.cols) > 1) 's' else ''), ': ', paste(unknown.cols, collapse = ', '), ".\nAllowed chromatographic column names are:\n", paste(dbcols, collapse = "\n"))) + } +} + +# Check that an RT column exists when using MZ/RT matching +if ( ! is.null(opt$rtcol) && ! opt[['input-col-names']][['rt']] %in% colnames(input)) + stop(paste0("You are running an MZ/RT match run on your input data, but no retention time column named '", opt[['input-col-names']][['rt']],"' can be found inside your input file.")) + +# Set streams +input.stream <- MsDbInputDataFrameStream$new(df = input, input.fields = opt[['input-col-names']]) +main.output <- MsDbOutputDataFrameStream$new(keep.unused = ! is.null(opt[['same-cols']]), output.fields = opt[['output-col-names']], one.line = ! is.null(opt[['same-rows']]), match.sep = opt[['molids-sep']], first.val = ! is.null(opt[['first-val']]), ascii = ! is.null(opt[['excel2011comp']]), nogreek = ! is.null(opt[['excel2011comp']]), noapostrophe = ! is.null(opt[['excel2011comp']]), noplusminus = ! is.null(opt[['excel2011comp']])) +peaks.output <- MsDbOutputDataFrameStream$new(keep.unused = ! is.null(opt[['same-cols']]), output.fields = opt[['output-col-names']], first.val = ! is.null(opt[['first-val']]), ascii = ! is.null(opt[['excel2011comp']]), nogreek = ! is.null(opt[['excel2011comp']]), noapostrophe = ! is.null(opt[['excel2011comp']]), noplusminus = ! is.null(opt[['excel2011comp']])) +invisible(db$setInputStream(input.stream)) +db$addOutputStreams(c(main.output, peaks.output)) + +# Set M/Z tolerance unit +db$setMzTolUnit(opt$mztolunit) + +# Search database +mode <- if (opt$mode == POS_MODE) MSDB.TAG.POS else MSDB.TAG.NEG +db$searchForMzRtList(mode = mode, shift = opt$mzshift, prec = opt$mzprec, rt.tol = opt$rttol, rt.tol.x = opt$rttolx, rt.tol.y = opt$rttoly, col = opt$rtcol, precursor.match = ! is.null(opt[['precursor-match']]), precursor.rt.tol = opt[['precursor-rt-tol']]) + +# Write output +# TODO Create a class MsDbOutputCsvFileStream +df.write.tsv(main.output$getDataFrame(), file = opt[['output-file']], row.names = FALSE) +if ( ! is.null(opt[['peak-output-file']])) + # TODO Create a class MsDbOutputCsvFileStream + df.write.tsv(peaks.output$getDataFrame(), file = opt[['peak-output-file']], row.names = FALSE) +if ( ! is.null(opt[['html-output-file']])) + # TODO Create a class MsDbOutputHtmlFileStream + output.html(db = db, main = main.output$getDataFrame(), peaks = peaks.output$getDataFrame(), file = opt[['html-output-file']], opt = opt, output.fields = opt[['output-col-names']])