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

#include "unit/common/arithmetic.hpp"
#include "unit/common/references.hpp"
#include "unit/tests.hpp"

TTS_CASE_TPL("Test dynamic map transposition-like operations",
             rotgen::tests::types)<typename T, typename O>(
  tts::type<tts::types<T, O>>)
{
  auto const cases = rotgen::tests::generate_matrix_references();
  for (auto const& [rows, cols, fn] : cases)
  {
    using mat_t = rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value>;
    mat_t base(rows, cols);
    rotgen::tests::prepare(rows, cols, fn, base);

    rotgen::map<mat_t> input(base.data(), rows, cols);
    rotgen::tests::check_shape_functions(input);
  }
};

TTS_CASE_TPL("Test static map transposition-like operations",
             rotgen::tests::types)<typename T, typename O>(
  tts::type<tts::types<T, O>>)
{
  auto const cases = rotgen::tests::generate_static_matrix_references();

  auto process = []<typename D>(D const& desc) {
    rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value> base(D::rows,
                                                                       D::cols);
    rotgen::tests::prepare(base.rows(), base.cols(), desc.init_fn, base);

    rotgen::map<rotgen::matrix<T, D::rows, D::cols, O::value>> input(
      base.data());
    rotgen::tests::check_shape_functions(input);
  };

  std::apply([&](auto const&... d) { (process(d), ...); }, cases);
};

TTS_CASE_TPL("Test dynamic map reduction-like operations",
             rotgen::tests::types)<typename T, typename O>(
  tts::type<tts::types<T, O>>)
{
  auto const cases = rotgen::tests::generate_matrix_references();
  for (auto const& [rows, cols, fn] : cases)
  {
    rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value> base(rows,
                                                                       cols);
    rotgen::tests::prepare(rows, cols, fn, base);

    rotgen::map<rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value>>
      input(base.data(), rows, cols);
    rotgen::tests::check_reduction_functions(input);
  }
};

TTS_CASE_TPL("Test static map reduction-like operations",
             rotgen::tests::types)<typename T, typename O>(
  tts::type<tts::types<T, O>>)
{
  auto const cases = rotgen::tests::generate_static_matrix_references();

  auto process = []<typename D>(D const& desc) {
    rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value> base(D::rows,
                                                                       D::cols);
    rotgen::tests::prepare(base.rows(), base.cols(), desc.init_fn, base);

    rotgen::map<rotgen::matrix<T, D::rows, D::cols, O::value>> input(
      base.data());
    rotgen::tests::check_reduction_functions(input);
  };

  std::apply([&](auto const&... d) { (process(d), ...); }, cases);
};

TTS_CASE_TPL("Test dot product", float, double)<typename T>(tts::type<T>){
  {auto v = rotgen::setConstant<rotgen::matrix<T, 1, rotgen::Dynamic>>(1,
                                                                       16,
                                                                       2);
auto a = rotgen::map<decltype(v)>(v.data(), 1, 8);
auto b = rotgen::map<decltype(v)>(v.data() + 8, 1, 8);

TTS_EQUAL(rotgen::dot(a, b), 32);
}

{
  auto v = rotgen::setConstant<rotgen::matrix<T, rotgen::Dynamic, 1>>(16, 1, 2);
  auto a = rotgen::map<decltype(v)>(v.data(), 8, 1);
  auto b = rotgen::map<decltype(v)>(v.data() + 8, 8, 1);

  TTS_EQUAL(rotgen::dot(a, b), 32);
}

{
  auto v = rotgen::setConstant<rotgen::matrix<T, 1, rotgen::Dynamic>>(1, 16, 2);
  auto a = rotgen::map<rotgen::matrix<T, 1, 8>>(v.data());
  auto b = rotgen::map<rotgen::matrix<T, 1, 8>>(v.data() + 8);

  TTS_EQUAL(rotgen::dot(a, b), 32);
}

{
  auto v = rotgen::setConstant<rotgen::matrix<T, 1, rotgen::Dynamic>>(1, 16, 2);
  auto a = rotgen::map<rotgen::matrix<T, 8, 1>>(v.data());
  auto b = rotgen::map<rotgen::matrix<T, 8, 1>>(v.data() + 8);

  TTS_EQUAL(rotgen::dot(a, b), 32);
}
}
;