Integrating the Cooked Matrix tracker into the commom reconstruction framework
[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITSUTrackerGlo.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 /*
17   TO CHECK
18   GetBz usage
19
20  */
21
22 //-------------------------------------------------------------------------
23 //  Implementation of the ITS Upgrade tracker mother class.              
24 //-------------------------------------------------------------------------
25 #include <TTree.h>
26 #include <Riostream.h> 
27 #include <TMath.h>
28 #include <TFile.h>
29
30 #include "AliITSUTrackerGlo.h"
31 #include "AliESDEvent.h"
32 #include "AliESDtrack.h"
33 #include "AliITSURecoDet.h"
34 #include "AliITSURecoSens.h"
35 #include "AliITSUReconstructor.h"
36 #include "AliITSReconstructor.h"
37 #include "AliITSUSeed.h"
38 #include "AliITSUClusterPix.h"
39 #include "AliITSUGeomTGeo.h"
40 #include "AliCodeTimer.h"
41 #include "AliRefArray.h"
42 using namespace AliITSUAux;
43 using namespace TMath;
44
45 //----------------- tmp stuff -----------------
46
47 ClassImp(AliITSUTrackerGlo)
48
49 const Double_t AliITSUTrackerGlo::fgkToler =  1e-6;// tolerance for layer on-surface check
50
51
52 //_________________________________________________________________________
53 AliITSUTrackerGlo::AliITSUTrackerGlo(AliITSUReconstructor* rec)
54 :  fReconstructor(rec)
55   ,fITS(0)
56   ,fMatLUT(0)
57   ,fCurrESDtrack(0)
58   ,fCurrESDtrMClb(kDummyLabel)
59   ,fCurrMass(kPionMass)
60   ,fCountProlongationTrials(0)
61   ,fCountITSin(0)
62   ,fCountITSout(0)
63   ,fCountITSrefit(0)
64   ,fNTracksESD(0)
65   ,fHypStore(100)
66   ,fLayerMaxCandidates(1000)
67   ,fLayerCandidates(0)
68   ,fNBranchesAdded(0)
69   ,fNCandidatesAdded(0)
70   ,fCurrHyp(0)
71   ,fWorkHyp(0)
72   ,fSeedsPool("AliITSUSeed",0)
73   ,fFreeSeedsID(0)
74   ,fESDIndex(0)
75   ,fNFreeSeeds(0)
76   ,fLastSeedID(0)
77   ,fNLrActive(0)
78   ,fDefTrackConds(0)
79   ,fCurrTrackCond(0)
80   ,fCurrActLrID(-1)
81   ,fCurrLayer(0)
82   ,fTrackPhaseID(-1)
83   ,fCurrPassID(-1)
84   ,fUseMatLUT(kFALSE)
85 #ifdef  _ITSU_TUNING_MODE_
86   ,fCHistoArrCorr(0)
87   ,fCHistoArrFake(0)
88 #endif
89 {
90   // Default constructor
91   if (rec) Init(rec);
92 }
93
94 //_________________________________________________________________________
95 AliITSUTrackerGlo::~AliITSUTrackerGlo()
96 {
97  // Default destructor
98  //  
99   delete[] fLayerCandidates;
100   if (fWorkHyp) fWorkHyp->SetTPCSeed(0); // this hypothesis does not own the seed
101   delete fWorkHyp;
102   delete fMatLUT;
103   //
104 #ifdef  _ITSU_TUNING_MODE_
105   if (fCHistoArrCorr || fCHistoArrFake) {
106     TFile* ctrOut = TFile::Open("itsuControlHistos.root","recreate");
107     ctrOut->cd();
108     AliInfo("Storing control histos");
109     //    ctrOut->WriteObject(fCHistoArr,"controlH","kSingleKey");
110     if (fCHistoArrCorr) {
111       fCHistoArrCorr->Write();
112       delete fCHistoArrCorr;
113     }
114     if (fCHistoArrFake) {
115       fCHistoArrFake->Write();
116       delete fCHistoArrFake;
117     }
118     ctrOut->Close();
119     delete ctrOut;
120     fCHistoArrCorr = 0;
121     fCHistoArrFake = 0;    
122   }
123 #endif 
124   //
125 }
126
127 //_________________________________________________________________________
128 void AliITSUTrackerGlo::Init(AliITSUReconstructor* rec)
129 {
130   // init with external reconstructor
131   //
132   fITS = rec->GetITSInterface();
133   fNLrActive = fITS->GetNLayersActive();
134   fWorkHyp = new AliITSUTrackHyp(fNLrActive);
135   // create material lookup table
136   const int kNTest = 1000;
137   const double kStepsPerCM=5;
138   fMatLUT  = new AliITSUMatLUT(fITS->GetRMin(),fITS->GetRMax(),Nint(kStepsPerCM*(fITS->GetRMax()-fITS->GetRMin())));
139   double zmn = 1e6;
140   for (int ilr=fITS->GetNLayers();ilr--;) {
141     AliITSURecoLayer* lr = fITS->GetLayer(ilr);
142     if (zmn>Abs(lr->GetZMin())) zmn = Abs(lr->GetZMin());
143     if (zmn>Abs(lr->GetZMax())) zmn = Abs(lr->GetZMax());
144   }
145   fMatLUT->FillData(kNTest,-zmn,zmn);
146   //
147   if (fLayerMaxCandidates<1) fLayerMaxCandidates = 1000;
148   fLayerCandidates = new AliITSUSeed*[fLayerMaxCandidates];
149   fSeedsPool.ExpandCreateFast(1000); // RS TOCHECK
150   fFreeSeedsID.Set(1000);
151   fESDIndex.Set(1000);
152   //
153 }
154
155 //_________________________________________________________________________
156 void AliITSUTrackerGlo::CreateDefaultTrackCond()
157 {
158   // creates default tracking conditions to be used when recoparam does not provide them
159
160   AliITSUTrackCond* cond = new AliITSUTrackCond();
161   //
162   cond->SetNLayers(fNLrActive);
163   cond->AddNewCondition(fNLrActive);
164   cond->AddGroupPattern( 0xffff, fNLrActive ); // require all layers hit
165   cond->Init();
166   //
167   fDefTrackConds.AddLast(cond);
168   //
169   AliInfo("Created conditions: ");
170   for (int i=0;i<fDefTrackConds.GetEntriesFast();i++) fDefTrackConds[i]->Print();
171   //
172 }
173
174
175 //_________________________________________________________________________
176 Int_t AliITSUTrackerGlo::Clusters2Tracks(AliESDEvent *esdEv)
177 {
178   //
179   AliCodeTimerAuto("",0);
180   SetTrackingPhase(kClus2Tracks);
181   //
182 #ifdef  _ITSU_TUNING_MODE_
183   if (!fCHistoArrCorr) fCHistoArrCorr = BookControlHistos("Corr");
184   if (!fCHistoArrFake) fCHistoArrFake = BookControlHistos("Fake");
185 #endif
186   static int evID = 0;
187   static TArrayF esdTrPt(fESDIndex.GetSize()); 
188   //
189   TObjArray *trackConds = 0;
190   //
191   fCountProlongationTrials = 0;
192   fCountITSin = 0;
193   fCountITSout = 0;
194   fCountITSrefit = 0;
195   //
196   ResetSeedsPool();
197   fNTracksESD = esdEv->GetNumberOfTracks();
198   AliInfo(Form("Will try to find prolongations for %d tracks",fNTracksESD));
199   int nTrackCond = AliITSUReconstructor::GetRecoParam()->GetNTrackingConditions();
200   fUseMatLUT = AliITSUReconstructor::GetRecoParam()->GetUseMatLUT(fTrackPhaseID);
201   if (nTrackCond<1) {
202     if (!fDefTrackConds.GetEntriesFast()) {
203       AliInfo("No tracking conditions found in recoparams, creating default one requesting all layers hit");
204       CreateDefaultTrackCond();
205     }
206     trackConds = &fDefTrackConds;
207     nTrackCond = trackConds->GetEntriesFast();
208   } 
209   else trackConds = AliITSUReconstructor::GetRecoParam()->GetTrackingConditions();
210   //
211   static Bool_t first = kTRUE;
212   if (first) {
213     AliITSUReconstructor::GetRecoParam()->Print();
214     first = kFALSE;
215   }
216   fHypStore.Delete();
217   if (fHypStore.GetSize()<fNTracksESD) fHypStore.Expand(fNTracksESD+100);
218   //
219   fITS->ProcessClusters();
220   //
221 #ifdef  _ITSU_TUNING_MODE_
222   FlagSplitClusters(); // tmp RS
223 #endif
224   //
225   // the tracks will be reconstructed in decreasing pt order, sort them
226   if (fESDIndex.GetSize()<fNTracksESD) {
227     fESDIndex.Set(fNTracksESD+200);
228     esdTrPt.Set(fNTracksESD+200);
229   }
230   int* esdTrackIndex = fESDIndex.GetArray();
231   float* trPt = esdTrPt.GetArray();
232   for (int itr=fNTracksESD;itr--;) {
233     AliESDtrack* esdTr = esdEv->GetTrack(itr);
234     esdTr->SetStatus(AliESDtrack::kITSupg);   // Flag all tracks as participating in the ITSup reco
235     trPt[itr] = esdTr->Pt();
236   }
237   Sort(fNTracksESD,trPt,esdTrackIndex,kTRUE);    
238   //
239   for (int icnd=0;icnd<nTrackCond;icnd++) {
240     fCurrPassID = icnd;
241     fCurrTrackCond = (AliITSUTrackCond*)trackConds->UncheckedAt(icnd);
242     if (!fCurrTrackCond->IsInitDone()) fCurrTrackCond->Init();
243     // select ESD tracks to propagate
244     for (int itr=0;itr<fNTracksESD;itr++) {
245       int trID = esdTrackIndex[itr];
246       fCurrESDtrack = esdEv->GetTrack(trID);
247       fCurrESDtrMClb = fCurrESDtrack->GetLabel();
248       //
249       
250       if (!NeedToProlong(fCurrESDtrack,trID)) continue;  // are we interested in this track?
251       /*
252         // if specific tracks should be checked, create a global array int wtc[] = {evITS*10000+trID, ...};
253       Bool_t dbg = kFALSE;
254       int nwtc = sizeof(wtc)/sizeof(int);
255
256       for (int k=0;k<nwtc;k++) {
257         if (wtc[k]==evID*10000+trID) {
258           dbg = kTRUE;
259           printf("At esdTr: %d MC: %d\n",wtc[k],fCurrESDtrMClb);
260           break;
261         }
262       }
263       AliLog::SetClassDebugLevel("AliITSUTrackerGlo",dbg ? 10:0);
264       */
265 #ifdef _ITSU_DEBUG_
266       AliDebug(1,Form("Processing track %d(esd%d) | M=%.3f Pt=%.3f | MCLabel: %d",itr,trID,fCurrESDtrack->GetMassForTracking(kTRUE),fCurrESDtrack->Pt(),fCurrESDtrMClb));//RS
267 #endif
268       FindTrack(fCurrESDtrack, trID);
269     }   
270     //
271 #ifdef _ITSU_DEBUG_
272     if (AliDebugLevelClass()>+2) {
273       AliInfo(Form("SeedsPool: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
274       fHypStore.Print();
275     }
276 #endif
277     FinalizeHypotheses();
278 #ifdef  _ITSU_TUNING_MODE_
279     CheckClusterUsage(); //!!RS
280 #endif
281     //
282     //AliLog::SetClassDebugLevel("AliITSUTrackerGlo",0);     // in case wtc array was defined, uncomment this
283   }
284   //
285   AliInfo(Form("%d ITSin for %d tried TPC seeds out of %d ESD tracks\n",fCountITSin,fCountProlongationTrials,fNTracksESD));
286   evID++;
287   return 0;
288 }
289
290 //_________________________________________________________________________
291 Int_t AliITSUTrackerGlo::PropagateBack(AliESDEvent *esdEv)
292 {
293   //
294   // Do outward fits in ITS
295   AliCodeTimerAuto("",0);
296   //
297   SetTrackingPhase(kPropBack);
298   fNTracksESD = esdEv->GetNumberOfTracks();
299   fUseMatLUT = AliITSUReconstructor::GetRecoParam()->GetUseMatLUT(fTrackPhaseID);
300 #ifdef _ITSU_DEBUG_
301   AliDebug(1,Form("Will propagate back %d tracks",fNTracksESD));
302 #endif
303   //
304   double bz0 = GetBz();
305   Double_t xyzTrk[3],xyzVtx[3]={GetX(),GetY(),GetZ()};
306   AliITSUTrackHyp dummyTr;
307   const double kWatchStep=10.; // for larger steps watch arc vs segment difference
308   Double_t times[AliPID::kSPECIESC];
309   //
310   for (int itr=0;itr<fNTracksESD;itr++) {
311     fCurrESDtrack = esdEv->GetTrack(itr);
312     fCurrESDtrMClb = fCurrESDtrack->GetLabel();
313     // Start time integral and add distance from current position to vertex 
314     if (fCurrESDtrack->IsOn(AliESDtrack::kITSout)) continue; // already done
315     if (!fCurrESDtrack->IsOn(AliESDtrack::kTPCin)) continue; // skip ITS s.a.
316     //
317     fCurrESDtrack->GetXYZ(xyzTrk); 
318     Double_t dst = 0.;     // set initial track lenght, tof
319     {
320       double dxs = xyzTrk[0] - xyzVtx[0];
321       double dys = xyzTrk[1] - xyzVtx[1];
322       //      double dzs = xyzTrk[2] - xyzVtx[2];
323       // RS: for large segment steps d use approximation of cicrular arc s by
324       // s = 2R*asin(d/2R) = d/p asin(p) \approx d/p (p + 1/6 p^3) = d (1+1/6 p^2)
325       // where R is the track radius, p = d/2R = 2C*d (C is the abs curvature)
326       // Hence s^2/d^2 = (1+1/6 p^2)^2
327       dst = dxs*dxs + dys*dys;
328       if (dst > kWatchStep*kWatchStep) { // correct circular part for arc/segment factor
329         double crv = Abs(fCurrESDtrack->GetC(bz0));
330         double fcarc = 1.+crv*crv*dst/6.;
331         dst *= fcarc*fcarc;
332       }
333       // RS: temporary hack since we don't have SPD vertex:      
334       //      dst += dzs*dzs;
335       dst *= 1+fCurrESDtrack->GetTgl()*fCurrESDtrack->GetTgl();
336       dst = Sqrt(dst); 
337     }
338     //    
339     fCurrESDtrack->SetStatus(AliESDtrack::kTIME);
340     //
341     if (!fCurrESDtrack->IsOn(AliESDtrack::kITSin)) { // case of tracks w/o ITS prolongation: just set the time integration
342       dummyTr.AliExternalTrackParam::operator=(*fCurrESDtrack);
343       dummyTr.StartTimeIntegral();
344       dummyTr.AddTimeStep(dst);
345       dummyTr.GetIntegratedTimes(times,AliPID::kSPECIESC); 
346       fCurrESDtrack->SetIntegratedTimes(times);
347       fCurrESDtrack->SetIntegratedLength(dummyTr.GetIntegratedLength());
348       continue;
349     }
350     //
351     fCurrHyp = GetTrackHyp(itr);
352     fCurrMass = fCurrHyp->GetMass();
353     fCurrHyp->StartTimeIntegral();
354     fCurrHyp->AddTimeStep(dst);
355     fCurrHyp->ResetCovariance(10000);
356     int nclFit = 0;
357     double chi2 = RefitTrack(fCurrHyp,fITS->GetRMax(),nclFit);
358     if (chi2>0) { // propagate to exit from the ITS/TPC screen
359       int ndf = nclFit*2-5;
360       if (ndf>0) chi2 /= ndf;
361       fCurrHyp->SetChi2(chi2);
362       UpdateESDTrack(fCurrHyp,AliESDtrack::kITSout);
363     }
364     else {
365 #ifdef _ITSU_DEBUG_
366       AliDebug(2,Form("Refit Failed for track %d | ESDtrack#%d (MClb:%d)",itr,fCurrESDtrack->GetID(),fCurrESDtrMClb));
367 #endif
368       //fCurrHyp->AliExternalTrackParam::Print();
369       //fCurrHyp->GetWinner()->Print();
370     }
371     //
372   }
373   //
374   AliInfo(Form("%d ITSout in %d ITSin tracks for %d tried TPC seeds out of %d ESD tracks\n",
375                fCountITSout,fCountITSin,fCountProlongationTrials,fNTracksESD));
376   //
377   return 0;
378 }
379
380 //_________________________________________________________________________
381 Int_t AliITSUTrackerGlo::RefitInward(AliESDEvent *esdEv)
382 {
383   //
384   // refit the tracks inward, using current cov.matrix
385   AliCodeTimerAuto("",0);
386   //
387   SetTrackingPhase(kRefitInw);
388   fUseMatLUT = AliITSUReconstructor::GetRecoParam()->GetUseMatLUT(fTrackPhaseID);
389   fNTracksESD = esdEv->GetNumberOfTracks();
390   //  AliLog::SetClassDebugLevel("AliITSUTrackerGlo",10);
391
392   AliDebug(1,Form("Will refit inward %d tracks",fNTracksESD));
393   //  Bool_t uselogMS = AliExternalTrackParam::GetUseLogTermMS();
394   //  AliExternalTrackParam::SetUseLogTermMS(kTRUE);
395   //
396   for (int itr=0;itr<fNTracksESD;itr++) {
397     fCurrESDtrack = esdEv->GetTrack(itr);
398     fCurrESDtrMClb = fCurrESDtrack->GetLabel();
399     // Start time integral and add distance from current position to vertex 
400     UInt_t trStat = fCurrESDtrack->GetStatus();
401     if ( !(trStat & AliESDtrack::kITSout) ) continue;
402     if (   trStat & AliESDtrack::kITSrefit ) continue; // already done
403     if ( !(trStat & AliESDtrack::kTPCin)   ) continue; // skip ITS s.a.
404     if (  (trStat & AliESDtrack::kTPCout) && !(trStat & AliESDtrack::kTPCrefit) ) continue;
405     //
406     fCurrHyp  = GetTrackHyp(itr);
407     fCurrMass = fCurrHyp->GetMass();
408     *fCurrHyp = *fCurrESDtrack;  // fetch current ESDtrack kinematics
409     //
410     int nclFit = 0;
411     //    fCurrHyp->ResetCovariance(1000); // in case we want to do SA refit
412     double chi2 = RefitTrack(fCurrHyp,fITS->GetRMin(),nclFit);
413     if (chi2>0) { // propagate up to inside radius of the beam pipe      
414       int ndf = nclFit*2-5;
415       if (ndf>0) chi2 /= ndf;
416       fCurrHyp->SetChi2(chi2);
417       UpdateESDTrack(fCurrHyp,AliESDtrack::kITSrefit);
418     }
419     else {
420 #ifdef _ITSU_DEBUG_
421       AliDebug(2,Form("Refit Failed for track %d |ESDtrack#%d (MClb:%d)",itr,fCurrESDtrack->GetID(),fCurrESDtrMClb));
422 #endif
423     }
424   }    
425   //  AliExternalTrackParam::SetUseLogTermMS(uselogMS);
426   //
427   AliInfo(Form("%d ITSrefit in %d ITSout in %d ITSin tracks for %d tried TPC seeds out of %d ESD tracks\n",
428                fCountITSrefit,fCountITSout,fCountITSin,fCountProlongationTrials,fNTracksESD));
429   //
430   //  AliLog::SetClassDebugLevel("AliITSUTrackerGlo",0);
431   return 0;
432 }
433
434 //_________________________________________________________________________
435 Int_t AliITSUTrackerGlo::LoadClusters(TTree * treeRP)
436 {
437   // read from tree (if pointer provided) or directly from the ITS reco interface
438   //
439   return fReconstructor->LoadClusters(treeRP);
440
441
442 //_________________________________________________________________________
443 void AliITSUTrackerGlo::UnloadClusters()
444 {
445   //
446   // Remove clusters from the memory 
447   //
448   AliITSURecoDet *det=fReconstructor->GetITSInterface();
449   Int_t nlayers=det->GetNLayersActive();
450   for (Int_t i=0; i<nlayers; i++) {
451       TClonesArray *clusters=*(det->GetLayerActive(i)->GetClustersAddress());
452       clusters->Delete();
453   }
454
455
456 //_________________________________________________________________________
457 AliCluster * AliITSUTrackerGlo::GetCluster(Int_t /*index*/) const
458 {
459   //
460   // To be implemented 
461   //
462   
463   Info("GetCluster","To be implemented");
464   return 0x0;
465
466
467 //_________________________________________________________________________
468 Bool_t AliITSUTrackerGlo::NeedToProlong(AliESDtrack* esdTr, Int_t esdID)
469 {
470   // do we need to match this track to ITS?
471   //
472   static double bz = GetBz();
473   //
474   // is it already prolonged
475   if (esdTr->IsOn(AliESDtrack::kITSin)) return kFALSE;
476   AliITSUTrackHyp* hyp = GetTrackHyp(esdID); // in case there is unfinished hypothesis from prev.pass, clean it
477   if (hyp) {
478     CleanHypothesis(hyp);
479     if (hyp->GetSkip()) return kFALSE; // need to skip this
480     return kTRUE;
481   }
482   //
483   if (!esdTr->IsOn(AliESDtrack::kTPCin) ||
484       esdTr->IsOn(AliESDtrack::kTPCout) ||
485       esdTr->IsOn(AliESDtrack::kITSin)  ||
486       esdTr->GetKinkIndex(0)>0) return kFALSE;
487   //
488   if (esdTr->Pt()<AliITSUReconstructor::GetRecoParam()->GetMinPtForProlongation()) return kFALSE;
489   //
490   float dtz[2];
491   esdTr->GetDZ(GetX(),GetY(),GetZ(),bz,dtz); 
492   // if track is not V0 candidata but has large offset wrt IP, reject it. RS TOCHECK
493   if ( !(esdTr->GetV0Index(0)>0 && dtz[0]>AliITSUReconstructor::GetRecoParam()->GetMaxDforV0dghtrForProlongation()) 
494        && (Abs(dtz[0])>AliITSUReconstructor::GetRecoParam()->GetMaxDForProlongation() ||
495            Abs(dtz[1])>AliITSUReconstructor::GetRecoParam()->GetMaxDZForProlongation())) return kFALSE;
496   //
497   //  if (esdTr->Pt()<3) return kFALSE;//RS
498   return kTRUE;
499 }
500
501 //_________________________________________________________________________
502 void AliITSUTrackerGlo::FindTrack(AliESDtrack* esdTr, Int_t esdID)
503 {
504   // find prolongaion candidates finding for single seed
505   //
506   AliITSUSeed seedUC;  // copy of the seed from the upper layer
507   AliITSUSeed seedT;   // transient seed between the seedUC and new prolongation hypothesis
508   //
509   //  AliLog::SetClassDebugLevel("AliITSUTrackerGlo",10);
510   AliITSUTrackHyp* hypTr = InitHypothesis(esdTr,esdID);  // initialize prolongations hypotheses tree
511   if (!hypTr || hypTr->GetSkip()) return;
512   //
513   double bz = GetBz();
514   fCurrHyp = fWorkHyp;
515   fCurrHyp->InitFrom(hypTr);
516   //
517   AliITSURecoSens *hitSens[AliITSURecoLayer::kMaxSensMatching];
518   //
519   int ilaUp = fNLrActive;     // previous active layer really checked (some may be excluded!)
520   //
521   for (int ila=fNLrActive;ila--;ilaUp=fCurrActLrID) {
522     if (fCurrTrackCond->IsLayerExcluded(ila)) continue;
523     fCurrActLrID = ila;
524     fCurrLayer = fITS->GetLayerActive(ila);
525     Bool_t noClSharing = fCurrTrackCond->GetClSharing(ila)==0;
526     //
527     // for the outermost layer the seed is created from the ESD track
528     int nSeedsUp = (ilaUp==fNLrActive) ? 1 : fCurrHyp->GetNSeeds(ilaUp);
529     int maxNBranches   = fCurrTrackCond->GetMaxBranches(ila);
530     int maxNCandidates = fCurrTrackCond->GetMaxCandidates(ila);
531     //
532     for (int isd=0;isd<nSeedsUp;isd++) {
533       AliITSUSeed* seedU;
534       if (ilaUp==fNLrActive) {
535         seedU = 0;
536         seedUC.InitFromSeed(fCurrHyp->GetTPCSeed()); // init with TPC seed from ref.R
537       }
538       else {
539         seedU = fCurrHyp->GetSeed(ilaUp,isd);  // seed on prev.active layer to prolong  
540         if (seedU->IsKilled()) continue;
541         seedUC = *seedU;                       // its copy will be prolonged
542         seedUC.SetParent(seedU);        
543       }
544       seedUC.ResetFMatrix();                    // reset the matrix for propagation to next layer
545       // go till next active layer
546 #ifdef _ITSU_DEBUG_
547       AliDebug(2,Form("working on Lr:%d Seed:%d of %d for esdID=%d (MClb:%d) | pT=%.3f",ila,isd,nSeedsUp,esdID,fCurrESDtrMClb,seedUC.Pt()));
548 #endif
549       if (!TransportToLayer(&seedUC, fITS->GetLrIDActive(ilaUp), fITS->GetLrIDActive(ila)) ) { // external seed already prolonged
550         //
551 #ifdef _ITSU_DEBUG_
552         AliDebug(2,Form("Transport failed | esdID=%d (MClb:%d)",esdID,fCurrESDtrMClb));
553 #endif
554         // Check if the seed satisfies to track definition
555         if (NeedToKill(&seedUC,kTransportFailed) && seedU) KillSeed(seedU,kTRUE);
556         continue; // RS TODO: decide what to do with tracks stopped on higher layers w/o killing
557       }
558       if (!GetRoadWidth(&seedUC, ila)) { // failed to find road width on the layer
559         if (NeedToKill(&seedUC,kRWCheckFailed) && seedU) KillSeed(seedU,kTRUE);
560         continue;
561       }
562       /*
563       //RS toremove
564       int mcquest = -1;
565       if (!seedUC.ContainsFake() && AliDebugLevelClass()>2) {
566         mcquest = fCurrESDtrMClb;
567         seedUC.Print("etp");
568         printf("FSParams: "); for (int ip=0;ip<kNTrImpData;ip++) printf("%+e ",fTrImpData[ip]); printf("\n");
569       }
570       //
571       int nsens = fCurrLayer->FindSensors(&fTrImpData[kTrPhi0], hitSens, mcquest);  // find detectors which may be hit by the track
572       */
573       int nsens = fCurrLayer->FindSensors(&fTrImpData[kTrPhi0], hitSens);  // find detectors which may be hit by the track
574 #ifdef _ITSU_DEBUG_
575       AliDebug(2,Form("Will check %d sensors on lr:%d | esdID=%d (MClb:%d)",nsens,ila,esdID,fCurrESDtrMClb));
576 #endif
577       //
578       seedUC.SetLr(ila);
579       //
580       for (int isn=nsens;isn--;) {
581         AliITSURecoSens* sens = hitSens[isn];
582         int ncl = sens->GetNClusters();
583         if (!ncl) continue;
584         seedT = seedUC;
585         //
586         // We need to propagate the seed to sensor on fCurrLayer staying in the frame of the sensor from prev.layer,
587         // since the transport matrix should be defined in this frame.
588         double xs; // X in the TF of current seed, corresponding to intersection with sensor plane
589         if (!seedT.GetTrackingXAtXAlpha(sens->GetXTF(),sens->GetPhiTF(),bz, xs)) {
590 #ifdef _ITSU_DEBUG_
591           if (AliDebugLevelClass()>2) {
592             printf("Failed on GetTrackingXAtXAlpha: X=%.4f alp=%.3f\n",sens->GetXTF(),sens->GetPhiTF());
593             seedT.Print("etp");
594           }
595 #endif
596           continue;
597         }
598         if (xs<seedT.GetX()) {
599           if (!PropagateSeed(&seedT,xs,fCurrMass)) continue;
600         }
601         else { // some low precision tracks may hit the sensor plane outside of the layer radius
602 #ifdef _ITSU_DEBUG_
603           if (AliDebugLevelClass()>2) {
604             if (!seedT.ContainsFake()) {
605               printf("WRONG PROP on L%d, sens%d of %d: %.4f > %.4f\n",ila,isn,nsens,xs,seedT.GetX());
606               sens->Print();
607               seedT.Print("etp");           
608             }
609           }
610 #endif
611           if (!seedT.PropagateParamOnlyTo(xs,bz)) continue;
612         }
613         //      if (!seedT.PropagateToX(xs,bz)) continue;
614         //      if (!seedT.Rotate(sens->GetPhiTF())) continue;
615         if (!seedT.RotateToAlpha(sens->GetPhiTF())) continue;
616         //
617         int clID0 = sens->GetFirstClusterId();
618         for (int icl=ncl;icl--;) { // don't change the order, clusters are sorted
619           //AliLog::SetClassDebugLevel("AliITSUTrackerGlo",10);
620           int clID = clID0 + icl;
621           if (noClSharing && fCurrLayer->GetCluster(clID)->IsClusterUsed()) continue;
622           int res = CheckCluster(&seedT,ila,clID0+icl);
623           //AliLog::SetClassDebugLevel("AliITSUTrackerGlo", 0);
624           //
625           if (res==kStopSearchOnSensor) break;     // stop looking on this sensor
626           if (res==kClusterNotMatching) continue;  // cluster does not match
627           // cluster is matching and it was added to the hypotheses tree
628         }
629       }
630       // cluster search is done. Do we need to have a version of this seed skipping current layer
631       if (!NeedToKill(&seedUC,kMissingCluster)) {
632         AliITSUSeed* seedSkp = NewSeedFromPool(&seedUC);
633         if (nsens) { // to do: make penalty to account for probability to miss the cluster for good reason
634           double penalty = -fCurrTrackCond->GetMissPenalty(ila);
635           seedSkp->SetChi2Cl(penalty);
636         }
637         if (seedSkp->GetChi2GloNrm()>fCurrTrackCond->GetMaxChi2GloNrm(ila)) {
638           MarkSeedFree(seedSkp);
639         }
640         else AddSeedBranch(seedSkp);
641       }
642       // transfer the new branches of the seed to the hypothesis container
643       if (fNBranchesAdded) ValidateAllowedBranches(maxNBranches);
644       //
645     }
646     if (fNCandidatesAdded) ValidateAllowedCandidates(ila,maxNCandidates);
647     //    ((TObjArray*)fCurrHyp->GetLayerSeeds(ila))->Sort();
648 #ifdef _ITSU_DEBUG_
649     if (AliDebugLevelClass()>2) { //RS
650       printf(">>> All hypotheses on lr %d: \n",ila);
651       for (int ih=0;ih<fCurrHyp->GetNSeeds(ila);ih++) {printf(" #%3d ",ih); fCurrHyp->GetSeed(ila,ih)->Print();}
652     }
653 #endif
654     //
655     /*
656     for (int ih=0;ih<fCurrHyp->GetNSeeds(ila);ih++) {
657       if (ila!=0) continue;
658       double vecL[5] = {0};
659       double matL[15] = {0};
660       AliITSUSeed* sp = fCurrHyp->GetSeed(ila,ih);
661       while(sp->GetParent()) {
662         sp->Smooth(vecL,matL);
663         if (sp->GetLayerID()>=fNLrActive-1) break;
664         sp = (AliITSUSeed*)sp->GetParent();
665       }
666       */
667   }
668   //
669   SaveReducedHypothesesTree(hypTr);
670 #ifdef _ITSU_DEBUG_
671   if (AliDebugLevelClass()>1) {
672     printf("\nSaved hypotheses for esdTrack %d (MCLab:%d)\n",esdID,fCurrESDtrMClb);
673     hypTr->Print("l");
674   }
675 #endif
676   fCurrHyp = 0;
677   //  AliLog::SetClassDebugLevel("AliITSUTrackerGlo",0);
678   //
679 }
680
681 //_________________________________________________________________________
682 AliITSUTrackHyp* AliITSUTrackerGlo::InitHypothesis(AliESDtrack *esdTr, Int_t esdID)
683 {
684   // init prolongaion candidates finding for single seed
685   AliITSUTrackHyp* hyp = GetTrackHyp(esdID);
686   if (hyp) return hyp;
687   //
688   fCountProlongationTrials++;
689   //
690   fCurrMass = esdTr->GetMassForTracking();
691   //
692   hyp = new AliITSUTrackHyp(fNLrActive);
693   hyp->SetESDTrack(esdTr);
694   hyp->SetUniqueID(esdID);
695   hyp->SetMass(fCurrMass);
696   hyp->SetTPCSeed( new AliExternalTrackParam(*esdTr) );
697   SetTrackHyp(hyp,esdID);
698   Bool_t lut = fUseMatLUT; // for propagation from TPC use TGeo material budget always
699   fUseMatLUT = kFALSE;
700   if (!TransportToLayer(hyp->GetTPCSeed(),fITS->GetNLayers(), fITS->GetLrIDActive(fNLrActive-1), fITS->GetRITSTPCRef())) hyp->SetSkip(); // propagate to outer R of ITS
701   fUseMatLUT = lut;
702   //
703   return hyp;
704   // TO DO
705 }
706
707 //_________________________________________________________________________
708 Bool_t AliITSUTrackerGlo::TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo, Double_t rLim)
709 {
710   // transport seed from layerFrom to the entrance (along the direction of the track) of layerTo or to rLim (if>0), wathever is closer
711   //  
712   //
713   if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
714   //
715   int dir = lTo > lFrom ? 1:-1;
716   AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
717   Bool_t checkFirst = kTRUE;
718   Bool_t limReached = kFALSE;
719   while(lFrom!=lTo) {
720     if (lrFr) {
721       if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
722       checkFirst = kFALSE;
723     }
724     AliITSURecoLayer* lrTo =  fITS->GetLayer( (lFrom+=dir) );
725     if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
726     //
727     // go the entrance of the layer, assuming no materials in between
728     double xToGo = lrTo->GetR(-dir);
729     if (rLim>0) {
730       if (dir>0) {
731         if (rLim<xToGo) {xToGo = rLim; limReached = kTRUE;}
732       }
733       else {
734         if (rLim>xToGo) {xToGo = rLim; limReached = kTRUE;}
735       }
736     }
737     //    double xts = xToGo;
738     if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
739       //printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
740       //seed->Print("etp");
741       //      
742       return kFALSE;
743     }
744 #ifdef _ITSU_DEBUG_
745     AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
746 #endif
747     if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
748       //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
749       //seed->Print("etp");
750       return kFALSE;
751     }
752     lrFr = lrTo;
753     if (limReached) break;
754   }
755   return kTRUE;
756   //
757 }
758
759 //_________________________________________________________________________
760 Bool_t AliITSUTrackerGlo::TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t rLim)
761 {
762   // transport track from layerFrom to the entrance of layerTo or to rLim (if>0), wathever is closer
763   //  
764   if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
765   //
766   int dir = lTo > lFrom ? 1:-1;
767   AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
768   Bool_t checkFirst = kTRUE;
769   Bool_t limReached = kFALSE;
770   while(lFrom!=lTo) {
771     if (lrFr) {
772       if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
773       checkFirst = kFALSE;
774     }
775     AliITSURecoLayer* lrTo =  fITS->GetLayer( (lFrom+=dir) );
776     if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
777     //
778     // go the entrance of the layer, assuming no materials in between
779     double xToGo = lrTo->GetR(-dir);
780     if (rLim>0) {
781       if (dir>0) {
782         if (rLim<xToGo) {xToGo = rLim; limReached = kTRUE;}
783       }
784       else {
785         if (rLim>xToGo) {xToGo = rLim; limReached = kTRUE;}
786       }
787     }
788     //    double xts = xToGo;
789     if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
790       //      printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
791       //      seed->Print("etp");
792       return kFALSE;
793     }
794 #ifdef _ITSU_DEBUG_
795     AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
796 #endif
797     if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
798       //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
799       //seed->Print("etp");
800       return kFALSE;
801     }
802     lrFr = lrTo;
803     if (limReached) break;
804   }
805   return kTRUE;
806   //
807 }
808
809 //_________________________________________________________________________
810 Bool_t AliITSUTrackerGlo::TransportToLayerX(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t xStop)
811 {
812   // transport track from layerFrom to the entrance of layerTo but do not pass control parameter X
813   //  
814   if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
815   //
816   int dir = lTo > lFrom ? 1:-1;
817   AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
818   Bool_t checkFirst = kTRUE;
819   while(lFrom!=lTo) {
820     if (lrFr) {
821       if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
822       checkFirst = kFALSE;
823     }
824     AliITSURecoLayer* lrTo =  fITS->GetLayer( (lFrom+=dir) );
825     if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
826     //
827     // go the entrance of the layer, assuming no materials in between
828     double xToGo = lrTo->GetR(-dir); // R of the entrance to layer
829     //
830     //    double xts = xToGo;
831     if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
832       //      printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
833       //      seed->Print("etp");
834       return kFALSE;
835     }
836     if ( (dir>0&&xToGo>xStop) || (dir<0&&xToGo<xStop) ) xToGo = xStop;
837     //
838 #ifdef _ITSU_DEBUG_
839     AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
840 #endif
841     if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
842       //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
843       //seed->Print("etp");
844       return kFALSE;
845     }
846     lrFr = lrTo;
847   }
848   return kTRUE;
849   //
850 }
851
852 //_________________________________________________________________________
853 Bool_t AliITSUTrackerGlo::GoToExitFromLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
854 {
855   // go to the exit from lr in direction dir, applying material corrections in steps specific for this layer
856   // If check is requested, do this only provided the track has not exited the layer already
857   double xToGo = lr->GetR(dir);
858   if (check) { // do we need to track till the surface of the current layer ?
859     double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
860 #ifdef _ITSU_DEBUG_
861     AliDebug(3,Form(" dir:%d Cur: %e Tgt: %e",dir,Sqrt(curR2),xToGo));
862 #endif
863     if      (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // on the surface or outside of the layer
864     else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // on the surface or outside of the layer
865   }
866   if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
867   // go via layer to its boundary, applying material correction.
868   if (!PropagateSeed(seed,xToGo,fCurrMass, lr->GetMaxStep())) return kFALSE;
869   //
870   return kTRUE;
871   //
872 }
873
874 //_________________________________________________________________________
875 Bool_t AliITSUTrackerGlo::GoToExitFromLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
876 {
877   // go to the exit from lr in direction dir, applying material corrections in steps specific for this layer
878   // If check is requested, do this only provided the track has not exited the layer already
879   double xToGo = lr->GetR(dir);
880   if (check) { // do we need to track till the surface of the current layer ?
881     double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
882     if      (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // on the surface or outside of the layer
883     else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // on the surface or outside of the layer
884   }
885 #ifdef _ITSU_DEBUG_
886   AliDebug(2,Form(" dir=%d : from R=%.4f -> R=%.4f",dir,Sqrt(seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY()), xToGo));
887 #endif
888   if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
889   // go via layer to its boundary, applying material correction.
890   if (!PropagateSeed(seed,xToGo,fCurrMass, lr->GetMaxStep())) return kFALSE;
891   //
892   return kTRUE;
893   //
894 }
895
896
897 //_________________________________________________________________________
898 Bool_t AliITSUTrackerGlo::GoToEntranceToLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
899 {
900   // go to the entrance of lr in direction dir, w/o applying material corrections.
901   // If check is requested, do this only provided the track did not reach the layer already
902   double xToGo = lr->GetR(-dir);
903   if (check) { // do we need to track till the surface of the current layer ?
904     double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
905     if      (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // already passed
906     else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // already passed
907   }
908   if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
909   // go via layer to its boundary, applying material correction.
910   if (!PropagateSeed(seed,xToGo,fCurrMass, 100, kFALSE)) return kFALSE;
911   return kTRUE;
912   //
913 }
914
915 //_________________________________________________________________________
916 Bool_t AliITSUTrackerGlo::GoToEntranceToLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
917 {
918   // go to the entrance of lr in direction dir, w/o applying material corrections.
919   // If check is requested, do this only provided the track did not reach the layer already
920   double xToGo = lr->GetR(-dir);
921   if (check) { // do we need to track till the surface of the current layer ?
922     double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
923     if      (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // already passed
924     else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // already passed
925   }
926   if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
927   // go via layer to its boundary, applying material correction.
928   if (!PropagateSeed(seed,xToGo,fCurrMass, 100, kFALSE)) return kFALSE;
929   return kTRUE;
930   //
931 }
932
933 //_________________________________________________________________________
934 Bool_t AliITSUTrackerGlo::GetRoadWidth(AliITSUSeed* seed, int ilrA)
935 {
936   // calculate road width in terms of phi and z for the track which MUST be on the external radius of the layer
937   // as well as some aux info
938   double bz = GetBz();
939   AliITSURecoLayer* lrA = fITS->GetLayerActive(ilrA);
940   seed->GetXYZ(&fTrImpData[kTrXIn]);    // lab position at the entrance from above
941   static AliExternalTrackParam sc;   // seed copy for manipulations
942   sc = *seed;
943   //
944   fTrImpData[kTrPhiIn] = ATan2(fTrImpData[kTrYIn],fTrImpData[kTrXIn]);
945   if (!sc.Rotate(fTrImpData[kTrPhiIn])) return kFALSE; // go to the frame of the entry point into the layer
946   BringTo02Pi(fTrImpData[kTrPhiIn]);
947   double dr  = lrA->GetDR();                              // approximate X dist at the inner radius
948   if (!sc.GetXYZAt(sc.GetX()-dr, bz, fTrImpData + kTrXOut)) {
949     // special case: track does not reach inner radius, might be tangential
950     double r = sc.GetD(0,0,bz);
951     double x;
952     if (!sc.GetXatLabR(r,x,bz,-1)) return kFALSE;
953     dr = Abs(sc.GetX() - x);
954     if (!sc.GetXYZAt(x, bz, fTrImpData + kTrXOut)) return kFALSE;
955   }
956   //
957   fTrImpData[kTrPhiOut] = ATan2(fTrImpData[kTrYOut],fTrImpData[kTrXOut]);
958   BringTo02Pi(fTrImpData[kTrPhiOut]);
959   double sgy = sc.GetSigmaY2() + dr*dr*sc.GetSigmaSnp2() + AliITSUReconstructor::GetRecoParam()->GetSigmaY2(ilrA);
960   double sgz = sc.GetSigmaZ2() + dr*dr*sc.GetSigmaTgl2() + AliITSUReconstructor::GetRecoParam()->GetSigmaZ2(ilrA);
961   sgy = Sqrt(sgy)*fCurrTrackCond->GetNSigmaRoadY(ilrA);
962   sgz = Sqrt(sgz)*fCurrTrackCond->GetNSigmaRoadZ(ilrA);
963   double phi0  = MeanPhiSmall(fTrImpData[kTrPhiOut],fTrImpData[kTrPhiIn]);
964   double dphi0 = DeltaPhiSmall(fTrImpData[kTrPhiOut],fTrImpData[kTrPhiIn]);
965   //
966   fTrImpData[kTrPhi0] = phi0;
967   fTrImpData[kTrZ0]   = 0.5*(fTrImpData[kTrZOut]+fTrImpData[kTrZIn]);
968   dphi0 += sgy/lrA->GetR();
969   fTrImpData[kTrDPhi] =  dphi0<PiOver2() ? dphi0 : PiOver2();
970   fTrImpData[kTrDZ]   = 0.5*Abs(fTrImpData[kTrZOut]-fTrImpData[kTrZIn])   + sgz;
971   //  
972   return kTRUE;
973 }
974
975 //________________________________________
976 void AliITSUTrackerGlo::ResetSeedsPool()
977 {
978   // mark all seeds in the pool as unused
979   AliDebug(-1,Form("CurrentSize: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
980   fNFreeSeeds = 0;
981   fSeedsPool.Clear(); // seeds don't allocate memory
982 }
983
984
985 //________________________________________
986 void AliITSUTrackerGlo::MarkSeedFree(AliITSUSeed *sd) 
987 {
988   // account that this seed is "deleted" 
989   int id = sd->GetPoolID();
990   if (id<0) {
991     AliError(Form("Freeing of seed %p NOT from the pool is requested",sd)); 
992     return;
993   }
994   //  AliInfo(Form("%d %p",id, seed));
995   fSeedsPool.RemoveAt(id);
996   if (fFreeSeedsID.GetSize()<=fNFreeSeeds) fFreeSeedsID.Set( 2*fNFreeSeeds + 100 );
997   fFreeSeedsID.GetArray()[fNFreeSeeds++] = id;
998 }
999
1000 //_________________________________________________________________________
1001 Int_t AliITSUTrackerGlo::CheckCluster(AliITSUSeed* track, Int_t lr, Int_t clID) 
1002 {
1003   // Check if the cluster (in tracking frame!) is matching to track. 
1004   // The track must be already propagated to sensor tracking frame.
1005   // Returns:  kStopSearchOnSensor if the search on given sensor should be stopped, 
1006   //           kClusterMatching    if the cluster is matching
1007   //           kClusterMatching    otherwise
1008   //
1009   const double kTolerX = 5e-4;
1010   //  AliCluster *cl = fITS->GetLayerActive(lr)->GetCluster(clID);
1011   AliCluster *cl = fCurrLayer->GetCluster(clID);
1012   //
1013   Bool_t goodCl = kFALSE;
1014   int currLabel = Abs(fCurrESDtrMClb);
1015   //
1016   if (cl->GetLabel(0)>=0) {
1017     for (int i=0;i<3;i++) if (cl->GetLabel(i)>=0 && cl->GetLabel(i)==currLabel) {goodCl = kTRUE; break;}
1018   }
1019   //
1020   if (TMath::Abs(cl->GetX())>kTolerX) { // if due to the misalingment X is large, propagate track only
1021     if (!track->PropagateParamOnlyTo(track->GetX()+cl->GetX(),GetBz())) {
1022       //
1023 #ifdef _ITSU_DEBUG_
1024       if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1025         AliDebug(2,Form("Lost good cl on L:%d failed propagation. |ESDtrack#%d (MClb:%d)",lr,fCurrESDtrack->GetID(),fCurrESDtrMClb)); 
1026         track->Print("etp");
1027         cl->Print();
1028       }
1029 #endif
1030       return kStopSearchOnSensor; // propagation failed, seedT is intact
1031     }
1032   }
1033   double dy = cl->GetY()-track->GetY();
1034   double dz = cl->GetZ()-track->GetZ();
1035   //
1036   double dy2 = dy*dy;
1037   double tol2 = (track->GetSigmaY2() + AliITSUReconstructor::GetRecoParam()->GetSigmaY2(lr))*
1038     fCurrTrackCond->GetNSigmaRoadY(lr)*fCurrTrackCond->GetNSigmaRoadY(lr); // RS TOOPTIMIZE
1039   if (dy2>tol2) {                          // the clusters are sorted in Z(col) then in Y(row). 
1040     //
1041 #ifdef _ITSU_DEBUG_
1042     if (AliDebugLevelClass()>2 && goodCl &&  !track->ContainsFake()) {    
1043       AliDebug(2,Form("Lost good cl: dy2=%e > tol2=%e |ESDtrack#%d (MClb:%d)",dy2,tol2,fCurrESDtrack->GetID(),fCurrESDtrMClb)); 
1044       track->Print("etp");
1045       cl->Print();
1046       PrintSeedClusters(track);
1047     }
1048 #endif
1049     // clusters are sorted in increasing Y and the loop goes from last (highers Y) to 1st.
1050     // if the track has too large y for this cluster (dy<0) it will be so for all other clusters of the sensor
1051     if (dy<0) return kStopSearchOnSensor;  // No chance that other cluster of this sensor will match (all Y's will be even larger)
1052     else      return kClusterNotMatching;   // Other clusters may match
1053   }
1054   double dz2 = dz*dz;
1055   tol2 = (track->GetSigmaZ2() + AliITSUReconstructor::GetRecoParam()->GetSigmaZ2(lr))*
1056     fCurrTrackCond->GetNSigmaRoadZ(lr)*fCurrTrackCond->GetNSigmaRoadZ(lr); // RS TOOPTIMIZE
1057   if (dz2>tol2) {
1058 #ifdef _ITSU_DEBUG_
1059     if (AliDebugLevelClass()>2 && goodCl &&  !track->ContainsFake()) {
1060       AliDebug(2,Form("Lost good cl on L:%d : dz2=%e > tol2=%e |ESDtrack#%d (MClb:%d)",lr,dz2,tol2,fCurrESDtrack->GetID(),fCurrESDtrMClb)); 
1061       track->Print("etp");
1062       cl->Print();
1063       PrintSeedClusters(track);
1064     }
1065 #endif
1066     return kClusterNotMatching; // Other clusters may match
1067   }
1068   //
1069   // check chi2
1070   Double_t p[2]={cl->GetY(), cl->GetZ()};
1071   Double_t cov[3]={cl->GetSigmaY2(), cl->GetSigmaYZ(), cl->GetSigmaZ2()};
1072   double chi2 = track->GetPredictedChi2(p,cov);
1073   //
1074   if (chi2>fCurrTrackCond->GetMaxTr2ClChi2(lr)) {
1075 #ifdef _ITSU_DEBUG_
1076     if (AliDebugLevelClass()>2 && goodCl &&  !track->ContainsFake()) {
1077       AliDebug(2,Form("Lost good cl on L:%d : Chi2=%e > Chi2Max=%e |dy: %+.3e dz: %+.3e |ESDtrack#%d (MClb:%d)\n",
1078                       lr,chi2,fCurrTrackCond->GetMaxTr2ClChi2(lr),dy,dz,fCurrESDtrack->GetID(),fCurrESDtrMClb)); 
1079       track->Print("etp");
1080       cl->Print("");
1081       PrintSeedClusters(track);
1082     }
1083 #endif
1084     return kClusterNotMatching;
1085   }
1086   //
1087   track = NewSeedFromPool(track);  // input track will be reused, use its clone for updates
1088   if (!track->Update()) {
1089 #ifdef _ITSU_DEBUG_
1090     if (AliDebugLevelClass()>2 && goodCl &&  !track->ContainsFake()) {
1091       AliDebug(2,Form("Lost good cluster on L:%d : failed to update",lr));
1092       track->Print("etp");
1093       cl->Print("");
1094       PrintSeedClusters(track);
1095     }
1096 #endif
1097     MarkSeedFree(track);
1098     return kClusterNotMatching;
1099   }
1100   track->SetChi2Cl(chi2);
1101   track->SetLrClusterID(lr,clID);
1102   //
1103   if (track->GetChi2GloNrm()>fCurrTrackCond->GetMaxChi2GloNrm(lr)) {
1104 #ifdef _ITSU_DEBUG_
1105     if (AliDebugLevelClass()>2 && goodCl &&  !track->ContainsFake()) {
1106       AliDebug(2,Form("Lost good cl on L:%d : Chi2Glo=%e > Chi2Max=%e |dy: %+.3e dz: %+.3e |ESDtrack#%d (MClb:%d)\n",
1107                       lr,track->GetChi2GloNrm(),fCurrTrackCond->GetMaxChi2GloNrm(lr),dy,dz,fCurrESDtrack->GetID(),fCurrESDtrMClb)); 
1108       track->Print("etp");
1109       cl->Print("");
1110       PrintSeedClusters(track);
1111     }
1112 #endif
1113     MarkSeedFree(track);
1114     return kClusterNotMatching;
1115   }
1116   //  cl->IncreaseClusterUsage(); // do this only for winners
1117   //
1118   track->SetFake(!goodCl);
1119   //
1120 #ifdef _ITSU_DEBUG_
1121   AliDebug(2,Form("AddCl(%d) Cl%d lr:%d: dY:%+8.4f dZ:%+8.4f (MC: %5d %5d %5d) |Chi2=%f(%c)| ",
1122                goodCl,clID,lr,dy,dz2,cl->GetLabel(0),cl->GetLabel(1),cl->GetLabel(2), chi2, track->IsFake() ? '-':'+'));
1123 #endif
1124   //
1125   AddSeedBranch(track);
1126   //
1127   return kClusterMatching;
1128 }
1129
1130 //_________________________________________________________________________
1131 Bool_t AliITSUTrackerGlo::NeedToKill(AliITSUSeed *seed, Int_t flag)
1132 {
1133   // check if the seed should not be discarded
1134   const UShort_t kMask = 0xffff;
1135   if (flag==kMissingCluster) {
1136     int lastChecked = seed->GetLayerID();
1137     UShort_t patt = seed->GetHitsPattern();
1138     if (lastChecked) patt |= (~(kMask<<lastChecked))&fCurrTrackCond->GetAllowedLayers(); // not all layers were checked, complete unchecked ones by potential hits
1139     Bool_t seedOK = fCurrTrackCond->CheckPattern(patt);
1140     return !seedOK;
1141   }
1142   return kTRUE;
1143 }
1144
1145 //______________________________________________________________________________
1146 Bool_t AliITSUTrackerGlo::PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep, Bool_t matCorr) 
1147 {
1148   // propagate seed to given x applying material correction if requested
1149   const Double_t kEpsilon = 1e-5;
1150   Double_t xpos     = seed->GetX();
1151   Int_t dir         = (xpos<xToGo) ? 1:-1;
1152   Double_t xyz0[3],xyz1[3];
1153   //
1154   Bool_t updTime = dir>0 && seed->IsStartedTimeIntegral();
1155   if (matCorr || updTime) seed->GetXYZ(xyz1);   //starting global position
1156   while ( (xToGo-xpos)*dir > kEpsilon){
1157     Double_t step = dir*TMath::Min(TMath::Abs(xToGo-xpos), maxStep);
1158     Double_t x    = xpos+step;
1159     Double_t bz=GetBz();   // getting the local Bz
1160     if (!seed->PropagateToX(x,bz)) return kFALSE;
1161     double ds = 0;
1162     if (matCorr || updTime) {
1163       xyz0[0]=xyz1[0]; // global pos at the beginning of step
1164       xyz0[1]=xyz1[1];
1165       xyz0[2]=xyz1[2];
1166       seed->GetXYZ(xyz1);    //  // global pos at the end of step
1167       if (matCorr) {
1168         Double_t xrho,xx0;
1169         ds = GetMaterialBudget(xyz0,xyz1,xx0,xrho);     
1170         if (dir>0) xrho = -xrho; // outward should be negative
1171         if (!seed->ApplyMaterialCorrection(xx0,xrho,mass,kFALSE)) {AliInfo("Fail2"); seed->Print("etp"); return kFALSE;}
1172       }
1173        else { // matCorr is not requested but time integral is
1174         double d0 = xyz1[0]-xyz0[0];
1175         double d1 = xyz1[1]-xyz0[1];
1176         double d2 = xyz1[2]-xyz0[2];    
1177         ds = TMath::Sqrt(d0*d0+d1*d1+d2*d2);
1178       }     
1179     }
1180     if (updTime) seed->AddTimeStep(ds);
1181     xpos = seed->GetX();
1182   }
1183   return kTRUE;
1184 }
1185
1186 //______________________________________________________________________________
1187 Bool_t AliITSUTrackerGlo::PropagateSeed(AliExternalTrackParam *seed, Double_t xToGo, Double_t mass, Double_t maxStep, Bool_t matCorr) 
1188 {
1189   // propagate seed to given x applying material correction if requested
1190   const Double_t kEpsilon = 1e-5;
1191   Double_t xpos     = seed->GetX();
1192   Int_t dir         = (xpos<xToGo) ? 1:-1;
1193   Double_t xyz0[3],xyz1[3];
1194   //
1195 #ifdef _ITSU_DEBUG_
1196   if (AliDebugLevelClass()>1) {
1197     AliDebug(2,Form("Before Propagate to X=%f with M=%.3f MaxStep=%.4f MatCorr=%d",xToGo,mass,maxStep,matCorr));
1198     seed->AliExternalTrackParam::Print();
1199   }
1200 #endif
1201   Bool_t updTime = dir>0 && seed->IsStartedTimeIntegral();
1202   if (matCorr || updTime) seed->GetXYZ(xyz1);   //starting global position
1203   while ( (xToGo-xpos)*dir > kEpsilon){
1204     Double_t step = dir*TMath::Min(TMath::Abs(xToGo-xpos), maxStep);
1205     Double_t x    = xpos+step;
1206     Double_t bz=GetBz();   // getting the local Bz
1207     if (!seed->PropagateTo(x,bz))  return kFALSE;
1208     double ds = 0;
1209     if (matCorr || updTime) {
1210       xyz0[0]=xyz1[0]; // global pos at the beginning of step
1211       xyz0[1]=xyz1[1];
1212       xyz0[2]=xyz1[2];
1213       seed->GetXYZ(xyz1);    //  // global pos at the end of step
1214       //
1215       if (matCorr) {
1216         Double_t xrho,xx0;
1217         ds = GetMaterialBudget(xyz0,xyz1,xx0,xrho);     
1218         if (dir>0) xrho = -xrho; // outward should be negative
1219         if (!seed->CorrectForMeanMaterial(xx0,xrho,mass)) return kFALSE;
1220       }
1221       else { // matCorr is not requested but time integral is
1222         double d0 = xyz1[0]-xyz0[0];
1223         double d1 = xyz1[1]-xyz0[1];
1224         double d2 = xyz1[2]-xyz0[2];    
1225         ds = TMath::Sqrt(d0*d0+d1*d1+d2*d2);
1226       }
1227     }
1228     if (updTime) seed->AddTimeStep(ds);
1229     //
1230     xpos = seed->GetX();
1231   }
1232 #ifdef _ITSU_DEBUG_
1233   if (AliDebugLevelClass()>1) {
1234     AliDebug(2,Form("After Propagate to X=%f",xToGo));
1235     seed->AliExternalTrackParam::Print();
1236   }
1237 #endif
1238   return kTRUE;
1239 }
1240
1241 //______________________________________________________________________________
1242 Bool_t AliITSUTrackerGlo::FinalizeHypothesis(AliITSUTrackHyp* hyp)
1243 {
1244   // finalize single TPC track hypothesis
1245   if (!hyp || hyp->GetESDTrack()->IsOn(AliESDtrack::kITSin)) return kFALSE;
1246   AliITSUSeed* winner = 0;
1247   fCurrHyp = hyp;
1248   fCurrMass = hyp->GetMass();
1249   if (!(winner=hyp->DefineWinner(fCurrTrackCond->GetActiveLrInner()))) return kFALSE;
1250   CookMCLabel(hyp);
1251   //
1252 #ifdef  _ITSU_TUNING_MODE_  // fill tuning histos
1253   TObjArray* dest = hyp->GetLabel()>=0 ? fCHistoArrCorr : fCHistoArrFake;
1254   if (dest) {
1255     AliITSUSeed* sd = winner;
1256     double htrPt = hyp->Pt();
1257     do {
1258       int clID,lrID;
1259       if ( (clID=sd->GetLrCluster(lrID))<0 ) continue;
1260       ((TH2*)dest->At( GetHistoID(lrID,kHChi2Nrm     ,fCurrPassID) ))->Fill(htrPt,sd->GetChi2GloNrm());
1261       ((TH2*)dest->At( GetHistoID(lrID,kHBestInBranch,fCurrPassID) ))->Fill(htrPt,sd->GetOrdBranch());
1262       ((TH2*)dest->At( GetHistoID(lrID,kHBestInCand  ,fCurrPassID) ))->Fill(htrPt,sd->GetOrdCand());
1263       //
1264       if (dest==fCHistoArrFake && !sd->IsFake()) continue; // for the fake seeds fill only fake clusters part
1265       //
1266       ((TH2*)dest->At( GetHistoID(lrID,kHResY        ,fCurrPassID) ))->Fill(htrPt,sd->GetResidY());
1267       ((TH2*)dest->At( GetHistoID(lrID,kHResZ        ,fCurrPassID) ))->Fill(htrPt,sd->GetResidZ());
1268       ((TH2*)dest->At( GetHistoID(lrID,kHResYP       ,fCurrPassID) ))->Fill(htrPt,sd->GetPullY());
1269       ((TH2*)dest->At( GetHistoID(lrID,kHResZP       ,fCurrPassID) ))->Fill(htrPt,sd->GetPullZ());
1270       ((TH2*)dest->At( GetHistoID(lrID,kHChi2Cl      ,fCurrPassID) ))->Fill(htrPt,sd->GetChi2Cl());
1271       //
1272     } while((sd=(AliITSUSeed*)sd->GetParent()));
1273     //
1274     ((TH2*)dest->At( GetHistoID(-1,kHChiMatch,fCurrPassID) ))->Fill(htrPt,winner->GetChi2ITSTPC());
1275     ((TH2*)dest->At( GetHistoID(-1,kHChiITSSA,fCurrPassID) ))->Fill(htrPt,winner->GetChi2ITSSA());
1276   }
1277   //
1278 #endif
1279   //
1280   CheckClusterSharingConflicts(hyp);
1281   return hyp->GetWinner();  // winner might change of disappear after resolving conflicts
1282   //
1283 }
1284
1285 //______________________________________________________________________________
1286 void AliITSUTrackerGlo::FinalizeHypotheses()
1287 {
1288   // select winner for each hypothesis, remove cl. sharing conflicts
1289   AliCodeTimerAuto("",0); 
1290   //
1291   int* esdTrackIndex = fESDIndex.GetArray();
1292   for (int ih=0;ih<fNTracksESD;ih++) FinalizeHypothesis(GetTrackHyp(esdTrackIndex[ih])); // finlize in decreasing pt order
1293   //
1294   for (int ih=fNTracksESD;ih--;) UpdateESDTrack(GetTrackHyp(esdTrackIndex[ih]),AliESDtrack::kITSin); 
1295   //
1296 }
1297
1298 //______________________________________________________________________________
1299 void AliITSUTrackerGlo::CheckClusterSharingConflicts(AliITSUTrackHyp* hyp)
1300 {
1301   // remove eventual cluster sharing conflicts
1302   AliITSUSeed* winner = 0;
1303   if (!(winner=hyp->GetWinner())) return;
1304   UShort_t idH = (UShort_t)hyp->GetUniqueID()+1;
1305   int lrID,clID;
1306   int inLr = fCurrTrackCond->GetActiveLrInner();
1307   AliITSUSeed* winSD=winner;
1308   do {
1309     if ( (clID=winSD->GetLrCluster(lrID))<0 ) continue;
1310     AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1311     Int_t refID = cl->GetRecoInfo();                // was it already referred by some track (with refID-1 if >0)
1312     if (!refID) {cl->SetRecoInfo(idH); continue;}   // if not, refer from track to cluster
1313     if (refID==idH) continue;                          // ignore reference to itself
1314     // the cluster is already used by other track, need to resolve the conflict
1315     AliITSUTrackHyp* hypC = GetTrackHyp(refID-1);   // competitor
1316     //    printf("Ref to %d (%p) from %d Cl %d %d\n",refID-1,hypC, idH-1,clID,lrID);
1317     AliITSUSeed* winnerC = hypC->GetWinner();
1318     if (!winnerC) {
1319       printf("Missing winner, ERROR\n");
1320       continue;
1321     }
1322     //
1323 #ifdef _ITSU_DEBUG_
1324     if (AliDebugLevelClass()>1) { //------------------------ DEBUG -------------------
1325       AliInfo(Form("Shared cl#%4d on Lr:%d: Hyp%3d/Q:%6.3f vs Hyp%3d/Q:%6.3f",
1326                    clID,lrID,idH-1,winner->GetQualityVar(),refID-1,winnerC->GetQualityVar()));
1327       // dump winner of hyp
1328       printf("#%4d Pt:%.3f Chi:%6.3f/%6.3f/%6.3f Ncl:%d MCits%+5d MCtpc:%+5d |",
1329              idH-1,winner->Pt(),winner->GetChi2GloNrm(),winner->GetChi2ITSTPC(),winner->GetChi2ITSSA(),
1330              winner->GetNLayersHit(),hyp->GetITSLabel(),hyp->GetESDTrack()->GetTPCLabel());
1331       int prevL=-1;
1332       AliITSUSeed* sd = winner;
1333       do {
1334         int lrs;
1335         int clIDt = sd->GetLrCluster(lrs);
1336         if (clIDt<0) continue;
1337         while( lrs>++prevL ) printf("%4s        ","----");
1338         printf("%4d (%5.1f)",clIDt,sd->GetChi2Cl());
1339       } while ((sd=(AliITSUSeed*)sd->GetParent()));
1340       printf("|\n");
1341       // dump winner of hypC
1342       printf("#%4d Pt:%.3f Chi:%6.3f/%6.3f/%6.3f Ncl:%d MCits%+5d MCtpc:%+5d |",
1343              refID-1,winnerC->Pt(),winnerC->GetChi2GloNrm(),winnerC->GetChi2ITSTPC(),winnerC->GetChi2ITSSA(),
1344              winnerC->GetNLayersHit(),hypC->GetITSLabel(),hypC->GetESDTrack()->GetTPCLabel());
1345       prevL=-1;
1346       sd = winnerC;
1347       do {
1348         int lrs;
1349         int clIDt = sd->GetLrCluster(lrs);
1350         if (clIDt<0) continue;
1351         while( lrs>++prevL ) printf("%4s        ","----");
1352         printf("%4d (%5.1f)",clIDt,sd->GetChi2Cl());
1353       } while ((sd=(AliITSUSeed*)sd->GetParent()));
1354       printf("|\n");
1355     }  //-------------------------------------------------- DEBUG -------------------
1356 #endif
1357     //
1358     if (winner->GetQualityVar()<winnerC->GetQualityVar()) { // current tracks is better then competitor track
1359       FlagSeedClusters(winnerC,kFALSE,refID); // unflag cluster usage by loser
1360       cl->SetRecoInfo(idH);
1361       //
1362       //if ( hypC->GetLabel()>=0) AliInfo(Form("KILLING CORRECT: %d",refID-1));
1363       winnerC->Kill();
1364       if (hypC->DefineWinner(inLr)) {             // find new winner instead of suppressed candidate
1365         CookMCLabel(hypC);
1366         CheckClusterSharingConflicts(hypC);   // and check its sharing conflicts again
1367         if (winner->IsKilled()) break;        // the current winner might have been killed during check of new winner of hypC!
1368       }
1369     }    
1370     else { // competitor hypC is better than the hyp
1371       FlagSeedClusters(winner,kFALSE,idH); // unflag cluster usage by loser
1372       winner->Kill();
1373       //if ( hyp->GetLabel()>=0) AliInfo(Form("KILLING CORRECT: %d",idH-1));
1374       if (hyp->DefineWinner(inLr)) {
1375         CookMCLabel(hyp);
1376         CheckClusterSharingConflicts(hyp);
1377       }
1378       break;
1379     }
1380     //   
1381   } while ((winSD=(AliITSUSeed*)winSD->GetParent()));
1382   //
1383   return;
1384 }
1385
1386 //______________________________________________________________________________
1387 void AliITSUTrackerGlo::UpdateESDTrack(AliITSUTrackHyp* hyp,Int_t flag)
1388 {
1389   // update ESD track with current best hypothesis
1390   if (!hyp) return;
1391   AliESDtrack* esdTr = hyp->GetESDTrack();
1392   if (!esdTr || esdTr->IsOn(flag)) return;
1393   AliITSUSeed* win = hyp->GetWinner();
1394   if (!win || win->IsKilled()) return;
1395   double chiSav;
1396   //
1397   switch (flag) {
1398   case AliESDtrack::kITSin: 
1399     esdTr->UpdateTrackParams(hyp,flag); // update kinematics
1400     fCountITSin++;
1401     // set fakes cluster info
1402     {
1403       UShort_t clfake = 0;
1404       Int_t    clSplit = 0;
1405       AliITSUSeed* sd = win;
1406       int ip=0;
1407       do {
1408         int lr, clID = sd->GetLrCluster(lr);
1409         if (sd->IsFake()) clfake |= 0x1<<lr;
1410         if (clID>=0) {
1411           esdTr->SetITSModuleIndex(ip++, sd->GetLrClusterID());
1412           AliITSUClusterPix *cl = (AliITSUClusterPix*)fITS->GetLayerActive(lr)->GetCluster(clID);
1413 #ifdef  _ITSU_TUNING_MODE_
1414           if (cl->IsSplit()) clSplit |= 0x1<<lr;
1415 #endif
1416         }
1417       } while ((sd=(AliITSUSeed*)sd->GetParent()));
1418       //
1419       // RS: Temporary set special flag for tracks from the afterburner
1420       if (fCurrPassID>0) clfake |= 0x1<<7;
1421       //
1422       esdTr->SetITSSharedMap(clfake);
1423       esdTr->SetITSModuleIndex(10,clSplit);
1424     }
1425     // TEMPORARY: store iteration id
1426     esdTr->SetITSModuleIndex(11,fCurrPassID);
1427     break;
1428     //
1429   case AliESDtrack::kITSout: 
1430     // here the stored chi2 will correspond to backward ITS-SA fit
1431     esdTr->UpdateTrackParams(hyp,flag); // update kinematics
1432     fCountITSout++;
1433     // TODO: avoid setting friend
1434     break;
1435     //
1436   case AliESDtrack::kITSrefit: 
1437
1438     // at the moment fill as a chi2 the TPC-ITS matching chi2
1439     chiSav = hyp->GetChi2();
1440     hyp->SetChi2(win->GetChi2ITSTPC());
1441     esdTr->UpdateTrackParams(hyp,flag); // update kinematics
1442     hyp->SetChi2(chiSav);
1443     fCountITSrefit++;
1444     // TODO: avoid setting cluster info
1445     break;
1446   default:
1447     AliFatal(Form("Unknown flag %d",flag));
1448   }
1449   //
1450   esdTr->SetITSLabel(hyp->GetITSLabel());
1451   // transfer chip indices
1452   // TODO
1453 }
1454
1455 //______________________________________________________________________________
1456 Double_t AliITSUTrackerGlo::RefitTrack(AliITSUTrackHyp* trc, Double_t rDest, Int_t &nclFit, Int_t stopCond)
1457 {
1458   // refit track till radius rDest. The cluster,mass info is taken from fCurrHyp (and its winner seed)
1459   // if stopCond<0 : propagate till last cluster then stop
1460   // if stopCond==0: propagate till last cluster then try to go till limiting rDest, don't mind if fail
1461   // if stopCond>0 : rDest must be reached
1462   //
1463   double rCurr = Sqrt(trc->GetX()*trc->GetX() + trc->GetY()*trc->GetY());
1464   int dir,lrStart,lrStop;
1465   //
1466   dir = rCurr<rDest ? 1 : -1;
1467   lrStart = fITS->FindFirstLayerID(rCurr,dir);
1468   lrStop  = fITS->FindLastLayerID(rDest,dir); // lr id before which we have to stop
1469   //
1470 #ifdef _ITSU_DEBUG_
1471   if (AliDebugLevelClass()>2) {
1472     printf("Refit %d: Lr:%d (%f) -> Lr:%d (%f)\n",dir,lrStart,rCurr, lrStop,rDest);
1473     printf("Before refit: "); trc->Print();
1474   }
1475 #endif
1476   if (lrStop<0 || lrStart<0) AliFatal(Form("Failed to find start(%d) or last(%d) layers. Track from %.3f to %.3f",lrStart,lrStop,rCurr,rDest));
1477   //
1478   Int_t clInfo[2*AliITSUAux::kMaxLayers];
1479   Int_t nCl = fCurrHyp->FetchClusterInfo(clInfo);
1480   fCurrMass = fCurrHyp->GetMass();
1481   AliITSUTrackHyp tmpTr(*(AliKalmanTrack*)trc);
1482   double chi2 = 0;
1483   int iclLr[2],nclLr;
1484   nclFit = 0;
1485   //
1486   int lrStop1 = lrStop+dir;
1487   for (int ilr=lrStart;ilr!=lrStop1;ilr+=dir) {
1488     AliITSURecoLayer* lr = fITS->GetLayer(ilr);
1489     if ( dir*(rCurr-lr->GetR(dir))>0) continue; // this layer is already passed
1490     int ilrA2,ilrA = lr->GetActiveID();
1491     // passive layer or active w/o hits will be traversed on the way to next cluster
1492     if (!lr->IsActive() || clInfo[ilrA2=(ilrA<<1)]<0) continue; 
1493     //
1494     // select the order in which possible 2 clusters (in case of the overlap) will be traversed and fitted
1495     nclLr=0;
1496     if (dir>0) { // clusters are stored in increasing radius order
1497       iclLr[nclLr++]=clInfo[ilrA2++];
1498       if (clInfo[ilrA2]>=0) iclLr[nclLr++]=clInfo[ilrA2];
1499     }
1500     else {
1501       if ( clInfo[ilrA2+1]>=0 ) iclLr[nclLr++]=clInfo[ilrA2+1];
1502       iclLr[nclLr++]=clInfo[ilrA2];
1503     }
1504     //
1505     Bool_t transportedToLayer = kFALSE;
1506     for (int icl=0;icl<nclLr;icl++) {
1507       AliITSUClusterPix* clus =  (AliITSUClusterPix*)lr->GetCluster(iclLr[icl]);
1508       AliITSURecoSens* sens = lr->GetSensorFromID(clus->GetVolumeId());
1509       if (!tmpTr.Rotate(sens->GetPhiTF())) {
1510 #ifdef _ITSU_DEBUG_
1511         AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1512         AliDebug(2,Form("Failed on rotate to %f | ESDtrack#%d (MClb:%d)",sens->GetPhiTF(),trESD->GetID(),trESD->GetTPCLabel()));
1513 #endif
1514         return -1;
1515       }
1516       //
1517       double xClus = sens->GetXTF()+clus->GetX();
1518       if (!transportedToLayer) {
1519         if (ilr!=lrStart && !TransportToLayerX(&tmpTr,lrStart,ilr,xClus)) {
1520 #ifdef _ITSU_DEBUG_
1521           AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1522           AliDebug(2,Form("Failed to transport %d -> %d | ESDtrack#%d (MClb:%d)\n",lrStart,ilr,trESD->GetID(),trESD->GetTPCLabel()));
1523 #endif
1524           return -1; // go to the entrance to the layer
1525         }
1526         lrStart = ilr;
1527         transportedToLayer = kTRUE;
1528       }
1529       //
1530 #ifdef _ITSU_DEBUG_
1531       if (AliDebugLevelClass()>1) {
1532         AliDebug(2,Form("Propagate to cl:%d on lr %d Need to go %.4f -> %.4f",icl,ilrA,tmpTr.GetX(),xClus));
1533       }
1534 #endif
1535       //
1536       if (!PropagateSeed(&tmpTr,xClus,fCurrMass)) {
1537 #ifdef _ITSU_DEBUG_
1538         AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1539         AliDebug(2,Form("Failed on propagate to %f (dir=%d) | ESDtrack#%d (MClb:%d)",xClus,dir,trESD->GetID(),trESD->GetTPCLabel()));
1540 #endif
1541         return -1;
1542       }
1543       //
1544       Double_t p[2]={clus->GetY(), clus->GetZ()};
1545       Double_t cov[3]={clus->GetSigmaY2(), clus->GetSigmaYZ(), clus->GetSigmaZ2()};
1546       double chi2cl = tmpTr.GetPredictedChi2(p,cov);
1547       chi2 += chi2cl;
1548       //
1549 #ifdef  _ITSU_TUNING_MODE_
1550       TObjArray* dest = trc->GetLabel()>=0 ? fCHistoArrCorr : fCHistoArrFake;
1551       if (dest && fTrackPhaseID>kClus2Tracks) {
1552         //
1553         double htrPt = tmpTr.Pt();
1554         double dy = p[0]-tmpTr.GetY();
1555         double dz = p[1]-tmpTr.GetZ();
1556         ((TH2*)dest->At( GetHistoID(ilrA,kHResY,0,fTrackPhaseID) ))->Fill(htrPt,dy);
1557         ((TH2*)dest->At( GetHistoID(ilrA,kHResZ,0,fTrackPhaseID) ))->Fill(htrPt,dz);
1558         double errY = tmpTr.GetSigmaY2();
1559         double errZ = tmpTr.GetSigmaZ2();
1560         if (errY>0) ((TH2*)dest->At( GetHistoID(ilrA,kHResYP,0,fTrackPhaseID) ))->Fill(htrPt,dy/Sqrt(errY));
1561         if (errZ>0) ((TH2*)dest->At( GetHistoID(ilrA,kHResZP,0,fTrackPhaseID) ))->Fill(htrPt,dz/Sqrt(errZ));
1562         ((TH2*)dest->At( GetHistoID(ilrA,kHChi2Cl,0,fTrackPhaseID) ))->Fill(htrPt,chi2cl);
1563       }
1564 #endif  
1565       //      
1566       if ( !tmpTr.Update(p,cov) ) {
1567 #ifdef _ITSU_DEBUG_
1568         AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1569         AliDebug(2,Form("Failed on Update | ESDtrack#%d (MClb:%d)",trESD->GetID(),trESD->GetTPCLabel()));
1570 #endif
1571         return -1;
1572       }
1573 #ifdef _ITSU_DEBUG_
1574       if (AliDebugLevelClass()>1) {
1575         printf("AfterRefit: "); tmpTr.Print();
1576       }
1577 #endif
1578       if (++nclFit==nCl && stopCond<0) {
1579         *trc = (AliKalmanTrack&)tmpTr;
1580         return chi2; // it was requested to not propagate after last update
1581       }
1582     }
1583     //
1584   }
1585   // All clusters were succesfully fitted. Even if the track does not reach rDest, this is enough to validate it.
1586   // Still, try to go as close as possible to rDest.
1587   //
1588   if (lrStart!=lrStop) {
1589     if (!TransportToLayer(&tmpTr,lrStart,lrStop)) {
1590 #ifdef _ITSU_DEBUG_
1591       AliDebug(1,Form("Failed on TransportToLayer %d->%d | ESDtrack#%d (MClb:%d), X=%f",lrStart,lrStop,fCurrESDtrack->GetID(),fCurrESDtrMClb,tmpTr.GetX()));
1592 #endif
1593       return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
1594     }    
1595     if (!GoToExitFromLayer(&tmpTr,fITS->GetLayer(lrStop),dir)) {
1596 #ifdef _ITSU_DEBUG_
1597       AliDebug(1,Form("Failed on GoToExitFromLayer %d | ESDtrack#%d (MClb:%d), X=%f",lrStop,fCurrESDtrack->GetID(),fCurrESDtrMClb,tmpTr.GetX()));
1598 #endif
1599       return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
1600     }
1601   }
1602   // go to the destination radius. Note that here we don't select direction to avoid precision problems
1603   if (!tmpTr.GetXatLabR(rDest,rDest,GetBz(),0) || !PropagateSeed(&tmpTr,rDest,fCurrMass, 100, kFALSE)) {
1604     return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
1605   }
1606   *trc = (AliKalmanTrack&)tmpTr;
1607 #ifdef _ITSU_DEBUG_
1608   if (AliDebugLevelClass()>2) {
1609     printf("After refit (now at lr %d): ",lrStop); trc->Print();
1610   }
1611 #endif
1612   return chi2;
1613 }
1614
1615 //__________________________________________________________________
1616 void AliITSUTrackerGlo::CookMCLabel(AliITSUTrackHyp* hyp) 
1617 {
1618   // build MC label
1619   //
1620   const int kMaxLbPerCl = 3;
1621   int lbID[kMaxLayers*6],lbStat[kMaxLayers*6];
1622   Int_t lr,nLab=0,nCl=0;
1623   AliITSUSeed *seed = hyp->GetWinner();
1624   while(seed) {
1625     int clID = seed->GetLrCluster(lr);
1626     if (clID>=0) {
1627       AliCluster *cl = fITS->GetLayerActive(lr)->GetCluster(clID);
1628       nCl++;
1629       for (int imc=0;imc<kMaxLbPerCl;imc++) { // labels within single cluster
1630         int trLb = cl->GetLabel(imc);
1631         if (trLb<0) break;
1632         // search this mc track in already accounted ones
1633         int iLab;
1634         for (iLab=0;iLab<nLab;iLab++) if (lbID[iLab]==trLb) break;
1635         if (iLab<nLab) lbStat[iLab]++;
1636         else {
1637           lbID[nLab] = trLb;
1638           lbStat[nLab++] = 1;
1639         }
1640       } // loop over given cluster's labels
1641     }
1642     seed = (AliITSUSeed*)seed->GetParent();
1643   } // loop over clusters
1644   // 
1645   AliESDtrack* esdTr = hyp->GetESDTrack();
1646   int tpcLab = esdTr ? Abs(esdTr->GetTPCLabel()) : -kDummyLabel;
1647   if (nCl && nLab) {
1648     int maxLab=0,nTPCok=0;
1649     for (int ilb=nLab;ilb--;) {
1650       int st = lbStat[ilb];
1651       if (lbStat[maxLab]<st) maxLab=ilb;
1652       if (tpcLab==lbID[ilb]) nTPCok += st;
1653     }
1654     hyp->SetFakeRatio(1.-float(nTPCok)/nCl);
1655     hyp->SetLabel( nTPCok==nCl ? tpcLab : -tpcLab);
1656     hyp->SetITSLabel( lbStat[maxLab]==nCl ? lbID[maxLab] : -lbID[maxLab]); // winner label
1657     return;
1658   }
1659   //
1660   hyp->SetFakeRatio(-1.);
1661   hyp->SetLabel( -tpcLab );
1662   hyp->SetITSLabel( kDummyLabel );
1663 }
1664
1665 //__________________________________________________________________
1666 Bool_t AliITSUTrackerGlo::AddSeedBranch(AliITSUSeed* seed)
1667 {
1668   // add new prolongation branch for the seed to temporary array. The seeds are sorted according to Compare f-n
1669   if (fNCandidatesAdded+fNBranchesAdded>=fLayerMaxCandidates ) {
1670     AliInfo(Form("Number of candidates at layer %d for current seed reached %d, increasing buffer",fCurrActLrID,fLayerMaxCandidates));
1671     fLayerMaxCandidates = 2*(fLayerMaxCandidates+1);
1672     AliITSUSeed** tmpArr = fLayerCandidates;
1673     fLayerCandidates = new AliITSUSeed*[fLayerMaxCandidates];
1674     memcpy(fLayerCandidates,tmpArr,(fNCandidatesAdded+fNBranchesAdded)*sizeof(AliITSUSeed*));
1675     delete[] tmpArr; // delete only array, not objects
1676   }
1677   AliITSUSeed** branches = &fLayerCandidates[fNCandidatesAdded]; // note: fNCandidatesAdded is incremented after adding all branches of current seed
1678   int slot=fNBranchesAdded++;
1679   for (int slotF=slot;slotF--;) { // slotF is always slot-1
1680     AliITSUSeed* si = branches[slotF];
1681     if (si->Compare(seed)<0) break; // found the last seed with better quality
1682     // otherwise, shift the worse seed to the next slot
1683     branches[slot] = si;
1684     slot = slotF; // slot should be slotF+1
1685   }
1686   // if needed, move worse seeds
1687   branches[slot] = seed;
1688   return kTRUE;
1689   //
1690 }
1691
1692 //__________________________________________________________________
1693 void AliITSUTrackerGlo::ValidateAllowedBranches(Int_t acceptMax)
1694 {
1695   // keep allowed number of branches for current seed and disable extras
1696   AliCodeTimerAuto("",0); 
1697   int nb = Min(fNBranchesAdded,acceptMax);
1698   //  if (nb<fNBranchesAdded) printf("ValidateAllowedBranches: %d of %d (%d) on lr %d\n",nb,fNBranchesAdded,fNCandidatesAdded,ilr);
1699   // disable unused branches
1700   AliITSUSeed** branches = &fLayerCandidates[fNCandidatesAdded];
1701   //
1702 #ifdef  _ITSU_TUNING_MODE_
1703   for (int ib=0;ib<fNBranchesAdded;ib++) branches[ib]->SetOrdBranch(ib);
1704 #endif
1705   //
1706   for (int ib=nb;ib<fNBranchesAdded;ib++) MarkSeedFree(branches[ib]);
1707   fNCandidatesAdded += nb; // update total candidates counter
1708   fNBranchesAdded = 0; // reset branches counter
1709   //
1710 }
1711
1712 //__________________________________________________________________
1713 void AliITSUTrackerGlo::ValidateAllowedCandidates(Int_t ilr, Int_t acceptMax)
1714 {
1715   // transfer allowed number of branches to hypothesis container
1716   //
1717   AliCodeTimerAuto("",0); 
1718   // sort candidates in increasing order of chi2
1719   static int lastSize = 0;
1720   static int *index = 0;
1721   static float *chi2 = 0;
1722   if (fLayerMaxCandidates>lastSize) {
1723     lastSize = fLayerMaxCandidates;
1724     delete[] index;
1725     delete[] chi2;
1726     index = new int[lastSize];
1727     chi2  = new float[lastSize];
1728   }
1729   for (int i=fNCandidatesAdded;i--;) chi2[i] = fLayerCandidates[i]->GetChi2GloNrm();
1730   Sort(fNCandidatesAdded,chi2,index,kFALSE);
1731   //
1732   int nacc=0,nb=0;
1733   if (ilr>fCurrTrackCond->GetActiveLrInner()) { // just take 1st acceptMax candidates
1734     nb = Min(fNCandidatesAdded,acceptMax);
1735     for (int ib=0;ib<nb;ib++) AddProlongationHypothesis(fLayerCandidates[index[ib]],ilr);
1736   }
1737   else { // for innermost layer test ITS SA backward chi2 and matching to TPC
1738     AliITSUSeed* wn0 = fCurrHyp->GetWinner(); // in principle, should be NULL
1739     for (nb=0;nb<fNCandidatesAdded;nb++) {
1740       AliITSUSeed* sd = fLayerCandidates[index[nb]];    
1741       fCurrHyp->SetWinner(sd);
1742       if (!CheckBackwardMatching(sd)) MarkSeedFree(sd); // discard bad backward tracks
1743       else {
1744         AddProlongationHypothesis(sd,ilr);
1745         if (++nacc==acceptMax) {nb++; break;} // the rest will be discarded
1746       }
1747     }
1748     fCurrHyp->SetWinner(wn0); // restore original winner (NULL?)
1749   }
1750   // disable unused candidates
1751   for (int ib=nb;ib<fNCandidatesAdded;ib++) MarkSeedFree(fLayerCandidates[index[ib]]);    
1752   fNCandidatesAdded = 0; // reset candidates counter
1753   //
1754 }
1755
1756 //__________________________________________________________________
1757 Bool_t AliITSUTrackerGlo::CheckBackwardMatching(AliITSUSeed* seed)
1758 {
1759   // check seed backward propagation chi2 and matching to TPC 
1760   double bz0 = GetBz();
1761   double rDest = fITS->GetRITSTPCRef(); // reference radius for comparison
1762   AliITSUTrackHyp trback;
1763   trback.AliExternalTrackParam::operator=(*seed);
1764   trback.ResetCovariance(10000);
1765   int nclFit = 0;
1766   double chi2sa = RefitTrack(&trback,rDest,nclFit,1);
1767   if (chi2sa<0) return kFALSE;
1768   int ndf = nclFit*2-5;
1769   if (ndf>0) chi2sa /= ndf;
1770   if (chi2sa>fCurrTrackCond->GetMaxITSSAChi2(nclFit)) return kFALSE;
1771   //
1772   // relate to TPC track at outer layer
1773   AliExternalTrackParam* tpcSeed = fCurrHyp->GetTPCSeed();
1774   if (!trback.Rotate(tpcSeed->GetAlpha()) || !trback.PropagateParamOnlyTo(tpcSeed->GetX(),bz0)) {
1775 #ifdef _ITSU_DEBUG_
1776     if (AliDebugLevelClass()>+1 && !seed->ContainsFake()) {
1777       AliInfo(Form("Failed to propagate seed to TPC track @ X:%.3f Alpha:%.3f",tpcSeed->GetX(),tpcSeed->GetAlpha()));
1778       seed->Print("etp");
1779       trback.Print();
1780     }
1781 #endif
1782     return kFALSE; 
1783   }
1784   double chi2Match = trback.GetPredictedChi2(tpcSeed)/5;
1785   if (chi2Match>fCurrTrackCond->GetMaxITSTPCMatchChi2()) return kFALSE;
1786   //
1787   seed->SetChi2ITSSA(chi2sa);
1788   seed->SetChi2ITSTPC(chi2Match);
1789   return kTRUE;
1790 }
1791
1792 //__________________________________________________________________
1793 void AliITSUTrackerGlo::SaveReducedHypothesesTree(AliITSUTrackHyp* dest)
1794 {
1795   // remove those hypothesis seeds which dont lead to candidates at final layer
1796   //
1797   // 1st, flag the seeds to save
1798   int lr0 = fCurrTrackCond->GetActiveLrInner();
1799   for (int isd=0;isd<fCurrHyp->GetNSeeds(lr0);isd++) {
1800     AliITSUSeed* seed = fCurrHyp->RemoveSeed(lr0,isd);
1801     if (!seed) continue;
1802     seed->FlagTree(AliITSUSeed::kSave);
1803     dest->AddSeed(seed,lr0);
1804   }
1805   for (int ilr=0;ilr<fNLrActive;ilr++) {
1806     int nsd = fCurrHyp->GetNSeeds(ilr);
1807     for (int isd=0;isd<nsd;isd++) {
1808       AliITSUSeed* seed = fCurrHyp->RemoveSeed(ilr,isd);
1809       if (!seed) continue; // already discarded or saved
1810       if (seed->IsSaved()) dest->AddSeed(seed,ilr);
1811       else MarkSeedFree(seed);
1812     }
1813   }
1814   //
1815   //  AliInfo(Form("SeedsPool: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
1816 }
1817
1818 //__________________________________________________________________
1819 void AliITSUTrackerGlo::CleanHypothesis(AliITSUTrackHyp* hyp)
1820 {
1821   // clean hypothesis
1822   hyp->SetWinner(0);
1823   for (int ilr=0;ilr<fNLrActive;ilr++) {
1824     int nsd = hyp->GetNSeeds(ilr);
1825     for (int isd=0;isd<nsd;isd++) {
1826       AliITSUSeed* seed = hyp->RemoveSeed(ilr,isd);
1827       if (!seed) continue; // already discarded or saved
1828       MarkSeedFree(seed);
1829     }
1830   }
1831 }
1832
1833 //__________________________________________________________________
1834 void AliITSUTrackerGlo::FlagSplitClusters()
1835 {
1836   // set special bit on split clusters using MC info
1837   for (int ilr=fNLrActive;ilr--;) {
1838     int nsplit=0;
1839     AliITSURecoLayer* lr = fITS->GetLayerActive(ilr);
1840     for (int isn=lr->GetNSensors();isn--;) {
1841       AliITSURecoSens* sens = lr->GetSensor(isn);
1842       int nCl = sens->GetNClusters();
1843       if (!nCl) continue;
1844       int cl0 = sens->GetFirstClusterId();
1845       for (int icl=nCl;icl--;) {
1846         AliITSUClusterPix *cl = (AliITSUClusterPix*)lr->GetCluster(cl0+icl);
1847         for (int icl1=icl;icl1--;) {
1848           AliITSUClusterPix *cl1 = (AliITSUClusterPix*)lr->GetCluster(cl0+icl1);
1849           if (cl->HasCommonTrack(cl1)) {
1850             if (!cl->IsSplit())  nsplit++;
1851             if (!cl1->IsSplit()) nsplit++;
1852             cl->SetSplit();
1853             cl1->SetSplit();
1854           }
1855         }
1856       }
1857     }
1858     AliInfo(Form("%4d out of %4d clusters are split",nsplit,lr->GetNClusters()));
1859   }
1860   //
1861 }
1862
1863 //__________________________________________________________________
1864 Bool_t AliITSUTrackerGlo::ContainsSplitCluster(const AliITSUSeed* seed, Int_t maxSize)
1865 {
1866   // check if the seed contains split cluster with size < maxSize
1867   int lrID,clID;
1868   if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
1869     AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1870     if (cl->IsSplit() && cl->GetNPix()<maxSize ) return kTRUE;
1871   }
1872   return seed->GetParent() ? ContainsSplitCluster((AliITSUSeed*)seed->GetParent(),maxSize) : kFALSE;
1873   //
1874 }
1875
1876 //__________________________________________________________________
1877 void AliITSUTrackerGlo::PrintSeedClusters(const AliITSUSeed* seed, Option_t* option)
1878 {
1879   // print seeds clusters
1880   int lrID,clID;
1881   if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
1882     AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1883     cl->Print(option);
1884   }
1885   if (seed->GetParent()) PrintSeedClusters((AliITSUSeed*)seed->GetParent(), option);
1886   //
1887 }
1888
1889 //__________________________________________________________________
1890 void AliITSUTrackerGlo::FlagSeedClusters(const AliITSUSeed* seed, Bool_t flg, UShort_t ref)
1891 {
1892   // mark used clusters
1893   int lrID,clID;
1894   while (seed) {
1895     if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
1896       AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1897       if (ref) { // do we need to set or delete cluster->track ref?
1898         if (flg) {
1899           if (!cl->GetRecoInfo()) cl->SetRecoInfo(ref);  // set ref only if cluster already does not refer to other track, inc.counter
1900         }
1901         else {
1902           if (cl->GetRecoInfo()==ref) cl->SetRecoInfo(0); // unset reference only if it refers to ref, decrease counter     
1903         }
1904       }
1905     }
1906     seed = (AliITSUSeed*)seed->GetParent();
1907   }
1908   //
1909 }
1910
1911 //__________________________________________________________________
1912 void AliITSUTrackerGlo::CheckClusterUsage()
1913 {
1914   // check cluster usage info
1915   printf("ClusterUsage at pass %d\n",fCurrPassID);
1916   for (int ilr=0;ilr<fNLrActive;ilr++) {
1917     AliITSURecoLayer* lr = fITS->GetLayerActive(ilr);
1918     int ncl = lr->GetNClusters();
1919     int nclUsed = 0;
1920     int nclUs0CntShr1 = 0;
1921     int nclUs1Cnt0 = 0;
1922     int nclUs1Shr1 = 0;
1923     int nclUseL = 0;
1924     for (int icl=0;icl<ncl;icl++) {
1925       AliITSUClusterPix *cl = (AliITSUClusterPix*)lr->GetCluster(icl);
1926       int usc = cl->GetClUsage();
1927       if (usc<1) {
1928         if (cl->IsClusterUsed() || cl->IsClusterShared()) {
1929           nclUs0CntShr1++;
1930           printf("Lr%d Cl%4d: UseCnt=0 but UseFlg=%d UseShr=%d\n",ilr,icl,cl->IsClusterUsed(),cl->IsClusterShared());
1931         }
1932         continue;
1933       }
1934       nclUsed++;
1935       if (!cl->IsClusterUsed()) {
1936         nclUs1Cnt0++;
1937         printf("Lr%d Cl%4d: UseCnt=%d but UseFlg=%d UseShr=%d\n",ilr,icl,usc, cl->IsClusterUsed(),cl->IsClusterShared());
1938       }
1939       if (cl->IsClusterShared()) {
1940         nclUs1Shr1++;
1941         printf("Lr%d Cl%4d: UseCnt=%d but UseFlg=%d UseShr=%d\n",ilr,icl,usc, cl->IsClusterUsed(),cl->IsClusterShared());
1942       }
1943       if (usc>1) {
1944         nclUseL++;
1945         printf("Lr%d Cl%4d: UseCnt=%d!! but UseFlg=%d UseShr=%d\n",ilr,icl,usc, cl->IsClusterUsed(),cl->IsClusterShared());
1946       }
1947     }
1948     printf("ClUsage Lr%d: Ncl=%5d Nusd=%5d (%5.2f) Use0CS1:%4d Use1C0:%4d Use1S1:%4d UseL:%d\n",
1949            ilr,ncl,nclUsed,ncl? float(nclUsed)/ncl : 0, nclUs0CntShr1,nclUs1Cnt0,nclUs1Shr1,nclUseL);
1950   }
1951 }
1952
1953
1954 #ifdef  _ITSU_TUNING_MODE_
1955 //__________________________________________________________________
1956 TObjArray* AliITSUTrackerGlo::BookControlHistos(const char* pref)
1957 {
1958   // book special control histos
1959   TString prefS = pref;
1960   prefS.ToLower();
1961   TObjArray* dest = new TObjArray;
1962   dest->SetOwner(kTRUE);
1963   //
1964   const int kNResDef=7;
1965   const double kResDef[kNResDef]={0.05,0.05,0.3, 0.05,1,0.5,1.5};
1966   const double ptMax=10;
1967   const double plMax=10;
1968   const double chiMax=100;
1969   const int nptbins=50;
1970   const int nresbins=400;
1971   const int nplbins=50;
1972   const int nchbins=200;
1973   const int maxBr  = 15;
1974   const int maxCand = 200;
1975   TString ttl;
1976   for (int phase=0;phase<kNTrackingPhases;phase++) {
1977     for (int pass=0;pass<AliITSUReconstructor::GetRecoParam()->GetNTrackingConditions();pass++) {
1978       //
1979       for (int ilr=0;ilr<fNLrActive;ilr++) {
1980         //
1981         // ----------------- These are histos to be filled in Cluster2Tracks of each pass. 
1982         // PropagateBack and RefitInward will be stored among the histos of 1st pass
1983         if (pass>0 && phase!=kClus2Tracks) continue;
1984         //
1985         double mxdf = ilr>=kNResDef ? kResDef[kNResDef-1] : kResDef[ilr];
1986         ttl = Form("Pass%d_S%d_residY%d_%s",pass,phase,ilr,pref);
1987         TH2F* hdy = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nresbins,-mxdf,mxdf);
1988         dest->AddAtAndExpand(hdy,GetHistoID(ilr,kHResY,pass,phase));
1989         hdy->SetDirectory(0);
1990         //
1991         ttl = Form("Pass%d_S%d_residYPull%d_%s",pass,phase,ilr,pref);   
1992         TH2F* hdyp = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nplbins,-plMax,plMax);
1993         dest->AddAtAndExpand(hdyp,GetHistoID(ilr,kHResYP,pass,phase));
1994         hdyp->SetDirectory(0);
1995         //
1996         ttl = Form("Pass%d_S%d_residZ%d_%s",pass,phase,ilr,pref);       
1997         TH2F* hdz = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nresbins,-mxdf,mxdf);
1998         dest->AddAtAndExpand(hdz,GetHistoID(ilr,kHResZ,pass,phase));
1999         hdz->SetDirectory(0);
2000         //
2001         ttl = Form("Pass%d_S%d_residZPull%d_%s",pass,phase,ilr,pref);           
2002         TH2F* hdzp = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nplbins,-plMax,plMax);
2003         hdzp->SetDirectory(0);
2004         dest->AddAtAndExpand(hdzp,GetHistoID(ilr,kHResZP,pass,phase));
2005         //
2006         ttl = Form("Pass%d_S%d_chi2Cl%d_%s",pass,phase,ilr,pref);               
2007         TH2F* hchi = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
2008         hchi->SetDirectory(0);
2009         dest->AddAtAndExpand(hchi,GetHistoID(ilr,kHChi2Cl,pass,phase));
2010         //
2011         // ------------------- These histos are filled for Clusters2Tracks only
2012         if (phase!=kClus2Tracks) continue;
2013         //
2014         ttl = Form("Pass%d_S%d_chi2Nrm%d_%s",pass,phase,ilr,pref);              
2015         TH2F* hchiN = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
2016         hchiN->SetDirectory(0);
2017         dest->AddAtAndExpand(hchiN,GetHistoID(ilr,kHChi2Nrm,pass,phase));
2018         //
2019         ttl = Form("Pass%d_S%d_bestInBranch%d_%s",pass,phase,ilr,pref);         
2020         TH2* hnbr = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, maxBr,-0.5,maxBr-0.5);
2021         hnbr->SetDirectory(0);
2022         dest->AddAtAndExpand(hnbr,GetHistoID(ilr,kHBestInBranch,pass,phase));
2023         //
2024         ttl = Form("Pass%d_S%d_bestInCands%d_%s",pass,phase,ilr,pref);          
2025         TH2* hncn = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, maxCand,-0.5,maxCand-0.5);
2026         hncn->SetDirectory(0);
2027         dest->AddAtAndExpand(hncn,GetHistoID(ilr,kHBestInCand,pass,phase));
2028         //
2029       } // loop over layers
2030       //
2031       // these are special histos, filled not per layer but in the end of track fit in Clusters2Tracks in EVERY pass
2032       //  
2033       if (phase==kClus2Tracks) {
2034         TH2* hchiMatch = 0; 
2035         ttl = Form("Pass%d_Chi2Match_%s",pass,pref);
2036         hchiMatch = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
2037         hchiMatch->SetDirectory(0);
2038         dest->AddAtAndExpand(hchiMatch,GetHistoID(-1,kHChiMatch,pass,phase));
2039         // 
2040         TH2* hchiSA = 0; 
2041         ttl = Form("Pass%d_Chi2ITSSA_%s",pass,pref);
2042         hchiSA = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins/2,0.,chiMax/2);
2043         hchiSA->SetDirectory(0);
2044         dest->AddAtAndExpand(hchiSA,GetHistoID(-1,kHChiITSSA,pass,phase));
2045         //      
2046       }
2047     } // loop over tracking passes
2048   }// loop over tracking phases
2049   //
2050   return dest;
2051 }
2052
2053 //__________________________________________________________________
2054 Int_t AliITSUTrackerGlo::GetHistoID(Int_t lr, Int_t hid, Int_t pass, Int_t phase)
2055 {
2056   // get id for the requested histo
2057   if (lr<0) lr=-1;
2058   lr++;
2059   return pass*kHistosPass + phase*kHistosPhase + lr*kMaxHID + hid;
2060   //
2061 }
2062
2063 #endif