KinResolutionsLoader.h

Go to the documentation of this file.

#ifndef PhysicsTools_PatAlgos_KinResolutionsLoader_h
#define PhysicsTools_PatAlgos_KinResolutionsLoader_h

#include "DataFormats/PatCandidates/interface/PATObject.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"

#include "FWCore/Utilities/interface/ESGetToken.h"

#include "PhysicsTools/PatAlgos/interface/KinematicResolutionProvider.h"
#include "PhysicsTools/PatAlgos/interface/KinematicResolutionRcd.h"

namespace pat {
  namespace helper {
    class KinResolutionsLoader {
    public:
      KinResolutionsLoader() {}

      KinResolutionsLoader(const edm::ParameterSet &iConfig, edm::ConsumesCollector);

      bool enabled() const { return !patlabels_.empty(); }

      void newEvent(const edm::Event &event, const edm::EventSetup &setup);

      template <typename T>
      void setResolutions(pat::PATObject<T> &obj) const;

      static void fillDescription(edm::ParameterSetDescription &iDesc);

    private:
      std::vector<std::string> patlabels_;
      std::vector<edm::ESGetToken<KinematicResolutionProvider, KinematicResolutionRcd>> estokens_;
      std::vector<KinematicResolutionProvider const *> resolutions_;
    };  // class

    template <typename T>
    void KinResolutionsLoader::setResolutions(pat::PATObject<T> &obj) const {
      for (size_t i = 0, n = patlabels_.size(); i < n; ++i) {
        obj.setKinResolution(resolutions_[i]->getResolution(obj), patlabels_[i]);
      }
    }

  }  // namespace helper
}  // namespace pat

#endif