mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-20 02:55:47 +00:00
8520d70e0d
VTK-m now provides the following filters with the default policy as part of the vtkm_filter library: - CellAverage - CleanGrid - ClipWithField - ClipWithImplicitFunction - Contour - ExternalFaces - ExtractStructured - PointAverage - Threshold - VectorMagnitude By building these as a library we hope to provide faster compile times for consumers of VTK-m when using common configurations.
72 lines
2.4 KiB
C++
72 lines
2.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.
|
|
//============================================================================
|
|
|
|
#include <vtkm/filter/CleanGrid.h>
|
|
|
|
#include <vtkm/worklet/CellDeepCopy.h>
|
|
#include <vtkm/worklet/RemoveUnusedPoints.h>
|
|
|
|
#include <vector>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace filter
|
|
{
|
|
|
|
template <typename Policy>
|
|
inline VTKM_CONT vtkm::cont::DataSet CleanGrid::DoExecute(const vtkm::cont::DataSet& inData,
|
|
vtkm::filter::PolicyBase<Policy> policy)
|
|
{
|
|
using CellSetType = vtkm::cont::CellSetExplicit<>;
|
|
|
|
CellSetType outputCellSet;
|
|
// Do a deep copy of the cells to new CellSetExplicit structures
|
|
const vtkm::cont::DynamicCellSet& inCellSet = inData.GetCellSet();
|
|
if (inCellSet.IsType<CellSetType>())
|
|
{
|
|
// Is expected type, do a shallow copy
|
|
outputCellSet = inCellSet.Cast<CellSetType>();
|
|
}
|
|
else
|
|
{ // Clean the grid
|
|
auto deducedCellSet = vtkm::filter::ApplyPolicyCellSet(inCellSet, policy);
|
|
vtkm::cont::ArrayHandle<vtkm::IdComponent> numIndices;
|
|
|
|
this->Invoke(worklet::CellDeepCopy::CountCellPoints{}, deducedCellSet, numIndices);
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::UInt8> shapes;
|
|
vtkm::cont::ArrayHandle<vtkm::Id> offsets;
|
|
vtkm::Id connectivitySize;
|
|
vtkm::cont::ConvertNumComponentsToOffsets(numIndices, offsets, connectivitySize);
|
|
numIndices.ReleaseResourcesExecution();
|
|
|
|
vtkm::cont::ArrayHandle<vtkm::Id> connectivity;
|
|
connectivity.Allocate(connectivitySize);
|
|
|
|
this->Invoke(worklet::CellDeepCopy::PassCellStructure{},
|
|
deducedCellSet,
|
|
shapes,
|
|
vtkm::cont::make_ArrayHandleGroupVecVariable(connectivity, offsets));
|
|
shapes.ReleaseResourcesExecution();
|
|
offsets.ReleaseResourcesExecution();
|
|
connectivity.ReleaseResourcesExecution();
|
|
|
|
outputCellSet.Fill(
|
|
deducedCellSet.GetNumberOfPoints(), shapes, numIndices, connectivity, offsets);
|
|
|
|
//Release the input grid from the execution space
|
|
deducedCellSet.ReleaseResourcesExecution();
|
|
}
|
|
|
|
return this->GenerateOutput(inData, outputCellSet);
|
|
}
|
|
}
|
|
}
|