Small changes

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

diff --git a/EMCAL/AliEMCALJet.h b/EMCAL/AliEMCALJet.h
index 3cb6e7f45a0069f3e27d07af10471d671a03e3a5..6db61e114cebdeb88f6f707a8514a736ff095867 100644 (file)
--- a/EMCAL/AliEMCALJet.h
+++ b/EMCAL/AliEMCALJet.h
@@ -15,24 +15,42 @@ class AliEMCALJet : public TObject {
   AliEMCALJet(Float_t energy, Float_t phi, Float_t eta); 
   virtual ~AliEMCALJet();
   void SetEnergy(Float_t val) {fEnergy = val;}
   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 SetEMCALEnergyBGSub(Float_t val){fEMCALEnergyBGSub = val;}
+  void SetTrackEnergy(Float_t val) {fTrackEnergy = val;}
+  void SetTrackEnergyPtCut(Float_t val){fTrackEnergyPtCut = val;}
+  void SetHCEnergy(Float_t val) {fHCEnergy = val;}

   void SetPhi(Float_t val)    {fPhi    = val;}  
   void SetEta(Float_t val)    {fEta    = val;}    
   void SetPhi(Float_t val)    {fPhi    = val;}  
   void SetEta(Float_t val)    {fEta    = val;}    

-  void SetTrackList(Int_t val, Float_t* pt, Float_t* eta, Float_t* phi);
-  Float_t Energy()  {return fEnergy;}
-  Float_t Phi()     {return fPhi;}
-  Float_t Eta()     {return fEta;}
-  Int_t   TrackList(Float_t* pt, Float_t* eta, Float_t* phi);
-  Int_t   NTracks() {return fNt;} 
+  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() const  {return fEnergy;}
+  Float_t EMCALEnergy() const  {return fEMCALEnergy;}
+  Float_t EMCALEnergyBGSub()  const {return fEMCALEnergyBGSub;}
+  Float_t TrackEnergy() const  {return fTrackEnergy;}
+  Float_t TrackEnergyPtCut() const  {return fTrackEnergyPtCut;}
+  Float_t HCEnergy() const  {return fHCEnergy;}
+  Float_t Phi() const     {return fPhi;}
+  Float_t Eta()  const    {return fEta;}
+  Int_t   TrackList(Float_t* pt, Float_t* eta, Float_t* phi, Int_t* pdg);
+  Int_t   NTracks() const {return fNt;} 

   
 protected:
   
 protected:

-  Float_t  fEnergy;   // Jet Energy
-  Float_t  fEta;      // Jet Phi
-  Float_t  fPhi;      // Jet Eta
+  Float_t  fEnergy;      // Jet Energy
+  Float_t  fEMCALEnergy; // EMCAL component of Energy inside Jet cone before BG subtraction
+  Float_t  fEMCALEnergyBGSub; // EMCAL component of Energy inside Jet cone after BG subtraction
+  Float_t  fTrackEnergy; // Charge tracks component of Energy inside Jet cone with no pT cut
+  Float_t  fTrackEnergyPtCut; // Charge tracks component of Energy inside Jet cone after pT cut
+  Float_t  fHCEnergy;    // HC  component of Energy inside Jet cone
+  Bool_t   fIsWeightedEnergy; // Store flag regarding energy calculation
+  Float_t  fEta;      // Jet Eta
+  Float_t  fPhi;      // Jet Phi

   Int_t    fNt;       // Number of associated tracks
   Int_t    fNt;       // Number of associated tracks

-  Float_t  fPtT [50]; //
-  Float_t  fEtaT[50]; //
-  Float_t  fPhiT[50]; //
-  ClassDef(AliEMCALJet,2) // Jet for EMCAL
+  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,7) // Jet for EMCAL

 
 } ;
 
 
 } ;