PuppiPhoton.h

Go to the documentation of this file.

#ifndef CommonTools_Puppi_PuppiPhoton_h_
#define CommonTools_Puppi_PuppiPhoton_h_
// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Math/interface/PtEtaPhiMass.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/Common/interface/ValueMap.h"

// ------------------------------------------------------------------------------------------
class PuppiPhoton : public edm::stream::EDProducer<> {
public:
  explicit PuppiPhoton(const edm::ParameterSet &);
  ~PuppiPhoton() override;

  static void fillDescriptions(edm::ConfigurationDescriptions &descriptions);
  typedef math::XYZTLorentzVector LorentzVector;
  typedef std::vector<LorentzVector> LorentzVectorCollection;
  typedef edm::View<reco::Candidate> CandidateView;
  typedef std::vector<reco::PFCandidate> PFOutputCollection;
  typedef edm::View<reco::PFCandidate> PFView;

private:
  void produce(edm::Event &, const edm::EventSetup &) override;
  bool matchPFCandidate(const reco::Candidate *iPF, const reco::Candidate *iPho);
  edm::EDGetTokenT<CandidateView> tokenPFCandidates_;
  edm::EDGetTokenT<CandidateView> tokenPuppiCandidates_;
  edm::EDGetTokenT<CandidateView> tokenPhotonCandidates_;
  edm::EDGetTokenT<edm::ValueMap<std::vector<reco::PFCandidateRef> > > reco2pf_;
  edm::EDGetTokenT<edm::ValueMap<float> > tokenWeights_;
  edm::EDGetTokenT<edm::ValueMap<bool> > tokenPhotonId_;
  double pt_;
  double eta_;
  bool usePFRef_;
  bool usePFphotons_;
  bool runOnMiniAOD_;
  bool usePhotonId_;
  std::vector<double> dRMatch_;
  std::vector<int32_t> pdgIds_;
  std::unique_ptr<PFOutputCollection> corrCandidates_;
  double weight_;
  bool useValueMap_;
};
#endif