L3MuonSumCaloPFIsolationProducer.cc

Go to the documentation of this file.

#include "L3MuonSumCaloPFIsolationProducer.h"

// Framework
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "DataFormats/Common/interface/Handle.h"

#include "FWCore/Framework/interface/ESHandle.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "DataFormats/Common/interface/AssociationMap.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidate.h"
#include "DataFormats/RecoCandidate/interface/RecoChargedCandidateFwd.h"

#include "DataFormats/RecoCandidate/interface/RecoChargedCandidateIsolation.h"

#include "DataFormats/Common/interface/ValueMap.h"

#include <string>

using namespace edm;
using namespace std;
using namespace reco;

L3MuonSumCaloPFIsolationProducer::L3MuonSumCaloPFIsolationProducer(const edm::ParameterSet& config) {
    
    recoChargedCandidateProducer_ = consumes<reco::RecoChargedCandidateCollection>(config.getParameter<edm::InputTag>("recoChargedCandidateProducer"));
    pfEcalClusterProducer_         = consumes<reco::RecoChargedCandidateIsolationMap>(config.getParameter<edm::InputTag>("pfEcalClusterProducer"));
    pfHcalClusterProducer_         = consumes<reco::RecoChargedCandidateIsolationMap>(config.getParameter<edm::InputTag>("pfHcalClusterProducer"));
    
    produces < edm::ValueMap<float> >();
    
}

L3MuonSumCaloPFIsolationProducer::~L3MuonSumCaloPFIsolationProducer()
{}

void L3MuonSumCaloPFIsolationProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
    edm::ParameterSetDescription desc;
    desc.add<edm::InputTag>("recoChargedCandidateProducer", edm::InputTag("hltL1SeededRecoChargedCandidatePF"));
    desc.add<edm::InputTag>("pfEcalClusterProducer", edm::InputTag("hltParticleFlowClusterECAL"));
    desc.add<edm::InputTag>("pfHcalClusterProducer", edm::InputTag("hltParticleFlowClusterHCAL"));
    descriptions.add(("hltL3MuonSumCaloPFIsolationProducer"), desc);
}

void L3MuonSumCaloPFIsolationProducer::produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const{
    
    edm::Handle<reco::RecoChargedCandidateCollection> recochargedcandHandle;
    iEvent.getByToken(recoChargedCandidateProducer_,recochargedcandHandle);
    
    edm::Handle<reco::RecoChargedCandidateIsolationMap> ecalIsolation;
    iEvent.getByToken (pfEcalClusterProducer_,ecalIsolation);
    
    edm::Handle<reco::RecoChargedCandidateIsolationMap> hcalIsolation;
    iEvent.getByToken (pfHcalClusterProducer_,hcalIsolation);
    
    auto caloIsoMap = std::make_unique<edm::ValueMap<float>>();
    std::vector<float> isoFloats(recochargedcandHandle->size(), 0);
    
    for (unsigned int iReco = 0; iReco < recochargedcandHandle->size(); iReco++) {
        reco::RecoChargedCandidateRef candRef(recochargedcandHandle, iReco);
        reco::RecoChargedCandidateIsolationMap::const_iterator mapiECAL = (*ecalIsolation).find( candRef );
        float valisoECAL = mapiECAL->val;
        reco::RecoChargedCandidateIsolationMap::const_iterator mapiHCAL = (*hcalIsolation).find( candRef );
        float valisoHCAL = mapiHCAL->val;
        float caloIso = valisoECAL + valisoHCAL;
        isoFloats[iReco] = caloIso;
    }
    
    edm::ValueMap<float> ::Filler isoFloatFiller(*caloIsoMap);
    isoFloatFiller.insert(recochargedcandHandle, isoFloats.begin(), isoFloats.end());
    isoFloatFiller.fill();
    iEvent.put(std::move(caloIsoMap));

}