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:
 
    using ParticleFinder::operator =;

    PrimaryParticles(std::initializer_list<int> pids, const Cut& c=Cuts::open())
      : ParticleFinder(c), _pdgIds(pids)
    {
      setName("PrimaryParticles");
    }
 
    // Clone on the heap.
    RIVET_DEFAULT_PROJ_CLONE(PrimaryParticles);

    using Projection::operator =;
 

    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;

    vector<int> _pdgIds;
 
  };
 
 
}
 
#endif