[u/mrichter/AliRoot.git] / PWGGA / EMCALJetTasks / AliAnalysisTaskEmcalJet.h

#ifndef ALIANALYSISTASKEMCALJET_H
#define ALIANALYSISTASKEMCALJET_H

// $Id$

class TClonesArray;
class TString;
class TList;
class AliVParticle;
class AliVCluster;
class AliEmcalJet;

#include "AliAnalysisTaskEmcal.h"

class AliAnalysisTaskEmcalJet : public AliAnalysisTaskEmcal {
 public:

  AliAnalysisTaskEmcalJet();
  AliAnalysisTaskEmcalJet(const char *name);
  AliAnalysisTaskEmcalJet(const char *name, Bool_t histo); 
  virtual ~AliAnalysisTaskEmcalJet();

  virtual void                UserExec(Option_t *option);
  virtual void                Init();

  void                        SetEtaLimits(Float_t min, Float_t max)               { fMinEta = min, fMaxEta = max ; }
  void                        SetInitialized(Bool_t ini = kTRUE)                   { fInitialized    = ini        ; }
  void                        SetJetAreaCut(Float_t cut)                           { fJetAreaCut     = cut        ; }
  void                        SetJetPtCut(Float_t cut)                             { fJetPtCut       = cut        ; }
  void                        SetJetRadius(Float_t r)                              { fJetRadius      = r          ; } 
  void                        SetJetsName(const char *n)                           { fJetsName       = n          ; }
  void                        SetMaxClusterPt(Float_t b)                           { fMaxClusterPt  = b           ; }
  void                        SetMaxTrackPt(Float_t b)                             { fMaxTrackPt = b              ; }
  void                        SetPhiLimits(Float_t min, Float_t max)               { fMinPhi = min, fMaxPhi = max ; }
  void                        SetPtBiasJetClus(Float_t b)                          { fPtBiasJetClus  = b          ; }
  void                        SetPtBiasJetTrack(Float_t b)                         { fPtBiasJetTrack = b          ; }
 
 protected:

  Bool_t                      AcceptJet(AliEmcalJet* jet, Bool_t bias = kTRUE, Bool_t upCut = kTRUE)   const;
  Bool_t                      AcceptBiasJet(AliEmcalJet* jet)                                          const;
  Bool_t                      IsJetTrack(AliEmcalJet* jet, Int_t itrack, Bool_t sorted = kTRUE)        const;
  Bool_t                      IsJetCluster(AliEmcalJet* jet, Int_t iclus, Bool_t sorted = kTRUE)       const;

  virtual Bool_t              RetrieveEventObjects();

  Float_t                     fJetRadius;                  // jet radius
  TString                     fJetsName;                   // name of jet collection
  Float_t                     fPtBiasJetTrack;             // select jets with a minimum pt track
  Float_t                     fPtBiasJetClus;              // select jets with a minimum pt cluster
  Float_t                     fJetPtCut;                   // cut on jet pt
  Float_t                     fJetAreaCut;                 // cut on jet area
  Float_t                     fMinEta;                     // minimum eta jet acceptance
  Float_t                     fMaxEta;                     // maximum eta jet acceptance
  Float_t                     fMinPhi;                     // minimum phi jet acceptance
  Float_t                     fMaxPhi;                     // maximum phi jet acceptance  
  Float_t                     fMaxClusterPt;               // maximum cluster constituent pt to accept the jet
  Float_t                     fMaxTrackPt;                 // maximum track constituent pt to accept the jet
  TClonesArray               *fJets;                       //!jets

 private:
  AliAnalysisTaskEmcalJet(const AliAnalysisTaskEmcalJet&);            // not implemented
  AliAnalysisTaskEmcalJet &operator=(const AliAnalysisTaskEmcalJet&); // not implemented

  ClassDef(AliAnalysisTaskEmcalJet, 3) // EMCAL Jet base analysis task
};
#endif
Commit	Line	Data
a546db49	1	#ifndef ALIANALYSISTASKEMCALJET_H
	2	#define ALIANALYSISTASKEMCALJET_H
	3
