PFLinker.h

Go to the documentation of this file.

#ifndef RecoParticleFlow_PFProducer_PFLinker_h
#define RecoParticleFlow_PFProducer_PFLinker_h

#include <iostream>
#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/EgammaCandidates/interface/GsfElectronFwd.h"
#include "DataFormats/EgammaCandidates/interface/PhotonFwd.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "DataFormats/MuonReco/interface/MuonToMuonMap.h"
#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"

#include <string>

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

  explicit PFLinker(const edm::ParameterSet&);

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

 private:
  template<typename TYPE>
    edm::ValueMap<reco::PFCandidatePtr> fillValueMap(edm::Event & event,
                             std::string label,
                             edm::Handle<TYPE>& inputObjCollection,
                             const std::map<edm::Ref<TYPE>, reco::PFCandidatePtr> & mapToTheCandidate,
                             const edm::OrphanHandle<reco::PFCandidateCollection> & newPFCandColl) const;    
  
 private:
 
  std::vector<edm::EDGetTokenT<reco::PFCandidateCollection> >       inputTagPFCandidates_;

  edm::EDGetTokenT<reco::GsfElectronCollection>    inputTagGsfElectrons_;

  edm::EDGetTokenT<reco::PhotonCollection>         inputTagPhotons_;

  edm::InputTag muonTag_;
  edm::EDGetTokenT<reco::MuonCollection>      inputTagMuons_;
  edm::EDGetTokenT<reco::MuonToMuonMap> inputTagMuonMap_;
  std::string nameOutputPF_;

  std::string nameOutputElectronsPF_;

  std::string nameOutputPhotonsPF_;

  std::string nameOutputMergedPF_;

  bool producePFCandidates_;

  bool fillMuonRefs_;

  bool forceElectronsInHGCAL_;
};

DEFINE_FWK_MODULE(PFLinker);

#endif