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

#ifndef ALIANALYSISTASKJETSPECTRUM_H
#define ALIANALYSISTASKJETSPECTRUM_H
 
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */
 
#include "AliAnalysisTaskSE.h"
#include "THnSparse.h"
////////////////
class AliJetHeader;
class AliESDEvent;
class AliAODEvent;
class AliAODJet;
class AliGenPythiaEventHeader;

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


class AliAnalysisTaskJetSpectrum : public AliAnalysisTaskSE
{
 public:
    AliAnalysisTaskJetSpectrum();
    AliAnalysisTaskJetSpectrum(const char* name);
    virtual ~AliAnalysisTaskJetSpectrum() {;}
    // 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 SetExternalWeight(Float_t f){fExternalWeight = f;}
    virtual void SetUseExternalWeightOnly(Bool_t b){fUseExternalWeightOnly = b;}
    virtual void SetAODInput(Bool_t b){fUseAODInput = b;}
    virtual void SetLimitGenJetEta(Bool_t b){fLimitGenJetEta = b;}
    virtual void SetAnalysisType(Int_t i){fAnalysisType = i;}
    virtual void SetBranchGen(char* c){fBranchGen = c;}
    virtual void SetBranchRec(char* c){fBranchRec = c;}

    // Helper
    static void GetClosestJets(AliAODJet *genJets,Int_t &nGenJets,
			       AliAODJet *recJets,Int_t &nRecJets,
			       Int_t *iGenIndex,Int_t *iRecIndex,Int_t iDebug = 0);

  //

    enum {kAnaMC =  0x1};
    enum {kMaxJets =  5};
    enum {kMaxCorrelation =  3};

 private:

    AliAnalysisTaskJetSpectrum(const AliAnalysisTaskJetSpectrum&);
    AliAnalysisTaskJetSpectrum& operator=(const AliAnalysisTaskJetSpectrum&);
    
    static const Float_t fgkJetNpartCut[kMaxCorrelation];


    AliJetHeader *fJetHeaderRec;
    AliJetHeader *fJetHeaderGen;
    AliAODEvent  *fAOD; // where we take the jets from can be input or output AOD

    TString       fBranchRec;  // AOD branch name for reconstructed
    TString       fConfigRec;  // Name of the Config file 
    TString       fBranchGen;  // AOD brnach for genereated
    TString       fConfigGen;  // Name of the Config file (if any)

    Bool_t        fUseAODInput;
    Bool_t        fUseExternalWeightOnly;
    Bool_t        fLimitGenJetEta;
    Int_t         fAnalysisType;
    Float_t       fExternalWeight;    

    TProfile*     fh1Xsec;   // pythia cross section and trials
    TH1F*         fh1Trials; // trials are added
    TH1F*         fh1PtHard;  // Pt har of the event...       
    TH1F*         fh1PtHard_NoW;  // Pt har of the event...       
    TH1F*         fh1PtHard_Trials;  // Number of trials 
    TH1F*         fh1NGenJets;
    TH1F*         fh1NRecJets;
    TH1F*         fh1E[kMaxJets];       // Jet Energy       
    TH1F*         fh1PtRecIn[kMaxJets];       // Jet pt for all      
    TH1F*         fh1PtRecOut[kMaxJets];      // Jet pt with corellated generated jet    
    TH1F*         fh1PtGenIn[kMaxJets];       // Detection efficiency for given p_T.gen
    TH1F*         fh1PtGenOut[kMaxJets];      //


    TH2F*         fh2PtFGen[kMaxJets];  //
    TH2F*         fh2PhiFGen[kMaxJets];  //
    TH2F*         fh2EtaFGen[kMaxJets];  //
    TH2F*         fh2Frag[kMaxJets];    // fragmentation function
    TH2F*         fh2FragLn[kMaxJets];  //
    TH2F*         fh2PtGenDeltaPhi[kMaxJets];  
    TH2F*         fh2PtGenDeltaEta[kMaxJets];  

