]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/AliExternalTrackParam.cxx
The GEANT parameterization of the Bethe-Bloch formula is now a function of STEER...
[u/mrichter/AliRoot.git] / STEER / AliExternalTrackParam.cxx
CommitLineData
51ad6848 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/* $Id$ */
17
18///////////////////////////////////////////////////////////////////////////////
19// //
49d13e89 20// Implementation of the external track parameterisation class. //
51ad6848 21// //
49d13e89 22// This parameterisation is used to exchange tracks between the detectors. //
23// A set of functions returning the position and the momentum of tracks //
24// in the global coordinate system as well as the track impact parameters //
25// are implemented.
26// Origin: I.Belikov, CERN, Jouri.Belikov@cern.ch //
51ad6848 27///////////////////////////////////////////////////////////////////////////////
4b189f98 28#include <TMatrixDSym.h>
d46683db 29#include <TPolyMarker3D.h>
30#include <TVector3.h>
31
51ad6848 32#include "AliExternalTrackParam.h"
d46683db 33#include "AliMathBase.h"
58e536c5 34#include "AliVVertex.h"
6c94f330 35#include "AliLog.h"
51ad6848 36
37ClassImp(AliExternalTrackParam)
38
ed5f2849 39Double32_t AliExternalTrackParam::fgMostProbablePt=kMostProbablePt;
40
51ad6848 41//_____________________________________________________________________________
90e48c0c 42AliExternalTrackParam::AliExternalTrackParam() :
4f6e22bd 43 AliVTrack(),
90e48c0c 44 fX(0),
c9ec41e8 45 fAlpha(0)
51ad6848 46{
90e48c0c 47 //
48 // default constructor
49 //
c9ec41e8 50 for (Int_t i = 0; i < 5; i++) fP[i] = 0;
51 for (Int_t i = 0; i < 15; i++) fC[i] = 0;
51ad6848 52}
53
6c94f330 54//_____________________________________________________________________________
55AliExternalTrackParam::AliExternalTrackParam(const AliExternalTrackParam &track):
4f6e22bd 56 AliVTrack(track),
6c94f330 57 fX(track.fX),
58 fAlpha(track.fAlpha)
59{
60 //
61 // copy constructor
62 //
63 for (Int_t i = 0; i < 5; i++) fP[i] = track.fP[i];
64 for (Int_t i = 0; i < 15; i++) fC[i] = track.fC[i];
65}
66
def9660e 67//_____________________________________________________________________________
68AliExternalTrackParam& AliExternalTrackParam::operator=(const AliExternalTrackParam &trkPar)
69{
70 //
71 // assignment operator
72 //
73
74 if (this!=&trkPar) {
4f6e22bd 75 AliVTrack::operator=(trkPar);
def9660e 76 fX = trkPar.fX;
77 fAlpha = trkPar.fAlpha;
78
79 for (Int_t i = 0; i < 5; i++) fP[i] = trkPar.fP[i];
80 for (Int_t i = 0; i < 15; i++) fC[i] = trkPar.fC[i];
81 }
82
83 return *this;
84}
85
51ad6848 86//_____________________________________________________________________________
87AliExternalTrackParam::AliExternalTrackParam(Double_t x, Double_t alpha,
88 const Double_t param[5],
90e48c0c 89 const Double_t covar[15]) :
4f6e22bd 90 AliVTrack(),
90e48c0c 91 fX(x),
c9ec41e8 92 fAlpha(alpha)
51ad6848 93{
90e48c0c 94 //
95 // create external track parameters from given arguments
96 //
c9ec41e8 97 for (Int_t i = 0; i < 5; i++) fP[i] = param[i];
98 for (Int_t i = 0; i < 15; i++) fC[i] = covar[i];
51ad6848 99}
100
4f6e22bd 101//_____________________________________________________________________________
102AliExternalTrackParam::AliExternalTrackParam(const AliVTrack *vTrack) :
103 AliVTrack(),
104 fX(0.),
105 fAlpha(0.)
106{
107 //
108 // constructor from virtual track
109 //
892be05f 110 Double_t xyz[3],pxpypz[3],cv[21];
111 vTrack->GetXYZ(xyz);
112 pxpypz[0]=vTrack->Px();
113 pxpypz[1]=vTrack->Py();
114 pxpypz[2]=vTrack->Pz();
4f6e22bd 115 vTrack->GetCovarianceXYZPxPyPz(cv);
116 Short_t sign = (Short_t)vTrack->Charge();
117
118 Set(xyz,pxpypz,cv,sign);
119}
120
90e48c0c 121//_____________________________________________________________________________
da4e3deb 122AliExternalTrackParam::AliExternalTrackParam(Double_t xyz[3],Double_t pxpypz[3],
123 Double_t cv[21],Short_t sign) :
4f6e22bd 124 AliVTrack(),
da4e3deb 125 fX(0.),
126 fAlpha(0.)
4f6e22bd 127{
128 //
129 // constructor from the global parameters
130 //
131
132 Set(xyz,pxpypz,cv,sign);
133}
134
135//_____________________________________________________________________________
136void AliExternalTrackParam::Set(Double_t xyz[3],Double_t pxpypz[3],
137 Double_t cv[21],Short_t sign)
da4e3deb 138{
139 //
140 // create external track parameters from the global parameters
141 // x,y,z,px,py,pz and their 6x6 covariance matrix
142 // A.Dainese 10.10.08
143
144 // Calculate alpha: the rotation angle of the corresponding local system
145 fAlpha = TMath::ATan2(pxpypz[1],pxpypz[0]);
146
147 // Get the vertex of origin and the momentum
148 TVector3 ver(xyz[0],xyz[1],xyz[2]);
149 TVector3 mom(pxpypz[0],pxpypz[1],pxpypz[2]);
150
151 // Rotate to the local coordinate system
152 ver.RotateZ(-fAlpha);
153 mom.RotateZ(-fAlpha);
154
155 // x of the reference plane
156 fX = ver.X();
157
158 Double_t charge = (Double_t)sign;
159
160 fP[0] = ver.Y();
161 fP[1] = ver.Z();
162 fP[2] = TMath::Sin(mom.Phi());
163 fP[3] = mom.Pz()/mom.Pt();
164 fP[4] = TMath::Sign(1/mom.Pt(),charge);
165
166 // Covariance matrix (formulas to be simplified)
167
168 Double_t pt=1./TMath::Abs(fP[4]);
169 Double_t cs=TMath::Cos(fAlpha), sn=TMath::Sin(fAlpha);
170 Double_t r=TMath::Sqrt((1.-fP[2])*(1.+fP[2]));
171
172 Double_t m00=-sn;// m10=cs;
173 Double_t m23=-pt*(sn + fP[2]*cs/r), m43=-pt*pt*(r*cs - fP[2]*sn);
174 Double_t m24= pt*(cs - fP[2]*sn/r), m44=-pt*pt*(r*sn + fP[2]*cs);
175 Double_t m35=pt, m45=-pt*pt*fP[3];
176
177 m43*=GetSign();
178 m44*=GetSign();
179 m45*=GetSign();
180
181 Double_t cv34 = TMath::Sqrt(cv[3 ]*cv[3 ]+cv[4 ]*cv[4 ]);
182 Double_t a1=cv[13]-cv[9]*(m23*m44+m43*m24)/m23/m43;
183 Double_t a2=m23*m24-m23*(m23*m44+m43*m24)/m43;
184 Double_t a3=m43*m44-m43*(m23*m44+m43*m24)/m23;
185 Double_t a4=cv[14]-2.*cv[9]*m24*m44/m23/m43;
186 Double_t a5=m24*m24-2.*m24*m44*m23/m43;
187 Double_t a6=m44*m44-2.*m24*m44*m43/m23;
188
189 fC[0 ] = cv[0 ]+cv[2 ];
190 fC[1 ] = TMath::Sign(cv34,cv[3 ]/m00);
191 fC[2 ] = cv[5 ];
192 fC[3 ] = (cv[10]/m44-cv[6]/m43)/(m24/m44-m23/m43)/m00;
193 fC[10] = (cv[6]/m00-fC[3 ]*m23)/m43;
194 fC[6 ] = (cv[15]/m00-fC[10]*m45)/m35;
195 fC[4 ] = (cv[12]-cv[8]*m44/m43)/(m24-m23*m44/m43);
196 fC[11] = (cv[8]-fC[4]*m23)/m43;
197 fC[7 ] = cv[17]/m35-fC[11]*m45/m35;
198 fC[5 ] = TMath::Abs((a4-a6*a1/a3)/(a5-a6*a2/a3));
199 fC[14] = TMath::Abs(a1/a3-a2*fC[5]/a3);
200 fC[12] = (cv[9]-fC[5]*m23*m23-fC[14]*m43*m43)/m23/m43;
201 Double_t b1=cv[18]-fC[12]*m23*m45-fC[14]*m43*m45;
202 Double_t b2=m23*m35;
203 Double_t b3=m43*m35;
204 Double_t b4=cv[19]-fC[12]*m24*m45-fC[14]*m44*m45;
205 Double_t b5=m24*m35;
206 Double_t b6=m44*m35;
207 fC[8 ] = (b4-b6*b1/b3)/(b5-b6*b2/b3);
208 fC[13] = b1/b3-b2*fC[8]/b3;
209 fC[9 ] = TMath::Abs((cv[20]-fC[14]*(m45*m45)-fC[13]*2.*m35*m45)/(m35*m35));
4f6e22bd 210
211 return;
da4e3deb 212}
213
51ad6848 214//_____________________________________________________________________________
c9ec41e8 215void AliExternalTrackParam::Reset() {
1530f89c 216 //
217 // Resets all the parameters to 0
218 //
c9ec41e8 219 fX=fAlpha=0.;
220 for (Int_t i = 0; i < 5; i++) fP[i] = 0;
221 for (Int_t i = 0; i < 15; i++) fC[i] = 0;
51ad6848 222}
223
3775b0ca 224//_____________________________________________________________________________
225void AliExternalTrackParam::AddCovariance(const Double_t c[15]) {
226 //
227 // Add "something" to the track covarince matrix.
228 // May be needed to account for unknown mis-calibration/mis-alignment
229 //
230 fC[0] +=c[0];
231 fC[1] +=c[1]; fC[2] +=c[2];
232 fC[3] +=c[3]; fC[4] +=c[4]; fC[5] +=c[5];
233 fC[6] +=c[6]; fC[7] +=c[7]; fC[8] +=c[8]; fC[9] +=c[9];
234 fC[10]+=c[10]; fC[11]+=c[11]; fC[12]+=c[12]; fC[13]+=c[13]; fC[14]+=c[14];
235}
236
237
c9ec41e8 238Double_t AliExternalTrackParam::GetP() const {
239 //---------------------------------------------------------------------
240 // This function returns the track momentum
241 // Results for (nearly) straight tracks are meaningless !
242 //---------------------------------------------------------------------
06fb4a2f 243 if (TMath::Abs(fP[4])<=kAlmost0) return kVeryBig;
c9ec41e8 244 return TMath::Sqrt(1.+ fP[3]*fP[3])/TMath::Abs(fP[4]);
51ad6848 245}
246
1d99986f 247Double_t AliExternalTrackParam::Get1P() const {
248 //---------------------------------------------------------------------
249 // This function returns the 1/(track momentum)
250 //---------------------------------------------------------------------
251 return TMath::Abs(fP[4])/TMath::Sqrt(1.+ fP[3]*fP[3]);
252}
253
c9ec41e8 254//_______________________________________________________________________
c7bafca9 255Double_t AliExternalTrackParam::GetD(Double_t x,Double_t y,Double_t b) const {
c9ec41e8 256 //------------------------------------------------------------------
257 // This function calculates the transverse impact parameter
258 // with respect to a point with global coordinates (x,y)
259 // in the magnetic field "b" (kG)
260 //------------------------------------------------------------------
5773defd 261 if (TMath::Abs(b) < kAlmost0Field) return GetLinearD(x,y);
1530f89c 262 Double_t rp4=GetC(b);
c9ec41e8 263
264 Double_t xt=fX, yt=fP[0];
265
266 Double_t sn=TMath::Sin(fAlpha), cs=TMath::Cos(fAlpha);
267 Double_t a = x*cs + y*sn;
268 y = -x*sn + y*cs; x=a;
269 xt-=x; yt-=y;
270
271 sn=rp4*xt - fP[2]; cs=rp4*yt + TMath::Sqrt(1.- fP[2]*fP[2]);
272 a=2*(xt*fP[2] - yt*TMath::Sqrt(1.- fP[2]*fP[2]))-rp4*(xt*xt + yt*yt);
1530f89c 273 return -a/(1 + TMath::Sqrt(sn*sn + cs*cs));
274}
275
276//_______________________________________________________________________
277void AliExternalTrackParam::
278GetDZ(Double_t x, Double_t y, Double_t z, Double_t b, Float_t dz[2]) const {
279 //------------------------------------------------------------------
280 // This function calculates the transverse and longitudinal impact parameters
281 // with respect to a point with global coordinates (x,y)
282 // in the magnetic field "b" (kG)
283 //------------------------------------------------------------------
284 Double_t f1 = fP[2], r1 = TMath::Sqrt(1. - f1*f1);
285 Double_t xt=fX, yt=fP[0];
286 Double_t sn=TMath::Sin(fAlpha), cs=TMath::Cos(fAlpha);
287 Double_t a = x*cs + y*sn;
288 y = -x*sn + y*cs; x=a;
289 xt-=x; yt-=y;
290
291 Double_t rp4=GetC(b);
292 if ((TMath::Abs(b) < kAlmost0Field) || (TMath::Abs(rp4) < kAlmost0)) {
293 dz[0] = -(xt*f1 - yt*r1);
294 dz[1] = fP[1] + (dz[0]*f1 - xt)/r1*fP[3] - z;
295 return;
296 }
297
298 sn=rp4*xt - f1; cs=rp4*yt + r1;
299 a=2*(xt*f1 - yt*r1)-rp4*(xt*xt + yt*yt);
300 Double_t rr=TMath::Sqrt(sn*sn + cs*cs);
301 dz[0] = -a/(1 + rr);
302 Double_t f2 = -sn/rr, r2 = TMath::Sqrt(1. - f2*f2);
303 dz[1] = fP[1] + fP[3]/rp4*TMath::ASin(f2*r1 - f1*r2) - z;
51ad6848 304}
305
49d13e89 306//_______________________________________________________________________
307Double_t AliExternalTrackParam::GetLinearD(Double_t xv,Double_t yv) const {
308 //------------------------------------------------------------------
309 // This function calculates the transverse impact parameter
310 // with respect to a point with global coordinates (xv,yv)
311 // neglecting the track curvature.
312 //------------------------------------------------------------------
313 Double_t sn=TMath::Sin(fAlpha), cs=TMath::Cos(fAlpha);
314 Double_t x= xv*cs + yv*sn;
315 Double_t y=-xv*sn + yv*cs;
316
317 Double_t d = (fX-x)*fP[2] - (fP[0]-y)*TMath::Sqrt(1.- fP[2]*fP[2]);
318
1530f89c 319 return -d;
49d13e89 320}
321
116b445b 322Bool_t AliExternalTrackParam::CorrectForMeanMaterial
7dded1d5 323(Double_t xOverX0, Double_t xTimesRho, Double_t mass, Bool_t anglecorr,
324 Double_t (*Bethe)(Double_t)) {
116b445b 325 //------------------------------------------------------------------
326 // This function corrects the track parameters for the crossed material.
327 // "xOverX0" - X/X0, the thickness in units of the radiation length.
328 // "xTimesRho" - is the product length*density (g/cm^2).
329 // "mass" - the mass of this particle (GeV/c^2).
330 //------------------------------------------------------------------
331 Double_t &fP2=fP[2];
332 Double_t &fP3=fP[3];
333 Double_t &fP4=fP[4];
334
335 Double_t &fC22=fC[5];
336 Double_t &fC33=fC[9];
337 Double_t &fC43=fC[13];
338 Double_t &fC44=fC[14];
339
7dded1d5 340 //Apply angle correction, if requested
341 if(anglecorr) {
342 Double_t angle=TMath::Sqrt((1.+ fP3*fP3)/(1.- fP2*fP2));
343 xOverX0 *=angle;
344 xTimesRho *=angle;
345 }
346
116b445b 347 Double_t p=GetP();
348 Double_t p2=p*p;
349 Double_t beta2=p2/(p2 + mass*mass);
116b445b 350
351 //Multiple scattering******************
352 if (xOverX0 != 0) {
353 Double_t theta2=14.1*14.1/(beta2*p2*1e6)*TMath::Abs(xOverX0);
9a53dcc6 354 if(theta2>TMath::Pi()*TMath::Pi()) return kFALSE;
116b445b 355 //Double_t theta2=1.0259e-6*14*14/28/(beta2*p2)*TMath::Abs(d)*9.36*2.33;
356 fC22 += theta2*(1.- fP2*fP2)*(1. + fP3*fP3);
357 fC33 += theta2*(1. + fP3*fP3)*(1. + fP3*fP3);
358 fC43 += theta2*fP3*fP4*(1. + fP3*fP3);
359 fC44 += theta2*fP3*fP4*fP3*fP4;
360 }
361
362 //Energy losses************************
363 if ((xTimesRho != 0.) && (beta2 < 1.)) {
d46683db 364 Double_t dE=Bethe(p/mass)*xTimesRho;
116b445b 365 Double_t e=TMath::Sqrt(p2 + mass*mass);
366 if ( TMath::Abs(dE) > 0.3*e ) return kFALSE; //30% energy loss is too much!
367 fP4*=(1.- e/p2*dE);
4b2fa3ce 368 if (TMath::Abs(fP4)>100.) return kFALSE; // Do not track below 10 MeV/c
369
116b445b 370
371 // Approximate energy loss fluctuation (M.Ivanov)
372 const Double_t knst=0.07; // To be tuned.
373 Double_t sigmadE=knst*TMath::Sqrt(TMath::Abs(dE));
374 fC44+=((sigmadE*e/p2*fP4)*(sigmadE*e/p2*fP4));
375
376 }
377
378 return kTRUE;
379}
380
381
ee5dba5e 382Bool_t AliExternalTrackParam::CorrectForMaterial
383(Double_t d, Double_t x0, Double_t mass, Double_t (*Bethe)(Double_t)) {
c7bafca9 384 //------------------------------------------------------------------
116b445b 385 // Deprecated function !
386 // Better use CorrectForMeanMaterial instead of it.
387 //
c7bafca9 388 // This function corrects the track parameters for the crossed material
389 // "d" - the thickness (fraction of the radiation length)
390 // "x0" - the radiation length (g/cm^2)
391 // "mass" - the mass of this particle (GeV/c^2)
392 //------------------------------------------------------------------
393 Double_t &fP2=fP[2];
394 Double_t &fP3=fP[3];
395 Double_t &fP4=fP[4];
396
397 Double_t &fC22=fC[5];
398 Double_t &fC33=fC[9];
399 Double_t &fC43=fC[13];
400 Double_t &fC44=fC[14];
401
7b5ef2e6 402 Double_t p=GetP();
403 Double_t p2=p*p;
c7bafca9 404 Double_t beta2=p2/(p2 + mass*mass);
405 d*=TMath::Sqrt((1.+ fP3*fP3)/(1.- fP2*fP2));
406
407 //Multiple scattering******************
408 if (d!=0) {
409 Double_t theta2=14.1*14.1/(beta2*p2*1e6)*TMath::Abs(d);
9a53dcc6 410 if(theta2>TMath::Pi()*TMath::Pi()) return kFALSE;
c7bafca9 411 //Double_t theta2=1.0259e-6*14*14/28/(beta2*p2)*TMath::Abs(d)*9.36*2.33;
412 fC22 += theta2*(1.- fP2*fP2)*(1. + fP3*fP3);
413 fC33 += theta2*(1. + fP3*fP3)*(1. + fP3*fP3);
414 fC43 += theta2*fP3*fP4*(1. + fP3*fP3);
415 fC44 += theta2*fP3*fP4*fP3*fP4;
416 }
417
418 //Energy losses************************
8fc1985d 419 if (x0!=0. && beta2<1) {
c7bafca9 420 d*=x0;
d46683db 421 Double_t dE=Bethe(p/mass)*d;
ee5dba5e 422 Double_t e=TMath::Sqrt(p2 + mass*mass);
ae666100 423 if ( TMath::Abs(dE) > 0.3*e ) return kFALSE; //30% energy loss is too much!
ee5dba5e 424 fP4*=(1.- e/p2*dE);
425
426 // Approximate energy loss fluctuation (M.Ivanov)
ed5f2849 427 const Double_t knst=0.07; // To be tuned.
428 Double_t sigmadE=knst*TMath::Sqrt(TMath::Abs(dE));
ee5dba5e 429 fC44+=((sigmadE*e/p2*fP4)*(sigmadE*e/p2*fP4));
430
c7bafca9 431 }
432
433 return kTRUE;
434}
435
d46683db 436Double_t AliExternalTrackParam::BetheBlochSolid(Double_t bg) {
ee5dba5e 437 //------------------------------------------------------------------
d46683db 438 // This is an approximation of the Bethe-Bloch formula,
439 // reasonable for solid materials.
440 // All the parameters are, in fact, for Si.
441 // The returned value is in [GeV]
ee5dba5e 442 //------------------------------------------------------------------
a821848c 443
d46683db 444 return AliMathBase::BetheBlochGeant(bg);
445}
ee5dba5e 446
d46683db 447Double_t AliExternalTrackParam::BetheBlochGas(Double_t bg) {
448 //------------------------------------------------------------------
449 // This is an approximation of the Bethe-Bloch formula,
450 // reasonable for gas materials.
451 // All the parameters are, in fact, for Ne.
452 // The returned value is in [GeV]
453 //------------------------------------------------------------------
454
455 const Double_t rho = 0.9e-3;
456 const Double_t x0 = 2.;
457 const Double_t x1 = 4.;
458 const Double_t mI = 140.e-9;
459 const Double_t mZA = 0.49555;
460
461 return AliMathBase::BetheBlochGeant(bg,rho,x0,x1,mI,mZA);
ee5dba5e 462}
463
49d13e89 464Bool_t AliExternalTrackParam::Rotate(Double_t alpha) {
465 //------------------------------------------------------------------
466 // Transform this track to the local coord. system rotated
467 // by angle "alpha" (rad) with respect to the global coord. system.
468 //------------------------------------------------------------------
dfcef74c 469 if (TMath::Abs(fP[2]) >= kAlmost1) {
470 AliError(Form("Precondition is not satisfied: |sin(phi)|>1 ! %f",fP[2]));
471 return kFALSE;
472 }
473
49d13e89 474 if (alpha < -TMath::Pi()) alpha += 2*TMath::Pi();
475 else if (alpha >= TMath::Pi()) alpha -= 2*TMath::Pi();
476
477 Double_t &fP0=fP[0];
478 Double_t &fP2=fP[2];
479 Double_t &fC00=fC[0];
480 Double_t &fC10=fC[1];
481 Double_t &fC20=fC[3];
482 Double_t &fC21=fC[4];
483 Double_t &fC22=fC[5];
484 Double_t &fC30=fC[6];
485 Double_t &fC32=fC[8];
486 Double_t &fC40=fC[10];
487 Double_t &fC42=fC[12];
488
489 Double_t x=fX;
490 Double_t ca=TMath::Cos(alpha-fAlpha), sa=TMath::Sin(alpha-fAlpha);
491 Double_t sf=fP2, cf=TMath::Sqrt(1.- fP2*fP2);
492
dfcef74c 493 Double_t tmp=sf*ca - cf*sa;
494 if (TMath::Abs(tmp) >= kAlmost1) return kFALSE;
495
49d13e89 496 fAlpha = alpha;
497 fX = x*ca + fP0*sa;
498 fP0= -x*sa + fP0*ca;
dfcef74c 499 fP2= tmp;
49d13e89 500
06fb4a2f 501 if (TMath::Abs(cf)<kAlmost0) {
502 AliError(Form("Too small cosine value %f",cf));
503 cf = kAlmost0;
504 }
505
49d13e89 506 Double_t rr=(ca+sf/cf*sa);
507
508 fC00 *= (ca*ca);
509 fC10 *= ca;
510 fC20 *= ca*rr;
511 fC21 *= rr;
512 fC22 *= rr*rr;
513 fC30 *= ca;
514 fC32 *= rr;
515 fC40 *= ca;
516 fC42 *= rr;
517
518 return kTRUE;
519}
520
521Bool_t AliExternalTrackParam::PropagateTo(Double_t xk, Double_t b) {
522 //----------------------------------------------------------------
523 // Propagate this track to the plane X=xk (cm) in the field "b" (kG)
524 //----------------------------------------------------------------
49d13e89 525 Double_t dx=xk-fX;
e421f556 526 if (TMath::Abs(dx)<=kAlmost0) return kTRUE;
18ebc5ef 527
1530f89c 528 Double_t crv=GetC(b);
5773defd 529 if (TMath::Abs(b) < kAlmost0Field) crv=0.;
530
49d13e89 531 Double_t f1=fP[2], f2=f1 + crv*dx;
bbefa4c4 532 if (TMath::Abs(f1) >= kAlmost1) return kFALSE;
49d13e89 533 if (TMath::Abs(f2) >= kAlmost1) return kFALSE;
534
535 Double_t &fP0=fP[0], &fP1=fP[1], &fP2=fP[2], &fP3=fP[3], &fP4=fP[4];
536 Double_t
537 &fC00=fC[0],
538 &fC10=fC[1], &fC11=fC[2],
539 &fC20=fC[3], &fC21=fC[4], &fC22=fC[5],
540 &fC30=fC[6], &fC31=fC[7], &fC32=fC[8], &fC33=fC[9],
541 &fC40=fC[10], &fC41=fC[11], &fC42=fC[12], &fC43=fC[13], &fC44=fC[14];
542
543 Double_t r1=TMath::Sqrt(1.- f1*f1), r2=TMath::Sqrt(1.- f2*f2);
544
545 fX=xk;
546 fP0 += dx*(f1+f2)/(r1+r2);
18ebc5ef 547 fP1 += dx*(r2 + f2*(f1+f2)/(r1+r2))*fP3; // Many thanks to P.Hristov !
49d13e89 548 fP2 += dx*crv;
549
550 //f = F - 1
551
552 Double_t f02= dx/(r1*r1*r1); Double_t cc=crv/fP4;
553 Double_t f04=0.5*dx*dx/(r1*r1*r1); f04*=cc;
554 Double_t f12= dx*fP3*f1/(r1*r1*r1);
555 Double_t f14=0.5*dx*dx*fP3*f1/(r1*r1*r1); f14*=cc;
556 Double_t f13= dx/r1;
557 Double_t f24= dx; f24*=cc;
558
559 //b = C*ft
560 Double_t b00=f02*fC20 + f04*fC40, b01=f12*fC20 + f14*fC40 + f13*fC30;
561 Double_t b02=f24*fC40;
562 Double_t b10=f02*fC21 + f04*fC41, b11=f12*fC21 + f14*fC41 + f13*fC31;
563 Double_t b12=f24*fC41;
564 Double_t b20=f02*fC22 + f04*fC42, b21=f12*fC22 + f14*fC42 + f13*fC32;
565 Double_t b22=f24*fC42;
566 Double_t b40=f02*fC42 + f04*fC44, b41=f12*fC42 + f14*fC44 + f13*fC43;
567 Double_t b42=f24*fC44;
568 Double_t b30=f02*fC32 + f04*fC43, b31=f12*fC32 + f14*fC43 + f13*fC33;
569 Double_t b32=f24*fC43;
570
571 //a = f*b = f*C*ft
572 Double_t a00=f02*b20+f04*b40,a01=f02*b21+f04*b41,a02=f02*b22+f04*b42;
573 Double_t a11=f12*b21+f14*b41+f13*b31,a12=f12*b22+f14*b42+f13*b32;
574 Double_t a22=f24*b42;
575
576 //F*C*Ft = C + (b + bt + a)
577 fC00 += b00 + b00 + a00;
578 fC10 += b10 + b01 + a01;
579 fC20 += b20 + b02 + a02;
580 fC30 += b30;
581 fC40 += b40;
582 fC11 += b11 + b11 + a11;
583 fC21 += b21 + b12 + a12;
584 fC31 += b31;
585 fC41 += b41;
586 fC22 += b22 + b22 + a22;
587 fC32 += b32;
588 fC42 += b42;
589
590 return kTRUE;
591}
592
052daaff 593void AliExternalTrackParam::Propagate(Double_t len, Double_t x[3],
594Double_t p[3], Double_t bz) const {
595 //+++++++++++++++++++++++++++++++++++++++++
596 // Origin: K. Shileev (Kirill.Shileev@cern.ch)
597 // Extrapolate track along simple helix in magnetic field
598 // Arguments: len -distance alogn helix, [cm]
599 // bz - mag field, [kGaus]
600 // Returns: x and p contain extrapolated positon and momentum
601 // The momentum returned for straight-line tracks is meaningless !
602 //+++++++++++++++++++++++++++++++++++++++++
603 GetXYZ(x);
604
def9660e 605 if (OneOverPt() < kAlmost0 || TMath::Abs(bz) < kAlmost0Field ){ //straight-line tracks
052daaff 606 Double_t unit[3]; GetDirection(unit);
607 x[0]+=unit[0]*len;
608 x[1]+=unit[1]*len;
609 x[2]+=unit[2]*len;
610
611 p[0]=unit[0]/kAlmost0;
612 p[1]=unit[1]/kAlmost0;
613 p[2]=unit[2]/kAlmost0;
614 } else {
615 GetPxPyPz(p);
616 Double_t pp=GetP();
617 Double_t a = -kB2C*bz*GetSign();
618 Double_t rho = a/pp;
619 x[0] += p[0]*TMath::Sin(rho*len)/a - p[1]*(1-TMath::Cos(rho*len))/a;
620 x[1] += p[1]*TMath::Sin(rho*len)/a + p[0]*(1-TMath::Cos(rho*len))/a;
621 x[2] += p[2]*len/pp;
622
623 Double_t p0=p[0];
624 p[0] = p0 *TMath::Cos(rho*len) - p[1]*TMath::Sin(rho*len);
625 p[1] = p[1]*TMath::Cos(rho*len) + p0 *TMath::Sin(rho*len);
626 }
627}
628
629Bool_t AliExternalTrackParam::Intersect(Double_t pnt[3], Double_t norm[3],
630Double_t bz) const {
631 //+++++++++++++++++++++++++++++++++++++++++
632 // Origin: K. Shileev (Kirill.Shileev@cern.ch)
633 // Finds point of intersection (if exists) of the helix with the plane.
634 // Stores result in fX and fP.
635 // Arguments: planePoint,planeNorm - the plane defined by any plane's point
636 // and vector, normal to the plane
637 // Returns: kTrue if helix intersects the plane, kFALSE otherwise.
638 //+++++++++++++++++++++++++++++++++++++++++
639 Double_t x0[3]; GetXYZ(x0); //get track position in MARS
640
641 //estimates initial helix length up to plane
642 Double_t s=
643 (pnt[0]-x0[0])*norm[0] + (pnt[1]-x0[1])*norm[1] + (pnt[2]-x0[2])*norm[2];
644 Double_t dist=99999,distPrev=dist;
645 Double_t x[3],p[3];
646 while(TMath::Abs(dist)>0.00001){
647 //calculates helix at the distance s from x0 ALONG the helix
648 Propagate(s,x,p,bz);
649
650 //distance between current helix position and plane
651 dist=(x[0]-pnt[0])*norm[0]+(x[1]-pnt[1])*norm[1]+(x[2]-pnt[2])*norm[2];
652
653 if(TMath::Abs(dist) >= TMath::Abs(distPrev)) {return kFALSE;}
654 distPrev=dist;
655 s-=dist;
656 }
657 //on exit pnt is intersection point,norm is track vector at that point,
658 //all in MARS
659 for (Int_t i=0; i<3; i++) {pnt[i]=x[i]; norm[i]=p[i];}
660 return kTRUE;
661}
662
49d13e89 663Double_t
664AliExternalTrackParam::GetPredictedChi2(Double_t p[2],Double_t cov[3]) const {
665 //----------------------------------------------------------------
666 // Estimate the chi2 of the space point "p" with the cov. matrix "cov"
667 //----------------------------------------------------------------
668 Double_t sdd = fC[0] + cov[0];
669 Double_t sdz = fC[1] + cov[1];
670 Double_t szz = fC[2] + cov[2];
671 Double_t det = sdd*szz - sdz*sdz;
672
673 if (TMath::Abs(det) < kAlmost0) return kVeryBig;
674
675 Double_t d = fP[0] - p[0];
676 Double_t z = fP[1] - p[1];
677
678 return (d*szz*d - 2*d*sdz*z + z*sdd*z)/det;
679}
680
4b189f98 681Double_t AliExternalTrackParam::
682GetPredictedChi2(Double_t p[3],Double_t covyz[3],Double_t covxyz[3]) const {
683 //----------------------------------------------------------------
684 // Estimate the chi2 of the 3D space point "p" and
1e023a36 685 // the full covariance matrix "covyz" and "covxyz"
4b189f98 686 //
687 // Cov(x,x) ... : covxyz[0]
688 // Cov(y,x) ... : covxyz[1] covyz[0]
689 // Cov(z,x) ... : covxyz[2] covyz[1] covyz[2]
690 //----------------------------------------------------------------
691
692 Double_t res[3] = {
693 GetX() - p[0],
694 GetY() - p[1],
695 GetZ() - p[2]
696 };
697
698 Double_t f=GetSnp();
699 if (TMath::Abs(f) >= kAlmost1) return kVeryBig;
700 Double_t r=TMath::Sqrt(1.- f*f);
701 Double_t a=f/r, b=GetTgl()/r;
702
703 Double_t s2=333.*333.; //something reasonably big (cm^2)
704
705 TMatrixDSym v(3);
706 v(0,0)= s2; v(0,1)= a*s2; v(0,2)= b*s2;;
707 v(1,0)=a*s2; v(1,1)=a*a*s2 + GetSigmaY2(); v(1,2)=a*b*s2 + GetSigmaZY();
708 v(2,0)=b*s2; v(2,1)=a*b*s2 + GetSigmaZY(); v(2,2)=b*b*s2 + GetSigmaZ2();
709
710 v(0,0)+=covxyz[0]; v(0,1)+=covxyz[1]; v(0,2)+=covxyz[2];
711 v(1,0)+=covxyz[1]; v(1,1)+=covyz[0]; v(1,2)+=covyz[1];
712 v(2,0)+=covxyz[2]; v(2,1)+=covyz[1]; v(2,2)+=covyz[2];
713
714 v.Invert();
715 if (!v.IsValid()) return kVeryBig;
716
717 Double_t chi2=0.;
718 for (Int_t i = 0; i < 3; i++)
719 for (Int_t j = 0; j < 3; j++) chi2 += res[i]*res[j]*v(i,j);
720
721 return chi2;
722
723
724}
725
1e023a36 726Bool_t AliExternalTrackParam::
727PropagateTo(Double_t p[3],Double_t covyz[3],Double_t covxyz[3],Double_t bz) {
728 //----------------------------------------------------------------
729 // Propagate this track to the plane
730 // the 3D space point "p" (with the covariance matrix "covyz" and "covxyz")
731 // belongs to.
732 // The magnetic field is "bz" (kG)
733 //
734 // The track curvature and the change of the covariance matrix
735 // of the track parameters are negleted !
736 // (So the "step" should be small compared with 1/curvature)
737 //----------------------------------------------------------------
738
739 Double_t f=GetSnp();
740 if (TMath::Abs(f) >= kAlmost1) return kFALSE;
741 Double_t r=TMath::Sqrt(1.- f*f);
742 Double_t a=f/r, b=GetTgl()/r;
743
744 Double_t s2=333.*333.; //something reasonably big (cm^2)
745
746 TMatrixDSym tV(3);
747 tV(0,0)= s2; tV(0,1)= a*s2; tV(0,2)= b*s2;
748 tV(1,0)=a*s2; tV(1,1)=a*a*s2; tV(1,2)=a*b*s2;
749 tV(2,0)=b*s2; tV(2,1)=a*b*s2; tV(2,2)=b*b*s2;
750
751 TMatrixDSym pV(3);
752 pV(0,0)=covxyz[0]; pV(0,1)=covxyz[1]; pV(0,2)=covxyz[2];
753 pV(1,0)=covxyz[1]; pV(1,1)=covyz[0]; pV(1,2)=covyz[1];
754 pV(2,0)=covxyz[2]; pV(2,1)=covyz[1]; pV(2,2)=covyz[2];
755
756 TMatrixDSym tpV(tV);
757 tpV+=pV;
758 tpV.Invert();
759 if (!tpV.IsValid()) return kFALSE;
760
761 TMatrixDSym pW(3),tW(3);
762 for (Int_t i=0; i<3; i++)
763 for (Int_t j=0; j<3; j++) {
764 pW(i,j)=tW(i,j)=0.;
765 for (Int_t k=0; k<3; k++) {
766 pW(i,j) += tV(i,k)*tpV(k,j);
767 tW(i,j) += pV(i,k)*tpV(k,j);
768 }
769 }
770
771 Double_t t[3] = {GetX(), GetY(), GetZ()};
772
773 Double_t x=0.;
774 for (Int_t i=0; i<3; i++) x += (tW(0,i)*t[i] + pW(0,i)*p[i]);
775 Double_t crv=GetC(bz);
776 if (TMath::Abs(b) < kAlmost0Field) crv=0.;
777 f += crv*(x-fX);
778 if (TMath::Abs(f) >= kAlmost1) return kFALSE;
779 fX=x;
780
781 fP[0]=0.;
782 for (Int_t i=0; i<3; i++) fP[0] += (tW(1,i)*t[i] + pW(1,i)*p[i]);
783 fP[1]=0.;
784 for (Int_t i=0; i<3; i++) fP[1] += (tW(2,i)*t[i] + pW(2,i)*p[i]);
785
786 return kTRUE;
787}
788
e23a38cb 789Double_t *AliExternalTrackParam::GetResiduals(
790Double_t *p,Double_t *cov,Bool_t updated) const {
791 //------------------------------------------------------------------
792 // Returns the track residuals with the space point "p" having
793 // the covariance matrix "cov".
794 // If "updated" is kTRUE, the track parameters expected to be updated,
795 // otherwise they must be predicted.
796 //------------------------------------------------------------------
797 static Double_t res[2];
798
799 Double_t r00=cov[0], r01=cov[1], r11=cov[2];
800 if (updated) {
801 r00-=fC[0]; r01-=fC[1]; r11-=fC[2];
802 } else {
803 r00+=fC[0]; r01+=fC[1]; r11+=fC[2];
804 }
805 Double_t det=r00*r11 - r01*r01;
806
807 if (TMath::Abs(det) < kAlmost0) return 0;
808
809 Double_t tmp=r00; r00=r11/det; r11=tmp/det;
f0fbf964 810
811 if (r00 < 0.) return 0;
812 if (r11 < 0.) return 0;
813
e23a38cb 814 Double_t dy = fP[0] - p[0];
815 Double_t dz = fP[1] - p[1];
816
817 res[0]=dy*TMath::Sqrt(r00);
818 res[1]=dz*TMath::Sqrt(r11);
819
820 return res;
821}
822
49d13e89 823Bool_t AliExternalTrackParam::Update(Double_t p[2], Double_t cov[3]) {
824 //------------------------------------------------------------------
825 // Update the track parameters with the space point "p" having
826 // the covariance matrix "cov"
827 //------------------------------------------------------------------
828 Double_t &fP0=fP[0], &fP1=fP[1], &fP2=fP[2], &fP3=fP[3], &fP4=fP[4];
829 Double_t
830 &fC00=fC[0],
831 &fC10=fC[1], &fC11=fC[2],
832 &fC20=fC[3], &fC21=fC[4], &fC22=fC[5],
833 &fC30=fC[6], &fC31=fC[7], &fC32=fC[8], &fC33=fC[9],
834 &fC40=fC[10], &fC41=fC[11], &fC42=fC[12], &fC43=fC[13], &fC44=fC[14];
835
836 Double_t r00=cov[0], r01=cov[1], r11=cov[2];
837 r00+=fC00; r01+=fC10; r11+=fC11;
838 Double_t det=r00*r11 - r01*r01;
839
840 if (TMath::Abs(det) < kAlmost0) return kFALSE;
841
842
843 Double_t tmp=r00; r00=r11/det; r11=tmp/det; r01=-r01/det;
844
845 Double_t k00=fC00*r00+fC10*r01, k01=fC00*r01+fC10*r11;
846 Double_t k10=fC10*r00+fC11*r01, k11=fC10*r01+fC11*r11;
847 Double_t k20=fC20*r00+fC21*r01, k21=fC20*r01+fC21*r11;
848 Double_t k30=fC30*r00+fC31*r01, k31=fC30*r01+fC31*r11;
849 Double_t k40=fC40*r00+fC41*r01, k41=fC40*r01+fC41*r11;
850
851 Double_t dy=p[0] - fP0, dz=p[1] - fP1;
852 Double_t sf=fP2 + k20*dy + k21*dz;
853 if (TMath::Abs(sf) > kAlmost1) return kFALSE;
854
855 fP0 += k00*dy + k01*dz;
856 fP1 += k10*dy + k11*dz;
857 fP2 = sf;
858 fP3 += k30*dy + k31*dz;
859 fP4 += k40*dy + k41*dz;
860
861 Double_t c01=fC10, c02=fC20, c03=fC30, c04=fC40;
862 Double_t c12=fC21, c13=fC31, c14=fC41;
863
864 fC00-=k00*fC00+k01*fC10; fC10-=k00*c01+k01*fC11;
865 fC20-=k00*c02+k01*c12; fC30-=k00*c03+k01*c13;
866 fC40-=k00*c04+k01*c14;
867
868 fC11-=k10*c01+k11*fC11;
869 fC21-=k10*c02+k11*c12; fC31-=k10*c03+k11*c13;
870 fC41-=k10*c04+k11*c14;
871
872 fC22-=k20*c02+k21*c12; fC32-=k20*c03+k21*c13;
873 fC42-=k20*c04+k21*c14;
874
875 fC33-=k30*c03+k31*c13;
876 fC43-=k30*c04+k31*c14;
877
878 fC44-=k40*c04+k41*c14;
879
880 return kTRUE;
881}
882
c7bafca9 883void
884AliExternalTrackParam::GetHelixParameters(Double_t hlx[6], Double_t b) const {
885 //--------------------------------------------------------------------
886 // External track parameters -> helix parameters
887 // "b" - magnetic field (kG)
888 //--------------------------------------------------------------------
889 Double_t cs=TMath::Cos(fAlpha), sn=TMath::Sin(fAlpha);
890
1530f89c 891 hlx[0]=fP[0]; hlx[1]=fP[1]; hlx[2]=fP[2]; hlx[3]=fP[3];
c7bafca9 892
893 hlx[5]=fX*cs - hlx[0]*sn; // x0
894 hlx[0]=fX*sn + hlx[0]*cs; // y0
895//hlx[1]= // z0
896 hlx[2]=TMath::ASin(hlx[2]) + fAlpha; // phi0
897//hlx[3]= // tgl
1530f89c 898 hlx[4]=GetC(b); // C
c7bafca9 899}
900
901
902static void Evaluate(const Double_t *h, Double_t t,
903 Double_t r[3], //radius vector
904 Double_t g[3], //first defivatives
905 Double_t gg[3]) //second derivatives
906{
907 //--------------------------------------------------------------------
908 // Calculate position of a point on a track and some derivatives
909 //--------------------------------------------------------------------
910 Double_t phase=h[4]*t+h[2];
911 Double_t sn=TMath::Sin(phase), cs=TMath::Cos(phase);
912
913 r[0] = h[5] + (sn - h[6])/h[4];
914 r[1] = h[0] - (cs - h[7])/h[4];
915 r[2] = h[1] + h[3]*t;
916
917 g[0] = cs; g[1]=sn; g[2]=h[3];
918
919 gg[0]=-h[4]*sn; gg[1]=h[4]*cs; gg[2]=0.;
920}
921
922Double_t AliExternalTrackParam::GetDCA(const AliExternalTrackParam *p,
923Double_t b, Double_t &xthis, Double_t &xp) const {
924 //------------------------------------------------------------
925 // Returns the (weighed !) distance of closest approach between
926 // this track and the track "p".
927 // Other returned values:
928 // xthis, xt - coordinates of tracks' reference planes at the DCA
929 //-----------------------------------------------------------
930 Double_t dy2=GetSigmaY2() + p->GetSigmaY2();
931 Double_t dz2=GetSigmaZ2() + p->GetSigmaZ2();
932 Double_t dx2=dy2;
933
c7bafca9 934 Double_t p1[8]; GetHelixParameters(p1,b);
935 p1[6]=TMath::Sin(p1[2]); p1[7]=TMath::Cos(p1[2]);
936 Double_t p2[8]; p->GetHelixParameters(p2,b);
937 p2[6]=TMath::Sin(p2[2]); p2[7]=TMath::Cos(p2[2]);
938
939
940 Double_t r1[3],g1[3],gg1[3]; Double_t t1=0.;
941 Evaluate(p1,t1,r1,g1,gg1);
942 Double_t r2[3],g2[3],gg2[3]; Double_t t2=0.;
943 Evaluate(p2,t2,r2,g2,gg2);
944
945 Double_t dx=r2[0]-r1[0], dy=r2[1]-r1[1], dz=r2[2]-r1[2];
946 Double_t dm=dx*dx/dx2 + dy*dy/dy2 + dz*dz/dz2;
947
948 Int_t max=27;
949 while (max--) {
950 Double_t gt1=-(dx*g1[0]/dx2 + dy*g1[1]/dy2 + dz*g1[2]/dz2);
951 Double_t gt2=+(dx*g2[0]/dx2 + dy*g2[1]/dy2 + dz*g2[2]/dz2);
952 Double_t h11=(g1[0]*g1[0] - dx*gg1[0])/dx2 +
953 (g1[1]*g1[1] - dy*gg1[1])/dy2 +
954 (g1[2]*g1[2] - dz*gg1[2])/dz2;
955 Double_t h22=(g2[0]*g2[0] + dx*gg2[0])/dx2 +
956 (g2[1]*g2[1] + dy*gg2[1])/dy2 +
957 (g2[2]*g2[2] + dz*gg2[2])/dz2;
958 Double_t h12=-(g1[0]*g2[0]/dx2 + g1[1]*g2[1]/dy2 + g1[2]*g2[2]/dz2);
959
960 Double_t det=h11*h22-h12*h12;
961
962 Double_t dt1,dt2;
963 if (TMath::Abs(det)<1.e-33) {
964 //(quasi)singular Hessian
965 dt1=-gt1; dt2=-gt2;
966 } else {
967 dt1=-(gt1*h22 - gt2*h12)/det;
968 dt2=-(h11*gt2 - h12*gt1)/det;
969 }
970
971 if ((dt1*gt1+dt2*gt2)>0) {dt1=-dt1; dt2=-dt2;}
972
973 //check delta(phase1) ?
974 //check delta(phase2) ?
975
976 if (TMath::Abs(dt1)/(TMath::Abs(t1)+1.e-3) < 1.e-4)
977 if (TMath::Abs(dt2)/(TMath::Abs(t2)+1.e-3) < 1.e-4) {
978 if ((gt1*gt1+gt2*gt2) > 1.e-4/dy2/dy2)
358f16ae 979 AliDebug(1," stopped at not a stationary point !");
c7bafca9 980 Double_t lmb=h11+h22; lmb=lmb-TMath::Sqrt(lmb*lmb-4*det);
981 if (lmb < 0.)
358f16ae 982 AliDebug(1," stopped at not a minimum !");
c7bafca9 983 break;
984 }
985
986 Double_t dd=dm;
987 for (Int_t div=1 ; ; div*=2) {
988 Evaluate(p1,t1+dt1,r1,g1,gg1);
989 Evaluate(p2,t2+dt2,r2,g2,gg2);
990 dx=r2[0]-r1[0]; dy=r2[1]-r1[1]; dz=r2[2]-r1[2];
991 dd=dx*dx/dx2 + dy*dy/dy2 + dz*dz/dz2;
992 if (dd<dm) break;
993 dt1*=0.5; dt2*=0.5;
994 if (div>512) {
358f16ae 995 AliDebug(1," overshoot !"); break;
c7bafca9 996 }
997 }
998 dm=dd;
999
1000 t1+=dt1;
1001 t2+=dt2;
1002
1003 }
1004
358f16ae 1005 if (max<=0) AliDebug(1," too many iterations !");
c7bafca9 1006
1007 Double_t cs=TMath::Cos(GetAlpha());
1008 Double_t sn=TMath::Sin(GetAlpha());
1009 xthis=r1[0]*cs + r1[1]*sn;
1010
1011 cs=TMath::Cos(p->GetAlpha());
1012 sn=TMath::Sin(p->GetAlpha());
1013 xp=r2[0]*cs + r2[1]*sn;
1014
1015 return TMath::Sqrt(dm*TMath::Sqrt(dy2*dz2));
1016}
1017
1018Double_t AliExternalTrackParam::
1019PropagateToDCA(AliExternalTrackParam *p, Double_t b) {
1020 //--------------------------------------------------------------
1021 // Propagates this track and the argument track to the position of the
1022 // distance of closest approach.
1023 // Returns the (weighed !) distance of closest approach.
1024 //--------------------------------------------------------------
1025 Double_t xthis,xp;
1026 Double_t dca=GetDCA(p,b,xthis,xp);
1027
1028 if (!PropagateTo(xthis,b)) {
1029 //AliWarning(" propagation failed !");
1030 return 1e+33;
1031 }
1032
1033 if (!p->PropagateTo(xp,b)) {
1034 //AliWarning(" propagation failed !";
1035 return 1e+33;
1036 }
1037
1038 return dca;
1039}
1040
1041
58e536c5 1042Bool_t AliExternalTrackParam::PropagateToDCA(const AliVVertex *vtx,
e99a34df 1043Double_t b, Double_t maxd, Double_t dz[2], Double_t covar[3]) {
f76701bf 1044 //
e99a34df 1045 // Propagate this track to the DCA to vertex "vtx",
f76701bf 1046 // if the (rough) transverse impact parameter is not bigger then "maxd".
1047 // Magnetic field is "b" (kG).
1048 //
1049 // a) The track gets extapolated to the DCA to the vertex.
1050 // b) The impact parameters and their covariance matrix are calculated.
1051 //
1052 // In the case of success, the returned value is kTRUE
1053 // (otherwise, it's kFALSE)
1054 //
1055 Double_t alpha=GetAlpha();
1056 Double_t sn=TMath::Sin(alpha), cs=TMath::Cos(alpha);
1057 Double_t x=GetX(), y=GetParameter()[0], snp=GetParameter()[2];
58e536c5 1058 Double_t xv= vtx->GetX()*cs + vtx->GetY()*sn;
1059 Double_t yv=-vtx->GetX()*sn + vtx->GetY()*cs, zv=vtx->GetZ();
f76701bf 1060 x-=xv; y-=yv;
1061
1062 //Estimate the impact parameter neglecting the track curvature
1063 Double_t d=TMath::Abs(x*snp - y*TMath::Sqrt(1.- snp*snp));
1064 if (d > maxd) return kFALSE;
1065
1066 //Propagate to the DCA
e99a34df 1067 Double_t crv=kB2C*b*GetParameter()[4];
1068 if (TMath::Abs(b) < kAlmost0Field) crv=0.;
1069
f76701bf 1070 Double_t tgfv=-(crv*x - snp)/(crv*y + TMath::Sqrt(1.-snp*snp));
1071 sn=tgfv/TMath::Sqrt(1.+ tgfv*tgfv); cs=TMath::Sqrt(1.- sn*sn);
e99a34df 1072 if (TMath::Abs(tgfv)>0.) cs = sn/tgfv;
1073 else cs=1.;
f76701bf 1074
1075 x = xv*cs + yv*sn;
1076 yv=-xv*sn + yv*cs; xv=x;
1077
1078 if (!Propagate(alpha+TMath::ASin(sn),xv,b)) return kFALSE;
e99a34df 1079
1080 if (dz==0) return kTRUE;
1081 dz[0] = GetParameter()[0] - yv;
1082 dz[1] = GetParameter()[1] - zv;
1083
1084 if (covar==0) return kTRUE;
58e536c5 1085 Double_t cov[6]; vtx->GetCovarianceMatrix(cov);
e99a34df 1086
1087 //***** Improvements by A.Dainese
1088 alpha=GetAlpha(); sn=TMath::Sin(alpha); cs=TMath::Cos(alpha);
1089 Double_t s2ylocvtx = cov[0]*sn*sn + cov[2]*cs*cs - 2.*cov[1]*cs*sn;
1090 covar[0] = GetCovariance()[0] + s2ylocvtx; // neglecting correlations
1091 covar[1] = GetCovariance()[1]; // between (x,y) and z
1092 covar[2] = GetCovariance()[2] + cov[5]; // in vertex's covariance matrix
1093 //*****
1094
29fbcc93 1095 return kTRUE;
f76701bf 1096}
1097
1098
b1149664 1099void AliExternalTrackParam::GetDirection(Double_t d[3]) const {
1100 //----------------------------------------------------------------
1101 // This function returns a unit vector along the track direction
1102 // in the global coordinate system.
1103 //----------------------------------------------------------------
1104 Double_t cs=TMath::Cos(fAlpha), sn=TMath::Sin(fAlpha);
1105 Double_t snp=fP[2];
92934324 1106 Double_t csp =TMath::Sqrt((1.- snp)*(1.+snp));
b1149664 1107 Double_t norm=TMath::Sqrt(1.+ fP[3]*fP[3]);
1108 d[0]=(csp*cs - snp*sn)/norm;
1109 d[1]=(snp*cs + csp*sn)/norm;
1110 d[2]=fP[3]/norm;
1111}
1112
c683ddc2 1113Bool_t AliExternalTrackParam::GetPxPyPz(Double_t p[3]) const {
c9ec41e8 1114 //---------------------------------------------------------------------
1115 // This function returns the global track momentum components
1116 // Results for (nearly) straight tracks are meaningless !
1117 //---------------------------------------------------------------------
1118 p[0]=fP[4]; p[1]=fP[2]; p[2]=fP[3];
1119 return Local2GlobalMomentum(p,fAlpha);
1120}
a5e407e9 1121
def9660e 1122Double_t AliExternalTrackParam::Px() const {
957fb479 1123 //---------------------------------------------------------------------
1124 // Returns x-component of momentum
1125 // Result for (nearly) straight tracks is meaningless !
1126 //---------------------------------------------------------------------
def9660e 1127
957fb479 1128 Double_t p[3]={kVeryBig,kVeryBig,kVeryBig};
def9660e 1129 GetPxPyPz(p);
1130
1131 return p[0];
1132}
1133
1134Double_t AliExternalTrackParam::Py() const {
957fb479 1135 //---------------------------------------------------------------------
1136 // Returns y-component of momentum
1137 // Result for (nearly) straight tracks is meaningless !
1138 //---------------------------------------------------------------------
def9660e 1139
957fb479 1140 Double_t p[3]={kVeryBig,kVeryBig,kVeryBig};
def9660e 1141 GetPxPyPz(p);
1142
1143 return p[1];
1144}
1145
1146Double_t AliExternalTrackParam::Pz() const {
957fb479 1147 //---------------------------------------------------------------------
1148 // Returns z-component of momentum
1149 // Result for (nearly) straight tracks is meaningless !
1150 //---------------------------------------------------------------------
def9660e 1151
957fb479 1152 Double_t p[3]={kVeryBig,kVeryBig,kVeryBig};
def9660e 1153 GetPxPyPz(p);
1154
1155 return p[2];
1156}
1157
c683ddc2 1158Double_t AliExternalTrackParam::Xv() const {
1159 //---------------------------------------------------------------------
1160 // Returns x-component of first track point
1161 //---------------------------------------------------------------------
1162
1163 Double_t r[3]={0.,0.,0.};
1164 GetXYZ(r);
1165
1166 return r[0];
1167}
1168
1169Double_t AliExternalTrackParam::Yv() const {
1170 //---------------------------------------------------------------------
1171 // Returns y-component of first track point
1172 //---------------------------------------------------------------------
1173
1174 Double_t r[3]={0.,0.,0.};
1175 GetXYZ(r);
1176
1177 return r[1];
1178}
1179
1180Double_t AliExternalTrackParam::Zv() const {
1181 //---------------------------------------------------------------------
1182 // Returns z-component of first track point
1183 //---------------------------------------------------------------------
1184
1185 Double_t r[3]={0.,0.,0.};
1186 GetXYZ(r);
1187
1188 return r[2];
1189}
1190
def9660e 1191Double_t AliExternalTrackParam::Theta() const {
1192 // return theta angle of momentum
1193
7cdd0c20 1194 return 0.5*TMath::Pi() - TMath::ATan(fP[3]);
def9660e 1195}
1196
1197Double_t AliExternalTrackParam::Phi() const {
957fb479 1198 //---------------------------------------------------------------------
1199 // Returns the azimuthal angle of momentum
1200 // 0 <= phi < 2*pi
1201 //---------------------------------------------------------------------
def9660e 1202
957fb479 1203 Double_t phi=TMath::ASin(fP[2]) + fAlpha;
1204 if (phi<0.) phi+=2.*TMath::Pi();
1205 else if (phi>=2.*TMath::Pi()) phi-=2.*TMath::Pi();
1206
1207 return phi;
def9660e 1208}
1209
1210Double_t AliExternalTrackParam::M() const {
1211 // return particle mass
1212
1213 // No mass information available so far.
1214 // Redifine in derived class!
1215
1216 return -999.;
1217}
1218
1219Double_t AliExternalTrackParam::E() const {
1220 // return particle energy
1221
1222 // No PID information available so far.
1223 // Redifine in derived class!
1224
1225 return -999.;
1226}
1227
1228Double_t AliExternalTrackParam::Eta() const {
1229 // return pseudorapidity
1230
1231 return -TMath::Log(TMath::Tan(0.5 * Theta()));
1232}
1233
1234Double_t AliExternalTrackParam::Y() const {
1235 // return rapidity
1236
1237 // No PID information available so far.
1238 // Redifine in derived class!
1239
1240 return -999.;
1241}
1242
c9ec41e8 1243Bool_t AliExternalTrackParam::GetXYZ(Double_t *r) const {
1244 //---------------------------------------------------------------------
1245 // This function returns the global track position
1246 //---------------------------------------------------------------------
1247 r[0]=fX; r[1]=fP[0]; r[2]=fP[1];
1248 return Local2GlobalPosition(r,fAlpha);
51ad6848 1249}
1250
c9ec41e8 1251Bool_t AliExternalTrackParam::GetCovarianceXYZPxPyPz(Double_t cv[21]) const {
1252 //---------------------------------------------------------------------
1253 // This function returns the global covariance matrix of the track params
1254 //
1255 // Cov(x,x) ... : cv[0]
1256 // Cov(y,x) ... : cv[1] cv[2]
1257 // Cov(z,x) ... : cv[3] cv[4] cv[5]
1258 // Cov(px,x)... : cv[6] cv[7] cv[8] cv[9]
1259 // Cov(py,x)... : cv[10] cv[11] cv[12] cv[13] cv[14]
1260 // Cov(pz,x)... : cv[15] cv[16] cv[17] cv[18] cv[19] cv[20]
a5e407e9 1261 //
c9ec41e8 1262 // Results for (nearly) straight tracks are meaningless !
1263 //---------------------------------------------------------------------
e421f556 1264 if (TMath::Abs(fP[4])<=kAlmost0) {
c9ec41e8 1265 for (Int_t i=0; i<21; i++) cv[i]=0.;
1266 return kFALSE;
a5e407e9 1267 }
49d13e89 1268 if (TMath::Abs(fP[2]) > kAlmost1) {
c9ec41e8 1269 for (Int_t i=0; i<21; i++) cv[i]=0.;
1270 return kFALSE;
1271 }
1272 Double_t pt=1./TMath::Abs(fP[4]);
1273 Double_t cs=TMath::Cos(fAlpha), sn=TMath::Sin(fAlpha);
92934324 1274 Double_t r=TMath::Sqrt((1.-fP[2])*(1.+fP[2]));
c9ec41e8 1275
1276 Double_t m00=-sn, m10=cs;
1277 Double_t m23=-pt*(sn + fP[2]*cs/r), m43=-pt*pt*(r*cs - fP[2]*sn);
1278 Double_t m24= pt*(cs - fP[2]*sn/r), m44=-pt*pt*(r*sn + fP[2]*cs);
1279 Double_t m35=pt, m45=-pt*pt*fP[3];
1280
854d5d49 1281 m43*=GetSign();
1282 m44*=GetSign();
1283 m45*=GetSign();
1284
c9ec41e8 1285 cv[0 ] = fC[0]*m00*m00;
1286 cv[1 ] = fC[0]*m00*m10;
1287 cv[2 ] = fC[0]*m10*m10;
1288 cv[3 ] = fC[1]*m00;
1289 cv[4 ] = fC[1]*m10;
1290 cv[5 ] = fC[2];
1291 cv[6 ] = m00*(fC[3]*m23 + fC[10]*m43);
1292 cv[7 ] = m10*(fC[3]*m23 + fC[10]*m43);
1293 cv[8 ] = fC[4]*m23 + fC[11]*m43;
1294 cv[9 ] = m23*(fC[5]*m23 + fC[12]*m43) + m43*(fC[12]*m23 + fC[14]*m43);
1295 cv[10] = m00*(fC[3]*m24 + fC[10]*m44);
1296 cv[11] = m10*(fC[3]*m24 + fC[10]*m44);
1297 cv[12] = fC[4]*m24 + fC[11]*m44;
1298 cv[13] = m23*(fC[5]*m24 + fC[12]*m44) + m43*(fC[12]*m24 + fC[14]*m44);
1299 cv[14] = m24*(fC[5]*m24 + fC[12]*m44) + m44*(fC[12]*m24 + fC[14]*m44);
1300 cv[15] = m00*(fC[6]*m35 + fC[10]*m45);
1301 cv[16] = m10*(fC[6]*m35 + fC[10]*m45);
1302 cv[17] = fC[7]*m35 + fC[11]*m45;
1303 cv[18] = m23*(fC[8]*m35 + fC[12]*m45) + m43*(fC[13]*m35 + fC[14]*m45);
1304 cv[19] = m24*(fC[8]*m35 + fC[12]*m45) + m44*(fC[13]*m35 + fC[14]*m45);
1305 cv[20] = m35*(fC[9]*m35 + fC[13]*m45) + m45*(fC[13]*m35 + fC[14]*m45);
51ad6848 1306
c9ec41e8 1307 return kTRUE;
51ad6848 1308}
1309
51ad6848 1310
c9ec41e8 1311Bool_t
1312AliExternalTrackParam::GetPxPyPzAt(Double_t x, Double_t b, Double_t *p) const {
1313 //---------------------------------------------------------------------
1314 // This function returns the global track momentum extrapolated to
1315 // the radial position "x" (cm) in the magnetic field "b" (kG)
1316 //---------------------------------------------------------------------
c9ec41e8 1317 p[0]=fP[4];
1530f89c 1318 p[1]=fP[2]+(x-fX)*GetC(b);
c9ec41e8 1319 p[2]=fP[3];
1320 return Local2GlobalMomentum(p,fAlpha);
51ad6848 1321}
1322
7cf7bb6c 1323Bool_t
1324AliExternalTrackParam::GetYAt(Double_t x, Double_t b, Double_t &y) const {
1325 //---------------------------------------------------------------------
1326 // This function returns the local Y-coordinate of the intersection
1327 // point between this track and the reference plane "x" (cm).
1328 // Magnetic field "b" (kG)
1329 //---------------------------------------------------------------------
1330 Double_t dx=x-fX;
1331 if(TMath::Abs(dx)<=kAlmost0) {y=fP[0]; return kTRUE;}
1332
1530f89c 1333 Double_t f1=fP[2], f2=f1 + dx*GetC(b);
7cf7bb6c 1334
1335 if (TMath::Abs(f1) >= kAlmost1) return kFALSE;
1336 if (TMath::Abs(f2) >= kAlmost1) return kFALSE;
1337
1338 Double_t r1=TMath::Sqrt(1.- f1*f1), r2=TMath::Sqrt(1.- f2*f2);
1339 y = fP[0] + dx*(f1+f2)/(r1+r2);
1340 return kTRUE;
1341}
1342
6c94f330 1343Bool_t
1344AliExternalTrackParam::GetZAt(Double_t x, Double_t b, Double_t &z) const {
1345 //---------------------------------------------------------------------
1346 // This function returns the local Z-coordinate of the intersection
1347 // point between this track and the reference plane "x" (cm).
1348 // Magnetic field "b" (kG)
1349 //---------------------------------------------------------------------
1350 Double_t dx=x-fX;
1351 if(TMath::Abs(dx)<=kAlmost0) {z=fP[1]; return kTRUE;}
1352
1353 Double_t f1=fP[2], f2=f1 + dx*fP[4]*b*kB2C;
1354
1355 if (TMath::Abs(f1) >= kAlmost1) return kFALSE;
1356 if (TMath::Abs(f2) >= kAlmost1) return kFALSE;
1357
1358 Double_t r1=sqrt(1.- f1*f1), r2=sqrt(1.- f2*f2);
1359 z = fP[1] + dx*(r2 + f2*(f1+f2)/(r1+r2))*fP[3]; // Many thanks to P.Hristov !
1360 return kTRUE;
1361}
1362
c9ec41e8 1363Bool_t
1364AliExternalTrackParam::GetXYZAt(Double_t x, Double_t b, Double_t *r) const {
1365 //---------------------------------------------------------------------
1366 // This function returns the global track position extrapolated to
1367 // the radial position "x" (cm) in the magnetic field "b" (kG)
1368 //---------------------------------------------------------------------
c9ec41e8 1369 Double_t dx=x-fX;
e421f556 1370 if(TMath::Abs(dx)<=kAlmost0) return GetXYZ(r);
1371
1530f89c 1372 Double_t f1=fP[2], f2=f1 + dx*GetC(b);
c9ec41e8 1373
e421f556 1374 if (TMath::Abs(f1) >= kAlmost1) return kFALSE;
49d13e89 1375 if (TMath::Abs(f2) >= kAlmost1) return kFALSE;
c9ec41e8 1376
1377 Double_t r1=TMath::Sqrt(1.- f1*f1), r2=TMath::Sqrt(1.- f2*f2);
1378 r[0] = x;
1379 r[1] = fP[0] + dx*(f1+f2)/(r1+r2);
f90a11c9 1380 r[2] = fP[1] + dx*(r2 + f2*(f1+f2)/(r1+r2))*fP[3];//Thanks to Andrea & Peter
1381
c9ec41e8 1382 return Local2GlobalPosition(r,fAlpha);
51ad6848 1383}
1384
51ad6848 1385//_____________________________________________________________________________
1386void AliExternalTrackParam::Print(Option_t* /*option*/) const
1387{
1388// print the parameters and the covariance matrix
1389
1390 printf("AliExternalTrackParam: x = %-12g alpha = %-12g\n", fX, fAlpha);
1391 printf(" parameters: %12g %12g %12g %12g %12g\n",
c9ec41e8 1392 fP[0], fP[1], fP[2], fP[3], fP[4]);
1393 printf(" covariance: %12g\n", fC[0]);
1394 printf(" %12g %12g\n", fC[1], fC[2]);
1395 printf(" %12g %12g %12g\n", fC[3], fC[4], fC[5]);
51ad6848 1396 printf(" %12g %12g %12g %12g\n",
c9ec41e8 1397 fC[6], fC[7], fC[8], fC[9]);
51ad6848 1398 printf(" %12g %12g %12g %12g %12g\n",
c9ec41e8 1399 fC[10], fC[11], fC[12], fC[13], fC[14]);
51ad6848 1400}
5b77d93c 1401
c194ba83 1402Double_t AliExternalTrackParam::GetSnpAt(Double_t x,Double_t b) const {
1403 //
1404 // Get sinus at given x
1405 //
1530f89c 1406 Double_t crv=GetC(b);
c194ba83 1407 if (TMath::Abs(b) < kAlmost0Field) crv=0.;
1408 Double_t dx = x-fX;
1409 Double_t res = fP[2]+dx*crv;
1410 return res;
1411}
bf00ebb8 1412
1413Bool_t AliExternalTrackParam::GetDistance(AliExternalTrackParam *param2, Double_t x, Double_t dist[3], Double_t bz){
1414 //------------------------------------------------------------------------
1415 // Get the distance between two tracks at the local position x
1416 // working in the local frame of this track.
1417 // Origin : Marian.Ivanov@cern.ch
1418 //-----------------------------------------------------------------------
1419 Double_t xyz[3];
1420 Double_t xyz2[3];
1421 xyz[0]=x;
1422 if (!GetYAt(x,bz,xyz[1])) return kFALSE;
1423 if (!GetZAt(x,bz,xyz[2])) return kFALSE;
1424 //
1425 //
1426 if (TMath::Abs(GetAlpha()-param2->GetAlpha())<kAlmost0){
1427 xyz2[0]=x;
1428 if (!param2->GetYAt(x,bz,xyz2[1])) return kFALSE;
1429 if (!param2->GetZAt(x,bz,xyz2[2])) return kFALSE;
1430 }else{
1431 //
1432 Double_t xyz1[3];
1433 Double_t dfi = param2->GetAlpha()-GetAlpha();
1434 Double_t ca = TMath::Cos(dfi), sa = TMath::Sin(dfi);
1435 xyz2[0] = xyz[0]*ca+xyz[1]*sa;
1436 xyz2[1] = -xyz[0]*sa+xyz[1]*ca;
1437 //
1438 xyz1[0]=xyz2[0];
1439 if (!param2->GetYAt(xyz2[0],bz,xyz1[1])) return kFALSE;
1440 if (!param2->GetZAt(xyz2[0],bz,xyz1[2])) return kFALSE;
1441 //
1442 xyz2[0] = xyz1[0]*ca-xyz1[1]*sa;
1443 xyz2[1] = +xyz1[0]*sa+xyz1[1]*ca;
1444 xyz2[2] = xyz1[2];
1445 }
1446 dist[0] = xyz[0]-xyz2[0];
1447 dist[1] = xyz[1]-xyz2[1];
1448 dist[2] = xyz[2]-xyz2[2];
1449
1450 return kTRUE;
1451}
0c19adf7 1452
1453
1454//
1455// Draw functionality.
1456// Origin: Marian Ivanov, Marian.Ivanov@cern.ch
1457//
1458
1459void AliExternalTrackParam::DrawTrack(Float_t magf, Float_t minR, Float_t maxR, Float_t stepR){
1460 //
1461 // Draw track line
1462 //
1463 if (minR>maxR) return ;
1464 if (stepR<=0) return ;
1465 Int_t npoints = TMath::Nint((maxR-minR)/stepR)+1;
1466 if (npoints<1) return;
1467 TPolyMarker3D *polymarker = new TPolyMarker3D(npoints);
1468 FillPolymarker(polymarker, magf,minR,maxR,stepR);
1469 polymarker->Draw();
1470}
1471
1472//
1473void AliExternalTrackParam::FillPolymarker(TPolyMarker3D *pol, Float_t magF, Float_t minR, Float_t maxR, Float_t stepR){
1474 //
1475 // Fill points in the polymarker
1476 //
1477 Int_t counter=0;
1478 for (Double_t r=minR; r<maxR; r+=stepR){
1479 Double_t point[3];
1480 GetXYZAt(r,magF,point);
1481 pol->SetPoint(counter,point[0],point[1], point[2]);
1482 printf("xyz\t%f\t%f\t%f\n",point[0], point[1],point[2]);
1483 counter++;
1484 }
1485}
0e8460af 1486
1487Int_t AliExternalTrackParam::GetIndex(Int_t i, Int_t j) const {
1488 //
1489 Int_t min = TMath::Min(i,j);
1490 Int_t max = TMath::Max(i,j);
1491
1492 return min+(max+1)*max/2;
1493}