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

#ifndef ALIANALYSISTASKJETSPECTRUM2_H
#define ALIANALYSISTASKJETSPECTRUM2_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

// **************************************
// Task used for the correction of determiantion of reconstructed jet spectra
// Compares input (gen) and output (rec) jets   
// *******************************************

#include  "AliAnalysisTaskSE.h"
#include  "THnSparse.h" // cannot forward declare ThnSparseF

////////////////
class AliJetHeader;
class AliESDEvent;
class AliAODEvent;
class AliAODJet;
class AliGenPythiaEventHeader;
class AliCFManager;

class TList;
class TChain;
class TH2F;
class TH3F;
class TProfile;
class TSTring;


class AliAnalysisTaskJetSpectrum2 : public AliAnalysisTaskSE
{
 public:
    AliAnalysisTaskJetSpectrum2();
    AliAnalysisTaskJetSpectrum2(const char* name);
    virtual ~AliAnalysisTaskJetSpectrum2() {;}
    // Implementation of interface methods
    virtual void UserCreateOutputObjects();
    virtual void Init();
    virtual void LocalInit() { Init(); }
    virtual void UserExec(Option_t *option);
    virtual void Terminate(Option_t *option);
    virtual Bool_t Notify();

    virtual void SetUseGlobalSelection(Bool_t b){fUseGlobalSelection = b;}
    virtual void SetExternalWeight(Float_t f){fExternalWeight = f;}
    virtual void SetUseExternalWeightOnly(Bool_t b){fUseExternalWeightOnly = 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 SetLimitGenJetEta(Bool_t b){fLimitGenJetEta = b;}
    virtual void SetRecEtaWindow(Float_t f){fRecEtaWindow = f;}
    virtual void SetMinJetPt(Float_t f){fMinJetPt = f;}
    virtual void SetDeltaPhiWindow(Float_t f){fDeltaPhiWindow = f;}
    virtual void SetAnalysisType(Int_t i){fAnalysisType = i;}
    virtual void SetBranchGen(const char* c){fBranchGen = c;}
    virtual void SetBranchRec(const char* c){fBranchRec = c;}
    virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
    virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
    virtual void SetFilterMask(UInt_t i){fFilterMask = i;}
    // use for the CF


    // Helper
    //

    // we have different cases
    // AOD reading -> MC from AOD
    // ESD reading -> MC from Kinematics
    // this has to match with our selection of input events
    enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
    enum {kAnaMC =  0x1, kAnaMCESD = 0x2};
    enum {kMaxJets = 4};
    enum {kMaxCorrelation =  3};
    
    // 
    // 0 all jets
    // 1 all jet in eta window
    // 2 all jets with partner
    // 3 all jets in eta window with partner
    // 4 all jets with partner in eta window
    enum {kStep0 = 0, kStep1, kStep2, kStep3, kStep4,kMaxStep};


 private:

    AliAnalysisTaskJetSpectrum2(const AliAnalysisTaskJetSpectrum2&);
    AliAnalysisTaskJetSpectrum2& operator=(const AliAnalysisTaskJetSpectrum2&);

    void MakeJetContainer();
    Int_t GetListOfTracks(TList *list,Int_t type);

    AliJetHeader *fJetHeaderRec;
    AliJetHeader *fJetHeaderGen;
    AliAODEvent  *fAOD; // where we take the jets from can be input or output AOD
    THnSparseF   *fhnJetContainer[kMaxStep*2];   // like particle container in corrfw with different steps need AliCFContainer with Scale(), and clone() to do the same
    THnSparseF   *fhnCorrelation;           // response matrix for unfolding 
    THnSparseF   *fhnCorrelationPhiZRec;       // response matrix for unfolding in max Z rec bins

    TF1          *f1PtScale;                // correction function to correct to the average true jet energy depending on p_T,rec

    TString       fBranchRec;  // AOD branch name for reconstructed
    TString       fBranchGen;  // AOD brnach for genereated

    Bool_t        fUseAODJetInput;        // take jet from input AOD not from ouptu AOD
    Bool_t        fUseAODTrackInput;      // take track from input AOD not from ouptu AOD
    Bool_t        fUseAODMCInput;         // take MC from input AOD not from ouptu AOD
    Bool_t        fUseGlobalSelection;    // Limit the eta of the generated jets
    Bool_t        fUseExternalWeightOnly; // use only external weight
    Bool_t        fLimitGenJetEta;        // Limit the eta of the generated jets
    UInt_t        fFilterMask;             // filter bit for slecected tracks
    Int_t         fAnalysisType;          // Analysis type 
    Int_t         fTrackTypeRec;          // type of tracks used for FF 
    Int_t         fTrackTypeGen;          // type of tracks used for FF 
    Float_t       fAvgTrials;             // Average nimber of trials
    Float_t       fExternalWeight;        // external weight
    Float_t       fRecEtaWindow;          // eta window used for corraltion plots between rec and gen 
    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
    Float_t       fDeltaPhiWindow;        // minium angle between dijets


    TProfile*     fh1Xsec;   // pythia cross section and trials
    TH1F*         fh1Trials; // trials are added
    TH1F*         fh1PtHard;  // Pt har of the event...       
    TH1F*         fh1PtHardNoW;  // Pt har of the event without weigt       
    TH1F*         fh1PtHardTrials;  // Number of trials 
    TH1F*         fh1NGenJets;      // nr of gen jets
    TH1F*         fh1NRecJets;      // nr of rec jets
    TH1F*         fh1PtTrackRec;    // track pt
    TH1F*         fh1SumPtTrackRec; // sum over all track pT    
    TH1F*         fh1SumPtTrackAreaRec; // sum over all track pT    
    TH1F*         fh1TmpRho;            // just temporary histo for calculation


    TH1F*         fh1PtRecIn[kMaxJets];  // Jet pt for all this info is also in the THNsparse      
    TH1F*         fh1PtGenIn[kMaxJets];  // Jet pt with corellated generated jet    

    TH1F*         fh1PtJetsRecIn;  // Jet pt for all jets
    TH1F*         fh1PtJetsLeadingRecIn;  // Jet pt for all jets
    TH1F*         fh1PtTracksRecIn;  // track pt for all tracks
    TH1F*         fh1PtTracksLeadingRecIn;  // track pt for all tracks
    TH1F*         fh1PtTracksGenIn;  // track pt for all tracks
    

    TH2F*         fh2NRecJetsPt;            // Number of found jets above threshold
    TH2F*         fh2NRecTracksPt;          // Number of found tracks above threshold
    TH2F*         fh2JetsLeadingPhiEta;     // jet correlation with leading jet
    TH2F*         fh2JetsLeadingPhiPt;      // jet correlation with leading jet
    TH2F*         fh2TracksLeadingPhiEta;   // track correlation with leading track
    TH2F*         fh2TracksLeadingPhiPt;    // track correlation with leading track
    TH2F*         fh2TracksLeadingJetPhiPt; // track correlation with leading track
    TH2F*         fh2JetPtJetPhi;           // Phi distribution of accepted jets 
    TH2F*         fh2TrackPtTrackPhi;       // phi distribution of accepted tracks
    TH2F*         fh2PhiPt[kMaxJets];    // delta phi correlation of tracks with the jet      
    TH2F*         fh2PhiEta[kMaxJets];   // eta   phi correlation of tracks with the jet      
    TH2F*         fh2RhoPtRec[kMaxJets];    // jet shape variable rho
    TH2F*         fh2PsiPtRec[kMaxJets];    // jet shape variable psi
    TH2F*         fh2RhoPtGen[kMaxJets];    //
    TH2F*         fh2PsiPtGen[kMaxJets];    //
    TH2F*         fh2FragRec[kMaxJets];     // fragmentation function
    TH2F*         fh2FragLnRec[kMaxJets];   // fragmetation in xi
    TH2F*         fh2FragGen[kMaxJets];     // fragmentation function
    TH2F*         fh2FragLnGen[kMaxJets];   // fragmetation in xi

    // Dijet histos
    TH2F*   fh2DijetDeltaPhiPt;      // dijet delta phi vs pt
    TH2F*   fh2DijetAsymPt;          // dijet asym vs pt
    TH2F*   fh2DijetAsymPtCut;       // dijet asym vs pt after delta phi cut
    TH2F*   fh2DijetDeltaPhiDeltaEta; // dijet delta phi delta eta
    TH2F*   fh2DijetPt2vsPt1;          // dijet pt2 vs pt1 
    TH2F*   fh2DijetDifvsSum;          // dijet dif vs sum
    TH1F*   fh1DijetMinv;            // dijet inv mass
    TH1F*   fh1DijetMinvCut;         // dijet inv after delta phi cut


    TList *fHistList; // Output list
   

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