104 lines
2.2 KiB
C++
104 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
|