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 // --- AliRoot header files ---
31 #include "AliJetUnitArray.h"
32 #include "AliJetESDFillUnitArrayTracks.h"
34 // --- ROOT system ---
39 // --- AliRoot header files ---
42 class AliJetESDReader;
44 ClassImp(AliJetESDFillUnitArrayTracks)
46 //_____________________________________________________________________________
47 AliJetESDFillUnitArrayTracks::AliJetESDFillUnitArrayTracks()
48 : AliJetFillUnitArray(),
51 fApplyMIPCorrection(kTRUE),
62 //_____________________________________________________________________________
63 AliJetESDFillUnitArrayTracks::AliJetESDFillUnitArrayTracks(AliESDEvent* esd)
64 : AliJetFillUnitArray(),
67 fApplyMIPCorrection(kTRUE),
78 //_____________________________________________________________________________
79 AliJetESDFillUnitArrayTracks::AliJetESDFillUnitArrayTracks(const AliJetESDFillUnitArrayTracks &det)
80 : AliJetFillUnitArray(det),
81 fNumUnits(det.fNumUnits),
82 fHadCorr(det.fHadCorr),
83 fApplyMIPCorrection(det.fApplyMIPCorrection),
94 //------------------------------------------------------------------
95 AliJetESDFillUnitArrayTracks& AliJetESDFillUnitArrayTracks::operator=(const AliJetESDFillUnitArrayTracks& other)
99 fNumUnits = other.fNumUnits;
100 fHadCorr = other.fHadCorr;
101 fApplyMIPCorrection = other.fApplyMIPCorrection;
103 fGrid0 = other.fGrid0;
104 fGrid1 = other.fGrid1;
105 fGrid2 = other.fGrid2;
106 fGrid3 = other.fGrid3;
107 fGrid4 = other.fGrid4;
112 //____________________________________________________________________________
113 void AliJetESDFillUnitArrayTracks::InitParameters()
115 fHadCorr = 0; // For hadron correction
116 fNumUnits = fGeom->GetNCells(); // Number of towers in EMCAL
117 cout << "In AliJetESDFillUnitArrayTracks:InitParameters(), Ncells : " << fNumUnits << endl;
119 fTPCGrid->GetAccParam(fNphi,fNeta,fPhiMin,
120 fPhiMax,fEtaMin,fEtaMax);
121 fTPCGrid->GetBinParam(fPhiBinInTPCAcc,fEtaBinInTPCAcc,
122 fPhiBinInEMCalAcc,fEtaBinInEMCalAcc,fNbinPhi);
124 fIndex = fTPCGrid->GetIndexObject();
127 for(Int_t i=0; i<fNphi+1; i++)
128 for(Int_t j=0; j<fNeta+1; j++) {cout << "fIndex[" << i << "," << j << "] : " <<
129 (*fIndex)(i,j) << endl; }
131 if(fDebug>1) printf("\n Parameters initiated ! \n");
134 //_____________________________________________________________________________
135 AliJetESDFillUnitArrayTracks::~AliJetESDFillUnitArrayTracks()
140 //_____________________________________________________________________________
141 void AliJetESDFillUnitArrayTracks::Exec(Option_t* const /*option*/)
145 // Fill the unit array with the charged particle information in ESD
148 fDebug = fReaderHeader->GetDebug();
153 // get number of tracks in event (for the loop)
156 //(not used ?) Int_t nt2 = 0;
162 if(fDebug>1) cout << "fESD in Fill array : " << fESD << endl;
164 nt = fESD->GetNumberOfTracks();
165 if(fDebug>1) cout << "Number of Tracks in ESD : " << nt << endl;
167 // temporary storage of signal and pt cut flag
168 Int_t* sflag = new Int_t[nt];
169 Int_t* cflag = new Int_t[nt];
171 // get cuts set by user
172 Float_t ptMin = fReaderHeader->GetPtCut();
173 Float_t etaMin = fReaderHeader->GetFiducialEtaMin();
174 Float_t etaMax = fReaderHeader->GetFiducialEtaMax();
175 fOpt = fReaderHeader->GetDetector();
176 fDZ = fReaderHeader->GetDZ();
178 Int_t nTracksEmcal = 0;
179 Int_t nTracksEmcalDZ = 0;
180 Int_t nTracksTpc = 0;
181 Int_t nTracksTpcOnly = 0;
182 Int_t nTracksEmcalCut = 0;
183 Int_t nTracksEmcalDZCut = 0;
184 Int_t nTracksTpcCut = 0;
185 Int_t nTracksTpcOnlyCut = 0;
187 //(not used ?) Int_t nElements = fTPCGrid->GetNEntries();
188 //(not used ?) Int_t nElements2 = fEMCalGrid->GetNEntries();
189 fGrid = fTPCGrid->GetGridType();
191 //(not used ?) Int_t nEntries = fUnitArray->GetEntries();
193 //(not used ?) Int_t nRefEnt = fRefArray->GetEntries();
197 for (Int_t it = 0; it < nmax; it++) {
199 track = fESD->GetTrack(it);
200 UInt_t status = track->GetStatus();
203 track->GetPxPyPz(mom);
205 p3.SetXYZ(mom[0],mom[1],mom[2]);
209 mass = track->GetMass();
211 if ((status & AliESDtrack::kTPCrefit) == 0) continue; // quality check
212 if ((status & AliESDtrack::kITSrefit) == 0) continue; // quality check
213 if (((AliJetESDReaderHeader*) fReaderHeader)->ReadSignalOnly()
214 && TMath::Abs(track->GetLabel()) > 10000) continue; // quality check
215 if (((AliJetESDReaderHeader*) fReaderHeader)->ReadBkgdOnly()
216 && TMath::Abs(track->GetLabel()) < 10000) continue; // quality check
218 phi = ( (p3.Phi()) < 0) ? (p3.Phi()) + 2. * TMath::Pi() : (p3.Phi());
220 if ( (eta > etaMax) || (eta < etaMin)) continue; // checking eta cut
223 if (TMath::Abs(track->GetLabel()) < 10000) sflag[goodTrack]=1;
225 if (pt > ptMin) cflag[goodTrack]=1; // pt cut
229 // Only TPC filled from its grid in its total acceptance
231 Int_t idTPC = fTPCGrid->GetIndex(phi,eta);
235 AliJetUnitArray *uArray = (AliJetUnitArray*)fUnitArray->At(idTPC-1);
236 TRefArray *reference = uArray->GetUnitTrackRef();
237 if (reference->GetEntries() == 0) {
238 new(reference) TRefArray(TProcessID::GetProcessWithUID(track));
241 reference->Add(track);
243 Float_t unitEnergy = 0.;
244 unitEnergy = uArray->GetUnitEnergy();
245 // nTracksTpcOnly is necessary to count the number of candidate cells
246 // but it doesn't give the real multiplicity -> it will be extracted
247 // in the jet finder through the reference to tracks
254 // Fill energy in TPC acceptance
255 uArray->SetUnitEnergy(unitEnergy + pt);
256 uArray->SetUnitMass(mass);
259 if(uArray->GetUnitEnergy()<ptMin){
260 uArray->SetUnitCutFlag(kPtSmaller);
263 uArray->SetUnitCutFlag(kPtHigher);
264 if(ok) nTracksTpcOnlyCut++;
268 if(sflag[goodTrack] == 1) {
269 uArray->SetUnitSignalFlag(kGood);
270 uArray->SetUnitSignalFlagC(kFALSE,kGood);
271 } else uArray->SetUnitSignalFlagC(kFALSE,kBad);
273 if(uArray->GetUnitEnergy()>0 && ref){
276 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray));
278 fRefArray->Add(uArray);
284 Int_t nElements = fTPCGrid->GetNEntries();
285 // Fill track information in EMCAL acceptance
286 if((eta >= fEtaMin && eta <= fEtaMax) &&
287 (phi >= fPhiMin && phi <= fPhiMax))// &&
290 // Include dead-zones
293 Double_t phimin0 = 0., phimin1 = 0., phimin2 = 0., phimin3 = 0., phimin4 = 0.;
294 Double_t phimax0 = 0., phimax1 = 0., phimax2 = 0., phimax3 = 0., phimax4 = 0.;
295 fGeom->GetPhiBoundariesOfSMGap(0,phimin0,phimax0);
296 fGeom->GetPhiBoundariesOfSMGap(1,phimin1,phimax1);
297 fGeom->GetPhiBoundariesOfSMGap(2,phimin2,phimax2);
298 fGeom->GetPhiBoundariesOfSMGap(3,phimin3,phimax3);
299 fGeom->GetPhiBoundariesOfSMGap(4,phimin4,phimax4);
300 Int_t n0 = fGrid0->GetNEntries();
301 Int_t n1 = fGrid1->GetNEntries();
302 Int_t n2 = fGrid2->GetNEntries();
303 Int_t n3 = fGrid3->GetNEntries();
304 //(not used ?) Int_t n4 = fGrid4->GetNEntries();
306 if(phi >= phimin0 && phi <= phimax0){
307 Int_t id0 = fGrid0->GetIndex(phi,eta)-1;
308 AliJetUnitArray *uArray0 = (AliJetUnitArray*)fUnitArray->At(id0+fNumUnits+nElements);
309 TRefArray *reference0 = uArray0->GetUnitTrackRef();
310 if (reference0->GetEntries() == 0) {
311 new(reference0) TRefArray(TProcessID::GetProcessWithUID(track));
314 reference0->Add(track);
316 Float_t uEnergy0 = uArray0->GetUnitEnergy();
318 Bool_t ref0 = kFALSE;
324 uArray0->SetUnitEnergy(uEnergy0+pt);
326 if(uArray0->GetUnitEnergy()<ptMin)
327 uArray0->SetUnitCutFlag(kPtSmaller);
329 uArray0->SetUnitCutFlag(kPtHigher);
330 if(ok0) nTracksEmcalDZCut++;
332 if(sflag[goodTrack] == 1) {
333 uArray0->SetUnitSignalFlag(kGood);
334 uArray0->SetUnitSignalFlagC(kFALSE,kGood);
335 } else uArray0->SetUnitSignalFlagC(kFALSE,kBad);
337 if(uArray0->GetUnitEnergy()>0 && ref0){
339 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray0));
342 fRefArray->Add(uArray0);
346 if(phi >= phimin1 && phi <= phimax1){
347 Int_t id1 = fGrid1->GetIndex(phi,eta)-1+n0;
348 AliJetUnitArray *uArray1 = (AliJetUnitArray*)fUnitArray->At(id1+fNumUnits+nElements);
349 TRefArray *reference1 = uArray1->GetUnitTrackRef();
350 if (reference1->GetEntries() == 0) {
351 new(reference1) TRefArray(TProcessID::GetProcessWithUID(track));
354 reference1->Add(track);
356 Float_t uEnergy1 = uArray1->GetUnitEnergy();
358 Bool_t ref1 = kFALSE;
364 uArray1->SetUnitEnergy(uEnergy1+pt);
366 if(uArray1->GetUnitEnergy()<ptMin)
367 uArray1->SetUnitCutFlag(kPtSmaller);
369 uArray1->SetUnitCutFlag(kPtHigher);
370 if(ok1) nTracksEmcalDZCut++;
372 if(sflag[goodTrack] == 1) {
373 uArray1->SetUnitSignalFlag(kGood);
374 uArray1->SetUnitSignalFlagC(kFALSE,kGood);
375 } else uArray1->SetUnitSignalFlagC(kFALSE,kBad);
377 if(uArray1->GetUnitEnergy()>0 && ref1){
379 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray1));
382 fRefArray->Add(uArray1);
386 if(phi >= phimin2 && phi <= phimax2){
387 Int_t id2 = fGrid2->GetIndex(phi,eta)-1+n0+n1;
388 AliJetUnitArray *uArray2 = (AliJetUnitArray*)fUnitArray->At(id2+fNumUnits+nElements);
389 TRefArray *reference2 = uArray2->GetUnitTrackRef();
390 if (reference2->GetEntries() == 0) {
391 new(reference2) TRefArray(TProcessID::GetProcessWithUID(track));
394 reference2->Add(track);
396 Float_t uEnergy2 = uArray2->GetUnitEnergy();
398 Bool_t ref2 = kFALSE;
404 uArray2->SetUnitEnergy(uEnergy2+pt);
406 if(uArray2->GetUnitEnergy()<ptMin)
407 uArray2->SetUnitCutFlag(kPtSmaller);
409 uArray2->SetUnitCutFlag(kPtHigher);
410 if(ok2) nTracksEmcalDZCut++;
412 if(sflag[goodTrack] == 1) {
413 uArray2->SetUnitSignalFlag(kGood);
414 uArray2->SetUnitSignalFlagC(kFALSE,kGood);
415 } else uArray2->SetUnitSignalFlagC(kFALSE,kBad);
417 if(uArray2->GetUnitEnergy()>0 && ref2){
419 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray2));
422 fRefArray->Add(uArray2);
426 if(phi >= phimin3 && phi <= phimax3){
427 Int_t id3 = fGrid3->GetIndex(phi,eta)-1+n0+n1+n2;
428 AliJetUnitArray *uArray3 = (AliJetUnitArray*)fUnitArray->At(id3+fNumUnits+nElements);
429 TRefArray *reference3 = uArray3->GetUnitTrackRef();
430 if (reference3->GetEntries() == 0) {
431 new(reference3) TRefArray(TProcessID::GetProcessWithUID(track));
434 reference3->Add(track);
436 Float_t uEnergy3 = uArray3->GetUnitEnergy();
438 Bool_t ref3 = kFALSE;
444 uArray3->SetUnitEnergy(uEnergy3+pt);
446 if(uArray3->GetUnitEnergy()<ptMin)
447 uArray3->SetUnitCutFlag(kPtSmaller);
449 uArray3->SetUnitCutFlag(kPtHigher);
450 if(ok3) nTracksEmcalDZCut++;
452 if(sflag[goodTrack] == 1) {
453 uArray3->SetUnitSignalFlag(kGood);
454 uArray3->SetUnitSignalFlagC(kFALSE,kGood);
455 } else uArray3->SetUnitSignalFlagC(kFALSE,kBad);
457 if(uArray3->GetUnitEnergy()>0 && ref3){
459 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray3));
462 fRefArray->Add(uArray3);
466 if(phi >= phimin4 && phi <= phimax4){
467 Int_t id4 = fGrid4->GetIndex(phi,eta)-1+n0+n1+n2+n3;
468 AliJetUnitArray *uArray4 = (AliJetUnitArray*)fUnitArray->At(id4+fNumUnits+nElements);
469 TRefArray *reference4 = uArray4->GetUnitTrackRef();
470 if (reference4->GetEntries() == 0) {
471 new(reference4) TRefArray(TProcessID::GetProcessWithUID(track));
474 reference4->Add(track);
476 Float_t uEnergy4 = uArray4->GetUnitEnergy();
478 Bool_t ref4 = kFALSE;
484 uArray4->SetUnitEnergy(uEnergy4+pt);
486 if(uArray4->GetUnitEnergy()<ptMin)
487 uArray4->SetUnitCutFlag(kPtSmaller);
489 uArray4->SetUnitCutFlag(kPtHigher);
490 if(ok4) nTracksEmcalDZCut++;
492 if(sflag[goodTrack] == 1) {
493 uArray4->SetUnitSignalFlag(kGood);
494 uArray4->SetUnitSignalFlagC(kFALSE,kGood);
495 } else uArray4->SetUnitSignalFlagC(kFALSE,kBad);
497 if(uArray4->GetUnitEnergy()>0 && ref4){
499 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray4));
502 fRefArray->Add(uArray4);
508 fGeom->GetAbsCellIdFromEtaPhi(eta,phi,towerID);
509 if(towerID==-1) continue;
511 AliJetUnitArray *uArray = (AliJetUnitArray*)fUnitArray->At(towerID);
512 TRefArray *reference = uArray->GetUnitTrackRef();
513 if (reference->GetEntries() == 0) {
514 new(reference) TRefArray(TProcessID::GetProcessWithUID(track));
517 reference->Add(track);
519 Float_t unitEnergy = uArray->GetUnitEnergy();
528 // Do Hadron Correction
529 // This is under construction !!!!!!!!!!!!!!!!!!!!!!!
530 // Parametrization to be added
531 if (fApplyMIPCorrection != 0)
533 // Float_t hCEnergy = fHadCorr->GetEnergy(p3.Mag(), (Double_t)eta,0);
534 // unitEnergy -= hCEnergy*TMath::Sin(2.0*TMath::ATan(TMath::Exp(-eta)));
535 } //end Hadron Correction loop
537 uArray->SetUnitEnergy(unitEnergy + pt);
540 if(uArray->GetUnitEnergy()<ptMin){
541 uArray->SetUnitCutFlag(kPtSmaller);
544 uArray->SetUnitCutFlag(kPtHigher);
545 if(ok) nTracksEmcalCut++;
549 if(sflag[goodTrack] == 1) {
550 uArray->SetUnitSignalFlag(kGood);
551 uArray->SetUnitSignalFlagC(kFALSE,kGood);
552 } else uArray->SetUnitSignalFlagC(kFALSE,kBad);
555 if(uArray->GetUnitEnergy()>0 && ref){
557 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray));
560 fRefArray->Add(uArray);
562 } // end loop on EMCal acceptance cut + tracks quality
564 // Outside EMCal acceptance
566 Int_t idTPC = fTPCGrid->GetIndex(phi,eta);
568 AliJetUnitArray *uArray = (AliJetUnitArray*)fUnitArray->At(fNumUnits-1+idTPC);
569 TRefArray *reference = uArray->GetUnitTrackRef();
570 if (reference->GetEntries() == 0) {
571 new(reference) TRefArray(TProcessID::GetProcessWithUID(track));
574 reference->Add(track);
576 Float_t unitEnergy2 = uArray->GetUnitEnergy();
577 Bool_t okout = kFALSE;
578 Bool_t refout = kFALSE;
584 // Fill energy outside emcal acceptance
585 uArray->SetUnitEnergy(unitEnergy2 + pt);
588 if(uArray->GetUnitEnergy()<ptMin){
589 uArray->SetUnitCutFlag(kPtSmaller);
592 uArray->SetUnitCutFlag(kPtHigher);
593 if(okout) nTracksTpcCut++;
596 if(sflag[goodTrack] == 1) {
597 uArray->SetUnitSignalFlag(kGood);
598 uArray->SetUnitSignalFlagC(kFALSE,kGood);
599 } else uArray->SetUnitSignalFlagC(kFALSE,kBad);
602 if(uArray->GetUnitEnergy()>0 && refout){
604 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray));
607 fRefArray->Add(uArray);
614 } // end loop on entries (tpc tracks)
618 cout << "End of Tracks %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%" << endl;
619 cout << "goodTracks: " << goodTrack << endl;
626 fNTracks = nTracksTpcOnly;
627 fNTracksCut = nTracksTpcOnlyCut;
629 cout << "fNTracks : " << fNTracks << endl;
630 cout << "fNTracksCut : " << fNTracksCut << endl;
634 fNTracks = nTracksEmcal+nTracksEmcalDZ+nTracksTpc;
635 fNTracksCut = nTracksEmcalCut+nTracksEmcalDZCut+nTracksTpcCut;
637 cout << "fNTracks : " << fNTracks << endl;
638 cout << "fNTracksCut : " << fNTracksCut << endl;