93 lines
2.7 KiB
C++
93 lines
2.7 KiB
C++
//==================================================================================================
|
|
/*
|
|
ROTGEN - Runtime Overlay for Eigen
|
|
Copyright : CODE RECKONS
|
|
SPDX-License-Identifier: BSL-1.0
|
|
*/
|
|
//==================================================================================================
|
|
#include <rotgen/rotgen.hpp>
|
|
#include <Eigen/Dense>
|
|
#include <vector>
|
|
#include <tuple>
|
|
|
|
namespace rotgen::tests
|
|
{
|
|
template<typename T>
|
|
void check_cwise_functions(const T& input)
|
|
{
|
|
using EigenMatrix = Eigen::Matrix<typename T::value_type, Eigen::Dynamic, Eigen::Dynamic>;
|
|
using mat_t = matrix<typename T::value_type,Dynamic,Dynamic,T::storage_order>;
|
|
|
|
TTS_WHEN("Unary Cwise operations")
|
|
{
|
|
EigenMatrix e_ref(input.rows(), input.cols());
|
|
prepare([&](auto r, auto c) { return input(r,c); }, e_ref);
|
|
|
|
mat_t ref(input.rows(), input.cols());
|
|
|
|
TTS_AND_THEN(".cwiseAbs")
|
|
{
|
|
e_ref = e_ref.cwiseAbs();
|
|
prepare([&](auto r, auto c) { return e_ref(r,c); }, ref);
|
|
TTS_EQUAL(input.cwiseAbs(), ref);
|
|
}
|
|
|
|
TTS_AND_THEN(".cwiseAbs2")
|
|
{
|
|
e_ref = e_ref.cwiseAbs2();
|
|
prepare([&](auto r, auto c) { return e_ref(r,c); }, ref);
|
|
TTS_EQUAL(input.cwiseAbs2(), ref);
|
|
}
|
|
|
|
TTS_AND_THEN(".cwiseInverse")
|
|
{
|
|
e_ref = e_ref.cwiseInverse();
|
|
prepare([&](auto r, auto c) { return e_ref(r,c); }, ref);
|
|
TTS_EQUAL(input.cwiseInverse(), ref);
|
|
}
|
|
|
|
TTS_AND_THEN(".cwiseSqrt")
|
|
{
|
|
e_ref = e_ref.cwiseAbs().cwiseSqrt();
|
|
auto mat = input.cwiseAbs();
|
|
auto proper_input = mat.cwiseSqrt();
|
|
|
|
for(rotgen::Index r=0;r<input.rows();++r)
|
|
for(rotgen::Index c=0;c<input.cols();++c)
|
|
TTS_ULP_EQUAL(proper_input(r,c), e_ref(r,c), 1);
|
|
}
|
|
|
|
TTS_AND_THEN("abs()")
|
|
{
|
|
e_ref = e_ref.cwiseAbs();
|
|
prepare([&](auto r, auto c) { return e_ref(r,c); }, ref);
|
|
TTS_EQUAL(rotgen::abs(input), ref);
|
|
}
|
|
|
|
TTS_AND_THEN("abs2()")
|
|
{
|
|
e_ref = e_ref.cwiseAbs2();
|
|
prepare([&](auto r, auto c) { return e_ref(r,c); }, ref);
|
|
TTS_EQUAL(rotgen::abs2(input), ref);
|
|
}
|
|
|
|
TTS_AND_THEN("rec")
|
|
{
|
|
e_ref = e_ref.cwiseInverse();
|
|
prepare([&](auto r, auto c) { return e_ref(r,c); }, ref);
|
|
TTS_EQUAL(rotgen::rec(input), ref);
|
|
}
|
|
|
|
TTS_AND_THEN("sqrt")
|
|
{
|
|
e_ref = e_ref.cwiseAbs().cwiseSqrt();
|
|
auto mat = input.cwiseAbs();
|
|
auto proper_input = rotgen::sqrt(mat);
|
|
|
|
for(rotgen::Index r=0;r<input.rows();++r)
|
|
for(rotgen::Index c=0;c<input.cols();++c)
|
|
TTS_ULP_EQUAL(proper_input(r,c), e_ref(r,c), 1);
|
|
}
|
|
}
|
|
}
|
|
}
|