blob: ba3e4aebea19d863d0d893fa9721714bc68461fa (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
#pragma once
#include <cstddef>
#include <initializer_list>
#include <vector>
namespace linalg {
class Matrix {
public:
Matrix() = default;
Matrix(std::size_t rows, std::size_t cols);
Matrix(std::size_t rows, std::size_t cols, double value);
Matrix(std::initializer_list<std::initializer_list<double>> values);
[[nodiscard]] std::size_t rows() const noexcept;
[[nodiscard]] std::size_t cols() const noexcept;
[[nodiscard]] bool empty() const noexcept;
double& operator()(std::size_t i, std::size_t j);
const double& operator()(std::size_t i, std::size_t j) const;
void fill(double value);
double* data() noexcept;
const double* data() const noexcept;
static Matrix identity(std::size_t n);
static Matrix zeros(std::size_t rows, std::size_t cols);
private:
[[nodiscard]] std::size_t index(std::size_t i, std::size_t j) const;
void check_bounds(std::size_t i, std::size_t j) const;
std::size_t rows_ = 0;
std::size_t cols_ = 0;
std::vector<double> data_;
};
} // namespace linalg
|