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