#include "IFGDHitCountingTask.hxx"

#include "IDigitContainer.hxx"
#include "ICOMETLog.hxx"
#include "IOADatabase.hxx"
#include "IDigitManager.hxx"

/// Initialization of task (at start of first file)
int IFGDHitCountingTask::Initialize(){

 std::cout << "Initialized FGD test" << std::endl;
  fNumberPulses = 0;
  fNumberHits = 0;
  fNumberEvents = 0;

  return 0;
}
  
/// Count up the number of pulses (before calibration) 
/// and the number of hits (after calibration)
int IFGDHitCountingTask::Process(COMET::ICOMETEvent& event){


  if (COMET::IOADatabase::Get().Digits().FactoryAvailable("fgd")){
    COMET::IHandle<COMET::IDigitContainer> fgdDigits 
      = COMET::IOADatabase::Get().Digits().CacheDigits("fgd");  
    fNumberPulses += fgdDigits->size();    

    //std::cout << fgdDigits->size() << std::endl;
  }   
  
  fNumberEvents++;

  return 0;

}
  
/// Finalization of task (at end of last file).
int IFGDHitCountingTask::Finalize(){

  double avgNumberPulses = 0.;
  if(fNumberEvents != 0)
     avgNumberPulses = fNumberPulses/static_cast<float>(fNumberEvents);

  COMETLog("Avg Number of FGD pulses " << avgNumberPulses);
  fNumberHits = 0;

  return 0;

}