structs_column_device_view.cuh

/*
 * SPDX-FileCopyrightText: Copyright (c) 2022-2024, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
#pragma once
 
#include <cudf/column/column_device_view.cuh>
#include <cudf/types.hpp>
 
namespace CUDF_EXPORT cudf {
 
namespace detail {
 
class structs_column_device_view : private column_device_view {
 public:
  structs_column_device_view()                                  = delete;
  ~structs_column_device_view()                                 = default;
  structs_column_device_view(structs_column_device_view const&) = default;  
  structs_column_device_view(structs_column_device_view&&)      = default;  
  structs_column_device_view& operator=(structs_column_device_view const&) = default;
 
  structs_column_device_view& operator=(structs_column_device_view&&) = default;
 
  CUDF_HOST_DEVICE structs_column_device_view(column_device_view const& underlying_)
    : column_device_view(underlying_)
  {
#ifdef __CUDA_ARCH__
    cudf_assert(underlying_.type().id() == type_id::STRUCT and
                "structs_column_device_view only supports structs");
#else
    CUDF_EXPECTS(underlying_.type().id() == type_id::STRUCT,
                 "structs_column_device_view only supports structs");
#endif
  }
 
  using column_device_view::child;
  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 get_sliced_child(size_type idx) const
  {
    return child(idx).slice(offset(), size());
  }
};
 
}  // namespace detail
 
}  // namespace CUDF_EXPORT cudf