2016-06-08 18:39:47 +00:00
|
|
|
//=============================================================================
|
|
|
|
//
|
|
|
|
// Copyright (c) Kitware, Inc.
|
|
|
|
// All rights reserved.
|
|
|
|
// See LICENSE.txt for details.
|
|
|
|
//
|
|
|
|
// This software is distributed WITHOUT ANY WARRANTY; without even
|
|
|
|
// the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
|
|
// PURPOSE. See the above copyright notice for more information.
|
|
|
|
//
|
2017-09-20 21:33:44 +00:00
|
|
|
// Copyright 2016 National Technology & Engineering Solutions of Sandia, LLC (NTESS).
|
2016-06-08 18:39:47 +00:00
|
|
|
// Copyright 2016 UT-Battelle, LLC.
|
|
|
|
// Copyright 2016 Los Alamos National Security.
|
|
|
|
//
|
2017-09-20 21:33:44 +00:00
|
|
|
// Under the terms of Contract DE-NA0003525 with NTESS,
|
2016-06-08 18:39:47 +00:00
|
|
|
// the U.S. Government retains certain rights in this software.
|
|
|
|
// Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
|
|
|
|
// Laboratory (LANL), the U.S. Government retains certain rights in
|
|
|
|
// this software.
|
|
|
|
//
|
|
|
|
//=============================================================================
|
|
|
|
|
|
|
|
#include <vtkm/Transform3D.h>
|
|
|
|
|
|
|
|
#include <vtkm/testing/Testing.h>
|
|
|
|
|
2016-09-02 18:44:30 +00:00
|
|
|
#include <ctime>
|
2016-09-08 15:54:41 +00:00
|
|
|
#include <random>
|
2016-09-02 18:44:30 +00:00
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
namespace
|
|
|
|
{
|
2016-06-08 18:39:47 +00:00
|
|
|
|
2016-09-08 15:54:41 +00:00
|
|
|
std::mt19937 g_RandomGenerator;
|
2016-06-08 18:39:47 +00:00
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
template <typename T>
|
2016-06-08 18:39:47 +00:00
|
|
|
struct TransformTests
|
|
|
|
{
|
2016-09-08 15:54:41 +00:00
|
|
|
std::uniform_real_distribution<T> RandomDistribution;
|
2016-06-08 18:39:47 +00:00
|
|
|
TransformTests()
|
2017-05-18 14:29:41 +00:00
|
|
|
: RandomDistribution(0.0f, 1.0f)
|
|
|
|
{
|
|
|
|
}
|
2016-06-08 18:39:47 +00:00
|
|
|
|
2016-09-08 15:54:41 +00:00
|
|
|
T RandomNum() { return this->RandomDistribution(g_RandomGenerator); }
|
2016-06-08 18:39:47 +00:00
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
typedef vtkm::Vec<T, 3> Vec;
|
|
|
|
typedef vtkm::Matrix<T, 4, 4> Transform;
|
2016-06-08 18:39:47 +00:00
|
|
|
|
2016-09-08 15:54:41 +00:00
|
|
|
Vec RandomVector()
|
2016-06-08 18:39:47 +00:00
|
|
|
{
|
|
|
|
Vec vec(this->RandomNum(), this->RandomNum(), this->RandomNum());
|
2017-05-18 14:29:41 +00:00
|
|
|
return T(2) * vec - Vec(1);
|
2016-06-08 18:39:47 +00:00
|
|
|
}
|
|
|
|
|
2016-09-08 15:54:41 +00:00
|
|
|
void CheckTranslate()
|
2016-06-08 18:39:47 +00:00
|
|
|
{
|
|
|
|
std::cout << "--- Checking translate" << std::endl;
|
|
|
|
|
|
|
|
Vec startPoint = this->RandomVector();
|
|
|
|
std::cout << " Starting point: " << startPoint << std::endl;
|
|
|
|
|
|
|
|
Vec translateAmount = this->RandomVector();
|
|
|
|
std::cout << " Translation amount: " << translateAmount << std::endl;
|
|
|
|
|
|
|
|
Transform translate = vtkm::Transform3DTranslate(translateAmount);
|
|
|
|
|
|
|
|
Vec translated1 = vtkm::Transform3DPoint(translate, startPoint);
|
|
|
|
std::cout << " First translation: " << translated1 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(translated1, startPoint + translateAmount), "Bad translation.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
Vec translated2 = vtkm::Transform3DPoint(translate, translated1);
|
|
|
|
std::cout << " Second translation: " << translated2 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(translated2, startPoint + T(2) * translateAmount),
|
2016-06-08 18:39:47 +00:00
|
|
|
"Bad translation.");
|
|
|
|
|
|
|
|
// Vectors should be invarient to translation.
|
|
|
|
translated1 = vtkm::Transform3DVector(translate, startPoint);
|
|
|
|
std::cout << " Translated vector: " << translated1 << std::endl;
|
|
|
|
VTKM_TEST_ASSERT(test_equal(translated1, startPoint), "Bad translation.");
|
|
|
|
}
|
|
|
|
|
2016-09-08 15:54:41 +00:00
|
|
|
void CheckScale()
|
2016-06-08 18:39:47 +00:00
|
|
|
{
|
|
|
|
std::cout << "--- Checking scale" << std::endl;
|
|
|
|
|
|
|
|
Vec startPoint = this->RandomVector();
|
|
|
|
std::cout << " Starting point: " << startPoint << std::endl;
|
|
|
|
|
|
|
|
Vec scaleAmount = this->RandomVector();
|
|
|
|
std::cout << " Scale amount: " << scaleAmount << std::endl;
|
|
|
|
|
|
|
|
Transform scale = vtkm::Transform3DScale(scaleAmount);
|
|
|
|
|
|
|
|
Vec scaled1 = vtkm::Transform3DPoint(scale, startPoint);
|
|
|
|
std::cout << " First scale: " << scaled1 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(scaled1, startPoint * scaleAmount), "Bad scale.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
Vec scaled2 = vtkm::Transform3DPoint(scale, scaled1);
|
|
|
|
std::cout << " Second scale: " << scaled2 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(scaled2, startPoint * scaleAmount * scaleAmount), "Bad scale.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
// Vectors should scale the same as points.
|
|
|
|
scaled1 = vtkm::Transform3DVector(scale, startPoint);
|
|
|
|
std::cout << " Scaled vector: " << scaled1 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(scaled1, startPoint * scaleAmount), "Bad scale.");
|
2016-06-08 18:39:47 +00:00
|
|
|
}
|
|
|
|
|
2016-09-08 15:54:41 +00:00
|
|
|
void CheckRotate()
|
2016-06-08 18:39:47 +00:00
|
|
|
{
|
|
|
|
std::cout << "--- Checking rotate" << std::endl;
|
|
|
|
|
|
|
|
Vec startPoint = this->RandomVector();
|
|
|
|
std::cout << " Starting point: " << startPoint << std::endl;
|
|
|
|
|
2016-06-14 19:52:55 +00:00
|
|
|
const T ninetyDegrees = T(90);
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
std::cout << "--Rotate 90 degrees around X" << std::endl;
|
|
|
|
Transform rotateX = vtkm::Transform3DRotateX(ninetyDegrees);
|
|
|
|
|
|
|
|
Vec rotated1 = vtkm::Transform3DPoint(rotateX, startPoint);
|
|
|
|
std::cout << " First rotate: " << rotated1 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(rotated1, Vec(startPoint[0], -startPoint[2], startPoint[1])),
|
|
|
|
"Bad rotate.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
Vec rotated2 = vtkm::Transform3DPoint(rotateX, rotated1);
|
|
|
|
std::cout << " Second rotate: " << rotated2 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(rotated2, Vec(startPoint[0], -startPoint[1], -startPoint[2])),
|
|
|
|
"Bad rotate.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
// Vectors should rotate the same as points.
|
|
|
|
rotated1 = vtkm::Transform3DVector(rotateX, startPoint);
|
|
|
|
std::cout << " Vector rotate: " << rotated1 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(rotated1, Vec(startPoint[0], -startPoint[2], startPoint[1])),
|
|
|
|
"Bad rotate.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
std::cout << "--Rotate 90 degrees around Y" << std::endl;
|
|
|
|
Transform rotateY = vtkm::Transform3DRotateY(ninetyDegrees);
|
|
|
|
|
|
|
|
rotated1 = vtkm::Transform3DPoint(rotateY, startPoint);
|
|
|
|
std::cout << " First rotate: " << rotated1 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(rotated1, Vec(startPoint[2], startPoint[1], -startPoint[0])),
|
|
|
|
"Bad rotate.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
rotated2 = vtkm::Transform3DPoint(rotateY, rotated1);
|
|
|
|
std::cout << " Second rotate: " << rotated2 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(rotated2, Vec(-startPoint[0], startPoint[1], -startPoint[2])),
|
|
|
|
"Bad rotate.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
// Vectors should rotate the same as points.
|
|
|
|
rotated1 = vtkm::Transform3DVector(rotateY, startPoint);
|
|
|
|
std::cout << " Vector rotate: " << rotated1 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(rotated1, Vec(startPoint[2], startPoint[1], -startPoint[0])),
|
|
|
|
"Bad rotate.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
std::cout << "--Rotate 90 degrees around Z" << std::endl;
|
|
|
|
Transform rotateZ = vtkm::Transform3DRotateZ(ninetyDegrees);
|
|
|
|
|
|
|
|
rotated1 = vtkm::Transform3DPoint(rotateZ, startPoint);
|
|
|
|
std::cout << " First rotate: " << rotated1 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(rotated1, Vec(-startPoint[1], startPoint[0], startPoint[2])),
|
|
|
|
"Bad rotate.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
rotated2 = vtkm::Transform3DPoint(rotateZ, rotated1);
|
|
|
|
std::cout << " Second rotate: " << rotated2 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(rotated2, Vec(-startPoint[0], -startPoint[1], startPoint[2])),
|
|
|
|
"Bad rotate.");
|
2016-06-08 18:39:47 +00:00
|
|
|
|
|
|
|
// Vectors should rotate the same as points.
|
|
|
|
rotated1 = vtkm::Transform3DVector(rotateZ, startPoint);
|
|
|
|
std::cout << " Vector rotate: " << rotated1 << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(rotated1, Vec(-startPoint[1], startPoint[0], startPoint[2])),
|
|
|
|
"Bad rotate.");
|
2016-06-08 18:39:47 +00:00
|
|
|
}
|
2016-09-01 15:46:37 +00:00
|
|
|
|
|
|
|
void CheckPerspective()
|
|
|
|
{
|
|
|
|
std::cout << "--- Checking Perspective" << std::endl;
|
|
|
|
|
|
|
|
Vec startPoint = this->RandomVector();
|
|
|
|
std::cout << " Starting point: " << startPoint << std::endl;
|
|
|
|
|
|
|
|
Transform perspective(0);
|
|
|
|
perspective(0, 0) = 1;
|
|
|
|
perspective(1, 1) = 1;
|
|
|
|
perspective(2, 2) = 1;
|
|
|
|
perspective(3, 2) = 1;
|
|
|
|
|
|
|
|
Vec projected = vtkm::Transform3DPointPerspective(perspective, startPoint);
|
|
|
|
std::cout << " Projected: " << projected << std::endl;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(projected, startPoint / startPoint[2]), "Bad perspective.");
|
2016-09-01 15:46:37 +00:00
|
|
|
}
|
2016-06-08 18:39:47 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
struct TryTransformsFunctor
|
|
|
|
{
|
2017-05-18 14:29:41 +00:00
|
|
|
template <typename T>
|
2016-06-08 18:39:47 +00:00
|
|
|
void operator()(T) const
|
|
|
|
{
|
|
|
|
TransformTests<T> tests;
|
|
|
|
tests.CheckTranslate();
|
|
|
|
tests.CheckScale();
|
|
|
|
tests.CheckRotate();
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
void TestTransforms()
|
|
|
|
{
|
2016-09-02 18:44:30 +00:00
|
|
|
vtkm::UInt32 seed = static_cast<vtkm::UInt32>(std::time(nullptr));
|
2016-06-08 18:39:47 +00:00
|
|
|
std::cout << "Seed: " << seed << std::endl;
|
|
|
|
g_RandomGenerator.seed(seed);
|
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
vtkm::testing::Testing::TryTypes(TryTransformsFunctor(), vtkm::TypeListTagFieldScalar());
|
2016-06-08 18:39:47 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
} // anonymous namespace
|
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
int UnitTestTransform3D(int, char* [])
|
2016-06-08 18:39:47 +00:00
|
|
|
{
|
|
|
|
return vtkm::testing::Testing::Run(TestTransforms);
|
|
|
|
}
|