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 **************************************************************************/
16 //-------------------------------------------------------------------------
18 // AOD reader for jet analysis
19 // This is the reader which must be used if the jet analysis task
20 // is executed after the ESD filter task, in order to read its output
22 // Author: Davide Perrino <davide.perrino@cern.ch>
23 //-------------------------------------------------------------------------
26 #include <Riostream.h>
28 #include <TLorentzVector.h>
33 #include <TGeoManager.h>
35 #include "AliJetAODReader.h"
36 #include "AliJetAODReaderHeader.h"
37 #include "AliAODEvent.h"
38 #include "AliAODTrack.h"
39 #include "AliJetDummyGeo.h"
40 #include "AliJetAODFillUnitArrayTracks.h"
41 #include "AliJetAODFillUnitArrayEMCalDigits.h"
42 #include "AliJetHadronCorrection.h"
43 #include "AliJetUnitArray.h"
45 ClassImp(AliJetAODReader)
47 AliJetAODReader::AliJetAODReader():
63 fApplyElectronCorrection(kFALSE),
64 fApplyMIPCorrection(kTRUE),
65 fApplyFractionHadronicCorrection(kFALSE),
66 fFractionHadronicCorrection(0.3),
80 //____________________________________________________________________________
82 AliJetAODReader::~AliJetAODReader()
100 //____________________________________________________________________________
102 void AliJetAODReader::OpenInputFiles()
104 // Open the necessary input files
105 // chain for the AODs
106 fChain = new TChain("aodTree");
108 // get directory and pattern name from the header
109 const char* dirName=fReaderHeader->GetDirectory();
110 const char* pattern=fReaderHeader->GetPattern();
112 // // Add files matching patters to the chain
114 void *dir = gSystem->OpenDirectory(dirName);
115 const char *name = 0x0;
116 int naod = ((AliJetAODReaderHeader*) fReaderHeader)->GetNaod();
118 while ((name = gSystem->GetDirEntry(dir))){
119 if (a>=naod) continue;
121 if (strstr(name,pattern)){
123 sprintf(path,"%s/%s/aod.root",dirName,name);
124 fChain->AddFile(path);
129 gSystem->FreeDirectory(dir);
132 fChain->SetBranchAddress("AOD",&fAOD);
134 int nMax = fChain->GetEntries();
136 printf("\n AliJetAODReader: Total number of events in chain= %d \n",nMax);
138 // set number of events in header
139 if (fReaderHeader->GetLastEvent() == -1)
140 fReaderHeader->SetLastEvent(nMax);
142 Int_t nUsr = fReaderHeader->GetLastEvent();
143 fReaderHeader->SetLastEvent(TMath::Min(nMax,nUsr));
147 //____________________________________________________________________________
149 void AliJetAODReader::ConnectTree(TTree* tree, TObject* /*data*/) {
151 // For AOD reader it's needed only to set the number of events
152 fChain = (TChain*) tree;
154 Int_t nMax = fChain->GetEntries();
155 printf("\n AliJetAODReader: Total number of events in chain= %5d \n", nMax);
156 // set number of events in header
157 if (fReaderHeader->GetLastEvent() == -1)
158 fReaderHeader->SetLastEvent(nMax);
160 Int_t nUsr = fReaderHeader->GetLastEvent();
161 fReaderHeader->SetLastEvent(TMath::Min(nMax,nUsr));
165 //____________________________________________________________________________
167 Bool_t AliJetAODReader::FillMomentumArray()
169 // Clear momentum array
172 fDebug = fReaderHeader->GetDebug();
178 // get number of tracks in event (for the loop)
179 Int_t nt = fAOD->GetNTracks();
180 printf("AOD tracks: %5d \t", nt);
182 // temporary storage of signal and pt cut flag
183 Int_t* sflag = new Int_t[nt];
184 Int_t* cflag = new Int_t[nt];
186 // get cuts set by user
187 Float_t ptMin = fReaderHeader->GetPtCut();
188 Float_t etaMin = fReaderHeader->GetFiducialEtaMin();
189 Float_t etaMax = fReaderHeader->GetFiducialEtaMax();
190 UInt_t filterMask = ((AliJetAODReaderHeader*)fReaderHeader)->GetTestFilterMask();
197 for (Int_t it = 0; it < nt; it++) {
198 AliAODTrack *track = fAOD->GetTrack(it);
199 UInt_t status = track->GetStatus();
201 Double_t mom[3] = {track->Px(),track->Py(),track->Pz()};
202 p3.SetXYZ(mom[0],mom[1],mom[2]);
205 if (status == 0) continue;
206 if((filterMask>0)&&!(track->TestFilterBit(filterMask)))continue;
207 if ( (eta > etaMax) || (eta < etaMin)) continue; // checking eta cut
209 new ((*fMomentumArray)[aodTrack]) TLorentzVector(p3,p3.Mag());
210 sflag[aodTrack] = (TMath::Abs(track->GetLabel()) < 10000) ? 1 : 0;
211 cflag[aodTrack] = ( pt > ptMin ) ? 1: 0;
215 printf("Used AOD tracks: %5d \n", aodTrack);
216 // set the signal flags
217 fSignalFlag.Set(aodTrack,sflag);
218 fCutFlag.Set(aodTrack,cflag);
226 //__________________________________________________________
227 void AliJetAODReader::SetApplyMIPCorrection(Bool_t val)
230 // Set flag to apply MIP correction fApplyMIPCorrection
231 // - exclusive with fApplyFractionHadronicCorrection
234 fApplyMIPCorrection = val;
235 if(fApplyMIPCorrection == kTRUE)
237 SetApplyFractionHadronicCorrection(kFALSE);
238 printf("Enabling MIP Correction \n");
242 printf("Disabling MIP Correction \n");
246 //__________________________________________________________
247 void AliJetAODReader::SetApplyFractionHadronicCorrection(Bool_t val)
250 // Set flag to apply EMC hadronic correction fApplyFractionHadronicCorrection
251 // - exclusive with fApplyMIPCorrection
254 fApplyFractionHadronicCorrection = val;
255 if(fApplyFractionHadronicCorrection == kTRUE)
257 SetApplyMIPCorrection(kFALSE);
258 printf("Enabling Fraction Hadronic Correction \n");
262 printf("Disabling Fraction Hadronic Correction \n");
266 //__________________________________________________________
267 void AliJetAODReader::SetFractionHadronicCorrection(Double_t val)
270 // Set value to fFractionHadronicCorrection (default is 0.3)
271 // apply EMC hadronic correction fApplyFractionHadronicCorrection
272 // - exclusive with fApplyMIPCorrection
275 fFractionHadronicCorrection = val;
276 if(fFractionHadronicCorrection > 0.0 && fFractionHadronicCorrection <= 1.0)
278 SetApplyFractionHadronicCorrection(kTRUE);
279 printf("Fraction Hadronic Correction %1.3f \n",fFractionHadronicCorrection);
283 SetApplyFractionHadronicCorrection(kFALSE);
287 //____________________________________________________________________________
288 void AliJetAODReader::CreateTasks(TChain* tree)
290 fDebug = fReaderHeader->GetDebug();
291 fDZ = fReaderHeader->GetDZ();
294 // Init EMCAL geometry and create UnitArray object
296 // cout << "In create task" << endl;
300 fFillUnitArray = new TTask("fFillUnitArray","Fill unit array jet finder");
301 fFillUAFromTracks = new AliJetAODFillUnitArrayTracks();
302 fFillUAFromTracks->SetReaderHeader(fReaderHeader);
303 fFillUAFromTracks->SetGeom(fGeom);
304 fFillUAFromTracks->SetTPCGrid(fTpcGrid);
305 fFillUAFromTracks->SetEMCalGrid(fEmcalGrid);
309 fFillUAFromTracks->SetGrid0(fGrid0);
310 fFillUAFromTracks->SetGrid1(fGrid1);
311 fFillUAFromTracks->SetGrid2(fGrid2);
312 fFillUAFromTracks->SetGrid3(fGrid3);
313 fFillUAFromTracks->SetGrid4(fGrid4);
315 fFillUAFromTracks->SetApplyMIPCorrection(fApplyMIPCorrection);
316 fFillUAFromTracks->SetHadCorrector(fHadCorr);
317 fFillUAFromEMCalDigits = new AliJetAODFillUnitArrayEMCalDigits();
318 fFillUAFromEMCalDigits->SetReaderHeader(fReaderHeader);
319 fFillUAFromEMCalDigits->SetGeom(fGeom);
320 fFillUAFromEMCalDigits->SetTPCGrid(fTpcGrid);
321 fFillUAFromEMCalDigits->SetEMCalGrid(fEmcalGrid);
322 fFillUAFromEMCalDigits->SetApplyFractionHadronicCorrection(fApplyFractionHadronicCorrection);
323 fFillUAFromEMCalDigits->SetFractionHadronicCorrection(fFractionHadronicCorrection);
324 fFillUAFromEMCalDigits->SetApplyElectronCorrection(fApplyElectronCorrection);
326 fFillUnitArray->Add(fFillUAFromTracks);
327 fFillUnitArray->Add(fFillUAFromEMCalDigits);
328 fFillUAFromTracks->SetActive(kFALSE);
329 fFillUAFromEMCalDigits->SetActive(kFALSE);
331 cout << "Tasks instantiated at that stage ! " << endl;
332 cout << "You can loop over events now ! " << endl;
336 //____________________________________________________________________________
337 Bool_t AliJetAODReader::ExecTasks(Bool_t procid, TRefArray* refArray)
341 // Fill the reader part
345 fRefArray = refArray;
346 //(not used ?) Int_t nEntRef = fRefArray->GetEntries();
347 //(not used ?) Int_t nEntUnit = fUnitArray->GetEntries();
349 // clear momentum array
352 fDebug = fReaderHeader->GetDebug();
353 fOpt = fReaderHeader->GetDetector();
359 // TPC only or Digits+TPC or Clusters+TPC
360 if(fOpt%2==!0 && fOpt!=0){
361 fFillUAFromTracks->SetAOD(fAOD);
362 fFillUAFromTracks->SetActive(kTRUE);
363 fFillUAFromTracks->SetUnitArray(fUnitArray);
364 fFillUAFromTracks->SetRefArray(fRefArray);
365 fFillUAFromTracks->SetProcId(fProcId);
366 // fFillUAFromTracks->ExecuteTask("tpc"); // => Temporarily changed
367 fFillUAFromTracks->Exec("tpc");
369 fNumCandidate = fFillUAFromTracks->GetMult();
370 fNumCandidateCut = fFillUAFromTracks->GetMultCut();
374 // Digits only or Digits+TPC
375 if(fOpt>=2 && fOpt<=3){
376 fFillUAFromEMCalDigits->SetAOD(fAOD);
377 fFillUAFromEMCalDigits->SetActive(kTRUE);
378 fFillUAFromEMCalDigits->SetUnitArray(fUnitArray);
379 fFillUAFromEMCalDigits->SetRefArray(fRefArray);
380 fFillUAFromEMCalDigits->SetProcId(fFillUAFromTracks->GetProcId());
381 fFillUAFromEMCalDigits->SetInitMult(fFillUAFromTracks->GetMult());
382 fFillUAFromEMCalDigits->SetInitMultCut(fFillUAFromTracks->GetMultCut());
383 fFillUAFromEMCalDigits->Exec("digits"); // => Temporarily added
384 fNumCandidate = fFillUAFromEMCalDigits->GetMult();
385 fNumCandidateCut = fFillUAFromEMCalDigits->GetMultCut();
388 // fFillUnitArray->ExecuteTask(); // => Temporarily commented
393 //____________________________________________________________________________
394 Bool_t AliJetAODReader::SetEMCALGeometry()
397 // Set the EMCal Geometry
400 if (!fTree->GetFile())
403 TString geomFile(fTree->GetFile()->GetName());
404 geomFile.ReplaceAll("AliESDs", "geometry");
406 // temporary workaround for PROOF bug #18505
407 geomFile.ReplaceAll("#geometry.root#geometry.root", "#geometry.root");
408 if(fDebug>1) printf("Current geometry file %s \n", geomFile.Data());
410 // Define EMCAL geometry to be able to read ESDs
411 fGeom = AliJetDummyGeo::GetInstance();
413 fGeom = AliJetDummyGeo::GetInstance("EMCAL_COMPLETE","EMCAL");
415 // To be setted to run some AliEMCALGeometry functions
416 TGeoManager::Import(geomFile);
417 fGeom->GetTransformationForSM();
418 printf("\n EMCal Geometry set ! \n");
424 //____________________________________________________________________________
425 void AliJetAODReader::InitParameters()
427 // Initialise parameters
428 fOpt = fReaderHeader->GetDetector();
429 // fHCorrection = 0; // For hadron correction
430 fHadCorr = 0; // For hadron correction
432 if(fOpt==0 || fOpt==1)
433 fECorrection = 0; // For electron correction
434 else fECorrection = 1; // For electron correction
436 fNumUnits = fGeom->GetNCells(); // Number of cells in EMCAL
437 if(fDebug>1) printf("\n EMCal parameters initiated ! \n");
440 //____________________________________________________________________________
441 void AliJetAODReader::InitUnitArray()
443 //Initialises unit arrays
444 Int_t nElements = fTpcGrid->GetNEntries();
445 Float_t eta = 0., phi = 0., deltaEta = 0., deltaPhi = 0.;
446 if(fArrayInitialised) fUnitArray->Delete();
448 if(fTpcGrid->GetGridType()==0)
449 { // Fill the following quantities :
450 // Good track ID, (Eta,Phi) position ID, eta, phi, energy, px, py, pz, deltaEta, deltaPhi,
451 // detector flag, in/out jet, pt cut, mass, cluster ID)
452 for(Int_t nBin = 1; nBin < nElements+1; nBin++)
454 // fTpcGrid->GetEtaPhiFromIndex2(nBin,eta,phi);
455 fTpcGrid->GetEtaPhiFromIndex2(nBin,phi,eta);
456 phi = ((phi < 0) ? phi + 2. * TMath::Pi() : phi);
457 deltaEta = fTpcGrid->GetDeta();
458 deltaPhi = fTpcGrid->GetDphi();
459 new ((*fUnitArray)[nBin-1]) AliJetUnitArray(nBin-1,0,eta,phi,0.,deltaEta,deltaPhi,kTpc,kOutJet,kPtSmaller,kPtSmaller,kBad,0.,-1);
463 if(fTpcGrid->GetGridType()==1)
466 Int_t n0 = 0, n1 = 0, n2 = 0, n3 = 0, n4 = 0;
470 // Define a grid of cell for the gaps between SM
471 Double_t phimin0 = 0., phimin1 = 0., phimin2 = 0., phimin3 = 0., phimin4 = 0.;
472 Double_t phimax0 = 0., phimax1 = 0., phimax2 = 0., phimax3 = 0., phimax4 = 0.;
473 fGeom->GetPhiBoundariesOfSMGap(0,phimin0,phimax0);
474 fGrid0 = new AliJetGrid(0,95,phimin0,phimax0,-0.7,0.7); // 0.015 x 0.015
475 fGrid0->SetGridType(0);
476 fGrid0->SetMatrixIndexes();
477 fGrid0->SetIndexIJ();
478 n0 = fGrid0->GetNEntries();
479 fGeom->GetPhiBoundariesOfSMGap(1,phimin1,phimax1);
480 fGrid1 = new AliJetGrid(0,95,phimin1,phimax1,-0.7,0.7); // 0.015 x 0.015
481 fGrid1->SetGridType(0);
482 fGrid1->SetMatrixIndexes();
483 fGrid1->SetIndexIJ();
484 n1 = fGrid1->GetNEntries();
485 fGeom->GetPhiBoundariesOfSMGap(2,phimin2,phimax2);
486 fGrid2 = new AliJetGrid(0,95,phimin2,phimax2,-0.7,0.7); // 0.015 x 0.015
487 fGrid2->SetGridType(0);
488 fGrid2->SetMatrixIndexes();
489 fGrid2->SetIndexIJ();
490 n2 = fGrid2->GetNEntries();
491 fGeom->GetPhiBoundariesOfSMGap(3,phimin3,phimax3);
492 fGrid3 = new AliJetGrid(0,95,phimin3,phimax3,-0.7,0.7); // 0.015 x 0.015
493 fGrid3->SetGridType(0);
494 fGrid3->SetMatrixIndexes();
495 fGrid3->SetIndexIJ();
496 n3 = fGrid3->GetNEntries();
497 fGeom->GetPhiBoundariesOfSMGap(4,phimin4,phimax4);
498 fGrid4 = new AliJetGrid(0,95,phimin4,phimax4,-0.7,0.7); // 0.015 x 0.015
499 fGrid4->SetGridType(0);
500 fGrid4->SetMatrixIndexes();
501 fGrid4->SetIndexIJ();
502 n4 = fGrid4->GetNEntries();
504 nGaps = n0+n1+n2+n3+n4;
508 for(Int_t nBin = 0; nBin < fNumUnits+nElements+nGaps; nBin++)
512 fGeom->EtaPhiFromIndex(nBin, eta, phi); // From EMCal geometry
513 // fEmcalGrid->GetEtaPhiFromIndex2(nBin,phi,eta); // My function from Grid
514 phi = ((phi < 0) ? phi + 2. * TMath::Pi() : phi);
515 deltaEta = fEmcalGrid->GetDeta(); // Modify with the exact detector values
516 deltaPhi = fEmcalGrid->GetDphi(); // Modify with the exact detector values
517 new ((*fUnitArray)[nBin]) AliJetUnitArray(nBin,0,eta,phi,0.,deltaEta,deltaPhi,kTpc,kOutJet,kPtSmaller,kPtSmaller,kBad,0.,-1);
520 if(nBin>=fNumUnits && nBin<fNumUnits+nElements){
521 fTpcGrid->GetEtaPhiFromIndex2(nBin+1-fNumUnits,phi,eta);
522 phi = ((phi < 0) ? phi + 2. * TMath::Pi() : phi);
523 deltaEta = fTpcGrid->GetDeta();
524 deltaPhi = fTpcGrid->GetDphi();
525 new ((*fUnitArray)[nBin]) AliJetUnitArray(nBin,0,eta,phi,0.,deltaEta,deltaPhi,kTpc,kOutJet,kPtSmaller,kPtSmaller,kBad,0.,-1);
529 if(nBin>=fNumUnits+nElements && nBin<fNumUnits+nElements+nGaps){
530 if(nBin<fNumUnits+nElements+n0)
533 fGrid0->GetEtaPhiFromIndex2(nBin+1-(fNumUnits+nElements),phi,eta);
534 deltaEta = fGrid0->GetDeta();
535 deltaPhi = fGrid0->GetDphi();
536 new ((*fUnitArray)[nBin]) AliJetUnitArray(nBin,0,eta,phi,0.,deltaEta,deltaPhi,kTpc,kOutJet,kPtSmaller,kPtSmaller,kBad,0.,-1);
538 else if(nBin>=fNumUnits+nElements+n0 && nBin<fNumUnits+nElements+n0+n1)
541 fGrid1->GetEtaPhiFromIndex2(nBin+1-(fNumUnits+nElements+n0),phi,eta);
542 deltaEta = fGrid1->GetDeta();
543 deltaPhi = fGrid1->GetDphi();
544 new ((*fUnitArray)[nBin]) AliJetUnitArray(nBin,0,eta,phi,0.,deltaEta,deltaPhi,kTpc,kOutJet,kPtSmaller,kPtSmaller,kBad,0.,-1);
546 else if(nBin>=fNumUnits+nElements+n0+n1 && nBin<fNumUnits+nElements+n0+n1+n2)
549 fGrid2->GetEtaPhiFromIndex2(nBin+1-(fNumUnits+nElements+n0+n1),phi,eta);
550 deltaEta = fGrid2->GetDeta();
551 deltaPhi = fGrid2->GetDphi();
552 new ((*fUnitArray)[nBin]) AliJetUnitArray(nBin,0,eta,phi,0.,deltaEta,deltaPhi,kTpc,kOutJet,kPtSmaller,kPtSmaller,kBad,0.,-1);
554 else if(nBin>=fNumUnits+nElements+n0+n1+n2 && nBin<fNumUnits+nElements+n0+n1+n2+n3)
557 fGrid3->GetEtaPhiFromIndex2(nBin+1-(fNumUnits+nElements+n0+n1+n2),phi,eta);
558 deltaEta = fGrid3->GetDeta();
559 deltaPhi = fGrid3->GetDphi();
560 new ((*fUnitArray)[nBin]) AliJetUnitArray(nBin,0,eta,phi,0.,deltaEta,deltaPhi,kTpc,kOutJet,kPtSmaller,kPtSmaller,kBad,0.,-1);
562 else if(nBin>=fNumUnits+nElements+n0+n1+n2+n3 && nBin<fNumUnits+nElements+nGaps)
565 fGrid4->GetEtaPhiFromIndex2(nBin+1-(fNumUnits+nElements+n0+n1+n2+n3),phi,eta);
566 deltaEta = fGrid4->GetDeta();
567 deltaPhi = fGrid4->GetDphi();
568 new ((*fUnitArray)[nBin]) AliJetUnitArray(nBin,0,eta,phi,0.,deltaEta,deltaPhi,kTpc,kOutJet,kPtSmaller,kPtSmaller,kBad,0.,-1);
574 } // end loop on nBin
575 } // end grid type == 1
576 fArrayInitialised = 1;