//==================================================================================================
/*
  ROTGEN - Runtime Overlay for Eigen
  Copyright : CODE RECKONS
  SPDX-License-Identifier: BSL-1.0
*/
//==================================================================================================
#include "unit/tests.hpp"
#include <rotgen/rotgen.hpp>

TTS_CASE_TPL("rowwise API", rotgen::tests::types)<typename T, typename O>(
  tts::type<tts::types<T, O>>)
{
  using e_t = Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic, O::value>;
  e_t ref = e_t::Random(4, 4);
  auto ref_rw = ref.rowwise();

  rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value> mat(4, 4);
  rotgen::tests::prepare([&](auto r, auto c) { return ref(r, c); }, mat);

  auto rw = rotgen::rowwise(mat);

  for (rotgen::Index i = 0; i < mat.rows(); ++i)
    TTS_EQUAL(rw.sum()(i), ref_rw.sum()(i));
  for (rotgen::Index i = 0; i < mat.rows(); ++i)
    TTS_EQUAL(rw.mean()(i), ref_rw.mean()(i));
  for (rotgen::Index i = 0; i < mat.rows(); ++i)
    TTS_EQUAL(rw.prod()(i), ref_rw.prod()(i));
  for (rotgen::Index i = 0; i < mat.rows(); ++i)
    TTS_EQUAL(rw.maxCoeff()(i), ref_rw.maxCoeff()(i));
  for (rotgen::Index i = 0; i < mat.rows(); ++i)
    TTS_EQUAL(rw.minCoeff()(i), ref_rw.minCoeff()(i));
  for (rotgen::Index i = 0; i < mat.rows(); ++i)
    TTS_EQUAL(rw.squaredNorm()(i), ref_rw.squaredNorm()(i));
  for (rotgen::Index i = 0; i < mat.rows(); ++i)
    TTS_EQUAL(rw.norm()(i), ref_rw.norm()(i));
};

TTS_CASE_TPL("colwise API", rotgen::tests::types)<typename T, typename O>(
  tts::type<tts::types<T, O>>)
{
  using e_t = Eigen::Matrix<T, Eigen::Dynamic, Eigen::Dynamic, O::value>;
  e_t ref = e_t::Random(4, 4);
  auto ref_rw = ref.colwise();

  rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value> mat(4, 4);
  rotgen::tests::prepare([&](auto r, auto c) { return ref(r, c); }, mat);

  auto rw = rotgen::colwise(mat);

  for (rotgen::Index i = 0; i < mat.cols(); ++i)
    TTS_EQUAL(rw.sum()(i), ref_rw.sum()(i));
  for (rotgen::Index i = 0; i < mat.cols(); ++i)
    TTS_EQUAL(rw.mean()(i), ref_rw.mean()(i));
  for (rotgen::Index i = 0; i < mat.cols(); ++i)
    TTS_EQUAL(rw.prod()(i), ref_rw.prod()(i));
  for (rotgen::Index i = 0; i < mat.cols(); ++i)
    TTS_EQUAL(rw.maxCoeff()(i), ref_rw.maxCoeff()(i));
  for (rotgen::Index i = 0; i < mat.cols(); ++i)
    TTS_EQUAL(rw.minCoeff()(i), ref_rw.minCoeff()(i));
  for (rotgen::Index i = 0; i < mat.cols(); ++i)
    TTS_EQUAL(rw.squaredNorm()(i), ref_rw.squaredNorm()(i));
  for (rotgen::Index i = 0; i < mat.cols(); ++i)
    TTS_EQUAL(rw.norm()(i), ref_rw.norm()(i));
};