MFEM
v4.6.0
Finite element discretization library
|
#include <blockmatrix.hpp>
Public Member Functions | |
BlockMatrix (const Array< int > &offsets) | |
Constructor for square block matrices. More... | |
BlockMatrix (const Array< int > &row_offsets, const Array< int > &col_offsets) | |
Constructor for rectangular block matrices. More... | |
void | SetBlock (int i, int j, SparseMatrix *mat) |
Set A(i,j) = mat. More... | |
int | NumRowBlocks () const |
Return the number of row blocks. More... | |
int | NumColBlocks () const |
Return the number of column blocks. More... | |
SparseMatrix & | GetBlock (int i, int j) |
Return a reference to block (i,j). Reference may be invalid if Aij(i,j) == NULL. More... | |
const SparseMatrix & | GetBlock (int i, int j) const |
int | IsZeroBlock (int i, int j) const |
Check if block (i,j) is a zero block. More... | |
Array< int > & | RowOffsets () |
Return the row offsets for block starts. More... | |
Array< int > & | ColOffsets () |
Return the columns offsets for block starts. More... | |
const Array< int > & | RowOffsets () const |
Return the row offsets for block starts (const version) More... | |
const Array< int > & | ColOffsets () const |
Return the row offsets for block starts (const version) More... | |
int | RowSize (const int i) const |
Return the number of non zeros in row i. More... | |
void | EliminateRowCol (int rc, DiagonalPolicy dpolicy=DIAG_ONE) |
Eliminate the row and column rc from the matrix. More... | |
void | EliminateRowCols (const Array< int > &vdofs, BlockMatrix *Ae, DiagonalPolicy dpolicy=DIAG_ONE) |
Eliminate the rows and columns corresponding to the entries in vdofs + save the eliminated entries into Ae so that (*this) + Ae is equal to the original matrix. More... | |
void | EliminateRowCol (Array< int > &ess_bc_dofs, Vector &sol, Vector &rhs) |
Symmetric elimination of the marked degree of freedom. More... | |
virtual void | Finalize (int skip_zeros=1) |
Finalize all the submatrices. More... | |
void | Finalize (int skip_zeros, bool fix_empty_rows) |
A slightly more general version of the Finalize(int) method. More... | |
SparseMatrix * | CreateMonolithic () const |
Returns a monolithic CSR matrix that represents this operator. More... | |
virtual void | PrintMatlab (std::ostream &os=mfem::out) const |
Export the monolithic matrix to file. More... | |
void | PartMult (const Array< int > &rows, const Vector &x, Vector &y) const |
Partial matrix vector multiplication of (*this) with x involving only the rows given by rows. The result is given in y. More... | |
void | PartAddMult (const Array< int > &rows, const Vector &x, Vector &y, const double a=1.0) const |
Partial matrix vector multiplication of (*this) with x involving only the rows given by rows. The result is multiplied by a and added to y. More... | |
virtual | ~BlockMatrix () |
Destructor. More... | |
virtual Type | GetType () const |
Matrix interface | |
virtual double & | Elem (int i, int j) |
Returns reference to a_{ij}. More... | |
virtual const double & | Elem (int i, int j) const |
Returns constant reference to a_{ij}. More... | |
virtual MatrixInverse * | Inverse () const |
Returns a pointer to (approximation) of the matrix inverse. More... | |
AbstractSparseMatrix interface | |
virtual int | NumNonZeroElems () const |
Returns the total number of non zeros in the matrix. More... | |
virtual int | GetRow (const int row, Array< int > &cols, Vector &srow) const |
Gets the columns indexes and values for row row. More... | |
virtual void | EliminateZeroRows (const double threshold=1e-12) |
If the matrix is square, this method will place 1 on the diagonal (i,i) if row i has "almost" zero l1-norm. More... | |
virtual void | Mult (const Vector &x, Vector &y) const |
Matrix-Vector Multiplication y = A*x. More... | |
virtual void | AddMult (const Vector &x, Vector &y, const double val=1.) const |
Matrix-Vector Multiplication y = y + val*A*x. More... | |
virtual void | MultTranspose (const Vector &x, Vector &y) const |
MatrixTranspose-Vector Multiplication y = A'*x. More... | |
virtual void | AddMultTranspose (const Vector &x, Vector &y, const double val=1.) const |
MatrixTranspose-Vector Multiplication y = y + val*A'*x. More... | |
Public Member Functions inherited from mfem::AbstractSparseMatrix | |
AbstractSparseMatrix (int s=0) | |
Creates a square matrix of the given size. More... | |
AbstractSparseMatrix (int h, int w) | |
Creates a matrix of the given height and width. More... | |
virtual | ~AbstractSparseMatrix () |
Destroys AbstractSparseMatrix. More... | |
Public Member Functions inherited from mfem::Matrix | |
Matrix (int s) | |
Creates a square matrix of size s. More... | |
Matrix (int h, int w) | |
Creates a matrix of the given height and width. More... | |
bool | IsSquare () const |
Returns whether the matrix is a square matrix. More... | |
virtual void | Print (std::ostream &out=mfem::out, int width_=4) const |
Prints matrix to stream out. More... | |
virtual | ~Matrix () |
Destroys matrix. More... | |
Public Member Functions inherited from mfem::Operator | |
void | InitTVectors (const Operator *Po, const Operator *Ri, const Operator *Pi, Vector &x, Vector &b, Vector &X, Vector &B) const |
Initializes memory for true vectors of linear system. More... | |
Operator (int s=0) | |
Construct a square Operator with given size s (default 0). More... | |
Operator (int h, int w) | |
Construct an Operator with the given height (output size) and width (input size). More... | |
int | Height () const |
Get the height (size of output) of the Operator. Synonym with NumRows(). More... | |
int | NumRows () const |
Get the number of rows (size of output) of the Operator. Synonym with Height(). More... | |
int | Width () const |
Get the width (size of input) of the Operator. Synonym with NumCols(). More... | |
int | NumCols () const |
Get the number of columns (size of input) of the Operator. Synonym with Width(). More... | |
virtual MemoryClass | GetMemoryClass () const |
Return the MemoryClass preferred by the Operator. More... | |
virtual void | ArrayMult (const Array< const Vector *> &X, Array< Vector *> &Y) const |
Operator application on a matrix: Y=A(X) . More... | |
virtual void | ArrayMultTranspose (const Array< const Vector *> &X, Array< Vector *> &Y) const |
Action of the transpose operator on a matrix: Y=A^t(X) . More... | |
virtual void | ArrayAddMult (const Array< const Vector *> &X, Array< Vector *> &Y, const double a=1.0) const |
Operator application on a matrix: Y+=A(X) (default) or Y+=a*A(X) . More... | |
virtual void | ArrayAddMultTranspose (const Array< const Vector *> &X, Array< Vector *> &Y, const double a=1.0) const |
Operator transpose application on a matrix: Y+=A^t(X) (default) or Y+=a*A^t(X) . More... | |
virtual Operator & | GetGradient (const Vector &x) const |
Evaluate the gradient operator at the point x. The default behavior in class Operator is to generate an error. More... | |
virtual void | AssembleDiagonal (Vector &diag) const |
Computes the diagonal entries into diag. Typically, this operation only makes sense for linear Operators. In some cases, only an approximation of the diagonal is computed. More... | |
virtual const Operator * | GetProlongation () const |
Prolongation operator from linear algebra (linear system) vectors, to input vectors for the operator. NULL means identity. More... | |
virtual const Operator * | GetRestriction () const |
Restriction operator from input vectors for the operator to linear algebra (linear system) vectors. NULL means identity. More... | |
virtual const Operator * | GetOutputProlongation () const |
Prolongation operator from linear algebra (linear system) vectors, to output vectors for the operator. NULL means identity. More... | |
virtual const Operator * | GetOutputRestrictionTranspose () const |
Transpose of GetOutputRestriction, directly available in this form to facilitate matrix-free RAP-type operators. More... | |
virtual const Operator * | GetOutputRestriction () const |
Restriction operator from output vectors for the operator to linear algebra (linear system) vectors. NULL means identity. More... | |
void | FormLinearSystem (const Array< int > &ess_tdof_list, Vector &x, Vector &b, Operator *&A, Vector &X, Vector &B, int copy_interior=0) |
Form a constrained linear system using a matrix-free approach. More... | |
void | FormRectangularLinearSystem (const Array< int > &trial_tdof_list, const Array< int > &test_tdof_list, Vector &x, Vector &b, Operator *&A, Vector &X, Vector &B) |
Form a column-constrained linear system using a matrix-free approach. More... | |
virtual void | RecoverFEMSolution (const Vector &X, const Vector &b, Vector &x) |
Reconstruct a solution vector x (e.g. a GridFunction) from the solution X of a constrained linear system obtained from Operator::FormLinearSystem() or Operator::FormRectangularLinearSystem(). More... | |
void | FormSystemOperator (const Array< int > &ess_tdof_list, Operator *&A) |
Return in A a parallel (on truedofs) version of this square operator. More... | |
void | FormRectangularSystemOperator (const Array< int > &trial_tdof_list, const Array< int > &test_tdof_list, Operator *&A) |
Return in A a parallel (on truedofs) version of this rectangular operator (including constraints). More... | |
void | FormDiscreteOperator (Operator *&A) |
Return in A a parallel (on truedofs) version of this rectangular operator. More... | |
void | PrintMatlab (std::ostream &out, int n, int m=0) const |
Prints operator with input size n and output size m in Matlab format. More... | |
virtual | ~Operator () |
Virtual destructor. More... | |
Type | GetType () const |
Return the type ID of the Operator class. More... | |
Public Attributes | |
int | owns_blocks |
If owns_blocks the SparseMatrix objects Aij will be deallocated. More... | |
Additional Inherited Members | |
Public Types inherited from mfem::Operator | |
enum | DiagonalPolicy { DIAG_ZERO, DIAG_ONE, DIAG_KEEP } |
Defines operator diagonal policy upon elimination of rows and/or columns. More... | |
enum | Type { ANY_TYPE, MFEM_SPARSEMAT, Hypre_ParCSR, PETSC_MATAIJ, PETSC_MATIS, PETSC_MATSHELL, PETSC_MATNEST, PETSC_MATHYPRE, PETSC_MATGENERIC, Complex_Operator, MFEM_ComplexSparseMat, Complex_Hypre_ParCSR, Complex_DenseMat, MFEM_Block_Matrix, MFEM_Block_Operator } |
Enumeration defining IDs for some classes derived from Operator. More... | |
Protected Member Functions inherited from mfem::Operator | |
void | FormConstrainedSystemOperator (const Array< int > &ess_tdof_list, ConstrainedOperator *&Aout) |
see FormSystemOperator() More... | |
void | FormRectangularConstrainedSystemOperator (const Array< int > &trial_tdof_list, const Array< int > &test_tdof_list, RectangularConstrainedOperator *&Aout) |
see FormRectangularSystemOperator() More... | |
Operator * | SetupRAP (const Operator *Pi, const Operator *Po) |
Returns RAP Operator of this, using input/output Prolongation matrices Pi corresponds to "P", Po corresponds to "Rt". More... | |
Protected Attributes inherited from mfem::Operator | |
int | height |
Dimension of the output / number of rows in the matrix. More... | |
int | width |
Dimension of the input / number of columns in the matrix. More... | |
Definition at line 24 of file blockmatrix.hpp.
mfem::BlockMatrix::BlockMatrix | ( | const Array< int > & | offsets | ) |
Constructor for square block matrices.
offsets | offsets that mark the start of each row/column block (size nRowBlocks+1). |
Definition at line 23 of file blockmatrix.cpp.
mfem::BlockMatrix::BlockMatrix | ( | const Array< int > & | row_offsets, |
const Array< int > & | col_offsets | ||
) |
Constructor for rectangular block matrices.
row_offsets | offsets that mark the start of each row block (size nRowBlocks+1). |
col_offsets | offsets that mark the start of each column block (size nColBlocks+1). |
Definition at line 35 of file blockmatrix.cpp.
|
virtual |
Destructor.
Definition at line 49 of file blockmatrix.cpp.
|
virtual |
Matrix-Vector Multiplication y = y + val*A*x.
Implements mfem::AbstractSparseMatrix.
Definition at line 453 of file blockmatrix.cpp.
|
virtual |
MatrixTranspose-Vector Multiplication y = y + val*A'*x.
Implements mfem::AbstractSparseMatrix.
Definition at line 492 of file blockmatrix.cpp.
|
inline |
Return the columns offsets for block starts.
Definition at line 59 of file blockmatrix.hpp.
|
inline |
Return the row offsets for block starts (const version)
Definition at line 63 of file blockmatrix.hpp.
SparseMatrix * mfem::BlockMatrix::CreateMonolithic | ( | ) | const |
Returns a monolithic CSR matrix that represents this operator.
Definition at line 560 of file blockmatrix.cpp.
|
virtual |
Returns reference to a_{ij}.
Implements mfem::Matrix.
Definition at line 133 of file blockmatrix.cpp.
|
virtual |
Returns constant reference to a_{ij}.
Implements mfem::Matrix.
Definition at line 149 of file blockmatrix.cpp.
void mfem::BlockMatrix::EliminateRowCol | ( | int | rc, |
DiagonalPolicy | dpolicy = DIAG_ONE |
||
) |
Eliminate the row and column rc from the matrix.
Eliminates the column and row rc, replacing the element (rc,rc) with 1.0. Assumes that element (i,rc) is assembled if and only if the element (rc,i) is assembled. If dpolicy is specified, the element (rc,rc) is treated according to that policy.
Definition at line 213 of file blockmatrix.cpp.
Symmetric elimination of the marked degree of freedom.
ess_bc_dofs | marker of the degree of freedom to be eliminated dof i is eliminated if ess_bc_dofs[i] = 1. |
sol | vector that stores the values of the degree of freedom that need to be eliminated |
rhs | vector that stores the rhs of the system. |
Definition at line 250 of file blockmatrix.cpp.
void mfem::BlockMatrix::EliminateRowCols | ( | const Array< int > & | vdofs, |
BlockMatrix * | Ae, | ||
DiagonalPolicy | dpolicy = DIAG_ONE |
||
) |
Eliminate the rows and columns corresponding to the entries in vdofs + save the eliminated entries into Ae so that (*this) + Ae is equal to the original matrix.
Definition at line 322 of file blockmatrix.cpp.
|
virtual |
If the matrix is square, this method will place 1 on the diagonal (i,i) if row i has "almost" zero l1-norm.
If entry (i,i) does not belong to the sparsity pattern of A, then a error will occur.
Implements mfem::AbstractSparseMatrix.
Definition at line 370 of file blockmatrix.cpp.
|
inlinevirtual |
Finalize all the submatrices.
Reimplemented from mfem::Matrix.
Definition at line 91 of file blockmatrix.hpp.
void mfem::BlockMatrix::Finalize | ( | int | skip_zeros, |
bool | fix_empty_rows | ||
) |
A slightly more general version of the Finalize(int) method.
Definition at line 423 of file blockmatrix.cpp.
SparseMatrix & mfem::BlockMatrix::GetBlock | ( | int | i, |
int | j | ||
) |
Return a reference to block (i,j). Reference may be invalid if Aij(i,j) == NULL.
Definition at line 82 of file blockmatrix.cpp.
const SparseMatrix & mfem::BlockMatrix::GetBlock | ( | int | i, |
int | j | ||
) | const |
Return a reference to block (i,j). Reference may be invalid if Aij(i,j) == NULL. (const version)
Definition at line 98 of file blockmatrix.cpp.
Gets the columns indexes and values for row row.
The return value is always 0 since cols and srow are copies of the values in the matrix.
Implements mfem::AbstractSparseMatrix.
Definition at line 183 of file blockmatrix.cpp.
|
inlinevirtual |
Definition at line 156 of file blockmatrix.hpp.
|
inlinevirtual |
Returns a pointer to (approximation) of the matrix inverse.
Implements mfem::Matrix.
Definition at line 108 of file blockmatrix.hpp.
|
inline |
Check if block (i,j) is a zero block.
Definition at line 55 of file blockmatrix.hpp.
Matrix-Vector Multiplication y = A*x.
Implements mfem::AbstractSparseMatrix.
Definition at line 438 of file blockmatrix.cpp.
MatrixTranspose-Vector Multiplication y = A'*x.
Implements mfem::AbstractSparseMatrix.
Definition at line 481 of file blockmatrix.cpp.
|
inline |
Return the number of column blocks.
Definition at line 48 of file blockmatrix.hpp.
|
virtual |
Returns the total number of non zeros in the matrix.
Implements mfem::AbstractSparseMatrix.
Definition at line 116 of file blockmatrix.cpp.
|
inline |
Return the number of row blocks.
Definition at line 46 of file blockmatrix.hpp.
void mfem::BlockMatrix::PartAddMult | ( | const Array< int > & | rows, |
const Vector & | x, | ||
Vector & | y, | ||
const double | a = 1.0 |
||
) | const |
Partial matrix vector multiplication of (*this) with x involving only the rows given by rows. The result is multiplied by a and added to y.
Definition at line 539 of file blockmatrix.cpp.
Partial matrix vector multiplication of (*this) with x involving only the rows given by rows. The result is given in y.
Definition at line 521 of file blockmatrix.cpp.
|
virtual |
Export the monolithic matrix to file.
Reimplemented from mfem::Operator.
Definition at line 640 of file blockmatrix.cpp.
|
inline |
Return the row offsets for block starts.
Definition at line 57 of file blockmatrix.hpp.
|
inline |
Return the row offsets for block starts (const version)
Definition at line 61 of file blockmatrix.hpp.
int mfem::BlockMatrix::RowSize | ( | const int | i | ) | const |
Return the number of non zeros in row i.
Definition at line 165 of file blockmatrix.cpp.
void mfem::BlockMatrix::SetBlock | ( | int | i, |
int | j, | ||
SparseMatrix * | mat | ||
) |
Set A(i,j) = mat.
Definition at line 61 of file blockmatrix.cpp.
int mfem::BlockMatrix::owns_blocks |
If owns_blocks the SparseMatrix objects Aij will be deallocated.
Definition at line 154 of file blockmatrix.hpp.