ProductFromFwdPtrProducer

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55
#ifndef CommonTools_UtilAlgos_ProductFromFwdPtrProducer_h
#define CommonTools_UtilAlgos_ProductFromFwdPtrProducer_h

/**
  \class    edm::ProductFromFwdPtrProducer ProductFromFwdPtrProducer.h "CommonTools/UtilAlgos/interface/ProductFromFwdPtrProducer.h"
  \brief    Produces a list of objects "by value" that correspond to the FwdPtr's from an input collection.


  \author   Salvatore Rappoccio
*/

#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