JetCorrFactorsProducer.h

Go to the documentation of this file.

#ifndef PhysicsTools_PatAlgos_JetCorrFactorsProducer_h
#define PhysicsTools_PatAlgos_JetCorrFactorsProducer_h

#include <map>
#include <string>
#include <boost/shared_ptr.hpp>

#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/EDProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

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


namespace pat {

  class JetCorrFactorsProducer : public edm::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() {};
    virtual void produce(edm::Event& event, const edm::EventSetup& setup);
    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, boost::shared_ptr<FactorizedJetCorrector>& corrector, int level);
    int numberOf(const edm::Handle<std::vector<reco::Vertex> >& primaryVertices);
    std::string payload();
    
  private:
    bool emf_;
    edm::InputTag src_;
    std::string type_;
    std::string label_;
    std::string payload_;
    edm::InputTag primaryVertices_;
    edm::InputTag rho_;
    bool useNPV_;
    bool useRho_;
    FlavorCorrLevelMap levels_;
  };

  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