]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWG4/JetTasks/AliAnalysisTaskJetSpectrum2.h
Coverity fix
[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;}
a31b8a87 61 virtual void SetMinTrackPt(Float_t f){fMinTrackPt = f;}
9280dfa6 62 virtual void SetDeltaPhiWindow(Float_t f){fDeltaPhiWindow = f;}
3b7ffecf 63 virtual void SetAnalysisType(Int_t i){fAnalysisType = i;}
64 virtual void SetBranchGen(const char* c){fBranchGen = c;}
65 virtual void SetBranchRec(const char* c){fBranchRec = c;}
37eb26ea 66 virtual void SetBranchBkgRec(const char* c){fBranchBkgRec = c;}
67 virtual void SetBranchBkgGen(const char* c){fBranchBkgGen = c;}
3b7ffecf 68 virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
69 virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
70 virtual void SetFilterMask(UInt_t i){fFilterMask = i;}
f2dd0695 71 virtual void SetEventSelectionMask(UInt_t i){fEventSelectionMask = i;}
a31b8a87 72 virtual void SetNonStdFile(char* c){fNonStdFile = c;}
3b7ffecf 73
74
75 // Helper
76 //
77
78 // we have different cases
79 // AOD reading -> MC from AOD
80 // ESD reading -> MC from Kinematics
81 // this has to match with our selection of input events
565584e8 82 enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
3b7ffecf 83 enum {kAnaMC = 0x1, kAnaMCESD = 0x2};
37eb26ea 84 enum {kMaxJets = 3};
a2522483 85 enum {kJetRec = 0, kJetGen, kJetRecFull, kJetGenFull, kJetTypes}; //
3b7ffecf 86 enum {kMaxCorrelation = 3};
87
88 //
89 // 0 all jets
90 // 1 all jet in eta window
91 // 2 all jets with partner
92 // 3 all jets in eta window with partner
93 // 4 all jets with partner in eta window
94 enum {kStep0 = 0, kStep1, kStep2, kStep3, kStep4,kMaxStep};
95
96
97 private:
98
99 AliAnalysisTaskJetSpectrum2(const AliAnalysisTaskJetSpectrum2&);
100 AliAnalysisTaskJetSpectrum2& operator=(const AliAnalysisTaskJetSpectrum2&);
101
a31b8a87 102 void MakeJetContainer();
103 Int_t GetListOfTracks(TList *list,Int_t type);
104 void FillTrackHistos(TList &particlesList,int iType);
3b7ffecf 105
a31b8a87 106 Int_t GetListOfJets(TList *list,TClonesArray* jarray,Int_t type);
107 void FillJetHistos(TList &jetsList,TList &particlesList,Int_t iType);
108
109 void FillMatchHistos(TList &recJetsList,TList &genJetsList);
110
111 Bool_t JetSelected(AliAODJet *jet);
37eb26ea 112 Int_t MultFromJetRefs(TClonesArray *jets);
d7117cbd 113 AliVParticle *LeadingTrackFromJetRefs(AliAODJet* jet);
114 AliVParticle *LeadingTrackInCone(AliAODJet* jet,TList *list,Float_t r = 0.4);
115
37eb26ea 116
3493c3a9 117 AliJetHeader *fJetHeaderRec;//! The jet header that can be fetched from the userinfo
118 AliJetHeader *fJetHeaderGen;//! The jet header that can fetched from the userinfo
a31b8a87 119 AliAODEvent *fAODIn; //! where we take the jets from
120 AliAODEvent *fAODOut; //! where we take the jets from
121 AliAODExtension *fAODExtension; //! where we take the jets from can be input or output AOD
3493c3a9 122 THnSparseF *fhnJetContainer[kMaxStep*2]; //! like particle container in corrfw with different steps need AliCFContainer with Scale(), and clone() to do the same
123 THnSparseF *fhnCorrelation; //! response matrix for unfolding
124 THnSparseF *fhnCorrelationPhiZRec; //! response matrix for unfolding in max Z rec bins
3b7ffecf 125
3493c3a9 126 TF1 *f1PtScale; //! correction function to correct to the average true jet energy depending on p_T,rec
3b7ffecf 127
128 TString fBranchRec; // AOD branch name for reconstructed
129 TString fBranchGen; // AOD brnach for genereated
37eb26ea 130 TString fBranchBkgRec; //AOD branch for background
131 TString fBranchBkgGen; //AOD branch for background
a31b8a87 132 TString fNonStdFile; // name of delta aod file to catch the extension
3b7ffecf 133
565584e8 134 Bool_t fUseAODJetInput; // take jet from input AOD not from ouptu AOD
135 Bool_t fUseAODTrackInput; // take track from input AOD not from ouptu AOD
136 Bool_t fUseAODMCInput; // take MC from input AOD not from ouptu AOD
b5cc0c6d 137 Bool_t fUseGlobalSelection; // Limit the eta of the generated jets
3b7ffecf 138 Bool_t fUseExternalWeightOnly; // use only external weight
139 Bool_t fLimitGenJetEta; // Limit the eta of the generated jets
a31b8a87 140 Short_t fNMatchJets; // number of leading jets considered from the list
f2dd0695 141 UInt_t fFilterMask; // filter bit for slecected tracks
142 UInt_t fEventSelectionMask; // Selection information used to filter events
3b7ffecf 143 Int_t fAnalysisType; // Analysis type
144 Int_t fTrackTypeRec; // type of tracks used for FF
145 Int_t fTrackTypeGen; // type of tracks used for FF
f4132e7d 146 Int_t fEventClass; // event class to be looked at for this instance of the task
3b7ffecf 147 Float_t fAvgTrials; // Average nimber of trials
148 Float_t fExternalWeight; // external weight
a31b8a87 149 Float_t fJetRecEtaWindow; // eta window for rec jets
150 Float_t fTrackRecEtaWindow; // eta window for rec tracks
9280dfa6 151 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 152 Float_t fMinTrackPt; // limits the track p_T
26fa06fb 153 Float_t fDeltaPhiWindow; // minium angle between dijets
37eb26ea 154 Int_t fMultRec; // ! reconstructed track multiplicity
155 Int_t fMultGen; // ! generated track multiplicity
3b7ffecf 156
3493c3a9 157 TProfile* fh1Xsec; //! pythia cross section and trials
158 TH1F* fh1Trials; //! trials are added
159 TH1F* fh1PtHard; //! Pt har of the event...
160 TH1F* fh1PtHardNoW; //! Pt har of the event without weigt
161 TH1F* fh1PtHardTrials; //! Number of trials
57ca1193 162 TH1F* fh1ZVtx; //! z-vtx distribution
37eb26ea 163 TH1F* fh1TmpRho; //! just temporary histo for calculation
164 TH2F* fh2MultRec; //! reconstructed track multiplicity
165 TH2F* fh2MultGen; //! generated track multiplicity
166
a31b8a87 167 TH2F* fh2PtFGen; //! found vs generated
3493c3a9 168 TH2F* fh2RelPtFGen; //! relative difference between generated and found
37eb26ea 169
9a83d4af 170
a31b8a87 171 // Jet histos second go
172
173 TH1F* fh1NJets[kJetTypes]; //! nr of gen jets
174 TH1F* fh1SumPtTrack[kJetTypes]; //! sum over all track pT
175
37eb26ea 176 TH1F* fh1PtIn[kJetTypes][kMaxJets+1]; //! Jet pt
a31b8a87 177 TH1F* fh1PtJetsIn[kJetTypes]; //! Jet pt for all jets
178 TH1F* fh1PtTracksIn[kJetTypes]; //! track pt for all tracks
cb883243 179 TH1F* fh1PtTracksInLow[kJetTypes]; //! track pt for all tracks
a31b8a87 180 TH1F* fh1PtTracksLeadingIn[kJetTypes]; //! track pt for all tracks
181
37eb26ea 182 TH2F* fh2MultJetPt[kJetTypes]; //! jet pt vs. mult
a31b8a87 183 TH2F* fh2NJetsPt[kJetTypes]; //! Number of found jets above threshold
a2522483 184 TH2F* fh2NTracksPt[kJetTypes]; //! Number of tracks above threshold
a31b8a87 185 TH2F* fh2LeadingTrackPtTrackPhi[kJetTypes]; //! phi distribution of accepted leading tracks
37eb26ea 186 TH2F* fh2RhoPt[kJetTypes][kMaxJets+1]; //! jet shape variable rho
187 TH2F* fh2PsiPt[kJetTypes][kMaxJets+1]; //! jet shape variable psi
188 TH2F* fh2PhiPt[kJetTypes][kMaxJets+1]; //! phi of jets
189 TH2F* fh2EtaPt[kJetTypes][kMaxJets+1]; //! eta of jets
190 TH2F* fh2AreaPt[kJetTypes][kMaxJets+1]; //! area distribution
191 TH2F* fh2EtaArea[kJetTypes][kMaxJets+1]; //! area vs eta distribution
192 TH2F* fh2PhiEta[kJetTypes][kMaxJets+1]; //! eta phi distribution of jet
d7117cbd 193 TH2F* fh2LTrackPtJetPt[kJetTypes][kMaxJets+1]; //! leading track within the jet vs jet pt
a31b8a87 194
195 TH1F* fh1DijetMinv[kJetTypes]; //! dijet inv mass
196 TH2F* fh2DijetDeltaPhiPt[kJetTypes]; //! dijet delta phi vs pt
197 TH2F* fh2DijetAsymPt[kJetTypes]; //! dijet asym vs pt after delta phi cut
198 TH2F* fh2DijetPt2vsPt1[kJetTypes]; //! dijet pt2 vs pt1
199 TH2F* fh2DijetDifvsSum[kJetTypes]; //! dijet dif vs sum
200
f4132e7d 201 TList *fHistList; //! Output list
3b7ffecf 202
203
d7117cbd 204 ClassDef(AliAnalysisTaskJetSpectrum2, 13) // Analysis task for standard jet analysis
3b7ffecf 205};
206
207#endif