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