ALPAKA_ACCELERATOR_NAMESPACE::SeedingTopoThresh Class Reference

Public Member Functions
template<typename TAcc , typename = std::enable_if_t<alpaka::isAccelerator<TAcc>>>
ALPAKA_FN_ACC void	operator() (const TAcc &acc, reco::PFClusteringVarsDeviceCollection::View pfClusteringVars, const reco::PFClusterParamsDeviceCollection::ConstView pfClusParams, const reco::PFRecHitHCALTopologyDeviceCollection::ConstView topology, const reco::PFRecHitHostCollection::ConstView pfRecHits, reco::PFClusterDeviceCollection::View clusterView, reco::PFRecHitFractionDeviceCollection::View fracView, uint32_t *__restrict__ nSeeds) const

Detailed Description

Definition at line 1084 of file PFClusterSoAProducerKernel.dev.cc.

Member Function Documentation

operator()()

template<typename TAcc , typename = std::enable_if_t<alpaka::isAccelerator<TAcc>>>

ALPAKA_FN_ACC void ALPAKA_ACCELERATOR_NAMESPACE::SeedingTopoThresh::operator() ( const TAcc &  acc, reco::PFClusteringVarsDeviceCollection::View  pfClusteringVars, const reco::PFClusterParamsDeviceCollection::ConstView  pfClusParams, const reco::PFRecHitHCALTopologyDeviceCollection::ConstView  topology, const reco::PFRecHitHostCollection::ConstView  pfRecHits, reco::PFClusterDeviceCollection::View  clusterView, reco::PFRecHitFractionDeviceCollection::View  fracView, uint32_t *__restrict__  nSeeds  ) const

inline

Definition at line 1087 of file PFClusterSoAProducerKernel.dev.cc.

References cms::cudacompat::atomicAdd(), hcalRecHitTable_cff::energy, PFLayer::HCAL_BARREL1, PFLayer::HCAL_ENDCAP, mps_fire::i, dqmdumpme::k, cms::alpakatools::once_per_grid(), HLT_2024v12_cff::pfRecHits, HLT_2024v12_cff::pt2, HLT_2024v12_cff::topology, and cms::alpakatools::uniform_elements().

                                                                       {
      const int nRH = pfRecHits.size();

      if (once_per_grid(acc)) {
        clusterView.size() = nRH;
      }

      for (auto i : uniform_elements(acc, nRH)) {
        // Initialize arrays
        pfClusteringVars[i].pfrh_isSeed() = 0;
        pfClusteringVars[i].rhCount() = 0;
        pfClusteringVars[i].topoSeedCount() = 0;
        pfClusteringVars[i].topoRHCount() = 0;
        pfClusteringVars[i].seedFracOffsets() = -1;
        pfClusteringVars[i].topoSeedOffsets() = -1;
        pfClusteringVars[i].topoSeedList() = -1;
        clusterView[i].seedRHIdx() = -1;

        int layer = pfRecHits[i].layer();
        int depthOffset = pfRecHits[i].depth() - 1;
        float energy = pfRecHits[i].energy();
        Position3 pos = Position3{pfRecHits[i].x(), pfRecHits[i].y(), pfRecHits[i].z()};
        float seedThreshold = 9999.;
        float topoThreshold = 9999.;

        if (topology.cutsFromDB()) {
          seedThreshold = topology[pfRecHits[i].denseId()].seedThreshold();
          topoThreshold = topology[pfRecHits[i].denseId()].noiseThreshold();
        } else {
          if (layer == PFLayer::HCAL_BARREL1) {
            seedThreshold = pfClusParams.seedEThresholdHB_vec()[depthOffset];
            topoThreshold = pfClusParams.topoEThresholdHB_vec()[depthOffset];
          } else if (layer == PFLayer::HCAL_ENDCAP) {
            seedThreshold = pfClusParams.seedEThresholdHE_vec()[depthOffset];
            topoThreshold = pfClusParams.topoEThresholdHE_vec()[depthOffset];
          }
        }

        // cmssdt.cern.ch/lxr/source/DataFormats/ParticleFlowReco/interface/PFRecHit.h#0108
        float pt2 = energy * energy * (pos.x * pos.x + pos.y * pos.y) / (pos.x * pos.x + pos.y * pos.y + pos.z * pos.z);

        // Seeding threshold test
        if ((layer == PFLayer::HCAL_BARREL1 && energy > seedThreshold && pt2 > pfClusParams.seedPt2ThresholdHB()) ||
            (layer == PFLayer::HCAL_ENDCAP && energy > seedThreshold && pt2 > pfClusParams.seedPt2ThresholdHE())) {
          pfClusteringVars[i].pfrh_isSeed() = 1;
          for (int k = 0; k < 4; k++) {  // Does this seed candidate have a higher energy than four neighbours
            if (pfRecHits[i].neighbours()(k) < 0)
              continue;
            if (energy < pfRecHits[pfRecHits[i].neighbours()(k)].energy()) {
              pfClusteringVars[i].pfrh_isSeed() = 0;
              break;
            }
          }
          if (pfClusteringVars[i].pfrh_isSeed())
            alpaka::atomicAdd(acc, nSeeds, 1u);
        }
        // Topo clustering threshold test

        if ((layer == PFLayer::HCAL_ENDCAP && energy > topoThreshold) ||
            (layer == PFLayer::HCAL_BARREL1 && energy > topoThreshold)) {
          pfClusteringVars[i].pfrh_passTopoThresh() = true;
          pfClusteringVars[i].pfrh_topoId() = i;
        } else {
          pfClusteringVars[i].pfrh_passTopoThresh() = false;
          pfClusteringVars[i].pfrh_topoId() = -1;
        }
      }
    }