First functional implementation.
[u/mrichter/AliRoot.git] / 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 "AliVirtualParticle.h"
15 #include "AliAODVertex.h"
16
17
18 class AliAODJet : public AliVirtualParticle {
19
20  public:
21     AliAODJet();
22     AliAODJet(Double_t px, Double_t py, Double_t pz, Double_t e);
23     AliAODJet(TLorentzVector p);  
24     virtual ~AliAODJet();
25     AliAODJet(const AliAODJet& jet); 
26     AliAODJet& operator=(const AliAODJet& jet);
27 // AliVirtualParticle methods
28     virtual Double_t Px()         const { return fMomentum->Px();      }
29     virtual Double_t Py()         const { return fMomentum->Py();      }
30     virtual Double_t Pz()         const { return fMomentum->Pz();      }
31     virtual Double_t Pt()         const { return fMomentum->Pt();      }
32     virtual Double_t P()          const { return fMomentum->P();       }
33     virtual Double_t OneOverPt()  const { return 1. / fMomentum->Pt(); }
34     virtual Double_t Phi()        const { return fMomentum->Phi();     }
35     virtual Double_t Theta()      const { return fMomentum->Theta();   }
36     virtual Double_t E()          const { return fMomentum->E();       }
37     virtual Double_t M()          const { return fMomentum->M();       }
38     virtual Double_t Eta()        const { return fMomentum->Eta();     }
39     virtual Double_t Y()          const { return fMomentum->Rapidity();}
40 //
41     virtual void     SetBgEnergy(Double_t bgEnCh, Double_t bgEnNe)
42         {fBackgEnergy[0] = bgEnCh; fBackgEnergy[1] = bgEnNe;}
43     virtual void     SetEffArea(Double_t effACh, Double_t effANe)
44         {fEffectiveArea[0] = effACh; fEffectiveArea[1] = effANe;}
45     
46     virtual Double_t ChargedBgEnergy()        const { return  fBackgEnergy[0];}
47     virtual Double_t NeutralBgEnergy()        const { return  fBackgEnergy[1];}
48     virtual Double_t TotalBgEnergy()          const { return (fBackgEnergy[0] + fBackgEnergy[1]);}
49
50     virtual Double_t EffectiveAreaCharged()   const { return  fEffectiveArea[0];}
51     virtual Double_t EffectiveAreaNeutral()   const { return  fEffectiveArea[1];}
52
53     virtual void     Print(Option_t* /*option*/) const;
54     
55 // Dummy  
56     virtual Short_t Charge()      const { return 0;}
57     virtual const Double_t* PID() const { return NULL;}
58 //
59     
60     
61  private:
62     TLorentzVector* fMomentum;           // Jet 4-momentum vector
63     Double_t        fBackgEnergy[2];     // Subtracted background energy
64     Double_t        fEffectiveArea[2];   // Effective jet area used for background subtraction
65     ClassDef(AliAODJet,1);
66 };
67
68 #endif