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, 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);
 
  private :
 
    set<PdgId> _stable;
 
    Particles _decaying;
 
    vector<unsigned int> _nStable;
 
    vector<map<PdgId,Particles> > _products;
  };
 
 
}
 
#endif