21b3b318ba
On one of my compile platforms, GCC was giving conversion warnings from any boost include that was not wrapped in pragmas to disable conversion warnings. To make things easier and more robust, I created a pair of macros, VTKM_BOOST_PRE_INCLUDE and VTKM_BOOST_POST_INCLUDE, that should be wrapped around any #include of a boost header file.
134 lines
4.6 KiB
C++
134 lines
4.6 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/CellAverage.h>
|
|
#include <vtkm/worklet/DispatcherMapTopology.h>
|
|
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
#include <vtkm/cont/testing/MakeTestDataSet.h>
|
|
|
|
VTKM_BOOST_PRE_INCLUDE
|
|
#include <boost/shared_ptr.hpp>
|
|
VTKM_BOOST_POST_INCLUDE
|
|
|
|
namespace {
|
|
|
|
void TestCellAverageRegular3D()
|
|
{
|
|
std::cout << "Testing CellAverage Worklet on 3D strucutred data" << std::endl;
|
|
|
|
vtkm::cont::testing::MakeTestDataSet testDataSet;
|
|
vtkm::cont::DataSet dataSet = testDataSet.Make3DRegularDataSet0();
|
|
|
|
vtkm::cont::Field result("avgvals",
|
|
1,
|
|
vtkm::cont::Field::ASSOC_CELL_SET,
|
|
std::string("cells"),
|
|
vtkm::Float32());
|
|
|
|
vtkm::worklet::DispatcherMapTopology<vtkm::worklet::CellAverage> dispatcher;
|
|
dispatcher.Invoke(dataSet.GetField("nodevar").GetData(),
|
|
dataSet.GetCellSet(),
|
|
result.GetData());
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::Float32> resultArrayHandle =
|
|
result.GetData().CastToArrayHandle(vtkm::Float32(), VTKM_DEFAULT_STORAGE_TAG());
|
|
|
|
vtkm::Float32 expected[4] = { 60.1875f, 70.2125f, 120.3375f, 130.3625f };
|
|
for (int i = 0; i < 4; ++i)
|
|
{
|
|
VTKM_TEST_ASSERT(test_equal(resultArrayHandle.GetPortalConstControl().Get(i),
|
|
expected[i]), "Wrong result for CellAverage worklet on 3D regular data");
|
|
}
|
|
}
|
|
|
|
void TestCellAverageRegular2D()
|
|
{
|
|
std::cout << "Testing CellAverage Worklet on 2D strucutred data" << std::endl;
|
|
|
|
vtkm::cont::testing::MakeTestDataSet testDataSet;
|
|
vtkm::cont::DataSet dataSet = testDataSet.Make3DExplicitDataSet1();
|
|
|
|
vtkm::cont::Field result("avgvals",
|
|
1,
|
|
vtkm::cont::Field::ASSOC_CELL_SET,
|
|
std::string("cells"),
|
|
vtkm::Float32());
|
|
|
|
vtkm::worklet::DispatcherMapTopology<vtkm::worklet::CellAverage> dispatcher;
|
|
dispatcher.Invoke(dataSet.GetField("nodevar").GetData(),
|
|
dataSet.GetCellSet(),
|
|
result.GetData());
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::Float32> resultArrayHandle =
|
|
result.GetData().CastToArrayHandle(vtkm::Float32(), VTKM_DEFAULT_STORAGE_TAG());
|
|
|
|
vtkm::Float32 expected[2] = { 20.1333f, 35.2f };
|
|
for (int i = 0; i < 2; ++i)
|
|
{
|
|
VTKM_TEST_ASSERT(test_equal(resultArrayHandle.GetPortalConstControl().Get(i),
|
|
expected[i]), "Wrong result for CellAverage worklet on 2D regular data");
|
|
}
|
|
}
|
|
|
|
void TestCellAverageExplicit()
|
|
{
|
|
std::cout << "Testing CellAverage Worklet on Explicit data" << std::endl;
|
|
|
|
vtkm::cont::testing::MakeTestDataSet testDataSet;
|
|
vtkm::cont::DataSet dataSet = testDataSet.Make2DRegularDataSet0();
|
|
|
|
vtkm::cont::Field result("avgvals",
|
|
1,
|
|
vtkm::cont::Field::ASSOC_CELL_SET,
|
|
std::string("cells"),
|
|
vtkm::Float32());
|
|
|
|
vtkm::worklet::DispatcherMapTopology<vtkm::worklet::CellAverage> dispatcher;
|
|
dispatcher.Invoke(dataSet.GetField("nodevar").GetData(),
|
|
dataSet.GetCellSet(),
|
|
result.GetData());
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::Float32> resultArrayHandle =
|
|
result.GetData().CastToArrayHandle(vtkm::Float32(), VTKM_DEFAULT_STORAGE_TAG());
|
|
|
|
vtkm::Float32 expected[2] = { 30.1f, 40.1f };
|
|
for (int i = 0; i < 2; ++i)
|
|
{
|
|
VTKM_TEST_ASSERT(test_equal(resultArrayHandle.GetPortalConstControl().Get(i),
|
|
expected[i]), "Wrong result for CellAverage worklet on 2D regular data");
|
|
}
|
|
}
|
|
|
|
|
|
void TestCellAverage()
|
|
{
|
|
TestCellAverageRegular2D();
|
|
TestCellAverageRegular3D();
|
|
TestCellAverageExplicit();
|
|
}
|
|
|
|
}
|
|
|
|
int UnitTestCellAverage(int, char *[])
|
|
{
|
|
return vtkm::cont::testing::Testing::Run(TestCellAverage);
|
|
}
|