#include <hpc/aux/repool.hpp>
#include <hpc/aux/slices.hpp>
#include <hpc/matvec/asum.hpp>
#include <hpc/matvec/gematrix.hpp>
#include <hpc/matvec/iterators.hpp>
#include <hpc/matvec/print.hpp>
#include <hpc/matvec/traits.hpp>

using namespace hpc;
using namespace hpc::aux;
using namespace hpc::matvec;

template <
   template<typename> class MatrixA, typename T,
   typename RandomEnginePool,
   Require< Ge<MatrixA<T>> > = true
>
void randomInit(MatrixA<T>& A, RandomEnginePool& repool) {
   using EngineType = typename RandomEnginePool::EngineType;
   RandomEngineGuard<EngineType> guard(repool);
   std::uniform_real_distribution<double> uniform(-100, 100);
   auto& engine = guard.get();

   for (auto [i, j, Aij]: A) {
      Aij = uniform(engine);
      (void) i; (void) j; // suppress gcc warning
   }
}

template <
   template<typename> class MatrixA, typename T,
   typename RandomEnginePool,
   Require< Ge<MatrixA<T>> > = true
>
void mt_randomInit(MatrixA<T>& A, RandomEnginePool& repool) {
   #pragma omp parallel for
   for (std::size_t row = 0; row < A.numRows(); ++row) {
      auto A_ = A.block(row, 0).dim(1, A.numCols());
      randomInit(A_, repool);
   }
}

int main() {
   RandomEnginePool<std::mt19937> repool(4);

   GeMatrix<double> A(1051, 781);
   mt_randomInit(A, repool);
}