#include "PndEmcClusterRemoveDuplCrys.h"
#include "PndEmcDigi.h"
#include "PndEmcCluster.h"

#include "FairRootManager.h"
#include "FairRunAna.h"

#include "TClonesArray.h"

using std::cout;
using std::cerr;
using std::endl;

PndEmcClusterRemoveDuplCrys::PndEmcClusterRemoveDuplCrys(Int_t verbose) :
	FairTask("PndEmcClusterRemoveDuplCrys", verbose) {
}


InitStatus PndEmcClusterRemoveDuplCrys::Init() {

	FairRootManager* ioman = FairRootManager::Instance();
	if (!ioman) {
		cout << "-E- PndEmcClusterRemoveDuplCrys::Init: "
			<< "RootManager not instantiated!" << endl;
		return kFATAL;
	}

	// Get inputarray
	fClusterArray = (TClonesArray*) ioman->GetObject("EmcCluster");
	if (!fClusterArray) {
		cout << "-W- PndEmcClusterRemoveDuplCrys::Init: "
			<< "No PndEmcHit array! Needed for MC Truth" << endl;
	}

	// Get input/output array
	fDigiArray = FairRunAna::Instance()->IsTimeStamp() ? (TClonesArray*) ioman->GetObject("EmcDigiClusterBase") : (TClonesArray*) ioman->GetObject("EmcDigi");
	if (!fDigiArray) {
		cout << "-W- PndEmcClusterRemoveDuplCrys::Init: "
			<< "No PndEmcDigi array!" << endl;
		return kERROR;
	}

	cout <<"-I- PndEmcClusterRemoveDuplCrys: "
		<< "Got both arrays" << endl;

	return kSUCCESS;
}

void PndEmcClusterRemoveDuplCrys::Exec(Option_t* opt) {

	for(Int_t iClu=0; iClu<fClusterArray->GetEntriesFast(); ++iClu) {
		PndEmcCluster* theCluster = static_cast<PndEmcCluster*>(fClusterArray->UncheckedAt(iClu));
		
		for(std::vector<Int_t>::reverse_iterator rit = theCluster->DigiList().rbegin(); rit!=theCluster->DigiList().rend(); ) {

			PndEmcDigi* theDigi = static_cast<PndEmcDigi*>(fDigiArray->UncheckedAt(*rit));
			Int_t detId = theDigi->GetDetectorId();

			//check wether other digi with same detector Id was already included in cluster
			//..we do a reverse iteration here, because only the first (i.e. with the smallest idx) digi out of all digis with same detId in fDigiArry will be seen in the MemberDigiMap
			//(cf PndEmcMakeCluster::addDigi...)
			
			Int_t tmp_pos = theCluster->MemberDigiMap().find(detId)->second;
			if(tmp_pos != *rit) { //in Member Map ..other digi is related to this detId...add digis up
				PndEmcDigi* oldDigi = static_cast<PndEmcDigi*>(fDigiArray->UncheckedAt(tmp_pos));
				oldDigi->AbsorbEnergy(*theDigi);

				//std::cerr << "Absorbing energy (ptr|detId): (" << oldDigi << " | "<< oldDigi->GetDetectorId() << ")" << std::flush;
				theCluster->removeDigi(fDigiArray, (++rit).base());	//remove theDigi from cluster
			} else {
				++rit;
			}
		}
	}
}