mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-19 10:35:42 +00:00
1889447d82
The previous implementation of the map field in the clip filters (`ClipWithField` and `ClipWithImplicitFunction`) checked for common field types and interpolated those. If the field value type did not match, it would either convert the field to floats (which is at odds with what VTK does) or fail outright if the `Vec` length is not supported. The map field function for clip has been changed to support all possible types. It does this by using the extract component functionality to get data from any type of array.
81 lines
2.5 KiB
C++
81 lines
2.5 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/CoordinateSystem.h>
|
|
#include <vtkm/cont/UnknownCellSet.h>
|
|
#include <vtkm/filter/MapFieldPermutation.h>
|
|
#include <vtkm/filter/contour/ClipWithImplicitFunction.h>
|
|
#include <vtkm/filter/contour/worklet/Clip.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace filter
|
|
{
|
|
namespace contour
|
|
{
|
|
namespace
|
|
{
|
|
|
|
bool DoMapField(vtkm::cont::DataSet& result,
|
|
const vtkm::cont::Field& field,
|
|
vtkm::worklet::Clip& worklet)
|
|
{
|
|
if (field.IsPointField())
|
|
{
|
|
auto resolve = [&](const auto& concrete) {
|
|
// use std::decay to remove const ref from the decltype of concrete.
|
|
using T = typename std::decay_t<decltype(concrete)>::ValueType;
|
|
vtkm::cont::ArrayHandle<T> outputArray;
|
|
worklet.ProcessPointField(concrete, outputArray);
|
|
result.AddPointField(field.GetName(), outputArray);
|
|
};
|
|
|
|
field.GetData()
|
|
.CastAndCallForTypesWithFloatFallback<VTKM_DEFAULT_TYPE_LIST, VTKM_DEFAULT_STORAGE_LIST>(
|
|
resolve);
|
|
return true;
|
|
}
|
|
else if (field.IsCellField())
|
|
{
|
|
// Use the precompiled field permutation function.
|
|
vtkm::cont::ArrayHandle<vtkm::Id> permutation = worklet.GetCellMapOutputToInput();
|
|
return vtkm::filter::MapFieldPermutation(field, permutation, result);
|
|
}
|
|
else if (field.IsWholeDataSetField())
|
|
{
|
|
result.AddField(field);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
} // anonymous namespace
|
|
|
|
//-----------------------------------------------------------------------------
|
|
vtkm::cont::DataSet ClipWithImplicitFunction::DoExecute(const vtkm::cont::DataSet& input)
|
|
{
|
|
const vtkm::cont::UnknownCellSet& inputCellSet = input.GetCellSet();
|
|
const vtkm::cont::CoordinateSystem& inputCoords =
|
|
input.GetCoordinateSystem(this->GetActiveCoordinateSystemIndex());
|
|
|
|
vtkm::worklet::Clip worklet;
|
|
|
|
vtkm::cont::CellSetExplicit<> outputCellSet =
|
|
worklet.Run(inputCellSet, this->Function, this->Offset, inputCoords, this->Invert);
|
|
|
|
auto mapper = [&](auto& result, const auto& f) { DoMapField(result, f, worklet); };
|
|
return this->CreateResult(input, outputCellSet, mapper);
|
|
}
|
|
} // namespace contour
|
|
} // namespace filter
|
|
} // namespace vtkm
|