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