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