#include "htofclusterfsim.h"
#include "hdebug.h"
#include "hades.h"
#include "hspectrometer.h"
#include "htofdetector.h"
#include "hevent.h"
#include "hcategory.h"
#include "hiterator.h"
#include "hlocation.h"
#include "htofhitsim.h"
#include "htofclustersim.h"
#include <vector>
#include <algorithm>
#include <iostream>
ClassImp(HTofClusterFSim)
HTofClusterFSim::~HTofClusterFSim(void) {
}
Bool_t HTofClusterFSim::init(void) {
initParContainer(gHades->getSetup(),gHades->getRuntimeDb());
fHitCat =gHades->getCurrentEvent()->getCategory(catTofHit);
if (!fHitCat) {
HTofDetector* tof=(HTofDetector*)(gHades->getSetup()->getDetector("Tof"));
fHitCat=tof->buildMatrixCategory("HTofHitSim",0.5F);
if (!fHitCat) return kFALSE;
else gHades->getCurrentEvent()->addCategory(catTofHit,fHitCat,"Tof");
} else {
if (fHitCat->getClass()!=HTofHitSim::Class()) {
Error("HTofClusterFSim::init()","Misconfigured input category");
return kFALSE;
}
}
iterh = (HIterator*)fHitCat->MakeIterator();
fClusterCat=gHades->getCurrentEvent()->getCategory(catTofCluster);
if (!fClusterCat) {
HTofDetector* tof=(HTofDetector*)(gHades->getSetup()->getDetector("Tof"));
fClusterCat=tof->buildMatrixCategory("HTofClusterSim",0.5F);
if (!fClusterCat) return kFALSE;
else gHades->getCurrentEvent()->addCategory(catTofCluster,fClusterCat,"Tof");
} else {
if (fClusterCat->getClass()!=HTofClusterSim::Class()) {
Error("HTofClusterFSim::init()","Misconfigured output category");
return kFALSE;
}
}
iterc = (HIterator*)fClusterCat->MakeIterator();
fActive = kTRUE;
return kTRUE;
}
Int_t HTofClusterFSim::execute() {
HTofClusterF::execute();
HTofHitSim *hit=NULL;
HTofClusterSim *cluster=NULL;
vector <Int_t> ntrack;
iterh->Reset();
iterc->Reset();
while((cluster = (HTofClusterSim*)iterc->Next()) != NULL) {
fCLoc[0]=cluster->getSector();
fCLoc[1]=cluster->getModule();
fCLoc[2]=cluster->getCell();
cluster->clear();
cluster->setNParticipants(1);
ntrack.clear();
for(Int_t i = 0; i < cluster->getClusterSize(); i ++){
hit = (HTofHitSim*)iterh->Next();
if(find(ntrack.begin(),ntrack.end(),hit->getNTrack1()) == ntrack.end()) ntrack.push_back(hit->getNTrack1());
if(find(ntrack.begin(),ntrack.end(),hit->getNTrack2()) == ntrack.end()) ntrack.push_back(hit->getNTrack2());
}
sort(ntrack.begin(),ntrack.end());
if(ntrack.size() < 4){
for(UInt_t i = 0; i < ntrack.size(); i ++){
cluster->setNTrack1( ntrack[i] );
cluster->setNTrack2( ntrack[i] );
cluster->incNParticipants();
}
} else {
for(UInt_t i = 0; i < ntrack.size() && i < 7; i += 2){
cluster->setNTrack1( ntrack[i] );
cluster->setNTrack2( ntrack[i+1] );
cluster->incNParticipants();
}
if(ntrack.size() == 5 ) {
cluster->setNTrack1( ntrack[4] );
cluster->setNTrack2( ntrack[4] );
cluster->incNParticipants();
}
if(ntrack.size() > 3 ) {
Warning("HTofClusterFSim::execute()","number of track numbers larger than 3! n=%i",(Int_t)ntrack.size());
}
if(ntrack.size() > 6 ) {
Warning("HTofClusterFSim::execute()","number of track numbers larger than maximum stored tracks! n=%i",(Int_t)ntrack.size());
}
}
}
return 0;
}