ModifiedObjectProducer.h

Go to the documentation of this file.

#ifndef __PhysicsTools_PatAlgos_ModifiedObjectProducer_h__
#define __PhysicsTools_PatAlgos_ModifiedObjectProducer_h__

#include "PhysicsTools/PatAlgos/interface/ObjectModifier.h"

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/Common/interface/View.h"

#include <memory>

namespace pat {

  template <class T>
  class ModifiedObjectProducer : public edm::stream::EDProducer<> {
  public:
    typedef std::vector<T> Collection;
    typedef pat::ObjectModifier<T> Modifier;

    ModifiedObjectProducer(const edm::ParameterSet& conf) {
      //set our input source
      src_ = consumes<edm::View<T> >(conf.getParameter<edm::InputTag>("src"));
      //setup modifier
      const edm::ParameterSet& mod_config = conf.getParameter<edm::ParameterSet>("modifierConfig");
      modifier_ = std::make_unique<Modifier>(mod_config, consumesCollector());
      //declare products
      produces<Collection>();
    }
    ~ModifiedObjectProducer() override {}

    void beginLuminosityBlock(const edm::LuminosityBlock&, const edm::EventSetup& evs) final {
      modifier_->setEventContent(evs);
    }

    void produce(edm::Event& evt, const edm::EventSetup& evs) final {
      edm::Handle<edm::View<T> > input;
      auto output = std::make_unique<Collection>();

      evt.getByToken(src_, input);
      output->reserve(input->size());

      modifier_->setEvent(evt);

      for (auto itr = input->begin(); itr != input->end(); ++itr) {
        output->push_back(*itr);
        T& obj = output->back();
        modifier_->modify(obj);
      }

      evt.put(std::move(output));
    }

  private:
    edm::EDGetTokenT<edm::View<T> > src_;
    std::unique_ptr<Modifier> modifier_;
  };
}  // namespace pat

#endif