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

    // 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 
    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; 

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