FDMMGLinearSystem3-Impl.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_IMPL_HPP
#define CUBBYFLOW_FDM_MG_LINEAR_SYSTEM3_IMPL_HPP
 
namespace CubbyFlow
{
template <typename T>
void FDMMGUtils3::ResizeArrayWithCoarsest(const Vector3UZ& coarsestResolution,
                                          size_t numberOfLevels,
                                          std::vector<Array3<T>>* levels)
{
    numberOfLevels = std::max(numberOfLevels, ONE_SIZE);
 
    levels->resize(numberOfLevels);
 
    // Level 0 is the finest level, thus takes coarsestResolution ^
    // numberOfLevels. Level numberOfLevels - 1 is the coarsest, taking
    // coarsestResolution.
    Vector3UZ res = coarsestResolution;
    for (size_t level = 0; level < numberOfLevels; ++level)
    {
        (*levels)[numberOfLevels - level - 1].Resize(res);
        res.x = res.x << 1;
        res.y = res.y << 1;
        res.z = res.z << 1;
    }
}
 
template <typename T>
void FDMMGUtils3::ResizeArrayWithFinest(const Vector3UZ& finestResolution,
                                        size_t maxNumberOfLevels,
                                        std::vector<Array3<T>>* levels)
{
    Vector3UZ res = finestResolution;
    size_t i = 1;
 
    for (; i < maxNumberOfLevels; ++i)
    {
        if (res.x % 2 == 0 && res.y % 2 == 0 && res.z % 2 == 0)
        {
            res.x = res.x >> 1;
            res.y = res.y >> 1;
            res.z = res.z >> 1;
        }
        else
        {
            break;
        }
    }
 
    ResizeArrayWithCoarsest(res, i, levels);
}
}  // namespace CubbyFlow
 
#endif