mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-08 13:23:51 +00:00
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.
102 lines
2.9 KiB
C++
102 lines
2.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.
|
|
//============================================================================
|
|
|
|
#ifndef vtk_m_filter_ExtractPoints_hxx
|
|
#define vtk_m_filter_ExtractPoints_hxx
|
|
|
|
#include <vtkm/cont/CoordinateSystem.h>
|
|
#include <vtkm/cont/DynamicCellSet.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace filter
|
|
{
|
|
|
|
//-----------------------------------------------------------------------------
|
|
inline VTKM_CONT ExtractPoints::ExtractPoints()
|
|
: vtkm::filter::FilterDataSet<ExtractPoints>()
|
|
, ExtractInside(true)
|
|
, CompactPoints(false)
|
|
{
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
template <typename DerivedPolicy>
|
|
inline vtkm::cont::DataSet ExtractPoints::DoExecute(const vtkm::cont::DataSet& input,
|
|
vtkm::filter::PolicyBase<DerivedPolicy> policy)
|
|
{
|
|
// extract the input cell set and coordinates
|
|
const vtkm::cont::DynamicCellSet& cells = input.GetCellSet();
|
|
const vtkm::cont::CoordinateSystem& coords =
|
|
input.GetCoordinateSystem(this->GetActiveCoordinateSystemIndex());
|
|
|
|
// run the worklet on the cell set
|
|
vtkm::cont::CellSetSingleType<> outCellSet;
|
|
vtkm::worklet::ExtractPoints worklet;
|
|
|
|
outCellSet = worklet.Run(vtkm::filter::ApplyPolicyCellSet(cells, policy, *this),
|
|
coords.GetData(),
|
|
this->Function,
|
|
this->ExtractInside);
|
|
|
|
// create the output dataset
|
|
vtkm::cont::DataSet output;
|
|
output.SetCellSet(outCellSet);
|
|
output.AddCoordinateSystem(input.GetCoordinateSystem(this->GetActiveCoordinateSystemIndex()));
|
|
|
|
// compact the unused points in the output dataset
|
|
if (this->CompactPoints)
|
|
{
|
|
this->Compactor.SetCompactPointFields(true);
|
|
this->Compactor.SetMergePoints(false);
|
|
return this->Compactor.Execute(output);
|
|
}
|
|
else
|
|
{
|
|
return output;
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
template <typename DerivedPolicy>
|
|
inline VTKM_CONT bool ExtractPoints::MapFieldOntoOutput(
|
|
vtkm::cont::DataSet& result,
|
|
const vtkm::cont::Field& field,
|
|
vtkm::filter::PolicyBase<DerivedPolicy> policy)
|
|
{
|
|
// point data is copied as is because it was not collapsed
|
|
if (field.IsFieldPoint())
|
|
{
|
|
if (this->CompactPoints)
|
|
{
|
|
return this->Compactor.MapFieldOntoOutput(result, field, policy);
|
|
}
|
|
else
|
|
{
|
|
result.AddField(field);
|
|
return true;
|
|
}
|
|
}
|
|
else if (field.IsFieldGlobal())
|
|
{
|
|
result.AddField(field);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
// cell data does not apply
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
#endif
|