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

// Helper: fill matrix from raw data
// NB: This function must not be turned into a lambda, otherwise
// `test-ubuntu-gcc-release` will not pass and the CI will fail.
// This is likely due to a compilation bug from GCC 13.3.0.
void fill(auto &m, int r, int c, auto data[]) {
  for (int k = 0; k < r * c; ++k)
    m.data()[k] = data[k];
}

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

  // 1x12 dynamic block at (0,0)
  rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value> dm(1,12);
  fill(dm,1,12,data);
  auto b1 = rotgen::block<decltype(dm), rotgen::Dynamic, rotgen::Dynamic>(dm, 0,0,1,12);
  TTS_EQUAL(b1.rows(), rotgen::Index{1});
  TTS_EQUAL(b1.cols(), rotgen::Index{12});

  // 1x5 dynamic block at (0,2)
  auto b2 = rotgen::block<decltype(dm), rotgen::Dynamic, rotgen::Dynamic>(dm, 0,2,1,5);
  TTS_EQUAL(b2.rows(), rotgen::Index{1});
  TTS_EQUAL(b2.cols(), rotgen::Index{5});

  // 3x2 dynamic block at (1,4) in 4x6
  rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value> dm2(4,6);
  fill(dm2,4,6,data);
  auto b3 = rotgen::block<decltype(dm2), rotgen::Dynamic, rotgen::Dynamic>(dm2, 1,4,3,2);
  TTS_EQUAL(b3.rows(), rotgen::Index{3});
  TTS_EQUAL(b3.cols(), rotgen::Index{2});
  TTS_EQUAL(b3.size(), rotgen::Index{6});

  // 3x4 static block
  rotgen::matrix<T,3,4,O::value> sm;
  fill(sm,3,4,data);
  auto b4 = rotgen::block<decltype(sm), 3, 4>(sm, 0,0);
  TTS_EQUAL(b4.rows(), rotgen::Index{3});
  TTS_EQUAL(b4.cols(), rotgen::Index{4});
  TTS_EQUAL(b4.size(), rotgen::Index{12});

  // 6x2 static block
  rotgen::matrix<T,6,2,O::value> sm2;
  fill(sm2,6,2,data);
  auto b5 = rotgen::block<decltype(sm2), 6, 2>(sm2, 0,0);
  TTS_EQUAL(b5.rows(), rotgen::Index{6});
  TTS_EQUAL(b5.cols(), rotgen::Index{2});
  TTS_EQUAL(b5.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};
  base mat(4,3);
  for(int k=0;k<12;++k) mat.data()[k] = data[k];

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

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

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

TTS_CASE_TPL("Test one index coefficient accessors", rotgen::tests::types)
<typename T, typename O>( tts::type< tts::types<T,O>> )
{
  auto vs = [&]()
  {
    if constexpr(O::value == rotgen::ColMajor)
    {
      using base = rotgen::matrix<T,1,rotgen::Dynamic>;
      base m(12);
      for(int k=0;k<12;++k) m(k) = k+1;
      return std::tuple{m,rotgen::block<base,1,rotgen::Dynamic>(m, 0, 0, 1, 12)};
    }
    else
    {
      using base = rotgen::matrix<T,rotgen::Dynamic,1>;
      base m(12);
      for(int k=0;k<12;++k) m(k) = k+1;
      return std::tuple{m,rotgen::block<base,rotgen::Dynamic,1>(m, 0, 0, 12, 1)};
    }
  }();

  auto mat = get<0>(vs);
  auto b   = get<1>(vs);
  TTS_EXPECT(b.IsVectorAtCompileTime);

  for(rotgen::Index i=0;i<b.size();i++)
    TTS_EQUAL(b(i), mat(i)) << "Index: " << i << "\n";

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

  b(1) = 42;
  TTS_EQUAL(mat(1), 42);

  T& ref = b(2);
  ref = 17;
  TTS_EQUAL(mat(2), 17);

  b[1] = 77;
  TTS_EQUAL(mat(1), 77);

  T& bref = b[3];
  bref = 331;
  TTS_EQUAL(mat(3), 331);
};