PrimaryParticles.hh

Go to the documentation of this file.

// -*- C++ -*-
#ifndef RIVET_PrimaryParticles_HH
#define RIVET_PrimaryParticles_HH

#include <Rivet/Projections/ParticleFinder.hh>
#include <Rivet/Tools/Cuts.hh>

namespace Rivet
{
  //
  class PrimaryParticles : public ParticleFinder
  {
  public:
    PrimaryParticles(std::initializer_list<int> pdgIds,
             const Cut& c=Cuts::open()) :
      ParticleFinder(c), _pdgIds(pdgIds) {
      setName("PrimaryParticles");
    }
    // Clone on the heap.
    DEFAULT_RIVET_PROJ_CLONE(PrimaryParticles);
      
    PrimaryParticles(const PrimaryParticles& other) : 
      ParticleFinder(other), _pdgIds(other._pdgIds) {
    }
    virtual int compare(const Projection& p) const
    {
      const PrimaryParticles* other = dynamic_cast<const PrimaryParticles*>(&p);
      if (!other) return UNDEFINED;
      if (_cuts != other->_cuts || _pdgIds != other->_pdgIds) return UNDEFINED;
      return EQUIVALENT;
      
    }
  protected:
    virtual void project(const Event& e);

    virtual bool isPrimary(const HepMC::GenParticle* p) const;
    virtual bool isIgnored(const HepMC::GenParticle* p) const;
    virtual bool isPrimaryPID(const HepMC::GenParticle* p) const;
    /*
     * Check if a particle @a p has decayed.
     *
     * @param p Pointer to HepMC particle 
     *
     * @return true if the particle has decayed according to the
     * status flag of the particle @a p
     */
    virtual bool hasDecayed(const HepMC::GenParticle* p) const;
    virtual bool isBeam(const HepMC::GenParticle* p) const;
    /*
     * Get the immediate ancestor of a particle.
     * 
     * @param p Particle for which to get the immediate ancestor 
     *
     * @return Pointer to immediate ancestor or null if there's no ancestor. 
     */
    const HepMC::GenParticle* ancestor(const HepMC::GenParticle* p) const;
    /*
     * Get the immediate ancestor of a particle, which is @e not an
     * ignored particle.
     * 
     * @param p Particle for which to get the immediate ancestor 
     *
     * @return Pointer to immediate ancestor or null if there's no ancestor. 
     */
    const HepMC::GenParticle* ancestor(const HepMC::GenParticle* p, bool) const;
    std::vector<int> _pdgIds;
  };
}
#endif