PFMultiDepthClusterProducer.cc

Go to the documentation of this file.

#include "PFMultiDepthClusterProducer.h"

#ifdef PFLOW_DEBUG
#define LOGVERB(x) edm::LogVerbatim(x)
#define LOGWARN(x) edm::LogWarning(x)
#define LOGERR(x) edm::LogError(x)
#define LOGDRESSED(x) edm::LogInfo(x)
#else
#define LOGVERB(x) LogTrace(x)
#define LOGWARN(x) edm::LogWarning(x)
#define LOGERR(x) edm::LogError(x)
#define LOGDRESSED(x) LogDebug(x)
#endif

PFMultiDepthClusterProducer::PFMultiDepthClusterProducer(const edm::ParameterSet& conf)
{
  _clustersLabel = consumes<reco::PFClusterCollection>(conf.getParameter<edm::InputTag>("clustersSource")); 
  _pfClusterBuilder.reset(nullptr);
  const edm::ParameterSet& pfcConf = conf.getParameterSet("pfClusterBuilder");
  if( !pfcConf.empty() ) {
    const std::string& pfcName = pfcConf.getParameter<std::string>("algoName");
    PFCBB* pfcb = PFClusterBuilderFactory::get()->create(pfcName,pfcConf);
    _pfClusterBuilder.reset(pfcb);
  }
  // see if new need to apply corrections, setup if there.
  const edm::ParameterSet& cConf =  conf.getParameterSet("energyCorrector");
  if( !cConf.empty() ) {
    const std::string& cName = cConf.getParameter<std::string>("algoName");
    PFClusterEnergyCorrectorBase* eCorr =
      PFClusterEnergyCorrectorFactory::get()->create(cName,cConf);
    _energyCorrector.reset(eCorr);
  }
  
  produces<reco::PFClusterCollection>();
}

void PFMultiDepthClusterProducer::beginLuminosityBlock(const edm::LuminosityBlock& lumi, 
                         const edm::EventSetup& es) {
  _pfClusterBuilder->update(es);
  
}

void PFMultiDepthClusterProducer::produce(edm::Event& e, const edm::EventSetup& es) {
  _pfClusterBuilder->reset();

  edm::Handle<reco::PFClusterCollection> inputClusters;
  e.getByToken(_clustersLabel,inputClusters);  
  
  std::vector<bool> seedable;

  auto pfClusters = std::make_unique<reco::PFClusterCollection>();
  _pfClusterBuilder->buildClusters(*inputClusters, seedable, *pfClusters);
  LOGVERB("PFMultiDepthClusterProducer::produce()") << *_pfClusterBuilder;

  if( _energyCorrector ) {
    _energyCorrector->correctEnergies(*pfClusters);
  }
  e.put(std::move(pfClusters));

}