Fixed a bug in the Digit reader, moved clusterfinder to initialize from OCDB, Added...
[u/mrichter/AliRoot.git] / ITS / AliITStrackerMI.cxx
CommitLineData
e43c066c 1/**************************************************************************
9a5326b1 2 * Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. *
e43c066c 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 **************************************************************************/
fddf8459 15/* $Id$ */
bf6adc12 16
e43c066c 17//-------------------------------------------------------------------------
18// Implementation of the ITS tracker class
00a7cc50 19// It reads AliITSRecPoint clusters and creates AliITStrackMI tracks
e43c066c 20// and fills with them the ESD
15dd636f 21// Origin: Marian Ivanov, CERN, Marian.Ivanov@cern.ch
23197852 22// Current support and development:
23// Andrea Dainese, andrea.dainese@lnl.infn.it
afd25725 24// dE/dx analysis by: Boris Batyunya, JINR, Boris.Batiounia@cern.ch
25// Params moved to AliITSRecoParam by: Andrea Dainese, INFN
26// Material budget from TGeo by: Ludovic Gaudichet & Andrea Dainese, INFN
e43c066c 27//-------------------------------------------------------------------------
bf6adc12 28
29#include <TMatrixD.h>
e43c066c 30#include <TTree.h>
afd25725 31#include <TDatabasePDG.h>
9a5326b1 32#include <TString.h>
e50912db 33#include <TRandom.h>
a70ed6ad 34#include <TTreeStream.h>
bf6adc12 35
4187a601 36#include "AliLog.h"
a70ed6ad 37#include "AliITSPlaneEff.h"
38#include "AliITSCalibrationSPD.h"
39#include "AliITSCalibrationSDD.h"
40#include "AliITSCalibrationSSD.h"
41#include "AliCDBEntry.h"
42#include "AliCDBManager.h"
43#include "AliAlignObj.h"
44#include "AliTrackPointArray.h"
45#include "AliESDVertex.h"
af885e0f 46#include "AliESDEvent.h"
aad72f45 47#include "AliESDtrack.h"
6c94f330 48#include "AliV0.h"
bf6adc12 49#include "AliHelix.h"
a70ed6ad 50#include "AliITSChannelStatus.h"
51#include "AliITSDetTypeRec.h"
00a7cc50 52#include "AliITSRecPoint.h"
e341247d 53#include "AliITSgeomTGeo.h"
44347160 54#include "AliITSReconstructor.h"
572f41f9 55#include "AliITSClusterParam.h"
23197852 56#include "AliITSsegmentation.h"
57#include "AliITSCalibration.h"
7167ae53 58#include "AliITSPlaneEffSPD.h"
59#include "AliITSPlaneEffSDD.h"
60#include "AliITSPlaneEffSSD.h"
44396017 61#include "AliITSV0Finder.h"
ae00569a 62#include "AliITStrackerMI.h"
e43c066c 63
64ClassImp(AliITStrackerMI)
e43c066c 65
e50912db 66AliITStrackerMI::AliITSlayer AliITStrackerMI::fgLayers[AliITSgeomTGeo::kNLayers]; // ITS layers
44347160 67
8221b41b 68AliITStrackerMI::AliITStrackerMI():AliTracker(),
69fI(0),
70fBestTrack(),
71fTrackToFollow(),
72fTrackHypothesys(),
73fBestHypothesys(),
74fOriginal(),
75fCurrentEsdTrack(),
76fPass(0),
77fAfterV0(kFALSE),
78fLastLayerToTrackTo(0),
afd25725 79fCoefficients(0),
8221b41b 80fEsd(0),
e50912db 81fTrackingPhase("Default"),
6518a6c5 82fUseTGeo(3),
e50912db 83fNtracks(0),
84fxOverX0Pipe(-1.),
85fxTimesRhoPipe(-1.),
86fxOverX0PipeTrks(0),
87fxTimesRhoPipeTrks(0),
88fxOverX0ShieldTrks(0),
89fxTimesRhoShieldTrks(0),
90fxOverX0LayerTrks(0),
91fxTimesRhoLayerTrks(0),
4a66240a 92fDebugStreamer(0),
23197852 93fITSChannelStatus(0),
a70ed6ad 94fkDetTypeRec(0),
ae00569a 95fPlaneEff(0) {
8221b41b 96 //Default constructor
e50912db 97 Int_t i;
98 for(i=0;i<4;i++) fSPDdetzcentre[i]=0.;
99 for(i=0;i<2;i++) {fxOverX0Shield[i]=-1.;fxTimesRhoShield[i]=-1.;}
100 for(i=0;i<6;i++) {fxOverX0Layer[i]=-1.;fxTimesRhoLayer[i]=-1.;}
8221b41b 101}
44347160 102//------------------------------------------------------------------------
1f3e997f 103AliITStrackerMI::AliITStrackerMI(const Char_t *geom) : AliTracker(),
e50912db 104fI(AliITSgeomTGeo::GetNLayers()),
8221b41b 105fBestTrack(),
106fTrackToFollow(),
107fTrackHypothesys(),
108fBestHypothesys(),
109fOriginal(),
110fCurrentEsdTrack(),
111fPass(0),
112fAfterV0(kFALSE),
e50912db 113fLastLayerToTrackTo(AliITSRecoParam::GetLastLayerToTrackTo()),
afd25725 114fCoefficients(0),
8221b41b 115fEsd(0),
e50912db 116fTrackingPhase("Default"),
6518a6c5 117fUseTGeo(3),
e50912db 118fNtracks(0),
119fxOverX0Pipe(-1.),
120fxTimesRhoPipe(-1.),
121fxOverX0PipeTrks(0),
122fxTimesRhoPipeTrks(0),
123fxOverX0ShieldTrks(0),
124fxTimesRhoShieldTrks(0),
125fxOverX0LayerTrks(0),
126fxTimesRhoLayerTrks(0),
4a66240a 127fDebugStreamer(0),
23197852 128fITSChannelStatus(0),
a70ed6ad 129fkDetTypeRec(0),
ae00569a 130fPlaneEff(0) {
e43c066c 131 //--------------------------------------------------------------------
132 //This is the AliITStrackerMI constructor
133 //--------------------------------------------------------------------
e341247d 134 if (geom) {
135 AliWarning("\"geom\" is actually a dummy argument !");
136 }
137
afd25725 138 fCoefficients = 0;
628e7bb0 139 fAfterV0 = kFALSE;
e341247d 140
e50912db 141 for (Int_t i=1; i<AliITSgeomTGeo::GetNLayers()+1; i++) {
e341247d 142 Int_t nlad=AliITSgeomTGeo::GetNLadders(i);
143 Int_t ndet=AliITSgeomTGeo::GetNDetectors(i);
144
145 Double_t xyz[3], &x=xyz[0], &y=xyz[1], &z=xyz[2];
146 AliITSgeomTGeo::GetOrigTranslation(i,1,1,xyz);
e43c066c 147 Double_t poff=TMath::ATan2(y,x);
148 Double_t zoff=z;
e341247d 149 Double_t r=TMath::Sqrt(x*x + y*y);
e43c066c 150
e341247d 151 AliITSgeomTGeo::GetOrigTranslation(i,1,2,xyz);
e43c066c 152 r += TMath::Sqrt(x*x + y*y);
e341247d 153 AliITSgeomTGeo::GetOrigTranslation(i,2,1,xyz);
e43c066c 154 r += TMath::Sqrt(x*x + y*y);
e341247d 155 AliITSgeomTGeo::GetOrigTranslation(i,2,2,xyz);
e43c066c 156 r += TMath::Sqrt(x*x + y*y);
157 r*=0.25;
158
159 new (fgLayers+i-1) AliITSlayer(r,poff,zoff,nlad,ndet);
160
161 for (Int_t j=1; j<nlad+1; j++) {
162 for (Int_t k=1; k<ndet+1; k++) { //Fill this layer with detectors
e341247d 163 TGeoHMatrix m; AliITSgeomTGeo::GetOrigMatrix(i,j,k,m);
1f3e997f 164 const TGeoHMatrix *tm=AliITSgeomTGeo::GetTracking2LocalMatrix(i,j,k);
165 m.Multiply(tm);
a5d0fae6 166 Double_t txyz[3]={0.};
167 xyz[0]=0.;xyz[1]=0.;xyz[2]=0.;
1f3e997f 168 m.LocalToMaster(txyz,xyz);
a5d0fae6 169 r=TMath::Sqrt(xyz[0]*xyz[0] + xyz[1]*xyz[1]);
1f3e997f 170 Double_t phi=TMath::ATan2(xyz[1],xyz[0]);
e341247d 171
172 if (phi<0) phi+=TMath::TwoPi();
173 else if (phi>=TMath::TwoPi()) phi-=TMath::TwoPi();
174
e43c066c 175 AliITSdetector &det=fgLayers[i-1].GetDetector((j-1)*ndet + k-1);
176 new(&det) AliITSdetector(r,phi);
1c97ce2f 177 // compute the real radius (with misalignment)
178 TGeoHMatrix mmisal(*(AliITSgeomTGeo::GetMatrix(i,j,k)));
179 mmisal.Multiply(tm);
180 xyz[0]=0.;xyz[1]=0.;xyz[2]=0.;
181 mmisal.LocalToMaster(txyz,xyz);
182 Double_t rmisal=TMath::Sqrt(xyz[0]*xyz[0] + xyz[1]*xyz[1]);
183 det.SetRmisal(rmisal);
184
23197852 185 } // end loop on detectors
186 } // end loop on ladders
4fa7f7d1 187 fForceSkippingOfLayer[i] = 0;
23197852 188 } // end loop on layers
e43c066c 189
e43c066c 190
e50912db 191 fI=AliITSgeomTGeo::GetNLayers();
e43c066c 192
193 fPass=0;
194 fConstraint[0]=1; fConstraint[1]=0;
195
afd25725 196 Double_t xyzVtx[]={AliITSReconstructor::GetRecoParam()->GetXVdef(),
197 AliITSReconstructor::GetRecoParam()->GetYVdef(),
198 AliITSReconstructor::GetRecoParam()->GetZVdef()};
199 Double_t ersVtx[]={AliITSReconstructor::GetRecoParam()->GetSigmaXVdef(),
200 AliITSReconstructor::GetRecoParam()->GetSigmaYVdef(),
201 AliITSReconstructor::GetRecoParam()->GetSigmaZVdef()};
202 SetVertex(xyzVtx,ersVtx);
e43c066c 203
e50912db 204 fLastLayerToTrackTo=AliITSRecoParam::GetLastLayerToTrackTo();
628e7bb0 205 for (Int_t i=0;i<100000;i++){
206 fBestTrackIndex[i]=0;
207 }
afd25725 208
209 // store positions of centre of SPD modules (in z)
210 Double_t tr[3];
211 AliITSgeomTGeo::GetTranslation(1,1,1,tr);
212 fSPDdetzcentre[0] = tr[2];
213 AliITSgeomTGeo::GetTranslation(1,1,2,tr);
214 fSPDdetzcentre[1] = tr[2];
215 AliITSgeomTGeo::GetTranslation(1,1,3,tr);
216 fSPDdetzcentre[2] = tr[2];
217 AliITSgeomTGeo::GetTranslation(1,1,4,tr);
218 fSPDdetzcentre[3] = tr[2];
219
afd25725 220 fUseTGeo = AliITSReconstructor::GetRecoParam()->GetUseTGeoInTracker();
6518a6c5 221 if(AliITSReconstructor::GetRecoParam()->GetExtendedEtaAcceptance() && fUseTGeo!=1 && fUseTGeo!=3) {
222 AliWarning("fUseTGeo changed to 3 because fExtendedEtaAcceptance is kTRUE");
223 fUseTGeo = 3;
224 }
e50912db 225
226 for(Int_t i=0;i<2;i++) {fxOverX0Shield[i]=-1.;fxTimesRhoShield[i]=-1.;}
227 for(Int_t i=0;i<6;i++) {fxOverX0Layer[i]=-1.;fxTimesRhoLayer[i]=-1.;}
228
81e97e0d 229 fDebugStreamer = new TTreeSRedirector("ITSdebug.root");
230
7167ae53 231 // only for plane efficiency evaluation
58e8dc31 232 if (AliITSReconstructor::GetRecoParam()->GetComputePlaneEff() &&
233 AliITSReconstructor::GetRecoParam()->GetIPlanePlaneEff()>=0) {
0ed58a47 234 Int_t iplane=AliITSReconstructor::GetRecoParam()->GetIPlanePlaneEff();
1a64bda6 235 if(!AliITSReconstructor::GetRecoParam()->GetLayersToSkip(iplane)==1)
0ed58a47 236 AliWarning(Form("Evaluation of Plane Eff for layer %d will be attempted without removing it from tracker",iplane));
237 if (iplane<2) fPlaneEff = new AliITSPlaneEffSPD();
238 else if (iplane<4) fPlaneEff = new AliITSPlaneEffSDD();
239 else fPlaneEff = new AliITSPlaneEffSSD();
275a301c 240 if(AliITSReconstructor::GetRecoParam()->GetReadPlaneEffFromOCDB())
0ed58a47 241 if(!fPlaneEff->ReadFromCDB()) {AliWarning("AliITStrackerMI reading of AliITSPlaneEff from OCDB failed") ;}
242 if(AliITSReconstructor::GetRecoParam()->GetHistoPlaneEff()) fPlaneEff->SetCreateHistos(kTRUE);
7167ae53 243 }
628e7bb0 244}
44347160 245//------------------------------------------------------------------------
8221b41b 246AliITStrackerMI::AliITStrackerMI(const AliITStrackerMI &tracker):AliTracker(tracker),
247fI(tracker.fI),
248fBestTrack(tracker.fBestTrack),
249fTrackToFollow(tracker.fTrackToFollow),
250fTrackHypothesys(tracker.fTrackHypothesys),
251fBestHypothesys(tracker.fBestHypothesys),
252fOriginal(tracker.fOriginal),
253fCurrentEsdTrack(tracker.fCurrentEsdTrack),
254fPass(tracker.fPass),
255fAfterV0(tracker.fAfterV0),
256fLastLayerToTrackTo(tracker.fLastLayerToTrackTo),
afd25725 257fCoefficients(tracker.fCoefficients),
8221b41b 258fEsd(tracker.fEsd),
e50912db 259fTrackingPhase(tracker.fTrackingPhase),
afd25725 260fUseTGeo(tracker.fUseTGeo),
e50912db 261fNtracks(tracker.fNtracks),
262fxOverX0Pipe(tracker.fxOverX0Pipe),
263fxTimesRhoPipe(tracker.fxTimesRhoPipe),
264fxOverX0PipeTrks(0),
265fxTimesRhoPipeTrks(0),
266fxOverX0ShieldTrks(0),
267fxTimesRhoShieldTrks(0),
268fxOverX0LayerTrks(0),
269fxTimesRhoLayerTrks(0),
4a66240a 270fDebugStreamer(tracker.fDebugStreamer),
23197852 271fITSChannelStatus(tracker.fITSChannelStatus),
a70ed6ad 272fkDetTypeRec(tracker.fkDetTypeRec),
ae00569a 273fPlaneEff(tracker.fPlaneEff) {
8221b41b 274 //Copy constructor
e50912db 275 Int_t i;
276 for(i=0;i<4;i++) {
277 fSPDdetzcentre[i]=tracker.fSPDdetzcentre[i];
278 }
279 for(i=0;i<6;i++) {
280 fxOverX0Layer[i]=tracker.fxOverX0Layer[i];
281 fxTimesRhoLayer[i]=tracker.fxTimesRhoLayer[i];
282 }
283 for(i=0;i<2;i++) {
284 fxOverX0Shield[i]=tracker.fxOverX0Shield[i];
285 fxTimesRhoShield[i]=tracker.fxTimesRhoShield[i];
286 }
8221b41b 287}
44347160 288//------------------------------------------------------------------------
8221b41b 289AliITStrackerMI & AliITStrackerMI::operator=(const AliITStrackerMI &tracker){
290 //Assignment operator
291 this->~AliITStrackerMI();
292 new(this) AliITStrackerMI(tracker);
293 return *this;
294}
44347160 295//------------------------------------------------------------------------
628e7bb0 296AliITStrackerMI::~AliITStrackerMI()
297{
298 //
299 //destructor
300 //
e50912db 301 if (fCoefficients) delete [] fCoefficients;
302 DeleteTrksMaterialLUT();
81e97e0d 303 if (fDebugStreamer) {
304 //fDebugStreamer->Close();
305 delete fDebugStreamer;
306 }
23197852 307 if(fITSChannelStatus) delete fITSChannelStatus;
308 if(fPlaneEff) delete fPlaneEff;
e43c066c 309}
44347160 310//------------------------------------------------------------------------
23197852 311void AliITStrackerMI::ReadBadFromDetTypeRec() {
312 //--------------------------------------------------------------------
313 //This function read ITS bad detectors, chips, channels from AliITSDetTypeRec
314 //i.e. from OCDB
315 //--------------------------------------------------------------------
316
317 if(!AliITSReconstructor::GetRecoParam()->GetUseBadZonesFromOCDB()) return;
318
4187a601 319 Info("ReadBadFromDetTypeRec","Reading info about bad ITS detectors and channels");
23197852 320
a70ed6ad 321 if(!fkDetTypeRec) Error("ReadBadFromDetTypeRec","AliITSDetTypeRec nof found!\n");
23197852 322
323 // ITS channels map
324 if(fITSChannelStatus) delete fITSChannelStatus;
a70ed6ad 325 fITSChannelStatus = new AliITSChannelStatus(fkDetTypeRec);
23197852 326
327 // ITS detectors and chips
328 Int_t i=0,j=0,k=0,ndet=0;
329 for (i=1; i<AliITSgeomTGeo::GetNLayers()+1; i++) {
4187a601 330 Int_t nBadDetsPerLayer=0;
23197852 331 ndet=AliITSgeomTGeo::GetNDetectors(i);
332 for (j=1; j<AliITSgeomTGeo::GetNLadders(i)+1; j++) {
333 for (k=1; k<ndet+1; k++) {
334 AliITSdetector &det=fgLayers[i-1].GetDetector((j-1)*ndet + k-1);
a70ed6ad 335 det.ReadBadDetectorAndChips(i-1,(j-1)*ndet + k-1,fkDetTypeRec);
4187a601 336 if(det.IsBad()) {nBadDetsPerLayer++;}
23197852 337 } // end loop on detectors
338 } // end loop on ladders
4187a601 339 Info("ReadBadFromDetTypeRec",Form("Layer %d: %d bad out of %d",i-1,nBadDetsPerLayer,ndet*AliITSgeomTGeo::GetNLadders(i)));
23197852 340 } // end loop on layers
341
342 return;
343}
344//------------------------------------------------------------------------
e43c066c 345Int_t AliITStrackerMI::LoadClusters(TTree *cTree) {
346 //--------------------------------------------------------------------
347 //This function loads ITS clusters
348 //--------------------------------------------------------------------
00a7cc50 349 TBranch *branch=cTree->GetBranch("ITSRecPoints");
e43c066c 350 if (!branch) {
351 Error("LoadClusters"," can't get the branch !\n");
352 return 1;
353 }
354
23197852 355 static TClonesArray dummy("AliITSRecPoint",10000), *clusters=&dummy;
e43c066c 356 branch->SetAddress(&clusters);
357
23197852 358 Int_t i=0,j=0,ndet=0;
e43c066c 359 Int_t detector=0;
23197852 360 for (i=0; i<AliITSgeomTGeo::GetNLayers(); i++) {
361 ndet=fgLayers[i].GetNdetectors();
e43c066c 362 Int_t jmax = j + fgLayers[i].GetNladders()*ndet;
363 for (; j<jmax; j++) {
364 if (!cTree->GetEvent(j)) continue;
365 Int_t ncl=clusters->GetEntriesFast();
366 SignDeltas(clusters,GetZ());
1f3e997f 367
e43c066c 368 while (ncl--) {
00a7cc50 369 AliITSRecPoint *c=(AliITSRecPoint*)clusters->UncheckedAt(ncl);
75fb37cc 370 detector=c->GetDetectorIndex();
a504d56f 371
75fb37cc 372 if (!c->Misalign()) AliWarning("Can't misalign this cluster !");
a504d56f 373
00a7cc50 374 fgLayers[i].InsertCluster(new AliITSRecPoint(*c));
e43c066c 375 }
19e17f92 376 clusters->Clear();
afd25725 377 // add dead zone "virtual" cluster in SPD, if there is a cluster within
378 // zwindow cm from the dead zone
379 if (i<2 && AliITSReconstructor::GetRecoParam()->GetAddVirtualClustersInDeadZone()) {
e50912db 380 for (Float_t xdead = 0; xdead < AliITSRecoParam::GetSPDdetxlength(); xdead += (i+1.)*AliITSReconstructor::GetRecoParam()->GetXPassDeadZoneHits()) {
afd25725 381 Int_t lab[4] = {0,0,0,detector};
382 Int_t info[3] = {0,0,i};
6518a6c5 383 Float_t q = 0.; // this identifies virtual clusters
afd25725 384 Float_t hit[5] = {xdead,
385 0.,
386 AliITSReconstructor::GetRecoParam()->GetSigmaXDeadZoneHit2(),
387 AliITSReconstructor::GetRecoParam()->GetSigmaZDeadZoneHit2(),
388 q};
389 Bool_t local = kTRUE;
390 Double_t zwindow = AliITSReconstructor::GetRecoParam()->GetZWindowDeadZone();
e50912db 391 hit[1] = fSPDdetzcentre[0]+0.5*AliITSRecoParam::GetSPDdetzlength();
afd25725 392 if (TMath::Abs(fgLayers[i].GetDetector(detector).GetZmax()-hit[1])<zwindow)
393 fgLayers[i].InsertCluster(new AliITSRecPoint(lab,hit,info,local));
e50912db 394 hit[1] = fSPDdetzcentre[1]-0.5*AliITSRecoParam::GetSPDdetzlength();
afd25725 395 if (TMath::Abs(fgLayers[i].GetDetector(detector).GetZmax()-hit[1])<zwindow)
396 fgLayers[i].InsertCluster(new AliITSRecPoint(lab,hit,info,local));
e50912db 397 hit[1] = fSPDdetzcentre[1]+0.5*AliITSRecoParam::GetSPDdetzlength();
afd25725 398 if (TMath::Abs(fgLayers[i].GetDetector(detector).GetZmax()-hit[1])<zwindow)
399 fgLayers[i].InsertCluster(new AliITSRecPoint(lab,hit,info,local));
e50912db 400 hit[1] = fSPDdetzcentre[2]-0.5*AliITSRecoParam::GetSPDdetzlength();
afd25725 401 if (TMath::Abs(fgLayers[i].GetDetector(detector).GetZmax()-hit[1])<zwindow)
402 fgLayers[i].InsertCluster(new AliITSRecPoint(lab,hit,info,local));
e50912db 403 hit[1] = fSPDdetzcentre[2]+0.5*AliITSRecoParam::GetSPDdetzlength();
afd25725 404 if (TMath::Abs(fgLayers[i].GetDetector(detector).GetZmax()-hit[1])<zwindow)
405 fgLayers[i].InsertCluster(new AliITSRecPoint(lab,hit,info,local));
e50912db 406 hit[1] = fSPDdetzcentre[3]-0.5*AliITSRecoParam::GetSPDdetzlength();
afd25725 407 if (TMath::Abs(fgLayers[i].GetDetector(detector).GetZmax()-hit[1])<zwindow)
408 fgLayers[i].InsertCluster(new AliITSRecPoint(lab,hit,info,local));
e43c066c 409 }
6518a6c5 410 } // "virtual" clusters in SPD
e43c066c 411
412 }
413 //
414 fgLayers[i].ResetRoad(); //road defined by the cluster density
415 fgLayers[i].SortClusters();
416 }
417
4fa7f7d1 418 // check whether we have to skip some layers
419 SetForceSkippingOfLayer();
420
e43c066c 421 return 0;
422}
44347160 423//------------------------------------------------------------------------
e43c066c 424void AliITStrackerMI::UnloadClusters() {
425 //--------------------------------------------------------------------
426 //This function unloads ITS clusters
427 //--------------------------------------------------------------------
e50912db 428 for (Int_t i=0; i<AliITSgeomTGeo::GetNLayers(); i++) fgLayers[i].ResetClusters();
e43c066c 429}
44347160 430//------------------------------------------------------------------------
23197852 431void AliITStrackerMI::FillClusterArray(TObjArray* array) const {
432 //--------------------------------------------------------------------
433 // Publishes all pointers to clusters known to the tracker into the
434 // passed object array.
435 // The ownership is not transfered - the caller is not expected to delete
436 // the clusters.
437 //--------------------------------------------------------------------
438
439 for(Int_t i=0; i<AliITSgeomTGeo::GetNLayers(); i++) {
440 for(Int_t icl=0; icl<fgLayers[i].GetNumberOfClusters(); icl++) {
441 AliCluster *cl = (AliCluster*)fgLayers[i].GetCluster(icl);
442 array->AddLast(cl);
443 }
444 }
445
446 return;
447}
448//------------------------------------------------------------------------
44396017 449Int_t AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(AliITStrackMI *t) {
e43c066c 450 //--------------------------------------------------------------------
451 // Correction for the material between the TPC and the ITS
e43c066c 452 //--------------------------------------------------------------------
e50912db 453 if (t->GetX() > AliITSRecoParam::Getriw()) { // inward direction
454 if (!t->PropagateToTGeo(AliITSRecoParam::Getriw(),1)) return 0;// TPC inner wall
455 if (!t->PropagateToTGeo(AliITSRecoParam::Getrcd(),1)) return 0;// TPC central drum
456 if (!t->PropagateToTGeo(AliITSRecoParam::Getrs(),1)) return 0;// ITS screen
457 } else if (t->GetX() < AliITSRecoParam::Getrs()) { // outward direction
458 if (!t->PropagateToTGeo(AliITSRecoParam::Getrs(),1)) return 0;// ITS screen
459 if (!t->PropagateToTGeo(AliITSRecoParam::Getrcd(),1)) return 0;// TPC central drum
460 if (!t->PropagateToTGeo(AliITSRecoParam::Getriw()+0.001,1)) return 0;// TPC inner wall
e43c066c 461 } else {
44396017 462 printf("CorrectForTPCtoITSDeadZoneMaterial: Track is already in the dead zone !\n");
e50912db 463 return 0;
e43c066c 464 }
465
e50912db 466 return 1;
e43c066c 467}
44347160 468//------------------------------------------------------------------------
af885e0f 469Int_t AliITStrackerMI::Clusters2Tracks(AliESDEvent *event) {
e43c066c 470 //--------------------------------------------------------------------
471 // This functions reconstructs ITS tracks
472 // The clusters must be already loaded !
473 //--------------------------------------------------------------------
fddf8459 474
4fa7f7d1 475 AliDebug(2,Form("SKIPPING %d %d %d %d %d %d",ForceSkippingOfLayer(0),ForceSkippingOfLayer(1),ForceSkippingOfLayer(2),ForceSkippingOfLayer(3),ForceSkippingOfLayer(4),ForceSkippingOfLayer(5)));
fddf8459 476
e50912db 477 fTrackingPhase="Clusters2Tracks";
478
e43c066c 479 TObjArray itsTracks(15000);
628e7bb0 480 fOriginal.Clear();
81e97e0d 481 fEsd = event; // store pointer to the esd
afd25725 482
9a5326b1 483 // temporary (for cosmics)
484 if(event->GetVertex()) {
485 TString title = event->GetVertex()->GetTitle();
486 if(title.Contains("cosmics")) {
487 Double_t xyz[3]={GetX(),GetY(),GetZ()};
488 Double_t exyz[3]={0.1,0.1,0.1};
489 SetVertex(xyz,exyz);
490 }
491 }
492 // temporary
afd25725 493
e43c066c 494 {/* Read ESD tracks */
afd25725 495 Double_t pimass = TDatabasePDG::Instance()->GetParticle(211)->Mass();
e43c066c 496 Int_t nentr=event->GetNumberOfTracks();
497 Info("Clusters2Tracks", "Number of ESD tracks: %d\n", nentr);
498 while (nentr--) {
499 AliESDtrack *esd=event->GetTrack(nentr);
4187a601 500 // ---- for debugging:
501 //if(TMath::Abs(esd->GetX()-83.65)<0.1) { FILE *f=fopen("tpc.dat","a"); fprintf(f,"%f %f %f %f %f %f\n",(Float_t)event->GetEventNumberInFile(),(Float_t)TMath::Abs(esd->GetLabel()),(Float_t)esd->GetX(),(Float_t)esd->GetY(),(Float_t)esd->GetZ(),(Float_t)esd->Pt()); fclose(f); }
e43c066c 502
503 if ((esd->GetStatus()&AliESDtrack::kTPCin)==0) continue;
504 if (esd->GetStatus()&AliESDtrack::kTPCout) continue;
505 if (esd->GetStatus()&AliESDtrack::kITSin) continue;
51ad6848 506 if (esd->GetKinkIndex(0)>0) continue; //kink daughter
15dd636f 507 AliITStrackMI *t=0;
e43c066c 508 try {
15dd636f 509 t=new AliITStrackMI(*esd);
e43c066c 510 } catch (const Char_t *msg) {
628e7bb0 511 //Warning("Clusters2Tracks",msg);
e43c066c 512 delete t;
513 continue;
514 }
b9671574 515 t->GetDZ(GetX(),GetY(),GetZ(),t->GetDP()); //I.B.
516 Double_t vdist = TMath::Sqrt(t->GetD(0)*t->GetD(0)+t->GetD(1)*t->GetD(1));
6518a6c5 517
518
afd25725 519 // look at the ESD mass hypothesys !
520 if (t->GetMass()<0.9*pimass) t->SetMass(pimass);
e43c066c 521 // write expected q
b9671574 522 t->SetExpQ(TMath::Max(0.8*t->GetESDtrack()->GetTPCsignal(),30.));
e43c066c 523
afd25725 524 if (esd->GetV0Index(0)>0 && t->GetD(0)<AliITSReconstructor::GetRecoParam()->GetMaxDforV0dghtrForProlongation()){
81e97e0d 525 //track - can be V0 according to TPC
afd25725 526 } else {
527 if (TMath::Abs(t->GetD(0))>AliITSReconstructor::GetRecoParam()->GetMaxDForProlongation()) {
81e97e0d 528 delete t;
529 continue;
e50912db 530 }
afd25725 531 if (TMath::Abs(vdist)>AliITSReconstructor::GetRecoParam()->GetMaxDZForProlongation()) {
81e97e0d 532 delete t;
533 continue;
534 }
6c23ffed 535 if (t->Pt()<AliITSReconstructor::GetRecoParam()->GetMinPtForProlongation()) {
81e97e0d 536 delete t;
537 continue;
538 }
e50912db 539 if (!CorrectForTPCtoITSDeadZoneMaterial(t)) {
81e97e0d 540 delete t;
541 continue;
542 }
e43c066c 543 }
b9671574 544 t->SetReconstructed(kFALSE);
e43c066c 545 itsTracks.AddLast(t);
628e7bb0 546 fOriginal.AddLast(t);
e43c066c 547 }
548 } /* End Read ESD tracks */
549
550 itsTracks.Sort();
628e7bb0 551 fOriginal.Sort();
e43c066c 552 Int_t nentr=itsTracks.GetEntriesFast();
553 fTrackHypothesys.Expand(nentr);
81e97e0d 554 fBestHypothesys.Expand(nentr);
afd25725 555 MakeCoefficients(nentr);
513ab112 556 if(fUseTGeo==3 || fUseTGeo==4) MakeTrksMaterialLUT(event->GetNumberOfTracks());
e43c066c 557 Int_t ntrk=0;
afd25725 558 // THE TWO TRACKING PASSES
e43c066c 559 for (fPass=0; fPass<2; fPass++) {
560 Int_t &constraint=fConstraint[fPass]; if (constraint<0) continue;
e50912db 561 for (fCurrentEsdTrack=0; fCurrentEsdTrack<nentr; fCurrentEsdTrack++) {
e50912db 562 AliITStrackMI *t=(AliITStrackMI*)itsTracks.UncheckedAt(fCurrentEsdTrack);
e43c066c 563 if (t==0) continue; //this track has been already tracked
1c97ce2f 564 //cout<<"========== "<<fPass<<" "<<fCurrentEsdTrack<<" =========\n";
b9671574 565 if (t->GetReconstructed()&&(t->GetNUsed()<1.5)) continue; //this track was already "succesfully" reconstructed
791f9a2a 566 Float_t dz[2]; t->GetDZ(GetX(),GetY(),GetZ(),dz); //I.B.
afd25725 567 if (fConstraint[fPass]) {
568 if (TMath::Abs(dz[0])>AliITSReconstructor::GetRecoParam()->GetMaxDZToUseConstraint() ||
569 TMath::Abs(dz[1])>AliITSReconstructor::GetRecoParam()->GetMaxDZToUseConstraint()) continue;
570 }
e43c066c 571
572 Int_t tpcLabel=t->GetLabel(); //save the TPC track label
4187a601 573 AliDebug(2,Form("LABEL %d pass %d",tpcLabel,fPass));
e43c066c 574 fI = 6;
575 ResetTrackToFollow(*t);
576 ResetBestTrack();
6518a6c5 577
e50912db 578 FollowProlongationTree(t,fCurrentEsdTrack,fConstraint[fPass]);
1c97ce2f 579
e43c066c 580
581 SortTrackHypothesys(fCurrentEsdTrack,20,0); //MI change
582 //
ae00569a 583 AliITStrackMI *besttrack = GetBestHypothesys(fCurrentEsdTrack,t,15);
e43c066c 584 if (!besttrack) continue;
585 besttrack->SetLabel(tpcLabel);
586 // besttrack->CookdEdx();
587 CookdEdx(besttrack);
b9671574 588 besttrack->SetFakeRatio(1.);
e43c066c 589 CookLabel(besttrack,0.); //For comparison only
e43c066c 590 UpdateESDtrack(besttrack,AliESDtrack::kITSin);
628e7bb0 591
628e7bb0 592 if (fConstraint[fPass]&&(!besttrack->IsGoldPrimary())) continue; //to be tracked also without vertex constrain
593
b9671574 594 t->SetReconstructed(kTRUE);
1c97ce2f 595 ntrk++;
4187a601 596 AliDebug(2,Form("TRACK! (label %d) ncls %d",besttrack->GetLabel(),besttrack->GetNumberOfClusters()));
e43c066c 597 }
598 GetBestHypothesysMIP(itsTracks);
afd25725 599 } // end loop on the two tracking passes
e43c066c 600
cd90f0a2 601 if(event->GetNumberOfV0s()>0) AliITSV0Finder::UpdateTPCV0(event,this);
602 if(AliITSReconstructor::GetRecoParam()->GetFindV0s()) AliITSV0Finder::FindV02(event,this);
628e7bb0 603 fAfterV0 = kTRUE;
628e7bb0 604 //
e43c066c 605 itsTracks.Delete();
606 //
607 Int_t entries = fTrackHypothesys.GetEntriesFast();
e50912db 608 for (Int_t ientry=0; ientry<entries; ientry++) {
e43c066c 609 TObjArray * array =(TObjArray*)fTrackHypothesys.UncheckedAt(ientry);
610 if (array) array->Delete();
611 delete fTrackHypothesys.RemoveAt(ientry);
612 }
613
614 fTrackHypothesys.Delete();
81e97e0d 615 fBestHypothesys.Delete();
628e7bb0 616 fOriginal.Clear();
afd25725 617 delete [] fCoefficients;
618 fCoefficients=0;
e50912db 619 DeleteTrksMaterialLUT();
620
e43c066c 621 Info("Clusters2Tracks","Number of prolonged tracks: %d\n",ntrk);
e50912db 622
623 fTrackingPhase="Default";
628e7bb0 624
e43c066c 625 return 0;
626}
44347160 627//------------------------------------------------------------------------
af885e0f 628Int_t AliITStrackerMI::PropagateBack(AliESDEvent *event) {
e43c066c 629 //--------------------------------------------------------------------
630 // This functions propagates reconstructed ITS tracks back
631 // The clusters must be loaded !
632 //--------------------------------------------------------------------
e50912db 633 fTrackingPhase="PropagateBack";
e43c066c 634 Int_t nentr=event->GetNumberOfTracks();
635 Info("PropagateBack", "Number of ESD tracks: %d\n", nentr);
636
637 Int_t ntrk=0;
638 for (Int_t i=0; i<nentr; i++) {
639 AliESDtrack *esd=event->GetTrack(i);
640
641 if ((esd->GetStatus()&AliESDtrack::kITSin)==0) continue;
642 if (esd->GetStatus()&AliESDtrack::kITSout) continue;
643
15dd636f 644 AliITStrackMI *t=0;
e43c066c 645 try {
15dd636f 646 t=new AliITStrackMI(*esd);
e43c066c 647 } catch (const Char_t *msg) {
628e7bb0 648 //Warning("PropagateBack",msg);
e43c066c 649 delete t;
650 continue;
651 }
b9671574 652 t->SetExpQ(TMath::Max(0.8*t->GetESDtrack()->GetTPCsignal(),30.));
e43c066c 653
654 ResetTrackToFollow(*t);
4fa7f7d1 655
67178f34 656 /*
9a5326b1 657 // propagate to vertex [SR, GSI 17.02.2003]
e43c066c 658 // Start Time measurement [SR, GSI 17.02.2003], corrected by I.Belikov
e50912db 659 if (CorrectForPipeMaterial(&fTrackToFollow,"inward")) {
660 if (fTrackToFollow.PropagateToVertex(event->GetVertex()))
661 fTrackToFollow.StartTimeIntegral();
662 // from vertex to outside pipe
663 CorrectForPipeMaterial(&fTrackToFollow,"outward");
67178f34 664 }*/
665 // Start time integral and add distance from current position to vertex
666 Double_t xyzTrk[3],xyzVtx[3]={GetX(),GetY(),GetZ()};
667 fTrackToFollow.GetXYZ(xyzTrk);
668 Double_t dst2 = 0.;
669 for (Int_t icoord=0; icoord<3; icoord++) {
670 Double_t di = xyzTrk[icoord] - xyzVtx[icoord];
671 dst2 += di*di;
e43c066c 672 }
67178f34 673 fTrackToFollow.StartTimeIntegral();
674 fTrackToFollow.AddTimeStep(TMath::Sqrt(dst2));
e43c066c 675
6c94f330 676 fTrackToFollow.ResetCovariance(10.); fTrackToFollow.ResetClusters();
e50912db 677 if (RefitAt(AliITSRecoParam::GetrInsideITSscreen(),&fTrackToFollow,t)) {
678 if (!CorrectForTPCtoITSDeadZoneMaterial(&fTrackToFollow)) {
e43c066c 679 delete t;
680 continue;
681 }
682 fTrackToFollow.SetLabel(t->GetLabel());
683 //fTrackToFollow.CookdEdx();
684 CookLabel(&fTrackToFollow,0.); //For comparison only
685 fTrackToFollow.UpdateESDtrack(AliESDtrack::kITSout);
686 //UseClusters(&fTrackToFollow);
687 ntrk++;
688 }
689 delete t;
690 }
691
692 Info("PropagateBack","Number of back propagated ITS tracks: %d\n",ntrk);
693
e50912db 694 fTrackingPhase="Default";
695
e43c066c 696 return 0;
697}
44347160 698//------------------------------------------------------------------------
af885e0f 699Int_t AliITStrackerMI::RefitInward(AliESDEvent *event) {
e43c066c 700 //--------------------------------------------------------------------
701 // This functions refits ITS tracks using the
702 // "inward propagated" TPC tracks
703 // The clusters must be loaded !
704 //--------------------------------------------------------------------
e50912db 705 fTrackingPhase="RefitInward";
44396017 706
cd90f0a2 707 if(AliITSReconstructor::GetRecoParam()->GetFindV0s()) AliITSV0Finder::RefitV02(event,this);
44396017 708
e43c066c 709 Int_t nentr=event->GetNumberOfTracks();
710 Info("RefitInward", "Number of ESD tracks: %d\n", nentr);
711
712 Int_t ntrk=0;
713 for (Int_t i=0; i<nentr; i++) {
714 AliESDtrack *esd=event->GetTrack(i);
715
716 if ((esd->GetStatus()&AliESDtrack::kITSout) == 0) continue;
717 if (esd->GetStatus()&AliESDtrack::kITSrefit) continue;
718 if (esd->GetStatus()&AliESDtrack::kTPCout)
719 if ((esd->GetStatus()&AliESDtrack::kTPCrefit)==0) continue;
720
15dd636f 721 AliITStrackMI *t=0;
e43c066c 722 try {
15dd636f 723 t=new AliITStrackMI(*esd);
e43c066c 724 } catch (const Char_t *msg) {
628e7bb0 725 //Warning("RefitInward",msg);
e43c066c 726 delete t;
727 continue;
728 }
b9671574 729 t->SetExpQ(TMath::Max(0.8*t->GetESDtrack()->GetTPCsignal(),30.));
e50912db 730 if (!CorrectForTPCtoITSDeadZoneMaterial(t)) {
e43c066c 731 delete t;
732 continue;
733 }
734
735 ResetTrackToFollow(*t);
736 fTrackToFollow.ResetClusters();
737
738 if ((esd->GetStatus()&AliESDtrack::kTPCin)==0)
6c94f330 739 fTrackToFollow.ResetCovariance(10.);
e43c066c 740
741 //Refitting...
58e8dc31 742 Bool_t pe=(AliITSReconstructor::GetRecoParam()->GetComputePlaneEff() &&
743 AliITSReconstructor::GetRecoParam()->GetIPlanePlaneEff()>=0);
1a64bda6 744
4187a601 745 AliDebug(2,Form("Refit LABEL %d %d",t->GetLabel(),t->GetNumberOfClusters()));
7167ae53 746 if (RefitAt(AliITSRecoParam::GetrInsideSPD1(),&fTrackToFollow,t,kTRUE,pe)) {
4187a601 747 AliDebug(2," refit OK");
e43c066c 748 fTrackToFollow.SetLabel(t->GetLabel());
749 // fTrackToFollow.CookdEdx();
750 CookdEdx(&fTrackToFollow);
751
752 CookLabel(&fTrackToFollow,0.0); //For comparison only
753
9a5326b1 754 //The beam pipe
e50912db 755 if (CorrectForPipeMaterial(&fTrackToFollow,"inward")) {
ae00569a 756 fTrackToFollow.UpdateESDtrack(AliESDtrack::kITSrefit);
9a5326b1 757 AliESDtrack *esdTrack =fTrackToFollow.GetESDtrack();
ae00569a 758 //printf(" %d\n",esdTrack->GetITSModuleIndex(0));
759 //esdTrack->UpdateTrackParams(&fTrackToFollow,AliESDtrack::kITSrefit); //original line
f7a1cc68 760 Double_t r[3]={0.,0.,0.};
9a5326b1 761 Double_t maxD=3.;
762 esdTrack->RelateToVertex(event->GetVertex(),GetBz(r),maxD);
763 ntrk++;
e43c066c 764 }
765 }
766 delete t;
767 }
768
769 Info("RefitInward","Number of refitted tracks: %d\n",ntrk);
770
e50912db 771 fTrackingPhase="Default";
772
e43c066c 773 return 0;
774}
44347160 775//------------------------------------------------------------------------
e43c066c 776AliCluster *AliITStrackerMI::GetCluster(Int_t index) const {
777 //--------------------------------------------------------------------
778 // Return pointer to a given cluster
779 //--------------------------------------------------------------------
780 Int_t l=(index & 0xf0000000) >> 28;
781 Int_t c=(index & 0x0fffffff) >> 00;
782 return fgLayers[l].GetCluster(c);
783}
9a5326b1 784//------------------------------------------------------------------------
df29e9a4 785Bool_t AliITStrackerMI::GetTrackPoint(Int_t index, AliTrackPoint& p) const {
9a5326b1 786 //--------------------------------------------------------------------
df29e9a4 787 // Get track space point with index i
9a5326b1 788 //--------------------------------------------------------------------
60066577 789
78b03929 790 Int_t l=(index & 0xf0000000) >> 28;
791 Int_t c=(index & 0x0fffffff) >> 00;
00a7cc50 792 AliITSRecPoint *cl = fgLayers[l].GetCluster(c);
df29e9a4 793 Int_t idet = cl->GetDetectorIndex();
4e65ff9e 794
60066577 795 Float_t xyz[3];
796 Float_t cov[6];
797 cl->GetGlobalXYZ(xyz);
798 cl->GetGlobalCov(cov);
799 p.SetXYZ(xyz, cov);
be77c8c9 800 p.SetCharge(cl->GetQ());
ebbcf1ea 801 p.SetDriftTime(cl->GetDriftTime());
b7bcc8ed 802 p.SetChargeRatio(cl->GetChargeRatio());
e3901fd8 803 p.SetClusterType(cl->GetClusterType());
ae079791 804 AliGeomManager::ELayerID iLayer = AliGeomManager::kInvalidLayer;
df29e9a4 805 switch (l) {
806 case 0:
ae079791 807 iLayer = AliGeomManager::kSPD1;
df29e9a4 808 break;
809 case 1:
ae079791 810 iLayer = AliGeomManager::kSPD2;
df29e9a4 811 break;
812 case 2:
ae079791 813 iLayer = AliGeomManager::kSDD1;
df29e9a4 814 break;
815 case 3:
ae079791 816 iLayer = AliGeomManager::kSDD2;
9a5326b1 817 break;
818 case 4:
819 iLayer = AliGeomManager::kSSD1;
820 break;
821 case 5:
822 iLayer = AliGeomManager::kSSD2;
823 break;
824 default:
825 AliWarning(Form("Wrong layer index in ITS (%d) !",l));
826 break;
827 };
828 UShort_t volid = AliGeomManager::LayerToVolUID(iLayer,idet);
829 p.SetVolumeID((UShort_t)volid);
830 return kTRUE;
831}
832//------------------------------------------------------------------------
833Bool_t AliITStrackerMI::GetTrackPointTrackingError(Int_t index,
834 AliTrackPoint& p, const AliESDtrack *t) {
835 //--------------------------------------------------------------------
836 // Get track space point with index i
837 // (assign error estimated during the tracking)
838 //--------------------------------------------------------------------
839
840 Int_t l=(index & 0xf0000000) >> 28;
841 Int_t c=(index & 0x0fffffff) >> 00;
842 const AliITSRecPoint *cl = fgLayers[l].GetCluster(c);
843 Int_t idet = cl->GetDetectorIndex();
4187a601 844
9a5326b1 845 const AliITSdetector &det=fgLayers[l].GetDetector(idet);
846
847 // tgphi and tglambda of the track in tracking frame with alpha=det.GetPhi
848 Float_t detxy[2];
849 detxy[0] = det.GetR()*TMath::Cos(det.GetPhi());
850 detxy[1] = det.GetR()*TMath::Sin(det.GetPhi());
851 Double_t alpha = t->GetAlpha();
852 Double_t xdetintrackframe = detxy[0]*TMath::Cos(alpha)+detxy[1]*TMath::Sin(alpha);
9be1d1c7 853 Float_t phi = TMath::ASin(t->GetSnpAt(xdetintrackframe,GetBz()));
9a5326b1 854 phi += alpha-det.GetPhi();
855 Float_t tgphi = TMath::Tan(phi);
856
857 Float_t tgl = t->GetTgl(); // tgl about const along track
858 Float_t expQ = TMath::Max(0.8*t->GetTPCsignal(),30.);
859
d9ead1a0 860 Float_t errtrky,errtrkz,covyz;
8c139cf3 861 Bool_t addMisalErr=kFALSE;
d9ead1a0 862 AliITSClusterParam::GetError(l,cl,tgl,tgphi,expQ,errtrky,errtrkz,covyz,addMisalErr);
9a5326b1 863
864 Float_t xyz[3];
865 Float_t cov[6];
866 cl->GetGlobalXYZ(xyz);
867 // cl->GetGlobalCov(cov);
868 Float_t pos[3] = {0.,0.,0.};
d9ead1a0 869 AliCluster tmpcl((UShort_t)cl->GetVolumeId(),pos[0],pos[1],pos[2],errtrky*errtrky,errtrkz*errtrkz,covyz);
9a5326b1 870 tmpcl.GetGlobalCov(cov);
871
872 p.SetXYZ(xyz, cov);
be77c8c9 873 p.SetCharge(cl->GetQ());
ebbcf1ea 874 p.SetDriftTime(cl->GetDriftTime());
b7bcc8ed 875 p.SetChargeRatio(cl->GetChargeRatio());
e3901fd8 876 p.SetClusterType(cl->GetClusterType());
9a5326b1 877
878 AliGeomManager::ELayerID iLayer = AliGeomManager::kInvalidLayer;
879 switch (l) {
880 case 0:
881 iLayer = AliGeomManager::kSPD1;
882 break;
883 case 1:
884 iLayer = AliGeomManager::kSPD2;
885 break;
886 case 2:
887 iLayer = AliGeomManager::kSDD1;
888 break;
889 case 3:
890 iLayer = AliGeomManager::kSDD2;
df29e9a4 891 break;
892 case 4:
ae079791 893 iLayer = AliGeomManager::kSSD1;
df29e9a4 894 break;
895 case 5:
ae079791 896 iLayer = AliGeomManager::kSSD2;
df29e9a4 897 break;
898 default:
899 AliWarning(Form("Wrong layer index in ITS (%d) !",l));
900 break;
901 };
ae079791 902 UShort_t volid = AliGeomManager::LayerToVolUID(iLayer,idet);
4187a601 903
df29e9a4 904 p.SetVolumeID((UShort_t)volid);
905 return kTRUE;
906}
44347160 907//------------------------------------------------------------------------
81e97e0d 908void AliITStrackerMI::FollowProlongationTree(AliITStrackMI * otrack, Int_t esdindex, Bool_t constrain)
e43c066c 909{
910 //--------------------------------------------------------------------
911 // Follow prolongation tree
912 //--------------------------------------------------------------------
81e97e0d 913 //
afd25725 914 Double_t xyzVtx[]={GetX(),GetY(),GetZ()};
915 Double_t ersVtx[]={GetSigmaX(),GetSigmaY(),GetSigmaZ()};
916
2755f080 917
b9671574 918 AliESDtrack * esd = otrack->GetESDtrack();
afd25725 919 if (esd->GetV0Index(0)>0) {
81e97e0d 920 // TEMPORARY SOLLUTION: map V0 indexes to point to proper track
afd25725 921 // mapping of ESD track is different as ITS track in Containers
81e97e0d 922 // Need something more stable
44347160 923 // Indexes are set back again to the ESD track indexes in UpdateTPCV0
81e97e0d 924 for (Int_t i=0;i<3;i++){
925 Int_t index = esd->GetV0Index(i);
926 if (index==0) break;
d6a49f20 927 AliESDv0 * vertex = fEsd->GetV0(index);
81e97e0d 928 if (vertex->GetStatus()<0) continue; // rejected V0
929 //
afd25725 930 if (esd->GetSign()>0) {
931 vertex->SetIndex(0,esdindex);
932 } else {
933 vertex->SetIndex(1,esdindex);
934 }
81e97e0d 935 }
936 }
937 TObjArray *bestarray = (TObjArray*)fBestHypothesys.At(esdindex);
938 if (!bestarray){
939 bestarray = new TObjArray(5);
940 fBestHypothesys.AddAt(bestarray,esdindex);
941 }
e43c066c 942
81e97e0d 943 //
e43c066c 944 //setup tree of the prolongations
945 //
15dd636f 946 static AliITStrackMI tracks[7][100];
947 AliITStrackMI *currenttrack;
948 static AliITStrackMI currenttrack1;
949 static AliITStrackMI currenttrack2;
950 static AliITStrackMI backuptrack;
e43c066c 951 Int_t ntracks[7];
952 Int_t nindexes[7][100];
953 Float_t normalizedchi2[100];
954 for (Int_t ilayer=0;ilayer<6;ilayer++) ntracks[ilayer]=0;
b9671574 955 otrack->SetNSkipped(0);
15dd636f 956 new (&(tracks[6][0])) AliITStrackMI(*otrack);
e43c066c 957 ntracks[6]=1;
81e97e0d 958 for (Int_t i=0;i<7;i++) nindexes[i][0]=0;
ae00569a 959 Int_t modstatus = 1; // found
960 Float_t xloc,zloc;
e43c066c 961 //
962 //
963 // follow prolongations
e50912db 964 for (Int_t ilayer=5; ilayer>=0; ilayer--) {
4187a601 965 AliDebug(2,Form("FollowProlongationTree: layer %d",ilayer));
2755f080 966 fI = ilayer;
e43c066c 967 //
2755f080 968 AliITSlayer &layer=fgLayers[ilayer];
969 Double_t r = layer.GetR();
e43c066c 970 ntracks[ilayer]=0;
971 //
972 //
2755f080 973 Int_t nskipped=0;
e43c066c 974 Float_t nused =0;
e50912db 975 for (Int_t itrack =0; itrack<ntracks[ilayer+1]; itrack++) {
e43c066c 976 //set current track
977 if (ntracks[ilayer]>=100) break;
b9671574 978 if (tracks[ilayer+1][nindexes[ilayer+1][itrack]].GetNSkipped()>0) nskipped++;
979 if (tracks[ilayer+1][nindexes[ilayer+1][itrack]].GetNUsed()>2.) nused++;
e43c066c 980 if (ntracks[ilayer]>15+ilayer){
b9671574 981 if (itrack>1&&tracks[ilayer+1][nindexes[ilayer+1][itrack]].GetNSkipped()>0 && nskipped>4+ilayer) continue;
982 if (itrack>1&&tracks[ilayer+1][nindexes[ilayer+1][itrack]].GetNUsed()>2. && nused>3) continue;
e43c066c 983 }
984
15dd636f 985 new(&currenttrack1) AliITStrackMI(tracks[ilayer+1][nindexes[ilayer+1][itrack]]);
44347160 986
afd25725 987 // material between SSD and SDD, SDD and SPD
e50912db 988 if (ilayer==3)
989 if(!CorrectForShieldMaterial(&currenttrack1,"SDD","inward")) continue;
990 if (ilayer==1)
991 if(!CorrectForShieldMaterial(&currenttrack1,"SPD","inward")) continue;
8602c008 992
2755f080 993 // detector number
8602c008 994 Double_t phi,z;
995 if (!currenttrack1.GetPhiZat(r,phi,z)) continue;
e43c066c 996 Int_t idet=layer.FindDetectorIndex(phi,z);
e50912db 997
998 Double_t trackGlobXYZ1[3];
d1181f10 999 if (!currenttrack1.GetXYZ(trackGlobXYZ1)) continue;
e50912db 1000
2755f080 1001 // Get the budget to the primary vertex for the current track being prolonged
1002 Double_t budgetToPrimVertex = GetEffectiveThickness();
1003
1004 // check if we allow a prolongation without point
6518a6c5 1005 Int_t skip = CheckSkipLayer(&currenttrack1,ilayer,idet);
2755f080 1006 if (skip) {
1007 AliITStrackMI* vtrack = new (&tracks[ilayer][ntracks[ilayer]]) AliITStrackMI(currenttrack1);
6518a6c5 1008 // propagate to the layer radius
d1181f10 1009 Double_t xToGo; if (!vtrack->GetLocalXat(r,xToGo)) continue;
9be1d1c7 1010 if(!vtrack->Propagate(xToGo)) continue;
2755f080 1011 // apply correction for material of the current layer
1012 CorrectForLayerMaterial(vtrack,ilayer,trackGlobXYZ1,"inward");
1013 vtrack->SetNDeadZone(vtrack->GetNDeadZone()+1);
f8720bda 1014 vtrack->SetDeadZoneProbability(ilayer,1.); // no penalty for missing cluster
9fd412a8 1015 vtrack->SetClIndex(ilayer,-1);
ae00569a 1016 modstatus = (skip==1 ? 3 : 4); // skipped : out in z
d1181f10 1017 if(LocalModuleCoord(ilayer,idet,vtrack,xloc,zloc)) { // local module coords
1018 vtrack->SetModuleIndexInfo(ilayer,idet,modstatus,xloc,zloc);
1019 }
2755f080 1020 if(constrain) vtrack->Improve(budgetToPrimVertex,xyzVtx,ersVtx);
1021 ntracks[ilayer]++;
6518a6c5 1022 continue;
2755f080 1023 }
1024
1025 // track outside layer acceptance in z
1026 if (idet<0) continue;
1027
1028 //propagate to the intersection with the detector plane
1029 const AliITSdetector &det=layer.GetDetector(idet);
1030 new(&currenttrack2) AliITStrackMI(currenttrack1);
44347160 1031 if (!currenttrack1.Propagate(det.GetPhi(),det.GetR())) continue;
d1181f10 1032 if (!currenttrack2.Propagate(det.GetPhi(),det.GetR())) continue;
e43c066c 1033 currenttrack1.SetDetectorIndex(idet);
1034 currenttrack2.SetDetectorIndex(idet);
d1181f10 1035 if(!LocalModuleCoord(ilayer,idet,&currenttrack1,xloc,zloc)) continue; // local module coords
6518a6c5 1036
afd25725 1037 //***************
1c97ce2f 1038 // DEFINITION OF SEARCH ROAD AND CLUSTERS SELECTION
e43c066c 1039 //
2755f080 1040 // road in global (rphi,z) [i.e. in tracking ref. system]
1c97ce2f 1041 Double_t zmin,zmax,ymin,ymax;
1042 if (!ComputeRoad(&currenttrack1,ilayer,idet,zmin,zmax,ymin,ymax)) continue;
2755f080 1043
afd25725 1044 // select clusters in road
e43c066c 1045 layer.SelectClusters(zmin,zmax,ymin,ymax);
afd25725 1046 //********************
44347160 1047
afd25725 1048 // Define criteria for track-cluster association
44347160 1049 Double_t msz = currenttrack1.GetSigmaZ2() +
1050 AliITSReconstructor::GetRecoParam()->GetNSigmaZLayerForRoadZ()*
1051 AliITSReconstructor::GetRecoParam()->GetNSigmaZLayerForRoadZ()*
1052 AliITSReconstructor::GetRecoParam()->GetSigmaZ2(ilayer);
1053 Double_t msy = currenttrack1.GetSigmaY2() +
1054 AliITSReconstructor::GetRecoParam()->GetNSigmaYLayerForRoadY()*
1055 AliITSReconstructor::GetRecoParam()->GetNSigmaYLayerForRoadY()*
1056 AliITSReconstructor::GetRecoParam()->GetSigmaY2(ilayer);
1057 if (constrain) {
1058 msz *= AliITSReconstructor::GetRecoParam()->GetNSigma2RoadZC();
1059 msy *= AliITSReconstructor::GetRecoParam()->GetNSigma2RoadYC();
1060 } else {
1061 msz *= AliITSReconstructor::GetRecoParam()->GetNSigma2RoadZNonC();
1062 msy *= AliITSReconstructor::GetRecoParam()->GetNSigma2RoadYNonC();
1063 }
1064 msz = 1./msz; // 1/RoadZ^2
1065 msy = 1./msy; // 1/RoadY^2
1c97ce2f 1066
e43c066c 1067 //
e43c066c 1068 //
afd25725 1069 // LOOP OVER ALL POSSIBLE TRACK PROLONGATIONS ON THIS LAYER
e43c066c 1070 //
afd25725 1071 const AliITSRecPoint *cl=0;
1072 Int_t clidx=-1;
1073 Double_t chi2trkcl=AliITSReconstructor::GetRecoParam()->GetMaxChi2(); // init with big value
6518a6c5 1074 Bool_t deadzoneSPD=kFALSE;
e43c066c 1075 currenttrack = &currenttrack1;
6518a6c5 1076
ae00569a 1077 // check if the road contains a dead zone
23197852 1078 Bool_t noClusters = kFALSE;
1079 if (!layer.GetNextCluster(clidx,kTRUE)) noClusters=kTRUE;
4187a601 1080 if (noClusters) AliDebug(2,"no clusters in road");
1c97ce2f 1081 Double_t dz=0.5*(zmax-zmin);
1082 Double_t dy=0.5*(ymax-ymin);
1083 Int_t dead = CheckDeadZone(&currenttrack1,ilayer,idet,dz,dy,noClusters);
4187a601 1084 if(dead) AliDebug(2,Form("DEAD (%d)\n",dead));
ae00569a 1085 // create a prolongation without clusters (check also if there are no clusters in the road)
1086 if (dead ||
23197852 1087 (noClusters &&
ae00569a 1088 AliITSReconstructor::GetRecoParam()->GetAllowProlongationWithEmptyRoad())) {
6518a6c5 1089 AliITStrackMI * updatetrack = new (&tracks[ilayer][ntracks[ilayer]]) AliITStrackMI(*currenttrack);
9fd412a8 1090 updatetrack->SetClIndex(ilayer,-1);
ae00569a 1091 if (dead==0) {
1092 modstatus = 5; // no cls in road
1093 } else if (dead==1) {
1094 modstatus = 7; // holes in z in SPD
f8720bda 1095 } else if (dead==2 || dead==3 || dead==4) {
ae00569a 1096 modstatus = 2; // dead from OCDB
1097 }
1098 updatetrack->SetModuleIndexInfo(ilayer,idet,modstatus,xloc,zloc);
6518a6c5 1099 // apply correction for material of the current layer
1100 CorrectForLayerMaterial(updatetrack,ilayer,trackGlobXYZ1,"inward");
1101 if (constrain) { // apply vertex constrain
1102 updatetrack->SetConstrain(constrain);
1103 Bool_t isPrim = kTRUE;
1104 if (ilayer<4) { // check that it's close to the vertex
1105 updatetrack->GetDZ(GetX(),GetY(),GetZ(),updatetrack->GetDP()); //I.B.
1106 if (TMath::Abs(updatetrack->GetD(0)/(1.+ilayer)) > // y
1107 AliITSReconstructor::GetRecoParam()->GetMaxDZforPrimTrk() ||
1108 TMath::Abs(updatetrack->GetD(1)/(1.+ilayer)) > // z
1109 AliITSReconstructor::GetRecoParam()->GetMaxDZforPrimTrk()) isPrim=kFALSE;
1110 }
1111 if (isPrim) updatetrack->Improve(budgetToPrimVertex,xyzVtx,ersVtx);
1112 }
f8720bda 1113 updatetrack->SetNDeadZone(updatetrack->GetNDeadZone()+1);
ae00569a 1114 if (dead) {
ae00569a 1115 if (dead==1) { // dead zone at z=0,+-7cm in SPD
f8720bda 1116 updatetrack->SetDeadZoneProbability(ilayer,GetSPDDeadZoneProbability(updatetrack->GetZ(),TMath::Sqrt(updatetrack->GetSigmaZ2())));
ae00569a 1117 deadzoneSPD=kTRUE;
f8720bda 1118 } else if (dead==2 || dead==3) { // dead module or chip from OCDB
1119 updatetrack->SetDeadZoneProbability(ilayer,1.);
1120 } else if (dead==4) { // at least a single dead channel from OCDB
1121 updatetrack->SetDeadZoneProbability(ilayer,0.);
1122 }
6518a6c5 1123 }
1124 ntracks[ilayer]++;
1125 }
1126
ae00569a 1127 clidx=-1;
6518a6c5 1128 // loop over clusters in the road
afd25725 1129 while ((cl=layer.GetNextCluster(clidx))!=0) {
e43c066c 1130 if (ntracks[ilayer]>95) break; //space for skipped clusters
afd25725 1131 Bool_t changedet =kFALSE;
6518a6c5 1132 if (cl->GetQ()==0 && deadzoneSPD==kTRUE) continue;
a5d0fae6 1133 Int_t idetc=cl->GetDetectorIndex();
afd25725 1134
a5d0fae6 1135 if (currenttrack->GetDetectorIndex()==idetc) { // track already on the cluster's detector
1c97ce2f 1136 // take into account misalignment (bring track to real detector plane)
1137 Double_t xTrOrig = currenttrack->GetX();
9be1d1c7 1138 if (!currenttrack->Propagate(xTrOrig+cl->GetX())) continue;
afd25725 1139 // a first cut on track-cluster distance
1140 if ( (currenttrack->GetZ()-cl->GetZ())*(currenttrack->GetZ()-cl->GetZ())*msz +
1141 (currenttrack->GetY()-cl->GetY())*(currenttrack->GetY()-cl->GetY())*msy > 1. )
1c97ce2f 1142 { // cluster not associated to track
4187a601 1143 AliDebug(2,"not associated");
1c97ce2f 1144 continue;
1145 }
1146 // bring track back to ideal detector plane
9be1d1c7 1147 if (!currenttrack->Propagate(xTrOrig)) continue;
afd25725 1148 } else { // have to move track to cluster's detector
a5d0fae6 1149 const AliITSdetector &detc=layer.GetDetector(idetc);
afd25725 1150 // a first cut on track-cluster distance
a5d0fae6 1151 Double_t y;
1c97ce2f 1152 if (!currenttrack2.GetProlongationFast(detc.GetPhi(),detc.GetR()+cl->GetX(),y,z)) continue;
afd25725 1153 if ( (z-cl->GetZ())*(z-cl->GetZ())*msz +
1154 (y-cl->GetY())*(y-cl->GetY())*msy > 1. )
44347160 1155 continue; // cluster not associated to track
e43c066c 1156 //
15dd636f 1157 new (&backuptrack) AliITStrackMI(currenttrack2);
afd25725 1158 changedet = kTRUE;
e43c066c 1159 currenttrack =&currenttrack2;
a5d0fae6 1160 if (!currenttrack->Propagate(detc.GetPhi(),detc.GetR())) {
15dd636f 1161 new (currenttrack) AliITStrackMI(backuptrack);
afd25725 1162 changedet = kFALSE;
e43c066c 1163 continue;
1164 }
a5d0fae6 1165 currenttrack->SetDetectorIndex(idetc);
e50912db 1166 // Get again the budget to the primary vertex
1167 // for the current track being prolonged, if had to change detector
1168 //budgetToPrimVertex = GetEffectiveThickness();// not needed at the moment because anyway we take a mean material for this correction
e43c066c 1169 }
1170
afd25725 1171 // calculate track-clusters chi2
1172 chi2trkcl = GetPredictedChi2MI(currenttrack,cl,ilayer);
1173 // chi2 cut
4187a601 1174 AliDebug(2,Form("chi2 %f max %f",chi2trkcl,AliITSReconstructor::GetRecoParam()->GetMaxChi2s(ilayer)));
afd25725 1175 if (chi2trkcl < AliITSReconstructor::GetRecoParam()->GetMaxChi2s(ilayer)) {
6518a6c5 1176 if (cl->GetQ()==0) deadzoneSPD=kTRUE; // only 1 prolongation with virtual cluster
e43c066c 1177 if (ntracks[ilayer]>=100) continue;
15dd636f 1178 AliITStrackMI * updatetrack = new (&tracks[ilayer][ntracks[ilayer]]) AliITStrackMI(*currenttrack);
9fd412a8 1179 updatetrack->SetClIndex(ilayer,-1);
afd25725 1180 if (changedet) new (&currenttrack2) AliITStrackMI(backuptrack);
ae00569a 1181
6518a6c5 1182 if (cl->GetQ()!=0) { // real cluster
1c97ce2f 1183 if (!UpdateMI(updatetrack,cl,chi2trkcl,(ilayer<<28)+clidx)) {
4187a601 1184 AliDebug(2,"update failed");
1c97ce2f 1185 continue;
1186 }
09cf9d66 1187 updatetrack->SetSampledEdx(cl->GetQ(),ilayer-2);
ae00569a 1188 modstatus = 1; // found
6518a6c5 1189 } else { // virtual cluster in dead zone
b9671574 1190 updatetrack->SetNDeadZone(updatetrack->GetNDeadZone()+1);
f8720bda 1191 updatetrack->SetDeadZoneProbability(ilayer,GetSPDDeadZoneProbability(updatetrack->GetZ(),TMath::Sqrt(updatetrack->GetSigmaZ2())));
ae00569a 1192 modstatus = 7; // holes in z in SPD
1193 }
1194
1195 if (changedet) {
1196 Float_t xlocnewdet,zlocnewdet;
d1181f10 1197 if(LocalModuleCoord(ilayer,idet,updatetrack,xlocnewdet,zlocnewdet)) { // local module coords
1198 updatetrack->SetModuleIndexInfo(ilayer,idet,modstatus,xlocnewdet,zlocnewdet);
1199 }
ae00569a 1200 } else {
1201 updatetrack->SetModuleIndexInfo(ilayer,idet,modstatus,xloc,zloc);
e43c066c 1202 }
afd25725 1203 if (cl->IsUsed()) updatetrack->IncrementNUsed();
1204
1205 // apply correction for material of the current layer
e50912db 1206 CorrectForLayerMaterial(updatetrack,ilayer,trackGlobXYZ1,"inward");
1207
afd25725 1208 if (constrain) { // apply vertex constrain
b9671574 1209 updatetrack->SetConstrain(constrain);
e43c066c 1210 Bool_t isPrim = kTRUE;
afd25725 1211 if (ilayer<4) { // check that it's close to the vertex
b9671574 1212 updatetrack->GetDZ(GetX(),GetY(),GetZ(),updatetrack->GetDP()); //I.B.
afd25725 1213 if (TMath::Abs(updatetrack->GetD(0)/(1.+ilayer)) > // y
1214 AliITSReconstructor::GetRecoParam()->GetMaxDZforPrimTrk() ||
1215 TMath::Abs(updatetrack->GetD(1)/(1.+ilayer)) > // z
1216 AliITSReconstructor::GetRecoParam()->GetMaxDZforPrimTrk()) isPrim=kFALSE;
e43c066c 1217 }
afd25725 1218 if (isPrim) updatetrack->Improve(budgetToPrimVertex,xyzVtx,ersVtx);
e43c066c 1219 } //apply vertex constrain
1220 ntracks[ilayer]++;
afd25725 1221 } // create new hypothesis
4187a601 1222 else {
1223 AliDebug(2,"chi2 too large");
1224 }
1225
2755f080 1226 } // loop over possible prolongations
1227
1228 // allow one prolongation without clusters
6518a6c5 1229 if (constrain && itrack<=1 && currenttrack1.GetNSkipped()==0 && deadzoneSPD==kFALSE && ntracks[ilayer]<100) {
15dd636f 1230 AliITStrackMI* vtrack = new (&tracks[ilayer][ntracks[ilayer]]) AliITStrackMI(currenttrack1);
2755f080 1231 // apply correction for material of the current layer
1232 CorrectForLayerMaterial(vtrack,ilayer,trackGlobXYZ1,"inward");
9fd412a8 1233 vtrack->SetClIndex(ilayer,-1);
ae00569a 1234 modstatus = 3; // skipped
1235 vtrack->SetModuleIndexInfo(ilayer,idet,modstatus,xloc,zloc);
afd25725 1236 vtrack->Improve(budgetToPrimVertex,xyzVtx,ersVtx);
b9671574 1237 vtrack->IncrementNSkipped();
e43c066c 1238 ntracks[ilayer]++;
1239 }
628e7bb0 1240
2755f080 1241 // allow one prolongation without clusters for tracks with |tgl|>1.1
1242 if (constrain && itrack==0 && TMath::Abs(currenttrack1.GetTgl())>1.1) { //big theta - for low flux
628e7bb0 1243 AliITStrackMI* vtrack = new (&tracks[ilayer][ntracks[ilayer]]) AliITStrackMI(currenttrack1);
2755f080 1244 // apply correction for material of the current layer
1245 CorrectForLayerMaterial(vtrack,ilayer,trackGlobXYZ1,"inward");
9fd412a8 1246 vtrack->SetClIndex(ilayer,-1);
ae00569a 1247 modstatus = 3; // skipped
1248 vtrack->SetModuleIndexInfo(ilayer,idet,modstatus,xloc,zloc);
afd25725 1249 vtrack->Improve(budgetToPrimVertex,xyzVtx,ersVtx);
b9671574 1250 vtrack->SetNDeadZone(vtrack->GetNDeadZone()+1);
628e7bb0 1251 ntracks[ilayer]++;
1252 }
1253
e43c066c 1254
e50912db 1255 } // loop over tracks in layer ilayer+1
afd25725 1256
1257 //loop over track candidates for the current layer
e43c066c 1258 //
1259 //
1260 Int_t accepted=0;
1261
afd25725 1262 Int_t golden=0;
e43c066c 1263 for (Int_t itrack=0;itrack<ntracks[ilayer];itrack++){
1264 normalizedchi2[itrack] = NormalizedChi2(&tracks[ilayer][itrack],ilayer);
afd25725 1265 if (normalizedchi2[itrack] <
1266 AliITSReconstructor::GetRecoParam()->GetMaxNormChi2ForGolden(ilayer)) golden++;
44347160 1267 if (ilayer>4) {
1268 accepted++;
1269 } else {
1270 if (constrain) { // constrain
afd25725 1271 if (normalizedchi2[itrack]<AliITSReconstructor::GetRecoParam()->GetMaxNormChi2C(ilayer)+1)
1272 accepted++;
44347160 1273 } else { // !constrain
afd25725 1274 if (normalizedchi2[itrack]<AliITSReconstructor::GetRecoParam()->GetMaxNormChi2NonC(ilayer)+1)
1275 accepted++;
44347160 1276 }
e43c066c 1277 }
1278 }
afd25725 1279 // sort tracks by increasing normalized chi2
1280 TMath::Sort(ntracks[ilayer],normalizedchi2,nindexes[ilayer],kFALSE);
e43c066c 1281 ntracks[ilayer] = TMath::Min(accepted,7+2*ilayer);
afd25725 1282 if (ntracks[ilayer]<golden+2+ilayer) ntracks[ilayer]=TMath::Min(golden+2+ilayer,accepted);
e43c066c 1283 if (ntracks[ilayer]>90) ntracks[ilayer]=90;
afd25725 1284 } // end loop over layers
1285
afd25725 1286
1287 //
1288 // Now select tracks to be kept
1289 //
44347160 1290 Int_t max = constrain ? 20 : 5;
e43c066c 1291
afd25725 1292 // tracks that reach layer 0 (SPD inner)
44347160 1293 for (Int_t i=0; i<TMath::Min(max,ntracks[0]); i++) {
15dd636f 1294 AliITStrackMI & track= tracks[0][nindexes[0][i]];
628e7bb0 1295 if (track.GetNumberOfClusters()<2) continue;
afd25725 1296 if (!constrain && track.GetNormChi2(0) >
6518a6c5 1297 AliITSReconstructor::GetRecoParam()->GetMaxNormChi2NonCForHypothesis()) {
1298 continue;
1299 }
15dd636f 1300 AddTrackHypothesys(new AliITStrackMI(track), esdindex);
e43c066c 1301 }
afd25725 1302
1303 // tracks that reach layer 1 (SPD outer)
628e7bb0 1304 for (Int_t i=0;i<TMath::Min(2,ntracks[1]);i++) {
15dd636f 1305 AliITStrackMI & track= tracks[1][nindexes[1][i]];
e43c066c 1306 if (track.GetNumberOfClusters()<4) continue;
afd25725 1307 if (!constrain && track.GetNormChi2(1) >
1308 AliITSReconstructor::GetRecoParam()->GetMaxNormChi2NonCForHypothesis()) continue;
b9671574 1309 if (constrain) track.IncrementNSkipped();
81e97e0d 1310 if (!constrain) {
b9671574 1311 track.SetD(0,track.GetD(GetX(),GetY()));
afd25725 1312 track.SetNSkipped(track.GetNSkipped()+4./(4.+8.*TMath::Abs(track.GetD(0))));
b9671574 1313 if (track.GetNumberOfClusters()+track.GetNDeadZone()+track.GetNSkipped()>6) {
1314 track.SetNSkipped(6-track.GetNumberOfClusters()+track.GetNDeadZone());
e43c066c 1315 }
1316 }
15dd636f 1317 AddTrackHypothesys(new AliITStrackMI(track), esdindex);
e43c066c 1318 }
afd25725 1319
6518a6c5 1320 // tracks that reach layer 2 (SDD inner), only during non-constrained pass
81e97e0d 1321 if (!constrain){
628e7bb0 1322 for (Int_t i=0;i<TMath::Min(2,ntracks[2]);i++) {
15dd636f 1323 AliITStrackMI & track= tracks[2][nindexes[2][i]];
628e7bb0 1324 if (track.GetNumberOfClusters()<3) continue;
afd25725 1325 if (!constrain && track.GetNormChi2(2) >
1326 AliITSReconstructor::GetRecoParam()->GetMaxNormChi2NonCForHypothesis()) continue;
b9671574 1327 if (constrain) track.SetNSkipped(track.GetNSkipped()+2);
81e97e0d 1328 if (!constrain){
b9671574 1329 track.SetD(0,track.GetD(GetX(),GetY()));
afd25725 1330 track.SetNSkipped(track.GetNSkipped()+7./(7.+8.*TMath::Abs(track.GetD(0))));
b9671574 1331 if (track.GetNumberOfClusters()+track.GetNDeadZone()+track.GetNSkipped()>6) {
1332 track.SetNSkipped(6-track.GetNumberOfClusters()+track.GetNDeadZone());
e43c066c 1333 }
1334 }
15dd636f 1335 AddTrackHypothesys(new AliITStrackMI(track), esdindex);
e43c066c 1336 }
1337 }
81e97e0d 1338
afd25725 1339 if (!constrain) {
81e97e0d 1340 //
afd25725 1341 // register best track of each layer - important for V0 finder
81e97e0d 1342 //
1343 for (Int_t ilayer=0;ilayer<5;ilayer++){
1344 if (ntracks[ilayer]==0) continue;
1345 AliITStrackMI & track= tracks[ilayer][nindexes[ilayer][0]];
1346 if (track.GetNumberOfClusters()<1) continue;
1347 CookLabel(&track,0);
1348 bestarray->AddAt(new AliITStrackMI(track),ilayer);
1349 }
1350 }
1351 //
1352 // update TPC V0 information
1353 //
b9671574 1354 if (otrack->GetESDtrack()->GetV0Index(0)>0){
81e97e0d 1355 Float_t fprimvertex[3]={GetX(),GetY(),GetZ()};
1356 for (Int_t i=0;i<3;i++){
b9671574 1357 Int_t index = otrack->GetESDtrack()->GetV0Index(i);
81e97e0d 1358 if (index==0) break;
44347160 1359 AliV0 *vertex = (AliV0*)fEsd->GetV0(index);
81e97e0d 1360 if (vertex->GetStatus()<0) continue; // rejected V0
1361 //
6c94f330 1362 if (otrack->GetSign()>0) {
81e97e0d 1363 vertex->SetIndex(0,esdindex);
1364 }
1365 else{
1366 vertex->SetIndex(1,esdindex);
1367 }
1368 //find nearest layer with track info
b75d63a7 1369 Double_t xrp[3]; vertex->GetXYZ(xrp[0],xrp[1],xrp[2]); //I.B.
1370 Int_t nearestold = GetNearestLayer(xrp); //I.B.
81e97e0d 1371 Int_t nearest = nearestold;
1372 for (Int_t ilayer =nearest;ilayer<8;ilayer++){
1373 if (ntracks[nearest]==0){
1374 nearest = ilayer;
1375 }
1376 }
1377 //
1378 AliITStrackMI & track= tracks[nearest][nindexes[nearest][0]];
1379 if (nearestold<5&&nearest<5){
b9671574 1380 Bool_t accept = track.GetNormChi2(nearest)<10;
81e97e0d 1381 if (accept){
6c94f330 1382 if (track.GetSign()>0) {
b75d63a7 1383 vertex->SetParamP(track);
81e97e0d 1384 vertex->Update(fprimvertex);
44347160 1385 //vertex->SetIndex(0,track.fESDtrack->GetID());
81e97e0d 1386 if (track.GetNumberOfClusters()>2) AddTrackHypothesys(new AliITStrackMI(track), esdindex);
1387 }else{
b75d63a7 1388 vertex->SetParamN(track);
81e97e0d 1389 vertex->Update(fprimvertex);
1390 //vertex->SetIndex(1,track.fESDtrack->GetID());
1391 if (track.GetNumberOfClusters()>2) AddTrackHypothesys(new AliITStrackMI(track), esdindex);
1392 }
1393 vertex->SetStatus(vertex->GetStatus()+1);
1394 }else{
44347160 1395 //vertex->SetStatus(-2); // reject V0 - not enough clusters
81e97e0d 1396 }
1397 }
81e97e0d 1398 }
afd25725 1399 }
1400
e43c066c 1401}
44347160 1402//------------------------------------------------------------------------
e43c066c 1403AliITStrackerMI::AliITSlayer & AliITStrackerMI::GetLayer(Int_t layer) const
1404{
1405 //--------------------------------------------------------------------
1406 //
1407 //
1408 return fgLayers[layer];
1409}
44347160 1410//------------------------------------------------------------------------
8221b41b 1411AliITStrackerMI::AliITSlayer::AliITSlayer():
1412fR(0),
1413fPhiOffset(0),
1414fNladders(0),
1415fZOffset(0),
1416fNdetectors(0),
1417fDetectors(0),
1418fN(0),
1419fDy5(0),
1420fDy10(0),
1421fDy20(0),
1422fClustersCs(0),
1423fClusterIndexCs(0),
1424fYcs(0),
1425fZcs(0),
1426fNcs(0),
1427fCurrentSlice(-1),
d9ead1a0 1428fZmin(0),
8221b41b 1429fZmax(0),
1430fYmin(0),
1431fYmax(0),
1432fI(0),
1433fImax(0),
1434fSkip(0),
1435fAccepted(0),
d9ead1a0 1436fRoad(0),
1437fMaxSigmaClY(0),
1438fMaxSigmaClZ(0),
1439fNMaxSigmaCl(3)
1440{
e43c066c 1441 //--------------------------------------------------------------------
1442 //default AliITSlayer constructor
1443 //--------------------------------------------------------------------
e50912db 1444 for (Int_t i=0; i<AliITSRecoParam::GetMaxClusterPerLayer(); i++) {
e43c066c 1445 fClusterWeight[i]=0;
1446 fClusterTracks[0][i]=-1;
1447 fClusterTracks[1][i]=-1;
1448 fClusterTracks[2][i]=-1;
1449 fClusterTracks[3][i]=-1;
1450 }
1451}
44347160 1452//------------------------------------------------------------------------
e43c066c 1453AliITStrackerMI::AliITSlayer::
8221b41b 1454AliITSlayer(Double_t r,Double_t p,Double_t z,Int_t nl,Int_t nd):
1455fR(r),
1456fPhiOffset(p),
1457fNladders(nl),
1458fZOffset(z),
1459fNdetectors(nd),
1460fDetectors(0),
1461fN(0),
1462fDy5(0),
1463fDy10(0),
1464fDy20(0),
1465fClustersCs(0),
1466fClusterIndexCs(0),
1467fYcs(0),
1468fZcs(0),
1469fNcs(0),
1470fCurrentSlice(-1),
d9ead1a0 1471fZmin(0),
8221b41b 1472fZmax(0),
1473fYmin(0),
1474fYmax(0),
1475fI(0),
1476fImax(0),
1477fSkip(0),
1478fAccepted(0),
d9ead1a0 1479fRoad(0),
1480fMaxSigmaClY(0),
1481fMaxSigmaClZ(0),
1482fNMaxSigmaCl(3) {
e43c066c 1483 //--------------------------------------------------------------------
1484 //main AliITSlayer constructor
1485 //--------------------------------------------------------------------
e43c066c 1486 fDetectors=new AliITSdetector[fNladders*fNdetectors];
afd25725 1487 fRoad=2*fR*TMath::Sqrt(TMath::Pi()/1.);//assuming that there's only one cluster
e43c066c 1488}
44347160 1489//------------------------------------------------------------------------
8221b41b 1490AliITStrackerMI::AliITSlayer::AliITSlayer(const AliITSlayer& layer):
1491fR(layer.fR),
1492fPhiOffset(layer.fPhiOffset),
1493fNladders(layer.fNladders),
1494fZOffset(layer.fZOffset),
1495fNdetectors(layer.fNdetectors),
1496fDetectors(layer.fDetectors),
1497fN(layer.fN),
1498fDy5(layer.fDy5),
1499fDy10(layer.fDy10),
1500fDy20(layer.fDy20),
1501fClustersCs(layer.fClustersCs),
1502fClusterIndexCs(layer.fClusterIndexCs),
1503fYcs(layer.fYcs),
1504fZcs(layer.fZcs),
1505fNcs(layer.fNcs),
1506fCurrentSlice(layer.fCurrentSlice),
d9ead1a0 1507fZmin(layer.fZmin),
8221b41b 1508fZmax(layer.fZmax),
1509fYmin(layer.fYmin),
1510fYmax(layer.fYmax),
1511fI(layer.fI),
1512fImax(layer.fImax),
1513fSkip(layer.fSkip),
1514fAccepted(layer.fAccepted),
d9ead1a0 1515fRoad(layer.fRoad),
1516fMaxSigmaClY(layer.fMaxSigmaClY),
1517fMaxSigmaClZ(layer.fMaxSigmaClZ),
1518fNMaxSigmaCl(layer.fNMaxSigmaCl)
1519{
8221b41b 1520 //Copy constructor
1521}
44347160 1522//------------------------------------------------------------------------
e43c066c 1523AliITStrackerMI::AliITSlayer::~AliITSlayer() {
1524 //--------------------------------------------------------------------
1525 // AliITSlayer destructor
1526 //--------------------------------------------------------------------
23197852 1527 delete [] fDetectors;
e43c066c 1528 for (Int_t i=0; i<fN; i++) delete fClusters[i];
e50912db 1529 for (Int_t i=0; i<AliITSRecoParam::GetMaxClusterPerLayer(); i++) {
e43c066c 1530 fClusterWeight[i]=0;
1531 fClusterTracks[0][i]=-1;
1532 fClusterTracks[1][i]=-1;
1533 fClusterTracks[2][i]=-1;
1534 fClusterTracks[3][i]=-1;
1535 }
1536}
44347160 1537//------------------------------------------------------------------------
e43c066c 1538void AliITStrackerMI::AliITSlayer::ResetClusters() {
1539 //--------------------------------------------------------------------
1540 // This function removes loaded clusters
1541 //--------------------------------------------------------------------
1542 for (Int_t i=0; i<fN; i++) delete fClusters[i];
e50912db 1543 for (Int_t i=0; i<AliITSRecoParam::GetMaxClusterPerLayer(); i++){
e43c066c 1544 fClusterWeight[i]=0;
1545 fClusterTracks[0][i]=-1;
1546 fClusterTracks[1][i]=-1;
1547 fClusterTracks[2][i]=-1;
1548 fClusterTracks[3][i]=-1;
1549 }
1550
1551 fN=0;
1552 fI=0;
1553}
44347160 1554//------------------------------------------------------------------------
e43c066c 1555void AliITStrackerMI::AliITSlayer::ResetWeights() {
1556 //--------------------------------------------------------------------
1557 // This function reset weights of the clusters
1558 //--------------------------------------------------------------------
e50912db 1559 for (Int_t i=0; i<AliITSRecoParam::GetMaxClusterPerLayer(); i++) {
e43c066c 1560 fClusterWeight[i]=0;
1561 fClusterTracks[0][i]=-1;
1562 fClusterTracks[1][i]=-1;
1563 fClusterTracks[2][i]=-1;
1564 fClusterTracks[3][i]=-1;
1565 }
1566 for (Int_t i=0; i<fN;i++) {
00a7cc50 1567 AliITSRecPoint * cl = (AliITSRecPoint*)GetCluster(i);
e43c066c 1568 if (cl&&cl->IsUsed()) cl->Use();
1569 }
1570
1571}
44347160 1572//------------------------------------------------------------------------
e43c066c 1573void AliITStrackerMI::AliITSlayer::ResetRoad() {
1574 //--------------------------------------------------------------------
1575 // This function calculates the road defined by the cluster density
1576 //--------------------------------------------------------------------
1577 Int_t n=0;
1578 for (Int_t i=0; i<fN; i++) {
1579 if (TMath::Abs(fClusters[i]->GetZ())<fR) n++;
1580 }
afd25725 1581 if (n>1) fRoad=2*fR*TMath::Sqrt(TMath::Pi()/n);
e43c066c 1582}
44347160 1583//------------------------------------------------------------------------
afd25725 1584Int_t AliITStrackerMI::AliITSlayer::InsertCluster(AliITSRecPoint *cl) {
e43c066c 1585 //--------------------------------------------------------------------
1586 //This function adds a cluster to this layer
1587 //--------------------------------------------------------------------
e50912db 1588 if (fN==AliITSRecoParam::GetMaxClusterPerLayer()) {
e43c066c 1589 ::Error("InsertCluster","Too many clusters !\n");
1590 return 1;
1591 }
1592 fCurrentSlice=-1;
afd25725 1593 fClusters[fN]=cl;
1d4090b7 1594 fN++;
afd25725 1595 AliITSdetector &det=GetDetector(cl->GetDetectorIndex());
d9ead1a0 1596 //AD
1597 Double_t nSigmaY=fNMaxSigmaCl*TMath::Sqrt(cl->GetSigmaY2());
1598 Double_t nSigmaZ=fNMaxSigmaCl*TMath::Sqrt(cl->GetSigmaZ2());
1599 if (cl->GetY()-nSigmaY<det.GetYmin()) det.SetYmin(cl->GetY()-nSigmaY);
1600 if (cl->GetY()+nSigmaY>det.GetYmax()) det.SetYmax(cl->GetY()+nSigmaY);
1601 if (cl->GetZ()-nSigmaZ<det.GetZmin()) det.SetZmin(cl->GetZ()-nSigmaZ);
1602 if (cl->GetZ()+nSigmaZ>det.GetZmax()) det.SetZmax(cl->GetZ()+nSigmaZ);
1603 //AD
1604 /*
afd25725 1605 if (cl->GetY()<det.GetYmin()) det.SetYmin(cl->GetY());
1606 if (cl->GetY()>det.GetYmax()) det.SetYmax(cl->GetY());
1607 if (cl->GetZ()<det.GetZmin()) det.SetZmin(cl->GetZ());
1608 if (cl->GetZ()>det.GetZmax()) det.SetZmax(cl->GetZ());
d9ead1a0 1609 */
e43c066c 1610 return 0;
1611}
44347160 1612//------------------------------------------------------------------------
e43c066c 1613void AliITStrackerMI::AliITSlayer::SortClusters()
1614{
1615 //
1616 //sort clusters
1617 //
00a7cc50 1618 AliITSRecPoint **clusters = new AliITSRecPoint*[fN];
1d4090b7 1619 Float_t *z = new Float_t[fN];
1620 Int_t * index = new Int_t[fN];
1621 //
d9ead1a0 1622 fMaxSigmaClY=0.; //AD
1623 fMaxSigmaClZ=0.; //AD
1624
1d4090b7 1625 for (Int_t i=0;i<fN;i++){
1626 z[i] = fClusters[i]->GetZ();
d9ead1a0 1627 // save largest errors in y and z for this layer
1628 fMaxSigmaClY=TMath::Max(fMaxSigmaClY,TMath::Sqrt(fClusters[i]->GetSigmaY2()));
1629 fMaxSigmaClZ=TMath::Max(fMaxSigmaClZ,TMath::Sqrt(fClusters[i]->GetSigmaZ2()));
1d4090b7 1630 }
1631 TMath::Sort(fN,z,index,kFALSE);
1632 for (Int_t i=0;i<fN;i++){
1633 clusters[i] = fClusters[index[i]];
1634 }
1635 //
1636 for (Int_t i=0;i<fN;i++){
1637 fClusters[i] = clusters[i];
1638 fZ[i] = fClusters[i]->GetZ();
1639 AliITSdetector &det=GetDetector(fClusters[i]->GetDetectorIndex());
1640 Double_t y=fR*det.GetPhi() + fClusters[i]->GetY();
1641 if (y>2.*fR*TMath::Pi()) y -= 2.*fR*TMath::Pi();
1642 fY[i] = y;
1643 }
1644 delete[] index;
1645 delete[] z;
1646 delete[] clusters;
1647 //
1648
e43c066c 1649 fYB[0]=10000000;
1650 fYB[1]=-10000000;
1651 for (Int_t i=0;i<fN;i++){
1652 if (fY[i]<fYB[0]) fYB[0]=fY[i];
1653 if (fY[i]>fYB[1]) fYB[1]=fY[i];
1654 fClusterIndex[i] = i;
1655 }
1656 //
1657 // fill slices
1658 fDy5 = (fYB[1]-fYB[0])/5.;
1659 fDy10 = (fYB[1]-fYB[0])/10.;
1660 fDy20 = (fYB[1]-fYB[0])/20.;
1661 for (Int_t i=0;i<6;i++) fN5[i] =0;
1662 for (Int_t i=0;i<11;i++) fN10[i]=0;
1663 for (Int_t i=0;i<21;i++) fN20[i]=0;
1664 //
1665 for (Int_t i=0;i<6;i++) {fBy5[i][0] = fYB[0]+(i-0.75)*fDy5; fBy5[i][1] = fYB[0]+(i+0.75)*fDy5;}
1666 for (Int_t i=0;i<11;i++) {fBy10[i][0] = fYB[0]+(i-0.75)*fDy10; fBy10[i][1] = fYB[0]+(i+0.75)*fDy10;}
1667 for (Int_t i=0;i<21;i++) {fBy20[i][0] = fYB[0]+(i-0.75)*fDy20; fBy20[i][1] = fYB[0]+(i+0.75)*fDy20;}
1668 //
1669 //
1d4090b7 1670 for (Int_t i=0;i<fN;i++)
1671 for (Int_t irot=-1;irot<=1;irot++){
1672 Float_t curY = fY[i]+irot*TMath::TwoPi()*fR;
1673 // slice 5
1674 for (Int_t slice=0; slice<6;slice++){
e50912db 1675 if (fBy5[slice][0]<curY && curY<fBy5[slice][1]&&fN5[slice]<AliITSRecoParam::GetMaxClusterPerLayer5()){
1d4090b7 1676 fClusters5[slice][fN5[slice]] = fClusters[i];
1677 fY5[slice][fN5[slice]] = curY;
1678 fZ5[slice][fN5[slice]] = fZ[i];
1679 fClusterIndex5[slice][fN5[slice]]=i;
1680 fN5[slice]++;
1681 }
e43c066c 1682 }
1d4090b7 1683 // slice 10
1684 for (Int_t slice=0; slice<11;slice++){
e50912db 1685 if (fBy10[slice][0]<curY && curY<fBy10[slice][1]&&fN10[slice]<AliITSRecoParam::GetMaxClusterPerLayer10()){
1d4090b7 1686 fClusters10[slice][fN10[slice]] = fClusters[i];
1687 fY10[slice][fN10[slice]] = curY;
1688 fZ10[slice][fN10[slice]] = fZ[i];
1689 fClusterIndex10[slice][fN10[slice]]=i;
1690 fN10[slice]++;
1691 }
e43c066c 1692 }
1d4090b7 1693 // slice 20
1694 for (Int_t slice=0; slice<21;slice++){
e50912db 1695 if (fBy20[slice][0]<curY && curY<fBy20[slice][1]&&fN20[slice]<AliITSRecoParam::GetMaxClusterPerLayer20()){
1d4090b7 1696 fClusters20[slice][fN20[slice]] = fClusters[i];
1697 fY20[slice][fN20[slice]] = curY;
1698 fZ20[slice][fN20[slice]] = fZ[i];
1699 fClusterIndex20[slice][fN20[slice]]=i;
1700 fN20[slice]++;
1701 }
1702 }
e43c066c 1703 }
1d4090b7 1704
1705 //
1706 // consistency check
1707 //
1708 for (Int_t i=0;i<fN-1;i++){
1709 if (fZ[i]>fZ[i+1]){
ae00569a 1710 printf("Bug\n");
e43c066c 1711 }
1712 }
1d4090b7 1713 //
1714 for (Int_t slice=0;slice<21;slice++)
1715 for (Int_t i=0;i<fN20[slice]-1;i++){
1716 if (fZ20[slice][i]>fZ20[slice][i+1]){
ae00569a 1717 printf("Bug\n");
1d4090b7 1718 }
1719 }
1720
1721
e43c066c 1722}
44347160 1723//------------------------------------------------------------------------
e43c066c 1724Int_t AliITStrackerMI::AliITSlayer::FindClusterIndex(Float_t z) const {
1725 //--------------------------------------------------------------------
1726 // This function returns the index of the nearest cluster
1727 //--------------------------------------------------------------------
1728 Int_t ncl=0;
1729 const Float_t *zcl;
1730 if (fCurrentSlice<0) {
1731 ncl = fN;
1732 zcl = fZ;
1733 }
1734 else{
1735 ncl = fNcs;
1736 zcl = fZcs;;
1737 }
1738
1739 if (ncl==0) return 0;
1740 Int_t b=0, e=ncl-1, m=(b+e)/2;
1741 for (; b<e; m=(b+e)/2) {
1742 // if (z > fClusters[m]->GetZ()) b=m+1;
1743 if (z > zcl[m]) b=m+1;
1744 else e=m;
1745 }
1746 return m;
1747}
44347160 1748//------------------------------------------------------------------------
1c97ce2f 1749Bool_t AliITStrackerMI::ComputeRoad(AliITStrackMI* track,Int_t ilayer,Int_t idet,Double_t &zmin,Double_t &zmax,Double_t &ymin,Double_t &ymax) const {
1750 //--------------------------------------------------------------------
1751 // This function computes the rectangular road for this track
1752 //--------------------------------------------------------------------
1753
1754
1755 AliITSdetector &det = fgLayers[ilayer].GetDetector(idet);
1756 // take into account the misalignment: propagate track to misaligned detector plane
1757 if (!track->Propagate(det.GetPhi(),det.GetRmisal())) return kFALSE;
1758
1759 Double_t dz=AliITSReconstructor::GetRecoParam()->GetNSigmaRoadZ()*
1760 TMath::Sqrt(track->GetSigmaZ2() +
1761 AliITSReconstructor::GetRecoParam()->GetNSigmaZLayerForRoadZ()*
1762 AliITSReconstructor::GetRecoParam()->GetNSigmaZLayerForRoadZ()*
1763 AliITSReconstructor::GetRecoParam()->GetSigmaZ2(ilayer));
1764 Double_t dy=AliITSReconstructor::GetRecoParam()->GetNSigmaRoadY()*
1765 TMath::Sqrt(track->GetSigmaY2() +
1766 AliITSReconstructor::GetRecoParam()->GetNSigmaYLayerForRoadY()*
1767 AliITSReconstructor::GetRecoParam()->GetNSigmaYLayerForRoadY()*
1768 AliITSReconstructor::GetRecoParam()->GetSigmaY2(ilayer));
1769
1770 // track at boundary between detectors, enlarge road
1771 Double_t boundaryWidth=AliITSRecoParam::GetBoundaryWidth();
1772 if ( (track->GetY()-dy < det.GetYmin()+boundaryWidth) ||
1773 (track->GetY()+dy > det.GetYmax()-boundaryWidth) ||
1774 (track->GetZ()-dz < det.GetZmin()+boundaryWidth) ||
1775 (track->GetZ()+dz > det.GetZmax()-boundaryWidth) ) {
1776 Float_t tgl = TMath::Abs(track->GetTgl());
1777 if (tgl > 1.) tgl=1.;
1778 Double_t deltaXNeighbDets=AliITSRecoParam::GetDeltaXNeighbDets();
1779 dz = TMath::Sqrt(dz*dz+deltaXNeighbDets*deltaXNeighbDets*tgl*tgl);
1780 Float_t snp = TMath::Abs(track->GetSnp());
1781 if (snp > AliITSReconstructor::GetRecoParam()->GetMaxSnp()) return kFALSE;
1782 dy = TMath::Sqrt(dy*dy+deltaXNeighbDets*deltaXNeighbDets*snp*snp);
1783 } // boundary
1784
1785 // add to the road a term (up to 2-3 mm) to deal with misalignments
1786 dy = TMath::Sqrt(dy*dy + AliITSReconstructor::GetRecoParam()->GetRoadMisal()*AliITSReconstructor::GetRecoParam()->GetRoadMisal());
1787 dz = TMath::Sqrt(dz*dz + AliITSReconstructor::GetRecoParam()->GetRoadMisal()*AliITSReconstructor::GetRecoParam()->GetRoadMisal());
1788
1789 Double_t r = fgLayers[ilayer].GetR();
1790 zmin = track->GetZ() - dz;
1791 zmax = track->GetZ() + dz;
1792 ymin = track->GetY() + r*det.GetPhi() - dy;
1793 ymax = track->GetY() + r*det.GetPhi() + dy;
1794
1795 // bring track back to idead detector plane
1796 if (!track->Propagate(det.GetPhi(),det.GetR())) return kFALSE;
1797
1798 return kTRUE;
1799}
1800//------------------------------------------------------------------------
e43c066c 1801void AliITStrackerMI::AliITSlayer::
1802SelectClusters(Double_t zmin,Double_t zmax,Double_t ymin, Double_t ymax) {
1803 //--------------------------------------------------------------------
1804 // This function sets the "window"
1805 //--------------------------------------------------------------------
1806
628e7bb0 1807 Double_t circle=2*TMath::Pi()*fR;
d9ead1a0 1808 fYmin = ymin;
1809 fYmax = ymax;
1810 fZmin = zmin;
1811 fZmax = zmax;
1812 // AD
1813 // enlarge road in y by maximum cluster error on this layer (3 sigma)
1814 fYmin -= fNMaxSigmaCl*fMaxSigmaClY;
1815 fYmax += fNMaxSigmaCl*fMaxSigmaClY;
1816 fZmin -= fNMaxSigmaCl*fMaxSigmaClZ;
1817 fZmax += fNMaxSigmaCl*fMaxSigmaClZ;
1818
e43c066c 1819 Float_t ymiddle = (fYmin+fYmax)*0.5;
ae00569a 1820 if (ymiddle<fYB[0]) {
1821 fYmin+=circle; fYmax+=circle; ymiddle+=circle;
1822 } else if (ymiddle>fYB[1]) {
1823 fYmin-=circle; fYmax-=circle; ymiddle-=circle;
e43c066c 1824 }
ae00569a 1825
e43c066c 1826 //
1827 fCurrentSlice =-1;
1828 // defualt take all
1829 fClustersCs = fClusters;
1830 fClusterIndexCs = fClusterIndex;
1831 fYcs = fY;
1832 fZcs = fZ;
1833 fNcs = fN;
628e7bb0 1834 //
e43c066c 1835 //is in 20 slice?
1836 if (fCurrentSlice<0&&TMath::Abs(fYmax-fYmin)<1.49*fDy20){
1837 Int_t slice = int(0.5+(ymiddle-fYB[0])/fDy20);
1838 if (slice<0) slice=0;
1839 if (slice>20) slice=20;
1840 Bool_t isOK = (fYmin>fBy20[slice][0]&&fYmax<fBy20[slice][1]);
1841 if (isOK) {
1842 fCurrentSlice=slice;
1843 fClustersCs = fClusters20[fCurrentSlice];
1844 fClusterIndexCs = fClusterIndex20[fCurrentSlice];
1845 fYcs = fY20[fCurrentSlice];
1846 fZcs = fZ20[fCurrentSlice];
1847 fNcs = fN20[fCurrentSlice];
1848 }
1849 }
1850 //
1851 //is in 10 slice?
1852 if (fCurrentSlice<0&&TMath::Abs(fYmax-fYmin)<1.49*fDy10){
1853 Int_t slice = int(0.5+(ymiddle-fYB[0])/fDy10);
1854 if (slice<0) slice=0;
1855 if (slice>10) slice=10;
1856 Bool_t isOK = (fYmin>fBy10[slice][0]&&fYmax<fBy10[slice][1]);
1857 if (isOK) {
1858 fCurrentSlice=slice;
1859 fClustersCs = fClusters10[fCurrentSlice];
1860 fClusterIndexCs = fClusterIndex10[fCurrentSlice];
1861 fYcs = fY10[fCurrentSlice];
1862 fZcs = fZ10[fCurrentSlice];
1863 fNcs = fN10[fCurrentSlice];
1864 }
1865 }
1866 //
1867 //is in 5 slice?
1868 if (fCurrentSlice<0&&TMath::Abs(fYmax-fYmin)<1.49*fDy5){
1869 Int_t slice = int(0.5+(ymiddle-fYB[0])/fDy5);
1870 if (slice<0) slice=0;
1871 if (slice>5) slice=5;
1872 Bool_t isOK = (fYmin>fBy5[slice][0]&&fYmax<fBy5[slice][1]);
ae00569a 1873 if (isOK) {
e43c066c 1874 fCurrentSlice=slice;
1875 fClustersCs = fClusters5[fCurrentSlice];
1876 fClusterIndexCs = fClusterIndex5[fCurrentSlice];
1877 fYcs = fY5[fCurrentSlice];
1878 fZcs = fZ5[fCurrentSlice];
1879 fNcs = fN5[fCurrentSlice];
1880 }
1881 }
628e7bb0 1882 //
d9ead1a0 1883 fI = FindClusterIndex(fZmin);
1884 fImax = TMath::Min(FindClusterIndex(fZmax)+1,fNcs);
1885 fSkip = 0;
1886 fAccepted = 0;
6518a6c5 1887
1888 return;
e43c066c 1889}
44347160 1890//------------------------------------------------------------------------
628e7bb0 1891Int_t AliITStrackerMI::AliITSlayer::
1892FindDetectorIndex(Double_t phi, Double_t z) const {
1893 //--------------------------------------------------------------------
1894 //This function finds the detector crossed by the track
1895 //--------------------------------------------------------------------
31fb8575 1896 Double_t dphi;
1897 if (fZOffset<0) // old geometry
1898 dphi = -(phi-fPhiOffset);
1899 else // new geometry
1900 dphi = phi-fPhiOffset;
1901
6518a6c5 1902
628e7bb0 1903 if (dphi < 0) dphi += 2*TMath::Pi();
1904 else if (dphi >= 2*TMath::Pi()) dphi -= 2*TMath::Pi();
1905 Int_t np=Int_t(dphi*fNladders*0.5/TMath::Pi()+0.5);
1906 if (np>=fNladders) np-=fNladders;
1907 if (np<0) np+=fNladders;
1908
6518a6c5 1909
628e7bb0 1910 Double_t dz=fZOffset-z;
6518a6c5 1911 Double_t nnz = dz*(fNdetectors-1)*0.5/fZOffset+0.5;
1912 Int_t nz = (nnz<0 ? -1 : (Int_t)nnz);
628e7bb0 1913 if (nz>=fNdetectors) return -1;
1914 if (nz<0) return -1;
1915
23197852 1916 // ad hoc correction for 3rd ladder of SDD inner layer,
1917 // which is reversed (rotated by pi around local y)
1918 // this correction is OK only from AliITSv11Hybrid onwards
1919 if (GetR()>12. && GetR()<20.) { // SDD inner
1920 if(np==2) { // 3rd ladder
1921 nz = (fNdetectors-1) - nz;
1922 }
1923 }
1924 //printf("ndet %d phi %f z %f np %d nz %d\n",fNdetectors,phi,z,np,nz);
1925
1926
628e7bb0 1927 return np*fNdetectors + nz;
1928}
44347160 1929//------------------------------------------------------------------------
ae00569a 1930const AliITSRecPoint *AliITStrackerMI::AliITSlayer::GetNextCluster(Int_t &ci,Bool_t test)
1931{
e43c066c 1932 //--------------------------------------------------------------------
1933 // This function returns clusters within the "window"
1934 //--------------------------------------------------------------------
1935
ae00569a 1936 if (fCurrentSlice<0) {
e43c066c 1937 Double_t rpi2 = 2.*fR*TMath::Pi();
1938 for (Int_t i=fI; i<fImax; i++) {
1939 Double_t y = fY[i];
d9ead1a0 1940 Double_t z = fZ[i];
e43c066c 1941 if (fYmax<y) y -= rpi2;
1d4090b7 1942 if (fYmin>y) y += rpi2;
e43c066c 1943 if (y<fYmin) continue;
1944 if (y>fYmax) continue;
d9ead1a0 1945 // AD
1946 // skip clusters that are in "extended" road but they
1947 // 3sigma error does not touch the original road
1948 if (z+fNMaxSigmaCl*TMath::Sqrt(fClusters[i]->GetSigmaZ2())<fZmin+fNMaxSigmaCl*fMaxSigmaClZ) continue;
1949 if (z-fNMaxSigmaCl*TMath::Sqrt(fClusters[i]->GetSigmaZ2())>fZmax-fNMaxSigmaCl*fMaxSigmaClZ) continue;
1950 //
e43c066c 1951 if (fClusters[i]->GetQ()==0&&fSkip==2) continue;
1952 ci=i;
ae00569a 1953 if (!test) fI=i+1;
e43c066c 1954 return fClusters[i];
1955 }
ae00569a 1956 } else {
e43c066c 1957 for (Int_t i=fI; i<fImax; i++) {
1958 if (fYcs[i]<fYmin) continue;
1959 if (fYcs[i]>fYmax) continue;
1960 if (fClustersCs[i]->GetQ()==0&&fSkip==2) continue;
1961 ci=fClusterIndexCs[i];
ae00569a 1962 if (!test) fI=i+1;
e43c066c 1963 return fClustersCs[i];
1964 }
1965 }
1966 return 0;
1967}
44347160 1968//------------------------------------------------------------------------
e43c066c 1969Double_t AliITStrackerMI::AliITSlayer::GetThickness(Double_t y,Double_t z,Double_t &x0)
1970const {
1971 //--------------------------------------------------------------------
afd25725 1972 // This function returns the layer thickness at this point (units X0)
e43c066c 1973 //--------------------------------------------------------------------
1974 Double_t d=0.0085;
e50912db 1975 x0=AliITSRecoParam::GetX0Air();
e43c066c 1976 if (43<fR&&fR<45) { //SSD2
1977 Double_t dd=0.0034;
1978 d=dd;
1979 if (TMath::Abs(y-0.00)>3.40) d+=dd;
1980 if (TMath::Abs(y-1.90)<0.45) {d+=(0.013-0.0034);}
1981 if (TMath::Abs(y+1.90)<0.45) {d+=(0.013-0.0034);}
1982 for (Int_t i=0; i<12; i++) {
1983 if (TMath::Abs(z-3.9*(i+0.5))<0.15) {
1984 if (TMath::Abs(y-0.00)>3.40) d+=dd;
1985 d+=0.0034;
1986 break;
1987 }
1988 if (TMath::Abs(z+3.9*(i+0.5))<0.15) {
1989 if (TMath::Abs(y-0.00)>3.40) d+=dd;
1990 d+=0.0034;
1991 break;
1992 }
1993 if (TMath::Abs(z-3.4-3.9*i)<0.50) {d+=(0.016-0.0034); break;}
1994 if (TMath::Abs(z+0.5+3.9*i)<0.50) {d+=(0.016-0.0034); break;}
1995 }
1996 } else
1997 if (37<fR&&fR<41) { //SSD1
1998 Double_t dd=0.0034;
1999 d=dd;
2000 if (TMath::Abs(y-0.00)>3.40) d+=dd;
2001 if (TMath::Abs(y-1.90)<0.45) {d+=(0.013-0.0034);}
2002 if (TMath::Abs(y+1.90)<0.45) {d+=(0.013-0.0034);}
2003 for (Int_t i=0; i<11; i++) {
2004 if (TMath::Abs(z-3.9*i)<0.15) {
2005 if (TMath::Abs(y-0.00)>3.40) d+=dd;
2006 d+=dd;
2007 break;
2008 }
2009 if (TMath::Abs(z+3.9*i)<0.15) {
2010 if (TMath::Abs(y-0.00)>3.40) d+=dd;
2011 d+=dd;
2012 break;
2013 }
2014 if (TMath::Abs(z-1.85-3.9*i)<0.50) {d+=(0.016-0.0034); break;}
2015 if (TMath::Abs(z+2.05+3.9*i)<0.50) {d+=(0.016-0.0034); break;}
2016 }
2017 } else
2018 if (13<fR&&fR<26) { //SDD
2019 Double_t dd=0.0033;
2020 d=dd;
2021 if (TMath::Abs(y-0.00)>3.30) d+=dd;
2022
2023 if (TMath::Abs(y-1.80)<0.55) {
2024 d+=0.016;
2025 for (Int_t j=0; j<20; j++) {
2026 if (TMath::Abs(z+0.7+1.47*j)<0.12) {d+=0.08; x0=9.; break;}
2027 if (TMath::Abs(z-0.7-1.47*j)<0.12) {d+=0.08; x0=9.; break;}
2028 }
2029 }
2030 if (TMath::Abs(y+1.80)<0.55) {
2031 d+=0.016;
2032 for (Int_t j=0; j<20; j++) {
2033 if (TMath::Abs(z-0.7-1.47*j)<0.12) {d+=0.08; x0=9.; break;}
2034 if (TMath::Abs(z+0.7+1.47*j)<0.12) {d+=0.08; x0=9.; break;}
2035 }
2036 }
2037
2038 for (Int_t i=0; i<4; i++) {
2039 if (TMath::Abs(z-7.3*i)<0.60) {
2040 d+=dd;
2041 if (TMath::Abs(y-0.00)>3.30) d+=dd;
2042 break;
2043 }
2044 if (TMath::Abs(z+7.3*i)<0.60) {
2045 d+=dd;
2046 if (TMath::Abs(y-0.00)>3.30) d+=dd;
2047 break;
2048 }
2049 }
2050 } else
2051 if (6<fR&&fR<8) { //SPD2
2052 Double_t dd=0.0063; x0=21.5;
2053 d=dd;
2054 if (TMath::Abs(y-3.08)>0.5) d+=dd;
afd25725 2055 if (TMath::Abs(y-3.03)<0.10) d+=0.014;
e43c066c 2056 } else
2057 if (3<fR&&fR<5) { //SPD1
2058 Double_t dd=0.0063; x0=21.5;
2059 d=dd;
2060 if (TMath::Abs(y+0.21)>0.6) d+=dd;
afd25725 2061 if (TMath::Abs(y+0.10)<0.10) d+=0.014;
e43c066c 2062 }
2063
2064 return d;
2065}
44347160 2066//------------------------------------------------------------------------
23197852 2067AliITStrackerMI::AliITSdetector::AliITSdetector(const AliITSdetector& det):
2068fR(det.fR),
1c97ce2f 2069fRmisal(det.fRmisal),
23197852 2070fPhi(det.fPhi),
2071fSinPhi(det.fSinPhi),
2072fCosPhi(det.fCosPhi),
2073fYmin(det.fYmin),
2074fYmax(det.fYmax),
2075fZmin(det.fZmin),
2076fZmax(det.fZmax),
2077fIsBad(det.fIsBad),
2078fNChips(det.fNChips),
2079fChipIsBad(det.fChipIsBad)
2080{
2081 //Copy constructor
2082}
2083//------------------------------------------------------------------------
2084void AliITStrackerMI::AliITSdetector::ReadBadDetectorAndChips(Int_t ilayer,Int_t idet,
a70ed6ad 2085 const AliITSDetTypeRec *detTypeRec)
23197852 2086{
2087 //--------------------------------------------------------------------
2088 // Read bad detectors and chips from calibration objects in AliITSDetTypeRec
2089 //--------------------------------------------------------------------
2090
2091 // In AliITSDetTypeRec, detector numbers go from 0 to 2197
2092 // while in the tracker they start from 0 for each layer
2093 for(Int_t il=0; il<ilayer; il++)
2094 idet += AliITSgeomTGeo::GetNLadders(il+1)*AliITSgeomTGeo::GetNDetectors(il+1);
2095
2096 Int_t detType;
2097 if (ilayer==0 || ilayer==1) { // ---------- SPD
2098 detType = 0;
2099 } else if (ilayer==2 || ilayer==3) { // ---------- SDD
2100 detType = 1;
2101 } else if (ilayer==4 || ilayer==5) { // ---------- SSD
2102 detType = 2;
2103 } else {
2104 printf("AliITStrackerMI::AliITSdetector::InitBadFromOCDB: Wrong layer number %d\n",ilayer);
2105 return;
2106 }
2107
2108 // Get calibration from AliITSDetTypeRec
2109 AliITSCalibration *calib = (AliITSCalibration*)detTypeRec->GetCalibrationModel(idet);
d1181f10 2110 calib->SetModuleIndex(idet);
23197852 2111 AliITSCalibration *calibSPDdead = 0;
2112 if(detType==0) calibSPDdead = (AliITSCalibration*)detTypeRec->GetSPDDeadModel(idet); // TEMPORARY
2113 if (calib->IsBad() ||
2114 (detType==0 && calibSPDdead->IsBad())) // TEMPORARY
2115 {
2116 SetBad();
359ac74d 2117 // printf("lay %d bad %d\n",ilayer,idet);
23197852 2118 }
2119
2120 // Get segmentation from AliITSDetTypeRec
2121 AliITSsegmentation *segm = (AliITSsegmentation*)detTypeRec->GetSegmentationModel(detType);
2122
2123 // Read info about bad chips
2124 fNChips = segm->GetMaximumChipIndex()+1;
2125 //printf("ilayer %d detType %d idet %d fNChips %d %d GetNumberOfChips %d\n",ilayer,detType,idet,fNChips,segm->GetMaximumChipIndex(),segm->GetNumberOfChips());
2126 if(fChipIsBad) { delete [] fChipIsBad; fChipIsBad=NULL; }
2127 fChipIsBad = new Bool_t[fNChips];
2128 for (Int_t iCh=0;iCh<fNChips;iCh++) {
2129 fChipIsBad[iCh] = calib->IsChipBad(iCh);
2130 if (detType==0 && calibSPDdead->IsChipBad(iCh)) fChipIsBad[iCh] = kTRUE; // TEMPORARY
4187a601 2131 //if(fChipIsBad[iCh]) {printf("lay %d det %d bad chip %d\n",ilayer,idet,iCh);}
23197852 2132 }
2133
2134 return;
2135}
2136//------------------------------------------------------------------------
e50912db 2137Double_t AliITStrackerMI::GetEffectiveThickness()
e43c066c 2138{
2139 //--------------------------------------------------------------------
afd25725 2140 // Returns the thickness between the current layer and the vertex (units X0)
e43c066c 2141 //--------------------------------------------------------------------
e43c066c 2142
e50912db 2143 if(fUseTGeo!=0) {
2144 if(fxOverX0Layer[0]<0) BuildMaterialLUT("Layers");
2145 if(fxOverX0Shield[0]<0) BuildMaterialLUT("Shields");
2146 if(fxOverX0Pipe<0) BuildMaterialLUT("Pipe");
2147 }
2148
afd25725 2149 // beam pipe
e50912db 2150 Double_t dPipe = (fUseTGeo==0 ? AliITSRecoParam::GetdPipe() : fxOverX0Pipe);
2151 Double_t d=dPipe*AliITSRecoParam::GetrPipe()*AliITSRecoParam::GetrPipe();
afd25725 2152
2153 // layers
2154 Double_t x0=0;
e43c066c 2155 Double_t xn=fgLayers[fI].GetR();
2156 for (Int_t i=0; i<fI; i++) {
2157 Double_t xi=fgLayers[i].GetR();
e50912db 2158 Double_t dLayer = (fUseTGeo==0 ? fgLayers[i].GetThickness(0,0,x0) : fxOverX0Layer[i]);
2159 d+=dLayer*xi*xi;
e43c066c 2160 }
2161
afd25725 2162 // shields
e43c066c 2163 if (fI>1) {
e50912db 2164 Double_t dshieldSPD = (fUseTGeo==0 ? AliITSRecoParam::Getdshield(0) : fxOverX0Shield[0]);
2165 d+=dshieldSPD*AliITSRecoParam::GetrInsideShield(0)*AliITSRecoParam::GetrInsideShield(0);
e43c066c 2166 }
e43c066c 2167 if (fI>3) {
e50912db 2168 Double_t dshieldSDD = (fUseTGeo==0 ? AliITSRecoParam::Getdshield(1) : fxOverX0Shield[1]);
2169 d+=dshieldSDD*AliITSRecoParam::GetrInsideShield(1)*AliITSRecoParam::GetrInsideShield(1);
e43c066c 2170 }
e43c066c 2171 return d/(xn*xn);
2172}
44347160 2173//------------------------------------------------------------------------
e43c066c 2174Int_t AliITStrackerMI::AliITSlayer::InRoad() const {
ae00569a 2175 //-------------------------------------------------------------------
e43c066c 2176 // This function returns number of clusters within the "window"
2177 //--------------------------------------------------------------------
2178 Int_t ncl=0;
2179 for (Int_t i=fI; i<fN; i++) {
00a7cc50 2180 const AliITSRecPoint *c=fClusters[i];
e43c066c 2181 if (c->GetZ() > fZmax) break;
2182 if (c->IsUsed()) continue;
2183 const AliITSdetector &det=GetDetector(c->GetDetectorIndex());
2184 Double_t y=fR*det.GetPhi() + c->GetY();
2185
2186 if (y>2.*fR*TMath::Pi()) y -= 2*fR*TMath::Pi();
2187 if (y>1.*fR*TMath::Pi() && fYmax<y) y -= 2*fR*TMath::Pi();
2188
2189 if (y<fYmin) continue;
2190 if (y>fYmax) continue;
2191 ncl++;
2192 }
2193 return ncl;
2194}
44347160 2195//------------------------------------------------------------------------
ae00569a 2196Bool_t AliITStrackerMI::RefitAt(Double_t xx,AliITStrackMI *track,
7167ae53 2197 const AliITStrackMI *clusters,Bool_t extra, Bool_t planeeff)
ae00569a 2198{
e43c066c 2199 //--------------------------------------------------------------------
ae00569a 2200 // This function refits the track "track" at the position "x" using
2201 // the clusters from "clusters"
ef7253ac 2202 // If "extra"==kTRUE,
ae00569a 2203 // the clusters from overlapped modules get attached to "track"
7167ae53 2204 // If "planeff"==kTRUE,
2205 // special approach for plane efficiency evaluation is applyed
e43c066c 2206 //--------------------------------------------------------------------
6518a6c5 2207
e50912db 2208 Int_t index[AliITSgeomTGeo::kNLayers];
e43c066c 2209 Int_t k;
e50912db 2210 for (k=0; k<AliITSgeomTGeo::GetNLayers(); k++) index[k]=-1;
ae00569a 2211 Int_t nc=clusters->GetNumberOfClusters();
e43c066c 2212 for (k=0; k<nc; k++) {
ae00569a 2213 Int_t idx=clusters->GetClusterIndex(k);
2214 Int_t ilayer=(idx&0xf0000000)>>28;
2215 index[ilayer]=idx;
e43c066c 2216 }
67c1e979 2217
7167ae53 2218 return RefitAt(xx,track,index,extra,planeeff); // call the method below
67c1e979 2219}
44347160 2220//------------------------------------------------------------------------
ae00569a 2221Bool_t AliITStrackerMI::RefitAt(Double_t xx,AliITStrackMI *track,
7167ae53 2222 const Int_t *clusters,Bool_t extra, Bool_t planeeff)
ae00569a 2223{
67c1e979 2224 //--------------------------------------------------------------------
ae00569a 2225 // This function refits the track "track" at the position "x" using
67c1e979 2226 // the clusters from array
ae00569a 2227 // If "extra"==kTRUE,
2228 // the clusters from overlapped modules get attached to "track"
7167ae53 2229 // If "planeff"==kTRUE,
2230 // special approach for plane efficiency evaluation is applyed
67c1e979 2231 //--------------------------------------------------------------------
e50912db 2232 Int_t index[AliITSgeomTGeo::kNLayers];
67c1e979 2233 Int_t k;
e50912db 2234 for (k=0; k<AliITSgeomTGeo::GetNLayers(); k++) index[k]=-1;
67c1e979 2235 //
e50912db 2236 for (k=0; k<AliITSgeomTGeo::GetNLayers(); k++) {
ae00569a 2237 index[k]=clusters[k];
67c1e979 2238 }
2239
afd25725 2240 // special for cosmics: check which the innermost layer crossed
2241 // by the track
2242 Int_t innermostlayer=5;
ae00569a 2243 Double_t drphi = TMath::Abs(track->GetD(0.,0.));
2244 for(innermostlayer=0; innermostlayer<AliITSgeomTGeo::GetNLayers(); innermostlayer++) {
2245 if(drphi < fgLayers[innermostlayer].GetR()) break;
2246 }
4fa7f7d1 2247 AliDebug(2,Form(" drphi %f innermost %d",drphi,innermostlayer));
afd25725 2248
ae00569a 2249 Int_t modstatus=1; // found
2250 Float_t xloc,zloc;
67c1e979 2251 Int_t from, to, step;
ae00569a 2252 if (xx > track->GetX()) {
e50912db 2253 from=innermostlayer; to=AliITSgeomTGeo::GetNLayers();
67c1e979 2254 step=+1;
2255 } else {
e50912db 2256 from=AliITSgeomTGeo::GetNLayers()-1; to=innermostlayer-1;
67c1e979 2257 step=-1;
2258 }
ae00569a 2259 TString dir = (step>0 ? "outward" : "inward");
67c1e979 2260
ae00569a 2261 for (Int_t ilayer = from; ilayer != to; ilayer += step) {
6518a6c5 2262 AliITSlayer &layer=fgLayers[ilayer];
67c1e979 2263 Double_t r=layer.GetR();
4fa7f7d1 2264
afd25725 2265 if (step<0 && xx>r) break;
2266
2267 // material between SSD and SDD, SDD and SPD
6518a6c5 2268 Double_t hI=ilayer-0.5*step;
e50912db 2269 if (TMath::Abs(hI-3.5)<0.01) // SDDouter
ae00569a 2270 if(!CorrectForShieldMaterial(track,"SDD",dir)) return kFALSE;
e50912db 2271 if (TMath::Abs(hI-1.5)<0.01) // SPDouter
ae00569a 2272 if(!CorrectForShieldMaterial(track,"SPD",dir)) return kFALSE;
67c1e979 2273
ae00569a 2274
afd25725 2275 Double_t oldGlobXYZ[3];
d1181f10 2276 if (!track->GetXYZ(oldGlobXYZ)) return kFALSE;
afd25725 2277
67178f34 2278 // continue if we are already beyond this layer
2279 Double_t oldGlobR = TMath::Sqrt(oldGlobXYZ[0]*oldGlobXYZ[0]+oldGlobXYZ[1]*oldGlobXYZ[1]);
2280 if(step>0 && oldGlobR > r) continue; // going outward
2281 if(step<0 && oldGlobR < r) continue; // going inward
2282
8602c008 2283 Double_t phi,z;
ae00569a 2284 if (!track->GetPhiZat(r,phi,z)) return kFALSE;
67c1e979 2285
67c1e979 2286 Int_t idet=layer.FindDetectorIndex(phi,z);
6518a6c5 2287
2288 // check if we allow a prolongation without point for large-eta tracks
ae00569a 2289 Int_t skip = CheckSkipLayer(track,ilayer,idet);
6518a6c5 2290 if (skip==2) {
ae00569a 2291 modstatus = 4; // out in z
d1181f10 2292 if(LocalModuleCoord(ilayer,idet,track,xloc,zloc)) { // local module coords
2293 track->SetModuleIndexInfo(ilayer,idet,modstatus,xloc,zloc);
2294 }
7131593b 2295 // cross layer
2296 // apply correction for material of the current layer
2297 // add time if going outward
2298 CorrectForLayerMaterial(track,ilayer,oldGlobXYZ,dir);
6518a6c5 2299 continue;
2300 }
2301
e50912db 2302 if (idet<0) return kFALSE;
ae00569a 2303
4fa7f7d1 2304
67c1e979 2305 const AliITSdetector &det=layer.GetDetector(idet);
1c97ce2f 2306 if (!track->Propagate(det.GetPhi(),det.GetR())) return kFALSE;
23197852 2307
ae00569a 2308 track->SetDetectorIndex(idet);
d1181f10 2309 if(!LocalModuleCoord(ilayer,idet,track,xloc,zloc)) return kFALSE; // local module coords
67c1e979 2310
23197852 2311 Double_t dz,zmin,zmax,dy,ymin,ymax;
ae00569a 2312
2313 const AliITSRecPoint *clAcc=0;
44347160 2314 Double_t maxchi2=1000.*AliITSReconstructor::GetRecoParam()->GetMaxChi2();
67c1e979 2315
6518a6c5 2316 Int_t idx=index[ilayer];
ae00569a 2317 if (idx>=0) { // cluster in this layer
2318 modstatus = 6; // no refit
2319 const AliITSRecPoint *cl=(AliITSRecPoint *)GetCluster(idx);
2320 if (cl) {
2321 if (idet != cl->GetDetectorIndex()) {
2322 idet=cl->GetDetectorIndex();
a5d0fae6 2323 const AliITSdetector &detc=layer.GetDetector(idet);
2324 if (!track->Propagate(detc.GetPhi(),detc.GetR())) return kFALSE;
ae00569a 2325 track->SetDetectorIndex(idet);
d1181f10 2326 if(!LocalModuleCoord(ilayer,idet,track,xloc,zloc)) return kFALSE; // local module coords
ae00569a 2327 }
ae00569a 2328 Int_t cllayer = (idx & 0xf0000000) >> 28;;
2329 Double_t chi2=GetPredictedChi2MI(track,cl,cllayer);
2330 if (chi2<maxchi2) {
2331 clAcc=cl;
2332 maxchi2=chi2;
2333 modstatus = 1; // found
2334 } else {
7167ae53 2335 return kFALSE; //
ae00569a 2336 }
2337 }
2338 } else { // no cluster in this layer
2339 if (skip==1) {
2340 modstatus = 3; // skipped
0ed58a47 2341 // Plane Eff determination:
2342 if (planeeff && ilayer==AliITSReconstructor::GetRecoParam()->GetIPlanePlaneEff()) {
2343 if (IsOKForPlaneEff(track,clusters,ilayer)) // only adequate track for plane eff. evaluation
2344 UseTrackForPlaneEff(track,ilayer);
7167ae53 2345 }
ae00569a 2346 } else {
2347 modstatus = 5; // no cls in road
2348 // check dead
1c97ce2f 2349 if (!ComputeRoad(track,ilayer,idet,zmin,zmax,ymin,ymax)) return kFALSE;
2350 dz = 0.5*(zmax-zmin);
2351 dy = 0.5*(ymax-ymin);
2352 Int_t dead = CheckDeadZone(track,ilayer,idet,dz,dy,kTRUE);
ae00569a 2353 if (dead==1) modstatus = 7; // holes in z in SPD
f8720bda 2354 if (dead==2 || dead==3 || dead==4) modstatus = 2; // dead from OCDB
ae00569a 2355 }
e43c066c 2356 }
ae00569a 2357
2358 if (clAcc) {
2359 if (!UpdateMI(track,clAcc,maxchi2,idx)) return kFALSE;
09cf9d66 2360 track->SetSampledEdx(clAcc->GetQ(),ilayer-2);
e43c066c 2361 }
ae00569a 2362 track->SetModuleIndexInfo(ilayer,idet,modstatus,xloc,zloc);
2363
2364
2365 if (extra) { // search for extra clusters in overlapped modules
2366 AliITStrackV2 tmp(*track);
1c97ce2f 2367 if (!ComputeRoad(track,ilayer,idet,zmin,zmax,ymin,ymax)) return kFALSE;
2368 layer.SelectClusters(zmin,zmax,ymin,ymax);
ae00569a 2369
2370 const AliITSRecPoint *clExtra=0; Int_t ci=-1,cci=-1;
2371 Int_t idetExtra=-1;
a5d0fae6 2372 maxchi2=1000.*AliITSReconstructor::GetRecoParam()->GetMaxChi2();
2373 Double_t tolerance=0.1;
ae00569a 2374 while ((clExtra=layer.GetNextCluster(ci))!=0) {
2375 // only clusters in another module! (overlaps)
2376 idetExtra = clExtra->GetDetectorIndex();
2377 if (idet == idetExtra) continue;
2378
a5d0fae6 2379 const AliITSdetector &detx=layer.GetDetector(idetExtra);
ae00569a 2380
1c97ce2f 2381 if (!tmp.Propagate(detx.GetPhi(),detx.GetR()+clExtra->GetX())) continue;
ae00569a 2382 if (TMath::Abs(tmp.GetZ() - clExtra->GetZ()) > tolerance) continue;
2383 if (TMath::Abs(tmp.GetY() - clExtra->GetY()) > tolerance) continue;
1c97ce2f 2384 if (!tmp.Propagate(detx.GetPhi(),detx.GetR())) continue;
ae00569a 2385
2386 Double_t chi2=tmp.GetPredictedChi2(clExtra);
2387 if (chi2<maxchi2) { maxchi2=chi2; cci=ci; }
2388 }
2389 if (cci>=0) {
2390 track->SetExtraCluster(ilayer,(ilayer<<28)+cci);
2391 track->SetExtraModule(ilayer,idetExtra);
2392 }
2393 } // end search for extra clusters in overlapped modules
1c97ce2f 2394
afd25725 2395 // Correct for material of the current layer
7131593b 2396 // cross material
2397 // add time if going outward
ae00569a 2398 if(!CorrectForLayerMaterial(track,ilayer,oldGlobXYZ,dir)) return kFALSE;
e43c066c 2399
ae00569a 2400 } // end loop on layers
2401
db355ee7 2402 if (!track->PropagateTo(xx,0.,0.)) return kFALSE;
e43c066c 2403
e43c066c 2404 return kTRUE;
2405}
44347160 2406//------------------------------------------------------------------------
15dd636f 2407Double_t AliITStrackerMI::GetNormalizedChi2(AliITStrackMI * track, Int_t mode)
e43c066c 2408{
2409 //
2410 // calculate normalized chi2
2411 // return NormalizedChi2(track,0);
2412 Float_t chi2 = 0;
2413 Float_t sum=0;
2414 Float_t *erry = GetErrY(fCurrentEsdTrack), *errz = GetErrZ(fCurrentEsdTrack);
2415 // track->fdEdxMismatch=0;
2416 Float_t dedxmismatch =0;
2417 Float_t *ny = GetNy(fCurrentEsdTrack), *nz = GetNz(fCurrentEsdTrack);
2418 if (mode<100){
2419 for (Int_t i = 0;i<6;i++){
9fd412a8 2420 if (track->GetClIndex(i)>=0){
e43c066c 2421 Float_t cerry, cerrz;
2422 if (ny[i]>0) {cerry = erry[i]; cerrz=errz[i];}
2423 else
b9671574 2424 { cerry= track->GetSigmaY(i); cerrz = track->GetSigmaZ(i);}
e43c066c 2425 cerry*=cerry;
2426 cerrz*=cerrz;
b9671574 2427 Float_t cchi2 = (track->GetDy(i)*track->GetDy(i)/cerry)+(track->GetDz(i)*track->GetDz(i)/cerrz);
572f41f9 2428 if (i>1 && AliITSReconstructor::GetRecoParam()->GetUseAmplitudeInfo(i)) {
b9671574 2429 Float_t ratio = track->GetNormQ(i)/track->GetExpQ();
e43c066c 2430 if (ratio<0.5) {
2431 cchi2+=(0.5-ratio)*10.;
2432 //track->fdEdxMismatch+=(0.5-ratio)*10.;
2433 dedxmismatch+=(0.5-ratio)*10.;
2434 }
2435 }
2436 if (i<2 ||i>3){
b9671574 2437 AliITSRecPoint * cl = (AliITSRecPoint*)GetCluster( track->GetClIndex(i));
e43c066c 2438 Double_t delta = cl->GetNy()+cl->GetNz()-ny[i]-nz[i];
2439 if (delta>1) chi2 +=0.5*TMath::Min(delta/2,2.);
2440 if (i<2) chi2+=2*cl->GetDeltaProbability();
2441 }
2442 chi2+=cchi2;
2443 sum++;
2444 }
2445 }
b9671574 2446 if (TMath::Abs(track->GetdEdxMismatch()-dedxmismatch)>0.0001){
2447 track->SetdEdxMismatch(dedxmismatch);
e43c066c 2448 }
2449 }
2450 else{
2451 for (Int_t i = 0;i<4;i++){
9fd412a8 2452 if (track->GetClIndex(i)>=0){
e43c066c 2453 Float_t cerry, cerrz;
2454 if (ny[i]>0) {cerry = erry[i]; cerrz=errz[i];}
b9671574 2455 else { cerry= track->GetSigmaY(i); cerrz = track->GetSigmaZ(i);}
e43c066c 2456 cerry*=cerry;
2457 cerrz*=cerrz;
b9671574 2458 chi2+= (track->GetDy(i)*track->GetDy(i)/cerry);
2459 chi2+= (track->GetDz(i)*track->GetDz(i)/cerrz);
e43c066c 2460 sum++;
2461 }
2462 }
2463 for (Int_t i = 4;i<6;i++){
9fd412a8 2464 if (track->GetClIndex(i)>=0){
e43c066c 2465 Float_t cerry, cerrz;
2466 if (ny[i]>0) {cerry = erry[i]; cerrz=errz[i];}
b9671574 2467 else { cerry= track->GetSigmaY(i); cerrz = track->GetSigmaZ(i);}
e43c066c 2468 cerry*=cerry;
2469 cerrz*=cerrz;
2470 Float_t cerryb, cerrzb;
2471 if (ny[i+6]>0) {cerryb = erry[i+6]; cerrzb=errz[i+6];}
b9671574 2472 else { cerryb= track->GetSigmaY(i+6); cerrzb = track->GetSigmaZ(i+6);}
e43c066c 2473 cerryb*=cerryb;
2474 cerrzb*=cerrzb;
b9671574 2475 chi2+= TMath::Min((track->GetDy(i+6)*track->GetDy(i+6)/cerryb),track->GetDy(i)*track->GetDy(i)/cerry);
2476 chi2+= TMath::Min((track->GetDz(i+6)*track->GetDz(i+6)/cerrzb),track->GetDz(i)*track->GetDz(i)/cerrz);
e43c066c 2477 sum++;
2478 }
2479 }
2480 }
b9671574 2481 if (track->GetESDtrack()->GetTPCsignal()>85){
2482 Float_t ratio = track->GetdEdx()/track->GetESDtrack()->GetTPCsignal();
e43c066c 2483 if (ratio<0.5) {
2484 chi2+=(0.5-ratio)*5.;
2485 }
2486 if (ratio>2){
2487 chi2+=(ratio-2.0)*3;
2488 }
2489 }
2490 //
b9671574 2491 Double_t match = TMath::Sqrt(track->GetChi22());
2492 if (track->GetConstrain()) match/=track->GetNumberOfClusters();
2755f080 2493 if (!track->GetConstrain()) {
2494 if (track->GetNumberOfClusters()>2) {
2495 match/=track->GetNumberOfClusters()-2.;
2496 } else {
2497 match=0;
2498 }
2499 }
e43c066c 2500 if (match<0) match=0;
f8720bda 2501
2502 // penalty factor for missing points (NDeadZone>0), but no penalty
2503 // for layer with deadZoneProb close to 1 (either we wanted to skip layer
2504 // or there is a dead from OCDB)
2505 Float_t deadzonefactor = 0.;
2506 if (track->GetNDeadZone()>0.) {
2507 Float_t sumDeadZoneProbability=0;
2508 for(Int_t ilay=0;ilay<6;ilay++) sumDeadZoneProbability+=track->GetDeadZoneProbability(ilay);
2509 Float_t nDeadZoneWithProbNot1=(Float_t)(track->GetNDeadZone())-sumDeadZoneProbability;
2510 if(nDeadZoneWithProbNot1>0.) {
2511 Float_t deadZoneProbability = sumDeadZoneProbability - (Float_t)((Int_t)sumDeadZoneProbability);
2512 deadZoneProbability /= nDeadZoneWithProbNot1;
2513 deadzonefactor = 3.*(1.1-deadZoneProbability);
2514 }
2515 }
2516
b9671574 2517 Double_t normchi2 = 2*track->GetNSkipped()+match+deadzonefactor+(1+(2*track->GetNSkipped()+deadzonefactor)/track->GetNumberOfClusters())*
2518 (chi2)/TMath::Max(double(sum-track->GetNSkipped()),
2519 1./(1.+track->GetNSkipped()));
e43c066c 2520
2521 return normchi2;
2522}
44347160 2523//------------------------------------------------------------------------
a70ed6ad 2524Double_t AliITStrackerMI::GetMatchingChi2(const AliITStrackMI * track1,const AliITStrackMI * track2)
e43c066c 2525{
2526 //
2527 // return matching chi2 between two tracks
d1181f10 2528 Double_t largeChi2=1000.;
2529
15dd636f 2530 AliITStrackMI track3(*track2);
d1181f10 2531 if (!track3.Propagate(track1->GetAlpha(),track1->GetX())) return largeChi2;
e43c066c 2532 TMatrixD vec(5,1);
6c94f330 2533 vec(0,0)=track1->GetY() - track3.GetY();
2534 vec(1,0)=track1->GetZ() - track3.GetZ();
2535 vec(2,0)=track1->GetSnp() - track3.GetSnp();
2536 vec(3,0)=track1->GetTgl() - track3.GetTgl();
6c23ffed 2537 vec(4,0)=track1->GetSigned1Pt() - track3.GetSigned1Pt();
e43c066c 2538 //
2539 TMatrixD cov(5,5);
6c94f330 2540 cov(0,0) = track1->GetSigmaY2()+track3.GetSigmaY2();
2541 cov(1,1) = track1->GetSigmaZ2()+track3.GetSigmaZ2();
2542 cov(2,2) = track1->GetSigmaSnp2()+track3.GetSigmaSnp2();
2543 cov(3,3) = track1->GetSigmaTgl2()+track3.GetSigmaTgl2();
2544 cov(4,4) = track1->GetSigma1Pt2()+track3.GetSigma1Pt2();
e43c066c 2545
6c94f330 2546 cov(0,1)=cov(1,0) = track1->GetSigmaZY()+track3.GetSigmaZY();
2547 cov(0,2)=cov(2,0) = track1->GetSigmaSnpY()+track3.GetSigmaSnpY();
2548 cov(0,3)=cov(3,0) = track1->GetSigmaTglY()+track3.GetSigmaTglY();
2549 cov(0,4)=cov(4,0) = track1->GetSigma1PtY()+track3.GetSigma1PtY();
e43c066c 2550 //
6c94f330 2551 cov(1,2)=cov(2,1) = track1->GetSigmaSnpZ()+track3.GetSigmaSnpZ();
2552 cov(1,3)=cov(3,1) = track1->GetSigmaTglZ()+track3.GetSigmaTglZ();
2553 cov(1,4)=cov(4,1) = track1->GetSigma1PtZ()+track3.GetSigma1PtZ();
e43c066c 2554 //
6c94f330 2555 cov(2,3)=cov(3,2) = track1->GetSigmaTglSnp()+track3.GetSigmaTglSnp();
2556 cov(2,4)=cov(4,2) = track1->GetSigma1PtSnp()+track3.GetSigma1PtSnp();
e43c066c 2557 //
6c94f330 2558 cov(3,4)=cov(4,3) = track1->GetSigma1PtTgl()+track3.GetSigma1PtTgl();
e43c066c 2559
2560 cov.Invert();
2561 TMatrixD vec2(cov,TMatrixD::kMult,vec);
2562 TMatrixD chi2(vec2,TMatrixD::kTransposeMult,vec);
2563 return chi2(0,0);
2564}
44347160 2565//------------------------------------------------------------------------
a70ed6ad 2566Double_t AliITStrackerMI::GetSPDDeadZoneProbability(Double_t zpos, Double_t zerr) const
e43c066c 2567{
2568 //
afd25725 2569 // return probability that given point (characterized by z position and error)
2570 // is in SPD dead zone
e43c066c 2571 //
afd25725 2572 Double_t probability = 0.;
e43c066c 2573 Double_t absz = TMath::Abs(zpos);
afd25725 2574 Double_t nearestz = (absz<2.) ? 0.5*(fSPDdetzcentre[1]+fSPDdetzcentre[2]) :
2575 0.5*(fSPDdetzcentre[2]+fSPDdetzcentre[3]);
2576 if (TMath::Abs(absz-nearestz)>0.25+3.*zerr) return probability;
2577 Double_t zmin, zmax;
2578 if (zpos<-6.) { // dead zone at z = -7
e50912db 2579 zmin = fSPDdetzcentre[0] + 0.5*AliITSRecoParam::GetSPDdetzlength();
2580 zmax = fSPDdetzcentre[1] - 0.5*AliITSRecoParam::GetSPDdetzlength();
afd25725 2581 } else if (zpos>6.) { // dead zone at z = +7
e50912db 2582 zmin = fSPDdetzcentre[2] + 0.5*AliITSRecoParam::GetSPDdetzlength();
2583 zmax = fSPDdetzcentre[3] - 0.5*AliITSRecoParam::GetSPDdetzlength();
afd25725 2584 } else if (absz<2.) { // dead zone at z = 0
e50912db 2585 zmin = fSPDdetzcentre[1] + 0.5*AliITSRecoParam::GetSPDdetzlength();
2586 zmax = fSPDdetzcentre[2] - 0.5*AliITSRecoParam::GetSPDdetzlength();
afd25725 2587 } else {
2588 zmin = 0.;
2589 zmax = 0.;
e43c066c 2590 }
afd25725 2591 // probability that the true z is in the range [zmin,zmax] (i.e. inside
2592 // dead zone)
2593 probability = 0.5*( TMath::Erf((zpos-zmin)/zerr/TMath::Sqrt(2.)) -
2594 TMath::Erf((zpos-zmax)/zerr/TMath::Sqrt(2.)) );
e43c066c 2595 return probability;
2596}
44347160 2597//------------------------------------------------------------------------
a70ed6ad 2598Double_t AliITStrackerMI::GetTruncatedChi2(const AliITStrackMI * track, Float_t fac)
e43c066c 2599{
2600 //
2601 // calculate normalized chi2
2602 Float_t chi2[6];
2603 Float_t *erry = GetErrY(fCurrentEsdTrack), *errz = GetErrZ(fCurrentEsdTrack);
2604 Float_t ncl = 0;
2605 for (Int_t i = 0;i<6;i++){
b9671574 2606 if (TMath::Abs(track->GetDy(i))>0){
2607 chi2[i]= (track->GetDy(i)/erry[i])*(track->GetDy(i)/erry[i]);
2608 chi2[i]+= (track->GetDz(i)/errz[i])*(track->GetDz(i)/errz[i]);
e43c066c 2609 ncl++;
2610 }
2611 else{chi2[i]=10000;}
2612 }
2613 Int_t index[6];
2614 TMath::Sort(6,chi2,index,kFALSE);
2615 Float_t max = float(ncl)*fac-1.;
2616 Float_t sumchi=0, sumweight=0;
2617 for (Int_t i=0;i<max+1;i++){
2618 Float_t weight = (i<max)?1.:(max+1.-i);
2619 sumchi+=weight*chi2[index[i]];
2620 sumweight+=weight;
2621 }
2622 Double_t normchi2 = sumchi/sumweight;
2623 return normchi2;
2624}
44347160 2625//------------------------------------------------------------------------
a70ed6ad 2626Double_t AliITStrackerMI::GetInterpolatedChi2(const AliITStrackMI * forwardtrack,const AliITStrackMI * backtrack)
e43c066c 2627{
2628 //
2629 // calculate normalized chi2
2630 // if (forwardtrack->fNUsed>0.3*float(forwardtrack->GetNumberOfClusters())) return 10000;
2631 Int_t npoints = 0;
2632 Double_t res =0;
2633 for (Int_t i=0;i<6;i++){
b9671574 2634 if ( (backtrack->GetSigmaY(i)<0.000000001) || (forwardtrack->GetSigmaY(i)<0.000000001)) continue;
2635 Double_t sy1 = forwardtrack->GetSigmaY(i);
2636 Double_t sz1 = forwardtrack->GetSigmaZ(i);
2637 Double_t sy2 = backtrack->GetSigmaY(i);
2638 Double_t sz2 = backtrack->GetSigmaZ(i);
e43c066c 2639 if (i<2){ sy2=1000.;sz2=1000;}
2640 //
b9671574 2641 Double_t dy0 = (forwardtrack->GetDy(i)/(sy1*sy1) +backtrack->GetDy(i)/(sy2*sy2))/(1./(sy1*sy1)+1./(sy2*sy2));
2642 Double_t dz0 = (forwardtrack->GetDz(i)/(sz1*sz1) +backtrack->GetDz(i)/(sz2*sz2))/(1./(sz1*sz1)+1./(sz2*sz2));
e43c066c 2643 //
2644 Double_t nz0 = dz0*TMath::Sqrt((1./(sz1*sz1)+1./(sz2*sz2)));
2645 Double_t ny0 = dy0*TMath::Sqrt((1./(sy1*sy1)+1./(sy2*sy2)));
2646 //
2647 res+= nz0*nz0+ny0*ny0;
2648 npoints++;
2649 }
2650 if (npoints>1) return
6c23ffed 2651 TMath::Max(0.3*forwardtrack->OneOverPt()-0.5,0.)+
e43c066c 2652 //2*forwardtrack->fNUsed+
b9671574 2653 res/TMath::Max(double(npoints-forwardtrack->GetNSkipped()),
2654 1./(1.+forwardtrack->GetNSkipped()));
e43c066c 2655 return 1000;
2656}
44347160 2657//------------------------------------------------------------------------
e43c066c 2658Float_t *AliITStrackerMI::GetWeight(Int_t index) {
2659 //--------------------------------------------------------------------
2660 // Return pointer to a given cluster
2661 //--------------------------------------------------------------------
2662 Int_t l=(index & 0xf0000000) >> 28;
2663 Int_t c=(index & 0x0fffffff) >> 00;
2664 return fgLayers[l].GetWeight(c);
2665}
44347160 2666//------------------------------------------------------------------------
a70ed6ad 2667void AliITStrackerMI::RegisterClusterTracks(const AliITStrackMI* track,Int_t id)
e43c066c 2668{
2669 //---------------------------------------------
2670 // register track to the list
628e7bb0 2671 //
b9671574 2672 if (track->GetESDtrack()->GetKinkIndex(0)!=0) return; //don't register kink tracks
628e7bb0 2673 //
2674 //
e43c066c 2675 for (Int_t icluster=0;icluster<track->GetNumberOfClusters();icluster++){
2676 Int_t index = track->GetClusterIndex(icluster);
2677 Int_t l=(index & 0xf0000000) >> 28;
2678 Int_t c=(index & 0x0fffffff) >> 00;
b9671574 2679 if (c>fgLayers[l].GetNumberOfClusters()) continue;
e43c066c 2680 for (Int_t itrack=0;itrack<4;itrack++){
b9671574 2681 if (fgLayers[l].GetClusterTracks(itrack,c)<0){
2682 fgLayers[l].SetClusterTracks(itrack,c,id);
e43c066c 2683 break;
2684 }
2685 }
2686 }
2687}
44347160 2688//------------------------------------------------------------------------
a70ed6ad 2689void AliITStrackerMI::UnRegisterClusterTracks(const AliITStrackMI* track, Int_t id)
e43c066c 2690{
2691 //---------------------------------------------
2692 // unregister track from the list
2693 for (Int_t icluster=0;icluster<track->GetNumberOfClusters();icluster++){
2694 Int_t index = track->GetClusterIndex(icluster);
2695 Int_t l=(index & 0xf0000000) >> 28;
2696 Int_t c=(index & 0x0fffffff) >> 00;
b9671574 2697 if (c>fgLayers[l].GetNumberOfClusters()) continue;
e43c066c 2698 for (Int_t itrack=0;itrack<4;itrack++){
b9671574 2699 if (fgLayers[l].GetClusterTracks(itrack,c)==id){
2700 fgLayers[l].SetClusterTracks(itrack,c,-1);
e43c066c 2701 }
2702 }
2703 }
2704}
44347160 2705//------------------------------------------------------------------------
00a7cc50 2706Float_t AliITStrackerMI::GetNumberOfSharedClusters(AliITStrackMI* track,Int_t id, Int_t list[6], AliITSRecPoint *clist[6])
e43c066c 2707{
2708 //-------------------------------------------------------------
2709 //get number of shared clusters
2710 //-------------------------------------------------------------
2711 Float_t shared=0;
2712 for (Int_t i=0;i<6;i++) { list[i]=-1, clist[i]=0;}
2713 // mean number of clusters
2714 Float_t *ny = GetNy(id), *nz = GetNz(id);
2715
2716
2717 for (Int_t icluster=0;icluster<track->GetNumberOfClusters();icluster++){
2718 Int_t index = track->GetClusterIndex(icluster);
2719 Int_t l=(index & 0xf0000000) >> 28;
2720 Int_t c=(index & 0x0fffffff) >> 00;
b9671574 2721 if (c>fgLayers[l].GetNumberOfClusters()) continue;
e43c066c 2722 if (ny[l]==0){
2723 printf("problem\n");
2724 }
00a7cc50 2725 AliITSRecPoint *cl = (AliITSRecPoint*)GetCluster(index);
e43c066c 2726 Float_t weight=1;
2727 //
2728 Float_t deltan = 0;
2729 if (l>3&&cl->GetNy()+cl->GetNz()>6) continue;
572f41f9 2730 if (l>2&&AliITSReconstructor::GetRecoParam()->GetUseAmplitudeInfo(l))
2731 if (track->GetNormQ(l)/track->GetExpQ()>3.5) continue;
e43c066c 2732 if (l<2 || l>3){
2733 deltan = (cl->GetNy()+cl->GetNz()-ny[l]-nz[l]);
2734 }
2735 else{
2736 deltan = (cl->GetNz()-nz[l]);
2737 }
2738 if (deltan>2.0) continue; // extended - highly probable shared cluster
2739 weight = 2./TMath::Max(3.+deltan,2.);
2740 //
2741 for (Int_t itrack=0;itrack<4;itrack++){
b9671574 2742 if (fgLayers[l].GetClusterTracks(itrack,c)>=0 && fgLayers[l].GetClusterTracks(itrack,c)!=id){
e43c066c 2743 list[l]=index;
00a7cc50 2744 clist[l] = (AliITSRecPoint*)GetCluster(index);
e43c066c 2745 shared+=weight;
2746 break;
2747 }
2748 }
2749 }
b9671574 2750 track->SetNUsed(shared);
e43c066c 2751 return shared;
2752}
44347160 2753//------------------------------------------------------------------------
a70ed6ad 2754Int_t AliITStrackerMI::GetOverlapTrack(const AliITStrackMI *track, Int_t trackID, Int_t &shared, Int_t clusterlist[6],Int_t overlist[6])
e43c066c 2755{
2756 //
2757 // find first shared track
2758 //
2759 // mean number of clusters
2760 Float_t *ny = GetNy(trackID), *nz = GetNz(trackID);
2761 //
2762 for (Int_t i=0;i<6;i++) overlist[i]=-1;
2763 Int_t sharedtrack=100000;
2764 Int_t tracks[24],trackindex=0;
2765 for (Int_t i=0;i<24;i++) {tracks[i]=-1;}
2766 //
2767 for (Int_t icluster=0;icluster<6;icluster++){
2768 if (clusterlist[icluster]<0) continue;
2769 Int_t index = clusterlist[icluster];
2770 Int_t l=(index & 0xf0000000) >> 28;
2771 Int_t c=(index & 0x0fffffff) >> 00;
2772 if (ny[l]==0){
2773 printf("problem\n");
2774 }
b9671574 2775 if (c>fgLayers[l].GetNumberOfClusters()) continue;
e43c066c 2776 //if (l>3) continue;
00a7cc50 2777 AliITSRecPoint *cl = (AliITSRecPoint*)GetCluster(index);
e43c066c 2778 //
2779 Float_t deltan = 0;
2780 if (l>3&&cl->GetNy()+cl->GetNz()>6) continue;
572f41f9 2781 if (l>2&&AliITSReconstructor::GetRecoParam()->GetUseAmplitudeInfo(l))
2782 if (track->GetNormQ(l)/track->GetExpQ()>3.5) continue;
e43c066c 2783 if (l<2 || l>3){
2784 deltan = (cl->GetNy()+cl->GetNz()-ny[l]-nz[l]);
2785 }
2786 else{
2787 deltan = (cl->GetNz()-nz[l]);
2788 }
2789 if (deltan>2.0) continue; // extended - highly probable shared cluster
2790 //
2791 for (Int_t itrack=3;itrack>=0;itrack--){
b9671574 2792 if (fgLayers[l].GetClusterTracks(itrack,c)<0) continue;
2793 if (fgLayers[l].GetClusterTracks(itrack,c)!=trackID){
2794 tracks[trackindex] = fgLayers[l].GetClusterTracks(itrack,c);
e43c066c 2795 trackindex++;
2796 }
2797 }
2798 }
2799 if (trackindex==0) return -1;
2800 if (trackindex==1){
2801 sharedtrack = tracks[0];
2802 }else{
2803 if (trackindex==2) sharedtrack =TMath::Min(tracks[0],tracks[1]);
2804 else{
2805 //
a5d0fae6 2806 Int_t tracks2[24], cluster[24];
2807 for (Int_t i=0;i<trackindex;i++){ tracks2[i]=-1; cluster[i]=0;}
e43c066c 2808 Int_t index =0;
2809 //
2810 for (Int_t i=0;i<trackindex;i++){
2811 if (tracks[i]<0) continue;
a5d0fae6 2812 tracks2[index] = tracks[i];
e43c066c 2813 cluster[index]++;
2814 for (Int_t j=i+1;j<trackindex;j++){
2815 if (tracks[j]<0) continue;
2816 if (tracks[j]==tracks[i]){
2817 cluster[index]++;
2818 tracks[j]=-1;
2819 }
2820 }
2821 index++;
2822 }
2823 Int_t max=0;
2824 for (Int_t i=0;i<index;i++){
2825 if (cluster[index]>max) {
a5d0fae6 2826 sharedtrack=tracks2[index];
e43c066c 2827 max=cluster[index];
2828 }
2829 }
2830 }
2831 }
2832
2833 if (sharedtrack>=100000) return -1;
2834 //
2835 // make list of overlaps
2836 shared =0;
2837 for (Int_t icluster=0;icluster<6;icluster++){
2838 if (clusterlist[icluster]<0) continue;
2839 Int_t index = clusterlist[icluster];
2840 Int_t l=(index & 0xf0000000) >> 28;
2841 Int_t c=(index & 0x0fffffff) >> 00;
b9671574 2842 if (c>fgLayers[l].GetNumberOfClusters()) continue;
00a7cc50 2843 AliITSRecPoint *cl = (AliITSRecPoint*)GetCluster(index);
e43c066c 2844 if (l==0 || l==1){
2845 if (cl->GetNy()>2) continue;
2846 if (cl->GetNz()>2) continue;
2847 }
2848 if (l==4 || l==5){
2849 if (cl->GetNy()>3) continue;
2850 if (cl->GetNz()>3) continue;
2851 }
2852 //
2853 for (Int_t itrack=3;itrack>=0;itrack--){
b9671574 2854 if (fgLayers[l].GetClusterTracks(itrack,c)<0) continue;
2855 if (fgLayers[l].GetClusterTracks(itrack,c)==sharedtrack){
e43c066c 2856 overlist[l]=index;
2857 shared++;
2858 }
2859 }
2860 }
2861 return sharedtrack;
2862}
44347160 2863//------------------------------------------------------------------------
15dd636f 2864AliITStrackMI * AliITStrackerMI::GetBest2Tracks(Int_t trackID1, Int_t trackID2, Float_t th0, Float_t th1){
e43c066c 2865 //
2866 // try to find track hypothesys without conflicts
2867 // with minimal chi2;
2868 TClonesArray *arr1 = (TClonesArray*)fTrackHypothesys.At(trackID1);
2869 Int_t entries1 = arr1->GetEntriesFast();
2870 TClonesArray *arr2 = (TClonesArray*)fTrackHypothesys.At(trackID2);
15dd636f 2871 if (!arr2) return (AliITStrackMI*) arr1->UncheckedAt(0);
e43c066c 2872 Int_t entries2 = arr2->GetEntriesFast();
15dd636f 2873 if (entries2<=0) return (AliITStrackMI*) arr1->UncheckedAt(0);
e43c066c 2874 //
15dd636f 2875 AliITStrackMI * track10=(AliITStrackMI*) arr1->UncheckedAt(0);
2876 AliITStrackMI * track20=(AliITStrackMI*) arr2->UncheckedAt(0);
6c23ffed 2877 if (track10->Pt()>0.5+track20->Pt()) return track10;
e43c066c 2878
2879 for (Int_t itrack=0;itrack<entries1;itrack++){
15dd636f 2880 AliITStrackMI * track=(AliITStrackMI*) arr1->UncheckedAt(itrack);
e43c066c 2881 UnRegisterClusterTracks(track,trackID1);
2882 }
2883 //
2884 for (Int_t itrack=0;itrack<entries2;itrack++){
15dd636f 2885 AliITStrackMI * track=(AliITStrackMI*) arr2->UncheckedAt(itrack);
e43c066c 2886 UnRegisterClusterTracks(track,trackID2);
2887 }
2888 Int_t index1=0;
2889 Int_t index2=0;
2890 Float_t maxconflicts=6;
2891 Double_t maxchi2 =1000.;
2892 //
2893 // get weight of hypothesys - using best hypothesy
2894 Double_t w1,w2;
2895
2896 Int_t list1[6],list2[6];
00a7cc50 2897 AliITSRecPoint *clist1[6], *clist2[6] ;
e43c066c 2898 RegisterClusterTracks(track10,trackID1);
2899 RegisterClusterTracks(track20,trackID2);
2900 Float_t conflict1 = GetNumberOfSharedClusters(track10,trackID1,list1,clist1);
2901 Float_t conflict2 = GetNumberOfSharedClusters(track20,trackID2,list2,clist2);
2902 UnRegisterClusterTracks(track10,trackID1);
2903 UnRegisterClusterTracks(track20,trackID2);
2904 //
2905 // normalized chi2
2906 Float_t chi21 =0,chi22=0,ncl1=0,ncl2=0;
2907 Float_t nerry[6],nerrz[6];
2908 Float_t *erry1=GetErrY(trackID1),*errz1 = GetErrZ(trackID1);
2909 Float_t *erry2=GetErrY(trackID2),*errz2 = GetErrZ(trackID2);
2910 for (Int_t i=0;i<6;i++){
2911 if ( (erry1[i]>0) && (erry2[i]>0)) {
2912 nerry[i] = TMath::Min(erry1[i],erry2[i]);
2913 nerrz[i] = TMath::Min(errz1[i],errz2[i]);
2914 }else{
2915 nerry[i] = TMath::Max(erry1[i],erry2[i]);
2916 nerrz[i] = TMath::Max(errz1[i],errz2[i]);
2917 }
b9671574 2918 if (TMath::Abs(track10->GetDy(i))>0.000000000000001){
2919 chi21 += track10->GetDy(i)*track10->GetDy(i)/(nerry[i]*nerry[i]);
2920 chi21 += track10->GetDz(i)*track10->GetDz(i)/(nerrz[i]*nerrz[i]);
e43c066c 2921 ncl1++;
2922 }
b9671574 2923 if (TMath::Abs(track20->GetDy(i))>0.000000000000001){
2924 chi22 += track20->GetDy(i)*track20->GetDy(i)/(nerry[i]*nerry[i]);
2925 chi22 += track20->GetDz(i)*track20->GetDz(i)/(nerrz[i]*nerrz[i]);
e43c066c 2926 ncl2++;
2927 }
2928 }
2929 chi21/=ncl1;
2930 chi22/=ncl2;
2931 //
2932 //
b9671574 2933 Float_t d1 = TMath::Sqrt(track10->GetD(0)*track10->GetD(0)+track10->GetD(1)*track10->GetD(1))+0.1;
2934 Float_t d2 = TMath::Sqrt(track20->GetD(0)*track20->GetD(0)+track20->GetD(1)*track20->GetD(1))+0.1;
e43c066c 2935 Float_t s1 = TMath::Sqrt(track10->GetSigmaY2()*track10->GetSigmaZ2());
2936 Float_t s2 = TMath::Sqrt(track20->GetSigmaY2()*track20->GetSigmaZ2());
2937 //
2938 w1 = (d2/(d1+d2)+ 2*s2/(s1+s2)+
2939 +s2/(s1+s2)*0.5*(chi22+2.)/(chi21+chi22+4.)
6c23ffed 2940 +1.*track10->Pt()/(track10->Pt()+track20->Pt())
e43c066c 2941 );
2942 w2 = (d1/(d1+d2)+ 2*s1/(s1+s2)+
2943 s1/(s1+s2)*0.5*(chi21+2.)/(chi21+chi22+4.)
6c23ffed 2944 +1.*track20->Pt()/(track10->Pt()+track20->Pt())
e43c066c 2945 );
2946
2947 Double_t sumw = w1+w2;
2948 w1/=sumw;
2949 w2/=sumw;
2950 if (w1<w2*0.5) {
2951 w1 = (d2+0.5)/(d1+d2+1.);
2952 w2 = (d1+0.5)/(d1+d2+1.);
2953 }
2954 // Float_t maxmax = w1*track10->fChi2MIP[0]+w2*track20->fChi2MIP[0]+w1*conflict1+w2*conflict2+1.;
2955 //Float_t maxconflicts0 = w1*conflict1+w2*conflict2;
2956 //
2957 // get pair of "best" hypothesys
2958 //
2959 Float_t * ny1 = GetNy(trackID1), * nz1 = GetNz(trackID1);
2960 Float_t * ny2 = GetNy(trackID2), * nz2 = GetNz(trackID2);
2961
2962 for (Int_t itrack1=0;itrack1<entries1;itrack1++){
15dd636f 2963 AliITStrackMI * track1=(AliITStrackMI*) arr1->UncheckedAt(itrack1);
e43c066c 2964 //if (track1->fFakeRatio>0) continue;
2965 RegisterClusterTracks(track1,trackID1);
2966 for (Int_t itrack2=0;itrack2<entries2;itrack2++){
15dd636f 2967 AliITStrackMI * track2=(AliITStrackMI*) arr2->UncheckedAt(itrack2);
e43c066c 2968
2969 // Float_t current = w1*track1->fChi2MIP[0]+w2*track2->fChi2MIP[0];
2970 //if (track2->fFakeRatio>0) continue;
2971 Float_t nskipped=0;
2972 RegisterClusterTracks(track2,trackID2);
e43c066c 2973 Float_t cconflict1 = GetNumberOfSharedClusters(track1,trackID1,list1,clist1);
2974 Float_t cconflict2 = GetNumberOfSharedClusters(track2,trackID2,list2,clist2);