#include <bitset>
#include <vector>

#include "IGeometryDatabase.hxx"
#include "IECALReconEventFunction.hxx"
#include "IHitSelection.hxx"
#include "IDataHit.hxx"
#include "IMCHit.hxx"

//======================================================================
// Constructor
//======================================================================
IECALReconEventFunction::IECALReconEventFunction() :
  IReconEventFunction("ECAL")
{
  // Prepare algorithm
  fASumupEnergies = new COMET::IECALSumupEnergies();
}

//======================================================================
// Destructor
//======================================================================
IECALReconEventFunction::~IECALReconEventFunction()
{
  delete fASumupEnergies;
}

//======================================================================
// Usage
//======================================================================
void IECALReconEventFunction::Usage()
{
  std::cout << "Waveform analysis -> Energy clustering." << std::endl
            << std::endl
            << "Options:\n"
            << "    No options yet.\n"
            << std::endl;
}

//======================================================================
// SetOption
//======================================================================
bool IECALReconEventFunction::SetOption(TString option, TString value)
{
  // Now no specific option here
  return false;
}

//======================================================================
// BeginOfEvent
//======================================================================
void IECALReconEventFunction::BeginOfEvent()
{
  
}

//======================================================================
// EndOfEvent
//======================================================================
void IECALReconEventFunction::EndOfEvent()
{
  
}

//======================================================================
// PrepareHits
//======================================================================
COMET::IHandle<COMET::IAlgorithmResult> 
IECALReconEventFunction::PrepareHits(COMET::ICOMETEvent& event)
{
  COMET::IHandle<COMET::IHitSelection> hits = event.GetHitSelection("ECAL");
  if(hits){
    return MakeBasicAlgorithmResult(hits);
  }
  
  // Return NULL
  return COMET::IHandle<COMET::IAlgorithmResult>();
}

//======================================================================
// ProcesAlgorithms
//======================================================================
COMET::IHandle<COMET::IAlgorithmResult>
IECALReconEventFunction::ProcessAlgorithms(const COMET::IAlgorithmResult& input,
                                           COMET::ICOMETEvent& event)
{
  COMET::IHandle<COMET::IAlgorithmResult> 
    sumenergy = fASumupEnergies->Process(input);
  event.AddFit(sumenergy);
  return sumenergy;
}