ShiftedParticleProducer.h

Go to the documentation of this file.

#ifndef PhysicsTools_PatUtils_ShiftedParticleProducer_h
#define PhysicsTools_PatUtils_ShiftedParticleProducer_h

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "CommonTools/Utils/interface/StringCutObjectSelector.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/LeafCandidate.h"

#include <string>
#include <vector>

#include <TF2.h>

class ShiftedParticleProducer : public edm::stream::EDProducer<>
{
  typedef edm::View<reco::Candidate> CandidateView;

 public:

  explicit ShiftedParticleProducer(const edm::ParameterSet& cfg);
  ~ShiftedParticleProducer() override;
  
 private:

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

  double getUncShift(const CandidateView::const_iterator& originalParticle);


  std::string moduleLabel_;

  edm::EDGetTokenT<CandidateView> srcToken_;

  struct binningEntryType
  {
  binningEntryType(std::string uncertainty, std::string moduleLabel)
      : binSelection_(nullptr),
      binUncertainty_(uncertainty),
      energyDep_(false)
    {
      binUncFormula_ = std::make_unique<TF2>(std::string(moduleLabel).append("_uncFormula").c_str(), binUncertainty_.c_str() );
    }
  binningEntryType(const edm::ParameterSet& cfg, std::string moduleLabel)
  : binSelection_(new StringCutObjectSelector<reco::Candidate>(cfg.getParameter<std::string>("binSelection"))),
      binUncertainty_(cfg.getParameter<std::string>("binUncertainty")),
      energyDep_(false)
    {
      binUncFormula_ = std::make_unique<TF2>(std::string(moduleLabel).append("_uncFormula").c_str(), binUncertainty_.c_str() );
      if(cfg.exists("energyDependency") ) {energyDep_=cfg.getParameter<bool>("energyDependency");
      }
    }
    ~binningEntryType()
    {
    }
    std::unique_ptr<StringCutObjectSelector<reco::Candidate> > binSelection_;
    //double binUncertainty_;
    std::string binUncertainty_;
    std::unique_ptr<TF2> binUncFormula_;
    bool energyDep_;
  };
  std::vector<binningEntryType*> binning_;

  double shiftBy_; // set to +1.0/-1.0 for up/down variation of energy scale
};

#endif