1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
16 //-----------------------------------------------------------------
17 // Implementation of the ESD track class
18 // This is the class to deal with during the phisical analysis of data
20 // Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
21 //-----------------------------------------------------------------
25 #include "AliESDtrack.h"
26 #include "AliKalmanTrack.h"
30 //_______________________________________________________________________
31 AliESDtrack::AliESDtrack() :
37 // The default ESD constructor
39 for (Int_t i=0; i<kSPECIES; i++) fR[i]=0.;
42 //_______________________________________________________________________
43 Float_t AliESDtrack::GetMass() const {
46 for (Int_t i=0; i<kSPECIES; i++) {
47 if (fR[i]>max) {k=i; max=fR[i];}
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 !");
58 //_______________________________________________________________________
59 Bool_t AliESDtrack::UpdateTrackParams(const AliKalmanTrack *t, ULong_t flags) {
61 // This function updates track's running parameters
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();
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
80 case kTRDin: case kTRDout: case kTRDrefit:
81 fTRDncls=t->GetNumberOfClusters();
82 fTRDchi2=t->GetChi2();
83 fTRDsignal=t->GetPIDsignal();
86 Error("UpdateTrackParams()","Wrong flag !\n");
93 if (t->IsStartedTimeIntegral()) {
95 Double_t times[10];t->GetIntegratedTimes(times); SetIntegratedTimes(times);
96 SetIntegratedLength(t->GetIntegratedLength());
99 fRalpha=t->GetAlpha();
100 t->GetExternalParameters(fRx,fRp);
101 t->GetExternalCovariance(fRc);
105 //_______________________________________________________________________
106 void AliESDtrack::GetExternalParameters(Double_t &x, Double_t p[5]) const {
107 //---------------------------------------------------------------------
108 // This function returns external representation of the track parameters
109 //---------------------------------------------------------------------
111 for (Int_t i=0; i<5; i++) p[i]=fRp[i];
114 Double_t AliESDtrack::GetP() const {
115 //---------------------------------------------------------------------
116 // This function returns the track momentum
117 //---------------------------------------------------------------------
118 Double_t lam=TMath::ATan(fRp[3]);
119 Double_t pt=1./TMath::Abs(fRp[4]);
120 return pt/TMath::Cos(lam);
123 void AliESDtrack::GetPxPyPz(Double_t *p) const {
124 //---------------------------------------------------------------------
125 // This function returns the global track momentum components
126 //---------------------------------------------------------------------
127 Double_t phi=TMath::ASin(fRp[2]) + fRalpha;
128 Double_t pt=1./TMath::Abs(fRp[4]);
129 p[0]=pt*TMath::Cos(phi); p[1]=pt*TMath::Sin(phi); p[2]=pt*fRp[3];
132 void AliESDtrack::GetXYZ(Double_t *xyz) const {
133 //---------------------------------------------------------------------
134 // This function returns the global track position
135 //---------------------------------------------------------------------
136 Double_t phi=TMath::ASin(fRp[2]) + fRalpha;
137 Double_t r=TMath::Sqrt(fRx*fRx + fRp[0]*fRp[0]);
138 xyz[0]=r*TMath::Cos(phi); xyz[1]=r*TMath::Sin(phi); xyz[2]=fRp[1];
141 //_______________________________________________________________________
142 void AliESDtrack::GetExternalCovariance(Double_t c[15]) const {
143 //---------------------------------------------------------------------
144 // This function returns external representation of the cov. matrix
145 //---------------------------------------------------------------------
146 for (Int_t i=0; i<15; i++) c[i]=fRc[i];
149 //_______________________________________________________________________
150 void AliESDtrack::GetIntegratedTimes(Double_t *times) const {
151 for (Int_t i=0; i<kSPECIES; i++) times[i]=fTrackTime[i];
154 //_______________________________________________________________________
155 void AliESDtrack::SetIntegratedTimes(const Double_t *times) {
156 for (Int_t i=0; i<kSPECIES; i++) fTrackTime[i]=times[i];
159 //_______________________________________________________________________
160 void AliESDtrack::SetITSpid(const Double_t *p) {
161 for (Int_t i=0; i<kSPECIES; i++) fITSr[i]=p[i];
162 SetStatus(AliESDtrack::kITSpid);
165 //_______________________________________________________________________
166 void AliESDtrack::GetITSpid(Double_t *p) const {
167 for (Int_t i=0; i<kSPECIES; i++) p[i]=fITSr[i];
170 //_______________________________________________________________________
171 Int_t AliESDtrack::GetITSclusters(UInt_t *idx) const {
172 //---------------------------------------------------------------------
173 // This function returns indices of the assgined ITS clusters
174 //---------------------------------------------------------------------
175 for (Int_t i=0; i<fITSncls; i++) idx[i]=fITSindex[i];
179 //_______________________________________________________________________
180 Int_t AliESDtrack::GetTPCclusters(UInt_t *idx) const {
181 //---------------------------------------------------------------------
182 // This function returns indices of the assgined ITS clusters
183 //---------------------------------------------------------------------
184 for (Int_t i=0; i<fTPCncls; i++) idx[i]=fTPCindex[i];
188 //_______________________________________________________________________
189 void AliESDtrack::SetTPCpid(const Double_t *p) {
190 for (Int_t i=0; i<kSPECIES; i++) fTPCr[i]=p[i];
191 SetStatus(AliESDtrack::kTPCpid);
194 //_______________________________________________________________________
195 void AliESDtrack::GetTPCpid(Double_t *p) const {
196 for (Int_t i=0; i<kSPECIES; i++) p[i]=fTPCr[i];
199 //_______________________________________________________________________
200 void AliESDtrack::SetTRDpid(const Double_t *p) {
201 for (Int_t i=0; i<kSPECIES; i++) fTRDr[i]=p[i];
202 SetStatus(AliESDtrack::kTRDpid);
205 //_______________________________________________________________________
206 void AliESDtrack::GetTRDpid(Double_t *p) const {
207 for (Int_t i=0; i<kSPECIES; i++) p[i]=fTRDr[i];
210 //_______________________________________________________________________
211 void AliESDtrack::SetTRDpid(Int_t iSpecies, Float_t p)
216 Float_t AliESDtrack::GetTRDpid(Int_t iSpecies) const
218 return fTRDr[iSpecies];
221 //_______________________________________________________________________
222 void AliESDtrack::SetTOFpid(const Double_t *p) {
223 for (Int_t i=0; i<kSPECIES; i++) fTOFr[i]=p[i];
224 SetStatus(AliESDtrack::kTOFpid);
227 //_______________________________________________________________________
228 void AliESDtrack::GetTOFpid(Double_t *p) const {
229 for (Int_t i=0; i<kSPECIES; i++) p[i]=fTOFr[i];
232 //_______________________________________________________________________
233 void AliESDtrack::SetESDpid(const Double_t *p) {
234 for (Int_t i=0; i<kSPECIES; i++) fR[i]=p[i];
235 SetStatus(AliESDtrack::kESDpid);
238 //_______________________________________________________________________
239 void AliESDtrack::GetESDpid(Double_t *p) const {
240 for (Int_t i=0; i<kSPECIES; i++) p[i]=fR[i];