ShiftedParticleProducer.cc

Go to the documentation of this file.

#include "PhysicsTools/PatUtils/plugins/ShiftedParticleProducer.h"
#include "FWCore/Utilities/interface/isFinite.h"

ShiftedParticleProducer::ShiftedParticleProducer(const edm::ParameterSet& cfg)
{
  moduleLabel_=cfg.getParameter<std::string>("@module_label");
  srcToken_ = consumes<CandidateView>(cfg.getParameter<edm::InputTag>("src"));
  shiftBy_ = cfg.getParameter<double>("shiftBy");

  if ( cfg.exists("binning") ) {
    typedef std::vector<edm::ParameterSet> vParameterSet;
    vParameterSet cfgBinning = cfg.getParameter<vParameterSet>("binning");
    for ( vParameterSet::const_iterator cfgBinningEntry = cfgBinning.begin();
      cfgBinningEntry != cfgBinning.end(); ++cfgBinningEntry ) {
      binning_.push_back(new binningEntryType(*cfgBinningEntry,moduleLabel_));
    }
  } else {
    std::string uncertainty = cfg.getParameter<std::string>("uncertainty");
    binning_.push_back(new binningEntryType(uncertainty, moduleLabel_));
  }

  produces<reco::CandidateCollection>();
}

ShiftedParticleProducer::~ShiftedParticleProducer()
{
  for(std::vector<binningEntryType*>::const_iterator it = binning_.begin();
    it != binning_.end(); ++it ) {
    delete (*it);
  }
}

void 
ShiftedParticleProducer::produce(edm::Event& evt, const edm::EventSetup& es)
{
  edm::Handle<CandidateView> originalParticles;
  evt.getByToken(srcToken_, originalParticles);

  auto shiftedParticles = std::make_unique<reco::CandidateCollection>();

  for(CandidateView::const_iterator originalParticle = originalParticles->begin();
      originalParticle != originalParticles->end(); ++originalParticle ) {

    double uncertainty = getUncShift(originalParticle);
    double shift = shiftBy_*uncertainty;

    reco::Candidate::LorentzVector shiftedParticleP4 = originalParticle->p4();
    //leave 0*nan = 0
    if (! (edm::isNotFinite(shift) && shiftedParticleP4.mag2()==0)) shiftedParticleP4 *= (1. + shift);

    std::unique_ptr<reco::Candidate> shiftedParticle = std::make_unique<reco::LeafCandidate>(*originalParticle);
    shiftedParticle->setP4(shiftedParticleP4);
    
    shiftedParticles->push_back( shiftedParticle.release() );
  }

  evt.put(std::move(shiftedParticles));
}

double 
ShiftedParticleProducer::getUncShift(const edm::View<reco::Candidate>::const_iterator& originalParticle) {
  double valx=0;
  double valy=0;
  for(std::vector<binningEntryType*>::iterator binningEntry=binning_.begin();
      binningEntry!=binning_.end(); ++binningEntry ) {
    if( (!(*binningEntry)->binSelection_) || (*(*binningEntry)->binSelection_)(*originalParticle) ) {

      if( (*binningEntry)->energyDep_ ) valx=originalParticle->energy();
      else valx=originalParticle->pt();

      valy=originalParticle->eta();
      return (*binningEntry)->binUncFormula_->Eval(valx, valy);
    }
  }
  return 0;
}

#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(ShiftedParticleProducer);