5#ifndef GKO_PUBLIC_CORE_BASE_INDEX_SET_HPP_
6#define GKO_PUBLIC_CORE_BASE_INDEX_SET_HPP_
10#include <initializer_list>
15#include <ginkgo/core/base/array.hpp>
16#include <ginkgo/core/base/exception_helpers.hpp>
17#include <ginkgo/core/base/executor.hpp>
18#include <ginkgo/core/base/types.hpp>
19#include <ginkgo/core/base/utils.hpp>
56template <
typename IndexType =
int32>
69 explicit index_set(std::shared_ptr<const Executor> exec) noexcept
70 : exec_(std::move(exec)),
72 num_stored_indices_{0},
73 subsets_begin_{array<index_type>(exec_)},
74 subsets_end_{array<index_type>(exec_)},
75 superset_cumulative_indices_{array<index_type>(exec_)}
87 explicit index_set(std::shared_ptr<const gko::Executor> exec,
88 std::initializer_list<IndexType> init_list,
89 const bool is_sorted =
false)
90 : exec_(std::move(exec)),
91 index_space_size_(init_list.size() > 0
92 ? *(std::max_element(std::begin(init_list),
93 std::end(init_list))) +
96 num_stored_indices_{static_cast<IndexType>(init_list.size())}
98 GKO_ASSERT(index_space_size_ > 0);
99 this->populate_subsets(
113 explicit index_set(std::shared_ptr<const gko::Executor> exec,
116 const bool is_sorted =
false)
117 : exec_(std::move(exec)), index_space_size_(size)
119 GKO_ASSERT(index_space_size_ >= indices.
get_size());
120 this->populate_subsets(indices, is_sorted);
152 *
this = std::move(other);
176 if (&other ==
this) {
179 this->index_space_size_ = other.index_space_size_;
180 this->num_stored_indices_ = other.num_stored_indices_;
181 this->subsets_begin_ = other.subsets_begin_;
182 this->subsets_end_ = other.subsets_end_;
183 this->superset_cumulative_indices_ = other.superset_cumulative_indices_;
200 if (&other ==
this) {
203 this->index_space_size_ = std::exchange(other.index_space_size_, 0);
204 this->num_stored_indices_ = std::exchange(other.num_stored_indices_, 0);
205 this->subsets_begin_ = std::move(other.subsets_begin_);
206 this->subsets_end_ = std::move(other.subsets_end_);
207 this->superset_cumulative_indices_ =
208 std::move(other.superset_cumulative_indices_);
222 this->index_space_size_ = 0;
223 this->num_stored_indices_ = 0;
224 this->subsets_begin_.
clear();
225 this->subsets_end_.
clear();
226 this->superset_cumulative_indices_.
clear();
234 std::shared_ptr<const Executor>
get_executor()
const {
return this->exec_; }
316 const bool is_sorted =
false)
const;
332 const bool is_sorted =
false)
const;
353 const bool is_sorted =
false)
const;
374 return this->subsets_begin_.
get_size();
411 const bool is_sorted);
413 std::shared_ptr<const Executor> exec_;
An array is a container which encapsulates fixed-sized arrays, stored on the Executor tied to the arr...
Definition logger.hpp:25
void clear() noexcept
Deallocates all data used by the array.
Definition array.hpp:605
const value_type * get_const_data() const noexcept
Returns a constant pointer to the block of memory used to store the elements of the array.
Definition array.hpp:683
size_type get_size() const noexcept
Returns the number of elements in the array.
Definition array.hpp:657
An index set class represents an ordered set of intervals.
Definition index_set.hpp:57
index_set(const index_set &other)
Creates a copy of the input index_set.
Definition index_set.hpp:140
index_set & operator=(index_set &&other)
Moves data from another index_set.
Definition index_set.hpp:198
array< index_type > to_global_indices() const
This function allows the user obtain a decompressed global_indices array from the indices stored in t...
const index_type * get_subsets_begin() const
Returns a pointer to the beginning indices of the subsets.
Definition index_set.hpp:382
index_type get_num_elems() const
Return the actual number of indices stored in the index set.
Definition index_set.hpp:255
index_type get_global_index(index_type local_index) const
Return the global index given a local index.
array< bool > contains(const array< index_type > &global_indices, const bool is_sorted=false) const
Checks if the individual global indeices exist in the index set.
IndexType index_type
The type of elements stored in the index set.
Definition index_set.hpp:62
const index_type * get_subsets_end() const
Returns a pointer to the end indices of the subsets.
Definition index_set.hpp:392
std::shared_ptr< const Executor > get_executor() const
Returns the executor of the index_set.
Definition index_set.hpp:234
void clear() noexcept
Deallocates all data used by the index_set.
Definition index_set.hpp:220
index_set(std::shared_ptr< const Executor > exec) noexcept
Creates an empty index_set tied to the specified Executor.
Definition index_set.hpp:69
index_type get_local_index(index_type global_index) const
Return the local index given a global index.
index_set(index_set &&other)
Moves the input index_set.
Definition index_set.hpp:160
array< index_type > map_global_to_local(const array< index_type > &global_indices, const bool is_sorted=false) const
This is an array version of the scalar function above.
index_set(std::shared_ptr< const gko::Executor > exec, std::initializer_list< IndexType > init_list, const bool is_sorted=false)
Creates an index set on the specified executor from the initializer list.
Definition index_set.hpp:87
index_type get_num_subsets() const
Returns the number of subsets stored in the index set.
Definition index_set.hpp:372
index_set(std::shared_ptr< const Executor > exec, index_set &&other)
Moves the input index_set to a different executor.
Definition index_set.hpp:149
bool contains(const index_type global_index) const
Checks if the global index exists in the index set.
index_set & operator=(const index_set &other)
Copies data from another index_set.
Definition index_set.hpp:174
bool is_contiguous() const
Returns if the index set is contiguous.
Definition index_set.hpp:248
index_set(std::shared_ptr< const gko::Executor > exec, const index_type size, const gko::array< index_type > &indices, const bool is_sorted=false)
Creates an index set on the specified executor and the given size.
Definition index_set.hpp:113
array< index_type > map_local_to_global(const array< index_type > &local_indices, const bool is_sorted=false) const
This is an array version of the scalar function above.
const index_type * get_superset_indices() const
Returns a pointer to the cumulative indices of the superset of the subsets.
Definition index_set.hpp:404
index_set(std::shared_ptr< const Executor > exec, const index_set &other)
Creates a copy of the input index_set on a different executor.
Definition index_set.hpp:129
index_type get_size() const
Returns the size of the index set space.
Definition index_set.hpp:241
The Ginkgo namespace.
Definition abstract_factory.hpp:20