vtk-m/vtkm/worklet/testing/UnitTestWorkletPointNeighborhood.cxx

357 lines
12 KiB
C++
Raw Normal View History

//============================================================================
// Copyright (c) Kitware, Inc.
// All rights reserved.
// See LICENSE.txt for details.
2019-04-15 23:24:21 +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/DispatcherPointNeighborhood.h>
#include <vtkm/worklet/WorkletPointNeighborhood.h>
#include <vtkm/worklet/ScatterIdentity.h>
#include <vtkm/worklet/ScatterUniform.h>
#include <vtkm/Math.h>
#include <vtkm/VecAxisAlignedPointCoordinates.h>
#include <vtkm/cont/ArrayHandleUniformPointCoordinates.h>
#include <vtkm/cont/CellSetStructured.h>
#include <vtkm/cont/DataSet.h>
2019-04-03 20:28:31 +00:00
#include <vtkm/cont/DeviceAdapterTag.h>
#include <vtkm/cont/Invoker.h>
#include <vtkm/cont/RuntimeDeviceTracker.h>
#include <vtkm/cont/testing/MakeTestDataSet.h>
#include <vtkm/cont/testing/Testing.h>
namespace test_pointneighborhood
{
struct MaxNeighborValue : public vtkm::worklet::WorkletPointNeighborhood
{
using ControlSignature = void(FieldInNeighborhood neighbors, CellSetIn, FieldOut maxV);
using ExecutionSignature = void(Boundary, _1, _3);
//verify input domain can be something other than first parameter
2018-02-22 13:29:13 +00:00
using InputDomain = _2;
template <typename FieldIn, typename FieldOut>
VTKM_EXEC void operator()(const vtkm::exec::BoundaryState& boundary,
const vtkm::exec::FieldNeighborhood<FieldIn>& inputField,
FieldOut& output) const
{
using ValueType = typename FieldIn::ValueType;
auto* nboundary = inputField.Boundary;
if (!(nboundary->IsRadiusInXBoundary(1) == boundary.IsRadiusInXBoundary(1)))
{
this->RaiseError("Got invalid XPos boundary state");
}
if (!(nboundary->IsRadiusInYBoundary(1) == boundary.IsRadiusInYBoundary(1)))
{
this->RaiseError("Got invalid YPos boundary state");
}
if (!(nboundary->IsRadiusInZBoundary(1) == boundary.IsRadiusInZBoundary(1)))
{
this->RaiseError("Got invalid ZPos boundary state");
}
if (!(nboundary->IsRadiusInBoundary(1) == boundary.IsRadiusInBoundary(1)))
{
this->RaiseError("Got invalid boundary state");
}
if (nboundary->IsRadiusInXBoundary(1) !=
(boundary.IsNeighborInXBoundary(-1) && boundary.IsNeighborInXBoundary(1)))
{
this->RaiseError("Neighbor/Radius boundary mismatch in X dimension.");
}
if (nboundary->IsRadiusInYBoundary(1) !=
(boundary.IsNeighborInYBoundary(-1) && boundary.IsNeighborInYBoundary(1)))
{
this->RaiseError("Neighbor/Radius boundary mismatch in Y dimension.");
}
if (nboundary->IsRadiusInZBoundary(1) !=
(boundary.IsNeighborInZBoundary(-1) && boundary.IsNeighborInZBoundary(1)))
{
this->RaiseError("Neighbor/Radius boundary mismatch in Z dimension.");
}
if (nboundary->IsRadiusInBoundary(1) !=
(boundary.IsNeighborInBoundary({ -1 }) && boundary.IsNeighborInBoundary({ 1 })))
{
this->RaiseError("Neighbor/Radius boundary mismatch.");
}
auto minNeighbors = boundary.MinNeighborIndices(1);
auto maxNeighbors = boundary.MaxNeighborIndices(1);
ValueType maxV = inputField.Get(0, 0, 0); //our value
for (vtkm::IdComponent k = minNeighbors[2]; k <= maxNeighbors[2]; ++k)
{
for (vtkm::IdComponent j = minNeighbors[1]; j <= maxNeighbors[1]; ++j)
{
for (vtkm::IdComponent i = minNeighbors[0]; i <= maxNeighbors[0]; ++i)
{
maxV = vtkm::Max(maxV, inputField.Get(i, j, k));
}
}
}
output = static_cast<FieldOut>(maxV);
}
};
struct ScatterIdentityNeighbor : public vtkm::worklet::WorkletPointNeighborhood
{
using ControlSignature = void(CellSetIn topology, FieldIn pointCoords);
using ExecutionSignature =
void(_2, WorkIndex, InputIndex, OutputIndex, ThreadIndices, VisitIndex);
VTKM_CONT
ScatterIdentityNeighbor() {}
template <typename T>
VTKM_EXEC void operator()(
const vtkm::Vec<T, 3>& vtkmNotUsed(coords),
const vtkm::Id& workIndex,
const vtkm::Id& inputIndex,
const vtkm::Id& outputIndex,
const vtkm::exec::arg::ThreadIndicesPointNeighborhood& vtkmNotUsed(threadIndices),
const vtkm::Id& visitIndex) const
{
if (workIndex != inputIndex)
{
this->RaiseError("Got wrong input value.");
}
if (outputIndex != workIndex)
{
this->RaiseError("Got work and output index don't match.");
}
if (visitIndex != 0)
{
this->RaiseError("Got wrong visit value1.");
}
}
using ScatterType = vtkm::worklet::ScatterIdentity;
};
struct ScatterUniformNeighbor : public vtkm::worklet::WorkletPointNeighborhood
{
using ControlSignature = void(CellSetIn topology, FieldIn pointCoords);
using ExecutionSignature =
void(_2, WorkIndex, InputIndex, OutputIndex, ThreadIndices, VisitIndex);
VTKM_CONT
ScatterUniformNeighbor() {}
template <typename T>
VTKM_EXEC void operator()(
const vtkm::Vec<T, 3>& vtkmNotUsed(coords),
const vtkm::Id& workIndex,
const vtkm::Id& inputIndex,
const vtkm::Id& outputIndex,
const vtkm::exec::arg::ThreadIndicesPointNeighborhood& vtkmNotUsed(threadIndices),
const vtkm::Id& visitIndex) const
{
if ((workIndex / 3) != inputIndex)
{
this->RaiseError("Got wrong input value.");
}
if (outputIndex != workIndex)
{
this->RaiseError("Got work and output index don't match.");
}
if ((workIndex % 3) != visitIndex)
{
this->RaiseError("Got wrong visit value2.");
}
}
using ScatterType = vtkm::worklet::ScatterUniform<3>;
};
// An example of using WorkletPointNeighborhood to iterate over a structured 3D cell
// domain rather than look at an actual neighborhood. It reduces a domain by subsampling
// every other item in the input field.
struct Subsample : public vtkm::worklet::WorkletPointNeighborhood
{
using ControlSignature =
void(WholeCellSetIn<vtkm::TopologyElementTagPoint, vtkm::TopologyElementTagCell> inputTopology,
WholeArrayIn inputField,
CellSetIn outputTopology,
FieldOut sampledField);
using ExecutionSignature = void(_1, _2, Boundary, _4);
using InputDomain = _3;
template <typename InFieldPortal, typename T>
VTKM_EXEC void operator()(const vtkm::exec::ConnectivityStructured<vtkm::TopologyElementTagPoint,
vtkm::TopologyElementTagCell,
3>& inputTopology,
const InFieldPortal& inFieldPortal,
const vtkm::exec::BoundaryState& boundary,
T& sample) const
{
sample =
inFieldPortal.Get(inputTopology.LogicalToFlatVisitIndex(2 * boundary.GetCenterIndex()));
}
};
} // namespace test_pointneighborhood
namespace
{
static void TestMaxNeighborValue();
static void TestScatterIdentityNeighbor();
static void TestScatterUnfiormNeighbor();
static void TestIndexing();
void TestWorkletPointNeighborhood(vtkm::cont::DeviceAdapterId id)
{
std::cout << "Testing Point Neighborhood Worklet on device adapter: " << id.GetName()
<< std::endl;
vtkm::cont::ScopedRuntimeDeviceTracker deviceScope(id);
TestMaxNeighborValue();
TestScatterIdentityNeighbor();
TestScatterUnfiormNeighbor();
TestIndexing();
}
static void TestMaxNeighborValue()
{
std::cout << "Testing MaxNeighborValue worklet" << std::endl;
vtkm::cont::testing::MakeTestDataSet testDataSet;
vtkm::worklet::DispatcherPointNeighborhood<::test_pointneighborhood::MaxNeighborValue> dispatcher;
vtkm::cont::ArrayHandle<vtkm::Float32> output;
vtkm::cont::DataSet dataSet3D = testDataSet.Make3DUniformDataSet0();
dispatcher.Invoke(dataSet3D.GetField("pointvar")
.GetData()
.ResetTypes<vtkm::TypeListFieldScalar, VTKM_DEFAULT_STORAGE_LIST>(),
dataSet3D.GetCellSet(),
output);
vtkm::Float32 expected3D[18] = { 110.3f, 120.3f, 120.3f, 110.3f, 120.3f, 120.3f,
170.5f, 180.5f, 180.5f, 170.5f, 180.5f, 180.5f,
170.5f, 180.5f, 180.5f, 170.5f, 180.5f, 180.5f };
for (int i = 0; i < 18; ++i)
{
VTKM_TEST_ASSERT(test_equal(output.ReadPortal().Get(i), expected3D[i]),
"Wrong result for MaxNeighborValue worklet");
}
vtkm::cont::DataSet dataSet2D = testDataSet.Make2DUniformDataSet1();
dispatcher.Invoke(dataSet2D.GetField("pointvar")
.GetData()
.ResetTypes<vtkm::TypeListFieldScalar, VTKM_DEFAULT_STORAGE_LIST>(),
dataSet2D.GetCellSet(),
output);
vtkm::Float32 expected2D[25] = { 100.0f, 100.0f, 78.0f, 49.0f, 33.0f, 100.0f, 100.0f,
78.0f, 50.0f, 48.0f, 94.0f, 94.0f, 91.0f, 91.0f,
91.0f, 52.0f, 52.0f, 91.0f, 91.0f, 91.0f, 12.0f,
51.0f, 91.0f, 91.0f, 91.0f };
for (int i = 0; i < 25; ++i)
{
VTKM_TEST_ASSERT(test_equal(output.ReadPortal().Get(i), expected2D[i]),
"Wrong result for MaxNeighborValue worklet");
}
}
static void TestScatterIdentityNeighbor()
{
std::cout << "Testing identity scatter with PointNeighborhood" << std::endl;
vtkm::cont::testing::MakeTestDataSet testDataSet;
vtkm::worklet::DispatcherPointNeighborhood<::test_pointneighborhood::ScatterIdentityNeighbor>
dispatcher;
vtkm::cont::DataSet dataSet3D = testDataSet.Make3DUniformDataSet0();
dispatcher.Invoke(dataSet3D.GetCellSet(), dataSet3D.GetCoordinateSystem());
vtkm::cont::DataSet dataSet2D = testDataSet.Make2DUniformDataSet0();
dispatcher.Invoke(dataSet2D.GetCellSet(), dataSet2D.GetCoordinateSystem());
}
static void TestScatterUnfiormNeighbor()
{
std::cout << "Testing uniform scatter with PointNeighborhood" << std::endl;
vtkm::cont::testing::MakeTestDataSet testDataSet;
vtkm::worklet::DispatcherPointNeighborhood<::test_pointneighborhood::ScatterUniformNeighbor>
dispatcher;
vtkm::cont::DataSet dataSet3D = testDataSet.Make3DUniformDataSet0();
dispatcher.Invoke(dataSet3D.GetCellSet(), dataSet3D.GetCoordinateSystem());
vtkm::cont::DataSet dataSet2D = testDataSet.Make2DUniformDataSet0();
dispatcher.Invoke(dataSet2D.GetCellSet(), dataSet2D.GetCoordinateSystem());
}
static void TestIndexing()
{
std::cout << "Testing using PointNeighborhood for 3D indexing." << std::endl;
constexpr vtkm::Id outDim = 4;
constexpr vtkm::Id inDim = outDim * 2;
vtkm::cont::CellSetStructured<3> inCellSet;
inCellSet.SetPointDimensions({ inDim });
vtkm::cont::CellSetStructured<3> outCellSet;
outCellSet.SetPointDimensions({ outDim });
vtkm::cont::ArrayHandleUniformPointCoordinates inField(vtkm::Id3{ inDim });
vtkm::cont::ArrayHandle<vtkm::Vec3f> outField;
vtkm::cont::Invoker invoke;
invoke(::test_pointneighborhood::Subsample{}, inCellSet, inField, outCellSet, outField);
VTKM_TEST_ASSERT(outField.GetNumberOfValues() == (outDim * outDim * outDim));
vtkm::Id flatIndex = 0;
vtkm::Id3 IJK;
auto outFieldPortal = outField.WritePortal();
for (IJK[2] = 0; IJK[2] < outDim; ++IJK[2])
{
for (IJK[1] = 0; IJK[1] < outDim; ++IJK[1])
{
for (IJK[0] = 0; IJK[0] < outDim; ++IJK[0])
{
vtkm::Vec3f computed = outFieldPortal.Get(flatIndex);
VTKM_TEST_ASSERT(test_equal(computed, 2 * IJK));
++flatIndex;
}
}
}
}
} // anonymous namespace
int UnitTestWorkletPointNeighborhood(int argc, char* argv[])
{
return vtkm::cont::testing::Testing::RunOnDevice(TestWorkletPointNeighborhood, argc, argv);
}