MuonMET.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    METProducers
// Class:      MuonMET
//
 
//____________________________________________________________________________||
#include "RecoMET/METProducers/interface/MuonMET.h"
 
#include "DataFormats/METReco/interface/METCollection.h"
#include "DataFormats/METReco/interface/CaloMETCollection.h"
 
//____________________________________________________________________________||
namespace cms {
 
  //____________________________________________________________________________||
  MuonMET::MuonMET(const edm::ParameterSet& iConfig)
      : alg_(), metTypeInputTag_(iConfig.getParameter<edm::InputTag>("metTypeInputTag")) {
    edm::InputTag muonsInputTag = iConfig.getParameter<edm::InputTag>("muonsInputTag");
    inputMuonToken_ = consumes<edm::View<reco::Muon> >(muonsInputTag);
 
    edm::InputTag muonDepValueMap = iConfig.getParameter<edm::InputTag>("muonMETDepositValueMapInputTag");
    inputValueMapMuonMetCorrToken_ = consumes<edm::ValueMap<reco::MuonMETCorrectionData> >(muonDepValueMap);
 
    edm::InputTag uncorMETInputTag = iConfig.getParameter<edm::InputTag>("uncorMETInputTag");
    if (metTypeInputTag_.label() == "CaloMET") {
      inputCaloMETToken_ = consumes<edm::View<reco::CaloMET> >(uncorMETInputTag);
      produces<reco::CaloMETCollection>();
    } else {
      inputMETToken_ = consumes<edm::View<reco::MET> >(uncorMETInputTag);
      produces<reco::METCollection>();
    }
  }
 
  MuonMET::MuonMET() : alg_() {}
 
  void MuonMET::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
    edm::Handle<edm::View<reco::Muon> > inputMuons;
    iEvent.getByToken(inputMuonToken_, inputMuons);
 
    edm::Handle<edm::ValueMap<reco::MuonMETCorrectionData> > vm_muCorrData_h;
 
    iEvent.getByToken(inputValueMapMuonMetCorrToken_, vm_muCorrData_h);
 
    if (metTypeInputTag_.label() == "CaloMET") {
      edm::Handle<edm::View<reco::CaloMET> > inputUncorMet;
      iEvent.getByToken(inputCaloMETToken_, inputUncorMet);
      auto output = std::make_unique<reco::CaloMETCollection>();
      alg_.run(*(inputMuons.product()), *(vm_muCorrData_h.product()), *(inputUncorMet.product()), &*output);
      iEvent.put(std::move(output));
    } else {
      edm::Handle<edm::View<reco::MET> > inputUncorMet;
      iEvent.getByToken(inputMETToken_, inputUncorMet);
      auto output = std::make_unique<reco::METCollection>();
      alg_.run(*(inputMuons.product()), *(vm_muCorrData_h.product()), *(inputUncorMet.product()), &*output);
      iEvent.put(std::move(output));
    }
  }
  //____________________________________________________________________________||
  DEFINE_FWK_MODULE(MuonMET);
 
}  // namespace cms
 
//____________________________________________________________________________||