library(msPurity)
library(optparse)
print(sessionInfo())

option_list <- list(
    make_option(c("-o", "--out_dir"), type = "character"),
    make_option("--mzML_files", type = "character"),
    make_option("--galaxy_names", type = "character"),
    make_option("--minOffset", type = "numeric"),
    make_option("--maxOffset", type = "numeric"),
    make_option("--ilim", type = "numeric"),
    make_option("--iwNorm", default = "none", type = "character"),
    make_option("--exclude_isotopes", action = "store_true"),
    make_option("--isotope_matrix", type = "character"),
    make_option("--mostIntense", action = "store_true"),
    make_option("--plotP", action = "store_true"),
    make_option("--nearest", action = "store_true"),
    make_option("--cores", default = 4),
    make_option("--ppmInterp", default = 7)
)

opt <- parse_args(OptionParser(option_list = option_list))
print(opt)

if (opt$iwNorm == "none") {
    iwNorm <- FALSE
    iwNormFun <- NULL
} else if (opt$iwNorm == "gauss") {
    iwNorm <- TRUE
    if (is.null(opt$minOffset) || is.null(opt$maxOffset)) {
        print("User has to define offsets if using Gaussian normalisation")
    } else {
        iwNormFun <- msPurity::iwNormGauss(
            minOff = -as.numeric(opt$minOffset),
            maxOff = as.numeric(opt$maxOffset)
        )
    }
} else if (opt$iwNorm == "rcosine") {
    iwNorm <- TRUE
    if (is.null(opt$minOffset) || is.null(opt$maxOffset)) {
        print("User has to define offsets if using R-cosine normalisation")
    } else {
        iwNormFun <- msPurity::iwNormRcosine(
            minOff = -as.numeric(opt$minOffset),
            maxOff = as.numeric(opt$maxOffset)
        )
    }
} else if (opt$iwNorm == "QE5") {
    iwNorm <- TRUE
    iwNormFun <- msPurity::iwNormQE.5()
}

filepaths <- trimws(strsplit(opt$mzML_files, ",")[[1]])
filepaths <- filepaths[filepaths != ""]



if (is.null(opt$minOffset) || is.null(opt$maxOffset)) {
    offsets <- NA
} else {
    offsets <- as.numeric(c(opt$minOffset, opt$maxOffset))
}


if (is.null(opt$mostIntense)) {
    mostIntense <- FALSE
} else {
    mostIntense <- TRUE
}

if (is.null(opt$nearest)) {
    nearest <- FALSE
} else {
    nearest <- TRUE
}

if (is.null(opt$plotP)) {
    plotP <- FALSE
    plotdir <- NULL
} else {
    plotP <- TRUE
    plotdir <- opt$out_dir
}


if (is.null(opt$isotope_matrix)) {
    im <- NULL
} else {
    im <- read.table(opt$isotope_matrix,
        header = TRUE, sep = "\t", stringsAsFactors = FALSE
    )
}

if (is.null(opt$exclude_isotopes)) {
    isotopes <- FALSE
} else {
    isotopes <- TRUE
}

pa <- msPurity::purityA(filepaths,
    cores = opt$cores,
    mostIntense = mostIntense,
    nearest = nearest,
    offsets = offsets,
    plotP = plotP,
    plotdir = plotdir,
    interpol = "linear",
    iwNorm = iwNorm,
    iwNormFun = iwNormFun,
    ilim = opt$ilim,
    mzRback = "pwiz",
    isotopes = isotopes,
    im = im,
    ppmInterp = opt$ppmInterp
)


if (!is.null(opt$galaxy_names)) {
    galaxy_names <- trimws(strsplit(opt$galaxy_names, ",")[[1]])
    galaxy_names <- galaxy_names[galaxy_names != ""]
    names(pa@fileList) <- galaxy_names
}

print(pa)
save(pa, file = file.path(opt$out_dir, "purityA_output.RData"))

pa@puritydf$filename <- sapply(pa@puritydf$fileid, function(x) names(pa@fileList)[as.integer(x)])

print(head(pa@puritydf))
write.table(pa@puritydf, file.path(opt$out_dir, "purityA_output.tsv"), row.names = FALSE, sep = "\t")