range.cuh

前往此文件的文档。

/*
 * 版权所有 (c) 2022, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证版本 2.0 (以下简称“许可证”) 获得许可；
 * 除非符合许可证的规定，否则您不得使用此文件。
 * 您可以在以下位置获得许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则软件
 * 根据许可证分发，是按“原样”分发的，
 * 不附带任何形式的明示或暗示保证或条件。
 * 有关管理权限和
 * 限制的特定语言，请参阅许可证。
 */
 
#pragma once
 
#include <cuspatial/cuda_utils.hpp>
#include <cuspatial/traits.hpp>
 
#include <thrust/detail/raw_reference_cast.h>
#include <thrust/distance.h>
 
namespace cuspatial {

template <typename IteratorType>
class range {
 public
 using value_type = iterator_value_type<IteratorType>;
 range(IteratorType begin, IteratorType end) : _begin(begin), _end(end) {}

 auto CUSPATIAL_HOST_DEVICE begin() { return _begin; }
 auto CUSPATIAL_HOST_DEVICE end() { return _end; }
 auto CUSPATIAL_HOST_DEVICE size() { return thrust::distance(_begin, _end); }

 template <typename IndexType>
 auto& CUSPATIAL_HOST_DEVICE operator[](IndexType i)
  {
 return thrust::raw_reference_cast(_begin[i]);
  }
 
 private
 IteratorType _begin;
 IteratorType _end;
};
 
} // namespace cuspatial