Record charged and neutral energy component separately.

[u/mrichter/AliRoot.git] / EMCAL / AliEMCALJet.h

diff --git a/EMCAL/AliEMCALJet.h b/EMCAL/AliEMCALJet.h
index f5d48b1310511ff93c2f0ca8552a5eb2d5beeca3..1f9b8f052694355f9369a68de8ed727b15808319 100644 (file)
--- a/EMCAL/AliEMCALJet.h
+++ b/EMCAL/AliEMCALJet.h
@@ -15,17 +15,36 @@ class AliEMCALJet : public TObject {
   AliEMCALJet(Float_t energy, Float_t phi, Float_t eta); 
   virtual ~AliEMCALJet();
   void SetEnergy(Float_t val) {fEnergy = val;}
+  void SetEMCALEnergy(Float_t val) {fEMCALEnergy = val;}
+  void SetTrackEnergy(Float_t val) {fTrackEnergy = val;}
+  void SetHCEnergy(Float_t val) {fHCEnergy = val;}
   void SetPhi(Float_t val)    {fPhi    = val;}  
   void SetEta(Float_t val)    {fEta    = val;}    
-
-  Float_t Energy() {return fEnergy;}
-  Float_t Phi()    {return fPhi;}
-  Float_t Eta()    {return fEta;}
+  void SetIsWeightedEnergy(Bool_t flag)    {fIsWeightedEnergy    = flag;}    
+  void SetTrackList(Int_t val, Float_t* pt, Float_t* eta, Float_t* phi, Int_t* pdg);
+  Float_t Energy()  {return fEnergy;}
+  Float_t EMCALEnergy()  {return fEMCALEnergy;}
+  Float_t TrackEnergy()  {return fTrackEnergy;}
+  Float_t HCEnergy()  {return fHCEnergy;}
+  Float_t Phi()     {return fPhi;}
+  Float_t Eta()     {return fEta;}
+  Int_t   TrackList(Float_t* pt, Float_t* eta, Float_t* phi, Int_t* pdg);
+  Int_t   NTracks() {return fNt;} 
+  
 protected:
-  Float_t fEnergy;   // Jet Energy
-  Float_t fEta;      // Jet Phi
-  Float_t fPhi;      // Jet Eta 
-  ClassDef(AliEMCALJet,1) // Jet for EMCAL
+  Float_t  fEnergy;      // Jet Energy
+  Float_t  fEMCALEnergy; // EMCAL component of Energy inside Jet cone
+  Float_t  fTrackEnergy; // Charge tracks component of Energy inside Jet cone
+  Float_t  fHCEnergy;    // HC  component of Energy inside Jet cone
+  Bool_t   fIsWeightedEnergy; // Store flag regarding energy calculation
+  Float_t  fEta;      // Jet Phi
+  Float_t  fPhi;      // Jet Eta
+  Int_t    fNt;       // Number of associated tracks
+  Float_t  fPtT [1000]; // Track pt 
+  Float_t  fEtaT[1000]; // Track eta
+  Float_t  fPhiT[1000]; // Track phi
+  Int_t    fPdgT[1000]; // Track pdg code
+  ClassDef(AliEMCALJet,5) // Jet for EMCAL
 
 } ;