0810af47	4	// $Id$
a546db49	5
	6	class TClonesArray;
	7	class TString;
	8	class TList;
	9	class AliVParticle;
	10	class AliVCluster;
	11	class AliEmcalJet;
	12
	13	#include "AliAnalysisTaskEmcal.h"
	14
	15	class AliAnalysisTaskEmcalJet : public AliAnalysisTaskEmcal {
	16	public:
	17
	18	AliAnalysisTaskEmcalJet();
	19	AliAnalysisTaskEmcalJet(const char *name);
	20	AliAnalysisTaskEmcalJet(const char *name, Bool_t histo);
	21	virtual ~AliAnalysisTaskEmcalJet();
	22
1f6fff78	23	virtual void UserExec(Option_t *option);
	24	virtual void Init();
	25
411b11e0	26	void SetEtaLimits(Float_t min, Float_t max) { fMinEta = min, fMaxEta = max ; }
411b11e0	27	void SetInitialized(Bool_t ini = kTRUE) { fInitialized = ini ; }
a546db49	28	void SetJetAreaCut(Float_t cut) { fJetAreaCut = cut ; }
	29	void SetJetPtCut(Float_t cut) { fJetPtCut = cut ; }
	30	void SetJetRadius(Float_t r) { fJetRadius = r ; }
	31	void SetJetsName(const char *n) { fJetsName = n ; }
411b11e0	32	void SetMaxClusterPt(Float_t b) { fMaxClusterPt = b ; }
	33	void SetMaxTrackPt(Float_t b) { fMaxTrackPt = b ; }
	34	void SetPhiLimits(Float_t min, Float_t max) { fMinPhi = min, fMaxPhi = max ; }
a546db49	35	void SetPtBiasJetClus(Float_t b) { fPtBiasJetClus = b ; }
a546db49	36	void SetPtBiasJetTrack(Float_t b) { fPtBiasJetTrack = b ; }
411b11e0	37
a546db49	38	protected:
a546db49	39
2bddb6ae	40	Bool_t AcceptJet(AliEmcalJet* jet, Bool_t bias = kTRUE, Bool_t upCut = kTRUE) const;
	41	Bool_t AcceptBiasJet(AliEmcalJet* jet) const;
	42	Bool_t IsJetTrack(AliEmcalJet* jet, Int_t itrack, Bool_t sorted = kTRUE) const;
	43	Bool_t IsJetCluster(AliEmcalJet* jet, Int_t iclus, Bool_t sorted = kTRUE) const;
a546db49	44
	45	virtual Bool_t RetrieveEventObjects();
	46
	47	Float_t fJetRadius; // jet radius
	48	TString fJetsName; // name of jet collection
	49	Float_t fPtBiasJetTrack; // select jets with a minimum pt track
	50	Float_t fPtBiasJetClus; // select jets with a minimum pt cluster
	51	Float_t fJetPtCut; // cut on jet pt
	52	Float_t fJetAreaCut; // cut on jet area
411b11e0	53	Float_t fMinEta; // minimum eta jet acceptance
	54	Float_t fMaxEta; // maximum eta jet acceptance
	55	Float_t fMinPhi; // minimum phi jet acceptance
	56	Float_t fMaxPhi; // maximum phi jet acceptance
	57	Float_t fMaxClusterPt; // maximum cluster constituent pt to accept the jet
	58	Float_t fMaxTrackPt; // maximum track constituent pt to accept the jet
a546db49	59	TClonesArray *fJets; //!jets
	60
	61	private:
	62	AliAnalysisTaskEmcalJet(const AliAnalysisTaskEmcalJet&); // not implemented
	63	AliAnalysisTaskEmcalJet &operator=(const AliAnalysisTaskEmcalJet&); // not implemented
	64
00a4f405	65	ClassDef(AliAnalysisTaskEmcalJet, 3) // EMCAL Jet base analysis task
a546db49	66	};
a546db49	67	#endif