1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
17 //_________________________________________________________________________
18 // Class containing methods for the isolation cut.
19 // An AOD candidate (AliAODPWG4ParticleCorrelation type)
20 // is passed. Look in a cone around the candidate and study
21 // the hadronic activity inside to decide if the candidate is isolated
24 //*-- Author: Gustavo Conesa (LNF-INFN)
25 //////////////////////////////////////////////////////////////////////////////
28 // --- ROOT system ---
29 //#include <Riostream.h>
30 #include <TLorentzVector.h>
31 #include <TObjArray.h>
33 // --- AliRoot system ---
34 #include "AliIsolationCut.h"
35 #include "AliAODPWG4ParticleCorrelation.h"
36 #include "AliAODTrack.h"
37 #include "AliAODCaloCluster.h"
38 #include "AliCaloTrackReader.h"
40 ClassImp(AliIsolationCut)
42 //____________________________________________________________________________
43 AliIsolationCut::AliIsolationCut() :
45 fConeSize(0.),fPtThreshold(0.), fPtFraction(0.), fICMethod(0),fPartInCone(0)
50 //Initialize parameters
55 //____________________________________________________________________________
56 AliIsolationCut::AliIsolationCut(const AliIsolationCut & g) :
58 fConeSize(g.fConeSize),
59 fPtThreshold(g.fPtThreshold),
60 fPtFraction(g.fPtFraction),
61 fICMethod(g.fICMethod)
67 //_________________________________________________________________________
68 AliIsolationCut & AliIsolationCut::operator = (const AliIsolationCut & source)
70 // assignment operator
72 if(&source == this) return *this;
74 fConeSize = source.fConeSize ;
75 fPtThreshold = source.fPtThreshold ;
76 fICMethod = source.fICMethod ;
77 fPtFraction = source.fPtFraction ;
83 //____________________________________________________________________________
84 TString AliIsolationCut::GetICParametersList()
86 //Put data member values in string to keep in output container
88 TString parList ; //this will be list of parameters used for this analysis.
91 sprintf(onePar,"--- AliIsolationCut ---\n") ;
93 sprintf(onePar,"fConeSize: (isolation cone size) %1.2f\n",fConeSize) ;
95 sprintf(onePar,"fPtThreshold =%1.2f (isolation pt threshold) \n",fPtThreshold) ;
97 sprintf(onePar,"fPtFraction=%1.2f (isolation pt threshold fraction ) \n",fPtFraction) ;
99 sprintf(onePar,"fICMethod=%d (isolation cut case) \n",fICMethod) ;
101 sprintf(onePar,"fPartInCone=%d \n",fPartInCone) ;
107 //____________________________________________________________________________
108 void AliIsolationCut::InitParameters()
110 //Initialize the parameters of the analysis.
115 fPartInCone = kNeutralAndCharged;
116 fICMethod = kPtThresIC; // 0 pt threshol method, 1 cone pt sum method
120 //__________________________________________________________________
121 void AliIsolationCut::MakeIsolationCut(TObjArray * const plCTS, TObjArray * const plNe, AliCaloTrackReader * const reader,
122 const Bool_t fillAOD, AliAODPWG4ParticleCorrelation *pCandidate,
123 const TString aodArrayRefName,
124 Int_t & n, Int_t & nfrac, Float_t &coneptsum, Bool_t &isolated) const
126 //Search in cone around a candidate particle if it is isolated
127 Float_t phiC = pCandidate->Phi() ;
128 Float_t etaC = pCandidate->Eta() ;
129 Float_t ptC = pCandidate->Pt() ;
131 Float_t eta = -100. ;
132 Float_t phi = -100. ;
138 //Initialize the array with refrences
139 TObjArray * refclusters = 0x0;
140 TObjArray * reftracks = 0x0;
141 Int_t ntrackrefs = 0;
142 Int_t nclusterrefs = 0;
144 //Check charged particles in cone.
145 if(plCTS && (fPartInCone==kOnlyCharged || fPartInCone==kNeutralAndCharged)){
147 for(Int_t ipr = 0;ipr < plCTS->GetEntries() ; ipr ++ ){
148 AliAODTrack* track = (AliAODTrack *)(plCTS->At(ipr)) ;
149 //Do not count the candidate (pion, conversion photon) or the daughters of the candidate
150 if(track->GetID() == pCandidate->GetTrackLabel(0) || track->GetID() == pCandidate->GetTrackLabel(1)) continue ;
151 p3.SetXYZ(track->Px(),track->Py(),track->Pz());
155 if(phi<0) phi+=TMath::TwoPi();
157 //Check if there is any particle inside cone with pt larger than fPtThreshold
158 rad = TMath::Sqrt((eta-etaC)*(eta-etaC)+ (phi-phiC)*(phi-phiC));
164 reftracks = new TObjArray(0);
165 reftracks->SetName(aodArrayRefName+"Tracks");
166 reftracks->SetOwner(kFALSE);
168 reftracks->Add(track);
170 //printf("charged in isolation cone pt %f, phi %f, eta %f, R %f \n",pt,phi,eta,rad);
172 if(pt > fPtThreshold ) n++;
173 if(pt > fPtFraction*ptC ) nfrac++;
175 }// charged particle loop
178 //Check neutral particles in cone.
179 if(plNe && (fPartInCone==kOnlyNeutral || fPartInCone==kNeutralAndCharged)){
181 //Get vertex for photon momentum calculation
182 Double_t vertex[] = {0,0,0} ; //vertex ;
183 Double_t vertex2[] = {0,0,0} ; //vertex second AOD input ;
184 if(!reader->GetDataType()== AliCaloTrackReader::kMC)
186 reader->GetVertex(vertex);
187 if(reader->GetSecondInputAODTree()) reader->GetSecondInputAODVertex(vertex2);
190 for(Int_t ipr = 0;ipr < plNe->GetEntries() ; ipr ++ ){
191 AliAODCaloCluster * calo = (AliAODCaloCluster *)(plNe->At(ipr)) ;
193 //Do not count the candidate (photon or pi0) or the daughters of the candidate
194 if(calo->GetID() == pCandidate->GetCaloLabel(0) || calo->GetID() == pCandidate->GetCaloLabel(1)) continue ; //Skip matched clusters with tracks
196 if(calo->GetNTracksMatched() > 0) continue ;
197 //Input from second AOD?
199 if (pCandidate->GetDetector() == "EMCAL" && reader->GetAODEMCALNormalInputEntries() <= ipr) input = 1 ;
200 else if(pCandidate->GetDetector() == "PHOS" && reader->GetAODPHOSNormalInputEntries() <= ipr) input = 1;
202 //Get Momentum vector,
203 if (input == 0) calo->GetMomentum(mom,vertex) ;//Assume that come from vertex in straight line
204 else if(input == 1) calo->GetMomentum(mom,vertex2);//Assume that come from vertex in straight line
209 if(phi<0) phi+=TMath::TwoPi();
211 //Check if there is any particle inside cone with pt larger than fPtThreshold
212 rad = TMath::Sqrt((eta-etaC)*(eta-etaC)+ (phi-phiC)*(phi-phiC));
217 refclusters = new TObjArray(0);
218 refclusters->SetName(aodArrayRefName+"Clusters");
219 refclusters->SetOwner(kFALSE);
221 refclusters->Add(calo);
223 //printf("neutral in isolation cone pt %f, phi %f, eta %f, R %f \n",pt,phi,eta,rad);
225 if(pt > fPtThreshold ) n++;
226 if(pt > fPtFraction*ptC ) nfrac++;
228 }// neutral particle loop
231 //printf("Isolation Cut: in cone with: pT>pTthres %d, pT > pTfrac*pTcandidate %d \n",n,nfrac);
233 //Add reference arrays to AOD when filling AODs only
235 if(refclusters) pCandidate->AddObjArray(refclusters);
236 if(reftracks) pCandidate->AddObjArray(reftracks);
239 //Check isolation, depending on method.
240 if( fICMethod == kPtThresIC){
241 if(n==0) isolated = kTRUE ;
243 else if( fICMethod == kSumPtIC){
244 if(coneptsum < fPtThreshold)
247 else if( fICMethod == kPtFracIC){
248 if(nfrac==0) isolated = kTRUE ;
250 else if( fICMethod == kSumPtFracIC){
251 if(coneptsum < fPtFraction*ptC)
255 //if(refclusters) delete refclusters;
256 //if(reftracks) delete reftracks;
260 //__________________________________________________________________
261 void AliIsolationCut::Print(const Option_t * opt) const
264 //Print some relevant parameters set for the analysis
268 printf("**** Print %s %s **** \n", GetName(), GetTitle() ) ;
270 printf("IC method = %d\n", fICMethod) ;
271 printf("Cone Size = %1.2f\n", fConeSize) ;
272 printf("pT threshold = %2.1f\n", fPtThreshold) ;
273 printf("pT fraction = %3.1f\n", fPtFraction) ;
274 printf("particle type in cone = %d\n",fPartInCone);