FDMMGSolver2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: FDMMGSolver2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 2-D finite difference-type linear system solver using Multigrid.
> Created Time: 2017/11/03
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_FDM_MG_SOLVER2_H
#define CUBBYFLOW_FDM_MG_SOLVER2_H

#include <Core/FDM/FDMMGLinearSystem2.h>
#include <Core/Solver/FDM/FDMLinearSystemSolver2.h>
#include <Core/Utils/MG.h>

namespace CubbyFlow
{
    class FDMMGSolver2 : public FDMLinearSystemSolver2
    {
    public:
        FDMMGSolver2(
            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);

        const MGParameters<FDMBLAS2>& GetParams() const;

        double GetSORFactor() const;

        bool GetUseRedBlackOrdering() const;

        bool Solve(FDMLinearSystem2* system) final;

        virtual bool Solve(FDMMGLinearSystem2* system);

    private:
        MGParameters<FDMBLAS2> m_mgParams;
        double m_sorFactor;
        bool m_useRedBlackOrdering;
    };

    using FDMMGSolver2Ptr = std::shared_ptr<FDMMGSolver2>;
}

#endif