MuPFIsoEmbedder.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    MuPFIsoEmbedder
// Class:      MuPFIsoEmbedder
// 
//
// Original Author:  Michail Bachtis,32 3-B16,+41227675567,
//         Created:  Thu Jun  9 01:36:17 CEST 2011
//
//


// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "RecoMuon/MuonIsolation/interface/MuPFIsoHelper.h"
#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

//
// class declaration
//

class MuPFIsoEmbedder : public edm::EDProducer {
   public:
  explicit MuPFIsoEmbedder(const edm::ParameterSet&);
      ~MuPFIsoEmbedder() override;

      static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

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

      // ----------member data ---------------------------
  edm::InputTag muons_;
  edm::EDGetTokenT<reco::MuonCollection> muonToken_;
  MuPFIsoHelper *helper_;

};




//
MuPFIsoEmbedder::MuPFIsoEmbedder(const edm::ParameterSet& iConfig):
  muons_(iConfig.getParameter<edm::InputTag>("src"))
{

  //decide what to read
    //Define a map between the isolation and the PSet for the PFHelper
    std::map<std::string,edm::ParameterSet> psetMap;

    //First declare what isolation you are going to read
    std::vector<std::string> isolationLabels;
    isolationLabels.push_back("pfIsolationR03");
    isolationLabels.push_back("pfIsoMeanDRProfileR03");
    isolationLabels.push_back("pfIsoSumDRProfileR03");
    isolationLabels.push_back("pfIsolationR04");
    isolationLabels.push_back("pfIsoMeanDRProfileR04");
    isolationLabels.push_back("pfIsoSumDRProfileR04");

    //Fill the label,pet map and initialize MuPFIsoHelper
    for( std::vector<std::string>::const_iterator label = isolationLabels.begin();label != isolationLabels.end();++label)
      psetMap[*label] =iConfig.getParameter<edm::ParameterSet >(*label); 
    helper_ = new MuPFIsoHelper(psetMap,consumesCollector());
    muonToken_ = consumes<reco::MuonCollection>(muons_);
  produces<reco::MuonCollection>();
}


MuPFIsoEmbedder::~MuPFIsoEmbedder()
{
 
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}


//
// member functions
//

// ------------ method called to produce the data  ------------
void
MuPFIsoEmbedder::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
   using namespace edm;
   using namespace reco;


   helper_->beginEvent(iEvent);
  
   edm::Handle<reco::MuonCollection > muons;
   iEvent.getByToken(muonToken_,muons);


   auto out = std::make_unique<MuonCollection>();

   for(unsigned int i=0;i<muons->size();++i) {
     MuonRef muonRef(muons,i);
     Muon muon = muons->at(i);
     helper_->embedPFIsolation(muon,muonRef);
     out->push_back(muon);
   }

   iEvent.put(std::move(out));
}

// ------------ method fills 'descriptions' with the allowed parameters for the module  ------------
void
MuPFIsoEmbedder::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  //The following says we do not know what parameters are allowed so do no validation
  // Please change this to state exactly what you do use, even if it is no parameters
  edm::ParameterSetDescription desc;
  desc.setUnknown();
  descriptions.addDefault(desc);
}

//define this as a plug-in
DEFINE_FWK_MODULE(MuPFIsoEmbedder);