RecoParticleFlow/PFClusterProducer/plugins/LocalMaximumSeedFinder.cc - master - githubmirror/cmssw

RecoParticleFlow/PFClusterProducer/plugins/LocalMaximumSeedFinder.cc
212 строк8 KB
Shahzad Malik Muzaffar
[RECONSTRUCTION] [LLVM21] Apply clang-tidy changes
26 ноя 2025, 17:10
26 ноя 2025, 17:107863ba0
#include "CommonTools/Utils/interface/DynArray.h"
#include "DataFormats/Math/interface/deltaPhi.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "RecoParticleFlow/PFClusterProducer/interface/SeedFinderBase.h"
#include "CondFormats/DataRecord/interface/HcalPFCutsRcd.h"
#include "CondTools/Hcal/interface/HcalPFCutsHandler.h"

#include <algorithm>
#include <cfloat>
#include <tuple>
#include <unordered_map>
#include <queue>

class LocalMaximumSeedFinder final : public SeedFinderBase {
public:
  LocalMaximumSeedFinder(const edm::ParameterSet& conf);
  LocalMaximumSeedFinder(const LocalMaximumSeedFinder&) = delete;
  LocalMaximumSeedFinder& operator=(const LocalMaximumSeedFinder&) = delete;

  void findSeeds(const edm::Handle<reco::PFRecHitCollection>& input,
                 const std::vector<bool>& mask,
                 std::vector<bool>& seedable,
                 const HcalPFCuts*) override;

private:
  const int _nNeighbours;

  const std::unordered_map<std::string, int> _layerMap;

  typedef std::tuple<std::vector<int>, std::vector<double>, std::vector<double> > I3tuple;

  std::array<I3tuple, 35> _thresholds;
  static constexpr int layerOffset = 15;

  static constexpr double detacut = 0.01;
  static constexpr double dphicut = 0.01;
};

DEFINE_EDM_PLUGIN(SeedFinderFactory, LocalMaximumSeedFinder, "LocalMaximumSeedFinder");

namespace {
  const reco::PFRecHit::Neighbours _noNeighbours(nullptr, 0);
}

LocalMaximumSeedFinder::LocalMaximumSeedFinder(const edm::ParameterSet& conf)
    : SeedFinderBase(conf),
      _nNeighbours(conf.getParameter<int>("nNeighbours")),
      _layerMap({{"PS2", (int)PFLayer::PS2},
                 {"PS1", (int)PFLayer::PS1},
                 {"ECAL_ENDCAP", (int)PFLayer::ECAL_ENDCAP},
                 {"ECAL_BARREL", (int)PFLayer::ECAL_BARREL},
                 {"NONE", (int)PFLayer::NONE},
                 {"HCAL_BARREL1", (int)PFLayer::HCAL_BARREL1},
                 {"HCAL_BARREL2_RING0", (int)PFLayer::HCAL_BARREL2},
                 // hack to deal with ring1 in HO
                 {"HCAL_BARREL2_RING1", 19},
                 {"HCAL_ENDCAP", (int)PFLayer::HCAL_ENDCAP},
                 {"HF_EM", (int)PFLayer::HF_EM},
                 {"HF_HAD", (int)PFLayer::HF_HAD}}) {
  const std::vector<edm::ParameterSet>& thresholds = conf.getParameterSetVector("thresholdsByDetector");
  for (const auto& pset : thresholds) {
    const std::string& det = pset.getParameter<std::string>("detector");
    std::vector<int> depths;
    std::vector<double> thresh_E;
    std::vector<double> thresh_pT;
    std::vector<double> thresh_pT2;

    if (det == std::string("HCAL_BARREL1") || det == std::string("HCAL_ENDCAP")) {
      depths = pset.getParameter<std::vector<int> >("depths");
      thresh_E = pset.getParameter<std::vector<double> >("seedingThreshold");
      thresh_pT = pset.getParameter<std::vector<double> >("seedingThresholdPt");
      if (thresh_E.size() != depths.size() || thresh_pT.size() != depths.size()) {
        throw cms::Exception("InvalidGatheringThreshold") << "gatheringThresholds mismatch with the numbers of depths";
      }
    } else {
      depths.push_back(0);
      thresh_E.push_back(pset.getParameter<double>("seedingThreshold"));
      thresh_pT.push_back(pset.getParameter<double>("seedingThresholdPt"));
    }

    thresh_pT2.reserve(thresh_pT.size());
    for (unsigned int i = 0; i < thresh_pT.size(); ++i) {
      thresh_pT2.push_back(thresh_pT[i] * thresh_pT[i]);
    }

    auto entry = _layerMap.find(det);
    if (entry == _layerMap.end()) {
      throw cms::Exception("InvalidDetectorLayer") << "Detector layer : " << det << " is not in the list of recognized"
                                                   << " detector layers!";
    }

    _thresholds[entry->second + layerOffset] = std::make_tuple(depths, thresh_E, thresh_pT2);
  }
}

