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() {}

    virtual 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_;
  };
}

#endif