5#ifndef GKO_PUBLIC_CORE_BASE_MATRIX_ASSEMBLY_DATA_HPP_
6#define GKO_PUBLIC_CORE_BASE_MATRIX_ASSEMBLY_DATA_HPP_
13#include <unordered_map>
16#include <ginkgo/core/base/dim.hpp>
17#include <ginkgo/core/base/math.hpp>
18#include <ginkgo/core/base/matrix_data.hpp>
19#include <ginkgo/core/base/types.hpp>
20#include <ginkgo/core/base/utils.hpp>
27template <
typename IndexType>
28struct symbolic_nonzero_hash {
29 symbolic_nonzero_hash() =
default;
31 explicit symbolic_nonzero_hash(
size_type num_cols) noexcept
35 std::size_t operator()(std::pair<IndexType, IndexType> nnz)
const noexcept
37 return static_cast<std::size_t
>(nnz.first) * num_cols_ + nnz.second;
59template <
typename ValueType = default_precision,
typename IndexType =
int32>
62 using value_type = ValueType;
63 using index_type = IndexType;
67 nonzeros_(0, detail::symbolic_nonzero_hash<index_type>(size_[1]))
79 void add_value(index_type row, index_type col, value_type val)
81 auto ind = std::make_pair(row, col);
82 nonzeros_[ind] += val;
93 void set_value(index_type row, index_type col, value_type val)
95 auto ind = std::make_pair(row, col);
108 const auto it = nonzeros_.find(std::make_pair(row, col));
109 if (it == nonzeros_.end()) {
125 return nonzeros_.find(std::make_pair(row, col)) != nonzeros_.end();
134 return nonzeros_.size();
144 using nonzero_type =
typename output_type::nonzero_type;
146 std::pair<std::pair<index_type, index_type>, value_type>;
147 output_type data{size_};
148 data.
nonzeros.reserve(nonzeros_.size());
149 std::transform(nonzeros_.begin(), nonzeros_.end(),
150 std::back_inserter(data.nonzeros), [](entry_type entry) {
151 return nonzero_type{entry.first.first,
155 data.sort_row_major();
172 std::unordered_map<std::pair<index_type, index_type>, value_type,
173 detail::symbolic_nonzero_hash<index_type>>
This structure is used as an intermediate type to assemble a sparse matrix.
Definition matrix_assembly_data.hpp:60
void add_value(index_type row, index_type col, value_type val)
Sets the matrix value at (row, col).
Definition matrix_assembly_data.hpp:79
void set_value(index_type row, index_type col, value_type val)
Sets the matrix value at (row, col).
Definition matrix_assembly_data.hpp:93
size_type get_num_stored_elements() const noexcept
Definition matrix_assembly_data.hpp:132
bool contains(index_type row, index_type col)
Returns true iff the matrix contains an entry at (row, col).
Definition matrix_assembly_data.hpp:123
value_type get_value(index_type row, index_type col)
Gets the matrix value at (row, col).
Definition matrix_assembly_data.hpp:106
dim< 2 > get_size() const noexcept
Definition matrix_assembly_data.hpp:129
matrix_data< ValueType, IndexType > get_ordered_data() const
Definition matrix_assembly_data.hpp:141
The Ginkgo namespace.
Definition abstract_factory.hpp:20
constexpr T zero()
Returns the additive identity for T.
Definition math.hpp:747
std::size_t size_type
Integral type used for allocation quantities.
Definition types.hpp:86
A type representing the dimensions of a multidimensional object.
Definition dim.hpp:27
This structure is used as an intermediate data type to store a sparse matrix.
Definition matrix_data.hpp:127
std::vector< nonzero_type > nonzeros
A vector of tuples storing the non-zeros of the matrix.
Definition matrix_data.hpp:454