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 "AliJetHadronCorrectionv1.h"
33 #include "AliJetESDFillUnitArrayTracks.h"
35 // --- ROOT system ---
40 // --- AliRoot header files ---
43 class AliJetESDReader;
47 ClassImp(AliJetESDFillUnitArrayTracks)
49 //_____________________________________________________________________________
50 AliJetESDFillUnitArrayTracks::AliJetESDFillUnitArrayTracks()
51 : AliJetFillUnitArray(),
54 fApplyMIPCorrection(kTRUE),
65 //_____________________________________________________________________________
66 AliJetESDFillUnitArrayTracks::AliJetESDFillUnitArrayTracks(AliESDEvent* esd)
67 : AliJetFillUnitArray(),
70 fApplyMIPCorrection(kTRUE),
81 //_____________________________________________________________________________
82 AliJetESDFillUnitArrayTracks::AliJetESDFillUnitArrayTracks(const AliJetESDFillUnitArrayTracks &det)
83 : AliJetFillUnitArray(det),
84 fNumUnits(det.fNumUnits),
85 fHadCorr(det.fHadCorr),
86 fApplyMIPCorrection(det.fApplyMIPCorrection),
97 //------------------------------------------------------------------
98 AliJetESDFillUnitArrayTracks& AliJetESDFillUnitArrayTracks::operator=(const AliJetESDFillUnitArrayTracks& other)
102 fNumUnits = other.fNumUnits;
103 fHadCorr = other.fHadCorr;
104 fApplyMIPCorrection = other.fApplyMIPCorrection;
106 fGrid0 = other.fGrid0;
107 fGrid1 = other.fGrid1;
108 fGrid2 = other.fGrid2;
109 fGrid3 = other.fGrid3;
110 fGrid4 = other.fGrid4;
115 //____________________________________________________________________________
116 void AliJetESDFillUnitArrayTracks::InitParameters()
118 // fHadCorr = 0; // For hadron correction
119 fNumUnits = fGeom->GetEMCGeometry()->GetNCells(); // Number of towers in EMCAL
120 cout << "In AliJetESDFillUnitArrayTracks:InitParameters(), Ncells : " << fNumUnits << endl;
122 fTPCGrid->GetAccParam(fNphi,fNeta,fPhiMin,
123 fPhiMax,fEtaMin,fEtaMax);
124 fTPCGrid->GetBinParam(fPhiBinInTPCAcc,fEtaBinInTPCAcc,
125 fPhiBinInEMCalAcc,fEtaBinInEMCalAcc,fNbinPhi);
127 fIndex = fTPCGrid->GetIndexObject();
130 for(Int_t i=0; i<fNphi+1; i++)
131 for(Int_t j=0; j<fNeta+1; j++) {cout << "fIndex[" << i << "," << j << "] : " <<
132 (*fIndex)(i,j) << endl; }
134 if(fDebug>1) printf("\n Parameters initiated ! \n");
137 //_____________________________________________________________________________
138 AliJetESDFillUnitArrayTracks::~AliJetESDFillUnitArrayTracks()
143 //_____________________________________________________________________________
144 void AliJetESDFillUnitArrayTracks::Exec(Option_t* const /*option*/)
148 // Fill the unit array with the charged particle information in ESD
151 fDebug = fReaderHeader->GetDebug();
157 // get number of tracks in event (for the loop)
160 //(not used ?) Int_t nt2 = 0;
166 if(fDebug>1) cout << "fESD in Fill array : " << fESD << endl;
168 nt = fESD->GetNumberOfTracks();
169 if(fDebug>1) cout << "Number of Tracks in ESD : " << nt << endl;
171 // temporary storage of signal and pt cut flag
172 Int_t* sflag = new Int_t[nt];
173 Int_t* cflag = new Int_t[nt];
175 // get cuts set by user
176 Float_t ptMin = fReaderHeader->GetPtCut();
177 Float_t etaMin = fReaderHeader->GetFiducialEtaMin();
178 Float_t etaMax = fReaderHeader->GetFiducialEtaMax();
179 fOpt = fReaderHeader->GetDetector();
180 fDZ = fReaderHeader->GetDZ();
182 Int_t nTracksEmcal = 0;
183 Int_t nTracksEmcalDZ = 0;
184 Int_t nTracksTpc = 0;
185 Int_t nTracksTpcOnly = 0;
186 Int_t nTracksEmcalCut = 0;
187 Int_t nTracksEmcalDZCut = 0;
188 Int_t nTracksTpcCut = 0;
189 Int_t nTracksTpcOnlyCut = 0;
191 //(not used ?) Int_t nElements = fTPCGrid->GetNEntries();
192 //(not used ?) Int_t nElements2 = fEMCalGrid->GetNEntries();
193 fGrid = fTPCGrid->GetGridType();
195 //(not used ?) Int_t nEntries = fUnitArray->GetEntries();
197 //(not used ?) Int_t nRefEnt = fRefArray->GetEntries();
201 for (Int_t it = 0; it < nmax; it++) {
203 track = fESD->GetTrack(it);
204 UInt_t status = track->GetStatus();
207 track->GetPxPyPz(mom);
209 p3.SetXYZ(mom[0],mom[1],mom[2]);
213 mass = track->GetMass();
215 if ((status & AliESDtrack::kTPCrefit) == 0) continue; // quality check
216 if ((status & AliESDtrack::kITSrefit) == 0) continue; // quality check
217 if (((AliJetESDReaderHeader*) fReaderHeader)->ReadSignalOnly()
218 && TMath::Abs(track->GetLabel()) > 10000) continue; // quality check
219 if (((AliJetESDReaderHeader*) fReaderHeader)->ReadBkgdOnly()
220 && TMath::Abs(track->GetLabel()) < 10000) continue; // quality check
222 phi = ( (p3.Phi()) < 0) ? (p3.Phi()) + 2. * TMath::Pi() : (p3.Phi());
224 if ( (eta > etaMax) || (eta < etaMin)) continue; // checking eta cut
227 if (TMath::Abs(track->GetLabel()) < 10000) sflag[goodTrack]=1;
229 if (pt > ptMin) cflag[goodTrack]=1; // pt cut
234 // Only TPC filled from its grid in its total acceptance
236 Int_t idTPC = fTPCGrid->GetIndex(phi,eta);
240 AliJetUnitArray *uArray = (AliJetUnitArray*)fUnitArray->At(idTPC-1);
241 TRefArray *reference = uArray->GetUnitTrackRef();
242 if (reference->GetEntries() == 0) {
243 new(reference) TRefArray(TProcessID::GetProcessWithUID(track));
246 reference->Add(track);
248 Float_t unitEnergy = 0.;
249 unitEnergy = uArray->GetUnitEnergy();
250 // nTracksTpcOnly is necessary to count the number of candidate cells
251 // but it doesn't give the real multiplicity -> it will be extracted
252 // in the jet finder through the reference to tracks
259 // Fill energy in TPC acceptance
260 uArray->SetUnitEnergy(unitEnergy + pt);
261 uArray->SetUnitMass(mass);
264 if(uArray->GetUnitEnergy()<ptMin){
265 uArray->SetUnitCutFlag(kPtSmaller);
268 uArray->SetUnitCutFlag(kPtHigher);
269 if(ok) nTracksTpcOnlyCut++;
273 if(sflag[goodTrack] == 1) {
274 uArray->SetUnitSignalFlag(kGood);
275 uArray->SetUnitSignalFlagC(kFALSE,kGood);
276 } else uArray->SetUnitSignalFlagC(kFALSE,kBad);
278 if(uArray->GetUnitEnergy()>0 && ref){
281 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray));
283 fRefArray->Add(uArray);
289 Int_t nElements = fTPCGrid->GetNEntries();
290 // Fill track information in EMCAL acceptance
291 if((eta >= fEtaMin && eta <= fEtaMax) &&
292 (phi >= fPhiMin && phi <= fPhiMax))// &&
295 // Include dead-zones
298 Double_t phimin0 = 0., phimin1 = 0., phimin2 = 0., phimin3 = 0., phimin4 = 0.;
299 Double_t phimax0 = 0., phimax1 = 0., phimax2 = 0., phimax3 = 0., phimax4 = 0.;
300 fGeom->GetEMCGeometry()->GetPhiBoundariesOfSMGap(0,phimin0,phimax0);
301 fGeom->GetEMCGeometry()->GetPhiBoundariesOfSMGap(1,phimin1,phimax1);
302 fGeom->GetEMCGeometry()->GetPhiBoundariesOfSMGap(2,phimin2,phimax2);
303 fGeom->GetEMCGeometry()->GetPhiBoundariesOfSMGap(3,phimin3,phimax3);
304 fGeom->GetEMCGeometry()->GetPhiBoundariesOfSMGap(4,phimin4,phimax4);
305 Int_t n0 = fGrid0->GetNEntries();
306 Int_t n1 = fGrid1->GetNEntries();
307 Int_t n2 = fGrid2->GetNEntries();
308 Int_t n3 = fGrid3->GetNEntries();
309 //(not used ?) Int_t n4 = fGrid4->GetNEntries();
311 if(phi >= phimin0 && phi <= phimax0){
312 Int_t id0 = fGrid0->GetIndex(phi,eta)-1;
313 AliJetUnitArray *uArray0 = (AliJetUnitArray*)fUnitArray->At(id0+fNumUnits+nElements);
314 TRefArray *reference0 = uArray0->GetUnitTrackRef();
315 if (reference0->GetEntries() == 0) {
316 new(reference0) TRefArray(TProcessID::GetProcessWithUID(track));
319 reference0->Add(track);
321 Float_t uEnergy0 = uArray0->GetUnitEnergy();
323 Bool_t ref0 = kFALSE;
329 uArray0->SetUnitEnergy(uEnergy0+pt);
331 if(uArray0->GetUnitEnergy()<ptMin)
332 uArray0->SetUnitCutFlag(kPtSmaller);
334 uArray0->SetUnitCutFlag(kPtHigher);
335 if(ok0) nTracksEmcalDZCut++;
337 if(sflag[goodTrack] == 1) {
338 uArray0->SetUnitSignalFlag(kGood);
339 uArray0->SetUnitSignalFlagC(kFALSE,kGood);
340 } else uArray0->SetUnitSignalFlagC(kFALSE,kBad);
342 if(uArray0->GetUnitEnergy()>0 && ref0){
344 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray0));
347 fRefArray->Add(uArray0);
351 if(phi >= phimin1 && phi <= phimax1){
352 Int_t id1 = fGrid1->GetIndex(phi,eta)-1+n0;
353 AliJetUnitArray *uArray1 = (AliJetUnitArray*)fUnitArray->At(id1+fNumUnits+nElements);
354 TRefArray *reference1 = uArray1->GetUnitTrackRef();
355 if (reference1->GetEntries() == 0) {
356 new(reference1) TRefArray(TProcessID::GetProcessWithUID(track));
359 reference1->Add(track);
361 Float_t uEnergy1 = uArray1->GetUnitEnergy();
363 Bool_t ref1 = kFALSE;
369 uArray1->SetUnitEnergy(uEnergy1+pt);
371 if(uArray1->GetUnitEnergy()<ptMin)
372 uArray1->SetUnitCutFlag(kPtSmaller);
374 uArray1->SetUnitCutFlag(kPtHigher);
375 if(ok1) nTracksEmcalDZCut++;
377 if(sflag[goodTrack] == 1) {
378 uArray1->SetUnitSignalFlag(kGood);
379 uArray1->SetUnitSignalFlagC(kFALSE,kGood);
380 } else uArray1->SetUnitSignalFlagC(kFALSE,kBad);
382 if(uArray1->GetUnitEnergy()>0 && ref1){
384 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray1));
387 fRefArray->Add(uArray1);
391 if(phi >= phimin2 && phi <= phimax2){
392 Int_t id2 = fGrid2->GetIndex(phi,eta)-1+n0+n1;
393 AliJetUnitArray *uArray2 = (AliJetUnitArray*)fUnitArray->At(id2+fNumUnits+nElements);
394 TRefArray *reference2 = uArray2->GetUnitTrackRef();
395 if (reference2->GetEntries() == 0) {
396 new(reference2) TRefArray(TProcessID::GetProcessWithUID(track));
399 reference2->Add(track);
401 Float_t uEnergy2 = uArray2->GetUnitEnergy();
403 Bool_t ref2 = kFALSE;
409 uArray2->SetUnitEnergy(uEnergy2+pt);
411 if(uArray2->GetUnitEnergy()<ptMin)
412 uArray2->SetUnitCutFlag(kPtSmaller);
414 uArray2->SetUnitCutFlag(kPtHigher);
415 if(ok2) nTracksEmcalDZCut++;
417 if(sflag[goodTrack] == 1) {
418 uArray2->SetUnitSignalFlag(kGood);
419 uArray2->SetUnitSignalFlagC(kFALSE,kGood);
420 } else uArray2->SetUnitSignalFlagC(kFALSE,kBad);
422 if(uArray2->GetUnitEnergy()>0 && ref2){
424 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray2));
427 fRefArray->Add(uArray2);
431 if(phi >= phimin3 && phi <= phimax3){
432 Int_t id3 = fGrid3->GetIndex(phi,eta)-1+n0+n1+n2;
433 AliJetUnitArray *uArray3 = (AliJetUnitArray*)fUnitArray->At(id3+fNumUnits+nElements);
434 TRefArray *reference3 = uArray3->GetUnitTrackRef();
435 if (reference3->GetEntries() == 0) {
436 new(reference3) TRefArray(TProcessID::GetProcessWithUID(track));
439 reference3->Add(track);
441 Float_t uEnergy3 = uArray3->GetUnitEnergy();
443 Bool_t ref3 = kFALSE;
449 uArray3->SetUnitEnergy(uEnergy3+pt);
451 if(uArray3->GetUnitEnergy()<ptMin)
452 uArray3->SetUnitCutFlag(kPtSmaller);
454 uArray3->SetUnitCutFlag(kPtHigher);
455 if(ok3) nTracksEmcalDZCut++;
457 if(sflag[goodTrack] == 1) {
458 uArray3->SetUnitSignalFlag(kGood);
459 uArray3->SetUnitSignalFlagC(kFALSE,kGood);
460 } else uArray3->SetUnitSignalFlagC(kFALSE,kBad);
462 if(uArray3->GetUnitEnergy()>0 && ref3){
464 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray3));
467 fRefArray->Add(uArray3);
471 if(phi >= phimin4 && phi <= phimax4){
472 Int_t id4 = fGrid4->GetIndex(phi,eta)-1+n0+n1+n2+n3;
473 AliJetUnitArray *uArray4 = (AliJetUnitArray*)fUnitArray->At(id4+fNumUnits+nElements);
474 TRefArray *reference4 = uArray4->GetUnitTrackRef();
475 if (reference4->GetEntries() == 0) {
476 new(reference4) TRefArray(TProcessID::GetProcessWithUID(track));
479 reference4->Add(track);
481 Float_t uEnergy4 = uArray4->GetUnitEnergy();
483 Bool_t ref4 = kFALSE;
489 uArray4->SetUnitEnergy(uEnergy4+pt);
491 if(uArray4->GetUnitEnergy()<ptMin)
492 uArray4->SetUnitCutFlag(kPtSmaller);
494 uArray4->SetUnitCutFlag(kPtHigher);
495 if(ok4) nTracksEmcalDZCut++;
497 if(sflag[goodTrack] == 1) {
498 uArray4->SetUnitSignalFlag(kGood);
499 uArray4->SetUnitSignalFlagC(kFALSE,kGood);
500 } else uArray4->SetUnitSignalFlagC(kFALSE,kBad);
502 if(uArray4->GetUnitEnergy()>0 && ref4){
504 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray4));
507 fRefArray->Add(uArray4);
513 fGeom->GetAbsCellIdFromEtaPhi(eta,phi,towerID);
514 if(towerID==-1) continue;
516 AliJetUnitArray *uArray = (AliJetUnitArray*)fUnitArray->At(towerID);
517 TRefArray *reference = uArray->GetUnitTrackRef();
518 if (reference->GetEntries() == 0) {
519 new(reference) TRefArray(TProcessID::GetProcessWithUID(track));
522 reference->Add(track);
524 Float_t unitEnergy = uArray->GetUnitEnergy();
533 // Do Hadron Correction
534 // This is under construction !!!!!!!!!!!!!!!!!!!!!!!
535 // For the moment I apply MIP correction if p >= 0.5 GeV/c
536 if (fApplyMIPCorrection != 0 && p3.Mag() >= 0.5)
538 ((AliJetHadronCorrectionv1*)fHadCorr)->SetGeometry("EMCAL_COMPLETEV1",1.);
540 // Get track position at the outer part of the reconstruction ~ TRD
541 Double_t phiOut = track->GetOuterParam()->Phi();
542 Double_t etaOut = track->GetOuterParam()->Eta();
544 // If the track in the outer part of the TPC/TDR ? is inside
545 // the calorimeter, it can deposit part of its energy
546 // We can then correct on average for these particles
547 if((etaOut >= fEtaMin && etaOut <= fEtaMax) &&
548 (phiOut >= fPhiMin && phiOut <= fPhiMax))// &&
550 Double_t hCEnergy = (Double_t)fHadCorr->GetEnergy(p3.Mag(), (Double_t)eta,0);
551 unitEnergy -= hCEnergy*TMath::Sin(2.0*TMath::ATan(TMath::Exp(-eta)));
553 } //end Hadron Correction loop
555 cout << "unitEnergy + pt = " << unitEnergy << " + " << pt << " = " << unitEnergy + pt << endl;
557 if((unitEnergy + pt) > 0.) uArray->SetUnitEnergy(unitEnergy + pt);
558 else uArray->SetUnitEnergy(0.);
561 if(uArray->GetUnitEnergy()<ptMin){
562 uArray->SetUnitCutFlag(kPtSmaller);
565 uArray->SetUnitCutFlag(kPtHigher);
566 if(ok) nTracksEmcalCut++;
570 if(sflag[goodTrack] == 1) {
571 uArray->SetUnitSignalFlag(kGood);
572 uArray->SetUnitSignalFlagC(kFALSE,kGood);
573 } else uArray->SetUnitSignalFlagC(kFALSE,kBad);
576 if(uArray->GetUnitEnergy()>0 && ref){
578 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray));
581 fRefArray->Add(uArray);
583 } // end loop on EMCal acceptance cut + tracks quality
585 // Outside EMCal acceptance
587 Int_t idTPC = fTPCGrid->GetIndex(phi,eta);
589 AliJetUnitArray *uArray = (AliJetUnitArray*)fUnitArray->At(fNumUnits-1+idTPC);
590 TRefArray *reference = uArray->GetUnitTrackRef();
591 if (reference->GetEntries() == 0) {
592 new(reference) TRefArray(TProcessID::GetProcessWithUID(track));
595 reference->Add(track);
597 Float_t unitEnergy2 = uArray->GetUnitEnergy();
598 Bool_t okout = kFALSE;
599 Bool_t refout = kFALSE;
605 // Fill energy outside emcal acceptance
606 uArray->SetUnitEnergy(unitEnergy2 + pt);
609 if(uArray->GetUnitEnergy()<ptMin){
610 uArray->SetUnitCutFlag(kPtSmaller);
613 uArray->SetUnitCutFlag(kPtHigher);
614 if(okout) nTracksTpcCut++;
617 if(sflag[goodTrack] == 1) {
618 uArray->SetUnitSignalFlag(kGood);
619 uArray->SetUnitSignalFlagC(kFALSE,kGood);
620 } else uArray->SetUnitSignalFlagC(kFALSE,kBad);
623 if(uArray->GetUnitEnergy()>0 && refout){
625 new(fRefArray) TRefArray(TProcessID::GetProcessWithUID(uArray));
628 fRefArray->Add(uArray);
635 } // end loop on entries (tpc tracks)
639 cout << "End of Tracks %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%" << endl;
640 cout << "goodTracks: " << goodTrack << endl;
643 // fSignalFlag.Set(goodTrack,sflag);
644 // fCutFlag.Set(goodTrack,cflag);
650 fNTracks = nTracksTpcOnly;
651 fNTracksCut = nTracksTpcOnlyCut;
653 cout << "fNTracks : " << fNTracks << endl;
654 cout << "fNTracksCut : " << fNTracksCut << endl;
658 fNTracks = nTracksEmcal+nTracksEmcalDZ+nTracksTpc;
659 fNTracksCut = nTracksEmcalCut+nTracksEmcalDZCut+nTracksTpcCut;
661 cout << "fNTracks : " << fNTracks << endl;
662 cout << "fNTracksCut : " << fNTracksCut << endl;