mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-20 02:55:47 +00:00
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
|