//==================================================================================================
/*
  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("Chains of extraction", rotgen::tests::types)
<typename T, typename O>( tts::type< tts::types<T,O>> )
{
  constexpr int N = 8;
  auto a = rotgen::matrix<T,N,N,O::value>::Random();

  auto b = topLeftCorner(a,5,5);
  TTS_EQUAL(b.startRow(), 0);
  TTS_EQUAL(b.startCol(), 0);
  setConstant(b,-7);

  for(rotgen::Index r=0;r<5;r++)
    for(rotgen::Index c=0;c<5;c++)
      TTS_EQUAL(a(r,c), -7);

  auto bb = bottomRightCorner(b,3,3);
  TTS_EQUAL(bb.startRow(), 2);
  TTS_EQUAL(bb.startCol(), 2);
  setConstant(bb,42);

  for(rotgen::Index r=2;r<5;r++)
    for(rotgen::Index c=2;c<5;c++)
      TTS_EQUAL(a(r,c), 42);

  auto bbb = row(bb,1);
  TTS_EQUAL(bbb.startRow(), 1);
  TTS_EQUAL(bbb.startCol(), 0);
  setConstant(bbb,99.5);

  for(rotgen::Index c=3;c<5;c++)
    TTS_EQUAL(a(3,c), 99.5);

  auto bbbb = col(bbb,1);
  TTS_EQUAL(bbbb.startRow(), 0);
  TTS_EQUAL(bbbb.startCol(), 1);
  setConstant(bbbb,0.125);

  TTS_EQUAL(a(3,3), 0.125);
};

auto ref_extract ( rotgen::ref<rotgen::matrix<float> >       m) { return rotgen::extract(m,0,0,3,4); }
auto ref_cextract( rotgen::ref<const rotgen::matrix<float> > m) { return rotgen::extract(m,3,4,4,3); }

TTS_CASE("Extraction of ref/ref const")
{
  auto m = rotgen::setRandom<rotgen::matrix<float,7,7>>();

  auto extracted = ref_extract(m);
  extracted = rotgen::setOnes<rotgen::matrix<float,3,4>>();

  for(rotgen::Index r=0;r<3;r++)
    for(rotgen::Index c=0;c<4;c++)
      TTS_EQUAL(m(r,c), 1.f);

  auto sliced = ref_cextract(m);
  rotgen::extract(m,3,4,4,3) = rotgen::setConstant<rotgen::matrix<float,4,3>>(5);
  for(rotgen::Index r=0;r<4;r++)
    for(rotgen::Index c=0;c<3;c++)
      TTS_EQUAL(sliced(r,c), 5.f);
};