    TH3F*         fh3PtRecGenHard[kMaxJets];  //                              
    TH3F*         fh3PtRecGenHard_NoW[kMaxJets];  //                  
    TH3F*         fh3RecEtaPhiPt[kMaxJets]; // 
    TH3F*         fh3RecEtaPhiPt_NoGen[kMaxJets]; // 
    TH3F*         fh3GenEtaPhiPt_NoFound[kMaxJets]; //                    
    TH3F*         fh3GenEtaPhiPt[kMaxJets]; //                                                                                                        
    // ========= Multiplicity dependence ======

    // ==========TODO , flavaor dependence ========                           
    // ============================================                       
                                     

    // ============= TODO , phi dependence ========
    // ============================================                            
  
    TList *fHistList; // Output list
    
    ///////// For 2 dimensional unfolding //////////////////
    TH1F*         fh1JetMultiplicity;   
    TH2F*         fh2ERecZRec;
    TH2F*         fh2EGenZGen;
    TH2F*         fh2Efficiency;
    TH3F*         fh3EGenERecN;
    THnSparseF*   fhnCorrelation[kMaxCorrelation];
    ///////////////////////////////////////////////////////


    ClassDef(AliAnalysisTaskJetSpectrum, 1) // Analysis task for standard jet analysis
};
 
#endif
Commit	Line	Data
f3050824	1	#ifndef ALIANALYSISTASKJETSPECTRUM_H
	2	#define ALIANALYSISTASKJETSPECTRUM_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	#include "AliAnalysisTaskSE.h"
4dbfdecc	8	#include "THnSparse.h"
df65bddb	9	////////////////
f3050824	10	class AliJetHeader;
	11	class AliESDEvent;
	12	class AliAODEvent;
	13	class AliAODJet;
	14	class AliGenPythiaEventHeader;
	15
	16	class TList;
	17	class TChain;
	18	class TH2F;
	19	class TH3F;
