LeadingParticlesFinalState.hh

// -*- C++ -*-
#ifndef RIVET_LeadingParticlesFinalState_HH
#define RIVET_LeadingParticlesFinalState_HH
 
#include "Rivet/Event.hh"
#include "Rivet/Projection.hh"
#include "Rivet/Projections/FinalState.hh"
 
namespace Rivet {
 

  class LeadingParticlesFinalState : public FinalState {
  public:

    LeadingParticlesFinalState(const FinalState& fsp)
      :  FinalState(), _leading_only(false)
    {
      setName("LeadingParticlesFinalState");
      declare(fsp, "FS");
    }

    RIVET_DEFAULT_PROJ_CLONE(LeadingParticlesFinalState);

    using Projection::operator =;
 

    LeadingParticlesFinalState& addParticleId(PdgId id) {
      _ids.insert(id);
      return *this;
    }

    LeadingParticlesFinalState& addParticleIds(vector<PdgId> ids) {
      for (PdgId id : ids) _ids.insert(id);
      return *this;
    }

    LeadingParticlesFinalState& addParticleIdPair(PdgId id) {
      _ids.insert(id);
      _ids.insert(-id);
      return *this;
    }

    void setLeadingOnly(bool leadingonly) {
      _leading_only = leadingonly;
    }
 
    // /// Check if a particle of a particular ID was found in the current event
    // bool hasParticleId(const PdgId pid) const;
 
    // /// Get a particle of a particular ID (check it exists first)
    // bool get(const PdgId pid) const;
 

    void project(const Event& e);

    CmpState compare(const Projection& p) const;
 
 
  protected:

    std::set<long> _ids;
    bool _leading_only;
 
  };
 
 
}
 
#endif