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);
    ~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 setConsumes(edm::ConsumesCollector& sumes) {
      for( unsigned i = 0; i < modifiers_.size(); ++i )
        modifiers_[i]->setConsumes(sumes);
    }

    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) {
    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");
      ModifyObjectValueBase* plugin = 
        ModifyObjectValueFactory::get()->create(mname,iconf);
      if( nullptr != plugin ) {
        modifiers_.push_back(ModifierPointer(plugin));
      } else {
        throw cms::Exception("BadPluginName")
          << "The requested modifier: " << mname << " is not available!";
      }
    }
  }
}

#endif