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

RecoParticleFlow/PFClusterProducer/plugins/Basic2DClusterForEachSeed.cc

57 строк2 KB

swagata87

review comments

09 ноя 2023, 21:56

09 ноя 2023, 21:562fbc415

#include "DataFormats/ParticleFlowReco/interface/PFRecHitFraction.h"
#include "RecoParticleFlow/PFClusterProducer/interface/InitialClusteringStepBase.h"
#include "CondFormats/DataRecord/interface/HcalPFCutsRcd.h"
#include "CondTools/Hcal/interface/HcalPFCutsHandler.h"

class Basic2DClusterForEachSeed : public InitialClusteringStepBase {
public:
  Basic2DClusterForEachSeed(const edm::ParameterSet& conf, edm::ConsumesCollector& cc)
      : InitialClusteringStepBase(conf, cc) {}
  ~Basic2DClusterForEachSeed() override = default;

  void buildClusters(const edm::Handle<reco::PFRecHitCollection>&,
                     const std::vector<bool>&,
                     const std::vector<bool>&,
                     reco::PFClusterCollection&,
                     const HcalPFCuts*) override;
};

DEFINE_EDM_PLUGIN(InitialClusteringStepFactory, Basic2DClusterForEachSeed, "Basic2DClusterForEachSeed");

void Basic2DClusterForEachSeed::buildClusters(const edm::Handle<reco::PFRecHitCollection>& input,
                                              const std::vector<bool>& rechitMask,
                                              const std::vector<bool>& seedable,
                                              reco::PFClusterCollection& output,
                                              const HcalPFCuts*) {
  auto const& hits = *input;

  // loop over seeds and make clusters
  reco::PFCluster cluster;
  for (unsigned int hit = 0; hit < hits.size(); ++hit) {
    if (!rechitMask[hit] || !seedable[hit])
      continue;  // if not seed, ignore.
    cluster.reset();

    // seed
    auto refhit = makeRefhit(input, hit);
    auto rhf = reco::PFRecHitFraction(refhit, 1.0);  // entire rechit energy should go to a cluster

    // add the hit to the cluster
    cluster.addRecHitFraction(rhf);

    // extract
    const auto rh_energy = refhit->energy();

    // fill cluster information
    cluster.setSeed(refhit->detId());
    cluster.setEnergy(rh_energy);
    cluster.setTime(refhit->time());
    cluster.setLayer(refhit->layer());
    cluster.setPosition(math::XYZPoint(refhit->position().x(), refhit->position().y(), refhit->position().z()));
    cluster.calculatePositionREP();
    cluster.setDepth(refhit->depth());

    output.push_back(cluster);

  }  // looping over seeds ends
}