lists_column_device_view.cuh

/*
 * 版权所有 (c) 2020-2024, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版 (以下简称“许可证”) 获得许可；
 * 除非符合许可证的规定，否则您不得使用此文件。
 * 您可在以下网址获取许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则根据许可证分发的软件
 * 是基于“现状”分发的，
 * 不附带任何形式的明示或默示担保或条件。
 * 请参见许可证以了解具体管理权限和
 * 限制。
 */
#pragma once

#include <cudf/column/column_device_view.cuh>
#include <cudf/lists/lists_column_view.hpp>
#include <cudf/types.hpp>

#include <cuda_runtime.h>

namespace cudf::detail {

class lists_column_device_view : private column_device_view {
 public
 lists_column_device_view() = delete;
 ~lists_column_device_view() = default;
 lists_column_device_view(lists_column_device_view const&) = default;
 lists_column_device_view(lists_column_device_view&&) = default;
 lists_column_device_view& operator=(lists_column_device_view const&) = default;
 lists_column_device_view& operator=(lists_column_device_view&&) = default;

 CUDF_HOST_DEVICE lists_column_device_view(column_device_view const& underlying_)
 : column_device_view(underlying_)
 {
#ifdef __CUDA_ARCH__
 cudf_assert(underlying_.type().id() == type_id::LIST and
 "lists_column_device_view 仅支持列表");
#else
 CUDF_EXPECTS(underlying_.type().id() == type_id::LIST,
 "lists_column_device_view 仅支持列表");
#endif
 }

 using column_device_view::is_null;
 using column_device_view::nullable;
 using column_device_view::offset;
 using column_device_view::size;

 [[nodiscard]] __device__ inline column_device_view offsets() const
 {
 return column_device_view::child(lists_column_view::offsets_column_index);
 }

 [[nodiscard]] __device__ inline size_type offset_at(size_type idx) const
 {
 return offsets().size() > 0 ? offsets().element<size_type>(offset() + idx) : 0;
 }

 [[nodiscard]] __device__ inline column_device_view child() const
 {
 return column_device_view::child(lists_column_view::child_column_index);
 }

 [[nodiscard]] __device__ inline column_device_view get_sliced_child() const
 {
 auto start = offset_at(0);
 auto end = offset_at(size());
 return child().slice(start, end - start);
 }
};

} // 命名空间 cudf::detail