70 lines
No EOL
2 KiB
C++
70 lines
No EOL
2 KiB
C++
//==================================================================================================
|
|
/*
|
|
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);
|
|
}; |