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