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

RecoParticleFlow/PFClusterProducer/plugins/ECALPFSeedCleaner.cc

49 строк2 KB

Jonathan Samudio

moved PFClusterProducer from beginLuminosityBlock to beginRun

01 фев 2023, 19:57

01 фев 2023, 19:57ed3768d

#include "CondFormats/DataRecord/interface/EcalPFSeedingThresholdsRcd.h"
#include "CondFormats/EcalObjects/interface/EcalPFSeedingThresholds.h"
#include "RecoParticleFlow/PFClusterProducer/interface/RecHitTopologicalCleanerBase.h"

class ECALPFSeedCleaner : public RecHitTopologicalCleanerBase {
public:
  ECALPFSeedCleaner(const edm::ParameterSet& conf, edm::ConsumesCollector& cc);
  ECALPFSeedCleaner(const ECALPFSeedCleaner&) = delete;
  ECALPFSeedCleaner& operator=(const ECALPFSeedCleaner&) = delete;

  void update(const edm::EventSetup&) override;

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

private:
  edm::ESHandle<EcalPFSeedingThresholds> ths_;
  edm::ESGetToken<EcalPFSeedingThresholds, EcalPFSeedingThresholdsRcd> thsToken_;
};

DEFINE_EDM_PLUGIN(RecHitTopologicalCleanerFactory, ECALPFSeedCleaner, "ECALPFSeedCleaner");

ECALPFSeedCleaner::ECALPFSeedCleaner(const edm::ParameterSet& conf, edm::ConsumesCollector& cc)
    : RecHitTopologicalCleanerBase(conf, cc), thsToken_(cc.esConsumes<edm::Transition::BeginRun>()) {}

void ECALPFSeedCleaner::update(const edm::EventSetup& iSetup) { ths_ = iSetup.getHandle(thsToken_); }

void ECALPFSeedCleaner::clean(const edm::Handle<reco::PFRecHitCollection>& input, std::vector<bool>& mask) {
  //need to run over energy sorted rechits, as this is order used in seeding step
  // this can cause ambiguity, isn't it better to index by detid ?
  auto const& hits = *input;
  std::vector<unsigned> ordered_hits(hits.size());
  for (unsigned i = 0; i < hits.size(); ++i)
    ordered_hits[i] = i;

  std::sort(ordered_hits.begin(), ordered_hits.end(), [&](unsigned i, unsigned j) {
    return hits[i].energy() > hits[j].energy();
  });

  for (const auto& idx : ordered_hits) {
    if (!mask[idx])
      continue;  // is it useful ?
    const reco::PFRecHit& rechit = hits[idx];

    float threshold = (*ths_)[rechit.detId()];
    if (rechit.energy() < threshold)
      mask[idx] = false;

  }  // rechit loop
}