mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-10-05 18:08:59 +00:00
6fdc7eb8c0
Following what was done with ArrayRangeCompute, the GetRange and GetBounds methods are embedded into the vtkm_cont library for the most common type lists. Also, and probably more importantly, the device adapter is no longer one of the arguments for either of these methods. It is no longer needed as ArrayRangeCompute no longer needs it.
94 lines
4.2 KiB
C++
94 lines
4.2 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 <iostream>
|
|
|
|
#include <vtkm/cont/DataSet.h>
|
|
#include <vtkm/cont/DeviceAdapter.h>
|
|
#include <vtkm/cont/testing/MakeTestDataSet.h>
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
|
|
#include <vtkm/worklet/DispatcherMapField.h>
|
|
#include <vtkm/worklet/VertexClustering.h>
|
|
|
|
|
|
void TestVertexClustering()
|
|
{
|
|
const vtkm::Id3 divisions(3, 3, 3);
|
|
vtkm::cont::testing::MakeTestDataSet maker;
|
|
vtkm::cont::DataSet dataSet = maker.Make3DExplicitDataSetCowNose();
|
|
|
|
//compute the bounds before calling the algorithm
|
|
vtkm::Bounds bounds = dataSet.GetCoordinateSystem().GetBounds();
|
|
|
|
// run
|
|
vtkm::worklet::VertexClustering clustering;
|
|
vtkm::cont::DataSet outDataSet = clustering.Run(dataSet.GetCellSet(),
|
|
dataSet.GetCoordinateSystem(),
|
|
bounds,
|
|
divisions,
|
|
VTKM_DEFAULT_DEVICE_ADAPTER_TAG());
|
|
|
|
// test
|
|
const vtkm::Id output_pointIds = 9;
|
|
vtkm::Id output_pointId[output_pointIds] = {0,1,3, 1,5,4, 1,2,5};
|
|
const vtkm::Id output_points = 6;
|
|
vtkm::Float64 output_point[output_points][3] = {{0.0174716003,0.0501927994,0.0930275023}, {0.0320714004,0.14704667,0.0952706337}, {0.0268670674,0.246195346,0.119720004}, {0.00215422804,0.0340906903,0.180881709}, {0.0108188,0.152774006,0.167914003}, {0.0202241503,0.225427493,0.140208006}};
|
|
|
|
VTKM_TEST_ASSERT(outDataSet.GetNumberOfCoordinateSystems() == 1,
|
|
"Number of output coordinate systems mismatch");
|
|
typedef vtkm::Vec<vtkm::Float64, 3> PointType;
|
|
vtkm::cont::ArrayHandle<PointType> pointArray;
|
|
outDataSet.GetCoordinateSystem(0).GetData().CopyTo(pointArray);
|
|
VTKM_TEST_ASSERT(pointArray.GetNumberOfValues() == output_points,
|
|
"Number of output points mismatch" );
|
|
for (vtkm::Id i = 0; i < pointArray.GetNumberOfValues(); ++i)
|
|
{
|
|
const PointType &p1 = pointArray.GetPortalConstControl().Get(i);
|
|
PointType p2 = vtkm::make_Vec(output_point[i][0],
|
|
output_point[i][1],
|
|
output_point[i][2]);
|
|
std::cout << "point: " << p1 << " " << p2 << std::endl;
|
|
VTKM_TEST_ASSERT(test_equal(p1, p2), "Point Array mismatch");
|
|
}
|
|
|
|
typedef vtkm::cont::CellSetSingleType<> CellSetType;
|
|
VTKM_TEST_ASSERT(outDataSet.GetNumberOfCellSets() == 1, "Number of output cellsets mismatch");
|
|
CellSetType cellSet;
|
|
outDataSet.GetCellSet(0).CopyTo(cellSet);
|
|
VTKM_TEST_ASSERT(
|
|
cellSet.GetConnectivityArray(vtkm::TopologyElementTagPoint(),vtkm::TopologyElementTagCell()).GetNumberOfValues() == output_pointIds,
|
|
"Number of connectivity array elements mismatch");
|
|
for (vtkm::Id i=0; i<cellSet.GetConnectivityArray(vtkm::TopologyElementTagPoint(),vtkm::TopologyElementTagCell()).GetNumberOfValues(); i++)
|
|
{
|
|
vtkm::Id id1 = cellSet.GetConnectivityArray(vtkm::TopologyElementTagPoint(),vtkm::TopologyElementTagCell()).GetPortalConstControl().Get(i) ;
|
|
vtkm::Id id2 = output_pointId[i] ;
|
|
std::cout << "pointid: " << id1 << " " << id2 << std::endl;
|
|
//VTKM_TEST_ASSERT( id1 == id2, "Connectivity Array mismatch" ) ;
|
|
}
|
|
|
|
} // TestVertexClustering
|
|
|
|
|
|
int UnitTestVertexClustering(int, char *[])
|
|
{
|
|
return vtkm::cont::testing::Testing::Run(TestVertexClustering);
|
|
}
|