PyMatrix.h

Go to the documentation of this file.

#pragma once
 
#include "mfem.hpp"
#include "pybind11/pybind11.h"
 
namespace serif::pybind {

    class PyMatrix : public mfem::Matrix {
    public:
        // Inherit constructors from the base mfem::Matrix class.
        using mfem::Matrix::Matrix;
 
        // --- Trampolines for new mfem::Matrix pure virtual methods ---
        mfem::real_t& Elem(int i, int j) override;

        const mfem::real_t& Elem(int i, int j) const override;

        mfem::MatrixInverse* Inverse() const override;
 
        // --- Trampoline for new mfem::Matrix regular virtual methods ---
        void Finalize(int skip_zeros) override;
 
        // --- Trampolines for inherited mfem::Operator virtual methods ---
        // These must be repeated here to allow Python classes inheriting from
        // Matrix to override methods originally from the Operator base class.

        void Mult(const mfem::Vector &x, mfem::Vector &y) const override;

        void MultTranspose(const mfem::Vector &x, mfem::Vector &y) const override;

        void AddMult(const mfem::Vector &x, mfem::Vector &y, const mfem::real_t a = 1.0) const override;

        void AddMultTranspose(const mfem::Vector &x, mfem::Vector &y, const mfem::real_t a = 1.0) const override;

        mfem::Operator& GetGradient(const mfem::Vector &x) const override;

        void AssembleDiagonal(mfem::Vector &diag) const override;

        const mfem::Operator* GetProlongation() const override;

        const mfem::Operator* GetRestriction() const override;

        void RecoverFEMSolution(const mfem::Vector &X, const mfem::Vector &b, mfem::Vector &x) override;
    };
 
} // namespace serif::pybind