GenAccessIO.cxx

Go to the documentation of this file.

#include "TruthHelper/GenAccessIO.h"
#include "HepMC/GenEvent.h"
#include "HepMC/GenParticle.h"
#include "TruthHelper/GenAll.h"
#include "TruthHelper/GenIMCselector.h"
#include <algorithm>
#include <cmath>
 
StatusCode GenAccessIO::getMC( MCparticleCollection& mcParticles ) const {
  const GenIMCselector* selector = new GenAll();
  StatusCode            stat     = this->getMC( mcParticles, selector );
  return stat;
}
 
StatusCode GenAccessIO::getMC( MCparticleCollection& mcParticles,
                               const GenIMCselector* selector ) const {
 
  const McEventCollection* mcCollptr;
  StatusCode               stat = this->getDH( mcCollptr );
  if ( !stat ) { return StatusCode::FAILURE; }
 
  // Iterate over all McEvent records
  McEventCollection::const_iterator itr;
  for ( itr = mcCollptr->begin(); itr != mcCollptr->end(); ++itr )
  {
    // Access the HepMC record which is wrapped within McEvent
    const HepMC::GenEvent* genEvt = ( *itr );
    if ( genEvt == 0 ) return StatusCode::FAILURE;
 
    HepMC::GenEvent::particle_const_iterator it = genEvt->particles_begin();
    HepMC::GenEvent::particle_const_iterator en = genEvt->particles_end();
    for ( ; it != en; ++it )
    {
      if ( ( *selector )( *it ) ) { mcParticles.push_back( *it ); }
    }
  }
 
  return StatusCode::SUCCESS;
}