Support for references to tracks added.

[u/mrichter/AliRoot.git] / STEER / AliAODJet.h

diff --git a/STEER/AliAODJet.h b/STEER/AliAODJet.h
index 0d7a319b090613c1188d6f81e181e1b2e5ca9474..0d67782c4a1e3b961ebdb0aff79bdb179f070235 100644 (file)
--- a/STEER/AliAODJet.h
+++ b/STEER/AliAODJet.h
@@ -11,6 +11,7 @@
 //-------------------------------------------------------------------------
 
 #include <TLorentzVector.h>
+#include <TArrayI.h>
 #include "AliVirtualParticle.h"
 #include "AliAODVertex.h"
 
@@ -38,17 +39,20 @@ class AliAODJet : public AliVirtualParticle {
     virtual Double_t Eta()        const { return fMomentum->Eta();     }
     virtual Double_t Y()          const { return fMomentum->Rapidity();}
 //
+    virtual void     AddTrack(TObject *tr) {fRefTracks->Add(tr);}
+    TObject* GetTrack(Int_t i) {return fRefTracks->At(i);}
     virtual void     SetBgEnergy(Double_t bgEnCh, Double_t bgEnNe)
        {fBackgEnergy[0] = bgEnCh; fBackgEnergy[1] = bgEnNe;}
     virtual void     SetEffArea(Double_t effACh, Double_t effANe)
        {fEffectiveArea[0] = effACh; fEffectiveArea[1] = effANe;}
     
-    virtual Double_t ChargedBgEnergy()        const { return  fBackgEnergy[0];}
-    virtual Double_t NeutralBgEnergy()        const { return  fBackgEnergy[1];}
-    virtual Double_t TotalBgEnergy()          const { return (fBackgEnergy[0] + fBackgEnergy[1]);}
+    virtual TRefArray* GetRefTracks()           const { return  fRefTracks;}
+    virtual Double_t   ChargedBgEnergy()        const { return  fBackgEnergy[0];}
+    virtual Double_t   NeutralBgEnergy()        const { return  fBackgEnergy[1];}
+    virtual Double_t   TotalBgEnergy()          const { return (fBackgEnergy[0] + fBackgEnergy[1]);}
 
-    virtual Double_t EffectiveAreaCharged()   const { return  fEffectiveArea[0];}
-    virtual Double_t EffectiveAreaNeutral()   const { return  fEffectiveArea[1];}
+    virtual Double_t   EffectiveAreaCharged()   const { return  fEffectiveArea[0];}
+    virtual Double_t   EffectiveAreaNeutral()   const { return  fEffectiveArea[1];}
 
     virtual void     Print(Option_t* /*option*/) const;
     
@@ -60,9 +64,10 @@ class AliAODJet : public AliVirtualParticle {
     
  private:
     TLorentzVector* fMomentum;           // Jet 4-momentum vector
+    TRefArray*      fRefTracks;          // array of references to the tracks belonging to the jet
     Double_t        fBackgEnergy[2];     // Subtracted background energy
     Double_t        fEffectiveArea[2];   // Effective jet area used for background subtraction
-    ClassDef(AliAODJet,1);
+    ClassDef(AliAODJet, 2);
 };
 
 #endif