2019-04-15 23:24:21 +00:00
|
|
|
//============================================================================
|
2015-09-15 04:11:09 +00:00
|
|
|
// 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.
|
2019-04-15 23:24:21 +00:00
|
|
|
//============================================================================
|
2015-09-15 04:11:09 +00:00
|
|
|
#ifndef vtk_m_cont_ArrayHandleIndex_h
|
|
|
|
#define vtk_m_cont_ArrayHandleIndex_h
|
|
|
|
|
|
|
|
#include <vtkm/cont/ArrayHandleImplicit.h>
|
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
namespace vtkm
|
|
|
|
{
|
|
|
|
namespace cont
|
|
|
|
{
|
2015-09-15 04:11:09 +00:00
|
|
|
|
2020-01-04 06:46:43 +00:00
|
|
|
struct VTKM_ALWAYS_EXPORT StorageTagIndex
|
2020-01-02 00:11:00 +00:00
|
|
|
{
|
|
|
|
};
|
|
|
|
|
|
|
|
namespace internal
|
2017-05-18 14:29:41 +00:00
|
|
|
{
|
2015-09-15 04:11:09 +00:00
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
struct VTKM_ALWAYS_EXPORT IndexFunctor
|
|
|
|
{
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_EXEC_CONT
|
2015-09-15 04:11:09 +00:00
|
|
|
vtkm::Id operator()(vtkm::Id index) const { return index; }
|
|
|
|
};
|
|
|
|
|
2020-01-02 00:11:00 +00:00
|
|
|
using StorageTagIndexSuperclass =
|
|
|
|
typename vtkm::cont::ArrayHandleImplicit<IndexFunctor>::StorageTag;
|
|
|
|
|
|
|
|
template <>
|
2020-01-04 06:46:43 +00:00
|
|
|
struct Storage<vtkm::Id, vtkm::cont::StorageTagIndex> : Storage<vtkm::Id, StorageTagIndexSuperclass>
|
2020-01-02 00:11:00 +00:00
|
|
|
{
|
|
|
|
using Superclass = Storage<vtkm::Id, StorageTagIndexSuperclass>;
|
|
|
|
using Superclass::Superclass;
|
|
|
|
};
|
|
|
|
|
|
|
|
template <typename Device>
|
2020-01-04 06:46:43 +00:00
|
|
|
struct ArrayTransfer<vtkm::Id, vtkm::cont::StorageTagIndex, Device>
|
2020-01-02 00:11:00 +00:00
|
|
|
: ArrayTransfer<vtkm::Id, StorageTagIndexSuperclass, Device>
|
|
|
|
{
|
|
|
|
using Superclass = ArrayTransfer<vtkm::Id, StorageTagIndexSuperclass, Device>;
|
|
|
|
using Superclass::Superclass;
|
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace internal
|
2015-09-15 04:11:09 +00:00
|
|
|
|
|
|
|
/// \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.
|
|
|
|
///
|
2020-01-02 00:11:00 +00:00
|
|
|
class ArrayHandleIndex : public vtkm::cont::ArrayHandle<vtkm::Id, StorageTagIndex>
|
2015-09-15 04:11:09 +00:00
|
|
|
{
|
|
|
|
public:
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_ARRAY_HANDLE_SUBCLASS_NT(ArrayHandleIndex,
|
2020-01-02 00:11:00 +00:00
|
|
|
(vtkm::cont::ArrayHandle<vtkm::Id, StorageTagIndex>));
|
2015-09-20 06:28:11 +00:00
|
|
|
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_CONT
|
2015-10-21 19:36:27 +00:00
|
|
|
ArrayHandleIndex(vtkm::Id length)
|
2020-01-28 19:14:32 +00:00
|
|
|
: Superclass(
|
|
|
|
typename internal::Storage<vtkm::Id, StorageTagIndex>::PortalType(internal::IndexFunctor{},
|
|
|
|
length))
|
2017-05-18 14:29:41 +00:00
|
|
|
{
|
|
|
|
}
|
2015-09-15 04:11:09 +00:00
|
|
|
};
|
2019-12-26 19:48:51 +00:00
|
|
|
|
|
|
|
/// 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);
|
|
|
|
}
|
2015-09-15 04:11:09 +00:00
|
|
|
}
|
|
|
|
} // namespace vtkm::cont
|
|
|
|
|
2018-06-08 15:56:40 +00:00
|
|
|
//=============================================================================
|
|
|
|
// Specializations of serialization related classes
|
2019-09-12 21:19:36 +00:00
|
|
|
/// @cond SERIALIZATION
|
2018-06-08 15:56:40 +00:00
|
|
|
|
|
|
|
namespace vtkm
|
|
|
|
{
|
|
|
|
namespace cont
|
|
|
|
{
|
|
|
|
|
|
|
|
template <>
|
2020-01-02 00:11:00 +00:00
|
|
|
struct SerializableTypeString<vtkm::cont::ArrayHandleIndex>
|
2018-06-08 15:56:40 +00:00
|
|
|
{
|
2020-01-02 00:11:00 +00:00
|
|
|
static VTKM_CONT const std::string Get() { return "AH_Index"; }
|
2018-06-08 15:56:40 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
template <>
|
2020-01-02 00:11:00 +00:00
|
|
|
struct SerializableTypeString<vtkm::cont::ArrayHandle<vtkm::Id, vtkm::cont::StorageTagIndex>>
|
|
|
|
: SerializableTypeString<vtkm::cont::ArrayHandleIndex>
|
2018-06-08 15:56:40 +00:00
|
|
|
{
|
|
|
|
};
|
|
|
|
}
|
|
|
|
} // vtkm::cont
|
|
|
|
|
2019-02-05 16:02:29 +00:00
|
|
|
namespace mangled_diy_namespace
|
2018-06-08 15:56:40 +00:00
|
|
|
{
|
|
|
|
|
|
|
|
template <>
|
2020-01-02 00:11:00 +00:00
|
|
|
struct Serialization<vtkm::cont::ArrayHandleIndex>
|
2018-06-08 15:56:40 +00:00
|
|
|
{
|
2020-01-02 00:11:00 +00:00
|
|
|
private:
|
|
|
|
using BaseType = vtkm::cont::ArrayHandle<vtkm::Id, vtkm::cont::StorageTagIndex>;
|
2018-06-08 15:56:40 +00:00
|
|
|
|
2020-01-02 00:11:00 +00:00
|
|
|
public:
|
|
|
|
static VTKM_CONT void save(BinaryBuffer& bb, const BaseType& obj)
|
|
|
|
{
|
|
|
|
vtkmdiy::save(bb, obj.GetNumberOfValues());
|
|
|
|
}
|
2018-06-08 15:56:40 +00:00
|
|
|
|
2020-01-02 00:11:00 +00:00
|
|
|
static VTKM_CONT void load(BinaryBuffer& bb, BaseType& obj)
|
|
|
|
{
|
|
|
|
vtkm::Id length = 0;
|
|
|
|
vtkmdiy::load(bb, length);
|
|
|
|
|
|
|
|
obj = vtkm::cont::ArrayHandleIndex(length);
|
|
|
|
}
|
2018-06-08 15:56:40 +00:00
|
|
|
};
|
|
|
|
|
2020-01-06 19:05:15 +00:00
|
|
|
template <>
|
|
|
|
struct Serialization<vtkm::cont::ArrayHandle<vtkm::Id, vtkm::cont::StorageTagIndex>>
|
|
|
|
: Serialization<vtkm::cont::ArrayHandleIndex>
|
|
|
|
{
|
|
|
|
};
|
2018-06-08 15:56:40 +00:00
|
|
|
} // diy
|
2019-09-12 21:19:36 +00:00
|
|
|
/// @endcond SERIALIZATION
|
2018-06-08 15:56:40 +00:00
|
|
|
|
2015-09-15 04:11:09 +00:00
|
|
|
#endif //vtk_m_cont_ArrayHandleIndex_h
|