[325390] | 1 | /*
|
---|
| 2 | * Matrix.hpp
|
---|
| 3 | *
|
---|
| 4 | * Created on: Jun 25, 2010
|
---|
| 5 | * Author: crueger
|
---|
| 6 | */
|
---|
| 7 |
|
---|
| 8 | #ifndef MATRIX_HPP_
|
---|
| 9 | #define MATRIX_HPP_
|
---|
| 10 |
|
---|
| 11 | #include <gsl/gsl_matrix.h>
|
---|
[c49c96] | 12 | #include <iostream>
|
---|
[325390] | 13 | #include "defs.hpp"
|
---|
| 14 |
|
---|
[cd82ec] | 15 | /**
|
---|
| 16 | * Simple class to store matrices with a few extra functions
|
---|
| 17 | */
|
---|
| 18 |
|
---|
[325390] | 19 | class Vector;
|
---|
| 20 |
|
---|
| 21 | class Matrix
|
---|
| 22 | {
|
---|
| 23 | friend Vector operator*(const Matrix&,const Vector&);
|
---|
| 24 | public:
|
---|
| 25 | Matrix();
|
---|
[cd82ec] | 26 |
|
---|
| 27 | /**
|
---|
| 28 | * construct a matrix from a 3x3 double array that contains all Elements.
|
---|
| 29 | *
|
---|
| 30 | * The elements are laid out in the following way:
|
---|
| 31 | * array -> matrix
|
---|
| 32 | * 0 -> (0,0)
|
---|
| 33 | * 1 -> (1,0)
|
---|
| 34 | * 2 -> (2,0)
|
---|
| 35 | * 3 -> (0,1)
|
---|
| 36 | * 4 -> (1,1)
|
---|
| 37 | * 5 -> (2,1)
|
---|
| 38 | * 6 -> (0,2)
|
---|
| 39 | * 7 -> (1,2)
|
---|
| 40 | * 8 -> (2,2)
|
---|
| 41 | *
|
---|
| 42 | */
|
---|
[325390] | 43 | Matrix(const double*);
|
---|
| 44 | Matrix(const Matrix&);
|
---|
| 45 | virtual ~Matrix();
|
---|
| 46 |
|
---|
[cd82ec] | 47 | /**
|
---|
| 48 | * Set the matrix to a unit matrix.
|
---|
| 49 | */
|
---|
[1da5f5] | 50 | void one();
|
---|
| 51 |
|
---|
[cd82ec] | 52 | /**
|
---|
| 53 | * Access the matrix at index (i,j)
|
---|
| 54 | */
|
---|
[325390] | 55 | double &at(size_t i, size_t j);
|
---|
[cd82ec] | 56 | /**
|
---|
| 57 | * Access the matrix at index (i,j)
|
---|
| 58 | */
|
---|
[436f04] | 59 | const double at(size_t i, size_t j) const;
|
---|
| 60 |
|
---|
[cd82ec] | 61 | /**
|
---|
| 62 | * Set the matrix at index (i,j).
|
---|
| 63 | *
|
---|
| 64 | * Slightly faster than at(i,j)=x
|
---|
| 65 | */
|
---|
[436f04] | 66 | void set(size_t i, size_t j, const double value);
|
---|
[325390] | 67 |
|
---|
[cd82ec] | 68 | /**
|
---|
| 69 | * Calculate the determinant of the matrix
|
---|
| 70 | */
|
---|
[cadbc1] | 71 | double determinant() const;
|
---|
[325390] | 72 |
|
---|
[cd82ec] | 73 | /**
|
---|
| 74 | * Calculate the inverse of the matrix.
|
---|
| 75 | *
|
---|
| 76 | * Rather costly, so use precomputation as often as possible.
|
---|
| 77 | */
|
---|
[cadbc1] | 78 | Matrix invert() const;
|
---|
[325390] | 79 |
|
---|
| 80 | // operators
|
---|
| 81 | Matrix &operator=(const Matrix&);
|
---|
| 82 |
|
---|
| 83 | Matrix &operator+=(const Matrix&);
|
---|
| 84 | Matrix &operator-=(const Matrix&);
|
---|
| 85 | Matrix &operator*=(const Matrix&);
|
---|
| 86 |
|
---|
| 87 | Matrix &operator*=(const double);
|
---|
| 88 |
|
---|
| 89 | Matrix operator+(const Matrix&) const;
|
---|
| 90 | Matrix operator-(const Matrix&) const;
|
---|
| 91 | Matrix operator*(const Matrix&) const;
|
---|
| 92 |
|
---|
| 93 | private:
|
---|
| 94 | Matrix(gsl_matrix*);
|
---|
| 95 | gsl_matrix *content;
|
---|
| 96 | };
|
---|
| 97 |
|
---|
| 98 | Matrix operator*(const double,const Matrix&);
|
---|
| 99 | Matrix operator*(const Matrix&,const double);
|
---|
| 100 |
|
---|
[cd82ec] | 101 | /**
|
---|
| 102 | * Takes a symmetric matrix that stores the lower diagonal and produces a
|
---|
| 103 | * full matrix.
|
---|
| 104 | *
|
---|
| 105 | * The array is laid out as follows:
|
---|
| 106 | *
|
---|
| 107 | * array -> matrix
|
---|
| 108 | * 0 -> (0,0)
|
---|
| 109 | * 1 -> (1,0);(0,1)
|
---|
| 110 | * 2 -> (1,1)
|
---|
| 111 | * 3 -> (2,0);(0,2)
|
---|
| 112 | * 4 -> (2,1);(1,2)
|
---|
| 113 | * 5 -> (2,2)
|
---|
| 114 | */
|
---|
[d10eb6] | 115 | Matrix ReturnFullMatrixforSymmetric(const double * const cell_size);
|
---|
| 116 |
|
---|
[c49c96] | 117 | std::ostream &operator<<(std::ostream&,const Matrix&);
|
---|
[4b94bb] | 118 | Vector operator*(const Matrix&,const Vector&);
|
---|
| 119 | Vector& operator*=(Vector&,const Matrix&);
|
---|
[c49c96] | 120 |
|
---|
[325390] | 121 | #endif /* MATRIX_HPP_ */
|
---|