JetCorrFactorsProducer.h

Go to the documentation of this file.

#ifndef PhysicsTools_PatAlgos_JetCorrFactorsProducer_h
#define PhysicsTools_PatAlgos_JetCorrFactorsProducer_h

#include <map>
#include <string>
#include <memory>

#include "DataFormats/Common/interface/View.h"
#include "DataFormats/JetReco/interface/Jet.h"

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

#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/PatCandidates/interface/JetCorrFactors.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "CondFormats/JetMETObjects/interface/FactorizedJetCorrector.h"
#include "CondFormats/JetMETObjects/interface/JetCorrectorParameters.h"


namespace pat {

  class JetCorrFactorsProducer : public edm::stream::EDProducer<> {
  public:
    typedef edm::ValueMap<pat::JetCorrFactors> JetCorrFactorsMap;
    typedef std::map<JetCorrFactors::Flavor, std::vector<std::string> > FlavorCorrLevelMap;

  public:
    explicit JetCorrFactorsProducer(const edm::ParameterSet& cfg);
    ~JetCorrFactorsProducer() override {};
    void produce(edm::Event& event, const edm::EventSetup& setup) override;
    static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);

  private:
    bool flavorDependent() const { return (levels_.size()>1); };
    std::vector<JetCorrectorParameters> params(const JetCorrectorParametersCollection& parameters, const std::vector<std::string>& levels) const;
    std::vector<std::string> expand(const std::vector<std::string>& levels, const JetCorrFactors::Flavor& flavor);
    float evaluate(edm::View<reco::Jet>::const_iterator& jet, const JetCorrFactors::Flavor& flavor, int level);
    int numberOf(const edm::Handle<std::vector<reco::Vertex> >& primaryVertices);
    std::string payload();

  private:
    bool emf_;
    edm::EDGetTokenT<edm::View<reco::Jet> > srcToken_;
    std::string type_;
    std::string label_;
    std::string payload_;
    std::vector<std::string> extraJPTOffset_;
    edm::InputTag primaryVertices_;
    edm::EDGetTokenT<std::vector<reco::Vertex> > primaryVerticesToken_;
    edm::InputTag rho_;
    edm::EDGetTokenT<double> rhoToken_;
    bool useNPV_;
    bool useRho_;
    FlavorCorrLevelMap levels_;
    unsigned long long cacheId_;
    std::map<JetCorrFactors::Flavor, std::unique_ptr<FactorizedJetCorrector> > correctors_;
    std::unique_ptr<FactorizedJetCorrector> extraJPTOffsetCorrector_;
  };

  inline int
  JetCorrFactorsProducer::numberOf(const edm::Handle<std::vector<reco::Vertex> >& primaryVertices)
  {
    int npv=0;
    for(std::vector<reco::Vertex>::const_iterator pv=primaryVertices->begin(); pv!=primaryVertices->end(); ++pv){
      if(pv->ndof()>=4) ++npv;
    }
    return npv;
  }
}

#endif