mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-21 03:25:48 +00:00
296 lines
9.4 KiB
C++
296 lines
9.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 2015 Sandia Corporation.
|
|
// Copyright 2015 UT-Battelle, LLC.
|
|
// Copyright 2015 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/cont/DataSetBuilderExplicit.h>
|
|
#include <vtkm/cont/DeviceAdapterAlgorithm.h>
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
#include <vtkm/cont/testing/MakeTestDataSet.h>
|
|
#include <vtkm/cont/testing/ExplicitTestData.h>
|
|
|
|
#include <vector>
|
|
|
|
namespace DataSetBuilderExplicitNamespace {
|
|
|
|
typedef vtkm::cont::DeviceAdapterAlgorithm<VTKM_DEFAULT_DEVICE_ADAPTER_TAG> DFA;
|
|
typedef VTKM_DEFAULT_DEVICE_ADAPTER_TAG DeviceAdapter;
|
|
|
|
template <typename T>
|
|
vtkm::Bounds ComputeBounds(std::size_t numPoints, const T *coords)
|
|
{
|
|
vtkm::Bounds bounds;
|
|
|
|
for (std::size_t i = 0; i < numPoints; i++)
|
|
{
|
|
bounds.Include(vtkm::Vec<T,3>(coords[i*3+0],coords[i*3+1],coords[i*3+2]));
|
|
}
|
|
|
|
return bounds;
|
|
}
|
|
|
|
void ValidateDataSet(const vtkm::cont::DataSet &ds,
|
|
vtkm::Id numPoints,
|
|
vtkm::Id numCells,
|
|
const vtkm::Bounds &bounds)
|
|
{
|
|
//Verify basics..
|
|
VTKM_TEST_ASSERT(ds.GetNumberOfCellSets() == 1,
|
|
"Wrong number of cell sets.");
|
|
VTKM_TEST_ASSERT(ds.GetNumberOfFields() == 2,
|
|
"Wrong number of fields.");
|
|
VTKM_TEST_ASSERT(ds.GetNumberOfCoordinateSystems() == 1,
|
|
"Wrong number of coordinate systems.");
|
|
VTKM_TEST_ASSERT(ds.GetCoordinateSystem().GetData().GetNumberOfValues() == numPoints,
|
|
"Wrong number of coordinates.");
|
|
VTKM_TEST_ASSERT(ds.GetCellSet().GetNumberOfCells() == numCells,
|
|
"Wrong number of cells.");
|
|
|
|
// test various field-getting methods and associations
|
|
try
|
|
{
|
|
ds.GetField("cellvar", vtkm::cont::Field::ASSOC_CELL_SET);
|
|
}
|
|
catch (...)
|
|
{
|
|
VTKM_TEST_FAIL("Failed to get field 'cellvar' with ASSOC_CELL_SET.");
|
|
}
|
|
|
|
try
|
|
{
|
|
ds.GetField("pointvar", vtkm::cont::Field::ASSOC_POINTS);
|
|
}
|
|
catch (...)
|
|
{
|
|
VTKM_TEST_FAIL("Failed to get field 'pointvar' with ASSOC_POINT_SET.");
|
|
}
|
|
|
|
//Make sure bounds are correct.
|
|
vtkm::Bounds computedBounds =
|
|
ds.GetCoordinateSystem().GetBounds(DeviceAdapter());
|
|
VTKM_TEST_ASSERT(test_equal(bounds, computedBounds),
|
|
"Bounds of coordinates do not match");
|
|
}
|
|
|
|
template <typename T>
|
|
std::vector<T>
|
|
createVec(std::size_t n, const T *data)
|
|
{
|
|
std::vector<T> vec(n);
|
|
for (std::size_t i = 0; i < n; i++)
|
|
{
|
|
vec[i] = data[i];
|
|
}
|
|
return vec;
|
|
}
|
|
|
|
template <typename T>
|
|
vtkm::cont::ArrayHandle<T>
|
|
createAH(std::size_t n, const T *data)
|
|
{
|
|
vtkm::cont::ArrayHandle<T> arr;
|
|
DFA::Copy(vtkm::cont::make_ArrayHandle(data, static_cast<vtkm::Id>(n)), arr);
|
|
return arr;
|
|
}
|
|
|
|
template <typename T>
|
|
vtkm::cont::DataSet
|
|
CreateDataSetArr(bool useSeparatedCoords,
|
|
std::size_t numPoints, const T *coords,
|
|
std::size_t numCells, std::size_t numConn,
|
|
const vtkm::Id *conn,
|
|
const vtkm::IdComponent *indices,
|
|
const vtkm::UInt8 *shape)
|
|
{
|
|
vtkm::cont::DataSet dataSet;
|
|
vtkm::cont::DataSetFieldAdd dsf;
|
|
vtkm::cont::DataSetBuilderExplicit dsb;
|
|
if (useSeparatedCoords)
|
|
{
|
|
std::vector<T> xvals(numPoints), yvals(numPoints), zvals(numPoints);
|
|
std::vector<T> varP(numPoints), varC(numCells);
|
|
for (std::size_t i = 0; i < numPoints; i++)
|
|
{
|
|
xvals[i] = coords[i*3 + 0];
|
|
yvals[i] = coords[i*3 + 1];
|
|
zvals[i] = coords[i*3 + 2];
|
|
varP[i] = static_cast<T>(i*1.1f);
|
|
}
|
|
for (std::size_t i = 0; i < numCells; i++)
|
|
{
|
|
varC[i] = static_cast<T>(i*1.1f);
|
|
}
|
|
vtkm::cont::ArrayHandle<T> X,Y,Z,P,C;
|
|
DFA::Copy(vtkm::cont::make_ArrayHandle(xvals), X);
|
|
DFA::Copy(vtkm::cont::make_ArrayHandle(yvals), Y);
|
|
DFA::Copy(vtkm::cont::make_ArrayHandle(zvals), Z);
|
|
DFA::Copy(vtkm::cont::make_ArrayHandle(varP), P);
|
|
DFA::Copy(vtkm::cont::make_ArrayHandle(varC), C);
|
|
dataSet = dsb.Create(X,Y,Z,
|
|
createAH(numCells, shape),
|
|
createAH(numCells, indices),
|
|
createAH(numConn, conn));
|
|
dsf.AddPointField(dataSet, "pointvar", P);
|
|
dsf.AddCellField(dataSet, "cellvar", C);
|
|
return dataSet;
|
|
}
|
|
else
|
|
{
|
|
std::vector<vtkm::Vec<T,3> > tmp(numPoints);
|
|
std::vector<vtkm::Vec<T,1> > varP(numPoints), varC(numCells);
|
|
for (std::size_t i = 0; i < numPoints; i++)
|
|
{
|
|
tmp[i][0] = coords[i*3 + 0];
|
|
tmp[i][1] = coords[i*3 + 1];
|
|
tmp[i][2] = coords[i*3 + 2];
|
|
varP[i][0] = static_cast<T>(i*1.1f);
|
|
}
|
|
for (std::size_t i = 0; i < numCells; i++)
|
|
{
|
|
varC[i][0] = static_cast<T>(i*1.1f);
|
|
}
|
|
vtkm::cont::ArrayHandle<vtkm::Vec<T,3> > pts;
|
|
DFA::Copy(vtkm::cont::make_ArrayHandle(tmp), pts);
|
|
dataSet = dsb.Create(pts,
|
|
createAH(numCells, shape),
|
|
createAH(numCells, indices),
|
|
createAH(numConn, conn));
|
|
dsf.AddPointField(dataSet, "pointvar", varP);
|
|
dsf.AddCellField(dataSet, "cellvar", varC);
|
|
return dataSet;
|
|
}
|
|
}
|
|
|
|
template <typename T>
|
|
vtkm::cont::DataSet
|
|
CreateDataSetVec(bool useSeparatedCoords,
|
|
std::size_t numPoints, const T *coords,
|
|
std::size_t numCells, std::size_t numConn,
|
|
const vtkm::Id *conn,
|
|
const vtkm::IdComponent *indices,
|
|
const vtkm::UInt8 *shape)
|
|
{
|
|
vtkm::cont::DataSet dataSet;
|
|
vtkm::cont::DataSetFieldAdd dsf;
|
|
vtkm::cont::DataSetBuilderExplicit dsb;
|
|
|
|
if (useSeparatedCoords)
|
|
{
|
|
std::vector<T> X(numPoints), Y(numPoints), Z(numPoints), varP(numPoints), varC(numCells);
|
|
for (std::size_t i = 0; i < numPoints; i++)
|
|
{
|
|
X[i] = coords[i*3 + 0];
|
|
Y[i] = coords[i*3 + 1];
|
|
Z[i] = coords[i*3 + 2];
|
|
varP[i] = static_cast<T>(i*1.1f);
|
|
}
|
|
for (std::size_t i = 0; i < numCells; i++)
|
|
{
|
|
varC[i] = static_cast<T>(i*1.1f);
|
|
}
|
|
dataSet = dsb.Create(X,Y,Z,
|
|
createVec(numCells, shape),
|
|
createVec(numCells, indices),
|
|
createVec(numConn, conn));
|
|
dsf.AddPointField(dataSet, "pointvar", varP);
|
|
dsf.AddCellField(dataSet, "cellvar", varC);
|
|
return dataSet;
|
|
}
|
|
else
|
|
{
|
|
std::vector<vtkm::Vec<T,3> > pts(numPoints);
|
|
std::vector<vtkm::Vec<T,1> > varP(numPoints), varC(numCells);
|
|
for (std::size_t i = 0; i < numPoints; i++)
|
|
{
|
|
pts[i][0] = coords[i*3 + 0];
|
|
pts[i][1] = coords[i*3 + 1];
|
|
pts[i][2] = coords[i*3 + 2];
|
|
varP[i][0] = static_cast<T>(i*1.1f);
|
|
}
|
|
for (std::size_t i = 0; i < numCells; i++)
|
|
{
|
|
varC[i][0] = static_cast<T>(i*1.1f);
|
|
}
|
|
dataSet = dsb.Create(pts,
|
|
createVec(numCells, shape),
|
|
createVec(numCells, indices),
|
|
createVec(numConn, conn));
|
|
dsf.AddPointField(dataSet, "pointvar", varP);
|
|
dsf.AddCellField(dataSet, "cellvar", varC);
|
|
return dataSet;
|
|
}
|
|
}
|
|
|
|
#define TEST_DATA(num) \
|
|
vtkm::cont::testing::ExplicitData##num::numPoints, \
|
|
vtkm::cont::testing::ExplicitData##num::coords, \
|
|
vtkm::cont::testing::ExplicitData##num::numCells, \
|
|
vtkm::cont::testing::ExplicitData##num::numConn, \
|
|
vtkm::cont::testing::ExplicitData##num::conn, \
|
|
vtkm::cont::testing::ExplicitData##num::numIndices, \
|
|
vtkm::cont::testing::ExplicitData##num::shapes
|
|
#define TEST_NUMS(num) \
|
|
vtkm::cont::testing::ExplicitData##num::numPoints, \
|
|
vtkm::cont::testing::ExplicitData##num::numCells
|
|
#define TEST_BOUNDS(num) \
|
|
vtkm::cont::testing::ExplicitData##num::numPoints, \
|
|
vtkm::cont::testing::ExplicitData##num::coords
|
|
|
|
void
|
|
TestDataSetBuilderExplicit()
|
|
{
|
|
vtkm::cont::DataSetBuilderExplicit dsb;
|
|
vtkm::cont::DataSet ds;
|
|
vtkm::Bounds bounds;
|
|
|
|
//Iterate over organization of coordinates.
|
|
for (int i = 0; i < 2; i++)
|
|
{
|
|
//Test ExplicitData0
|
|
bounds = ComputeBounds(TEST_BOUNDS(0));
|
|
ds = CreateDataSetArr(i==0,TEST_DATA(0));
|
|
ValidateDataSet(ds, TEST_NUMS(0), bounds);
|
|
ds = CreateDataSetVec(i==0, TEST_DATA(0));
|
|
ValidateDataSet(ds, TEST_NUMS(0), bounds);
|
|
|
|
//Test ExplicitData1
|
|
bounds = ComputeBounds(TEST_BOUNDS(1));
|
|
ds = CreateDataSetArr(i==0,TEST_DATA(1));
|
|
ValidateDataSet(ds, TEST_NUMS(1), bounds);
|
|
ds = CreateDataSetVec(i==0, TEST_DATA(1));
|
|
ValidateDataSet(ds, TEST_NUMS(1), bounds);
|
|
|
|
//Test ExplicitData2
|
|
bounds = ComputeBounds(TEST_BOUNDS(2));
|
|
ds = CreateDataSetArr(i==0,TEST_DATA(2));
|
|
ValidateDataSet(ds, TEST_NUMS(2), bounds);
|
|
ds = CreateDataSetVec(i==0, TEST_DATA(2));
|
|
ValidateDataSet(ds, TEST_NUMS(2), bounds);
|
|
}
|
|
}
|
|
|
|
} // namespace DataSetBuilderExplicitNamespace
|
|
|
|
int UnitTestDataSetBuilderExplicit(int, char *[])
|
|
{
|
|
using namespace DataSetBuilderExplicitNamespace;
|
|
return vtkm::cont::testing::Testing::Run(TestDataSetBuilderExplicit);
|
|
}
|