#include <iomanip>
#include <iostream>
#include "matrix.hpp"

template<typename Matrix>
void print_matrix(const Matrix& A) {
   for (std::size_t i = 0; i < A.m; ++i) {
      for (std::size_t j = 0; j < A.n; ++j) {
	 std::cout << " " << std::setw(8) << A(i, j);
      }
      std::cout << std::endl;
   }
}

template<typename Matrix, typename Initializer>
void init_matrix(Matrix& A, Initializer&& initializer) {
   for (std::size_t i = 0; i < A.m; ++i) {
      for (std::size_t j = 0; j < A.n; ++j) {
	 A(i, j) = initializer(i, j);
      }
   }
}

int main() {
   Matrix<double> A(7, 8, StorageOrder::ColMajor);
   init_matrix(A, [&A](std::size_t i, std::size_t j) {
      return i * A.n + j;
   });
   std::cout << "A:" << std::endl; print_matrix(A);

   Matrix<float> B(3, 3, StorageOrder::RowMajor);
   init_matrix(B, [&B](std::size_t i, std::size_t j) {
      return j * B.m + i;
   });
   std::cout << std::endl << "B:" << std::endl; print_matrix(B);
}