mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-16 17:22:55 +00:00
63 lines
2.1 KiB
C++
63 lines
2.1 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/CellSetExtrude.h>
|
|
#include <vtkm/cont/ErrorFilterExecution.h>
|
|
#include <vtkm/cont/UncertainCellSet.h>
|
|
#include <vtkm/cont/UnknownCellSet.h>
|
|
#include <vtkm/filter/field_conversion/PointAverage.h>
|
|
#include <vtkm/filter/field_conversion/worklet/PointAverage.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace filter
|
|
{
|
|
namespace field_conversion
|
|
{
|
|
vtkm::cont::DataSet PointAverage::DoExecute(const vtkm::cont::DataSet& input)
|
|
{
|
|
const auto& field = GetFieldFromDataSet(input);
|
|
if (!field.IsCellField())
|
|
{
|
|
throw vtkm::cont::ErrorFilterExecution("Cell field expected.");
|
|
}
|
|
|
|
vtkm::cont::UnknownCellSet cellSet = input.GetCellSet();
|
|
vtkm::cont::UnknownArrayHandle outArray;
|
|
|
|
auto resolveType = [&](const auto& concrete) {
|
|
using T = typename std::decay_t<decltype(concrete)>::ValueType;
|
|
using SupportedCellSets =
|
|
vtkm::ListAppend<vtkm::List<vtkm::cont::CellSetExtrude>, VTKM_DEFAULT_CELL_SET_LIST>;
|
|
|
|
vtkm::cont::ArrayHandle<T> result;
|
|
this->Invoke(vtkm::worklet::PointAverage{},
|
|
cellSet.ResetCellSetList<SupportedCellSets>(),
|
|
concrete,
|
|
result);
|
|
outArray = result;
|
|
};
|
|
// TODO: Do we need to deal with XCG storage type explicitly?
|
|
// using AdditionalFieldStorage = vtkm::List<vtkm::cont::StorageTagXGCCoordinates>;
|
|
field.GetData()
|
|
.CastAndCallForTypesWithFloatFallback<vtkm::TypeListField, VTKM_DEFAULT_STORAGE_LIST>(
|
|
resolveType);
|
|
|
|
std::string outputName = this->GetOutputFieldName();
|
|
if (outputName.empty())
|
|
{
|
|
// Default name is name of input.
|
|
outputName = field.GetName();
|
|
}
|
|
return this->CreateResultFieldPoint(input, outputName, outArray);
|
|
}
|
|
}
|
|
}
|
|
}
|