mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-10-05 18:08:59 +00:00
5ef4e7eeee
What was previously declared as `ArrayHandleNewStyle` is now just the implementation of `ArrayHandle`. The old implementation of `ArrayHandle` has been moved to `ArrayHandleDeprecated`, and `ArrayHandle`s still using this implementation must declare `VTKM_ARRAY_HANDLE_DEPRECATED` to use it.
133 lines
3.3 KiB
C++
133 lines
3.3 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.
|
|
//============================================================================
|
|
#ifndef vtk_m_cont_ArrayHandleIndex_h
|
|
#define vtk_m_cont_ArrayHandleIndex_h
|
|
|
|
#include <vtkm/cont/ArrayHandleImplicit.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
|
|
namespace internal
|
|
{
|
|
|
|
struct VTKM_ALWAYS_EXPORT IndexFunctor
|
|
{
|
|
VTKM_EXEC_CONT vtkm::Id operator()(vtkm::Id index) const { return index; }
|
|
};
|
|
|
|
} // namespace internal
|
|
|
|
namespace cont
|
|
{
|
|
|
|
struct VTKM_ALWAYS_EXPORT StorageTagIndex
|
|
{
|
|
};
|
|
|
|
namespace internal
|
|
{
|
|
|
|
using StorageTagIndexSuperclass =
|
|
typename vtkm::cont::ArrayHandleImplicit<vtkm::internal::IndexFunctor>::StorageTag;
|
|
|
|
template <>
|
|
struct Storage<vtkm::Id, vtkm::cont::StorageTagIndex> : Storage<vtkm::Id, StorageTagIndexSuperclass>
|
|
{
|
|
};
|
|
|
|
} // namespace internal
|
|
|
|
/// \brief An implicit array handle containing the its own indices.
|
|
///
|
|
/// \c ArrayHandleIndex is an implicit array handle containing the values
|
|
/// 0, 1, 2, 3,... to a specified size. Every value in the array is the same
|
|
/// as the index to that value.
|
|
///
|
|
class ArrayHandleIndex : public vtkm::cont::ArrayHandle<vtkm::Id, StorageTagIndex>
|
|
{
|
|
public:
|
|
VTKM_ARRAY_HANDLE_SUBCLASS_NT(ArrayHandleIndex,
|
|
(vtkm::cont::ArrayHandle<vtkm::Id, StorageTagIndex>));
|
|
|
|
VTKM_CONT
|
|
ArrayHandleIndex(vtkm::Id length)
|
|
: Superclass(
|
|
internal::FunctorToArrayHandleImplicitBuffers(vtkm::internal::IndexFunctor{}, length))
|
|
{
|
|
}
|
|
};
|
|
|
|
/// A convenience function for creating an ArrayHandleIndex. It takes the
|
|
/// size of the array and generates an array holding vtkm::Id from [0, size - 1]
|
|
VTKM_CONT inline vtkm::cont::ArrayHandleIndex make_ArrayHandleIndex(vtkm::Id length)
|
|
{
|
|
return vtkm::cont::ArrayHandleIndex(length);
|
|
}
|
|
}
|
|
} // namespace vtkm::cont
|
|
|
|
//=============================================================================
|
|
// Specializations of serialization related classes
|
|
/// @cond SERIALIZATION
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace cont
|
|
{
|
|
|
|
template <>
|
|
struct SerializableTypeString<vtkm::cont::ArrayHandleIndex>
|
|
{
|
|
static VTKM_CONT const std::string Get() { return "AH_Index"; }
|
|
};
|
|
|
|
template <>
|
|
struct SerializableTypeString<vtkm::cont::ArrayHandle<vtkm::Id, vtkm::cont::StorageTagIndex>>
|
|
: SerializableTypeString<vtkm::cont::ArrayHandleIndex>
|
|
{
|
|
};
|
|
}
|
|
} // vtkm::cont
|
|
|
|
namespace mangled_diy_namespace
|
|
{
|
|
|
|
template <>
|
|
struct Serialization<vtkm::cont::ArrayHandleIndex>
|
|
{
|
|
private:
|
|
using BaseType = vtkm::cont::ArrayHandle<vtkm::Id, vtkm::cont::StorageTagIndex>;
|
|
|
|
public:
|
|
static VTKM_CONT void save(BinaryBuffer& bb, const BaseType& obj)
|
|
{
|
|
vtkmdiy::save(bb, obj.GetNumberOfValues());
|
|
}
|
|
|
|
static VTKM_CONT void load(BinaryBuffer& bb, BaseType& obj)
|
|
{
|
|
vtkm::Id length = 0;
|
|
vtkmdiy::load(bb, length);
|
|
|
|
obj = vtkm::cont::ArrayHandleIndex(length);
|
|
}
|
|
};
|
|
|
|
template <>
|
|
struct Serialization<vtkm::cont::ArrayHandle<vtkm::Id, vtkm::cont::StorageTagIndex>>
|
|
: Serialization<vtkm::cont::ArrayHandleIndex>
|
|
{
|
|
};
|
|
} // diy
|
|
/// @endcond SERIALIZATION
|
|
|
|
#endif //vtk_m_cont_ArrayHandleIndex_h
|