StGenEventReco.cc

Go to the documentation of this file.

#include "TopQuarkAnalysis/TopEventProducers/interface/StGenEventReco.h"

StGenEventReco::StGenEventReco(const edm::ParameterSet& cfg):
  srcToken_ ( consumes<reco::GenParticleCollection>(cfg.getParameter<edm::InputTag>( "src"  ) ) ),
  initToken_( consumes<reco::GenParticleCollection>(cfg.getParameter<edm::InputTag>( "init" ) ) )
{
  produces<StGenEvent>();
}

StGenEventReco::~StGenEventReco()
{
}

void
StGenEventReco::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
  StGenEvent* genEvt = new StGenEvent( cands, initParts );
  std::unique_ptr<StGenEvent> gen( genEvt );
  evt.put(std::move(gen));
}