mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-08 21:33:55 +00:00
760553725c
`ArrayExtractComponent` allows you to get a component of an array. Unlike `ArrayHandleExtractComponent`, the type you get is always the same: an `ArrayHandleStride`. This way, you can get an array that contains the data of an extracted component with less templating and potentially dramatically reduce the amount of code generated (although some runtime integer arithmetic is added).
93 lines
3.1 KiB
C++
93 lines
3.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/ArrayHandleUniformPointCoordinates.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace cont
|
|
{
|
|
|
|
ArrayHandleUniformPointCoordinates::ArrayHandleUniformPointCoordinates(vtkm::Id3 dimensions,
|
|
ValueType origin,
|
|
ValueType spacing)
|
|
: Superclass(internal::PortalToArrayHandleImplicitBuffers(
|
|
vtkm::internal::ArrayPortalUniformPointCoordinates(dimensions, origin, spacing)))
|
|
{
|
|
}
|
|
|
|
ArrayHandleUniformPointCoordinates::~ArrayHandleUniformPointCoordinates() = default;
|
|
|
|
vtkm::Id3 ArrayHandleUniformPointCoordinates::GetDimensions() const
|
|
{
|
|
return this->ReadPortal().GetDimensions();
|
|
}
|
|
|
|
vtkm::Vec3f ArrayHandleUniformPointCoordinates::GetOrigin() const
|
|
{
|
|
return this->ReadPortal().GetOrigin();
|
|
}
|
|
|
|
vtkm::Vec3f ArrayHandleUniformPointCoordinates::GetSpacing() const
|
|
{
|
|
return this->ReadPortal().GetSpacing();
|
|
}
|
|
|
|
namespace internal
|
|
{
|
|
|
|
vtkm::cont::ArrayHandleStride<vtkm::FloatDefault>
|
|
ArrayExtractComponentImpl<vtkm::cont::StorageTagUniformPoints>::operator()(
|
|
const vtkm::cont::ArrayHandleUniformPointCoordinates& src,
|
|
vtkm::IdComponent componentIndex,
|
|
vtkm::CopyFlag allowCopy) const
|
|
{
|
|
if (allowCopy != vtkm::CopyFlag::On)
|
|
{
|
|
throw vtkm::cont::ErrorBadValue(
|
|
"Cannot extract component of ArrayHandleUniformPointCoordinates without copying. "
|
|
"(However, the whole array does not need to be copied.)");
|
|
}
|
|
|
|
vtkm::Id3 dims = src.GetDimensions();
|
|
vtkm::Vec3f origin = src.GetOrigin();
|
|
vtkm::Vec3f spacing = src.GetSpacing();
|
|
|
|
// A "slow" way to create the data, but the array is probably short. It would probably take
|
|
// longer to schedule something on a device. (Can change that later if use cases change.)
|
|
vtkm::cont::ArrayHandleBasic<vtkm::FloatDefault> componentArray;
|
|
componentArray.Allocate(dims[componentIndex]);
|
|
auto portal = componentArray.WritePortal();
|
|
for (vtkm::Id i = 0; i < dims[componentIndex]; ++i)
|
|
{
|
|
portal.Set(i, origin[componentIndex] + (i * spacing[componentIndex]));
|
|
}
|
|
|
|
switch (componentIndex)
|
|
{
|
|
case 0:
|
|
return vtkm::cont::ArrayHandleStride<vtkm::FloatDefault>(
|
|
componentArray, src.GetNumberOfValues(), 1, 0, dims[0], 1);
|
|
case 1:
|
|
return vtkm::cont::ArrayHandleStride<vtkm::FloatDefault>(
|
|
componentArray, src.GetNumberOfValues(), 1, 0, dims[1], dims[0]);
|
|
case 2:
|
|
return vtkm::cont::ArrayHandleStride<vtkm::FloatDefault>(
|
|
componentArray, src.GetNumberOfValues(), 1, 0, 0, dims[0] * dims[1]);
|
|
default:
|
|
throw vtkm::cont::ErrorBadValue("Bad index given to ArrayExtractComponent.");
|
|
}
|
|
}
|
|
|
|
} // namespace internal
|
|
|
|
}
|
|
} // namespace vtkm::cont
|