// -*- 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