STEER/AliAODJet.h

   1 #ifndef AliAODJet_H
   2 #define AliAODJet_H
   3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 //-------------------------------------------------------------------------
   9 //     AOD jet class
  10 //     Author: Andreas Morsch, CERN
  11 //-------------------------------------------------------------------------
  12
  13 #include <TLorentzVector.h>
  14 #include <TArrayI.h>
  15 #include "AliVirtualParticle.h"
  16 #include "AliAODVertex.h"
  17
  18
  19 class AliAODJet : public AliVirtualParticle {
  20
  21  public:
  22     AliAODJet();
  23     AliAODJet(Double_t px, Double_t py, Double_t pz, Double_t e);
  24     AliAODJet(TLorentzVector & p);
  25     virtual ~AliAODJet();
  26     AliAODJet(const AliAODJet& jet);
  27     AliAODJet& operator=(const AliAODJet& jet);
  28 // AliVirtualParticle methods
  29     virtual Double_t Px()         const { return fMomentum->Px();      }
  30     virtual Double_t Py()         const { return fMomentum->Py();      }
  31     virtual Double_t Pz()         const { return fMomentum->Pz();      }
  32     virtual Double_t Pt()         const { return fMomentum->Pt();      }
  33     virtual Double_t P()          const { return fMomentum->P();       }
  34     virtual Double_t OneOverPt()  const { return 1. / fMomentum->Pt(); }
  35     virtual Double_t Phi()        const { return fMomentum->Phi();     }
  36     virtual Double_t Theta()      const { return fMomentum->Theta();   }
  37     virtual Double_t E()          const { return fMomentum->E();       }
  38     virtual Double_t M()          const { return fMomentum->M();       }
  39     virtual Double_t Eta()        const { return fMomentum->Eta();     }
  40     virtual Double_t Y()          const { return fMomentum->Rapidity();}
  41 //
  42     virtual void     AddTrack(TObject *tr) {fRefTracks->Add(tr);}
  43     TObject* GetTrack(Int_t i) {return fRefTracks->At(i);}
  44     virtual void     SetBgEnergy(Double_t bgEnCh, Double_t bgEnNe)
  45         {fBackgEnergy[0] = bgEnCh; fBackgEnergy[1] = bgEnNe;}
  46     virtual void     SetEffArea(Double_t effACh, Double_t effANe)
  47         {fEffectiveArea[0] = effACh; fEffectiveArea[1] = effANe;}
  48
  49     virtual TRefArray* GetRefTracks()           const { return  fRefTracks;}
  50     virtual Double_t   ChargedBgEnergy()        const { return  fBackgEnergy[0];}
  51     virtual Double_t   NeutralBgEnergy()        const { return  fBackgEnergy[1];}
  52     virtual Double_t   TotalBgEnergy()          const { return (fBackgEnergy[0] + fBackgEnergy[1]);}
  53
  54     virtual Double_t   EffectiveAreaCharged()   const { return  fEffectiveArea[0];}
  55     virtual Double_t   EffectiveAreaNeutral()   const { return  fEffectiveArea[1];}
  56
  57     virtual void     Print(Option_t* /*option*/) const;
  58
  59 // Dummy
  60     virtual Short_t Charge()      const { return 0;}
  61     virtual const Double_t* PID() const { return NULL;}
  62 //
  63
  64
  65  private:
  66     TLorentzVector* fMomentum;           // Jet 4-momentum vector
  67     TRefArray*      fRefTracks;          // array of references to the tracks belonging to the jet
  68     Double_t        fBackgEnergy[2];     // Subtracted background energy
  69     Double_t        fEffectiveArea[2];   // Effective jet area used for background subtraction
  70     ClassDef(AliAODJet, 2);
  71 };
  72
  73 #endif