6e1d3a84f0
how did any of this work? match other CellSet file layouts. ??? compile in CUDA. unit tests. also only serial. make error message accurate Well, this compiles and works now. Did it ever? use CellShapeTagGeneric UnitTest matches previous changes. whoops Fix linking problems. Need the same interface as other ThreadIndices. add filter test okay, let's try duplicating CellSetStructure. okay inching... change to wedge in CellSetListTag Means changing these to support it. switch back to wedge from generic compiles and runs remove ExtrudedType need vtkm_worklet vtkm_worklet needs to be included fix segment count for wedge specialization need to actually save the index for the other constructor. specialize on Explicit clean up warning angled brackets not quotes. formatting
150 lines
5.0 KiB
C++
150 lines
5.0 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.
|
|
//============================================================================
|
|
|
|
#ifndef vtk_m_worklet_TetrahedralizeExplicit_h
|
|
#define vtk_m_worklet_TetrahedralizeExplicit_h
|
|
|
|
#include <vtkm/cont/ArrayHandle.h>
|
|
#include <vtkm/cont/ArrayHandleGroupVec.h>
|
|
#include <vtkm/cont/CellSetExplicit.h>
|
|
#include <vtkm/cont/DataSet.h>
|
|
#include <vtkm/cont/DeviceAdapter.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 tetrahedralize cells for an explicit grid data set
|
|
class TetrahedralizeExplicit
|
|
{
|
|
public:
|
|
TetrahedralizeExplicit() {}
|
|
|
|
//
|
|
// Worklet to count the number of tetrahedra generated per cell
|
|
//
|
|
class TetrahedraPerCell : public vtkm::worklet::WorkletMapField
|
|
{
|
|
public:
|
|
using ControlSignature = void(FieldIn shapes, ExecObject tables, FieldOut tetrahedronCount);
|
|
using ExecutionSignature = _3(_1, _2);
|
|
using InputDomain = _1;
|
|
|
|
VTKM_CONT
|
|
TetrahedraPerCell() {}
|
|
|
|
template <typename DeviceAdapter>
|
|
VTKM_EXEC vtkm::IdComponent operator()(
|
|
vtkm::UInt8 shape,
|
|
const vtkm::worklet::internal::TetrahedralizeTablesExecutionObject<DeviceAdapter>& tables)
|
|
const
|
|
{
|
|
return tables.GetCount(vtkm::CellShapeTagGeneric(shape));
|
|
}
|
|
};
|
|
|
|
//
|
|
// Worklet to turn cells into tetrahedra
|
|
// Vertices remain the same and each cell is processed with needing topology
|
|
//
|
|
class TetrahedralizeCell : 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 CellArrayType>
|
|
VTKM_CONT static ScatterType MakeScatter(const CellArrayType& cellArray)
|
|
{
|
|
return ScatterType(cellArray);
|
|
}
|
|
|
|
// Each cell produces tetrahedra and write result at the offset
|
|
template <typename CellShapeTag,
|
|
typename ConnectivityInVec,
|
|
typename DeviceAdapter,
|
|
typename ConnectivityOutVec>
|
|
VTKM_EXEC void operator()(
|
|
CellShapeTag shape,
|
|
const ConnectivityInVec& connectivityIn,
|
|
const vtkm::worklet::internal::TetrahedralizeTablesExecutionObject<DeviceAdapter>& tables,
|
|
ConnectivityOutVec& connectivityOut,
|
|
vtkm::IdComponent visitIndex) const
|
|
{
|
|
vtkm::Vec<vtkm::IdComponent, 4> tetIndices = tables.GetIndices(shape, visitIndex);
|
|
connectivityOut[0] = connectivityIn[tetIndices[0]];
|
|
connectivityOut[1] = connectivityIn[tetIndices[1]];
|
|
connectivityOut[2] = connectivityIn[tetIndices[2]];
|
|
connectivityOut[3] = connectivityIn[tetIndices[3]];
|
|
}
|
|
};
|
|
|
|
template <typename CellSetType>
|
|
vtkm::cont::CellSetSingleType<> Run(
|
|
const CellSetType& cellSet,
|
|
vtkm::cont::ArrayHandle<vtkm::IdComponent>& vtkmNotUsed(outCellsPerCell))
|
|
{
|
|
vtkm::cont::CellSetSingleType<> outCellSet(cellSet.GetName());
|
|
|
|
return outCellSet;
|
|
}
|
|
};
|
|
template <>
|
|
vtkm::cont::CellSetSingleType<> TetrahedralizeExplicit::Run(
|
|
const vtkm::cont::CellSetExplicit<>& 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::TetrahedralizeTables tables;
|
|
|
|
// Determine the number of output cells each input cell will generate
|
|
vtkm::worklet::DispatcherMapField<TetrahedraPerCell> tetPerCellDispatcher;
|
|
tetPerCellDispatcher.Invoke(inShapes, tables.PrepareForInput(), outCellsPerCell);
|
|
|
|
// Build new cells
|
|
vtkm::worklet::DispatcherMapTopology<TetrahedralizeCell> tetrahedralizeDispatcher(
|
|
TetrahedralizeCell::MakeScatter(outCellsPerCell));
|
|
tetrahedralizeDispatcher.Invoke(
|
|
cellSet, tables.PrepareForInput(), vtkm::cont::make_ArrayHandleGroupVec<4>(outConnectivity));
|
|
|
|
// Add cells to output cellset
|
|
outCellSet.Fill(cellSet.GetNumberOfPoints(), vtkm::CellShapeTagTetra::Id, 4, outConnectivity);
|
|
return outCellSet;
|
|
}
|
|
}
|
|
} // namespace vtkm::worklet
|
|
|
|
#endif // vtk_m_worklet_TetrahedralizeExplicit_h
|