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