mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-08 21:33:55 +00:00
7d681fb585
Instead, always use precompiled versions of range computing. This means you won't be able to specify the type. Currently, types are limited to scalars vecs up to size 4. The main motivation for this change is to allow you to include Field.h with a non-device compiler. This is an important feature for our customers. I plan in the future to implement a mechanism to pull out a component of most ArrayHandle's as a single array. This would enable us to support a precompiled version that can compute the range of arbitrarily sized Vecs.
73 lines
2.6 KiB
C++
73 lines
2.6 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/FieldRangeCompute.h>
|
|
|
|
#include <vtkm/cont/Algorithm.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace cont
|
|
{
|
|
|
|
//-----------------------------------------------------------------------------
|
|
VTKM_CONT
|
|
vtkm::cont::ArrayHandle<vtkm::Range> FieldRangeCompute(const vtkm::cont::DataSet& dataset,
|
|
const std::string& name,
|
|
vtkm::cont::Field::Association assoc)
|
|
{
|
|
vtkm::cont::Field field;
|
|
try
|
|
{
|
|
field = dataset.GetField(name, assoc);
|
|
}
|
|
catch (vtkm::cont::ErrorBadValue&)
|
|
{
|
|
// field missing, return empty range.
|
|
return vtkm::cont::ArrayHandle<vtkm::Range>();
|
|
}
|
|
|
|
return field.GetRange();
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
VTKM_CONT
|
|
vtkm::cont::ArrayHandle<vtkm::Range> FieldRangeCompute(const vtkm::cont::PartitionedDataSet& pds,
|
|
const std::string& name,
|
|
vtkm::cont::Field::Association assoc)
|
|
{
|
|
std::vector<vtkm::Range> result_vector = std::accumulate(
|
|
pds.begin(),
|
|
pds.end(),
|
|
std::vector<vtkm::Range>(),
|
|
[&](const std::vector<vtkm::Range>& accumulated_value, const vtkm::cont::DataSet& dataset) {
|
|
vtkm::cont::ArrayHandle<vtkm::Range> partition_range =
|
|
vtkm::cont::FieldRangeCompute(dataset, name, assoc);
|
|
|
|
std::vector<vtkm::Range> result = accumulated_value;
|
|
|
|
// if the current partition has more components than we have seen so far,
|
|
// resize the result to fit all components.
|
|
result.resize(
|
|
std::max(result.size(), static_cast<size_t>(partition_range.GetNumberOfValues())));
|
|
|
|
auto portal = partition_range.ReadPortal();
|
|
std::transform(vtkm::cont::ArrayPortalToIteratorBegin(portal),
|
|
vtkm::cont::ArrayPortalToIteratorEnd(portal),
|
|
result.begin(),
|
|
result.begin(),
|
|
std::plus<vtkm::Range>());
|
|
return result;
|
|
});
|
|
|
|
return vtkm::cont::make_ArrayHandleMove(std::move(result_vector));
|
|
}
|
|
}
|
|
} // namespace vtkm::cont
|