]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWG4/JetTasks/AliAnalysisTaskJetSpectrum2.h
preps for using the reactions plane
[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliAnalysisTaskJetSpectrum2.h
CommitLineData
3b7ffecf 1#ifndef ALIANALYSISTASKJETSPECTRUM2_H
2#define ALIANALYSISTASKJETSPECTRUM2_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// Task used for the correction of determiantion of reconstructed jet spectra
9// Compares input (gen) and output (rec) jets
10// *******************************************
11
12#include "AliAnalysisTaskSE.h"
13#include "THnSparse.h" // cannot forward declare ThnSparseF
14
15////////////////
16class AliJetHeader;
17class AliESDEvent;
18class AliAODEvent;
a31b8a87 19class AliAODExtension;
3b7ffecf 20class AliAODJet;
d7117cbd 21class AliVParticle;
c2785065 22class AliAODJetEventBackground;
3b7ffecf 23class AliGenPythiaEventHeader;
24class AliCFManager;
25
26class TList;
27class TChain;
c2785065 28class TH1F;
3b7ffecf 29class TH2F;
30class TH3F;
31class TProfile;
b1cd90b5 32class TSTring;
3b7ffecf 33
34
35class AliAnalysisTaskJetSpectrum2 : public AliAnalysisTaskSE
36{
37 public:
38 AliAnalysisTaskJetSpectrum2();
39 AliAnalysisTaskJetSpectrum2(const char* name);
40 virtual ~AliAnalysisTaskJetSpectrum2() {;}
41 // Implementation of interface methods
42 virtual void UserCreateOutputObjects();
43 virtual void Init();
44 virtual void LocalInit() { Init(); }
45 virtual void UserExec(Option_t *option);
46 virtual void Terminate(Option_t *option);
47 virtual Bool_t Notify();
48
b5cc0c6d 49 virtual void SetUseGlobalSelection(Bool_t b){fUseGlobalSelection = b;}
f4132e7d 50 virtual void SetEventClass(Int_t i){fEventClass = i;}
3b7ffecf 51 virtual void SetExternalWeight(Float_t f){fExternalWeight = f;}
52 virtual void SetUseExternalWeightOnly(Bool_t b){fUseExternalWeightOnly = b;}
565584e8 53 virtual void SetAODJetInput(Bool_t b){fUseAODJetInput = b;}
54 virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
55 virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
3b7ffecf 56 virtual void SetLimitGenJetEta(Bool_t b){fLimitGenJetEta = b;}
a31b8a87 57 virtual void SetJetEtaWindow(Float_t f){fJetRecEtaWindow = f;}
58 virtual void SetTrackEtaWindow(Float_t f){fTrackRecEtaWindow = f;}
59 virtual void SetNMatchJets(Short_t f){fNMatchJets = f;}
9280dfa6 60 virtual void SetMinJetPt(Float_t f){fMinJetPt = f;}
7b822bc5 61 virtual void SetNRPBins(Short_t i){fNRPBins = i;}
a31b8a87 62 virtual void SetMinTrackPt(Float_t f){fMinTrackPt = f;}
9280dfa6 63 virtual void SetDeltaPhiWindow(Float_t f){fDeltaPhiWindow = f;}
3b7ffecf 64 virtual void SetAnalysisType(Int_t i){fAnalysisType = i;}
65 virtual void SetBranchGen(const char* c){fBranchGen = c;}
66 virtual void SetBranchRec(const char* c){fBranchRec = c;}
37eb26ea 67 virtual void SetBranchBkgRec(const char* c){fBranchBkgRec = c;}
68 virtual void SetBranchBkgGen(const char* c){fBranchBkgGen = c;}
3b7ffecf 69 virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
70 virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
71 virtual void SetFilterMask(UInt_t i){fFilterMask = i;}
f2dd0695 72 virtual void SetEventSelectionMask(UInt_t i){fEventSelectionMask = i;}
a31b8a87 73 virtual void SetNonStdFile(char* c){fNonStdFile = c;}
3b7ffecf 74
75
76 // Helper
77 //
78
79 // we have different cases
80 // AOD reading -> MC from AOD
81 // ESD reading -> MC from Kinematics
82 // this has to match with our selection of input events
565584e8 83 enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
3b7ffecf 84 enum {kAnaMC = 0x1, kAnaMCESD = 0x2};
37eb26ea 85 enum {kMaxJets = 3};
a2522483 86 enum {kJetRec = 0, kJetGen, kJetRecFull, kJetGenFull, kJetTypes}; //
3b7ffecf 87 enum {kMaxCorrelation = 3};
88
89 //
90 // 0 all jets
91 // 1 all jet in eta window
92 // 2 all jets with partner
93 // 3 all jets in eta window with partner
94 // 4 all jets with partner in eta window
95 enum {kStep0 = 0, kStep1, kStep2, kStep3, kStep4,kMaxStep};
96
97
98 private:
99
100 AliAnalysisTaskJetSpectrum2(const AliAnalysisTaskJetSpectrum2&);
101 AliAnalysisTaskJetSpectrum2& operator=(const AliAnalysisTaskJetSpectrum2&);
102
a31b8a87 103 void MakeJetContainer();
104 Int_t GetListOfTracks(TList *list,Int_t type);
105 void FillTrackHistos(TList &particlesList,int iType);
3b7ffecf 106
a31b8a87 107 Int_t GetListOfJets(TList *list,TClonesArray* jarray,Int_t type);
108 void FillJetHistos(TList &jetsList,TList &particlesList,Int_t iType);
109
110 void FillMatchHistos(TList &recJetsList,TList &genJetsList);
111
112 Bool_t JetSelected(AliAODJet *jet);
37eb26ea 113 Int_t MultFromJetRefs(TClonesArray *jets);
d7117cbd 114 AliVParticle *LeadingTrackFromJetRefs(AliAODJet* jet);
115 AliVParticle *LeadingTrackInCone(AliAODJet* jet,TList *list,Float_t r = 0.4);
116
37eb26ea 117
3493c3a9 118 AliJetHeader *fJetHeaderRec;//! The jet header that can be fetched from the userinfo
119 AliJetHeader *fJetHeaderGen;//! The jet header that can fetched from the userinfo
a31b8a87 120 AliAODEvent *fAODIn; //! where we take the jets from
121 AliAODEvent *fAODOut; //! where we take the jets from
122 AliAODExtension *fAODExtension; //! where we take the jets from can be input or output AOD
3493c3a9 123 THnSparseF *fhnJetContainer[kMaxStep*2]; //! like particle container in corrfw with different steps need AliCFContainer with Scale(), and clone() to do the same
124 THnSparseF *fhnCorrelation; //! response matrix for unfolding
125 THnSparseF *fhnCorrelationPhiZRec; //! response matrix for unfolding in max Z rec bins
3b7ffecf 126
3493c3a9 127 TF1 *f1PtScale; //! correction function to correct to the average true jet energy depending on p_T,rec
3b7ffecf 128
129 TString fBranchRec; // AOD branch name for reconstructed
130 TString fBranchGen; // AOD brnach for genereated
37eb26ea 131 TString fBranchBkgRec; //AOD branch for background
132 TString fBranchBkgGen; //AOD branch for background
a31b8a87 133 TString fNonStdFile; // name of delta aod file to catch the extension
3b7ffecf 134
565584e8 135 Bool_t fUseAODJetInput; // take jet from input AOD not from ouptu AOD
136 Bool_t fUseAODTrackInput; // take track from input AOD not from ouptu AOD
137 Bool_t fUseAODMCInput; // take MC from input AOD not from ouptu AOD
b5cc0c6d 138 Bool_t fUseGlobalSelection; // Limit the eta of the generated jets
3b7ffecf 139 Bool_t fUseExternalWeightOnly; // use only external weight
140 Bool_t fLimitGenJetEta; // Limit the eta of the generated jets
a31b8a87 141 Short_t fNMatchJets; // number of leading jets considered from the list
7b822bc5 142 Short_t fNRPBins; // number of bins with respect to RP
f2dd0695 143 UInt_t fFilterMask; // filter bit for slecected tracks
144 UInt_t fEventSelectionMask; // Selection information used to filter events
3b7ffecf 145 Int_t fAnalysisType; // Analysis type
146 Int_t fTrackTypeRec; // type of tracks used for FF
147 Int_t fTrackTypeGen; // type of tracks used for FF
f4132e7d 148 Int_t fEventClass; // event class to be looked at for this instance of the task
3b7ffecf 149 Float_t fAvgTrials; // Average nimber of trials
150 Float_t fExternalWeight; // external weight
a31b8a87 151 Float_t fJetRecEtaWindow; // eta window for rec jets
152 Float_t fTrackRecEtaWindow; // eta window for rec tracks
9280dfa6 153 Float_t fMinJetPt; // limits the jet p_T in addition to what already is done in the jet finder, this is important for jet matching for JF with lo threshold
a31b8a87 154 Float_t fMinTrackPt; // limits the track p_T
26fa06fb 155 Float_t fDeltaPhiWindow; // minium angle between dijets
7b822bc5 156 Float_t fRPAngle; // ! angle of the reaction plane
37eb26ea 157 Int_t fMultRec; // ! reconstructed track multiplicity
158 Int_t fMultGen; // ! generated track multiplicity
7b822bc5 159
3b7ffecf 160
3493c3a9 161 TProfile* fh1Xsec; //! pythia cross section and trials
162 TH1F* fh1Trials; //! trials are added
163 TH1F* fh1PtHard; //! Pt har of the event...
164 TH1F* fh1PtHardNoW; //! Pt har of the event without weigt
165 TH1F* fh1PtHardTrials; //! Number of trials
57ca1193 166 TH1F* fh1ZVtx; //! z-vtx distribution
7b822bc5 167 TH1F* fh1RP; //! RP distribution
37eb26ea 168 TH1F* fh1TmpRho; //! just temporary histo for calculation
169 TH2F* fh2MultRec; //! reconstructed track multiplicity
170 TH2F* fh2MultGen; //! generated track multiplicity
171
a31b8a87 172 TH2F* fh2PtFGen; //! found vs generated
3493c3a9 173 TH2F* fh2RelPtFGen; //! relative difference between generated and found
37eb26ea 174
9a83d4af 175
a31b8a87 176 // Jet histos second go
177
178 TH1F* fh1NJets[kJetTypes]; //! nr of gen jets
179 TH1F* fh1SumPtTrack[kJetTypes]; //! sum over all track pT
180
37eb26ea 181 TH1F* fh1PtIn[kJetTypes][kMaxJets+1]; //! Jet pt
a31b8a87 182 TH1F* fh1PtJetsIn[kJetTypes]; //! Jet pt for all jets
183 TH1F* fh1PtTracksIn[kJetTypes]; //! track pt for all tracks
cb883243 184 TH1F* fh1PtTracksInLow[kJetTypes]; //! track pt for all tracks
a31b8a87 185 TH1F* fh1PtTracksLeadingIn[kJetTypes]; //! track pt for all tracks
186
187 TH2F* fh2NJetsPt[kJetTypes]; //! Number of found jets above threshold
a2522483 188 TH2F* fh2NTracksPt[kJetTypes]; //! Number of tracks above threshold
a31b8a87 189 TH2F* fh2LeadingTrackPtTrackPhi[kJetTypes]; //! phi distribution of accepted leading tracks
37eb26ea 190 TH2F* fh2RhoPt[kJetTypes][kMaxJets+1]; //! jet shape variable rho
191 TH2F* fh2PsiPt[kJetTypes][kMaxJets+1]; //! jet shape variable psi
7b822bc5 192
37eb26ea 193 TH2F* fh2PhiPt[kJetTypes][kMaxJets+1]; //! phi of jets
194 TH2F* fh2EtaPt[kJetTypes][kMaxJets+1]; //! eta of jets
195 TH2F* fh2AreaPt[kJetTypes][kMaxJets+1]; //! area distribution
196 TH2F* fh2EtaArea[kJetTypes][kMaxJets+1]; //! area vs eta distribution
197 TH2F* fh2PhiEta[kJetTypes][kMaxJets+1]; //! eta phi distribution of jet
d7117cbd 198 TH2F* fh2LTrackPtJetPt[kJetTypes][kMaxJets+1]; //! leading track within the jet vs jet pt
a31b8a87 199
7b822bc5 200
201 TH3F* fh3MultPtRP[kJetTypes][kMaxJets+1]; //! RP vs. mult of jets
202 TH3F* fh3MultTrackPtRP[kJetTypes]; //! mult vs pt
203
204
a31b8a87 205 TH1F* fh1DijetMinv[kJetTypes]; //! dijet inv mass
206 TH2F* fh2DijetDeltaPhiPt[kJetTypes]; //! dijet delta phi vs pt
207 TH2F* fh2DijetAsymPt[kJetTypes]; //! dijet asym vs pt after delta phi cut
208 TH2F* fh2DijetPt2vsPt1[kJetTypes]; //! dijet pt2 vs pt1
209 TH2F* fh2DijetDifvsSum[kJetTypes]; //! dijet dif vs sum
210
7b822bc5 211
f4132e7d 212 TList *fHistList; //! Output list
3b7ffecf 213
214
d7117cbd 215 ClassDef(AliAnalysisTaskJetSpectrum2, 13) // Analysis task for standard jet analysis
3b7ffecf 216};
217
218#endif