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>
29 #include "AliITSUTrackerGlo.h"
30 #include "AliESDEvent.h"
31 #include "AliESDtrack.h"
32 #include "AliITSURecoDet.h"
33 #include "AliITSURecoSens.h"
34 #include "AliITSUReconstructor.h"
35 #include "AliITSReconstructor.h"
36 #include "AliITSUSeed.h"
37 #include "AliITSUAux.h"
38 #include "AliITSUClusterPix.h"
39 using namespace AliITSUAux;
40 using namespace TMath;
43 //----------------- tmp stuff -----------------
45 ClassImp(AliITSUTrackerGlo)
46 //_________________________________________________________________________
47 AliITSUTrackerGlo::AliITSUTrackerGlo(AliITSUReconstructor* rec)
54 ,fSeedsPool("AliITSUSeed",0)
57 // Default constructor
61 //_________________________________________________________________________
62 AliITSUTrackerGlo::~AliITSUTrackerGlo()
70 //_________________________________________________________________________
71 void AliITSUTrackerGlo::Init(AliITSUReconstructor* rec)
73 // init with external reconstructor
75 fITS = new AliITSURecoDet(rec->GetGeom(),"ITSURecoInterface");
76 for (int ilr=fITS->GetNLayersActive();ilr--;) {
77 fITS->GetLayerActive(ilr)->SetClusters(rec->GetClusters(ilr));
80 fSeedsPool.ExpandCreateFast(1000); // RS TOCHECK
82 fTrCond.SetNLayers(fITS->GetNLayersActive());
83 fTrCond.AddNewCondition(5);
84 fTrCond.AddGroupPattern( (0x1<<0)|(0x1<<1) );
85 fTrCond.AddGroupPattern( (0x1<<3)|(0x1<<4) );
86 fTrCond.AddGroupPattern( (0x1<<5)|(0x1<<6) );
88 fTrCond.AddNewCondition(5);
89 fTrCond.AddGroupPattern( (0x1<<0)|(0x1<<2) );
90 fTrCond.AddGroupPattern( (0x1<<3)|(0x1<<4) );
91 fTrCond.AddGroupPattern( (0x1<<5)|(0x1<<6) );
93 fTrCond.AddNewCondition(5);
94 fTrCond.AddGroupPattern( (0x1<<1)|(0x1<<2) );
95 fTrCond.AddGroupPattern( (0x1<<3)|(0x1<<4) );
96 fTrCond.AddGroupPattern( (0x1<<5)|(0x1<<6) );
98 printf("Tracking Conditions: ");
102 //_________________________________________________________________________
103 Int_t AliITSUTrackerGlo::Clusters2Tracks(AliESDEvent *esdEv)
107 AliITSUReconstructor::GetRecoParam()->Print();
108 int nTrESD = esdEv->GetNumberOfTracks();
110 if (fHypStore.GetSize()<nTrESD) fHypStore.Expand(nTrESD+100);
112 fITS->ProcessClusters();
113 // select ESD tracks to propagate
114 for (int itr=0;itr<nTrESD;itr++) {
115 AliESDtrack *esdTr = esdEv->GetTrack(itr);
116 AliInfo(Form("Processing track %d | MCLabel: %d",itr,esdTr->GetTPCLabel()));
117 FindTrack(esdTr, itr);
120 printf("Hypotheses for current event (N seeds in pool: %d, size: %d)\n",fSeedsPool.GetEntriesFast(),fSeedsPool.GetSize());
126 //_________________________________________________________________________
127 Int_t AliITSUTrackerGlo::PropagateBack(AliESDEvent * /*event*/)
133 Info("PropagateBack","To be implemented");
137 //_________________________________________________________________________
138 Int_t AliITSUTrackerGlo::RefitInward(AliESDEvent * /*event*/)
144 Info("RefitInward","To be implemented");
148 //_________________________________________________________________________
149 Int_t AliITSUTrackerGlo::LoadClusters(TTree * treeRP)
151 // read from tree (if pointer provided) or directly from the ITS reco interface
153 return fReconstructor->LoadClusters(treeRP);
156 //_________________________________________________________________________
157 void AliITSUTrackerGlo::UnloadClusters()
163 Info("UnloadClusters","To be implemented");
165 //_________________________________________________________________________
166 AliCluster * AliITSUTrackerGlo::GetCluster(Int_t /*index*/) const
172 Info("GetCluster","To be implemented");
176 //_________________________________________________________________________
177 Bool_t AliITSUTrackerGlo::NeedToProlong(AliESDtrack* esdTr)
179 // do we need to match this track to ITS?
181 static double bz = GetBz();
182 if (!esdTr->IsOn(AliESDtrack::kTPCin) ||
183 esdTr->IsOn(AliESDtrack::kTPCout) ||
184 esdTr->IsOn(AliESDtrack::kITSin) ||
185 esdTr->GetKinkIndex(0)>0) return kFALSE;
187 if (esdTr->Pt()<AliITSUReconstructor::GetRecoParam()->GetMinPtForProlongation()) return kFALSE;
190 esdTr->GetDZ(GetX(),GetY(),GetZ(),bz,dtz);
191 // if track is not V0 candidata but has large offset wrt IP, reject it. RS TOCHECK
192 if ( !(esdTr->GetV0Index(0)>0 && dtz[0]>AliITSUReconstructor::GetRecoParam()->GetMaxDforV0dghtrForProlongation())
193 && (Abs(dtz[0])>AliITSUReconstructor::GetRecoParam()->GetMaxDForProlongation() ||
194 Abs(dtz[1])>AliITSUReconstructor::GetRecoParam()->GetMaxDZForProlongation())) return kFALSE;
199 //_________________________________________________________________________
200 void AliITSUTrackerGlo::FindTrack(AliESDtrack* esdTr, Int_t esdID)
202 // find prolongaion candidates finding for single seed
204 if (!NeedToProlong(esdTr)) return; // are we interested in this track?
205 if (!InitSeed(esdTr,esdID)) return; // initialize prolongations hypotheses tree
207 AliITSURecoSens *hitSens[AliITSURecoSens::kNNeighbors+1];
208 AliITSUSeed seedUC; // copy of the seed from the upper layer
209 AliITSUSeed seedT; // transient seed between the seedUC and new prolongation hypothesis
211 TObjArray clArr; // container for transfer of clusters matching to seed
213 for (int ila=fITS->GetNLayersActive();ila--;) {
214 int ilaUp = ila+1; // prolong seeds from layer above
215 int nSeedsUp = fCurrHyp->GetNSeeds(ilaUp);
216 for (int isd=0;isd<nSeedsUp;isd++) {
217 AliITSUSeed* seedU = fCurrHyp->GetSeed(ilaUp,isd); // seed on prev.active layer to prolong
218 seedUC = *seedU; // its copy will be prolonged
219 seedUC.SetParent(seedU);
220 seedUC.ResetFMatrix(); // reset the matrix for propagation to next layer
221 // go till next active layer
222 AliInfo(Form("working on Lr:%d Seed:%d of %d",ila,isd,nSeedsUp));
223 if (!TransportToLayer(&seedUC, fITS->GetLrIDActive(ilaUp), fITS->GetLrIDActive(ila)) ) {
225 AliInfo("Transport failed");
226 // Check if the seed satisfies to track definition
227 if (NeedToKill(&seedUC,kTransportFailed)) seedU->Kill();
228 continue; // RS TODO: decide what to do with tracks stopped on higher layers w/o killing
230 AliITSURecoLayer* lrA = fITS->GetLayerActive(ila);
231 if (!GetRoadWidth(&seedUC, ila)) { // failed to find road width on the layer
232 if (NeedToKill(&seedUC,kRWCheckFailed)) seedU->Kill();
235 int nsens = lrA->FindSensors(&fTrImpData[kTrPhi0], hitSens); // find detectors which may be hit by the track
236 AliInfo(Form("Will check %d sensors on lr:%d ",nsens,ila));
239 for (int isn=nsens;isn--;) {
241 AliITSURecoSens* sens = hitSens[isn];
243 // We need to propagate the seed to sensor on lrA staying the frame of the sensor from prev.layer,
244 // since the transport matrix should be defined in this frame.
245 double xs; // X in the TF of current seed, corresponding to intersection with sensor plane
246 if (!seedT.GetTrackingXAtXAlpha(sens->GetXTF(),sens->GetPhiTF(),bz, xs)) continue;
247 if (!seedT.PropagateToX(xs,bz)) continue;
248 // if (!seedT.Rotate(sens->GetPhiTF())) continue;
249 if (!seedT.RotateToAlpha(sens->GetPhiTF())) continue;
251 int clID0 = sens->GetFirstClusterId();
252 for (int icl=sens->GetNClusters();icl--;) {
253 int res = CheckCluster(&seedT,ila,clID0+icl);
255 if (res==kStopSearchOnSensor) break; // stop looking on this sensor
256 if (res==kClusterNotMatching) continue; // cluster does not match
257 // cluster is matching and it was added to the hypotheses tree
260 // cluster search is done. Do we need ta have a version of this seed skipping current layer
262 if (!NeedToKill(&seedT,kMissingCluster)) {
263 AliITSUSeed* seedSkp = NewSeedFromPool(&seedT);
264 double penalty = -AliITSUReconstructor::GetRecoParam()->GetMissPenalty(ila);
265 // to do: make penalty to account for probability to miss the cluster for good reason
266 seedSkp->SetChi2Cl(penalty);
267 AddProlongationHypothesis(seedSkp,ila);
270 ((TObjArray*)fCurrHyp->GetLayerSeeds(ila))->Sort();
271 printf(">>> All hypotheses on lr %d: \n",ila);
272 for (int ih=0;ih<fCurrHyp->GetNSeeds(ila);ih++) {
273 printf(" #%3d ",ih); fCurrHyp->GetSeed(ila,ih)->Print();
276 if (ila!=0) continue;
277 double vecL[5] = {0};
278 double matL[15] = {0};
279 AliITSUSeed* sp = fCurrHyp->GetSeed(ila,ih);
280 while(sp->GetParent()) {
281 sp->Smooth(vecL,matL);
282 if (sp->GetLayerID()>=fITS->GetNLayersActive()-1) break;
283 sp = (AliITSUSeed*)sp->GetParent();
289 SaveCurrentTrackHypotheses();
293 //_________________________________________________________________________
294 Bool_t AliITSUTrackerGlo::InitSeed(AliESDtrack *esdTr, Int_t esdID)
296 // init prolongaion candidates finding for single seed
297 fCurrHyp = GetTrackHyp(esdID);
298 if (fCurrHyp) return kTRUE;
300 fCurrMass = esdTr->GetMass();
301 fCurrESDtrack = esdTr;
302 if (fCurrMass<kPionMass*0.9) fCurrMass = kPionMass; // don't trust to mu, e identification from TPCin
304 AliITSUSeed* seed = NewSeedFromPool();
305 seed->SetLr(fITS->GetNLayersActive()); // fake layer
306 seed->AliExternalTrackParam::operator=(*esdTr);
307 seed->SetParent(esdTr);
308 int nl = fITS->GetNLayersActive();
309 fCurrHyp = new AliITSUTrackHyp(nl);
310 fCurrHyp->SetESDSeed(seed);
311 AddProlongationHypothesis(seed,nl);
312 fCurrHyp->SetUniqueID(esdID);
313 SetTrackHyp(fCurrHyp,esdID);
318 //_________________________________________________________________________
319 Bool_t AliITSUTrackerGlo::TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo)
321 // transport seed from layerFrom to the entrance of layerTo
323 const double kToler = 1e-6; // tolerance for layer on-surface check
325 int dir = lTo > lFrom ? 1:-1;
326 AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
327 Bool_t checkFirst = kTRUE;
329 double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
331 Bool_t doLayer = kTRUE;
332 double xToGo = dir>0 ? lrFr->GetRMax() : lrFr->GetRMin();
333 if (checkFirst) { // do we need to track till the surface of the current layer ?
335 if (dir>0) { if (curR2-xToGo*xToGo>kToler) doLayer = kFALSE; } // on the surface or outside of the layer
336 else if (dir<0) { if (xToGo*xToGo-curR2>kToler) doLayer = kFALSE; } // on the surface or outside of the layer
339 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
340 // go via layer to its boundary, applying material correction.
341 if (!PropagateSeed(seed,xToGo,fCurrMass, lrFr->GetMaxStep())) return kFALSE;
344 AliITSURecoLayer* lrTo = fITS->GetLayer( (lFrom+=dir) );
345 if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
347 // go the entrance of the layer, assuming no materials in between
348 double xToGo = dir>0 ? lrTo->GetRMin() : lrTo->GetRMax();
349 if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
350 if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) return kFALSE;
357 //_________________________________________________________________________
358 Bool_t AliITSUTrackerGlo::GetRoadWidth(AliITSUSeed* seed, int ilrA)
360 // calculate road width in terms of phi and z for the track which MUST be on the external radius of the layer
361 // as well as some aux info
363 AliITSURecoLayer* lrA = fITS->GetLayerActive(ilrA);
364 seed->GetXYZ(&fTrImpData[kTrXIn]); // lab position at the entrance from above
365 static AliExternalTrackParam sc; // seed copy for manipalitions
368 fTrImpData[kTrPhiIn] = ATan2(fTrImpData[kTrYIn],fTrImpData[kTrXIn]);
369 if (!sc.Rotate(fTrImpData[kTrPhiIn])) return kFALSE; // go to the frame of the entry point into the layer
370 double dr = lrA->GetDR(); // approximate X dist at the inner radius
371 if (!sc.GetXYZAt(sc.GetX()-dr, bz, fTrImpData + kTrXOut)) {
372 // special case: track does not reach inner radius, might be tangential
373 double r = sc.GetD(0,0,bz);
375 if (!sc.GetXatLabR(r,x,bz,-1)) {
377 AliFatal(Form("This should not happen: r=%f",r));
379 dr = Abs(sc.GetX() - x);
380 if (!sc.GetXYZAt(x, bz, fTrImpData + kTrXOut)) {
382 AliFatal(Form("This should not happen: x=%f",x));
386 fTrImpData[kTrPhiOut] = ATan2(fTrImpData[kTrYOut],fTrImpData[kTrXOut]);
387 double sgy = sc.GetSigmaY2() + dr*dr*sc.GetSigmaSnp2() + AliITSUReconstructor::GetRecoParam()->GetSigmaY2(ilrA);
388 double sgz = sc.GetSigmaZ2() + dr*dr*sc.GetSigmaTgl2() + AliITSUReconstructor::GetRecoParam()->GetSigmaZ2(ilrA);
389 sgy = Sqrt(sgy)*AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadY();
390 sgz = Sqrt(sgz)*AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadZ();
391 fTrImpData[kTrPhi0] = 0.5*(fTrImpData[kTrPhiOut]+fTrImpData[kTrPhiIn]);
392 fTrImpData[kTrZ0] = 0.5*(fTrImpData[kTrZOut]+fTrImpData[kTrZIn]);
393 fTrImpData[kTrDPhi] = 0.5*Abs(fTrImpData[kTrPhiOut]-fTrImpData[kTrPhiIn]) + sgy/lrA->GetR();
394 fTrImpData[kTrDZ] = 0.5*Abs(fTrImpData[kTrZOut]-fTrImpData[kTrZIn]) + sgz;
399 //_________________________________________________________________________
400 AliITSUSeed* AliITSUTrackerGlo::NewSeedFromPool(const AliITSUSeed* src)
402 // create new seed, optionally copying from the source
404 new(fSeedsPool[fSeedsPool.GetEntriesFast()]) AliITSUSeed(*src) :
405 new(fSeedsPool[fSeedsPool.GetEntriesFast()]) AliITSUSeed();
408 //_________________________________________________________________________
409 Int_t AliITSUTrackerGlo::CheckCluster(AliITSUSeed* track, Int_t lr, Int_t clID)
411 // Check if the cluster (in tracking frame!) is matching to track.
412 // The track must be already propagated to sensor tracking frame.
413 // Returns: kStopSearchOnSensor if the search on given sensor should be stopped,
414 // kClusterMatching if the cluster is matching
415 // kClusterMatching otherwise
417 const double kTolerX = 5e-4;
418 AliCluster *cl = fITS->GetLayerActive(lr)->GetCluster(clID);
420 Bool_t goodCl = kFALSE;
421 int currLabel = Abs(fCurrESDtrack->GetTPCLabel());
423 if (cl->GetLabel(0)>=0) {for (int i=0;i<3;i++) if (cl->GetLabel(i)>=0 && cl->GetLabel(i)==currLabel) {goodCl = kTRUE; break;}}
426 if (TMath::Abs(cl->GetX())>kTolerX) { // if due to the misalingment X is large, propagate track only
427 if (!track->PropagateParamOnlyTo(track->GetX()+cl->GetX(),GetBz())) {
428 if (goodCl) {printf("Loose good cl: Failed propagation. |"); cl->Print();}
429 return kStopSearchOnSensor; // propagation failed, seedT is intact
432 double dy = cl->GetY()-track->GetY();
433 double dz = cl->GetZ()-track->GetZ();
436 double tol2 = (track->GetSigmaY2() + AliITSUReconstructor::GetRecoParam()->GetSigmaY2(lr))*
437 AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadY()*AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadY(); // RS TOOPTIMIZE
438 if (dy2>tol2) { // the clusters are sorted in Z(col) then in Y(row).
439 if (goodCl) {printf("Loose good cl: dy2=%e > tol2=%e |",dy2,tol2); cl->Print();}
440 if (dy>0) return kStopSearchOnSensor; // No chance that other cluster of this sensor will match (all Y's will be even larger)
441 else return kClusterNotMatching; // Other clusters may match
444 tol2 = (track->GetSigmaZ2() + AliITSUReconstructor::GetRecoParam()->GetSigmaZ2(lr))*
445 AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadZ()*AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadZ(); // RS TOOPTIMIZE
447 if (goodCl) {printf("Loose good cl: dz2=%e > tol2=%e |",dz2,tol2); cl->Print();}
448 return kClusterNotMatching; // Other clusters may match
452 Double_t p[2]={cl->GetY(), cl->GetZ()};
453 Double_t cov[3]={cl->GetSigmaY2(), cl->GetSigmaYZ(), cl->GetSigmaZ2()};
454 double chi2 = track->GetPredictedChi2(p,cov);
455 if (chi2>AliITSUReconstructor::GetRecoParam()->GetMaxTr2ClChi2(lr)) {
457 printf("Loose good cl: Chi2=%e > Chi2Max=%e |dy: %+.3e dz: %+.3e\n",
458 chi2,AliITSUReconstructor::GetRecoParam()->GetMaxTr2ClChi2(lr),dy,dz);
462 return kClusterNotMatching;
465 track = NewSeedFromPool(track); // input track will be reused, use its clone for updates
466 if (!track->Update()) {
467 if (goodCl) {printf("Loose good cl: Failed update |"); cl->Print();}
468 return kClusterNotMatching;
470 track->SetChi2Cl(chi2);
471 track->SetLrClusterID(lr,clID);
472 cl->IncreaseClusterUsage();
474 track->SetFake(!goodCl);
476 AliInfo(Form("AddCl(%d) Cl%d lr:%d: dY:%+8.4f dZ:%+8.4f (MC: %5d %5d %5d) |Chi2=%f(%c)",
477 goodCl,clID,lr,dy,dz2,cl->GetLabel(0),cl->GetLabel(1),cl->GetLabel(2), chi2, track->IsFake() ? '-':'+'));
479 AddProlongationHypothesis(track,lr);
481 return kClusterMatching;
484 //_________________________________________________________________________
485 Bool_t AliITSUTrackerGlo::NeedToKill(AliITSUSeed *seed, Int_t flag)
487 // check if the seed should not be discarded
488 const UShort_t kMask = 0xffff;
489 if (flag==kMissingCluster) {
490 int lastChecked = seed->GetLayerID();
491 UShort_t patt = seed->GetHitsPattern();
492 if (lastChecked) patt |= ~(kMask<<lastChecked); // not all layers were checked, complete unchecked once by potential hits
493 Bool_t seedOK = fTrCond.CheckPattern(patt);
499 //______________________________________________________________________________
500 Bool_t AliITSUTrackerGlo::PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep, Bool_t matCorr)
502 // propagate seed to given x applying material correction if requested
503 const Double_t kEpsilon = 1e-5;
504 Double_t xpos = seed->GetX();
505 Int_t dir = (xpos<xToGo) ? 1:-1;
506 Double_t xyz0[3],xyz1[3],param[7];
508 if (matCorr) seed->GetXYZ(xyz1); //starting global position
509 while ( (xToGo-xpos)*dir > kEpsilon){
510 Double_t step = dir*TMath::Min(TMath::Abs(xToGo-xpos), maxStep);
511 Double_t x = xpos+step;
512 Double_t bz=GetBz(); // getting the local Bz
513 if (!seed->PropagateToX(x,bz)) return kFALSE;
515 xyz0[0]=xyz1[0]; // global pos at the beginning of step
518 seed->GetXYZ(xyz1); // // global pos at the end of step
519 MeanMaterialBudget(xyz0,xyz1,param);
520 Double_t xrho=param[0]*param[4], xx0=param[1];
521 if (dir>0) xrho = -xrho; // outward should be negative
522 if (!seed->ApplyMaterialCorrection(xx0,xrho,mass,kFALSE)) return kFALSE;
529 //______________________________________________________________________________
530 void AliITSUTrackerGlo::SaveCurrentTrackHypotheses()
532 // RS: shall we clean up killed seeds?