From e520b464e4610ca4caa8825f180e69e82c6bc943 Mon Sep 17 00:00:00 2001
From: lisyarus <lisyarus@gmail.com>
Date: Wed, 30 Sep 2020 07:23:05 +0300
Subject: [PATCH] Add util::array iterators over range of indices

---
 libs/util/include/psemek/util/array.hpp | 73 +++++++++++++++++++++++++
 1 file changed, 73 insertions(+)

diff --git a/libs/util/include/psemek/util/array.hpp b/libs/util/include/psemek/util/array.hpp
index 6a799d42..e8bf74b7 100644
--- a/libs/util/include/psemek/util/array.hpp
+++ b/libs/util/include/psemek/util/array.hpp
@@ -96,6 +96,10 @@ namespace psemek::util
 
 		void fill(T const & value);
 
+		auto index_begin() const;
+		auto index_end() const;
+		auto indices() const;
+
 	private:
 		std::unique_ptr<T[]> data_;
 		std::array<std::size_t, N> dims_;
@@ -148,6 +152,75 @@ namespace psemek::util
 			return false;
 		}
 
+		template <std::size_t N>
+		struct array_index_iterator
+		{
+			std::array<std::size_t, N> idx;
+			std::array<std::size_t, N> dims;
+
+			bool end;
+
+			std::array<std::size_t, N> const & operator*() const { return idx; }
+
+			array_index_iterator & operator ++()
+			{
+				end = next(idx, dims);
+				return *this;
+			}
+		};
+
+		struct array_index_iterator_sentinel
+		{};
+
+		template <std::size_t N>
+		bool operator == (array_index_iterator<N> const & it, array_index_iterator_sentinel)
+		{
+			return it.end;
+		}
+
+		template <std::size_t N>
+		bool operator != (array_index_iterator<N> const & it, array_index_iterator_sentinel)
+		{
+			return !it.end;
+		}
+
+		template <std::size_t N>
+		struct array_index_range
+		{
+			std::array<std::size_t, N> dims;
+
+			array_index_iterator<N> begin() const
+			{
+				std::array<std::size_t, N> idx;
+				for (std::size_t i = 0; i < N; ++i)
+					idx[i] = 0;
+				return {idx, dims, false};
+			}
+
+			array_index_iterator_sentinel end() const
+			{
+				return {};
+			}
+		};
+
+	}
+
+	template <typename T, std::size_t N>
+	auto array<T, N>::index_begin() const
+	{
+		return indices().begin();
+	}
+
+	template <typename T, std::size_t N>
+	auto array<T, N>::index_end() const
+	{
+		return indices().end();
+	}
+
+	template <typename T, std::size_t N>
+	auto array<T, N>::indices() const
+	{
+		return detail::array_index_range<N>{dims_};
 	}
 
 	template <typename T, std::size_t N>