]>
Commit | Line | Data |
---|---|---|
4a01bb2c | 1 | #ifndef ALIJETFILLUNITARRAYTRACKS_H |
2 | #define ALIJETFILLUNITARRAYTRACKS_H | |
3 | ||
4 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
5 | * See cxx source for full Copyright notice */ | |
6 | ||
7 | //--------------------------------------------------------------------- | |
8 | // Jet Fill Unit Array | |
9 | // Called by ESD Reader for jet analysis | |
10 | // Author: Magali Estienne (magali.estienne@subatech.in2p3.fr) | |
11 | //--------------------------------------------------------------------- | |
12 | ||
13 | #ifndef ROOT_TTask | |
14 | #include "TTask.h" | |
15 | #endif | |
16 | ||
17 | #include <TMatrixD.h> | |
18 | #include <TArrayD.h> | |
19 | ||
4a01bb2c | 20 | class AliJetHadronCorrection; |
21 | class AliJetReader; | |
22 | class AliJetESDReader; | |
23 | class TClonesArray; | |
ee7de0dd | 24 | class TRefArray; |
25 | //class AliJetUnitArray; | |
4a01bb2c | 26 | //class AliJetReaderHeader; |
4a01bb2c | 27 | class AliJetGrid; |
ee7de0dd | 28 | class AliJetDummyGeo; |
29 | class AliESD; | |
30 | class AliESDEvent; | |
4a01bb2c | 31 | |
32 | class AliJetFillUnitArrayTracks : public TTask | |
33 | { | |
34 | public: | |
35 | AliJetFillUnitArrayTracks(); | |
ee7de0dd | 36 | AliJetFillUnitArrayTracks(AliESD *fESD); |
37 | AliJetFillUnitArrayTracks(AliESDEvent *fESD); | |
4a01bb2c | 38 | virtual ~AliJetFillUnitArrayTracks(); |
39 | ||
40 | // Setter | |
41 | void SetReaderHeader(AliJetReaderHeader *readerHeader) {fReaderHeader = readerHeader;} | |
ee7de0dd | 42 | void SetGeom(AliJetDummyGeo *geom) {fGeom = geom;} |
4a01bb2c | 43 | void SetMomentumArray(TClonesArray *momentumArray) {fMomentumArray = momentumArray;} |
ee7de0dd | 44 | void SetUnitArray(TClonesArray *unitArray) {fUnitArray = unitArray;} |
45 | void SetRefArray(TRefArray *refArray) {fRefArray = refArray;} | |
4a01bb2c | 46 | void SetHadCorrection(Int_t flag = 1) {fHCorrection = flag;} |
47 | void SetHadCorrector(AliJetHadronCorrectionv1* corr) {fHadCorr = corr;} | |
48 | void SetTPCGrid(AliJetGrid *grid) {fTPCGrid = grid;} | |
49 | void SetEMCalGrid(AliJetGrid *grid) {fEMCalGrid = grid;} | |
50 | void SetGrid(Double_t phiMin,Double_t phiMax,Double_t etaMin,Double_t etaMax); | |
ee7de0dd | 51 | // void SetESD(AliESD *esd) {fESD = esd;} |
52 | void SetESD(AliESDEvent *esd) {fESD = esd;} | |
53 | void SetGrid0(AliJetGrid *grid0){fGrid0 = grid0;} | |
54 | void SetGrid1(AliJetGrid *grid1){fGrid1 = grid1;} | |
55 | void SetGrid2(AliJetGrid *grid2){fGrid2 = grid2;} | |
56 | void SetGrid3(AliJetGrid *grid3){fGrid3 = grid3;} | |
57 | void SetGrid4(AliJetGrid *grid4){fGrid4 = grid4;} | |
4a01bb2c | 58 | |
59 | // Getter | |
ee7de0dd | 60 | TClonesArray* GetUnitArray() {return fUnitArray;} |
61 | TRefArray* GetRefArray() {return fRefArray;} | |
62 | // Int_t GetIndexFromEtaPhi(Double_t eta,Double_t phi) const; | |
63 | void GetEtaPhiFromIndex(Int_t index,Float_t &eta,Float_t &phi); | |
64 | Int_t GetNeta() {return fNeta;} | |
65 | Int_t GetNphi() {return fNphi;} | |
66 | Int_t GetHadCorrection() {return fHCorrection;} | |
67 | Int_t GetMult() {return fNTracks;} | |
68 | Int_t GetMultCut() {return fNTracksCut;} | |
69 | void Exec(Option_t*); | |
4a01bb2c | 70 | |
71 | protected: | |
72 | Int_t fNumUnits; // Number of units in the unit object array (same as num towers in EMCAL) | |
73 | Float_t fEtaMinCal; // Define EMCal acceptance in Eta | |
74 | Float_t fEtaMaxCal; // Define EMCal acceptance in Eta | |
75 | Float_t fPhiMinCal; // Define EMCal acceptance in Phi | |
76 | Float_t fPhiMaxCal; // Define EMCal acceptance in Phi | |
77 | AliJetHadronCorrectionv1 *fHadCorr; // Pointer to Hadron Correction Object | |
ee7de0dd | 78 | Int_t fHCorrection; // Hadron correction flag |
79 | Int_t fNTracks; // Number of tracks stored in UnitArray | |
80 | Int_t fNTracksCut; // Number of tracks stored in UnitArray with a pt cut | |
4a01bb2c | 81 | Int_t fOpt; // Detector to be used for jet reconstruction |
ee7de0dd | 82 | Bool_t fDZ; // Use or not dead zones |
4a01bb2c | 83 | Int_t fDebug; // Debug option |
84 | ||
85 | AliJetReaderHeader *fReaderHeader; // ReaderHeader | |
86 | TClonesArray *fMomentumArray; // MomentumArray | |
ee7de0dd | 87 | TClonesArray *fUnitArray; // UnitArray |
88 | TRefArray *fRefArray; // UnitArray | |
4a01bb2c | 89 | AliJetGrid *fTPCGrid; // Define filled grid |
90 | AliJetGrid *fEMCalGrid; // Define filled grid | |
ee7de0dd | 91 | AliJetDummyGeo *fGeom; // Define EMCal geometry |
92 | // AliESD *fESD; // ESD | |
93 | AliESDEvent *fESD; // ESD | |
94 | AliJetGrid *fGrid0; | |
95 | AliJetGrid *fGrid1; | |
96 | AliJetGrid *fGrid2; | |
97 | AliJetGrid *fGrid3; | |
98 | AliJetGrid *fGrid4; | |
4a01bb2c | 99 | |
100 | Int_t fNphi; // number of points in the grid: phi | |
101 | Int_t fNeta; // " eta | |
102 | TArrayD* fPhi2; // grid points in phi | |
103 | TArrayD* fEta2; // grid points in eta | |
104 | TArrayD* fPhi; // grid points in phi | |
105 | TArrayD* fEta; // grid points in eta | |
106 | TMatrixD* fIndex; // grid points in (phi,eta) | |
107 | TMatrixD* fParams; // matrix of parameters in the grid points | |
108 | Int_t fGrid; // Select the grid acceptance you want to fill | |
109 | // 0 = TPC acceptance, 1 = TPC-EMCal acceptance | |
110 | Float_t fPhiMin; | |
111 | Float_t fPhiMax; | |
112 | Float_t fEtaMin; | |
113 | Float_t fEtaMax; | |
114 | Int_t fEtaBinInTPCAcc; | |
115 | Int_t fPhiBinInTPCAcc; | |
116 | Int_t fEtaBinInEMCalAcc; | |
117 | Int_t fPhiBinInEMCalAcc; | |
118 | Int_t fNbinPhi; | |
119 | ||
120 | private: | |
ee7de0dd | 121 | // void SetEMCALGeometry(); |
4a01bb2c | 122 | void InitParameters(); |
123 | ||
4a01bb2c | 124 | ClassDef(AliJetFillUnitArrayTracks,1) // Fill Unit Array with tpc and/or emcal information |
125 | }; | |
126 | ||
127 | #endif |