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