1 #ifndef AliJetContainer_H
2 #define AliJetContainer_H
9 class AliEMCALGeometry;
12 class AliParticleContainer;
13 class AliClusterContainer;
14 class AliLocalRhoParameter;
18 #include "AliRhoParameter.h"
19 #include "AliEmcalContainer.h"
21 #include "AliVEvent.h"
23 class AliJetContainer : public AliEmcalContainer {
26 enum JetAcceptanceType {
27 kTPC = 0, // TPC acceptance
28 kEMCAL = 1, // EMCal acceptance
29 kUser = 2 // User defined acceptance
34 AliJetContainer(const char *name);
35 virtual ~AliJetContainer() {;}
37 void SetEMCALGeometry();
38 void SetEMCALGeometry(AliEMCALGeometry *p) {fGeom = p;}
39 void LoadRho(AliVEvent *event);
40 void LoadLocalRho(AliVEvent *event);
41 void LoadRhoMass(AliVEvent *event);
42 void LoadPythiaInfo(AliVEvent *event);
44 void SetJetAcceptanceType(JetAcceptanceType type) { fJetAcceptanceType = type ; }
47 void SetJetEtaPhiEMCAL() ;
48 void SetJetEtaPhiTPC() ;
49 void SetRunNumber(Int_t r) { fRunNumber = r; }
50 void SetJetEtaLimits(Float_t min, Float_t max) { fJetMinEta = min, fJetMaxEta = max ; }
51 void SetJetPhiLimits(Float_t min, Float_t max) { fJetMinPhi = min, fJetMaxPhi = max ; }
52 void SetJetPtCut(Float_t cut) { fJetPtCut = cut ; }
53 void SetJetRadius(Float_t r) { fJetRadius = r ; }
54 void SetJetAreaCut(Float_t cut) { fJetAreaCut = cut ; }
55 void SetPercAreaCut(Float_t p) { if(fJetRadius==0.) AliWarning("JetRadius not set. Area cut will be 0");
56 fJetAreaCut = p*TMath::Pi()*fJetRadius*fJetRadius; }
57 void SetAreaEmcCut(Double_t a = 0.99) { fAreaEmcCut = a ; }
58 void SetZLeadingCut(Float_t zemc, Float_t zch) { fZLeadingEmcCut = zemc; fZLeadingChCut = zch ; }
59 void SetNEFCut(Float_t min = 0., Float_t max = 1.) { fNEFMinCut = min; fNEFMaxCut = max; }
60 void SetFlavourCut(Int_t myflavour) { fFlavourSelection = myflavour;}
61 void SetMaxClusterPt(Float_t b) { fMaxClusterPt = b ; }
62 void SetMaxTrackPt(Float_t b) { fMaxTrackPt = b ; }
63 void SetPtBiasJetClus(Float_t b) { fPtBiasJetClus = b ; }
64 void SetNLeadingJets(Int_t t) { fNLeadingJets = t ; }
65 void SetPtBiasJetTrack(Float_t b) { fPtBiasJetTrack = b ; }
66 void SetLeadingHadronType(Int_t t) { fLeadingHadronType = t ; }
67 void SetJetBitMap(UInt_t m) { fJetBitMap = m ; }
68 void SetJetTrigger(UInt_t t=AliVEvent::kEMCEJE) { fJetTrigger = t ; }
69 void SetTagStatus(Int_t i) { fTagStatus = i ; }
71 virtual void SetRhoName(const char *n) { fRhoName = n ; }
72 virtual void SetLocalRhoName(const char *n) { fLocalRhoName = n ; }
73 virtual void SetRhoMassName(const char *n) { fRhoMassName = n ; }
74 virtual void SetPythiaInfoName(const char *n) { fPythiaInfoName = n; }
76 void ConnectParticleContainer(AliParticleContainer *c) { fParticleContainer = c ; }
77 void ConnectClusterContainer(AliClusterContainer *c) { fClusterContainer = c ; }
79 AliEmcalJet *GetLeadingJet(const char* opt="") ;
80 AliEmcalJet *GetJet(Int_t i) const;
81 AliEmcalJet *GetAcceptJet(Int_t i) ;
82 AliEmcalJet *GetJetWithLabel(Int_t lab) const;
83 AliEmcalJet *GetAcceptJetWithLabel(Int_t lab) ;
84 AliEmcalJet *GetNextAcceptJet(Int_t i=-1) ;
85 AliEmcalJet *GetNextJet(Int_t i=-1) ;
86 void GetMomentum(TLorentzVector &mom, Int_t i) const;
87 Bool_t AcceptJet(const AliEmcalJet* jet) ;
88 Bool_t AcceptBiasJet(const AliEmcalJet* jet) ;
89 Int_t GetFlavourCut() const {return fFlavourSelection;}
90 Int_t GetNJets() const {return GetNEntries();}
91 Double_t GetLeadingHadronPt(const AliEmcalJet* jet) const;
92 void GetLeadingHadronMomentum(TLorentzVector &mom, const AliEmcalJet* jet) const;
93 Double_t GetZ(const AliEmcalJet *jet, TLorentzVector mom) const;
94 Double_t GetZLeadingEmc(const AliEmcalJet *jet) const;
95 Double_t GetZLeadingCharged(const AliEmcalJet *jet) const;
96 AliRhoParameter *GetRhoParameter() {return fRho;}
97 Double_t GetRhoVal() const {if (fRho) return fRho->GetVal(); else return 0;}
98 const TString& GetRhoName() const {return fRhoName;}
99 AliLocalRhoParameter *GetLocalRhoParameter() const {return fLocalRho;}
100 const TString& GetLocalRhoName() const {return fLocalRhoName;}
101 AliRhoParameter *GetRhoMassParameter() {return fRhoMass;}
102 Double_t GetRhoMassVal() const {if (fRhoMass) return fRhoMass->GetVal(); else return 0;}
103 const TString& GetRhoMassName() const {return fRhoMassName;}
104 const TString& GetPythiaInfoName() const {return fPythiaInfoName;}
105 AliPythiaInfo *GetPythiaInfo() const {return fPythiaInfo;}
106 Double_t GetJetPtCorr(Int_t i) const;
107 Double_t GetJetPtCorrLocal(Int_t i) const;
108 Float_t GetJetRadius() const {return fJetRadius;}
109 Float_t GetJetEtaMin() const {return fJetMinEta;}
110 Float_t GetJetEtaMax() const {return fJetMaxEta;}
111 Float_t GetJetPhiMin() const {return fJetMinPhi;}
112 Float_t GetJetPhiMax() const {return fJetMaxPhi;}
113 Float_t GetJetPtCut() const {return fJetPtCut;}
114 void SetClassName(const char *clname);
115 void SetArray(AliVEvent *event);
116 AliParticleContainer *GetParticleContainer() const {return fParticleContainer;}
117 AliClusterContainer *GetClusterContainer() const {return fClusterContainer;}
118 Double_t GetFractionSharedPt(const AliEmcalJet *jet) const;
122 JetAcceptanceType fJetAcceptanceType; // acceptance type
123 Float_t fJetRadius; // jet radius
124 TString fRhoName; // Name of rho object
125 TString fLocalRhoName; // Name of local rho object
126 TString fRhoMassName; // Name of rho mass object
127 TString fPythiaInfoName; // Name of pythia info object
128 Int_t fFlavourSelection; // selection on jet flavour
129 Float_t fPtBiasJetTrack; // select jets with a minimum pt track
130 Float_t fPtBiasJetClus; // select jets with a minimum pt cluster
131 Float_t fJetPtCut; // cut on jet pt
132 Float_t fJetAreaCut; // cut on jet area
133 Float_t fAreaEmcCut; // minimum cut on jet emcal area
134 Float_t fJetMinEta; // minimum eta jet acceptance
135 Float_t fJetMaxEta; // maximum eta jet acceptance
136 Float_t fJetMinPhi; // minimum phi jet acceptance
137 Float_t fJetMaxPhi; // maximum phi jet acceptance
138 Float_t fMaxClusterPt; // maximum cluster constituent pt to accept the jet
139 Float_t fMaxTrackPt; // maximum track constituent pt to accept the jet
140 Float_t fZLeadingEmcCut; // maximum z,leading neutral
141 Float_t fZLeadingChCut; // maximum z,leading charged
142 Float_t fNEFMinCut; // minimum NEF in a jet
143 Float_t fNEFMaxCut; // maximum NEF in a jet
144 Int_t fLeadingHadronType; // 0 = charged, 1 = neutral, 2 = both
145 Int_t fNLeadingJets; // how many jets are to be considered the leading jet(s)
146 UInt_t fJetBitMap; // bit map of accepted jets
147 UInt_t fJetTrigger; // jet trigger
148 Int_t fTagStatus; // jet tag status
149 AliParticleContainer *fParticleContainer; // particle container (jet constituents)
150 AliClusterContainer *fClusterContainer; // cluster container (jet constituents)
151 AliRhoParameter *fRho; //! event rho for these jets
152 AliLocalRhoParameter *fLocalRho; //! event local rho for these jets
153 AliRhoParameter *fRhoMass; //! event rho mass for these jets
154 AliPythiaInfo *fPythiaInfo; //! event parton info
155 AliEMCALGeometry *fGeom; //! emcal geometry
156 Int_t fRunNumber; //! run number
159 AliJetContainer(const AliJetContainer& obj); // copy constructor
160 AliJetContainer& operator=(const AliJetContainer& other); // assignment
162 ClassDef(AliJetContainer,10);