FwdPtrProducer.h

Go to the documentation of this file.

#ifndef CommonTools_UtilAlgos_FwdPtrProducer_h
#define CommonTools_UtilAlgos_FwdPtrProducer_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 "DataFormats/Common/interface/FwdPtr.h"
#include "CommonTools/UtilAlgos/interface/FwdPtrConversionFactory.h"
#include <vector>
 
namespace edm {
 
  template <class T, class H = FwdPtrFromProductFactory<T> >
  class FwdPtrProducer : public edm::stream::EDProducer<> {
  public:
    explicit FwdPtrProducer(edm::ParameterSet const& params)
        : srcToken_(consumes<edm::View<T> >(params.getParameter<edm::InputTag>("src"))) {
      produces<std::vector<edm::FwdPtr<T> > >();
    }
 
    ~FwdPtrProducer() override {}
 
    void produce(edm::Event& iEvent, const edm::EventSetup& iSetup) override {
      edm::Handle<edm::View<T> > hSrc;
      iEvent.getByToken(srcToken_, hSrc);
 
      std::unique_ptr<std::vector<edm::FwdPtr<T> > > pOutputFwdPtr(new std::vector<edm::FwdPtr<T> >);
 
      for (typename edm::View<T>::const_iterator ibegin = hSrc->begin(), iend = hSrc->end(), i = ibegin; i != iend;
           ++i) {
        H factory;
        FwdPtr<T> ptr = factory(*hSrc, i - ibegin);
        pOutputFwdPtr->push_back(ptr);
      }
 
      iEvent.put(std::move(pOutputFwdPtr));
    }
 
  protected:
    edm::EDGetTokenT<edm::View<T> > srcToken_;
  };
}  // namespace edm
 
#endif