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

#ifndef ALIANALYSISTASKJETRESPONSEV2_H
#define ALIANALYSISTASKJETRESPONSEV2_H

class TH1F;
class TH2F;
class TH3F;
class THnSparse;
class AliESDEvent;
class AliAODEvent;

#include "AliAnalysisTaskSE.h"
#include "AliVEvent.h"

class AliAnalysisTaskJetResponseV2 : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskJetResponseV2();
  AliAnalysisTaskJetResponseV2(const char *name);
  virtual ~AliAnalysisTaskJetResponseV2();

  virtual void     LocalInit() {Init();}
  virtual void     Init();
  virtual void     UserCreateOutputObjects();
  virtual void     UserExec(Option_t *option);
  virtual void     Terminate(const Option_t*);

  virtual Int_t      GetNInputTracks();
  virtual THnSparse* NewTHnSparseF(const char* name, UInt_t entries, UInt_t opt);
  virtual void       GetDimParams(Int_t iEntry, Bool_t hr, TString &label, Int_t &nbins, Double_t &xmin, Double_t &xmax);

  virtual AliVEvent::EOfflineTriggerTypes GetOfflineTrgMask() const { return fOfflineTrgMask; }
  virtual void     GetBranchNames(TString &branch1, TString &branch2) const { branch1 = fJetBranchName[0]; branch2 = fJetBranchName[1]; }
  virtual Bool_t   GetIsPbPb() const { return fIsPbPb; }
  virtual Int_t    GetMinContribVtx() const { return fMinContribVtx; };
  virtual Float_t  GetVtxZMin() const { return fVtxZMin; }
  virtual Float_t  GetVtxZMax() const { return fVtxZMax; }
  virtual Int_t    GetEvtClassMin() const { return fEvtClassMin; }
  virtual Int_t    GetEvtClassMax() const { return fEvtClassMax; }
  virtual Float_t  GetCentMin() const { return fCentMin; }
  virtual Float_t  GetCentMax() const { return fCentMax; }
  virtual Int_t    GetNInputTracksMin() const { return fNInputTracksMin; }
  virtual Int_t    GetNInputTracksMax() const { return fNInputTracksMax; } 
  virtual Float_t  GetJetEtaMin() const { return fJetEtaMin; }
  virtual Float_t  GetJetEtaMax() const { return fJetEtaMax; }
  virtual Float_t  GetJetPtMin() const { return fJetPtMin; }
  virtual Float_t  GetJetPtFractionMin() const { return fJetPtFractionMin; }
  virtual Int_t    GetNMatchJets() const { return fNMatchJets; }

  virtual void     SetBranchNames(const TString &branch1, const TString &branch2);
  virtual void     SetBackgroundBranch(TString &branch) { fBackgroundBranch = branch;}
  virtual void     SetIsPbPb(Bool_t b=kTRUE) { fIsPbPb = b; }
  virtual void     SetOfflineTrgMask(AliVEvent::EOfflineTriggerTypes mask) { fOfflineTrgMask = mask; }
  virtual void     SetMinContribVtx(Int_t n) { fMinContribVtx = n; }
  virtual void     SetVtxZMin(Float_t z) { fVtxZMin = z; }
  virtual void     SetVtxZMax(Float_t z) { fVtxZMax = z; }
  virtual void     SetEvtClassMin(Int_t evtClass) { fEvtClassMin = evtClass; }
  virtual void     SetEvtClassMax(Int_t evtClass) { fEvtClassMax = evtClass; }
  virtual void     SetCentMin(Float_t cent) { fCentMin = cent; }
  virtual void     SetCentMax(Float_t cent) { fCentMax = cent; }
  virtual void     SetNInputTracksMin(Int_t nTr) { fNInputTracksMin = nTr; }
  virtual void     SetNInputTracksMax(Int_t nTr) { fNInputTracksMax = nTr; }
  virtual void     SetJetEtaMin(Float_t eta) { fJetEtaMin = eta; }
  virtual void     SetJetEtaMax(Float_t eta) { fJetEtaMax = eta; }
  virtual void     SetJetPtMin(Float_t pt) { fJetPtMin = pt; }
  virtual void     SetJetPtFractionMin(Float_t frac) { fJetPtFractionMin = frac; }
  virtual void     SetNMatchJets(Int_t n) { fNMatchJets = n; }
  virtual void     SetFillEvent(Bool_t b) { fbEvent = b; }
  virtual void     SetFillJetsMismatch1(Bool_t b) { fbJetsMismatch1 = b; }
  virtual void     SetFillJetsMismatch2(Bool_t b) { fbJetsMismatch2 = b; }
  virtual void     SetFillJetsRp(Bool_t b) { fbJetsRp = b; }
  virtual void     SetFillJetsDeltaPt(Bool_t b) { fbJetsDeltaPt = b; }
  virtual void     SetFillJetsEta(Bool_t b) { fbJetsEta = b; }
  virtual void     SetFillJetsPhi(Bool_t b) { fbJetsPhi = b; }
  virtual void     SetFillJetsArea(Bool_t b) { fbJetsArea = b; }
  virtual void     SetFillJetsBeforeCut1(Bool_t b) { fbJetsBeforeCut1 = b; }
  virtual void     SetFillJetsBeforeCut2(Bool_t b) { fbJetsBeforeCut2 = b; }
  
 private:
  // ESD/AOD events
  AliESDEvent *fESD;    //! ESD object
  AliAODEvent *fAOD;    //! AOD event

  // jets to compare
  TString fJetBranchName[2]; //  name of jet branches to compare
  TList *fListJets[2];       //! jet lists
  
  TString fBackgroundBranch;

  // event selection
  Bool_t fIsPbPb;         // is Pb-Pb (fast embedding) or p-p (detector response)
  AliVEvent::EOfflineTriggerTypes fOfflineTrgMask; // mask of offline triggers to accept
  Int_t   fMinContribVtx; // minimum number of track contributors for primary vertex
  Float_t fVtxZMin;	  // lower bound on vertex z
  Float_t fVtxZMax;	  // upper bound on vertex z
  Int_t   fEvtClassMin;	  // lower bound on event class
  Int_t   fEvtClassMax;	  // upper bound on event class
  Float_t fCentMin;	  // lower bound on centrality
  Float_t fCentMax;	  // upper bound on centrality
  Int_t   fNInputTracksMin;  // lower bound of nb. of input tracks
  Int_t   fNInputTracksMax;  // upper bound of nb. of input tracks
  Float_t fJetEtaMin;     // lower bound on eta for found jets
  Float_t fJetEtaMax;     // upper bound on eta for found jets
  Float_t fJetPtMin;      // minimum jet pT
  Float_t fJetPtFractionMin; // minimum fraction for positiv match of jets
  Int_t   fNMatchJets;       // maximal nb. of jets taken for matching
  Double_t fMatchMaxDist;     // maximal distance of matching jets
  

  // output objects
  const Int_t fkNbranches;                   //! number of branches to be read
  const Int_t fkEvtClasses;                  //! number of event classes
  TList *fOutputList;                        //! output data container
  Bool_t fbEvent;                            // fill fhnEvent
  Bool_t fbJetsMismatch1;                    // fill fhnJetsMismatch1
  Bool_t fbJetsMismatch2;                    // fill fhnJetsMismatch2
  Bool_t fbJetsRp;                           // fill fhnJetsRp
  Bool_t fbJetsDeltaPt;                      // fill fhnJetsDeltaPt
  Bool_t fbJetsEta;                          // fill fhnJetsEta
  Bool_t fbJetsPhi;                          // fill fhnJetsEta
  Bool_t fbJetsArea;                         // fill fhnJetsArea
  Bool_t fbJetsBeforeCut1;                   // fill fhnJetsBeforeCut1
  Bool_t fbJetsBeforeCut2;                   // fill fhnJetsBeforeCut2
  TH1I  *fHistEvtSelection;                  //! event selection statistic
  TH1I  *fHistJetSelection;                  //! jet selection statistic
  THnSparse *fhnEvent;                       //! variables per event
  THnSparse *fhnJetsMismatch1;               //! variables per jet
  THnSparse *fhnJetsMismatch2;               //! variables per jet
  THnSparse *fhnJetsRp;                      //! variables per jet
  THnSparse *fhnJetsDeltaPt;                 //! variables per jet
  THnSparse *fhnJetsEta;                     //! variables per jet
  THnSparse *fhnJetsPhi;                     //! variables per jet
  THnSparse *fhnJetsArea;                    //! variables per jet
  THnSparse *fhnJetsBeforeCut1;               //! variables per jet before acceptance cut
  THnSparse *fhnJetsBeforeCut2;               //! variables per jet before acceptance cut
  
  AliAnalysisTaskJetResponseV2(const AliAnalysisTaskJetResponseV2&); // not implemented
  AliAnalysisTaskJetResponseV2& operator=(const AliAnalysisTaskJetResponseV2&); // not implemented

  ClassDef(AliAnalysisTaskJetResponseV2, 3);
};

