2017-06-07 16:21:46 +00:00
|
|
|
//============================================================================
|
|
|
|
// Copyright (c) Kitware, Inc.
|
|
|
|
// All rights reserved.
|
|
|
|
// See LICENSE.txt for details.
|
2019-04-15 23:24:21 +00:00
|
|
|
//
|
2017-06-07 16:21:46 +00:00
|
|
|
// 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/worklet/SurfaceNormals.h>
|
|
|
|
|
|
|
|
#include <vtkm/cont/testing/MakeTestDataSet.h>
|
|
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
|
|
#include <vtkm/worklet/DispatcherMapTopology.h>
|
|
|
|
|
|
|
|
namespace
|
|
|
|
{
|
|
|
|
|
2019-07-31 16:20:38 +00:00
|
|
|
using NormalsArrayHandle = vtkm::cont::ArrayHandle<vtkm::Vec3f>;
|
2017-06-07 16:21:46 +00:00
|
|
|
|
2017-06-09 15:49:31 +00:00
|
|
|
void TestFacetedSurfaceNormals(const vtkm::cont::DataSet& dataset, NormalsArrayHandle& normals)
|
|
|
|
{
|
|
|
|
std::cout << "Testing FacetedSurfaceNormals:\n";
|
2017-06-07 16:21:46 +00:00
|
|
|
|
2017-06-09 15:49:31 +00:00
|
|
|
vtkm::worklet::FacetedSurfaceNormals faceted;
|
2018-08-30 15:53:18 +00:00
|
|
|
faceted.Run(dataset.GetCellSet(), dataset.GetCoordinateSystem().GetData(), normals);
|
2017-06-07 16:21:46 +00:00
|
|
|
|
2019-07-31 16:20:38 +00:00
|
|
|
vtkm::Vec3f expected[8] = { { -0.707f, -0.500f, 0.500f }, { -0.707f, -0.500f, 0.500f },
|
|
|
|
{ 0.707f, 0.500f, -0.500f }, { 0.000f, -0.707f, -0.707f },
|
|
|
|
{ 0.000f, -0.707f, -0.707f }, { 0.000f, 0.707f, 0.707f },
|
|
|
|
{ -0.707f, 0.500f, -0.500f }, { 0.707f, -0.500f, 0.500f } };
|
2017-06-07 16:21:46 +00:00
|
|
|
auto portal = normals.GetPortalConstControl();
|
2017-06-09 15:49:31 +00:00
|
|
|
VTKM_TEST_ASSERT(portal.GetNumberOfValues() == 8, "incorrect faceNormals array length");
|
2017-06-07 16:21:46 +00:00
|
|
|
for (vtkm::Id i = 0; i < 8; ++i)
|
|
|
|
{
|
2017-06-09 15:49:31 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(portal.Get(i), expected[i], 0.001),
|
2017-06-07 16:21:46 +00:00
|
|
|
"result does not match expected value");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-06-09 15:49:31 +00:00
|
|
|
void TestSmoothSurfaceNormals(const vtkm::cont::DataSet& dataset,
|
|
|
|
const NormalsArrayHandle& faceNormals)
|
|
|
|
{
|
|
|
|
std::cout << "Testing SmoothSurfaceNormals:\n";
|
|
|
|
|
|
|
|
NormalsArrayHandle pointNormals;
|
|
|
|
vtkm::worklet::SmoothSurfaceNormals smooth;
|
2018-08-30 15:53:18 +00:00
|
|
|
smooth.Run(dataset.GetCellSet(), faceNormals, pointNormals);
|
2017-06-09 15:49:31 +00:00
|
|
|
|
2019-07-31 16:20:38 +00:00
|
|
|
vtkm::Vec3f expected[8] = { { -0.8165f, -0.4082f, -0.4082f }, { -0.2357f, -0.9714f, 0.0286f },
|
|
|
|
{ 0.0000f, -0.1691f, 0.9856f }, { -0.8660f, 0.0846f, 0.4928f },
|
|
|
|
{ 0.0000f, -0.1691f, -0.9856f }, { 0.0000f, 0.9856f, -0.1691f },
|
|
|
|
{ 0.8165f, 0.4082f, 0.4082f }, { 0.8165f, -0.4082f, -0.4082f } };
|
2017-06-09 15:49:31 +00:00
|
|
|
auto portal = pointNormals.GetPortalConstControl();
|
|
|
|
VTKM_TEST_ASSERT(portal.GetNumberOfValues() == 8, "incorrect pointNormals array length");
|
|
|
|
for (vtkm::Id i = 0; i < 8; ++i)
|
|
|
|
{
|
|
|
|
VTKM_TEST_ASSERT(test_equal(portal.Get(i), expected[i], 0.001),
|
|
|
|
"result does not match expected value");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void TestSurfaceNormals()
|
|
|
|
{
|
2018-08-30 15:53:18 +00:00
|
|
|
|
2017-06-09 15:49:31 +00:00
|
|
|
vtkm::cont::DataSet dataset =
|
|
|
|
vtkm::cont::testing::MakeTestDataSet().Make3DExplicitDataSetPolygonal();
|
|
|
|
NormalsArrayHandle faceNormals;
|
|
|
|
|
|
|
|
TestFacetedSurfaceNormals(dataset, faceNormals);
|
|
|
|
TestSmoothSurfaceNormals(dataset, faceNormals);
|
|
|
|
}
|
|
|
|
|
2017-06-07 16:21:46 +00:00
|
|
|
} // anonymous namespace
|
|
|
|
|
2018-10-17 19:17:29 +00:00
|
|
|
int UnitTestSurfaceNormals(int argc, char* argv[])
|
2017-06-07 16:21:46 +00:00
|
|
|
{
|
2018-10-17 19:17:29 +00:00
|
|
|
return vtkm::cont::testing::Testing::Run(TestSurfaceNormals, argc, argv);
|
2017-06-07 16:21:46 +00:00
|
|
|
}
|