// --- ROOT system ---
-//#include <Riostream.h>
#include <TLorentzVector.h>
#include <TObjArray.h>
#include "AliVCluster.h"
#include "AliCaloTrackReader.h"
#include "AliMixedEvent.h"
+#include "AliCaloPID.h"
ClassImp(AliIsolationCut)
-//____________________________________________________________________________
- AliIsolationCut::AliIsolationCut() :
- TObject(),
- fConeSize(0.),fPtThreshold(0.), fSumPtThreshold(0.), fPtFraction(0.), fICMethod(0),fPartInCone(0)
-
+//____________________________________
+AliIsolationCut::AliIsolationCut() :
+TObject(),
+fConeSize(0.),
+fPtThreshold(0.),
+fSumPtThreshold(0.),
+fPtFraction(0.),
+fICMethod(0),
+fPartInCone(0)
+
{
//default ctor
//Initialize parameters
InitParameters();
-
-}
-/*
-//____________________________________________________________________________
-AliIsolationCut::AliIsolationCut(const AliIsolationCut & g) :
- TObject(g),
- fConeSize(g.fConeSize),
- fPtThreshold(g.fPtThreshold),
- fPtFraction(g.fPtFraction),
- fICMethod(g.fICMethod)
-{
- // cpy ctor
}
-//_________________________________________________________________________
-AliIsolationCut & AliIsolationCut::operator = (const AliIsolationCut & source)
-{
- // assignment operator
-
- if(&source == this) return *this;
-
- fConeSize = source.fConeSize ;
- fPtThreshold = source.fPtThreshold ;
- fICMethod = source.fICMethod ;
- fPtFraction = source.fPtFraction ;
-
- return *this;
-
-}
-*/
-//____________________________________________________________________________
+//____________________________________________
TString AliIsolationCut::GetICParametersList()
{
//Put data member values in string to keep in output container
parList+=onePar ;
snprintf(onePar,buffersize,"fPartInCone=%d \n",fPartInCone) ;
parList+=onePar ;
-
+
return parList;
}
-//____________________________________________________________________________
+//____________________________________
void AliIsolationCut::InitParameters()
{
//Initialize the parameters of the analysis.
- fConeSize = 0.4 ;
- fPtThreshold = 1. ;
+ fConeSize = 0.4 ;
+ fPtThreshold = 1. ;
fSumPtThreshold = 0.5 ;
- fPtFraction = 0.1 ;
- fPartInCone = kNeutralAndCharged;
- fICMethod = kPtThresIC; // 0 pt threshol method, 1 cone pt sum method
+ fPtFraction = 0.1 ;
+ fPartInCone = kOnlyCharged;
+ fICMethod = kSumPtFracIC; // 0 pt threshol method, 1 cone pt sum method
}
-//__________________________________________________________________
-void AliIsolationCut::MakeIsolationCut(TObjArray * const plCTS, TObjArray * const plNe, AliCaloTrackReader * const reader,
- const Bool_t fillAOD, AliAODPWG4ParticleCorrelation *pCandidate,
- const TString aodArrayRefName,
- Int_t & n, Int_t & nfrac, Float_t &coneptsum, Bool_t &isolated) const
+//________________________________________________________________________________
+void AliIsolationCut::MakeIsolationCut(const TObjArray * plCTS,
+ const TObjArray * plNe,
+ const AliCaloTrackReader * reader,
+ const AliCaloPID * pid,
+ const Bool_t bFillAOD,
+ AliAODPWG4ParticleCorrelation *pCandidate,
+ const TString & aodArrayRefName,
+ Int_t & n,
+ Int_t & nfrac,
+ Float_t &coneptsum,
+ Bool_t &isolated) const
{
//Search in cone around a candidate particle if it is isolated
Float_t phiC = pCandidate->Phi() ;
+ if(phiC<0) phiC+=TMath::TwoPi();
Float_t etaC = pCandidate->Eta() ;
Float_t ptC = pCandidate->Pt() ;
Float_t pt = -100. ;
- Float_t eta = -100. ;
- Float_t phi = -100. ;
- Float_t rad = -100 ;
- n = 0 ;
+ Float_t eta = -100. ;
+ Float_t phi = -100. ;
+ Float_t rad = -100. ;
+
+ n = 0 ;
+ nfrac = 0 ;
coneptsum = 0.;
- isolated = kFALSE;
-
+ isolated = kFALSE;
+
//Initialize the array with refrences
TObjArray * refclusters = 0x0;
TObjArray * reftracks = 0x0;
Int_t ntrackrefs = 0;
Int_t nclusterrefs = 0;
+
//Check charged particles in cone.
if(plCTS && (fPartInCone==kOnlyCharged || fPartInCone==kNeutralAndCharged)){
TVector3 p3;
for(Int_t ipr = 0;ipr < plCTS->GetEntries() ; ipr ++ ){
AliAODTrack* track = (AliAODTrack *)(plCTS->At(ipr)) ;
//Do not count the candidate (pion, conversion photon) or the daughters of the candidate
- if(track->GetID() == pCandidate->GetTrackLabel(0) || track->GetID() == pCandidate->GetTrackLabel(1)) continue ;
+ if(track->GetID() == pCandidate->GetTrackLabel(0) || track->GetID() == pCandidate->GetTrackLabel(1)
+ || track->GetID() == pCandidate->GetTrackLabel(2) || track->GetID() == pCandidate->GetTrackLabel(3)
+ ) continue ;
p3.SetXYZ(track->Px(),track->Py(),track->Pz());
pt = p3.Pt();
eta = p3.Eta();
if(TMath::Abs(phi-phiC)>TMath::PiOver2()) continue ;
//if at the same side has particle larger than candidate, then candidate can not be the leading, skip such events
if(pt > ptC){
- n = -1;
- nfrac = -1;
+ n = -1;
+ nfrac = -1;
coneptsum = -1;
- isolated = kFALSE;
- if(fillAOD && reftracks) reftracks->Clear();
+ isolated = kFALSE;
+ if(bFillAOD && reftracks) {
+ reftracks->Clear();
+ delete reftracks;
+ }
return ;
}
+
//Check if there is any particle inside cone with pt larger than fPtThreshold
- rad = TMath::Sqrt((eta-etaC)*(eta-etaC)+ (phi-phiC)*(phi-phiC));
+
+ rad = Radius(etaC, phiC, eta, phi);
if(rad < fConeSize){
- if(fillAOD) {
+ if(bFillAOD) {
ntrackrefs++;
if(ntrackrefs == 1){
reftracks = new TObjArray(0);
- reftracks->SetName(aodArrayRefName+"Tracks");
+ //reftracks->SetName(Form("Tracks%s",aodArrayRefName.Data()));
+ TString tempo(aodArrayRefName) ;
+ tempo += "Tracks" ;
+ reftracks->SetName(tempo);
reftracks->SetOwner(kFALSE);
}
reftracks->Add(track);
}
//printf("charged in isolation cone pt %f, phi %f, eta %f, R %f \n",pt,phi,eta,rad);
coneptsum+=pt;
- if(pt > fPtThreshold ) n++;
+ if(pt > fPtThreshold ) n++;
if(pt > fPtFraction*ptC ) nfrac++;
- }
+ } // Inside cone
}// charged particle loop
}//Tracks
//Check neutral particles in cone.
if(plNe && (fPartInCone==kOnlyNeutral || fPartInCone==kNeutralAndCharged)){
- //Get vertex for photon momentum calculation
- //Double_t vertex2[] = {0,0,0} ; //vertex second AOD input ;
- //if(reader->GetDataType()!= AliCaloTrackReader::kMC)
- //{
- //if(reader->GetSecondInputAODTree()) reader->GetSecondInputAODVertex(vertex2);
- //}
+
TLorentzVector mom ;
for(Int_t ipr = 0;ipr < plNe->GetEntries() ; ipr ++ ){
AliVCluster * calo = (AliVCluster *)(plNe->At(ipr)) ;
}
//Do not count the candidate (photon or pi0) or the daughters of the candidate
- if(calo->GetID() == pCandidate->GetCaloLabel(0) || calo->GetID() == pCandidate->GetCaloLabel(1)) continue ; //Skip matched clusters with tracks
-
- if(calo->GetNTracksMatched() > 0) continue ;
-
- //Input from second AOD?
- //Int_t input = 0;
- // if (pCandidate->GetDetector() == "EMCAL" && reader->GetAODEMCALNormalInputEntries() <= ipr) input = 1 ;
- // else if(pCandidate->GetDetector() == "PHOS" && reader->GetAODPHOSNormalInputEntries() <= ipr) input = 1;
+ if(calo->GetID() == pCandidate->GetCaloLabel(0) ||
+ calo->GetID() == pCandidate->GetCaloLabel(1)) continue ;
- //Get Momentum vector,
- //if (input == 0)
- calo->GetMomentum(mom,reader->GetVertex(evtIndex)) ;//Assume that come from vertex in straight line
- //else if(input == 1) calo->GetMomentum(mom,vertex2);//Assume that come from vertex in straight line
+ //Skip matched clusters with tracks
+ if( pid->IsTrackMatched(calo,reader->GetCaloUtils(),reader->GetInputEvent()) ) continue ;
+
+ //Assume that come from vertex in straight line
+ calo->GetMomentum(mom,reader->GetVertex(evtIndex)) ;
pt = mom.Pt();
eta = mom.Eta();
if(phi<0) phi+=TMath::TwoPi();
//only loop the particle at the same side of candidate
- if(TMath::Abs(phi-phiC)>TMath::PiOver2()) continue ;
//if at the same side has particle larger than candidate, then candidate can not be the leading, skip such events
+ if(TMath::Abs(phi-phiC)>TMath::PiOver2()) continue ;
+
if(pt > ptC){
- n = -1;
- nfrac = -1;
+ n = -1;
+ nfrac = -1;
coneptsum = -1;
- isolated = kFALSE;
- if(fillAOD){
- if(reftracks) reftracks->Clear();
- if(refclusters)refclusters->Clear();
+ isolated = kFALSE;
+ if(bFillAOD){
+ if(reftracks){
+ reftracks ->Clear();
+ delete reftracks;
+ }
+ if(refclusters){
+ refclusters->Clear();
+ delete refclusters;
+ }
}
return ;
}
//Check if there is any particle inside cone with pt larger than fPtThreshold
- rad = TMath::Sqrt((eta-etaC)*(eta-etaC)+ (phi-phiC)*(phi-phiC));
+
+ rad = Radius(etaC, phiC, eta, phi);
+
if(rad < fConeSize){
- if(fillAOD) {
+ if(bFillAOD) {
nclusterrefs++;
if(nclusterrefs==1){
refclusters = new TObjArray(0);
- refclusters->SetName(aodArrayRefName+"Clusters");
+ //refclusters->SetName(Form("Clusters%s",aodArrayRefName.Data()));
+ TString tempo(aodArrayRefName) ;
+ tempo += "Clusters" ;
+ refclusters->SetName(tempo);
refclusters->SetOwner(kFALSE);
}
refclusters->Add(calo);
}
//printf("neutral in isolation cone pt %f, phi %f, eta %f, R %f \n",pt,phi,eta,rad);
coneptsum+=pt;
- if(pt > fPtThreshold ) n++;
+ if(pt > fPtThreshold ) n++;
//if fPtFraction*ptC<fPtThreshold then consider the fPtThreshold directly
- if(pt > fPtFraction*ptC && pt>fPtThreshold) nfrac++;
+ if(fPtFraction*ptC<fPtThreshold) {
+ if(pt>fPtThreshold) nfrac++ ;
+ }
+ else {
+ if(pt>fPtFraction*ptC) nfrac++;
+ }
}//in cone
}// neutral particle loop
}//neutrals
-
+
//printf("Isolation Cut: in cone with: pT>pTthres %d, pT > pTfrac*pTcandidate %d \n",n,nfrac);
//Add reference arrays to AOD when filling AODs only
- if(fillAOD) {
+ if(bFillAOD) {
if(refclusters) pCandidate->AddObjArray(refclusters);
if(reftracks) pCandidate->AddObjArray(reftracks);
}
}
else if( fICMethod == kSumPtFracIC){
//when the fPtFraction*ptC < fSumPtThreshold then consider the later case
- if(coneptsum < fPtFraction*ptC && coneptsum < fSumPtThreshold) isolated = kTRUE ;
+ if(fPtFraction*ptC < fSumPtThreshold && coneptsum < fSumPtThreshold) isolated = kTRUE ;
+ if(fPtFraction*ptC > fSumPtThreshold && coneptsum < fPtFraction*ptC) isolated = kTRUE ;
}
- //if(refclusters) delete refclusters;
- //if(reftracks) delete reftracks;
-
}
-//__________________________________________________________________
+//_____________________________________________________
void AliIsolationCut::Print(const Option_t * opt) const
{
printf("**** Print %s %s **** \n", GetName(), GetTitle() ) ;
- printf("IC method = %d\n", fICMethod) ;
- printf("Cone Size = %1.2f\n", fConeSize) ;
+ printf("IC method = %d\n", fICMethod ) ;
+ printf("Cone Size = %1.2f\n", fConeSize ) ;
printf("pT threshold = %2.1f\n", fPtThreshold) ;
- printf("pT fraction = %3.1f\n", fPtFraction) ;
- printf("particle type in cone = %d\n",fPartInCone);
+ printf("pT fraction = %3.1f\n", fPtFraction ) ;
+ printf("particle type in cone = %d\n", fPartInCone ) ;
printf(" \n") ;
}
+
+//___________________________________________________________________________
+Float_t AliIsolationCut::Radius(const Float_t etaC, const Float_t phiC,
+ const Float_t eta , const Float_t phi) const
+{
+ // Calculate the distance to trigger from any particle
+
+ Float_t dEta = etaC-eta;
+ Float_t dPhi = phiC-phi;
+
+ if(TMath::Abs(dPhi) >= TMath::Pi())
+ dPhi = TMath::TwoPi()-TMath::Abs(dPhi);
+
+ return TMath::Sqrt( dEta*dEta + dPhi*dPhi );
+
+}
+
+
+