Corrected protection.
[u/mrichter/AliRoot.git] / STEER / AliAODDimuon.h
1 #ifndef AliAODDimuon_H
2 #define AliAODDimuon_H
3
4 // AliAODDimuon: a class for AODs for the MUON Arm of the ALICE Experiment
5 // Author: P. Cortese, Universita' del Piemonte Orientale in Alessandria and
6 // INFN of Torino - Italy
7 //
8 // The class defines a dimuon pair object from two AliAODTrack objects.
9 // AliAODDimuon objects are supposed to be added to the AliAODEvent structure
10 // during analysis. They would then allow to calculate the dimuon-related
11 // kinematic variables with a minimal disk occupancy.
12 // The payload of the class has been reduced to two pointers to the two
13 // tracks. An instance of this class has also to be added to the AliAODEvent 
14 // structure to provide additional information that is specific to MUON and 
15 // therefore has not been included into the AOD header.
16 // Two transient data members are not stored on file as they can be recomputed
17 // at runtime.
18 //
19
20 // 2007/07/07 v1.00 Initial version
21 // 2007/12/06 v1.01 Introduction of AliAODEventInfo
22 // 2007/12/18 v1.02 Corrected CostCS for Like-Sign, added CostKh, CostHe and xf
23 // 2008/02/01 v1.03 Apply coding conventions
24
25 #include "TRef.h"
26 #include "AliVParticle.h"
27 #include "AliAODTrack.h"
28
29 class TLorentzVector;
30
31 class AliAODDimuon: public AliVParticle {
32 public:
33   AliAODDimuon();
34   AliAODDimuon(const AliAODDimuon& dimu);
35   AliAODDimuon &operator=(const AliAODDimuon& dimu);
36   AliAODDimuon(TObject *mu0, TObject *mu1);
37   virtual ~AliAODDimuon();
38
39   // Methods to access kinematics
40   virtual Double_t Px() const;
41   virtual Double_t Py() const;
42   virtual Double_t Pz() const;
43   virtual Bool_t PxPyPz(Double_t* p) const { p[0]=Px(); p[1]=Py(); p[2]=Pz(); return 1;}
44   virtual Double_t Pt() const;
45   virtual Double_t P() const;
46
47   virtual Double_t OneOverPt() const {return Pt()>0 ? 1./Pt() : -999999999;}
48   virtual Double_t Phi() const;
49   virtual Double_t Theta() const;
50
51   virtual Double_t E() const;
52   virtual Double_t M() const;
53   
54   virtual Double_t Eta() const;
55   virtual Double_t Y() const;
56   
57   virtual Short_t Charge() const;
58
59   // Dimuon vertex will be implemented when the muon track covariance matrix 
60   // at vertex will be included in the ESD (and AOD)
61   // It would require also the information about magnetic field when filling AOD
62   virtual Double_t Xv() const {return -999999999;}
63   virtual Double_t Yv() const {return -999999999;}
64   virtual Double_t Zv() const {return -999999999;}
65   virtual Bool_t XvYvZv(Double_t* v) const { v[0]=-999999999; v[1]=-999999999; v[2]=-999999999; return 0;}
66
67   Double_t P();
68   Double_t Phi();
69   Double_t Theta();
70   Double_t M();
71   Double_t Mass();
72   Double_t Eta();
73   Double_t Y();
74
75   // Added functions
76   Double_t XF();     // Feynman x
77   Double_t CostCS(); // Cosinus of the Collins-Soper polar decay angle
78   Double_t CostHe(); // Cosinus of the Helicity polar decay angle
79   Double_t PhiCS();  // Azimuthal angle in the Collins-Soper frame
80   Double_t PhiHe();  // Azimuthal angle in the Helicity frame
81   Int_t AnyPt();
82   Int_t LowPt();
83   Int_t HighPt();
84   Double_t MaxChi2Match();
85   // PID
86   virtual const Double_t *PID() const {return 0;} // return PID object (to be defined, still)
87   
88   //
89   Int_t GetLabel() const {return -1;}
90   // Additional getters and setters
91   AliAODTrack* GetMu(Int_t imu=0) const {return (imu==0||imu==1)&&(fMu[imu]!=0) ? (AliAODTrack*)fMu[imu].GetObject() : 0; } // Get a pointer to a muon
92   AliAODTrack* Mu(Int_t imu=0) const {return (imu==0||imu==1)&&(fMu[imu]!=0) ? (AliAODTrack*)fMu[imu].GetObject() : 0; } // Get a pointer to a muon
93
94   void SetMu(Int_t imu=0, AliAODTrack *mu=0);
95   void SetMuons(AliAODTrack *mu0=0, AliAODTrack *mu1=0);
96   // Dummy
97   virtual Int_t PdgCode() const {return 0;}
98
99 private:
100   Int_t CheckPointers() const;
101   void BookP();
102
103   // Data members
104   TRef fMu[2];  // Pointers to the reconstructed muons
105   TLorentzVector *fP; //! TLorentzVector of dimuon momentum (not stored into file)
106
107   // Useful constants
108   Double_t fMProton; //! Proton mass (not stored into file)
109
110   ClassDef(AliAODDimuon,1)  // AliAODDimuon track
111 };
112
113 #endif