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() override;

private:
  void produce(edm::Event&, const edm::EventSetup&) override;
  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::unique_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(std::move(coll));
}

#endif