variant.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <cudf/column/column.hpp>
#include <cudf/column/column_view.hpp>
#include <cudf/io/experimental/variant_spec.hpp>
#include <cudf/types.hpp>
#include <cudf/utilities/default_stream.hpp>
#include <cudf/utilities/memory_resource.hpp>
 
#include <cuda/stream>
 
#include <memory>
#include <optional>
#include <string_view>
 
namespace CUDF_EXPORT cudf {
namespace io::parquet::experimental {
[[nodiscard]] std::unique_ptr<column> get_variant_field(
  column_view const& variant_column,
  std::string_view path,
  std::optional<mutable_column_view> status = std::nullopt,
  cuda::stream_ref stream                   = cudf::get_default_stream(),
  rmm::device_async_resource_ref mr         = cudf::get_current_device_resource_ref());
 
[[nodiscard]] std::unique_ptr<column> cast_variant(
  column_view const& values,
  data_type desired_type,
  std::optional<mutable_column_view> status = std::nullopt,
  cuda::stream_ref stream                   = cudf::get_default_stream(),
  rmm::device_async_resource_ref mr         = cudf::get_current_device_resource_ref());
 
[[nodiscard]] std::unique_ptr<column> extract_variant_field(
  column_view const& variant_column,
  std::string_view path,
  data_type desired_type,
  std::optional<mutable_column_view> status = std::nullopt,
  cuda::stream_ref stream                   = cudf::get_default_stream(),
  rmm::device_async_resource_ref mr         = cudf::get_current_device_resource_ref());
 
[[nodiscard]] std::unique_ptr<column> get_variant_type_id(
  column_view const& values,
  cuda::stream_ref stream           = cudf::get_default_stream(),
  rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
 
}  // namespace io::parquet::experimental
}  // namespace CUDF_EXPORT cudf