#endif
Commit	Line	Data
3c6a60f7	1	#ifndef ALIANALYSISTASKJETRESPONSEV2_H
	2	#define ALIANALYSISTASKJETRESPONSEV2_H
	3
	4	class TH1F;
	5	class TH2F;
	6	class TH3F;
	7	class THnSparse;
	8	class AliESDEvent;
	9	class AliAODEvent;
	10
	11	#include "AliAnalysisTaskSE.h"
	12	#include "AliVEvent.h"
	13
	14	class AliAnalysisTaskJetResponseV2 : public AliAnalysisTaskSE {
	15	public:
	16	AliAnalysisTaskJetResponseV2();
	17	AliAnalysisTaskJetResponseV2(const char *name);
	18	virtual ~AliAnalysisTaskJetResponseV2();
	19
	20	virtual void LocalInit() {Init();}
	21	virtual void Init();
	22	virtual void UserCreateOutputObjects();
	23	virtual void UserExec(Option_t *option);
	24	virtual void Terminate(const Option_t*);
d314de7b	25
	26	virtual Int_t GetNInputTracks();
	27	virtual THnSparse* NewTHnSparseF(const char* name, UInt_t entries, UInt_t opt);
	28	virtual void GetDimParams(Int_t iEntry, Bool_t hr, TString &label, Int_t &nbins, Double_t &xmin, Double_t &xmax);
3c6a60f7	29
	30	virtual AliVEvent::EOfflineTriggerTypes GetOfflineTrgMask() const { return fOfflineTrgMask; }
	31	virtual void GetBranchNames(TString &branch1, TString &branch2) const { branch1 = fJetBranchName[0]; branch2 = fJetBranchName[1]; }
739e606a	32	virtual Bool_t GetIsPbPb() const { return fIsPbPb; }
3c6a60f7	33	virtual Int_t GetMinContribVtx() const { return fMinContribVtx; };
	34	virtual Float_t GetVtxZMin() const { return fVtxZMin; }
	35	virtual Float_t GetVtxZMax() const { return fVtxZMax; }
	36	virtual Int_t GetEvtClassMin() const { return fEvtClassMin; }
	37	virtual Int_t GetEvtClassMax() const { return fEvtClassMax; }
	38	virtual Float_t GetCentMin() const { return fCentMin; }
	39	virtual Float_t GetCentMax() const { return fCentMax; }
	40	virtual Int_t GetNInputTracksMin() const { return fNInputTracksMin; }
	41	virtual Int_t GetNInputTracksMax() const { return fNInputTracksMax; }
3c6a60f7	42	virtual Float_t GetJetEtaMin() const { return fJetEtaMin; }
	43	virtual Float_t GetJetEtaMax() const { return fJetEtaMax; }
	44	virtual Float_t GetJetPtMin() const { return fJetPtMin; }
	45	virtual Float_t GetJetPtFractionMin() const { return fJetPtFractionMin; }
	46	virtual Int_t GetNMatchJets() const { return fNMatchJets; }
	47
	48	virtual void SetBranchNames(const TString &branch1, const TString &branch2);
d314de7b	49	virtual void SetBackgroundBranch(TString &branch) { fBackgroundBranch = branch;}
739e606a	50	virtual void SetIsPbPb(Bool_t b=kTRUE) { fIsPbPb = b; }
3c6a60f7	51	virtual void SetOfflineTrgMask(AliVEvent::EOfflineTriggerTypes mask) { fOfflineTrgMask = mask; }
	52	virtual void SetMinContribVtx(Int_t n) { fMinContribVtx = n; }
	53	virtual void SetVtxZMin(Float_t z) { fVtxZMin = z; }
	54	virtual void SetVtxZMax(Float_t z) { fVtxZMax = z; }
	55	virtual void SetEvtClassMin(Int_t evtClass) { fEvtClassMin = evtClass; }
	56	virtual void SetEvtClassMax(Int_t evtClass) { fEvtClassMax = evtClass; }
	57	virtual void SetCentMin(Float_t cent) { fCentMin = cent; }
	58	virtual void SetCentMax(Float_t cent) { fCentMax = cent; }
	59	virtual void SetNInputTracksMin(Int_t nTr) { fNInputTracksMin = nTr; }
	60	virtual void SetNInputTracksMax(Int_t nTr) { fNInputTracksMax = nTr; }
3c6a60f7	61	virtual void SetJetEtaMin(Float_t eta) { fJetEtaMin = eta; }
	62	virtual void SetJetEtaMax(Float_t eta) { fJetEtaMax = eta; }
	63	virtual void SetJetPtMin(Float_t pt) { fJetPtMin = pt; }
739e606a	64	virtual void SetJetPtFractionMin(Float_t frac) { fJetPtFractionMin = frac; }
3c6a60f7	65	virtual void SetNMatchJets(Int_t n) { fNMatchJets = n; }
d314de7b	66	virtual void SetFillEvent(Bool_t b) { fbEvent = b; }
	67	virtual void SetFillJetsMismatch1(Bool_t b) { fbJetsMismatch1 = b; }
	68	virtual void SetFillJetsMismatch2(Bool_t b) { fbJetsMismatch2 = b; }
	69	virtual void SetFillJetsRp(Bool_t b) { fbJetsRp = b; }
	70	virtual void SetFillJetsDeltaPt(Bool_t b) { fbJetsDeltaPt = b; }
	71	virtual void SetFillJetsEta(Bool_t b) { fbJetsEta = b; }
	72	virtual void SetFillJetsPhi(Bool_t b) { fbJetsPhi = b; }
	73	virtual void SetFillJetsArea(Bool_t b) { fbJetsArea = b; }
	74	virtual void SetFillJetsBeforeCut1(Bool_t b) { fbJetsBeforeCut1 = b; }
	75	virtual void SetFillJetsBeforeCut2(Bool_t b) { fbJetsBeforeCut2 = b; }
	76
3c6a60f7	77	private:
	78	// ESD/AOD events
	79	AliESDEvent *fESD; //! ESD object
	80	AliAODEvent *fAOD; //! AOD event
d314de7b	81
3c6a60f7	82	// jets to compare
	83	TString fJetBranchName[2]; // name of jet branches to compare
	84	TList *fListJets[2]; //! jet lists
739e606a	85
d314de7b	86	TString fBackgroundBranch;
d314de7b	87
3c6a60f7	88	// event selection
d314de7b	89	Bool_t fIsPbPb; // is Pb-Pb (fast embedding) or p-p (detector response)
3c6a60f7	90	AliVEvent::EOfflineTriggerTypes fOfflineTrgMask; // mask of offline triggers to accept
	91	Int_t fMinContribVtx; // minimum number of track contributors for primary vertex
	92	Float_t fVtxZMin; // lower bound on vertex z
	93	Float_t fVtxZMax; // upper bound on vertex z
	94	Int_t fEvtClassMin; // lower bound on event class
	95	Int_t fEvtClassMax; // upper bound on event class
	96	Float_t fCentMin; // lower bound on centrality
	97	Float_t fCentMax; // upper bound on centrality
	98	Int_t fNInputTracksMin; // lower bound of nb. of input tracks
	99	Int_t fNInputTracksMax; // upper bound of nb. of input tracks
3c6a60f7	100	Float_t fJetEtaMin; // lower bound on eta for found jets
	101	Float_t fJetEtaMax; // upper bound on eta for found jets
	102	Float_t fJetPtMin; // minimum jet pT
	103	Float_t fJetPtFractionMin; // minimum fraction for positiv match of jets
	104	Int_t fNMatchJets; // maximal nb. of jets taken for matching
	105	Double_t fMatchMaxDist; // maximal distance of matching jets
	106
	107
	108	// output objects
	109	const Int_t fkNbranches; //! number of branches to be read
	110	const Int_t fkEvtClasses; //! number of event classes
	111	TList *fOutputList; //! output data container
d314de7b	112	Bool_t fbEvent; // fill fhnEvent
	113	Bool_t fbJetsMismatch1; // fill fhnJetsMismatch1
	114	Bool_t fbJetsMismatch2; // fill fhnJetsMismatch2
	115	Bool_t fbJetsRp; // fill fhnJetsRp
	116	Bool_t fbJetsDeltaPt; // fill fhnJetsDeltaPt
	117	Bool_t fbJetsEta; // fill fhnJetsEta
	118	Bool_t fbJetsPhi; // fill fhnJetsEta
	119	Bool_t fbJetsArea; // fill fhnJetsArea
	120	Bool_t fbJetsBeforeCut1; // fill fhnJetsBeforeCut1
	121	Bool_t fbJetsBeforeCut2; // fill fhnJetsBeforeCut2
3c6a60f7	122	TH1I *fHistEvtSelection; //! event selection statistic
d314de7b	123	TH1I *fHistJetSelection; //! jet selection statistic
3c6a60f7	124	THnSparse *fhnEvent; //! variables per event
d314de7b	125	THnSparse *fhnJetsMismatch1; //! variables per jet
d314de7b	126	THnSparse *fhnJetsMismatch2; //! variables per jet
3c6a60f7	127	THnSparse *fhnJetsRp; //! variables per jet
	128	THnSparse *fhnJetsDeltaPt; //! variables per jet
	129	THnSparse *fhnJetsEta; //! variables per jet
d314de7b	130	THnSparse *fhnJetsPhi; //! variables per jet
3c6a60f7	131	THnSparse *fhnJetsArea; //! variables per jet
	132	THnSparse *fhnJetsBeforeCut1; //! variables per jet before acceptance cut
	133	THnSparse *fhnJetsBeforeCut2; //! variables per jet before acceptance cut
	134
	135	AliAnalysisTaskJetResponseV2(const AliAnalysisTaskJetResponseV2&); // not implemented
	136	AliAnalysisTaskJetResponseV2& operator=(const AliAnalysisTaskJetResponseV2&); // not implemented
	137
d314de7b	138	ClassDef(AliAnalysisTaskJetResponseV2, 3);
3c6a60f7	139	};
	140
	141	#endif