mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-16 17:22:55 +00:00
Add MapFieldPermutation function
This function is compiled into the vtkm_filter library, so all filters can implement this type of mapping without providing their own version of compiling it. Because we only compile it once, we provide more types to convert. Hopefully the total compile time balances out.
This commit is contained in:
parent
0d2678368b
commit
934732bb64
@ -45,6 +45,7 @@ set(headers
|
|||||||
ImageMedian.h
|
ImageMedian.h
|
||||||
Lagrangian.h
|
Lagrangian.h
|
||||||
LagrangianStructures.h
|
LagrangianStructures.h
|
||||||
|
MapFieldPermutation.h
|
||||||
Mask.h
|
Mask.h
|
||||||
MaskPoints.h
|
MaskPoints.h
|
||||||
MeshQuality.h
|
MeshQuality.h
|
||||||
@ -154,6 +155,7 @@ set(sources_device
|
|||||||
GradientScalar.cxx
|
GradientScalar.cxx
|
||||||
GradientUniformPoints.cxx
|
GradientUniformPoints.cxx
|
||||||
GradientVector.cxx
|
GradientVector.cxx
|
||||||
|
MapFieldPermutation.cxx
|
||||||
PointAverage.cxx
|
PointAverage.cxx
|
||||||
Threshold.cxx
|
Threshold.cxx
|
||||||
VectorMagnitude.cxx
|
VectorMagnitude.cxx
|
||||||
|
73
vtkm/filter/MapFieldPermutation.cxx
Normal file
73
vtkm/filter/MapFieldPermutation.cxx
Normal file
@ -0,0 +1,73 @@
|
|||||||
|
//============================================================================
|
||||||
|
// 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/filter/MapFieldPermutation.h>
|
||||||
|
|
||||||
|
#include <vtkm/TypeList.h>
|
||||||
|
|
||||||
|
#include <vtkm/cont/ArrayCopy.h>
|
||||||
|
#include <vtkm/cont/ArrayHandlePermutation.h>
|
||||||
|
#include <vtkm/cont/Logging.h>
|
||||||
|
|
||||||
|
#include <vtkm/filter/PolicyDefault.h>
|
||||||
|
|
||||||
|
namespace
|
||||||
|
{
|
||||||
|
|
||||||
|
struct DoMapFieldPermutation
|
||||||
|
{
|
||||||
|
bool CalledMap = false;
|
||||||
|
|
||||||
|
template <typename T, typename S>
|
||||||
|
void operator()(const vtkm::cont::ArrayHandle<T, S>& inputArray,
|
||||||
|
const vtkm::cont::ArrayHandle<vtkm::Id>& permutation,
|
||||||
|
vtkm::cont::VariantArrayHandle& output)
|
||||||
|
{
|
||||||
|
vtkm::cont::ArrayHandle<T> outputArray;
|
||||||
|
vtkm::cont::ArrayCopy(vtkm::cont::make_ArrayHandlePermutation(permutation, inputArray),
|
||||||
|
outputArray);
|
||||||
|
output = vtkm::cont::VariantArrayHandle(outputArray);
|
||||||
|
this->CalledMap = true;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
} // anonymous namespace
|
||||||
|
|
||||||
|
bool vtkm::filter::MapFieldPermutation(const vtkm::cont::Field& inputField,
|
||||||
|
const vtkm::cont::ArrayHandle<vtkm::Id>& permutation,
|
||||||
|
vtkm::cont::Field& outputField)
|
||||||
|
{
|
||||||
|
vtkm::cont::VariantArrayHandle outputArray;
|
||||||
|
DoMapFieldPermutation functor;
|
||||||
|
inputField.GetData().ResetTypes<vtkm::TypeListAll>().CastAndCall(
|
||||||
|
vtkm::filter::PolicyDefault::StorageList{}, functor, permutation, outputArray);
|
||||||
|
if (functor.CalledMap)
|
||||||
|
{
|
||||||
|
outputField = vtkm::cont::Field(inputField.GetName(), inputField.GetAssociation(), outputArray);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
VTKM_LOG_S(vtkm::cont::LogLevel::Warn, "Faild to map field " << inputField.GetName());
|
||||||
|
}
|
||||||
|
return functor.CalledMap;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool MapFieldPermutation(const vtkm::cont::Field& inputField,
|
||||||
|
const vtkm::cont::ArrayHandle<vtkm::Id>& permutation,
|
||||||
|
vtkm::cont::DataSet& outputData)
|
||||||
|
{
|
||||||
|
vtkm::cont::Field outputField;
|
||||||
|
bool success = vtkm::filter::MapFieldPermutation(inputField, permutation, outputField);
|
||||||
|
if (success)
|
||||||
|
{
|
||||||
|
outputData.AddField(outputField);
|
||||||
|
}
|
||||||
|
return success;
|
||||||
|
}
|
72
vtkm/filter/MapFieldPermutation.h
Normal file
72
vtkm/filter/MapFieldPermutation.h
Normal file
@ -0,0 +1,72 @@
|
|||||||
|
//============================================================================
|
||||||
|
// 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_MapFieldPermutation_h
|
||||||
|
#define vtk_m_filter_MapFieldPermutation_h
|
||||||
|
|
||||||
|
#include <vtkm/cont/ArrayHandle.h>
|
||||||
|
#include <vtkm/cont/DataSet.h>
|
||||||
|
#include <vtkm/cont/Field.h>
|
||||||
|
|
||||||
|
#include <vtkm/filter/vtkm_filter_export.h>
|
||||||
|
|
||||||
|
namespace vtkm
|
||||||
|
{
|
||||||
|
namespace filter
|
||||||
|
{
|
||||||
|
|
||||||
|
/// \brief Maps a field by permuting it by a given index array.
|
||||||
|
///
|
||||||
|
/// This method will create a new field containing the data from the provided `inputField` but
|
||||||
|
/// reorded by the given `permutation` index array. The value in the resulting field for index _i_
|
||||||
|
/// will be be a value from `inputField`, but comes from the index that comes from `permutation` at
|
||||||
|
/// position _i_. The result is placed in `outputField`.
|
||||||
|
///
|
||||||
|
/// The intention of this method is to implement the `MapFieldOntoOutput` methods in filters (many
|
||||||
|
/// of which require this permutation of a field), but can be used in other places as well.
|
||||||
|
///
|
||||||
|
/// `outputField` is set to have the same metadata as the input. If the metadata needs to change
|
||||||
|
/// (such as the name or the association) that should be done after the function returns.
|
||||||
|
///
|
||||||
|
/// This function returns whether the field was successfully permuted. If the returned result is
|
||||||
|
/// `true`, then the results in `outputField` are valid. If it is `false`, then `outputField`
|
||||||
|
/// should not be used.
|
||||||
|
///
|
||||||
|
VTKM_FILTER_EXPORT VTKM_CONT bool MapFieldPermutation(
|
||||||
|
const vtkm::cont::Field& inputField,
|
||||||
|
const vtkm::cont::ArrayHandle<vtkm::Id>& permutation,
|
||||||
|
vtkm::cont::Field& outputField);
|
||||||
|
|
||||||
|
/// \brief Maps a field by permuting it by a given index array.
|
||||||
|
///
|
||||||
|
/// This method will create a new field containing the data from the provided `inputField` but
|
||||||
|
/// reorded by the given `permutation` index array. The value in the resulting field for index _i_
|
||||||
|
/// will be be a value from `inputField`, but comes from the index that comes from `permutation` at
|
||||||
|
/// position _i_.
|
||||||
|
///
|
||||||
|
/// The intention of this method is to implement the `MapFieldOntoOutput` methods in filters (many
|
||||||
|
/// of which require this permutation of a field), but can be used in other places as well. The
|
||||||
|
/// resulting field is put in the given `DataSet`.
|
||||||
|
///
|
||||||
|
/// The returned `Field` has the same metadata as the input. If the metadata needs to change (such
|
||||||
|
/// as the name or the association), then a different form of `MapFieldPermutation` should be used.
|
||||||
|
///
|
||||||
|
/// This function returns whether the field was successfully permuted. If the returned result is
|
||||||
|
/// `true`, then `outputData` has the permuted field. If it is `false`, then the field is not
|
||||||
|
/// placed in `outputData`.
|
||||||
|
///
|
||||||
|
VTKM_FILTER_EXPORT VTKM_CONT bool MapFieldPermutation(
|
||||||
|
const vtkm::cont::Field& inputField,
|
||||||
|
const vtkm::cont::ArrayHandle<vtkm::Id>& permutation,
|
||||||
|
vtkm::cont::DataSet& outputData);
|
||||||
|
}
|
||||||
|
} // namespace vtkm::filter
|
||||||
|
|
||||||
|
#endif //vtk_m_filter_MapFieldPermutation_h
|
@ -39,6 +39,7 @@ set(unit_tests
|
|||||||
UnitTestImageMedianFilter.cxx
|
UnitTestImageMedianFilter.cxx
|
||||||
UnitTestLagrangianFilter.cxx
|
UnitTestLagrangianFilter.cxx
|
||||||
UnitTestLagrangianStructuresFilter.cxx
|
UnitTestLagrangianStructuresFilter.cxx
|
||||||
|
UnitTestMapFieldPermutation.cxx
|
||||||
UnitTestMaskFilter.cxx
|
UnitTestMaskFilter.cxx
|
||||||
UnitTestMaskPointsFilter.cxx
|
UnitTestMaskPointsFilter.cxx
|
||||||
UnitTestMeshQualityFilter.cxx
|
UnitTestMeshQualityFilter.cxx
|
||||||
|
114
vtkm/filter/testing/UnitTestMapFieldPermutation.cxx
Normal file
114
vtkm/filter/testing/UnitTestMapFieldPermutation.cxx
Normal file
@ -0,0 +1,114 @@
|
|||||||
|
//============================================================================
|
||||||
|
// 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/filter/MapFieldPermutation.h>
|
||||||
|
|
||||||
|
#include <vtkm/cont/ArrayCopy.h>
|
||||||
|
#include <vtkm/cont/ArrayHandle.h>
|
||||||
|
#include <vtkm/cont/ArrayHandleCartesianProduct.h>
|
||||||
|
#include <vtkm/cont/ArrayHandleCounting.h>
|
||||||
|
#include <vtkm/cont/ArrayHandlePermutation.h>
|
||||||
|
#include <vtkm/cont/ArrayHandleUniformPointCoordinates.h>
|
||||||
|
#include <vtkm/cont/Field.h>
|
||||||
|
|
||||||
|
#include <vtkm/cont/testing/Testing.h>
|
||||||
|
|
||||||
|
namespace
|
||||||
|
{
|
||||||
|
|
||||||
|
constexpr vtkm::Id ARRAY_SIZE = 26;
|
||||||
|
constexpr vtkm::Id3 ARRAY3_DIM = { 3, 3, 3 };
|
||||||
|
|
||||||
|
template <typename T, typename S>
|
||||||
|
void TryArray(const vtkm::cont::ArrayHandle<T, S>& inputArray)
|
||||||
|
{
|
||||||
|
std::cout << "Input" << std::endl;
|
||||||
|
vtkm::cont::printSummary_ArrayHandle(inputArray, std::cout);
|
||||||
|
|
||||||
|
vtkm::cont::Field::Association association =
|
||||||
|
((sizeof(T) < 8) ? vtkm::cont::Field::Association::POINTS
|
||||||
|
: vtkm::cont::Field::Association::CELL_SET);
|
||||||
|
|
||||||
|
vtkm::cont::Field inputField("my-array", association, inputArray);
|
||||||
|
|
||||||
|
vtkm::cont::ArrayHandle<vtkm::Id> permutationArray;
|
||||||
|
vtkm::cont::ArrayCopy(
|
||||||
|
vtkm::cont::make_ArrayHandleCounting<vtkm::Id>(0, 2, inputArray.GetNumberOfValues() / 2),
|
||||||
|
permutationArray);
|
||||||
|
|
||||||
|
vtkm::cont::ArrayHandle<T> expectedOutputArray;
|
||||||
|
vtkm::cont::ArrayCopy(vtkm::cont::make_ArrayHandlePermutation(permutationArray, inputArray),
|
||||||
|
expectedOutputArray);
|
||||||
|
std::cout << "Expected output" << std::endl;
|
||||||
|
vtkm::cont::printSummary_ArrayHandle(expectedOutputArray, std::cout);
|
||||||
|
|
||||||
|
vtkm::cont::Field outputField;
|
||||||
|
bool result = vtkm::filter::MapFieldPermutation(inputField, permutationArray, outputField);
|
||||||
|
VTKM_TEST_ASSERT(result, "Could not permute the array.");
|
||||||
|
|
||||||
|
VTKM_TEST_ASSERT(outputField.GetAssociation() == association);
|
||||||
|
VTKM_TEST_ASSERT(outputField.GetName() == "my-array");
|
||||||
|
|
||||||
|
vtkm::cont::ArrayHandle<T> outputArray;
|
||||||
|
outputField.GetData().CopyTo(outputArray);
|
||||||
|
std::cout << "Actual output" << std::endl;
|
||||||
|
vtkm::cont::printSummary_ArrayHandle(outputArray, std::cout);
|
||||||
|
|
||||||
|
VTKM_TEST_ASSERT(test_equal_portals(expectedOutputArray.ReadPortal(), outputArray.ReadPortal()));
|
||||||
|
}
|
||||||
|
|
||||||
|
template <typename T>
|
||||||
|
void TryType(T)
|
||||||
|
{
|
||||||
|
vtkm::cont::ArrayHandle<T> inputArray;
|
||||||
|
inputArray.Allocate(ARRAY_SIZE);
|
||||||
|
SetPortal(inputArray.WritePortal());
|
||||||
|
TryArray(inputArray);
|
||||||
|
}
|
||||||
|
|
||||||
|
struct TryTypeFunctor
|
||||||
|
{
|
||||||
|
template <typename T>
|
||||||
|
void operator()(T x) const
|
||||||
|
{
|
||||||
|
TryType(x);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
void TryCartesianProduct()
|
||||||
|
{
|
||||||
|
vtkm::cont::ArrayHandle<vtkm::FloatDefault> axes[3];
|
||||||
|
for (vtkm::IdComponent i = 0; i < 3; ++i)
|
||||||
|
{
|
||||||
|
axes[i].Allocate(ARRAY3_DIM[i]);
|
||||||
|
SetPortal(axes[i].WritePortal());
|
||||||
|
}
|
||||||
|
|
||||||
|
TryArray(vtkm::cont::make_ArrayHandleCartesianProduct(axes[0], axes[1], axes[2]));
|
||||||
|
}
|
||||||
|
|
||||||
|
void DoTest()
|
||||||
|
{
|
||||||
|
std::cout << "**** Test Basic Arrays *****" << std::endl;
|
||||||
|
vtkm::testing::Testing::TryTypes(TryTypeFunctor{});
|
||||||
|
|
||||||
|
std::cout << std::endl << "**** Test Uniform Point Coordiantes *****" << std::endl;
|
||||||
|
TryArray(vtkm::cont::ArrayHandleUniformPointCoordinates(ARRAY3_DIM));
|
||||||
|
|
||||||
|
std::cout << std::endl << "**** Test Cartesian Product *****" << std::endl;
|
||||||
|
TryCartesianProduct();
|
||||||
|
}
|
||||||
|
|
||||||
|
} // anonymous namespace
|
||||||
|
|
||||||
|
int UnitTestMapFieldPermutation(int argc, char* argv[])
|
||||||
|
{
|
||||||
|
return vtkm::cont::testing::Testing::Run(DoTest, argc, argv);
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user