2014-02-11 17:34:56 +00:00
|
|
|
//============================================================================
|
|
|
|
// Copyright (c) Kitware, Inc.
|
|
|
|
// All rights reserved.
|
|
|
|
// See LICENSE.txt for details.
|
2019-04-15 23:24:21 +00:00
|
|
|
//
|
2014-02-11 17:34:56 +00:00
|
|
|
// 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.
|
|
|
|
//============================================================================
|
2014-06-11 16:43:36 +00:00
|
|
|
#ifndef vtk_m_cont_ArrayHandleCounting_h
|
|
|
|
#define vtk_m_cont_ArrayHandleCounting_h
|
2014-02-11 17:34:56 +00:00
|
|
|
|
2020-08-26 13:58:59 +00:00
|
|
|
#include <vtkm/cont/ArrayHandleImplicit.h>
|
2014-06-23 23:33:04 +00:00
|
|
|
|
2021-01-11 23:28:26 +00:00
|
|
|
#include <vtkm/TypeTraits.h>
|
2015-05-07 22:34:06 +00:00
|
|
|
#include <vtkm/VecTraits.h>
|
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
namespace vtkm
|
|
|
|
{
|
|
|
|
namespace cont
|
|
|
|
{
|
2014-02-11 17:34:56 +00:00
|
|
|
|
2020-01-04 06:46:43 +00:00
|
|
|
struct VTKM_ALWAYS_EXPORT StorageTagCounting
|
2019-12-30 21:23:57 +00:00
|
|
|
{
|
|
|
|
};
|
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
namespace internal
|
|
|
|
{
|
2014-02-11 17:34:56 +00:00
|
|
|
|
|
|
|
/// \brief An implicit array portal that returns an counting value.
|
|
|
|
template <class CountingValueType>
|
2016-12-19 19:39:17 +00:00
|
|
|
class VTKM_ALWAYS_EXPORT ArrayPortalCounting
|
2014-02-11 17:34:56 +00:00
|
|
|
{
|
2017-06-23 18:50:34 +00:00
|
|
|
using ComponentType = typename vtkm::VecTraits<CountingValueType>::ComponentType;
|
2015-05-07 22:34:06 +00:00
|
|
|
|
2014-02-11 17:34:56 +00:00
|
|
|
public:
|
2017-06-23 18:50:34 +00:00
|
|
|
using ValueType = CountingValueType;
|
2014-02-11 17:34:56 +00:00
|
|
|
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_EXEC_CONT
|
2017-05-18 14:29:41 +00:00
|
|
|
ArrayPortalCounting()
|
|
|
|
: Start(0)
|
|
|
|
, Step(1)
|
|
|
|
, NumberOfValues(0)
|
|
|
|
{
|
|
|
|
}
|
2014-02-11 17:34:56 +00:00
|
|
|
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_EXEC_CONT
|
2017-05-18 14:29:41 +00:00
|
|
|
ArrayPortalCounting(ValueType start, ValueType step, vtkm::Id numValues)
|
|
|
|
: Start(start)
|
|
|
|
, Step(step)
|
|
|
|
, NumberOfValues(numValues)
|
|
|
|
{
|
|
|
|
}
|
2014-02-11 17:34:56 +00:00
|
|
|
|
2018-06-08 15:56:40 +00:00
|
|
|
VTKM_EXEC_CONT
|
|
|
|
ValueType GetStart() const { return this->Start; }
|
|
|
|
|
|
|
|
VTKM_EXEC_CONT
|
|
|
|
ValueType GetStep() const { return this->Step; }
|
|
|
|
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_EXEC_CONT
|
2014-02-11 17:34:56 +00:00
|
|
|
vtkm::Id GetNumberOfValues() const { return this->NumberOfValues; }
|
|
|
|
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_EXEC_CONT
|
2017-05-18 14:29:41 +00:00
|
|
|
ValueType Get(vtkm::Id index) const
|
|
|
|
{
|
|
|
|
return ValueType(this->Start + this->Step * ValueType(static_cast<ComponentType>(index)));
|
2015-05-07 22:34:06 +00:00
|
|
|
}
|
2014-02-11 17:34:56 +00:00
|
|
|
|
|
|
|
private:
|
2015-09-15 04:55:44 +00:00
|
|
|
ValueType Start;
|
|
|
|
ValueType Step;
|
2014-02-11 17:34:56 +00:00
|
|
|
vtkm::Id NumberOfValues;
|
|
|
|
};
|
|
|
|
|
2021-01-11 23:28:26 +00:00
|
|
|
namespace detail
|
|
|
|
{
|
|
|
|
|
|
|
|
template <typename T, typename UseVecTraits = vtkm::HasVecTraits<T>>
|
|
|
|
struct CanCountImpl;
|
|
|
|
|
|
|
|
template <typename T>
|
|
|
|
struct CanCountImpl<T, std::false_type>
|
|
|
|
{
|
|
|
|
using TTraits = vtkm::TypeTraits<T>;
|
|
|
|
static constexpr bool IsNumeric =
|
|
|
|
!std::is_same<typename TTraits::NumericTag, vtkm::TypeTraitsUnknownTag>::value;
|
|
|
|
|
|
|
|
static constexpr bool value = IsNumeric;
|
|
|
|
};
|
|
|
|
|
|
|
|
template <typename T>
|
|
|
|
struct CanCountImpl<T, std::true_type>
|
|
|
|
{
|
|
|
|
using VTraits = vtkm::VecTraits<T>;
|
|
|
|
using BaseType = typename VTraits::BaseComponentType;
|
|
|
|
static constexpr bool IsBool = std::is_same<BaseType, bool>::value;
|
|
|
|
|
|
|
|
static constexpr bool value = CanCountImpl<BaseType, std::false_type>::value && !IsBool;
|
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace detail
|
|
|
|
|
|
|
|
// Not all types can be counted.
|
|
|
|
template <typename T>
|
|
|
|
struct CanCount
|
|
|
|
{
|
|
|
|
static constexpr bool value = detail::CanCountImpl<T>::value;
|
|
|
|
};
|
|
|
|
|
2019-12-30 21:23:57 +00:00
|
|
|
template <typename T>
|
|
|
|
using StorageTagCountingSuperclass =
|
|
|
|
vtkm::cont::StorageTagImplicit<internal::ArrayPortalCounting<T>>;
|
|
|
|
|
|
|
|
template <typename T>
|
2021-01-11 23:28:26 +00:00
|
|
|
struct Storage<T, typename std::enable_if<CanCount<T>::value, vtkm::cont::StorageTagCounting>::type>
|
|
|
|
: Storage<T, StorageTagCountingSuperclass<T>>
|
2019-12-30 21:23:57 +00:00
|
|
|
{
|
2014-02-11 17:34:56 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
} // namespace internal
|
|
|
|
|
2015-05-13 20:24:51 +00:00
|
|
|
/// ArrayHandleCounting is a specialization of ArrayHandle. By default it
|
2014-02-11 17:34:56 +00:00
|
|
|
/// contains a increment value, that is increment for each step between zero
|
|
|
|
/// and the passed in length
|
|
|
|
template <typename CountingValueType>
|
2019-12-30 21:23:57 +00:00
|
|
|
class ArrayHandleCounting
|
|
|
|
: public vtkm::cont::ArrayHandle<CountingValueType, vtkm::cont::StorageTagCounting>
|
2014-02-11 17:34:56 +00:00
|
|
|
{
|
|
|
|
public:
|
2019-12-30 21:23:57 +00:00
|
|
|
VTKM_ARRAY_HANDLE_SUBCLASS(ArrayHandleCounting,
|
|
|
|
(ArrayHandleCounting<CountingValueType>),
|
|
|
|
(vtkm::cont::ArrayHandle<CountingValueType, StorageTagCounting>));
|
2014-02-11 17:34:56 +00:00
|
|
|
|
2016-10-19 22:42:58 +00:00
|
|
|
VTKM_CONT
|
2017-05-18 14:29:41 +00:00
|
|
|
ArrayHandleCounting(CountingValueType start, CountingValueType step, vtkm::Id length)
|
2020-08-26 13:58:59 +00:00
|
|
|
: Superclass(internal::PortalToArrayHandleImplicitBuffers(
|
|
|
|
internal::ArrayPortalCounting<CountingValueType>(start, step, length)))
|
2014-02-11 17:34:56 +00:00
|
|
|
{
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
/// A convenience function for creating an ArrayHandleCounting. It takes the
|
|
|
|
/// value to start counting from and and the number of times to increment.
|
2017-05-18 14:29:41 +00:00
|
|
|
template <typename CountingValueType>
|
2017-05-26 17:53:28 +00:00
|
|
|
VTKM_CONT vtkm::cont::ArrayHandleCounting<CountingValueType>
|
|
|
|
make_ArrayHandleCounting(CountingValueType start, CountingValueType step, vtkm::Id length)
|
2014-02-11 17:34:56 +00:00
|
|
|
{
|
2017-05-18 14:29:41 +00:00
|
|
|
return vtkm::cont::ArrayHandleCounting<CountingValueType>(start, step, length);
|
2014-02-11 17:34:56 +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 <typename T>
|
2019-02-28 18:15:27 +00:00
|
|
|
struct SerializableTypeString<vtkm::cont::ArrayHandleCounting<T>>
|
2018-06-08 15:56:40 +00:00
|
|
|
{
|
|
|
|
static VTKM_CONT const std::string& Get()
|
|
|
|
{
|
2019-02-28 18:15:27 +00:00
|
|
|
static std::string name = "AH_Counting<" + SerializableTypeString<T>::Get() + ">";
|
2018-06-08 15:56:40 +00:00
|
|
|
return name;
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
template <typename T>
|
2019-12-30 21:23:57 +00:00
|
|
|
struct SerializableTypeString<vtkm::cont::ArrayHandle<T, vtkm::cont::StorageTagCounting>>
|
2019-02-28 18:15:27 +00:00
|
|
|
: SerializableTypeString<vtkm::cont::ArrayHandleCounting<T>>
|
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 <typename T>
|
|
|
|
struct Serialization<vtkm::cont::ArrayHandleCounting<T>>
|
|
|
|
{
|
|
|
|
private:
|
|
|
|
using Type = vtkm::cont::ArrayHandleCounting<T>;
|
|
|
|
using BaseType = vtkm::cont::ArrayHandle<typename Type::ValueType, typename Type::StorageTag>;
|
|
|
|
|
|
|
|
public:
|
|
|
|
static VTKM_CONT void save(BinaryBuffer& bb, const BaseType& obj)
|
|
|
|
{
|
2020-01-28 19:14:32 +00:00
|
|
|
auto portal = obj.ReadPortal();
|
2019-02-05 16:02:29 +00:00
|
|
|
vtkmdiy::save(bb, portal.GetStart());
|
|
|
|
vtkmdiy::save(bb, portal.GetStep());
|
|
|
|
vtkmdiy::save(bb, portal.GetNumberOfValues());
|
2018-06-08 15:56:40 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static VTKM_CONT void load(BinaryBuffer& bb, BaseType& obj)
|
|
|
|
{
|
|
|
|
T start{}, step{};
|
|
|
|
vtkm::Id count = 0;
|
|
|
|
|
2019-02-05 16:02:29 +00:00
|
|
|
vtkmdiy::load(bb, start);
|
|
|
|
vtkmdiy::load(bb, step);
|
|
|
|
vtkmdiy::load(bb, count);
|
2018-06-08 15:56:40 +00:00
|
|
|
|
|
|
|
obj = vtkm::cont::make_ArrayHandleCounting(start, step, count);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
template <typename T>
|
2019-12-30 21:23:57 +00:00
|
|
|
struct Serialization<vtkm::cont::ArrayHandle<T, vtkm::cont::StorageTagCounting>>
|
2018-06-08 15:56:40 +00:00
|
|
|
: Serialization<vtkm::cont::ArrayHandleCounting<T>>
|
|
|
|
{
|
|
|
|
};
|
|
|
|
} // diy
|
2019-09-12 21:19:36 +00:00
|
|
|
/// @endcond SERIALIZATION
|
2018-06-08 15:56:40 +00:00
|
|
|
|
2014-06-11 16:43:36 +00:00
|
|
|
#endif //vtk_m_cont_ArrayHandleCounting_h
|