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