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