#pragma once

#include <gsl/pointers>

template <typename Element>
class MatrixRowIterator
{
public:
	using ColumnPtr = gsl::owner<Element *>;

	explicit MatrixRowIterator(ColumnPtr column_ptr) noexcept;

	MatrixRowIterator &operator++() noexcept;
	MatrixRowIterator operator++(int) noexcept;

	Element &operator*() const noexcept;

	bool operator==(const MatrixRowIterator &rhs) const noexcept;
	bool operator!=(const MatrixRowIterator &rhs) const noexcept;

private:
	ColumnPtr _column_ptr;
};

template <typename Element>
class MatrixRow
{
public:
	using RowPtr = gsl::owner<Element *>;

	explicit MatrixRow(RowPtr row_ptr, const uint32_t &column_cnt) noexcept;

	[[nodiscard]] MatrixRowIterator<Element> begin() const noexcept;

	[[nodiscard]] MatrixRowIterator<Element> end() const noexcept;

private:
	RowPtr _row_ptr;

	const uint32_t &_column_cnt;
};

template <typename Element>
class MatrixIterator
{
public:
	using RowPtr = gsl::owner<Element **>;

	explicit MatrixIterator(RowPtr row_ptr, const uint32_t &column_cnt) noexcept;

	MatrixIterator &operator++() noexcept;
	MatrixIterator operator++(int) noexcept;

	MatrixRow<Element> operator*() const noexcept;

	bool operator==(const MatrixIterator &rhs) const noexcept;
	bool operator!=(const MatrixIterator &rhs) const noexcept;

private:
	RowPtr _row_ptr;

	const uint32_t &_column_cnt;
};

#include "matrix_iterator.tpp"