#include "CbmEcalMaximumLocator.h"

#include "CbmEcalStructure.h"
#include "CbmEcalMaximum.h"
#include "CbmEcalCell.h"

#include "FairRootManager.h"

#include "TClonesArray.h"

#include <list>
#include <iostream>

using namespace std;

CbmEcalMaximumLocator::CbmEcalMaximumLocator()
  : FairTask()
{
  fMaximums=NULL;
  fStr=NULL;
  fECut=0.005;
}

CbmEcalMaximumLocator::CbmEcalMaximumLocator(const char* name, Int_t verbose)
  : FairTask(name, verbose)
{
  fMaximums=NULL;
  fStr=NULL;
  fECut=0.005;
}

CbmEcalMaximumLocator::~CbmEcalMaximumLocator()
{
  if (fMaximums)
    delete fMaximums;
}

InitStatus CbmEcalMaximumLocator::Init()
{
  FairRootManager* fManager=FairRootManager::Instance();
  if (!fManager)
  {
    Fatal("Init", "Can't find a Root Manager.");
    return kFATAL;
  }
  fStr=(CbmEcalStructure*)fManager->GetObject("EcalStructure");
  if (!fStr)
  {
    Fatal("Init()", "Can't find calorimeter structure in the system.");
    return kFATAL;
  }
  fMaximums=new TClonesArray("CbmEcalMaximum", 3000);
  fManager->Register("EcalMaximums", "ECAL", fMaximums, kFALSE);
  fEvent=0;
  return kSUCCESS;
}

void CbmEcalMaximumLocator::Finish()
{
  fMaximums->Clear();
}

void CbmEcalMaximumLocator::Exec(const Option_t* opt)
{
  list<CbmEcalCell*> all;
  list<CbmEcalCell*> cells;
  list<CbmEcalCell*>::const_iterator p;
  list<CbmEcalCell*>::const_iterator r;
  Double_t e;
  Int_t n=0;

  fEvent++;
  fMaximums->Clear();
  fStr->GetCells(all);
  for(p=all.begin();p!=all.end();++p)
  {
    e=(*p)->GetTotalEnergy();
    if (e<fECut)
      continue;
    (*p)->GetNeighborsList(0, cells);
    for(r=cells.begin();r!=cells.end();++r)
      if ((*r)->GetTotalEnergy()>e)
	break;
    if (r!=cells.end())
      continue;
    new ((*fMaximums)[n++]) CbmEcalMaximum(*p);
  }
  if (fVerbose>9)
    Info("Exec", "%d maximums found", n);
}

ClassImp(CbmEcalMaximumLocator)