[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliAnalysisTaskFragFuncBB.h

#ifndef AliAnalysisTaskFragFuncBB_h
#define AliAnalysisTaskFragFuncBB_h

class AliESDEvent;
class AliAODEvent;
class TList;
class TH1F;
class TH2F;
class TH3F;


#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskFragFuncBB : public AliAnalysisTaskSE {

	public:
	
	AliAnalysisTaskFragFuncBB();
	AliAnalysisTaskFragFuncBB(const char *name);
	virtual ~AliAnalysisTaskFragFuncBB();

	virtual void   UserCreateOutputObjects();
	virtual void   Init();
	virtual void   LocalInit() {Init();}
	virtual void   UserExec(Option_t *option);
	virtual void   Terminate(Option_t* );
	
	virtual void   SetFilterMask(UInt_t i) {fFilterMask = i;}
	virtual void   SetUseGlobalSelection(Bool_t b){fUseGlobalSelection = b;}
	virtual void   SetAODJetInput(Bool_t b){fUseAODJetInput = b;}
	virtual void   SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
	virtual void   SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
	virtual void   SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
	virtual void   SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
	virtual void   SetBranchGenJets(const char* c){fBranchGenJets = c;}
	virtual void   SetBranchRecJets(const char* c){fBranchRecJets = c;}
	

	private:

	// Consts
	enum{ fgkMaxJets=4 }; // max. nb. of stored jets
	
	enum {kTrackUndef=0, kTrackAOD, kTrackKineAll, kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};

	// 
	Int_t   GetListOfTracks(TList *list, Int_t type);

	//
	AliESDEvent* fESD;
	AliAODEvent* fAOD;
	AliMCEvent*  fMCEvent;
	
	TString fBranchRecJets; // branch name for reconstructed jets
	TString fBranchGenJets; // branch name for generated jets

        Int_t   fLeadingRecJet; //
        Int_t   fLeadingGenJet; //
	
	Int_t  	fTrackTypeRec;       // type of reconstructed tracks
	Int_t  	fTrackTypeGen;       // type of generated tracks
	UInt_t 	fFilterMask;	     // filter bit for selected tracks
	Bool_t 	fUseGlobalSelection; // use selection of JetHelperTask
	Bool_t  fUseAODJetInput;	 // take jet from input AOD not from ouptut AOD
	Bool_t  fUseAODTrackInput;	 // take track from input AOD not from ouptut AOD
	Bool_t  fUseAODMCInput;		 // take MC from input AOD not from ouptut AOD
	
	Float_t fJetRadius;  // jet radius, used for fragmentation function 
	Float_t	fEtaMaxPart; // eta acceptance range for particles
	Float_t	fEtaMaxJets; // eta acceptance range for jets

	
	// Histograms
	TList		*fHistList;		  // List of histos

        TH1F            *fh1_eventSelection;
        TH1F		*fh1_vertexNContributors;
        TH1F		*fh1_vertexZ;

        // all jets 
	TH1F		*fh1_recJets_Et;
	TH1F		*fh1_genJets_Et;
	TH2F		*fh2_recJets_EtaPhi;
	TH2F		*fh2_genJets_EtaPhi;

        // jets, without acceptance cuts
	TH1F		*fh1_recJetsWoC_Et;
	TH1F		*fh1_genJetsWoC_Et;
	TH2F		*fh2_recJetsWoC_EtaPhi;
	TH2F		*fh2_genJetsWoC_EtaPhi;

        // leading jets
	TH1F		*fh1_recLJets_Et;
	TH1F		*fh1_genLJets_Et;
	TH2F		*fh2_recLJets_EtaPhi;
	TH2F		*fh2_genLJets_EtaPhi;
	
        // FF, all jets
	TH2F		*fh2_recFF_JetEt;
	TH2F		*fh2_genFF_JetEt;
	TH2F		*fh2_recHumpBacked_JetEt;
	TH2F		*fh2_genHumpBacked_JetEt;


        // FF, all jets without acceptance cuts
	TH2F		*fh2_recFF_JetWoCEt;
	TH2F		*fh2_genFF_JetWoCEt;
	TH2F		*fh2_recHumpBacked_JetWoCEt;
	TH2F		*fh2_genHumpBacked_JetWoCEt;

        // FF, leading jets
	TH2F		*fh2_recFF_LJetEt;
	TH2F		*fh2_genFF_LJetEt;
	TH2F		*fh2_recHumpBacked_LJetEt;
	TH2F		*fh2_genHumpBacked_LJetEt;

 
        // particles
	TH1F		*fh1_recPart_Pt;
	TH1F		*fh1_genPart_Pt;
	TH2F		*fh2_recPart_EtaPhi;
	TH2F		*fh2_genPart_EtaPhi;

	TH1F		*fh1_recJetPart_Pt;
	TH1F		*fh1_genJetPart_Pt;
	TH2F		*fh2_recJetPart_EtaPhi;
	TH2F		*fh2_genJetPart_EtaPhi;

        TH2F            *fh2_recJetPart_RJetPt;
        TH2F            *fh2_genJetPart_RJetPt;

	TH1F		*fh1_recJetWoCPart_Pt;
	TH1F		*fh1_genJetWoCPart_Pt;
	TH2F		*fh2_recJetWoCPart_EtaPhi;
	TH2F		*fh2_genJetWoCPart_EtaPhi;

        TH2F            *fh2_recJetWoCPart_RJetPt;
        TH2F            *fh2_genJetWoCPart_RJetPt;

	TH1F		*fh1_recLJetPart_Pt;
	TH1F		*fh1_genLJetPart_Pt;
	TH2F		*fh2_recLJetPart_EtaPhi;
	TH2F		*fh2_genLJetPart_EtaPhi;

        TH2F            *fh2_recLJetPart_RJetPt;
        TH2F            *fh2_genLJetPart_RJetPt;

        TH3F		*fh3_recPart_EtaPhiPt;
	
	ClassDef(AliAnalysisTaskFragFuncBB, 1);
};