4dbfdecc	20	class TProfile;
4dbfdecc	21
f3050824	22
	23	class AliAnalysisTaskJetSpectrum : public AliAnalysisTaskSE
	24	{
	25	public:
	26	AliAnalysisTaskJetSpectrum();
	27	AliAnalysisTaskJetSpectrum(const char* name);
	28	virtual ~AliAnalysisTaskJetSpectrum() {;}
	29	// Implementation of interface methods
	30	virtual void UserCreateOutputObjects();
	31	virtual void Init();
	32	virtual void LocalInit() { Init(); }
	33	virtual void UserExec(Option_t *option);
	34	virtual void Terminate(Option_t *option);
4dbfdecc	35	virtual Bool_t Notify();
4dbfdecc	36
f3050824	37
	38	virtual void SetExternalWeight(Float_t f){fExternalWeight = f;}
	39	virtual void SetUseExternalWeightOnly(Bool_t b){fUseExternalWeightOnly = b;}
	40	virtual void SetAODInput(Bool_t b){fUseAODInput = b;}
	41	virtual void SetLimitGenJetEta(Bool_t b){fLimitGenJetEta = b;}
	42	virtual void SetAnalysisType(Int_t i){fAnalysisType = i;}
	43	virtual void SetBranchGen(char* c){fBranchGen = c;}
	44	virtual void SetBranchRec(char* c){fBranchRec = c;}
	45
	46	// Helper
	47	static void GetClosestJets(AliAODJet *genJets,Int_t &nGenJets,
	48	AliAODJet *recJets,Int_t &nRecJets,
	49	Int_t iGenIndex,Int_t iRecIndex,Int_t iDebug = 0);
	50
	51	//
	52
	53	enum {kAnaMC = 0x1};
df65bddb	54	enum {kMaxJets = 5};
df65bddb	55	enum {kMaxCorrelation = 3};
f3050824	56
	57	private:
	58
	59	AliAnalysisTaskJetSpectrum(const AliAnalysisTaskJetSpectrum&);
	60	AliAnalysisTaskJetSpectrum& operator=(const AliAnalysisTaskJetSpectrum&);
	61
df65bddb	62	static const Float_t fgkJetNpartCut[kMaxCorrelation];
f3050824	63
f3050824	64
	65	AliJetHeader *fJetHeaderRec;
	66	AliJetHeader *fJetHeaderGen;
	67	AliAODEvent *fAOD; // where we take the jets from can be input or output AOD
	68
	69	TString fBranchRec; // AOD branch name for reconstructed
	70	TString fConfigRec; // Name of the Config file
	71	TString fBranchGen; // AOD brnach for genereated
	72	TString fConfigGen; // Name of the Config file (if any)
	73
	74	Bool_t fUseAODInput;
	75	Bool_t fUseExternalWeightOnly;
	76	Bool_t fLimitGenJetEta;
	77	Int_t fAnalysisType;
	78	Float_t fExternalWeight;
	79
4dbfdecc	80	TProfile* fh1Xsec; // pythia cross section and trials
4dbfdecc	81	TH1F* fh1Trials; // trials are added
f3050824	82	TH1F* fh1PtHard; // Pt har of the event...
	83	TH1F* fh1PtHard_NoW; // Pt har of the event...
	84	TH1F* fh1PtHard_Trials; // Number of trials
	85	TH1F* fh1NGenJets;
	86	TH1F* fh1NRecJets;
	87	TH1F* fh1E[kMaxJets]; // Jet Energy
	88	TH1F* fh1PtRecIn[kMaxJets]; // Jet pt for all
	89	TH1F* fh1PtRecOut[kMaxJets]; // Jet pt with corellated generated jet
	90	TH1F* fh1PtGenIn[kMaxJets]; // Detection efficiency for given p_T.gen
	91	TH1F* fh1PtGenOut[kMaxJets]; //
	92
	93
	94
	95	TH2F* fh2PtFGen[kMaxJets]; //
22ceb537	96	TH2F* fh2PhiFGen[kMaxJets]; //
22ceb537	97	TH2F* fh2EtaFGen[kMaxJets]; //
f3050824	98	TH2F* fh2Frag[kMaxJets]; // fragmentation function
f3050824	99	TH2F* fh2FragLn[kMaxJets]; //
22ceb537	100	TH2F* fh2PtGenDeltaPhi[kMaxJets];
22ceb537	101	TH2F* fh2PtGenDeltaEta[kMaxJets];
f3050824	102
	103	TH3F* fh3PtRecGenHard[kMaxJets]; //
	104	TH3F* fh3PtRecGenHard_NoW[kMaxJets]; //
	105	TH3F* fh3RecEtaPhiPt[kMaxJets]; //
	106	TH3F* fh3RecEtaPhiPt_NoGen[kMaxJets]; //
	107	TH3F* fh3GenEtaPhiPt_NoFound[kMaxJets]; //
	108	TH3F* fh3GenEtaPhiPt[kMaxJets]; //
	109	// ========= Multiplicity dependence ======
	110
	111	// ==========TODO , flavaor dependence ========
	112	// ============================================
	113
	114
	115	// ============= TODO , phi dependence ========
	116	// ============================================
	117
	118	TList *fHistList; // Output list
df65bddb	119
	120	///////// For 2 dimensional unfolding //////////////////
	121	TH1F* fh1JetMultiplicity;
	122	TH2F* fh2ERecZRec;
	123	TH2F* fh2EGenZGen;
	124	TH2F* fh2Efficiency;
	125	TH3F* fh3EGenERecN;
	126	THnSparseF* fhnCorrelation[kMaxCorrelation];
	127	///////////////////////////////////////////////////////
f3050824	128
	129
	130	ClassDef(AliAnalysisTaskJetSpectrum, 1) // Analysis task for standard jet analysis
	131	};
	132
	133	#endif