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/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 <string>

class PFLinker : public edm::EDProducer {
 public:

  explicit PFLinker(const edm::ParameterSet&);

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

  virtual void beginRun(edm::Run& run,const edm::EventSetup & es);

 private:

  template<typename T>
    bool fetchCollection(edm::Handle<T>& c, 
             const edm::InputTag& tag, 
             const edm::Event& iEvent) const ;
  
  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::InputTag>       inputTagPFCandidates_;

  edm::InputTag       inputTagGsfElectrons_;

  edm::InputTag       inputTagPhotons_;

  edm::InputTag       inputTagMuons_;

  std::string nameOutputPF_;

  std::string nameOutputElectronsPF_;

  std::string nameOutputPhotonsPF_;

  std::string nameOutputMergedPF_;

  bool producePFCandidates_;

  bool fillMuonRefs_;

};

#endif