AliAnaInsideClusterInvariantMass: Finalize the factorization of histogram filling...
[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / AliJetContainer.h
CommitLineData
421b12a3 1#ifndef AliJetContainer_H
2#define AliJetContainer_H
3
4//
5// container with name, TClonesArray and cuts for jets
6//
7
8class TObjArray;
9class TClonesArray;
10class TString;
11class TList;
12class AliEmcalParticle;
13class AliMCParticle;
14class AliVCluster;
15class AliVTrack;
16class AliVParticle;
17class AliVCaloCells;
18class TH1F;
19class AliEMCALGeometry;
20class AliEmcalJet;
21class AliVEvent;
22
23#include "Rtypes.h"
24#include <TArrayS.h>
25#include "TString.h"
26#include "AliLog.h"
27#include "AliRhoParameter.h"
28#include "AliEmcalContainer.h"
29
30
31class AliJetContainer : public AliEmcalContainer {
32 public:
33 enum JetAcceptanceType {
34 kTPC = 0, // TPC acceptance
35 kEMCAL = 1, // EMCal acceptance
4c9493a7 36 kUser = 2 // User defined acceptance
421b12a3 37 };
38
39
40 AliJetContainer();
41 AliJetContainer(const char *name);
42 virtual ~AliJetContainer();
43
44 void SetJetArray(AliVEvent *event);
45 void SetEMCALGeometry();
46 void SetEMCALGeometry(AliEMCALGeometry *p) {fGeom = p;}
47 void LoadRho(AliVEvent *event);
48
49 void SetJetAcceptanceType(JetAcceptanceType type) { fJetAcceptanceType = type ; }
50
51 void ResetCuts();
52
53 void SetJetEtaPhiEMCAL() ;
54 void SetJetEtaPhiTPC() ;
56f370b0 55 void SetRunNumber(Int_t r) { fRunNumber = r; }
421b12a3 56
57 void SetJetEtaLimits(Float_t min, Float_t max) { fJetMinEta = min, fJetMaxEta = max ; }
58 void SetJetPhiLimits(Float_t min, Float_t max) { fJetMinPhi = min, fJetMaxPhi = max ; }
59 void SetJetAreaCut(Float_t cut) { fJetAreaCut = cut ; }
60 void SetPercAreaCut(Float_t p) { if(fJetRadius==0.) AliWarning("JetRadius not set. Area cut will be 0"); fJetAreaCut = p*TMath::Pi()*fJetRadius*fJetRadius; }
61
62 void SetAreaEmcCut(Double_t a = 0.99) { fAreaEmcCut = a ; }
63 void SetJetPtCut(Float_t cut) { fJetPtCut = cut ; }
64 void SetJetRadius(Float_t r) { fJetRadius = r ; }
65 virtual void SetRhoName(const char *n) { fRhoName = n ; }
66 void SetMaxClusterPt(Float_t b) { fMaxClusterPt = b ; }
67 void SetMaxTrackPt(Float_t b) { fMaxTrackPt = b ; }
68 void SetPtBiasJetClus(Float_t b) { fPtBiasJetClus = b ; }
69 void SetNLeadingJets(Int_t t) { fNLeadingJets = t ; }
70 void SetPtBiasJetTrack(Float_t b) { fPtBiasJetTrack = b ; }
71
72 void SetLeadingHadronType(Int_t t) { fLeadingHadronType = t ; }
73 void SetJetBitMap(UInt_t m) { fJetBitMap = m ; }
74
75 AliEmcalJet *GetJet(Int_t i) const;
76 AliEmcalJet *GetAcceptJet(Int_t i) const;
77 virtual Bool_t AcceptJet(AliEmcalJet* jet) const;
78 Bool_t AcceptBiasJet(AliEmcalJet* jet) const;
79 Int_t GetNJets() const {return GetNEntries();}
80 Double_t GetLeadingHadronPt(AliEmcalJet* jet) const;
81 AliRhoParameter *GetRhoParameter() {return fRho;}
82 Double_t GetRhoVal() const {return fRho->GetVal();}
83 const TString& GetRhoName() const {return fRhoName;}
84 Double_t GetJetPtCorr(Int_t i) const;
56f370b0 85 Float_t GetJetRadius() const {return fJetRadius;}
4c9493a7 86 Float_t GetJetEtaMin() const {return fJetMinEta;}
87 Float_t GetJetEtaMax() const {return fJetMaxEta;}
88 Float_t GetJetPhiMin() const {return fJetMinPhi;}
89 Float_t GetJetPhiMax() const {return fJetMaxPhi;}
421b12a3 90
91 protected:
92 JetAcceptanceType fJetAcceptanceType; // acceptance type
93 Float_t fJetRadius; // jet radius
94 TString fRhoName; // Name of rho object
95 Float_t fPtBiasJetTrack; // select jets with a minimum pt track
96 Float_t fPtBiasJetClus; // select jets with a minimum pt cluster
97 Float_t fJetPtCut; // cut on jet pt
98 Float_t fJetAreaCut; // cut on jet area
99 Float_t fAreaEmcCut; // minimum cut on jet emcal area
100 Float_t fJetMinEta; // minimum eta jet acceptance
101 Float_t fJetMaxEta; // maximum eta jet acceptance
102 Float_t fJetMinPhi; // minimum phi jet acceptance
103 Float_t fJetMaxPhi; // maximum phi jet acceptance
104 Float_t fMaxClusterPt; // maximum cluster constituent pt to accept the jet
105 Float_t fMaxTrackPt; // maximum track constituent pt to accept the jet
106 Int_t fLeadingHadronType; // 0 = charged, 1 = neutral, 2 = both
107 Int_t fNLeadingJets; // how many jets are to be considered the leading jet(s)
108 UInt_t fJetBitMap; // bit map of accepted jets
109
110 AliRhoParameter *fRho; //!event rho for these jets
111
112 AliEMCALGeometry *fGeom; //!emcal geometry
56f370b0 113 Int_t fRunNumber; // run number
421b12a3 114
115 private:
116 AliJetContainer(const AliJetContainer& obj); // copy constructor
117 AliJetContainer& operator=(const AliJetContainer& other); // assignment
118
119 ClassDef(AliJetContainer,1);
120
121};
122
123#endif
124
125