lists_column_view.hpp

转到此文件的文档。

/*
 * 版权所有 (c) 2020-2024, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版（“许可证”）获得许可；
 * 除非遵守许可证，否则您不得使用此文件。
 * 您可以在以下网址获取许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则根据许可证分发的软件按“原样”分发，
 * 不附带任何明示或默示的保证或条件。
 * 请参阅许可证以了解特定的语言权限和限制。
 *
 */

#pragma once

#include <cudf/column/column.hpp>
#include <cudf/column/column_view.hpp>
#include <cudf/utilities/export.hpp>

#include <rmm/cuda_stream_view.hpp>

namespace CUDF_EXPORT cudf {

class lists_column_view : private column_view {
 public
 lists_column_view() = default;
 lists_column_view(column_view const& lists_column);
 lists_column_view(lists_column_view&&) = default;
 lists_column_view(lists_column_view const&) = default;
 ~lists_column_view() override = default;
 lists_column_view& operator=(lists_column_view const&) = default;
 lists_column_view& operator=(lists_column_view&&) = default;

 static constexpr size_type offsets_column_index{0};
 static constexpr size_type child_column_index{1};

 using column_view::child_begin;
 using column_view::child_end;
 using column_view::has_nulls;
 using column_view::is_empty;
 using column_view::null_count;
 using column_view::null_mask;
 using column_view::offset;
 using column_view::size;
 using offset_iterator = size_type const*;

 [[nodiscard]] column_view parent() const;

 [[nodiscard]] column_view offsets() const;

 [[nodiscard]] column_view child() const;

 [[nodiscard]] column_view get_sliced_child(rmm::cuda_stream_view stream) const;

 [[nodiscard]] offset_iterator offsets_begin() const noexcept
 {
 return offsets().begin<size_type>() + offset();
 }

 [[nodiscard]] offset_iterator offsets_end() const noexcept
 {
 return offsets_begin() + size() + 1;
 }
}; // end of group
} // namespace CUDF_EXPORT cudf