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


get_av_spectra <- function(x) {

  if (length(x$av_intra) > 0) {
    av_intra_df <- plyr::ldply(x$av_intra)

    if (nrow(av_intra_df) == 0) {
      av_intra_df <- NULL
    }else{
      av_intra_df$method <- "intra"
    }

  }else{
    av_intra_df <- NULL
  }

  if ((is.null(x$av_inter)) || (nrow(x$av_inter) == 0)) {
    av_inter_df <- NULL
  }else{
    av_inter_df <- x$av_inter
    av_inter_df$method <- "inter"
  }

  if ((is.null(x$av_all)) || (nrow(x$av_all) == 0)) {
    av_all_df <- NULL
  }else{
    av_all_df <- x$av_all
    av_all_df$method <- "all"
  }

  combined <- plyr::rbind.fill(av_intra_df, av_inter_df, av_all_df)

  return(combined)
}


option_list <- list(
  make_option("--out_rdata", type = "character"),
  make_option("--out_peaklist", type = "character"),
  make_option("--pa", type = "character"),
  make_option("--av_level", type = "character"),
  make_option("--minfrac", default = 0.5),
  make_option("--minnum", default = 1),
  make_option("--ppm", default = 5.0),
  make_option("--snr", default = 0),
  make_option("--ra", default = 0),
  make_option("--av", default = "median", type = "character"),
  make_option("--sumi", action = "store_true"),
  make_option("--rmp", action = "store_true"),
  make_option("--cores", default = 1)
)

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


load_r_data <- function(rdata_path, name) {
    #loads an RData file, and returns the named xset object if it is there
    load(rdata_path)
    return(get(ls()[ls() %in% name]))
}

# Requires
pa <- load_r_data(opt$pa, "pa")

pa@cores <- opt$cores

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

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

if (opt$av_level == "intra") {
  pa <- msPurity::averageIntraFragSpectra(pa,
                                          minfrac = opt$minfrac,
                                          minnum = opt$minnum,
                                          ppm = opt$ppm,
                                          snr = opt$snr,
                                          ra = opt$ra,
                                          av = opt$av,
                                          sumi = sumi,
                                          rmp = rmp,
                                          cores = opt$cores)

} else if (opt$av_level == "inter") {

  pa <- msPurity::averageInterFragSpectra(pa,
                                          minfrac = opt$minfrac,
                                          minnum = opt$minnum,
                                          ppm = opt$ppm,
                                          snr = opt$snr,
                                          ra = opt$ra,
                                          av = opt$av,
                                          sumi = sumi,
                                          rmp = rmp,
                                          cores = opt$cores)
} else if (opt$av_level == "all") {

  pa <- msPurity::averageAllFragSpectra(pa,
                                        minfrac = opt$minfrac,
                                        minnum = opt$minnum,
                                        ppm = opt$ppm,
                                        snr = opt$snr,
                                        ra = opt$ra,
                                        av = opt$av,
                                        sumi = sumi,
                                        rmp = rmp,
                                        cores = opt$cores)
}

print(pa)
save(pa, file = opt$out_rdata)

if (length(pa) > 0) {

  av_spectra <- plyr::ldply(pa@av_spectra, get_av_spectra)

  if (nrow(av_spectra) == 0) {
    message("No average spectra available")
  } else {
    colnames(av_spectra)[1] <- "grpid"
    av_spectra$grpid <- names(pa@av_spectra)[av_spectra$grpid]

    if ((length(pa@av_intra_params) > 0) || (length(pa@av_inter_params) > 0)) {
      # Add some extra info (only required if av_intra or av_inter performed)
      colnames(av_spectra)[2] <- "fileid"
      av_spectra$avid <- seq_len(nrow(av_spectra))

      filenames <- sapply(av_spectra$fileid,
                          function(x) names(pa@fileList)[as.integer(x)])
      # filenames_galaxy <- sapply(
      #    av_spectra$fileid, function(x) basename(pa@fileList[as.integer(x)]))

        av_spectra <- as.data.frame(
          append(av_spectra, list(filename = filenames), after = 2))
    }


    print(head(av_spectra))
    write.table(av_spectra, opt$out_peaklist, row.names = FALSE, sep = "\t")

  }
}