]>
Commit | Line | Data |
---|---|---|
b2a297fa | 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 | ||
ba15fdfb | 30 | class AliVVertex; |
b2a297fa | 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 | ||
1201a1a9 | 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 | |
b2a297fa | 51 | |
52 | void SetTracks(AliVTrack * const particle1, Int_t pid1, | |
53 | AliVTrack * const particle2, Int_t pid2); | |
54 | ||
1201a1a9 | 55 | void SetTracks(const AliKFParticle * const particle1, |
56 | const AliKFParticle * const particle2, | |
57 | AliVTrack * const refParticle1, | |
58 | AliVTrack * const refParticle2); | |
59 | ||
b2a297fa | 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(); } | |
61d106d3 | 72 | virtual Bool_t XvYvZv(Double_t x[3]) const { x[0]=Xv(); x[1]=Yv(); x[2]=Zv(); return kTRUE; } |
b2a297fa | 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();} | |
8df8e382 | 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 | } | |
b2a297fa | 87 | |
88 | virtual Short_t Charge() const { return fPair.GetQ();} | |
a655b716 | 89 | virtual Int_t GetLabel() const { return fLabel; } |
b2a297fa | 90 | // PID |
91 | virtual const Double_t *PID() const { return 0;} //TODO: check | |
a655b716 | 92 | // Dummy |
93 | Int_t PdgCode() const {return 0;} | |
ba15fdfb | 94 | // |
95 | Double_t GetLXY(const AliVVertex * const vtx) const; | |
96 | Double_t GetPseudoProperTime(const AliVVertex * const vtx) const; | |
572b0139 | 97 | |
b2a297fa | 98 | |
99 | UChar_t GetType() const { return fType; } | |
100 | void SetType(Char_t type) { fType=type; } | |
a655b716 | 101 | |
102 | void SetLabel(Int_t label) {fLabel=label;} | |
ba15fdfb | 103 | |
104 | void SetProductionVertex(const AliKFParticle &Vtx) { fPair.SetProductionVertex(Vtx); } | |
572b0139 | 105 | |
106 | //inter leg information | |
61d106d3 | 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; } | |
8df8e382 | 115 | // calculate cos(theta*) and phi* in HE and CS pictures |
61d106d3 | 116 | void GetThetaPhiCM(Double_t &thetaHE, Double_t &phiHE, Double_t &thetaCS, Double_t &phiCS) const; |
117 | ||
8df8e382 | 118 | Double_t ThetaPhiCM(Bool_t isHE, Bool_t isTheta) const; |
119 | static Double_t ThetaPhiCM(const AliVParticle* d1, const AliVParticle* d2, | |
61d106d3 | 120 | const Bool_t isHE, const Bool_t isTheta); |
b2a297fa | 121 | // internal KF particle |
572b0139 | 122 | const AliKFParticle& GetKFParticle() const { return fPair; } |
123 | const AliKFParticle& GetKFFirstDaughter() const { return fD1; } | |
124 | const AliKFParticle& GetKFSecondDaughter() const { return fD2; } | |
125 | ||
b2a297fa | 126 | // daughter references |
8df8e382 | 127 | void SetRefFirstDaughter(AliVParticle * const track) {fRefD1 = track;} |
128 | void SetRefSecondDaughter(AliVParticle * const track) {fRefD2 = track;} | |
129 | ||
b2a297fa | 130 | AliVParticle* GetFirstDaughter() const { return dynamic_cast<AliVParticle*>(fRefD1.GetObject()); } |
131 | AliVParticle* GetSecondDaughter() const { return dynamic_cast<AliVParticle*>(fRefD2.GetObject()); } | |
a655b716 | 132 | |
b2a297fa | 133 | |
134 | private: | |
a655b716 | 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 | ||
b2a297fa | 138 | AliKFParticle fPair; // KF particle internally used for pair calculation |
572b0139 | 139 | AliKFParticle fD1; // KF particle first daughter |
140 | AliKFParticle fD2; // KF particle1 second daughter | |
141 | ||
b2a297fa | 142 | TRef fRefD1; // Reference to first daughter |
143 | TRef fRefD2; // Reference to second daughter | |
144 | ||
572b0139 | 145 | ClassDef(AliDielectronPair,3) |
b2a297fa | 146 | }; |
147 | ||
148 | #endif |