5498ecd35b
Generally, fields that have a WHOLE_MESH association might be valid even if the structure of the mesh changes. Thus, it makes sense for filters to pass this data pretty much all the time. Also cleaned up some code and comments to make the relationship between `MapFieldOntoOutput` and `DoMapField` more clear.
119 lines
3.3 KiB
C++
119 lines
3.3 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_filter_Triangulate_hxx
|
|
#define vtk_m_filter_Triangulate_hxx
|
|
|
|
#include <vtkm/filter/Triangulate.h>
|
|
|
|
#include <vtkm/filter/MapFieldPermutation.h>
|
|
|
|
namespace
|
|
{
|
|
|
|
class DeduceCellSet
|
|
{
|
|
vtkm::worklet::Triangulate& Worklet;
|
|
vtkm::cont::CellSetSingleType<>& OutCellSet;
|
|
|
|
public:
|
|
DeduceCellSet(vtkm::worklet::Triangulate& worklet, vtkm::cont::CellSetSingleType<>& outCellSet)
|
|
: Worklet(worklet)
|
|
, OutCellSet(outCellSet)
|
|
{
|
|
}
|
|
|
|
template <typename CellSetType>
|
|
void operator()(const CellSetType& vtkmNotUsed(cellset)) const
|
|
{
|
|
}
|
|
};
|
|
template <>
|
|
void DeduceCellSet::operator()(const vtkm::cont::CellSetExplicit<>& cellset) const
|
|
{
|
|
this->OutCellSet = Worklet.Run(cellset);
|
|
}
|
|
template <>
|
|
void DeduceCellSet::operator()(const vtkm::cont::CellSetStructured<2>& cellset) const
|
|
{
|
|
this->OutCellSet = Worklet.Run(cellset);
|
|
}
|
|
template <>
|
|
void DeduceCellSet::operator()(const vtkm::cont::CellSetStructured<3>& cellset) const
|
|
{
|
|
this->OutCellSet = Worklet.Run(cellset);
|
|
}
|
|
}
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace filter
|
|
{
|
|
|
|
//-----------------------------------------------------------------------------
|
|
inline VTKM_CONT Triangulate::Triangulate()
|
|
: vtkm::filter::FilterDataSet<Triangulate>()
|
|
, Worklet()
|
|
{
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
template <typename DerivedPolicy>
|
|
inline VTKM_CONT vtkm::cont::DataSet Triangulate::DoExecute(
|
|
const vtkm::cont::DataSet& input,
|
|
vtkm::filter::PolicyBase<DerivedPolicy> policy)
|
|
{
|
|
const vtkm::cont::DynamicCellSet& cells = input.GetCellSet();
|
|
|
|
vtkm::cont::CellSetSingleType<> outCellSet;
|
|
DeduceCellSet triangulate(this->Worklet, outCellSet);
|
|
|
|
vtkm::cont::CastAndCall(vtkm::filter::ApplyPolicyCellSet(cells, policy, *this), triangulate);
|
|
|
|
// create the output dataset
|
|
vtkm::cont::DataSet output;
|
|
output.SetCellSet(outCellSet);
|
|
output.AddCoordinateSystem(input.GetCoordinateSystem(this->GetActiveCoordinateSystemIndex()));
|
|
|
|
return output;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
template <typename DerivedPolicy>
|
|
inline VTKM_CONT bool Triangulate::MapFieldOntoOutput(vtkm::cont::DataSet& result,
|
|
const vtkm::cont::Field& field,
|
|
vtkm::filter::PolicyBase<DerivedPolicy>)
|
|
{
|
|
if (field.IsFieldPoint())
|
|
{
|
|
// point data is copied as is because it was not collapsed
|
|
result.AddField(field);
|
|
return true;
|
|
}
|
|
else if (field.IsFieldCell())
|
|
{
|
|
// cell data must be scattered to the cells created per input cell
|
|
vtkm::cont::ArrayHandle<vtkm::Id> permutation =
|
|
this->Worklet.GetOutCellScatter().GetOutputToInputMap();
|
|
return vtkm::filter::MapFieldPermutation(field, permutation, result);
|
|
}
|
|
else if (field.IsFieldGlobal())
|
|
{
|
|
result.AddField(field);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
#endif
|