//==================================================================================================
/*
  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("Function size", rotgen::tests::types)<typename T, typename O>(
  tts::type<tts::types<T, O>>)
{
  T data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12};

  rotgen::map<rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value>>
    dyn_map(data, 1, 12);
  TTS_EQUAL(dyn_map.rows(), rotgen::Index{1});
  TTS_EQUAL(dyn_map.cols(), rotgen::Index{12});

  rotgen::map<rotgen::matrix<T, 1, 12, rotgen::RowMajor>> s112_map(data);
  TTS_EQUAL(s112_map.rows(), rotgen::Index{1});
  TTS_EQUAL(s112_map.cols(), rotgen::Index{12});
  TTS_EQUAL(s112_map.size(), rotgen::Index{12});

  rotgen::map<rotgen::matrix<T, 12, 1, rotgen::ColMajor>> s121_map(data);
  TTS_EQUAL(s121_map.rows(), rotgen::Index{12});
  TTS_EQUAL(s121_map.cols(), rotgen::Index{1});
  TTS_EQUAL(s121_map.size(), rotgen::Index{12});

  rotgen::map<rotgen::matrix<T, 3, 4, O::value>> s34_map(data);
  TTS_EQUAL(s34_map.rows(), rotgen::Index{3});
  TTS_EQUAL(s34_map.cols(), rotgen::Index{4});
  TTS_EQUAL(s34_map.size(), rotgen::Index{12});

  rotgen::map<rotgen::matrix<T, 6, 2, O::value>> s62_map(data);
  TTS_EQUAL(s62_map.rows(), rotgen::Index{6});
  TTS_EQUAL(s62_map.cols(), rotgen::Index{2});
  TTS_EQUAL(s62_map.size(), rotgen::Index{12});
};

TTS_CASE_TPL("Test coefficient accessors",
             rotgen::tests::types)<typename T, typename O>(
  tts::type<tts::types<T, O>>)
{
  using base = rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value>;

  T data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12};

  rotgen::map<base> a(data, 4, 3);
  for (rotgen::Index i = 0; i < 4; i++)
  {
    for (rotgen::Index j = 0; j < 3; j++)
    {
      if constexpr (O::value) TTS_EQUAL(a(i, j), data[j + 3 * i]);
      else TTS_EQUAL(a(i, j), data[i + 4 * j]);
    }
  }

  a(1, 1) = 42;
  TTS_EQUAL(a(1, 1), 42);

  T& ref = a(2, 2);
  ref = 17;
  TTS_EQUAL(a(2, 2), 17);
};

TTS_CASE_TPL("Test one index coefficient accessors",
             rotgen::tests::types)<typename T, typename O>(
  tts::type<tts::types<T, O>>)
{
  T data[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12};

  auto m = [&]() {
    if constexpr (O::value == rotgen::ColMajor)
    {
      using base = rotgen::matrix<T, 1, rotgen::Dynamic>;
      rotgen::map<base> a(data, 12);
      return a;
    }
    else
    {
      using base = rotgen::matrix<T, rotgen::Dynamic, 1>;
      rotgen::map<base> a(data, 12);
      return a;
    }
  }();

  TTS_EXPECT(m.IsVectorAtCompileTime);

  for (rotgen::Index i = 0; i < m.size(); i++)
    TTS_EQUAL(m(i), data[i]) << "Index: " << i << "\n";

  for (rotgen::Index i = 0; i < m.size(); i++)
    TTS_EQUAL(m[i], data[i]) << "Index: " << i << "\n";

  m(1) = 42;
  TTS_EQUAL(data[1], 42);

  T& ref = m(2);
  ref = 17;
  TTS_EQUAL(data[2], 17);

  m[1] = 77;
  TTS_EQUAL(data[1], 77);

  T& bref = m[3];
  bref = 331;
  TTS_EQUAL(data[3], 331);
};