148 lines
4.1 KiB
C++
148 lines
4.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.
|
|
//============================================================================
|
|
#ifndef vtk_m_cont_ArrayHandleExtrudeField_h
|
|
#define vtk_m_cont_ArrayHandleExtrudeField_h
|
|
|
|
#include <vtkm/cont/ArrayHandle.h>
|
|
#include <vtkm/cont/StorageExtrude.h>
|
|
|
|
namespace vtkm
|
|
{
|
|
namespace cont
|
|
{
|
|
|
|
template <typename T>
|
|
class VTKM_ALWAYS_EXPORT ArrayHandleExtrudeField
|
|
: public vtkm::cont::ArrayHandle<T, internal::StorageTagExtrude>
|
|
{
|
|
using StorageType = vtkm::cont::internal::Storage<T, internal::StorageTagExtrude>;
|
|
|
|
public:
|
|
VTKM_ARRAY_HANDLE_SUBCLASS(ArrayHandleExtrudeField,
|
|
(ArrayHandleExtrudeField<T>),
|
|
(vtkm::cont::ArrayHandle<T, internal::StorageTagExtrude>));
|
|
|
|
ArrayHandleExtrudeField(const StorageType& storage)
|
|
: Superclass(storage)
|
|
{
|
|
}
|
|
|
|
vtkm::Int32 GetNumberOfValuesPerPlane() const
|
|
{
|
|
return this->GetStorage().GetNumberOfValuesPerPlane();
|
|
}
|
|
|
|
vtkm::Int32 GetNumberOfPlanes() const { return this->GetStorage().GetNumberOfPlanes(); }
|
|
bool GetUseCylindrical() const { return this->GetStorage().GetUseCylindrical(); }
|
|
const vtkm::cont::ArrayHandle<T>& GetArray() const { return this->GetStorage().Array; }
|
|
};
|
|
|
|
template <typename T>
|
|
vtkm::cont::ArrayHandleExtrudeField<T> make_ArrayHandleExtrudeField(
|
|
const vtkm::cont::ArrayHandle<T>& array,
|
|
vtkm::Int32 numberOfPlanes,
|
|
bool cylindrical)
|
|
{
|
|
using StorageType = vtkm::cont::internal::Storage<T, internal::StorageTagExtrude>;
|
|
auto storage = StorageType{ array, numberOfPlanes, cylindrical };
|
|
return ArrayHandleExtrudeField<T>(storage);
|
|
}
|
|
|
|
template <typename T>
|
|
vtkm::cont::ArrayHandleExtrudeField<T> make_ArrayHandleExtrudeField(
|
|
const T* array,
|
|
vtkm::Id length,
|
|
vtkm::Int32 numberOfPlanes,
|
|
bool cylindrical,
|
|
vtkm::CopyFlag copy = vtkm::CopyFlag::Off)
|
|
{
|
|
using StorageType = vtkm::cont::internal::Storage<T, internal::StorageTagExtrude>;
|
|
auto storage =
|
|
StorageType(vtkm::cont::make_ArrayHandle(array, length, copy), numberOfPlanes, cylindrical);
|
|
return ArrayHandleExtrudeField<T>(storage);
|
|
}
|
|
|
|
template <typename T>
|
|
vtkm::cont::ArrayHandleExtrudeField<T> make_ArrayHandleExtrudeField(
|
|
const std::vector<T>& array,
|
|
vtkm::Int32 numberOfPlanes,
|
|
bool cylindrical,
|
|
vtkm::CopyFlag copy = vtkm::CopyFlag::Off)
|
|
{
|
|
if (!array.empty())
|
|
{
|
|
return make_ArrayHandleExtrudeField(
|
|
array.data(), static_cast<vtkm::Id>(array.size()), numberOfPlanes, cylindrical, copy);
|
|
}
|
|
else
|
|
{
|
|
// Vector empty. Just return an empty array handle.
|
|
return ArrayHandleExtrudeField<T>();
|
|
}
|
|
}
|
|
}
|
|
} // vtkm::cont
|
|
|
|
//=============================================================================
|
|
// Specializations of serialization related classes
|
|
/// @cond SERIALIZATION
|
|
namespace vtkm
|
|
{
|
|
namespace cont
|
|
{
|
|
|
|
template <typename T>
|
|
struct SerializableTypeString<vtkm::cont::ArrayHandleExtrudeField<T>>
|
|
{
|
|
static VTKM_CONT const std::string& Get()
|
|
{
|
|
static std::string name = "AH_ExtrudeField<" + SerializableTypeString<T>::Get() + ">";
|
|
return name;
|
|
}
|
|
};
|
|
}
|
|
} // vtkm::cont
|
|
|
|
namespace mangled_diy_namespace
|
|
{
|
|
|
|
template <typename T>
|
|
struct Serialization<vtkm::cont::ArrayHandleExtrudeField<T>>
|
|
{
|
|
private:
|
|
using Type = vtkm::cont::ArrayHandleExtrudeField<T>;
|
|
|
|
public:
|
|
static VTKM_CONT void save(BinaryBuffer& bb, const Type& ah)
|
|
{
|
|
vtkmdiy::save(bb, ah.GetNumberOfPlanes());
|
|
vtkmdiy::save(bb, ah.GetUseCylindrical());
|
|
vtkmdiy::save(bb, ah.GetArray());
|
|
}
|
|
|
|
static VTKM_CONT void load(BinaryBuffer& bb, Type& ah)
|
|
{
|
|
vtkm::Int32 numberOfPlanes;
|
|
bool isCylindrical;
|
|
vtkm::cont::ArrayHandle<T> array;
|
|
|
|
vtkmdiy::load(bb, numberOfPlanes);
|
|
vtkmdiy::load(bb, isCylindrical);
|
|
vtkmdiy::load(bb, array);
|
|
|
|
ah = vtkm::cont::make_ArrayHandleExtrudeField(array, numberOfPlanes, isCylindrical);
|
|
}
|
|
};
|
|
|
|
} // diy
|
|
/// @endcond SERIALIZATION
|
|
|
|
#endif
|