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->GetMass(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::kSPECIES];
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);
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[AliITSURecoSens::kNNeighbors+1];
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->GetMass();
684 if (fCurrMass<kPionMass*0.9) fCurrMass = kPionMass; // don't trust to mu, e identification from TPCin
686 hyp = new AliITSUTrackHyp(fNLrActive);
687 hyp->SetESDTrack(esdTr);
688 hyp->SetUniqueID(esdID);
689 hyp->SetMass(fCurrMass);
690 hyp->SetTPCSeed( new AliExternalTrackParam(*esdTr) );
691 SetTrackHyp(hyp,esdID);
692 Bool_t lut = fUseMatLUT; // for propagation from TPC use TGeo material budget always
694 if (!TransportToLayer(hyp->GetTPCSeed(),fITS->GetNLayers(), fITS->GetLrIDActive(fNLrActive-1), fITS->GetRITSTPCRef())) hyp->SetSkip(); // propagate to outer R of ITS
701 //_________________________________________________________________________
702 Bool_t AliITSUTrackerGlo::TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo, Double_t rLim)
704 // transport seed from layerFrom to the entrance (along the direction of the track) of layerTo or to rLim (if>0), wathever is closer
707 if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
709 int dir = lTo > lFrom ? 1:-1;
710 AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
711 Bool_t checkFirst = kTRUE;
712 Bool_t limReached = kFALSE;
715 if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
718 AliITSURecoLayer* lrTo = fITS->GetLayer( (lFrom+=dir) );
719 if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
721 // go the entrance of the layer, assuming no materials in between
722 double xToGo = lrTo->GetR(-dir);
725 if (rLim<xToGo) {xToGo = rLim; limReached = kTRUE;}
728 if (rLim>xToGo) {xToGo = rLim; limReached = kTRUE;}
731 // double xts = xToGo;
732 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
733 //printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
734 //seed->Print("etp");
739 AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
741 if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
742 //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
743 //seed->Print("etp");
747 if (limReached) break;
753 //_________________________________________________________________________
754 Bool_t AliITSUTrackerGlo::TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t rLim)
756 // transport track from layerFrom to the entrance of layerTo or to rLim (if>0), wathever is closer
758 if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
760 int dir = lTo > lFrom ? 1:-1;
761 AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
762 Bool_t checkFirst = kTRUE;
763 Bool_t limReached = kFALSE;
766 if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
769 AliITSURecoLayer* lrTo = fITS->GetLayer( (lFrom+=dir) );
770 if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
772 // go the entrance of the layer, assuming no materials in between
773 double xToGo = lrTo->GetR(-dir);
776 if (rLim<xToGo) {xToGo = rLim; limReached = kTRUE;}
779 if (rLim>xToGo) {xToGo = rLim; limReached = kTRUE;}
782 // double xts = xToGo;
783 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
784 // printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
785 // seed->Print("etp");
789 AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
791 if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
792 //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
793 //seed->Print("etp");
797 if (limReached) break;
803 //_________________________________________________________________________
804 Bool_t AliITSUTrackerGlo::TransportToLayerX(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t xStop)
806 // transport track from layerFrom to the entrance of layerTo but do not pass control parameter X
808 if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
810 int dir = lTo > lFrom ? 1:-1;
811 AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
812 Bool_t checkFirst = kTRUE;
815 if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
818 AliITSURecoLayer* lrTo = fITS->GetLayer( (lFrom+=dir) );
819 if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
821 // go the entrance of the layer, assuming no materials in between
822 double xToGo = lrTo->GetR(-dir); // R of the entrance to layer
824 // double xts = xToGo;
825 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
826 // printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
827 // seed->Print("etp");
830 if ( (dir>0&&xToGo>xStop) || (dir<0&&xToGo<xStop) ) xToGo = xStop;
833 AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
835 if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
836 //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
837 //seed->Print("etp");
846 //_________________________________________________________________________
847 Bool_t AliITSUTrackerGlo::GoToExitFromLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
849 // go to the exit from lr in direction dir, applying material corrections in steps specific for this layer
850 // If check is requested, do this only provided the track has not exited the layer already
851 double xToGo = lr->GetR(dir);
852 if (check) { // do we need to track till the surface of the current layer ?
853 double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
855 AliDebug(3,Form(" dir:%d Cur: %e Tgt: %e",dir,Sqrt(curR2),xToGo));
857 if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // on the surface or outside of the layer
858 else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // on the surface or outside of the layer
860 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
861 // go via layer to its boundary, applying material correction.
862 if (!PropagateSeed(seed,xToGo,fCurrMass, lr->GetMaxStep())) return kFALSE;
868 //_________________________________________________________________________
869 Bool_t AliITSUTrackerGlo::GoToExitFromLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
871 // go to the exit from lr in direction dir, applying material corrections in steps specific for this layer
872 // If check is requested, do this only provided the track has not exited the layer already
873 double xToGo = lr->GetR(dir);
874 if (check) { // do we need to track till the surface of the current layer ?
875 double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
876 if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // on the surface or outside of the layer
877 else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // on the surface or outside of the layer
880 AliDebug(2,Form(" dir=%d : from R=%.4f -> R=%.4f",dir,Sqrt(seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY()), xToGo));
882 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
883 // go via layer to its boundary, applying material correction.
884 if (!PropagateSeed(seed,xToGo,fCurrMass, lr->GetMaxStep())) return kFALSE;
891 //_________________________________________________________________________
892 Bool_t AliITSUTrackerGlo::GoToEntranceToLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
894 // go to the entrance of lr in direction dir, w/o applying material corrections.
895 // If check is requested, do this only provided the track did not reach the layer already
896 double xToGo = lr->GetR(-dir);
897 if (check) { // do we need to track till the surface of the current layer ?
898 double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
899 if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // already passed
900 else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // already passed
902 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
903 // go via layer to its boundary, applying material correction.
904 if (!PropagateSeed(seed,xToGo,fCurrMass, 100, kFALSE)) return kFALSE;
909 //_________________________________________________________________________
910 Bool_t AliITSUTrackerGlo::GoToEntranceToLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check)
912 // go to the entrance of lr in direction dir, w/o applying material corrections.
913 // If check is requested, do this only provided the track did not reach the layer already
914 double xToGo = lr->GetR(-dir);
915 if (check) { // do we need to track till the surface of the current layer ?
916 double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
917 if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // already passed
918 else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // already passed
920 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
921 // go via layer to its boundary, applying material correction.
922 if (!PropagateSeed(seed,xToGo,fCurrMass, 100, kFALSE)) return kFALSE;
927 //_________________________________________________________________________
928 Bool_t AliITSUTrackerGlo::GetRoadWidth(AliITSUSeed* seed, int ilrA)
930 // calculate road width in terms of phi and z for the track which MUST be on the external radius of the layer
931 // as well as some aux info
933 AliITSURecoLayer* lrA = fITS->GetLayerActive(ilrA);
934 seed->GetXYZ(&fTrImpData[kTrXIn]); // lab position at the entrance from above
935 static AliExternalTrackParam sc; // seed copy for manipulations
938 fTrImpData[kTrPhiIn] = ATan2(fTrImpData[kTrYIn],fTrImpData[kTrXIn]);
939 if (!sc.Rotate(fTrImpData[kTrPhiIn])) return kFALSE; // go to the frame of the entry point into the layer
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 double sgy = sc.GetSigmaY2() + dr*dr*sc.GetSigmaSnp2() + AliITSUReconstructor::GetRecoParam()->GetSigmaY2(ilrA);
952 double sgz = sc.GetSigmaZ2() + dr*dr*sc.GetSigmaTgl2() + AliITSUReconstructor::GetRecoParam()->GetSigmaZ2(ilrA);
953 sgy = Sqrt(sgy)*fCurrTrackCond->GetNSigmaRoadY(ilrA);
954 sgz = Sqrt(sgz)*fCurrTrackCond->GetNSigmaRoadZ(ilrA);
955 double dphi0 = 0.5*Abs(fTrImpData[kTrPhiOut]-fTrImpData[kTrPhiIn]);
956 double phi0 = 0.5*(fTrImpData[kTrPhiOut]+fTrImpData[kTrPhiIn]);
957 if ( dphi0>(0.5*Pi()) ) {
958 // special case of angles around pi
963 fTrImpData[kTrPhi0] = phi0;
964 fTrImpData[kTrZ0] = 0.5*(fTrImpData[kTrZOut]+fTrImpData[kTrZIn]);
965 dphi0 += sgy/lrA->GetR();
966 fTrImpData[kTrDPhi] = dphi0<PiOver2() ? dphi0 : PiOver2();
967 fTrImpData[kTrDZ] = 0.5*Abs(fTrImpData[kTrZOut]-fTrImpData[kTrZIn]) + sgz;
972 //________________________________________
973 void AliITSUTrackerGlo::ResetSeedsPool()
975 // mark all seeds in the pool as unused
976 AliDebug(-1,Form("CurrentSize: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
978 fSeedsPool.Clear(); // seeds don't allocate memory
982 //________________________________________
983 void AliITSUTrackerGlo::MarkSeedFree(AliITSUSeed *sd)
985 // account that this seed is "deleted"
986 int id = sd->GetPoolID();
988 AliError(Form("Freeing of seed %p NOT from the pool is requested",sd));
991 // AliInfo(Form("%d %p",id, seed));
992 fSeedsPool.RemoveAt(id);
993 if (fFreeSeedsID.GetSize()<=fNFreeSeeds) fFreeSeedsID.Set( 2*fNFreeSeeds + 100 );
994 fFreeSeedsID.GetArray()[fNFreeSeeds++] = id;
997 //_________________________________________________________________________
998 Int_t AliITSUTrackerGlo::CheckCluster(AliITSUSeed* track, Int_t lr, Int_t clID)
1000 // Check if the cluster (in tracking frame!) is matching to track.
1001 // The track must be already propagated to sensor tracking frame.
1002 // Returns: kStopSearchOnSensor if the search on given sensor should be stopped,
1003 // kClusterMatching if the cluster is matching
1004 // kClusterMatching otherwise
1006 const double kTolerX = 5e-4;
1007 // AliCluster *cl = fITS->GetLayerActive(lr)->GetCluster(clID);
1008 AliCluster *cl = fCurrLayer->GetCluster(clID);
1010 Bool_t goodCl = kFALSE;
1011 int currLabel = Abs(fCurrESDtrMClb);
1013 if (cl->GetLabel(0)>=0) {
1014 for (int i=0;i<3;i++) if (cl->GetLabel(i)>=0 && cl->GetLabel(i)==currLabel) {goodCl = kTRUE; break;}
1017 if (TMath::Abs(cl->GetX())>kTolerX) { // if due to the misalingment X is large, propagate track only
1018 if (!track->PropagateParamOnlyTo(track->GetX()+cl->GetX(),GetBz())) {
1021 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1022 AliDebug(2,Form("Lost good cl on L:%d failed propagation. |ESDtrack#%d (MClb:%d)",lr,fCurrESDtrack->GetID(),fCurrESDtrMClb));
1023 track->Print("etp");
1027 return kStopSearchOnSensor; // propagation failed, seedT is intact
1030 double dy = cl->GetY()-track->GetY();
1031 double dz = cl->GetZ()-track->GetZ();
1034 double tol2 = (track->GetSigmaY2() + AliITSUReconstructor::GetRecoParam()->GetSigmaY2(lr))*
1035 fCurrTrackCond->GetNSigmaRoadY(lr)*fCurrTrackCond->GetNSigmaRoadY(lr); // RS TOOPTIMIZE
1036 if (dy2>tol2) { // the clusters are sorted in Z(col) then in Y(row).
1039 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1040 AliDebug(2,Form("Lost good cl: dy2=%e > tol2=%e |ESDtrack#%d (MClb:%d)",dy2,tol2,fCurrESDtrack->GetID(),fCurrESDtrMClb));
1041 track->Print("etp");
1043 PrintSeedClusters(track);
1046 // clusters are sorted in increasing Y and the loop goes from last (highers Y) to 1st.
1047 // if the track has too large y for this cluster (dy<0) it will be so for all other clusters of the sensor
1048 if (dy<0) return kStopSearchOnSensor; // No chance that other cluster of this sensor will match (all Y's will be even larger)
1049 else return kClusterNotMatching; // Other clusters may match
1052 tol2 = (track->GetSigmaZ2() + AliITSUReconstructor::GetRecoParam()->GetSigmaZ2(lr))*
1053 fCurrTrackCond->GetNSigmaRoadZ(lr)*fCurrTrackCond->GetNSigmaRoadZ(lr); // RS TOOPTIMIZE
1056 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1057 AliDebug(2,Form("Lost good cl on L:%d : dz2=%e > tol2=%e |ESDtrack#%d (MClb:%d)",lr,dz2,tol2,fCurrESDtrack->GetID(),fCurrESDtrMClb));
1058 track->Print("etp");
1060 PrintSeedClusters(track);
1063 return kClusterNotMatching; // Other clusters may match
1067 Double_t p[2]={cl->GetY(), cl->GetZ()};
1068 Double_t cov[3]={cl->GetSigmaY2(), cl->GetSigmaYZ(), cl->GetSigmaZ2()};
1069 double chi2 = track->GetPredictedChi2(p,cov);
1071 if (chi2>fCurrTrackCond->GetMaxTr2ClChi2(lr)) {
1073 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1074 AliDebug(2,Form("Lost good cl on L:%d : Chi2=%e > Chi2Max=%e |dy: %+.3e dz: %+.3e |ESDtrack#%d (MClb:%d)\n",
1075 lr,chi2,fCurrTrackCond->GetMaxTr2ClChi2(lr),dy,dz,fCurrESDtrack->GetID(),fCurrESDtrMClb));
1076 track->Print("etp");
1078 PrintSeedClusters(track);
1081 return kClusterNotMatching;
1084 track = NewSeedFromPool(track); // input track will be reused, use its clone for updates
1085 if (!track->Update()) {
1087 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1088 AliDebug(2,Form("Lost good cluster on L:%d : failed to update",lr));
1089 track->Print("etp");
1091 PrintSeedClusters(track);
1094 MarkSeedFree(track);
1095 return kClusterNotMatching;
1097 track->SetChi2Cl(chi2);
1098 track->SetLrClusterID(lr,clID);
1100 if (track->GetChi2GloNrm()>fCurrTrackCond->GetMaxChi2GloNrm(lr)) {
1102 if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
1103 AliDebug(2,Form("Lost good cl on L:%d : Chi2Glo=%e > Chi2Max=%e |dy: %+.3e dz: %+.3e |ESDtrack#%d (MClb:%d)\n",
1104 lr,track->GetChi2GloNrm(),fCurrTrackCond->GetMaxChi2GloNrm(lr),dy,dz,fCurrESDtrack->GetID(),fCurrESDtrMClb));
1105 track->Print("etp");
1107 PrintSeedClusters(track);
1110 MarkSeedFree(track);
1111 return kClusterNotMatching;
1113 // cl->IncreaseClusterUsage(); // do this only for winners
1115 track->SetFake(!goodCl);
1118 AliDebug(2,Form("AddCl(%d) Cl%d lr:%d: dY:%+8.4f dZ:%+8.4f (MC: %5d %5d %5d) |Chi2=%f(%c)| ",
1119 goodCl,clID,lr,dy,dz2,cl->GetLabel(0),cl->GetLabel(1),cl->GetLabel(2), chi2, track->IsFake() ? '-':'+'));
1122 AddSeedBranch(track);
1124 return kClusterMatching;
1127 //_________________________________________________________________________
1128 Bool_t AliITSUTrackerGlo::NeedToKill(AliITSUSeed *seed, Int_t flag)
1130 // check if the seed should not be discarded
1131 const UShort_t kMask = 0xffff;
1132 if (flag==kMissingCluster) {
1133 int lastChecked = seed->GetLayerID();
1134 UShort_t patt = seed->GetHitsPattern();
1135 if (lastChecked) patt |= (~(kMask<<lastChecked))&fCurrTrackCond->GetAllowedLayers(); // not all layers were checked, complete unchecked ones by potential hits
1136 Bool_t seedOK = fCurrTrackCond->CheckPattern(patt);
1142 //______________________________________________________________________________
1143 Bool_t AliITSUTrackerGlo::PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep, Bool_t matCorr)
1145 // propagate seed to given x applying material correction if requested
1146 const Double_t kEpsilon = 1e-5;
1147 Double_t xpos = seed->GetX();
1148 Int_t dir = (xpos<xToGo) ? 1:-1;
1149 Double_t xyz0[3],xyz1[3];
1151 Bool_t updTime = dir>0 && seed->IsStartedTimeIntegral();
1152 if (matCorr || updTime) seed->GetXYZ(xyz1); //starting global position
1153 while ( (xToGo-xpos)*dir > kEpsilon){
1154 Double_t step = dir*TMath::Min(TMath::Abs(xToGo-xpos), maxStep);
1155 Double_t x = xpos+step;
1156 Double_t bz=GetBz(); // getting the local Bz
1157 if (!seed->PropagateToX(x,bz)) return kFALSE;
1159 if (matCorr || updTime) {
1160 xyz0[0]=xyz1[0]; // global pos at the beginning of step
1163 seed->GetXYZ(xyz1); // // global pos at the end of step
1166 ds = GetMaterialBudget(xyz0,xyz1,xx0,xrho);
1167 if (dir>0) xrho = -xrho; // outward should be negative
1168 if (!seed->ApplyMaterialCorrection(xx0,xrho,mass,kFALSE)) {AliInfo("Fail2"); seed->Print("etp"); return kFALSE;}
1170 else { // matCorr is not requested but time integral is
1171 double d0 = xyz1[0]-xyz0[0];
1172 double d1 = xyz1[1]-xyz0[1];
1173 double d2 = xyz1[2]-xyz0[2];
1174 ds = TMath::Sqrt(d0*d0+d1*d1+d2*d2);
1177 if (updTime) seed->AddTimeStep(ds);
1178 xpos = seed->GetX();
1183 //______________________________________________________________________________
1184 Bool_t AliITSUTrackerGlo::PropagateSeed(AliExternalTrackParam *seed, Double_t xToGo, Double_t mass, Double_t maxStep, Bool_t matCorr)
1186 // propagate seed to given x applying material correction if requested
1187 const Double_t kEpsilon = 1e-5;
1188 Double_t xpos = seed->GetX();
1189 Int_t dir = (xpos<xToGo) ? 1:-1;
1190 Double_t xyz0[3],xyz1[3];
1193 if (AliDebugLevelClass()>1) {
1194 AliDebug(2,Form("Before Propagate to X=%f with M=%.3f MaxStep=%.4f MatCorr=%d",xToGo,mass,maxStep,matCorr));
1195 seed->AliExternalTrackParam::Print();
1198 Bool_t updTime = dir>0 && seed->IsStartedTimeIntegral();
1199 if (matCorr || updTime) seed->GetXYZ(xyz1); //starting global position
1200 while ( (xToGo-xpos)*dir > kEpsilon){
1201 Double_t step = dir*TMath::Min(TMath::Abs(xToGo-xpos), maxStep);
1202 Double_t x = xpos+step;
1203 Double_t bz=GetBz(); // getting the local Bz
1204 if (!seed->PropagateTo(x,bz)) return kFALSE;
1206 if (matCorr || updTime) {
1207 xyz0[0]=xyz1[0]; // global pos at the beginning of step
1210 seed->GetXYZ(xyz1); // // global pos at the end of step
1214 ds = GetMaterialBudget(xyz0,xyz1,xx0,xrho);
1215 if (dir>0) xrho = -xrho; // outward should be negative
1216 if (!seed->CorrectForMeanMaterial(xx0,xrho,mass)) return kFALSE;
1218 else { // matCorr is not requested but time integral is
1219 double d0 = xyz1[0]-xyz0[0];
1220 double d1 = xyz1[1]-xyz0[1];
1221 double d2 = xyz1[2]-xyz0[2];
1222 ds = TMath::Sqrt(d0*d0+d1*d1+d2*d2);
1225 if (updTime) seed->AddTimeStep(ds);
1227 xpos = seed->GetX();
1230 if (AliDebugLevelClass()>1) {
1231 AliDebug(2,Form("After Propagate to X=%f",xToGo));
1232 seed->AliExternalTrackParam::Print();
1238 //______________________________________________________________________________
1239 Bool_t AliITSUTrackerGlo::FinalizeHypothesis(AliITSUTrackHyp* hyp)
1241 // finalize single TPC track hypothesis
1242 if (!hyp || hyp->GetESDTrack()->IsOn(AliESDtrack::kITSin)) return kFALSE;
1243 AliITSUSeed* winner = 0;
1245 fCurrMass = hyp->GetMass();
1246 if (!(winner=hyp->DefineWinner(fCurrTrackCond->GetActiveLrInner()))) return kFALSE;
1249 #ifdef _ITSU_TUNING_MODE_ // fill tuning histos
1250 TObjArray* dest = hyp->GetLabel()>=0 ? fCHistoArrCorr : fCHistoArrFake;
1252 AliITSUSeed* sd = winner;
1253 double htrPt = hyp->Pt();
1256 if ( (clID=sd->GetLrCluster(lrID))<0 ) continue;
1257 ((TH2*)dest->At( GetHistoID(lrID,kHChi2Nrm ,fCurrPassID) ))->Fill(htrPt,sd->GetChi2GloNrm());
1258 ((TH2*)dest->At( GetHistoID(lrID,kHBestInBranch,fCurrPassID) ))->Fill(htrPt,sd->GetOrdBranch());
1259 ((TH2*)dest->At( GetHistoID(lrID,kHBestInCand ,fCurrPassID) ))->Fill(htrPt,sd->GetOrdCand());
1261 if (dest==fCHistoArrFake && !sd->IsFake()) continue; // for the fake seeds fill only fake clusters part
1263 ((TH2*)dest->At( GetHistoID(lrID,kHResY ,fCurrPassID) ))->Fill(htrPt,sd->GetResidY());
1264 ((TH2*)dest->At( GetHistoID(lrID,kHResZ ,fCurrPassID) ))->Fill(htrPt,sd->GetResidZ());
1265 ((TH2*)dest->At( GetHistoID(lrID,kHResYP ,fCurrPassID) ))->Fill(htrPt,sd->GetPullY());
1266 ((TH2*)dest->At( GetHistoID(lrID,kHResZP ,fCurrPassID) ))->Fill(htrPt,sd->GetPullZ());
1267 ((TH2*)dest->At( GetHistoID(lrID,kHChi2Cl ,fCurrPassID) ))->Fill(htrPt,sd->GetChi2Cl());
1269 } while((sd=(AliITSUSeed*)sd->GetParent()));
1271 ((TH2*)dest->At( GetHistoID(-1,kHChiMatch,fCurrPassID) ))->Fill(htrPt,winner->GetChi2ITSTPC());
1272 ((TH2*)dest->At( GetHistoID(-1,kHChiITSSA,fCurrPassID) ))->Fill(htrPt,winner->GetChi2ITSSA());
1277 CheckClusterSharingConflicts(hyp);
1278 return hyp->GetWinner(); // winner might change of disappear after resolving conflicts
1282 //______________________________________________________________________________
1283 void AliITSUTrackerGlo::FinalizeHypotheses()
1285 // select winner for each hypothesis, remove cl. sharing conflicts
1286 AliCodeTimerAuto("",0);
1288 int* esdTrackIndex = fESDIndex.GetArray();
1289 for (int ih=0;ih<fNTracksESD;ih++) FinalizeHypothesis(GetTrackHyp(esdTrackIndex[ih])); // finlize in decreasing pt order
1291 for (int ih=fNTracksESD;ih--;) UpdateESDTrack(GetTrackHyp(esdTrackIndex[ih]),AliESDtrack::kITSin);
1295 //______________________________________________________________________________
1296 void AliITSUTrackerGlo::CheckClusterSharingConflicts(AliITSUTrackHyp* hyp)
1298 // remove eventual cluster sharing conflicts
1299 AliITSUSeed* winner = 0;
1300 if (!(winner=hyp->GetWinner())) return;
1301 UShort_t idH = (UShort_t)hyp->GetUniqueID()+1;
1303 int inLr = fCurrTrackCond->GetActiveLrInner();
1304 AliITSUSeed* winSD=winner;
1306 if ( (clID=winSD->GetLrCluster(lrID))<0 ) continue;
1307 AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1308 Int_t refID = cl->GetRecoInfo(); // was it already referred by some track (with refID-1 if >0)
1309 if (!refID) {cl->SetRecoInfo(idH); continue;} // if not, refer from track to cluster
1310 if (refID==idH) continue; // ignore reference to itself
1311 // the cluster is already used by other track, need to resolve the conflict
1312 AliITSUTrackHyp* hypC = GetTrackHyp(refID-1); // competitor
1313 // printf("Ref to %d (%p) from %d Cl %d %d\n",refID-1,hypC, idH-1,clID,lrID);
1314 AliITSUSeed* winnerC = hypC->GetWinner();
1316 printf("Missing winner, ERROR\n");
1321 if (AliDebugLevelClass()>1) { //------------------------ DEBUG -------------------
1322 AliInfo(Form("Shared cl#%4d on Lr:%d: Hyp%3d/Q:%6.3f vs Hyp%3d/Q:%6.3f",
1323 clID,lrID,idH-1,winner->GetQualityVar(),refID-1,winnerC->GetQualityVar()));
1324 // dump winner of hyp
1325 printf("#%4d Pt:%.3f Chi:%6.3f/%6.3f/%6.3f Ncl:%d MCits%+5d MCtpc:%+5d |",
1326 idH-1,winner->Pt(),winner->GetChi2GloNrm(),winner->GetChi2ITSTPC(),winner->GetChi2ITSSA(),
1327 winner->GetNLayersHit(),hyp->GetITSLabel(),hyp->GetESDTrack()->GetTPCLabel());
1329 AliITSUSeed* sd = winner;
1332 int clIDt = sd->GetLrCluster(lrs);
1333 if (clIDt<0) continue;
1334 while( lrs>++prevL ) printf("%4s ","----");
1335 printf("%4d (%5.1f)",clIDt,sd->GetChi2Cl());
1336 } while ((sd=(AliITSUSeed*)sd->GetParent()));
1338 // dump winner of hypC
1339 printf("#%4d Pt:%.3f Chi:%6.3f/%6.3f/%6.3f Ncl:%d MCits%+5d MCtpc:%+5d |",
1340 refID-1,winnerC->Pt(),winnerC->GetChi2GloNrm(),winnerC->GetChi2ITSTPC(),winnerC->GetChi2ITSSA(),
1341 winnerC->GetNLayersHit(),hypC->GetITSLabel(),hypC->GetESDTrack()->GetTPCLabel());
1346 int clIDt = sd->GetLrCluster(lrs);
1347 if (clIDt<0) continue;
1348 while( lrs>++prevL ) printf("%4s ","----");
1349 printf("%4d (%5.1f)",clIDt,sd->GetChi2Cl());
1350 } while ((sd=(AliITSUSeed*)sd->GetParent()));
1352 } //-------------------------------------------------- DEBUG -------------------
1355 if (winner->GetQualityVar()<winnerC->GetQualityVar()) { // current tracks is better then competitor track
1356 FlagSeedClusters(winnerC,kFALSE,refID); // unflag cluster usage by loser
1357 cl->SetRecoInfo(idH);
1359 //if ( hypC->GetLabel()>=0) AliInfo(Form("KILLING CORRECT: %d",refID-1));
1361 if (hypC->DefineWinner(inLr)) { // find new winner instead of suppressed candidate
1363 CheckClusterSharingConflicts(hypC); // and check its sharing conflicts again
1364 if (winner->IsKilled()) break; // the current winner might have been killed during check of new winner of hypC!
1367 else { // competitor hypC is better than the hyp
1368 FlagSeedClusters(winner,kFALSE,idH); // unflag cluster usage by loser
1370 //if ( hyp->GetLabel()>=0) AliInfo(Form("KILLING CORRECT: %d",idH-1));
1371 if (hyp->DefineWinner(inLr)) {
1373 CheckClusterSharingConflicts(hyp);
1378 } while ((winSD=(AliITSUSeed*)winSD->GetParent()));
1383 //______________________________________________________________________________
1384 void AliITSUTrackerGlo::UpdateESDTrack(AliITSUTrackHyp* hyp,Int_t flag)
1386 // update ESD track with current best hypothesis
1388 AliESDtrack* esdTr = hyp->GetESDTrack();
1389 if (!esdTr || esdTr->IsOn(flag)) return;
1390 AliITSUSeed* win = hyp->GetWinner();
1391 if (!win || win->IsKilled()) return;
1395 case AliESDtrack::kITSin:
1396 esdTr->UpdateTrackParams(hyp,flag); // update kinematics
1398 // set fakes cluster info
1400 UShort_t clfake = 0;
1402 AliITSUSeed* sd = win;
1405 int lr, clID = sd->GetLrCluster(lr);
1406 if (sd->IsFake()) clfake |= 0x1<<lr;
1408 esdTr->SetITSModuleIndex(ip++, sd->GetLrClusterID());
1409 AliITSUClusterPix *cl = (AliITSUClusterPix*)fITS->GetLayerActive(lr)->GetCluster(clID);
1410 #ifdef _ITSU_TUNING_MODE_
1411 if (cl->IsSplit()) clSplit |= 0x1<<lr;
1414 } while ((sd=(AliITSUSeed*)sd->GetParent()));
1416 // RS: Temporary set special flag for tracks from the afterburner
1417 if (fCurrPassID>0) clfake |= 0x1<<7;
1419 esdTr->SetITSSharedMap(clfake);
1420 esdTr->SetITSModuleIndex(10,clSplit);
1422 // TEMPORARY: store iteration id
1423 esdTr->SetITSModuleIndex(11,fCurrPassID);
1426 case AliESDtrack::kITSout:
1427 // here the stored chi2 will correspond to backward ITS-SA fit
1428 esdTr->UpdateTrackParams(hyp,flag); // update kinematics
1430 // TODO: avoid setting friend
1433 case AliESDtrack::kITSrefit:
1435 // at the moment fill as a chi2 the TPC-ITS matching chi2
1436 chiSav = hyp->GetChi2();
1437 hyp->SetChi2(win->GetChi2ITSTPC());
1438 esdTr->UpdateTrackParams(hyp,flag); // update kinematics
1439 hyp->SetChi2(chiSav);
1441 // TODO: avoid setting cluster info
1444 AliFatal(Form("Unknown flag %d",flag));
1447 esdTr->SetITSLabel(hyp->GetITSLabel());
1448 // transfer module indices
1452 //______________________________________________________________________________
1453 Double_t AliITSUTrackerGlo::RefitTrack(AliITSUTrackHyp* trc, Double_t rDest, Int_t &nclFit, Int_t stopCond)
1455 // refit track till radius rDest. The cluster,mass info is taken from fCurrHyp (and its winner seed)
1456 // if stopCond<0 : propagate till last cluster then stop
1457 // if stopCond==0: propagate till last cluster then try to go till limiting rDest, don't mind if fail
1458 // if stopCond>0 : rDest must be reached
1460 double rCurr = Sqrt(trc->GetX()*trc->GetX() + trc->GetY()*trc->GetY());
1461 int dir,lrStart,lrStop;
1463 dir = rCurr<rDest ? 1 : -1;
1464 lrStart = fITS->FindFirstLayerID(rCurr,dir);
1465 lrStop = fITS->FindLastLayerID(rDest,dir); // lr id before which we have to stop
1468 if (AliDebugLevelClass()>2) {
1469 printf("Refit %d: Lr:%d (%f) -> Lr:%d (%f)\n",dir,lrStart,rCurr, lrStop,rDest);
1470 printf("Before refit: "); trc->Print();
1473 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));
1475 Int_t clInfo[2*AliITSUAux::kMaxLayers];
1476 Int_t nCl = fCurrHyp->FetchClusterInfo(clInfo);
1477 fCurrMass = fCurrHyp->GetMass();
1478 AliITSUTrackHyp tmpTr(*(AliKalmanTrack*)trc);
1483 int lrStop1 = lrStop+dir;
1484 for (int ilr=lrStart;ilr!=lrStop1;ilr+=dir) {
1485 AliITSURecoLayer* lr = fITS->GetLayer(ilr);
1486 if ( dir*(rCurr-lr->GetR(dir))>0) continue; // this layer is already passed
1487 int ilrA2,ilrA = lr->GetActiveID();
1488 // passive layer or active w/o hits will be traversed on the way to next cluster
1489 if (!lr->IsActive() || clInfo[ilrA2=(ilrA<<1)]<0) continue;
1491 // select the order in which possible 2 clusters (in case of the overlap) will be traversed and fitted
1493 if (dir>0) { // clusters are stored in increasing radius order
1494 iclLr[nclLr++]=clInfo[ilrA2++];
1495 if (clInfo[ilrA2]>=0) iclLr[nclLr++]=clInfo[ilrA2];
1498 if ( clInfo[ilrA2+1]>=0 ) iclLr[nclLr++]=clInfo[ilrA2+1];
1499 iclLr[nclLr++]=clInfo[ilrA2];
1502 Bool_t transportedToLayer = kFALSE;
1503 for (int icl=0;icl<nclLr;icl++) {
1504 AliITSUClusterPix* clus = (AliITSUClusterPix*)lr->GetCluster(iclLr[icl]);
1505 AliITSURecoSens* sens = lr->GetSensorFromID(clus->GetVolumeId());
1506 if (!tmpTr.Rotate(sens->GetPhiTF())) {
1508 AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1509 AliDebug(2,Form("Failed on rotate to %f | ESDtrack#%d (MClb:%d)",sens->GetPhiTF(),trESD->GetID(),trESD->GetTPCLabel()));
1514 double xClus = sens->GetXTF()+clus->GetX();
1515 if (!transportedToLayer) {
1516 if (ilr!=lrStart && !TransportToLayerX(&tmpTr,lrStart,ilr,xClus)) {
1518 AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1519 AliDebug(2,Form("Failed to transport %d -> %d | ESDtrack#%d (MClb:%d)\n",lrStart,ilr,trESD->GetID(),trESD->GetTPCLabel()));
1521 return -1; // go to the entrance to the layer
1524 transportedToLayer = kTRUE;
1528 if (AliDebugLevelClass()>1) {
1529 AliDebug(2,Form("Propagate to cl:%d on lr %d Need to go %.4f -> %.4f",icl,ilrA,tmpTr.GetX(),xClus));
1533 if (!PropagateSeed(&tmpTr,xClus,fCurrMass)) {
1535 AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1536 AliDebug(2,Form("Failed on propagate to %f (dir=%d) | ESDtrack#%d (MClb:%d)",xClus,dir,trESD->GetID(),trESD->GetTPCLabel()));
1541 Double_t p[2]={clus->GetY(), clus->GetZ()};
1542 Double_t cov[3]={clus->GetSigmaY2(), clus->GetSigmaYZ(), clus->GetSigmaZ2()};
1543 double chi2cl = tmpTr.GetPredictedChi2(p,cov);
1546 #ifdef _ITSU_TUNING_MODE_
1547 TObjArray* dest = trc->GetLabel()>=0 ? fCHistoArrCorr : fCHistoArrFake;
1548 if (dest && fTrackPhaseID>kClus2Tracks) {
1550 double htrPt = tmpTr.Pt();
1551 double dy = p[0]-tmpTr.GetY();
1552 double dz = p[1]-tmpTr.GetZ();
1553 ((TH2*)dest->At( GetHistoID(ilrA,kHResY,0,fTrackPhaseID) ))->Fill(htrPt,dy);
1554 ((TH2*)dest->At( GetHistoID(ilrA,kHResZ,0,fTrackPhaseID) ))->Fill(htrPt,dz);
1555 double errY = tmpTr.GetSigmaY2();
1556 double errZ = tmpTr.GetSigmaZ2();
1557 if (errY>0) ((TH2*)dest->At( GetHistoID(ilrA,kHResYP,0,fTrackPhaseID) ))->Fill(htrPt,dy/Sqrt(errY));
1558 if (errZ>0) ((TH2*)dest->At( GetHistoID(ilrA,kHResZP,0,fTrackPhaseID) ))->Fill(htrPt,dz/Sqrt(errZ));
1559 ((TH2*)dest->At( GetHistoID(ilrA,kHChi2Cl,0,fTrackPhaseID) ))->Fill(htrPt,chi2cl);
1563 if ( !tmpTr.Update(p,cov) ) {
1565 AliESDtrack* trESD = fCurrHyp->GetESDTrack();
1566 AliDebug(2,Form("Failed on Update | ESDtrack#%d (MClb:%d)",trESD->GetID(),trESD->GetTPCLabel()));
1571 if (AliDebugLevelClass()>1) {
1572 printf("AfterRefit: "); tmpTr.Print();
1575 if (++nclFit==nCl && stopCond<0) {
1576 *trc = (AliKalmanTrack&)tmpTr;
1577 return chi2; // it was requested to not propagate after last update
1582 // All clusters were succesfully fitted. Even if the track does not reach rDest, this is enough to validate it.
1583 // Still, try to go as close as possible to rDest.
1585 if (lrStart!=lrStop) {
1586 if (!TransportToLayer(&tmpTr,lrStart,lrStop)) {
1588 AliDebug(1,Form("Failed on TransportToLayer %d->%d | ESDtrack#%d (MClb:%d), X=%f",lrStart,lrStop,fCurrESDtrack->GetID(),fCurrESDtrMClb,tmpTr.GetX()));
1590 return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
1592 if (!GoToExitFromLayer(&tmpTr,fITS->GetLayer(lrStop),dir)) {
1594 AliDebug(1,Form("Failed on GoToExitFromLayer %d | ESDtrack#%d (MClb:%d), X=%f",lrStop,fCurrESDtrack->GetID(),fCurrESDtrMClb,tmpTr.GetX()));
1596 return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
1599 // go to the destination radius. Note that here we don't select direction to avoid precision problems
1600 if (!tmpTr.GetXatLabR(rDest,rDest,GetBz(),0) || !PropagateSeed(&tmpTr,rDest,fCurrMass, 100, kFALSE)) {
1601 return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
1603 *trc = (AliKalmanTrack&)tmpTr;
1605 if (AliDebugLevelClass()>2) {
1606 printf("After refit (now at lr %d): ",lrStop); trc->Print();
1612 //__________________________________________________________________
1613 void AliITSUTrackerGlo::CookMCLabel(AliITSUTrackHyp* hyp)
1617 const int kMaxLbPerCl = 3;
1618 int lbID[kMaxLayers*6],lbStat[kMaxLayers*6];
1619 Int_t lr,nLab=0,nCl=0;
1620 AliITSUSeed *seed = hyp->GetWinner();
1622 int clID = seed->GetLrCluster(lr);
1624 AliCluster *cl = fITS->GetLayerActive(lr)->GetCluster(clID);
1626 for (int imc=0;imc<kMaxLbPerCl;imc++) { // labels within single cluster
1627 int trLb = cl->GetLabel(imc);
1629 // search this mc track in already accounted ones
1631 for (iLab=0;iLab<nLab;iLab++) if (lbID[iLab]==trLb) break;
1632 if (iLab<nLab) lbStat[iLab]++;
1637 } // loop over given cluster's labels
1639 seed = (AliITSUSeed*)seed->GetParent();
1640 } // loop over clusters
1642 AliESDtrack* esdTr = hyp->GetESDTrack();
1643 int tpcLab = esdTr ? Abs(esdTr->GetTPCLabel()) : -kDummyLabel;
1645 int maxLab=0,nTPCok=0;
1646 for (int ilb=nLab;ilb--;) {
1647 int st = lbStat[ilb];
1648 if (lbStat[maxLab]<st) maxLab=ilb;
1649 if (tpcLab==lbID[ilb]) nTPCok += st;
1651 hyp->SetFakeRatio(1.-float(nTPCok)/nCl);
1652 hyp->SetLabel( nTPCok==nCl ? tpcLab : -tpcLab);
1653 hyp->SetITSLabel( lbStat[maxLab]==nCl ? lbID[maxLab] : -lbID[maxLab]); // winner label
1657 hyp->SetFakeRatio(-1.);
1658 hyp->SetLabel( -tpcLab );
1659 hyp->SetITSLabel( kDummyLabel );
1662 //__________________________________________________________________
1663 Bool_t AliITSUTrackerGlo::AddSeedBranch(AliITSUSeed* seed)
1665 // add new prolongation branch for the seed to temporary array. The seeds are sorted according to Compare f-n
1666 if (fNCandidatesAdded+fNBranchesAdded>=fLayerMaxCandidates ) {
1667 AliInfo(Form("Number of candidates at layer %d for current seed reached %d, increasing buffer",fCurrActLrID,fLayerMaxCandidates));
1668 fLayerMaxCandidates = 2*(fLayerMaxCandidates+1);
1669 AliITSUSeed** tmpArr = fLayerCandidates;
1670 fLayerCandidates = new AliITSUSeed*[fLayerMaxCandidates];
1671 memcpy(fLayerCandidates,tmpArr,(fNCandidatesAdded+fNBranchesAdded)*sizeof(AliITSUSeed*));
1672 delete tmpArr; // delete only array, not objects
1674 AliITSUSeed** branches = &fLayerCandidates[fNCandidatesAdded]; // note: fNCandidatesAdded is incremented after adding all branches of current seed
1675 int slot=fNBranchesAdded++;
1676 for (int slotF=slot;slotF--;) { // slotF is always slot-1
1677 AliITSUSeed* si = branches[slotF];
1678 if (si->Compare(seed)<0) break; // found the last seed with better quality
1679 // otherwise, shift the worse seed to the next slot
1680 branches[slot] = si;
1681 slot = slotF; // slot should be slotF+1
1683 // if needed, move worse seeds
1684 branches[slot] = seed;
1689 //__________________________________________________________________
1690 void AliITSUTrackerGlo::ValidateAllowedBranches(Int_t acceptMax)
1692 // keep allowed number of branches for current seed and disable extras
1693 AliCodeTimerAuto("",0);
1694 int nb = Min(fNBranchesAdded,acceptMax);
1695 // if (nb<fNBranchesAdded) printf("ValidateAllowedBranches: %d of %d (%d) on lr %d\n",nb,fNBranchesAdded,fNCandidatesAdded,ilr);
1696 // disable unused branches
1697 AliITSUSeed** branches = &fLayerCandidates[fNCandidatesAdded];
1699 #ifdef _ITSU_TUNING_MODE_
1700 for (int ib=0;ib<fNBranchesAdded;ib++) branches[ib]->SetOrdBranch(ib);
1703 for (int ib=nb;ib<fNBranchesAdded;ib++) MarkSeedFree(branches[ib]);
1704 fNCandidatesAdded += nb; // update total candidates counter
1705 fNBranchesAdded = 0; // reset branches counter
1709 //__________________________________________________________________
1710 void AliITSUTrackerGlo::ValidateAllowedCandidates(Int_t ilr, Int_t acceptMax)
1712 // transfer allowed number of branches to hypothesis container
1714 AliCodeTimerAuto("",0);
1715 // sort candidates in increasing order of chi2
1716 static int lastSize = 0;
1717 static int *index = 0;
1718 static float *chi2 = 0;
1719 if (fLayerMaxCandidates>lastSize) {
1720 lastSize = fLayerMaxCandidates;
1723 index = new int[lastSize];
1724 chi2 = new float[lastSize];
1726 for (int i=fNCandidatesAdded;i--;) chi2[i] = fLayerCandidates[i]->GetChi2GloNrm();
1727 Sort(fNCandidatesAdded,chi2,index,kFALSE);
1730 if (ilr>fCurrTrackCond->GetActiveLrInner()) { // just take 1st acceptMax candidates
1731 nb = Min(fNCandidatesAdded,acceptMax);
1732 for (int ib=0;ib<nb;ib++) AddProlongationHypothesis(fLayerCandidates[index[ib]],ilr);
1734 else { // for innermost layer test ITS SA backward chi2 and matching to TPC
1735 AliITSUSeed* wn0 = fCurrHyp->GetWinner(); // in principle, should be NULL
1736 for (nb=0;nb<fNCandidatesAdded;nb++) {
1737 AliITSUSeed* sd = fLayerCandidates[index[nb]];
1738 fCurrHyp->SetWinner(sd);
1739 if (!CheckBackwardMatching(sd)) MarkSeedFree(sd); // discard bad backward tracks
1741 AddProlongationHypothesis(sd,ilr);
1742 if (++nacc==acceptMax) {nb++; break;} // the rest will be discarded
1745 fCurrHyp->SetWinner(wn0); // restore original winner (NULL?)
1747 // disable unused candidates
1748 for (int ib=nb;ib<fNCandidatesAdded;ib++) MarkSeedFree(fLayerCandidates[index[ib]]);
1749 fNCandidatesAdded = 0; // reset candidates counter
1753 //__________________________________________________________________
1754 Bool_t AliITSUTrackerGlo::CheckBackwardMatching(AliITSUSeed* seed)
1756 // check seed backward propagation chi2 and matching to TPC
1757 double bz0 = GetBz();
1758 double rDest = fITS->GetRITSTPCRef(); // reference radius for comparison
1759 AliITSUTrackHyp trback;
1760 trback.AliExternalTrackParam::operator=(*seed);
1761 trback.ResetCovariance(10000);
1763 double chi2sa = RefitTrack(&trback,rDest,nclFit,1);
1764 if (chi2sa<0) return kFALSE;
1765 int ndf = nclFit*2-5;
1766 if (ndf>0) chi2sa /= ndf;
1767 if (chi2sa>fCurrTrackCond->GetMaxITSSAChi2(nclFit)) return kFALSE;
1769 // relate to TPC track at outer layer
1770 AliExternalTrackParam* tpcSeed = fCurrHyp->GetTPCSeed();
1771 if (!trback.Rotate(tpcSeed->GetAlpha()) || !trback.PropagateParamOnlyTo(tpcSeed->GetX(),bz0)) {
1773 if (AliDebugLevelClass()>+1 && !seed->ContainsFake()) {
1774 AliInfo(Form("Failed to propagate seed to TPC track @ X:%.3f Alpha:%.3f",tpcSeed->GetX(),tpcSeed->GetAlpha()));
1781 double chi2Match = trback.GetPredictedChi2(tpcSeed)/5;
1782 if (chi2Match>fCurrTrackCond->GetMaxITSTPCMatchChi2()) return kFALSE;
1784 seed->SetChi2ITSSA(chi2sa);
1785 seed->SetChi2ITSTPC(chi2Match);
1789 //__________________________________________________________________
1790 void AliITSUTrackerGlo::SaveReducedHypothesesTree(AliITSUTrackHyp* dest)
1792 // remove those hypothesis seeds which dont lead to candidates at final layer
1794 // 1st, flag the seeds to save
1795 int lr0 = fCurrTrackCond->GetActiveLrInner();
1796 for (int isd=0;isd<fCurrHyp->GetNSeeds(lr0);isd++) {
1797 AliITSUSeed* seed = fCurrHyp->RemoveSeed(lr0,isd);
1798 if (!seed) continue;
1799 seed->FlagTree(AliITSUSeed::kSave);
1800 dest->AddSeed(seed,lr0);
1802 for (int ilr=0;ilr<fNLrActive;ilr++) {
1803 int nsd = fCurrHyp->GetNSeeds(ilr);
1804 for (int isd=0;isd<nsd;isd++) {
1805 AliITSUSeed* seed = fCurrHyp->RemoveSeed(ilr,isd);
1806 if (!seed) continue; // already discarded or saved
1807 if (seed->IsSaved()) dest->AddSeed(seed,ilr);
1808 else MarkSeedFree(seed);
1812 // AliInfo(Form("SeedsPool: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
1815 //__________________________________________________________________
1816 void AliITSUTrackerGlo::CleanHypothesis(AliITSUTrackHyp* hyp)
1820 for (int ilr=0;ilr<fNLrActive;ilr++) {
1821 int nsd = hyp->GetNSeeds(ilr);
1822 for (int isd=0;isd<nsd;isd++) {
1823 AliITSUSeed* seed = hyp->RemoveSeed(ilr,isd);
1824 if (!seed) continue; // already discarded or saved
1830 //__________________________________________________________________
1831 void AliITSUTrackerGlo::FlagSplitClusters()
1833 // set special bit on split clusters using MC info
1834 for (int ilr=fNLrActive;ilr--;) {
1836 AliITSURecoLayer* lr = fITS->GetLayerActive(ilr);
1837 for (int isn=lr->GetNSensors();isn--;) {
1838 AliITSURecoSens* sens = lr->GetSensor(isn);
1839 int nCl = sens->GetNClusters();
1841 int cl0 = sens->GetFirstClusterId();
1842 for (int icl=nCl;icl--;) {
1843 AliITSUClusterPix *cl = (AliITSUClusterPix*)lr->GetCluster(cl0+icl);
1844 for (int icl1=icl;icl1--;) {
1845 AliITSUClusterPix *cl1 = (AliITSUClusterPix*)lr->GetCluster(cl0+icl1);
1846 if (cl->HasCommonTrack(cl1)) {
1847 if (!cl->IsSplit()) nsplit++;
1848 if (!cl1->IsSplit()) nsplit++;
1855 AliInfo(Form("%4d out of %4d clusters are split",nsplit,lr->GetNClusters()));
1860 //__________________________________________________________________
1861 Bool_t AliITSUTrackerGlo::ContainsSplitCluster(const AliITSUSeed* seed, Int_t maxSize)
1863 // check if the seed contains split cluster with size < maxSize
1865 if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
1866 AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1867 if (cl->IsSplit() && cl->GetNPix()<maxSize ) return kTRUE;
1869 return seed->GetParent() ? ContainsSplitCluster((AliITSUSeed*)seed->GetParent(),maxSize) : kFALSE;
1873 //__________________________________________________________________
1874 void AliITSUTrackerGlo::PrintSeedClusters(const AliITSUSeed* seed, Option_t* option)
1876 // print seeds clusters
1878 if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
1879 AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1882 if (seed->GetParent()) PrintSeedClusters((AliITSUSeed*)seed->GetParent(), option);
1886 //__________________________________________________________________
1887 void AliITSUTrackerGlo::FlagSeedClusters(const AliITSUSeed* seed, Bool_t flg, UShort_t ref)
1889 // mark used clusters
1892 if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
1893 AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
1894 if (ref) { // do we need to set or delete cluster->track ref?
1896 if (!cl->GetRecoInfo()) cl->SetRecoInfo(ref); // set ref only if cluster already does not refer to other track, inc.counter
1899 if (cl->GetRecoInfo()==ref) cl->SetRecoInfo(0); // unset reference only if it refers to ref, decrease counter
1903 seed = (AliITSUSeed*)seed->GetParent();
1908 //__________________________________________________________________
1909 void AliITSUTrackerGlo::CheckClusterUsage()
1911 // check cluster usage info
1912 printf("ClusterUsage at pass %d\n",fCurrPassID);
1913 for (int ilr=0;ilr<fNLrActive;ilr++) {
1914 AliITSURecoLayer* lr = fITS->GetLayerActive(ilr);
1915 int ncl = lr->GetNClusters();
1917 int nclUs0CntShr1 = 0;
1921 for (int icl=0;icl<ncl;icl++) {
1922 AliITSUClusterPix *cl = (AliITSUClusterPix*)lr->GetCluster(icl);
1923 int usc = cl->GetClUsage();
1925 if (cl->IsClusterUsed() || cl->IsClusterShared()) {
1927 printf("Lr%d Cl%4d: UseCnt=0 but UseFlg=%d UseShr=%d\n",ilr,icl,cl->IsClusterUsed(),cl->IsClusterShared());
1932 if (!cl->IsClusterUsed()) {
1934 printf("Lr%d Cl%4d: UseCnt=%d but UseFlg=%d UseShr=%d\n",ilr,icl,usc, cl->IsClusterUsed(),cl->IsClusterShared());
1936 if (cl->IsClusterShared()) {
1938 printf("Lr%d Cl%4d: UseCnt=%d but UseFlg=%d UseShr=%d\n",ilr,icl,usc, cl->IsClusterUsed(),cl->IsClusterShared());
1942 printf("Lr%d Cl%4d: UseCnt=%d!! but UseFlg=%d UseShr=%d\n",ilr,icl,usc, cl->IsClusterUsed(),cl->IsClusterShared());
1945 printf("ClUsage Lr%d: Ncl=%5d Nusd=%5d (%5.2f) Use0CS1:%4d Use1C0:%4d Use1S1:%4d UseL:%d\n",
1946 ilr,ncl,nclUsed,ncl? float(nclUsed)/ncl : 0, nclUs0CntShr1,nclUs1Cnt0,nclUs1Shr1,nclUseL);
1951 #ifdef _ITSU_TUNING_MODE_
1952 //__________________________________________________________________
1953 TObjArray* AliITSUTrackerGlo::BookControlHistos(const char* pref)
1955 // book special control histos
1956 TString prefS = pref;
1958 TObjArray* dest = new TObjArray;
1959 dest->SetOwner(kTRUE);
1961 const int kNResDef=7;
1962 const double kResDef[kNResDef]={0.05,0.05,0.3, 0.05,1,0.5,1.5};
1963 const double ptMax=10;
1964 const double plMax=10;
1965 const double chiMax=100;
1966 const int nptbins=50;
1967 const int nresbins=400;
1968 const int nplbins=50;
1969 const int nchbins=200;
1970 const int maxBr = 15;
1971 const int maxCand = 200;
1973 for (int phase=0;phase<kNTrackingPhases;phase++) {
1974 for (int pass=0;pass<AliITSUReconstructor::GetRecoParam()->GetNTrackingConditions();pass++) {
1976 for (int ilr=0;ilr<fNLrActive;ilr++) {
1978 // ----------------- These are histos to be filled in Cluster2Tracks of each pass.
1979 // PropagateBack and RefitInward will be stored among the histos of 1st pass
1980 if (pass>0 && phase!=kClus2Tracks) continue;
1982 double mxdf = ilr>=kNResDef ? kResDef[kNResDef-1] : kResDef[ilr];
1983 ttl = Form("Pass%d_S%d_residY%d_%s",pass,phase,ilr,pref);
1984 TH2F* hdy = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nresbins,-mxdf,mxdf);
1985 dest->AddAtAndExpand(hdy,GetHistoID(ilr,kHResY,pass,phase));
1986 hdy->SetDirectory(0);
1988 ttl = Form("Pass%d_S%d_residYPull%d_%s",pass,phase,ilr,pref);
1989 TH2F* hdyp = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nplbins,-plMax,plMax);
1990 dest->AddAtAndExpand(hdyp,GetHistoID(ilr,kHResYP,pass,phase));
1991 hdyp->SetDirectory(0);
1993 ttl = Form("Pass%d_S%d_residZ%d_%s",pass,phase,ilr,pref);
1994 TH2F* hdz = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nresbins,-mxdf,mxdf);
1995 dest->AddAtAndExpand(hdz,GetHistoID(ilr,kHResZ,pass,phase));
1996 hdz->SetDirectory(0);
1998 ttl = Form("Pass%d_S%d_residZPull%d_%s",pass,phase,ilr,pref);
1999 TH2F* hdzp = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nplbins,-plMax,plMax);
2000 hdzp->SetDirectory(0);
2001 dest->AddAtAndExpand(hdzp,GetHistoID(ilr,kHResZP,pass,phase));
2003 ttl = Form("Pass%d_S%d_chi2Cl%d_%s",pass,phase,ilr,pref);
2004 TH2F* hchi = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
2005 hchi->SetDirectory(0);
2006 dest->AddAtAndExpand(hchi,GetHistoID(ilr,kHChi2Cl,pass,phase));
2008 // ------------------- These histos are filled for Clusters2Tracks only
2009 if (phase!=kClus2Tracks) continue;
2011 ttl = Form("Pass%d_S%d_chi2Nrm%d_%s",pass,phase,ilr,pref);
2012 TH2F* hchiN = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
2013 hchiN->SetDirectory(0);
2014 dest->AddAtAndExpand(hchiN,GetHistoID(ilr,kHChi2Nrm,pass,phase));
2016 ttl = Form("Pass%d_S%d_bestInBranch%d_%s",pass,phase,ilr,pref);
2017 TH2* hnbr = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, maxBr,-0.5,maxBr-0.5);
2018 hnbr->SetDirectory(0);
2019 dest->AddAtAndExpand(hnbr,GetHistoID(ilr,kHBestInBranch,pass,phase));
2021 ttl = Form("Pass%d_S%d_bestInCands%d_%s",pass,phase,ilr,pref);
2022 TH2* hncn = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, maxCand,-0.5,maxCand-0.5);
2023 hncn->SetDirectory(0);
2024 dest->AddAtAndExpand(hncn,GetHistoID(ilr,kHBestInCand,pass,phase));
2026 } // loop over layers
2028 // these are special histos, filled not per layer but in the end of track fit in Clusters2Tracks in EVERY pass
2030 if (phase==kClus2Tracks) {
2032 ttl = Form("Pass%d_Chi2Match_%s",pass,pref);
2033 hchiMatch = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
2034 hchiMatch->SetDirectory(0);
2035 dest->AddAtAndExpand(hchiMatch,GetHistoID(-1,kHChiMatch,pass,phase));
2038 ttl = Form("Pass%d_Chi2ITSSA_%s",pass,pref);
2039 hchiSA = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins/2,0.,chiMax/2);
2040 hchiSA->SetDirectory(0);
2041 dest->AddAtAndExpand(hchiSA,GetHistoID(-1,kHChiITSSA,pass,phase));
2044 } // loop over tracking passes
2045 }// loop over tracking phases
2050 //__________________________________________________________________
2051 Int_t AliITSUTrackerGlo::GetHistoID(Int_t lr, Int_t hid, Int_t pass, Int_t phase)
2053 // get id for the requested histo
2056 return pass*kHistosPass + phase*kHistosPhase + lr*kMaxHID + hid;