TopInitSubset.cc

Go to the documentation of this file.

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

TopInitSubset::TopInitSubset(const edm::ParameterSet& cfg):
  srcToken_ ( consumes<reco::GenParticleCollection>(cfg.getParameter<edm::InputTag>( "src" ) ) )
{
  produces<reco::GenParticleCollection>();
}

TopInitSubset::~TopInitSubset()
{
}

void
TopInitSubset::produce(edm::Event& evt, const edm::EventSetup& setup)
{
  edm::Handle<reco::GenParticleCollection> src;
  evt.getByToken(srcToken_, src);

  const reco::GenParticleRefProd ref = evt.getRefBeforePut<reco::GenParticleCollection>();
  std::auto_ptr<reco::GenParticleCollection> sel( new reco::GenParticleCollection );

  //fill output collection
  fillOutput( *src, *sel );

  evt.put( sel );
}

void TopInitSubset::fillOutput(const reco::GenParticleCollection& src, reco::GenParticleCollection& sel)
{
  for(reco::GenParticleCollection::const_iterator t=src.begin(); t!=src.end(); ++t){
    if( std::abs(t->pdgId())==TopInitID::tID ){
      bool hasTopMother = false;
      for(unsigned idx=0; idx<t->numberOfMothers(); ++idx)
    if( std::abs(t->mother(idx)->pdgId())==TopInitID::tID )
      hasTopMother = true;
      if(hasTopMother)
    continue;
      for(unsigned idx=0; idx<t->numberOfMothers(); ++idx){
    reco::GenParticle* cand = new reco::GenParticle( t->mother(idx)->threeCharge(), t->mother(idx)->p4(),
                             t->mother(idx)->vertex(), t->mother(idx)->pdgId(),
                             t->mother(idx)->status(), false );
    std::auto_ptr<reco::GenParticle> ptr( cand );
    sel.push_back( *ptr );
      }
      break;
    }
  }
}