GSparseMatrix.i
5.39 KB
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
/***************************************************************************
* GSparseMatrix.i - Sparse Matrix class SWIG file *
* ----------------------------------------------------------------------- *
* copyright (C) 2008-2012 by Juergen Knoedlseder *
* ----------------------------------------------------------------------- *
* *
* This program is free software: you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation, either version 3 of the License, or *
* (at your option) any later version. *
* *
* This program is distributed in the hope that it will be useful, *
* but WITHOUT ANY WARRANTY; without even the implied warranty of *
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
* GNU General Public License for more details. *
* *
* You should have received a copy of the GNU General Public License *
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
* *
***************************************************************************/
/**
* @file GSparseMatrix.i
* @brief Sparse matrix class definition.
* @author Juergen Knoedlseder
*/
%{
/* Put headers and other declarations here that are needed for compilation */
#include "GSparseMatrix.hpp"
#include "GTools.hpp"
%}
/* __ Includes ___________________________________________________________ */
%include "GTypemaps.i"
/***********************************************************************//**
* @class GSparseMatrix
*
* @brief Sparse matrix class definition
*
* GSparseMatrix implements a sparse matrix storage class. It derives from
* the abstract base class GMatrixBase.
***************************************************************************/
class GSparseMatrix : public GMatrixBase {
public:
// Constructors and destructors
GSparseMatrix(void);
GSparseMatrix(const int& rows, const int& cols, const int& elements = 0);
GSparseMatrix(const GMatrix& matrix);
GSparseMatrix(const GSymMatrix& matrix);
GSparseMatrix(const GSparseMatrix& matrix);
virtual ~GSparseMatrix(void);
// Implemented pure virtual base class operators
GVector operator*(const GVector& vector) const;
// Overloaded virtual base class operators
bool operator==(const GSparseMatrix& matrix) const;
bool operator!=(const GSparseMatrix& matrix) const;
// Other operators
GSparseMatrix operator+(const GSparseMatrix& matrix) const;
GSparseMatrix operator-(const GSparseMatrix& matrix) const;
GSparseMatrix operator*(const GSparseMatrix& matrix) const;
GSparseMatrix operator-(void) const;
GSparseMatrix& operator+=(const GSparseMatrix& matrix);
GSparseMatrix& operator-=(const GSparseMatrix& matrix);
GSparseMatrix& operator*=(const GSparseMatrix& matrix);
GSparseMatrix& operator*=(const double& scalar);
GSparseMatrix& operator/=(const double& scalar);
// Implemented pure virtual base class methods
void clear(void);
GSparseMatrix* clone(void) const;
void transpose(void);
void invert(void);
void add_col(const GVector& vector, const int& col);
void insert_col(const GVector& vector, const int& col);
GVector extract_row(const int& row) const;
GVector extract_col(const int& col) const;
double fill(void) const;
double min(void) const;
double max(void) const;
double sum(void) const;
// Other methods
void add_col(const double* values, const int* rows,
int number, const int& col);
void insert_col(const double* values, const int* rows,
int number, const int& col);
void cholesky_decompose(bool compress = true);
GVector cholesky_solver(const GVector& vector, bool compress = true);
void cholesky_invert(bool compress = true);
void set_mem_block(const int& block);
void stack_init(const int& size = 0, const int& entries = 0);
int stack_push_column(const GVector& vector, const int& col);
int stack_push_column(const double* values, const int* rows,
const int& number, const int& col);
void stack_flush(void);
void stack_destroy(void);
};
/***********************************************************************//**
* @brief GSparseMatrix class extension
***************************************************************************/
%extend GSparseMatrix {
char *__str__() {
return tochar(self->print());
}
double __getitem__(int GTuple[2]) {
return (*self)(GTuple[0], GTuple[1]);
}
void __setitem__(int GTuple[2], double value) {
(*self)(GTuple[0], GTuple[1]) = value;
}
GSparseMatrix __mul__(const double &a) {
return (*self) * a;
}
GSparseMatrix __div__(const double &a) {
return (*self) / a;
}
GSparseMatrix copy() {
return (*self);
}
};