feat: [] indexing operator for vec<> class

mrdcvlsc · Aug 1, 2023 · fad7420 · fad7420
1 parent c0e5cb6
commit fad7420
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/include/cyfre/vectors.hpp b/include/cyfre/vectors.hpp
@@ -35,6 +35,9 @@ namespace cyfre {
 
     constexpr vec(std::initializer_list<T> sequence);
 
+    constexpr T &operator[](size_t i);
+    constexpr const T &operator[](size_t i) const;
+
     /// @returns true if the matrix is equal, false otherwise.
     constexpr bool operator==(vec const &that) const;
 

diff --git a/src/cyfre/vectors.cpp b/src/cyfre/vectors.cpp
@@ -44,6 +44,24 @@ namespace cyfre {
     }
   }
 
+  template <concepts::scalars T, typename Dim, axis_t Axis, order_t Order, typename Blas>
+  constexpr T &vec<T, Dim, Axis, Order, Blas>::operator[](size_t i) {
+    if constexpr (Axis == axis_t::x) {
+      return mat<T, typename VecOrient<Dim, Axis>::OrientDim, Order, Blas>::operator()(0, i);
+    } else {
+      return mat<T, typename VecOrient<Dim, Axis>::OrientDim, Order, Blas>::operator()(i, 0);
+    }
+  }
+
+  template <concepts::scalars T, typename Dim, axis_t Axis, order_t Order, typename Blas>
+  constexpr const T &vec<T, Dim, Axis, Order, Blas>::operator[](size_t i) const {
+    if constexpr (Axis == axis_t::x) {
+      return mat<T, typename VecOrient<Dim, Axis>::OrientDim, Order, Blas>::operator()(0, i);
+    } else {
+      return mat<T, typename VecOrient<Dim, Axis>::OrientDim, Order, Blas>::operator()(i, 0);
+    }
+  }
+
   /// @returns true if the matrix is equal, false otherwise.
   template <concepts::scalars T, typename Dim, axis_t Axis, order_t Order, typename Blas>
   constexpr bool vec<T, Dim, Axis, Order, Blas>::operator==(vec const &that) const {
@@ -56,7 +74,7 @@ namespace cyfre {
   constexpr bool vec<T, Dim, Axis, Order, Blas>::operator==(Matrix const &that) const {
     return mat<T, typename VecOrient<Dim, Axis>::OrientDim, Order, Blas>::operator==(that);
   }
-
+  
   template <concepts::scalars T, typename Dim, axis_t Axis, order_t Order, typename Blas>
   constexpr size_t vec<T, Dim, Axis, Order, Blas>::size() const {
     return this->rows() * this->cols();