FDMGaussSeidelSolver2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: FDMGaussSeidelSolver2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 2-D finite difference-type linear system solver using Gauss-Seidel method.
> Created Time: 2017/08/17
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_FDM_GAUSS_SEIDEL_SOLVER2_H
#define CUBBYFLOW_FDM_GAUSS_SEIDEL_SOLVER2_H

#include <Core/Solver/FDM/FDMLinearSystemSolver2.h>

namespace CubbyFlow
{
    class FDMGaussSeidelSolver2 final : public FDMLinearSystemSolver2
    {
    public:
        FDMGaussSeidelSolver2(
            unsigned int maxNumberOfIterations,
            unsigned int residualCheckInterval,
            double tolerance,
            double sorFactor = 1.0,
            bool useRedBlackOrdering = false);

        bool Solve(FDMLinearSystem2* system) override;

        bool SolveCompressed(FDMCompressedLinearSystem2* system) override;

        unsigned int GetMaxNumberOfIterations() const;

        unsigned int GetLastNumberOfIterations() const;

        double GetTolerance() const;

        double GetLastResidual() const;

        double GetSORFactor() const;

        bool GetUseRedBlackOrdering() const;

        static void Relax(const FDMMatrix2& A, const FDMVector2& b, double sorFactor, FDMVector2* x);

        static void Relax(const MatrixCSRD& A, const VectorND& b, double sorFactor, VectorND* x);

        static void RelaxRedBlack(const FDMMatrix2& A, const FDMVector2& b, double sorFactor, FDMVector2* x);

    private:
        unsigned int m_maxNumberOfIterations;
        unsigned int m_lastNumberOfIterations;
        unsigned int m_residualCheckInterval;
        double m_tolerance;
        double m_lastResidual;
        double m_sorFactor;
        double m_useRedBlackOrdering;

        // Uncompressed vectors
        FDMVector2 m_residual;

        // Compressed vectors
        VectorND m_residualComp;

        void ClearUncompressedVectors();
        void ClearCompressedVectors();
    };

    using FDMGaussSeidelSolver2Ptr = std::shared_ptr<FDMGaussSeidelSolver2>;
}

#endif