structs_column_device_view.cuh

/*
 * Copyright (c) 2022-2024, NVIDIA CORPORATION.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#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