IsGenType.cxx

Go to the documentation of this file.

#include <vector>
 
#include "HepMC/GenParticle.h"
#include "TruthHelper/IsGenType.h"
using std::abs;
using std::vector;
 
//****************************************************************
//*                    IsGenType                             *
//****************************************************************
IsGenType::IsGenType( int ParticleType ) { m_TypeList.push_back( ParticleType ); }
IsGenType::IsGenType( std::vector<int> ParticleTypeList ) {
  vector<int>::const_iterator itype = ParticleTypeList.begin();
  for ( ; itype != ParticleTypeList.end(); ++itype ) { m_TypeList.push_back( ( *itype ) ); }
}
IsGenType::IsGenType( const IsGenType& rhs )
    : GenIMCselector(), m_TypeList( rhs.m_TypeList ) {}
 
bool IsGenType::operator()( const Particle* const p ) const {
  vector<int>::const_iterator itype = m_TypeList.begin();
  for ( ; itype != m_TypeList.end(); ++itype )
  {
    if ( abs( p->pdg_id() ) == ( *itype ) ) return true;
  }
  return false;
}
 
bool IsGenType::operator()( const Particle& p ) const { return this->operator()( &p ); }
GenIMCselector* IsGenType::create() const { return new IsGenType( *this ); }