overload + and - operations for valarray_base

cp-algorithms · May 3, 2024 · 4d117b8 · 4d117b8
1 parent 65e1680
commit 4d117b8
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/cp-algo/linalg/vector.hpp b/cp-algo/linalg/vector.hpp
@@ -32,6 +32,17 @@ namespace cp_algo::linalg {
         }
     };
 
+    template<class vec, typename base>
+    vec operator+(valarray_base<vec, base> const& a, valarray_base<vec, base> const& b) {
+        return *static_cast<std::valarray<base> const*>(&a) 
+             + *static_cast<std::valarray<base> const*>(&b);
+    }
+    template<class vec, typename base>
+    vec operator-(valarray_base<vec, base> const& a, valarray_base<vec, base> const& b) {
+        return *static_cast<std::valarray<base> const*>(&a) 
+             - *static_cast<std::valarray<base> const*>(&b);
+    }
+
     template<class vec, typename base>
     struct vec_base: valarray_base<vec, base> {
         using Base = valarray_base<vec, base>;