Add fast merging option (Diego)
[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliDielectronPair.h
1 #ifndef ALIDIELECTRONPAIR_H
2 #define ALIDIELECTRONPAIR_H
3 /* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 //#############################################################
7 //#                                                           # 
8 //#                  AliDielectronPair                        #
9 //#               Class to store pair information             #
10 //#                                                           #
11 //#                                                           #
12 //#  Authors:                                                 #
13 //#   Anton     Andronic, GSI / A.Andronic@gsi.de             #
14 //#   Ionut C.  Arsene,   GSI / I.C.Arsene@gsi.de             #
15 //#   Julian    Book,     Uni Ffm / Julian.Book@cern.ch       #
16 //#   Frederick Kramer,   Uni Ffm, / Frederick.Kramer@cern.ch #
17 //#   Magnus    Mager,    CERN / Magnus.Mager@cern.ch         #
18 //#   WooJin J. Park,     GSI / W.J.Park@gsi.de               #
19 //#   Jens      Wiechula, Uni HD / Jens.Wiechula@cern.ch      #
20 //#                                                           #
21 //#############################################################
22
23 #include <TMath.h>
24 #include <TRef.h>
25 #include <TLorentzVector.h>
26
27 #include <AliKFParticle.h>
28 #include <AliVParticle.h>
29
30 class AliVVertex;
31 class AliVTrack;
32
33 //TODO
34 //TODO: Should we inherit from AliVTrack in order to built another AliDielectronPair of KF with it?
35 //TODO
36 class AliDielectronPair : public AliVParticle {
37 public:
38   AliDielectronPair();
39   virtual ~AliDielectronPair();
40   
41   AliDielectronPair(AliVTrack * const particle1, Int_t pid1,
42                     AliVTrack * const particle2, Int_t pid2, Char_t type);
43
44   AliDielectronPair(const AliKFParticle * const particle1,
45                     const AliKFParticle * const particle2,
46                     AliVTrack * const refParticle1,
47                     AliVTrack * const refParticle2,
48                     Char_t type);
49   
50 //TODO:  copy constructor + assignment operator
51   
52   void SetTracks(AliVTrack * const particle1, Int_t pid1,
53                  AliVTrack * const particle2, Int_t pid2);
54
55   void SetTracks(const AliKFParticle * const particle1,
56                  const AliKFParticle * const particle2,
57                  AliVTrack * const refParticle1,
58                  AliVTrack * const refParticle2);
59     
60   //AliVParticle interface
61   // kinematics
62   virtual Double_t Px() const { return fPair.GetPx(); }
63   virtual Double_t Py() const { return fPair.GetPy(); }
64   virtual Double_t Pz() const { return fPair.GetPz(); }
65   virtual Double_t Pt() const { return fPair.GetPt(); }
66   virtual Double_t P() const  { return fPair.GetP();  }
67   virtual Bool_t   PxPyPz(Double_t p[3]) const { p[0]=Px(); p[1]=Py(); p[2]=Pz(); return kTRUE; }
68   
69   virtual Double_t Xv() const { return fPair.GetX(); }
70   virtual Double_t Yv() const { return fPair.GetY(); }
71   virtual Double_t Zv() const { return fPair.GetZ(); }
72   virtual Bool_t   XvYvZv(Double_t x[3]) const { x[0]=Xv(); x[1]=Yv(); x[2]=Zv(); return kTRUE; }
73   
74   virtual Double_t OneOverPt() const { return Pt()>0.?1./Pt():0.; }  //TODO: check
75   virtual Double_t Phi()       const { return fPair.GetPhi();}
76   virtual Double_t Theta()     const { return Pz()!=0?TMath::ATan(Pt()/Pz()):0.; } //TODO: check
77   
78   
79   virtual Double_t E() const { return fPair.GetE();    }
80   virtual Double_t M() const { return fPair.GetMass(); }
81   
82   virtual Double_t Eta() const { return fPair.GetEta();}
83   virtual Double_t Y()  const  { 
84     if((E()*E()-Px()*Px()-Py()*Py()-Pz()*Pz())>0.) return TLorentzVector(Px(),Py(),Pz(),E()).Rapidity();
85     else return -1111.;
86   }
87   
88   virtual Short_t Charge() const    { return fPair.GetQ();}
89   virtual Int_t   GetLabel() const  { return fLabel;      }
90   // PID
91   virtual const Double_t *PID() const { return 0;} //TODO: check
92   // Dummy
93   Int_t PdgCode() const {return 0;}
94   //
95   Double_t GetLXY(const AliVVertex * const vtx) const;
96   Double_t GetPseudoProperTime(const AliVVertex * const vtx) const;
97
98
99   UChar_t GetType() const { return fType; }
100   void SetType(Char_t type) { fType=type; }
101
102   void SetLabel(Int_t label) {fLabel=label;}
103
104   void SetProductionVertex(const AliKFParticle &Vtx) { fPair.SetProductionVertex(Vtx); }
105   
106   //inter leg information
107   Double_t OpeningAngle()         const { return fD1.GetAngle(fD2);                             }
108   Double_t DistanceDaughters()    const { return fD1.GetDistanceFromParticle(fD2);              }
109   Double_t DistanceDaughtersXY()  const { return fD1.GetDistanceFromParticleXY(fD2);            }
110   Double_t DeviationDaughters()   const { return fD1.GetDeviationFromParticle(fD2);             }
111   Double_t DeviationDaughtersXY() const { return fD1.GetDeviationFromParticleXY(fD2);           }
112   Double_t DeltaEta()             const { return TMath::Abs(fD1.GetEta()-fD2.GetEta());         }
113   Double_t DeltaPhi()             const { Double_t dphi=TMath::Abs(fD1.GetPhi()-fD2.GetPhi());
114                                           return (dphi>TMath::Pi())?dphi-TMath::Pi():dphi;      }
115   // calculate cos(theta*) and phi* in HE and CS pictures
116   void GetThetaPhiCM(Double_t &thetaHE, Double_t &phiHE, Double_t &thetaCS, Double_t &phiCS) const;
117   
118   Double_t ThetaPhiCM(Bool_t isHE, Bool_t isTheta) const;
119   static Double_t ThetaPhiCM(const AliVParticle* d1, const AliVParticle* d2, 
120                                                const Bool_t isHE, const Bool_t isTheta);
121   // internal KF particle
122   const AliKFParticle& GetKFParticle()       const { return fPair; }
123   const AliKFParticle& GetKFFirstDaughter()  const { return fD1;   }
124   const AliKFParticle& GetKFSecondDaughter() const { return fD2;   }
125   
126   // daughter references
127   void SetRefFirstDaughter(AliVParticle * const track)  {fRefD1 = track;}
128   void SetRefSecondDaughter(AliVParticle * const track) {fRefD2 = track;}
129   
130   AliVParticle* GetFirstDaughter()   const { return dynamic_cast<AliVParticle*>(fRefD1.GetObject()); }
131   AliVParticle* GetSecondDaughter()  const { return dynamic_cast<AliVParticle*>(fRefD2.GetObject()); }
132
133   
134 private:
135   Char_t   fType;         // type of the pair e.g. like sign SE, unlike sign SE, ... see AliDielectron
136   Int_t    fLabel;        // MC label
137   
138   AliKFParticle fPair;   // KF particle internally used for pair calculation
139   AliKFParticle fD1;     // KF particle first daughter
140   AliKFParticle fD2;     // KF particle1 second daughter
141   
142   TRef fRefD1;           // Reference to first daughter
143   TRef fRefD2;           // Reference to second daughter
144   
145   ClassDef(AliDielectronPair,3)
146 };
147
148 #endif