mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-19 18:45:43 +00:00
2d30e6d45a
Previously, `DataSet` managed `CoordinateSystem`s separately from `Field`s. However, a `CoordinateSystem` is really just a `Field` with some special attributes. Thus, coordiante systems are now just listed along with the rest of the fields, and the coordinate systems are simply strings that point back to the appropriate field. (This was actually the original concept for `DataSet`, but the coordinate systems were separated from fields for some now obsolete reasons.)
81 lines
2.3 KiB
C++
81 lines
2.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.
|
|
//============================================================================
|
|
#include <vtkm/cont/UnknownCellSet.h>
|
|
#include <vtkm/filter/clean_grid/CleanGrid.h>
|
|
#include <vtkm/filter/entity_extraction/ExtractPoints.h>
|
|
#include <vtkm/filter/entity_extraction/worklet/ExtractPoints.h>
|
|
|
|
namespace
|
|
{
|
|
bool DoMapField(vtkm::cont::DataSet& result, const vtkm::cont::Field& field)
|
|
{
|
|
// point data is copied as is because it was not collapsed
|
|
if (field.IsPointField())
|
|
{
|
|
result.AddField(field);
|
|
return true;
|
|
}
|
|
else if (field.IsWholeDataSetField())
|
|
{
|
|
result.AddField(field);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
// cell data does not apply
|
|
return false;
|
|
}
|
|
}
|
|
} // anonymous namespace
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace filter
|
|
{
|
|
namespace entity_extraction
|
|
{
|
|
//-----------------------------------------------------------------------------
|
|
VTKM_CONT
|
|
vtkm::cont::DataSet ExtractPoints::DoExecute(const vtkm::cont::DataSet& input)
|
|
{
|
|
// extract the input cell set and coordinates
|
|
const vtkm::cont::UnknownCellSet& 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;
|
|
|
|
// FIXME: is the other overload of .Run ever used?
|
|
outCellSet = worklet.Run(cells, coords.GetData(), this->Function, this->ExtractInside);
|
|
|
|
// create the output dataset
|
|
auto mapper = [&](auto& result, const auto& f) { DoMapField(result, f); };
|
|
vtkm::cont::DataSet output = this->CreateResult(input, outCellSet, mapper);
|
|
|
|
// compact the unused points in the output dataset
|
|
if (this->CompactPoints)
|
|
{
|
|
vtkm::filter::clean_grid::CleanGrid compactor;
|
|
compactor.SetCompactPointFields(true);
|
|
compactor.SetMergePoints(false);
|
|
return compactor.Execute(output);
|
|
}
|
|
else
|
|
{
|
|
return output;
|
|
}
|
|
}
|
|
|
|
} // namespace entity_extraction
|
|
} // namespace filter
|
|
} // namespace vtkm
|