/******************************************************************************
 *
 * Copyright (C) 2020 by
 * The Salk Institute for Biological Studies
 *
 * Use of this source code is governed by an MIT-style
 * license that can be found in the LICENSE file or at
 * https://opensource.org/licenses/MIT.
 *
******************************************************************************/

#include "nfsim_species_unifier.h"

#include <iostream>
#include <fstream>

using namespace std;

extern int bngldebug;

namespace MCell {

bool NFSimSpeciesUnifier::read_input_file(const std::string& input_file, const bool is_dat) {

  //bngldebug = 1;

  ifstream ifs;
  ifs.open(input_file, ifstream::in);
  if (!ifs.is_open()) {
    cerr << "Could not open input file " << input_file << ".\n";
    return false;
  }

  string line;
  uint linenr = 0;
  while (getline(ifs, line)) {
    linenr++;
    if (line.empty()) {
      continue;
    }
    if (line[0] == '#') {
      // comment
      continue;
    }

    // expecting that each line will be in this form:
    // NFSim Species:
    //    BNGL_COMPLEX count
    // .dat (MCell viz output)
    //    BNGL_COMPLEX position normal
    size_t cplx_end = line.find(' ');
    if (cplx_end == string::npos) {
      cerr << "Did not find a separator ' ' on line " << linenr << ".\n";
      return false;
    }

    string cplx_string = line.substr(0, cplx_end);
    string count_string;
    if (!is_dat) {
      count_string = line.substr(cplx_end);
    }
    else {
      count_string = "1"; // each line in .dat files is one instance
    }

    // parse cplx string
    BNG::Cplx cplx_inst(&bng_engine.get_data());
    int num_errors = BNG::parse_single_cplx_string(
        cplx_string, bng_engine.get_data(),
        cplx_inst
    );
    if (num_errors != 0) {
      cerr << "Could not parse complex instance '" << cplx_string << "' on line " << linenr << ".\n";
      ifs.close();
      return false;
    }
    assert(!cplx_inst.elem_mols.empty());

    // see what species it is
    BNG::Species new_species = BNG::Species(cplx_inst, bng_engine.get_data(), bng_engine.get_config(), false);
    BNG::species_id_t id = bng_engine.get_all_species().find_or_add(new_species);

    // convert count
    double count;
    try {
      count = stod(count_string);
    }
    catch (const std::invalid_argument& ia) {
      cerr << "Could not convert count '" << count_string << "' on line " << linenr << ".\n";
      ifs.close();
      return false;
    }

    // remember this count
    auto it = counts_per_unique_species.find(id);
    if (it == counts_per_unique_species.end()) {
      counts_per_unique_species[id] = count;
    }
    else {
      counts_per_unique_species[id] += count;
    }
  }

  return true;
}


// out_file may be an empty string, in this case the output is printed to
bool NFSimSpeciesUnifier::print_unified_species(const std::string& out_file) {
  ostream* out;
  ofstream ofile;
  if (out_file != "") {
    ofile.open(out_file, ofstream::out);
    if (!ofile.is_open()) {
      cerr << "Could not open output file " << out_file << ".\n";
      return false;
    }
    out = &ofile;
  }
  else {
    out = &cout;
  }

  for (auto it: counts_per_unique_species) {
    BNG::Species& s = bng_engine.get_all_species().get(it.first);
    *out << s.to_str() << " " << it.second << "\n";
  }

  if (out_file != "") {
    ofile.close();
  }
  return true;
}

} // namespace MCell