/**
 *  \file fitting_solutions_reader_writer.cpp
 *  \brief handles reading and writing of fitting solutions
 *           volume calculation.
 *
 *  Copyright 2007-2010 IMP Inventors. All rights reserved.
 *
 */

#include <IMP/multifit/fitting_solutions_reader_writer.h>
#include <boost/algorithm/string.hpp>
#include <boost/lexical_cast.hpp>
#include <boost/format.hpp>

IMPMULTIFIT_BEGIN_NAMESPACE

FittingSolutionRecord parse_fitting_line(const std::string &line) {
  FittingSolutionRecord fit_sol;
  typedef boost::split_iterator<std::string::iterator> string_split_iterator;
  IMP_USAGE_CHECK(line.size() > 0,"no data to parse"<<std::endl);
  IMP_LOG(VERBOSE,"going to parse:"<<line);
  std::vector<std::string> line_split,rotation_split,translation_split;
  boost::split(line_split, line, boost::is_any_of("|"));
  IMP_USAGE_CHECK(line_split.size() == 8,
     "FittingSolutionRecord::parse_fitting_line Wrong format of input line : "<<
      "not enough fields:"<<line);
  boost::split(rotation_split, line_split[2], boost::is_any_of(" "));
  IMP_USAGE_CHECK(rotation_split.size() == 4,
     "Wrong format of input line: wrong rotation format "<<
     "(expected 4 blocks and got "<< rotation_split.size()<<")"<<std::endl);
  IMP_LOG(VERBOSE,"going to parse translation:"<<line_split[3]<<std::endl);
  boost::split(translation_split, line_split[3], boost::is_any_of(" "));
  IMP_USAGE_CHECK(translation_split.size() == 3,
           "Wrong format of input line: wrong translation format"<<std::endl);
  fit_sol.set_index(boost::lexical_cast<int>(line_split[0]));
  fit_sol.set_solution_filename(
    boost::lexical_cast<std::string>(line_split[1]));
  fit_sol.set_transformation(
     algebra::Transformation3D(
       algebra::Rotation3D(
         boost::lexical_cast<float>(rotation_split[0]),
         boost::lexical_cast<float>(rotation_split[1]),
         boost::lexical_cast<float>(rotation_split[2]),
         boost::lexical_cast<float>(rotation_split[3])),
       algebra::Vector3D(
         boost::lexical_cast<float>(translation_split[0]),
         boost::lexical_cast<float>(translation_split[1]),
         boost::lexical_cast<float>(translation_split[2]))));
  IMP_LOG(VERBOSE,"tranformation is set:"
     <<fit_sol.get_transformation()<<std::endl);
  fit_sol.set_match_size(boost::lexical_cast<int>(line_split[4]));
  fit_sol.set_match_average_distance(
     boost::lexical_cast<float>(line_split[5]));
  fit_sol.set_fitting_score(boost::lexical_cast<float>(line_split[6]));
  fit_sol.set_rmsd_to_reference(boost::lexical_cast<float>(line_split[7]));
  IMP_LOG(VERBOSE,"finish parsing line"<<std::endl);
  return fit_sol;
}


FittingSolutionRecords read_fitting_solutions(const char *fitting_fn) {
  std::fstream in;
  FittingSolutionRecords sols;
  in.open(fitting_fn, std::fstream::in);
  if (! in.good()) {
    IMP_WARN("Problem openning file " << fitting_fn <<
                  " for reading; returning 0 solutions" << std::endl);
    in.close();
    return sols;
  }
  std::string line;
  getline(in, line); //skip header line
  while (!in.eof()) {
    if (!getline(in, line)) break;
    sols.push_back(parse_fitting_line(line));
  }
  in.close();
  return sols;
}

void write_fitting_solutions(const char *fitting_fn,
                             const FittingSolutionRecords &fit_sols,
                             int num_sols) {
  if (num_sols==-1) {
    num_sols = fit_sols.size();
  }
  std::fstream out;
  out.open(fitting_fn, std::fstream::out);
  IMP_USAGE_CHECK(out.good(), "Problem openning file " <<
                  fitting_fn << " for writing"<<std::endl);
  //write header
  out<<"solution index | solution filename | rotation | translation  |"
     <<" match size | match average distance | cluster size |"
     <<" fitting score | RMSD to reference"<<std::endl;
  for(int i=0;i<num_sols;i++) {
    fit_sols[i].show(out);
    out<<std::endl;
  }
  out.close();
}

IMPMULTIFIT_END_NAMESPACE