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 AliHelix class
20 // Origin: Marian Ivanov, CERN, marian.ivanov@cern.ch
21 //-------------------------------------------------------------------------
25 #include "AliKalmanTrack.h"
26 #include "AliTracker.h"
31 //_______________________________________________________________________
35 // Default constructor
37 for (Int_t i =0;i<9;i++) fHelix[i]=0;
40 //_______________________________________________________________________
41 AliHelix::AliHelix(const AliHelix &t):TObject(t){
44 for (Int_t i=0;i<9;i++)
45 fHelix[i]=t.fHelix[i];
48 AliHelix::AliHelix(const AliKalmanTrack &t)
52 Double_t alpha,x,cs,sn;
53 t.GetExternalParameters(x,fHelix);
57 //PH Sometimes fP4 and fHelix[4] are very big and the calculation
58 //PH of the Sqrt cannot be done. To be investigated...
59 fHelix[4]=fHelix[4]/(-1000/0.299792458/AliTracker::GetBz()); // C
60 cs=TMath::Cos(alpha); sn=TMath::Sin(alpha);
65 Double_t dummy = 1-(x-xc)*(x-xc)*fHelix[4]*fHelix[4];
67 AliError(Form("The argument of the Sqrt is %f => set to 0\n",dummy));
70 yc = fHelix[0]+TMath::Sqrt(dummy)/fHelix[4];
72 fHelix[6] = xc*cs - yc*sn;
73 fHelix[7] = xc*sn + yc*cs;
74 fHelix[8] = TMath::Abs(rc);
77 fHelix[5]=x*cs - fHelix[0]*sn; // x0
78 fHelix[0]=x*sn + fHelix[0]*cs; // y0
80 fHelix[2]=TMath::ATan2(-(fHelix[5]-fHelix[6]),fHelix[0]-fHelix[7]); // phi0
81 if (fHelix[4]>0) fHelix[2]-=TMath::Pi();
86 fHelix[5] = fHelix[6];
87 fHelix[0] = fHelix[7];
91 AliHelix::AliHelix(const AliExternalTrackParam &t)
95 Double_t alpha,x,cs,sn;
96 const Double_t *param =t.GetParameter();
97 for (Int_t i=0;i<5;i++) fHelix[i]=param[i];
102 //PH Sometimes fP4 and fHelix[4] are very big and the calculation
103 //PH of the Sqrt cannot be done. To be investigated...
104 fHelix[4]=fHelix[4]/(-1000/0.299792458/AliTracker::GetBz()); // C
105 cs=TMath::Cos(alpha); sn=TMath::Sin(alpha);
110 Double_t dummy = 1-(x-xc)*(x-xc)*fHelix[4]*fHelix[4];
112 AliError(Form("The argument of the Sqrt is %f => set to 0\n",dummy));
115 yc = fHelix[0]+TMath::Sqrt(dummy)/fHelix[4];
117 fHelix[6] = xc*cs - yc*sn;
118 fHelix[7] = xc*sn + yc*cs;
119 fHelix[8] = TMath::Abs(rc);
122 fHelix[5]=x*cs - fHelix[0]*sn; // x0
123 fHelix[0]=x*sn + fHelix[0]*cs; // y0
125 fHelix[2]=TMath::ASin(fHelix[2]) + alpha; // phi0
129 fHelix[5] = fHelix[6];
130 fHelix[0] = fHelix[7];
133 AliHelix::AliHelix(Double_t x[3], Double_t p[3], Double_t charge, Double_t conversion)
138 Double_t pt = TMath::Sqrt(p[0]*p[0]+p[1]*p[1]);
139 if (TMath::Abs(conversion)<0.00000001)
140 conversion = -1000/0.299792458/AliTracker::GetBz();
143 fHelix[4] = charge/(conversion*pt); // C
144 fHelix[3] = p[2]/pt; // tgl
148 xc = x[0] -rc*p[1]/pt;
149 yc = x[1] +rc*p[0]/pt;
151 fHelix[5] = x[0]; // x0
152 fHelix[0] = x[1]; // y0
153 fHelix[1] = x[2]; // z0
157 fHelix[8] = TMath::Abs(rc);
162 if (TMath::Abs(p[1])<TMath::Abs(p[0])){
163 fHelix[2]=TMath::ASin(p[1]/pt);
164 if (charge*yc<charge*x[1]) fHelix[2] = TMath::Pi()-fHelix[2];
167 fHelix[2]=TMath::ACos(p[0]/pt);
168 if (charge*xc>charge*x[0]) fHelix[2] = -fHelix[2];
173 void AliHelix::GetMomentum(Double_t phase, Double_t p[4],Double_t conversion, Double_t *xr)
175 // return momentum at given phase
176 Double_t x[3],g[3],gg[3];
177 Evaluate(phase,x,g,gg);
178 // if (TMath::Abs(conversion)<0.0001) conversion = -1000/0.299792458/AliTracker::GetBz();
179 if (TMath::Abs(conversion)<0.0001) conversion = TMath::Abs(1./kB2C/AliTracker::GetBz());
181 Double_t mt = TMath::Sqrt(g[0]*g[0]+g[1]*g[1]);
182 p[0] = fHelix[8]*g[0]/(mt*conversion);
183 p[1] = fHelix[8]*g[1]/(mt*conversion);
184 p[2] = fHelix[8]*g[2]/(mt*conversion);
186 xr[0] = x[0]; xr[1] = x[1]; xr[2] = x[2];
190 void AliHelix::GetAngle(Double_t t1, AliHelix &h, Double_t t2, Double_t angle[3])
195 Double_t x1[3],g1[3],gg1[3];
196 Double_t x2[3],g2[3],gg2[3];
197 Evaluate(t1,x1,g1,gg1);
198 h.Evaluate(t2,x2,g2,gg2);
201 Double_t norm1r = g1[0]*g1[0]+g1[1]*g1[1];
202 Double_t norm1 = TMath::Sqrt(norm1r+g1[2]*g1[2]);
203 norm1r = TMath::Sqrt(norm1r);
205 Double_t norm2r = g2[0]*g2[0]+g2[1]*g2[1];
206 Double_t norm2 = TMath::Sqrt(norm2r+g2[2]*g2[2]);
207 norm2r = TMath::Sqrt(norm2r);
209 angle[0] = (g1[0]*g2[0]+g1[1]*g2[1])/(norm1r*norm2r); // angle in phi projection
210 if (TMath::Abs(angle[0])<1.) angle[0] = TMath::ACos(angle[0]);
212 if (angle[0]>0) angle[0] = 0;
213 if (angle[0]<0) angle[0] = TMath::Pi();
216 angle[1] = ((norm1r*norm2r)+g1[2]*g2[2])/(norm1*norm2); // angle in rz projection
217 if (TMath::Abs(angle[1])<1.) angle[1] = TMath::ACos(angle[1]);
221 angle[2] = (g1[0]*g2[0]+g1[1]*g2[1]+g1[2]*g2[2])/(norm1*norm2); //3D angle
222 if (TMath::Abs(angle[2])<1.) angle[2] = TMath::ACos(angle[2]);
232 void AliHelix::Evaluate(Double_t t,
233 Double_t r[3], //radius vector
234 Double_t g[3], //first defivatives
235 Double_t gg[3]) //second derivatives
237 //--------------------------------------------------------------------
238 // Calculate position of a point on a track and some derivatives at given phase
239 //--------------------------------------------------------------------
240 Double_t phase=fHelix[4]*t+fHelix[2];
241 Double_t sn=TMath::Sin(phase), cs=TMath::Cos(phase);
243 r[0] = fHelix[5] + sn/fHelix[4];
244 r[1] = fHelix[0] - cs/fHelix[4];
245 r[2] = fHelix[1] + fHelix[3]*t;
247 g[0] = cs; g[1]=sn; g[2]=fHelix[3];
249 gg[0]=-fHelix[4]*sn; gg[1]=fHelix[4]*cs; gg[2]=0.;
252 Int_t AliHelix::GetClosestPhases(AliHelix &h, Double_t phase[2][2])
255 // get phases to minimize distances
260 for (Int_t i=0;i<2;i++){
261 Evaluate(phase[i][0] ,xyz0);
262 h.Evaluate(phase[i][1],xyz1);
263 Double_t mindist = TMath::Sqrt((xyz0[0]-xyz1[0])*(xyz0[0]-xyz1[0])+
264 (xyz0[1]-xyz1[1])*(xyz0[1]-xyz1[1])+
265 (xyz0[2]-xyz1[2])*(xyz0[2]-xyz1[2]));
266 Double_t tbest[2]={phase[i][0],phase[i][1]};
267 for (Int_t i0=-1;i0<=1;i0++){
268 Double_t t0 = ((phase[i][0]*fHelix[4])+i0*2.*TMath::Pi())/fHelix[4];
270 for (Int_t i1=-1;i1<=1;i1++){
271 Double_t t1 = ((phase[i][1]*h.fHelix[4])+i1*2.*TMath::Pi())/h.fHelix[4];
273 Double_t dist = TMath::Sqrt((xyz0[0]-xyz1[0])*(xyz0[0]-xyz1[0])+
274 (xyz0[1]-xyz1[1])*(xyz0[1]-xyz1[1])+
275 (xyz0[2]-xyz1[2])*(xyz0[2]-xyz1[2]));
283 phase[i][0] = tbest[0];
284 phase[i][1] = tbest[1];
289 Double_t AliHelix::GetPointAngle(AliHelix &h, Double_t phase[2], const Float_t * vertex)
292 // get point angle bettwen two helixes
294 Double_t r0[3],p0[4];
295 Double_t r1[3],p1[4];
296 GetMomentum(phase[0],p0,1,r0);
297 h.GetMomentum(phase[1],p1,1,r1);
299 Double_t r[3] = {(r0[0]+r1[0])*0.5-vertex[0],(r0[1]+r1[1])*0.5-vertex[1],(r0[2]+r1[2])*0.5-vertex[2]};
300 //intersection point - relative to the prim vertex
301 Double_t p[3] = { p0[0]+p1[0], p0[1]+p1[1],p0[2]+p1[2]};
303 Double_t normr = TMath::Sqrt(r[0]*r[0]+r[1]*r[1]+r[2]*r[2]);
304 Double_t normp = TMath::Sqrt(p[0]*p[0]+p[1]*p[1]+p[2]*p[2]);
305 Double_t pointAngle = (r[0]*p[0]+r[1]*p[1]+r[2]*p[2])/(normr*normp);
309 Double_t AliHelix::GetPhase(Double_t x, Double_t y )
313 //calculate helix param at given x,y point
315 //Double_t phase2 = TMath::ATan2((y-fHelix[0]), (x-fHelix[5]))- TMath::Pi()/2.;
316 Double_t phase2 = TMath::ATan2(-(x-fHelix[5]),(y-fHelix[0]));
317 Int_t sign = (fHelix[4]>0)? 1:-1;
318 if (sign>0) phase2 = phase2-TMath::Pi();
320 Float_t delta = TMath::Nint((phase2-fHelix[2])/(2.*TMath::Pi()));
321 phase2-= 2*TMath::Pi()*delta;
322 if ( (phase2-fHelix[2])>TMath::Pi()) phase2 -=2.*TMath::Pi();
323 if ( (phase2-fHelix[2])<-TMath::Pi()) phase2+=2.*TMath::Pi();
325 Double_t t = (phase2-fHelix[2]);
330 Int_t AliHelix::GetPhase(Double_t /*r0*/, Double_t * /*t[2]*/)
333 //calculate helix param at given r point - return nearest point ()
335 // not implemented yet
342 Double_t AliHelix::GetPhaseZ(Double_t z0)
346 return (z0-fHelix[1])/fHelix[3];
350 Int_t AliHelix::GetRPHIintersections(AliHelix &h, Double_t phase[2][2], Double_t ri[2], Double_t cut)
352 //--------------------------------------------------------------------
353 // This function returns phase vectors with intesection between helix (0, 1 or 2)
354 // in x-y plane projection
355 //--------------------------------------------------------------------
357 // Double_t * c1 = &fHelix[6];
358 //Double_t * c2 = &(h.fHelix[6]);
359 // Double_t c1[3] = {fHelix[5],fHelix[0],fHelix[8]};
361 // PH initiaziation in case of return
362 phase[0][0]=phase[0][1]=phase[1][0]=phase[1][1]=0;
365 Double_t c1[3] = {0,0,fHelix[8]};
366 Double_t c2[3] = {h.fHelix[5]-fHelix[5],h.fHelix[0]-fHelix[0],h.fHelix[8]};
368 Double_t d = TMath::Sqrt(c2[0]*c2[0]+c2[1]*c2[1]);
369 if (d<0.000000000001) return 0;
374 if ( d>=(c1[2]+c2[2])){
375 if (d>=(c1[2]+c2[2]+cut)) return 0;
376 x0[0] = (d+c1[2]-c2[2])*c2[0]/(2*d)+ fHelix[5];
377 y0[0] = (d+c1[2]-c2[2])*c2[1]/(2*d)+ fHelix[0];
379 phase[1][0] = phase[0][0] = GetPhase(x0[0],y0[0]);
380 phase[1][1] = phase[0][1] = h.GetPhase(x0[0],y0[0]);
381 ri[1] = ri[0] = x0[0]*x0[0]+y0[0]*y0[0];
384 if ( (d+c2[2])<c1[2]){
385 if ( (d+c2[2])+cut<c1[2]) return 0;
387 Double_t xx = c2[0]+ c2[0]*c2[2]/d+ fHelix[5];
388 Double_t yy = c2[1]+ c2[1]*c2[2]/d+ fHelix[0];
389 phase[1][1] = phase[0][1] = h.GetPhase(xx,yy);
391 Double_t xx2 = c2[0]*c1[2]/d+ fHelix[5];
392 Double_t yy2 = c2[1]*c1[2]/d+ fHelix[0];
393 phase[1][0] = phase[0][0] = GetPhase(xx2,yy2);
394 ri[1] = ri[0] = xx*xx+yy*yy;
398 if ( (d+c1[2])<c2[2]){
399 if ( (d+c1[2])+cut<c2[2]) return 0;
401 Double_t xx = -c2[0]*c1[2]/d+ fHelix[5];
402 Double_t yy = -c2[1]*c1[2]/d+ fHelix[0];
403 phase[1][1] = phase[0][1] = GetPhase(xx,yy);
405 Double_t xx2 = c2[0]- c2[0]*c2[2]/d+ fHelix[5];
406 Double_t yy2 = c2[1]- c2[1]*c2[2]/d+ fHelix[0];
407 phase[1][0] = phase[0][0] = h.GetPhase(xx2,yy2);
408 ri[1] = ri[0] = xx*xx+yy*yy;
412 Double_t d1 = (d*d+c1[2]*c1[2]-c2[2]*c2[2])/(2.*d);
413 Double_t v1 = c1[2]*c1[2]-d1*d1;
415 v1 = TMath::Sqrt(v1);
417 x0[0] = (c2[0]*d1+c2[1]*v1)/d + fHelix[5];
418 y0[0] = (c2[1]*d1-c2[0]*v1)/d + fHelix[0];
420 x0[1] = (c2[0]*d1-c2[1]*v1)/d + fHelix[5];
421 y0[1] = (c2[1]*d1+c2[0]*v1)/d + fHelix[0];
423 for (Int_t i=0;i<2;i++){
424 phase[i][0] = GetPhase(x0[i],y0[i]);
425 phase[i][1] = h.GetPhase(x0[i],y0[i]);
426 ri[i] = x0[i]*x0[i]+y0[i]*y0[i];
433 Int_t AliHelix::LinearDCA(AliHelix &h, Double_t &t1, Double_t &t2,
434 Double_t &R, Double_t &dist)
438 // find intersection using linear approximation
439 Double_t r1[3],g1[3],gg1[3];
440 Double_t r2[3],g2[3],gg2[3];
442 Evaluate(t1,r1,g1,gg1);
443 h.Evaluate(t2,r2,g2,gg2);
445 Double_t g1_2 = g1[0]*g1[0] +g1[1]*g1[1] +g1[2]*g1[2];
446 Double_t g2_2 = g2[0]*g2[0] +g2[1]*g2[1] +g2[2]*g2[2];
447 Double_t g1x2 = g1[0]*g2[0] +g1[1]*g2[1] +g1[2]*g2[2];
448 Double_t det = g1_2*g2_2 - g1x2*g1x2;
450 if (TMath::Abs(det)>0){
452 Double_t r1g1 = r1[0]*g1[0] +r1[1]*g1[1] +r1[2]*g1[2];
453 Double_t r2g1 = r2[0]*g1[0] +r2[1]*g1[1] +r2[2]*g1[2];
454 Double_t r1g2 = r1[0]*g2[0] +r1[1]*g2[1] +r1[2]*g2[2];
455 Double_t r2g2 = r2[0]*g2[0] +r2[1]*g2[1] +r2[2]*g2[2];
457 Double_t dt = - ( g2_2*(r1g1-r2g1) - g1x2*(r1g2-r2g2)) / det;
458 Double_t dp = - ( g1_2*(r2g2-r1g2) - g1x2*(r2g1-r1g1)) / det;
465 dist = (r1[0]-r2[0])*(r1[0]-r2[0])+
466 (r1[1]-r2[1])*(r1[1]-r2[1])+
467 (r1[2]-r2[2])*(r1[2]-r2[2]);
468 R = ((r1[0]+r2[0])*(r1[0]+r2[0])+(r1[1]+r2[1])*(r1[1]+r2[1]))/4.;
474 Int_t AliHelix::ParabolicDCA(AliHelix&h, //helixes
475 Double_t &t1, Double_t &t2,
476 Double_t &R, Double_t &dist, Int_t iter)
480 // find intersection using linear fit
481 Double_t r1[3],g1[3],gg1[3];
482 Double_t r2[3],g2[3],gg2[3];
484 Evaluate(t1,r1,g1,gg1);
485 h.Evaluate(t2,r2,g2,gg2);
492 Double_t dx=r2[0]-r1[0], dy=r2[1]-r1[1], dz=r2[2]-r1[2];
493 Double_t dm=dx*dx/dx2 + dy*dy/dy2 + dz*dz/dz2;
498 Double_t gt1=-(dx*g1[0]/dx2 + dy*g1[1]/dy2 + dz*g1[2]/dz2);
499 Double_t gt2=+(dx*g2[0]/dx2 + dy*g2[1]/dy2 + dz*g2[2]/dz2);
501 Double_t h11=(g1[0]*g1[0] - dx*gg1[0])/dx2 +
502 (g1[1]*g1[1] - dy*gg1[1])/dy2 +
503 (g1[2]*g1[2] - dz*gg1[2])/dz2;
504 Double_t h22=(g2[0]*g2[0] + dx*gg2[0])/dx2 +
505 (g2[1]*g2[1] + dy*gg2[1])/dy2 +
506 (g2[2]*g2[2] + dz*gg2[2])/dz2;
507 Double_t h12=-(g1[0]*g2[0]/dx2 + g1[1]*g2[1]/dy2 + g1[2]*g2[2]/dz2);
509 Double_t det=h11*h22-h12*h12;
512 if (TMath::Abs(det)<1.e-33) {
513 //(quasi)singular Hessian
516 dt1=-(gt1*h22 - gt2*h12)/det;
517 dt2=-(h11*gt2 - h12*gt1)/det;
520 if ((dt1*gt1+dt2*gt2)>0) {dt1=-dt1; dt2=-dt2;}
522 //if (TMath::Abs(dt1)/(TMath::Abs(t1)+1.e-3) < 1.e-4)
523 // if (TMath::Abs(dt2)/(TMath::Abs(t2)+1.e-3) < 1.e-4) {
528 for (Int_t div=1 ; div<512 ; div*=2) {
529 Evaluate(t1+dt1,r1,g1,gg1);
530 h.Evaluate(t2+dt2,r2,g2,gg2);
531 dx=r2[0]-r1[0]; dy=r2[1]-r1[1]; dz=r2[2]-r1[2];
532 dd=dx*dx/dx2 + dy*dy/dy2 + dz*dz/dz2;
546 Evaluate(t1,r1,g1,gg1);
547 h.Evaluate(t2,r2,g2,gg2);
549 dist = (r1[0]-r2[0])*(r1[0]-r2[0])+
550 (r1[1]-r2[1])*(r1[1]-r2[1])+
551 (r1[2]-r2[2])*(r1[2]-r2[2]);
553 R = ((r1[0]+r2[0])*(r1[0]+r2[0])+(r1[1]+r2[1])*(r1[1]+r2[1]))/4;
559 Int_t AliHelix::ParabolicDCA2(AliHelix&h, //helixes
560 Double_t &t1, Double_t &t2,
561 Double_t &R, Double_t &dist, Double_t err[3], Int_t iter)
565 // find intersection using linear fit
566 Double_t r1[3],g1[3],gg1[3];
567 Double_t r2[3],g2[3],gg2[3];
569 Evaluate(t1,r1,g1,gg1);
570 h.Evaluate(t2,r2,g2,gg2);
577 Double_t dx=r2[0]-r1[0], dy=r2[1]-r1[1], dz=r2[2]-r1[2];
578 Double_t dm=dx*dx/dx2 + dy*dy/dy2 + dz*dz/dz2;
583 Double_t gt1=-(dx*g1[0]/dx2 + dy*g1[1]/dy2 + dz*g1[2]/dz2);
584 Double_t gt2=+(dx*g2[0]/dx2 + dy*g2[1]/dy2 + dz*g2[2]/dz2);
586 Double_t h11=(g1[0]*g1[0] - dx*gg1[0])/dx2 +
587 (g1[1]*g1[1] - dy*gg1[1])/dy2 +
588 (g1[2]*g1[2] - dz*gg1[2])/dz2;
589 Double_t h22=(g2[0]*g2[0] + dx*gg2[0])/dx2 +
590 (g2[1]*g2[1] + dy*gg2[1])/dy2 +
591 (g2[2]*g2[2] + dz*gg2[2])/dz2;
592 Double_t h12=-(g1[0]*g2[0]/dx2 + g1[1]*g2[1]/dy2 + g1[2]*g2[2]/dz2);
594 Double_t det=h11*h22-h12*h12;
597 if (TMath::Abs(det)<1.e-33) {
598 //(quasi)singular Hessian
601 dt1=-(gt1*h22 - gt2*h12)/det;
602 dt2=-(h11*gt2 - h12*gt1)/det;
605 if ((dt1*gt1+dt2*gt2)>0) {dt1=-dt1; dt2=-dt2;}
607 //if (TMath::Abs(dt1)/(TMath::Abs(t1)+1.e-3) < 1.e-4)
608 // if (TMath::Abs(dt2)/(TMath::Abs(t2)+1.e-3) < 1.e-4) {
613 for (Int_t div=1 ; div<512 ; div*=2) {
614 Evaluate(t1+dt1,r1,g1,gg1);
615 h.Evaluate(t2+dt2,r2,g2,gg2);
616 dx=r2[0]-r1[0]; dy=r2[1]-r1[1]; dz=r2[2]-r1[2];
617 dd=dx*dx/dx2 + dy*dy/dy2 + dz*dz/dz2;
631 Evaluate(t1,r1,g1,gg1);
632 h.Evaluate(t2,r2,g2,gg2);
634 dist = (r1[0]-r2[0])*(r1[0]-r2[0])+
635 (r1[1]-r2[1])*(r1[1]-r2[1])+
636 (r1[2]-r2[2])*(r1[2]-r2[2]);
638 R = ((r1[0]+r2[0])*(r1[0]+r2[0])+(r1[1]+r2[1])*(r1[1]+r2[1]))/4;