mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-19 18:45:43 +00:00
f9750e83f7
I ran into a few minor issues with the constructors to the Field class. The big change I made was that I removed the Field constructors that take an example type and create an empty field of that type. The problem was that the example type was easily confused with some other type that was supposed to describe an array. This lead to some odd behavior in the compiler and resulted in errors in unexpected places. The use case for this constructor is dubious. There were several tests in the code that would create an empty field, add it to a data set, then get it back out to pass to the worklet. The code is much simpler if you just make an ArrayHandle of the right type and use that in the worklet invoke directly. It is also faster to compile with smaller code because the type is known statically (whereas it is lost the other way). The other change was to declare references to ArrayHandle and DynamicArrayHandle as const. There is nothing in the behavior that invalidates the const, and it accepts arrays constructed in the parameter.
220 lines
7.4 KiB
C++
220 lines
7.4 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.
|
|
//
|
|
// Copyright 2014 Sandia Corporation.
|
|
// Copyright 2014 UT-Battelle, LLC.
|
|
// Copyright 2014 Los Alamos National Security.
|
|
//
|
|
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
|
|
// 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/worklet/DispatcherMapTopology.h>
|
|
#include <vtkm/worklet/WorkletMapTopology.h>
|
|
|
|
#include <vtkm/Math.h>
|
|
|
|
#include <vtkm/cont/DataSet.h>
|
|
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
#include <vtkm/cont/testing/MakeTestDataSet.h>
|
|
|
|
namespace test_explicit {
|
|
|
|
class MaxPointOrCellValue : public vtkm::worklet::WorkletMapPointToCell
|
|
{
|
|
public:
|
|
typedef void ControlSignature(FieldInCell<Scalar> inCells,
|
|
FieldInPoint<Scalar> inPoints,
|
|
TopologyIn topology,
|
|
FieldOutCell<Scalar> outCells);
|
|
typedef void ExecutionSignature(_1, _4, _2, PointCount, CellShape, PointIndices);
|
|
typedef _3 InputDomain;
|
|
|
|
VTKM_CONT_EXPORT
|
|
MaxPointOrCellValue() { }
|
|
|
|
template<typename InCellType,
|
|
typename OutCellType,
|
|
typename InPointVecType,
|
|
typename CellShapeTag,
|
|
typename PointIndexType>
|
|
VTKM_EXEC_EXPORT
|
|
void operator()(const InCellType &cellValue,
|
|
OutCellType &maxValue,
|
|
const InPointVecType &pointValues,
|
|
const vtkm::IdComponent &numPoints,
|
|
const CellShapeTag &vtkmNotUsed(type),
|
|
const PointIndexType &vtkmNotUsed(pointIDs)) const
|
|
{
|
|
//simple functor that returns the max of cellValue and pointValue
|
|
maxValue = static_cast<OutCellType>(cellValue);
|
|
for (vtkm::IdComponent pointIndex = 0; pointIndex < numPoints; ++pointIndex)
|
|
{
|
|
maxValue = vtkm::Max(maxValue,
|
|
static_cast<OutCellType>(pointValues[pointIndex]));
|
|
}
|
|
}
|
|
};
|
|
|
|
class AveragePointToCellValue : public vtkm::worklet::WorkletMapPointToCell
|
|
{
|
|
public:
|
|
typedef void ControlSignature(FieldInPoint<Scalar> inPoints,
|
|
TopologyIn topology,
|
|
FieldOutCell<Scalar> outCells);
|
|
typedef void ExecutionSignature(_1, _3, PointCount);
|
|
typedef _2 InputDomain;
|
|
|
|
VTKM_CONT_EXPORT
|
|
AveragePointToCellValue() { }
|
|
|
|
template<typename PointVecType, typename OutType>
|
|
VTKM_EXEC_EXPORT
|
|
void operator()(const PointVecType &pointValues,
|
|
OutType &avgVal,
|
|
const vtkm::IdComponent &numPoints) const
|
|
{
|
|
//simple functor that returns the average pointValue.
|
|
avgVal = static_cast<OutType>(pointValues[0]);
|
|
for (vtkm::IdComponent pointIndex = 1; pointIndex < numPoints; ++pointIndex)
|
|
{
|
|
avgVal += static_cast<OutType>(pointValues[pointIndex]);
|
|
}
|
|
avgVal = avgVal / static_cast<OutType>(numPoints);
|
|
}
|
|
};
|
|
|
|
class AverageCellToPointValue : public vtkm::worklet::WorkletMapCellToPoint
|
|
{
|
|
public:
|
|
typedef void ControlSignature(FieldInCell<Scalar> inCells,
|
|
TopologyIn topology,
|
|
FieldOut<Scalar> outPoints);
|
|
typedef void ExecutionSignature(_1, _3, CellCount);
|
|
typedef _2 InputDomain;
|
|
|
|
VTKM_CONT_EXPORT
|
|
AverageCellToPointValue() { }
|
|
|
|
template<typename CellVecType, typename OutType>
|
|
VTKM_EXEC_EXPORT
|
|
void operator()(const CellVecType &cellValues,
|
|
OutType &avgVal,
|
|
const vtkm::IdComponent &numCellIDs) const
|
|
{
|
|
//simple functor that returns the average cell Value.
|
|
avgVal = static_cast<OutType>(cellValues[0]);
|
|
for (vtkm::IdComponent cellIndex = 1; cellIndex < numCellIDs; ++cellIndex)
|
|
{
|
|
avgVal += static_cast<OutType>(cellValues[cellIndex]);
|
|
}
|
|
avgVal = avgVal / static_cast<OutType>(numCellIDs);
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
namespace {
|
|
|
|
static void TestMaxPointOrCell();
|
|
static void TestAvgPointToCell();
|
|
static void TestAvgCellToPoint();
|
|
|
|
void TestWorkletMapTopologyExplicit()
|
|
{
|
|
typedef vtkm::cont::DeviceAdapterTraits<
|
|
VTKM_DEFAULT_DEVICE_ADAPTER_TAG> DeviceAdapterTraits;
|
|
std::cout << "Testing Topology Worklet ( Explicit ) on device adapter: "
|
|
<< DeviceAdapterTraits::GetName() << std::endl;
|
|
|
|
TestMaxPointOrCell();
|
|
TestAvgPointToCell();
|
|
TestAvgCellToPoint();
|
|
}
|
|
|
|
|
|
static void
|
|
TestMaxPointOrCell()
|
|
{
|
|
std::cout<<"Testing MaxPointOfCell worklet"<<std::endl;
|
|
vtkm::cont::testing::MakeTestDataSet testDataSet;
|
|
vtkm::cont::DataSet dataSet = testDataSet.Make3DExplicitDataSet0();
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::Float32> result;
|
|
|
|
vtkm::worklet::DispatcherMapTopology< ::test_explicit::MaxPointOrCellValue >
|
|
dispatcher;
|
|
dispatcher.Invoke(dataSet.GetField("cellvar").GetData(),
|
|
dataSet.GetField("pointvar").GetData(),
|
|
dataSet.GetCellSet(0),
|
|
result);
|
|
|
|
//Make sure we got the right answer.
|
|
VTKM_TEST_ASSERT(test_equal(result.GetPortalConstControl().Get(0), 100.1f),
|
|
"Wrong result for PointToCellMax worklet");
|
|
VTKM_TEST_ASSERT(test_equal(result.GetPortalConstControl().Get(1), 100.2f),
|
|
"Wrong result for PointToCellMax worklet");
|
|
}
|
|
|
|
static void
|
|
TestAvgPointToCell()
|
|
{
|
|
std::cout<<"Testing AvgPointToCell worklet"<<std::endl;
|
|
|
|
vtkm::cont::testing::MakeTestDataSet testDataSet;
|
|
vtkm::cont::DataSet dataSet = testDataSet.Make3DExplicitDataSet0();
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::Float32> result;
|
|
|
|
vtkm::worklet::DispatcherMapTopology< ::test_explicit::AveragePointToCellValue > dispatcher;
|
|
dispatcher.Invoke(dataSet.GetField("pointvar").GetData(),
|
|
dataSet.GetCellSet(),
|
|
result);
|
|
|
|
//make sure we got the right answer.
|
|
VTKM_TEST_ASSERT(test_equal(result.GetPortalConstControl().Get(0), 20.1333f),
|
|
"Wrong result for PointToCellAverage worklet");
|
|
VTKM_TEST_ASSERT(test_equal(result.GetPortalConstControl().Get(1), 35.2f),
|
|
"Wrong result for PointToCellAverage worklet");
|
|
}
|
|
|
|
static void
|
|
TestAvgCellToPoint()
|
|
{
|
|
std::cout<<"Testing AvgCellToPoint worklet"<<std::endl;
|
|
|
|
vtkm::cont::testing::MakeTestDataSet testDataSet;
|
|
vtkm::cont::DataSet dataSet = testDataSet.Make3DExplicitDataSet1();
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::Float32> result;
|
|
|
|
|
|
vtkm::worklet::DispatcherMapTopology< ::test_explicit::AverageCellToPointValue > dispatcher;
|
|
dispatcher.Invoke(dataSet.GetField("cellvar").GetData(),
|
|
dataSet.GetCellSet(),
|
|
result);
|
|
|
|
//make sure we got the right answer.
|
|
VTKM_TEST_ASSERT(test_equal(result.GetPortalConstControl().Get(0), 100.1f),
|
|
"Wrong result for CellToPointAverage worklet");
|
|
VTKM_TEST_ASSERT(test_equal(result.GetPortalConstControl().Get(1), 100.15f),
|
|
"Wrong result for CellToPointAverage worklet");
|
|
}
|
|
|
|
} // anonymous namespace
|
|
|
|
int UnitTestWorkletMapTopologyExplicit(int, char *[])
|
|
{
|
|
return vtkm::cont::testing::Testing::Run(TestWorkletMapTopologyExplicit);
|
|
}
|