[u/mrichter/AliRoot.git] / JETAN / AliAnalysisTaskJetCluster.h

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

// **************************************
// task used for comapring different jets D parmaters from fastjet 
// *******************************************

#include  "AliAnalysisTaskSE.h"
#include  "THnSparse.h" // cannot forward declare ThnSparseF
#include "fastjet/ClusterSequenceArea.hh"
#include "fastjet/AreaDefinition.hh"
#include "fastjet/JetDefinition.hh"
#include "fastjet/PseudoJet.hh"

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

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


class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
{
 public:
    AliAnalysisTaskJetCluster();
    AliAnalysisTaskJetCluster(const char* name);
    virtual ~AliAnalysisTaskJetCluster() {;}
    // 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 SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
    virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
    virtual void SetRecEtaWindow(Float_t f){fRecEtaWindow = f;}
    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
    

    // for Fast Jet
    fastjet::JetAlgorithm        GetAlgorithm()         const {return fAlgorithm;}
    fastjet::Strategy            GetStrategy()          const {return fStrategy;}
    fastjet::RecombinationScheme GetRecombScheme()      const {return fRecombScheme;}
    fastjet::AreaType            GetAreaType()          const {return fAreaType;}
    // Setters
    void SetRparam(Double_t f)                           {fRparam = f;}
    void SetAlgorithm(fastjet::JetAlgorithm f)           {fAlgorithm = f;}
    void SetStrategy(fastjet::Strategy f)                {fStrategy = f;}
    void SetRecombScheme(fastjet::RecombinationScheme f) {fRecombScheme = f;}
    void SetAreaType(fastjet::AreaType f)                {fAreaType = f;}

    // 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 {kMaxJets = 4};
    enum {kMaxCorrelation =  3};
    enum {kMaxRadius =       5};
    

 private:

    AliAnalysisTaskJetCluster(const AliAnalysisTaskJetCluster&);
    AliAnalysisTaskJetCluster& operator=(const AliAnalysisTaskJetCluster&);

    Int_t GetListOfTracks(TList *list,Int_t type);

    AliAODEvent  *fAOD; // where we take the jets from can be input or output 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
    UInt_t        fFilterMask;             // filter bit for slecected tracks
    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 
    // Fast jet
    Double_t fRparam;
    fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
    fastjet::Strategy fStrategy;  //= fastjet::Best;
    fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
    fastjet::AreaType fAreaType; 

    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*         fh1NJetsRec; // number of reconstructed jets
    TH1F*         fh1NConstRec;// number of constiutens in leading jet
    TH1F*         fh1NConstLeadingRec;// number of constiutens in leading jet
    TH1F*         fh1PtJetsRecIn;  // Jet pt for all jets
    TH1F*         fh1PtJetsLeadingRecIn;  // Jet pt for all jets
    TH1F*         fh1PtJetConstRec;// pt of constituents
    TH1F*         fh1PtJetConstLeadingRec;// pt of constituents
    TH1F*         fh1PtTracksRecIn;  // track pt for all tracks
    TH1F*         fh1PtTracksLeadingRecIn;  // track pt for all tracks

    // Randomized track histos
    TH1F*         fh1NJetsRecRan; // number of reconstructed jets from randomized
    TH1F*         fh1NConstRecRan;// number of constiutens in leading jet
    TH1F*         fh1PtJetsLeadingRecInRan;  // Jet pt for all jets
    TH1F*         fh1NConstLeadingRecRan;// number of constiutens in leading jet
    TH1F*         fh1PtJetsRecInRan;  // Jet pt for all jets

    TH1F*         fh1PtTracksGenIn;  // track pt for all tracks


