// SanityCheckProc.cc
// Contact person: Gabriel Orebi Gann <gorebigann@lbl.gov>
// See SanityCheckProc.hh for more details
//———————————————————————//

#include <iostream>
#include <sstream>
#include <cmath>
#include <algorithm>
#include <vector>
#include <Randomize.hh>
#include <RAT/SanityCheckProc.hh>
#include <RAT/DB.hh>
#include <RAT/Log.hh>
#include <G4UImessenger.hh>

using namespace std;

namespace RAT {

SanityCheckProc::SanityCheckProc() : Processor("sanitychecks")
{
  fHitPMTs.resize(0);
  nPMTs = 0;
}

SanityCheckProc::~SanityCheckProc()
{
  fHitPMTs.resize(0);
}

void SanityCheckProc::BeginOfRun(DS::Run& run) {
  // Cheat: define size of PMT array based on DQCH array
  DBLinkPtr dqxxBank = DB::Get()->GetLink("PMT_DQXX");
  std::vector<int> dqch = dqxxBank->GetIArray("dqch");
  Log::Assert(!dqch.empty(), "SanityCheckProc: Unable to determine PMT count from DQCH data");
  nPMTs = (int) dqch.size();
  dqch.clear();
}

Processor::Result SanityCheckProc::DSEvent( DS::Run&,
                                            DS::Entry& ds)
{
  fHitPMTs.resize(0);

  // Loop over events
  for (size_t iev=0; iev<ds.GetEVCount(); ++iev){

    DS::EV& ev = ds.GetEV(iev);
    int eventID = ev.GetGTID();
    SanityCheckProc::EventID(eventID);

    int npmtcal = ev.GetCalPMTs().GetCount();

    for(int ihit = 0; ihit<npmtcal; ++ihit){
      // Get MCTruth info
      const DS::PMTCal& pmt = ev.GetCalPMTs().GetPMT(ihit);
      int pmtid = pmt.GetID();
      SanityCheckProc::PMTID(pmtid);
      fHitPMTs.push_back(pmtid);
    }


  }  // ! ev loop


  return OK;
}

// The eventID should be in the range [0, (2^24 - 1)]
// And, certain event IDs should never occur
void SanityCheckProc::EventID(int eventID)
{
  if(eventID<0 || eventID>( (1<<24)-1))SanityCheckProc::Die("",3,eventID,(1<<24)-1);

  // Event IDs that should never occur are those with bits 0-15 all set to 1
  // So, increment eventID by 1 and then test for divisibility by 2^16
  eventID += 1;
  int ibtest = (1<<16);
  if(eventID % ibtest == 0)SanityCheckProc::Die("",4,eventID-1,0);
}

// The PMT ID should be within the range of nPMTs
// And, each PMT should only be hit once within a single detector event
void SanityCheckProc::PMTID(int pmtID)
{
  if(pmtID<0 || pmtID>(nPMTs-1))SanityCheckProc::Die("",2,pmtID,nPMTs-1);

  int nToDate = fHitPMTs.size();
  for(int ih = 0; ih<nToDate; ++ih){
    if(pmtID == fHitPMTs[ih])SanityCheckProc::Die("",1,pmtID,0);
  }
}

// Handle errors
void SanityCheckProc::DontDie(std::string message, int flag_err, int info1, int info2)
{
  warn << YELLOW << " Sanity SETBACK: " << CLR << newline;
  if(info1==0||info2==0||message==""){/*remove warnings*/}

  if(flag_err==1){ //
  }

}

// Handle FATAL errors
void SanityCheckProc::Die(std::string message, int flag_err, int info1, int info2, int return_code)
{
  warn << RED << " Sanity PROBLEM:" << CLR << newline;
  if(message==""){/*remove warnings*/}

  if(flag_err==1){  // This PMT was hit more than once in an event
    warn << BBLUE << " PMT ID " << info1
         << " was hit twice in this event" << newline
         << "  This shouldn't occur; " << RED << " DYING NOW " << CLR
         << newline;
  }

  if(flag_err==2){  // PMT ID is outside range
    warn << BBLUE << " PMT ID " << info1
         << " is outside the expected range of 0 - " << info2 << newline
         << "  This shouldn't occur; " << RED << " DYING NOW " << CLR
         << newline;
  }

  if(flag_err==3){  // Event ID is outside range
    warn << BBLUE << " Event ID " << info1
         << " is outside the expected range of 0 - " << info2 << newline
         << "  This shouldn't occur; " << RED << " DYING NOW " << CLR
         << newline;
  }

  if(flag_err==4){  // Event ID is one that should have been skipped
    warn << BBLUE << " Event ID " << info1
         << " should have been skipped due to SYNC CLEAR" << newline
         << "  This shouldn't occur; " << RED << " DYING NOW " << CLR
         << newline;
  }

  exit(return_code);
}


} // namespace RAT