15#define GPUCA_CADEBUG 0
16#define GPUCA_MERGE_LOOPER_MC 0
20#if !defined(GPUCA_GPUCODE) && (defined(GPUCA_MERGER_BY_MC_LABEL) || defined(GPUCA_CADEBUG_ENABLED) || GPUCA_MERGE_LOOPER_MC)
25#ifndef GPUCA_GPUCODE_DEVICE
40#include "GPUDefParametersRuntime.h"
62static constexpr int32_t kMaxParts = 400;
67using namespace gputpcgmmergertypes;
94 GPUd() bool operator()(const int32_t
aa, const int32_t
bb)
99 return a.CCE() >
b.CCE();
101 if (
a.Legs() !=
b.Legs()) {
102 return a.Legs() >
b.Legs();
105 if (
a.NClusters() !=
b.NClusters()) {
106 return a.NClusters() > b.NClusters();
107 }
if (CAMath::Abs(
a.GetParam().GetQPt()) != CAMath::Abs(
b.GetParam().GetQPt())) {
108 return CAMath::Abs(a.GetParam().GetQPt()) > CAMath::Abs(b.GetParam().GetQPt());
109 }
if (
a.GetParam().GetY() !=
b.GetParam().GetY()) {
110 return a.GetParam().GetY() > b.GetParam().GetY();
114 return a.NClusters() >
b.NClusters();
122 GPUd() bool operator()(const int32_t
aa, const int32_t
bb)
128 return CAMath::Abs(
a.GetParam().GetQPt()) > CAMath::Abs(
b.GetParam().GetQPt());
129 }
if (
a.GetParam().GetY() !=
b.GetParam().GetY()) {
130 return a.GetParam().GetY() >
b.GetParam().GetY();
132 return a.GetParam().GetZ() >
b.GetParam().GetZ();
134 return CAMath::Abs(
a.GetParam().GetQPt()) > CAMath::Abs(
b.GetParam().GetQPt());
142 return GPUCA_DETERMINISTIC_CODE(CAMath::Abs(
a.refz) != CAMath::Abs(
b.refz) ? CAMath::Abs(
a.refz) < CAMath::Abs(
b.refz) :
a.id <
b.id, CAMath::Abs(
a.refz) < CAMath::Abs(
b.refz));
163 mNextSectorInd[mid] = 0;
164 mPrevSectorInd[0] = mid;
165 mNextSectorInd[last] =
NSECTORS / 2;
166 mPrevSectorInd[
NSECTORS / 2] = last;
170#if !defined(GPUCA_GPUCODE) && (defined(GPUCA_MERGER_BY_MC_LABEL) || defined(GPUCA_CADEBUG_ENABLED) || GPUCA_MERGE_LOOPER_MC)
173void GPUTPCGMMerger::CheckMergedTracks()
175 std::vector<bool> trkUsed(SectorTrackInfoLocalTotal());
176 for (int32_t
i = 0;
i < SectorTrackInfoLocalTotal();
i++) {
180 for (int32_t itr = 0; itr < SectorTrackInfoLocalTotal(); itr++) {
182 if (track.PrevSegmentNeighbour() >= 0) {
185 if (track.PrevNeighbour() >= 0) {
191 int32_t iTrk = tr - mSectorTrackInfos;
193 GPUError(
"FAILURE: double use");
195 trkUsed[iTrk] =
true;
197 int32_t jtr = tr->NextSegmentNeighbour();
199 tr = &(mSectorTrackInfos[jtr]);
202 jtr = trbase->NextNeighbour();
204 trbase = &(mSectorTrackInfos[jtr]);
206 if (tr->PrevSegmentNeighbour() >= 0) {
215 for (int32_t
i = 0;
i < SectorTrackInfoLocalTotal();
i++) {
216 if (trkUsed[
i] ==
false) {
217 GPUError(
"FAILURE: trk missed");
233template <
class T,
class S>
234int64_t GPUTPCGMMerger::GetTrackLabelA(
const S& trk)
const
238 if constexpr (std::is_same_v<S, GPUTPCGMBorderTrack&>) {
239 sectorTrack = &mSectorTrackInfos[trk.TrackID()];
240 nClusters = sectorTrack->OrigTrack()->NHits();
244 auto acc =
GPUTPCTrkLbl<false, GPUTPCTrkLbl_ret>(resolveMCLabels<T>(GetConstantMem()->ioPtrs.clustersNative ? GetConstantMem()->ioPtrs.clustersNative->clustersMCTruth : nullptr, GetConstantMem()->ioPtrs.mcLabelsTPC), 0.5f);
247 if constexpr (std::is_same_v<S, GPUTPCGMBorderTrack&>) {
248 const GPUTPCTracker& tracker = GetConstantMem()->tpcTrackers[sectorTrack->Sector()];
249 const GPUTPCHitId& ic = tracker.TrackHits()[sectorTrack->OrigTrack()->FirstHitID() +
i];
250 id = tracker.Data().ClusterDataIndex(tracker.Data().Row(ic.RowIndex()), ic.HitIndex()) + GetConstantMem()->ioPtrs.clustersNative->clusterOffset[sectorTrack->Sector()][0];
252 id = mClusters[trk.FirstClusterRef() +
i].
num;
256 return acc.computeLabel().id;
260int64_t GPUTPCGMMerger::GetTrackLabel(
const S& trk)
const
262#ifdef GPUCA_TPC_GEOMETRY_O2
263 if (GetConstantMem()->ioPtrs.clustersNative->clustersMCTruth) {
264 return GetTrackLabelA<o2::dataformats::ConstMCTruthContainerView<o2::MCCompLabel>,
S>(trk);
268 return GetTrackLabelA<AliHLTTPCClusterMCLabel, S>(trk);
275void GPUTPCGMMerger::PrintMergeGraph(
const GPUTPCGMSectorTrack* trk, std::ostream& out)
const
278 while (trk->PrevSegmentNeighbour() >= 0) {
279 trk = &mSectorTrackInfos[trk->PrevSegmentNeighbour()];
282 while (trk->PrevNeighbour() >= 0) {
283 trk = &mSectorTrackInfos[trk->PrevNeighbour()];
286 int32_t nextId = trk - mSectorTrackInfos;
287 out <<
"Graph of track " << (orgTrack - mSectorTrackInfos) <<
"\n";
288 while (nextId >= 0) {
289 trk = &mSectorTrackInfos[nextId];
290 if (trk->PrevSegmentNeighbour() >= 0) {
291 out <<
"TRACK TREE INVALID!!! " << trk->PrevSegmentNeighbour() <<
" --> " << nextId <<
"\n";
293 out << (trk == orgTower ?
"--" :
" ");
294 while (nextId >= 0) {
296 if (trk != trk2 && (trk2->PrevNeighbour() >= 0 || trk2->NextNeighbour() >= 0)) {
297 out <<
" (TRACK TREE INVALID!!! " << trk2->PrevNeighbour() <<
" <-- " << nextId <<
" --> " << trk2->NextNeighbour() <<
") ";
300 snprintf(tmp, 128,
" %s%5d(%5.2f)", trk2 == orgTrack ?
"!" :
" ", nextId, trk2->QPt());
302 nextId = trk2->NextSegmentNeighbour();
305 nextId = trk->NextNeighbour();
325 mBorder[
iSector] = mBorderMemory + 2 * nTracks;
327 mBorderRange[
iSector] = mBorderRangeMemory + 2 * nTracks;
335 memMax = (
void*)std::max((
size_t)mem, (size_t)memMax);
339 memMax = (
void*)std::max((
size_t)mem, (size_t)memMax);
343 memMax = (
void*)std::max((
size_t)mem, (size_t)memMax);
346 memMax = (
void*)std::max((
size_t)mem, (size_t)memMax);
360 if (mergerSortTracks) {
388 mClusterStateExt =
nullptr;
438 mNTotalSectorTracks = 0;
440 mNMaxSingleSectorTracks = 0;
443 mNTotalSectorTracks += ntrk;
445 if (mNMaxSingleSectorTracks < ntrk) {
446 mNMaxSingleSectorTracks = ntrk;
450 if (CAMath::Abs(
Param().polynomialField.GetNominalBz()) < (gpu_common_constants::kZeroFieldCut * gpu_common_constants::kCLight)) {
463 mNMaxClusters = mNClusters;
465 mNMaxLooperMatches = mNMaxClusters / 4;
472 throw std::runtime_error(
"mNMaxSingleSectorTracks too small");
476 throw std::runtime_error(
"Must run also sector tracking");
485 const int32_t
n =
output ? mMemory->nMergedTracks : SectorTrackInfoLocalTotal();
486 for (int32_t
i = iBlock * nThreads + iThread;
i <
n;
i += nThreads * nBlocks) {
494 prop.SetMaterialTPC();
496 prop.SetSeedingErrors(
true);
497 prop.SetFitInProjections(
false);
498 prop.SetPolynomialField(&
Param().polynomialField);
500 trk.X() = inTrack->Param().GetX();
501 trk.
Y() = inTrack->Param().GetY();
502 trk.
Z() = inTrack->Param().GetZ();
503 trk.SinPhi() = inTrack->Param().GetSinPhi();
504 trk.DzDs() = inTrack->Param().GetDzDs();
505 trk.QPt() = inTrack->Param().GetQPt();
506 trk.TZOffset() =
Param().par.earlyTpcTransform ? inTrack->Param().GetZOffset() : GetConstantMem()->calibObjects.fastTransformHelper->getCorrMap()->convZOffsetToVertexTime(sector, inTrack->Param().GetZOffset(),
Param().continuousMaxTimeBin);
507 trk.ShiftZ(
this, sector, sectorTrack.ClusterZT0(), sectorTrack.ClusterZTN(), inTrack->Param().GetX(), inTrack->Param().GetX());
508 sectorTrack.SetX2(0.f);
509 for (int32_t way = 0; way < 2; way++) {
511 prop.SetFitInProjections(
true);
512 prop.SetPropagateBzOnly(
true);
514 trk.ResetCovariance();
515 prop.SetTrack(&trk,
alpha);
516 int32_t
start = way ? inTrack->NHits() - 1 : 0;
517 int32_t
end = way ? 0 : (inTrack->NHits() - 1);
518 int32_t incr = way ? -1 : 1;
522 const GPUTPCTracker& tracker = GetConstantMem()->tpcTrackers[sector];
523 const GPUTPCHitId& ic = tracker.TrackHits()[inTrack->FirstHitID() +
i];
524 int32_t clusterIndex = tracker.Data().ClusterDataIndex(tracker.Data().Row(ic.RowIndex()), ic.HitIndex());
526 const ClusterNative& cl = GetConstantMem()->ioPtrs.clustersNative->clustersLinear[GetConstantMem()->ioPtrs.clustersNative->clusterOffset[sector][0] + clusterIndex];
527 flags = cl.getFlags();
528 if (
Param().
par.earlyTpcTransform) {
529 x = tracker.Data().ClusterData()[clusterIndex].x;
530 y = tracker.Data().ClusterData()[clusterIndex].y;
531 z = tracker.Data().ClusterData()[clusterIndex].
z - trk.TZOffset();
533 GetConstantMem()->calibObjects.fastTransformHelper->Transform(sector,
row, cl.getPad(), cl.getTime(),
x,
y,
z, trk.TZOffset());
535 if (prop.PropagateToXAlpha(
x,
alpha,
true)) {
538 trk.ConstrainSinPhi();
539 if (prop.Update(
y,
z,
row,
Param(),
flags &
GPUTPCGMMergedTrackHit::clustererAndSharedFlags, 0,
nullptr,
false, sector, -1.f, 0.f, 0.f)) {
542 trk.ConstrainSinPhi();
545 sectorTrack.SetParam2(trk);
547 sectorTrack.Set(trk, inTrack,
alpha, sector);
555 const GPUTPCTracker& trk = GetConstantMem()->tpcTrackers[iSector];
556 const GPUTPCHitId& ic1 = trk.TrackHits()[sectorTr->FirstHitID()];
557 const GPUTPCHitId& ic2 = trk.TrackHits()[sectorTr->FirstHitID() + sectorTr->NHits() - 1];
558 int32_t clusterIndex1 = trk.Data().ClusterDataIndex(trk.Data().Row(ic1.RowIndex()), ic1.HitIndex());
559 int32_t clusterIndex2 = trk.Data().ClusterDataIndex(trk.Data().Row(ic2.RowIndex()), ic2.HitIndex());
560 if (
Param().
par.earlyTpcTransform) {
561 track.SetClusterZT(trk.Data().ClusterData()[clusterIndex1].
z, trk.Data().ClusterData()[clusterIndex2].
z);
563 const ClusterNative* cl = GetConstantMem()->ioPtrs.clustersNative->clustersLinear + GetConstantMem()->ioPtrs.clustersNative->clusterOffset[iSector][0];
564 track.SetClusterZT(cl[clusterIndex1].
getTime(), cl[clusterIndex2].
getTime());
570 mSectorTrackInfoIndex[
id] = mMemory->nUnpackedTracks;
573GPUd()
void GPUTPCGMMerger::UnpackSectorGlobal(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t iSector)
575 const GPUTPCTracker& trk = GetConstantMem()->tpcTrackers[iSector];
577 const GPUTPCTrack* sectorTr = mMemory->firstExtrapolatedTracks[iSector];
578 uint32_t nLocalTracks = trk.CommonMemory()->nLocalTracks;
579 uint32_t nTracks = *trk.NTracks();
580 for (uint32_t itr = nLocalTracks + iBlock * nThreads + iThread; itr < nTracks; itr += nBlocks * nThreads) {
581 sectorTr = &trk.Tracks()[itr];
582 int32_t localId = mTrackIDs[(sectorTr->LocalTrackId() >> 24) * mNMaxSingleSectorTracks + (sectorTr->LocalTrackId() & 0xFFFFFF)];
586 uint32_t myTrack = CAMath::AtomicAdd(&mMemory->nUnpackedTracks, 1u);
588 SetTrackClusterZT(track, iSector, sectorTr);
589 track.Set(
this, sectorTr,
alpha, iSector);
590 track.SetGlobalSectorTrackCov();
591 track.SetPrevNeighbour(-1);
592 track.SetNextNeighbour(-1);
593 track.SetNextSegmentNeighbour(-1);
594 track.SetPrevSegmentNeighbour(-1);
595 track.SetLocalTrackId(localId);
599GPUd()
void GPUTPCGMMerger::UnpackResetIds(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t iSector)
601 const GPUTPCTracker& trk = GetConstantMem()->tpcTrackers[iSector];
602 uint32_t nLocalTracks = trk.CommonMemory()->nLocalTracks;
603 for (uint32_t
i = iBlock * nThreads + iThread;
i < nLocalTracks;
i += nBlocks * nThreads) {
604 mTrackIDs[iSector * mNMaxSingleSectorTracks +
i] = -1;
608GPUd()
void GPUTPCGMMerger::RefitSectorTracks(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t iSector)
610 const GPUTPCTracker& trk = GetConstantMem()->tpcTrackers[iSector];
611 uint32_t nLocalTracks = trk.CommonMemory()->nLocalTracks;
616 for (uint32_t itr = iBlock * nThreads + iThread; itr < nLocalTracks; itr += nBlocks * nThreads) {
617 sectorTr = &trk.Tracks()[itr];
619 SetTrackClusterZT(track, iSector, sectorTr);
620 if (
Param().
rec.tpc.mergerCovSource == 0) {
621 track.Set(
this, sectorTr,
alpha, iSector);
625 }
else if (
Param().
rec.tpc.mergerCovSource == 1) {
626 track.Set(
this, sectorTr,
alpha, iSector);
627 track.CopyBaseTrackCov();
628 }
else if (
Param().
rec.tpc.mergerCovSource == 2) {
629 if (RefitSectorTrack(track, sectorTr,
alpha, iSector)) {
630 track.Set(
this, sectorTr,
alpha, iSector);
637 CADEBUG(GPUInfo(
"INPUT Sector %d, Track %u, QPt %f DzDs %f", iSector, itr, track.QPt(), track.DzDs()));
638 track.SetPrevNeighbour(-1);
639 track.SetNextNeighbour(-1);
640 track.SetNextSegmentNeighbour(-1);
641 track.SetPrevSegmentNeighbour(-1);
642 track.SetExtrapolatedTrackId(0, -1);
643 track.SetExtrapolatedTrackId(1, -1);
644 uint32_t myTrack = CAMath::AtomicAdd(&mMemory->nUnpackedTracks, 1u);
645 mTrackIDs[iSector * mNMaxSingleSectorTracks + sectorTr->LocalTrackId()] = myTrack;
646 mSectorTrackInfos[myTrack] = track;
650GPUd()
void GPUTPCGMMerger::LinkExtrapolatedTracks(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
652 for (int32_t itr = SectorTrackInfoGlobalFirst(0) + iBlock * nThreads + iThread; itr < SectorTrackInfoGlobalLast(NSECTORS - 1); itr += nThreads * nBlocks) {
655 if (localTrack.ExtrapolatedTrackId(0) != -1 || !CAMath::AtomicCAS(&localTrack.ExtrapolatedTrackIds()[0], -1, itr)) {
656 localTrack.SetExtrapolatedTrackId(1, itr);
666 float fieldBz =
Param().bzCLight;
668 float dAlpha =
Param().dAlpha / 2;
672 dAlpha = dAlpha - CAMath::Pi() / 2;
673 }
else if (iBorder == 1) {
674 dAlpha = -dAlpha - CAMath::Pi() / 2;
675 }
else if (iBorder == 2) {
676 x0 = GPUTPCGeometry::Row2X(63);
677 }
else if (iBorder == 3) {
679 x0 = GPUTPCGeometry::Row2X(63);
680 }
else if (iBorder == 4) {
682 x0 = GPUTPCGeometry::Row2X(63);
685 const float maxSin = CAMath::Sin(60.f / 180.f * CAMath::Pi());
686 float cosAlpha = CAMath::Cos(dAlpha);
687 float sinAlpha = CAMath::Sin(dAlpha);
690 for (int32_t itr = iBlock * nThreads + iThread; itr < SectorTrackInfoLocalTotal(); itr += nThreads * nBlocks) {
692 int32_t iSector = track->Sector();
694 if (track->PrevSegmentNeighbour() >= 0 && track->Sector() == mSectorTrackInfos[track->PrevSegmentNeighbour()].Sector()) {
697 if (useOrigTrackParam) {
698 if (CAMath::Abs(track->QPt()) *
Param().qptB5Scaler <
Param().rec.tpc.mergerLooperQPtB5Limit) {
702 while (track->NextSegmentNeighbour() >= 0 && track->Sector() == mSectorTrackInfos[track->NextSegmentNeighbour()].Sector()) {
703 track = &mSectorTrackInfos[track->NextSegmentNeighbour()];
704 if (track->OrigTrack()->Param().X() < trackMin->OrigTrack()->Param().X()) {
708 trackTmp = *trackMin;
710 if (
Param().rec.tpc.mergerCovSource == 2 && trackTmp.X2() != 0.f) {
711 trackTmp.UseParam2();
713 trackTmp.Set(
this, trackMin->OrigTrack(), trackMin->Alpha(), trackMin->Sector());
716 if (CAMath::Abs(track->QPt()) *
Param().qptB5Scaler <
Param().rec.tpc.mergerLooperSecondHorizontalQPtB5Limit) {
717 if (iBorder == 0 && track->NextNeighbour() >= 0) {
720 if (iBorder == 1 && track->PrevNeighbour() >= 0) {
727 if (track->TransportToXAlpha(
this,
x0, sinAlpha, cosAlpha, fieldBz,
b, maxSin)) {
729 b.SetNClusters(track->NClusters());
730 for (int32_t
i = 0;
i < 4;
i++) {
731 if (CAMath::Abs(
b.Cov()[
i]) >= 5.0f) {
735 if (CAMath::Abs(
b.Cov()[4]) >= 0.5f) {
738 uint32_t myTrack = CAMath::AtomicAdd(&nB[iSector], 1u);
739 B[iSector][myTrack] =
b;
745GPUd()
void GPUTPCGMMerger::MergeBorderTracks<0>(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t iSector1,
GPUTPCGMBorderTrack* B1, int32_t N1, int32_t iSector2,
GPUTPCGMBorderTrack* B2, int32_t N2, int32_t mergeMode)
747 CADEBUG(GPUInfo(
"\nMERGING Sectors %d %d NTracks %d %d CROSS %d", iSector1, iSector2, N1, N2, mergeMode));
749 GPUTPCGMBorderRange* range2 = mBorderRange[iSector2] + *GetConstantMem()->tpcTrackers[iSector2].NTracks();
750 bool sameSector = (iSector1 == iSector2);
751 for (int32_t itr = iBlock * nThreads + iThread; itr < N1; itr += nThreads * nBlocks) {
753 float d = CAMath::Max(0.5f, 3.5f * CAMath::Sqrt(
b.Cov()[1]));
754 if (CAMath::Abs(
b.Par()[4]) *
Param().qptB5Scaler >= 20) {
759 CADEBUG(printf(
" Input Sector 1 %d Track %d: ", iSector1, itr);
for (int32_t
i = 0;
i < 5;
i++) { printf(
"%8.3f ",
b.Par()[
i]); } printf(
" - ");
for (int32_t
i = 0;
i < 5;
i++) { printf(
"%8.3f ",
b.Cov()[
i]); } printf(
" - D %8.3f\n", d));
762 range.fMin =
b.Par()[1] +
b.ZOffsetLinear() - d;
763 range.fMax =
b.Par()[1] +
b.ZOffsetLinear() + d;
770 for (int32_t itr = iBlock * nThreads + iThread; itr < N2; itr += nThreads * nBlocks) {
772 float d = CAMath::Max(0.5f, 3.5f * CAMath::Sqrt(
b.Cov()[1]));
773 if (CAMath::Abs(
b.Par()[4]) *
Param().qptB5Scaler >= 20) {
778 CADEBUG(printf(
" Input Sector 2 %d Track %d: ", iSector2, itr);
for (int32_t
i = 0;
i < 5;
i++) { printf(
"%8.3f ",
b.Par()[
i]); } printf(
" - ");
for (int32_t
i = 0;
i < 5;
i++) { printf(
"%8.3f ",
b.Cov()[
i]); } printf(
" - D %8.3f\n", d));
781 range.fMin =
b.Par()[1] +
b.ZOffsetLinear() - d;
782 range.fMax =
b.Par()[1] +
b.ZOffsetLinear() + d;
789GPUd()
void GPUTPCGMMerger::MergeBorderTracks<1>(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t iSector1,
GPUTPCGMBorderTrack* B1, int32_t N1, int32_t iSector2,
GPUTPCGMBorderTrack* B2, int32_t N2, int32_t mergeMode)
791#if !defined(GPUCA_GPUCODE_COMPILEKERNELS)
793 GPUTPCGMBorderRange* range2 = mBorderRange[iSector2] + *GetConstantMem()->tpcTrackers[iSector2].NTracks();
798 }
else if (iBlock == 1) {
803 printf(
"This sorting variant is disabled for RTC");
810#ifndef GPUCA_SPECIALIZE_THRUST_SORTS
811 if (iThread == 0 && iBlock == 0) {
822GPUd()
void GPUTPCGMMerger::MergeBorderTracks<2>(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t iSector1,
GPUTPCGMBorderTrack* B1, int32_t N1, int32_t iSector2,
GPUTPCGMBorderTrack* B2, int32_t N2, int32_t mergeMode)
825 float factor2ys =
Param().rec.tpc.trackMergerFactor2YS;
826 float factor2zt =
Param().rec.tpc.trackMergerFactor2ZT;
827 float factor2k =
Param().rec.tpc.trackMergerFactor2K;
828 float factor2General =
Param().rec.tpc.trackMergerFactor2General;
830 factor2k = factor2General * factor2k;
831 factor2ys = factor2General * factor2ys;
832 factor2zt = factor2General * factor2zt;
834 int32_t minNPartHits =
Param().rec.tpc.trackMergerMinPartHits;
835 int32_t minNTotalHits =
Param().rec.tpc.trackMergerMinTotalHits;
837 bool sameSector = (iSector1 == iSector2);
840 GPUTPCGMBorderRange* range2 = mBorderRange[iSector2] + *GetConstantMem()->tpcTrackers[iSector2].NTracks();
843 for (int32_t i1 = iBlock * nThreads + iThread; i1 < N1; i1 += nThreads * nBlocks) {
845 while (i2 < N2 && range2[i2].fMax < r1.
fMin) {
850 if (
b1.NClusters() < minNPartHits) {
856 for (int32_t k2 = i2; k2 < N2; k2++) {
861 if (sameSector && (r1.
fId >= r2.
fId)) {
867#if defined(GPUCA_MERGER_BY_MC_LABEL) && !defined(GPUCA_GPUCODE)
868 int64_t label1 = GetTrackLabel(
b1);
869 int64_t label2 = GetTrackLabel(b2);
870 if (label1 != label2 && label1 != -1)
873 CADEBUG(
if (GetConstantMem()->ioPtrs.mcLabelsTPC) {printf(
"Comparing track %3d to %3d: ", r1.fId, r2.fId); for (int32_t i = 0; i < 5; i++) { printf(
"%8.3f ", b1.Par()[i]); } printf(
" - ");
for (int32_t
i = 0;
i < 5;
i++) { printf(
"%8.3f ",
b1.Cov()[
i]); } printf(
"\n%28s",
""); });
874 CADEBUG(
if (GetConstantMem()->ioPtrs.mcLabelsTPC) {for (int32_t i = 0; i < 5; i++) { printf(
"%8.3f ", b2.Par()[i]); } printf(
" - ");
for (int32_t
i = 0;
i < 5;
i++) { printf(
"%8.3f ", b2.Cov()[
i]); } printf(
" - %5s - ", GetTrackLabel(
b1) == GetTrackLabel(b2) ?
"CLONE" :
"FAKE"); });
875 if (b2.NClusters() < lBest2) {
876 CADEBUG2(
continue, printf(
"!NCl1\n"));
880 int32_t maxRowDiff = mergeMode == 2 ? 1 : 3;
881 if (CAMath::Abs(
b1.Row() - b2.Row()) > maxRowDiff) {
882 CADEBUG2(
continue, printf(
"!ROW\n"));
884 if (CAMath::Abs(
b1.Par()[2] - b2.Par()[2]) > 0.5f || CAMath::Abs(
b1.Par()[3] - b2.Par()[3]) > 0.5f) {
885 CADEBUG2(
continue, printf(
"!CE SinPhi/Tgl\n"));
891 if (!
b1.CheckChi2Y(b2, factor2ys)) {
895 if (!
b1.CheckChi2QPt(b2, factor2k)) {
896 CADEBUG2(
continue, printf(
"!QPt\n"));
898 float fys = CAMath::Abs(
b1.Par()[4]) *
Param().qptB5Scaler < 20 ? factor2ys : (2.f * factor2ys);
899 float fzt = CAMath::Abs(
b1.Par()[4]) *
Param().qptB5Scaler < 20 ? factor2zt : (2.f * factor2zt);
900 if (!
b1.CheckChi2YS(b2, fys)) {
901 CADEBUG2(
continue, printf(
"!YS\n"));
903 if (!
b1.CheckChi2ZT(b2, fzt)) {
904 CADEBUG2(
continue, printf(
"!ZT\n"));
906 if (CAMath::Abs(
b1.Par()[4]) *
Param().qptB5Scaler < 20) {
907 if (b2.NClusters() < minNPartHits) {
908 CADEBUG2(
continue, printf(
"!NCl2\n"));
910 if (
b1.NClusters() + b2.NClusters() < minNTotalHits) {
911 CADEBUG2(
continue, printf(
"!NCl3\n"));
914 CADEBUG(printf(
"OK: dZ %8.3f D1 %8.3f D2 %8.3f\n", CAMath::Abs(
b1.Par()[1] - b2.Par()[1]), 3.5f * sqrt(
b1.Cov()[1]), 3.5f * sqrt(b2.Cov()[1])));
916 lBest2 = b2.NClusters();
917 iBest2 = b2.TrackID();
923 GPUCA_DEBUG_STREAMER_CHECK(
float weight =
b1.Par()[4] *
b1.Par()[4];
if (o2::utils::DebugStreamer::checkStream(
o2::utils::StreamFlags::streamMergeBorderTracksBest,
b1.TrackID(),
weight)) { MergedTrackStreamer(
b1, MergedTrackStreamerFindBorderTrack(B2, N2, iBest2),
"merge_best_track", iSector1, iSector2, mergeMode,
weight, o2::utils::DebugStreamer::getSamplingFrequency(
o2::utils::StreamFlags::streamMergeBorderTracksBest)); });
927 CADEBUG(GPUInfo(
"Found match %d %d",
b1.TrackID(), iBest2));
929 mTrackLinks[
b1.TrackID()] = iBest2;
939 if (withinSector == 1) {
941 n1 = n2 = mMemory->tmpCounter[iSector];
942 b1 = b2 = mBorder[iSector];
943 }
else if (withinSector == -1) {
944 jSector = (iSector + NSECTORS / 2);
945 const int32_t
offset = mergeMode == 2 ? NSECTORS : 0;
946 n1 = mMemory->tmpCounter[iSector +
offset];
947 n2 = mMemory->tmpCounter[jSector +
offset];
949 b2 = mBorder[jSector +
offset];
951 jSector = mNextSectorInd[iSector];
952 n1 = mMemory->tmpCounter[iSector];
953 n2 = mMemory->tmpCounter[NSECTORS + jSector];
954 b1 = mBorder[iSector];
955 b2 = mBorder[NSECTORS + jSector];
960GPUd()
void GPUTPCGMMerger::MergeBorderTracks(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t iSector, int8_t withinSector, int8_t mergeMode)
965 MergeBorderTracksSetup(n1, n2,
b1, b2, jSector, iSector, withinSector, mergeMode);
966 MergeBorderTracks<I>(nBlocks, nThreads, iBlock, iThread, iSector,
b1, n1, jSector, b2, n2, mergeMode);
969#if !defined(GPUCA_GPUCODE) || defined(GPUCA_GPUCODE_DEVICE)
970template GPUdni()
void GPUTPCGMMerger::MergeBorderTracks<0>(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t iSector, int8_t withinSector, int8_t mergeMode);
971template
GPUdni()
void GPUTPCGMMerger::MergeBorderTracks<1>(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t iSector, int8_t withinSector, int8_t mergeMode);
972template
GPUdni()
void GPUTPCGMMerger::MergeBorderTracks<2>(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t iSector, int8_t withinSector, int8_t mergeMode);
975GPUd()
void GPUTPCGMMerger::MergeWithinSectorsPrepare(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
977 float x0 = GPUTPCGeometry::Row2X(63);
978 const float maxSin = CAMath::Sin(60.f / 180.f * CAMath::Pi());
980 for (int32_t itr = iBlock * nThreads + iThread; itr < SectorTrackInfoLocalTotal(); itr += nThreads * nBlocks) {
982 int32_t iSector = track.Sector();
984 if (track.TransportToX(
this,
x0,
Param().bzCLight,
b, maxSin)) {
986 CADEBUG(printf(
"WITHIN SECTOR %d Track %d - ", iSector, itr);
for (int32_t
i = 0;
i < 5;
i++) { printf(
"%8.3f ",
b.Par()[
i]); } printf(
" - ");
for (int32_t
i = 0;
i < 5;
i++) { printf(
"%8.3f ",
b.Cov()[
i]); } printf(
"\n"));
987 b.SetNClusters(track.NClusters());
988 uint32_t myTrack = CAMath::AtomicAdd(&mMemory->tmpCounter[iSector], 1u);
989 mBorder[iSector][myTrack] =
b;
994GPUd()
void GPUTPCGMMerger::MergeSectorsPrepare(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread, int32_t border0, int32_t border1, int8_t useOrigTrackParam)
996 bool part2 = iBlock & 1;
997 int32_t
border = part2 ? border1 : border0;
1004 MergeSectorsPrepareStep2((nBlocks + !part2) >> 1, nThreads, iBlock >> 1, iThread,
border,
b,
n, useOrigTrackParam);
1009 start = (elems + nBlocks - 1) / nBlocks * iBlock;
1010 end = (elems + nBlocks - 1) / nBlocks * (iBlock + 1);
1011 end = CAMath::Min(elems,
end);
1020 u = mTrackCCRoots[u];
1021 v = mTrackCCRoots[
v];
1025 int32_t
h = CAMath::Max(u,
v);
1026 int32_t l = CAMath::Min(u,
v);
1028 int32_t old = CAMath::AtomicCAS(&mTrackCCRoots[
h],
h, l);
1033 u = mTrackCCRoots[
h];
1038GPUd()
void GPUTPCGMMerger::ResolveFindConnectedComponentsSetup(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1041 setBlockRange(SectorTrackInfoLocalTotal(), nBlocks, iBlock,
start,
end);
1042 for (int32_t
i =
start + iThread;
i <
end;
i += nThreads) {
1043 mTrackCCRoots[
i] =
i;
1047GPUd()
void GPUTPCGMMerger::ResolveFindConnectedComponentsHookLinks(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1052 setBlockRange(SectorTrackInfoLocalTotal(), nBlocks, iBlock,
start,
end);
1053 for (int32_t itr =
start + iThread; itr <
end; itr += nThreads) {
1054 hookEdge(itr, mTrackLinks[itr]);
1058GPUd()
void GPUTPCGMMerger::ResolveFindConnectedComponentsHookNeighbors(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1061 nBlocks = nBlocks / 4 * 4;
1062 if (iBlock >= nBlocks) {
1067 setBlockRange(SectorTrackInfoLocalTotal(), nBlocks / 4, iBlock / 4,
start,
end);
1069 int32_t myNeighbor = iBlock % 4;
1071 for (int32_t itr =
start + iThread; itr <
end; itr += nThreads) {
1072 int32_t
v = mSectorTrackInfos[itr].AnyNeighbour(myNeighbor);
1077GPUd()
void GPUTPCGMMerger::ResolveFindConnectedComponentsMultiJump(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1081 setBlockRange(SectorTrackInfoLocalTotal(), nBlocks, iBlock,
start,
end);
1082 for (int32_t itr =
start + iThread; itr <
end; itr += nThreads) {
1084 int32_t next = mTrackCCRoots[root];
1090 next = mTrackCCRoots[next];
1091 }
while (root != next);
1092 mTrackCCRoots[itr] = root;
1129 setBlockRange(SectorTrackInfoLocalTotal(), nBlocks, iBlock,
start,
end);
1131 for (int32_t baseIdx = 0; baseIdx < SectorTrackInfoLocalTotal(); baseIdx += nThreads) {
1132 int32_t itr = baseIdx + iThread;
1133 bool inRange = itr < SectorTrackInfoLocalTotal();
1137 itr2 = mTrackLinks[itr];
1140 bool resolveSector = (itr2 > -1);
1141 if (resolveSector) {
1142 int32_t root = mTrackCCRoots[itr];
1143 resolveSector &= (
start <= root) && (root <
end);
1146 int16_t smemIdx = work_group_scan_inclusive_add(int16_t(resolveSector));
1148 if (resolveSector) {
1149 smem.iTrack1[smemIdx - 1] = itr;
1150 smem.iTrack2[smemIdx - 1] = itr2;
1154 if (iThread < nThreads - 1) {
1158 const int32_t nSectors = smemIdx;
1160 for (int32_t
i = 0;
i < nSectors;
i++) {
1161 itr = smem.iTrack1[
i];
1162 itr2 = smem.iTrack2[
i];
1169 bool sameSegment = CAMath::Abs(track1->NClusters() > track2->NClusters() ? track1->QPt() : track2->QPt()) *
Param().qptB5Scaler < 2 || track1->QPt() * track2->QPt() > 0;
1174 while (track2->PrevSegmentNeighbour() >= 0) {
1175 track2 = &mSectorTrackInfos[track2->PrevSegmentNeighbour()];
1178 if (track1 == track2) {
1181 while (track1->PrevSegmentNeighbour() >= 0) {
1182 track1 = &mSectorTrackInfos[track1->PrevSegmentNeighbour()];
1183 if (track1 == track2) {
1187 GPUCommonAlgorithm::swap(track1, track1Base);
1188 for (int32_t k = 0; k < 2; k++) {
1190 while (tmp->Neighbour(k) >= 0) {
1191 tmp = &mSectorTrackInfos[tmp->Neighbour(k)];
1192 if (tmp == track2) {
1198 while (track1->NextSegmentNeighbour() >= 0) {
1199 track1 = &mSectorTrackInfos[track1->NextSegmentNeighbour()];
1200 if (track1 == track2) {
1205 while (track1->PrevSegmentNeighbour() >= 0) {
1206 track1 = &mSectorTrackInfos[track1->PrevSegmentNeighbour()];
1209 if (track1 == track2) {
1212 for (int32_t k = 0; k < 2; k++) {
1214 while (tmp->Neighbour(k) >= 0) {
1215 tmp = &mSectorTrackInfos[tmp->Neighbour(k)];
1216 if (tmp == track2) {
1222 float z1min, z1max, z2min, z2max;
1223 z1min = track1->MinClusterZT();
1224 z1max = track1->MaxClusterZT();
1225 z2min = track2->MinClusterZT();
1226 z2max = track2->MaxClusterZT();
1227 if (track1 != track1Base) {
1228 z1min = CAMath::Min(z1min, track1Base->MinClusterZT());
1229 z1max = CAMath::Max(z1max, track1Base->MaxClusterZT());
1231 if (track2 != track2Base) {
1232 z2min = CAMath::Min(z2min, track2Base->MinClusterZT());
1233 z2max = CAMath::Max(z2max, track2Base->MaxClusterZT());
1235 bool goUp = z2max - z1min > z1max - z2min;
1237 if (track1->Neighbour(goUp) < 0 && track2->Neighbour(!goUp) < 0) {
1238 track1->SetNeighbor(track2 - mSectorTrackInfos, goUp);
1239 track2->SetNeighbor(track1 - mSectorTrackInfos, !goUp);
1243 }
else if (track1->Neighbour(goUp) < 0) {
1244 track2 = &mSectorTrackInfos[track2->Neighbour(!goUp)];
1245 GPUCommonAlgorithm::swap(track1, track2);
1246 }
else if (track2->Neighbour(!goUp) < 0) {
1247 track1 = &mSectorTrackInfos[track1->Neighbour(goUp)];
1249 track1 = &mSectorTrackInfos[track1->Neighbour(goUp)];
1251 track1Base = track1;
1254 track2Base = track2;
1256 while (track1->NextSegmentNeighbour() >= 0) {
1257 track1 = &mSectorTrackInfos[track1->NextSegmentNeighbour()];
1260 track1->SetNextSegmentNeighbour(track2 - mSectorTrackInfos);
1261 track2->SetPrevSegmentNeighbour(track1 - mSectorTrackInfos);
1264 for (int32_t k = 0; k < 2; k++) {
1265 track1 = track1Base;
1266 track2 = track2Base;
1267 while (track2->Neighbour(k) >= 0) {
1268 if (track1->Neighbour(k) >= 0) {
1271 track2->SetNeighbor(-1, k);
1272 track2new->SetNeighbor(-1, k ^ 1);
1274 while (track1->NextSegmentNeighbour() >= 0) {
1275 track1 = &mSectorTrackInfos[track1->NextSegmentNeighbour()];
1277 track1->SetNextSegmentNeighbour(track2new - mSectorTrackInfos);
1278 track2new->SetPrevSegmentNeighbour(track1 - mSectorTrackInfos);
1283 track1->SetNeighbor(track2->Neighbour(k), k);
1284 track2->SetNeighbor(-1, k);
1285 track2new->SetNeighbor(track1 - mSectorTrackInfos, k ^ 1);
1303 if (
Param().
par.earlyTpcTransform) {
1307 auto& cln = mConstantMem->ioPtrs.clustersNative->clustersLinear[cls.num];
1308 GPUTPCConvertImpl::convert(*mConstantMem, cls.sector, cls.row, cln.getPad(), cln.getTime(),
x,
y,
z);
1311 if (!
Param().
par.continuousTracking && CAMath::Abs(
z) > 10) {
1314 int32_t sector = track->Sector();
1315 for (int32_t attempt = 0; attempt < 2; attempt++) {
1317 const float x0 = GPUTPCGeometry::Row2X(attempt == 0 ? 63 : cls.
row);
1320 b.SetNClusters(mMergedTracks[itr].NClusters());
1321 if (CAMath::Abs(
b.Cov()[4]) >= 0.5f) {
1324 if (track->CSide()) {
1325 b.SetPar(1,
b.Par()[1] - 2 * (
z -
b.ZOffsetLinear()));
1326 b.SetZOffsetLinear(-
b.ZOffsetLinear());
1329 uint32_t
id = sector + attempt * NSECTORS;
1330 uint32_t myTrack = CAMath::AtomicAdd(&mMemory->tmpCounter[
id], 1u);
1331 mBorder[
id][myTrack] =
b;
1339 const ClusterNative* cls =
Param().par.earlyTpcTransform ? nullptr : mConstantMem->ioPtrs.clustersNative->clustersLinear;
1340 for (uint32_t
i = iBlock * nThreads + iThread;
i < mMemory->nMergedTracks;
i += nThreads * nBlocks) {
1341 if (mMergedTracks[
i].CSide() == 0 && mTrackLinks[
i] >= 0) {
1342 if (mTrackLinks[mTrackLinks[
i]] != (int32_t)
i) {
1347 if (!trk[1]->OK() || trk[1]->CCE()) {
1350 bool celooper = (trk[0]->GetParam().GetQPt() *
Param().qptB5Scaler > 1 && trk[0]->GetParam().GetQPt() * trk[1]->GetParam().GetQPt() < 0);
1351 bool looper = trk[0]->Looper() || trk[1]->Looper() || celooper;
1352 if (!looper && trk[0]->GetParam().GetPar(3) * trk[1]->GetParam().GetPar(3) < 0) {
1356 uint32_t newRef = CAMath::AtomicAdd(&mMemory->nMergedTrackClusters, trk[0]->NClusters() + trk[1]->NClusters());
1357 if (newRef + trk[0]->NClusters() + trk[1]->NClusters() >= mNMaxMergedTrackClusters) {
1358 raiseError(GPUErrors::ERROR_MERGER_CE_HIT_OVERFLOW, newRef + trk[0]->NClusters() + trk[1]->NClusters(), mNMaxMergedTrackClusters);
1359 for (uint32_t k = newRef; k < mNMaxMergedTrackClusters; k++) {
1360 mClusters[k].num = 0;
1361 mClusters[k].state = 0;
1363 CAMath::AtomicExch(&mMemory->nMergedTrackClusters, mNMaxMergedTrackClusters);
1367 bool needswap =
false;
1370 if (
Param().
par.earlyTpcTransform) {
1371 z0max = CAMath::Max(CAMath::Abs(mClustersXYZ[trk[0]->FirstClusterRef()].
z), CAMath::Abs(mClustersXYZ[trk[0]->FirstClusterRef() + trk[0]->NClusters() - 1].
z));
1372 z1max = CAMath::Max(CAMath::Abs(mClustersXYZ[trk[1]->FirstClusterRef()].
z), CAMath::Abs(mClustersXYZ[trk[1]->FirstClusterRef() + trk[1]->NClusters() - 1].
z));
1374 z0max = -CAMath::Min(cls[mClusters[trk[0]->FirstClusterRef()].
num].
getTime(), cls[mClusters[trk[0]->FirstClusterRef() + trk[0]->NClusters() - 1].
num].
getTime());
1375 z1max = -CAMath::Min(cls[mClusters[trk[1]->FirstClusterRef()].
num].
getTime(), cls[mClusters[trk[1]->FirstClusterRef() + trk[1]->NClusters() - 1].
num].
getTime());
1377 if (z1max < z0max) {
1381 if (mClusters[trk[0]->FirstClusterRef()].
row > mClusters[trk[1]->FirstClusterRef()].
row) {
1386 GPUCommonAlgorithm::swap(trk[0], trk[1]);
1389 bool reverse[2] = {
false,
false};
1391 if (
Param().
par.earlyTpcTransform) {
1392 reverse[0] = (mClustersXYZ[trk[0]->FirstClusterRef()].z > mClustersXYZ[trk[0]->FirstClusterRef() + trk[0]->NClusters() - 1].z) ^ (trk[0]->CSide() > 0);
1393 reverse[1] = (mClustersXYZ[trk[1]->FirstClusterRef()].z < mClustersXYZ[trk[1]->FirstClusterRef() + trk[1]->NClusters() - 1].z) ^ (trk[1]->CSide() > 0);
1395 reverse[0] = cls[mClusters[trk[0]->FirstClusterRef()].num].getTime() < cls[mClusters[trk[0]->FirstClusterRef() + trk[0]->NClusters() - 1].num].getTime();
1396 reverse[1] = cls[mClusters[trk[1]->FirstClusterRef()].num].getTime() > cls[mClusters[trk[1]->FirstClusterRef() + trk[1]->NClusters() - 1].num].getTime();
1400 if (
Param().par.continuousTracking) {
1401 if (
Param().par.earlyTpcTransform) {
1402 const float z0 = trk[0]->CSide() ? CAMath::Max(mClustersXYZ[trk[0]->FirstClusterRef()].
z, mClustersXYZ[trk[0]->FirstClusterRef() + trk[0]->NClusters() - 1].
z) :
CAMath::Min(mClustersXYZ[trk[0]->FirstClusterRef()].
z, mClustersXYZ[trk[0]->FirstClusterRef() + trk[0]->NClusters() - 1].
z);
1403 const float z1 = trk[1]->CSide() ? CAMath::Max(mClustersXYZ[trk[1]->FirstClusterRef()].
z, mClustersXYZ[trk[1]->FirstClusterRef() + trk[1]->NClusters() - 1].
z) :
CAMath::Min(mClustersXYZ[trk[1]->FirstClusterRef()].
z, mClustersXYZ[trk[1]->FirstClusterRef() + trk[1]->NClusters() - 1].
z);
1404 const float offset = CAMath::Abs(z1) > CAMath::Abs(z0) ? -z0 :
z1;
1405 trk[1]->Param().Z() += trk[1]->Param().TZOffset() -
offset;
1406 trk[1]->Param().TZOffset() =
offset;
1409 const float tmax = CAMath::MaxWithRef(cls[mClusters[trk[0]->FirstClusterRef()].
num].
getTime(), cls[mClusters[trk[0]->FirstClusterRef() + trk[0]->NClusters() - 1].
num].
getTime(),
1410 cls[mClusters[trk[1]->FirstClusterRef()].
num].
getTime(), cls[mClusters[trk[1]->FirstClusterRef() + trk[1]->NClusters() - 1].
num].
getTime(),
1411 &mClusters[trk[0]->FirstClusterRef()], &mClusters[trk[0]->FirstClusterRef() + trk[0]->NClusters() - 1],
1412 &mClusters[trk[1]->FirstClusterRef()], &mClusters[trk[1]->FirstClusterRef() + trk[1]->NClusters() - 1], clsmax);
1413 const float offset = CAMath::Max(tmax - mConstantMem->calibObjects.fastTransformHelper->getCorrMap()->getMaxDriftTime(clsmax->
sector, clsmax->
row, cls[clsmax->
num].getPad()), 0.f);
1414 trk[1]->Param().Z() += mConstantMem->calibObjects.fastTransformHelper->getCorrMap()->convDeltaTimeToDeltaZinTimeFrame(trk[1]->CSide() * NSECTORS / 2, trk[1]->
Param().TZOffset() -
offset);
1415 trk[1]->Param().TZOffset() =
offset;
1419 int32_t
pos = newRef;
1421 int32_t lastLeg = -1;
1423 for (int32_t k = 1; k >= 0; k--) {
1424 int32_t loopstart =
reverse[k] ? (trk[k]->NClusters() - 1) : 0;
1425 int32_t loopend =
reverse[k] ? -1 : (int32_t)trk[k]->NClusters();
1426 int32_t loopinc =
reverse[k] ? -1 : 1;
1427 for (int32_t
j = loopstart;
j != loopend;
j += loopinc) {
1428 if (
Param().par.earlyTpcTransform) {
1429 mClustersXYZ[
pos] = mClustersXYZ[trk[k]->FirstClusterRef() +
j];
1431 mClusters[
pos] = mClusters[trk[k]->FirstClusterRef() +
j];
1433 if (mClusters[trk[k]->FirstClusterRef() +
j].leg != lastLeg) {
1435 lastLeg = mClusters[trk[k]->FirstClusterRef() +
j].leg;
1437 mClusters[
pos].leg =
leg;
1445 trk[1]->SetFirstClusterRef(newRef);
1446 trk[1]->SetNClusters(trk[0]->NClusters() + trk[1]->NClusters());
1451 trk[1]->SetCCE(
true);
1453 trk[1]->SetLooper(
true);
1454 trk[1]->SetLegs(
leg + 1);
1456 trk[0]->SetNClusters(0);
1457 trk[0]->SetOK(
false);
1468struct GPUTPCGMMerger_CompareClusterIdsLooper {
1469 struct clcomparestruct {
1478 GPUd() bool operator()(const int16_t
aa, const int16_t
bb)
1480 const clcomparestruct&
a =
cmp2[
aa];
1484 if (
a.leg !=
b.leg) {
1485 return ((
leg > 0) ^ (
a.leg >
b.leg));
1494struct GPUTPCGMMerger_CompareClusterIds {
1497 GPUd() bool operator()(const int16_t
aa, const int16_t
bb)
1501 if (
a.row !=
b.row) {
1502 return (
a.row >
b.row);
1510GPUd()
void GPUTPCGMMerger::CollectMergedTracks(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1514 for (int32_t itr = iBlock * nThreads + iThread; itr < SectorTrackInfoLocalTotal(); itr += nThreads * nBlocks) {
1517 if (track.PrevSegmentNeighbour() >= 0) {
1520 if (track.PrevNeighbour() >= 0) {
1527 tr->SetPrevSegmentNeighbour(1000000000);
1529 if (nParts >= kMaxParts) {
1532 if (nHits + tr->NClusters() > kMaxClusters) {
1535 nHits += tr->NClusters();
1538 trackParts[nParts++] = tr;
1539 for (int32_t
i = 0;
i < 2;
i++) {
1540 if (tr->ExtrapolatedTrackId(
i) != -1) {
1541 if (nParts >= kMaxParts) {
1544 if (nHits + mSectorTrackInfos[tr->ExtrapolatedTrackId(
i)].NClusters() > kMaxClusters) {
1547 trackParts[nParts] = &mSectorTrackInfos[tr->ExtrapolatedTrackId(
i)];
1548 trackParts[nParts++]->SetLeg(
leg);
1549 nHits += mSectorTrackInfos[tr->ExtrapolatedTrackId(
i)].NClusters();
1552 int32_t jtr = tr->NextSegmentNeighbour();
1554 tr = &(mSectorTrackInfos[jtr]);
1555 tr->SetPrevSegmentNeighbour(1000000002);
1558 jtr = trbase->NextNeighbour();
1560 trbase = &(mSectorTrackInfos[jtr]);
1562 if (tr->PrevSegmentNeighbour() >= 0) {
1565 tr->SetPrevSegmentNeighbour(1000000001);
1573 if (nParts > 1 &&
leg == 0) {
1576 if (
a->X() !=
b->X()) {
1577 return (a->X() > b->X());
1579 if (
a->Y() !=
b->Y()) {
1580 return (a->Y() > b->Y());
1582 if (
a->Z() !=
b->Z()) {
1583 return (a->Z() > b->Z());
1585 return a->QPt() >
b->QPt();
1587 return (
a->X() >
b->X());
1592 if (
Param().rec.tpc.dropLoopers &&
leg > 0) {
1597 trackCluster trackClusters[kMaxClusters];
1599 for (int32_t ipart = 0; ipart < nParts; ipart++) {
1601 CADEBUG(printf(
"Collect Track %d Part %d QPt %f DzDs %f\n", mMemory->nMergedTracks, ipart, t->QPt(), t->DzDs()));
1602 int32_t nTrackHits = t->NClusters();
1603 trackCluster*
c2 = trackClusters + nHits + nTrackHits - 1;
1604 for (int32_t
i = 0;
i < nTrackHits;
i++,
c2--) {
1605 const GPUTPCTracker& trk = GetConstantMem()->tpcTrackers[t->Sector()];
1606 const GPUTPCHitId& ic = trk.TrackHits()[t->OrigTrack()->FirstHitID() +
i];
1607 uint32_t
id = trk.Data().ClusterDataIndex(trk.Data().Row(ic.RowIndex()), ic.HitIndex()) + GetConstantMem()->ioPtrs.clustersNative->clusterOffset[t->Sector()][0];
1608 *
c2 = trackCluster{
id, (
uint8_t)ic.RowIndex(), t->Sector(), t->Leg()};
1610 nHits += nTrackHits;
1618 for (int32_t
i = 1;
i < nHits;
i++) {
1619 if (trackClusters[
i].
row > trackClusters[
i - 1].
row || trackClusters[
i].
id == trackClusters[
i - 1].
id) {
1625 int32_t firstTrackIndex = 0;
1626 int32_t lastTrackIndex = nParts - 1;
1628 int32_t nTmpHits = 0;
1629 trackCluster trackClustersUnsorted[kMaxClusters];
1630 int16_t clusterIndices[kMaxClusters];
1631 for (int32_t
i = 0;
i < nHits;
i++) {
1632 trackClustersUnsorted[
i] = trackClusters[
i];
1633 clusterIndices[
i] =
i;
1640 for (int32_t
i = 0;
i < nParts;
i++) {
1641 if (trackParts[
i]->Leg() == 0 || trackParts[
i]->Leg() ==
leg) {
1643 if (
Param().par.earlyTpcTransform) {
1644 zt = CAMath::Min(CAMath::Abs(trackParts[
i]->ClusterZT0()), CAMath::Abs(trackParts[
i]->ClusterZTN()));
1646 zt = -trackParts[
i]->MinClusterZT();
1650 baseLeg = trackParts[
i]->Leg();
1654 int32_t iLongest = 1e9;
1656 for (int32_t
i = (baseLeg ? (nParts - 1) : 0); baseLeg ? (
i >= 0) : (
i < nParts); baseLeg ?
i-- :
i++) {
1657 if (trackParts[
i]->Leg() != baseLeg) {
1660 if (trackParts[
i]->OrigTrack()->NHits() >
length) {
1662 length = trackParts[
i]->OrigTrack()->NHits();
1666 if (
Param().par.earlyTpcTransform) {
1667 outwards = (trackParts[iLongest]->ClusterZT0() > trackParts[iLongest]->ClusterZTN()) ^ trackParts[iLongest]->CSide();
1669 outwards = trackParts[iLongest]->ClusterZT0() < trackParts[iLongest]->ClusterZTN();
1671 GPUTPCGMMerger_CompareClusterIdsLooper::clcomparestruct clusterSort[kMaxClusters];
1672 for (int32_t iPart = 0; iPart < nParts; iPart++) {
1674 int32_t nTrackHits = t->NClusters();
1675 for (int32_t
j = 0;
j < nTrackHits;
j++) {
1676 int32_t
i = nTmpHits +
j;
1677 clusterSort[
i].leg = t->Leg();
1679 nTmpHits += nTrackHits;
1682 GPUCommonAlgorithm::sort(clusterIndices, clusterIndices + nHits, GPUTPCGMMerger_CompareClusterIdsLooper(baseLeg,
outwards, trackClusters, clusterSort));
1684 GPUCommonAlgorithm::sort(clusterIndices, clusterIndices + nHits, GPUTPCGMMerger_CompareClusterIds(trackClusters));
1687 firstTrackIndex = lastTrackIndex = -1;
1688 for (int32_t
i = 0;
i < nParts;
i++) {
1689 nTmpHits += trackParts[
i]->NClusters();
1690 if (nTmpHits > clusterIndices[0] && firstTrackIndex == -1) {
1691 firstTrackIndex =
i;
1693 if (nTmpHits > clusterIndices[nHits - 1] && lastTrackIndex == -1) {
1698 int32_t nFilteredHits = 0;
1699 int32_t indPrev = -1;
1700 for (int32_t
i = 0;
i < nHits;
i++) {
1701 int32_t ind = clusterIndices[
i];
1702 if (indPrev >= 0 && trackClustersUnsorted[ind].
id == trackClustersUnsorted[indPrev].
id) {
1706 trackClusters[nFilteredHits] = trackClustersUnsorted[ind];
1709 nHits = nFilteredHits;
1712 const uint32_t iMergedTrackFirstCluster = CAMath::AtomicAdd(&mMemory->nMergedTrackClusters, (uint32_t)nHits);
1713 if (iMergedTrackFirstCluster + nHits > mNMaxMergedTrackClusters) {
1714 raiseError(GPUErrors::ERROR_MERGER_HIT_OVERFLOW, iMergedTrackFirstCluster, mNMaxMergedTrackClusters);
1715 CAMath::AtomicExch(&mMemory->nMergedTrackClusters, mNMaxMergedTrackClusters);
1721 for (int32_t
i = 0;
i < nHits;
i++) {
1723 if (
Param().
par.earlyTpcTransform) {
1724 const GPUTPCClusterData&
c = GetConstantMem()->tpcTrackers[trackClusters[
i].sector].ClusterData()[trackClusters[
i].id - GetConstantMem()->tpcTrackers[trackClusters[
i].sector].Data().ClusterIdOffset()];
1729 clXYZ[
i].
amp =
c.amp;
1730#ifdef GPUCA_TPC_RAW_PROPAGATE_PAD_ROW_TIME
1731 clXYZ[
i].pad =
c.mPad;
1732 clXYZ[
i].time =
c.mTime;
1736 const ClusterNative&
c = GetConstantMem()->ioPtrs.clustersNative->clustersLinear[trackClusters[
i].id];
1740 cl[
i].
row = trackClusters[
i].row;
1741 cl[
i].
num = trackClusters[
i].id;
1742 cl[
i].
sector = trackClusters[
i].sector;
1743 cl[
i].
leg = trackClusters[
i].leg;
1746 uint32_t iOutputTrack = CAMath::AtomicAdd(&mMemory->nMergedTracks, 1u);
1747 if (iOutputTrack >= mNMaxTracks) {
1748 raiseError(GPUErrors::ERROR_MERGER_TRACK_OVERFLOW, iOutputTrack, mNMaxTracks);
1749 CAMath::AtomicExch(&mMemory->nMergedTracks, mNMaxTracks);
1755 mergedTrack.SetFlags(0);
1756 mergedTrack.SetOK(1);
1757 mergedTrack.SetLooper(
leg > 0);
1758 mergedTrack.SetLegs(
leg);
1759 mergedTrack.SetNClusters(nHits);
1760 mergedTrack.SetFirstClusterRef(iMergedTrackFirstCluster);
1763 mergedTrack.SetCSide(
p2.CSide());
1766 const float toX =
Param().par.earlyTpcTransform ? mClustersXYZ[iMergedTrackFirstCluster].x : GPUTPCGeometry::Row2X(cl[0].
row);
1769 p1.Y() =
b.Par()[0];
1770 p1.Z() =
b.Par()[1];
1771 p1.SinPhi() =
b.Par()[2];
1776 p1.SinPhi() =
p2.SinPhi();
1778 p1.TZOffset() =
p2.TZOffset();
1779 p1.DzDs() =
p2.DzDs();
1780 p1.QPt() =
p2.QPt();
1781 mergedTrack.SetAlpha(
p2.Alpha());
1782 if (CAMath::Abs(
Param().polynomialField.GetNominalBz()) < (gpu_common_constants::kZeroFieldCut * gpu_common_constants::kCLight)) {
1783 p1.QPt() = 100.f /
Param().rec.bz0Pt10MeV;
1796 if (
Param().
par.earlyTpcTransform) {
1798 CEside = (mergedTrack.CSide() != 0) ^ (clXYZ[0].
z > clXYZ[nHits - 1].
z);
1800 auto& cls = mConstantMem->ioPtrs.clustersNative->clustersLinear;
1801 CEside = cls[cl[0].
num].getTime() < cls[cl[nHits - 1].
num].getTime();
1803 MergeCEFill(trackParts[CEside ? lastTrackIndex : firstTrackIndex], cl[CEside ? (nHits - 1) : 0],
Param().
par.earlyTpcTransform ? &(mClustersXYZ + iMergedTrackFirstCluster)[CEside ? (nHits - 1) : 0] : nullptr, iOutputTrack);
1808GPUd()
void GPUTPCGMMerger::SortTracksPrepare(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1810 for (uint32_t
i = iBlock * nThreads + iThread;
i < mMemory->nMergedTracks;
i += nThreads * nBlocks) {
1811 mTrackOrderProcess[
i] =
i;
1815GPUd()
void GPUTPCGMMerger::PrepareClustersForFit0(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1817 for (uint32_t
i = iBlock * nThreads + iThread;
i < mMemory->nMergedTracks;
i += nBlocks * nThreads) {
1822GPUd()
void GPUTPCGMMerger::SortTracks(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1824#ifndef GPUCA_SPECIALIZE_THRUST_SORTS
1825 if (iThread == 0 && iBlock == 0) {
1826 GPUCommonAlgorithm::sortDeviceDynamic(mTrackOrderProcess, mTrackOrderProcess + mMemory->nMergedTracks,
GPUTPCGMMergerSortTracks_comp(mMergedTracks));
1831GPUd()
void GPUTPCGMMerger::SortTracksQPt(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1833#ifndef GPUCA_SPECIALIZE_THRUST_SORTS
1834 if (iThread == 0 && iBlock == 0) {
1840GPUd()
void GPUTPCGMMerger::PrepareClustersForFit1(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1842 for (uint32_t
i = iBlock * nThreads + iThread;
i < mMemory->nMergedTracks;
i += nBlocks * nThreads) {
1843 mTrackOrderAttach[mTrackSort[
i]] =
i;
1846 for (uint32_t
j = 0;
j < trk.NClusters();
j++) {
1848 CAMath::AtomicAdd(&mSharedCount[mClusters[trk.FirstClusterRef() +
j].num], 1u);
1854GPUd()
void GPUTPCGMMerger::PrepareClustersForFit2(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1856 for (uint32_t
i = iBlock * nThreads + iThread;
i < mMemory->nMergedTrackClusters;
i += nBlocks * nThreads) {
1857 if (mSharedCount[mClusters[
i].
num] > 1) {
1861 if (mClusterStateExt) {
1862 for (uint32_t
i = iBlock * nThreads + iThread;
i < mNMaxClusters;
i += nBlocks * nThreads) {
1863 uint8_t state = GetConstantMem()->ioPtrs.clustersNative->clustersLinear[
i].getFlags();
1864 if (mSharedCount[
i] > 1) {
1867 mClusterStateExt[
i] =
state;
1874 for (uint32_t
i = iBlock * nThreads + iThread;
i < mMemory->nMergedTracks;
i += nThreads * nBlocks) {
1875 mTrackSort[mTrackOrderAttach[
i]] =
i;
1877 for (uint32_t
i = iBlock * nThreads + iThread;
i < mMemory->nMergedTrackClusters;
i += nThreads * nBlocks) {
1878 mClusterAttachment[mClusters[
i].num] = 0;
1884 for (uint32_t
i = iBlock * nThreads + iThread;
i < mMemory->nMergedTracks;
i += nThreads * nBlocks) {
1886 if (!trk.OK() || trk.NClusters() == 0) {
1889 uint8_t goodLeg = mClusters[trk.FirstClusterRef() + trk.NClusters() - 1].leg;
1890 for (uint32_t
j = 0;
j < trk.NClusters();
j++) {
1891 int32_t
id = mClusters[trk.FirstClusterRef() +
j].num;
1893 uint8_t clusterState = mClusters[trk.FirstClusterRef() +
j].state;
1899 if (mClusters[trk.FirstClusterRef() +
j].leg == goodLeg) {
1902 CAMath::AtomicMax(&mClusterAttachment[
id],
weight);
1909 for (uint32_t
i = iBlock * nThreads + iThread;
i < mNMaxClusters;
i += nThreads * nBlocks) {
1910 if (mClusterAttachment[
i] != 0) {
1916GPUd()
void GPUTPCGMMerger::MergeLoopersInit(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1918 const float lowPtThresh =
Param().rec.tpc.rejectQPtB5 * 1.1f;
1920 const auto& trk = mMergedTracks[
i];
1921 const auto& p = trk.GetParam();
1922 const float qptabs = CAMath::Abs(p.GetQPt());
1923 if (trk.NClusters() && qptabs *
Param().qptB5Scaler > 5.f && qptabs *
Param().qptB5Scaler <= lowPtThresh) {
1924 const int32_t sector = mClusters[trk.FirstClusterRef() + trk.NClusters() - 1].sector;
1925 const float refz = p.GetZ() + (
Param().par.earlyTpcTransform ? p.GetTZOffset() : GetConstantMem()->calibObjects.fastTransformHelper->getCorrMap()->convVertexTimeToZOffset(sector, p.GetTZOffset(),
Param().continuousMaxTimeBin)) + (trk.CSide() ? -100 : 100);
1927 CAMath::SinCos(trk.GetAlpha(), sinA, cosA);
1928 float gx = cosA * p.GetX() - sinA * p.GetY();
1929 float gy = cosA * p.GetY() + sinA * p.GetX();
1930 float bz =
Param().polynomialField.GetFieldBz(gx, gy, p.GetZ());
1931 const float r1 = p.GetQPt() * bz;
1932 const float r = CAMath::Abs(r1) > 0.0001f ? (1.f / r1) : 10000;
1933 const float mx = p.GetX() +
r * p.GetSinPhi();
1934 const float my = p.GetY() -
r * CAMath::Sqrt(1 - p.GetSinPhi() * p.GetSinPhi());
1935 const float gmx = cosA * mx - sinA * my;
1936 const float gmy = cosA * my + sinA * mx;
1937 uint32_t myId = CAMath::AtomicAdd(&mMemory->nLooperMatchCandidates, 1u);
1938 if (myId >= mNMaxLooperMatches) {
1939 raiseError(GPUErrors::ERROR_LOOPER_MATCH_OVERFLOW, myId, mNMaxLooperMatches);
1940 CAMath::AtomicExch(&mMemory->nLooperMatchCandidates, mNMaxLooperMatches);
1968GPUd()
void GPUTPCGMMerger::MergeLoopersSort(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1970#ifndef GPUCA_SPECIALIZE_THRUST_SORTS
1971 if (iThread == 0 && iBlock == 0) {
1977GPUd()
void GPUTPCGMMerger::MergeLoopersMain(int32_t nBlocks, int32_t nThreads, int32_t iBlock, int32_t iThread)
1981#if GPUCA_MERGE_LOOPER_MC && !defined(GPUCA_GPUCODE)
1982 std::vector<int64_t> paramLabels(mMemory->nLooperMatchCandidates);
1983 for (uint32_t
i = 0;
i < mMemory->nLooperMatchCandidates;
i++) {
1984 paramLabels[
i] = GetTrackLabel(mMergedTracks[
params[
i].
id]);
1990 if (!mRec->GetProcessingSettings().runQA) {
1991 throw std::runtime_error(
"Need QA enabled for the Merge Loopers MC QA");
1996 for (uint32_t
j =
i + 1;
j < mMemory->nLooperMatchCandidates;
j++) {
1998 if (CAMath::Abs(
params[
j].refz) > CAMath::Abs(
params[
i].refz) + 100.f) {
2006 const auto& trk1 = mMergedTracks[
params[
i].id];
2007 const auto& trk2 = mMergedTracks[
params[
j].id];
2008 const auto& param1 = trk1.GetParam();
2009 const auto& param2 = trk2.GetParam();
2010 if (CAMath::Abs(param1.GetDzDs()) > 0.03f && CAMath::Abs(param2.GetDzDs()) > 0.03f && param1.GetDzDs() * param2.GetDzDs() * param1.GetQPt() * param2.GetQPt() < 0) {
2015 const float dznormalized = (CAMath::Abs(
params[
j].refz) - CAMath::Abs(
params[
i].refz)) / (CAMath::TwoPi() * 0.5f * (CAMath::Abs(param1.GetDzDs()) + CAMath::Abs(param2.GetDzDs())) * 1.f / (0.5f * (CAMath::Abs(param1.GetQPt()) + CAMath::Abs(param2.GetQPt())) * CAMath::Abs(
Param().polynomialField.GetNominalBz())));
2016 const float phasecorr = CAMath::Modf((CAMath::ASin(param1.GetSinPhi()) + trk1.GetAlpha() - CAMath::ASin(param2.GetSinPhi()) - trk2.GetAlpha()) / CAMath::TwoPi() + 5.5f, 1.f) - 0.5f;
2017 const float phasecorrdirection = (
params[
j].refz * param1.GetQPt() * param1.GetDzDs()) > 0 ? 1 : -1;
2018 const float dzcorr = dznormalized + phasecorr * phasecorrdirection;
2019 const bool sameside = !(trk1.CSide() ^ trk2.CSide());
2020 const float dzcorrlimit[4] = {sameside ? 0.018f : 0.012f, sameside ? 0.12f : 0.025f, 0.14f, 0.15f};
2021 const int32_t dzcorrcount = sameside ? 4 : 2;
2022 bool dzcorrok =
false;
2024 for (int32_t k = 0; k < dzcorrcount; k++) {
2025 const float d = CAMath::Abs(dzcorr - 0.5f * k);
2026 if (d <= dzcorrlimit[k]) {
2028 dznorm = d / dzcorrlimit[k];
2037 const float dtgl = param1.GetDzDs() - (param1.GetQPt() * param2.GetQPt() > 0 ? param2.GetDzDs() : -param2.GetDzDs());
2038 const float dqpt = (CAMath::Abs(param1.GetQPt()) - CAMath::Abs(param2.GetQPt())) / CAMath::Min(param1.GetQPt(), param2.GetQPt());
2039 float d = CAMath::Sum2(dtgl * (1.f / 0.03f), dqpt * (1.f / 0.04f)) + d2xy * (1.f / 4.f) + dznorm * (1.f / 0.3f);
2041#if GPUCA_MERGE_LOOPER_MC && !defined(GPUCA_GPUCODE)
2042 const int64_t label1 = paramLabels[
i];
2043 const int64_t label2 = paramLabels[
j];
2044 bool labelEQ = label1 != -1 && label1 == label2;
2045 if (1 || EQ || labelEQ) {
2047 static auto& tup =
GPUROOTDump<TNtuple>::get(
"mergeloopers",
"labeleq:sides:d2xy:tgl1:tgl2:qpt1:qpt2:dz:dzcorr:dtgl:dqpt:dznorm:bs");
2048 tup.Fill((
float)labelEQ, (trk1.CSide() ? 1 : 0) | (trk2.CSide() ? 2 : 0), d2xy, param1.GetDzDs(), param2.GetDzDs(), param1.GetQPt(), param2.GetQPt(),
CAMath::
Abs(
params[
j].refz) -
CAMath::
Abs(
params[
i].refz), dzcorr, dtgl, dqpt, dznorm, bs);
2049 static auto tup2 =
GPUROOTDump<TNtuple>::getNew(
"mergeloopers2",
"labeleq:refz1:refz2:tgl1:tgl2:qpt1:qpt2:snp1:snp2:a1:a2:dzn:phasecor:phasedir:dzcorr");
2050 tup2.Fill((
float)labelEQ,
params[
i].refz,
params[
j].refz, param1.GetDzDs(), param2.GetDzDs(), param1.GetQPt(), param2.GetQPt(), param1.GetSinPhi(), param2.GetSinPhi(), trk1.GetAlpha(), trk2.GetAlpha(), dznormalized, phasecorr, phasecorrdirection, dzcorr);
2064 mMergedTracks[
params[
j].id].SetMergedLooper(
true);
2065 if (CAMath::Abs(param2.GetQPt() *
Param().qptB5Scaler) >=
Param().
rec.tpc.rejectQPtB5) {
2066 mMergedTracks[
params[
i].id].SetMergedLooper(
true);
Class of a TPC cluster in TPC-native coordinates (row, time)
A const (ready only) version of MCTruthContainer.
#define get_global_size(dim)
#define get_global_id(dim)
#define GPUCA_DEBUG_STREAMER_CHECK(...)
#define GPUCA_DETERMINISTIC_CODE(det, indet)
#define GPUCA_MERGER_MAX_TRACK_CLUSTERS
#define GPUCA_MAX_SIN_PHI_LOW
#define GPUCA_TRACKLET_SELECTOR_MIN_HITS_B5(QPTB5)
#define GPUCA_MAX_SIN_PHI
#define CADEBUG2(cmd,...)
const GPUTPCGMMerger::trackCluster & b1
const GPUTPCGMMerger::trackCluster & a1
const GPUTPCGMMerger::trackCluster *const mCmp
const clcomparestruct *const cmp2
const GPUTPCGMMerger::trackCluster *const cmp1
bool const GPUTPCGMMerger::trackCluster * c1
bool const GPUTPCGMMerger::trackCluster const clcomparestruct * c2
constexpr int p1()
constexpr to accelerate the coordinates changing
std::array< int, 64 > reverse(std::array< int, 64 > a)
Class for time synchronization of RawReader instances.
static void computePointerWithAlignment(T *&basePtr, S *&objPtr, size_t nEntries=1)
void AllocateAndInitializeLate()
static GPUROOTDump< T, Args... > getNew(const char *name1, Names... names)
static GPUROOTDump< T, Args... > & get(const char *name1, Names... names)
RecoStepField GetRecoStepsGPU() const
int16_t RegisterMemoryAllocation(T *proc, void *(T::*setPtr)(void *), int32_t type, const char *name="", const GPUMemoryReuse &re=GPUMemoryReuse())
RecoStepField GetRecoSteps() const
const GPUParam & GetParam() const
const GPUConstantMem & GetConstantMem() const
GPUMemorySizeScalers * MemoryScalers()
const GPUSettingsProcessing & GetProcessingSettings() const
virtual const GPUDefParameters & getGPUParameters(bool doGPU) const =0
void * SetPointersRefitScratch(void *mem)
void RegisterMemoryAllocation()
void * SetPointersMemory(void *mem)
static constexpr const int32_t NSECTORS
void * SetPointersOutput(void *mem)
void InitializeProcessor()
void * SetPointersOutputO2MC(void *mem)
void * SetPointersOutputO2Scratch(void *mem)
void * SetPointersMerger(void *mem)
void * SetPointersOutputO2Clus(void *mem)
void * SetPointersOutputState(void *mem)
void SetMaxData(const GPUTrackingInOutPointers &io)
void * SetPointersOutputO2(void *mem)
int32_t int32_t int32_t bool float float Z
int32_t int32_t int32_t bool float Y
GLint GLint GLsizei GLint border
GLfloat GLfloat GLfloat alpha
GLuint GLuint GLfloat weight
GLboolean GLboolean GLboolean b
GLenum const GLfloat * params
GLuint GLsizei GLsizei * length
typedef void(APIENTRYP PFNGLCULLFACEPROC)(GLenum mode)
GLboolean GLboolean GLboolean GLboolean a
GLdouble GLdouble GLdouble z
uint8_t itsSharedClusterMap uint8_t
Node par(int index)
Parameters.
GPUd() const expr uint32_t MultivariatePolynomialHelper< Dim
float float float float z1
if(!okForPhiMin(phi0, phi1))
Global TPC definitions and constants.
@ streamMergeBorderTracksBest
stream MergeBorderTracks best track
@ streamMergeBorderTracksAll
stream MergeBorderTracks all tracks
std::string getTime(uint64_t ts)
GPUTPCTracker tpcTrackers[GPUCA_NSECTORS]
size_t getValue(size_t maxVal, size_t val)
size_t NTPCMergedTrackHits(size_t tpcSectorTrackHitss)
size_t NTPCMergedTracks(size_t tpcSectorTracks)
@ clustererAndSharedFlags
const o2::tpc::ClusterNativeAccess * clustersNative
const GPUTPCGMMergedTrack & GPUrestrict() b
GPUd() bool operator()(const int32_t aa
const GPUTPCGMMergedTrack *const mCmp
GPUhd() GPUTPCGMMergerSortTracksQPt_comp(GPUTPCGMMergedTrack *cmp)
const GPUTPCGMMergedTrack & GPUrestrict() b
GPUhd() GPUTPCGMMergerSortTracks_comp(GPUTPCGMMergedTrack *cmp)
GPUd() bool operator()(const int32_t aa
const GPUTPCGMMergedTrack *const mCmp
unsigned int nClustersTotal
char const *restrict const cmp