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

  virtual 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::auto_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( pOutput );
    }

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

#endif