[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 AliAODExtension;
class AliAODJet;
class AliGenPythiaEventHeader;
class AliCFManager;
class AliAODJetEventBackground;
class AliJetFinder;
class TList;
class TChain;
class TH2F;
class TH1F;
class TH3F;
class TProfile;
class TRefArray;


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 SetTrackPtCut(Float_t x){fTrackPtCut = x;}
    virtual void SetFilterMask(UInt_t i){fFilterMask = i;}
    
    virtual void SetNSkipLeadingRan(Int_t x){fNSkipLeadingRan = x;}

    virtual void SetJetOutputBranch(const char *c){fNonStdBranch = c;}
    virtual void SetJetOutputFile(const char *c){fNonStdFile = c;}
    virtual void SetJetOutputMinPt(Float_t x){fJetOutputMinPt = x;}

    // 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;}
    void SetGhostArea(Double_t f) {fGhostArea = f;}
    void SetActiveAreaRepeats(Int_t f) {fActiveAreaRepeats = f;}
    void SetGhostEtamax(Double_t f) {fGhostEtamax = 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
    AliAODExtension *fAODExtension;       // ! AOD extension in case we write a non-sdt branch to a separate file and the aod is standard
    TRefArray       *fRef;               // ! trefarray for track references within the jet
    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 
    Int_t         fNSkipLeadingRan;        // number of leading tracks to be skipped in the randomized event
    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       fTrackPtCut;            // minimum track pt to be accepted
    Float_t       fJetOutputMinPt;        // minimum p_t for jets to be written out

    // output configurartion
    TString       fNonStdBranch;      // the name of the non-std branch name, if empty no branch is filled
    TString       fNonStdFile;        // The optional name of the output file the non-std brnach is written to
    

    // 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; 
    Double_t fGhostArea;
    Int_t fActiveAreaRepeats;
    Double_t fGhostEtamax;
    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
    TH1F*         fh1BiARandomCones[3]; //! Residual distribtion from reandom cones on real event
    TH1F*         fh1BiARandomConesRan[3]; //! Residual distribtion from reandom cones on random event


    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, 6) 
};
 
#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;
54424110	22	class AliAODExtension;
dbf053f0	23	class AliAODJet;
	24	class AliGenPythiaEventHeader;
	25	class AliCFManager;
d6e66a82	26	class AliAODJetEventBackground;
d6e66a82	27	class AliJetFinder;
dbf053f0	28	class TList;
	29	class TChain;
	30	class TH2F;
	31	class TH1F;
	32	class TH3F;
	33	class TProfile;
