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 "DataFormats/Common/interface/ValueMap.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 reco::Candidate& originalParticle);
 
  std::string moduleLabel_;
 
  edm::EDGetTokenT<CandidateView> srcToken_;
  edm::EDGetTokenT<edm::ValueMap<float>> weightsToken_;
 
  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