GenericClusterImporter.cc

Go to the documentation of this file.

#include "RecoParticleFlow/PFProducer/interface/BlockElementImporterBase.h"
#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"
#include "DataFormats/ParticleFlowReco/interface/PFBlockElementCluster.h"

class GenericClusterImporter : public BlockElementImporterBase {
public:
  GenericClusterImporter(const edm::ParameterSet& conf, edm::ConsumesCollector& cc)
      : BlockElementImporterBase(conf, cc),
        _src(cc.consumes<reco::PFClusterCollection>(conf.getParameter<edm::InputTag>("source"))) {}

  void importToBlock(const edm::Event&, ElementList&) const override;

private:
  edm::EDGetTokenT<reco::PFClusterCollection> _src;
};

DEFINE_EDM_PLUGIN(BlockElementImporterFactory, GenericClusterImporter, "GenericClusterImporter");

void GenericClusterImporter::importToBlock(const edm::Event& e, BlockElementImporterBase::ElementList& elems) const {
  auto clusters = e.getHandle(_src);
  auto cbegin = clusters->cbegin();
  auto cend = clusters->cend();
  for (auto clus = cbegin; clus != cend; ++clus) {
    reco::PFBlockElement::Type type = reco::PFBlockElement::NONE;
    reco::PFClusterRef cref(clusters, std::distance(cbegin, clus));
    switch (clus->layer()) {
      case PFLayer::PS1:
        type = reco::PFBlockElement::PS1;
        break;
      case PFLayer::PS2:
        type = reco::PFBlockElement::PS2;
        break;
      case PFLayer::ECAL_BARREL:
      case PFLayer::ECAL_ENDCAP:
        type = reco::PFBlockElement::ECAL;
        break;
      case PFLayer::HCAL_BARREL1:
      case PFLayer::HCAL_ENDCAP:
        type = reco::PFBlockElement::HCAL;
        break;
      case PFLayer::HCAL_BARREL2:
        type = reco::PFBlockElement::HO;
        break;
      case PFLayer::HF_EM:
        type = reco::PFBlockElement::HFEM;
        break;
      case PFLayer::HF_HAD:
        type = reco::PFBlockElement::HFHAD;
        break;
      case PFLayer::HGCAL:
        type = reco::PFBlockElement::HGCAL;
        break;
      default:
        throw cms::Exception("InvalidPFLayer") << "Layer given, " << clus->layer() << " is not a valid PFLayer!";
    }
    reco::PFBlockElement* cptr = new reco::PFBlockElementCluster(cref, type);
    elems.emplace_back(cptr);
  }
}