]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWG3/dielectron/AliDielectronPair.h
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Fix in the last caall to CleanOwnPrimaryVertex
[u/mrichter/AliRoot.git] / PWG3 / dielectron / AliDielectronPair.h
CommitLineData
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 30class AliVVertex;
b2a297fa 31class AliVTrack;
32
33//TODO
34//TODO: Should we inherit from AliVTrack in order to built another AliDielectronPair of KF with it?
35//TODO
36class AliDielectronPair : public AliVParticle {
37public:
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
134private:
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