54424110	34	class TRefArray;
dbf053f0	35
	36
	37	class AliAnalysisTaskJetCluster : public AliAnalysisTaskSE
	38	{
	39	public:
	40	AliAnalysisTaskJetCluster();
	41	AliAnalysisTaskJetCluster(const char* name);
54424110	42	virtual ~AliAnalysisTaskJetCluster();
dbf053f0	43	// Implementation of interface methods
	44	virtual void UserCreateOutputObjects();
	45	virtual void Init();
	46	virtual void LocalInit() { Init(); }
	47	virtual void UserExec(Option_t *option);
	48	virtual void Terminate(Option_t *option);
	49	virtual Bool_t Notify();
	50
d6e66a82	51
d6e66a82	52
dbf053f0	53	virtual void SetUseGlobalSelection(Bool_t b){fUseGlobalSelection = b;}
	54	virtual void SetAODTrackInput(Bool_t b){fUseAODTrackInput = b;}
	55	virtual void SetAODMCInput(Bool_t b){fUseAODMCInput = b;}
	56	virtual void SetRecEtaWindow(Float_t f){fRecEtaWindow = f;}
	57	virtual void SetTrackTypeGen(Int_t i){fTrackTypeGen = i;}
	58	virtual void SetTrackTypeRec(Int_t i){fTrackTypeRec = i;}
54424110	59	virtual void SetTrackPtCut(Float_t x){fTrackPtCut = x;}
dbf053f0	60	virtual void SetFilterMask(UInt_t i){fFilterMask = i;}
d6e66a82	61
bd80a748	62	virtual void SetNSkipLeadingRan(Int_t x){fNSkipLeadingRan = x;}
bd80a748	63
54424110	64	virtual void SetJetOutputBranch(const char *c){fNonStdBranch = c;}
	65	virtual void SetJetOutputFile(const char *c){fNonStdFile = c;}
	66	virtual void SetJetOutputMinPt(Float_t x){fJetOutputMinPt = x;}
	67
dbf053f0	68	// for Fast Jet
	69	fastjet::JetAlgorithm GetAlgorithm() const {return fAlgorithm;}
	70	fastjet::Strategy GetStrategy() const {return fStrategy;}
	71	fastjet::RecombinationScheme GetRecombScheme() const {return fRecombScheme;}
	72	fastjet::AreaType GetAreaType() const {return fAreaType;}
	73	// Setters
	74	void SetRparam(Double_t f) {fRparam = f;}
	75	void SetAlgorithm(fastjet::JetAlgorithm f) {fAlgorithm = f;}
	76	void SetStrategy(fastjet::Strategy f) {fStrategy = f;}
	77	void SetRecombScheme(fastjet::RecombinationScheme f) {fRecombScheme = f;}
	78	void SetAreaType(fastjet::AreaType f) {fAreaType = f;}
d6e66a82	79	void SetGhostArea(Double_t f) {fGhostArea = f;}
	80	void SetActiveAreaRepeats(Int_t f) {fActiveAreaRepeats = f;}
	81	void SetGhostEtamax(Double_t f) {fGhostEtamax = f;}
	82
	83
dbf053f0	84
	85	// Helper
	86	//
	87
	88	// we have different cases
	89	// AOD reading -> MC from AOD
	90	// ESD reading -> MC from Kinematics
	91	// this has to match with our selection of input events
	92	enum {kTrackUndef = 0, kTrackAOD, kTrackKineAll,kTrackKineCharged, kTrackAODMCAll, kTrackAODMCCharged, kTrackAODMCChargedAcceptance};
	93	enum {kMaxJets = 4};
	94	enum {kMaxCorrelation = 3};
	95	enum {kMaxRadius = 5};
	96
	97
	98	private:
	99
	100	AliAnalysisTaskJetCluster(const AliAnalysisTaskJetCluster&);
	101	AliAnalysisTaskJetCluster& operator=(const AliAnalysisTaskJetCluster&);
	102
	103	Int_t GetListOfTracks(TList *list,Int_t type);
	104
54424110	105	AliAODEvent *fAOD; // ! where we take the jets from can be input or output AOD
	106	AliAODExtension *fAODExtension; // ! AOD extension in case we write a non-sdt branch to a separate file and the aod is standard
	107	TRefArray *fRef; // ! trefarray for track references within the jet
dbf053f0	108	Bool_t fUseAODTrackInput; // take track from input AOD not from ouptu AOD
	109	Bool_t fUseAODMCInput; // take MC from input AOD not from ouptu AOD
	110	Bool_t fUseGlobalSelection; // Limit the eta of the generated jets
54424110	111	UInt_t fFilterMask; // filter bit for slecected tracks
dbf053f0	112	Int_t fTrackTypeRec; // type of tracks used for FF
dbf053f0	113	Int_t fTrackTypeGen; // type of tracks used for FF
bd80a748	114	Int_t fNSkipLeadingRan; // number of leading tracks to be skipped in the randomized event
dbf053f0	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
54424110	118	Float_t fTrackPtCut; // minimum track pt to be accepted
	119	Float_t fJetOutputMinPt; // minimum p_t for jets to be written out
	120
	121	// output configurartion
	122	TString fNonStdBranch; // the name of the non-std branch name, if empty no branch is filled
	123	TString fNonStdFile; // The optional name of the output file the non-std brnach is written to
	124
	125
dbf053f0	126	// Fast jet
	127	Double_t fRparam;
	128	fastjet::JetAlgorithm fAlgorithm; //fastjet::kt_algorithm
	129	fastjet::Strategy fStrategy; //= fastjet::Best;
	130	fastjet::RecombinationScheme fRecombScheme; // = fastjet::BIpt_scheme;
	131	fastjet::AreaType fAreaType;
d6e66a82	132	Double_t fGhostArea;
	133	Int_t fActiveAreaRepeats;
	134	Double_t fGhostEtamax;
	135	TProfile* fh1Xsec; //! pythia cross section and trials
	136	TH1F* fh1Trials; //! trials are added
	137	TH1F* fh1PtHard; //! Pt har of the event...
	138	TH1F* fh1PtHardNoW; //! Pt har of the event without weigt
	139	TH1F* fh1PtHardTrials; //! Number of trials
	140
	141	TH1F* fh1NJetsRec; //! number of reconstructed jets
	142	TH1F* fh1NConstRec;//! number of constiutens in leading jet
	143	TH1F* fh1NConstLeadingRec;//! number of constiutens in leading jet
	144	TH1F* fh1PtJetsRecIn; //! Jet pt for all jets
	145	TH1F* fh1PtJetsLeadingRecIn; //! Jet pt for all jets
	146	TH1F* fh1PtJetConstRec;//! pt of constituents
dbf053f0	147	TH1F* fh1PtJetConstLeadingRec;// pt of constituents
d6e66a82	148	TH1F* fh1PtTracksRecIn; //! track pt for all tracks
d6e66a82	149	TH1F* fh1PtTracksLeadingRecIn; //! track pt for all tracks
dbf053f0	150
dbf053f0	151	// Randomized track histos
d6e66a82	152	TH1F* fh1NJetsRecRan; //! number of reconstructed jets from randomized
	153	TH1F* fh1NConstRecRan;//! number of constiutens in leading jet
	154	TH1F* fh1PtJetsLeadingRecInRan; //! Jet pt for all jets
	155	TH1F* fh1NConstLeadingRecRan;//! number of constiutens in leading jet
	156	TH1F* fh1PtJetsRecInRan; //! Jet pt for all jets
	157
	158	TH1F* fh1PtTracksGenIn; //! track pt for all tracks
	159	TH1F* fh1Nch; //! charged particle mult
82a35b1a	160	TH1F* fh1BiARandomCones[3]; //! Residual distribtion from reandom cones on real event
	161	TH1F* fh1BiARandomConesRan[3]; //! Residual distribtion from reandom cones on random event
	162
d6e66a82	163
	164	TH2F* fh2NRecJetsPt; //! Number of found jets above threshold
	165	TH2F* fh2NRecTracksPt; //! Number of found tracks above threshold
	166	TH2F* fh2NConstPt; //! number of constituents vs. pt
	167	TH2F* fh2NConstLeadingPt; //! number of constituents vs. pt
	168	TH2F* fh2JetPhiEta; //! jet phi eta
	169	TH2F* fh2LeadingJetPhiEta; //! leading jet phi eta
	170	TH2F* fh2JetEtaPt; //! leading jet eta
	171	TH2F* fh2LeadingJetEtaPt; //! leading jet eta
	172	TH2F* fh2TrackEtaPt; //! track eta
	173	TH2F* fh2LeadingTrackEtaPt; //! leading track eta
	174	TH2F* fh2JetsLeadingPhiEta; //! jet phi eta
	175	TH2F* fh2JetsLeadingPhiPt; //! jet correlation with leading jet
	176	TH2F* fh2TracksLeadingPhiEta; //! track correlation with leading track
	177	TH2F* fh2TracksLeadingPhiPt; //! track correlation with leading track
	178	TH2F* fh2TracksLeadingJetPhiPt; //! track correlation with leading Jet
	179	TH2F* fh2JetsLeadingPhiPtW; //! jet correlation with leading jet
	180	TH2F* fh2TracksLeadingPhiPtW; //! track correlation with leading track
	181	TH2F* fh2TracksLeadingJetPhiPtW; //! track correlation with leading Jet
	182	TH2F* fh2NRecJetsPtRan; //! Number of found jets above threshold
	183	TH2F* fh2NConstPtRan; //! number of constituents vs. pt
	184	TH2F* fh2NConstLeadingPtRan; //! number of constituents vs. pt
	185	TH2F* fh2PtNch; //! p_T of cluster vs. multiplicity,
	186	TH2F* fh2PtNchRan; //! p_T of cluster vs. multiplicity,random
	187	TH2F* fh2PtNchN; //! p_T of cluster vs. multiplicity, weigthed with constituents
	188	TH2F* fh2PtNchNRan; //! p_T of cluster vs. multiplicity, weigthed with constituents random
	189	TH2F* fh2TracksLeadingJetPhiPtRan; //! track correlation with leading Jet
	190	TH2F* fh2TracksLeadingJetPhiPtWRan; //! track correlation with leading Jet
82a35b1a	191
dbf053f0	192	TList *fHistList; // Output list
	193
	194
82a35b1a	195	ClassDef(AliAnalysisTaskJetCluster, 6)
dbf053f0	196	};
	197
	198	#endif