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