mirror of
https://gitlab.com/libeigen/eigen.git
synced 2026-04-10 11:34:33 +08:00
release of tvmet (inactive for 2 years and developer unreachable) as the basis for eigen2, because it provides seemingly good expression template mechanisms, we want that, and it would take years to reinvent that wheel. We'll see. So this commit imports the last tvmet release.
325 lines
7.4 KiB
C++
325 lines
7.4 KiB
C++
/*
|
|
* Tiny Vector Matrix Library
|
|
* Dense Vector Matrix Libary of Tiny size using Expression Templates
|
|
*
|
|
* Copyright (C) 2001 - 2003 Olaf Petzold <opetzold@users.sourceforge.net>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*
|
|
* $Id: TestXprVectorFunctions.h,v 1.2 2005/03/25 07:12:07 opetzold Exp $
|
|
*/
|
|
|
|
#ifndef TVMET_TEST_XPR_VECTORFUNC_H
|
|
#define TVMET_TEST_XPR_VECTORFUNC_H
|
|
|
|
#include <limits>
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include <tvmet/Vector.h>
|
|
#include <tvmet/util/General.h>
|
|
|
|
template <class T>
|
|
class TestXprVectorFunctions : public CppUnit::TestFixture
|
|
{
|
|
CPPUNIT_TEST_SUITE( TestXprVectorFunctions );
|
|
CPPUNIT_TEST( scalarFuncs1 );
|
|
CPPUNIT_TEST( scalarFuncs2 );
|
|
CPPUNIT_TEST( globalXprVectorFuncs );
|
|
CPPUNIT_TEST( fn_sum );
|
|
CPPUNIT_TEST( fn_product );
|
|
CPPUNIT_TEST( fn_dot );
|
|
CPPUNIT_TEST( fn_cross );
|
|
CPPUNIT_TEST( fn_norm );
|
|
CPPUNIT_TEST_SUITE_END();
|
|
|
|
private:
|
|
typedef tvmet::Vector<T, 3> vector_type;
|
|
|
|
public:
|
|
TestXprVectorFunctions()
|
|
: vZero(0), vOne(1), scalar(10), scalar2(2) { }
|
|
|
|
public: // cppunit interface
|
|
/** cppunit hook for fixture set up. */
|
|
void setUp();
|
|
|
|
/** cppunit hook for fixture tear down. */
|
|
void tearDown();
|
|
|
|
protected:
|
|
void scalarFuncs1();
|
|
void scalarFuncs2();
|
|
void globalXprVectorFuncs();
|
|
void fn_sum();
|
|
void fn_product();
|
|
void fn_dot();
|
|
void fn_cross();
|
|
void fn_norm();
|
|
|
|
private:
|
|
const vector_type vZero;
|
|
const vector_type vOne;
|
|
vector_type v1;
|
|
vector_type vBig; /**< vector 10x bigger than v1 */
|
|
|
|
private:
|
|
const T scalar;
|
|
const T scalar2;
|
|
};
|
|
|
|
|
|
/*****************************************************************************
|
|
* Implementation Part I (cppunit part)
|
|
****************************************************************************/
|
|
|
|
|
|
template <class T>
|
|
void TestXprVectorFunctions<T>::setUp() {
|
|
v1 = 1,2,3;
|
|
vBig = 10,20,30;
|
|
}
|
|
|
|
|
|
template <class T>
|
|
void TestXprVectorFunctions<T>::tearDown() { }
|
|
|
|
|
|
/*****************************************************************************
|
|
* Implementation Part II
|
|
****************************************************************************/
|
|
|
|
|
|
/*
|
|
* global math operators with scalars
|
|
* function(XprVector, scalar)
|
|
*/
|
|
template <class T>
|
|
void
|
|
TestXprVectorFunctions<T>::scalarFuncs1() {
|
|
vector_type r1(v1), r2(v1), r3(v1), r4(vBig);
|
|
vector_type t1(0), t2(0), t3(0), t4(0);
|
|
|
|
r1 += scalar;
|
|
r2 -= scalar;
|
|
r3 *= scalar;
|
|
r4 /= scalar;
|
|
|
|
// all element wise
|
|
t1 = add(T(1)*v1, scalar);
|
|
t2 = sub(T(1)*v1, scalar);
|
|
t3 = mul(T(1)*v1, scalar);
|
|
t4 = div(T(1)*vBig, scalar);
|
|
|
|
CPPUNIT_ASSERT( all_elements(t1 == r1) );
|
|
CPPUNIT_ASSERT( all_elements(t2 == r2) );
|
|
CPPUNIT_ASSERT( all_elements(t3 == r3) );
|
|
CPPUNIT_ASSERT( all_elements(t4 == r4) );
|
|
}
|
|
|
|
|
|
/*
|
|
* global math operators with scalars, part II
|
|
* function(scalar, XprVector)
|
|
*/
|
|
template <class T>
|
|
void
|
|
TestXprVectorFunctions<T>::scalarFuncs2() {
|
|
vector_type r1(v1), r2(v1);
|
|
vector_type t1(0), t2(0);
|
|
|
|
r1 += scalar;
|
|
r2 *= scalar;
|
|
|
|
// all element wise
|
|
t1 = add(scalar, T(1)*v1);
|
|
t2 = mul(scalar, T(1)*v1);
|
|
CPPUNIT_ASSERT( all_elements(t1 == r1) );
|
|
CPPUNIT_ASSERT( all_elements(t2 == r2) );
|
|
}
|
|
|
|
|
|
/*
|
|
* global math operators with xpr vectors (using functions)
|
|
*/
|
|
template <class T>
|
|
void
|
|
TestXprVectorFunctions<T>::globalXprVectorFuncs() {
|
|
vector_type r1(v1), r2(v1), r3(v1), r4(v1);
|
|
vector_type t1(0), t2(0), t3(0), t4(0);
|
|
vector_type v2(v1);
|
|
|
|
CPPUNIT_ASSERT( all_elements( v1 == v2) );
|
|
|
|
r1 += v1;
|
|
r2 -= v1;
|
|
r3 *= v1;
|
|
|
|
{
|
|
using namespace tvmet::element_wise;
|
|
r4 /= v1;
|
|
}
|
|
|
|
CPPUNIT_ASSERT( all_elements(r2 == T(0)) );
|
|
CPPUNIT_ASSERT( all_elements(r4 == T(1)) );
|
|
|
|
t1 = add(T(1)*v1, T(1)*v2);
|
|
t2 = sub(T(1)*v1, T(1)*v2);
|
|
t3 = mul(T(1)*v1, T(1)*v2);
|
|
t4 = tvmet::element_wise::div(T(1)*v1, T(1)*v2);
|
|
|
|
CPPUNIT_ASSERT( all_elements(t1 == r1) );
|
|
CPPUNIT_ASSERT( all_elements(t2 == r2) );
|
|
CPPUNIT_ASSERT( all_elements(t3 == r3) );
|
|
CPPUNIT_ASSERT( all_elements(t4 == r4) );
|
|
}
|
|
|
|
|
|
/*
|
|
* sum of vector
|
|
*/
|
|
template <class T>
|
|
void
|
|
TestXprVectorFunctions<T>::fn_sum() {
|
|
T t = sum(scalar*v1); // alias vBig
|
|
|
|
CPPUNIT_ASSERT( t == (vBig(0) + vBig(1) + vBig(2)));
|
|
}
|
|
|
|
|
|
/*
|
|
* product of vector
|
|
*/
|
|
template <class T>
|
|
void
|
|
TestXprVectorFunctions<T>::fn_product() {
|
|
T t = product(scalar*v1); // alias vBig
|
|
|
|
CPPUNIT_ASSERT( t == (vBig(0) * vBig(1) * vBig(2)));
|
|
}
|
|
|
|
/*
|
|
* dot product
|
|
*/
|
|
template <class T>
|
|
void
|
|
TestXprVectorFunctions<T>::fn_dot() {
|
|
vector_type v2(v1);
|
|
|
|
T t1 = dot(T(1)*v1, T(1)*v2);
|
|
CPPUNIT_ASSERT( t1 == 14 );
|
|
|
|
T t2 = dot(T(1)*v1, T(1)*vBig);
|
|
CPPUNIT_ASSERT( t2 == 140 );
|
|
|
|
T t3 = dot(T(1)*v1, T(1)*vOne);
|
|
CPPUNIT_ASSERT( t3 == 6 );
|
|
|
|
T t4 = dot(T(1)*v1, T(1)*vZero);
|
|
CPPUNIT_ASSERT( t4 == 0 );
|
|
|
|
T t5 = dot(T(1)*v1, vOne);
|
|
CPPUNIT_ASSERT( t5 == 6 );
|
|
|
|
T t6 = dot(vOne, T(1)*v1);
|
|
CPPUNIT_ASSERT( t6 == 6 );
|
|
}
|
|
|
|
|
|
/*
|
|
* cross product
|
|
*/
|
|
template <class T>
|
|
void
|
|
TestXprVectorFunctions<T>::fn_cross() {
|
|
vector_type v2(v1);
|
|
|
|
vector_type t1 = cross(T(1)*v1, T(1)*v2);
|
|
CPPUNIT_ASSERT( all_elements(t1 == vZero) ); // orthogonal vectors
|
|
|
|
vector_type t2 = cross(T(1)*v1, T(1)*vBig);
|
|
CPPUNIT_ASSERT( all_elements(t2 == vZero) ); // orthogonal vectors
|
|
|
|
const vector_type r(-1,2,-1);
|
|
vector_type t3 = cross(T(1)*v1, T(1)*vOne);
|
|
CPPUNIT_ASSERT( all_elements(t3 == r) );
|
|
|
|
vector_type t4 = cross(T(1)*v1, T(1)*vZero);
|
|
CPPUNIT_ASSERT( all_elements(t4 == vZero) );
|
|
|
|
vector_type t5 = cross(T(1)*v2, v1); // orthogonal
|
|
CPPUNIT_ASSERT( all_elements(t5 == vZero) );
|
|
|
|
vector_type t6 = cross(v1, T(1)*v2); // orthogonal
|
|
CPPUNIT_ASSERT( all_elements(t6 == vZero) );
|
|
}
|
|
|
|
|
|
/*
|
|
* norm
|
|
* Note: norm2 for ints specialized
|
|
*/
|
|
template <class T>
|
|
void
|
|
TestXprVectorFunctions<T>::fn_norm() {
|
|
vector_type v2;
|
|
vector_type r;
|
|
vector_type t5;
|
|
|
|
T t1 = norm1(T(1)*v1);
|
|
T t2 = norm1(-v1);
|
|
T t3 = norm2(T(1)*v1);
|
|
T t4 = norm2(-v1);
|
|
|
|
CPPUNIT_ASSERT( t1 == sum(v1) );
|
|
CPPUNIT_ASSERT( t2 == sum(v1) );
|
|
CPPUNIT_ASSERT( std::abs(t3 - std::sqrt(static_cast<typename tvmet::NumericTraits<T>::float_type>(14)))
|
|
< std::numeric_limits<T>::epsilon() );
|
|
CPPUNIT_ASSERT( std::abs(t4 - std::sqrt(static_cast<typename tvmet::NumericTraits<T>::float_type>(14)))
|
|
< std::numeric_limits<T>::epsilon() );
|
|
|
|
r = v1/norm2(v1); // norm2 is checked before
|
|
t5 = normalize(T(1)*v1);
|
|
|
|
CPPUNIT_ASSERT( all_elements(t5 == r) );
|
|
}
|
|
|
|
|
|
/*****************************************************************************
|
|
* Implementation Part II (specialized for ints)
|
|
****************************************************************************/
|
|
|
|
|
|
/*
|
|
* norm on int specialized due to rounding problems
|
|
*/
|
|
template <>
|
|
void
|
|
TestXprVectorFunctions<int>::fn_norm() {
|
|
vector_type v2;
|
|
|
|
int t1 = norm1(int(1)*v1);
|
|
int t2 = norm1(-v1);
|
|
CPPUNIT_ASSERT( t1 == sum(v1) );
|
|
CPPUNIT_ASSERT( t2 == sum(v1) );
|
|
}
|
|
|
|
|
|
#endif // TVMET_TEST_XPR_VECTORFUNC_H
|
|
|
|
// Local Variables:
|
|
// mode:C++
|
|
// End:
|