#include <TTree.h>
#include <Riostream.h>
#include <TMath.h>
+#include <TFile.h>
#include "AliITSUTrackerGlo.h"
#include "AliESDEvent.h"
#include "AliITSUReconstructor.h"
#include "AliITSReconstructor.h"
#include "AliITSUSeed.h"
-#include "AliITSUAux.h"
#include "AliITSUClusterPix.h"
+#include "AliITSUGeomTGeo.h"
+#include "AliCodeTimer.h"
+#include "AliRefArray.h"
using namespace AliITSUAux;
using namespace TMath;
-
-
//----------------- tmp stuff -----------------
ClassImp(AliITSUTrackerGlo)
: fReconstructor(rec)
,fITS(0)
,fCurrESDtrack(0)
+ ,fCurrESDtrMClb(kDummyLabel)
,fCurrMass(kPionMass)
+ ,fCountProlongationTrials(0)
+ ,fCountITSin(0)
+ ,fCountITSout(0)
+ ,fCountITSrefit(0)
,fHypStore(100)
+ ,fLayerMaxCandidates(1000)
+ ,fLayerCandidates(0)
+ ,fNBranchesAdded(0)
+ ,fNCandidatesAdded(0)
,fCurrHyp(0)
+ ,fWorkHyp(0)
,fSeedsPool("AliITSUSeed",0)
,fFreeSeedsID(0)
+ ,fESDIndex(0)
,fNFreeSeeds(0)
,fLastSeedID(0)
+ ,fNLrActive(0)
,fDefTrackConds(0)
,fCurrTrackCond(0)
+ ,fCurrActLrID(-1)
+ ,fCurrLayer(0)
,fTrackPhase(-1)
- ,fClInfo(0)
+#ifdef _ITSU_TUNING_MODE_
+ ,fCHistoArrCorr(0)
+ ,fCHistoArrFake(0)
+#endif
{
// Default constructor
if (rec) Init(rec);
{
// Default destructor
//
- delete[] fClInfo;
+ delete[] fLayerCandidates;
+ if (fWorkHyp) fWorkHyp->SetTPCSeed(0); // this hypothesis does not own the seed
+ delete fWorkHyp;
+ //
+#ifdef _ITSU_TUNING_MODE_
+ if (fCHistoArrCorr || fCHistoArrFake) {
+ TFile* ctrOut = TFile::Open("itsuControlHistos.root","recreate");
+ ctrOut->cd();
+ AliInfo("Storing control histos");
+ // ctrOut->WriteObject(fCHistoArr,"controlH","kSingleKey");
+ if (fCHistoArrCorr) {
+ fCHistoArrCorr->Write();
+ delete fCHistoArrCorr;
+ }
+ if (fCHistoArrFake) {
+ fCHistoArrFake->Write();
+ delete fCHistoArrFake;
+ }
+ ctrOut->Close();
+ delete ctrOut;
+ fCHistoArrCorr = 0;
+ fCHistoArrFake = 0;
+ }
+#endif
//
}
// init with external reconstructor
//
fITS = rec->GetITSInterface();
- int nLr = fITS->GetNLayersActive();
- fClInfo = new Int_t[nLr<<1];
+ fNLrActive = fITS->GetNLayersActive();
+ fWorkHyp = new AliITSUTrackHyp(fNLrActive);
//
+ if (fLayerMaxCandidates<1) fLayerMaxCandidates = 1000;
+ fLayerCandidates = new AliITSUSeed*[fLayerMaxCandidates];
fSeedsPool.ExpandCreateFast(1000); // RS TOCHECK
fFreeSeedsID.Set(1000);
+ fESDIndex.Set(1000);
+
//
}
void AliITSUTrackerGlo::CreateDefaultTrackCond()
{
// creates default tracking conditions to be used when recoparam does not provide them
- int nLr = fITS->GetNLayersActive();
- fClInfo = new Int_t[nLr<<1];
- //
+
AliITSUTrackCond* cond = new AliITSUTrackCond();
//
- cond->SetNLayers(fITS->GetNLayersActive());
- cond->AddNewCondition(nLr);
- cond->AddGroupPattern( 0xffff ); // require all layers hit
+ cond->SetNLayers(fNLrActive);
+ cond->AddNewCondition(fNLrActive);
+ cond->AddGroupPattern( 0xffff, fNLrActive ); // require all layers hit
+ cond->Init();
//
fDefTrackConds.AddLast(cond);
//
//_________________________________________________________________________
Int_t AliITSUTrackerGlo::Clusters2Tracks(AliESDEvent *esdEv)
{
+ //
+ AliCodeTimerAuto("",0);
+ SetTrackingPhase(kClus2Tracks);
+ //
+#ifdef _ITSU_TUNING_MODE_
+ if (!fCHistoArrCorr) BookControlHistos("Corr");
+ if (!fCHistoArrFake) BookControlHistos("Fake");
+#endif
+ static int evID = 0;
+ static TArrayF esdTrPt(fESDIndex.GetSize());
//
TObjArray *trackConds = 0;
//
- SetTrackingPhase(kClus2Tracks);
+ fCountProlongationTrials = 0;
+ fCountITSin = 0;
+ fCountITSout = 0;
+ fCountITSrefit = 0;
+ //
ResetSeedsPool();
int nTrESD = esdEv->GetNumberOfTracks();
AliInfo(Form("Will try to find prolongations for %d tracks",nTrESD));
- AliITSUReconstructor::GetRecoParam()->Print();
- fHypStore.Delete();
- if (fHypStore.GetSize()<nTrESD) fHypStore.Expand(nTrESD+100);
- //
- fITS->ProcessClusters();
- //
int nTrackCond = AliITSUReconstructor::GetRecoParam()->GetNTrackingConditions();
if (nTrackCond<1) {
if (!fDefTrackConds.GetEntriesFast()) {
}
else trackConds = AliITSUReconstructor::GetRecoParam()->GetTrackingConditions();
//
+ static Bool_t first = kTRUE;
+ if (first) {
+ AliITSUReconstructor::GetRecoParam()->Print();
+ first = kFALSE;
+ }
+ fHypStore.Delete();
+ if (fHypStore.GetSize()<nTrESD) fHypStore.Expand(nTrESD+100);
+ //
+ fITS->ProcessClusters();
+ //
+#ifdef _ITSU_TUNING_MODE_
+ FlagSplitClusters(); // tmp RS
+#endif
+ //
+ // the tracks will be reconstructed in decreasing pt order, sort them
+ if (fESDIndex.GetSize()<nTrESD) {
+ fESDIndex.Set(nTrESD+200);
+ esdTrPt.Set(nTrESD+200);
+ }
+ int* esdTrackIndex = fESDIndex.GetArray();
+ float* trPt = esdTrPt.GetArray();
+ for (int itr=nTrESD;itr--;) trPt[itr] = esdEv->GetTrack(itr)->Pt();
+ Sort(nTrESD,trPt,esdTrackIndex,kTRUE);
+ //
for (int icnd=0;icnd<nTrackCond;icnd++) {
fCurrTrackCond = (AliITSUTrackCond*)trackConds->UncheckedAt(icnd);
+ if (!fCurrTrackCond->IsInitDone()) fCurrTrackCond->Init();
// select ESD tracks to propagate
for (int itr=0;itr<nTrESD;itr++) {
- AliESDtrack *esdTr = esdEv->GetTrack(itr);
- AliInfo(Form("Processing track %d | MCLabel: %d",itr,esdTr->GetTPCLabel()));
- if (!NeedToProlong(esdTr)) continue; // are we interested in this track?
- FindTrack(esdTr, itr);
+ int trID = esdTrackIndex[itr];
+ fCurrESDtrack = esdEv->GetTrack(trID);
+ fCurrESDtrMClb = fCurrESDtrack->GetLabel();
+ //
+ if (!NeedToProlong(fCurrESDtrack)) continue; // are we interested in this track?
+ /*
+ // if specific tracks should be checked, create a global array int wtc[] = {evITS*10000+trID, ...};
+ Bool_t dbg = kFALSE;
+ int nwtc = sizeof(wtc)/sizeof(int);
+
+ for (int k=0;k<nwtc;k++) {
+ if (wtc[k]==evID*10000+trID) {
+ dbg = kTRUE;
+ printf("\n\n\n\n\n\n\n");
+ printf("At esdTr: %d MC: %d\n",wtc[k],fCurrESDtrMClb);
+ printf("\n\n\n\n\n\n\n");
+ break;
+ }
+ }
+ AliLog::SetClassDebugLevel("AliITSUTrackerGlo",dbg ? 10:0);
+ */
+
+ AliDebug(1,Form("Processing track %d(esd%d) | M=%.3f Pt=%.3f | MCLabel: %d",itr,trID,fCurrESDtrack->GetMass(kTRUE),fCurrESDtrack->Pt(),fCurrESDtrMClb));//RS
+ FindTrack(fCurrESDtrack, trID);
}
//
- AliInfo(Form("SeedsPool: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
- fHypStore.Print();
+ if (AliDebugLevelClass()>+2) {
+ AliInfo(Form("SeedsPool: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
+ fHypStore.Print();
+ }
FinalizeHypotheses();
+ //
+ //AliLog::SetClassDebugLevel("AliITSUTrackerGlo",0); // in case wtc array was defined, uncomment this
}
//
+ AliInfo(Form("%d ITSin for %d tried TPC seeds out of %d ESD tracks\n",fCountITSin,fCountProlongationTrials,nTrESD));
+ evID++;
return 0;
}
{
//
// Do outward fits in ITS
+ AliCodeTimerAuto("",0);
//
SetTrackingPhase(kPropBack);
int nTrESD = esdEv->GetNumberOfTracks();
- AliInfo(Form("Will propagate back %d tracks",nTrESD));
+ AliDebug(1,Form("Will propagate back %d tracks",nTrESD));
//
double bz0 = GetBz();
Double_t xyzTrk[3],xyzVtx[3]={GetX(),GetY(),GetZ()};
- AliITSUTrackHyp dummyTr,*currTr=0;
+ AliITSUTrackHyp dummyTr;
const double kWatchStep=10.; // for larger steps watch arc vs segment difference
Double_t times[AliPID::kSPECIES];
//
- //
for (int itr=0;itr<nTrESD;itr++) {
- AliESDtrack *esdTr = esdEv->GetTrack(itr);
+ fCurrESDtrack = esdEv->GetTrack(itr);
+ fCurrESDtrMClb = fCurrESDtrack->GetLabel();
// Start time integral and add distance from current position to vertex
- if (esdTr->IsOn(AliESDtrack::kITSout)) continue; // already done
+ if (fCurrESDtrack->IsOn(AliESDtrack::kITSout)) continue; // already done
//
- esdTr->GetXYZ(xyzTrk);
+ fCurrESDtrack->GetXYZ(xyzTrk);
Double_t dst = 0.; // set initial track lenght, tof
{
double dxs = xyzTrk[0] - xyzVtx[0];
// Hence s^2/d^2 = (1+1/6 p^2)^2
dst = dxs*dxs + dys*dys;
if (dst > kWatchStep*kWatchStep) { // correct circular part for arc/segment factor
- double crv = Abs(esdTr->GetC(bz0));
+ double crv = Abs(fCurrESDtrack->GetC(bz0));
double fcarc = 1.+crv*crv*dst/6.;
dst *= fcarc*fcarc;
}
dst = Sqrt(dst);
}
//
- esdTr->SetStatus(AliESDtrack::kTIME);
+ fCurrESDtrack->SetStatus(AliESDtrack::kTIME);
//
- if (!esdTr->IsOn(AliESDtrack::kITSin)) { // case of tracks w/o ITS prolongation: just set the time integration
- dummyTr.AliExternalTrackParam::operator=(*esdTr);
+ if (!fCurrESDtrack->IsOn(AliESDtrack::kITSin)) { // case of tracks w/o ITS prolongation: just set the time integration
+ dummyTr.AliExternalTrackParam::operator=(*fCurrESDtrack);
dummyTr.StartTimeIntegral();
dummyTr.AddTimeStep(dst);
dummyTr.GetIntegratedTimes(times);
- esdTr->SetIntegratedTimes(times);
- esdTr->SetIntegratedLength(dummyTr.GetIntegratedLength());
+ fCurrESDtrack->SetIntegratedTimes(times);
+ fCurrESDtrack->SetIntegratedLength(dummyTr.GetIntegratedLength());
continue;
}
//
- currTr = GetTrackHyp(itr);
- currTr->StartTimeIntegral();
- currTr->AddTimeStep(dst);
- printf("Before resetCov: "); currTr->AliExternalTrackParam::Print();
- currTr->ResetCovariance(10000);
- if (RefitTrack(currTr,fITS->GetRMax())) { // propagate to exit from the ITS/TPC screen
- UpdateESDTrack(currTr,AliESDtrack::kITSout);
+ fCurrHyp = GetTrackHyp(itr);
+ fCurrMass = fCurrHyp->GetMass();
+ fCurrHyp->StartTimeIntegral();
+ fCurrHyp->AddTimeStep(dst);
+ fCurrHyp->ResetCovariance(10000);
+ double chi2 = RefitTrack(fCurrHyp,fITS->GetRMax());
+ if (chi2>0) { // propagate to exit from the ITS/TPC screen
+ int ndf = fCurrHyp->GetWinner()->GetNLayersHit()*2-5;
+ if (ndf>0) chi2 /= ndf;
+ fCurrHyp->SetChi2(chi2);
+ UpdateESDTrack(fCurrHyp,AliESDtrack::kITSout);
}
else {
- AliInfo(Form("Refit Failed for track %d",itr));
+ AliDebug(2,Form("Refit Failed for track %d | ESDtrack#%d (MClb:%d)",itr,fCurrESDtrack->GetID(),fCurrESDtrMClb));
+ //fCurrHyp->AliExternalTrackParam::Print();
+ //fCurrHyp->GetWinner()->Print();
}
//
}
//
+ AliInfo(Form("%d ITSout in %d ITSin tracks for %d tried TPC seeds out of %d ESD tracks\n",
+ fCountITSout,fCountITSin,fCountProlongationTrials,nTrESD));
+ //
return 0;
}
{
//
// refit the tracks inward, using current cov.matrix
+ AliCodeTimerAuto("",0);
//
SetTrackingPhase(kRefitInw);
Int_t nTrESD = esdEv->GetNumberOfTracks();
- AliInfo(Form("Will refit inward %d tracks",nTrESD));
- AliITSUTrackHyp *currTr=0;
+ // AliLog::SetClassDebugLevel("AliITSUTrackerGlo",10);
+
+ AliDebug(1,Form("Will refit inward %d tracks",nTrESD));
//
for (int itr=0;itr<nTrESD;itr++) {
- AliESDtrack *esdTr = esdEv->GetTrack(itr);
+ fCurrESDtrack = esdEv->GetTrack(itr);
+ fCurrESDtrMClb = fCurrESDtrack->GetLabel();
// Start time integral and add distance from current position to vertex
- UInt_t trStat = esdTr->GetStatus();
+ UInt_t trStat = fCurrESDtrack->GetStatus();
if ( !(trStat & AliESDtrack::kITSout) ) continue;
if ( trStat & AliESDtrack::kITSrefit ) continue; // already done
if ( (trStat & AliESDtrack::kTPCout) && !(trStat & AliESDtrack::kTPCrefit) ) continue;
//
- currTr = GetTrackHyp(itr);
- currTr->AliExternalTrackParam::operator=(*esdTr); // fetch current ESDtrack kinematics
- if (RefitTrack(currTr,fITS->GetRMin())) { // propagate up to inside radius of the beam pipe
- UpdateESDTrack(currTr,AliESDtrack::kITSrefit);
+ fCurrHyp = GetTrackHyp(itr);
+ fCurrHyp->AliExternalTrackParam::operator=(*fCurrESDtrack); // fetch current ESDtrack kinematics
+ fCurrMass = fCurrHyp->GetMass();
+ //
+ double chi2 = RefitTrack(fCurrHyp,fITS->GetRMin());
+ if (chi2>0) { // propagate up to inside radius of the beam pipe
+ fCurrHyp->SetChi2(chi2);
+ UpdateESDTrack(fCurrHyp,AliESDtrack::kITSrefit);
}
else {
- AliInfo(Form("Refit Failed for track %d",itr));
+ AliDebug(2,Form("Refit Failed for track %d |ESDtrack#%d (MClb:%d)",itr,fCurrESDtrack->GetID(),fCurrESDtrMClb));
}
}
//
+ AliInfo(Form("%d ITSrefit in %d ITSout in %d ITSin tracks for %d tried TPC seeds out of %d ESD tracks\n",
+ fCountITSrefit,fCountITSout,fCountITSin,fCountProlongationTrials,nTrESD));
+ //
+ // AliLog::SetClassDebugLevel("AliITSUTrackerGlo",0);
return 0;
}
&& (Abs(dtz[0])>AliITSUReconstructor::GetRecoParam()->GetMaxDForProlongation() ||
Abs(dtz[1])>AliITSUReconstructor::GetRecoParam()->GetMaxDZForProlongation())) return kFALSE;
//
+ // if (esdTr->Pt()<3) return kFALSE;//RS
return kTRUE;
}
AliITSUSeed seedUC; // copy of the seed from the upper layer
AliITSUSeed seedT; // transient seed between the seedUC and new prolongation hypothesis
//
- if (!InitHypothesis(esdTr,esdID)) return; // initialize prolongations hypotheses tree
+ AliITSUTrackHyp* hypTr = InitHypothesis(esdTr,esdID); // initialize prolongations hypotheses tree
+ if (!hypTr || hypTr->GetSkip()) return;
+ //
+ fCurrHyp = fWorkHyp;
+ fCurrHyp->InitFrom(hypTr);
//
AliITSURecoSens *hitSens[AliITSURecoSens::kNNeighbors+1];
//
TObjArray clArr; // container for transfer of clusters matching to seed
//
- int nLrActive = fITS->GetNLayersActive();
- for (int ila=nLrActive;ila--;) {
+ for (int ila=fNLrActive;ila--;) {
+ fCurrActLrID = ila;
+ fCurrLayer = fITS->GetLayerActive(ila);
+ Bool_t noClSharing = fCurrTrackCond->GetClSharing(ila)==0;
int ilaUp = ila+1; // prolong seeds from layer above
//
// for the outermost layer the seed is created from the ESD track
- int nSeedsUp = (ilaUp==nLrActive) ? 1 : fCurrHyp->GetNSeeds(ilaUp);
+ int nSeedsUp = (ilaUp==fNLrActive) ? 1 : fCurrHyp->GetNSeeds(ilaUp);
+ int maxNBranches = fCurrTrackCond->GetMaxBranches(ila);
+ int maxNCandidates = fCurrTrackCond->GetMaxCandidates(ila);
//
for (int isd=0;isd<nSeedsUp;isd++) {
AliITSUSeed* seedU;
- if (ilaUp==nLrActive) {
+ if (ilaUp==fNLrActive) {
seedU = 0;
- seedUC.InitFromESDTrack(esdTr);
+ seedUC.InitFromSeed(fCurrHyp->GetTPCSeed()); // init with TPC seed from ref.R
}
else {
seedU = fCurrHyp->GetSeed(ilaUp,isd); // seed on prev.active layer to prolong
+ if (seedU->IsKilled()) continue;
seedUC = *seedU; // its copy will be prolonged
seedUC.SetParent(seedU);
}
seedUC.ResetFMatrix(); // reset the matrix for propagation to next layer
// go till next active layer
- AliInfo(Form("working on Lr:%d Seed:%d of %d",ila,isd,nSeedsUp));
- if (!TransportToLayer(&seedUC, fITS->GetLrIDActive(ilaUp), fITS->GetLrIDActive(ila)) ) {
+ 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()));
+ if (!TransportToLayer(&seedUC, fITS->GetLrIDActive(ilaUp), fITS->GetLrIDActive(ila)) ) { // external seed already prolonged
//
- AliInfo("Transport failed");
+ AliDebug(2,Form("Transport failed | esdID=%d (MClb:%d)",esdID,fCurrESDtrMClb));
// Check if the seed satisfies to track definition
if (NeedToKill(&seedUC,kTransportFailed) && seedU) KillSeed(seedU,kTRUE);
continue; // RS TODO: decide what to do with tracks stopped on higher layers w/o killing
}
- AliITSURecoLayer* lrA = fITS->GetLayerActive(ila);
if (!GetRoadWidth(&seedUC, ila)) { // failed to find road width on the layer
if (NeedToKill(&seedUC,kRWCheckFailed) && seedU) KillSeed(seedU,kTRUE);
continue;
}
- int nsens = lrA->FindSensors(&fTrImpData[kTrPhi0], hitSens); // find detectors which may be hit by the track
- AliInfo(Form("Will check %d sensors on lr:%d ",nsens,ila));
+ /*
+ //RS toremove
+ int mcquest = -1;
+ if (!seedUC.ContainsFake() && AliDebugLevelClass()>2) {
+ mcquest = fCurrESDtrMClb;
+ seedUC.Print("etp");
+ printf("FSParams: "); for (int ip=0;ip<kNTrImpData;ip++) printf("%+e ",fTrImpData[ip]); printf("\n");
+ }
+ //
+ int nsens = fCurrLayer->FindSensors(&fTrImpData[kTrPhi0], hitSens, mcquest); // find detectors which may be hit by the track
+ */
+ int nsens = fCurrLayer->FindSensors(&fTrImpData[kTrPhi0], hitSens); // find detectors which may be hit by the track
+ AliDebug(2,Form("Will check %d sensors on lr:%d | esdID=%d (MClb:%d)",nsens,ila,esdID,fCurrESDtrMClb));
+ //
+ seedUC.SetLr(ila);
//
double bz = GetBz();
for (int isn=nsens;isn--;) {
- seedT = seedUC;
AliITSURecoSens* sens = hitSens[isn];
+ int ncl = sens->GetNClusters();
+ if (!ncl) continue;
+ seedT = seedUC;
//
- // We need to propagate the seed to sensor on lrA staying the frame of the sensor from prev.layer,
+ // We need to propagate the seed to sensor on fCurrLayer staying in the frame of the sensor from prev.layer,
// since the transport matrix should be defined in this frame.
double xs; // X in the TF of current seed, corresponding to intersection with sensor plane
- if (!seedT.GetTrackingXAtXAlpha(sens->GetXTF(),sens->GetPhiTF(),bz, xs)) continue;
- if (!PropagateSeed(&seedT,xs,fCurrMass)) continue;
+ if (!seedT.GetTrackingXAtXAlpha(sens->GetXTF(),sens->GetPhiTF(),bz, xs)) {
+ if (AliDebugLevelClass()>2) {
+ printf("Failed on GetTrackingXAtXAlpha: X=%.4f alp=%.3f\n",sens->GetXTF(),sens->GetPhiTF());
+ seedT.Print("etp");
+ }
+ continue;
+ }
+ if (xs<seedT.GetX()) {
+ if (!PropagateSeed(&seedT,xs,fCurrMass)) continue;
+ }
+ else { // some low precision tracks may hit the sensor plane outside of the layer radius
+ if (AliDebugLevelClass()>2) {
+ if (!seedT.ContainsFake()) {
+ printf("WRONG PROP on L%d, sens%d of %d: %.4f > %.4f\n",ila,isn,nsens,xs,seedT.GetX());
+ sens->Print();
+ seedT.Print("etp");
+ }
+ }
+ if (!seedT.PropagateParamOnlyTo(xs,bz)) continue;
+ }
// if (!seedT.PropagateToX(xs,bz)) continue;
// if (!seedT.Rotate(sens->GetPhiTF())) continue;
if (!seedT.RotateToAlpha(sens->GetPhiTF())) continue;
//
int clID0 = sens->GetFirstClusterId();
- for (int icl=sens->GetNClusters();icl--;) {
+ for (int icl=ncl;icl--;) { // don't change the order, clusters are sorted
+ //AliLog::SetClassDebugLevel("AliITSUTrackerGlo",10);
+ int clID = clID0 + icl;
+ if (noClSharing && fCurrLayer->GetCluster(clID)->IsClusterUsed()) continue;
int res = CheckCluster(&seedT,ila,clID0+icl);
+ //AliLog::SetClassDebugLevel("AliITSUTrackerGlo", 0);
//
if (res==kStopSearchOnSensor) break; // stop looking on this sensor
if (res==kClusterNotMatching) continue; // cluster does not match
// cluster is matching and it was added to the hypotheses tree
}
}
- // cluster search is done. Do we need ta have a version of this seed skipping current layer
- seedT.SetLr(ila);
- if (!NeedToKill(&seedT,kMissingCluster)) {
- AliITSUSeed* seedSkp = NewSeedFromPool(&seedT);
- double penalty = -AliITSUReconstructor::GetRecoParam()->GetMissPenalty(ila);
+ // cluster search is done. Do we need to have a version of this seed skipping current layer
+ if (!NeedToKill(&seedUC,kMissingCluster)) {
+ AliITSUSeed* seedSkp = NewSeedFromPool(&seedUC);
+ double penalty = -fCurrTrackCond->GetMissPenalty(ila);
// to do: make penalty to account for probability to miss the cluster for good reason
seedSkp->SetChi2Cl(penalty);
- AddProlongationHypothesis(seedSkp,ila);
+ if (seedSkp->GetChi2GloNrm()>fCurrTrackCond->GetMaxChi2GloNrm(ila)) {
+ MarkSeedFree(seedSkp);
+ }
+ else AddSeedBranch(seedSkp);
}
+ // transfer the new branches of the seed to the hypothesis container
+ if (fNBranchesAdded) ValidateAllowedBranches(maxNBranches);
+ //
+ }
+ if (fNCandidatesAdded) ValidateAllowedCandidates(ila,maxNCandidates);
+ // ((TObjArray*)fCurrHyp->GetLayerSeeds(ila))->Sort();
+ if (AliDebugLevelClass()>2) { //RS
+ printf(">>> All hypotheses on lr %d: \n",ila);
+ for (int ih=0;ih<fCurrHyp->GetNSeeds(ila);ih++) {printf(" #%3d ",ih); fCurrHyp->GetSeed(ila,ih)->Print();}
}
- ((TObjArray*)fCurrHyp->GetLayerSeeds(ila))->Sort();
- printf(">>> All hypotheses on lr %d: \n",ila);
+ //
+ /*
for (int ih=0;ih<fCurrHyp->GetNSeeds(ila);ih++) {
- printf(" #%3d ",ih); fCurrHyp->GetSeed(ila,ih)->Print();
- //
- /*
if (ila!=0) continue;
double vecL[5] = {0};
double matL[15] = {0};
AliITSUSeed* sp = fCurrHyp->GetSeed(ila,ih);
while(sp->GetParent()) {
sp->Smooth(vecL,matL);
- if (sp->GetLayerID()>=fITS->GetNLayersActive()-1) break;
+ if (sp->GetLayerID()>=fNLrActive-1) break;
sp = (AliITSUSeed*)sp->GetParent();
}
*/
- }
}
//
- SaveCurrentTrackHypotheses();
+ SaveReducedHypothesesTree(hypTr);
+ if (AliDebugLevelClass()>1) {
+ printf("\nSaved hypotheses for esdTrack %d (MCLab:%d)\n",esdID,fCurrESDtrMClb);
+ hypTr->Print("l");
+ }
+ fCurrHyp = 0;
//
}
//_________________________________________________________________________
-Bool_t AliITSUTrackerGlo::InitHypothesis(AliESDtrack *esdTr, Int_t esdID)
+AliITSUTrackHyp* AliITSUTrackerGlo::InitHypothesis(AliESDtrack *esdTr, Int_t esdID)
{
// init prolongaion candidates finding for single seed
- fCurrHyp = GetTrackHyp(esdID);
- if (fCurrHyp) return kTRUE;
+ AliITSUTrackHyp* hyp = GetTrackHyp(esdID);
+ if (hyp) return hyp;
+ //
+ fCountProlongationTrials++;
//
fCurrMass = esdTr->GetMass();
- fCurrESDtrack = esdTr;
if (fCurrMass<kPionMass*0.9) fCurrMass = kPionMass; // don't trust to mu, e identification from TPCin
//
- fCurrHyp = new AliITSUTrackHyp(fITS->GetNLayersActive());
- fCurrHyp->SetESDTrack(esdTr);
- fCurrHyp->SetUniqueID(esdID);
- fCurrHyp->SetMass(fCurrMass);
- SetTrackHyp(fCurrHyp,esdID);
+ hyp = new AliITSUTrackHyp(fNLrActive);
+ hyp->SetESDTrack(esdTr);
+ hyp->SetUniqueID(esdID);
+ hyp->SetMass(fCurrMass);
+ hyp->SetTPCSeed( new AliExternalTrackParam(*esdTr) );
+ SetTrackHyp(hyp,esdID);
+
+ if (!TransportToLayer(hyp->GetTPCSeed(),fITS->GetNLayers(), fITS->GetLrIDActive(fNLrActive-1), fITS->GetRITSTPCRef())) hyp->SetSkip(); // propagate to outer R of ITS
//
- return kTRUE;
+ return hyp;
// TO DO
}
//_________________________________________________________________________
-Bool_t AliITSUTrackerGlo::TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo)
+Bool_t AliITSUTrackerGlo::TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo, Double_t rLim)
{
- // transport seed from layerFrom to the entrance of layerTo
+ // transport seed from layerFrom to the entrance (along the direction of the track) of layerTo or to rLim (if>0), wathever is closer
//
//
if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
int dir = lTo > lFrom ? 1:-1;
AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
Bool_t checkFirst = kTRUE;
+ Bool_t limReached = kFALSE;
+ while(lFrom!=lTo) {
+ if (lrFr) {
+ if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
+ checkFirst = kFALSE;
+ }
+ AliITSURecoLayer* lrTo = fITS->GetLayer( (lFrom+=dir) );
+ if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
+ //
+ // go the entrance of the layer, assuming no materials in between
+ double xToGo = lrTo->GetR(-dir);
+ if (rLim>0) {
+ if (dir>0) {
+ if (rLim<xToGo) {xToGo = rLim; limReached = kTRUE;}
+ }
+ else {
+ if (rLim>xToGo) {xToGo = rLim; limReached = kTRUE;}
+ }
+ }
+ // double xts = xToGo;
+ if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
+ //printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
+ //seed->Print("etp");
+ //
+ return kFALSE;
+ }
+ AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
+ if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
+ //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
+ //seed->Print("etp");
+ return kFALSE;
+ }
+ lrFr = lrTo;
+ if (limReached) break;
+ }
+ return kTRUE;
+ //
+}
+
+//_________________________________________________________________________
+Bool_t AliITSUTrackerGlo::TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t rLim)
+{
+ // transport track from layerFrom to the entrance of layerTo or to rLim (if>0), wathever is closer
+ //
+ if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
+ //
+ int dir = lTo > lFrom ? 1:-1;
+ AliITSURecoLayer* lrFr = fITS->GetLayer(lFrom); // this can be 0 when extrapolation from TPC to ITS is requested
+ Bool_t checkFirst = kTRUE;
+ Bool_t limReached = kFALSE;
while(lFrom!=lTo) {
if (lrFr) {
if (!GoToExitFromLayer(seed,lrFr,dir,checkFirst)) return kFALSE; // go till the end of current layer
//
// go the entrance of the layer, assuming no materials in between
double xToGo = lrTo->GetR(-dir);
- if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
- if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) return kFALSE;
+ if (rLim>0) {
+ if (dir>0) {
+ if (rLim<xToGo) {xToGo = rLim; limReached = kTRUE;}
+ }
+ else {
+ if (rLim>xToGo) {xToGo = rLim; limReached = kTRUE;}
+ }
+ }
+ // double xts = xToGo;
+ if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
+ // printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
+ // seed->Print("etp");
+ return kFALSE;
+ }
+ AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
+ if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
+ //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
+ //seed->Print("etp");
+ return kFALSE;
+ }
lrFr = lrTo;
+ if (limReached) break;
}
return kTRUE;
//
}
//_________________________________________________________________________
-Bool_t AliITSUTrackerGlo::TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo)
+Bool_t AliITSUTrackerGlo::TransportToLayerX(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t xStop)
{
- // transport track from layerFrom to the entrance of layerTo
+ // transport track from layerFrom to the entrance of layerTo but do not pass control parameter X
//
if (lTo==lFrom) AliFatal(Form("was called with lFrom=%d lTo=%d",lFrom,lTo));
//
AliITSURecoLayer* lrTo = fITS->GetLayer( (lFrom+=dir) );
if (!lrTo) AliFatal(Form("Layer %d does not exist",lFrom));
//
- if (!GoToExitFromLayer(seed,lrTo,dir)) return kFALSE; // go the entrance of the layer, assuming no materials in between
+ // go the entrance of the layer, assuming no materials in between
+ double xToGo = lrTo->GetR(-dir); // R of the entrance to layer
+ //
+ // double xts = xToGo;
+ if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) {
+ // printf("FailHere1: %f %f %d\n",xts,xToGo,dir);
+ // seed->Print("etp");
+ return kFALSE;
+ }
+ if ( (dir>0&&xToGo>xStop) || (dir<0&&xToGo<xStop) ) xToGo = xStop;
+ //
+ AliDebug(2,Form("go in dir=%d to R=%.4f(X:%.4f)",dir,lrTo->GetR(-dir), xToGo));
+ if (!PropagateSeed(seed,xToGo,fCurrMass,100, kFALSE )) {
+ //printf("FailHere2: %f %f %d\n",xts,xToGo,dir);
+ //seed->Print("etp");
+ return kFALSE;
+ }
lrFr = lrTo;
}
return kTRUE;
double xToGo = lr->GetR(dir);
if (check) { // do we need to track till the surface of the current layer ?
double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
- // AliInfo(Form(" dir:%d Cur: %e Tgt: %e",dir,Sqrt(curR2),xToGo));
- if (dir>0) { if (curR2-xToGo*xToGo>fgkToler) return kTRUE; } // on the surface or outside of the layer
- else if (dir<0) { if (xToGo*xToGo-curR2>fgkToler) return kTRUE; } // on the surface or outside of the layer
+ AliDebug(3,Form(" dir:%d Cur: %e Tgt: %e",dir,Sqrt(curR2),xToGo));
+ if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // on the surface or outside of the layer
+ else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // on the surface or outside of the layer
}
if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
// go via layer to its boundary, applying material correction.
double xToGo = lr->GetR(dir);
if (check) { // do we need to track till the surface of the current layer ?
double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
- if (dir>0) { if (curR2-xToGo*xToGo>fgkToler) return kTRUE; } // on the surface or outside of the layer
- else if (dir<0) { if (xToGo*xToGo-curR2>fgkToler) return kTRUE; } // on the surface or outside of the layer
+ if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // on the surface or outside of the layer
+ else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // on the surface or outside of the layer
}
+ AliDebug(2,Form(" dir=%d : from R=%.4f -> R=%.4f",dir,Sqrt(seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY()), xToGo));
if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
// go via layer to its boundary, applying material correction.
if (!PropagateSeed(seed,xToGo,fCurrMass, lr->GetMaxStep())) return kFALSE;
double xToGo = lr->GetR(-dir);
if (check) { // do we need to track till the surface of the current layer ?
double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
- if (dir>0) { if (curR2-xToGo*xToGo>fgkToler) return kTRUE; } // already passed
- else if (dir<0) { if (xToGo*xToGo-curR2>fgkToler) return kTRUE; } // already passed
+ if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // already passed
+ else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // already passed
}
if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
// go via layer to its boundary, applying material correction.
double xToGo = lr->GetR(-dir);
if (check) { // do we need to track till the surface of the current layer ?
double curR2 = seed->GetX()*seed->GetX() + seed->GetY()*seed->GetY(); // current radius
- if (dir>0) { if (curR2-xToGo*xToGo>fgkToler) return kTRUE; } // already passed
- else if (dir<0) { if (xToGo*xToGo-curR2>fgkToler) return kTRUE; } // already passed
+ if (dir>0) { if (curR2-xToGo*xToGo>-fgkToler) return kTRUE; } // already passed
+ else if (dir<0) { if (xToGo*xToGo-curR2>-fgkToler) return kTRUE; } // already passed
}
if (!seed->GetXatLabR(xToGo,xToGo,GetBz(),dir)) return kFALSE;
// go via layer to its boundary, applying material correction.
double bz = GetBz();
AliITSURecoLayer* lrA = fITS->GetLayerActive(ilrA);
seed->GetXYZ(&fTrImpData[kTrXIn]); // lab position at the entrance from above
- static AliExternalTrackParam sc; // seed copy for manipalitions
+ static AliExternalTrackParam sc; // seed copy for manipulations
sc = *seed;
//
fTrImpData[kTrPhiIn] = ATan2(fTrImpData[kTrYIn],fTrImpData[kTrXIn]);
// special case: track does not reach inner radius, might be tangential
double r = sc.GetD(0,0,bz);
double x;
- if (!sc.GetXatLabR(r,x,bz,-1)) {
- sc.Print();
- AliFatal(Form("This should not happen: r=%f",r));
- }
+ if (!sc.GetXatLabR(r,x,bz,-1)) return kFALSE;
dr = Abs(sc.GetX() - x);
- if (!sc.GetXYZAt(x, bz, fTrImpData + kTrXOut)) {
- sc.Print();
- AliFatal(Form("This should not happen: x=%f",x));
- }
+ if (!sc.GetXYZAt(x, bz, fTrImpData + kTrXOut)) return kFALSE;
}
//
fTrImpData[kTrPhiOut] = ATan2(fTrImpData[kTrYOut],fTrImpData[kTrXOut]);
double sgy = sc.GetSigmaY2() + dr*dr*sc.GetSigmaSnp2() + AliITSUReconstructor::GetRecoParam()->GetSigmaY2(ilrA);
double sgz = sc.GetSigmaZ2() + dr*dr*sc.GetSigmaTgl2() + AliITSUReconstructor::GetRecoParam()->GetSigmaZ2(ilrA);
- sgy = Sqrt(sgy)*AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadY();
- sgz = Sqrt(sgz)*AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadZ();
- fTrImpData[kTrPhi0] = 0.5*(fTrImpData[kTrPhiOut]+fTrImpData[kTrPhiIn]);
+ sgy = Sqrt(sgy)*fCurrTrackCond->GetNSigmaRoadY(ilrA);
+ sgz = Sqrt(sgz)*fCurrTrackCond->GetNSigmaRoadZ(ilrA);
+ double dphi0 = 0.5*Abs(fTrImpData[kTrPhiOut]-fTrImpData[kTrPhiIn]);
+ double phi0 = 0.5*(fTrImpData[kTrPhiOut]+fTrImpData[kTrPhiIn]);
+ if ( dphi0>(0.5*Pi()) ) {
+ // special case of angles around pi
+ dphi0 = Abs(phi0);
+ phi0 += Pi();
+ }
+
+ fTrImpData[kTrPhi0] = phi0;
fTrImpData[kTrZ0] = 0.5*(fTrImpData[kTrZOut]+fTrImpData[kTrZIn]);
- fTrImpData[kTrDPhi] = 0.5*Abs(fTrImpData[kTrPhiOut]-fTrImpData[kTrPhiIn]) + sgy/lrA->GetR();
+ dphi0 += sgy/lrA->GetR();
+ fTrImpData[kTrDPhi] = dphi0<PiOver2() ? dphi0 : PiOver2();
fTrImpData[kTrDZ] = 0.5*Abs(fTrImpData[kTrZOut]-fTrImpData[kTrZIn]) + sgz;
//
return kTRUE;
void AliITSUTrackerGlo::ResetSeedsPool()
{
// mark all seeds in the pool as unused
- AliInfo(Form("CurrentSize: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
+ AliDebug(-1,Form("CurrentSize: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
fNFreeSeeds = 0;
fSeedsPool.Clear(); // seeds don't allocate memory
}
// kClusterMatching otherwise
//
const double kTolerX = 5e-4;
- AliCluster *cl = fITS->GetLayerActive(lr)->GetCluster(clID);
+ // AliCluster *cl = fITS->GetLayerActive(lr)->GetCluster(clID);
+ AliCluster *cl = fCurrLayer->GetCluster(clID);
//
Bool_t goodCl = kFALSE;
- int currLabel = Abs(fCurrESDtrack->GetTPCLabel());
+ int currLabel = Abs(fCurrESDtrMClb);
//
if (cl->GetLabel(0)>=0) {
for (int i=0;i<3;i++) if (cl->GetLabel(i)>=0 && cl->GetLabel(i)==currLabel) {goodCl = kTRUE; break;}
}
- else goodCl = kTRUE;
//
if (TMath::Abs(cl->GetX())>kTolerX) { // if due to the misalingment X is large, propagate track only
if (!track->PropagateParamOnlyTo(track->GetX()+cl->GetX(),GetBz())) {
- if (goodCl) {printf("Loose good cl: Failed propagation. |"); cl->Print();}
+ //
+ if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
+ AliDebug(2,Form("Lost good cl on L:%d failed propagation. |ESDtrack#%d (MClb:%d)",lr,fCurrESDtrack->GetID(),fCurrESDtrMClb));
+ track->Print("etp");
+ cl->Print();
+ }
return kStopSearchOnSensor; // propagation failed, seedT is intact
}
}
//
double dy2 = dy*dy;
double tol2 = (track->GetSigmaY2() + AliITSUReconstructor::GetRecoParam()->GetSigmaY2(lr))*
- AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadY()*AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadY(); // RS TOOPTIMIZE
+ fCurrTrackCond->GetNSigmaRoadY(lr)*fCurrTrackCond->GetNSigmaRoadY(lr); // RS TOOPTIMIZE
if (dy2>tol2) { // the clusters are sorted in Z(col) then in Y(row).
- if (goodCl) {printf("Loose good cl: dy2=%e > tol2=%e |",dy2,tol2); cl->Print();}
- if (dy>0) return kStopSearchOnSensor; // No chance that other cluster of this sensor will match (all Y's will be even larger)
+ //
+ if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
+ AliDebug(2,Form("Lost good cl: dy2=%e > tol2=%e |ESDtrack#%d (MClb:%d)",dy2,tol2,fCurrESDtrack->GetID(),fCurrESDtrMClb));
+ track->Print("etp");
+ cl->Print();
+ PrintSeedClusters(track);
+ }
+ // clusters are sorted in increasing Y and the loop goes from last (highers Y) to 1st.
+ // if the track has too large y for this cluster (dy<0) it will be so for all other clusters of the sensor
+ if (dy<0) return kStopSearchOnSensor; // No chance that other cluster of this sensor will match (all Y's will be even larger)
else return kClusterNotMatching; // Other clusters may match
}
double dz2 = dz*dz;
tol2 = (track->GetSigmaZ2() + AliITSUReconstructor::GetRecoParam()->GetSigmaZ2(lr))*
- AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadZ()*AliITSUReconstructor::GetRecoParam()->GetNSigmaRoadZ(); // RS TOOPTIMIZE
+ fCurrTrackCond->GetNSigmaRoadZ(lr)*fCurrTrackCond->GetNSigmaRoadZ(lr); // RS TOOPTIMIZE
if (dz2>tol2) {
- if (goodCl) {printf("Loose good cl: dz2=%e > tol2=%e |",dz2,tol2); cl->Print();}
+ if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
+ AliDebug(2,Form("Lost good cl on L:%d : dz2=%e > tol2=%e |ESDtrack#%d (MClb:%d)",lr,dz2,tol2,fCurrESDtrack->GetID(),fCurrESDtrMClb));
+ track->Print("etp");
+ cl->Print();
+ PrintSeedClusters(track);
+ }
return kClusterNotMatching; // Other clusters may match
}
//
Double_t p[2]={cl->GetY(), cl->GetZ()};
Double_t cov[3]={cl->GetSigmaY2(), cl->GetSigmaYZ(), cl->GetSigmaZ2()};
double chi2 = track->GetPredictedChi2(p,cov);
- if (chi2>AliITSUReconstructor::GetRecoParam()->GetMaxTr2ClChi2(lr)) {
- if (goodCl) {
- printf("Loose good cl: Chi2=%e > Chi2Max=%e |dy: %+.3e dz: %+.3e\n",
- chi2,AliITSUReconstructor::GetRecoParam()->GetMaxTr2ClChi2(lr),dy,dz);
+ //
+ if (chi2>fCurrTrackCond->GetMaxTr2ClChi2(lr)) {
+ if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
+ AliDebug(2,Form("Lost good cl on L:%d : Chi2=%e > Chi2Max=%e |dy: %+.3e dz: %+.3e |ESDtrack#%d (MClb:%d)\n",
+ lr,chi2,fCurrTrackCond->GetMaxTr2ClChi2(lr),dy,dz,fCurrESDtrack->GetID(),fCurrESDtrMClb));
track->Print("etp");
cl->Print("");
+ PrintSeedClusters(track);
}
return kClusterNotMatching;
}
//
track = NewSeedFromPool(track); // input track will be reused, use its clone for updates
if (!track->Update()) {
- if (goodCl) {printf("Loose good cl: Failed update |"); cl->Print();}
+ if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
+ AliDebug(2,Form("Lost good cluster on L:%d : failed to update",lr));
+ track->Print("etp");
+ cl->Print("");
+ PrintSeedClusters(track);
+ }
MarkSeedFree(track);
return kClusterNotMatching;
}
track->SetChi2Cl(chi2);
track->SetLrClusterID(lr,clID);
- cl->IncreaseClusterUsage();
+ //
+ if (track->GetChi2GloNrm()>fCurrTrackCond->GetMaxChi2GloNrm(lr)) {
+ if (AliDebugLevelClass()>2 && goodCl && !track->ContainsFake()) {
+ AliDebug(2,Form("Lost good cl on L:%d : Chi2Glo=%e > Chi2Max=%e |dy: %+.3e dz: %+.3e |ESDtrack#%d (MClb:%d)\n",
+ lr,track->GetChi2GloNrm(),fCurrTrackCond->GetMaxChi2GloNrm(lr),dy,dz,fCurrESDtrack->GetID(),fCurrESDtrMClb));
+ track->Print("etp");
+ cl->Print("");
+ PrintSeedClusters(track);
+ }
+ MarkSeedFree(track);
+ return kClusterNotMatching;
+ }
+ // cl->IncreaseClusterUsage(); // do this only for winners
//
track->SetFake(!goodCl);
//
- AliInfo(Form("AddCl(%d) Cl%d lr:%d: dY:%+8.4f dZ:%+8.4f (MC: %5d %5d %5d) |Chi2=%f(%c)",
+ AliDebug(2,Form("AddCl(%d) Cl%d lr:%d: dY:%+8.4f dZ:%+8.4f (MC: %5d %5d %5d) |Chi2=%f(%c)| ",
goodCl,clID,lr,dy,dz2,cl->GetLabel(0),cl->GetLabel(1),cl->GetLabel(2), chi2, track->IsFake() ? '-':'+'));
//
- AddProlongationHypothesis(track,lr);
+ AddSeedBranch(track);
//
return kClusterMatching;
}
if (flag==kMissingCluster) {
int lastChecked = seed->GetLayerID();
UShort_t patt = seed->GetHitsPattern();
- if (lastChecked) patt |= ~(kMask<<lastChecked); // not all layers were checked, complete unchecked once by potential hits
+ if (lastChecked) patt |= ~(kMask<<lastChecked); // not all layers were checked, complete unchecked ones by potential hits
Bool_t seedOK = fCurrTrackCond->CheckPattern(patt);
return !seedOK;
}
Double_t step = dir*TMath::Min(TMath::Abs(xToGo-xpos), maxStep);
Double_t x = xpos+step;
Double_t bz=GetBz(); // getting the local Bz
- if (!seed->PropagateToX(x,bz)) return kFALSE;
+ if (!seed->PropagateToX(x,bz)) return kFALSE;
double ds = 0;
if (matCorr || updTime) {
xyz0[0]=xyz1[0]; // global pos at the beginning of step
MeanMaterialBudget(xyz0,xyz1,param);
Double_t xrho=param[0]*param[4], xx0=param[1];
if (dir>0) xrho = -xrho; // outward should be negative
- if (!seed->ApplyMaterialCorrection(xx0,xrho,mass,kFALSE)) return kFALSE;
+ if (!seed->ApplyMaterialCorrection(xx0,xrho,mass,kFALSE)) {AliInfo("Fail2"); seed->Print("etp"); return kFALSE;}
ds = param[4];
}
else { // matCorr is not requested but time integral is
Int_t dir = (xpos<xToGo) ? 1:-1;
Double_t xyz0[3],xyz1[3],param[7];
//
+ if (AliDebugLevelClass()>1) {
+ AliDebug(2,Form("Before Propagate to X=%f with M=%.3f MaxStep=%.4f MatCorr=%d",xToGo,mass,maxStep,matCorr));
+ seed->AliExternalTrackParam::Print();
+ }
Bool_t updTime = dir>0 && seed->IsStartedTimeIntegral();
if (matCorr || updTime) seed->GetXYZ(xyz1); //starting global position
while ( (xToGo-xpos)*dir > kEpsilon){
//
xpos = seed->GetX();
}
+ if (AliDebugLevelClass()>1) {
+ AliDebug(2,Form("After Propagate to X=%f",xToGo));
+ seed->AliExternalTrackParam::Print();
+ }
return kTRUE;
}
//______________________________________________________________________________
-void AliITSUTrackerGlo::SaveCurrentTrackHypotheses()
+Bool_t AliITSUTrackerGlo::FinalizeHypothesis(AliITSUTrackHyp* hyp)
{
- // RS: shall we clean up killed seeds?
- fCurrHyp = 0;
- // TODO
-
+ // finalize single TPC track hypothesis
+ if (!hyp || hyp->GetESDTrack()->IsOn(AliESDtrack::kITSin)) return kFALSE;
+ AliITSUSeed* winner = 0;
+ fCurrHyp = hyp;
+ fCurrMass = hyp->GetMass();
+ if (!(winner=hyp->DefineWinner())) return kFALSE;
+ CookMCLabel(hyp);
+ //
+#ifdef _ITSU_TUNING_MODE_ // fill tuning histos
+ TObjArray* dest = hyp->GetLabel()>=0 ? fCHistoArrCorr : fCHistoArrFake;
+ if (dest) {
+ AliITSUSeed* sd = winner;
+ double htrPt = hyp->Pt();
+ do {
+ int clID,lrID;
+ if ( (clID=sd->GetLrCluster(lrID))<0 ) continue;
+ int hcOffs = (1+fTrackPhase)*kHistosPhase + lrID;
+ ((TH2*)dest->At(kHChi2Nrm+hcOffs))->Fill(htrPt,sd->GetChi2GloNrm());
+ ((TH2*)dest->At(kHBestInBranch+hcOffs))->Fill(htrPt,sd->GetOrdBranch());
+ ((TH2*)dest->At(kHBestInCand+hcOffs))->Fill(htrPt, sd->GetOrdCand());
+ //
+ if (dest==fCHistoArrFake && !sd->IsFake()) continue; // for the fake seeds fill only fake clusters part
+ //
+ ((TH2*)dest->At(kHResY+hcOffs))->Fill(htrPt,sd->GetResidY());
+ ((TH2*)dest->At(kHResZ+hcOffs))->Fill(htrPt,sd->GetResidZ());
+ ((TH2*)dest->At(kHResYP+hcOffs))->Fill(htrPt,sd->GetPullY());
+ ((TH2*)dest->At(kHResZP+hcOffs))->Fill(htrPt,sd->GetPullZ());
+ ((TH2*)dest->At(kHChi2Cl+hcOffs))->Fill(htrPt,sd->GetChi2Cl());
+ //
+ } while((sd=(AliITSUSeed*)sd->GetParent()));
+ //
+ ((TH2*)dest->At(kHChiMatch))->Fill(htrPt,winner->GetChi2ITSTPC());
+ ((TH2*)dest->At(kHChiITSSA))->Fill(htrPt,winner->GetChi2ITSSA());
+ }
+ //
+#endif
+ //
+ CheckClusterSharingConflicts(hyp);
+ return hyp->GetWinner(); // winner might change of disappear after resolving conflicts
+ //
}
//______________________________________________________________________________
void AliITSUTrackerGlo::FinalizeHypotheses()
{
// select winner for each hypothesis, remove cl. sharing conflicts
- AliInfo("TODO");
+ AliCodeTimerAuto("",0);
+ AliDebug(2,"TODO");
//
int nh = fHypStore.GetEntriesFast();
- for (int ih=0;ih<nh;ih++) {
- AliITSUTrackHyp* hyp = (AliITSUTrackHyp*) fHypStore.UncheckedAt(ih);
- if (!hyp || !hyp->DefineWinner()) continue; // TODO
- CookMCLabel(hyp);
- UpdateESDTrack(hyp,AliESDtrack::kITSin);
- }
+ for (int ih=0;ih<nh;ih++) FinalizeHypothesis(GetTrackHyp(ih));
+ //
+ for (int ih=0;ih<nh;ih++) UpdateESDTrack(GetTrackHyp(ih),AliESDtrack::kITSin);
+ //
+}
+//______________________________________________________________________________
+void AliITSUTrackerGlo::CheckClusterSharingConflicts(AliITSUTrackHyp* hyp)
+{
+ // remove eventual cluster sharing conflicts
+ AliITSUSeed* winner = 0;
+ if (!(winner=hyp->GetWinner())) return;
+ UShort_t idH = (UShort_t)hyp->GetUniqueID()+1;
+ int lrID,clID;
+ AliITSUSeed* winSD=winner;
+ do {
+ if ( (clID=winSD->GetLrCluster(lrID))<0 ) continue;
+ AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
+ Int_t refID = cl->GetRecoInfo(); // was it already referred by some track (with refID-1 if >0)
+ if (!refID) {cl->SetRecoInfo(idH); continue;} // if not, refer from track to cluster
+ if (refID==idH) continue; // ignore reference to itself
+ // the cluster is already used by other track, need to resolve the conflict
+ AliITSUTrackHyp* hypC = GetTrackHyp(refID-1); // competitor
+ // printf("Ref to %d (%p) from %d Cl %d %d\n",refID-1,hypC, idH-1,clID,lrID);
+ AliITSUSeed* winnerC = hypC->GetWinner();
+ if (!winnerC) {
+ printf("Missing winner, ERROR\n");
+ continue;
+ }
+ //
+ if (AliDebugLevelClass()>1) { //------------------------ DEBUG -------------------
+ AliInfo(Form("Shared cl#%4d on Lr:%d: Hyp%3d/Q:%6.3f vs Hyp%3d/Q:%6.3f",
+ clID,lrID,idH-1,winner->GetQualityVar(),refID-1,winnerC->GetQualityVar()));
+ // dump winner of hyp
+ printf("#%4d Pt:%.3f Chi:%6.3f/%6.3f/%6.3f Ncl:%d MCits%+5d MCtpc:%+5d |",
+ idH-1,winner->Pt(),winner->GetChi2GloNrm(),winner->GetChi2ITSTPC(),winner->GetChi2ITSSA(),
+ winner->GetNLayersHit(),hyp->GetITSLabel(),hyp->GetESDTrack()->GetTPCLabel());
+ int prevL=-1;
+ AliITSUSeed* sd = winner;
+ do {
+ int lrs;
+ int clIDt = sd->GetLrCluster(lrs);
+ if (clIDt<0) continue;
+ while( lrs>++prevL ) printf("%4s ","----");
+ printf("%4d (%5.1f)",clIDt,sd->GetChi2Cl());
+ } while ((sd=(AliITSUSeed*)sd->GetParent()));
+ printf("|\n");
+ // dump winner of hypC
+ printf("#%4d Pt:%.3f Chi:%6.3f/%6.3f/%6.3f Ncl:%d MCits%+5d MCtpc:%+5d |",
+ refID-1,winnerC->Pt(),winnerC->GetChi2GloNrm(),winnerC->GetChi2ITSTPC(),winnerC->GetChi2ITSSA(),
+ winnerC->GetNLayersHit(),hypC->GetITSLabel(),hypC->GetESDTrack()->GetTPCLabel());
+ prevL=-1;
+ sd = winnerC;
+ do {
+ int lrs;
+ int clIDt = sd->GetLrCluster(lrs);
+ if (clIDt<0) continue;
+ while( lrs>++prevL ) printf("%4s ","----");
+ printf("%4d (%5.1f)",clIDt,sd->GetChi2Cl());
+ } while ((sd=(AliITSUSeed*)sd->GetParent()));
+ printf("|\n");
+ } //-------------------------------------------------- DEBUG -------------------
+ //
+ if (winner->GetQualityVar()<winnerC->GetQualityVar()) { // current tracks is better then competitor track
+ FlagSeedClusters(winnerC,kFALSE,refID); // unflag cluster usage by loser
+ cl->SetRecoInfo(idH);
+ //
+ //if ( hypC->GetLabel()>=0) AliInfo(Form("KILLING CORRECT: %d",refID-1));
+ winnerC->Kill();
+ if (hypC->DefineWinner()) { // find new winner instead of suppressed candidate
+ CookMCLabel(hypC);
+ CheckClusterSharingConflicts(hypC); // and check its sharing conflicts again
+ if (winner->IsKilled()) break; // the current winner might have been killed during check of new winner of hypC!
+ }
+ }
+ else { // competitor hypC is better than the hyp
+ FlagSeedClusters(winner,kFALSE,idH); // unflag cluster usage by loser
+ winner->Kill();
+ //if ( hyp->GetLabel()>=0) AliInfo(Form("KILLING CORRECT: %d",idH-1));
+ if (hyp->DefineWinner()) {
+ CookMCLabel(hyp);
+ CheckClusterSharingConflicts(hyp);
+ }
+ break;
+ }
+ //
+ } while ((winSD=(AliITSUSeed*)winSD->GetParent()));
+ //
+ return;
}
//______________________________________________________________________________
void AliITSUTrackerGlo::UpdateESDTrack(AliITSUTrackHyp* hyp,Int_t flag)
{
// update ESD track with current best hypothesis
+ if (!hyp) return;
AliESDtrack* esdTr = hyp->GetESDTrack();
if (!esdTr) return;
AliITSUSeed* win = hyp->GetWinner();
- if (!win) return;
+ if (!win || win->IsKilled()) return;
+ double chiSav;
+ //
switch (flag) {
case AliESDtrack::kITSin:
esdTr->UpdateTrackParams(hyp,flag); // update kinematics
+ fCountITSin++;
// TODO: set cluster info
break;
//
case AliESDtrack::kITSout:
+ // here the stored chi2 will correspond to backward ITS-SA fit
esdTr->UpdateTrackParams(hyp,flag); // update kinematics
+ fCountITSout++;
// TODO: avoid setting friend
break;
//
case AliESDtrack::kITSrefit:
+
+ // at the moment fill as a chi2 the TPC-ITS matching chi2
+ chiSav = hyp->GetChi2();
+ hyp->SetChi2(win->GetChi2ITSTPC());
esdTr->UpdateTrackParams(hyp,flag); // update kinematics
+ hyp->SetChi2(chiSav);
+ fCountITSrefit++;
// TODO: avoid setting cluster info
break;
default:
AliFatal(Form("Unknown flag %d",flag));
}
//
+ esdTr->SetITSLabel(hyp->GetITSLabel());
// transfer module indices
// TODO
}
//______________________________________________________________________________
-Bool_t AliITSUTrackerGlo::RefitTrack(AliITSUTrackHyp* trc, Double_t rDest, Bool_t stopAtLastCl)
+Double_t AliITSUTrackerGlo::RefitTrack(AliExternalTrackParam* trc, Double_t rDest, Int_t stopCond)
{
- // refit track till radius rDest
- AliITSUTrackHyp tmpTr;
+ // refit track till radius rDest. The cluster,mass info is taken from fCurrHyp (and its winner seed)
+ // if stopCond<0 : propagate till last cluster then stop
+ // if stopCond==0: propagate till last cluster then try to go till limiting rDest, don't mind if fail
+ // if stopCond>0 : rDest must be reached
//
double rCurr = Sqrt(trc->GetX()*trc->GetX() + trc->GetY()*trc->GetY());
int dir,lrStart,lrStop;
//
dir = rCurr<rDest ? 1 : -1;
lrStart = fITS->FindFirstLayerID(rCurr,dir);
- lrStop = fITS->FindLastLayerID(rDest,dir);
- printf("Refit %d: Lr:%d (%f) -> Lr:%d (%f)\n",dir,lrStart,rCurr, lrStop,rDest);
- printf("Before refit: "); trc->AliExternalTrackParam::Print();
+ lrStop = fITS->FindLastLayerID(rDest,dir); // lr id before which we have to stop
+ //
+ if (AliDebugLevelClass()>2) {
+ printf("Refit %d: Lr:%d (%f) -> Lr:%d (%f)\n",dir,lrStart,rCurr, lrStop,rDest);
+ printf("Before refit: "); trc->Print();
+ }
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));
//
- int ncl = trc->FetchClusterInfo(fClInfo);
- fCurrMass = trc->GetMass();
- tmpTr.AliKalmanTrack::operator=(*trc);
- tmpTr.SetChi2(0);
+ Int_t clInfo[2*AliITSUAux::kMaxLayers];
+ Int_t nCl = fCurrHyp->FetchClusterInfo(clInfo);
+ fCurrMass = fCurrHyp->GetMass();
+ AliExternalTrackParam tmpTr(*trc);
+ double chi2 = 0;
int iclLr[2],nclLr,clCount=0;
//
- for (int ilr=lrStart;ilr!=lrStop;ilr+=dir) {
+ int lrStop1 = lrStop+1;
+ for (int ilr=lrStart;ilr!=lrStop1;ilr+=dir) {
AliITSURecoLayer* lr = fITS->GetLayer(ilr);
if ( dir*(rCurr-lr->GetR(dir))>0) continue; // this layer is already passed
int ilrA2,ilrA = lr->GetActiveID();
// passive layer or active w/o hits will be traversed on the way to next cluster
- if (!lr->IsActive() || fClInfo[ilrA2=(ilrA<<1)]<0) continue;
- //
- if (ilr!=lrStart && !TransportToLayer(&tmpTr,lrStart,ilr)) {
- AliInfo(Form("Failed to transport %d -> %d\n",lrStart,ilr));
- return kFALSE; // go to the entrance to the layer
- }
- lrStart = ilr;
+ if (!lr->IsActive() || clInfo[ilrA2=(ilrA<<1)]<0) continue;
//
// select the order in which possible 2 clusters (in case of the overlap) will be traversed and fitted
nclLr=0;
if (dir>0) { // clusters are stored in increasing radius order
- iclLr[nclLr++]=fClInfo[ilrA2++];
- if (fClInfo[ilrA2]>=0) iclLr[nclLr++]=fClInfo[ilrA2];
+ iclLr[nclLr++]=clInfo[ilrA2++];
+ if (clInfo[ilrA2]>=0) iclLr[nclLr++]=clInfo[ilrA2];
}
else {
- if ( fClInfo[ilrA2+1]>=0 ) iclLr[nclLr++]=fClInfo[ilrA2+1];
- iclLr[nclLr++]=fClInfo[ilrA2];
+ if ( clInfo[ilrA2+1]>=0 ) iclLr[nclLr++]=clInfo[ilrA2+1];
+ iclLr[nclLr++]=clInfo[ilrA2];
}
//
+ Bool_t transportedToLayer = kFALSE;
for (int icl=0;icl<nclLr;icl++) {
AliITSUClusterPix* clus = (AliITSUClusterPix*)lr->GetCluster(iclLr[icl]);
AliITSURecoSens* sens = lr->GetSensorFromID(clus->GetVolumeId());
if (!tmpTr.Rotate(sens->GetPhiTF())) {
- AliInfo(Form("Failed on rotate to %f",sens->GetPhiTF()));
- return kFALSE;
+ AliESDtrack* trESD = fCurrHyp->GetESDTrack();
+ AliDebug(2,Form("Failed on rotate to %f | ESDtrack#%d (MClb:%d)",sens->GetPhiTF(),trESD->GetID(),trESD->GetTPCLabel()));
+ return -1;
+ }
+ //
+ double xClus = sens->GetXTF()+clus->GetX();
+ if (!transportedToLayer) {
+ if (ilr!=lrStart && !TransportToLayerX(&tmpTr,lrStart,ilr,xClus)) {
+ AliESDtrack* trESD = fCurrHyp->GetESDTrack();
+ AliDebug(2,Form("Failed to transport %d -> %d | ESDtrack#%d (MClb:%d)\n",lrStart,ilr,trESD->GetID(),trESD->GetTPCLabel()));
+ return -1; // go to the entrance to the layer
+ }
+ lrStart = ilr;
+ transportedToLayer = kTRUE;
+ }
+ //
+ if (AliDebugLevelClass()>1) {
+ AliDebug(2,Form("Propagate to cl:%d on lr %d Need to go %.4f -> %.4f",icl,ilrA,tmpTr.GetX(),xClus));
+ }
+ //
+ if (!PropagateSeed(&tmpTr,xClus,fCurrMass)) {
+ AliESDtrack* trESD = fCurrHyp->GetESDTrack();
+ AliDebug(2,Form("Failed on propagate to %f (dir=%d) | ESDtrack#%d (MClb:%d)",xClus,dir,trESD->GetID(),trESD->GetTPCLabel()));
+ return -1;
+ }
+ //
+ Double_t p[2]={clus->GetY(), clus->GetZ()};
+ Double_t cov[3]={clus->GetSigmaY2(), clus->GetSigmaYZ(), clus->GetSigmaZ2()};
+ double chi2cl = tmpTr.GetPredictedChi2(p,cov);
+ chi2 += chi2cl;
+ //
+#ifdef _ITSU_TUNING_MODE_
+ TObjArray* dest = trc->GetLabel()>=0 ? fCHistoArrCorr : fCHistoArrFake;
+ if (dest && fTrackPhase>kClus2Tracks) {
+ int hcOffs = (1+fTrackPhase)*kHistosPhase + ilrA;
+ //
+ double htrPt = tmpTr.Pt();
+ double dy = p[0]-tmpTr.GetY();
+ double dz = p[1]-tmpTr.GetZ();
+ ((TH2*)dest->At(kHResY+hcOffs))->Fill(htrPt,dy);
+ ((TH2*)dest->At(kHResZ+hcOffs))->Fill(htrPt,dz);
+ double errY = tmpTr.GetSigmaY2();
+ double errZ = tmpTr.GetSigmaZ2();
+ if (errY>0) ((TH2*)dest->At(kHResYP+hcOffs))->Fill(htrPt,dy/Sqrt(errY));
+ if (errZ>0) ((TH2*)dest->At(kHResZP+hcOffs))->Fill(htrPt,dz/Sqrt(errZ));
+ ((TH2*)dest->At(kHChi2Cl+hcOffs))->Fill(htrPt,chi2cl);
}
- //printf("Refit cl:%d on lr %d Need to go %.4f -> %.4f\n",icl,ilrA,tmpTr.GetX(),sens->GetXTF()+clus->GetX());
- if (!PropagateSeed(&tmpTr,sens->GetXTF()+clus->GetX(),fCurrMass)) {
- AliInfo(Form("Failed on propagate to %f",sens->GetXTF()+clus->GetX()));
- return kFALSE;
+#endif
+ //
+ if ( !tmpTr.Update(p,cov) ) {
+ AliESDtrack* trESD = fCurrHyp->GetESDTrack();
+ AliDebug(2,Form("Failed on Update | ESDtrack#%d (MClb:%d)",trESD->GetID(),trESD->GetTPCLabel()));
+ return -1;
}
- if (!tmpTr.Update(clus)) {
- AliInfo(Form("Failed on Update"));
- return kFALSE;
+ if (AliDebugLevelClass()>1) {
+ printf("AfterRefit: "); tmpTr.Print();
+ }
+ if (++clCount==nCl && stopCond<0) {
+ *trc = tmpTr;
+ return chi2; // it was requested to not propagate after last update
}
- //printf("AfterRefit: "); tmpTr.AliExternalTrackParam::Print();
- if (stopAtLastCl && ++clCount==ncl) return kTRUE; // it was requested to not propagate after last update
}
//
}
// Still, try to go as close as possible to rDest.
//
if (lrStart!=lrStop) {
- //printf("Going to last layer %d -> %d\n",lrStart,lrStop);
if (!TransportToLayer(&tmpTr,lrStart,lrStop)) {
- AliInfo(Form("Failed on TransportToLayer %d->%d",lrStart,lrStop));
- return kTRUE;
+ AliDebug(1,Form("Failed on TransportToLayer %d->%d | ESDtrack#%d (MClb:%d), X=%f",lrStart,lrStop,fCurrESDtrack->GetID(),fCurrESDtrMClb,tmpTr.GetX()));
+ return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
}
if (!GoToExitFromLayer(&tmpTr,fITS->GetLayer(lrStop),dir)) {
- AliFatal(Form("Failed on GoToExitFromLayer %d",lrStop));
- return kTRUE; // go till the exit from layer
+ AliDebug(1,Form("Failed on GoToExitFromLayer %d | ESDtrack#%d (MClb:%d), X=%f",lrStop,fCurrESDtrack->GetID(),fCurrESDtrMClb,tmpTr.GetX()));
+ return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
}
- //
- //printf("On exit from last layer\n");
- tmpTr.AliExternalTrackParam::Print();
- // go to the destination radius
- if (!tmpTr.GetXatLabR(rDest,rDest,GetBz(),dir)) return kTRUE;
- if (!PropagateSeed(&tmpTr,rDest,fCurrMass, 100, kFALSE)) return kTRUE;
- }
- trc->AliKalmanTrack::operator=(tmpTr);
- printf("After refit (now at lr %d): ",lrStart); trc->AliExternalTrackParam::Print();
- return kTRUE;
+ }
+ // go to the destination radius. Note that here we don't select direction to avoid precision problems
+ if (!tmpTr.GetXatLabR(rDest,rDest,GetBz(),0) || !PropagateSeed(&tmpTr,rDest,fCurrMass, 100, kFALSE)) {
+ return (stopCond>0) ? -chi2 : chi2; // rDest was obligatory
+ }
+ *trc=tmpTr;
+ if (AliDebugLevelClass()>2) {
+ printf("After refit (now at lr %d): ",lrStop); trc->Print();
+ }
+ return chi2;
}
//__________________________________________________________________
seed = (AliITSUSeed*)seed->GetParent();
} // loop over clusters
//
+ AliESDtrack* esdTr = hyp->GetESDTrack();
+ int tpcLab = esdTr ? Abs(esdTr->GetTPCLabel()) : -kDummyLabel;
if (nCl && nLab) {
int maxLab=0,nTPCok=0;
- AliESDtrack* esdTr = hyp->GetESDTrack();
- int tpcLab = esdTr ? Abs(esdTr->GetTPCLabel()) : -kDummyLabel;
for (int ilb=nLab;ilb--;) {
int st = lbStat[ilb];
if (lbStat[maxLab]<st) maxLab=ilb;
}
hyp->SetFakeRatio(1.-float(nTPCok)/nCl);
hyp->SetLabel( nTPCok==nCl ? tpcLab : -tpcLab);
- hyp->SetITSLabel( lbStat[maxLab]==nCl ? lbStat[maxLab] : -lbStat[maxLab]); // winner label
+ hyp->SetITSLabel( lbStat[maxLab]==nCl ? lbID[maxLab] : -lbID[maxLab]); // winner label
return;
}
//
hyp->SetFakeRatio(-1.);
- hyp->SetLabel( kDummyLabel );
+ hyp->SetLabel( -tpcLab );
hyp->SetITSLabel( kDummyLabel );
}
+
+//__________________________________________________________________
+Bool_t AliITSUTrackerGlo::AddSeedBranch(AliITSUSeed* seed)
+{
+ // add new prolongation branch for the seed to temporary array. The seeds are sorted according to Compare f-n
+ if (fNCandidatesAdded+fNBranchesAdded>=fLayerMaxCandidates ) {
+ AliInfo(Form("Number of candidates at layer %d for current seed reached %d, increasing buffer",fCurrActLrID,fLayerMaxCandidates));
+ fLayerMaxCandidates = 2*(fLayerMaxCandidates+1);
+ AliITSUSeed** tmpArr = fLayerCandidates;
+ fLayerCandidates = new AliITSUSeed*[fLayerMaxCandidates];
+ memcpy(fLayerCandidates,tmpArr,(fNCandidatesAdded+fNBranchesAdded)*sizeof(AliITSUSeed*));
+ delete tmpArr; // delete only array, not objects
+ }
+ AliITSUSeed** branches = &fLayerCandidates[fNCandidatesAdded]; // note: fNCandidatesAdded is incremented after adding all branches of current seed
+ int slot=fNBranchesAdded++;
+ for (int slotF=slot;slotF--;) { // slotF is always slot-1
+ AliITSUSeed* si = branches[slotF];
+ if (si->Compare(seed)<0) break; // found the last seed with better quality
+ // otherwise, shift the worse seed to the next slot
+ branches[slot] = si;
+ slot = slotF; // slot should be slotF+1
+ }
+ // if needed, move worse seeds
+ branches[slot] = seed;
+ return kTRUE;
+ //
+}
+
+//__________________________________________________________________
+void AliITSUTrackerGlo::ValidateAllowedBranches(Int_t acceptMax)
+{
+ // keep allowed number of branches for current seed and disable extras
+ AliCodeTimerAuto("",0);
+ int nb = Min(fNBranchesAdded,acceptMax);
+ // if (nb<fNBranchesAdded) printf("ValidateAllowedBranches: %d of %d (%d) on lr %d\n",nb,fNBranchesAdded,fNCandidatesAdded,ilr);
+ // disable unused branches
+ AliITSUSeed** branches = &fLayerCandidates[fNCandidatesAdded];
+ //
+#ifdef _ITSU_TUNING_MODE_
+ for (int ib=0;ib<fNBranchesAdded;ib++) branches[ib]->SetOrdBranch(ib);
+#endif
+ //
+ for (int ib=nb;ib<fNBranchesAdded;ib++) MarkSeedFree(branches[ib]);
+ fNCandidatesAdded += nb; // update total candidates counter
+ fNBranchesAdded = 0; // reset branches counter
+ //
+}
+
+//__________________________________________________________________
+void AliITSUTrackerGlo::ValidateAllowedCandidates(Int_t ilr, Int_t acceptMax)
+{
+ // transfer allowed number of branches to hypothesis container
+ //
+ AliCodeTimerAuto("",0);
+ // sort candidates in increasing order of chi2
+ static int lastSize = 0;
+ static int *index = 0;
+ static float *chi2 = 0;
+ if (fLayerMaxCandidates>lastSize) {
+ lastSize = fLayerMaxCandidates;
+ delete[] index;
+ delete[] chi2;
+ index = new int[lastSize];
+ chi2 = new float[lastSize];
+ }
+ for (int i=fNCandidatesAdded;i--;) chi2[i] = fLayerCandidates[i]->GetChi2GloNrm();
+ Sort(fNCandidatesAdded,chi2,index,kFALSE);
+ //
+ int nacc=0,nb=0;
+ if (ilr>0) { // just take 1st acceptMax candidates
+ nb = Min(fNCandidatesAdded,acceptMax);
+ for (int ib=0;ib<nb;ib++) AddProlongationHypothesis(fLayerCandidates[index[ib]],ilr);
+ }
+ else { // for innermost layer test ITS SA backward chi2 and matching to TPC
+ AliITSUSeed* wn0 = fCurrHyp->GetWinner(); // in principle, should be NULL
+ for (nb=0;nb<fNCandidatesAdded;nb++) {
+ AliITSUSeed* sd = fLayerCandidates[index[nb]];
+ fCurrHyp->SetWinner(sd);
+ if (!CheckBackwardMatching(sd)) MarkSeedFree(sd); // discard bad backward tracks
+ else {
+ AddProlongationHypothesis(sd,ilr);
+ if (++nacc==acceptMax) {nb++; break;} // the rest will be discarded
+ }
+ }
+ fCurrHyp->SetWinner(wn0); // restore original winner (NULL?)
+ }
+ // disable unused candidates
+ for (int ib=nb;ib<fNCandidatesAdded;ib++) MarkSeedFree(fLayerCandidates[index[ib]]);
+ fNCandidatesAdded = 0; // reset candidates counter
+ //
+}
+
+//__________________________________________________________________
+Bool_t AliITSUTrackerGlo::CheckBackwardMatching(AliITSUSeed* seed)
+{
+ // check seed backward propagation chi2 and matching to TPC
+ double bz0 = GetBz();
+ double rDest = fITS->GetRITSTPCRef(); // reference radius for comparison
+ static AliExternalTrackParam trback;
+ trback = *seed;
+ trback.ResetCovariance(10000);
+ int ndf = seed->GetNLayersHit()*2-5;
+ double chi2sa = RefitTrack(&trback,rDest,1);
+ if (chi2sa<0) return kFALSE;
+ if (ndf>0) chi2sa /= ndf;
+ if (chi2sa>fCurrTrackCond->GetMaxITSSAChi2()) return kFALSE;
+ //
+ // relate to TPC track at outer layer
+ AliExternalTrackParam* tpcSeed = fCurrHyp->GetTPCSeed();
+ if (!trback.Rotate(tpcSeed->GetAlpha()) || !trback.PropagateParamOnlyTo(tpcSeed->GetX(),bz0)) {
+ if (AliDebugLevelClass()>+1 && !seed->ContainsFake()) {
+ AliInfo(Form("Failed to propagate seed to TPC track @ X:%.3f Alpha:%.3f",tpcSeed->GetX(),tpcSeed->GetAlpha()));
+ seed->Print("etp");
+ trback.Print();
+ }
+ return kFALSE;
+ }
+ double chi2Match = trback.GetPredictedChi2(tpcSeed)/5;
+ if (chi2Match>fCurrTrackCond->GetMaxITSTPCMatchChi2()) return kFALSE;
+ //
+ seed->SetChi2ITSSA(chi2sa);
+ seed->SetChi2ITSTPC(chi2Match);
+ return kTRUE;
+}
+
+//__________________________________________________________________
+void AliITSUTrackerGlo::SaveReducedHypothesesTree(AliITSUTrackHyp* dest)
+{
+ // remove those hypothesis seeds which dont lead to candidates at final layer
+ //
+ // 1st, flag the seeds to save
+ int lr0 = 0;
+ for (int isd=0;isd<fCurrHyp->GetNSeeds(lr0);isd++) {
+ AliITSUSeed* seed = fCurrHyp->RemoveSeed(lr0,isd);
+ if (!seed) continue;
+ seed->FlagTree(AliITSUSeed::kSave);
+ dest->AddSeed(seed,lr0);
+ }
+ for (int ilr=1;ilr<fNLrActive;ilr++) {
+ int nsd = fCurrHyp->GetNSeeds(ilr);
+ for (int isd=0;isd<nsd;isd++) {
+ AliITSUSeed* seed = fCurrHyp->RemoveSeed(ilr,isd);
+ if (!seed) continue; // already discarded or saved
+ if (seed->IsSaved()) dest->AddSeed(seed,ilr);
+ else MarkSeedFree(seed);
+ }
+ }
+ //
+ // AliInfo(Form("SeedsPool: %d, BookedUpTo: %d, free: %d",fSeedsPool.GetSize(),fSeedsPool.GetEntriesFast(),fNFreeSeeds));
+}
+
+//__________________________________________________________________
+void AliITSUTrackerGlo::FlagSplitClusters()
+{
+ // set special bit on split clusters using MC info
+ for (int ilr=fNLrActive;ilr--;) {
+ int nsplit=0;
+ AliITSURecoLayer* lr = fITS->GetLayerActive(ilr);
+ for (int isn=lr->GetNSensors();isn--;) {
+ AliITSURecoSens* sens = lr->GetSensor(isn);
+ int nCl = sens->GetNClusters();
+ if (!nCl) continue;
+ int cl0 = sens->GetFirstClusterId();
+ for (int icl=nCl;icl--;) {
+ AliITSUClusterPix *cl = (AliITSUClusterPix*)lr->GetCluster(cl0+icl);
+ for (int icl1=icl;icl1--;) {
+ AliITSUClusterPix *cl1 = (AliITSUClusterPix*)lr->GetCluster(cl0+icl1);
+ if (cl->HasCommonTrack(cl1)) {
+ if (!cl->IsSplit()) nsplit++;
+ if (!cl1->IsSplit()) nsplit++;
+ cl->SetSplit();
+ cl1->SetSplit();
+ }
+ }
+ }
+ }
+ AliInfo(Form("%4d out of %4d clusters are split",nsplit,lr->GetNClusters()));
+ }
+ //
+}
+
+//__________________________________________________________________
+Bool_t AliITSUTrackerGlo::ContainsSplitCluster(const AliITSUSeed* seed, Int_t maxSize)
+{
+ // check if the seed contains split cluster with size < maxSize
+ int lrID,clID;
+ if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
+ AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
+ if (cl->IsSplit() && cl->GetNPix()<maxSize ) return kTRUE;
+ }
+ return seed->GetParent() ? ContainsSplitCluster((AliITSUSeed*)seed->GetParent(),maxSize) : kFALSE;
+ //
+}
+
+//__________________________________________________________________
+void AliITSUTrackerGlo::PrintSeedClusters(const AliITSUSeed* seed, Option_t* option)
+{
+ // print seeds clusters
+ int lrID,clID;
+ if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
+ AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
+ cl->Print(option);
+ }
+ if (seed->GetParent()) PrintSeedClusters((AliITSUSeed*)seed->GetParent(), option);
+ //
+}
+
+//__________________________________________________________________
+void AliITSUTrackerGlo::FlagSeedClusters(const AliITSUSeed* seed, Bool_t flg, UShort_t ref)
+{
+ // mark used clusters
+ int lrID,clID;
+ while (seed) {
+ if ( (clID=seed->GetLrCluster(lrID))>=0 ) {
+ AliITSUClusterPix* cl = (AliITSUClusterPix*)fITS->GetLayerActive(lrID)->GetCluster(clID);
+ if (ref) { // do we need to set or delete cluster->track ref?
+ if (flg) {
+ if (!cl->GetRecoInfo()) cl->SetRecoInfo(ref); // set ref only if cluster already does not refer to other track, inc.counter
+ }
+ else {
+ if (cl->GetRecoInfo()==ref) cl->SetRecoInfo(0); // unset reference only if it refers to ref, decrease counter
+ }
+ }
+ }
+ seed = (AliITSUSeed*)seed->GetParent();
+ }
+ //
+}
+
+#ifdef _ITSU_TUNING_MODE_
+//__________________________________________________________________
+void AliITSUTrackerGlo::BookControlHistos(const char* pref)
+{
+ // book special control histos
+ TString prefS = pref;
+ prefS.ToLower();
+ TObjArray* dest = 0;
+ if (prefS=="corr") dest = fCHistoArrCorr = new TObjArray();
+ else if (prefS=="fake") dest = fCHistoArrFake = new TObjArray();
+ else {AliError(Form("Unknown histo set %s is requested",pref)); return;}
+ dest->SetOwner(kTRUE);
+ //
+ const int kNResDef=7;
+ const double kResDef[kNResDef]={0.05,0.05,0.3, 0.05,1,0.5,1.5};
+ const double ptMax=10;
+ const double plMax=10;
+ const double chiMax=100;
+ const int nptbins=50;
+ const int nresbins=400;
+ const int nplbins=50;
+ const int nchbins=200;
+ const int maxBr = 15;
+ const int maxCand = 200;
+ TString ttl;
+ for (int stp=0;stp<kNTrackingPhases;stp++) {
+ for (int ilr=0;ilr<fNLrActive;ilr++) {
+ int hoffs = (1+stp)*kHistosPhase + ilr;
+ double mxdf = ilr>=kNResDef ? kResDef[kNResDef-1] : kResDef[ilr];
+ ttl = Form("S%d_residY%d_%s",stp,ilr,pref);
+ TH2F* hdy = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nresbins,-mxdf,mxdf);
+ dest->AddAtAndExpand(hdy,hoffs + kHResY);
+ hdy->SetDirectory(0);
+ //
+ ttl = Form("S%d_residYPull%d_%s",stp,ilr,pref);
+ TH2F* hdyp = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nplbins,-plMax,plMax);
+ dest->AddAtAndExpand(hdyp,hoffs + kHResYP);
+ hdyp->SetDirectory(0);
+ //
+ ttl = Form("S%d_residZ%d_%s",stp,ilr,pref);
+ TH2F* hdz = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nresbins,-mxdf,mxdf);
+ dest->AddAtAndExpand(hdz,hoffs + kHResZ);
+ hdz->SetDirectory(0);
+ //
+ ttl = Form("S%d_residZPull%d_%s",stp,ilr,pref);
+ TH2F* hdzp = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax,nplbins,-plMax,plMax);
+ hdzp->SetDirectory(0);
+ dest->AddAtAndExpand(hdzp,hoffs + kHResZP);
+ //
+ ttl = Form("S%d_chi2Cl%d_%s",stp,ilr,pref);
+ TH2F* hchi = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
+ hchi->SetDirectory(0);
+ dest->AddAtAndExpand(hchi,hoffs + kHChi2Cl);
+ //
+ ttl = Form("S%d_chi2Nrm%d_%s",stp,ilr,pref);
+ TH2F* hchiN = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
+ hchiN->SetDirectory(0);
+ dest->AddAtAndExpand(hchiN,hoffs + kHChi2Nrm);
+ //
+ if (stp==0) { // these histos make sense only for clusters2tracks stage
+ ttl = Form("S%d_bestInBranch%d_%s",stp,ilr,pref);
+ TH2* hnbr = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, maxBr,-0.5,maxBr-0.5);
+ hnbr->SetDirectory(0);
+ dest->AddAtAndExpand(hnbr,hoffs + kHBestInBranch);
+ //
+ ttl = Form("S%d_bestInCands%d_%s",stp,ilr,pref);
+ TH2* hncn = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, maxCand,-0.5,maxCand-0.5);
+ hncn->SetDirectory(0);
+ dest->AddAtAndExpand(hncn,hoffs + kHBestInCand);
+ //
+ }
+ }
+ }
+ // custom histos
+ //
+ TH2* hchiMatch = 0;
+ ttl = Form("Chi2Match_%s",pref);
+ hchiMatch = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins,0.,chiMax);
+ hchiMatch->SetDirectory(0);
+ dest->AddAtAndExpand(hchiMatch,kHChiMatch);
+ //
+ TH2* hchiSA = 0;
+ ttl = Form("Chi2ITSSA_%s",pref);
+ hchiSA = new TH2F(ttl.Data(),ttl.Data(),nptbins,0,ptMax, nchbins/2,0.,chiMax/2);
+ hchiSA->SetDirectory(0);
+ dest->AddAtAndExpand(hchiSA,kHChiITSSA);
+ //
+}
+#endif