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);