Increase speed of SDD raw data decoding (F.Prino)
[u/mrichter/AliRoot.git] / ITS / AliITStrackV2.cxx
CommitLineData
006b5f7f 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
ae00569a 16/* $Id$ */
17
66e811e2 18///////////////////////////////////////////////////////////////////////////
006b5f7f 19// Implementation of the ITS track class
20//
21// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
a9a2d814 22// dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
66e811e2 23///////////////////////////////////////////////////////////////////////////
006b5f7f 24#include <TMath.h>
006b5f7f 25
26#include "AliCluster.h"
6c94f330 27#include "AliTracker.h"
83d73500 28#include "AliESDtrack.h"
006b5f7f 29#include "AliITStrackV2.h"
30
8602c008 31const Int_t AliITStrackV2::fgkWARN = 5;
32
1aedd96e 33ClassImp(AliITStrackV2)
8676d691 34
e43c066c 35
006b5f7f 36//____________________________________________________________________________
6c94f330 37AliITStrackV2::AliITStrackV2() : AliKalmanTrack(),
22b13da0 38 fdEdx(0),
83d73500 39 fESDtrack(0)
e43c066c 40{
ae00569a 41 for(Int_t i=0; i<2*AliITSgeomTGeo::kNLayers; i++) {fIndex[i]=-1; fModule[i]=-1;}
42 for(Int_t i=0; i<4; i++) fdEdxSample[i]=0;
e43c066c 43}
44
83d73500 45
22b13da0 46//____________________________________________________________________________
67c3dcbe 47AliITStrackV2::AliITStrackV2(AliESDtrack& t,Bool_t c) throw (const Char_t *) :
6c94f330 48 AliKalmanTrack(),
49 fdEdx(t.GetITSsignal()),
50 fESDtrack(&t)
51{
83d73500 52 //------------------------------------------------------------------
67c3dcbe 53 // Conversion ESD track -> ITS track.
54 // If c==kTRUE, create the ITS track out of the constrained params.
83d73500 55 //------------------------------------------------------------------
6c94f330 56 const AliExternalTrackParam *par=&t;
57 if (c) {
6c4ef2ed 58 par=t.GetConstrainedParam();
6c94f330 59 if (!par) throw "AliITStrackV2: conversion failed !\n";
60 }
61 Set(par->GetX(),par->GetAlpha(),par->GetParameter(),par->GetCovariance());
62
63 //if (!Invariant()) throw "AliITStrackV2: conversion failed !\n";
64
83d73500 65 SetLabel(t.GetLabel());
66 SetMass(t.GetMass());
6c94f330 67 SetNumberOfClusters(t.GetITSclusters(fIndex));
006b5f7f 68
83d73500 69 if (t.GetStatus()&AliESDtrack::kTIME) {
70 StartTimeIntegral();
71 Double_t times[10]; t.GetIntegratedTimes(times); SetIntegratedTimes(times);
72 SetIntegratedLength(t.GetIntegratedLength());
73 }
e43c066c 74
68b8060b 75 for(Int_t i=0; i<4; i++) fdEdxSample[i]=0;
006b5f7f 76}
77
7ac23097 78//____________________________________________________________________________
79void AliITStrackV2::ResetClusters() {
80 //------------------------------------------------------------------
81 // Reset the array of attached clusters.
82 //------------------------------------------------------------------
83 for (Int_t i=0; i<2*AliITSgeomTGeo::kNLayers; i++) fIndex[i]=-1;
84 SetChi2(0.);
85 SetNumberOfClusters(0);
86}
87
15dd636f 88void AliITStrackV2::UpdateESDtrack(ULong_t flags) const {
83d73500 89 fESDtrack->UpdateTrackParams(this,flags);
ae00569a 90 // copy the module indices
91 for(Int_t i=0;i<12;i++) {
92 // printf(" %d\n",GetModuleIndex(i));
93 fESDtrack->SetITSModuleIndex(i,GetModuleIndex(i));
94 }
83d73500 95}
96
006b5f7f 97//____________________________________________________________________________
6c94f330 98AliITStrackV2::AliITStrackV2(const AliITStrackV2& t) :
99 AliKalmanTrack(t),
100 fdEdx(t.fdEdx),
101 fESDtrack(t.fESDtrack)
102{
006b5f7f 103 //------------------------------------------------------------------
104 //Copy constructor
105 //------------------------------------------------------------------
ef7253ac 106 Int_t i;
ef7253ac 107 for (i=0; i<4; i++) fdEdxSample[i]=t.fdEdxSample[i];
ae00569a 108 for (i=0; i<2*AliITSgeomTGeo::GetNLayers(); i++) {
109 fIndex[i]=t.fIndex[i];
110 fModule[i]=t.fModule[i];
111 }
006b5f7f 112}
a9a2d814 113
006b5f7f 114//_____________________________________________________________________________
115Int_t AliITStrackV2::Compare(const TObject *o) const {
116 //-----------------------------------------------------------------
117 // This function compares tracks according to the their curvature
118 //-----------------------------------------------------------------
7f6ddf58 119 AliITStrackV2 *t=(AliITStrackV2*)o;
6c23ffed 120 //Double_t co=OneOverPt();
121 //Double_t c =OneOverPt();
15dd636f 122 Double_t co=t->GetSigmaY2()*t->GetSigmaZ2();
123 Double_t c =GetSigmaY2()*GetSigmaZ2();
006b5f7f 124 if (c>co) return 1;
125 else if (c<co) return -1;
126 return 0;
127}
128
006b5f7f 129//____________________________________________________________________________
6c94f330 130Bool_t
131AliITStrackV2::PropagateToVertex(const AliESDVertex *v,Double_t d,Double_t x0)
132{
006b5f7f 133 //------------------------------------------------------------------
134 //This function propagates a track to the minimal distance from the origin
6c94f330 135 //------------------------------------------------------------------
136 Double_t bz=GetBz();
e50912db 137 if (PropagateToDCA(v,bz,kVeryBig)) {
138 Double_t xOverX0,xTimesRho;
139 xOverX0 = d; xTimesRho = d*x0;
140 if (CorrectForMeanMaterial(xOverX0,xTimesRho,kTRUE)) return kTRUE;
141 }
6c94f330 142 return kFALSE;
006b5f7f 143}
144
145//____________________________________________________________________________
6c94f330 146Bool_t AliITStrackV2::
e50912db 147GetGlobalXYZat(Double_t xloc, Double_t &x, Double_t &y, Double_t &z) const {
006b5f7f 148 //------------------------------------------------------------------
149 //This function returns a track position in the global system
150 //------------------------------------------------------------------
6c94f330 151 Double_t r[3];
e50912db 152 Bool_t rc=GetXYZAt(xloc, AliTracker::GetBz(), r);
6c94f330 153 x=r[0]; y=r[1]; z=r[2];
154 return rc;
006b5f7f 155}
156
157//_____________________________________________________________________________
6c94f330 158Double_t AliITStrackV2::GetPredictedChi2(const AliCluster *c) const {
006b5f7f 159 //-----------------------------------------------------------------
160 // This function calculates a predicted chi2 increment.
161 //-----------------------------------------------------------------
6c94f330 162 Double_t p[2]={c->GetY(), c->GetZ()};
163 Double_t cov[3]={c->GetSigmaY2(), 0., c->GetSigmaZ2()};
164 return AliExternalTrackParam::GetPredictedChi2(p,cov);
a9a2d814 165}
166
167//____________________________________________________________________________
6c94f330 168Bool_t AliITStrackV2::PropagateTo(Double_t xk, Double_t d, Double_t x0) {
006b5f7f 169 //------------------------------------------------------------------
170 //This function propagates a track
171 //------------------------------------------------------------------
afd25725 172
6c94f330 173 Double_t oldX=GetX(), oldY=GetY(), oldZ=GetZ();
006b5f7f 174
6c94f330 175 Double_t bz=GetBz();
176 if (!AliExternalTrackParam::PropagateTo(xk,bz)) return kFALSE;
e50912db 177 Double_t xOverX0,xTimesRho;
178 xOverX0 = d; xTimesRho = d*x0;
179 if (!CorrectForMeanMaterial(xOverX0,xTimesRho,kTRUE)) return kFALSE;
006b5f7f 180
dd44614b 181 Double_t x=GetX(), y=GetY(), z=GetZ();
6c94f330 182 if (IsStartedTimeIntegral() && x>oldX) {
183 Double_t l2 = (x-oldX)*(x-oldX) + (y-oldY)*(y-oldY) + (z-oldZ)*(z-oldZ);
f29dbb4b 184 AddTimeStep(TMath::Sqrt(l2));
185 }
f29dbb4b 186
6c94f330 187 return kTRUE;
006b5f7f 188}
189
190//____________________________________________________________________________
e50912db 191Bool_t AliITStrackV2::PropagateToTGeo(Double_t xToGo, Int_t nstep, Double_t &xOverX0, Double_t &xTimesRho) {
afd25725 192 //-------------------------------------------------------------------
193 // Propagates the track to a reference plane x=xToGo in n steps.
194 // These n steps are only used to take into account the curvature.
195 // The material is calculated with TGeo. (L.Gaudichet)
196 //-------------------------------------------------------------------
197
198 Double_t startx = GetX(), starty = GetY(), startz = GetZ();
199 Double_t sign = (startx<xToGo) ? -1.:1.;
200 Double_t step = (xToGo-startx)/TMath::Abs(nstep);
201
202 Double_t start[3], end[3], mparam[7], bz = GetBz();
203 Double_t x = startx;
204
205 for (Int_t i=0; i<nstep; i++) {
206
207 GetXYZ(start); //starting global position
208 x += step;
209 if (!GetXYZAt(x, bz, end)) return kFALSE;
210 if (!AliExternalTrackParam::PropagateTo(x, bz)) return kFALSE;
211 AliTracker::MeanMaterialBudget(start, end, mparam);
afd25725 212 if (mparam[1]<900000) {
e50912db 213 xTimesRho = sign*mparam[4]*mparam[0];
214 xOverX0 = mparam[1];
afd25725 215 if (!AliExternalTrackParam::CorrectForMeanMaterial(xOverX0,
e50912db 216 xTimesRho,GetMass())) return kFALSE;
afd25725 217 }
218 }
219
220 if (IsStartedTimeIntegral() && GetX()>startx) {
221 Double_t l2 = ( (GetX()-startx)*(GetX()-startx) +
222 (GetY()-starty)*(GetY()-starty) +
223 (GetZ()-startz)*(GetZ()-startz) );
224 AddTimeStep(TMath::Sqrt(l2));
225 }
226
227 return kTRUE;
228}
229
230//____________________________________________________________________________
6c94f330 231Bool_t AliITStrackV2::Update(const AliCluster* c, Double_t chi2, Int_t index)
232{
006b5f7f 233 //------------------------------------------------------------------
234 //This function updates track parameters
235 //------------------------------------------------------------------
6c94f330 236 Double_t p[2]={c->GetY(), c->GetZ()};
237 Double_t cov[3]={c->GetSigmaY2(), 0., c->GetSigmaZ2()};
006b5f7f 238
6c94f330 239 if (!AliExternalTrackParam::Update(p,cov)) return kFALSE;
006b5f7f 240
28d5b13d 241 Int_t n=GetNumberOfClusters();
006b5f7f 242 if (!Invariant()) {
28d5b13d 243 if (n>fgkWARN) AliWarning("Wrong invariant !");
6c94f330 244 return kFALSE;
006b5f7f 245 }
246
6c94f330 247 if (chi2<0) return kTRUE;
67c3dcbe 248
150f264c 249 AliTracker::FillResiduals(this,p,cov,c->GetVolumeId());
250
006b5f7f 251 fIndex[n]=index;
252 SetNumberOfClusters(n+1);
253 SetChi2(GetChi2()+chi2);
254
6c94f330 255 return kTRUE;
006b5f7f 256}
257
6c94f330 258Bool_t AliITStrackV2::Invariant() const {
006b5f7f 259 //------------------------------------------------------------------
260 // This function is for debugging purpose only
261 //------------------------------------------------------------------
7f6ddf58 262 Int_t n=GetNumberOfClusters();
6c94f330 263
264 Double_t sP2=GetParameter()[2];
265 if (TMath::Abs(sP2) >= kAlmost1){
8602c008 266 if (n>fgkWARN) Warning("Invariant","fP2=%f\n",sP2);
6c94f330 267 return kFALSE;
a9a2d814 268 }
6c94f330 269 Double_t sC00=GetCovariance()[0];
270 if (sC00<=0 || sC00>9.) {
8602c008 271 if (n>fgkWARN) Warning("Invariant","fC00=%f\n",sC00);
6c94f330 272 return kFALSE;
a9a2d814 273 }
6c94f330 274 Double_t sC11=GetCovariance()[2];
275 if (sC11<=0 || sC11>9.) {
8602c008 276 if (n>fgkWARN) Warning("Invariant","fC11=%f\n",sC11);
6c94f330 277 return kFALSE;
a9a2d814 278 }
6c94f330 279 Double_t sC22=GetCovariance()[5];
280 if (sC22<=0 || sC22>1.) {
8602c008 281 if (n>fgkWARN) Warning("Invariant","fC22=%f\n",sC22);
6c94f330 282 return kFALSE;
a9a2d814 283 }
6c94f330 284 Double_t sC33=GetCovariance()[9];
285 if (sC33<=0 || sC33>1.) {
8602c008 286 if (n>fgkWARN) Warning("Invariant","fC33=%f\n",sC33);
6c94f330 287 return kFALSE;
a9a2d814 288 }
6c94f330 289 Double_t sC44=GetCovariance()[14];
290 if (sC44<=0 /*|| sC44>6e-5*/) {
8602c008 291 if (n>fgkWARN) Warning("Invariant","fC44=%f\n",sC44);
6c94f330 292 return kFALSE;
14825d5a 293 }
6c94f330 294
295 return kTRUE;
006b5f7f 296}
297
298//____________________________________________________________________________
6c94f330 299Bool_t AliITStrackV2::Propagate(Double_t alp,Double_t xk) {
006b5f7f 300 //------------------------------------------------------------------
301 //This function propagates a track
302 //------------------------------------------------------------------
6c94f330 303 Double_t bz=GetBz();
304 if (!AliExternalTrackParam::Propagate(alp,xk,bz)) return kFALSE;
006b5f7f 305
006b5f7f 306 if (!Invariant()) {
6c94f330 307 AliWarning("Wrong invariant !");
308 return kFALSE;
791f9a2a 309 }
310
6c94f330 311 return kTRUE;
a9a2d814 312}
006b5f7f 313
afd25725 314Bool_t AliITStrackV2::MeanBudgetToPrimVertex(Double_t xyz[3], Double_t step, Double_t &d) const {
315
316 //-------------------------------------------------------------------
317 // Get the mean material budget between the actual point and the
318 // primary vertex. (L.Gaudichet)
319 //-------------------------------------------------------------------
320
321 Double_t cs=TMath::Cos(GetAlpha()), sn=TMath::Sin(GetAlpha());
322 Double_t vertexX = xyz[0]*cs + xyz[1]*sn;
323
324 Int_t nstep = Int_t((GetX()-vertexX)/step);
325 if (nstep<1) nstep = 1;
326 step = (GetX()-vertexX)/nstep;
327
c7719399 328 // Double_t mparam[7], densMean=0, radLength=0, length=0;
329 Double_t mparam[7];
afd25725 330 Double_t p1[3], p2[3], x = GetX(), bz = GetBz();
331 GetXYZ(p1);
332
333 d=0.;
334
335 for (Int_t i=0; i<nstep; i++) {
336 x += step;
337 if (!GetXYZAt(x, bz, p2)) return kFALSE;
338 AliTracker::MeanMaterialBudget(p1, p2, mparam);
339 if (mparam[1]>900000) return kFALSE;
340 d += mparam[1];
341
342 p1[0] = p2[0];
343 p1[1] = p2[1];
344 p1[2] = p2[2];
345 }
346
347 return kTRUE;
348}
349
6c94f330 350Bool_t AliITStrackV2::Improve(Double_t x0,Double_t xyz[3],Double_t ers[3]) {
babd135a 351 //------------------------------------------------------------------
6c94f330 352 //This function improves angular track parameters
babd135a 353 //------------------------------------------------------------------
6c94f330 354 Double_t cs=TMath::Cos(GetAlpha()), sn=TMath::Sin(GetAlpha());
355//Double_t xv = xyz[0]*cs + xyz[1]*sn; // vertex
356 Double_t yv =-xyz[0]*sn + xyz[1]*cs; // in the
357 Double_t zv = xyz[2]; // local frame
babd135a 358
6c94f330 359 Double_t dy = Par(0) - yv, dz = Par(1) - zv;
360 Double_t r2=GetX()*GetX() + dy*dy;
6c23ffed 361 Double_t p2=(1.+ GetTgl()*GetTgl())/(GetSigned1Pt()*GetSigned1Pt());
a9a2d814 362 Double_t beta2=p2/(p2 + GetMass()*GetMass());
363 x0*=TMath::Sqrt((1.+ GetTgl()*GetTgl())/(1.- GetSnp()*GetSnp()));
8676d691 364 Double_t theta2=14.1*14.1/(beta2*p2*1e6)*x0;
365 //Double_t theta2=1.0259e-6*14*14/28/(beta2*p2)*x0*9.36*2.33;
6c94f330 366
367 Double_t cnv=GetBz()*kB2C;
a9a2d814 368 {
6c94f330 369 Double_t dummy = 4/r2 - GetC()*GetC();
370 if (dummy < 0) return kFALSE;
371 Double_t parp = 0.5*(GetC()*GetX() + dy*TMath::Sqrt(dummy));
372 Double_t sigma2p = theta2*(1.- GetSnp()*GetSnp())*(1. + GetTgl()*GetTgl());
373 sigma2p += Cov(0)/r2*(1.- dy*dy/r2)*(1.- dy*dy/r2);
374 sigma2p += ers[1]*ers[1]/r2;
375 sigma2p += 0.25*Cov(14)*cnv*cnv*GetX()*GetX();
00709873 376 Double_t eps2p=sigma2p/(Cov(5) + sigma2p);
6c94f330 377 Par(0) += Cov(3)/(Cov(5) + sigma2p)*(parp - GetSnp());
378 Par(2) = eps2p*GetSnp() + (1 - eps2p)*parp;
379 Cov(5) *= eps2p;
380 Cov(3) *= eps2p;
a9a2d814 381 }
382 {
6c94f330 383 Double_t parl=0.5*GetC()*dz/TMath::ASin(0.5*GetC()*TMath::Sqrt(r2));
384 Double_t sigma2l=theta2;
385 sigma2l += Cov(2)/r2 + Cov(0)*dy*dy*dz*dz/(r2*r2*r2);
386 sigma2l += ers[2]*ers[2]/r2;
387 Double_t eps2l = sigma2l/(Cov(9) + sigma2l);
388 Par(1) += Cov(7 )/(Cov(9) + sigma2l)*(parl - Par(3));
389 Par(4) += Cov(13)/(Cov(9) + sigma2l)*(parl - Par(3));
390 Par(3) = eps2l*Par(3) + (1-eps2l)*parl;
391 Cov(9) *= eps2l;
00709873 392 Cov(13)*= eps2l;
6c94f330 393 Cov(7) *= eps2l;
a9a2d814 394 }
6c94f330 395 if (!Invariant()) return kFALSE;
14825d5a 396
6c94f330 397 return kTRUE;
14825d5a 398}
23efe5f1 399
400void AliITStrackV2::CookdEdx(Double_t low, Double_t up) {
401 //-----------------------------------------------------------------
a9a2d814 402 // This function calculates dE/dX within the "low" and "up" cuts.
403 // Origin: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
23efe5f1 404 //-----------------------------------------------------------------
23efe5f1 405 // The clusters order is: SSD-2, SSD-1, SDD-2, SDD-1, SPD-2, SPD-1
c0dd5278 406
407 Int_t i;
408 Int_t nc=0;
409 for (i=0; i<GetNumberOfClusters(); i++) {
410 Int_t idx=GetClusterIndex(i);
411 idx=(idx&0xf0000000)>>28;
412 if (idx>1) nc++; // Take only SSD and SDD
413 }
23efe5f1 414
a9a2d814 415 Int_t swap;//stupid sorting
23efe5f1 416 do {
417 swap=0;
418 for (i=0; i<nc-1; i++) {
419 if (fdEdxSample[i]<=fdEdxSample[i+1]) continue;
420 Float_t tmp=fdEdxSample[i];
421 fdEdxSample[i]=fdEdxSample[i+1]; fdEdxSample[i+1]=tmp;
422 swap++;
423 }
424 } while (swap);
425
a9a2d814 426 Int_t nl=Int_t(low*nc), nu=Int_t(up*nc); //b.b. to take two lowest dEdX
427 // values from four ones choose
428 // nu=2
23efe5f1 429 Float_t dedx=0;
a9a2d814 430 for (i=nl; i<nu; i++) dedx += fdEdxSample[i];
c0dd5278 431 if (nu-nl>0) dedx /= (nu-nl);
23efe5f1 432
23efe5f1 433 SetdEdx(dedx);
434}
c7bafca9 435
6c94f330 436Double_t AliITStrackV2::GetBz() const {
437 //
438 // returns Bz component of the magnetic field (kG)
439 //
440 if (AliTracker::UniformField()) return AliTracker::GetBz();
441 Double_t r[3]; GetXYZ(r);
442 return AliTracker::GetBz(r);
5773defd 443}
6c94f330 444
8602c008 445//____________________________________________________________________________
446Bool_t AliITStrackV2::
e50912db 447GetPhiZat(Double_t r, Double_t &phi, Double_t &z) const {
448 //------------------------------------------------------------------
449 // This function returns the global cylindrical (phi,z) of the track
450 // position estimated at the radius r.
451 // The track curvature is neglected.
452 //------------------------------------------------------------------
453 Double_t d=GetD(0.,0.);
454 if (TMath::Abs(d) > r) return kFALSE;
455
456 Double_t rcurr=TMath::Sqrt(GetX()*GetX() + GetY()*GetY());
457 if (TMath::Abs(d) > rcurr) return kFALSE;
458 Double_t phicurr=GetAlpha()+TMath::ASin(GetSnp());
459
460 phi=phicurr+TMath::ASin(d/r)-TMath::ASin(d/rcurr);
c7de5c65 461 z=GetZ()+GetTgl()*(TMath::Sqrt((r-d)*(r+d))-TMath::Sqrt((rcurr-d)*(rcurr+d)));
e50912db 462 return kTRUE;
463}
464//____________________________________________________________________________
465Bool_t AliITStrackV2::
466GetLocalXat(Double_t r,Double_t &xloc) const {
8602c008 467 //------------------------------------------------------------------
e50912db 468 // This function returns the local x of the track
469 // position estimated at the radius r.
8602c008 470 // The track curvature is neglected.
471 //------------------------------------------------------------------
472 Double_t d=GetD(0.,0.);
e50912db 473 if (TMath::Abs(d) > r) return kFALSE;
8602c008 474
e50912db 475 Double_t rcurr=TMath::Sqrt(GetX()*GetX() + GetY()*GetY());
476 Double_t phicurr=GetAlpha()+TMath::ASin(GetSnp());
477 Double_t phi=phicurr+TMath::ASin(d/r)-TMath::ASin(d/rcurr);
8602c008 478
e50912db 479 xloc=r*(TMath::Cos(phi)*TMath::Cos(GetAlpha())
480 +TMath::Sin(phi)*TMath::Sin(GetAlpha()));
8602c008 481
482 return kTRUE;
483}