DiagonalSolver.h

Go to the documentation of this file.

#ifndef included_AMP_DiagonalSolver
#define included_AMP_DiagonalSolver
 
#include "AMP/solvers/SolverStrategy.h"
#include "AMP/solvers/SolverStrategyParameters.h"
#include "AMP/utils/AMP_MPI.h"
 
#include <string>
 
namespace AMP::Solver {
 
template<typename T = double>
class DiagonalSolver : public SolverStrategy
{
public:
    DiagonalSolver() = default;
 
    explicit DiagonalSolver( std::shared_ptr<SolverStrategyParameters> params );
 
    static std::unique_ptr<SolverStrategy>
    createSolver( std::shared_ptr<SolverStrategyParameters> params )
    {
        return std::make_unique<DiagonalSolver<T>>( params );
    }
 
    virtual ~DiagonalSolver() = default;
 
    std::string type() const override { return "DiagonalSolver"; }
 
    void apply( std::shared_ptr<const AMP::LinearAlgebra::Vector> f,
                std::shared_ptr<AMP::LinearAlgebra::Vector> u ) override;
 
    void initialize( std::shared_ptr<const SolverStrategyParameters> params ) override;
 
    void registerOperator( std::shared_ptr<AMP::Operator::Operator> op ) override;
 
protected:
    void getFromInput( std::shared_ptr<AMP::Database> db );
 
private:
    bool d_bUsesNestedSolver = false;
    std::shared_ptr<AMP::LinearAlgebra::Vector> d_pDiagonalInverse;
};
} // namespace AMP::Solver
 
#endif