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 //======================================================================
19 // Fill Unit Array class
20 // Class used by AliJetESDReader to fill a UnitArray from the information extracted
21 // from the particle tracks
22 // Author: magali.estienne@ires.in2p3.fr
23 //======================================================================
26 // --- ROOT system ---
28 #include <TProcessID.h>
30 #include "AliJetUnitArray.h"
31 #include "AliJetAODFillUnitArrayTracks.h"
33 // --- ROOT system ---
38 // --- AliRoot header files ---
41 ClassImp(AliJetAODFillUnitArrayTracks)
43 //_____________________________________________________________________________
44 AliJetAODFillUnitArrayTracks::AliJetAODFillUnitArrayTracks()
45 : AliJetFillUnitArray(),
48 fApplyMIPCorrection(kTRUE),
59 //_____________________________________________________________________________
60 AliJetAODFillUnitArrayTracks::AliJetAODFillUnitArrayTracks(AliAODEvent* aod)
61 : AliJetFillUnitArray(),
64 fApplyMIPCorrection(kTRUE),
75 //_____________________________________________________________________________
76 AliJetAODFillUnitArrayTracks::AliJetAODFillUnitArrayTracks(const AliJetAODFillUnitArrayTracks &det)
77 : AliJetFillUnitArray(det),
78 fNumUnits(det.fNumUnits),
79 fHadCorr(det.fHadCorr),
80 fApplyMIPCorrection(det.fApplyMIPCorrection),
91 //_____________________________________________________________________________
92 AliJetAODFillUnitArrayTracks& AliJetAODFillUnitArrayTracks::operator=(const AliJetAODFillUnitArrayTracks& other)
96 fNumUnits = other.fNumUnits;
97 fHadCorr = other.fHadCorr;
98 fApplyMIPCorrection = other.fApplyMIPCorrection;
100 fGrid0 = other.fGrid0;
101 fGrid1 = other.fGrid1;
102 fGrid2 = other.fGrid2;
103 fGrid3 = other.fGrid3;
104 fGrid4 = other.fGrid4;
109 //____________________________________________________________________________
110 void AliJetAODFillUnitArrayTracks::InitParameters()
112 fHadCorr = 0; // For hadron correction
113 fNumUnits = fGeom->GetNCells(); // Number of towers in EMCAL
115 fTPCGrid->GetAccParam(fNphi,fNeta,fPhiMin,
116 fPhiMax,fEtaMin,fEtaMax);
117 fTPCGrid->GetBinParam(fPhiBinInTPCAcc,fEtaBinInTPCAcc,
118 fPhiBinInEMCalAcc,fEtaBinInEMCalAcc,fNbinPhi);
120 fIndex = fTPCGrid->GetIndexObject();
123 for(Int_t i=0; i<fNphi+1; i++)
124 for(Int_t j=0; j<fNeta+1; j++) {cout << "fIndex[" << i << "," << j << "] : " <<
125 (*fIndex)(i,j) << endl; }
127 if(fDebug>1) printf("\n Parameters initiated ! \n");
130 //_____________________________________________________________________________
131 AliJetAODFillUnitArrayTracks::~AliJetAODFillUnitArrayTracks()
136 //_____________________________________________________________________________
137 void AliJetAODFillUnitArrayTracks::Exec(Option_t* const /*option*/)
138 //void AliJetAODFillUnitArrayTracks::Exec(Option_t* option)
142 // Fill the unit array with the charged particle information in AOD
145 fDebug = fReaderHeader->GetDebug();
150 // get number of tracks in event (for the loop)
153 //(not used ?) Int_t nt2 = 0;
159 nt = fAOD->GetNumberOfTracks();
160 if(fDebug>1) cout << "Number of Tracks in ESD : " << nt << endl;
162 // temporary storage of signal and pt cut flag
163 Int_t* sflag = new Int_t[nt];
164 Int_t* cflag = new Int_t[nt];
166 // get cuts set by user
167 Float_t ptMin = fReaderHeader->GetPtCut();
168 Float_t etaMin = fReaderHeader->GetFiducialEtaMin();
169 Float_t etaMax = fReaderHeader->GetFiducialEtaMax();
170 fOpt = fReaderHeader->GetDetector();
171 fDZ = fReaderHeader->GetDZ();
172 UInt_t filterMask = ((AliJetAODReaderHeader*)fReaderHeader)->GetTestFilterMask();
174 Int_t nTracksEmcal = 0;
175 Int_t nTracksEmcalDZ = 0;
176 Int_t nTracksTpc = 0;
177 Int_t nTracksTpcOnly = 0;
178 Int_t nTracksEmcalCut = 0;
179 Int_t nTracksEmcalDZCut = 0;
180 Int_t nTracksTpcCut = 0;
181 Int_t nTracksTpcOnlyCut = 0;
183 //(not used ?) Int_t nElements = fTPCGrid->GetNEntries();
184 //(not used ?) Int_t nElements2 = fEMCalGrid->GetNEntries();
185 fGrid = fTPCGrid->GetGridType();
187 //(not used ?) Int_t nEntries = fUnitArray->GetEntries();
189 //(not used ?) Int_t nRefEnt = fRefArray->GetEntries();
193 for (Int_t it = 0; it < nmax; it++) {
195 track = fAOD->GetTrack(it);
196 UInt_t status = track->GetStatus();
199 track->GetPxPyPz(mom);
201 p3.SetXYZ(mom[0],mom[1],mom[2]);
204 //(not used ?) Float_t mass = 0.;
207 phi = ( (p3.Phi()) < 0) ? (p3.Phi()) + 2. * TMath::Pi() : (p3.Phi());
208 if (status == 0) continue;
209 if((filterMask>0)&&!(track->TestFilterBit(filterMask)))continue;
211 if ( (eta > etaMax) || (eta < etaMin)) continue; // checking eta cut
214 if (TMath::Abs(track->GetLabel()) < 10000) sflag[goodTrack]=1;
216 if (pt > ptMin) cflag[goodTrack]=1; // pt cut
220 // Only TPC filled from its grid in its total acceptance
222 Int_t idTPC = fTPCGrid->GetIndex(phi,eta);
226 AliJetUnitArray *uArray = (AliJetUnitArray*)fUnitArray->At(idTPC-1);
227 TRefArray *reference = uArray->GetUnitTrackRef();
228 if (reference->GetEntries() == 0) {
229 new(reference) TRefArray(TProcessID::GetProcessWithUID(track));
232 reference->Add(track);
234 Float_t unitEnergy = 0.;
235 unitEnergy = uArray->GetUnitEnergy();
236 // nTracksTpcOnly is necessary to count the number of candidate cells
237 // but it doesn't give the real multiplicity -> it will be extracted
238 // in the jet finder through the reference to tracks
245 // Fill energy in TPC acceptance
246 uArray->SetUnitEnergy(unitEnergy + pt);
249 if(uArray->GetUnitEnergy()<ptMin){
250 uArray->SetUnitCutFlag(kPtSmaller);
253 uArray->SetUnitCutFlag(kPtHigher);
254 if(ok) nTracksTpcOnlyCut++;
258 if(sflag[goodTrack] == 1) {
259 uArray->SetUnitSignalFlag(kGood);
260 uArray->SetUnitSignalFlagC(kFALSE,kGood);
261 } else uArray->SetUnitSignalFlagC(kFALSE,kBad);
263 if(uArray->GetUnitEnergy()>0 && ref){
266 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray));
268 fRefArray->Add(uArray);
275 Int_t nElements = fTPCGrid->GetNEntries();
276 // Fill track information in EMCAL acceptance
277 if((eta >= fEtaMin && eta <= fEtaMax) &&
278 (phi >= fPhiMin && phi <= fPhiMax))// &&
280 // Include dead-zones
283 Double_t phimin0 = 0., phimin1 = 0., phimin2 = 0., phimin3 = 0., phimin4 = 0.;
284 Double_t phimax0 = 0., phimax1 = 0., phimax2 = 0., phimax3 = 0., phimax4 = 0.;
285 fGeom->GetPhiBoundariesOfSMGap(0,phimin0,phimax0);
286 fGeom->GetPhiBoundariesOfSMGap(1,phimin1,phimax1);
287 fGeom->GetPhiBoundariesOfSMGap(2,phimin2,phimax2);
288 fGeom->GetPhiBoundariesOfSMGap(3,phimin3,phimax3);
289 fGeom->GetPhiBoundariesOfSMGap(4,phimin4,phimax4);
290 Int_t n0 = fGrid0->GetNEntries();
291 Int_t n1 = fGrid1->GetNEntries();
292 Int_t n2 = fGrid2->GetNEntries();
293 Int_t n3 = fGrid3->GetNEntries();
294 //(not used ?) Int_t n4 = fGrid4->GetNEntries();
296 // cout << "phimin0: " << phimin0 << ", phimax0: " << phimax0 << endl;
297 if(phi >= phimin0 && phi <= phimax0){
298 Int_t id0 = fGrid0->GetIndex(phi,eta)-1;
299 AliJetUnitArray *uArray0 = (AliJetUnitArray*)fUnitArray->At(id0+fNumUnits+nElements);
300 TRefArray *reference0 = uArray0->GetUnitTrackRef();
301 if (reference0->GetEntries() == 0) {
302 new(reference0) TRefArray(TProcessID::GetProcessWithUID(track));
305 reference0->Add(track);
306 uArray0->SetUnitTrackID(it);
308 Float_t uEnergy0 = uArray0->GetUnitEnergy();
310 Bool_t ref0 = kFALSE;
316 uArray0->SetUnitEnergy(uEnergy0+pt);
318 if(uArray0->GetUnitEnergy()<ptMin)
319 uArray0->SetUnitCutFlag(kPtSmaller);
321 uArray0->SetUnitCutFlag(kPtHigher);
322 if(ok0) nTracksEmcalDZCut++;
324 if(sflag[goodTrack] == 1) {
325 uArray0->SetUnitSignalFlag(kGood);
326 uArray0->SetUnitSignalFlagC(kFALSE,kGood);
327 } else uArray0->SetUnitSignalFlagC(kFALSE,kBad);
329 if(uArray0->GetUnitEnergy()>0 && ref0){
331 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray0));
334 fRefArray->Add(uArray0);
338 if(phi >= phimin1 && phi <= phimax1){
339 Int_t id1 = fGrid1->GetIndex(phi,eta)-1+n0;
340 AliJetUnitArray *uArray1 = (AliJetUnitArray*)fUnitArray->At(id1+fNumUnits+nElements);
341 TRefArray *reference1 = uArray1->GetUnitTrackRef();
342 if (reference1->GetEntries() == 0) {
343 new(reference1) TRefArray(TProcessID::GetProcessWithUID(track));
346 reference1->Add(track);
348 Float_t uEnergy1 = uArray1->GetUnitEnergy();
350 Bool_t ref1 = kFALSE;
356 uArray1->SetUnitEnergy(uEnergy1+pt);
358 if(uArray1->GetUnitEnergy()<ptMin)
359 uArray1->SetUnitCutFlag(kPtSmaller);
361 uArray1->SetUnitCutFlag(kPtHigher);
362 if(ok1) nTracksEmcalDZCut++;
364 if(sflag[goodTrack] == 1) {
365 uArray1->SetUnitSignalFlag(kGood);
366 uArray1->SetUnitSignalFlagC(kFALSE,kGood);
367 } else uArray1->SetUnitSignalFlagC(kFALSE,kBad);
369 if(uArray1->GetUnitEnergy()>0 && ref1){
371 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray1));
374 fRefArray->Add(uArray1);
378 if(phi >= phimin2 && phi <= phimax2){
379 Int_t id2 = fGrid2->GetIndex(phi,eta)-1+n0+n1;
380 AliJetUnitArray *uArray2 = (AliJetUnitArray*)fUnitArray->At(id2+fNumUnits+nElements);
381 TRefArray *reference2 = uArray2->GetUnitTrackRef();
382 if (reference2->GetEntries() == 0) {
383 new(reference2) TRefArray(TProcessID::GetProcessWithUID(track));
386 reference2->Add(track);
388 Float_t uEnergy2 = uArray2->GetUnitEnergy();
390 Bool_t ref2 = kFALSE;
396 uArray2->SetUnitEnergy(uEnergy2+pt);
398 if(uArray2->GetUnitEnergy()<ptMin)
399 uArray2->SetUnitCutFlag(kPtSmaller);
401 uArray2->SetUnitCutFlag(kPtHigher);
402 if(ok2) nTracksEmcalDZCut++;
404 if(sflag[goodTrack] == 1) {
405 uArray2->SetUnitSignalFlag(kGood);
406 uArray2->SetUnitSignalFlagC(kFALSE,kGood);
407 } else uArray2->SetUnitSignalFlagC(kFALSE,kBad);
409 if(uArray2->GetUnitEnergy()>0 && ref2){
411 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray2));
414 fRefArray->Add(uArray2);
418 if(phi >= phimin3 && phi <= phimax3){
419 Int_t id3 = fGrid3->GetIndex(phi,eta)-1+n0+n1+n2;
420 AliJetUnitArray *uArray3 = (AliJetUnitArray*)fUnitArray->At(id3+fNumUnits+nElements);
421 TRefArray *reference3 = uArray3->GetUnitTrackRef();
422 if (reference3->GetEntries() == 0) {
423 new(reference3) TRefArray(TProcessID::GetProcessWithUID(track));
426 reference3->Add(track);
428 Float_t uEnergy3 = uArray3->GetUnitEnergy();
430 Bool_t ref3 = kFALSE;
436 uArray3->SetUnitEnergy(uEnergy3+pt);
438 if(uArray3->GetUnitEnergy()<ptMin)
439 uArray3->SetUnitCutFlag(kPtSmaller);
441 uArray3->SetUnitCutFlag(kPtHigher);
442 if(ok3) nTracksEmcalDZCut++;
444 if(sflag[goodTrack] == 1) {
445 uArray3->SetUnitSignalFlag(kGood);
446 uArray3->SetUnitSignalFlagC(kFALSE,kGood);
447 } else uArray3->SetUnitSignalFlagC(kFALSE,kBad);
449 if(uArray3->GetUnitEnergy()>0 && ref3){
451 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray3));
454 fRefArray->Add(uArray3);
458 if(phi >= phimin4 && phi <= phimax4){
459 Int_t id4 = fGrid4->GetIndex(phi,eta)-1+n0+n1+n2+n3;
460 AliJetUnitArray *uArray4 = (AliJetUnitArray*)fUnitArray->At(id4+fNumUnits+nElements);
461 TRefArray *reference4 = uArray4->GetUnitTrackRef();
462 if (reference4->GetEntries() == 0) {
463 new(reference4) TRefArray(TProcessID::GetProcessWithUID(track));
466 reference4->Add(track);
468 Float_t uEnergy4 = uArray4->GetUnitEnergy();
470 Bool_t ref4 = kFALSE;
476 uArray4->SetUnitEnergy(uEnergy4+pt);
478 if(uArray4->GetUnitEnergy()<ptMin)
479 uArray4->SetUnitCutFlag(kPtSmaller);
481 uArray4->SetUnitCutFlag(kPtHigher);
482 if(ok4) nTracksEmcalDZCut++;
484 if(sflag[goodTrack] == 1) {
485 uArray4->SetUnitSignalFlag(kGood);
486 uArray4->SetUnitSignalFlagC(kFALSE,kGood);
487 } else uArray4->SetUnitSignalFlagC(kFALSE,kBad);
489 if(uArray4->GetUnitEnergy()>0 && ref4){
491 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray4));
494 fRefArray->Add(uArray4);
500 fGeom->GetAbsCellIdFromEtaPhi(eta,phi,towerID);
501 if(towerID==-1) continue;
503 AliJetUnitArray *uArray = (AliJetUnitArray*)fUnitArray->At(towerID);
504 TRefArray *reference = uArray->GetUnitTrackRef();
505 if (reference->GetEntries() == 0) {
506 new(reference) TRefArray(TProcessID::GetProcessWithUID(track));
509 reference->Add(track);
511 Float_t unitEnergy = uArray->GetUnitEnergy();
520 // Do Hadron Correction
521 // This is under construction !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
522 // Parametrization to be added
523 if (fApplyMIPCorrection != 0)
525 // Float_t hCEnergy = fHadCorr->GetEnergy(p3.Mag(), (Double_t)eta,0);
526 // unitEnergy -= hCEnergy*TMath::Sin(2.0*TMath::ATan(TMath::Exp(-eta)));
527 } //end Hadron Correction loop
529 uArray->SetUnitEnergy(unitEnergy + pt);
532 if(uArray->GetUnitEnergy()<ptMin){
533 uArray->SetUnitCutFlag(kPtSmaller);
536 uArray->SetUnitCutFlag(kPtHigher);
537 if(ok) nTracksEmcalCut++;
541 if(sflag[goodTrack] == 1) {
542 uArray->SetUnitSignalFlag(kGood);
543 uArray->SetUnitSignalFlagC(kFALSE,kGood);
544 } else uArray->SetUnitSignalFlagC(kFALSE,kBad);
546 if(uArray->GetUnitEnergy()>0 && ref){
548 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray));
551 fRefArray->Add(uArray);
553 } // end loop on EMCal acceptance cut + tracks quality
555 // Outside EMCal acceptance
556 // Int_t idTPC = GetIndexFromEtaPhi(etaT[i],phiT[i]);
557 Int_t idTPC = fTPCGrid->GetIndex(phi,eta);
559 AliJetUnitArray *uArray = (AliJetUnitArray*)fUnitArray->At(fNumUnits-1+idTPC);
560 TRefArray *reference = uArray->GetUnitTrackRef();
561 if (reference->GetEntries() == 0) {
562 new(reference) TRefArray(TProcessID::GetProcessWithUID(track));
564 reference->Add(track);
566 Float_t unitEnergy2 = uArray->GetUnitEnergy(); // check if fNumUnits or fNumUnits-1
567 Bool_t okout = kFALSE;
568 Bool_t refout = kFALSE;
574 // Fill energy outside emcal acceptance
575 uArray->SetUnitEnergy(unitEnergy2 + pt);
578 if(uArray->GetUnitEnergy()<ptMin){
579 uArray->SetUnitCutFlag(kPtSmaller);
582 uArray->SetUnitCutFlag(kPtHigher);
583 if(okout) nTracksTpcCut++;
586 if(sflag[goodTrack] == 1) {
587 uArray->SetUnitSignalFlag(kGood);
588 uArray->SetUnitSignalFlagC(kFALSE,kGood);
589 } else uArray->SetUnitSignalFlagC(kFALSE,kBad);
591 if(uArray->GetUnitEnergy()>0 && refout){
593 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray));
596 fRefArray->Add(uArray);
603 } // end loop on entries (tpc tracks)
608 cout << "End of Tracks %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%" << endl;
609 cout << "goodTracks: " << goodTrack << endl;
615 // } // end loop on entries (tpc tracks)
618 fNTracks = nTracksTpcOnly;
619 fNTracksCut = nTracksTpcOnlyCut;
621 cout << "fNTracks : " << fNTracks << endl;
622 cout << "fNTracksCut : " << fNTracksCut << endl;
626 fNTracks = nTracksEmcal+nTracksEmcalDZ+nTracksTpc;
627 fNTracksCut = nTracksEmcalCut+nTracksEmcalDZCut+nTracksTpcCut;
629 cout << "fNTracks : " << fNTracks << endl;
630 cout << "fNTracksCut : " << fNTracksCut << endl;