#endif
Commit	Line	Data
3b4e1a15	1	#ifndef AliAnalysisTaskFragFuncBB_h
	2	#define AliAnalysisTaskFragFuncBB_h
	3
	4	class AliESDEvent;
	5	class AliAODEvent;
	6	class TList;
	7	class TH1F;
	8	class TH2F;
	9	class TH3F;
	10
	11
	12	#include "AliAnalysisTaskSE.h"
	13
	14	class AliAnalysisTaskFragFuncBB : public AliAnalysisTaskSE {
	15
	16	public:
	17
	18	AliAnalysisTaskFragFuncBB();
	19	AliAnalysisTaskFragFuncBB(const char *name);
	20	virtual ~AliAnalysisTaskFragFuncBB();
	21
	22	virtual void UserCreateOutputObjects();
	23	virtual void Init();
	24	virtual void LocalInit() {Init();}
	25	virtual void UserExec(Option_t *option);
	26	virtual void Terminate(Option_t* );
	27
	28	virtual void SetFilterMask(UInt_t i) {fFilterMask = i;}
	29	virtual void SetUseGlobalSelection(Bool_t b){fUseGlobalSelection = b;}
	30	virtual void SetAODJetInput(Bool_t b){fUseAODJetInput = b;}
	31	virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
	32	virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
	33	virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
	34	virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
	35	virtual void SetBranchGenJets(const char* c){fBranchGenJets = c;}
	36	virtual void SetBranchRecJets(const char* c){fBranchRecJets = c;}
	37
	38
	39	private:
	40
	41	// Consts
	42	enum{ fgkMaxJets=4 }; // max. nb. of stored jets
	43
	44	enum {kTrackUndef=0, kTrackAOD, kTrackKineAll, kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
	45
	46	//
	47	Int_t GetListOfTracks(TList *list, Int_t type);
	48
	49	//
	50	AliESDEvent* fESD;
	51	AliAODEvent* fAOD;
	52	AliMCEvent* fMCEvent;
	53
	54	TString fBranchRecJets; // branch name for reconstructed jets
	55	TString fBranchGenJets; // branch name for generated jets
	56
	57	Int_t fLeadingRecJet; //
	58	Int_t fLeadingGenJet; //
	59
	60	Int_t fTrackTypeRec; // type of reconstructed tracks
	61	Int_t fTrackTypeGen; // type of generated tracks
	62	UInt_t fFilterMask; // filter bit for selected tracks
	63	Bool_t fUseGlobalSelection; // use selection of JetHelperTask
	64	Bool_t fUseAODJetInput; // take jet from input AOD not from ouptut AOD
65	Bool_t fUseAODTrackInput; // take track from input AOD not from ouptut AOD
66	Bool_t fUseAODMCInput; // take MC from input AOD not from ouptut AOD
67
68	Float_t fJetRadius; // jet radius, used for fragmentation function
69	Float_t fEtaMaxPart; // eta acceptance range for particles
70	Float_t fEtaMaxJets; // eta acceptance range for jets
71
72
73	// Histograms
74	TList *fHistList; // List of histos
75
76	TH1F *fh1_eventSelection;
77	TH1F *fh1_vertexNContributors;
78	TH1F *fh1_vertexZ;
79
80	// all jets
81	TH1F *fh1_recJets_Et;
82	TH1F *fh1_genJets_Et;
83	TH2F *fh2_recJets_EtaPhi;
84	TH2F *fh2_genJets_EtaPhi;
85
86	// jets, without acceptance cuts
87	TH1F *fh1_recJetsWoC_Et;
88	TH1F *fh1_genJetsWoC_Et;
89	TH2F *fh2_recJetsWoC_EtaPhi;
90	TH2F *fh2_genJetsWoC_EtaPhi;
91
92	// leading jets
93	TH1F *fh1_recLJets_Et;
94	TH1F *fh1_genLJets_Et;
95	TH2F *fh2_recLJets_EtaPhi;
96	TH2F *fh2_genLJets_EtaPhi;
97
98	// FF, all jets
99	TH2F *fh2_recFF_JetEt;
100	TH2F *fh2_genFF_JetEt;
101	TH2F *fh2_recHumpBacked_JetEt;
102	TH2F *fh2_genHumpBacked_JetEt;
103
104
105	// FF, all jets without acceptance cuts
106	TH2F *fh2_recFF_JetWoCEt;
107	TH2F *fh2_genFF_JetWoCEt;
108	TH2F *fh2_recHumpBacked_JetWoCEt;
109	TH2F *fh2_genHumpBacked_JetWoCEt;
110
111	// FF, leading jets
112	TH2F *fh2_recFF_LJetEt;
113	TH2F *fh2_genFF_LJetEt;
114	TH2F *fh2_recHumpBacked_LJetEt;
115	TH2F *fh2_genHumpBacked_LJetEt;
116
117
118	// particles
119	TH1F *fh1_recPart_Pt;
120	TH1F *fh1_genPart_Pt;
121	TH2F *fh2_recPart_EtaPhi;
122	TH2F *fh2_genPart_EtaPhi;
123
124	TH1F *fh1_recJetPart_Pt;
125	TH1F *fh1_genJetPart_Pt;
126	TH2F *fh2_recJetPart_EtaPhi;
127	TH2F *fh2_genJetPart_EtaPhi;
128
129	TH2F *fh2_recJetPart_RJetPt;
130	TH2F *fh2_genJetPart_RJetPt;
131
132	TH1F *fh1_recJetWoCPart_Pt;
133	TH1F *fh1_genJetWoCPart_Pt;
134	TH2F *fh2_recJetWoCPart_EtaPhi;
135	TH2F *fh2_genJetWoCPart_EtaPhi;
136
137	TH2F *fh2_recJetWoCPart_RJetPt;
138	TH2F *fh2_genJetWoCPart_RJetPt;
139
140	TH1F *fh1_recLJetPart_Pt;
141	TH1F *fh1_genLJetPart_Pt;
142	TH2F *fh2_recLJetPart_EtaPhi;
143	TH2F *fh2_genLJetPart_EtaPhi;
144
145	TH2F *fh2_recLJetPart_RJetPt;
146	TH2F *fh2_genLJetPart_RJetPt;
147
148	TH3F *fh3_recPart_EtaPhiPt;
149
150	ClassDef(AliAnalysisTaskFragFuncBB, 1);
151	};
152
153	#endif