DecayedParticles.hh

// -*- C++ -*-
#ifndef RIVET_DecayedParticles_HH
#define RIVET_DecayedParticles_HH
 
#include "Rivet/Projections/ParticleFinder.hh"
 
namespace Rivet {
 

  class DecayedParticles : public Projection {
  public:


    DecayedParticles() { }
 
    DecayedParticles(const ParticleFinder& particles) {
      setName("DecayedParticles");
      declare(particles, "PARTICLES");
    }

    RIVET_DEFAULT_PROJ_CLONE(DecayedParticles);

    using Projection::operator=;

    virtual ~DecayedParticles() { }

 
  public:

    DecayedParticles& addStable(PdgId pid) {
      _stable.insert(pid);
      return *this;
    }

    const Particles& decaying() const {
      return _decaying;
    }

    const vector<unsigned int>& nStable() const {
      return _nStable;
    }

    const vector<map<PdgId, Particles>>& decayProducts() const {
      return _products;
    }

    bool modeMatches(size_t imode, unsigned int nstable, const map<PdgId, unsigned int>& prod) const {
      // same no of stable particles
      if (nstable != _nStable[imode]) return false;
      for (const auto& kv : prod) {
        // check if same decay products
        map<PdgId, Particles>::const_iterator iloc = _products[imode].find(kv.first);
        // same type of product
        if (iloc == _products[imode].end()) return false;
        // and same number
        if (iloc->second.size() != kv.second) return false;
      }
      // pass all the tests
      return true;
    }
 
  protected:

    virtual void project(const Event& e);

    virtual CmpState compare(const Projection& p) const;
 
  private:

    void findDecayProducts(const Particle& mother,
                           unsigned int& nstable,
                           map<PdgId, Particles>& products) const;
 
  private:

    set<PdgId> _stable;

    Particles _decaying;

    vector<unsigned int> _nStable;

    vector<map<PdgId, Particles>> _products;
  };
 
 
}
 
#endif