NonlinearSystem.h

Go to the documentation of this file.

 
#pragma once
 
#include "EquationSystem.h"
#include "Types.h"
 
namespace NumLib
{
 
template <NonlinearSolverTag NLTag>
class NonlinearSystem;
 
template <>
class NonlinearSystem<NonlinearSolverTag::Newton> : public EquationSystem
{
public:
    virtual void assemble(std::vector<GlobalVector*> const& x,
                          std::vector<GlobalVector*> const& x_prev,
                          int const process_id) = 0;
 
    virtual void getResidual(GlobalVector const& x,
                             GlobalVector const& x_prev,
                             GlobalVector& res) const = 0;
 
    virtual void getJacobian(GlobalMatrix& Jac) const = 0;
 
    virtual void computeKnownSolutions(GlobalVector const& x,
                                       int const process_id) = 0;
 
    virtual void applyKnownSolutions(GlobalVector& x) const = 0;
 
    virtual void applyKnownSolutionsNewton(
        GlobalMatrix& Jac, GlobalVector& res,
        GlobalVector& minus_delta_x) const = 0;
 
    virtual void updateConstraints(GlobalVector& /*lower*/,
                                   GlobalVector& /*upper*/,
                                   int /*process_id*/) = 0;
};
 
template <>
class NonlinearSystem<NonlinearSolverTag::Picard> : public EquationSystem
{
public:
    virtual void assemble(std::vector<GlobalVector*> const& x,
                          std::vector<GlobalVector*> const& x_prev,
                          int const process_id) = 0;
 
    virtual void getA(GlobalMatrix& A) const = 0;
 
    virtual void getRhs(GlobalVector const& x_prev,
                        GlobalVector& rhs) const = 0;
 
    virtual void computeKnownSolutions(GlobalVector const& x,
                                       int const process_id) = 0;
 
    virtual void applyKnownSolutions(GlobalVector& x) const = 0;
 
    virtual void applyKnownSolutionsPicard(GlobalMatrix& A, GlobalVector& rhs,
                                           GlobalVector& x) const = 0;
};
 
}  // namespace NumLib