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