table.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2019-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 */
#pragma once
 
#include <cudf/column/column.hpp>
#include <cudf/table/table_view.hpp>
#include <cudf/utilities/default_stream.hpp>
#include <cudf/utilities/memory_resource.hpp>
 
#include <rmm/cuda_stream_view.hpp>
 
#include <memory>
#include <vector>
 
namespace CUDF_EXPORT cudf {
 
class table {
 public:
  table()                        = default;
  ~table()                       = default;
  table(table&&)                 = default;  
  table& operator=(table const&) = delete;
  table& operator=(table&&)      = delete;
 
  explicit table(table const& other,
                 rmm::cuda_stream_view stream      = cudf::get_default_stream(),
                 rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
  table(std::vector<std::unique_ptr<column>>&& columns);
 
  table(std::vector<std::unique_ptr<column>>&& columns, size_type num_rows);
 
  table(table_view view,
        rmm::cuda_stream_view stream      = cudf::get_default_stream(),
        rmm::device_async_resource_ref mr = cudf::get_current_device_resource_ref());
 
  [[nodiscard]] size_type num_columns() const noexcept { return _columns.size(); }
 
  [[nodiscard]] size_type num_rows() const noexcept { return _num_rows; }
 
  [[nodiscard]] std::size_t alloc_size() const;
 
  [[nodiscard]] table_view view() const;
 
  operator table_view() const { return this->view(); };
 
  mutable_table_view mutable_view();
 
  operator mutable_table_view() { return this->mutable_view(); };
 
  std::vector<std::unique_ptr<column>> release();
 
  template <typename InputIterator>
  [[nodiscard]] table_view select(InputIterator begin, InputIterator end) const
  {
    std::vector<column_view> columns(std::distance(begin, end));
    std::transform(
      begin, end, columns.begin(), [this](auto index) { return _columns.at(index)->view(); });
    return table_view{columns, num_rows()};
  }
 
  [[nodiscard]] table_view select(std::vector<cudf::size_type> const& column_indices) const
  {
    return select(column_indices.begin(), column_indices.end());
  };
 
  column& get_column(cudf::size_type column_index) { return *(_columns.at(column_index)); }
 
  [[nodiscard]] column const& get_column(cudf::size_type i) const { return *(_columns.at(i)); }
 
 private:
  std::vector<std::unique_ptr<column>> _columns{};
  size_type _num_rows{};
};
 
}  // namespace CUDF_EXPORT cudf