01757ea136
Most of VTK-m follows the convention of calling the 0D topology elements "points" (which follows the convention of VTK). However, there were several places where they were referred to as "nodes." Make things consistent by calling them points everywhere. Also merged some redundant ExecutionSignature tags.
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("pointvar").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("pointvar").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("pointvar").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);
|
|
}
|