VertexerTraits.cxx

Go to the documentation of this file.

// Copyright 2019-2026 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
#include <algorithm>
#include <memory>
#include <ranges>
#include <span>
#include <unordered_map>
 
#include <oneapi/tbb/blocked_range.h>
#include <oneapi/tbb/parallel_for.h>
#include <oneapi/tbb/combinable.h>
 
#include "ITStracking/VertexerTraits.h"
#include "ITStracking/BoundedAllocator.h"
#include "ITStracking/ClusterLines.h"
#include "ITStracking/Definitions.h"
#include "ITStracking/LineVertexerHelpers.h"
#include "ITStracking/Tracklet.h"
#include "SimulationDataFormat/DigitizationContext.h"
#include "SimulationDataFormat/O2DatabasePDG.h"
#include "Steer/MCKinematicsReader.h"
#include "DataFormatsITSMFT/DPLAlpideParam.h"
#include "DetectorsRaw/HBFUtils.h"
 
namespace o2::its
{
namespace
{
 
template <TrackletMode Mode, bool EvalRun, int NLayers>
void trackleterKernelHost(
  const gsl::span<const Cluster>& clustersNextLayer,    // 0 2
  const gsl::span<const Cluster>& clustersCurrentLayer, // 1 1
  const gsl::span<uint8_t>& usedClustersNextLayer,      // 0 2
  const int* indexTableNext,
  const float phiCut,
  bounded_vector<Tracklet>& tracklets,
  gsl::span<int> foundTracklets,
  const IndexTableUtils<NLayers>& utils,
  const TimeEstBC& timErr,
  gsl::span<int> rofFoundTrackletsOffsets,
  const int globalOffsetNextLayer,
  const int globalOffsetCurrentLayer,
  const int maxTrackletsPerCluster)
{
  const int PhiBins{utils.getNphiBins()};
  const int ZBins{utils.getNzBins()};
  // loop on layer1 clusters
  for (int iCurrentLayerClusterIndex = 0; iCurrentLayerClusterIndex < clustersCurrentLayer.size(); ++iCurrentLayerClusterIndex) {
    int storedTracklets{0};
    const Cluster& currentCluster{clustersCurrentLayer[iCurrentLayerClusterIndex]};
    const int4 selectedBinsRect{o2::its::getBinsRect(currentCluster, (int)Mode + 1, 0.f, 0.f, 100.f, phiCut / 2, utils)};
    if (selectedBinsRect.x >= 0) {
      int phiBinsNum{selectedBinsRect.w - selectedBinsRect.y + 1};
      if (phiBinsNum < 0) {
        phiBinsNum += PhiBins;
      }
      // loop on phi bins next layer
      for (int iPhiBin{selectedBinsRect.y}, iPhiCount{0}; iPhiCount < phiBinsNum && storedTracklets < maxTrackletsPerCluster; iPhiBin = ++iPhiBin == PhiBins ? 0 : iPhiBin, iPhiCount++) {
        const int firstBinIndex{utils.getBinIndex(selectedBinsRect.x, iPhiBin)};
        const int firstRowClusterIndex{indexTableNext[firstBinIndex]};
        const int maxRowClusterIndex{indexTableNext[firstBinIndex + ZBins]};
        // loop on clusters next layer
        for (int iNextLayerClusterIndex{firstRowClusterIndex}; iNextLayerClusterIndex < maxRowClusterIndex && iNextLayerClusterIndex < static_cast<int>(clustersNextLayer.size()) && storedTracklets < maxTrackletsPerCluster; ++iNextLayerClusterIndex) {
          if (usedClustersNextLayer[iNextLayerClusterIndex]) {
            continue;
          }
          const Cluster& nextCluster{clustersNextLayer[iNextLayerClusterIndex]};
          if (math_utils::isPhiDifferenceBelow(currentCluster.phi, nextCluster.phi, phiCut)) {
            if (storedTracklets < maxTrackletsPerCluster) {
              if constexpr (!EvalRun) {
                if constexpr (Mode == TrackletMode::Layer0Layer1) {
                  tracklets[rofFoundTrackletsOffsets[iCurrentLayerClusterIndex] + storedTracklets] = Tracklet{globalOffsetNextLayer + iNextLayerClusterIndex, globalOffsetCurrentLayer + iCurrentLayerClusterIndex, nextCluster, currentCluster, timErr};
                } else {
                  tracklets[rofFoundTrackletsOffsets[iCurrentLayerClusterIndex] + storedTracklets] = Tracklet{globalOffsetCurrentLayer + iCurrentLayerClusterIndex, globalOffsetNextLayer + iNextLayerClusterIndex, currentCluster, nextCluster, timErr};
                }
              }
              ++storedTracklets;
            }
          }
        }
      }
    }
    if constexpr (EvalRun) {
      foundTracklets[iCurrentLayerClusterIndex] += storedTracklets;
    } else {
      rofFoundTrackletsOffsets[iCurrentLayerClusterIndex] += storedTracklets;
    }
  }
}
 
void trackletSelectionKernelHost(
  const Cluster* clusters0,               // global layer 0 clusters
  const Cluster* clusters1,               // global layer 1 clusters
  gsl::span<unsigned char> usedClusters0, // global layer 0 used clusters
  gsl::span<unsigned char> usedClusters2, // global layer 2 used clusters
  const gsl::span<const Tracklet>& tracklets01,
  const gsl::span<const Tracklet>& tracklets12,
  bounded_vector<uint8_t>& usedTracklets,
  const gsl::span<int> foundTracklets01,
  const gsl::span<int> foundTracklets12,
  bounded_vector<Line>& lines,
  const gsl::span<const o2::MCCompLabel>& trackletLabels,
  bounded_vector<o2::MCCompLabel>& linesLabels,
  const int nLayer1Clusters,
  const float tanLambdaCut,
  const float phiCut,
  const int maxTracklets)
{
  int offset01{0}, offset12{0};
  for (int iCurrentLayerClusterIndex{0}; iCurrentLayerClusterIndex < nLayer1Clusters; ++iCurrentLayerClusterIndex) {
    int validTracklets{0};
    const int endTracklet01 = offset01 + foundTracklets01[iCurrentLayerClusterIndex];
    const int endTracklet12 = offset12 + foundTracklets12[iCurrentLayerClusterIndex];
    for (int iTracklet12{offset12}; iTracklet12 < endTracklet12 && validTracklets != maxTracklets; ++iTracklet12) {
      const auto& tracklet12{tracklets12[iTracklet12]};
      for (int iTracklet01{offset01}; iTracklet01 < endTracklet01 && validTracklets != maxTracklets; ++iTracklet01) {
        if (usedTracklets[iTracklet01]) {
          continue;
        }
 
        const auto& tracklet01{tracklets01[iTracklet01]};
        if (!tracklet01.getTimeStamp().isCompatible(tracklet12.getTimeStamp())) {
          continue;
        }
 
        const float deltaTanLambda{o2::gpu::GPUCommonMath::Abs(tracklet01.tanLambda - tracklet12.tanLambda)};
        if (deltaTanLambda >= tanLambdaCut) {
          continue;
        }
        if (math_utils::isPhiDifferenceBelow(tracklet01.phi, tracklet12.phi, phiCut) && validTracklets != maxTracklets) {
          usedClusters0[tracklet01.firstClusterIndex] = 1;
          usedClusters2[tracklet12.secondClusterIndex] = 1;
          usedTracklets[iTracklet01] = true;
          lines.emplace_back(tracklet01, clusters0, clusters1);
          if (!trackletLabels.empty()) {
            linesLabels.emplace_back(trackletLabels[iTracklet01]);
          }
          ++validTracklets;
        }
      }
    }
    offset01 += foundTracklets01[iCurrentLayerClusterIndex];
    offset12 += foundTracklets12[iCurrentLayerClusterIndex];
  }
}
} // namespace
 
template <int NLayers>
void VertexerTraits<NLayers>::initialise(const TrackingParameters& trackingParams)
{
  mTimeFrame->initialise(trackingParams, 3);
}
 
template <int NLayers>
void VertexerTraits<NLayers>::updateVertexingParameters(const std::vector<VertexingParameters>& vrtPar)
{
  mVrtParams = vrtPar;
  mIndexTableUtils.setTrackingParameters(vrtPar[0]);
  for (auto& par : mVrtParams) {
    par.phiSpan = static_cast<int>(std::ceil(mIndexTableUtils.getNphiBins() * par.phiCut / o2::constants::math::TwoPI));
    par.zSpan = static_cast<int>(std::ceil(par.zCut * mIndexTableUtils.getInverseZCoordinate(0)));
  }
}
 
// Main functions
template <int NLayers>
void VertexerTraits<NLayers>::computeTracklets(const int iteration)
{
  mTaskArena->execute([&] {
    tbb::parallel_for(0, mTimeFrame->getNrof(1), [&](const short pivotRofId) {
      bool skip = skipROF(iteration, pivotRofId);
      const auto& rofRange01 = mTimeFrame->getROFOverlapTableView().getOverlap(1, 0, pivotRofId);
      for (auto targetRofId = rofRange01.getFirstEntry(); targetRofId < rofRange01.getEntriesBound(); ++targetRofId) {
        const auto timeErr = mTimeFrame->getROFOverlapTableView().getTimeStamp(0, targetRofId, 1, pivotRofId);
        trackleterKernelHost<TrackletMode::Layer0Layer1, true>(
          !skip ? mTimeFrame->getClustersOnLayer(targetRofId, 0) : gsl::span<Cluster>(), // Clusters to be matched with the next layer in target rof
          !skip ? mTimeFrame->getClustersOnLayer(pivotRofId, 1) : gsl::span<Cluster>(),  // Clusters to be matched with the current layer in pivot rof
          mTimeFrame->getUsedClustersROF(targetRofId, 0),                                // Span of the used clusters in the target rof
          mTimeFrame->getIndexTable(targetRofId, 0).data(),                              // Index table to access the data on the next layer in target rof
          mVrtParams[iteration].phiCut,
          mTimeFrame->getTracklets()[0],                   // Flat tracklet buffer
          mTimeFrame->getNTrackletsCluster(pivotRofId, 0), // Span of the number of tracklets per each cluster in pivot rof
          mIndexTableUtils,
          timeErr,
          gsl::span<int>(), // Offset in the tracklet buffer
          0,
          0,
          mVrtParams[iteration].maxTrackletsPerCluster);
      }
      const auto& rofRange12 = mTimeFrame->getROFOverlapTableView().getOverlap(1, 2, pivotRofId);
      for (auto targetRofId = rofRange12.getFirstEntry(); targetRofId < rofRange12.getEntriesBound(); ++targetRofId) {
        const auto timeErr = mTimeFrame->getROFOverlapTableView().getTimeStamp(2, targetRofId, 1, pivotRofId);
        trackleterKernelHost<TrackletMode::Layer1Layer2, true>(
          !skip ? mTimeFrame->getClustersOnLayer(targetRofId, 2) : gsl::span<Cluster>(),
          !skip ? mTimeFrame->getClustersOnLayer(pivotRofId, 1) : gsl::span<Cluster>(),
          mTimeFrame->getUsedClustersROF(targetRofId, 2),
          mTimeFrame->getIndexTable(targetRofId, 2).data(),
          mVrtParams[iteration].phiCut,
          mTimeFrame->getTracklets()[1],
          mTimeFrame->getNTrackletsCluster(pivotRofId, 1), // Span of the number of tracklets per each cluster in pivot rof
          mIndexTableUtils,
          timeErr,
          gsl::span<int>(), // Offset in the tracklet buffer
          0,
          0,
          mVrtParams[iteration].maxTrackletsPerCluster);
      }
      mTimeFrame->getNTrackletsROF(pivotRofId, 0) = std::accumulate(mTimeFrame->getNTrackletsCluster(pivotRofId, 0).begin(), mTimeFrame->getNTrackletsCluster(pivotRofId, 0).end(), 0);
      mTimeFrame->getNTrackletsROF(pivotRofId, 1) = std::accumulate(mTimeFrame->getNTrackletsCluster(pivotRofId, 1).begin(), mTimeFrame->getNTrackletsCluster(pivotRofId, 1).end(), 0);
    });
 
    mTimeFrame->computeTrackletsPerROFScans();
    if (auto tot0 = mTimeFrame->getTotalTrackletsTF(0), tot1 = mTimeFrame->getTotalTrackletsTF(1);
        tot0 == 0 || tot1 == 0) {
      return;
    } else {
      mTimeFrame->getTracklets()[0].resize(tot0);
      mTimeFrame->getTracklets()[1].resize(tot1);
    }
 
    tbb::parallel_for(0, mTimeFrame->getNrof(1), [&](const short pivotRofId) {
      bool skip = skipROF(iteration, pivotRofId);
      const int globalOffsetPivot = mTimeFrame->getSortedStartIndex(pivotRofId, 1);
      const auto& rofRange01 = mTimeFrame->getROFOverlapTableView().getOverlap(1, 0, pivotRofId);
      for (auto targetRofId = rofRange01.getFirstEntry(); targetRofId < rofRange01.getEntriesBound(); ++targetRofId) {
        const auto timeErr = mTimeFrame->getROFOverlapTableView().getTimeStamp(0, targetRofId, 1, pivotRofId);
        trackleterKernelHost<TrackletMode::Layer0Layer1, false>(
          !skip ? mTimeFrame->getClustersOnLayer(targetRofId, 0) : gsl::span<Cluster>(),
          !skip ? mTimeFrame->getClustersOnLayer(pivotRofId, 1) : gsl::span<Cluster>(),
          mTimeFrame->getUsedClustersROF(targetRofId, 0),
          mTimeFrame->getIndexTable(targetRofId, 0).data(),
          mVrtParams[iteration].phiCut,
          mTimeFrame->getTracklets()[0],
          mTimeFrame->getNTrackletsCluster(pivotRofId, 0),
          mIndexTableUtils,
          timeErr,
          mTimeFrame->getExclusiveNTrackletsCluster(pivotRofId, 0),
          mTimeFrame->getSortedStartIndex(targetRofId, 0),
          globalOffsetPivot,
          mVrtParams[iteration].maxTrackletsPerCluster);
      }
      const auto& rofRange12 = mTimeFrame->getROFOverlapTableView().getOverlap(1, 2, pivotRofId);
      for (auto targetRofId = rofRange12.getFirstEntry(); targetRofId < rofRange12.getEntriesBound(); ++targetRofId) {
        const auto timeErr = mTimeFrame->getROFOverlapTableView().getTimeStamp(2, targetRofId, 1, pivotRofId);
        trackleterKernelHost<TrackletMode::Layer1Layer2, false>(
          !skip ? mTimeFrame->getClustersOnLayer(targetRofId, 2) : gsl::span<Cluster>(),
          !skip ? mTimeFrame->getClustersOnLayer(pivotRofId, 1) : gsl::span<Cluster>(),
          mTimeFrame->getUsedClustersROF(targetRofId, 2),
          mTimeFrame->getIndexTable(targetRofId, 2).data(),
          mVrtParams[iteration].phiCut,
          mTimeFrame->getTracklets()[1],
          mTimeFrame->getNTrackletsCluster(pivotRofId, 1),
          mIndexTableUtils,
          timeErr,
          mTimeFrame->getExclusiveNTrackletsCluster(pivotRofId, 1),
          mTimeFrame->getSortedStartIndex(targetRofId, 2),
          globalOffsetPivot,
          mVrtParams[iteration].maxTrackletsPerCluster);
      }
    });
  });
 
  if (mTimeFrame->hasMCinformation()) {
    for (const auto& trk : mTimeFrame->getTracklets()[0]) {
      o2::MCCompLabel label;
      int sortedId0{trk.firstClusterIndex};
      int sortedId1{trk.secondClusterIndex};
      for (const auto& lab0 : mTimeFrame->getClusterLabels(0, mTimeFrame->getClusters()[0][sortedId0].clusterId)) {
        for (const auto& lab1 : mTimeFrame->getClusterLabels(1, mTimeFrame->getClusters()[1][sortedId1].clusterId)) {
          if (lab0 == lab1 && lab0.isValid()) {
            label = lab0;
            break;
          }
        }
        if (label.isValid()) {
          break;
        }
      }
      mTimeFrame->getTrackletsLabel(0).emplace_back(label);
    }
  }
}
 
template <int NLayers>
void VertexerTraits<NLayers>::computeTrackletMatching(const int iteration)
{
  mTaskArena->execute([&] {
    tbb::combinable<int> totalLines{0};
    tbb::parallel_for(
      tbb::blocked_range<short>(0, (short)mTimeFrame->getNrof(1)),
      [&](const tbb::blocked_range<short>& Rofs) {
        for (short pivotRofId = Rofs.begin(); pivotRofId < Rofs.end(); ++pivotRofId) {
          if (mTimeFrame->getFoundTracklets(pivotRofId, 0).empty() || skipROF(iteration, pivotRofId)) {
            continue;
          }
          mTimeFrame->getLines(pivotRofId).reserve(std::min(mTimeFrame->getFoundTracklets(pivotRofId, 0).size(), mTimeFrame->getNTrackletsCluster(pivotRofId, 0).size() * constants::MaxSelectedTrackletsPerCluster));
          bounded_vector<uint8_t> usedTracklets(mTimeFrame->getFoundTracklets(pivotRofId, 0).size(), 0, mMemoryPool.get());
          trackletSelectionKernelHost(
            mTimeFrame->getClusters()[0].data(),
            mTimeFrame->getClusters()[1].data(),
            mTimeFrame->getUsedClusters(0),
            mTimeFrame->getUsedClusters(2),
            mTimeFrame->getFoundTracklets(pivotRofId, 0),
            mTimeFrame->getFoundTracklets(pivotRofId, 1),
            usedTracklets,
            mTimeFrame->getNTrackletsCluster(pivotRofId, 0),
            mTimeFrame->getNTrackletsCluster(pivotRofId, 1),
            mTimeFrame->getLines(pivotRofId),
            mTimeFrame->getLabelsFoundTracklets(pivotRofId, 0),
            mTimeFrame->getLinesLabel(pivotRofId),
            static_cast<int>(mTimeFrame->getClustersOnLayer(pivotRofId, 1).size()),
            mVrtParams[iteration].tanLambdaCut,
            mVrtParams[iteration].phiCut,
            constants::MaxSelectedTrackletsPerCluster);
          totalLines.local() += mTimeFrame->getLines(pivotRofId).size();
        }
      });
    mTimeFrame->setNLinesTotal(totalLines.combine(std::plus<int>()));
  });
 
  // from here on we do not use tracklets anymore, so let's free them
  deepVectorClear(mTimeFrame->getTracklets());
}
 
template <int NLayers>
void VertexerTraits<NLayers>::computeVertices(const int iteration)
{
  const int nRofs = mTimeFrame->getNrof(1);
  std::vector<std::vector<Vertex>> rofVertices(nRofs);
  std::vector<std::vector<VertexLabel>> rofLabels(nRofs);
  const float pairCut2 = mVrtParams[iteration].pairCut * mVrtParams[iteration].pairCut;
  const float duplicateZCut = mVrtParams[iteration].duplicateZCut > 0.f ? mVrtParams[iteration].duplicateZCut : std::max(4.f * mVrtParams[iteration].pairCut, 0.5f * mVrtParams[iteration].clusterCut);
  const float duplicateDistance2Cut = mVrtParams[iteration].duplicateDistance2Cut > 0.f ? mVrtParams[iteration].duplicateDistance2Cut : std::max(16.f * pairCut2, 0.0625f * mVrtParams[iteration].clusterCut * mVrtParams[iteration].clusterCut);
  line_vertexer::Settings settings;
  settings.beamX = mTimeFrame->getBeamX();
  settings.beamY = mTimeFrame->getBeamY();
  settings.pairCut = mVrtParams[iteration].pairCut;
  settings.pairCut2 = pairCut2;
  settings.clusterCut = mVrtParams[iteration].clusterCut;
  settings.coarseZWindow = mVrtParams[iteration].coarseZWindow;
  settings.seedDedupZCut = mVrtParams[iteration].seedDedupZCut;
  settings.refitDedupZCut = mVrtParams[iteration].refitDedupZCut;
  settings.duplicateZCut = duplicateZCut;
  settings.duplicateDistance2Cut = duplicateDistance2Cut;
  settings.finalSelectionZCut = mVrtParams[iteration].finalSelectionZCut;
  settings.maxZ = mVrtParams[iteration].maxZPositionAllowed;
  settings.seedMemberRadiusTime = mVrtParams[iteration].seedMemberRadiusTime;
  settings.seedMemberRadiusZ = mVrtParams[iteration].seedMemberRadiusZ;
  settings.memoryPool = mMemoryPool;
 
  const auto processROF = [&](const int rofId) {
    if (skipROF(iteration, rofId)) {
      return;
    }
    auto& lines = mTimeFrame->getLines(rofId);
    auto clusters = line_vertexer::buildClusters(std::span<const Line>{lines.data(), lines.size()}, settings);
    deepVectorClear(lines); // not needed after
    auto clusterBeamDistance2 = [&](const ClusterLines& cluster) {
      return (mTimeFrame->getBeamX() - cluster.getVertex()[0]) * (mTimeFrame->getBeamX() - cluster.getVertex()[0]) +
             (mTimeFrame->getBeamY() - cluster.getVertex()[1]) * (mTimeFrame->getBeamY() - cluster.getVertex()[1]);
    };
    auto clusterBetter = [&](const ClusterLines& lhs, const ClusterLines& rhs) {
      if (lhs.getSize() != rhs.getSize()) {
        return lhs.getSize() > rhs.getSize();
      }
      if (o2::gpu::GPUCommonMath::Abs(lhs.getAvgDistance2() - rhs.getAvgDistance2()) > constants::Tolerance) {
        return lhs.getAvgDistance2() < rhs.getAvgDistance2();
      }
      const auto lhsBeam = clusterBeamDistance2(lhs);
      const auto rhsBeam = clusterBeamDistance2(rhs);
      if (o2::gpu::GPUCommonMath::Abs(lhsBeam - rhsBeam) > constants::Tolerance) {
        return lhsBeam < rhsBeam;
      }
      return lhs.getVertex()[2] < rhs.getVertex()[2];
    };
 
    // Cluster deduplication by local non-maximum suppression in time/space
    std::sort(clusters.begin(), clusters.end(), clusterBetter);
    float minClusterZ = std::numeric_limits<float>::max();
    for (const auto& cluster : clusters) {
      minClusterZ = std::min(minClusterZ, cluster.getVertex()[2]);
    }
    bounded_vector<ClusterLines> deduplicated(mMemoryPool.get());
    deduplicated.reserve(clusters.size());
    std::unordered_map<int, std::vector<int>> keptByZBin;
    for (auto& candidate : clusters) {
      bool duplicate = false;
      const auto candidateZ = candidate.getVertex()[2];
      const auto zBin = static_cast<int>(std::floor((candidateZ - minClusterZ) / settings.duplicateZCut));
      for (int neighborBin = zBin - 1; neighborBin <= zBin + 1 && !duplicate; ++neighborBin) {
        const auto found = keptByZBin.find(neighborBin);
        if (found == keptByZBin.end()) {
          continue;
        }
        for (const auto ownerId : found->second) {
          const auto& owner = deduplicated[ownerId];
          if (!candidate.getTimeStamp().isCompatible(owner.getTimeStamp())) {
            continue;
          }
          if (o2::gpu::GPUCommonMath::Abs(candidate.getVertex()[2] - owner.getVertex()[2]) >= settings.duplicateZCut) {
            continue;
          }
          const auto dx = candidate.getVertex()[0] - owner.getVertex()[0];
          const auto dy = candidate.getVertex()[1] - owner.getVertex()[1];
          const auto dz = candidate.getVertex()[2] - owner.getVertex()[2];
          const auto distance2 = math_utils::SqSum(dx, dy, dz);
          if (distance2 < settings.duplicateDistance2Cut) {
            duplicate = true;
            break;
          }
        }
      }
      if (duplicate) {
        continue;
      }
 
      const auto ownerId = static_cast<int>(deduplicated.size());
      keptByZBin[zBin].push_back(ownerId);
      deduplicated.push_back(std::move(candidate));
    }
    clusters = std::move(deduplicated);
    int nClusters = static_cast<int>(clusters.size());
 
    // Vertex filtering with score-based local NMS
    std::sort(clusters.begin(), clusters.end(), clusterBetter);
    std::vector<int> candidateIndices;
    candidateIndices.reserve(nClusters);
    for (int iCluster{0}; iCluster < nClusters; ++iCluster) {
      const bool zCompatible = o2::gpu::GPUCommonMath::Abs(clusters[iCluster].getVertex()[2]) < mVrtParams[iteration].maxZPositionAllowed;
 
      if (zCompatible) {
        candidateIndices.push_back(iCluster);
      }
    }
 
    if (candidateIndices.empty()) {
      return;
    }
 
    auto countSharedLabels = [](const ClusterLines& lhs, const ClusterLines& rhs) {
      size_t shared = 0;
      auto lhsIt = lhs.getLabels().begin();
      auto rhsIt = rhs.getLabels().begin();
      while (lhsIt != lhs.getLabels().end() && rhsIt != rhs.getLabels().end()) {
        if (*lhsIt == *rhsIt) {
          ++shared;
          ++lhsIt;
          ++rhsIt;
        } else if (*lhsIt < *rhsIt) {
          ++lhsIt;
        } else {
          ++rhsIt;
        }
      }
      return shared;
    };
 
    float minCandidateZ = std::numeric_limits<float>::max();
    for (const auto clusterId : candidateIndices) {
      minCandidateZ = std::min(minCandidateZ, clusters[clusterId].getVertex()[2]);
    }
    std::unordered_map<int, std::vector<int>> selectedByZBin;
    std::vector<int> selectedIndices;
    selectedIndices.reserve(candidateIndices.size());
    for (const auto clusterId : candidateIndices) {
      const auto& candidate = clusters[clusterId];
      const auto candidateZ = candidate.getVertex()[2];
      const auto zBin = static_cast<int>((candidateZ - minCandidateZ) / settings.finalSelectionZCut);
      bool suppressed = false;
      for (int neighborBin = zBin - 1; neighborBin <= zBin + 1 && !suppressed; ++neighborBin) {
        const auto found = selectedByZBin.find(neighborBin);
        if (found == selectedByZBin.end()) {
          continue;
        }
        for (const auto selectedId : found->second) {
          const auto& selected = clusters[selectedId];
          if (!candidate.getTimeStamp().isCompatible(selected.getTimeStamp())) {
            continue;
          }
          const auto zDelta = o2::gpu::GPUCommonMath::Abs(candidateZ - selected.getVertex()[2]);
          const auto sharedLabels = countSharedLabels(candidate, selected);
          const auto minSize = std::min(candidate.getSize(), selected.getSize());
          const bool overlapDuplicate = sharedLabels > 0 && sharedLabels * 4 >= minSize;
          const bool strongZDuplicate = zDelta < settings.finalSelectionZCut;
          const bool clearlyBetterMultiplicity = selected.getSize() >= candidate.getSize() + 3;
          const bool clearlyBetterQuality = selected.getSize() > candidate.getSize() &&
                                            selected.getAvgDistance2() + constants::Tolerance < 0.8f * candidate.getAvgDistance2();
          const bool weakCandidate = clearlyBetterMultiplicity || clearlyBetterQuality;
          if (overlapDuplicate || (strongZDuplicate && weakCandidate)) {
            suppressed = true;
            break;
          }
        }
      }
      if (suppressed) {
        continue;
      }
      selectedByZBin[zBin].push_back(clusterId);
      selectedIndices.push_back(clusterId);
    }
 
    // sort vertices by their multiplicity to opt. suppress lower mult. debris
    std::vector<int> sortedIndices(selectedIndices.size());
    std::iota(sortedIndices.begin(), sortedIndices.end(), 0);
    std::sort(sortedIndices.begin(), sortedIndices.end(), [&selectedIndices, &clusters](int i, int j) {
      return clusters[selectedIndices[i]].getSize() > clusters[selectedIndices[j]].getSize();
    });
    for (const auto sortedId : sortedIndices) {
      const auto& cluster = clusters[selectedIndices[sortedId]];
      const auto beamDistance2 = clusterBeamDistance2(cluster);
      if (!(beamDistance2 < mVrtParams[iteration].NSigmaCut)) {
        continue;
      }
      if (cluster.getSize() < mVrtParams[iteration].clusterContributorsCut) {
        continue;
      }
      if (!rofVertices[rofId].empty() && cluster.getSize() < mVrtParams[iteration].suppressLowMultDebris) {
        continue;
      }
 
      Vertex vertex{cluster.getVertex().data(),
                    cluster.getRMS2(),
                    (ushort)cluster.getSize(),
                    cluster.getAvgDistance2()};
      if (mVrtParams[iteration].PassFlags[IterationStep::MarkVerticesAsUPC]) {
        vertex.setFlags(Vertex::UPCMode);
      }
      vertex.setTimeStamp(cluster.getTimeStamp());
      rofVertices[rofId].push_back(vertex);
      if (mTimeFrame->hasMCinformation()) {
        auto& lineLabels = mTimeFrame->getLinesLabel(rofId);
        bounded_vector<o2::MCCompLabel> labels(mMemoryPool.get());
        for (auto& index : cluster.getLabels()) {
          labels.push_back(lineLabels[index]);
        }
        const auto mainLabel = computeMain(labels);
        rofLabels[rofId].push_back(mainLabel);
      }
    }
  };
 
  if (mTaskArena->max_concurrency() <= 1) {
    for (int rofId{0}; rofId < nRofs; ++rofId) {
      processROF(rofId);
    }
  } else {
    mTaskArena->execute([&] {
      tbb::parallel_for(0, nRofs, [&](const int rofId) {
        processROF(rofId);
      });
    });
  }
  // add vertices, these anyways get sorted afterward
  for (int rofId{0}; rofId < nRofs; ++rofId) {
    for (auto& vertex : rofVertices[rofId]) {
      mTimeFrame->addPrimaryVertex(vertex);
    }
    if (mTimeFrame->hasMCinformation()) {
      for (auto& label : rofLabels[rofId]) {
        mTimeFrame->addPrimaryVertexLabel(label);
      }
    }
  }
}
 
template <int NLayers>
void VertexerTraits<NLayers>::addTruthSeedingVertices()
{
  LOGP(info, "Using truth seeds as vertices; will skip computations");
  const auto dc = o2::steer::DigitizationContext::loadFromFile("collisioncontext.root");
  const auto irs = dc->getEventRecords();
  int64_t roFrameBiasInBC = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::ITS>::Instance().getROFBiasInBC(1);
  int64_t roFrameLengthInBC = o2::itsmft::DPLAlpideParam<o2::detectors::DetID::ITS>::Instance().getROFLengthInBC(1);
  o2::steer::MCKinematicsReader mcReader(dc);
  const int iSrc = 0; // take only events from collision generator
  auto eveId2colId = dc->getCollisionIndicesForSource(iSrc);
  for (int iEve{0}; iEve < mcReader.getNEvents(iSrc); ++iEve) {
    const auto& ir = irs[eveId2colId[iEve]];
    if (!ir.isDummy()) { // do we need this, is this for diffractive events?
      const auto& eve = mcReader.getMCEventHeader(iSrc, iEve);
      auto bc = (ir - raw::HBFUtils::Instance().getFirstSampledTFIR()).toLong() - roFrameBiasInBC;
      if (bc < 0) { // event happened before TF
        continue;
      }
      Vertex vert;
      vert.getTimeStamp().setTimeStamp(bc);
      vert.getTimeStamp().setTimeStampError(roFrameLengthInBC / 2);
      // set minimum to 1 sometimes for diffractive events there is nothing acceptance
      vert.setNContributors(std::max(1L, std::ranges::count_if(mcReader.getTracks(iSrc, iEve), [](const auto& trk) {
                                       if (!trk.isPrimary() || trk.GetPt() < 0.05 || std::abs(trk.GetEta()) > 1.1) {
                                         return false;
                                       }
                                       const auto* p = o2::O2DatabasePDG::Instance()->GetParticle(trk.GetPdgCode());
                                       return (!p) ? false : p->Charge() != 0;
                                     })));
      vert.setXYZ((float)eve.GetX(), (float)eve.GetY(), (float)eve.GetZ());
      vert.setChi2(1); // not used as constraint
      constexpr float cov = 25e-4;
      vert.setSigmaX(cov);
      vert.setSigmaY(cov);
      vert.setSigmaZ(cov);
      mTimeFrame->addPrimaryVertex(vert);
      o2::MCCompLabel mcLbl(o2::MCCompLabel::maxTrackID(), iEve, iSrc, false);
      VertexLabel lbl(mcLbl, 1.0);
      mTimeFrame->addPrimaryVertexLabel(lbl);
    }
    mcReader.releaseTracksForSourceAndEvent(iSrc, iEve);
  }
  LOGP(info, "Imposed {} pv collisions from mc-truth", mTimeFrame->getPrimaryVertices().size());
}
 
template <int NLayers>
void VertexerTraits<NLayers>::setNThreads(int n, std::shared_ptr<tbb::task_arena>& arena)
{
  if (arena == nullptr) {
    mTaskArena = std::make_shared<tbb::task_arena>(std::abs(n));
    LOGP(info, "Setting seeding vertexer with {} threads.", n);
  } else {
    mTaskArena = arena;
  }
}
 
template <int NLayers>
bool VertexerTraits<NLayers>::skipROF(int iteration, int rof) const
{
  return mVrtParams[iteration].PassFlags[IterationStep::SkipROFsAboveThreshold] &&
         (int)mTimeFrame->getROFVertexLookupTableView().getVertices(1, rof).getEntries() > mVrtParams[iteration].vertPerRofThreshold;
}
 
template class VertexerTraits<7>;
} // namespace o2::its