matrixequation.hpp

#pragma once
 
#include <armadillo>
 
class BoundaryConditions;
 
class MatrixEquation
{
public:
    ~MatrixEquation();
 
    void fillCoefficientMatrixAndConstantsVector(
            const arma::uword nGridPoints,
            const arma::uword nEquationsAndVariables,
            const BoundaryConditions& boundaryConditions,
            const arma::cube& term_i,
            const arma::cube& term_ipp,
            const arma::mat& boundaryTerms); // fills self matrix and vector
 
    arma::mat solve(
            const arma::uword nGridPoints,
            const arma::uword nEquationsAndVariables,
            const BoundaryConditions& boundaryConditions) const;
 
    const arma::sp_mat& coefficients() const { return m_coefficients; }
    const arma::vec& constants() const { return m_constants; }
 
private:
    arma::sp_mat m_coefficients; 
    arma::vec m_constants; 
 
    arma::mat reshapeSolverOutput(
            const arma::vec& x,
            const BoundaryConditions& boundaryConditions,
            const arma::uword nGridPoints,
            const arma::uword nVariables) const;
 
    arma::vec solveMatrixEquation() const;
};