New class AliPHOSFastGlobalReconstruction is added
[u/mrichter/AliRoot.git] / STEER / AliESDtrack.cxx
CommitLineData
ae982df3 1/**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 * *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
6 * *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
15
16//-----------------------------------------------------------------
17// Implementation of the ESD track class
18// This is the class to deal with during the phisical analysis of data
19//
20// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
21//-----------------------------------------------------------------
22
23#include "TMath.h"
24
25#include "AliESDtrack.h"
26#include "AliKalmanTrack.h"
27
28ClassImp(AliESDtrack)
29
30//_______________________________________________________________________
31AliESDtrack::AliESDtrack() :
32fFlags(0),
33fITSncls(0),
34fTPCncls(0)
35{
36 //
37 // The default ESD constructor
38 //
39 for (Int_t i=0; i<kSPECIES; i++) fR[i]=0.;
40}
41
42//_______________________________________________________________________
43Float_t AliESDtrack::GetMass() const {
44 Float_t max=0.;
45 Int_t k=-1;
46 for (Int_t i=0; i<kSPECIES; i++) {
47 if (fR[i]>max) {k=i; max=fR[i];}
48 }
49 if (k==0) return 0.00051;
50 if (k==1) return 0.10566;
51 if (k==2||k==-1) return 0.13957;
52 if (k==3) return 0.49368;
53 if (k==4) return 0.93827;
54 Warning("GetMass()","Undefined mass !");
55 return 0.13957;
56}
57
58//_______________________________________________________________________
59Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags) {
60 //
61 // This function updates track's running parameters
62 //
63 switch (flags) {
64 case kITSin: case kITSout: case kITSrefit:
65 fITSncls=t->GetNumberOfClusters();
66 fITSchi2=t->GetChi2();
67 for (Int_t i=0;i<fITSncls;i++) fITSindex[i]=t->GetClusterIndex(i);
68 fITSsignal=t->GetPIDsignal();
69 break;
70 case kTPCin: case kTPCout: case kTPCrefit:
71 fTPCncls=t->GetNumberOfClusters();
72 fTPCchi2=t->GetChi2();
73 for (Int_t i=0;i<fTPCncls;i++) fTPCindex[i]=t->GetClusterIndex(i);
74 fTPCsignal=t->GetPIDsignal();
75 {Double_t mass=t->GetMass(); // preliminary mass setting
76 if (mass>0.5) fR[4]=1.; // used by
77 else if (mass<0.4) fR[2]=1.; // the ITS reconstruction
78 else fR[3]=1.;} //
79 break;
80 default:
81 Error("UpdateTrackParams()","Wrong flag !\n");
82 return kFALSE;
83 }
84
85 SetStatus(flags);
86 fLabel=t->GetLabel();
87
88 if (t->IsStartedTimeIntegral()) {
89 SetStatus(kTIME);
90 Double_t times[10];t->GetIntegratedTimes(times); SetIntegratedTimes(times);
91 SetIntegratedLength(t->GetIntegratedLength());
92 }
93
94 fRalpha=t->GetAlpha();
95 t->GetExternalParameters(fRx,fRp);
96 t->GetExternalCovariance(fRc);
97 return kTRUE;
98}
99
100//_______________________________________________________________________
101void AliESDtrack::GetExternalParameters(Double_t &x, Double_t p[5]) const {
102 //---------------------------------------------------------------------
103 // This function returns external representation of the track parameters
104 //---------------------------------------------------------------------
105 x=fRx;
106 for (Int_t i=0; i<5; i++) p[i]=fRp[i];
107}
108
109Double_t AliESDtrack::GetP() const {
110 //---------------------------------------------------------------------
111 // This function returns the track momentum
112 //---------------------------------------------------------------------
113 Double_t lam=TMath::ATan(fRp[3]);
114 Double_t pt=1./TMath::Abs(fRp[4]);
115 return pt/TMath::Cos(lam);
116}
117
118void AliESDtrack::GetPxPyPz(Double_t *p) const {
119 //---------------------------------------------------------------------
120 // This function returns the global track momentum components
121 //---------------------------------------------------------------------
122 Double_t phi=TMath::ASin(fRp[2]) + fRalpha;
123 Double_t pt=1./TMath::Abs(fRp[4]);
124 p[0]=pt*TMath::Cos(phi); p[1]=pt*TMath::Sin(phi); p[2]=pt*fRp[3];
125}
126
127void AliESDtrack::GetXYZ(Double_t *xyz) const {
128 //---------------------------------------------------------------------
129 // This function returns the global track position
130 //---------------------------------------------------------------------
131 Double_t phi=TMath::ASin(fRp[2]) + fRalpha;
132 Double_t r=TMath::Sqrt(fRx*fRx + fRp[0]*fRp[0]);
133 xyz[0]=r*TMath::Cos(phi); xyz[1]=r*TMath::Sin(phi); xyz[2]=fRp[1];
134}
135
136//_______________________________________________________________________
137void AliESDtrack::GetExternalCovariance(Double_t c[15]) const {
138 //---------------------------------------------------------------------
139 // This function returns external representation of the cov. matrix
140 //---------------------------------------------------------------------
141 for (Int_t i=0; i<15; i++) c[i]=fRc[i];
142}
143
144//_______________________________________________________________________
145void AliESDtrack::GetIntegratedTimes(Double_t *times) const {
146 for (Int_t i=0; i<kSPECIES; i++) times[i]=fTrackTime[i];
147}
148
149//_______________________________________________________________________
150void AliESDtrack::SetIntegratedTimes(const Double_t *times) {
151 for (Int_t i=0; i<kSPECIES; i++) fTrackTime[i]=times[i];
152}
153
154//_______________________________________________________________________
155Int_t AliESDtrack::GetITSclusters(UInt_t *idx) const {
156 //---------------------------------------------------------------------
157 // This function returns indices of the assgined ITS clusters
158 //---------------------------------------------------------------------
159 for (Int_t i=0; i<fITSncls; i++) idx[i]=fITSindex[i];
160 return fITSncls;
161}
162
163//_______________________________________________________________________
164Int_t AliESDtrack::GetTPCclusters(UInt_t *idx) const {
165 //---------------------------------------------------------------------
166 // This function returns indices of the assgined ITS clusters
167 //---------------------------------------------------------------------
168 for (Int_t i=0; i<fTPCncls; i++) idx[i]=fTPCindex[i];
169 return fTPCncls;
170}