Merge branch 'master' of https://git.cern.ch/reps/AliRoot
[u/mrichter/AliRoot.git] / ITS / AliITStrackerV2.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 **************************************************************************/
a5d0fae6 15/* $Id$ */
006b5f7f 16//-------------------------------------------------------------------------
17// Implementation of the ITS tracker class
00a7cc50 18// It reads AliITSRecPoint clusters and creates AliITStrackV2 tracks
18856a77 19// and fills with them the ESD
006b5f7f 20// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
a9a2d814 21// dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
006b5f7f 22//-------------------------------------------------------------------------
1f9a65c4 23
9c369bfe 24#include <new>
25
75fb37cc 26#include <TError.h>
006b5f7f 27#include <TFile.h>
28#include <TTree.h>
29#include <TRandom.h>
a504d56f 30#include <TGeoMatrix.h>
006b5f7f 31
e341247d 32#include "AliITSgeomTGeo.h"
a504d56f 33#include "AliAlignObj.h"
006b5f7f 34#include "AliITSRecPoint.h"
af885e0f 35#include "AliESDEvent.h"
aad72f45 36#include "AliESDtrack.h"
00a7cc50 37#include "AliITSRecPoint.h"
44347160 38#include "AliITSReconstructor.h"
006b5f7f 39#include "AliITStrackerV2.h"
40
8676d691 41ClassImp(AliITStrackerV2)
006b5f7f 42
e50912db 43AliITStrackerV2::AliITSlayer AliITStrackerV2::fgLayers[AliITSgeomTGeo::kNLayers]; //ITS layers
006b5f7f 44
6c94f330 45AliITStrackerV2::AliITStrackerV2():
46 AliTracker(),
e50912db 47 fI(AliITSgeomTGeo::GetNLayers()),
6c94f330 48 fBestTrack(),
49 fTrackToFollow(),
50 fPass(0),
e50912db 51 fLastLayerToTrackTo(AliITSRecoParam::GetLastLayerToTrackTo())
6c94f330 52{
53 //--------------------------------------------------------------------
54 //This is the AliITStrackerV2 default constructor
55 //--------------------------------------------------------------------
56
e50912db 57 for (Int_t i=1; i<AliITSgeomTGeo::GetNLayers()+1; i++) new(fgLayers+i-1) AliITSlayer();
6c94f330 58
59 fConstraint[0]=1; fConstraint[1]=0;
60
44347160 61 Double_t xyz[]={AliITSReconstructor::GetRecoParam()->GetXVdef(),
62 AliITSReconstructor::GetRecoParam()->GetYVdef(),
63 AliITSReconstructor::GetRecoParam()->GetZVdef()};
64 Double_t ers[]={AliITSReconstructor::GetRecoParam()->GetSigmaXVdef(),
65 AliITSReconstructor::GetRecoParam()->GetSigmaYVdef(),
66 AliITSReconstructor::GetRecoParam()->GetSigmaZVdef()};
6c94f330 67 SetVertex(xyz,ers);
68
e50912db 69 for (Int_t i=0; i<AliITSgeomTGeo::GetNLayers(); i++) fLayersNotToSkip[i]=AliITSRecoParam::GetLayersNotToSkip(i);
6c94f330 70
71}
72
73AliITStrackerV2::AliITStrackerV2(const AliITStrackerV2 &t):
74 AliTracker(t),
75 fI(t.fI),
76 fBestTrack(t.fBestTrack),
77 fTrackToFollow(t.fTrackToFollow),
78 fPass(t.fPass),
79 fLastLayerToTrackTo(t.fLastLayerToTrackTo)
80{
81 //--------------------------------------------------------------------
82 //This is the AliITStrackerV2 copy constructor
83 //--------------------------------------------------------------------
84
e50912db 85 //for (Int_t i=1; i<AliITSgeomTGeo::GetNLayers()+1; i++) new(fgLayers+i-1) AliITSlayer();
6c94f330 86
87 fConstraint[0]=t.fConstraint[0]; fConstraint[1]=t.fConstraint[1];
88
44347160 89 Double_t xyz[]={AliITSReconstructor::GetRecoParam()->GetXVdef(),
90 AliITSReconstructor::GetRecoParam()->GetYVdef(),
91 AliITSReconstructor::GetRecoParam()->GetZVdef()};
92 Double_t ers[]={AliITSReconstructor::GetRecoParam()->GetSigmaXVdef(),
93 AliITSReconstructor::GetRecoParam()->GetSigmaYVdef(),
94 AliITSReconstructor::GetRecoParam()->GetSigmaZVdef()};
6c94f330 95 xyz[0]=t.GetX(); xyz[1]=t.GetY(); xyz[2]=t.GetZ();
96 ers[0]=t.GetSigmaX(); ers[1]=t.GetSigmaY(); ers[2]=t.GetSigmaZ();
97 SetVertex(xyz,ers);
98
e50912db 99 for (Int_t i=0; i<AliITSgeomTGeo::GetNLayers(); i++) fLayersNotToSkip[i]=t.fLayersNotToSkip[i];
6c94f330 100
101}
102
1f3e997f 103AliITStrackerV2::AliITStrackerV2(const Char_t *geom) :
6c94f330 104 AliTracker(),
e50912db 105 fI(AliITSgeomTGeo::GetNLayers()),
6c94f330 106 fBestTrack(),
107 fTrackToFollow(),
108 fPass(0),
e50912db 109 fLastLayerToTrackTo(AliITSRecoParam::GetLastLayerToTrackTo())
6c94f330 110{
006b5f7f 111 //--------------------------------------------------------------------
61ab8ea8 112 //This is the AliITStrackerV2 constructor
006b5f7f 113 //--------------------------------------------------------------------
e341247d 114 if (geom) {
115 AliWarning("\"geom\" is actually a dummy argument !");
116 }
006b5f7f 117
e50912db 118 for (Int_t i=1; i<AliITSgeomTGeo::GetNLayers()+1; i++) {
e341247d 119 Int_t nlad=AliITSgeomTGeo::GetNLadders(i);
120 Int_t ndet=AliITSgeomTGeo::GetNDetectors(i);
006b5f7f 121
e341247d 122 Double_t xyz[3], &x=xyz[0], &y=xyz[1], &z=xyz[2];
123 AliITSgeomTGeo::GetOrigTranslation(i,1,1,xyz);
006b5f7f 124 Double_t poff=TMath::ATan2(y,x);
125 Double_t zoff=z;
a504d56f 126 Double_t r=TMath::Sqrt(x*x + y*y);
006b5f7f 127
e341247d 128 AliITSgeomTGeo::GetOrigTranslation(i,1,2,xyz);
006b5f7f 129 r += TMath::Sqrt(x*x + y*y);
e341247d 130 AliITSgeomTGeo::GetOrigTranslation(i,2,1,xyz);
006b5f7f 131 r += TMath::Sqrt(x*x + y*y);
e341247d 132 AliITSgeomTGeo::GetOrigTranslation(i,2,2,xyz);
006b5f7f 133 r += TMath::Sqrt(x*x + y*y);
134 r*=0.25;
135
18856a77 136 new (fgLayers+i-1) AliITSlayer(r,poff,zoff,nlad,ndet);
006b5f7f 137
138 for (Int_t j=1; j<nlad+1; j++) {
139 for (Int_t k=1; k<ndet+1; k++) { //Fill this layer with detectors
e341247d 140 TGeoHMatrix m; AliITSgeomTGeo::GetOrigMatrix(i,j,k,m);
1f3e997f 141 const TGeoHMatrix *tm=AliITSgeomTGeo::GetTracking2LocalMatrix(i,j,k);
142 m.Multiply(tm);
a5d0fae6 143 Double_t txyz[3]={0.};
144 xyz[0]=0.; xyz[1]=0.; xyz[2]=0.;
1f3e997f 145 m.LocalToMaster(txyz,xyz);
a5d0fae6 146 r=TMath::Sqrt(xyz[0]*xyz[0] + xyz[1]*xyz[1]);
1f3e997f 147 Double_t phi=TMath::ATan2(xyz[1],xyz[0]);
c219fd63 148
149 if (phi<0) phi+=TMath::TwoPi();
150 else if (phi>=TMath::TwoPi()) phi-=TMath::TwoPi();
151
18856a77 152 AliITSdetector &det=fgLayers[i-1].GetDetector((j-1)*ndet + k-1);
006b5f7f 153 new(&det) AliITSdetector(r,phi);
154 }
155 }
c630aafd 156
006b5f7f 157 }
158
c219fd63 159 fConstraint[0]=1; fConstraint[1]=0;
8676d691 160
44347160 161 Double_t xyz[]={AliITSReconstructor::GetRecoParam()->GetXVdef(),
162 AliITSReconstructor::GetRecoParam()->GetYVdef(),
163 AliITSReconstructor::GetRecoParam()->GetZVdef()};
164 Double_t ers[]={AliITSReconstructor::GetRecoParam()->GetSigmaXVdef(),
165 AliITSReconstructor::GetRecoParam()->GetSigmaYVdef(),
166 AliITSReconstructor::GetRecoParam()->GetSigmaZVdef()};
8676d691 167 SetVertex(xyz,ers);
c0dd5278 168
e50912db 169 for (Int_t i=0; i<AliITSgeomTGeo::GetNLayers(); i++) fLayersNotToSkip[i]=AliITSRecoParam::GetLayersNotToSkip(i);
c0dd5278 170
171}
172
173void AliITStrackerV2::SetLayersNotToSkip(Int_t *l) {
174 //--------------------------------------------------------------------
175 //This function set masks of the layers which must be not skipped
176 //--------------------------------------------------------------------
e50912db 177 for (Int_t i=0; i<AliITSgeomTGeo::GetNLayers(); i++) fLayersNotToSkip[i]=l[i];
61ab8ea8 178}
006b5f7f 179
c630aafd 180Int_t AliITStrackerV2::LoadClusters(TTree *cTree) {
61ab8ea8 181 //--------------------------------------------------------------------
182 //This function loads ITS clusters
183 //--------------------------------------------------------------------
00a7cc50 184 TBranch *branch=cTree->GetBranch("ITSRecPoints");
61ab8ea8 185 if (!branch) {
c630aafd 186 Error("LoadClusters"," can't get the branch !\n");
8676d691 187 return 1;
61ab8ea8 188 }
006b5f7f 189
00a7cc50 190 TClonesArray dummy("AliITSRecPoint",10000), *clusters=&dummy;
61ab8ea8 191 branch->SetAddress(&clusters);
006b5f7f 192
61ab8ea8 193 Int_t j=0;
e50912db 194 for (Int_t i=0; i<AliITSgeomTGeo::GetNLayers(); i++) {
18856a77 195 Int_t ndet=fgLayers[i].GetNdetectors();
196 Int_t jmax = j + fgLayers[i].GetNladders()*ndet;
c219fd63 197
198 Double_t r=fgLayers[i].GetR();
199 Double_t circ=TMath::TwoPi()*r;
200
61ab8ea8 201 for (; j<jmax; j++) {
202 if (!cTree->GetEvent(j)) continue;
203 Int_t ncl=clusters->GetEntriesFast();
1f3e997f 204
61ab8ea8 205 while (ncl--) {
00a7cc50 206 AliITSRecPoint *c=(AliITSRecPoint*)clusters->UncheckedAt(ncl);
c219fd63 207
75fb37cc 208 if (!c->Misalign()) AliWarning("Can't misalign this cluster !");
1f3e997f 209
c219fd63 210 Int_t idx=c->GetDetectorIndex();
a504d56f 211 AliITSdetector &det=fgLayers[i].GetDetector(idx);
212
a504d56f 213 Double_t y=r*det.GetPhi()+c->GetY();
c219fd63 214 if (y>circ) y-=circ; else if (y<0) y+=circ;
215 c->SetPhiR(y);
216
00a7cc50 217 fgLayers[i].InsertCluster(new AliITSRecPoint(*c));
61ab8ea8 218 }
219 clusters->Delete();
220 }
18856a77 221 fgLayers[i].ResetRoad(); //road defined by the cluster density
006b5f7f 222 }
c630aafd 223
8676d691 224 return 0;
61ab8ea8 225}
006b5f7f 226
61ab8ea8 227void AliITStrackerV2::UnloadClusters() {
228 //--------------------------------------------------------------------
229 //This function unloads ITS clusters
230 //--------------------------------------------------------------------
e50912db 231 for (Int_t i=0; i<AliITSgeomTGeo::GetNLayers(); i++) fgLayers[i].ResetClusters();
006b5f7f 232}
233
880f41b9 234static Int_t CorrectForDeadZoneMaterial(AliITStrackV2 *t) {
235 //--------------------------------------------------------------------
236 // Correction for the material between the TPC and the ITS
237 // (should it belong to the TPC code ?)
238 //--------------------------------------------------------------------
239 Double_t riw=80., diw=0.0053, x0iw=30; // TPC inner wall ?
240 Double_t rcd=61., dcd=0.0053, x0cd=30; // TPC "central drum" ?
241 Double_t yr=12.8, dr=0.03; // rods ?
242 Double_t zm=0.2, dm=0.40; // membrane
243 //Double_t rr=52., dr=0.19, x0r=24., yyr=7.77; //rails
244 Double_t rs=50., ds=0.001; // something belonging to the ITS (screen ?)
245
246 if (t->GetX() > riw) {
247 if (!t->PropagateTo(riw,diw,x0iw)) return 1;
248 if (TMath::Abs(t->GetY())>yr) t->CorrectForMaterial(dr);
249 if (TMath::Abs(t->GetZ())<zm) t->CorrectForMaterial(dm);
250 if (!t->PropagateTo(rcd,dcd,x0cd)) return 1;
251 //Double_t x,y,z; t->GetGlobalXYZat(rr,x,y,z);
252 //if (TMath::Abs(y)<yyr) t->PropagateTo(rr,dr,x0r);
253 if (!t->PropagateTo(rs,ds)) return 1;
254 } else if (t->GetX() < rs) {
255 if (!t->PropagateTo(rs,-ds)) return 1;
256 //Double_t x,y,z; t->GetGlobalXYZat(rr,x,y,z);
257 //if (TMath::Abs(y)<yyr) t->PropagateTo(rr,-dr,x0r);
258 if (!t->PropagateTo(rcd,-dcd,x0cd)) return 1;
ba1a3e1a 259 if (!t->PropagateTo(riw+0.001,-diw,x0iw)) return 1;
880f41b9 260 } else {
261 ::Error("CorrectForDeadZoneMaterial","track is already in the dead zone !");
262 return 1;
263 }
264
265 return 0;
266}
267
af885e0f 268Int_t AliITStrackerV2::Clusters2Tracks(AliESDEvent *event) {
006b5f7f 269 //--------------------------------------------------------------------
c630aafd 270 // This functions reconstructs ITS tracks
271 // The clusters must be already loaded !
006b5f7f 272 //--------------------------------------------------------------------
c630aafd 273 TObjArray itsTracks(15000);
006b5f7f 274
c630aafd 275 {/* Read ESD tracks */
276 Int_t nentr=event->GetNumberOfTracks();
277 Info("Clusters2Tracks", "Number of ESD tracks: %d\n", nentr);
278 while (nentr--) {
279 AliESDtrack *esd=event->GetTrack(nentr);
280
ba1a3e1a 281 if ((esd->GetStatus()&AliESDtrack::kTPCin)==0) continue;
282 if (esd->GetStatus()&AliESDtrack::kTPCout) continue;
283 if (esd->GetStatus()&AliESDtrack::kITSin) continue;
c630aafd 284
763bdf3e 285 AliITStrackV2 *t = new AliITStrackV2(*esd);
286
6c94f330 287 if (TMath::Abs(t->GetD(GetX(),GetY()))>4) {
ebd8369c 288 delete t;
289 continue;
290 }
006b5f7f 291
c630aafd 292 if (CorrectForDeadZoneMaterial(t)!=0) {
293 Warning("Clusters2Tracks",
294 "failed to correct for the material in the dead zone !\n");
295 delete t;
296 continue;
297 }
298 itsTracks.AddLast(t);
299 }
300 } /* End Read ESD tracks */
301
302 itsTracks.Sort();
303 Int_t nentr=itsTracks.GetEntriesFast();
15dd636f 304
c630aafd 305 Int_t ntrk=0;
306 for (fPass=0; fPass<2; fPass++) {
307 Int_t &constraint=fConstraint[fPass]; if (constraint<0) continue;
308 for (Int_t i=0; i<nentr; i++) {
309 AliITStrackV2 *t=(AliITStrackV2*)itsTracks.UncheckedAt(i);
15dd636f 310 if (t==0) continue; //this track has been already tracked
c630aafd 311 Int_t tpcLabel=t->GetLabel(); //save the TPC track label
312
313 ResetTrackToFollow(*t);
314 ResetBestTrack();
315
e50912db 316 for (FollowProlongation(); fI<AliITSgeomTGeo::GetNLayers(); fI++) {
15dd636f 317 while (TakeNextProlongation()) FollowProlongation();
c630aafd 318 }
319
de5d3595 320 if (fBestTrack.GetNumberOfClusters() == 0) continue;
15dd636f 321
c630aafd 322 if (fConstraint[fPass]) {
15dd636f 323 ResetTrackToFollow(*t);
324 if (!RefitAt(3.7, &fTrackToFollow, &fBestTrack)) continue;
325 ResetBestTrack();
326 }
327
c630aafd 328 fBestTrack.SetLabel(tpcLabel);
329 fBestTrack.CookdEdx();
330 CookLabel(&fBestTrack,0.); //For comparison only
331 fBestTrack.UpdateESDtrack(AliESDtrack::kITSin);
15dd636f 332 UseClusters(&fBestTrack);
c630aafd 333 delete itsTracks.RemoveAt(i);
334 ntrk++;
335 }
336 }
babd135a 337
c630aafd 338 itsTracks.Delete();
2257f27e 339
c630aafd 340 Info("Clusters2Tracks","Number of prolonged tracks: %d\n",ntrk);
341
342 return 0;
343}
344
af885e0f 345Int_t AliITStrackerV2::PropagateBack(AliESDEvent *event) {
c630aafd 346 //--------------------------------------------------------------------
347 // This functions propagates reconstructed ITS tracks back
348 // The clusters must be loaded !
349 //--------------------------------------------------------------------
350 Int_t nentr=event->GetNumberOfTracks();
351 Info("PropagateBack", "Number of ESD tracks: %d\n", nentr);
352
353 Int_t ntrk=0;
354 for (Int_t i=0; i<nentr; i++) {
355 AliESDtrack *esd=event->GetTrack(i);
356
ba1a3e1a 357 if ((esd->GetStatus()&AliESDtrack::kITSin)==0) continue;
358 if (esd->GetStatus()&AliESDtrack::kITSout) continue;
c630aafd 359
763bdf3e 360 AliITStrackV2 *t = new AliITStrackV2(*esd);
c630aafd 361
362 ResetTrackToFollow(*t);
363
364 // propagete to vertex [SR, GSI 17.02.2003]
a9459f9c 365 // Start Time measurement [SR, GSI 17.02.2003], corrected by I.Belikov
366 if (fTrackToFollow.PropagateTo(3.,0.0028,65.19)) {
6c94f330 367 if (fTrackToFollow.PropagateToVertex(event->GetVertex())) {
a9459f9c 368 fTrackToFollow.StartTimeIntegral();
369 }
f9a4a033 370 Bool_t okProp=fTrackToFollow.PropagateTo(3.,-0.0028,65.19);
371 if(!okProp){
372 AliWarning("Propagation to beam pipe radius failed");
373 }
a9459f9c 374 }
c630aafd 375
6c94f330 376 fTrackToFollow.ResetCovariance(10.); fTrackToFollow.ResetClusters();
c630aafd 377 if (RefitAt(49.,&fTrackToFollow,t)) {
378 if (CorrectForDeadZoneMaterial(&fTrackToFollow)!=0) {
379 Warning("PropagateBack",
380 "failed to correct for the material in the dead zone !\n");
381 delete t;
382 continue;
383 }
384 fTrackToFollow.SetLabel(t->GetLabel());
ba1a3e1a 385 //fTrackToFollow.CookdEdx();
c630aafd 386 CookLabel(&fTrackToFollow,0.); //For comparison only
387 fTrackToFollow.UpdateESDtrack(AliESDtrack::kITSout);
15dd636f 388 UseClusters(&fTrackToFollow);
c630aafd 389 ntrk++;
390 }
391 delete t;
392 }
393
394 Info("PropagateBack","Number of back propagated ITS tracks: %d\n",ntrk);
61ab8ea8 395
83d73500 396 return 0;
397}
398
af885e0f 399Int_t AliITStrackerV2::RefitInward(AliESDEvent *event) {
83d73500 400 //--------------------------------------------------------------------
401 // This functions refits ITS tracks using the
402 // "inward propagated" TPC tracks
c630aafd 403 // The clusters must be loaded !
83d73500 404 //--------------------------------------------------------------------
c630aafd 405 Int_t nentr=event->GetNumberOfTracks();
406 Info("RefitInward", "Number of ESD tracks: %d\n", nentr);
407
408 Int_t ntrk=0;
409 for (Int_t i=0; i<nentr; i++) {
410 AliESDtrack *esd=event->GetTrack(i);
411
ba1a3e1a 412 if ((esd->GetStatus()&AliESDtrack::kITSout) == 0) continue;
413 if (esd->GetStatus()&AliESDtrack::kITSrefit) continue;
414 if (esd->GetStatus()&AliESDtrack::kTPCout)
15dd636f 415 if ((esd->GetStatus()&AliESDtrack::kTPCrefit)==0) continue;
c630aafd 416
763bdf3e 417 AliITStrackV2 *t = new AliITStrackV2(*esd);
c630aafd 418
419 if (CorrectForDeadZoneMaterial(t)!=0) {
420 Warning("RefitInward",
421 "failed to correct for the material in the dead zone !\n");
422 delete t;
423 continue;
424 }
425
426 ResetTrackToFollow(*t);
427 fTrackToFollow.ResetClusters();
428
429 //Refitting...
ef7253ac 430 if (RefitAt(3.7, &fTrackToFollow, t, kTRUE)) {
c630aafd 431 fTrackToFollow.SetLabel(t->GetLabel());
432 fTrackToFollow.CookdEdx();
15dd636f 433 CookLabel(&fTrackToFollow,0.); //For comparison only
18856a77 434
49d13e89 435 if (fTrackToFollow.PropagateTo(3.,0.0028,65.19)) {//The beam pipe
09cf9d66 436 fTrackToFollow.UpdateESDtrack(AliESDtrack::kITSrefit);
49d13e89 437 AliESDtrack *esdTrack =fTrackToFollow.GetESDtrack();
f7a1cc68 438 Double_t r[3]={0.,0.,0.};
49d13e89 439 Double_t maxD=3.;
440 esdTrack->RelateToVertex(event->GetVertex(),GetBz(r),maxD);
441 ntrk++;
67c3dcbe 442 }
c630aafd 443 }
444 delete t;
445 }
446
447 Info("RefitInward","Number of refitted tracks: %d\n",ntrk);
83d73500 448
c630aafd 449 return 0;
450}
83d73500 451
006b5f7f 452AliCluster *AliITStrackerV2::GetCluster(Int_t index) const {
453 //--------------------------------------------------------------------
454 // Return pointer to a given cluster
455 //--------------------------------------------------------------------
456 Int_t l=(index & 0xf0000000) >> 28;
457 Int_t c=(index & 0x0fffffff) >> 00;
18856a77 458 return fgLayers[l].GetCluster(c);
006b5f7f 459}
460
461
462void AliITStrackerV2::FollowProlongation() {
463 //--------------------------------------------------------------------
464 //This function finds a track prolongation
465 //--------------------------------------------------------------------
c0dd5278 466 while (fI>fLastLayerToTrackTo) {
a9a2d814 467 Int_t i=fI-1;
8676d691 468
18856a77 469 AliITSlayer &layer=fgLayers[i];
a9a2d814 470 AliITStrackV2 &track=fTracks[i];
006b5f7f 471
14825d5a 472 Double_t r=layer.GetR();
61ab8ea8 473
a9a2d814 474 if (i==3 || i==1) {
18856a77 475 Double_t rs=0.5*(fgLayers[i+1].GetR() + r);
61ab8ea8 476 Double_t d=0.0034, x0=38.6;
477 if (i==1) {rs=9.; d=0.0097; x0=42;}
478 if (!fTrackToFollow.PropagateTo(rs,d,x0)) {
8676d691 479 //Warning("FollowProlongation","propagation failed !\n");
c0dd5278 480 return;
a9a2d814 481 }
006b5f7f 482 }
483
484 //find intersection
8602c008 485 Double_t phi,z;
486 if (!fTrackToFollow.GetPhiZat(r,phi,z)) {
8676d691 487 //Warning("FollowProlongation","failed to estimate track !\n");
c0dd5278 488 return;
006b5f7f 489 }
f363d377 490
006b5f7f 491 Int_t idet=layer.FindDetectorIndex(phi,z);
492 if (idet<0) {
8676d691 493 //Warning("FollowProlongation","failed to find a detector !\n");
c0dd5278 494 return;
006b5f7f 495 }
496
497 //propagate to the intersection
498 const AliITSdetector &det=layer.GetDetector(idet);
14825d5a 499 phi=det.GetPhi();
a9a2d814 500 if (!fTrackToFollow.Propagate(phi,det.GetR())) {
8676d691 501 //Warning("FollowProlongation","propagation failed !\n");
c0dd5278 502 return;
006b5f7f 503 }
504 fTrackToFollow.SetDetectorIndex(idet);
505
506 //Select possible prolongations and store the current track estimation
507 track.~AliITStrackV2(); new(&track) AliITStrackV2(fTrackToFollow);
44347160 508 Double_t dz=7*TMath::Sqrt(track.GetSigmaZ2() + AliITSReconstructor::GetRecoParam()->GetSigmaZ2(i));
509 Double_t dy=7*TMath::Sqrt(track.GetSigmaY2() + AliITSReconstructor::GetRecoParam()->GetSigmaY2(i));
8676d691 510 Double_t road=layer.GetRoad();
511 if (dz*dy>road*road) {
512 Double_t dd=TMath::Sqrt(dz*dy), scz=dz/dd, scy=dy/dd;
513 dz=road*scz; dy=road*scy;
514 }
b87bd7cd 515
44347160 516 //Double_t dz=4*TMath::Sqrt(track.GetSigmaZ2() + AliITSReconstructor::GetRecoParam()->GetSigmaZ2(i));
14825d5a 517 if (dz < 0.5*TMath::Abs(track.GetTgl())) dz=0.5*TMath::Abs(track.GetTgl());
44347160 518 if (dz > AliITSReconstructor::GetRecoParam()->GetMaxRoad()) {
8676d691 519 //Warning("FollowProlongation","too broad road in Z !\n");
c0dd5278 520 return;
006b5f7f 521 }
a9a2d814 522
15dd636f 523 if (TMath::Abs(fTrackToFollow.GetZ()-GetZ()) > r+dz) return;
a9a2d814 524
44347160 525 //Double_t dy=4*TMath::Sqrt(track.GetSigmaY2() + AliITSReconstructor::GetRecoParam()->GetSigmaY2(i));
14825d5a 526 if (dy < 0.5*TMath::Abs(track.GetSnp())) dy=0.5*TMath::Abs(track.GetSnp());
44347160 527 if (dy > AliITSReconstructor::GetRecoParam()->GetMaxRoad()) {
8676d691 528 //Warning("FollowProlongation","too broad road in Y !\n");
c0dd5278 529 return;
006b5f7f 530 }
a9a2d814 531
c219fd63 532 fI--;
533
7f6ddf58 534 Double_t zmin=track.GetZ() - dz;
006b5f7f 535 Double_t zmax=track.GetZ() + dz;
14825d5a 536 Double_t ymin=track.GetY() + r*phi - dy;
537 Double_t ymax=track.GetY() + r*phi + dy;
c219fd63 538 if (layer.SelectClusters(zmin,zmax,ymin,ymax)==0)
539 if (fLayersNotToSkip[fI]) return;
006b5f7f 540
15dd636f 541 if (!TakeNextProlongation())
542 if (fLayersNotToSkip[fI]) return;
543
006b5f7f 544 }
545
546 //deal with the best track
547 Int_t ncl=fTrackToFollow.GetNumberOfClusters();
548 Int_t nclb=fBestTrack.GetNumberOfClusters();
15dd636f 549 if (ncl)
550 if (ncl >= nclb) {
551 Double_t chi2=fTrackToFollow.GetChi2();
44347160 552 if (chi2/ncl < AliITSReconstructor::GetRecoParam()->GetChi2PerCluster()) {
15dd636f 553 if (ncl > nclb || chi2 < fBestTrack.GetChi2()) {
554 ResetBestTrack();
006b5f7f 555 }
15dd636f 556 }
006b5f7f 557 }
15dd636f 558
006b5f7f 559}
560
006b5f7f 561Int_t AliITStrackerV2::TakeNextProlongation() {
562 //--------------------------------------------------------------------
a9a2d814 563 // This function takes another track prolongation
564 //
565 // dEdx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
006b5f7f 566 //--------------------------------------------------------------------
18856a77 567 AliITSlayer &layer=fgLayers[fI];
4ab260d6 568 ResetTrackToFollow(fTracks[fI]);
006b5f7f 569
44347160 570 Double_t dz=7*TMath::Sqrt(fTrackToFollow.GetSigmaZ2() + AliITSReconstructor::GetRecoParam()->GetSigmaZ2(fI));
571 Double_t dy=7*TMath::Sqrt(fTrackToFollow.GetSigmaY2() + AliITSReconstructor::GetRecoParam()->GetSigmaY2(fI));
8676d691 572 Double_t road=layer.GetRoad();
573 if (dz*dy>road*road) {
574 Double_t dd=TMath::Sqrt(dz*dy), scz=dz/dd, scy=dy/dd;
575 dz=road*scz; dy=road*scy;
576 }
006b5f7f 577
00a7cc50 578 const AliITSRecPoint *c=0; Int_t ci=-1;
579 const AliITSRecPoint *cc=0; Int_t cci=-1;
44347160 580 Double_t chi2=AliITSReconstructor::GetRecoParam()->GetMaxChi2();
006b5f7f 581 while ((c=layer.GetNextCluster(ci))!=0) {
006b5f7f 582 Int_t idet=c->GetDetectorIndex();
15dd636f 583
4ab260d6 584 if (fTrackToFollow.GetDetectorIndex()!=idet) {
006b5f7f 585 const AliITSdetector &det=layer.GetDetector(idet);
4ab260d6 586 ResetTrackToFollow(fTracks[fI]);
587 if (!fTrackToFollow.Propagate(det.GetPhi(),det.GetR())) {
8676d691 588 //Warning("TakeNextProlongation","propagation failed !\n");
006b5f7f 589 continue;
590 }
4ab260d6 591 fTrackToFollow.SetDetectorIndex(idet);
592 if (TMath::Abs(fTrackToFollow.GetZ()-GetZ())>layer.GetR()+dz) continue;
006b5f7f 593 }
594
4ab260d6 595 if (TMath::Abs(fTrackToFollow.GetZ() - c->GetZ()) > dz) continue;
596 if (TMath::Abs(fTrackToFollow.GetY() - c->GetY()) > dy) continue;
006b5f7f 597
c219fd63 598 Double_t ch2=fTrackToFollow.GetPredictedChi2(c);
599 if (ch2 > chi2) continue;
600 chi2=ch2;
601 cc=c; cci=ci;
602 break;
006b5f7f 603 }
604
c219fd63 605 if (!cc) return 0;
006b5f7f 606
1f3e997f 607 {// Take into account the mis-alignment
608 Double_t x = fTrackToFollow.GetX() + cc->GetX();
609 if (!fTrackToFollow.PropagateTo(x,0.,0.)) return 0;
610 }
c219fd63 611 if (!fTrackToFollow.Update(cc,chi2,(fI<<28)+cci)) {
8676d691 612 //Warning("TakeNextProlongation","filtering failed !\n");
006b5f7f 613 return 0;
614 }
15dd636f 615
4ab260d6 616 if (fTrackToFollow.GetNumberOfClusters()>1)
6c94f330 617 if (TMath::Abs(fTrackToFollow.GetD(GetX(),GetY()))>4) return 0;
4ab260d6 618
a9a2d814 619 fTrackToFollow.
09cf9d66 620 SetSampledEdx(cc->GetQ(),fI-2); //b.b.
a9a2d814 621
622 {
880f41b9 623 Double_t x0;
61ab8ea8 624 Double_t d=layer.GetThickness(fTrackToFollow.GetY(),fTrackToFollow.GetZ(),x0);
880f41b9 625 fTrackToFollow.CorrectForMaterial(d,x0);
a9a2d814 626 }
4ab260d6 627
a9a2d814 628 if (fConstraint[fPass]) {
4ab260d6 629 Double_t d=GetEffectiveThickness(0,0); //Think of this !!!!
8676d691 630 Double_t xyz[]={GetX(),GetY(),GetZ()};
631 Double_t ers[]={GetSigmaX(),GetSigmaY(),GetSigmaZ()};
632 fTrackToFollow.Improve(d,xyz,ers);
a9a2d814 633 }
006b5f7f 634
006b5f7f 635 return 1;
636}
637
638
6c94f330 639AliITStrackerV2::AliITSlayer::AliITSlayer():
640 fR(0.),
641 fPhiOffset(0.),
642 fNladders(0),
643 fZOffset(0.),
644 fNdetectors(0),
645 fDetectors(0),
646 fNsel(0),
647 fRoad(2*fR*TMath::Sqrt(3.14/1.)) //assuming that there's only one cluster
648{
006b5f7f 649 //--------------------------------------------------------------------
650 //default AliITSlayer constructor
651 //--------------------------------------------------------------------
c219fd63 652
653 for (Int_t i=0; i<kNsector; i++) fN[i]=0;
4edc3c8c 654 for (Int_t i=0; i<AliITSRecoParam::kMaxClusterPerLayer; i++){
1e788092 655 fClusters[i]=0;
656 fIndex[i]=0;
657 }
006b5f7f 658}
659
660AliITStrackerV2::AliITSlayer::
6c94f330 661AliITSlayer(Double_t r,Double_t p,Double_t z,Int_t nl,Int_t nd):
662 fR(r),
663 fPhiOffset(p),
664 fNladders(nl),
665 fZOffset(z),
666 fNdetectors(nd),
667 fDetectors(new AliITSdetector[nl*nd]),
668 fNsel(0),
669 fRoad(2*r*TMath::Sqrt(3.14/1.)) //assuming that there's only one cluster
670{
006b5f7f 671 //--------------------------------------------------------------------
672 //main AliITSlayer constructor
673 //--------------------------------------------------------------------
006b5f7f 674
c219fd63 675 for (Int_t i=0; i<kNsector; i++) fN[i]=0;
c219fd63 676
4edc3c8c 677 for (Int_t i=0; i<AliITSRecoParam::kMaxClusterPerLayer; i++){
1e788092 678 fClusters[i]=0;
679 fIndex[i]=0;
680 }
006b5f7f 681}
682
683AliITStrackerV2::AliITSlayer::~AliITSlayer() {
684 //--------------------------------------------------------------------
685 // AliITSlayer destructor
686 //--------------------------------------------------------------------
687 delete[] fDetectors;
c219fd63 688 ResetClusters();
006b5f7f 689}
690
61ab8ea8 691void AliITStrackerV2::AliITSlayer::ResetClusters() {
692 //--------------------------------------------------------------------
693 // This function removes loaded clusters
694 //--------------------------------------------------------------------
c219fd63 695 for (Int_t s=0; s<kNsector; s++) {
696 Int_t &n=fN[s];
697 while (n) {
698 n--;
699 delete fClusters[s*kMaxClusterPerSector+n];
700 }
701 }
61ab8ea8 702}
703
b87bd7cd 704void AliITStrackerV2::AliITSlayer::ResetRoad() {
705 //--------------------------------------------------------------------
706 // This function calculates the road defined by the cluster density
707 //--------------------------------------------------------------------
708 Int_t n=0;
c219fd63 709 for (Int_t s=0; s<kNsector; s++) {
710 Int_t i=fN[s];
711 while (i--)
712 if (TMath::Abs(fClusters[s*kMaxClusterPerSector+i]->GetZ())<fR) n++;
b87bd7cd 713 }
714 if (n>1) fRoad=2*fR*TMath::Sqrt(3.14/n);
715}
716
00a7cc50 717Int_t AliITStrackerV2::AliITSlayer::InsertCluster(AliITSRecPoint *c) {
006b5f7f 718 //--------------------------------------------------------------------
c219fd63 719 // This function inserts a cluster to this layer in increasing
720 // order of the cluster's fZ
006b5f7f 721 //--------------------------------------------------------------------
c219fd63 722 Float_t circ=TMath::TwoPi()*fR;
723 Int_t sec=Int_t(kNsector*c->GetPhiR()/circ);
81e97e0d 724 if (sec>=kNsector) {
725 ::Error("InsertCluster","Wrong sector !\n");
726 return 1;
727 }
c219fd63 728 Int_t &n=fN[sec];
729 if (n>=kMaxClusterPerSector) {
730 ::Error("InsertCluster","Too many clusters !\n");
731 return 1;
006b5f7f 732 }
c219fd63 733 if (n==0) fClusters[sec*kMaxClusterPerSector]=c;
734 else {
735 Int_t i=FindClusterIndex(c->GetZ(),sec);
736 Int_t k=n-i+sec*kMaxClusterPerSector;
00a7cc50 737 memmove(fClusters+i+1 ,fClusters+i,k*sizeof(AliITSRecPoint*));
c219fd63 738 fClusters[i]=c;
739 }
740 n++;
006b5f7f 741 return 0;
742}
743
c219fd63 744Int_t
745AliITStrackerV2::AliITSlayer::FindClusterIndex(Float_t z,Int_t s) const {
006b5f7f 746 //--------------------------------------------------------------------
c219fd63 747 // For the sector "s", this function returns the index of the first
748 // with its fZ >= "z".
006b5f7f 749 //--------------------------------------------------------------------
c219fd63 750 Int_t nc=fN[s];
751 if (nc==0) return kMaxClusterPerSector*s;
752
753 Int_t b=kMaxClusterPerSector*s;
754 if (z <= fClusters[b]->GetZ()) return b;
755
756 Int_t e=b+nc-1;
757 if (z > fClusters[e]->GetZ()) return e+1;
758
759 Int_t m=(b+e)/2;
006b5f7f 760 for (; b<e; m=(b+e)/2) {
761 if (z > fClusters[m]->GetZ()) b=m+1;
762 else e=m;
763 }
764 return m;
765}
766
c219fd63 767Int_t AliITStrackerV2::AliITSlayer::
768SelectClusters(Float_t zmin,Float_t zmax,Float_t ymin, Float_t ymax) {
006b5f7f 769 //--------------------------------------------------------------------
c219fd63 770 // This function selects clusters within the "window"
006b5f7f 771 //--------------------------------------------------------------------
c219fd63 772 Float_t circ=fR*TMath::TwoPi();
773
774 if (ymin>circ) ymin-=circ; else if (ymin<0) ymin+=circ;
775 if (ymax>circ) ymax-=circ; else if (ymax<0) ymax+=circ;
776
521f4796 777 Int_t i1=Int_t(kNsector*ymin/circ); if (i1==kNsector) i1--;
c219fd63 778 if (fN[i1]!=0) {
779 Float_t ym = (ymax<ymin) ? ymax+circ : ymax;
780 Int_t i=FindClusterIndex(zmin,i1), imax=i1*kMaxClusterPerSector+fN[i1];
781 for (; i<imax; i++) {
00a7cc50 782 AliITSRecPoint *c=fClusters[i];
c219fd63 783 if (c->IsUsed()) continue;
784 if (c->GetZ()>zmax) break;
785 if (c->GetPhiR()<=ymin) continue;
786 if (c->GetPhiR()>ym) continue;
787 fIndex[fNsel++]=i;
788 }
789 }
790
521f4796 791 Int_t i2=Int_t(kNsector*ymax/circ); if (i2==kNsector) i2--;
c219fd63 792 if (i2==i1) return fNsel;
793
794 if (fN[i2]!=0) {
795 Float_t ym = (ymin>ymax) ? ymin-circ : ymin;
796 Int_t i=FindClusterIndex(zmin,i2), imax=i2*kMaxClusterPerSector+fN[i2];
797 for (; i<imax; i++) {
00a7cc50 798 AliITSRecPoint *c=fClusters[i];
c219fd63 799 if (c->IsUsed()) continue;
800 if (c->GetZ()>zmax) break;
801 if (c->GetPhiR()<=ym) continue;
802 if (c->GetPhiR()>ymax) continue;
803 fIndex[fNsel++]=i;
804 }
805 }
806
807 return fNsel;
006b5f7f 808}
809
00a7cc50 810const AliITSRecPoint *AliITStrackerV2::AliITSlayer::GetNextCluster(Int_t &ci){
006b5f7f 811 //--------------------------------------------------------------------
812 // This function returns clusters within the "window"
813 //--------------------------------------------------------------------
00a7cc50 814 AliITSRecPoint *c=0;
c219fd63 815 ci=-1;
816 if (fNsel) {
817 fNsel--;
818 ci=fIndex[fNsel];
819 c=fClusters[ci];
006b5f7f 820 }
c219fd63 821 return c;
822}
7f6ddf58 823
c219fd63 824Int_t AliITStrackerV2::AliITSlayer::GetNumberOfClusters() const {
825 Int_t n=0;
826 for (Int_t s=0; s<kNsector; s++) n+=fN[s];
827 return n;
006b5f7f 828}
829
c219fd63 830Int_t
831AliITStrackerV2::AliITSlayer::FindDetectorIndex(Double_t phi,Double_t z)const {
006b5f7f 832 //--------------------------------------------------------------------
833 //This function finds the detector crossed by the track
834 //--------------------------------------------------------------------
31fb8575 835 Double_t dphi;
836 if (fZOffset<0) // old geometry
837 dphi = -(phi-fPhiOffset);
838 else // new geometry
839 dphi = phi-fPhiOffset;
840
006b5f7f 841 if (dphi < 0) dphi += 2*TMath::Pi();
842 else if (dphi >= 2*TMath::Pi()) dphi -= 2*TMath::Pi();
843 Int_t np=Int_t(dphi*fNladders*0.5/TMath::Pi()+0.5);
3e3e8427 844 if (np>=fNladders) np-=fNladders;
845 if (np<0) np+=fNladders;
006b5f7f 846
847 Double_t dz=fZOffset-z;
848 Int_t nz=Int_t(dz*(fNdetectors-1)*0.5/fZOffset+0.5);
3e3e8427 849 if (nz>=fNdetectors) return -1;
850 if (nz<0) return -1;
006b5f7f 851
006b5f7f 852 return np*fNdetectors + nz;
853}
854
855Double_t
61ab8ea8 856AliITStrackerV2::AliITSlayer::GetThickness(Double_t y,Double_t z,Double_t &x0)
857const {
a9a2d814 858 //--------------------------------------------------------------------
859 //This function returns the layer thickness at this point (units X0)
860 //--------------------------------------------------------------------
861 Double_t d=0.0085;
61ab8ea8 862 x0=21.82;
a9a2d814 863
864 if (43<fR&&fR<45) { //SSD2
61ab8ea8 865 Double_t dd=0.0034;
866 d=dd;
867 if (TMath::Abs(y-0.00)>3.40) d+=dd;
868 if (TMath::Abs(y-1.90)<0.45) {d+=(0.013-0.0034);}
869 if (TMath::Abs(y+1.90)<0.45) {d+=(0.013-0.0034);}
a9a2d814 870 for (Int_t i=0; i<12; i++) {
61ab8ea8 871 if (TMath::Abs(z-3.9*(i+0.5))<0.15) {
872 if (TMath::Abs(y-0.00)>3.40) d+=dd;
873 d+=0.0034;
874 break;
875 }
876 if (TMath::Abs(z+3.9*(i+0.5))<0.15) {
877 if (TMath::Abs(y-0.00)>3.40) d+=dd;
878 d+=0.0034;
879 break;
880 }
881 if (TMath::Abs(z-3.4-3.9*i)<0.50) {d+=(0.016-0.0034); break;}
882 if (TMath::Abs(z+0.5+3.9*i)<0.50) {d+=(0.016-0.0034); break;}
a9a2d814 883 }
884 } else
885 if (37<fR&&fR<41) { //SSD1
61ab8ea8 886 Double_t dd=0.0034;
887 d=dd;
888 if (TMath::Abs(y-0.00)>3.40) d+=dd;
889 if (TMath::Abs(y-1.90)<0.45) {d+=(0.013-0.0034);}
890 if (TMath::Abs(y+1.90)<0.45) {d+=(0.013-0.0034);}
a9a2d814 891 for (Int_t i=0; i<11; i++) {
61ab8ea8 892 if (TMath::Abs(z-3.9*i)<0.15) {
893 if (TMath::Abs(y-0.00)>3.40) d+=dd;
894 d+=dd;
895 break;
896 }
897 if (TMath::Abs(z+3.9*i)<0.15) {
898 if (TMath::Abs(y-0.00)>3.40) d+=dd;
899 d+=dd;
900 break;
901 }
902 if (TMath::Abs(z-1.85-3.9*i)<0.50) {d+=(0.016-0.0034); break;}
903 if (TMath::Abs(z+2.05+3.9*i)<0.50) {d+=(0.016-0.0034); break;}
a9a2d814 904 }
905 } else
906 if (13<fR&&fR<26) { //SDD
61ab8ea8 907 Double_t dd=0.0033;
908 d=dd;
909 if (TMath::Abs(y-0.00)>3.30) d+=dd;
910
911 if (TMath::Abs(y-1.80)<0.55) {
912 d+=0.016;
913 for (Int_t j=0; j<20; j++) {
914 if (TMath::Abs(z+0.7+1.47*j)<0.12) {d+=0.08; x0=9.; break;}
915 if (TMath::Abs(z-0.7-1.47*j)<0.12) {d+=0.08; x0=9.; break;}
916 }
917 }
918 if (TMath::Abs(y+1.80)<0.55) {
919 d+=0.016;
920 for (Int_t j=0; j<20; j++) {
921 if (TMath::Abs(z-0.7-1.47*j)<0.12) {d+=0.08; x0=9.; break;}
922 if (TMath::Abs(z+0.7+1.47*j)<0.12) {d+=0.08; x0=9.; break;}
923 }
924 }
925
926 for (Int_t i=0; i<4; i++) {
927 if (TMath::Abs(z-7.3*i)<0.60) {
928 d+=dd;
929 if (TMath::Abs(y-0.00)>3.30) d+=dd;
930 break;
931 }
932 if (TMath::Abs(z+7.3*i)<0.60) {
933 d+=dd;
934 if (TMath::Abs(y-0.00)>3.30) d+=dd;
935 break;
936 }
a9a2d814 937 }
938 } else
939 if (6<fR&&fR<8) { //SPD2
61ab8ea8 940 Double_t dd=0.0063; x0=21.5;
941 d=dd;
942 if (TMath::Abs(y-3.08)>0.5) d+=dd;
943 //if (TMath::Abs(y-3.08)>0.45) d+=dd;
944 if (TMath::Abs(y-3.03)<0.10) {d+=0.014;}
a9a2d814 945 } else
946 if (3<fR&&fR<5) { //SPD1
61ab8ea8 947 Double_t dd=0.0063; x0=21.5;
948 d=dd;
949 if (TMath::Abs(y+0.21)>0.6) d+=dd;
950 //if (TMath::Abs(y+0.21)>0.45) d+=dd;
951 if (TMath::Abs(y+0.10)<0.10) {d+=0.014;}
a9a2d814 952 }
953
a9a2d814 954 return d;
955}
006b5f7f 956
a9a2d814 957Double_t AliITStrackerV2::GetEffectiveThickness(Double_t y,Double_t z) const
006b5f7f 958{
959 //--------------------------------------------------------------------
a9a2d814 960 //Returns the thickness between the current layer and the vertex (units X0)
006b5f7f 961 //--------------------------------------------------------------------
61ab8ea8 962 Double_t d=0.0028*3*3; //beam pipe
963 Double_t x0=0;
006b5f7f 964
18856a77 965 Double_t xn=fgLayers[fI].GetR();
006b5f7f 966 for (Int_t i=0; i<fI; i++) {
18856a77 967 Double_t xi=fgLayers[i].GetR();
968 d+=fgLayers[i].GetThickness(y,z,x0)*xi*xi;
006b5f7f 969 }
970
971 if (fI>1) {
61ab8ea8 972 Double_t xi=9.;
973 d+=0.0097*xi*xi;
006b5f7f 974 }
975
976 if (fI>3) {
18856a77 977 Double_t xi=0.5*(fgLayers[3].GetR()+fgLayers[4].GetR());
61ab8ea8 978 d+=0.0034*xi*xi;
006b5f7f 979 }
61ab8ea8 980
006b5f7f 981 return d/(xn*xn);
982}
983
ef7253ac 984Bool_t AliITStrackerV2::RefitAt(Double_t xx,AliITStrackV2 *t,
985 const AliITStrackV2 *c, Bool_t extra) {
4ab260d6 986 //--------------------------------------------------------------------
c630aafd 987 // This function refits the track "t" at the position "x" using
988 // the clusters from "c"
ef7253ac 989 // If "extra"==kTRUE,
990 // the clusters from overlapped modules get attached to "t"
4ab260d6 991 //--------------------------------------------------------------------
e50912db 992 Int_t index[AliITSgeomTGeo::kNLayers];
880f41b9 993 Int_t k;
e50912db 994 for (k=0; k<AliITSgeomTGeo::GetNLayers(); k++) index[k]=-1;
c630aafd 995 Int_t nc=c->GetNumberOfClusters();
880f41b9 996 for (k=0; k<nc; k++) {
c630aafd 997 Int_t idx=c->GetClusterIndex(k),nl=(idx&0xf0000000)>>28;
880f41b9 998 index[nl]=idx;
999 }
880f41b9 1000
4ab260d6 1001 Int_t from, to, step;
ee34184f 1002 if (xx > t->GetX()) {
e50912db 1003 from=0; to=AliITSgeomTGeo::GetNLayers();
4ab260d6 1004 step=+1;
1005 } else {
e50912db 1006 from=AliITSgeomTGeo::GetNLayers()-1; to=-1;
4ab260d6 1007 step=-1;
1008 }
1009
1010 for (Int_t i=from; i != to; i += step) {
18856a77 1011 AliITSlayer &layer=fgLayers[i];
4ab260d6 1012 Double_t r=layer.GetR();
1013
1014 {
1015 Double_t hI=i-0.5*step;
225cff59 1016 if (TMath::Abs(hI-1.5)<0.01 || TMath::Abs(hI-3.5)<0.01) {
1017 Int_t iLay = i-step;
1018 Double_t rs = 0.;
1019 if(iLay<0 || iLay>= AliITSgeomTGeo::kNLayers){
225cff59 1020 AliError(Form("Invalid layer %d ",iLay));
1021 return kFALSE;
1022 }
1e788092 1023 else{
1024 rs=0.5*(fgLayers[i-step].GetR() + r);
1025 }
225cff59 1026 Double_t d=0.0034, x0=38.6;
1027 if (TMath::Abs(hI-1.5)<0.01) {rs=9.; d=0.0097; x0=42;}
1028 if (!t->PropagateTo(rs,-step*d,x0)) {
1029 return kFALSE;
1030 }
4ab260d6 1031 }
1032 }
1033
880f41b9 1034 // remember old position [SR, GSI 18.02.2003]
1035 Double_t oldX=0., oldY=0., oldZ=0.;
1036 if (t->IsStartedTimeIntegral() && step==1) {
1037 t->GetGlobalXYZat(t->GetX(),oldX,oldY,oldZ);
1038 }
1039 //
1040
8602c008 1041 Double_t phi,z;
1042 if (!t->GetPhiZat(r,phi,z)) {
4ab260d6 1043 return kFALSE;
1044 }
8602c008 1045
4ab260d6 1046 Int_t idet=layer.FindDetectorIndex(phi,z);
1047 if (idet<0) {
1048 return kFALSE;
1049 }
1050 const AliITSdetector &det=layer.GetDetector(idet);
1051 phi=det.GetPhi();
1052 if (!t->Propagate(phi,det.GetR())) {
1053 return kFALSE;
1054 }
1055 t->SetDetectorIndex(idet);
1056
00a7cc50 1057 const AliITSRecPoint *cl=0;
44347160 1058 Double_t maxchi2=AliITSReconstructor::GetRecoParam()->GetMaxChi2();
4ab260d6 1059
1060 Int_t idx=index[i];
c0342e67 1061 if (idx>=0) {
a5d0fae6 1062 const AliITSRecPoint *ccc=(AliITSRecPoint *)GetCluster(idx);
1063 if (idet != ccc->GetDetectorIndex()) {
1064 idet=ccc->GetDetectorIndex();
1065 const AliITSdetector &det2=layer.GetDetector(idet);
1066 if (!t->Propagate(det2.GetPhi(),det2.GetR())) {
4ab260d6 1067 return kFALSE;
1068 }
1069 t->SetDetectorIndex(idet);
1070 }
a5d0fae6 1071 Double_t chi2=t->GetPredictedChi2(ccc);
c630aafd 1072 if (chi2<maxchi2) {
a5d0fae6 1073 cl=ccc;
c630aafd 1074 maxchi2=chi2;
1075 } else {
1076 return kFALSE;
1077 }
4ab260d6 1078 }
ef7253ac 1079
4ab260d6 1080 if (cl) {
1f3e997f 1081 // Take into account the mis-alignment
1082 Double_t x=t->GetX()+cl->GetX();
1083 if (!t->PropagateTo(x,0.,0.)) return kFALSE;
4ab260d6 1084 if (!t->Update(cl,maxchi2,idx)) {
1085 return kFALSE;
1086 }
09cf9d66 1087 t->SetSampledEdx(cl->GetQ(),i-2);
4ab260d6 1088 }
1089
1090 {
61ab8ea8 1091 Double_t x0;
1092 Double_t d=layer.GetThickness(t->GetY(),t->GetZ(),x0);
1093 t->CorrectForMaterial(-step*d,x0);
4ab260d6 1094 }
880f41b9 1095
ef7253ac 1096 if (extra) { //search for extra clusters
1097 AliITStrackV2 tmp(*t);
44347160 1098 Double_t dz=4*TMath::Sqrt(tmp.GetSigmaZ2()+AliITSReconstructor::GetRecoParam()->GetSigmaZ2(i));
ef7253ac 1099 if (dz < 0.5*TMath::Abs(tmp.GetTgl())) dz=0.5*TMath::Abs(tmp.GetTgl());
44347160 1100 Double_t dy=4*TMath::Sqrt(t->GetSigmaY2()+AliITSReconstructor::GetRecoParam()->GetSigmaY2(i));
ef7253ac 1101 if (dy < 0.5*TMath::Abs(tmp.GetSnp())) dy=0.5*TMath::Abs(tmp.GetSnp());
1102 Double_t zmin=t->GetZ() - dz;
1103 Double_t zmax=t->GetZ() + dz;
1104 Double_t ymin=t->GetY() + phi*r - dy;
1105 Double_t ymax=t->GetY() + phi*r + dy;
1106 layer.SelectClusters(zmin,zmax,ymin,ymax);
1107
a5d0fae6 1108 const AliITSRecPoint *cx=0; Int_t ci=-1,cci=-1;
1109 maxchi2=1000.*AliITSReconstructor::GetRecoParam()->GetMaxChi2();
1110 Double_t tolerance=0.1;
1111 while ((cx=layer.GetNextCluster(ci))!=0) {
1112 if (idet == cx->GetDetectorIndex()) continue;
ef7253ac 1113
a5d0fae6 1114 const AliITSdetector &detx=layer.GetDetector(cx->GetDetectorIndex());
ef7253ac 1115
a5d0fae6 1116 if (!tmp.Propagate(detx.GetPhi(),detx.GetR())) continue;
ef7253ac 1117
a5d0fae6 1118 if (TMath::Abs(tmp.GetZ() - cx->GetZ()) > tolerance) continue;
1119 if (TMath::Abs(tmp.GetY() - cx->GetY()) > tolerance) continue;
ef7253ac 1120
a5d0fae6 1121 Double_t chi2=tmp.GetPredictedChi2(cx);
ef7253ac 1122 if (chi2<maxchi2) { maxchi2=chi2; cci=ci; }
1123 }
1124 if (cci>=0) t->SetExtraCluster(i,(i<<28)+cci);
1125 }
1126
880f41b9 1127 // track time update [SR, GSI 17.02.2003]
1128 if (t->IsStartedTimeIntegral() && step==1) {
1129 Double_t newX, newY, newZ;
1130 t->GetGlobalXYZat(t->GetX(),newX,newY,newZ);
1131 Double_t dL2 = (oldX-newX)*(oldX-newX) + (oldY-newY)*(oldY-newY) +
1132 (oldZ-newZ)*(oldZ-newZ);
1133 t->AddTimeStep(TMath::Sqrt(dL2));
1134 }
1135 //
4ab260d6 1136
1137 }
1138
ee34184f 1139 if (!t->PropagateTo(xx,0.,0.)) return kFALSE;
4ab260d6 1140 return kTRUE;
1141}
1142
b87bd7cd 1143void AliITStrackerV2::UseClusters(const AliKalmanTrack *t, Int_t from) const {
1144 //--------------------------------------------------------------------
1145 // This function marks clusters assigned to the track
1146 //--------------------------------------------------------------------
1147 AliTracker::UseClusters(t,from);
880f41b9 1148
5482d29d 1149 Int_t clusterIndex = t->GetClusterIndex(0);
1150 AliITSRecPoint *c= 0x0;
1151
1152 if (clusterIndex>-1)
1153 c = (AliITSRecPoint *)GetCluster(clusterIndex);
d920bdee 1154 if (c && c->GetSigmaZ2()>0.1) c->UnUse();
5482d29d 1155
1156 c = 0x0;
1157 clusterIndex = t->GetClusterIndex(1);
1158 if (clusterIndex>-1)
1159 c=(AliITSRecPoint *)GetCluster(clusterIndex);
d920bdee 1160 if (c && c->GetSigmaZ2()>0.1) c->UnUse();
880f41b9 1161
b87bd7cd 1162}