#include <iostream>
#include <iomanip>
#include <vector>
#include <algorithm>

#include "TROOT.h"
#include "TFile.h"

#include "JAcoustics/JEvent.hh"
#include "JAcoustics/JSupport.hh"

#include "JSupport/JSingleFileScanner.hh"

#include "JLang/JException.hh"
#include "JLang/JComparator.hh"
#include "JLang/JComparison.hh"

#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"

namespace {

  using JLANG::JValueOutOfRange;
  using JACOUSTICS::JTransmission;
  using JACOUSTICS::JEvent;

  enum JCompare_t {
    ToA_t = 1,
    ToE_t = 2
  };

  int cta = 0;   // time selection
  
  const struct {

    /**
     * Compare transmissions.
     *
     * \param  first        first  transmission
     * \param  second       second transmission
     * \return              true if first transmission less than second; else false
     */
    inline bool operator()(const JTransmission& first, const JTransmission& second) const
    {
      if (cta != ToA_t && cta != ToE_t) {
	THROW(JValueOutOfRange, "Invalid time selection " << cta);
      }
      
      if (first.getRunNumber() == second.getRunNumber()) {

	if (first.getID() == second.getID()) {

	  if ((cta == ToA_t && first.getToA() == second.getToA()) ||
	      (cta == ToE_t && first.getToE() == second.getToE())) {

	    if (first.getQ() == second.getQ()) {

	      return first.getW() < second.getW();

	    } else {

	      return first.getQ() < second.getQ();
	    }

	  } else {

	    if (cta == ToA_t) { return first.getToA() < second.getToA(); }
	    if (cta == ToE_t) { return first.getToE() < second.getToE(); }

	    THROW(JValueOutOfRange, "Invalid time selection " << cta);
	  }

	} else {

	  return first.getID() < second.getID();
	}

      } else {

	return first.getRunNumber() < second.getRunNumber();
      }
    }


    /**
     * Compare events.
     *
     * \param  first        first  event
     * \param  second       second event
     * \return              true if first event less than second; else false
     */
    inline bool operator()(const JEvent& first, const JEvent& second) const
    {
      if (first.getOID() == second.getOID()) {

	if (first.getID() == second.getID()) {

	  if (first.size() == second.size()) {

	    for (JEvent::const_iterator
		   p0 = first .begin(),
		   p1 = second.begin(); p0 != first.end() && p1 != second.end(); ++p0, ++p1) {

	      if ((*this)(*p0, *p1)) {
		return true;
	      }
	    }

	    return false;
	    
	  } else {

	    return first.size() < second.size();
	  }

	} else {

	  return first.getID() < second.getID();
	}

      } else {

	return first.getOID() < second.getOID();
      }
    }
  } compare;


  /**
   * Printer.
   */
  struct printer {

    int    debug;
    size_t width;

    /**
     * Print data.
     *
     * \param  out            output stream
     * \param  filename       file name
     * \param  index          index
     * \param  evt            event
     * \param  prefix         prefix
     * \param  postfix        postfix
     * \return                output stream
     */
    std::ostream& operator()(std::ostream& out, const std::string& filename, const int index, const JEvent& evt, const std::string& prefix, const std::string& postfix) const
    {
      using namespace std;
      using namespace JPP;
      
      if (debug >= debug_t) {
	out << prefix << (prefix == "" ? "" : " ")
	    << setw(width)  << left << filename << right    << ' '
	    << "[" << FILL(6,'0') << index << "]" << FILL() << ' '
	    << evt.getOID()                                 << ' '
	    << setw(6)      << evt.getCounter()             << ' ' 
	    << setw(2)      << evt.getID()                  << ' '
	    << FIXED(12,6)  << evt. begin()->getToA()       << ' '
	    << FIXED(12,6)  << evt.rbegin()->getToA()
	    << (postfix == "" ? "" :  " ") << postfix       << endl;
      }

      return out;
    }


    /**
     * Write transmission to output stream.
     *
     * \param  out            output stream
     * \param  object         transmission
     * \param  prefix         prefix
     * \param  postfix        postfix
     * \return                output stream
     */
    std::ostream& operator()(std::ostream& out, const JTransmission& object, const std::string& prefix, const std::string& postfix) const
    {
      using namespace std;
      using namespace JPP;

      if (debug >= debug_t) {
	out << prefix << (prefix == "" ? "" : " ")
	    << setw(8) << object.getID()                     << ' '
	    << setw(8) << object.getRunNumber()              << ' '
	    << FIXED(13,7) << object.getToA()                << ' '
	    << FIXED(13,7) << object.getToE()                << ' '
	    << FIXED(8,0) << object.getQ() 
	    << (postfix == "" ? "" :  " ") << postfix        << endl;
      }

      return out;
    }
  };


  /**
   * Compare acoustics events by time of arrival of first hit.
   *
   * \param  first          first  event
   * \param  second         second event
   * \return                true if first event earliear than second; else false
   */
  static inline bool toa(const JEvent& first, const JEvent& second)
  {
    return first.begin()->getToA() < second.begin()->getToA();
  }


  /**
   * Compare acoustics events by time of emission of first hit.
   *
   * \param  first          first  event
   * \param  second         second event
   * \return                true if first event earliear than second; else false
   */
  static inline bool toe(const JEvent& first, const JEvent& second)
  {
    return first.begin()->getToE() < second.begin()->getToE();
  }


