101 lines
2.2 KiB
C++
101 lines
2.2 KiB
C++
#ifndef TOOLS_MATRIX_H
|
|
#define TOOLS_MATRIX_H
|
|
|
|
// some functions for matrix interaction
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
/**
|
|
* Vector is a class to represent
|
|
* a column vector with n rows.
|
|
*/
|
|
|
|
|
|
class Vector {
|
|
protected:
|
|
// Number of rows in vector
|
|
int rows;
|
|
// Number of columns in vector
|
|
int columns;
|
|
// Matrix representation as vector of vectors of integers
|
|
std::vector<double> vector;
|
|
public:
|
|
Vector(int n);
|
|
|
|
Vector(const Vector& other);
|
|
|
|
Vector(std::initializer_list<double>);
|
|
|
|
/* Getter for the number of rows */
|
|
int size() const;
|
|
|
|
double& operator[](int row);
|
|
|
|
Vector& operator=(const Vector& other);
|
|
|
|
Vector operator+(Vector& other);
|
|
|
|
Vector operator-(Vector& other);
|
|
|
|
/* Input operator reads element of vector */
|
|
friend std::istream& operator>>(std::istream& cin, Vector& vectorObj);
|
|
|
|
/* Output operator prints elements of the vector
|
|
* in a row separated with a space (no space at the end of the line)
|
|
*/
|
|
friend std::ostream& operator<<(std::ostream& cout, Vector& vectorObj);
|
|
};
|
|
|
|
/**
|
|
* Class Matrix represents
|
|
* a matrix of size n x m
|
|
* of type integer.
|
|
*/
|
|
class Matrix {
|
|
protected:
|
|
// Number of rows in matrix
|
|
int rows;
|
|
// Number of columns in matrix
|
|
int columns;
|
|
// Matrix representation as vector of vectors of integers
|
|
std::vector<Vector> matrix;
|
|
public:
|
|
Matrix(int n, int m);
|
|
|
|
Matrix(const Matrix& other);
|
|
|
|
Matrix(std::initializer_list<std::vector<double>>);
|
|
|
|
/* Getter for the number of rows */
|
|
int getRows() const;
|
|
|
|
/* Getter for the number of columns */
|
|
int getColumns() const;
|
|
|
|
Vector& operator[](int row);
|
|
|
|
Matrix& operator=(const Matrix& other);
|
|
|
|
Matrix operator+(Matrix& other) const;
|
|
|
|
Matrix operator-(Matrix& other) const;
|
|
|
|
Matrix operator*(Matrix& other) const;
|
|
|
|
/* Matrix-Vector multiplication */
|
|
Vector operator*(Vector other) const;
|
|
|
|
/* Produces transposed version of the matrix */
|
|
Matrix transpose() const;
|
|
|
|
/* Input operator reads element of matrix row by row */
|
|
friend std::istream& operator>>(std::istream& cin, Matrix& matrixObj);
|
|
|
|
/* Output operator prints elements of the matrix
|
|
* row by row separated with a space (no space at the end of each line)
|
|
*/
|
|
friend std::ostream& operator<<(std::ostream& cout, Matrix& matrixObj);
|
|
};
|
|
|
|
#endif // TOOLS_MATRIX_H
|