    TH2F*         fh2NRecJetsPt;            // Number of found jets above threshold
    TH2F*         fh2NRecTracksPt;          // Number of found tracks above threshold
    TH2F*         fh2NConstPt;           // number of constituents vs. pt
    TH2F*         fh2NConstLeadingPt;           // number of constituents vs. pt
    TH2F*         fh2JetPhiEta;             // jet phi eta
    TH2F*         fh2LeadingJetPhiEta;      // leading jet phi eta
    TH2F*         fh2JetEtaPt;              // leading jet eta
    TH2F*         fh2LeadingJetEtaPt;              // leading jet eta
    TH2F*         fh2TrackEtaPt;              // track eta
    TH2F*         fh2LeadingTrackEtaPt;       // leading track eta
    TH2F*         fh2JetsLeadingPhiEta;     // jet phi eta
    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 Jet
    TH2F*         fh2JetsLeadingPhiPtW;      // jet correlation with leading jet
    TH2F*         fh2TracksLeadingPhiPtW;   // track correlation with leading track
    TH2F*         fh2TracksLeadingJetPhiPtW; // track correlation with leading Jet
    TH2F*         fh2NRecJetsPtRan;            // Number of found jets above threshold
    TH2F*         fh2NConstPtRan;           // number of constituents vs. pt
    TH2F*         fh2NConstLeadingPtRan;           // number of constituents vs. pt
    TH2F*         fh2TracksLeadingJetPhiPtRan; // track correlation with leading Jet
    TH2F*         fh2TracksLeadingJetPhiPtWRan; // track correlation with leading Jet
    TList *fHistList; // Output list
   

    ClassDef(AliAnalysisTaskJetCluster, 2) // Analysis task for standard jet analysis
};
 
