add flag for the number of leading jets excluded for random cones
[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskJetCluster.h
1 #ifndef ALIANALYSISTASKJETCLUSTER_H
2 #define ALIANALYSISTASKJETCLUSTER_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 comparing different jets D parmaters from fastjet 
9 // *******************************************
10
11 #include  "AliAnalysisTaskSE.h"
12 #include  "THnSparse.h" // cannot forward declare ThnSparseF
13 #include "fastjet/ClusterSequenceArea.hh"
14 #include "fastjet/AreaDefinition.hh"
15 #include "fastjet/JetDefinition.hh"
16 #include "fastjet/PseudoJet.hh"
17
18 ////////////////
19 class AliJetHeader;
20 class AliESDEvent;
21 class AliAODEvent;
22 class AliAODExtension;
23 class AliAODJet;
24 class AliGenPythiaEventHeader;
25 class AliCFManager;
26 class AliAODJetEventBackground;
27 class AliJetFinder;
28 class TList;
29 class TChain;
30 class TH2F;
31 class TH1F;
32 class TH3F;
33 class TProfile;
34 class TRandom3;
35 class TRefArray;
36 class TClonesArray;
37
38 class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
39 {
40  public:
41     AliAnalysisTaskJetCluster();
42     AliAnalysisTaskJetCluster(const char* name);
43     virtual ~AliAnalysisTaskJetCluster();
44     // Implementation of interface methods
45     virtual void UserCreateOutputObjects();
46     virtual void Init();
47     virtual void LocalInit() { Init(); }
48     virtual void UserExec(Option_t *option);
49     virtual void Terminate(Option_t *option);
50     virtual Bool_t Notify();
51
52     
53
54     virtual void SetUseGlobalSelection(Bool_t b){fUseGlobalSelection = b;}
55     virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
56     virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
57     virtual void SetRecEtaWindow(Float_t f){fRecEtaWindow = f;}
58     virtual void SetTrackEtaWindow(Float_t f){fTrackEtaWindow = f;}
59     virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
60     virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
61     virtual void SetTrackPtCut(Float_t x){fTrackPtCut = x;}
62     virtual void SetCentralityCut(Float_t xLo,Float_t xUp){fCentCutLo = xLo; fCentCutUp = xUp;}
63     virtual void SetFilterMask(UInt_t i){fFilterMask = i;}
64     virtual void SetJetTriggerPtCut(Float_t x){fJetTriggerPtCut = x;}    
65     virtual void SetBackgroundBranch(const char* c){fBackgroundBranch = c;}
66     virtual const char* GetBackgroundBranch(){return fBackgroundBranch.Data();}    
67     virtual void SetNSkipLeadingRan(Int_t x){fNSkipLeadingRan = x;}
68     virtual void SetNSkipLeadingCone(Int_t x){fNSkipLeadingCone = x;}
69     virtual void SetNRandomCones(Int_t x){fNRandomCones = x;}
70
71     virtual void SetJetOutputBranch(const char *c){fNonStdBranch = c;}
72     virtual const char* GetJetOutputBranch(){return fNonStdBranch.Data();}
73     virtual void SetJetOutputFile(const char *c){fNonStdFile = c;}
74     virtual const char* GetJetOutputFile(){return fNonStdFile.Data();}
75     virtual void SetJetOutputMinPt(Float_t x){fJetOutputMinPt = x;}
76     virtual void SetBackgroundCalc(Bool_t b){fUseBackgroundCalc = b;} 
77
78     // for Fast Jet
79     fastjet::JetAlgorithm        GetAlgorithm()         const {return fAlgorithm;}
80     fastjet::Strategy            GetStrategy()          const {return fStrategy;}
81     fastjet::RecombinationScheme GetRecombScheme()      const {return fRecombScheme;}
82     fastjet::AreaType            GetAreaType()          const {return fAreaType;}
83     // Setters
84     void SetRparam(Double_t f)                           {fRparam = f;}
85     void SetAlgorithm(fastjet::JetAlgorithm f)           {fAlgorithm = f;}
86     void SetStrategy(fastjet::Strategy f)                {fStrategy = f;}
87     void SetRecombScheme(fastjet::RecombinationScheme f) {fRecombScheme = f;}
88     void SetAreaType(fastjet::AreaType f)                {fAreaType = f;}
89     void SetGhostArea(Double_t f) {fGhostArea = f;}
90     void SetActiveAreaRepeats(Int_t f) {fActiveAreaRepeats = f;}
91     void SetGhostEtamax(Double_t f) {fGhostEtamax = f;}
92
93
94
95     // Helper
96     //
97
98     // we have different cases
99     // AOD reading -> MC from AOD
100     // ESD reading -> MC from Kinematics
101     // this has to match with our selection of input events
102     enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance, kTrackAODextra, kTrackAODextraonly};
103     enum {kMaxJets = 4};
104     enum {kMaxCorrelation =  3};
105     enum {kMaxRadius =       5};
106     enum {kMaxCent =         4};
107     
108
109  private:
110
111     AliAnalysisTaskJetCluster(const AliAnalysisTaskJetCluster&);
112     AliAnalysisTaskJetCluster& operator=(const AliAnalysisTaskJetCluster&);
113
114     Int_t GetListOfTracks(TList *list,Int_t type);
115
116     AliAODEvent     *fAOD;                // ! where we take the jets from can be input or output AOD
117     AliAODExtension *fAODExtension;       // ! AOD extension in case we write a non-sdt branch to a separate file and the aod is standard
118     TRefArray       *fRef;               // ! trefarray for track references within the jet
119     Bool_t        fUseAODTrackInput;      // take track from input AOD not from ouptu AOD
120     Bool_t        fUseAODMCInput;         // take MC from input AOD not from ouptu AOD
121     Bool_t        fUseGlobalSelection;    // Limit the eta of the generated jets
122     Bool_t        fUseBackgroundCalc;     // switches on background calculations
123     UInt_t        fFilterMask;            // filter bit for slecected tracks
124     Int_t         fTrackTypeRec;          // type of tracks used for FF 
125     Int_t         fTrackTypeGen;          // type of tracks used for FF 
126     Int_t         fNSkipLeadingRan;       // number of leading tracks to be skipped in the randomized event
127     Int_t         fNSkipLeadingCone;      // number of leading jets to be for the random cones
128     Int_t         fNRandomCones;          // number of generated random cones
129     Float_t       fAvgTrials;             // Average nimber of trials
130     Float_t       fExternalWeight;        // external weight
131     Float_t       fTrackEtaWindow;          // eta window used for corraltion plots between rec and gen 
132     Float_t       fRecEtaWindow;          // eta window used for corraltion plots between rec and gen 
133     Float_t       fTrackPtCut;            // minimum track pt to be accepted
134     Float_t       fJetOutputMinPt;        // minimum p_t for jets to be written out
135     Float_t       fJetTriggerPtCut;       // minimum jwt pT for AOD to be written
136     Float_t       fCentCutUp;             // upper limit on centrality
137     Float_t       fCentCutLo;             // lower limit on centrality
138     // output configurartion
139     TString       fNonStdBranch;      // the name of the non-std branch name, if empty no branch is filled
140     TString       fBackgroundBranch;  // name of the branch used for background subtraction
141     TString       fNonStdFile;        // The optional name of the output file the non-std brnach is written to
142     
143
144     // Fast jet
145     Double_t fRparam;
146     fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
147     fastjet::Strategy fStrategy;  //= fastjet::Best;
148     fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
149     fastjet::AreaType fAreaType; 
150     Double_t fGhostArea;
151     Int_t fActiveAreaRepeats;
152     Double_t fGhostEtamax;
153
154     TClonesArray  *fTCAJetsOut; //! TCA of output jets
155     TClonesArray  *fTCAJetsOutRan; //! TCA of output jets in randomized event
156     TClonesArray  *fTCARandomConesOut; //! TCA of output jets in randomized event
157     TClonesArray  *fTCARandomConesOutRan; //! TCA of output jets in randomized event
158     AliAODJetEventBackground *fAODJetBackgroundOut; //! jet background to be written out
159
160     TRandom3*     fRandom;   //! random number generator
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 
166
167     TH1F*         fh1NJetsRec; //! number of reconstructed jets
168     TH1F*         fh1NConstRec;//! number of constiutens in leading jet
169     TH1F*         fh1NConstLeadingRec;//! number of constiutens in leading jet
170     TH1F*         fh1PtJetsRecIn;  //! Jet pt for all jets
171     TH1F*         fh1PtJetsLeadingRecIn;  //! Jet pt for all jets
172     TH1F*         fh1PtJetConstRec;//! pt of constituents
173     TH1F*         fh1PtJetConstLeadingRec;// pt of constituents
174     TH1F*         fh1PtTracksRecIn;  //! track pt for all tracks
175     TH1F*         fh1PtTracksLeadingRecIn;  //! track pt for all tracks
176
177     // Randomized track histos
178     TH1F*         fh1NJetsRecRan; //! number of reconstructed jets from randomized
179     TH1F*         fh1NConstRecRan;//! number of constiutens in leading jet
180     TH1F*         fh1PtJetsLeadingRecInRan;  //! Jet pt for all jets
181     TH1F*         fh1NConstLeadingRecRan;//! number of constiutens in leading jet
182     TH1F*         fh1PtJetsRecInRan;  //! Jet pt for all jets
183
184     TH1F*         fh1PtTracksGenIn;  //! track pt for all tracks
185     TH1F*         fh1Nch;            //! charged particle mult
186     TH1F*         fh1BiARandomCones[3]; //! Residual distribtion from reandom cones on real event
187     TH1F*         fh1BiARandomConesRan[3]; //! Residual distribtion from reandom cones on random event
188     TH1F*         fh1CentralityPhySel;          // ! centrality of anaylsed events 
189     TH1F*         fh1Centrality;                // ! centrality of anaylsed events 
190     TH1F*         fh1CentralitySelect;          // ! centrality of selected events 
191     TH1F*         fh1ZPhySel;          // ! centrality of anaylsed events 
192     TH1F*         fh1Z;                // ! centrality of anaylsed events 
193     TH1F*         fh1ZSelect;          // ! centrality of selected events 
194
195
196     TH2F*         fh2NRecJetsPt;            //! Number of found jets above threshold
197     TH2F*         fh2NRecTracksPt;          //! Number of found tracks above threshold
198     TH2F*         fh2NConstPt;           //! number of constituents vs. pt
199     TH2F*         fh2NConstLeadingPt;           //! number of constituents vs. pt
200     TH2F*         fh2JetPhiEta;             //! jet phi eta
201     TH2F*         fh2LeadingJetPhiEta;      //! leading jet phi eta
202     TH2F*         fh2JetEtaPt;              //! leading jet eta
203     TH2F*         fh2LeadingJetEtaPt;              //! leading jet eta
204     TH2F*         fh2TrackEtaPt;              //! track eta
205     TH2F*         fh2LeadingTrackEtaPt;       //! leading track eta
206     TH2F*         fh2JetsLeadingPhiEta;     //! jet phi eta
207     TH2F*         fh2JetsLeadingPhiPt;      //! jet correlation with leading jet
208     TH2F*         fh2TracksLeadingPhiEta;   //! track correlation with leading track
209     TH2F*         fh2TracksLeadingPhiPt;    //! track correlation with leading track
210     TH2F*         fh2TracksLeadingJetPhiPt; //! track correlation with leading Jet
211     TH2F*         fh2JetsLeadingPhiPtW;      //! jet correlation with leading jet
212     TH2F*         fh2TracksLeadingPhiPtW;   //! track correlation with leading track
213     TH2F*         fh2TracksLeadingJetPhiPtW; //! track correlation with leading Jet
214     TH2F*         fh2NRecJetsPtRan;            //! Number of found jets above threshold
215     TH2F*         fh2NConstPtRan;           //! number of constituents vs. pt
216     TH2F*         fh2NConstLeadingPtRan;           //! number of constituents vs. pt
217     TH2F*         fh2PtNch;               //! p_T of cluster vs. multiplicity,
218     TH2F*         fh2PtNchRan;            //! p_T of cluster vs. multiplicity,random
219     TH2F*         fh2PtNchN;               //! p_T of cluster vs. multiplicity, weigthed with constituents
220     TH2F*         fh2PtNchNRan;            //! p_T of cluster vs. multiplicity, weigthed with constituents random
221     TH2F*         fh2TracksLeadingJetPhiPtRan; //! track correlation with leading Jet
222     TH2F*         fh2TracksLeadingJetPhiPtWRan; //! track correlation with leading Jet
223
224
225     TH2F*         fh2JetsLeadingPhiPtC[kMaxCent]; //! jet correlation with leading jet    
226     TH2F*         fh2JetsLeadingPhiPtWC[kMaxCent];      //! jet correlation with leading jet
227     TH2F*         fh2TracksLeadingJetPhiPtC[kMaxCent]; //! track correlation with leading Jet
228     TH2F*         fh2TracksLeadingJetPhiPtWC[kMaxCent]; //! track correlation with leading Jet
229
230     TList *fHistList; //!leading tracks to be skipped in the randomized event Output list
231    
232
233     ClassDef(AliAnalysisTaskJetCluster, 16) 
234 };
235  
236 #endif