TtGenEventReco.cc

Go to the documentation of this file.

#include "AnalysisDataFormats/TopObjects/interface/TtGenEvent.h"
 
#include "TopQuarkAnalysis/TopEventProducers/interface/TtGenEventReco.h"
 
TtGenEventReco::TtGenEventReco(const edm::ParameterSet& cfg)
    : srcToken_(consumes<reco::GenParticleCollection>(cfg.getParameter<edm::InputTag>("src"))),
      initToken_(consumes<reco::GenParticleCollection>(cfg.getParameter<edm::InputTag>("init"))) {
  produces<TtGenEvent>();
}
 
TtGenEventReco::~TtGenEventReco() {}
 
void TtGenEventReco::produce(edm::Event& evt, const edm::EventSetup& setup) {
  edm::Handle<reco::GenParticleCollection> parts;
  evt.getByToken(srcToken_, parts);
 
  edm::Handle<reco::GenParticleCollection> inits;
  evt.getByToken(initToken_, inits);
 
  //add TopDecayTree
  reco::GenParticleRefProd cands(parts);
 
  //add InitialStatePartons
  reco::GenParticleRefProd initParts(inits);
 
  //add genEvt to the output stream
  TtGenEvent* genEvt = new TtGenEvent(cands, initParts);
  std::unique_ptr<TtGenEvent> gen(genEvt);
  evt.put(std::move(gen));
}