18#ifndef __DOLFIN_EIGEN_MATRIX_H
19#define __DOLFIN_EIGEN_MATRIX_H
26#include <Eigen/Sparse>
28#include <dolfin/common/MPI.h>
29#include <dolfin/common/types.h>
30#include "EigenVector.h"
31#include "GenericMatrix.h"
32#include "GenericVector.h"
33#include "TensorLayout.h"
72 {
return size(0) == 0; }
75 virtual std::size_t
size(std::size_t dim)
const;
78 virtual std::pair<std::int64_t, std::int64_t>
80 {
return {0,
size(dim)}; }
83 std::size_t
nnz()
const;
89 virtual void apply(std::string mode);
93 {
return _mpi_comm.
comm(); }
96 virtual std::string
str(
bool verbose)
const;
101 virtual std::shared_ptr<GenericMatrix>
copy()
const;
104 virtual void resize(std::size_t M, std::size_t N);
119 virtual void set(
const double* block, std::size_t m,
124 virtual void set_local(
const double* block, std::size_t m,
127 {
set(block, m, rows, n, cols); }
130 virtual void add(
const double* block, std::size_t m,
135 virtual void add_local(
const double* block, std::size_t m,
138 {
add(block, m, rows, n, cols); }
142 bool same_nonzero_pattern);
145 virtual double norm(std::string norm_type)
const;
148 virtual void getrow(std::size_t row, std::vector<std::size_t>& columns,
149 std::vector<double>& values)
const;
152 virtual void setrow(std::size_t row_idx,
153 const std::vector<std::size_t>& columns,
154 const std::vector<double>& values);
193 virtual std::tuple<const int*, const int*, const double*, std::size_t>
213 { _matA.makeCompressed(); }
217 {
return _matA.coeff(i, j); }
Definition EigenMatrix.h:47
virtual void add_local(const double *block, std::size_t m, const dolfin::la_index *rows, std::size_t n, const dolfin::la_index *cols)
Add block of values using local indices.
Definition EigenMatrix.h:135
virtual MPI_Comm mpi_comm() const
Return MPI communicator.
Definition EigenMatrix.h:92
virtual std::string str(bool verbose) const
Return informal string representation (pretty-print)
Definition EigenMatrix.cpp:357
virtual void mult(const GenericVector &x, GenericVector &y) const
Matrix-vector product, y = Ax.
Definition EigenMatrix.cpp:230
virtual void set_local(const double *block, std::size_t m, const dolfin::la_index *rows, std::size_t n, const dolfin::la_index *cols)
Set block of values using local indices.
Definition EigenMatrix.h:124
virtual void set_diagonal(const GenericVector &x)
Set diagonal of a matrix.
Definition EigenMatrix.cpp:271
virtual void axpy(double a, const GenericMatrix &A, bool same_nonzero_pattern)
Add multiple of given matrix (AXPY operation)
Definition EigenMatrix.cpp:422
double operator()(dolfin::la_index i, dolfin::la_index j) const
Access value of given entry.
Definition EigenMatrix.h:216
virtual const EigenMatrix & operator/=(double a)
Divide matrix by given number.
Definition EigenMatrix.cpp:320
virtual void ident(std::size_t m, const dolfin::la_index *rows)
Set given rows to identity matrix.
Definition EigenMatrix.cpp:195
const eigen_matrix_type & mat() const
Return reference to Eigen matrix (const version)
Definition EigenMatrix.h:204
virtual void add(const double *block, std::size_t m, const dolfin::la_index *rows, std::size_t n, const dolfin::la_index *cols)
Add block of values using global indices.
Definition EigenMatrix.cpp:156
virtual std::pair< std::int64_t, std::int64_t > local_range(std::size_t dim) const
Return local ownership range.
Definition EigenMatrix.h:79
virtual ~EigenMatrix()
Destructor.
Definition EigenMatrix.cpp:47
virtual void apply(std::string mode)
Finalize assembly of tensor.
Definition EigenMatrix.cpp:417
virtual GenericLinearAlgebraFactory & factory() const
Return linear algebra backend factory.
Definition EigenMatrix.cpp:25
virtual double norm(std::string norm_type) const
Return norm of matrix.
Definition EigenMatrix.cpp:77
Eigen::SparseMatrix< double, Eigen::RowMajor, int > eigen_matrix_type
Eigen Matrix type.
Definition EigenMatrix.h:51
virtual std::tuple< const int *, const int *, const double *, std::size_t > data() const
Definition EigenMatrix.cpp:342
virtual const EigenMatrix & operator*=(double a)
Multiply matrix by given number.
Definition EigenMatrix.cpp:314
virtual void init_vector(GenericVector &z, std::size_t dim) const
Definition EigenMatrix.cpp:139
virtual void zero()
Set all entries to zero and keep any sparse structure.
Definition EigenMatrix.cpp:180
virtual bool empty() const
Return true if empty.
Definition EigenMatrix.h:71
EigenMatrix()
Create empty matrix.
Definition EigenMatrix.cpp:30
virtual void setrow(std::size_t row_idx, const std::vector< std::size_t > &columns, const std::vector< double > &values)
Set values for given row.
Definition EigenMatrix.cpp:129
virtual void get(double *block, std::size_t m, const dolfin::la_index *rows, std::size_t n, const dolfin::la_index *cols) const
Get block of values.
Definition EigenMatrix.cpp:168
eigen_matrix_type & mat()
Return reference to Eigen matrix (non-const version)
Definition EigenMatrix.h:208
virtual void getrow(std::size_t row, std::vector< std::size_t > &columns, std::vector< double > &values) const
Get non-zero values of given row.
Definition EigenMatrix.cpp:107
virtual void ident_local(std::size_t m, const dolfin::la_index *rows)
Set given rows to identity matrix.
Definition EigenMatrix.h:167
virtual void init(const TensorLayout &tensor_layout)
Initialize zero tensor using tenor layout.
Definition EigenMatrix.cpp:384
std::size_t nnz() const
Return number of non-zero entries in matrix.
Definition EigenMatrix.cpp:412
virtual void zero_local(std::size_t m, const dolfin::la_index *rows)
Set given rows (local row indices) to zero.
Definition EigenMatrix.h:160
virtual void set(const double *block, std::size_t m, const dolfin::la_index *rows, std::size_t n, const dolfin::la_index *cols)
Set block of values using global indices.
Definition EigenMatrix.cpp:144
virtual void get_diagonal(GenericVector &x) const
Get diagonal of a matrix.
Definition EigenMatrix.cpp:257
virtual std::shared_ptr< GenericMatrix > copy() const
Return copy of matrix.
Definition EigenMatrix.cpp:52
void compress()
Compress matrix (eliminate all zeros from a sparse matrix)
Definition EigenMatrix.h:212
virtual const GenericMatrix & operator=(const GenericMatrix &A)
Assignment operator.
Definition EigenMatrix.cpp:326
virtual std::size_t size(std::size_t dim) const
Return size of given dimension.
Definition EigenMatrix.cpp:65
virtual void resize(std::size_t M, std::size_t N)
Resize matrix to M x N.
Definition EigenMatrix.cpp:57
virtual void transpmult(const GenericVector &x, GenericVector &y) const
Matrix-vector product, y = A^T x.
Definition EigenMatrix.cpp:285
Base class for LinearAlgebra factories.
Definition GenericLinearAlgebraFactory.h:47
This class defines a common interface for matrices.
Definition GenericMatrix.h:47
This class defines a common interface for vectors.
Definition GenericVector.h:48
MPI_Comm comm() const
Return the underlying MPI_Comm object.
Definition MPI.cpp:117
Definition TensorLayout.h:42
PetscInt la_index
Index type for compatibility with linear algebra backend(s)
Definition types.h:32