EGMEnergyVarProducer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    PhysicsTools/NanoAOD
// Class:      EGMEnergyVarProducer
//
//
// Original Author:  Emanuele Di Marco
//         Created:  Wed, 06 Sep 2017 12:34:38 GMT
//
//

// system include files
#include <memory>

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

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

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/StreamID.h"

#include "DataFormats/PatCandidates/interface/Electron.h"
#include "DataFormats/PatCandidates/interface/Photon.h"

#include "TLorentzVector.h"
#include "DataFormats/Common/interface/View.h"

#include "PhysicsTools/NanoAOD/interface/MatchingUtils.h"

//
// class declaration
//

template <typename T>
class EGMEnergyVarProducer : public edm::global::EDProducer<> {
public:
  explicit EGMEnergyVarProducer(const edm::ParameterSet& iConfig)
      : srcRaw_(consumes<edm::View<T>>(iConfig.getParameter<edm::InputTag>("srcRaw"))),
        srcCorr_(consumes<edm::View<T>>(iConfig.getParameter<edm::InputTag>("srcCorr"))) {
    produces<edm::ValueMap<float>>("eCorr");
  }
  ~EGMEnergyVarProducer() override{};

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

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

  // ----------member data ---------------------------

  edm::EDGetTokenT<edm::View<T>> srcRaw_;
  edm::EDGetTokenT<edm::View<T>> srcCorr_;
};

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// member functions
//

// ------------ method called to produce the data  ------------
template <typename T>
void EGMEnergyVarProducer<T>::produce(edm::StreamID streamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
  edm::Handle<edm::View<T>> srcRaw;
  iEvent.getByToken(srcRaw_, srcRaw);
  edm::Handle<edm::View<T>> srcCorr;
  iEvent.getByToken(srcCorr_, srcCorr);

  unsigned nSrcRaw = srcRaw->size();
  unsigned nSrcCorr = srcCorr->size();

  std::vector<float> eCorr(nSrcCorr, -1);

  for (unsigned int ir = 0; ir < nSrcRaw; ir++) {
    auto egm_raw = srcRaw->ptrAt(ir);
    for (unsigned int ic = 0; ic < nSrcCorr; ic++) {
      auto egm_corr = srcCorr->ptrAt(ic);
      if (matchByCommonParentSuperClusterRef(*egm_raw, *egm_corr)) {
        eCorr[ir] = egm_corr->energy() / egm_raw->energy();
        break;
      }
    }
  }

  std::unique_ptr<edm::ValueMap<float>> eCorrV(new edm::ValueMap<float>());
  edm::ValueMap<float>::Filler fillerCorr(*eCorrV);
  fillerCorr.insert(srcRaw, eCorr.begin(), eCorr.end());
  fillerCorr.fill();
  iEvent.put(std::move(eCorrV), "eCorr");
}

// ------------ method fills 'descriptions' with the allowed parameters for the module  ------------
template <typename T>
void EGMEnergyVarProducer<T>::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  desc.add<edm::InputTag>("srcRaw")->setComment("input raw physics object collection");
  desc.add<edm::InputTag>("srcCorr")->setComment("input corrected physics object collection");
  std::string modname;
  if (typeid(T) == typeid(pat::Electron))
    modname += "Electron";
  else if (typeid(T) == typeid(pat::Photon))
    modname += "Photon";
  modname += "EnergyVarProducer";
  descriptions.add(modname, desc);
}

typedef EGMEnergyVarProducer<pat::Electron> ElectronEnergyVarProducer;
typedef EGMEnergyVarProducer<pat::Photon> PhotonEnergyVarProducer;

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