[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;}

    // 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
    TH1F*         fh1Nch;            // charged particle mult

    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*         fh2PtNch;               // p_T of cluster vs. multiplicity,
    TH2F*         fh2PtNchRan;            // p_T of cluster vs. multiplicity,random
    TH2F*         fh2PtNchN;               // p_T of cluster vs. multiplicity, weigthed with constituents
    TH2F*         fh2PtNchNRan;            // p_T of cluster vs. multiplicity, weigthed with constituents random
    TH2F*         fh2TracksLeadingJetPhiPtRan; // track correlation with leading Jet
    TH2F*         fh2TracksLeadingJetPhiPtWRan; // track correlation with leading Jet
    TList *fHistList; // Output list
   

    ClassDef(AliAnalysisTaskJetCluster, 3) // 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;}
dbf053f0	56
	57	// for Fast Jet
	58	fastjet::JetAlgorithm GetAlgorithm() const {return fAlgorithm;}
	59	fastjet::Strategy GetStrategy() const {return fStrategy;}
	60	fastjet::RecombinationScheme GetRecombScheme() const {return fRecombScheme;}
	61	fastjet::AreaType GetAreaType() const {return fAreaType;}
	62	// Setters
	63	void SetRparam(Double_t f) {fRparam = f;}
	64	void SetAlgorithm(fastjet::JetAlgorithm f) {fAlgorithm = f;}
	65	void SetStrategy(fastjet::Strategy f) {fStrategy = f;}
	66	void SetRecombScheme(fastjet::RecombinationScheme f) {fRecombScheme = f;}
	67	void SetAreaType(fastjet::AreaType f) {fAreaType = f;}
	68
	69	// Helper
	70	//
	71
	72	// we have different cases
	73	// AOD reading -> MC from AOD
	74	// ESD reading -> MC from Kinematics
	75	// this has to match with our selection of input events
	76	enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
	77	enum {kMaxJets = 4};
	78	enum {kMaxCorrelation = 3};
	79	enum {kMaxRadius = 5};
	80
	81
	82	private:
	83
	84	AliAnalysisTaskJetCluster(const AliAnalysisTaskJetCluster&);
	85	AliAnalysisTaskJetCluster& operator=(const AliAnalysisTaskJetCluster&);
	86
	87	Int_t GetListOfTracks(TList *list,Int_t type);
	88
	89	AliAODEvent *fAOD; // where we take the jets from can be input or output AOD
	90
	91	Bool_t fUseAODTrackInput; // take track from input AOD not from ouptu AOD
	92	Bool_t fUseAODMCInput; // take MC from input AOD not from ouptu AOD
	93	Bool_t fUseGlobalSelection; // Limit the eta of the generated jets
	94	UInt_t fFilterMask; // filter bit for slecected tracks
	95	Int_t fTrackTypeRec; // type of tracks used for FF
	96	Int_t fTrackTypeGen; // type of tracks used for FF
	97	Float_t fAvgTrials; // Average nimber of trials
	98	Float_t fExternalWeight; // external weight
	99	Float_t fRecEtaWindow; // eta window used for corraltion plots between rec and gen
	100	// Fast jet
	101	Double_t fRparam;
	102	fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
	103	fastjet::Strategy fStrategy; //= fastjet::Best;
	104	fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
	105	fastjet::AreaType fAreaType;
	106
	107	TProfile* fh1Xsec; // pythia cross section and trials
	108	TH1F* fh1Trials; // trials are added
	109	TH1F* fh1PtHard; // Pt har of the event...
	110	TH1F* fh1PtHardNoW; // Pt har of the event without weigt
	111	TH1F* fh1PtHardTrials; // Number of trials
	112
	113	TH1F* fh1NJetsRec; // number of reconstructed jets
	114	TH1F* fh1NConstRec;// number of constiutens in leading jet
	115	TH1F* fh1NConstLeadingRec;// number of constiutens in leading jet
	116	TH1F* fh1PtJetsRecIn; // Jet pt for all jets
	117	TH1F* fh1PtJetsLeadingRecIn; // Jet pt for all jets
	118	TH1F* fh1PtJetConstRec;// pt of constituents
	119	TH1F* fh1PtJetConstLeadingRec;// pt of constituents
120	TH1F* fh1PtTracksRecIn; // track pt for all tracks
121	TH1F* fh1PtTracksLeadingRecIn; // track pt for all tracks
122
123	// Randomized track histos
124	TH1F* fh1NJetsRecRan; // number of reconstructed jets from randomized
125	TH1F* fh1NConstRecRan;// number of constiutens in leading jet
126	TH1F* fh1PtJetsLeadingRecInRan; // Jet pt for all jets
127	TH1F* fh1NConstLeadingRecRan;// number of constiutens in leading jet
128	TH1F* fh1PtJetsRecInRan; // Jet pt for all jets
129
130	TH1F* fh1PtTracksGenIn; // track pt for all tracks
a0d08b6d	131	TH1F* fh1Nch; // charged particle mult
dbf053f0	132
	133	TH2F* fh2NRecJetsPt; // Number of found jets above threshold
	134	TH2F* fh2NRecTracksPt; // Number of found tracks above threshold
	135	TH2F* fh2NConstPt; // number of constituents vs. pt
	136	TH2F* fh2NConstLeadingPt; // number of constituents vs. pt
	137	TH2F* fh2JetPhiEta; // jet phi eta
	138	TH2F* fh2LeadingJetPhiEta; // leading jet phi eta
	139	TH2F* fh2JetEtaPt; // leading jet eta
	140	TH2F* fh2LeadingJetEtaPt; // leading jet eta
	141	TH2F* fh2TrackEtaPt; // track eta
	142	TH2F* fh2LeadingTrackEtaPt; // leading track eta
	143	TH2F* fh2JetsLeadingPhiEta; // jet phi eta
	144	TH2F* fh2JetsLeadingPhiPt; // jet correlation with leading jet
	145	TH2F* fh2TracksLeadingPhiEta; // track correlation with leading track
	146	TH2F* fh2TracksLeadingPhiPt; // track correlation with leading track
	147	TH2F* fh2TracksLeadingJetPhiPt; // track correlation with leading Jet
	148	TH2F* fh2JetsLeadingPhiPtW; // jet correlation with leading jet
	149	TH2F* fh2TracksLeadingPhiPtW; // track correlation with leading track
	150	TH2F* fh2TracksLeadingJetPhiPtW; // track correlation with leading Jet
	151	TH2F* fh2NRecJetsPtRan; // Number of found jets above threshold
	152	TH2F* fh2NConstPtRan; // number of constituents vs. pt
	153	TH2F* fh2NConstLeadingPtRan; // number of constituents vs. pt
a0d08b6d	154	TH2F* fh2PtNch; // p_T of cluster vs. multiplicity,
	155	TH2F* fh2PtNchRan; // p_T of cluster vs. multiplicity,random
	156	TH2F* fh2PtNchN; // p_T of cluster vs. multiplicity, weigthed with constituents
	157	TH2F* fh2PtNchNRan; // p_T of cluster vs. multiplicity, weigthed with constituents random
dbf053f0	158	TH2F* fh2TracksLeadingJetPhiPtRan; // track correlation with leading Jet
	159	TH2F* fh2TracksLeadingJetPhiPtWRan; // track correlation with leading Jet
	160	TList *fHistList; // Output list
	161
	162
a0d08b6d	163	ClassDef(AliAnalysisTaskJetCluster, 3) // Analysis task for standard jet analysis
dbf053f0	164	};
	165
	166	#endif