EfficiencyLoader.h

Go to the documentation of this file.

#ifndef PhysicsTools_PatAlgos_EfficiencyLoader_h
#define PhysicsTools_PatAlgos_EfficiencyLoader_h
 
#include "DataFormats/PatCandidates/interface/PATObject.h"
#include "DataFormats/PatCandidates/interface/LookupTableRecord.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Common/interface/ValueMap.h"
 
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
 
namespace pat {
  namespace helper {
    class EfficiencyLoader {
    public:
      EfficiencyLoader() {}
 
      EfficiencyLoader(const edm::ParameterSet &iConfig, edm::ConsumesCollector &&iC);
 
      bool enabled() const { return !names_.empty(); }
 
      void newEvent(const edm::Event &event);
 
      template <typename T, typename R>
      void setEfficiencies(pat::PATObject<T> &obj, const R &originalRef) const;
 
    private:
      std::vector<std::string> names_;
      std::vector<edm::EDGetTokenT<edm::ValueMap<pat::LookupTableRecord> > > tokens_;
      std::vector<edm::Handle<edm::ValueMap<pat::LookupTableRecord> > > handles_;
    };  // class
 
    template <typename T, typename R>
    void EfficiencyLoader::setEfficiencies(pat::PATObject<T> &obj, const R &originalRef) const {
      for (size_t i = 0, n = names_.size(); i < n; ++i) {
        obj.setEfficiency(names_[i], (*handles_[i])[originalRef]);
      }
    }
 
  }  // namespace helper
}  // namespace pat
 
#endif