PrimaryParticles.hh

// -*- 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> pids,
             const Cut& c=Cuts::open()) :
      ParticleFinder(c), _pdgIds(pids) {
      setName("PrimaryParticles");
    }
 
    // Clone on the heap.
    DEFAULT_RIVET_PROJ_CLONE(PrimaryParticles);
 
    PrimaryParticles(const PrimaryParticles& other) :
      ParticleFinder(other), _pdgIds(other._pdgIds) {
    }
 
    virtual CmpState compare(const Projection& p) const
    {
      const PrimaryParticles* other = dynamic_cast<const PrimaryParticles*>(&p);
      if (!other) return CmpState::NEQ;
      if (_cuts != other->_cuts || _pdgIds != other->_pdgIds) return CmpState::NEQ;
      return CmpState::EQ;
    }
 
 
  protected:
 
    virtual void project(const Event& e);
 
    virtual bool isPrimary(ConstGenParticlePtr p) const;
 
    virtual bool isIgnored(ConstGenParticlePtr p) const;
 
    virtual bool isPrimaryPID(ConstGenParticlePtr p) const;
 
    virtual bool hasDecayed(ConstGenParticlePtr p) const;
 
    virtual bool isBeam(ConstGenParticlePtr p) const;
 
    ConstGenParticlePtr ancestor(ConstGenParticlePtr p) const;
 
    ConstGenParticlePtr ancestor(ConstGenParticlePtr p, bool) const;
 
    std::vector<int> _pdgIds;
 
  };
 
 
}
 
#endif