table.hpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2019-2023, NVIDIA CORPORATION.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 #pragma once
17 
18 #include <cudf/column/column.hpp>
20 #include <cudf/utilities/default_stream.hpp>
21 
22 #include <rmm/cuda_stream_view.hpp>
24 
25 #include <memory>
26 #include <vector>
27 
33 namespace cudf {
34 
40 class table {
41  public:
42  table() = default;
43  ~table() = default;
44  table(table&&) = default;
45  table& operator=(table const&) = delete;
46  table& operator=(table&&) = delete;
47 
53  table(table const& other);
54 
62  table(std::vector<std::unique_ptr<column>>&& columns);
63 
74 
80  [[nodiscard]] size_type num_columns() const noexcept { return _columns.size(); }
81 
87  [[nodiscard]] size_type num_rows() const noexcept { return _num_rows; }
88 
95  [[nodiscard]] table_view view() const;
96 
101  operator table_view() const { return this->view(); };
102 
110 
115  operator mutable_table_view() { return this->mutable_view(); };
116 
125  std::vector<std::unique_ptr<column>> release();
126 
139  template <typename InputIterator>
140  table_view select(InputIterator begin, InputIterator end) const
141  {
142  std::vector<column_view> columns(std::distance(begin, end));
144  begin, end, columns.begin(), [this](auto index) { return _columns.at(index)->view(); });
145  return table_view(columns);
146  }
147 
158  [[nodiscard]] table_view select(std::vector<cudf::size_type> const& column_indices) const
159  {
160  return select(column_indices.begin(), column_indices.end());
161  };
162 
172  column& get_column(cudf::size_type column_index) { return *(_columns.at(column_index)); }
173 
183  [[nodiscard]] column const& get_column(cudf::size_type i) const { return *(_columns.at(i)); }
184 
185  private:
186  std::vector<std::unique_ptr<column>> _columns{};
187  size_type _num_rows{};
188 };
189 
190 } // namespace cudf
A container of nullable device data as a column of elements.
Definition: column.hpp:48
A set of mutable_column_views of the same size.
Definition: table_view.hpp:255
A set of cudf::column_view's of the same size.
Definition: table_view.hpp:187
A set of cudf::column's of the same size.
Definition: table.hpp:40
mutable_table_view mutable_view()
Returns a mutable, non-owning mutable_table_view of the contents of this table.
std::vector< std::unique_ptr< column > > release()
Releases ownership of the columns by returning a vector of unique_ptrs to the constituent columns.
table_view select(std::vector< cudf::size_type > const &column_indices) const
Returns a table_view with set of specified columns.
Definition: table.hpp:158
column const & get_column(cudf::size_type i) const
Returns a const reference to the specified column.
Definition: table.hpp:183
table_view view() const
Returns an immutable, non-owning table_view of the contents of this table.
table(std::vector< std::unique_ptr< column >> &&columns)
Moves the contents from a vector of unique_ptrs to columns to construct a new table.
size_type num_rows() const noexcept
Returns the number of rows.
Definition: table.hpp:87
table(table &&)=default
Move constructor.
table_view select(InputIterator begin, InputIterator end) const
Returns a table_view built from a range of column indices.
Definition: table.hpp:140
table(table const &other)
Construct a new table by copying the contents of another table.
table(table_view view, rmm::cuda_stream_view stream=cudf::get_default_stream(), rmm::mr::device_memory_resource *mr=rmm::mr::get_current_device_resource())
Copy the contents of a table_view to construct a new table.
size_type num_columns() const noexcept
Returns the number of columns in the table.
Definition: table.hpp:80
column & get_column(cudf::size_type column_index)
Returns a reference to the specified column.
Definition: table.hpp:172
Class definition for cudf::column.
device_memory_resource * get_current_device_resource()
std::unique_ptr< column > transform(column_view const &input, std::string const &unary_udf, data_type output_type, bool is_ptx, rmm::mr::device_memory_resource *mr=rmm::mr::get_current_device_resource())
Creates a new column by applying a unary function against every element of an input column.
int32_t size_type
Row index type for columns and tables.
Definition: types.hpp:80
size_type distance(T f, T l)
Similar to std::distance but returns cudf::size_type and performs static_cast
Definition: types.hpp:94
cuDF interfaces
Definition: aggregation.hpp:34
rmm::cuda_stream_view const get_default_stream()
Get the current default stream.
Class definitions for (mutable)_table_view