FDMMGLinearSystem2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: FDMMGLinearSystem2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: Multigrid-syle 2-D linear system.
> Created Time: 2017/11/02
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_FDM_MG_LINEAR_SYSTEM2_H
#define CUBBYFLOW_FDM_MG_LINEAR_SYSTEM2_H

#include <Core/FDM/FDMLinearSystem2.h>
#include <Core/Utils/MG.h>

namespace CubbyFlow
{
    using FDMMGMatrix2 = MGMatrix<FDMBLAS2>;

    using FDMMGVector2 = MGVector<FDMBLAS2>;

    struct FDMMGLinearSystem2
    {
        FDMMGMatrix2 A;

        FDMMGVector2 x;

        FDMMGVector2 b;

        void Clear();

        size_t GetNumberOfLevels() const;

        void ResizeWithCoarsest(const Size2& coarsestResolution, size_t numberOfLevels);

        void ResizeWithFinest(const Size2& finestResolution, size_t maxNumberOfLevels);
    };

    class FDMMGUtils2
    {
    public:
        static void Restrict(const FDMVector2& finer, FDMVector2* coarser);

        static void Correct(const FDMVector2& coarser, FDMVector2* finer);

        template <typename T>
        static void ResizeArrayWithCoarsest(const Size2& coarsestResolution,
            size_t numberOfLevels, std::vector<Array2<T>>* levels);

        template <typename T>
        static void ResizeArrayWithFinest(const Size2& finestResolution,
            size_t maxNumberOfLevels, std::vector<Array2<T>>* levels);
    };
}

#include <Core/FDM/FDMMGLinearSystem2-Impl.h>

#endif