fdaccc22db
Change the VTKM_CONT_EXPORT to VTKM_CONT. (Likewise for EXEC and EXEC_CONT.) Remove the inline from these macros so that they can be applied to everything, including implementations in a library. Because inline is not declared in these modifies, you have to add the keyword to functions and methods where the implementation is not inlined in the class.
156 lines
4.8 KiB
C++
156 lines
4.8 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.
|
|
//
|
|
// Copyright 2014 Sandia Corporation.
|
|
// Copyright 2014 UT-Battelle, LLC.
|
|
// Copyright 2014 Los Alamos National Security.
|
|
//
|
|
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
|
|
// the U.S. Government retains certain rights in this software.
|
|
//
|
|
// Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
|
|
// Laboratory (LANL), the U.S. Government retains certain rights in
|
|
// this software.
|
|
//============================================================================
|
|
#ifndef vtk_m_cont_cuda_internal_MakeThrustIterator_h
|
|
#define vtk_m_cont_cuda_internal_MakeThrustIterator_h
|
|
|
|
#include <vtkm/Types.h>
|
|
#include <vtkm/Pair.h>
|
|
#include <vtkm/internal/ExportMacros.h>
|
|
#include <vtkm/cont/ArrayPortalToIterators.h>
|
|
|
|
#include <vtkm/exec/cuda/internal/ArrayPortalFromThrust.h>
|
|
#include <vtkm/exec/cuda/internal/WrappedOperators.h>
|
|
|
|
// Disable warnings we check vtkm for but Thrust does not.
|
|
VTKM_THIRDPARTY_PRE_INCLUDE
|
|
#include <thrust/system/cuda/memory.h>
|
|
#include <thrust/functional.h>
|
|
#include <thrust/iterator/counting_iterator.h>
|
|
#include <thrust/iterator/transform_iterator.h>
|
|
VTKM_THIRDPARTY_POST_INCLUDE
|
|
|
|
namespace vtkm {
|
|
namespace cont {
|
|
namespace cuda {
|
|
namespace internal {
|
|
namespace detail {
|
|
|
|
// Tags to specify what type of thrust iterator to use.
|
|
struct ThrustIteratorFromArrayPortalTag { };
|
|
struct ThrustIteratorDevicePtrTag { };
|
|
|
|
// Traits to help classify what thrust iterators will be used.
|
|
template<typename IteratorType>
|
|
struct ThrustIteratorTag {
|
|
typedef ThrustIteratorFromArrayPortalTag Type;
|
|
};
|
|
template<typename T>
|
|
struct ThrustIteratorTag< thrust::system::cuda::pointer<T> > {
|
|
typedef ThrustIteratorDevicePtrTag Type;
|
|
};
|
|
template<typename T>
|
|
struct ThrustIteratorTag< thrust::system::cuda::pointer<const T> > {
|
|
typedef ThrustIteratorDevicePtrTag Type;
|
|
};
|
|
|
|
template<typename PortalType, typename Tag> struct IteratorChooser;
|
|
template<typename PortalType>
|
|
struct IteratorChooser<PortalType, detail::ThrustIteratorFromArrayPortalTag> {
|
|
typedef vtkm::exec::cuda::internal::IteratorFromArrayPortal<PortalType> Type;
|
|
};
|
|
template<typename PortalType>
|
|
struct IteratorChooser<PortalType, detail::ThrustIteratorDevicePtrTag> {
|
|
typedef vtkm::cont::ArrayPortalToIterators<PortalType> PortalToIteratorType;
|
|
|
|
typedef typename PortalToIteratorType::IteratorType Type;
|
|
|
|
};
|
|
|
|
template<typename PortalType>
|
|
struct IteratorTraits
|
|
{
|
|
typedef vtkm::cont::ArrayPortalToIterators<PortalType> PortalToIteratorType;
|
|
typedef typename detail::ThrustIteratorTag<
|
|
typename PortalToIteratorType::IteratorType>::Type Tag;
|
|
typedef typename IteratorChooser<
|
|
PortalType,
|
|
Tag
|
|
>::Type IteratorType;
|
|
};
|
|
|
|
template<typename PortalType>
|
|
VTKM_CONT
|
|
typename IteratorTraits<PortalType>::IteratorType
|
|
MakeIteratorBegin(PortalType portal, detail::ThrustIteratorFromArrayPortalTag)
|
|
{
|
|
return vtkm::exec::cuda::internal::IteratorFromArrayPortal<PortalType>(portal);
|
|
}
|
|
|
|
template<typename PortalType>
|
|
VTKM_CONT
|
|
typename IteratorTraits<PortalType>::IteratorType
|
|
MakeIteratorBegin(PortalType portal, detail::ThrustIteratorDevicePtrTag)
|
|
{
|
|
vtkm::cont::ArrayPortalToIterators<PortalType> iterators(portal);
|
|
return iterators.GetBegin();
|
|
}
|
|
|
|
template<typename PortalType>
|
|
VTKM_CONT
|
|
typename IteratorTraits<PortalType>::IteratorType
|
|
MakeIteratorEnd(PortalType portal, detail::ThrustIteratorFromArrayPortalTag)
|
|
{
|
|
vtkm::exec::cuda::internal::IteratorFromArrayPortal<PortalType>
|
|
iterator(portal);
|
|
::thrust::advance(iterator,
|
|
static_cast<std::size_t>(portal.GetNumberOfValues()));
|
|
return iterator;
|
|
}
|
|
|
|
template<typename PortalType>
|
|
VTKM_CONT
|
|
typename IteratorTraits<PortalType>::IteratorType
|
|
MakeIteratorEnd(PortalType portal, detail::ThrustIteratorDevicePtrTag)
|
|
{
|
|
vtkm::cont::ArrayPortalToIterators<PortalType> iterators(portal);
|
|
return iterators.GetEnd();
|
|
}
|
|
|
|
} // namespace detail
|
|
|
|
|
|
|
|
template<typename PortalType>
|
|
VTKM_CONT
|
|
typename detail::IteratorTraits<PortalType>::IteratorType
|
|
IteratorBegin(PortalType portal)
|
|
{
|
|
typedef typename detail::IteratorTraits<PortalType>::Tag IteratorTag;
|
|
return detail::MakeIteratorBegin(portal, IteratorTag());
|
|
}
|
|
|
|
template<typename PortalType>
|
|
VTKM_CONT
|
|
typename detail::IteratorTraits<PortalType>::IteratorType
|
|
IteratorEnd(PortalType portal)
|
|
{
|
|
typedef typename detail::IteratorTraits<PortalType>::Tag IteratorTag;
|
|
return detail::MakeIteratorEnd(portal, IteratorTag());
|
|
}
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
} //namespace vtkm::cont::cuda::internal
|
|
|
|
#endif
|