ObjectModifier.h

Go to the documentation of this file.

#ifndef __PhysicsTools_PatAlgos_ObjectModifier_h__
#define __PhysicsTools_PatAlgos_ObjectModifier_h__

#include "CommonTools/CandAlgos/interface/ModifyObjectValueBase.h"
#include <memory>

namespace pat {
  template<class T>
  class ObjectModifier {
  public:
    typedef std::unique_ptr<ModifyObjectValueBase> ModifierPointer;

    ObjectModifier(const edm::ParameterSet& conf, edm::ConsumesCollector&& cc);
    ~ObjectModifier() {}

    void setEvent(const edm::Event& event) {
      for( unsigned i = 0; i < modifiers_.size(); ++i )
        modifiers_[i]->setEvent(event);
    }

    void setEventContent(const edm::EventSetup& setup) {
      for( unsigned i = 0; i < modifiers_.size(); ++i )
        modifiers_[i]->setEventContent(setup);
    }

    void modify(T& obj) const {
      for( unsigned i = 0; i < modifiers_.size(); ++i )
        modifiers_[i]->modifyObject(obj);
    }

  private:
    std::vector<ModifierPointer> modifiers_;
  };

  template<class T>
  ObjectModifier<T>::ObjectModifier(const edm::ParameterSet& conf, edm::ConsumesCollector&& cc) {
    const std::vector<edm::ParameterSet>& mods = 
      conf.getParameterSetVector("modifications");
    for(unsigned i = 0; i < mods.size(); ++i ) {
      const edm::ParameterSet& iconf = mods[i];
      const std::string& mname = iconf.getParameter<std::string>("modifierName");
      modifiers_.emplace_back(ModifyObjectValueFactory::get()->create(mname,iconf,cc));
    }
  }
}

#endif