ProductFromFwdPtrProducer.h

Go to the documentation of this file.

#ifndef CommonTools_UtilAlgos_ProductFromFwdPtrProducer_h
#define CommonTools_UtilAlgos_ProductFromFwdPtrProducer_h

#include "FWCore/Framework/interface/global/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 "DataFormats/Common/interface/FwdPtr.h"
#include "CommonTools/UtilAlgos/interface/FwdPtrConversionFactory.h"
#include <vector>

namespace edm {

  template <class T, class H = ProductFromFwdPtrFactory<T> >
  class ProductFromFwdPtrProducer : public edm::global::EDProducer<> {
  public:
    explicit ProductFromFwdPtrProducer(edm::ParameterSet const& params)
        : srcToken_(consumes<std::vector<edm::FwdPtr<T> > >(params.getParameter<edm::InputTag>("src"))) {
      produces<std::vector<T> >();
    }

    ~ProductFromFwdPtrProducer() override {}

    void produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const override {
      edm::Handle<std::vector<edm::FwdPtr<T> > > hSrc;
      iEvent.getByToken(srcToken_, hSrc);

      std::unique_ptr<std::vector<T> > pOutput(new std::vector<T>);

      for (typename std::vector<edm::FwdPtr<T> >::const_iterator ibegin = hSrc->begin(), iend = hSrc->end(), i = ibegin;
           i != iend;
           ++i) {
        H factory;
        T t = factory(*i);
        pOutput->push_back(t);
      }

      iEvent.put(std::move(pOutput));
    }

  protected:
    const edm::EDGetTokenT<std::vector<edm::FwdPtr<T> > > srcToken_;
  };
}  // namespace edm

#endif