ShallowCloneProducer.h

Go to the documentation of this file.

#ifndef CandAlgos_ShallowCloneProducer_h
#define CandAlgos_ShallowCloneProducer_h

#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/Candidate/interface/ShallowCloneCandidate.h"

template<typename C>
class ShallowCloneProducer : public edm::EDProducer {
public:
  explicit ShallowCloneProducer( const edm::ParameterSet& );
  ~ShallowCloneProducer();

private:
  virtual void produce( edm::Event&, const edm::EventSetup& );
  edm::EDGetTokenT<C> srcToken_;
};

template<typename C>
ShallowCloneProducer<C>::ShallowCloneProducer( const edm::ParameterSet& par ) :
  srcToken_( consumes<C>(par.template getParameter<edm::InputTag>( "src" ) ) ) {
  produces<reco::CandidateCollection>();
}

template<typename C>
ShallowCloneProducer<C>::~ShallowCloneProducer() {
}

template<typename C>
void ShallowCloneProducer<C>::produce( edm::Event& evt, const edm::EventSetup& ) {
  std::auto_ptr<reco::CandidateCollection> coll( new reco::CandidateCollection );
  edm::Handle<C> masterCollection;
  evt.getByToken( srcToken_, masterCollection );
  for( size_t i = 0; i < masterCollection->size(); ++i ) {
    reco::CandidateBaseRef masterClone( edm::Ref<C>( masterCollection, i ) );
    coll->push_back( new reco::ShallowCloneCandidate( masterClone ) );
  }
  evt.put( coll );
}

#endif