Loading...
Searching...
No Matches
BlockMatrix.h
// Copyright (C) 2008 Kent-Andre Mardal
//
// This file is part of DOLFIN.
//
// DOLFIN is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// DOLFIN 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with DOLFIN. If not, see <http://www.gnu.org/licenses/>.
//
// Modified by Anders Logg, 2008.
// Modified by Garth N. Wells, 2011.
//
// First added: 2008-08-25
// Last changed: 2011-01-22
#ifndef __BLOCKMATRIX_H
#define __BLOCKMATRIX_H
#include <boost/multi_array.hpp>
#include <memory>
#include "BlockVector.h"
namespace dolfin
{
// Forward declarations
class GenericMatrix;
class BlockMatrix
{
public:
BlockMatrix(std::size_t m=0, std::size_t n=0);
~BlockMatrix();
void set_block(std::size_t i, std::size_t j,
std::shared_ptr<GenericMatrix> m);
std::shared_ptr<const GenericMatrix>
get_block(std::size_t i, std::size_t j) const;
std::shared_ptr<GenericMatrix> get_block(std::size_t i, std::size_t j);
std::size_t size(std::size_t dim) const;
void zero();
void apply(std::string mode);
std::string str(bool verbose) const;
void mult(const BlockVector& x, BlockVector& y,
bool transposed=false) const;
std::shared_ptr<GenericMatrix>
schur_approximation(bool symmetry=true) const;
private:
boost::multi_array<std::shared_ptr<GenericMatrix>, 2> matrices;
};
}
#endif
dolfin::BlockMatrix::BlockMatrix
BlockMatrix(std::size_t m=0, std::size_t n=0)
Constructor.
Definition BlockMatrix.cpp:38
dolfin::BlockMatrix::~BlockMatrix
~BlockMatrix()
Destructor.
Definition BlockMatrix.cpp:46
dolfin::BlockMatrix::set_block
void set_block(std::size_t i, std::size_t j, std::shared_ptr< GenericMatrix > m)
Set block.
Definition BlockMatrix.cpp:51
dolfin::BlockMatrix::get_block
std::shared_ptr< const GenericMatrix > get_block(std::size_t i, std::size_t j) const
Get block (const version).
Definition BlockMatrix.cpp:60
dolfin::BlockMatrix::zero
void zero()
Set all entries to zero and keep any sparse structure.
Definition BlockMatrix.cpp:81
dolfin::BlockMatrix::str
std::string str(bool verbose) const
Return informal string representation (pretty-print).
Definition BlockMatrix.cpp:96
dolfin::BlockMatrix::mult
void mult(const BlockVector &x, BlockVector &y, bool transposed=false) const
Matrix-vector product, y = Ax.
Definition BlockMatrix.cpp:122
dolfin::BlockMatrix::apply
void apply(std::string mode)
Finalize assembly of tensor.
Definition BlockMatrix.cpp:88
dolfin::BlockMatrix::schur_approximation
std::shared_ptr< GenericMatrix > schur_approximation(bool symmetry=true) const
Definition BlockMatrix.cpp:161
dolfin::BlockMatrix::size
std::size_t size(std::size_t dim) const
Return size of given dimension.
Definition BlockMatrix.cpp:75
dolfin::BlockVector
Block vector.
Definition BlockVector.h:42
dolfin::GenericMatrix
This class defines a common interface for matrices.
Definition GenericMatrix.h:47
dolfin
Definition adapt.h:30