FDMMGLinearSystem3.h

Go to the documentation of this file.

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

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

namespace CubbyFlow
{
    using FDMMGMatrix3 = MGMatrix<FDMBLAS3>;

    using FDMMGVector3 = MGVector<FDMBLAS3>;

    struct FDMMGLinearSystem3
    {
        FDMMGMatrix3 A;

        FDMMGVector3 x;

        FDMMGVector3 b;

        void Clear();

        size_t GetNumberOfLevels() const;

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

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

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

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

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

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

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

#endif