]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGJE/UserTasks/AliAnalysisTaskJetHBOM.h
Changes to compile with Root6 on macosx64
[u/mrichter/AliRoot.git] / PWGJE / UserTasks / AliAnalysisTaskJetHBOM.h
CommitLineData
bcf7a476 1#ifndef ALIANALYSISTASKJETHBOM_H
2#define ALIANALYSISTASKJETHBOM_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 detector effects for background fluctuations in jet spectra by the HBOM method
9// *******************************************
10
11#include "AliAnalysisTaskSE.h"
12#include "THnSparse.h" // cannot forward declare ThnSparseF
dd3f006e 13#ifndef __CINT__
bcf7a476 14#include "fastjet/ClusterSequenceArea.hh"
15#include "fastjet/AreaDefinition.hh"
16#include "fastjet/JetDefinition.hh"
17#include "fastjet/PseudoJet.hh"
dd3f006e
CKB
18#else
19namespace fastjet {
20 class PseudoJet;
21 class GhostedAreaSpec;
22 class AreaType;
23 class JetDefinition;
24 class JetAlgorithm;
25 class Strategy;
26 class RecombinationScheme;
27}
28#endif
792cf46b 29#include "TH1F.h"
30#include "TH2D.h"
bcf7a476 31
32
33////////////////
34class AliJetHeader;
35class AliESDEvent;
36class AliAODEvent;
37class AliAODExtension;
38class AliAODJet;
39class AliGenPythiaEventHeader;
40class AliCFManager;
41class AliAODJetEventBackground;
42class AliJetFinder;
43class TList;
44class TChain;
bcf7a476 45class TH3F;
46class TProfile;
47class TRandom3;
48class TRefArray;
49class TClonesArray;
50class TF1;
51
52class AliAnalysisTaskJetHBOM : public AliAnalysisTaskSE
53{
54 public:
55 AliAnalysisTaskJetHBOM();
56 AliAnalysisTaskJetHBOM(const char* name);
57 virtual ~AliAnalysisTaskJetHBOM();
58 // Implementation of interface methods
59 virtual void UserCreateOutputObjects();
60 virtual void Init();
61 virtual void LocalInit() { Init(); }
62 virtual void UserExec(Option_t *option);
63 virtual void Terminate(Option_t *option);
64 virtual Bool_t Notify();
65
66
67
68 virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
69 virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
70 virtual void SetEventSelection(Bool_t b){fEventSelection = b;}
71 virtual void SetTrackEtaWindow(Float_t f){fTrackEtaWindow = f;}
72 virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
73 virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
74 virtual void SetTrackPtCut(Float_t x){fTrackPtCut = x;}
75 virtual void SetCentralityCut(Float_t xLo,Float_t xUp){fCentCutLo = xLo; fCentCutUp = xUp;}
76 virtual void SetFilterMask(UInt_t i,Int_t iType = 0){fFilterMask = i;
77 fFilterType = iType;}
78 virtual void SetJetTypes(UInt_t i){fJetTypes = i;}
79 virtual void SetVtxCuts(Float_t z,Float_t r = 1){fVtxZCut = z; fVtxR2Cut = r *r;}
80 virtual void SetBackgroundBranch(const char* c){fBackgroundBranch = c;}
81 virtual const char* GetBackgroundBranch(){return fBackgroundBranch.Data();}
82 virtual void SetNSkipLeadingRan(Int_t x){fNSkipLeadingRan = x;}
83 virtual void SetNSkipLeadingCone(Int_t x){fNSkipLeadingCone = x;}
84 virtual void SetNRandomCones(Int_t x){fNRandomCones = x;}
cfcde656 85 virtual void SetRandConePos(Double_t eta, Double_t phi){randCone_pos=1;randCone_Eta=eta;randCone_Phi=phi;}
bcf7a476 86
87 virtual void SetfNHBOM(Int_t x){fNHBOM = x;};
792cf46b 88 virtual void SetEfficiencyPt(TH1F *h){fh1efficiencyPt = (TH1F*)h->Clone("h1efficiencyPt");}
89 virtual void SetEfficiencyPhi(TH2D *h){fh2efficiencyPhi = (TH2D*)h->Clone("h2efficiencyPhi");}
bcf7a476 90
91 virtual void SetJetOutputBranch(const char *c){fNonStdBranch = c;}
92 virtual const char* GetJetOutputBranch(){return fNonStdBranch.Data();}
93 virtual void SetJetOutputFile(const char *c){fNonStdFile = c;}
94 virtual const char* GetJetOutputFile(){return fNonStdFile.Data();}
95 virtual void SetMaxTrackPtInJet(Float_t x){fMaxTrackPtInJet = x;}
96 virtual void SetJetOutputMinPt(Float_t x){fJetOutputMinPt = x;}
97
98 //Setters for detector level effects
99 virtual void SetSmearResolution(Bool_t b){fUseTrMomentumSmearing = b;}
100 virtual void SetDiceEfficiency(Bool_t b){fUseDiceEfficiency = b;}
101 virtual void SetMomentumResolutionHybrid(TProfile *p1, TProfile *p2, TProfile *p3);
102 virtual void SetEfficiencyHybrid(TH1 *h1, TH1 *h2, TH1 *h3);
103
104 Double_t GetMomentumSmearing(Int_t cat, Double_t pt);
105 void FitMomentumResolution();
106
107
108 // for Fast Jet
109 fastjet::JetAlgorithm GetAlgorithm() const {return fAlgorithm;}
110 fastjet::Strategy GetStrategy() const {return fStrategy;}
111 fastjet::RecombinationScheme GetRecombScheme() const {return fRecombScheme;}
112 fastjet::AreaType GetAreaType() const {return fAreaType;}
113 // Setters
114 void SetRparam(Double_t f) {fRparam = f;}
115 void SetAlgorithm(fastjet::JetAlgorithm f) {fAlgorithm = f;}
116 void SetStrategy(fastjet::Strategy f) {fStrategy = f;}
117 void SetRecombScheme(fastjet::RecombinationScheme f) {fRecombScheme = f;}
118 void SetAreaType(fastjet::AreaType f) {fAreaType = f;}
119 void SetGhostArea(Double_t f) {fGhostArea = f;}
120 void SetActiveAreaRepeats(Int_t f) {fActiveAreaRepeats = f;}
121 void SetGhostEtamax(Double_t f) {fGhostEtamax = f;}
122
123
124
125 // Helper
126 //
127
128 // we have different cases
129 // AOD reading -> MC from AOD
130 // ESD reading -> MC from Kinematics
131 // this has to match with our selection of input events
132 enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance, kTrackAODextra, kTrackAODextraonly};
133 enum {kMaxJets = 4};
134 enum {kMaxCorrelation = 3};
135 enum {kMaxRadius = 5};
136 enum {kMaxCent = 4};
137 enum {kJet = 1<<0,
138 kJetRan = 1<<1,
139 kRC = 1<<2,
140 kRCRan = 1<<3
141 };
142
143
144 private:
145
146 AliAnalysisTaskJetHBOM(const AliAnalysisTaskJetHBOM&);
147 AliAnalysisTaskJetHBOM& operator=(const AliAnalysisTaskJetHBOM&);
148
149 Int_t GetListOfTracks(TList *list,Int_t type);
150
151 AliAODEvent *fAOD; // ! where we take the jets from can be input or output AOD
152 AliAODExtension *fAODExtension; // ! AOD extension in case we write a non-sdt branch to a separate file and the aod is standard
153 TRefArray *fRef; // ! trefarray for track references within the jet
154 Bool_t fUseAODTrackInput; // take track from input AOD not from ouptu AOD
155 Bool_t fUseAODMCInput; // take MC from input AOD not from ouptu AOD
156 Bool_t fEventSelection; // use the event selection of this task, otherwise analyse all
157 UInt_t fFilterMask; // filter bit for slecected tracks
158 UInt_t fFilterMaskBestPt; // filter bit to mark jets with high quality leading tracks
159
160 UInt_t fFilterType; // filter type 0 = all, 1 = ITSTPC, 2 = TPC
161 UInt_t fJetTypes; // 1<<0 regular jets, 1<<1 << randomized event 1<<2 random cones 1<<3 random cones randomiuzed event
162 Int_t fTrackTypeRec; // type of tracks used for FF
163 Int_t fTrackTypeGen; // type of tracks used for FF
164 Int_t fNSkipLeadingRan; // number of leading tracks to be skipped in the randomized event
165 Int_t fNSkipLeadingCone; // number of leading jets to be for the random cones
166 Int_t fNRandomCones; // number of generated random cones
cfcde656 167 Bool_t randCone_pos; // use fixed position for random cones
168 Double_t randCone_Eta; // eta for random Cone at fixed position
169 Double_t randCone_Phi; // phi for random Cone at fixed position
bcf7a476 170 Int_t fNHBOM; // number of detector runs
171 Float_t fTrackEtaWindow; // eta window used for corraltion plots between rec and gen
bcf7a476 172 Float_t fTrackPtCut; // minimum track pt to be accepted
173 Float_t fJetOutputMinPt; // minimum p_t for jets to be written out
174 Float_t fMaxTrackPtInJet; // maximum track pt within a jet for flagging...
175 // Float_t fJetTriggerPtCut; // minimum jwt pT for AOD to be written
176 Float_t fVtxZCut; // zvtx cut
177 Float_t fVtxR2Cut; // R vtx cut (squared)
178 Float_t fCentCutUp; // upper limit on centrality
179 Float_t fCentCutLo; // lower limit on centrality
180 // output configurartion
181 TString fNonStdBranch; // the name of the non-std branch name, if empty no branch is filled
182 TString fBackgroundBranch; // name of the branch used for background subtraction
183 TString fNonStdFile; // The optional name of the output file the non-std branch is written to
184
185 //Detector level effects
186 TProfile *fMomResH1; // Momentum resolution from TrackQA Hybrid Category 1
187 TProfile *fMomResH2; // Momentum resolution from TrackQA Hybrid Category 2
188 TProfile *fMomResH3; // Momentum resolution from TrackQA Hybrid Category 3
189 TF1 *fMomResH1Fit; //fit
190 TF1 *fMomResH2Fit; //fit
191 TF1 *fMomResH3Fit; //fit
192
193 TH1 *fhEffH1; // Efficiency for Spectra Hybrid Category 1
194 TH1 *fhEffH2; // Efficiency for Spectra Hybrid Category 2
195 TH1 *fhEffH3; // Efficiency for Spectra Hybrid Category 3
196 Bool_t fUseTrMomentumSmearing; // Apply momentum smearing on track level
197 Bool_t fUseDiceEfficiency; // Apply efficiency on track level by dicing
198
199 // Fast jet
200 Double_t fRparam; // fastjet distance parameter
201 fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
202 fastjet::Strategy fStrategy; //= fastjet::Best;
203 fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
204 fastjet::AreaType fAreaType; // fastjet area type
205 Double_t fGhostArea; // fasjet ghost area
206 Int_t fActiveAreaRepeats; // fast jet active area repeats
207 Double_t fGhostEtamax; // fast jet ghost area
208
209 Double_t background; //background rho in the event
210
211 TClonesArray *fTCARandomConesOut; //! TCA of output jets in randomized event
bcf7a476 212
213 TRandom3* fRandom; //! random number generator
214 TProfile* fh1Xsec; //! pythia cross section and trials
215 TH1F* fh1Trials; //! trials are added
216 TH1F* fh1PtHard; //! Pt har of the event...
217 TH1F* fh1PtHardNoW; //! Pt har of the event without weigt
218 TH1F* fh1PtHardTrials; //! Number of trials
219
220 TH1F* fh1Nch; //! charged particle mult
221 TH1F* fh1CentralityPhySel; // ! centrality of anaylsed events
222 TH1F* fh1Centrality; // ! centrality of selected events
223 TH1F* fh1DeltapT; // pT of random Cone - background energy
224 TH1F* fh1Rho; //background rho
9b0a3be5 225 TH1F* fh1RhoSigma; //fluctuation of the background
bcf7a476 226 TH1F* fh1PtRandCone; //pT of random Cone
bcf7a476 227
228 TH1F* fh1efficiencyPt; //here efficiency is stored
229 TH2D* fh2efficiencyPhi; //here efficiency is stored
230
231 TH1F* fh1ZPhySel; // ! zvtx of anaylsed events
232 TH1F* fh1Z; // ! zvtx of selected events
233
234
235 TList *fHistList; //!leading tracks to be skipped in the randomized event Output list
236
237
238 ClassDef(AliAnalysisTaskJetHBOM, 1)
239};
240
241#endif