mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-20 11:05:44 +00:00
132 lines
4.9 KiB
C++
132 lines
4.9 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 <algorithm>
|
|
#include <vtkm/cont/DeviceAdapter.h>
|
|
#include <vtkm/cont/DataSetBuilderExplicit.h>
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
#include <vtkm/worklet/ExternalFaces.h>
|
|
|
|
namespace {
|
|
|
|
vtkm::cont::DataSet RunExternalFaces(vtkm::cont::DataSet &ds)
|
|
{
|
|
|
|
vtkm::cont::CellSetExplicit<> cellset;
|
|
ds.GetCellSet(0).CopyTo(cellset);
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::UInt8> shapes = cellset.GetShapesArray(
|
|
vtkm::TopologyElementTagPoint(),vtkm::TopologyElementTagCell());
|
|
vtkm::cont::ArrayHandle<vtkm::IdComponent> numIndices = cellset.GetNumIndicesArray(
|
|
vtkm::TopologyElementTagPoint(),vtkm::TopologyElementTagCell());
|
|
vtkm::cont::ArrayHandle<vtkm::Id> conn = cellset.GetConnectivityArray(
|
|
vtkm::TopologyElementTagPoint(),vtkm::TopologyElementTagCell());
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::UInt8> output_shapes;
|
|
vtkm::cont::ArrayHandle<vtkm::IdComponent> output_numIndices;
|
|
vtkm::cont::ArrayHandle<vtkm::Id> output_conn;
|
|
|
|
//Run the External Faces worklet
|
|
vtkm::worklet::ExternalFaces().run(
|
|
shapes,
|
|
numIndices,
|
|
conn,
|
|
output_shapes,
|
|
output_numIndices,
|
|
output_conn,
|
|
VTKM_DEFAULT_DEVICE_ADAPTER_TAG());
|
|
|
|
vtkm::cont::DataSet new_ds;
|
|
for(vtkm::IdComponent i=0; i < ds.GetNumberOfCoordinateSystems(); ++i)
|
|
{
|
|
new_ds.AddCoordinateSystem(ds.GetCoordinateSystem(i));
|
|
}
|
|
|
|
|
|
vtkm::cont::CellSetExplicit<> new_cs(cellset.GetNumberOfPoints(),"cells");
|
|
new_cs.Fill(output_shapes, output_numIndices, output_conn);
|
|
new_ds.AddCellSet(new_cs);
|
|
|
|
return new_ds;
|
|
}
|
|
|
|
void TestExternalFaces()
|
|
{
|
|
//--------------Construct a VTK-m Test Dataset----------------
|
|
const int nVerts = 8; //A cube that is tetrahedralized
|
|
typedef vtkm::Vec<vtkm::Float32,3> CoordType;
|
|
vtkm::cont::ArrayHandle< CoordType > coordinates;
|
|
coordinates.Allocate(nVerts);
|
|
coordinates.GetPortalControl().Set(0, CoordType(0.0f, 0.0f, 0.0f) );
|
|
coordinates.GetPortalControl().Set(1, CoordType(1.0f, 0.0f, 0.0f) );
|
|
coordinates.GetPortalControl().Set(2, CoordType(1.0f, 1.0f, 0.0f) );
|
|
coordinates.GetPortalControl().Set(3, CoordType(0.0f, 1.0f, 0.0f) );
|
|
coordinates.GetPortalControl().Set(4, CoordType(0.0f, 0.0f, 1.0f) );
|
|
coordinates.GetPortalControl().Set(5, CoordType(1.0f, 0.0f, 1.0f) );
|
|
coordinates.GetPortalControl().Set(6, CoordType(1.0f, 1.0f, 1.0f) );
|
|
coordinates.GetPortalControl().Set(7, CoordType(0.0f, 1.0f, 1.0f) );
|
|
|
|
//Construct the VTK-m shapes and numIndices connectivity arrays
|
|
const int nCells = 6; //The tetrahedrons of the cube
|
|
vtkm::IdComponent cellVerts[nCells][4] = {
|
|
{4,7,6,3}, {4,6,3,2}, {4,0,3,2},
|
|
{4,6,5,2}, {4,5,0,2}, {1,0,5,2}
|
|
};
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::UInt8> shapes;
|
|
vtkm::cont::ArrayHandle<vtkm::IdComponent> numIndices;
|
|
vtkm::cont::ArrayHandle<vtkm::Id> conn;
|
|
shapes.Allocate(static_cast<vtkm::Id>(nCells));
|
|
numIndices.Allocate(static_cast<vtkm::Id>(nCells));
|
|
conn.Allocate(static_cast<vtkm::Id>(4 * nCells));
|
|
|
|
int index = 0;
|
|
for(int j = 0; j < nCells; j++)
|
|
{
|
|
shapes.GetPortalControl().Set(j, static_cast<vtkm::UInt8>(vtkm::CELL_SHAPE_TETRA));
|
|
numIndices.GetPortalControl().Set(j, 4);
|
|
for(int k = 0; k < 4; k++)
|
|
conn.GetPortalControl().Set(index++, cellVerts[j][k]);
|
|
}
|
|
|
|
vtkm::cont::DataSetBuilderExplicit builder;
|
|
vtkm::cont::DataSet ds = builder.Create(coordinates, shapes, numIndices, conn);
|
|
|
|
//Run the External Faces worklet
|
|
vtkm::cont::DataSet new_ds = RunExternalFaces(ds);
|
|
vtkm::cont::CellSetExplicit<> new_cs;
|
|
new_ds.GetCellSet(0).CopyTo(new_cs);
|
|
|
|
vtkm::Id numExtFaces_out = new_cs.GetNumberOfCells();
|
|
|
|
//Validate the number of external faces (output) returned by the worklet
|
|
const vtkm::Id numExtFaces_actual = 12;
|
|
VTKM_TEST_ASSERT(numExtFaces_out == numExtFaces_actual, "Number of External Faces mismatch");
|
|
|
|
} // TestExternalFaces
|
|
|
|
}
|
|
|
|
int UnitTestExternalFaces(int, char *[])
|
|
{
|
|
return vtkm::cont::testing::Testing::Run(TestExternalFaces);
|
|
}
|