bddad9b386
Now that the dispatcher does its own TryExecute, filters do not need to do that. This change requires all worklets called by filters to be able to execute without knowing the device a priori.
153 lines
5.4 KiB
C++
153 lines
5.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 National Technology & Engineering Solutions of Sandia, LLC (NTESS).
|
|
// Copyright 2015 UT-Battelle, LLC.
|
|
// Copyright 2015 Los Alamos National Security.
|
|
//
|
|
// Under the terms of Contract DE-NA0003525 with NTESS,
|
|
// 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.
|
|
//============================================================================
|
|
|
|
#ifndef vtk_m_worklet_TriangulateExplicit_h
|
|
#define vtk_m_worklet_TriangulateExplicit_h
|
|
|
|
#include <vtkm/cont/ArrayHandle.h>
|
|
#include <vtkm/cont/ArrayHandleGroupVec.h>
|
|
#include <vtkm/cont/CellSetExplicit.h>
|
|
#include <vtkm/cont/DataSet.h>
|
|
#include <vtkm/cont/DynamicArrayHandle.h>
|
|
#include <vtkm/cont/Field.h>
|
|
|
|
#include <vtkm/worklet/DispatcherMapField.h>
|
|
#include <vtkm/worklet/DispatcherMapTopology.h>
|
|
#include <vtkm/worklet/ScatterCounting.h>
|
|
#include <vtkm/worklet/WorkletMapField.h>
|
|
#include <vtkm/worklet/WorkletMapTopology.h>
|
|
|
|
#include <vtkm/worklet/internal/TriangulateTables.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace worklet
|
|
{
|
|
|
|
/// \brief Compute the triangulate cells for an explicit grid data set
|
|
class TriangulateExplicit
|
|
{
|
|
public:
|
|
TriangulateExplicit() {}
|
|
|
|
//
|
|
// Worklet to count the number of triangles generated per cell
|
|
//
|
|
class TrianglesPerCell : public vtkm::worklet::WorkletMapField
|
|
{
|
|
public:
|
|
using ControlSignature = void(FieldIn<> shapes,
|
|
FieldIn<> numPoints,
|
|
ExecObject tables,
|
|
FieldOut<> triangleCount);
|
|
using ExecutionSignature = _4(_1, _2, _3);
|
|
using InputDomain = _1;
|
|
|
|
VTKM_CONT
|
|
TrianglesPerCell() {}
|
|
|
|
template <typename DeviceAdapter>
|
|
VTKM_EXEC vtkm::IdComponent operator()(
|
|
vtkm::UInt8 shape,
|
|
vtkm::IdComponent numPoints,
|
|
const vtkm::worklet::internal::TriangulateTablesExecutionObject<DeviceAdapter>& tables) const
|
|
{
|
|
return tables.GetCount(vtkm::CellShapeTagGeneric(shape), numPoints);
|
|
}
|
|
};
|
|
|
|
//
|
|
// Worklet to turn cells into triangles
|
|
// Vertices remain the same and each cell is processed with needing topology
|
|
//
|
|
class TriangulateCell : public vtkm::worklet::WorkletMapPointToCell
|
|
{
|
|
public:
|
|
using ControlSignature = void(CellSetIn cellset,
|
|
ExecObject tables,
|
|
FieldOutCell<> connectivityOut);
|
|
using ExecutionSignature = void(CellShape, PointIndices, _2, _3, VisitIndex);
|
|
using InputDomain = _1;
|
|
|
|
using ScatterType = vtkm::worklet::ScatterCounting;
|
|
|
|
template <typename CountArrayType>
|
|
VTKM_CONT static ScatterType MakeScatter(const CountArrayType& countArray)
|
|
{
|
|
return ScatterType(countArray);
|
|
}
|
|
|
|
// Each cell produces triangles and write result at the offset
|
|
template <typename CellShapeTag,
|
|
typename ConnectivityInVec,
|
|
typename ConnectivityOutVec,
|
|
typename DeviceAdapter>
|
|
VTKM_EXEC void operator()(
|
|
CellShapeTag shape,
|
|
const ConnectivityInVec& connectivityIn,
|
|
const vtkm::worklet::internal::TriangulateTablesExecutionObject<DeviceAdapter>& tables,
|
|
ConnectivityOutVec& connectivityOut,
|
|
vtkm::IdComponent visitIndex) const
|
|
{
|
|
vtkm::Vec<vtkm::IdComponent, 3> triIndices = tables.GetIndices(shape, visitIndex);
|
|
connectivityOut[0] = connectivityIn[triIndices[0]];
|
|
connectivityOut[1] = connectivityIn[triIndices[1]];
|
|
connectivityOut[2] = connectivityIn[triIndices[2]];
|
|
}
|
|
};
|
|
|
|
template <typename CellSetType>
|
|
vtkm::cont::CellSetSingleType<> Run(const CellSetType& cellSet,
|
|
vtkm::cont::ArrayHandle<vtkm::IdComponent>& outCellsPerCell)
|
|
{
|
|
vtkm::cont::CellSetSingleType<> outCellSet(cellSet.GetName());
|
|
|
|
// Input topology
|
|
auto inShapes =
|
|
cellSet.GetShapesArray(vtkm::TopologyElementTagPoint(), vtkm::TopologyElementTagCell());
|
|
auto inNumIndices =
|
|
cellSet.GetNumIndicesArray(vtkm::TopologyElementTagPoint(), vtkm::TopologyElementTagCell());
|
|
|
|
// Output topology
|
|
vtkm::cont::ArrayHandle<vtkm::Id> outConnectivity;
|
|
|
|
vtkm::worklet::internal::TriangulateTables tables;
|
|
|
|
// Determine the number of output cells each input cell will generate
|
|
vtkm::worklet::DispatcherMapField<TrianglesPerCell> triPerCellDispatcher;
|
|
triPerCellDispatcher.Invoke(inShapes, inNumIndices, tables.PrepareForInput(), outCellsPerCell);
|
|
|
|
// Build new cells
|
|
vtkm::worklet::DispatcherMapTopology<TriangulateCell> triangulateDispatcher(
|
|
TriangulateCell::MakeScatter(outCellsPerCell));
|
|
triangulateDispatcher.Invoke(
|
|
cellSet, tables.PrepareForInput(), vtkm::cont::make_ArrayHandleGroupVec<3>(outConnectivity));
|
|
|
|
// Add cells to output cellset
|
|
outCellSet.Fill(
|
|
cellSet.GetNumberOfPoints(), vtkm::CellShapeTagTriangle::Id, 3, outConnectivity);
|
|
return outCellSet;
|
|
}
|
|
};
|
|
}
|
|
} // namespace vtkm::worklet
|
|
|
|
#endif // vtk_m_worklet_TriangulateExplicit_h
|