#include <memory>
#include <iostream>
#include <sstream>
#include <limits>
#include <unistd.h>

// ROOT include files.
#include <TROOT.h>
#include <TString.h>

// COMET Include Files
#include <IMCHit.hxx>
#include <IDataHit.hxx>
#include "ICOMETEvent.hxx"
#include "ICOMETInput.hxx"
#include "ICOMETOutput.hxx"
#include "IOADatabase.hxx"
#include "IHandleHack.hxx"
#include "ICOMETLog.hxx"
#include "EoaCore.hxx"
#include "IRealDatum.hxx"
#include <IAlgorithm.hxx>
#include "IMidasFile.hxx"
#include <ITpcBank.hxx>
#include <TRawTpcChanItr.hxx>
#include <TRawTpcChan.hxx>
#include <TRawTpcPulseItr.hxx>
#include <TRawTpcPulse.hxx>
#include "ICOMETRawEvent.hxx"
#include "ICalibManager.hxx"

using namespace std;

///    \brief Converts raw events in midas files to comet events  
///    \author Christian Hansen, hansen@ific.uv.ca, Thomas Lindner, lindner@phas.ubc.ca
///
///    oaRawEventDecoder
///
/// An example how to read in midas files and convert the raw events 
/// and produce an cometevent file (for ReconGlobal)


void usage() {
  std::cerr<<"Usage: oaRawEventDecoder.exe [-c] <outputRootFileName> <inputMidasFileName1> <inputMidasFileName2> ... "<<std::endl;
}

int main(int argc, char *argv[])  {
  //Check if the number of input arguments are correct
  if (argc <= 2) {
    usage();
    exit(1);
  }
  
  //Find out if we want to calibrate or not
  bool calibrate = false;
  int inputArgNumberForOutput = 1;
  std::string firstArg = argv[1];
  if (firstArg == "-c") {
    inputArgNumberForOutput++;
    calibrate = true;
  }

  //First input argument after the calibration flag (if there is any) 
  //is the file name of the root file that we will create 
  std::string outputName = argv[inputArgNumberForOutput];
  COMET::ICOMETOutput *output = output = new COMET::ICOMETOutput(outputName.c_str(),"RECREATE");
  if (!output->IsOpen()) {
    std::cerr<<"ERROR: Output file, "<<outputName<<" not open"<<std::endl;
    exit(1);
  }    

  //Get an instante of the calibration manager (not yet initiate the TXxxCalibs)
  COMET::ICalibManager calibManager;
  //Initiate the TXxxCalibs that each own calib stores                                                                                       
  calibManager.Initiate();
  if (calibrate) {
    //Fill the calib stores with calibration parameters from the DB
    calibManager.RetrieveCalibParametersFromDB(0, 1000);
  }

  //This is a good way to get the geometry: 
  //TString geometryFile = getenv("CALIBGLOBALROOT") + TString("/geometry/singleFGDGeometryFile.root");
  //TEMP but for now (until I figured out how to set CALIBGLOBALROOT and how to make a only-geometry root file) just get a MC file
  TString geometryFileName = TString("/cometrep/Data/m11-mu-dist5.es.root" );
  COMET::ICOMETInput geoinput(geometryFileName, "OLD");
  if (!geoinput.IsOpen()) {
    std::cout<<"ERROR: File not found " <<geometryFileName<<std::endl;
    return -1;
  } 
  
  // Write the ROOT geometry to output file.
  if (output) {
    try {
      output->WriteGeometry(COMET::IOADatabase::Get().Geometry());
    }
    catch (COMET::ENoGeometry) {
      std::cout<<"ERROR: Geometry not written to output"<<std::endl;
    }
  } else std::cout<<"ERROR, output not opened"<<std::endl;

  // Loop over input MIDAS files
  int eventCounter = 0;
  COMET::IMidasFile midasFile;
  for (int i = inputArgNumberForOutput + 1; i < argc; i++){
    // Try to open the midas file number i + inputArgNumberForOutput
    if (!midasFile.Open(argv[i])) {
      std::cout<<"ERROR, cannot open midas input file "<<argv[i]<<std::endl;
      return -1;
    } else {
      std::cout<<"Opened midas file "<<argv[i]<<std::endl;
      
      //Read in all raw events from this file 
      while ( const COMET::ICOMETRawEvent* rawEvent = midasFile.ReadRawEvent() ) {
  
  //IF MIDAS FILES FROM DIFFERENT SUBDETECTORS ARE READ IN, E.G. midTPC.mid.gz and midFGD.mid.gz,
  //EVENTS WITH SAME EVENT NUMBERS SHOULD BE PUT IN SAME COMETEVENT, BUT FOR NOW I CREATE A 
  //NEW COMETEVENT FOR EACH MIDAS EVENT, EVEN IF THEY ARE FROM DIFFERENT SUBDETECTORS 
  //I KEEP IT LIKE THIS SINCE I DONT KNOW IF THERE WILL BE DIFFERENT MIDAS FILES FOR 
  //DIFFERENT SUBDETECTORS OR IF MIDAS FILES WILL CONTAIN DATA FROM THE WHOLE COMET

  // Create an COMET event   
  int serialNumber = rawEvent->GetHeader().GetSeqNo();
  std::auto_ptr<COMET::ICOMETEvent> cometEvent(new COMET::ICOMETEvent(0,serialNumber));
  //cout<<"made a new comet event with serial number "<<serialNumber<<endl;
  
  ///Fill an comet event with uncalibrated THitSelections (one for each TXxxCalib) converting info from a raw event
  calibManager.AddUncalibratedHits(*cometEvent, *rawEvent);

  if (calibrate) {
    // Add calibrated THitSelections (one for each TXxxCalib) using calibration parameters stored in the calib stores
    calibManager.AddCalibratedHits(*cometEvent);
  }

  // Write the COMET event to output file.
  if (output) {
    output->WriteEvent(*cometEvent);	
  }
  
  // Count events
  eventCounter++;
  if (eventCounter%20 == 0) std::cout << "Event: " << eventCounter<<std::endl;

  // Check for memory leak
  int lastRunId;
  cometEvent.reset(NULL);
  if (!COMET::CleanHandleRegistry()) {
    std::cout<<"WARNING: (COMET) Memory Leak in "<< std::endl;
    std::cout<<"    File(event,run): " <<outputName<<std::endl;
  }

  // Delete raw event
  delete rawEvent;
      }
    }
    
    // Close MIDAS file.
    midasFile.Close();
  
  }
  
  // Close output file
  if (output) output->Close();
  
  
  
  
} // AWESOMEE!!! DO TOMORROWWW!!!!