// the starting state of seedable is all false!
void LocalMaximumSeedFinder::findSeeds(const edm::Handle<reco::PFRecHitCollection>& input,
                                       const std::vector<bool>& mask,
                                       std::vector<bool>& seedable,
                                       const HcalPFCuts* hcalCuts) {
  auto nhits = input->size();
  initDynArray(bool, nhits, usable, true);
  //need to run over energy sorted rechits
  declareDynArray(float, nhits, energies);
  unInitDynArray(int, nhits, qst);  // queue storage
  auto cmp = [&](int i, int j) { return energies[i] < energies[j]; };
  std::priority_queue<int, DynArray<int>, decltype(cmp)> ordered_hits(cmp, std::move(qst));

  for (unsigned i = 0; i < nhits; ++i) {
    if (!mask[i])
      continue;  // cannot seed masked objects
    auto const& maybeseed = (*input)[i];
    energies[i] = maybeseed.energy();
    int seedlayer = (int)maybeseed.layer();
    if (seedlayer == PFLayer::HCAL_BARREL2 && std::abs(maybeseed.positionREP().eta()) > 0.34) {
      seedlayer = 19;
    }
    auto const& thresholds = _thresholds[seedlayer + layerOffset];

    double thresholdE = 0.;
    double thresholdPT2 = 0.;

    for (unsigned int j = 0; j < (std::get<2>(thresholds)).size(); ++j) {
      int depth = std::get<0>(thresholds)[j];
      if ((seedlayer == PFLayer::HCAL_BARREL1 && maybeseed.depth() == depth) ||
          (seedlayer == PFLayer::HCAL_ENDCAP && maybeseed.depth() == depth) ||
          (seedlayer != PFLayer::HCAL_BARREL1 && seedlayer != PFLayer::HCAL_ENDCAP)) {
        thresholdE = std::get<1>(thresholds)[j];
        thresholdPT2 = std::get<2>(thresholds)[j];
      }
    }

    if (hcalCuts != nullptr) {  // this means, cutsFromDB is set to True in PFClusterProducer.cc
      if ((seedlayer == PFLayer::HCAL_BARREL1) || (seedlayer == PFLayer::HCAL_ENDCAP)) {
        HcalDetId thisId = maybeseed.detId();
        const HcalPFCut* item = hcalCuts->getValues(thisId.rawId());
        thresholdE = item->seedThreshold();
      }
    }

    if (maybeseed.energy() < thresholdE || maybeseed.pt2() < thresholdPT2)
      usable[i] = false;
    if (!usable[i])
      continue;
    ordered_hits.push(i);
  }

  while (!ordered_hits.empty()) {
    auto idx = ordered_hits.top();
    ordered_hits.pop();
    if (!usable[idx])
      continue;
    //get the neighbours of this seed
    auto const& maybeseed = (*input)[idx];
    reco::PFRecHit::Neighbours myNeighbours;
    switch (_nNeighbours) {
      case -1:
        myNeighbours = maybeseed.neighbours();
        break;
      case 0:  // for HF clustering
        myNeighbours = _noNeighbours;
        break;
      case 4:
        myNeighbours = maybeseed.neighbours4();
        break;
      case 8:
        myNeighbours = maybeseed.neighbours8();
        break;
      default:
        throw cms::Exception("InvalidConfiguration") << "LocalMaximumSeedFinder only accepts nNeighbors = {-1,0,4,8}";
    }
    seedable[idx] = true;
    for (auto neighbour : myNeighbours) {
      if (!mask[neighbour])
        continue;
      if (energies[neighbour] > energies[idx]) {
        //        std::cout << "how this can be?" << std::endl;
        seedable[idx] = false;
        break;
      }
    }
    if (seedable[idx]) {
      for (auto neighbour : myNeighbours) {
        //
        // For HCAL,
        // even if channel a is a neighbor of channel b, channel b may not be a neighbor of channel a.
        // So, perform additional checks to ensure making a hit unusable for seeding is safe.
        int seedlayer = (int)maybeseed.layer();
        switch (seedlayer) {
          case PFLayer::HCAL_BARREL1:
          case PFLayer::HCAL_ENDCAP:
          case PFLayer::HF_EM:   // with the current HF setting, we won't see this case
                                 // but this can come in if we change _nNeighbours for HF.
          case PFLayer::HF_HAD:  // same as above
            // HO has only one depth and eta-phi segmentation is regular, so no need to make this check
            auto const& nei = (*input)[neighbour];
            if (maybeseed.depth() != nei.depth())
              continue;  // masking is done only if the neighbor is on the same depth layer as the seed
            if (std::abs(deltaPhi(maybeseed.positionREP().phi(), nei.positionREP().phi())) > dphicut &&
                std::abs(maybeseed.positionREP().eta() - nei.positionREP().eta()) > detacut)
              continue;  // masking is done only if the neighbor is on the swiss-cross w.r.t. the seed
            break;
        }

        usable[neighbour] = false;

      }  // for-loop
    }
  }

  LogDebug("LocalMaximumSeedFinder") << " found " << std::count(seedable.begin(), seedable.end(), true) << " seeds";
}