648dd768ee
Co-authored-by: Jules Pénuchot <jules@penuchot.com> Co-authored-by: Joel FALCOU <jfalcou@codereckons.com> See merge request oss/rotgen!41
65 lines
1.5 KiB
C++
65 lines
1.5 KiB
C++
//==================================================================================================
|
|
/*
|
|
ROTGEN - Runtime Overlay for Eigen
|
|
Copyright : CODE RECKONS
|
|
SPDX-License-Identifier: BSL-1.0
|
|
*/
|
|
//==================================================================================================
|
|
#include "unit/tests.hpp"
|
|
#include <rotgen/rotgen.hpp>
|
|
|
|
template<int size = rotgen::Dynamic,
|
|
typename Scalar = float,
|
|
int maxSize = size>
|
|
using column = rotgen::matrix<Scalar, size, 1, rotgen::ColMajor, maxSize, 1>;
|
|
|
|
template<int size = rotgen::Dynamic, typename Scalar = float>
|
|
using column_ref = rotgen::ref<column<size, Scalar>>;
|
|
|
|
template<int size = rotgen::Dynamic, typename Scalar = float>
|
|
using const_column_ref = rotgen::ref<column<size, Scalar> const>;
|
|
|
|
template<int N> void process(column_ref<N> v)
|
|
{
|
|
v(0) = rotgen::sum(v);
|
|
}
|
|
|
|
auto process(column_ref<> v)
|
|
{
|
|
auto sz = v.size();
|
|
switch (sz)
|
|
{
|
|
case 1:
|
|
process<1>(v);
|
|
break;
|
|
case 2:
|
|
process<2>(v);
|
|
break;
|
|
case 3:
|
|
process<3>(v);
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return sz;
|
|
}
|
|
|
|
TTS_CASE("Reference of reference check")
|
|
{
|
|
auto v1 = rotgen::matrix<float, 1, 1>::Ones();
|
|
auto v2 = rotgen::matrix<float, 2, 1>::Random();
|
|
auto v3 = rotgen::matrix<float, 3, 1>::Constant(6.66);
|
|
|
|
auto sum1 = v1(0);
|
|
auto sum2 = v2(0) + v2(1);
|
|
auto sum3 = v3(0) + v3(1) + v3(2);
|
|
|
|
TTS_EQUAL(process(v1), 1);
|
|
TTS_EQUAL(process(v2), 2);
|
|
TTS_EQUAL(process(v3), 3);
|
|
|
|
TTS_EQUAL(v1[0], sum1);
|
|
TTS_EQUAL(v2[0], sum2);
|
|
TTS_EQUAL(v3[0], sum3);
|
|
};
|