BeamConstraint.hh

Go to the documentation of this file.

// -*- C++ -*-
#ifndef RIVET_BeamConstraint_HH
#define RIVET_BeamConstraint_HH

#include "Rivet/Rivet.hh"
#include "Rivet/ParticleName.hh"
#include "Rivet/Particle.hh"
#include <iostream>


namespace Rivet {

  /// Find whether ParticleName @a p is compatible with the
  /// template ParticleName @a allowed. Effectively this is
  /// asking whether @a p is a subset of @a allowed.
  inline bool compatible(PdgId p, PdgId allowed) {
    //assert(p != ANY);
    return (allowed == ANY || p == allowed);
  }

  /// Find whether BeamPair @a pair is compatible with the template
  /// BeamPair @a allowedpair. This assesses whether either of the
  /// two possible pairings of @a pair's constituents is compatible.
  inline bool compatible(const BeamPair& pair, const BeamPair& allowedpair) {
    bool oneToOne = compatible(pair.first, allowedpair.first);
    bool twoToTwo = compatible(pair.second, allowedpair.second);
    bool oneToTwo = compatible(pair.first, allowedpair.second);
    bool twoToOne = compatible(pair.second, allowedpair.first);
    return (oneToOne && twoToTwo) || (oneToTwo && twoToOne);
  }


  /// Check particle compatibility of Particle pairs
  inline bool compatible(const ParticlePair& ppair,
                         const BeamPair& allowedpair) {
    return compatible(make_pdgid_pair(ppair.first.pdgId(),
                                      ppair.second.pdgId()), allowedpair);
  }
  /// Check particle compatibility of Particle pairs (for symmetric completeness)
  inline bool compatible(const BeamPair& allowedpair,
                         const ParticlePair& ppair) {
    return compatible(ppair, allowedpair);
  }


  /// Find whether a BeamPair @a pair is compatible with at least one template
  /// beam pair in a set @a allowedpairs.
  inline bool compatible(const BeamPair& pair, const set<BeamPair>& allowedpairs) {
    for (set<BeamPair>::const_iterator bp = allowedpairs.begin(); bp != allowedpairs.end(); ++bp) {
      if (compatible(pair, *bp)) return true;
    }
    return false;
  }

  /// Return the intersection of two sets of {@link BeamPair}s.
  inline set<BeamPair> intersection(const set<BeamPair>& a, const set<BeamPair>& b) {
    set<BeamPair> ret;
    for (set<BeamPair>::const_iterator bp = a.begin(); bp != a.end(); ++bp) {
      if (compatible(*bp, b)) ret.insert(*bp);
    }
    return ret;
  }


}

#endif