testHepMCIteration.h

// testHepMCIteration.h
//
// garren@fnal.gov, May 2007
//
// Define methods and classes used by testHepMCIteration

bool IsPhoton( const HepMC::GenParticle* p ) { 
    if ( p->pdg_id() == 22 
         && p->momentum().perp() > 10. ) return true;
    return false;
}

bool IsWBoson( const HepMC::GenParticle* p ) { 
    if ( abs(p->pdg_id()) == 24 ) return true;
    return false;
}

class IsFinalState {
public:
    bool operator()( const HepMC::GenParticle* p ) { 
        if ( !p->end_vertex() && p->status()==1 ) return true;
        return false;
    }
};

class PrintPhoton {
public:
    PrintPhoton( std::ostream & os ) : m_out( os ) {}
    void operator()( const HepMC::GenParticle* p ) { 
        if ( IsPhoton(p) ) p->print( m_out );
    }
private:
   std::ostream & m_out;
};

class PrintParticle {
public:
    PrintParticle( std::ostream & os ) : m_out( os ) {}
    void operator()( const HepMC::GenParticle* p ) { 
        m_out << "\t";
        p->print( m_out );
    }
private:
   std::ostream & m_out;
};


class PrintChildren {
public:
    PrintChildren( std::ostream & os ) : m_out( os ) {}
    void operator()( HepMC::GenParticle* p ) { 
        // make a copy
        HepMC::GenParticle* cp = p;
        // use the copy and the original
        m_out << "\t\t\t (id,barcode,status) " 
              << cp->pdg_id() << " " 
              << p->barcode() << " "
              << cp->status() << std::endl;
    }
private:
   std::ostream & m_out;
};


class PrintDescendants {
public:
    PrintDescendants( std::ostream & os ) : m_out( os ) {}
    void operator()( const HepMC::GenParticle* p ) { 
        m_out << "\t\t";
        p->print( m_out );
    }
private:
   std::ostream & m_out;
};