111 lines
3.5 KiB
C++
111 lines
3.5 KiB
C++
//============================================================================
|
|
// 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.
|
|
//============================================================================
|
|
|
|
#include <vtkm/VecFromPortalPermute.h>
|
|
|
|
#include <vtkm/VecVariable.h>
|
|
|
|
#include <vtkm/testing/Testing.h>
|
|
|
|
namespace UnitTestVecFromPortalPermuteNamespace
|
|
{
|
|
|
|
static constexpr vtkm::IdComponent ARRAY_SIZE = 10;
|
|
|
|
template <typename T>
|
|
void CheckType(T, T)
|
|
{
|
|
// Check passes if this function is called correctly.
|
|
}
|
|
|
|
template <typename T>
|
|
class TestPortal
|
|
{
|
|
public:
|
|
using ValueType = T;
|
|
|
|
VTKM_EXEC
|
|
vtkm::Id GetNumberOfValues() const { return ARRAY_SIZE; }
|
|
|
|
VTKM_EXEC
|
|
ValueType Get(vtkm::Id index) const { return TestValue(index, ValueType()); }
|
|
};
|
|
|
|
struct VecFromPortalPermuteTestFunctor
|
|
{
|
|
template <typename T>
|
|
void operator()(T) const
|
|
{
|
|
using PortalType = TestPortal<T>;
|
|
using IndexVecType = vtkm::VecVariable<vtkm::Id, ARRAY_SIZE>;
|
|
using VecType = vtkm::VecFromPortalPermute<IndexVecType, PortalType>;
|
|
using TTraits = vtkm::TypeTraits<VecType>;
|
|
using VTraits = vtkm::VecTraits<VecType>;
|
|
|
|
std::cout << "Checking VecFromPortal traits" << std::endl;
|
|
|
|
// The statements will fail to compile if the traits is not working as
|
|
// expected.
|
|
CheckType(typename TTraits::DimensionalityTag(), vtkm::TypeTraitsVectorTag());
|
|
CheckType(typename VTraits::ComponentType(), T());
|
|
CheckType(typename VTraits::HasMultipleComponents(), vtkm::VecTraitsTagMultipleComponents());
|
|
CheckType(typename VTraits::IsSizeStatic(), vtkm::VecTraitsTagSizeVariable());
|
|
|
|
std::cout << "Checking VecFromPortal contents" << std::endl;
|
|
|
|
PortalType portal;
|
|
|
|
for (vtkm::Id offset = 0; offset < ARRAY_SIZE; offset++)
|
|
{
|
|
for (vtkm::IdComponent length = 0; 2 * length + offset < ARRAY_SIZE; length++)
|
|
{
|
|
IndexVecType indices;
|
|
for (vtkm::IdComponent index = 0; index < length; index++)
|
|
{
|
|
indices.Append(offset + 2 * index);
|
|
}
|
|
|
|
VecType vec(&indices, portal);
|
|
|
|
VTKM_TEST_ASSERT(vec.GetNumberOfComponents() == length, "Wrong length.");
|
|
VTKM_TEST_ASSERT(VTraits::GetNumberOfComponents(vec) == length, "Wrong length.");
|
|
|
|
vtkm::Vec<T, ARRAY_SIZE> copyDirect;
|
|
vec.CopyInto(copyDirect);
|
|
|
|
vtkm::Vec<T, ARRAY_SIZE> copyTraits;
|
|
VTraits::CopyInto(vec, copyTraits);
|
|
|
|
for (vtkm::IdComponent index = 0; index < length; index++)
|
|
{
|
|
T expected = TestValue(2 * index + offset, T());
|
|
VTKM_TEST_ASSERT(test_equal(vec[index], expected), "Wrong value.");
|
|
VTKM_TEST_ASSERT(test_equal(VTraits::GetComponent(vec, index), expected), "Wrong value.");
|
|
VTKM_TEST_ASSERT(test_equal(copyDirect[index], expected), "Wrong copied value.");
|
|
VTKM_TEST_ASSERT(test_equal(copyTraits[index], expected), "Wrong copied value.");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
void VecFromPortalPermuteTest()
|
|
{
|
|
vtkm::testing::Testing::TryTypes(VecFromPortalPermuteTestFunctor(), vtkm::TypeListTagCommon());
|
|
}
|
|
|
|
} // namespace UnitTestVecFromPortalPermuteNamespace
|
|
|
|
int UnitTestVecFromPortalPermute(int argc, char* argv[])
|
|
{
|
|
return vtkm::testing::Testing::Run(
|
|
UnitTestVecFromPortalPermuteNamespace::VecFromPortalPermuteTest, argc, argv);
|
|
}
|