fdm_mgpcg_solver2.h

Go to the documentation of this file.

// Copyright (c) 2018 Doyub Kim
//
// I am making my contributions/submissions to this project solely in my
// personal capacity and am not conveying any rights to any intellectual
// property of any third parties.
 
#ifndef INCLUDE_JET_FDM_MGPCG_SOLVER2_H_
#define INCLUDE_JET_FDM_MGPCG_SOLVER2_H_
 
#include <jet/fdm_mg_solver2.h>
 
namespace jet {
 
class FdmMgpcgSolver2 final : public FdmMgSolver2 {
 public:
    FdmMgpcgSolver2(unsigned int numberOfCgIter, size_t maxNumberOfLevels,
                    unsigned int numberOfRestrictionIter = 5,
                    unsigned int numberOfCorrectionIter = 5,
                    unsigned int numberOfCoarsestIter = 20,
                    unsigned int numberOfFinalIter = 20,
                    double maxTolerance = 1e-9, double sorFactor = 1.5,
                    bool useRedBlackOrdering = false);
 
    bool solve(FdmMgLinearSystem2* system) override;
 
    unsigned int maxNumberOfIterations() const;
 
    unsigned int lastNumberOfIterations() const;
 
    double tolerance() const;
 
    double lastResidual() const;
 
 private:
    struct Preconditioner final {
        FdmMgLinearSystem2* system;
        MgParameters<FdmBlas2> mgParams;
 
        void build(FdmMgLinearSystem2* system, MgParameters<FdmBlas2> mgParams);
 
        void solve(const FdmVector2& b, FdmVector2* x);
    };
 
    unsigned int _maxNumberOfIterations;
    unsigned int _lastNumberOfIterations;
    double _tolerance;
    double _lastResidualNorm;
 
    FdmVector2 _r;
    FdmVector2 _d;
    FdmVector2 _q;
    FdmVector2 _s;
    Preconditioner _precond;
};
 
typedef std::shared_ptr<FdmMgpcgSolver2> FdmMgpcgSolver2Ptr;
 
}  // namespace jet
 
#endif  // INCLUDE_JET_FDM_MGPCG_SOLVER2_H_