PFRecHitCreatorBase.h

Go to the documentation of this file.

#ifndef RecoParticleFlow_PFClusterProducer_PFRecHitCreatorBase_h
#define RecoParticleFlow_PFClusterProducer_PFRecHitCreatorBase_h
 
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
 
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "DataFormats/ParticleFlowReco/interface/PFLayer.h"
#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecHit.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecHitFwd.h"
 
#include "RecoParticleFlow/PFClusterProducer/interface/PFRecHitQTestBase.h"
#include <memory>
 
class PFRecHitCreatorBase {
public:
  PFRecHitCreatorBase() {}
  PFRecHitCreatorBase(const edm::ParameterSet& iConfig, edm::ConsumesCollector& iC) {
    std::vector<edm::ParameterSet> qTests = iConfig.getParameter<std::vector<edm::ParameterSet> >("qualityTests");
    for (auto& qTest : qTests) {
      std::string name = qTest.getParameter<std::string>("name");
      qualityTests_.emplace_back(PFRecHitQTestFactory::get()->create(name, qTest));
    }
  }
  virtual ~PFRecHitCreatorBase() = default;
 
  virtual void init(const edm::EventSetup& es) {}
 
  virtual void importRecHits(std::unique_ptr<reco::PFRecHitCollection>&,
                             std::unique_ptr<reco::PFRecHitCollection>&,
                             const edm::Event&,
                             const edm::EventSetup&) = 0;
 
protected:
  void beginEvent(const edm::Event& event, const edm::EventSetup& setup) {
    for (auto& qualityTest : qualityTests_)
      qualityTest->beginEvent(event, setup);
  }
 
  std::vector<std::unique_ptr<PFRecHitQTestBase> > qualityTests_;
};
 
#include "FWCore/PluginManager/interface/PluginFactory.h"
typedef edmplugin::PluginFactory<PFRecHitCreatorBase*(const edm::ParameterSet&, edm::ConsumesCollector&)>
    PFRecHitFactory;
#endif