PFPileUp.h

Go to the documentation of this file.

#ifndef PhysicsTools_PFCandProducer_PFPileUp_
#define PhysicsTools_PFCandProducer_PFPileUp_

// system include files
#include <memory>
#include <string>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"

#include "CommonTools/ParticleFlow/interface/PFPileUpAlgo.h"

class PFPileUp : public edm::stream::EDProducer<> {
 public:

  typedef std::vector< edm::FwdPtr<reco::PFCandidate> >  PFCollection;
  typedef edm::View<reco::PFCandidate>                   PFView;
  typedef std::vector<reco::PFCandidate>                 PFCollectionByValue;

  explicit PFPileUp(const edm::ParameterSet&);

  ~PFPileUp();

  virtual void produce(edm::Event&, const edm::EventSetup&) override;

 private:

  PFPileUpAlgo    pileUpAlgo_;

  edm::EDGetTokenT<PFCollection>   tokenPFCandidates_;
  edm::EDGetTokenT<PFView>   tokenPFCandidatesView_;

  edm::EDGetTokenT<reco::VertexCollection>   tokenVertices_;

  bool   enable_;

  bool   verbose_;

  bool   checkClosestZVertex_;

};

#endif