VectorN.hh

#ifndef RIVET_MATH_VECTORN
#define RIVET_MATH_VECTORN
 
#include "Rivet/Math/MathConstants.hh"
#include "Rivet/Math/MathUtils.hh"
 
#include "Rivet/Math/eigen3/Dense"
 
namespace Rivet {
 
 
  template <size_t N>
  class Vector;
  template <size_t N>
  class Matrix;
 
  template <size_t N>
  Vector<N> multiply(const Matrix<N>& a, const Vector<N>& b);
 
 
  template <size_t N>
  class Vector {
 
    template <size_t M>
    friend Vector<M> multiply(const Matrix<M>& a, const Vector<M>& b);
 
 
  public:
 
    Vector() : _vec(EVector::Zero()) { }
 
    const double& get(const size_t index) const {
      if (index >= N) {
        throw std::runtime_error("Tried to access an invalid vector index.");
      } else {
        return _vec(index);
      }
    }
 
    double& get(const size_t index) {
      if (index >= N) {
        throw std::runtime_error("Tried to access an invalid vector index.");
      } else {
        return _vec(index);
      }
    }
 
    const double& operator[](const size_t index) const {
      return get(index);
    }
 
    double& operator[](const size_t index) {
      return get(index);
    }
 
    Vector<N>& set(const size_t index, const double value) {
      if (index >= N) {
        throw std::runtime_error("Tried to access an invalid vector index.");
      } else {
        _vec[index] = value;
      }
      return *this;
    }
 
    constexpr size_t size() const {
      return N;
    }
 
    bool isZero(double tolerance=1E-5) const {
      for (size_t i=0; i < N; ++i) {
        if (! Rivet::isZero(_vec[i], tolerance) ) return false;
      }
      return true;
    }
 
    double mod2() const {
      double mod2 = 0.0;
      for (size_t i = 0; i < size(); ++i) {
        const double element = get(i);
        mod2 += element*element;
      }
      return mod2;
    }
 
    double mod() const {
      const double norm = mod2();
      //assert(norm >= 0); //< *should* be impossible
      return sqrt(norm);
    }
 
    Vector<N> operator-() const {
      Vector<N> rtn;
      rtn._vec = -_vec;
      return rtn;
    }
 
    bool operator==(const Vector<N>& a) const {
      return _vec == a._vec;
    }
 
    bool operator!=(const Vector<N>& a) const {
      return _vec != a._vec;
    }
 
    // bool operator<(const Vector<N>& a) const {
    //   return _vec < a._vec;
    // }
 
    // bool operator<=(const Vector<N>& a) const {
    //   return _vec <= a._vec;
    // }
 
    // bool operator>(const Vector<N>& a) const {
    //   return _vec > a._vec;
    // }
 
    // bool operator>=(const Vector<N>& a) const {
    //   return _vec >= a._vec;
    // }
 
    using EVector = RivetEigen::Matrix<double,N,1>;
    EVector _vec;
 
  };
 
 
 
 
 
  template <size_t N>
  inline const string toString(const Vector<N>& v) {
    std::ostringstream out;
    out << "(";
    for (size_t i = 0; i < v.size(); ++i) {
      out << (fabs(v[i]) < 1E-30 ? 0.0 : v[i]);
      if (i < v.size()-1) out << ", ";
    }
    out << ")";
    return out.str();
  }
 
  template <size_t N>
  inline std::ostream& operator<<(std::ostream& out, const Vector<N>& v) {
    out << toString(v);
    return out;
  }
 
 
 
 
 
  template <size_t N>
  inline bool fuzzyEquals(const Vector<N>& va, const Vector<N>& vb, double tolerance=1E-5) {
    for (size_t i = 0; i < N; ++i) {
      const double a = va.get(i);
      const double b = vb.get(i);
      if (!Rivet::fuzzyEquals(a, b, tolerance)) return false;
    }
    return true;
  }
 
 
  template <size_t N>
  inline bool isZero(const Vector<N>& v, double tolerance=1E-5) {
    return v.isZero(tolerance);
  }
 
 
}
 
#endif