mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-20 02:55:47 +00:00
fdaccc22db
Change the VTKM_CONT_EXPORT to VTKM_CONT. (Likewise for EXEC and EXEC_CONT.) Remove the inline from these macros so that they can be applied to everything, including implementations in a library. Because inline is not declared in these modifies, you have to add the keyword to functions and methods where the implementation is not inlined in the class.
220 lines
7.3 KiB
C++
220 lines
7.3 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,
|
|
CellSetIn topology,
|
|
FieldOutCell<Scalar> outCells);
|
|
typedef void ExecutionSignature(_1, _4, _2, PointCount, CellShape, PointIndices);
|
|
typedef _3 InputDomain;
|
|
|
|
VTKM_CONT
|
|
MaxPointOrCellValue() { }
|
|
|
|
template<typename InCellType,
|
|
typename OutCellType,
|
|
typename InPointVecType,
|
|
typename CellShapeTag,
|
|
typename PointIndexType>
|
|
VTKM_EXEC
|
|
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,
|
|
CellSetIn topology,
|
|
FieldOutCell<Scalar> outCells);
|
|
typedef void ExecutionSignature(_1, _3, PointCount);
|
|
typedef _2 InputDomain;
|
|
|
|
VTKM_CONT
|
|
AveragePointToCellValue() { }
|
|
|
|
template<typename PointVecType, typename OutType>
|
|
VTKM_EXEC
|
|
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,
|
|
CellSetIn topology,
|
|
FieldOut<Scalar> outPoints);
|
|
typedef void ExecutionSignature(_1, _3, CellCount);
|
|
typedef _2 InputDomain;
|
|
|
|
VTKM_CONT
|
|
AverageCellToPointValue() { }
|
|
|
|
template<typename CellVecType, typename OutType>
|
|
VTKM_EXEC
|
|
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"),
|
|
dataSet.GetField("pointvar"),
|
|
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"),
|
|
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"),
|
|
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);
|
|
}
|