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 **************************************************************************/
18 //-----------------------------------------------------------------
19 // Implementation of the TPC track class
20 // This class is used by the AliTPCtracker class
21 // Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
22 //-----------------------------------------------------------------
24 #include <Riostream.h>
26 #include "AliTPCtrack.h"
27 #include "AliCluster.h"
28 #include "AliBarrelTrack.h"
29 #include "AliESDtrack.h"
33 //_________________________________________________________________________
34 AliTPCtrack::AliTPCtrack(): AliKalmanTrack()
36 fX = fP0 = fP1 = fP2 = fP3 = fP3 = fP4 = 0.0;
38 fNWrong = fNRotation = fNumber = 0; // [SR, 01.04.2003]
41 //_________________________________________________________________________
42 AliTPCtrack::AliTPCtrack(UInt_t index, const Double_t xx[5],
43 const Double_t cc[15], Double_t xref, Double_t alpha) : AliKalmanTrack() {
44 //-----------------------------------------------------------------
45 // This is the main track constructor.
46 //-----------------------------------------------------------------
49 if (fAlpha<-TMath::Pi()) fAlpha += 2*TMath::Pi();
50 if (fAlpha>=TMath::Pi()) fAlpha -= 2*TMath::Pi();
53 fP0=xx[0]; fP1=xx[1]; fP2=xx[2]; fP3=xx[3]; fP4=xx[4];
56 fC10=cc[1]; fC11=cc[2];
57 fC20=cc[3]; fC21=cc[4]; fC22=cc[5];
58 fC30=cc[6]; fC31=cc[7]; fC32=cc[8]; fC33=cc[9];
59 fC40=cc[10]; fC41=cc[11]; fC42=cc[12]; fC43=cc[13]; fC44=cc[14];
62 SetNumberOfClusters(1);
75 //_____________________________________________________________________________
76 AliTPCtrack::AliTPCtrack(const AliKalmanTrack& t,Double_t alpha) :
78 //-----------------------------------------------------------------
79 // Conversion AliKalmanTrack -> AliTPCtrack.
80 //-----------------------------------------------------------------
82 SetNumberOfClusters(0);
86 if (fAlpha < -TMath::Pi()) fAlpha += 2*TMath::Pi();
87 else if (fAlpha >= TMath::Pi()) fAlpha -= 2*TMath::Pi();
89 //Conversion of the track parameters
90 Double_t x,p[5]; t.GetExternalParameters(x,p);
91 fX=x; x=GetConvConst();
98 //Conversion of the covariance matrix
99 Double_t c[15]; t.GetExternalCovariance(c);
100 c[10]/=x; c[11]/=x; c[12]/=x; c[13]/=x; c[14]/=x*x;
102 Double_t c22=fX*fX*c[14] - 2*fX*c[12] + c[5];
103 Double_t c32=fX*c[13] - c[8];
104 Double_t c20=fX*c[10] - c[3], c21=fX*c[11] - c[4], c42=fX*c[14] - c[12];
107 fC10=c[1 ]; fC11=c[2 ];
108 fC20=c20; fC21=c21; fC22=c22;
109 fC30=c[6 ]; fC31=c[7 ]; fC32=c32; fC33=c[9 ];
110 fC40=c[10]; fC41=c[11]; fC42=c42; fC43=c[13]; fC44=c[14];
123 //_____________________________________________________________________________
124 AliTPCtrack::AliTPCtrack(const AliESDtrack& t) : AliKalmanTrack() {
125 //-----------------------------------------------------------------
126 // Conversion AliESDtrack -> AliTPCtrack.
127 //-----------------------------------------------------------------
128 SetNumberOfClusters(t.GetTPCclusters(fIndex));
129 SetLabel(t.GetLabel());
130 SetMass(t.GetMass());
132 fdEdx = t.GetTPCsignal();
133 fAlpha = t.GetAlpha();
134 if (fAlpha < -TMath::Pi()) fAlpha += 2*TMath::Pi();
135 else if (fAlpha >= TMath::Pi()) fAlpha -= 2*TMath::Pi();
137 //Conversion of the track parameters
138 Double_t x,p[5]; t.GetExternalParameters(x,p);
139 fX=x; x=GetConvConst();
146 //Conversion of the covariance matrix
147 Double_t c[15]; t.GetExternalCovariance(c);
148 c[10]/=x; c[11]/=x; c[12]/=x; c[13]/=x; c[14]/=x*x;
150 Double_t c22=fX*fX*c[14] - 2*fX*c[12] + c[5];
151 Double_t c32=fX*c[13] - c[8];
152 Double_t c20=fX*c[10] - c[3], c21=fX*c[11] - c[4], c42=fX*c[14] - c[12];
155 fC10=c[1 ]; fC11=c[2 ];
156 fC20=c20; fC21=c21; fC22=c22;
157 fC30=c[6 ]; fC31=c[7 ]; fC32=c32; fC33=c[9 ];
158 fC40=c[10]; fC41=c[11]; fC42=c42; fC43=c[13]; fC44=c[14];
160 if ((t.GetStatus()&AliESDtrack::kTIME) == 0) return;
162 Double_t times[10]; t.GetIntegratedTimes(times); SetIntegratedTimes(times);
163 SetIntegratedLength(t.GetIntegratedLength());
176 //_____________________________________________________________________________
177 AliTPCtrack::AliTPCtrack(const AliTPCtrack& t) : AliKalmanTrack(t) {
178 //-----------------------------------------------------------------
179 // This is a track copy constructor.
180 //-----------------------------------------------------------------
185 fP0=t.fP0; fP1=t.fP1; fP2=t.fP2; fP3=t.fP3; fP4=t.fP4;
188 fC10=t.fC10; fC11=t.fC11;
189 fC20=t.fC20; fC21=t.fC21; fC22=t.fC22;
190 fC30=t.fC30; fC31=t.fC31; fC32=t.fC32; fC33=t.fC33;
191 fC40=t.fC40; fC41=t.fC41; fC42=t.fC42; fC43=t.fC43; fC44=t.fC44;
193 //Int_t n=GetNumberOfClusters();
194 for (Int_t i=0; i<kMaxRow; i++) fIndex[i]=t.fIndex[i];
198 fNFoundable = t.fNFoundable;
199 fBConstrain = t.fBConstrain;
200 fLastPoint = t.fLastPoint;
201 fFirstPoint = t.fFirstPoint;
202 fRemoval = t.fRemoval ;
203 fTrackType = t.fTrackType;
207 //_____________________________________________________________________________
209 void AliTPCtrack::GetBarrelTrack(AliBarrelTrack *track) {
211 // Create a Barrel Track out of this track
212 // Current track is propagated to the reference plane
218 Double_t xr, vec[5], cov[15];
220 track->SetLabel(GetLabel());
221 track->SetX(fX, fAlpha);
222 track->SetNClusters(GetNumberOfClusters(), GetChi2());
224 GetIntegratedTimes(times);
225 track->SetTime(times, GetIntegratedLength());
227 track->SetMass(GetMass());
228 track->SetdEdX(GetdEdx());
230 track->SetNWrongClusters(fNWrong);
231 track->SetNRotate(fNRotation);
233 GetExternalParameters(xr, vec);
234 track->SetStateVector(vec);
236 GetExternalCovariance(cov);
237 track->SetCovarianceMatrix(cov);
240 //_____________________________________________________________________________
241 Int_t AliTPCtrack::Compare(const TObject *o) const {
242 //-----------------------------------------------------------------
243 // This function compares tracks according to the their curvature
244 //-----------------------------------------------------------------
245 AliTPCtrack *t=(AliTPCtrack*)o;
246 //Double_t co=TMath::Abs(t->Get1Pt());
247 //Double_t c =TMath::Abs(Get1Pt());
248 Double_t co=t->GetSigmaY2()*t->GetSigmaZ2();
249 Double_t c =GetSigmaY2()*GetSigmaZ2();
251 else if (c<co) return -1;
255 //_____________________________________________________________________________
256 void AliTPCtrack::GetExternalCovariance(Double_t cc[15]) const {
257 //-------------------------------------------------------------------------
258 // This function returns an external representation of the covriance matrix.
259 // (See comments in AliTPCtrack.h about external track representation)
260 //-------------------------------------------------------------------------
261 Double_t a=GetConvConst();
263 Double_t c22=fX*fX*fC44-2*fX*fC42+fC22;
264 Double_t c32=fX*fC43-fC32;
265 Double_t c20=fX*fC40-fC20, c21=fX*fC41-fC21, c42=fX*fC44-fC42;
268 cc[1 ]=fC10; cc[2 ]=fC11;
269 cc[3 ]=c20; cc[4 ]=c21; cc[5 ]=c22;
270 cc[6 ]=fC30; cc[7 ]=fC31; cc[8 ]=c32; cc[9 ]=fC33;
271 cc[10]=fC40*a; cc[11]=fC41*a; cc[12]=c42*a; cc[13]=fC43*a; cc[14]=fC44*a*a;
275 //_____________________________________________________________________________
276 Double_t AliTPCtrack::GetPredictedChi2(const AliCluster *c) const
278 //-----------------------------------------------------------------
279 // This function calculates a predicted chi2 increment.
280 //-----------------------------------------------------------------
281 Double_t r00=c->GetSigmaY2(), r01=0., r11=c->GetSigmaZ2();
282 r00+=fC00; r01+=fC10; r11+=fC11;
284 Double_t det=r00*r11 - r01*r01;
285 if (TMath::Abs(det) < 1.e-10) {
286 Int_t n=GetNumberOfClusters();
287 if (n>4) cerr<<n<<" AliKalmanTrack warning: Singular matrix !\n";
290 Double_t tmp=r00; r00=r11; r11=tmp; r01=-r01;
292 Double_t dy=c->GetY() - fP0, dz=c->GetZ() - fP1;
294 return (dy*r00*dy + 2*r01*dy*dz + dz*r11*dz)/det;
297 Double_t AliTPCtrack::GetYat(Double_t xk) const {
298 //-----------------------------------------------------------------
299 // This function calculates the Y-coordinate of a track at the plane x=xk.
300 //-----------------------------------------------------------------
301 Double_t c1=fP4*fX - fP2, r1=TMath::Sqrt(1.- c1*c1);
302 Double_t c2=fP4*xk - fP2;
304 Int_t n=GetNumberOfClusters();
305 if (n>4) cerr<<n<<"AliTPCtrack::GetYat: can't evaluate the y-coord !\n";
308 Double_t r2=TMath::Sqrt(1.- c2*c2);
309 return fP0 + (xk-fX)*(c1+c2)/(r1+r2);
312 //_____________________________________________________________________________
313 Int_t AliTPCtrack::PropagateTo(Double_t xk,Double_t /*x0*/,Double_t rho) {
314 //-----------------------------------------------------------------
315 // This function propagates a track to a reference plane x=xk.
316 //-----------------------------------------------------------------
317 if (TMath::Abs(fP4*xk - fP2) >= 0.9) {
318 // Int_t n=GetNumberOfClusters();
319 //if (n>4) cerr<<n<<" AliTPCtrack warning: Propagation failed !\n";
323 // old position for time [SR, GSI 17.02.2003]
329 Double_t x1=fX, x2=x1+(xk-x1), dx=x2-x1, y1=fP0, z1=fP1;
330 Double_t c1=fP4*x1 - fP2, r1=sqrt(1.- c1*c1);
331 Double_t c2=fP4*x2 - fP2, r2=sqrt(1.- c2*c2);
333 fP0 += dx*(c1+c2)/(r1+r2);
334 fP1 += dx*(c1+c2)/(c1*r2 + c2*r1)*fP3;
337 Double_t rr=r1+r2, cc=c1+c2, xx=x1+x2;
338 Double_t f02=-dx*(2*rr + cc*(c1/r1 + c2/r2))/(rr*rr);
339 Double_t f04= dx*(rr*xx + cc*(c1*x1/r1+c2*x2/r2))/(rr*rr);
340 Double_t cr=c1*r2+c2*r1;
341 Double_t f12=-dx*fP3*(2*cr + cc*(c2*c1/r1-r1 + c1*c2/r2-r2))/(cr*cr);
342 Double_t f13= dx*cc/cr;
343 Double_t f14=dx*fP3*(cr*xx-cc*(r1*x2-c2*c1*x1/r1+r2*x1-c1*c2*x2/r2))/(cr*cr);
346 Double_t b00=f02*fC20 + f04*fC40, b01=f12*fC20 + f14*fC40 + f13*fC30;
347 Double_t b10=f02*fC21 + f04*fC41, b11=f12*fC21 + f14*fC41 + f13*fC31;
348 Double_t b20=f02*fC22 + f04*fC42, b21=f12*fC22 + f14*fC42 + f13*fC32;
349 Double_t b30=f02*fC32 + f04*fC43, b31=f12*fC32 + f14*fC43 + f13*fC33;
350 Double_t b40=f02*fC42 + f04*fC44, b41=f12*fC42 + f14*fC44 + f13*fC43;
353 Double_t a00=f02*b20+f04*b40,a01=f02*b21+f04*b41,a11=f12*b21+f14*b41+f13*b31;
355 //F*C*Ft = C + (a + b + bt)
357 fC10 += a01 + b01 + b10;
368 //Multiple scattering******************
369 Double_t d=sqrt((x1-fX)*(x1-fX)+(y1-fP0)*(y1-fP0)+(z1-fP1)*(z1-fP1));
370 Double_t p2=(1.+ GetTgl()*GetTgl())/(Get1Pt()*Get1Pt());
371 Double_t beta2=p2/(p2 + GetMass()*GetMass());
372 //Double_t theta2=14.1*14.1/(beta2*p2*1e6)*d/x0*rho;
373 Double_t theta2=1.0259e-6*10*10/20/(beta2*p2)*d*rho;
375 Double_t ey=fP4*fX - fP2, ez=fP3;
376 Double_t xz=fP4*ez, zz1=ez*ez+1, xy=fP2+ey;
378 fC22 += (2*ey*ez*ez*fP2+1-ey*ey+ez*ez+fP2*fP2*ez*ez)*theta2;
379 fC32 += ez*zz1*xy*theta2;
380 fC33 += zz1*zz1*theta2;
381 fC42 += xz*ez*xy*theta2;
382 fC43 += xz*zz1*theta2;
383 fC44 += xz*xz*theta2;
385 //Energy losses************************
386 Double_t dE=0.153e-3/beta2*(log(5940*beta2/(1-beta2)) - beta2)*d*rho;
389 fP4*=(1.- sqrt(p2+GetMass()*GetMass())/p2*dE);
392 // Integrated Time [SR, GSI, 17.02.2003]
394 if (IsStartedTimeIntegral()) {
395 Double_t l2 = (fX-oldX)*(fX-oldX)+(fP0-oldY)*(fP0-oldY)+(fP1-oldZ)*(fP1-oldZ);
396 AddTimeStep(TMath::Sqrt(l2));
403 //_____________________________________________________________________________
404 Int_t AliTPCtrack::PropagateToVertex(Double_t x0,Double_t rho)
406 //-----------------------------------------------------------------
407 // This function propagates tracks to the "vertex".
408 //-----------------------------------------------------------------
409 Double_t c=fP4*fX - fP2;
410 Double_t tgf=-fP2/(fP4*fP0 + sqrt(1-c*c));
411 Double_t snf=tgf/sqrt(1.+ tgf*tgf);
412 Double_t xv=(fP2+snf)/fP4;
413 return PropagateTo(xv,x0,rho);
416 //_____________________________________________________________________________
417 Int_t AliTPCtrack::Update(const AliCluster *c, Double_t chisq, UInt_t index) {
418 //-----------------------------------------------------------------
419 // This function associates a cluster with this track.
420 //-----------------------------------------------------------------
422 // update the number of wrong SR[20.03.2003]
423 Int_t absLabel = TMath::Abs(GetLabel());
424 if ( (c->GetLabel(0) != absLabel) &&
425 (c->GetLabel(0) != absLabel) &&
426 (c->GetLabel(0) != absLabel)) fNWrong++;
429 Double_t r00=c->GetSigmaY2(), r01=0., r11=c->GetSigmaZ2();
430 r00+=fC00; r01+=fC10; r11+=fC11;
431 Double_t det=r00*r11 - r01*r01;
432 Double_t tmp=r00; r00=r11/det; r11=tmp/det; r01=-r01/det;
434 Double_t k00=fC00*r00+fC10*r01, k01=fC00*r01+fC10*r11;
435 Double_t k10=fC10*r00+fC11*r01, k11=fC10*r01+fC11*r11;
436 Double_t k20=fC20*r00+fC21*r01, k21=fC20*r01+fC21*r11;
437 Double_t k30=fC30*r00+fC31*r01, k31=fC30*r01+fC31*r11;
438 Double_t k40=fC40*r00+fC41*r01, k41=fC40*r01+fC41*r11;
440 Double_t dy=c->GetY() - fP0, dz=c->GetZ() - fP1;
441 Double_t cur=fP4 + k40*dy + k41*dz, eta=fP2 + k20*dy + k21*dz;
442 if (TMath::Abs(cur*fX-eta) >= 0.9) {
443 // Int_t n=GetNumberOfClusters();
444 //if (n>4) cerr<<n<<" AliTPCtrack warning: Filtering failed !\n";
448 fP0 += k00*dy + k01*dz;
449 fP1 += k10*dy + k11*dz;
451 fP3 += k30*dy + k31*dz;
454 Double_t c01=fC10, c02=fC20, c03=fC30, c04=fC40;
455 Double_t c12=fC21, c13=fC31, c14=fC41;
457 fC00-=k00*fC00+k01*fC10; fC10-=k00*c01+k01*fC11;
458 fC20-=k00*c02+k01*c12; fC30-=k00*c03+k01*c13;
459 fC40-=k00*c04+k01*c14;
461 fC11-=k10*c01+k11*fC11;
462 fC21-=k10*c02+k11*c12; fC31-=k10*c03+k11*c13;
463 fC41-=k10*c04+k11*c14;
465 fC22-=k20*c02+k21*c12; fC32-=k20*c03+k21*c13;
466 fC42-=k20*c04+k21*c14;
468 fC33-=k30*c03+k31*c13;
469 fC43-=k40*c03+k41*c13;
471 fC44-=k40*c04+k41*c14;
473 Int_t n=GetNumberOfClusters();
475 SetNumberOfClusters(n+1);
476 SetChi2(GetChi2()+chisq);
481 //_____________________________________________________________________________
482 Int_t AliTPCtrack::Rotate(Double_t alpha)
484 //-----------------------------------------------------------------
485 // This function rotates this track.
486 //-----------------------------------------------------------------
488 if (alpha != 0) fNRotation++; // [SR, 01.04.2003]
491 if (fAlpha<-TMath::Pi()) fAlpha += 2*TMath::Pi();
492 if (fAlpha>=TMath::Pi()) fAlpha -= 2*TMath::Pi();
494 Double_t x1=fX, y1=fP0;
495 Double_t ca=cos(alpha), sa=sin(alpha);
496 Double_t r1=fP4*fX - fP2;
500 fP2=fP2*ca + (fP4*y1 + sqrt(1.- r1*r1))*sa;
502 Double_t r2=fP4*fX - fP2;
503 if (TMath::Abs(r2) >= 0.99999) {
504 Int_t n=GetNumberOfClusters();
505 if (n>4) cerr<<n<<" AliTPCtrack warning: Rotation failed !\n";
509 Double_t y0=fP0 + sqrt(1.- r2*r2)/fP4;
510 if ((fP0-y0)*fP4 >= 0.) {
511 Int_t n=GetNumberOfClusters();
512 if (n>4) cerr<<n<<" AliTPCtrack warning: Rotation failed !!!\n";
517 Double_t f00=ca-1, f24=(y1 - r1*x1/sqrt(1.- r1*r1))*sa,
518 f20=fP4*sa, f22=(ca + sa*r1/sqrt(1.- r1*r1))-1;
521 Double_t b00=fC00*f00, b02=fC00*f20+fC40*f24+fC20*f22;
522 Double_t b10=fC10*f00, b12=fC10*f20+fC41*f24+fC21*f22;
523 Double_t b20=fC20*f00, b22=fC20*f20+fC42*f24+fC22*f22;
524 Double_t b30=fC30*f00, b32=fC30*f20+fC43*f24+fC32*f22;
525 Double_t b40=fC40*f00, b42=fC40*f20+fC44*f24+fC42*f22;
528 Double_t a00=f00*b00, a02=f00*b02, a22=f20*b02+f24*b42+f22*b22;
530 // *** Double_t dy2=fCyy;
532 //F*C*Ft = C + (a + b + bt)
543 // *** fCyy+=dy2*sa*sa*r1*r1/(1.- r1*r1);
544 // *** fCzz+=d2y*sa*sa*fT*fT/(1.- r1*r1);
549 void AliTPCtrack::ResetCovariance() {
550 //------------------------------------------------------------------
551 //This function makes a track forget its history :)
552 //------------------------------------------------------------------
556 fC20=0.; fC21=0.; fC22*=10.;
557 fC30=0.; fC31=0.; fC32=0.; fC33*=10.;
558 fC40=0.; fC41=0.; fC42=0.; fC43=0.; fC44*=10.;
562 ////////////////////////////////////////////////////////////////////////
563 Double_t AliTPCtrack::Phi() const {
567 Double_t phi = TMath::ASin(GetSnp()) + fAlpha;
568 if (phi<0) phi+=2*TMath::Pi();
569 if (phi>=2*TMath::Pi()) phi-=2*TMath::Pi();
572 ////////////////////////////////////////////////////////////////////////
576 ////////////////////////////////////////////////////////////////////////
579 Float_t AliTPCtrack::Density(Int_t row0, Int_t row1)
582 // calculate cluster density
585 //if (row0<fFirstPoint) row0 = fFirstPoint;
586 if (row1>fLastPoint) row1 = fLastPoint;
589 for (Int_t i=row0;i<=row1;i++){
590 // Int_t index = fClusterIndex[i];
591 Int_t index = fIndex[i];
592 if (index!=-1) good++;
593 if (index>0) found++;
596 if (good>0) density = Float_t(found)/Float_t(good);
601 Float_t AliTPCtrack::Density2(Int_t row0, Int_t row1)
604 // calculate cluster density
608 for (Int_t i=row0;i<=row1;i++){
609 Int_t index = fIndex[i];
610 if (index!=-1) good++;
611 if (index>0) found++;
614 if (good>0) density = Float_t(found)/Float_t(good);
619 Double_t AliTPCtrack::GetZat0() const
622 // return virtual z - supposing that x = 0
623 if (TMath::Abs(fP2)>1) return 0;
624 if (TMath::Abs(fX*fP4-fP2)>1) return 0;
625 Double_t vz = fP1+fP3/fP4*(asin(-fP2)-asin(fX*fP4-fP2));
630 Double_t AliTPCtrack::GetD(Double_t x, Double_t y) const {
631 //------------------------------------------------------------------
632 // This function calculates the transverse impact parameter
633 // with respect to a point with global coordinates (x,y)
634 //------------------------------------------------------------------
635 //Double_t xt=fX, yt=fP0;
637 Double_t sn=TMath::Sin(fAlpha), cs=TMath::Cos(fAlpha);
638 Double_t a = x*cs + y*sn;
639 y = -x*sn + y*cs; x=a;
641 Double_t r = TMath::Abs(1/fP4);
642 Double_t x0 = TMath::Abs(fP2*r);
644 y0= fP0+TMath::Sqrt(1-(fP4*fX-fP2)*(fP4*fX-fP2))/fP4;
646 Double_t delta = TMath::Sqrt((x-x0)*(x-x0)+(y-y0)*(y-y0));
647 // Double_t delta = TMath::Sqrt(TMath::Abs(x*x-2*x0*x+x0*x0+ y*y-2*y*y0+y0*y0));
648 delta -= TMath::Abs(r);