#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);
}