1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
22 //-------------------------------------------------------------------------
23 // Implementation of the ITS Upgrade tracker mother class.
24 //-------------------------------------------------------------------------
26 #include <Riostream.h>
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;
45 //----------------- tmp stuff -----------------
47 ClassImp(AliITSUTrackerGlo)
49 const Double_t AliITSUTrackerGlo::fgkToler = 1e-6;// tolerance for layer on-surface check
52 //_________________________________________________________________________
53 AliITSUTrackerGlo::AliITSUTrackerGlo(AliITSUReconstructor* rec)
58 ,fCurrESDtrMClb(kDummyLabel)
60 ,fCountProlongationTrials(0)
66 ,fLayerMaxCandidates(1000)
72 ,fSeedsPool("AliITSUSeed",0)
85 #ifdef _ITSU_TUNING_MODE_
90 // Default constructor
94 //_________________________________________________________________________
95 AliITSUTrackerGlo::~AliITSUTrackerGlo()
99 delete[] fLayerCandidates;
100 if (fWorkHyp) fWorkHyp->SetTPCSeed(0); // this hypothesis does not own the seed
104 #ifdef _ITSU_TUNING_MODE_
105 if (fCHistoArrCorr || fCHistoArrFake) {
106 TFile* ctrOut = TFile::Open("itsuControlHistos.root","recreate");
108 AliInfo("Storing control histos");
109 // ctrOut->WriteObject(fCHistoArr,"controlH","kSingleKey");
110 if (fCHistoArrCorr) {
111 fCHistoArrCorr->Write();
112 delete fCHistoArrCorr;
114 if (fCHistoArrFake) {
115 fCHistoArrFake->Write();
116 delete fCHistoArrFake;
127 //_________________________________________________________________________
128 void AliITSUTrackerGlo::Init(AliITSUReconstructor* rec)
130 // init with external reconstructor
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())));
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());
145 fMatLUT->FillData(kNTest,-zmn,zmn);
147 if (fLayerMaxCandidates<1) fLayerMaxCandidates = 1000;
148 fLayerCandidates = new AliITSUSeed*[fLayerMaxCandidates];
149 fSeedsPool.ExpandCreateFast(1000); // RS TOCHECK
150 fFreeSeedsID.Set(1000);
155 //_________________________________________________________________________
156 void AliITSUTrackerGlo::CreateDefaultTrackCond()
158 // creates default tracking conditions to be used when recoparam does not provide them
160 AliITSUTrackCond* cond = new AliITSUTrackCond();
162 cond->SetNLayers(fNLrActive);
163 cond->AddNewCondition(fNLrActive);
164 cond->AddGroupPattern( 0xffff, fNLrActive ); // require all layers hit
167 fDefTrackConds.AddLast(cond);
169 AliInfo("Created conditions: ");
170 for (int i=0;i<fDefTrackConds.GetEntriesFast();i++) fDefTrackConds[i]->Print();
175 //_________________________________________________________________________
176 Int_t AliITSUTrackerGlo::Clusters2Tracks(AliESDEvent *esdEv)
179 AliCodeTimerAuto("",0);
180 SetTrackingPhase(kClus2Tracks);
182 #ifdef _ITSU_TUNING_MODE_
183 if (!fCHistoArrCorr) fCHistoArrCorr = BookControlHistos("Corr");
184 if (!fCHistoArrFake) fCHistoArrFake = BookControlHistos("Fake");
187 static TArrayF esdTrPt(fESDIndex.GetSize());
189 TObjArray *trackConds = 0;
191 fCountProlongationTrials = 0;
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);
202 if (!fDefTrackConds.GetEntriesFast()) {
203 AliInfo("No tracking conditions found in recoparams, creating default one requesting all layers hit");
204 CreateDefaultTrackCond();
206 trackConds = &fDefTrackConds;
207 nTrackCond = trackConds->GetEntriesFast();
209 else trackConds = AliITSUReconstructor::GetRecoParam()->GetTrackingConditions();
211 static Bool_t first = kTRUE;
213 AliITSUReconstructor::GetRecoParam()->Print();
217 if (fHypStore.GetSize()<fNTracksESD) fHypStore.Expand(fNTracksESD+100);
219 fITS->ProcessClusters();
221 #ifdef _ITSU_TUNING_MODE_
222 FlagSplitClusters(); // tmp RS
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);
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();
237 Sort(fNTracksESD,trPt,esdTrackIndex,kTRUE);
239 for (int icnd=0;icnd<nTrackCond;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();
250 if (!NeedToProlong(fCurrESDtrack,trID)) continue; // are we interested in this track?
252 // if specific tracks should be checked, create a global array int wtc[] = {evITS*10000+trID, ...};
254 int nwtc = sizeof(wtc)/sizeof(int);
256 for (int k=0;k<nwtc;k++) {
257 if (wtc[k]==evID*10000+trID) {
259 printf("At esdTr: %d MC: %d\n",wtc[k],fCurrESDtrMClb);
263 AliLog::SetClassDebugLevel("AliITSUTrackerGlo",dbg ? 10:0);
266 AliDebug(1,Form("Processing track %d(esd%d) | M=%.3f Pt=%.3f | MCLabel: %d",itr,trID,fCurrESDtrack->GetMassForTracking(kTRUE),fCurrESDtrack->Pt(),fCurrESDtrMClb));//RS
268 FindTrack(fCurrESDtrack, trID);
272 if (AliDebugLevelClass()>+2) {
273 AliInfo(Form("SeedsPool: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
277 FinalizeHypotheses();
278 #ifdef _ITSU_TUNING_MODE_
279 CheckClusterUsage(); //!!RS
282 //AliLog::SetClassDebugLevel("AliITSUTrackerGlo",0); // in case wtc array was defined, uncomment this
285 AliInfo(Form("%d ITSin for %d tried TPC seeds out of %d ESD tracks\n",fCountITSin,fCountProlongationTrials,fNTracksESD));
290 //_________________________________________________________________________
291 Int_t AliITSUTrackerGlo::PropagateBack(AliESDEvent *esdEv)
294 // Do outward fits in ITS
295 AliCodeTimerAuto("",0);
297 SetTrackingPhase(kPropBack);
298 fNTracksESD = esdEv->GetNumberOfTracks();
299 fUseMatLUT = AliITSUReconstructor::GetRecoParam()->GetUseMatLUT(fTrackPhaseID);
301 AliDebug(1,Form("Will propagate back %d tracks",fNTracksESD));
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];
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
316 fCurrESDtrack->GetXYZ(xyzTrk);
317 Double_t dst = 0.; // set initial track lenght, tof
319 double dxs = xyzTrk[0] - xyzVtx[0];
320 double dys = xyzTrk[1] - xyzVtx[1];
321 // double dzs = xyzTrk[2] - xyzVtx[2];
322 // RS: for large segment steps d use approximation of cicrular arc s by
323 // s = 2R*asin(d/2R) = d/p asin(p) \approx d/p (p + 1/6 p^3) = d (1+1/6 p^2)
324 // where R is the track radius, p = d/2R = 2C*d (C is the abs curvature)
325 // Hence s^2/d^2 = (1+1/6 p^2)^2
326 dst = dxs*dxs + dys*dys;
327 if (dst > kWatchStep*kWatchStep) { // correct circular part for arc/segment factor
328 double crv = Abs(fCurrESDtrack->GetC(bz0));
329 double fcarc = 1.+crv*crv*dst/6.;
332 // RS: temporary hack since we don't have SPD vertex:
334 dst *= 1+fCurrESDtrack->GetTgl()*fCurrESDtrack->GetTgl();
338 fCurrESDtrack->SetStatus(AliESDtrack::kTIME);
340 if (!fCurrESDtrack->IsOn(AliESDtrack::kITSin)) { // case of tracks w/o ITS prolongation: just set the time integration
341 dummyTr.AliExternalTrackParam::operator=(*fCurrESDtrack);
342 dummyTr.StartTimeIntegral();
343 dummyTr.AddTimeStep(dst);
344 dummyTr.GetIntegratedTimes(times,AliPID::kSPECIESC);
345 fCurrESDtrack->SetIntegratedTimes(times);
346 fCurrESDtrack->SetIntegratedLength(dummyTr.GetIntegratedLength());
350 fCurrHyp = GetTrackHyp(itr);
351 fCurrMass = fCurrHyp->GetMass();
352 fCurrHyp->StartTimeIntegral();
353 fCurrHyp->AddTimeStep(dst);
354 fCurrHyp->ResetCovariance(10000);
356 double chi2 = RefitTrack(fCurrHyp,fITS->GetRMax(),nclFit);
357 if (chi2>0) { // propagate to exit from the ITS/TPC screen
358 int ndf = nclFit*2-5;
359 if (ndf>0) chi2 /= ndf;
360 fCurrHyp->SetChi2(chi2);
361 UpdateESDTrack(fCurrHyp,AliESDtrack::kITSout);
365 AliDebug(2,Form("Refit Failed for track %d | ESDtrack#%d (MClb:%d)",itr,fCurrESDtrack->GetID(),fCurrESDtrMClb));
367 //fCurrHyp->AliExternalTrackParam::Print();
368 //fCurrHyp->GetWinner()->Print();
373 AliInfo(Form("%d ITSout in %d ITSin tracks for %d tried TPC seeds out of %d ESD tracks\n",
374 fCountITSout,fCountITSin,fCountProlongationTrials,fNTracksESD));
379 //_________________________________________________________________________
380 Int_t AliITSUTrackerGlo::RefitInward(AliESDEvent *esdEv)
383 // refit the tracks inward, using current cov.matrix
384 AliCodeTimerAuto("",0);
386 SetTrackingPhase(kRefitInw);
387 fUseMatLUT = AliITSUReconstructor::GetRecoParam()->GetUseMatLUT(fTrackPhaseID);
388 fNTracksESD = esdEv->GetNumberOfTracks();
389 // AliLog::SetClassDebugLevel("AliITSUTrackerGlo",10);
391 AliDebug(1,Form("Will refit inward %d tracks",fNTracksESD));
392 // Bool_t uselogMS = AliExternalTrackParam::GetUseLogTermMS();
393 // AliExternalTrackParam::SetUseLogTermMS(kTRUE);
395 for (int itr=0;itr<fNTracksESD;itr++) {
396 fCurrESDtrack = esdEv->GetTrack(itr);
397 fCurrESDtrMClb = fCurrESDtrack->GetLabel();
398 // Start time integral and add distance from current position to vertex
399 UInt_t trStat = fCurrESDtrack->GetStatus();
400 if ( !(trStat & AliESDtrack::kITSout) ) continue;
401 if ( trStat & AliESDtrack::kITSrefit ) continue; // already done
402 if ( (trStat & AliESDtrack::kTPCout) && !(trStat & AliESDtrack::kTPCrefit) ) continue;
404 fCurrHyp = GetTrackHyp(itr);
405 fCurrMass = fCurrHyp->GetMass();
406 *fCurrHyp = *fCurrESDtrack; // fetch current ESDtrack kinematics
409 // fCurrHyp->ResetCovariance(1000); // in case we want to do SA refit
410 double chi2 = RefitTrack(fCurrHyp,fITS->GetRMin(),nclFit);
411 if (chi2>0) { // propagate up to inside radius of the beam pipe
412 int ndf = nclFit*2-5;
413 if (ndf>0) chi2 /= ndf;
414 fCurrHyp->SetChi2(chi2);
415 UpdateESDTrack(fCurrHyp,AliESDtrack::kITSrefit);
419 AliDebug(2,Form("Refit Failed for track %d |ESDtrack#%d (MClb:%d)",itr,fCurrESDtrack->GetID(),fCurrESDtrMClb));
423 // AliExternalTrackParam::SetUseLogTermMS(uselogMS);
425 AliInfo(Form("%d ITSrefit in %d ITSout in %d ITSin tracks for %d tried TPC seeds out of %d ESD tracks\n",
426 fCountITSrefit,fCountITSout,fCountITSin,fCountProlongationTrials,fNTracksESD));
428 // AliLog::SetClassDebugLevel("AliITSUTrackerGlo",0);
432 //_________________________________________________________________________
433 Int_t AliITSUTrackerGlo::LoadClusters(TTree * treeRP)
435 // read from tree (if pointer provided) or directly from the ITS reco interface
437 return fReconstructor->LoadClusters(treeRP);
440 //_________________________________________________________________________
441 void AliITSUTrackerGlo::UnloadClusters()
447 Info("UnloadClusters","To be implemented");
449 //_________________________________________________________________________
450 AliCluster * AliITSUTrackerGlo::GetCluster(Int_t /*index*/) const
456 Info("GetCluster","To be implemented");
460 //_________________________________________________________________________
461 Bool_t AliITSUTrackerGlo::NeedToProlong(AliESDtrack* esdTr, Int_t esdID)
463 // do we need to match this track to ITS?
465 static double bz = GetBz();
467 // is it already prolonged
468 if (esdTr->IsOn(AliESDtrack::kITSin)) return kFALSE;
469 AliITSUTrackHyp* hyp = GetTrackHyp(esdID); // in case there is unfinished hypothesis from prev.pass, clean it
471 CleanHypothesis(hyp);
472 if (hyp->GetSkip()) return kFALSE; // need to skip this
476 if (!esdTr->IsOn(AliESDtrack::kTPCin) ||
477 esdTr->IsOn(AliESDtrack::kTPCout) ||
478 esdTr->IsOn(AliESDtrack::kITSin) ||
479 esdTr->GetKinkIndex(0)>0) return kFALSE;
481 if (esdTr->Pt()<AliITSUReconstructor::GetRecoParam()->GetMinPtForProlongation()) return kFALSE;
484 esdTr->GetDZ(GetX(),GetY(),GetZ(),bz,dtz);
485 // if track is not V0 candidata but has large offset wrt IP, reject it. RS TOCHECK
486 if ( !(esdTr->GetV0Index(0)>0 && dtz[0]>AliITSUReconstructor::GetRecoParam()->GetMaxDforV0dghtrForProlongation())
487 && (Abs(dtz[0])>AliITSUReconstructor::GetRecoParam()->GetMaxDForProlongation() ||
488 Abs(dtz[1])>AliITSUReconstructor::GetRecoParam()->GetMaxDZForProlongation())) return kFALSE;
490 // if (esdTr->Pt()<3) return kFALSE;//RS
494 //_________________________________________________________________________
495 void AliITSUTrackerGlo::FindTrack(AliESDtrack* esdTr, Int_t esdID)
497 // find prolongaion candidates finding for single seed
499 AliITSUSeed seedUC; // copy of the seed from the upper layer
500 AliITSUSeed seedT; // transient seed between the seedUC and new prolongation hypothesis
502 // AliLog::SetClassDebugLevel("AliITSUTrackerGlo",10);
503 AliITSUTrackHyp* hypTr = InitHypothesis(esdTr,esdID); // initialize prolongations hypotheses tree
504 if (!hypTr || hypTr->GetSkip()) return;
508 fCurrHyp->InitFrom(hypTr);
510 AliITSURecoSens *hitSens[AliITSURecoLayer::kMaxSensMatching];
512 int ilaUp = fNLrActive; // previous active layer really checked (some may be excluded!)
514 for (int ila=fNLrActive;ila--;ilaUp=fCurrActLrID) {
515 if (fCurrTrackCond->IsLayerExcluded(ila)) continue;
517 fCurrLayer = fITS->GetLayerActive(ila);
518 Bool_t noClSharing = fCurrTrackCond->GetClSharing(ila)==0;
520 // for the outermost layer the seed is created from the ESD track
521 int nSeedsUp = (ilaUp==fNLrActive) ? 1 : fCurrHyp->GetNSeeds(ilaUp);
522 int maxNBranches = fCurrTrackCond->GetMaxBranches(ila);
523 int maxNCandidates = fCurrTrackCond->GetMaxCandidates(ila);
525 for (int isd=0;isd<nSeedsUp;isd++) {
527 if (ilaUp==fNLrActive) {
529 seedUC.InitFromSeed(fCurrHyp->GetTPCSeed()); // init with TPC seed from ref.R
532 seedU = fCurrHyp->GetSeed(ilaUp,isd); // seed on prev.active layer to prolong
533 if (seedU->IsKilled()) continue;
534 seedUC = *seedU; // its copy will be prolonged
535 seedUC.SetParent(seedU);
537 seedUC.ResetFMatrix(); // reset the matrix for propagation to next layer
538 // go till next active layer
540 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()));
542 if (!TransportToLayer(&seedUC, fITS->GetLrIDActive(ilaUp), fITS->GetLrIDActive(ila)) ) { // external seed already prolonged
545 AliDebug(2,Form("Transport failed | esdID=%d (MClb:%d)",esdID,fCurrESDtrMClb));
547 // Check if the seed satisfies to track definition
548 if (NeedToKill(&seedUC,kTransportFailed) && seedU) KillSeed(seedU,kTRUE);
549 continue; // RS TODO: decide what to do with tracks stopped on higher layers w/o killing
551 if (!GetRoadWidth(&seedUC, ila)) { // failed to find road width on the layer
552 if (NeedToKill(&seedUC,kRWCheckFailed) && seedU) KillSeed(seedU,kTRUE);
558 if (!seedUC.ContainsFake() && AliDebugLevelClass()>2) {
559 mcquest = fCurrESDtrMClb;
561 printf("FSParams: "); for (int ip=0;ip<kNTrImpData;ip++) printf("%+e ",fTrImpData[ip]); printf("\n");
564 int nsens = fCurrLayer->FindSensors(&fTrImpData[kTrPhi0], hitSens, mcquest); // find detectors which may be hit by the track
566 int nsens = fCurrLayer->FindSensors(&fTrImpData[kTrPhi0], hitSens); // find detectors which may be hit by the track
568 AliDebug(2,Form("Will check %d sensors on lr:%d | esdID=%d (MClb:%d)",nsens,ila,esdID,fCurrESDtrMClb));
573 for (int isn=nsens;isn--;) {
574 AliITSURecoSens* sens = hitSens[isn];
575 int ncl = sens->GetNClusters();
579 // We need to propagate the seed to sensor on fCurrLayer staying in the frame of the sensor from prev.layer,
580 // since the transport matrix should be defined in this frame.
581 double xs; // X in the TF of current seed, corresponding to intersection with sensor plane
582 if (!seedT.GetTrackingXAtXAlpha(sens->GetXTF(),sens->GetPhiTF(),bz, xs)) {
584 if (AliDebugLevelClass()>2) {
585 printf("Failed on GetTrackingXAtXAlpha: X=%.4f alp=%.3f\n",sens->GetXTF(),sens->GetPhiTF());
591 if (xs<seedT.GetX()) {
592 if (!PropagateSeed(&seedT,xs,fCurrMass)) continue;
594 else { // some low precision tracks may hit the sensor plane outside of the layer radius
596 if (AliDebugLevelClass()>2) {
597 if (!seedT.ContainsFake()) {
598 printf("WRONG PROP on L%d, sens%d of %d: %.4f > %.4f\n",ila,isn,nsens,xs,seedT.GetX());
604 if (!seedT.PropagateParamOnlyTo(xs,bz)) continue;
606 // if (!seedT.PropagateToX(xs,bz)) continue;
607 // if (!seedT.Rotate(sens->GetPhiTF())) continue;
608 if (!seedT.RotateToAlpha(sens->GetPhiTF())) continue;
610 int clID0 = sens->GetFirstClusterId();
611 for (int icl=ncl;icl--;) { // don't change the order, clusters are sorted
612 //AliLog::SetClassDebugLevel("AliITSUTrackerGlo",10);
613 int clID = clID0 + icl;
614 if (noClSharing && fCurrLayer->GetCluster(clID)->IsClusterUsed()) continue;
615 int res = CheckCluster(&seedT,ila,clID0+icl);
616 //AliLog::SetClassDebugLevel("AliITSUTrackerGlo", 0);
618 if (res==kStopSearchOnSensor) break; // stop looking on this sensor
619 if (res==kClusterNotMatching) continue; // cluster does not match
620 // cluster is matching and it was added to the hypotheses tree
623 // cluster search is done. Do we need to have a version of this seed skipping current layer
624 if (!NeedToKill(&seedUC,kMissingCluster)) {
625 AliITSUSeed* seedSkp = NewSeedFromPool(&seedUC);
626 if (nsens) { // to do: make penalty to account for probability to miss the cluster for good reason
627 double penalty = -fCurrTrackCond->GetMissPenalty(ila);
628 seedSkp->SetChi2Cl(penalty);
630 if (seedSkp->GetChi2GloNrm()>fCurrTrackCond->GetMaxChi2GloNrm(ila)) {
631 MarkSeedFree(seedSkp);
633 else AddSeedBranch(seedSkp);
635 // transfer the new branches of the seed to the hypothesis container
636 if (fNBranchesAdded) ValidateAllowedBranches(maxNBranches);
639 if (fNCandidatesAdded) ValidateAllowedCandidates(ila,maxNCandidates);
640 // ((TObjArray*)fCurrHyp->GetLayerSeeds(ila))->Sort();
642 if (AliDebugLevelClass()>2) { //RS
643 printf(">>> All hypotheses on lr %d: \n",ila);
644 for (int ih=0;ih<fCurrHyp->GetNSeeds(ila);ih++) {printf(" #%3d ",ih); fCurrHyp->GetSeed(ila,ih)->Print();}
649 for (int ih=0;ih<fCurrHyp->GetNSeeds(ila);ih++) {
650 if (ila!=0) continue;
651 double vecL[5] = {0};
652 double matL[15] = {0};
653 AliITSUSeed* sp = fCurrHyp->GetSeed(ila,ih);
654 while(sp->GetParent()) {
655 sp->Smooth(vecL,matL);
656 if (sp->GetLayerID()>=fNLrActive-1) break;
657 sp = (AliITSUSeed*)sp->GetParent();
662 SaveReducedHypothesesTree(hypTr);
664 if (AliDebugLevelClass()>1) {
665 printf("\nSaved hypotheses for esdTrack %d (MCLab:%d)\n",esdID,fCurrESDtrMClb);
670 // AliLog::SetClassDebugLevel("AliITSUTrackerGlo",0);
674 //_________________________________________________________________________
675 AliITSUTrackHyp* AliITSUTrackerGlo::InitHypothesis(AliESDtrack *esdTr, Int_t esdID)
677 // init prolongaion candidates finding for single seed
678 AliITSUTrackHyp* hyp = GetTrackHyp(esdID);
681 fCountProlongationTrials++;
683 fCurrMass = esdTr->GetMassForTracking();
685 hyp = new AliITSUTrackHyp(fNLrActive);
686 hyp->SetESDTrack(esdTr);
687 hyp->SetUniqueID(esdID);
688 hyp->SetMass(fCurrMass);
689 hyp->SetTPCSeed( new AliExternalTrackParam(*esdTr) );
690 SetTrackHyp(hyp,esdID);
691 Bool_t lut = fUseMatLUT; // for propagation from TPC use TGeo material budget always
693 if (!TransportToLayer(hyp->GetTPCSeed(),fITS->GetNLayers(), fITS->GetLrIDActive(fNLrActive-1), fITS->GetRITSTPCRef())) hyp->SetSkip(); // propagate to outer R of ITS
700 //_________________________________________________________________________
701 Bool_t AliITSUTrackerGlo::TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo, Double_t rLim)
703 // transport seed from layerFrom to the entrance (along the direction of the track) of layerTo or to rLim (if>0), wathever is closer
706 if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
708 int dir = lTo > lFrom ? 1:-1;
709 AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
710 Bool_t checkFirst = kTRUE;
711 Bool_t limReached = kFALSE;
714 if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
717 AliITSURecoLayer* lrTo = fITS->GetLayer( (lFrom+=dir) );
718 if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
720 // go the entrance of the layer, assuming no materials in between
721 double xToGo = lrTo->GetR(-dir);
724 if (rLim<xToGo) {xToGo = rLim; limReached = kTRUE;}
727 if (rLim>xToGo) {xToGo = rLim; limReached = kTRUE;}
730 // double xts = xToGo;
731 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
732 //printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
733 //seed->Print("etp");
738 AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
740 if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
741 //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
742 //seed->Print("etp");
746 if (limReached) break;
752 //_________________________________________________________________________
753 Bool_t AliITSUTrackerGlo::TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t rLim)
755 // transport track from layerFrom to the entrance of layerTo or to rLim (if>0), wathever is closer
757 if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
759 int dir = lTo > lFrom ? 1:-1;
760 AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
761 Bool_t checkFirst = kTRUE;
762 Bool_t limReached = kFALSE;
765 if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
768 AliITSURecoLayer* lrTo = fITS->GetLayer( (lFrom+=dir) );
769 if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
771 // go the entrance of the layer, assuming no materials in between
772 double xToGo = lrTo->GetR(-dir);
775 if (rLim<xToGo) {xToGo = rLim; limReached = kTRUE;}
778 if (rLim>xToGo) {xToGo = rLim; limReached = kTRUE;}
781 // double xts = xToGo;
782 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
783 // printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
784 // seed->Print("etp");
788 AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
790 if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
791 //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
792 //seed->Print("etp");
796 if (limReached) break;
802 //_________________________________________________________________________
803 Bool_t AliITSUTrackerGlo::TransportToLayerX(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t xStop)
805 // transport track from layerFrom to the entrance of layerTo but do not pass control parameter X
807 if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
809 int dir = lTo > lFrom ? 1:-1;
810 AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
811 Bool_t checkFirst = kTRUE;
814 if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
817 AliITSURecoLayer* lrTo = fITS->GetLayer( (lFrom+=dir) );
818 if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
820 // go the entrance of the layer, assuming no materials in between
821 double xToGo = lrTo->GetR(-dir); // R of the entrance to layer
823 // double xts = xToGo;
824 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
825 // printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
826 // seed->Print("etp");
829 if ( (dir>0&&xToGo>xStop) || (dir<0&&xToGo<xStop) ) xToGo = xStop;
832 AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
834 if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
835 //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
836 //seed->Print("etp");
845 //_________________________________________________________________________
846 Bool_t AliITSUTrackerGlo::GoToExitFromLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
848 // go to the exit from lr in direction dir, applying material corrections in steps specific for this layer
849 // If check is requested, do this only provided the track has not exited the layer already
850 double xToGo = lr->GetR(dir);
851 if (check) { // do we need to track till the surface of the current layer ?
852 double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
854 AliDebug(3,Form(" dir:%d Cur: %e Tgt: %e",dir,Sqrt(curR2),xToGo));
856 if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // on the surface or outside of the layer
857 else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // on the surface or outside of the layer
859 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
860 // go via layer to its boundary, applying material correction.
861 if (!PropagateSeed(seed,xToGo,fCurrMass, lr->GetMaxStep())) return kFALSE;
867 //_________________________________________________________________________
868 Bool_t AliITSUTrackerGlo::GoToExitFromLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
870 // go to the exit from lr in direction dir, applying material corrections in steps specific for this layer
871 // If check is requested, do this only provided the track has not exited the layer already
872 double xToGo = lr->GetR(dir);
873 if (check) { // do we need to track till the surface of the current layer ?
874 double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
875 if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // on the surface or outside of the layer
876 else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // on the surface or outside of the layer
879 AliDebug(2,Form(" dir=%d : from R=%.4f -> R=%.4f",dir,Sqrt(seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY()), xToGo));
881 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
882 // go via layer to its boundary, applying material correction.
883 if (!PropagateSeed(seed,xToGo,fCurrMass, lr->GetMaxStep())) return kFALSE;
890 //_________________________________________________________________________
891 Bool_t AliITSUTrackerGlo::GoToEntranceToLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
893 // go to the entrance of lr in direction dir, w/o applying material corrections.
894 // If check is requested, do this only provided the track did not reach the layer already
895 double xToGo = lr->GetR(-dir);
896 if (check) { // do we need to track till the surface of the current layer ?
897 double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
898 if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // already passed
899 else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // already passed
901 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
902 // go via layer to its boundary, applying material correction.
903 if (!PropagateSeed(seed,xToGo,fCurrMass, 100, kFALSE)) return kFALSE;
908 //_________________________________________________________________________
909 Bool_t AliITSUTrackerGlo::GoToEntranceToLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
911 // go to the entrance of lr in direction dir, w/o applying material corrections.
912 // If check is requested, do this only provided the track did not reach the layer already
913 double xToGo = lr->GetR(-dir);
914 if (check) { // do we need to track till the surface of the current layer ?
915 double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
916 if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // already passed
917 else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // already passed
919 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
920 // go via layer to its boundary, applying material correction.
921 if (!PropagateSeed(seed,xToGo,fCurrMass, 100, kFALSE)) return kFALSE;
926 //_________________________________________________________________________
927 Bool_t AliITSUTrackerGlo::GetRoadWidth(AliITSUSeed* seed, int ilrA)
929 // calculate road width in terms of phi and z for the track which MUST be on the external radius of the layer
930 // as well as some aux info
932 AliITSURecoLayer* lrA = fITS->GetLayerActive(ilrA);
933 seed->GetXYZ(&fTrImpData[kTrXIn]); // lab position at the entrance from above
934 static AliExternalTrackParam sc; // seed copy for manipulations
937 fTrImpData[kTrPhiIn] = ATan2(fTrImpData[kTrYIn],fTrImpData[kTrXIn]);
938 if (!sc.Rotate(fTrImpData[kTrPhiIn])) return kFALSE; // go to the frame of the entry point into the layer
939 BringTo02Pi(fTrImpData[kTrPhiIn]);
940 double dr = lrA->GetDR(); // approximate X dist at the inner radius
941 if (!sc.GetXYZAt(sc.GetX()-dr, bz, fTrImpData + kTrXOut)) {
942 // special case: track does not reach inner radius, might be tangential
943 double r = sc.GetD(0,0,bz);
945 if (!sc.GetXatLabR(r,x,bz,-1)) return kFALSE;
946 dr = Abs(sc.GetX() - x);
947 if (!sc.GetXYZAt(x, bz, fTrImpData + kTrXOut)) return kFALSE;
950 fTrImpData[kTrPhiOut] = ATan2(fTrImpData[kTrYOut],fTrImpData[kTrXOut]);
951 BringTo02Pi(fTrImpData[kTrPhiOut]);
952 double sgy = sc.GetSigmaY2() + dr*dr*sc.GetSigmaSnp2() + AliITSUReconstructor::GetRecoParam()->GetSigmaY2(ilrA);
953 double sgz = sc.GetSigmaZ2() + dr*dr*sc.GetSigmaTgl2() + AliITSUReconstructor::GetRecoParam()->GetSigmaZ2(ilrA);
954 sgy = Sqrt(sgy)*fCurrTrackCond->GetNSigmaRoadY(ilrA);
955 sgz = Sqrt(sgz)*fCurrTrackCond->GetNSigmaRoadZ(ilrA);
956 double phi0 = MeanPhiSmall(fTrImpData[kTrPhiOut],fTrImpData[kTrPhiIn]);
957 double dphi0 = DeltaPhiSmall(fTrImpData[kTrPhiOut],fTrImpData[kTrPhiIn]);
959 fTrImpData[kTrPhi0] = phi0;
960 fTrImpData[kTrZ0] = 0.5*(fTrImpData[kTrZOut]+fTrImpData[kTrZIn]);
961 dphi0 += sgy/lrA->GetR();
962 fTrImpData[kTrDPhi] = dphi0<PiOver2() ? dphi0 : PiOver2();
963 fTrImpData[kTrDZ] = 0.5*Abs(fTrImpData[kTrZOut]-fTrImpData[kTrZIn]) + sgz;
968 //________________________________________
969 void AliITSUTrackerGlo::ResetSeedsPool()
971 // mark all seeds in the pool as unused
972 AliDebug(-1,Form("CurrentSize: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
974 fSeedsPool.Clear(); // seeds don't allocate memory
978 //________________________________________
979 void AliITSUTrackerGlo::MarkSeedFree(AliITSUSeed *sd)
981 // account that this seed is "deleted"
982 int id = sd->GetPoolID();
984 AliError(Form("Freeing of seed %p NOT from the pool is requested",sd));
987 // AliInfo(Form("%d %p",id, seed));
988 fSeedsPool.RemoveAt(id);
989 if (fFreeSeedsID.GetSize()<=fNFreeSeeds) fFreeSeedsID.Set( 2*fNFreeSeeds + 100 );
990 fFreeSeedsID.GetArray()[fNFreeSeeds++] = id;
993 //_________________________________________________________________________
994 Int_t AliITSUTrackerGlo::CheckCluster(AliITSUSeed* track, Int_t lr, Int_t clID)
996 // Check if the cluster (in tracking frame!) is matching to track.
997 // The track must be already propagated to sensor tracking frame.
998 // Returns: kStopSearchOnSensor if the search on given sensor should be stopped,
999 // kClusterMatching if the cluster is matching
1000 // kClusterMatching otherwise
1002 const double kTolerX = 5e-4;
1003 // AliCluster *cl = fITS->GetLayerActive(lr)->GetCluster(clID);
1004 AliCluster *cl = fCurrLayer->GetCluster(clID);
1006 Bool_t goodCl = kFALSE;
1007 int currLabel = Abs(fCurrESDtrMClb);
1009 if (cl->GetLabel(0)>=0) {
1010 for (int i=0;i<3;i++) if (cl->GetLabel(i)>=0 && cl->GetLabel(i)==currLabel) {goodCl = kTRUE; break;}
1013 if (TMath::Abs(cl->GetX())>kTolerX) { // if due to the misalingment X is large, propagate track only
1014 if (!track->PropagateParamOnlyTo(track->GetX()+cl->GetX(),GetBz())) {
1017 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1018 AliDebug(2,Form("Lost good cl on L:%d failed propagation. |ESDtrack#%d (MClb:%d)",lr,fCurrESDtrack->GetID(),fCurrESDtrMClb));
1019 track->Print("etp");
1023 return kStopSearchOnSensor; // propagation failed, seedT is intact
1026 double dy = cl->GetY()-track->GetY();
1027 double dz = cl->GetZ()-track->GetZ();
1030 double tol2 = (track->GetSigmaY2() + AliITSUReconstructor::GetRecoParam()->GetSigmaY2(lr))*
1031 fCurrTrackCond->GetNSigmaRoadY(lr)*fCurrTrackCond->GetNSigmaRoadY(lr); // RS TOOPTIMIZE
1032 if (dy2>tol2) { // the clusters are sorted in Z(col) then in Y(row).
1035 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1036 AliDebug(2,Form("Lost good cl: dy2=%e > tol2=%e |ESDtrack#%d (MClb:%d)",dy2,tol2,fCurrESDtrack->GetID(),fCurrESDtrMClb));
1037 track->Print("etp");
1039 PrintSeedClusters(track);
1042 // clusters are sorted in increasing Y and the loop goes from last (highers Y) to 1st.
1043 // if the track has too large y for this cluster (dy<0) it will be so for all other clusters of the sensor
1044 if (dy<0) return kStopSearchOnSensor; // No chance that other cluster of this sensor will match (all Y's will be even larger)
1045 else return kClusterNotMatching; // Other clusters may match
1048 tol2 = (track->GetSigmaZ2() + AliITSUReconstructor::GetRecoParam()->GetSigmaZ2(lr))*
1049 fCurrTrackCond->GetNSigmaRoadZ(lr)*fCurrTrackCond->GetNSigmaRoadZ(lr); // RS TOOPTIMIZE
1052 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1053 AliDebug(2,Form("Lost good cl on L:%d : dz2=%e > tol2=%e |ESDtrack#%d (MClb:%d)",lr,dz2,tol2,fCurrESDtrack->GetID(),fCurrESDtrMClb));
1054 track->Print("etp");
1056 PrintSeedClusters(track);
1059 return kClusterNotMatching; // Other clusters may match
1063 Double_t p[2]={cl->GetY(), cl->GetZ()};
1064 Double_t cov[3]={cl->GetSigmaY2(), cl->GetSigmaYZ(), cl->GetSigmaZ2()};
1065 double chi2 = track->GetPredictedChi2(p,cov);
1067 if (chi2>fCurrTrackCond->GetMaxTr2ClChi2(lr)) {
1069 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1070 AliDebug(2,Form("Lost good cl on L:%d : Chi2=%e > Chi2Max=%e |dy: %+.3e dz: %+.3e |ESDtrack#%d (MClb:%d)\n",
1071 lr,chi2,fCurrTrackCond->GetMaxTr2ClChi2(lr),dy,dz,fCurrESDtrack->GetID(),fCurrESDtrMClb));
1072 track->Print("etp");
1074 PrintSeedClusters(track);
1077 return kClusterNotMatching;
1080 track = NewSeedFromPool(track); // input track will be reused, use its clone for updates
1081 if (!track->Update()) {
1083 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1084 AliDebug(2,Form("Lost good cluster on L:%d : failed to update",lr));
1085 track->Print("etp");
1087 PrintSeedClusters(track);
1090 MarkSeedFree(track);
1091 return kClusterNotMatching;
1093 track->SetChi2Cl(chi2);
1094 track->SetLrClusterID(lr,clID);
1096 if (track->GetChi2GloNrm()>fCurrTrackCond->GetMaxChi2GloNrm(lr)) {
1098 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1099 AliDebug(2,Form("Lost good cl on L:%d : Chi2Glo=%e > Chi2Max=%e |dy: %+.3e dz: %+.3e |ESDtrack#%d (MClb:%d)\n",
1100 lr,track->GetChi2GloNrm(),fCurrTrackCond->GetMaxChi2GloNrm(lr),dy,dz,fCurrESDtrack->GetID(),fCurrESDtrMClb));
1101 track->Print("etp");
1103 PrintSeedClusters(track);
1106 MarkSeedFree(track);
1107 return kClusterNotMatching;
1109 // cl->IncreaseClusterUsage(); // do this only for winners
1111 track->SetFake(!goodCl);
1114 AliDebug(2,Form("AddCl(%d) Cl%d lr:%d: dY:%+8.4f dZ:%+8.4f (MC: %5d %5d %5d) |Chi2=%f(%c)| ",
1115 goodCl,clID,lr,dy,dz2,cl->GetLabel(0),cl->GetLabel(1),cl->GetLabel(2), chi2, track->IsFake() ? '-':'+'));
1118 AddSeedBranch(track);
1120 return kClusterMatching;
1123 //_________________________________________________________________________
1124 Bool_t AliITSUTrackerGlo::NeedToKill(AliITSUSeed *seed, Int_t flag)
1126 // check if the seed should not be discarded
1127 const UShort_t kMask = 0xffff;
1128 if (flag==kMissingCluster) {
1129 int lastChecked = seed->GetLayerID();
1130 UShort_t patt = seed->GetHitsPattern();
1131 if (lastChecked) patt |= (~(kMask<<lastChecked))&fCurrTrackCond->GetAllowedLayers(); // not all layers were checked, complete unchecked ones by potential hits
1132 Bool_t seedOK = fCurrTrackCond->CheckPattern(patt);
1138 //______________________________________________________________________________
1139 Bool_t AliITSUTrackerGlo::PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep, Bool_t matCorr)
1141 // propagate seed to given x applying material correction if requested
1142 const Double_t kEpsilon = 1e-5;
1143 Double_t xpos = seed->GetX();
1144 Int_t dir = (xpos<xToGo) ? 1:-1;
1145 Double_t xyz0[3],xyz1[3];
1147 Bool_t updTime = dir>0 && seed->IsStartedTimeIntegral();
1148 if (matCorr || updTime) seed->GetXYZ(xyz1); //starting global position
1149 while ( (xToGo-xpos)*dir > kEpsilon){
1150 Double_t step = dir*TMath::Min(TMath::Abs(xToGo-xpos), maxStep);
1151 Double_t x = xpos+step;
1152 Double_t bz=GetBz(); // getting the local Bz
1153 if (!seed->PropagateToX(x,bz)) return kFALSE;
1155 if (matCorr || updTime) {
1156 xyz0[0]=xyz1[0]; // global pos at the beginning of step
1159 seed->GetXYZ(xyz1); // // global pos at the end of step
1162 ds = GetMaterialBudget(xyz0,xyz1,xx0,xrho);
1163 if (dir>0) xrho = -xrho; // outward should be negative
1164 if (!seed->ApplyMaterialCorrection(xx0,xrho,mass,kFALSE)) {AliInfo("Fail2"); seed->Print("etp"); return kFALSE;}
1166 else { // matCorr is not requested but time integral is
1167 double d0 = xyz1[0]-xyz0[0];
1168 double d1 = xyz1[1]-xyz0[1];
1169 double d2 = xyz1[2]-xyz0[2];
1170 ds = TMath::Sqrt(d0*d0+d1*d1+d2*d2);
1173 if (updTime) seed->AddTimeStep(ds);
1174 xpos = seed->GetX();
1179 //______________________________________________________________________________
1180 Bool_t AliITSUTrackerGlo::PropagateSeed(AliExternalTrackParam *seed, Double_t xToGo, Double_t mass, Double_t maxStep, Bool_t matCorr)
1182 // propagate seed to given x applying material correction if requested
1183 const Double_t kEpsilon = 1e-5;
1184 Double_t xpos = seed->GetX();
1185 Int_t dir = (xpos<xToGo) ? 1:-1;
1186 Double_t xyz0[3],xyz1[3];
1189 if (AliDebugLevelClass()>1) {
1190 AliDebug(2,Form("Before Propagate to X=%f with M=%.3f MaxStep=%.4f MatCorr=%d",xToGo,mass,maxStep,matCorr));
1191 seed->AliExternalTrackParam::Print();
1194 Bool_t updTime = dir>0 && seed->IsStartedTimeIntegral();
1195 if (matCorr || updTime) seed->GetXYZ(xyz1); //starting global position
1196 while ( (xToGo-xpos)*dir > kEpsilon){
1197 Double_t step = dir*TMath::Min(TMath::Abs(xToGo-xpos), maxStep);
1198 Double_t x = xpos+step;
1199 Double_t bz=GetBz(); // getting the local Bz
1200 if (!seed->PropagateTo(x,bz)) return kFALSE;
1202 if (matCorr || updTime) {
1203 xyz0[0]=xyz1[0]; // global pos at the beginning of step
1206 seed->GetXYZ(xyz1); // // global pos at the end of step
1210 ds = GetMaterialBudget(xyz0,xyz1,xx0,xrho);
1211 if (dir>0) xrho = -xrho; // outward should be negative
1212 if (!seed->CorrectForMeanMaterial(xx0,xrho,mass)) return kFALSE;
1214 else { // matCorr is not requested but time integral is
1215 double d0 = xyz1[0]-xyz0[0];
1216 double d1 = xyz1[1]-xyz0[1];
1217 double d2 = xyz1[2]-xyz0[2];
1218 ds = TMath::Sqrt(d0*d0+d1*d1+d2*d2);
1221 if (updTime) seed->AddTimeStep(ds);
1223 xpos = seed->GetX();
1226 if (AliDebugLevelClass()>1) {
1227 AliDebug(2,Form("After Propagate to X=%f",xToGo));
1228 seed->AliExternalTrackParam::Print();
1234 //______________________________________________________________________________
1235 Bool_t AliITSUTrackerGlo::FinalizeHypothesis(AliITSUTrackHyp* hyp)
1237 // finalize single TPC track hypothesis
1238 if (!hyp || hyp->GetESDTrack()->IsOn(AliESDtrack::kITSin)) return kFALSE;
1239 AliITSUSeed* winner = 0;
1241 fCurrMass = hyp->GetMass();
1242 if (!(winner=hyp->DefineWinner(fCurrTrackCond->GetActiveLrInner()))) return kFALSE;
1245 #ifdef _ITSU_TUNING_MODE_ // fill tuning histos
1246 TObjArray* dest = hyp->GetLabel()>=0 ? fCHistoArrCorr : fCHistoArrFake;
1248 AliITSUSeed* sd = winner;
1249 double htrPt = hyp->Pt();
1252 if ( (clID=sd->GetLrCluster(lrID))<0 ) continue;
1253 ((TH2*)dest->At( GetHistoID(lrID,kHChi2Nrm ,fCurrPassID) ))->Fill(htrPt,sd->GetChi2GloNrm());
1254 ((TH2*)dest->At( GetHistoID(lrID,kHBestInBranch,fCurrPassID) ))->Fill(htrPt,sd->GetOrdBranch());
1255 ((TH2*)dest->At( GetHistoID(lrID,kHBestInCand ,fCurrPassID) ))->Fill(htrPt,sd->GetOrdCand());
1257 if (dest==fCHistoArrFake && !sd->IsFake()) continue; // for the fake seeds fill only fake clusters part
1259 ((TH2*)dest->At( GetHistoID(lrID,kHResY ,fCurrPassID) ))->Fill(htrPt,sd->GetResidY());
1260 ((TH2*)dest->At( GetHistoID(lrID,kHResZ ,fCurrPassID) ))->Fill(htrPt,sd->GetResidZ());
1261 ((TH2*)dest->At( GetHistoID(lrID,kHResYP ,fCurrPassID) ))->Fill(htrPt,sd->GetPullY());
1262 ((TH2*)dest->At( GetHistoID(lrID,kHResZP ,fCurrPassID) ))->Fill(htrPt,sd->GetPullZ());
1263 ((TH2*)dest->At( GetHistoID(lrID,kHChi2Cl ,fCurrPassID) ))->Fill(htrPt,sd->GetChi2Cl());
1265 } while((sd=(AliITSUSeed*)sd->GetParent()));
1267 ((TH2*)dest->At( GetHistoID(-1,kHChiMatch,fCurrPassID) ))->Fill(htrPt,winner->GetChi2ITSTPC());
1268 ((TH2*)dest->At( GetHistoID(-1,kHChiITSSA,fCurrPassID) ))->Fill(htrPt,winner->GetChi2ITSSA());
1273 CheckClusterSharingConflicts(hyp);
1274 return hyp->GetWinner(); // winner might change of disappear after resolving conflicts
1278 //______________________________________________________________________________
1279 void AliITSUTrackerGlo::FinalizeHypotheses()
1281 // select winner for each hypothesis, remove cl. sharing conflicts
1282 AliCodeTimerAuto("",0);
1284 int* esdTrackIndex = fESDIndex.GetArray();
1285 for (int ih=0;ih<fNTracksESD;ih++) FinalizeHypothesis(GetTrackHyp(esdTrackIndex[ih])); // finlize in decreasing pt order
1287 for (int ih=fNTracksESD;ih--;) UpdateESDTrack(GetTrackHyp(esdTrackIndex[ih]),AliESDtrack::kITSin);
1291 //______________________________________________________________________________
1292 void AliITSUTrackerGlo::CheckClusterSharingConflicts(AliITSUTrackHyp* hyp)
1294 // remove eventual cluster sharing conflicts
1295 AliITSUSeed* winner = 0;
1296 if (!(winner=hyp->GetWinner())) return;
1297 UShort_t idH = (UShort_t)hyp->GetUniqueID()+1;
1299 int inLr = fCurrTrackCond->GetActiveLrInner();
1300 AliITSUSeed* winSD=winner;
1302 if ( (clID=winSD->GetLrCluster(lrID))<0 ) continue;
1303 AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1304 Int_t refID = cl->GetRecoInfo(); // was it already referred by some track (with refID-1 if >0)
1305 if (!refID) {cl->SetRecoInfo(idH); continue;} // if not, refer from track to cluster
1306 if (refID==idH) continue; // ignore reference to itself
1307 // the cluster is already used by other track, need to resolve the conflict
1308 AliITSUTrackHyp* hypC = GetTrackHyp(refID-1); // competitor
1309 // printf("Ref to %d (%p) from %d Cl %d %d\n",refID-1,hypC, idH-1,clID,lrID);
1310 AliITSUSeed* winnerC = hypC->GetWinner();
1312 printf("Missing winner, ERROR\n");
1317 if (AliDebugLevelClass()>1) { //------------------------ DEBUG -------------------
1318 AliInfo(Form("Shared cl#%4d on Lr:%d: Hyp%3d/Q:%6.3f vs Hyp%3d/Q:%6.3f",
1319 clID,lrID,idH-1,winner->GetQualityVar(),refID-1,winnerC->GetQualityVar()));
1320 // dump winner of hyp
1321 printf("#%4d Pt:%.3f Chi:%6.3f/%6.3f/%6.3f Ncl:%d MCits%+5d MCtpc:%+5d |",
1322 idH-1,winner->Pt(),winner->GetChi2GloNrm(),winner->GetChi2ITSTPC(),winner->GetChi2ITSSA(),
1323 winner->GetNLayersHit(),hyp->GetITSLabel(),hyp->GetESDTrack()->GetTPCLabel());
1325 AliITSUSeed* sd = winner;
1328 int clIDt = sd->GetLrCluster(lrs);
1329 if (clIDt<0) continue;
1330 while( lrs>++prevL ) printf("%4s ","----");
1331 printf("%4d (%5.1f)",clIDt,sd->GetChi2Cl());
1332 } while ((sd=(AliITSUSeed*)sd->GetParent()));
1334 // dump winner of hypC
1335 printf("#%4d Pt:%.3f Chi:%6.3f/%6.3f/%6.3f Ncl:%d MCits%+5d MCtpc:%+5d |",
1336 refID-1,winnerC->Pt(),winnerC->GetChi2GloNrm(),winnerC->GetChi2ITSTPC(),winnerC->GetChi2ITSSA(),
1337 winnerC->GetNLayersHit(),hypC->GetITSLabel(),hypC->GetESDTrack()->GetTPCLabel());
1342 int clIDt = sd->GetLrCluster(lrs);
1343 if (clIDt<0) continue;
1344 while( lrs>++prevL ) printf("%4s ","----");
1345 printf("%4d (%5.1f)",clIDt,sd->GetChi2Cl());
1346 } while ((sd=(AliITSUSeed*)sd->GetParent()));
1348 } //-------------------------------------------------- DEBUG -------------------
1351 if (winner->GetQualityVar()<winnerC->GetQualityVar()) { // current tracks is better then competitor track
1352 FlagSeedClusters(winnerC,kFALSE,refID); // unflag cluster usage by loser
1353 cl->SetRecoInfo(idH);
1355 //if ( hypC->GetLabel()>=0) AliInfo(Form("KILLING CORRECT: %d",refID-1));
1357 if (hypC->DefineWinner(inLr)) { // find new winner instead of suppressed candidate
1359 CheckClusterSharingConflicts(hypC); // and check its sharing conflicts again
1360 if (winner->IsKilled()) break; // the current winner might have been killed during check of new winner of hypC!
1363 else { // competitor hypC is better than the hyp
1364 FlagSeedClusters(winner,kFALSE,idH); // unflag cluster usage by loser
1366 //if ( hyp->GetLabel()>=0) AliInfo(Form("KILLING CORRECT: %d",idH-1));
1367 if (hyp->DefineWinner(inLr)) {
1369 CheckClusterSharingConflicts(hyp);
1374 } while ((winSD=(AliITSUSeed*)winSD->GetParent()));
1379 //______________________________________________________________________________
1380 void AliITSUTrackerGlo::UpdateESDTrack(AliITSUTrackHyp* hyp,Int_t flag)
1382 // update ESD track with current best hypothesis
1384 AliESDtrack* esdTr = hyp->GetESDTrack();
1385 if (!esdTr || esdTr->IsOn(flag)) return;
1386 AliITSUSeed* win = hyp->GetWinner();
1387 if (!win || win->IsKilled()) return;
1391 case AliESDtrack::kITSin:
1392 esdTr->UpdateTrackParams(hyp,flag); // update kinematics
1394 // set fakes cluster info
1396 UShort_t clfake = 0;
1398 AliITSUSeed* sd = win;
1401 int lr, clID = sd->GetLrCluster(lr);
1402 if (sd->IsFake()) clfake |= 0x1<<lr;
1404 esdTr->SetITSModuleIndex(ip++, sd->GetLrClusterID());
1405 AliITSUClusterPix *cl = (AliITSUClusterPix*)fITS->GetLayerActive(lr)->GetCluster(clID);
1406 #ifdef _ITSU_TUNING_MODE_
1407 if (cl->IsSplit()) clSplit |= 0x1<<lr;
1410 } while ((sd=(AliITSUSeed*)sd->GetParent()));
1412 // RS: Temporary set special flag for tracks from the afterburner
1413 if (fCurrPassID>0) clfake |= 0x1<<7;
1415 esdTr->SetITSSharedMap(clfake);
1416 esdTr->SetITSModuleIndex(10,clSplit);
1418 // TEMPORARY: store iteration id
1419 esdTr->SetITSModuleIndex(11,fCurrPassID);
1422 case AliESDtrack::kITSout:
1423 // here the stored chi2 will correspond to backward ITS-SA fit
1424 esdTr->UpdateTrackParams(hyp,flag); // update kinematics
1426 // TODO: avoid setting friend
1429 case AliESDtrack::kITSrefit:
1431 // at the moment fill as a chi2 the TPC-ITS matching chi2
1432 chiSav = hyp->GetChi2();
1433 hyp->SetChi2(win->GetChi2ITSTPC());
1434 esdTr->UpdateTrackParams(hyp,flag); // update kinematics
1435 hyp->SetChi2(chiSav);
1437 // TODO: avoid setting cluster info
1440 AliFatal(Form("Unknown flag %d",flag));
1443 esdTr->SetITSLabel(hyp->GetITSLabel());
1444 // transfer chip indices
1448 //______________________________________________________________________________
1449 Double_t AliITSUTrackerGlo::RefitTrack(AliITSUTrackHyp* trc, Double_t rDest, Int_t &nclFit, Int_t stopCond)
1451 // refit track till radius rDest. The cluster,mass info is taken from fCurrHyp (and its winner seed)
1452 // if stopCond<0 : propagate till last cluster then stop
1453 // if stopCond==0: propagate till last cluster then try to go till limiting rDest, don't mind if fail
1454 // if stopCond>0 : rDest must be reached
1456 double rCurr = Sqrt(trc->GetX()*trc->GetX() + trc->GetY()*trc->GetY());
1457 int dir,lrStart,lrStop;
1459 dir = rCurr<rDest ? 1 : -1;
1460 lrStart = fITS->FindFirstLayerID(rCurr,dir);
1461 lrStop = fITS->FindLastLayerID(rDest,dir); // lr id before which we have to stop
1464 if (AliDebugLevelClass()>2) {
1465 printf("Refit %d: Lr:%d (%f) -> Lr:%d (%f)\n",dir,lrStart,rCurr, lrStop,rDest);
1466 printf("Before refit: "); trc->Print();
1469 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));
1471 Int_t clInfo[2*AliITSUAux::kMaxLayers];
1472 Int_t nCl = fCurrHyp->FetchClusterInfo(clInfo);
1473 fCurrMass = fCurrHyp->GetMass();
1474 AliITSUTrackHyp tmpTr(*(AliKalmanTrack*)trc);
1479 int lrStop1 = lrStop+dir;
1480 for (int ilr=lrStart;ilr!=lrStop1;ilr+=dir) {
1481 AliITSURecoLayer* lr = fITS->GetLayer(ilr);
1482 if ( dir*(rCurr-lr->GetR(dir))>0) continue; // this layer is already passed
1483 int ilrA2,ilrA = lr->GetActiveID();
1484 // passive layer or active w/o hits will be traversed on the way to next cluster
1485 if (!lr->IsActive() || clInfo[ilrA2=(ilrA<<1)]<0) continue;
1487 // select the order in which possible 2 clusters (in case of the overlap) will be traversed and fitted
1489 if (dir>0) { // clusters are stored in increasing radius order
1490 iclLr[nclLr++]=clInfo[ilrA2++];
1491 if (clInfo[ilrA2]>=0) iclLr[nclLr++]=clInfo[ilrA2];
1494 if ( clInfo[ilrA2+1]>=0 ) iclLr[nclLr++]=clInfo[ilrA2+1];
1495 iclLr[nclLr++]=clInfo[ilrA2];
1498 Bool_t transportedToLayer = kFALSE;
1499 for (int icl=0;icl<nclLr;icl++) {
1500 AliITSUClusterPix* clus = (AliITSUClusterPix*)lr->GetCluster(iclLr[icl]);
1501 AliITSURecoSens* sens = lr->GetSensorFromID(clus->GetVolumeId());
1502 if (!tmpTr.Rotate(sens->GetPhiTF())) {
1504 AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1505 AliDebug(2,Form("Failed on rotate to %f | ESDtrack#%d (MClb:%d)",sens->GetPhiTF(),trESD->GetID(),trESD->GetTPCLabel()));
1510 double xClus = sens->GetXTF()+clus->GetX();
1511 if (!transportedToLayer) {
1512 if (ilr!=lrStart && !TransportToLayerX(&tmpTr,lrStart,ilr,xClus)) {
1514 AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1515 AliDebug(2,Form("Failed to transport %d -> %d | ESDtrack#%d (MClb:%d)\n",lrStart,ilr,trESD->GetID(),trESD->GetTPCLabel()));
1517 return -1; // go to the entrance to the layer
1520 transportedToLayer = kTRUE;
1524 if (AliDebugLevelClass()>1) {
1525 AliDebug(2,Form("Propagate to cl:%d on lr %d Need to go %.4f -> %.4f",icl,ilrA,tmpTr.GetX(),xClus));
1529 if (!PropagateSeed(&tmpTr,xClus,fCurrMass)) {
1531 AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1532 AliDebug(2,Form("Failed on propagate to %f (dir=%d) | ESDtrack#%d (MClb:%d)",xClus,dir,trESD->GetID(),trESD->GetTPCLabel()));
1537 Double_t p[2]={clus->GetY(), clus->GetZ()};
1538 Double_t cov[3]={clus->GetSigmaY2(), clus->GetSigmaYZ(), clus->GetSigmaZ2()};
1539 double chi2cl = tmpTr.GetPredictedChi2(p,cov);
1542 #ifdef _ITSU_TUNING_MODE_
1543 TObjArray* dest = trc->GetLabel()>=0 ? fCHistoArrCorr : fCHistoArrFake;
1544 if (dest && fTrackPhaseID>kClus2Tracks) {
1546 double htrPt = tmpTr.Pt();
1547 double dy = p[0]-tmpTr.GetY();
1548 double dz = p[1]-tmpTr.GetZ();
1549 ((TH2*)dest->At( GetHistoID(ilrA,kHResY,0,fTrackPhaseID) ))->Fill(htrPt,dy);
1550 ((TH2*)dest->At( GetHistoID(ilrA,kHResZ,0,fTrackPhaseID) ))->Fill(htrPt,dz);
1551 double errY = tmpTr.GetSigmaY2();
1552 double errZ = tmpTr.GetSigmaZ2();
1553 if (errY>0) ((TH2*)dest->At( GetHistoID(ilrA,kHResYP,0,fTrackPhaseID) ))->Fill(htrPt,dy/Sqrt(errY));
1554 if (errZ>0) ((TH2*)dest->At( GetHistoID(ilrA,kHResZP,0,fTrackPhaseID) ))->Fill(htrPt,dz/Sqrt(errZ));
1555 ((TH2*)dest->At( GetHistoID(ilrA,kHChi2Cl,0,fTrackPhaseID) ))->Fill(htrPt,chi2cl);
1559 if ( !tmpTr.Update(p,cov) ) {
1561 AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1562 AliDebug(2,Form("Failed on Update | ESDtrack#%d (MClb:%d)",trESD->GetID(),trESD->GetTPCLabel()));
1567 if (AliDebugLevelClass()>1) {
1568 printf("AfterRefit: "); tmpTr.Print();
1571 if (++nclFit==nCl && stopCond<0) {
1572 *trc = (AliKalmanTrack&)tmpTr;
1573 return chi2; // it was requested to not propagate after last update
1578 // All clusters were succesfully fitted. Even if the track does not reach rDest, this is enough to validate it.
1579 // Still, try to go as close as possible to rDest.
1581 if (lrStart!=lrStop) {
1582 if (!TransportToLayer(&tmpTr,lrStart,lrStop)) {
1584 AliDebug(1,Form("Failed on TransportToLayer %d->%d | ESDtrack#%d (MClb:%d), X=%f",lrStart,lrStop,fCurrESDtrack->GetID(),fCurrESDtrMClb,tmpTr.GetX()));
1586 return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
1588 if (!GoToExitFromLayer(&tmpTr,fITS->GetLayer(lrStop),dir)) {
1590 AliDebug(1,Form("Failed on GoToExitFromLayer %d | ESDtrack#%d (MClb:%d), X=%f",lrStop,fCurrESDtrack->GetID(),fCurrESDtrMClb,tmpTr.GetX()));
1592 return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
1595 // go to the destination radius. Note that here we don't select direction to avoid precision problems
1596 if (!tmpTr.GetXatLabR(rDest,rDest,GetBz(),0) || !PropagateSeed(&tmpTr,rDest,fCurrMass, 100, kFALSE)) {
1597 return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
1599 *trc = (AliKalmanTrack&)tmpTr;
1601 if (AliDebugLevelClass()>2) {
1602 printf("After refit (now at lr %d): ",lrStop); trc->Print();
1608 //__________________________________________________________________
1609 void AliITSUTrackerGlo::CookMCLabel(AliITSUTrackHyp* hyp)
1613 const int kMaxLbPerCl = 3;
1614 int lbID[kMaxLayers*6],lbStat[kMaxLayers*6];
1615 Int_t lr,nLab=0,nCl=0;
1616 AliITSUSeed *seed = hyp->GetWinner();
1618 int clID = seed->GetLrCluster(lr);
1620 AliCluster *cl = fITS->GetLayerActive(lr)->GetCluster(clID);
1622 for (int imc=0;imc<kMaxLbPerCl;imc++) { // labels within single cluster
1623 int trLb = cl->GetLabel(imc);
1625 // search this mc track in already accounted ones
1627 for (iLab=0;iLab<nLab;iLab++) if (lbID[iLab]==trLb) break;
1628 if (iLab<nLab) lbStat[iLab]++;
1633 } // loop over given cluster's labels
1635 seed = (AliITSUSeed*)seed->GetParent();
1636 } // loop over clusters
1638 AliESDtrack* esdTr = hyp->GetESDTrack();
1639 int tpcLab = esdTr ? Abs(esdTr->GetTPCLabel()) : -kDummyLabel;
1641 int maxLab=0,nTPCok=0;
1642 for (int ilb=nLab;ilb--;) {
1643 int st = lbStat[ilb];
1644 if (lbStat[maxLab]<st) maxLab=ilb;
1645 if (tpcLab==lbID[ilb]) nTPCok += st;
1647 hyp->SetFakeRatio(1.-float(nTPCok)/nCl);
1648 hyp->SetLabel( nTPCok==nCl ? tpcLab : -tpcLab);
1649 hyp->SetITSLabel( lbStat[maxLab]==nCl ? lbID[maxLab] : -lbID[maxLab]); // winner label
1653 hyp->SetFakeRatio(-1.);
1654 hyp->SetLabel( -tpcLab );
1655 hyp->SetITSLabel( kDummyLabel );
1658 //__________________________________________________________________
1659 Bool_t AliITSUTrackerGlo::AddSeedBranch(AliITSUSeed* seed)
1661 // add new prolongation branch for the seed to temporary array. The seeds are sorted according to Compare f-n
1662 if (fNCandidatesAdded+fNBranchesAdded>=fLayerMaxCandidates ) {
1663 AliInfo(Form("Number of candidates at layer %d for current seed reached %d, increasing buffer",fCurrActLrID,fLayerMaxCandidates));
1664 fLayerMaxCandidates = 2*(fLayerMaxCandidates+1);
1665 AliITSUSeed** tmpArr = fLayerCandidates;
1666 fLayerCandidates = new AliITSUSeed*[fLayerMaxCandidates];
1667 memcpy(fLayerCandidates,tmpArr,(fNCandidatesAdded+fNBranchesAdded)*sizeof(AliITSUSeed*));
1668 delete tmpArr; // delete only array, not objects
1670 AliITSUSeed** branches = &fLayerCandidates[fNCandidatesAdded]; // note: fNCandidatesAdded is incremented after adding all branches of current seed
1671 int slot=fNBranchesAdded++;
1672 for (int slotF=slot;slotF--;) { // slotF is always slot-1
1673 AliITSUSeed* si = branches[slotF];
1674 if (si->Compare(seed)<0) break; // found the last seed with better quality
1675 // otherwise, shift the worse seed to the next slot
1676 branches[slot] = si;
1677 slot = slotF; // slot should be slotF+1
1679 // if needed, move worse seeds
1680 branches[slot] = seed;
1685 //__________________________________________________________________
1686 void AliITSUTrackerGlo::ValidateAllowedBranches(Int_t acceptMax)
1688 // keep allowed number of branches for current seed and disable extras
1689 AliCodeTimerAuto("",0);
1690 int nb = Min(fNBranchesAdded,acceptMax);
1691 // if (nb<fNBranchesAdded) printf("ValidateAllowedBranches: %d of %d (%d) on lr %d\n",nb,fNBranchesAdded,fNCandidatesAdded,ilr);
1692 // disable unused branches
1693 AliITSUSeed** branches = &fLayerCandidates[fNCandidatesAdded];
1695 #ifdef _ITSU_TUNING_MODE_
1696 for (int ib=0;ib<fNBranchesAdded;ib++) branches[ib]->SetOrdBranch(ib);
1699 for (int ib=nb;ib<fNBranchesAdded;ib++) MarkSeedFree(branches[ib]);
1700 fNCandidatesAdded += nb; // update total candidates counter
1701 fNBranchesAdded = 0; // reset branches counter
1705 //__________________________________________________________________
1706 void AliITSUTrackerGlo::ValidateAllowedCandidates(Int_t ilr, Int_t acceptMax)
1708 // transfer allowed number of branches to hypothesis container
1710 AliCodeTimerAuto("",0);
1711 // sort candidates in increasing order of chi2
1712 static int lastSize = 0;
1713 static int *index = 0;
1714 static float *chi2 = 0;
1715 if (fLayerMaxCandidates>lastSize) {
1716 lastSize = fLayerMaxCandidates;
1719 index = new int[lastSize];
1720 chi2 = new float[lastSize];
1722 for (int i=fNCandidatesAdded;i--;) chi2[i] = fLayerCandidates[i]->GetChi2GloNrm();
1723 Sort(fNCandidatesAdded,chi2,index,kFALSE);
1726 if (ilr>fCurrTrackCond->GetActiveLrInner()) { // just take 1st acceptMax candidates
1727 nb = Min(fNCandidatesAdded,acceptMax);
1728 for (int ib=0;ib<nb;ib++) AddProlongationHypothesis(fLayerCandidates[index[ib]],ilr);
1730 else { // for innermost layer test ITS SA backward chi2 and matching to TPC
1731 AliITSUSeed* wn0 = fCurrHyp->GetWinner(); // in principle, should be NULL
1732 for (nb=0;nb<fNCandidatesAdded;nb++) {
1733 AliITSUSeed* sd = fLayerCandidates[index[nb]];
1734 fCurrHyp->SetWinner(sd);
1735 if (!CheckBackwardMatching(sd)) MarkSeedFree(sd); // discard bad backward tracks
1737 AddProlongationHypothesis(sd,ilr);
1738 if (++nacc==acceptMax) {nb++; break;} // the rest will be discarded
1741 fCurrHyp->SetWinner(wn0); // restore original winner (NULL?)
1743 // disable unused candidates
1744 for (int ib=nb;ib<fNCandidatesAdded;ib++) MarkSeedFree(fLayerCandidates[index[ib]]);
1745 fNCandidatesAdded = 0; // reset candidates counter
1749 //__________________________________________________________________
1750 Bool_t AliITSUTrackerGlo::CheckBackwardMatching(AliITSUSeed* seed)
1752 // check seed backward propagation chi2 and matching to TPC
1753 double bz0 = GetBz();
1754 double rDest = fITS->GetRITSTPCRef(); // reference radius for comparison
1755 AliITSUTrackHyp trback;
1756 trback.AliExternalTrackParam::operator=(*seed);
1757 trback.ResetCovariance(10000);
1759 double chi2sa = RefitTrack(&trback,rDest,nclFit,1);
1760 if (chi2sa<0) return kFALSE;
1761 int ndf = nclFit*2-5;
1762 if (ndf>0) chi2sa /= ndf;
1763 if (chi2sa>fCurrTrackCond->GetMaxITSSAChi2(nclFit)) return kFALSE;
1765 // relate to TPC track at outer layer
1766 AliExternalTrackParam* tpcSeed = fCurrHyp->GetTPCSeed();
1767 if (!trback.Rotate(tpcSeed->GetAlpha()) || !trback.PropagateParamOnlyTo(tpcSeed->GetX(),bz0)) {
1769 if (AliDebugLevelClass()>+1 && !seed->ContainsFake()) {
1770 AliInfo(Form("Failed to propagate seed to TPC track @ X:%.3f Alpha:%.3f",tpcSeed->GetX(),tpcSeed->GetAlpha()));
1777 double chi2Match = trback.GetPredictedChi2(tpcSeed)/5;
1778 if (chi2Match>fCurrTrackCond->GetMaxITSTPCMatchChi2()) return kFALSE;
1780 seed->SetChi2ITSSA(chi2sa);
1781 seed->SetChi2ITSTPC(chi2Match);
1785 //__________________________________________________________________
1786 void AliITSUTrackerGlo::SaveReducedHypothesesTree(AliITSUTrackHyp* dest)
1788 // remove those hypothesis seeds which dont lead to candidates at final layer
1790 // 1st, flag the seeds to save
1791 int lr0 = fCurrTrackCond->GetActiveLrInner();
1792 for (int isd=0;isd<fCurrHyp->GetNSeeds(lr0);isd++) {
1793 AliITSUSeed* seed = fCurrHyp->RemoveSeed(lr0,isd);
1794 if (!seed) continue;
1795 seed->FlagTree(AliITSUSeed::kSave);
1796 dest->AddSeed(seed,lr0);
1798 for (int ilr=0;ilr<fNLrActive;ilr++) {
1799 int nsd = fCurrHyp->GetNSeeds(ilr);
1800 for (int isd=0;isd<nsd;isd++) {
1801 AliITSUSeed* seed = fCurrHyp->RemoveSeed(ilr,isd);
1802 if (!seed) continue; // already discarded or saved
1803 if (seed->IsSaved()) dest->AddSeed(seed,ilr);
1804 else MarkSeedFree(seed);
1808 // AliInfo(Form("SeedsPool: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
1811 //__________________________________________________________________
1812 void AliITSUTrackerGlo::CleanHypothesis(AliITSUTrackHyp* hyp)
1816 for (int ilr=0;ilr<fNLrActive;ilr++) {
1817 int nsd = hyp->GetNSeeds(ilr);
1818 for (int isd=0;isd<nsd;isd++) {
1819 AliITSUSeed* seed = hyp->RemoveSeed(ilr,isd);
1820 if (!seed) continue; // already discarded or saved
1826 //__________________________________________________________________
1827 void AliITSUTrackerGlo::FlagSplitClusters()
1829 // set special bit on split clusters using MC info
1830 for (int ilr=fNLrActive;ilr--;) {
1832 AliITSURecoLayer* lr = fITS->GetLayerActive(ilr);
1833 for (int isn=lr->GetNSensors();isn--;) {
1834 AliITSURecoSens* sens = lr->GetSensor(isn);
1835 int nCl = sens->GetNClusters();
1837 int cl0 = sens->GetFirstClusterId();
1838 for (int icl=nCl;icl--;) {
1839 AliITSUClusterPix *cl = (AliITSUClusterPix*)lr->GetCluster(cl0+icl);
1840 for (int icl1=icl;icl1--;) {
1841 AliITSUClusterPix *cl1 = (AliITSUClusterPix*)lr->GetCluster(cl0+icl1);
1842 if (cl->HasCommonTrack(cl1)) {
1843 if (!cl->IsSplit()) nsplit++;
1844 if (!cl1->IsSplit()) nsplit++;
1851 AliInfo(Form("%4d out of %4d clusters are split",nsplit,lr->GetNClusters()));
1856 //__________________________________________________________________
1857 Bool_t AliITSUTrackerGlo::ContainsSplitCluster(const AliITSUSeed* seed, Int_t maxSize)
1859 // check if the seed contains split cluster with size < maxSize
1861 if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
1862 AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1863 if (cl->IsSplit() && cl->GetNPix()<maxSize ) return kTRUE;
1865 return seed->GetParent() ? ContainsSplitCluster((AliITSUSeed*)seed->GetParent(),maxSize) : kFALSE;
1869 //__________________________________________________________________
1870 void AliITSUTrackerGlo::PrintSeedClusters(const AliITSUSeed* seed, Option_t* option)
1872 // print seeds clusters
1874 if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
1875 AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1878 if (seed->GetParent()) PrintSeedClusters((AliITSUSeed*)seed->GetParent(), option);
1882 //__________________________________________________________________
1883 void AliITSUTrackerGlo::FlagSeedClusters(const AliITSUSeed* seed, Bool_t flg, UShort_t ref)
1885 // mark used clusters
1888 if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
1889 AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1890 if (ref) { // do we need to set or delete cluster->track ref?
1892 if (!cl->GetRecoInfo()) cl->SetRecoInfo(ref); // set ref only if cluster already does not refer to other track, inc.counter
1895 if (cl->GetRecoInfo()==ref) cl->SetRecoInfo(0); // unset reference only if it refers to ref, decrease counter
1899 seed = (AliITSUSeed*)seed->GetParent();
1904 //__________________________________________________________________
1905 void AliITSUTrackerGlo::CheckClusterUsage()
1907 // check cluster usage info
1908 printf("ClusterUsage at pass %d\n",fCurrPassID);
1909 for (int ilr=0;ilr<fNLrActive;ilr++) {
1910 AliITSURecoLayer* lr = fITS->GetLayerActive(ilr);
1911 int ncl = lr->GetNClusters();
1913 int nclUs0CntShr1 = 0;
1917 for (int icl=0;icl<ncl;icl++) {
1918 AliITSUClusterPix *cl = (AliITSUClusterPix*)lr->GetCluster(icl);
1919 int usc = cl->GetClUsage();
1921 if (cl->IsClusterUsed() || cl->IsClusterShared()) {
1923 printf("Lr%d Cl%4d: UseCnt=0 but UseFlg=%d UseShr=%d\n",ilr,icl,cl->IsClusterUsed(),cl->IsClusterShared());
1928 if (!cl->IsClusterUsed()) {
1930 printf("Lr%d Cl%4d: UseCnt=%d but UseFlg=%d UseShr=%d\n",ilr,icl,usc, cl->IsClusterUsed(),cl->IsClusterShared());
1932 if (cl->IsClusterShared()) {
1934 printf("Lr%d Cl%4d: UseCnt=%d but UseFlg=%d UseShr=%d\n",ilr,icl,usc, cl->IsClusterUsed(),cl->IsClusterShared());
1938 printf("Lr%d Cl%4d: UseCnt=%d!! but UseFlg=%d UseShr=%d\n",ilr,icl,usc, cl->IsClusterUsed(),cl->IsClusterShared());
1941 printf("ClUsage Lr%d: Ncl=%5d Nusd=%5d (%5.2f) Use0CS1:%4d Use1C0:%4d Use1S1:%4d UseL:%d\n",
1942 ilr,ncl,nclUsed,ncl? float(nclUsed)/ncl : 0, nclUs0CntShr1,nclUs1Cnt0,nclUs1Shr1,nclUseL);
1947 #ifdef _ITSU_TUNING_MODE_
1948 //__________________________________________________________________
1949 TObjArray* AliITSUTrackerGlo::BookControlHistos(const char* pref)
1951 // book special control histos
1952 TString prefS = pref;
1954 TObjArray* dest = new TObjArray;
1955 dest->SetOwner(kTRUE);
1957 const int kNResDef=7;
1958 const double kResDef[kNResDef]={0.05,0.05,0.3, 0.05,1,0.5,1.5};
1959 const double ptMax=10;
1960 const double plMax=10;
1961 const double chiMax=100;
1962 const int nptbins=50;
1963 const int nresbins=400;
1964 const int nplbins=50;
1965 const int nchbins=200;
1966 const int maxBr = 15;
1967 const int maxCand = 200;
1969 for (int phase=0;phase<kNTrackingPhases;phase++) {
1970 for (int pass=0;pass<AliITSUReconstructor::GetRecoParam()->GetNTrackingConditions();pass++) {
1972 for (int ilr=0;ilr<fNLrActive;ilr++) {
1974 // ----------------- These are histos to be filled in Cluster2Tracks of each pass.
1975 // PropagateBack and RefitInward will be stored among the histos of 1st pass
1976 if (pass>0 && phase!=kClus2Tracks) continue;
1978 double mxdf = ilr>=kNResDef ? kResDef[kNResDef-1] : kResDef[ilr];
1979 ttl = Form("Pass%d_S%d_residY%d_%s",pass,phase,ilr,pref);
1980 TH2F* hdy = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nresbins,-mxdf,mxdf);
1981 dest->AddAtAndExpand(hdy,GetHistoID(ilr,kHResY,pass,phase));
1982 hdy->SetDirectory(0);
1984 ttl = Form("Pass%d_S%d_residYPull%d_%s",pass,phase,ilr,pref);
1985 TH2F* hdyp = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nplbins,-plMax,plMax);
1986 dest->AddAtAndExpand(hdyp,GetHistoID(ilr,kHResYP,pass,phase));
1987 hdyp->SetDirectory(0);
1989 ttl = Form("Pass%d_S%d_residZ%d_%s",pass,phase,ilr,pref);
1990 TH2F* hdz = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nresbins,-mxdf,mxdf);
1991 dest->AddAtAndExpand(hdz,GetHistoID(ilr,kHResZ,pass,phase));
1992 hdz->SetDirectory(0);
1994 ttl = Form("Pass%d_S%d_residZPull%d_%s",pass,phase,ilr,pref);
1995 TH2F* hdzp = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nplbins,-plMax,plMax);
1996 hdzp->SetDirectory(0);
1997 dest->AddAtAndExpand(hdzp,GetHistoID(ilr,kHResZP,pass,phase));
1999 ttl = Form("Pass%d_S%d_chi2Cl%d_%s",pass,phase,ilr,pref);
2000 TH2F* hchi = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
2001 hchi->SetDirectory(0);
2002 dest->AddAtAndExpand(hchi,GetHistoID(ilr,kHChi2Cl,pass,phase));
2004 // ------------------- These histos are filled for Clusters2Tracks only
2005 if (phase!=kClus2Tracks) continue;
2007 ttl = Form("Pass%d_S%d_chi2Nrm%d_%s",pass,phase,ilr,pref);
2008 TH2F* hchiN = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
2009 hchiN->SetDirectory(0);
2010 dest->AddAtAndExpand(hchiN,GetHistoID(ilr,kHChi2Nrm,pass,phase));
2012 ttl = Form("Pass%d_S%d_bestInBranch%d_%s",pass,phase,ilr,pref);
2013 TH2* hnbr = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, maxBr,-0.5,maxBr-0.5);
2014 hnbr->SetDirectory(0);
2015 dest->AddAtAndExpand(hnbr,GetHistoID(ilr,kHBestInBranch,pass,phase));
2017 ttl = Form("Pass%d_S%d_bestInCands%d_%s",pass,phase,ilr,pref);
2018 TH2* hncn = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, maxCand,-0.5,maxCand-0.5);
2019 hncn->SetDirectory(0);
2020 dest->AddAtAndExpand(hncn,GetHistoID(ilr,kHBestInCand,pass,phase));
2022 } // loop over layers
2024 // these are special histos, filled not per layer but in the end of track fit in Clusters2Tracks in EVERY pass
2026 if (phase==kClus2Tracks) {
2028 ttl = Form("Pass%d_Chi2Match_%s",pass,pref);
2029 hchiMatch = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
2030 hchiMatch->SetDirectory(0);
2031 dest->AddAtAndExpand(hchiMatch,GetHistoID(-1,kHChiMatch,pass,phase));
2034 ttl = Form("Pass%d_Chi2ITSSA_%s",pass,pref);
2035 hchiSA = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins/2,0.,chiMax/2);
2036 hchiSA->SetDirectory(0);
2037 dest->AddAtAndExpand(hchiSA,GetHistoID(-1,kHChiITSSA,pass,phase));
2040 } // loop over tracking passes
2041 }// loop over tracking phases
2046 //__________________________________________________________________
2047 Int_t AliITSUTrackerGlo::GetHistoID(Int_t lr, Int_t hid, Int_t pass, Int_t phase)
2049 // get id for the requested histo
2052 return pass*kHistosPass + phase*kHistosPhase + lr*kMaxHID + hid;