FDMMGLinearSystem3.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_SYSTEM3_HPP
#define CUBBYFLOW_FDM_MG_LINEAR_SYSTEM3_HPP
 
#include <Core/FDM/FDMLinearSystem3.hpp>
#include <Core/Utils/MG.hpp>
 
namespace CubbyFlow
{
using FDMMGMatrix3 = MGMatrix<FDMBLAS3>;
 
using FDMMGVector3 = MGVector<FDMBLAS3>;
 
struct FDMMGLinearSystem3
{
    void Clear();
 
    [[nodiscard]] size_t GetNumberOfLevels() const;
 
    void ResizeWithCoarsest(const Vector3UZ& coarsestResolution,
                            size_t numberOfLevels);
 
    void ResizeWithFinest(const Vector3UZ& finestResolution,
                          size_t maxNumberOfLevels);
 
    FDMMGMatrix3 A;
 
    FDMMGVector3 x;
 
    FDMMGVector3 b;
};
 
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 Vector3UZ& coarsestResolution,
                                        size_t numberOfLevels,
                                        std::vector<Array3<T>>* levels);
 
    template <typename T>
    static void ResizeArrayWithFinest(const Vector3UZ& finestResolution,
                                      size_t maxNumberOfLevels,
                                      std::vector<Array3<T>>* levels);
};
}  // namespace CubbyFlow
 
#include <Core/FDM/FDMMGLinearSystem3-Impl.hpp>
 
#endif