#endif
Commit	Line	Data
dbf053f0	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 comapring 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 AliAODJet;
	23	class AliGenPythiaEventHeader;
	24	class AliCFManager;
	25
	26	class TList;
	27	class TChain;
	28	class TH2F;
	29	class TH1F;
	30	class TH3F;
	31	class TProfile;
	32
	33
	34
	35	class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
	36	{
	37	public:
	38	AliAnalysisTaskJetCluster();
	39	AliAnalysisTaskJetCluster(const char* name);
	40	virtual ~AliAnalysisTaskJetCluster() {;}
	41	// Implementation of interface methods
	42	virtual void UserCreateOutputObjects();
	43	virtual void Init();
	44	virtual void LocalInit() { Init(); }
	45	virtual void UserExec(Option_t *option);
	46	virtual void Terminate(Option_t *option);
	47	virtual Bool_t Notify();
	48
	49	virtual void SetUseGlobalSelection(Bool_t b){fUseGlobalSelection = b;}
	50	virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
	51	virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
	52	virtual void SetRecEtaWindow(Float_t f){fRecEtaWindow = f;}
	53	virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
	54	virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
	55	virtual void SetFilterMask(UInt_t i){fFilterMask = i;}
	56	// use for the CF
	57
	58
	59	// for Fast Jet
	60	fastjet::JetAlgorithm GetAlgorithm() const {return fAlgorithm;}
	61	fastjet::Strategy GetStrategy() const {return fStrategy;}
	62	fastjet::RecombinationScheme GetRecombScheme() const {return fRecombScheme;}
	63	fastjet::AreaType GetAreaType() const {return fAreaType;}
	64	// Setters
65	void SetRparam(Double_t f) {fRparam = f;}
66	void SetAlgorithm(fastjet::JetAlgorithm f) {fAlgorithm = f;}
67	void SetStrategy(fastjet::Strategy f) {fStrategy = f;}
68	void SetRecombScheme(fastjet::RecombinationScheme f) {fRecombScheme = f;}
69	void SetAreaType(fastjet::AreaType f) {fAreaType = f;}
70
71	// Helper
72	//
73
74	// we have different cases
75	// AOD reading -> MC from AOD
76	// ESD reading -> MC from Kinematics
77	// this has to match with our selection of input events
78	enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
79	enum {kMaxJets = 4};
80	enum {kMaxCorrelation = 3};
81	enum {kMaxRadius = 5};
82
83
84	private:
85
86	AliAnalysisTaskJetCluster(const AliAnalysisTaskJetCluster&);
87	AliAnalysisTaskJetCluster& operator=(const AliAnalysisTaskJetCluster&);
88
89	Int_t GetListOfTracks(TList *list,Int_t type);
90
91	AliAODEvent *fAOD; // where we take the jets from can be input or output AOD
92
93	Bool_t fUseAODTrackInput; // take track from input AOD not from ouptu AOD
94	Bool_t fUseAODMCInput; // take MC from input AOD not from ouptu AOD
95	Bool_t fUseGlobalSelection; // Limit the eta of the generated jets
96	UInt_t fFilterMask; // filter bit for slecected tracks
97	Int_t fTrackTypeRec; // type of tracks used for FF
98	Int_t fTrackTypeGen; // type of tracks used for FF
99	Float_t fAvgTrials; // Average nimber of trials
100	Float_t fExternalWeight; // external weight
101	Float_t fRecEtaWindow; // eta window used for corraltion plots between rec and gen
102	// Fast jet
103	Double_t fRparam;
104	fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
105	fastjet::Strategy fStrategy; //= fastjet::Best;
106	fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
107	fastjet::AreaType fAreaType;
108
109	TProfile* fh1Xsec; // pythia cross section and trials
110	TH1F* fh1Trials; // trials are added
111	TH1F* fh1PtHard; // Pt har of the event...
112	TH1F* fh1PtHardNoW; // Pt har of the event without weigt
113	TH1F* fh1PtHardTrials; // Number of trials
114
115	TH1F* fh1NJetsRec; // number of reconstructed jets
116	TH1F* fh1NConstRec;// number of constiutens in leading jet
117	TH1F* fh1NConstLeadingRec;// number of constiutens in leading jet
118	TH1F* fh1PtJetsRecIn; // Jet pt for all jets
119	TH1F* fh1PtJetsLeadingRecIn; // Jet pt for all jets
120	TH1F* fh1PtJetConstRec;// pt of constituents
121	TH1F* fh1PtJetConstLeadingRec;// pt of constituents
122	TH1F* fh1PtTracksRecIn; // track pt for all tracks
123	TH1F* fh1PtTracksLeadingRecIn; // track pt for all tracks
124
125	// Randomized track histos
126	TH1F* fh1NJetsRecRan; // number of reconstructed jets from randomized
127	TH1F* fh1NConstRecRan;// number of constiutens in leading jet
128	TH1F* fh1PtJetsLeadingRecInRan; // Jet pt for all jets
129	TH1F* fh1NConstLeadingRecRan;// number of constiutens in leading jet
130	TH1F* fh1PtJetsRecInRan; // Jet pt for all jets
131
132	TH1F* fh1PtTracksGenIn; // track pt for all tracks
133
134
135	TH2F* fh2NRecJetsPt; // Number of found jets above threshold
136	TH2F* fh2NRecTracksPt; // Number of found tracks above threshold
137	TH2F* fh2NConstPt; // number of constituents vs. pt
138	TH2F* fh2NConstLeadingPt; // number of constituents vs. pt
139	TH2F* fh2JetPhiEta; // jet phi eta
140	TH2F* fh2LeadingJetPhiEta; // leading jet phi eta
141	TH2F* fh2JetEtaPt; // leading jet eta
142	TH2F* fh2LeadingJetEtaPt; // leading jet eta
143	TH2F* fh2TrackEtaPt; // track eta
144	TH2F* fh2LeadingTrackEtaPt; // leading track eta
145	TH2F* fh2JetsLeadingPhiEta; // jet phi eta
146	TH2F* fh2JetsLeadingPhiPt; // jet correlation with leading jet
147	TH2F* fh2TracksLeadingPhiEta; // track correlation with leading track
148	TH2F* fh2TracksLeadingPhiPt; // track correlation with leading track
149	TH2F* fh2TracksLeadingJetPhiPt; // track correlation with leading Jet
150	TH2F* fh2JetsLeadingPhiPtW; // jet correlation with leading jet
151	TH2F* fh2TracksLeadingPhiPtW; // track correlation with leading track
152	TH2F* fh2TracksLeadingJetPhiPtW; // track correlation with leading Jet
153	TH2F* fh2NRecJetsPtRan; // Number of found jets above threshold
154	TH2F* fh2NConstPtRan; // number of constituents vs. pt
155	TH2F* fh2NConstLeadingPtRan; // number of constituents vs. pt
156	TH2F* fh2TracksLeadingJetPhiPtRan; // track correlation with leading Jet
157	TH2F* fh2TracksLeadingJetPhiPtWRan; // track correlation with leading Jet
158	TList *fHistList; // Output list
159
160
161	ClassDef(AliAnalysisTaskJetCluster, 2) // Analysis task for standard jet analysis
162	};
163
164	#endif