diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/ipa/libipa/vector.cpp | 12 | ||||
| -rw-r--r-- | src/ipa/libipa/vector.h | 7 |
2 files changed, 19 insertions, 0 deletions
diff --git a/src/ipa/libipa/vector.cpp b/src/ipa/libipa/vector.cpp index d29c1642..8019f8cf 100644 --- a/src/ipa/libipa/vector.cpp +++ b/src/ipa/libipa/vector.cpp @@ -290,6 +290,18 @@ namespace ipa { */ /** + * \fn Vector::sum() const + * \brief Calculate the sum of all the vector elements + * \tparam R The type of the sum + * + * The type R of the sum defaults to the type T of the elements, but can be set + * explicitly to use a different type in case the type T would risk + * overflowing. + * + * \return The sum of all the vector elements + */ + +/** * \fn Vector<T, Rows> operator*(const Matrix<T, Rows, Cols> &m, const Vector<T, Cols> &v) * \brief Multiply a matrix by a vector * \tparam T Numerical type of the contents of the matrix and vector diff --git a/src/ipa/libipa/vector.h b/src/ipa/libipa/vector.h index 937a28e9..9bdd54b6 100644 --- a/src/ipa/libipa/vector.h +++ b/src/ipa/libipa/vector.h @@ -10,6 +10,7 @@ #include <array> #include <cmath> #include <functional> +#include <numeric> #include <optional> #include <ostream> @@ -239,6 +240,12 @@ public: return std::sqrt(length2()); } + template<typename R = T> + constexpr R sum() const + { + return std::accumulate(data_.begin(), data_.end(), R{}); + } + private: template<class BinaryOp> static constexpr Vector apply(const Vector &lhs, const Vector &rhs, BinaryOp op) |