FDMMGLinearSystem2.hpp

Go to the documentation of this file.

// This code is based on Jet framework.
// Copyright (c) 2018 Doyub Kim
// CubbyFlow is voxel-based fluid simulation engine for computer games.
// Copyright (c) 2020 CubbyFlow Team
// Core Part: Chris Ohk, Junwoo Hwang, Jihong Sin, Seungwoo Yoo
// AI Part: Dongheon Cho, Minseo Kim
// We are making my contributions/submissions to this project solely in our
// personal capacity and are not conveying any rights to any intellectual
// property of any third parties.

#ifndef CUBBYFLOW_FDM_MG_LINEAR_SYSTEM2_HPP
#define CUBBYFLOW_FDM_MG_LINEAR_SYSTEM2_HPP

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

namespace CubbyFlow
{
using FDMMGMatrix2 = MGMatrix<FDMBLAS2>;

using FDMMGVector2 = MGVector<FDMBLAS2>;

struct FDMMGLinearSystem2
{
    void Clear();

    [[nodiscard]] size_t GetNumberOfLevels() const;

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

    void ResizeWithFinest(const Vector2UZ& finestResolution,
                          size_t maxNumberOfLevels);

    FDMMGMatrix2 A;

    FDMMGVector2 x;

    FDMMGVector2 b;
};

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 Vector2UZ& coarsestResolution,
                                        size_t numberOfLevels,
                                        std::vector<Array2<T>>* levels);

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

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

#endif