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 */
6 //#############################################################
8 //# AliDielectronPair #
9 //# Class to store pair information #
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 #
21 //#############################################################
25 #include <TLorentzVector.h>
27 #include <AliKFParticle.h>
28 #include <AliVParticle.h>
33 //TODO: Should we inherit from AliVTrack in order to built another AliDielectronPair of KF with it?
35 class AliDielectronPair : public AliVParticle {
38 virtual ~AliDielectronPair();
40 AliDielectronPair(AliVTrack * const particle1, Int_t pid1,
41 AliVTrack * const particle2, Int_t pid2, Char_t type);
43 AliDielectronPair(const AliKFParticle * const particle1,
44 const AliKFParticle * const particle2,
45 AliVTrack * const refParticle1,
46 AliVTrack * const refParticle2,
49 //TODO: copy constructor + assignment operator
51 void SetTracks(AliVTrack * const particle1, Int_t pid1,
52 AliVTrack * const particle2, Int_t pid2);
54 void SetTracks(const AliKFParticle * const particle1,
55 const AliKFParticle * const particle2,
56 AliVTrack * const refParticle1,
57 AliVTrack * const refParticle2);
59 //AliVParticle interface
61 virtual Double_t Px() const { return fPair.GetPx(); }
62 virtual Double_t Py() const { return fPair.GetPy(); }
63 virtual Double_t Pz() const { return fPair.GetPz(); }
64 virtual Double_t Pt() const { return fPair.GetPt(); }
65 virtual Double_t P() const { return fPair.GetP(); }
66 virtual Bool_t PxPyPz(Double_t p[3]) const { p[0]=Px(); p[1]=Py(); p[2]=Pz(); return kTRUE; }
68 virtual Double_t Xv() const { return fPair.GetX(); }
69 virtual Double_t Yv() const { return fPair.GetY(); }
70 virtual Double_t Zv() const { return fPair.GetZ(); }
71 virtual Bool_t XvYvZv(Double_t x[3]) const { x[0]=Xv(); x[1]=Yv(); x[2]=Zv(); return kTRUE; }
73 virtual Double_t OneOverPt() const { return Pt()>0.?1./Pt():0.; } //TODO: check
74 virtual Double_t Phi() const { return fPair.GetPhi();}
75 virtual Double_t Theta() const { return Pz()!=0?TMath::ATan(Pt()/Pz()):0.; } //TODO: check
78 virtual Double_t E() const { return fPair.GetE(); }
79 virtual Double_t M() const { return fPair.GetMass(); }
81 virtual Double_t Eta() const { return fPair.GetEta();}
82 virtual Double_t Y() const {
83 if((E()*E()-Px()*Px()-Py()*Py()-Pz()*Pz())>0.) return TLorentzVector(Px(),Py(),Pz(),E()).Rapidity();
87 virtual Short_t Charge() const { return fPair.GetQ();}
88 virtual Int_t GetLabel() const { return fLabel; }
90 virtual const Double_t *PID() const { return 0;} //TODO: check
92 Int_t PdgCode() const {return 0;}
95 UChar_t GetType() const { return fType; }
96 void SetType(Char_t type) { fType=type; }
98 void SetLabel(Int_t label) {fLabel=label;}
100 //inter leg information
101 Double_t OpeningAngle() const { return fD1.GetAngle(fD2); }
102 Double_t DistanceDaughters() const { return fD1.GetDistanceFromParticle(fD2); }
103 Double_t DistanceDaughtersXY() const { return fD1.GetDistanceFromParticleXY(fD2); }
104 Double_t DeviationDaughters() const { return fD1.GetDeviationFromParticle(fD2); }
105 Double_t DeviationDaughtersXY() const { return fD1.GetDeviationFromParticleXY(fD2); }
106 Double_t DeltaEta() const { return TMath::Abs(fD1.GetEta()-fD2.GetEta()); }
107 Double_t DeltaPhi() const { Double_t dphi=TMath::Abs(fD1.GetPhi()-fD2.GetPhi());
108 return (dphi>TMath::Pi())?dphi-TMath::Pi():dphi; }
109 // calculate cos(theta*) and phi* in HE and CS pictures
110 void GetThetaPhiCM(Double_t &thetaHE, Double_t &phiHE, Double_t &thetaCS, Double_t &phiCS) const;
112 Double_t ThetaPhiCM(Bool_t isHE, Bool_t isTheta) const;
113 static Double_t ThetaPhiCM(const AliVParticle* d1, const AliVParticle* d2,
114 const Bool_t isHE, const Bool_t isTheta);
115 // internal KF particle
116 const AliKFParticle& GetKFParticle() const { return fPair; }
117 const AliKFParticle& GetKFFirstDaughter() const { return fD1; }
118 const AliKFParticle& GetKFSecondDaughter() const { return fD2; }
120 // daughter references
121 void SetRefFirstDaughter(AliVParticle * const track) {fRefD1 = track;}
122 void SetRefSecondDaughter(AliVParticle * const track) {fRefD2 = track;}
124 AliVParticle* GetFirstDaughter() const { return dynamic_cast<AliVParticle*>(fRefD1.GetObject()); }
125 AliVParticle* GetSecondDaughter() const { return dynamic_cast<AliVParticle*>(fRefD2.GetObject()); }
129 Char_t fType; // type of the pair e.g. like sign SE, unlike sign SE, ... see AliDielectron
130 Int_t fLabel; // MC label
132 AliKFParticle fPair; // KF particle internally used for pair calculation
133 AliKFParticle fD1; // KF particle first daughter
134 AliKFParticle fD2; // KF particle1 second daughter
136 TRef fRefD1; // Reference to first daughter
137 TRef fRefD2; // Reference to second daughter
139 ClassDef(AliDielectronPair,3)