Technical fix: use include instead of forward declaration in case of objects
[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;
19class AliAODJet;
20class AliGenPythiaEventHeader;
21class AliCFManager;
22
23class TList;
24class TChain;
25class TH2F;
26class TH3F;
27class TProfile;
b1cd90b5 28class TSTring;
3b7ffecf 29
30
31class AliAnalysisTaskJetSpectrum2 : public AliAnalysisTaskSE
32{
33 public:
34 AliAnalysisTaskJetSpectrum2();
35 AliAnalysisTaskJetSpectrum2(const char* name);
36 virtual ~AliAnalysisTaskJetSpectrum2() {;}
37 // Implementation of interface methods
38 virtual void UserCreateOutputObjects();
39 virtual void Init();
40 virtual void LocalInit() { Init(); }
41 virtual void UserExec(Option_t *option);
42 virtual void Terminate(Option_t *option);
43 virtual Bool_t Notify();
44
b5cc0c6d 45 virtual void SetUseGlobalSelection(Bool_t b){fUseGlobalSelection = b;}
3b7ffecf 46 virtual void SetExternalWeight(Float_t f){fExternalWeight = f;}
47 virtual void SetUseExternalWeightOnly(Bool_t b){fUseExternalWeightOnly = b;}
565584e8 48 virtual void SetAODJetInput(Bool_t b){fUseAODJetInput = b;}
49 virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
50 virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
3b7ffecf 51 virtual void SetLimitGenJetEta(Bool_t b){fLimitGenJetEta = b;}
52 virtual void SetRecEtaWindow(Float_t f){fRecEtaWindow = f;}
53 virtual void SetAnalysisType(Int_t i){fAnalysisType = i;}
54 virtual void SetBranchGen(const char* c){fBranchGen = c;}
55 virtual void SetBranchRec(const char* c){fBranchRec = c;}
56 virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
57 virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
58 virtual void SetFilterMask(UInt_t i){fFilterMask = i;}
26fa06fb 59 virtual void SetDeltaPhiWindow(Float_t f){fDeltaPhiWindow = f;}
3b7ffecf 60 // use for the CF
61
62
63 // Helper
64 //
65
66 // we have different cases
67 // AOD reading -> MC from AOD
68 // ESD reading -> MC from Kinematics
69 // this has to match with our selection of input events
565584e8 70 enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
3b7ffecf 71 enum {kAnaMC = 0x1, kAnaMCESD = 0x2};
72 enum {kMaxJets = 4};
73 enum {kMaxCorrelation = 3};
74
75 //
76 // 0 all jets
77 // 1 all jet in eta window
78 // 2 all jets with partner
79 // 3 all jets in eta window with partner
80 // 4 all jets with partner in eta window
81 enum {kStep0 = 0, kStep1, kStep2, kStep3, kStep4,kMaxStep};
82
83
84 private:
85
86 AliAnalysisTaskJetSpectrum2(const AliAnalysisTaskJetSpectrum2&);
87 AliAnalysisTaskJetSpectrum2& operator=(const AliAnalysisTaskJetSpectrum2&);
88
89 void MakeJetContainer();
90 Int_t GetListOfTracks(TList *list,Int_t type);
91
92 AliJetHeader *fJetHeaderRec;
93 AliJetHeader *fJetHeaderGen;
94 AliAODEvent *fAOD; // where we take the jets from can be input or output AOD
95 THnSparseF *fhnJetContainer[kMaxStep*2]; // like particle container in corrfw with different steps need AliCFContainer with Scale(), and clone() to do the same
96 THnSparseF *fhnCorrelation; // response matrix for unfolding
97
98
99 TString fBranchRec; // AOD branch name for reconstructed
100 TString fBranchGen; // AOD brnach for genereated
101
565584e8 102 Bool_t fUseAODJetInput; // take jet from input AOD not from ouptu AOD
103 Bool_t fUseAODTrackInput; // take track from input AOD not from ouptu AOD
104 Bool_t fUseAODMCInput; // take MC from input AOD not from ouptu AOD
b5cc0c6d 105 Bool_t fUseGlobalSelection; // Limit the eta of the generated jets
3b7ffecf 106 Bool_t fUseExternalWeightOnly; // use only external weight
107 Bool_t fLimitGenJetEta; // Limit the eta of the generated jets
108 UInt_t fFilterMask; // filter bit for slecected tracks
109 Int_t fAnalysisType; // Analysis type
110 Int_t fTrackTypeRec; // type of tracks used for FF
111 Int_t fTrackTypeGen; // type of tracks used for FF
112 Float_t fAvgTrials; // Average nimber of trials
113 Float_t fExternalWeight; // external weight
114 Float_t fRecEtaWindow; // eta window used for corraltion plots between rec and gen
26fa06fb 115 Float_t fDeltaPhiWindow; // minium angle between dijets
116
3b7ffecf 117
118 TProfile* fh1Xsec; // pythia cross section and trials
119 TH1F* fh1Trials; // trials are added
120 TH1F* fh1PtHard; // Pt har of the event...
121 TH1F* fh1PtHardNoW; // Pt har of the event without weigt
122 TH1F* fh1PtHardTrials; // Number of trials
123 TH1F* fh1NGenJets; // nr of gen jets
124 TH1F* fh1NRecJets; // nr of rec jets
edfbe476 125 TH1F* fh1PtTrackRec; // track pt
126 TH1F* fh1SumPtTrackRec; // sum over all track pT
127 TH1F* fh1SumPtTrackAreaRec; // sum over all track pT
128
129
3b7ffecf 130 TH1F* fh1PtRecIn[kMaxJets]; // Jet pt for all this info is also in the THNsparse
131 TH1F* fh1PtGenIn[kMaxJets]; // Jet pt with corellated generated jet
132
cc0649e4 133 TH1F* fh1PtJetsRecIn; // Jet pt for all jets
565584e8 134 TH1F* fh1PtJetsLeadingRecIn; // Jet pt for all jets
cc0649e4 135 TH1F* fh1PtTracksRecIn; // track pt for all tracks
565584e8 136 TH1F* fh1PtTracksLeadingRecIn; // track pt for all tracks
cc0649e4 137 TH1F* fh1PtTracksGenIn; // track pt for all tracks
138
139
140 TH2F* fh2NRecJetsPt; // Number of found jets above threshold
141 TH2F* fh2NRecTracksPt; // Number of found tracks above threshold
565584e8 142 TH2F* fh2JetsLeadingPhiEta; // jet correlation with leading jet
143 TH2F* fh2JetsLeadingPhiPt; // jet correlation with leading jet
144 TH2F* fh2TracksLeadingPhiEta; // track correlation with leading track
145 TH2F* fh2TracksLeadingPhiPt; // track correlation with leading track
cf049e94 146 TH2F* fh2TracksLeadingJetPhiPt; // track correlation with leading track
edfbe476 147 TH2F* fh2PhiPt[kMaxJets]; // delta phi correlation of tracks with the jet
148 TH2F* fh2PhiEta[kMaxJets]; // eta phi correlation of tracks with the jet
149
3b7ffecf 150 TH2F* fh2FragRec[kMaxJets]; // fragmentation function
151 TH2F* fh2FragLnRec[kMaxJets]; // fragmetation in xi
152
153 TH2F* fh2FragGen[kMaxJets]; // fragmentation function
154 TH2F* fh2FragLnGen[kMaxJets]; // fragmetation in xi
155
26fa06fb 156 // Dijet histos
157 TH2F* fh2DijetDeltaPhiPt; // dijet delta phi vs pt
158 TH2F* fh2DijetAsymPt; // dijet asym vs pt
159 TH2F* fh2DijetAsymPtCut; // dijet asym vs pt after delta phi cut
160 TH2F* fh2DijetDeltaPhiDeltaEta; // dijet delta phi delta eta
161 TH2F* fh2DijetPt2vsPt1; // dijet pt2 vs pt1
162 TH2F* fh2DijetDifvsSum; // dijet dif vs sum
163 TH1F* fh1DijetMinv; // dijet inv mass
164 TH1F* fh1DijetMinvCut; // dijet inv after delta phi cut
165
3b7ffecf 166
167 TList *fHistList; // Output list
168
169
170 ClassDef(AliAnalysisTaskJetSpectrum2, 1) // Analysis task for standard jet analysis
171};
172
173#endif