  /**
   * Options.
   */
  enum JOption_t {
    Null_t = 0,
    Sort_t = 1
  };
}

/**
 * \file
 *
 * Program to compare acoustics event data.
 * \author mdejong
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  
  vector<string> inputFile;
  JLimit         numberOfEvents;
  int            option;
  int            debug;

  try {

    JParser<> zap("Program to compare acoustics event data.");
    
    zap['f'] = make_field(inputFile,  "two outputs of JAcousticsEventBuilder[.sh]");
    zap['n'] = make_field(numberOfEvents)                                    = JLimit::max();    
    zap['O'] = make_field(option,     Null_t << " -> nothing " << Sort_t << " -> sort")  = Null_t, Sort_t;
    zap['C'] = make_field(cta,        ToA_t  << " -> ToA "     << ToE_t  << " -> ToE")   = ToA_t,  ToE_t;
    zap['d'] = make_field(debug)                                             = 2;

    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }

  if (inputFile.size() != 2u) {
    FATAL("Wrong number of input files " << inputFile.size() << endl); 
  }

  const size_t  width = max(inputFile[0].size(), inputFile[1].size());
  const printer print = { debug, width };
  
  vector<JEvent> buffer[2];

  for (int i = 0; i != 2; ++i) {
    
    for (JSingleFileScanner<JEvent> in(inputFile[i], numberOfEvents); in.hasNext(); ) {
      buffer[i].push_back(*in.next());
    }
  }

  if (option == Sort_t) {

    for (int i = 0; i != 2; ++i) {

      for (JEvent& evt : buffer[i]) {
	sort(evt.begin(), evt.end(), compare);
      }

      if (cta == ToA_t) { sort(buffer[i].begin(), buffer[i].end(), toa); }
      if (cta == ToE_t) { sort(buffer[i].begin(), buffer[i].end(), toe); }

      for (vector<JEvent>::iterator p = buffer[i].begin(); p != buffer[i].end(); ++p) {
	sort(p->begin(), p->end(), compare);
      }
    }
  }

  int count[] = { 0, 0 };
  
  for (vector<JEvent>::const_iterator
	 p0 = buffer[0].begin(),
	 p1 = buffer[1].begin(); p0 != buffer[0].end() && p1 != buffer[1].end(); ) {

    for ( ; p0 != buffer[0].end() && p1 != buffer[1].end() && ((cta == ToA_t && toa(*p0,*p1)) || (cta == ToE_t && toe(*p0,*p1))); ++p0, ++count[1]) {
      print(cout, inputFile[0], distance(buffer[0].cbegin(),p0), *p0, ">>", "");
    }

    for ( ; p0 != buffer[0].end() && p1 != buffer[1].end() && ((cta == ToA_t && toa(*p1,*p0)) || (cta == ToE_t && toe(*p1,*p0))); ++p1, ++count[1]) {
      print(cout, inputFile[1], distance(buffer[1].cbegin(),p1), *p1, "<<", "");
    }

    if (p0 != buffer[0].end() && p1 != buffer[1].end()) {

      if (!compare(*p0,*p1) && !compare(*p1,*p0)) {

	++count[0];

	print(cout, inputFile[0], distance(buffer[0].cbegin(),p0), *p0, "", "\\");
	print(cout, inputFile[1], distance(buffer[1].cbegin(),p1), *p1, "", "/ ");

      } else {
	
	++count[1];

	print(cout, inputFile[0], distance(buffer[0].cbegin(),p0), *p0, "", "*");
	print(cout, inputFile[1], distance(buffer[1].cbegin(),p1), *p1, "", "*");

	if (p0->getOID()     == p1->getOID()     &&
	    p0->getCounter() == p1->getCounter() &&
	    p0->getID()      == p1->getID()) {

	  JEvent::const_iterator i0 = p0->begin();
	  JEvent::const_iterator i1 = p1->begin();

	  while (i0 != p0->end() && i1 != p1->end()) {

	    for ( ; i0 != p0->end() && i1 != p1->end() && compare(*i0,*i1); ++i0) {
	      print(cout, *i0, ">>", "");
	    }

	    for ( ; i0 != p0->end() && i1 != p1->end() && compare(*i1,*i0); ++i1) {
	      print(cout, *i1, "<<", "");
	    }

	    if (i0 != p0->end() && i1 != p1->end()) {
	      if (!compare(*i0, *i1) && !compare(*i1,*i0)) {
		++i0;
		++i1;
	      }
	    }
	  }

	  for ( ; i0 != p0->end(); ++i0) {
	    print(cout, *i0, ">>", "");
	  }

	  for ( ; i1 != p1->end(); ++i1) {
	    print(cout, *i1, "<<", "");
	  }

	} else {

	  print(cout, inputFile[0], distance(buffer[0].cbegin(),p0), *p0, ">>", "");
	  print(cout, inputFile[1], distance(buffer[1].cbegin(),p1), *p1, "<<", "");
	}
      }
      
      if (toa(*p0,*p1) || toa(*p1,*p0)) {

      } else {

	++p0;
	++p1;
      }
    }
  }

  STATUS("Number of differences / events: " << count[1] << " / " << count[0] << endl);
    
  if (buffer[0].size() != buffer[1].size()) {
    FATAL("Different size " << buffer[0].size() << ' ' << buffer[1].size() << endl);
  }
  
  if (count[1] != 0) {
    FATAL("Number of differences " << count[1] << endl);
  }
}