GridFractionalSinglePhasePressureSolver2.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_FRACTIONAL_SINGLE_PHASE_PRESSURE_SOLVER2_HPP
#define CUBBYFLOW_FRACTIONAL_SINGLE_PHASE_PRESSURE_SOLVER2_HPP

#include <Core/FDM/FDMMGLinearSystem2.hpp>
#include <Core/Solver/FDM/FDMLinearSystemSolver2.hpp>
#include <Core/Solver/FDM/FDMMGSolver2.hpp>
#include <Core/Solver/Grid/GridPressureSolver2.hpp>

namespace CubbyFlow
{
class GridFractionalSinglePhasePressureSolver2 final
    : public GridPressureSolver2
{
 public:
    GridFractionalSinglePhasePressureSolver2();

    GridFractionalSinglePhasePressureSolver2(
        const GridFractionalSinglePhasePressureSolver2&) = delete;

    GridFractionalSinglePhasePressureSolver2(
        GridFractionalSinglePhasePressureSolver2&&) noexcept = delete;

    ~GridFractionalSinglePhasePressureSolver2() override = default;

    GridFractionalSinglePhasePressureSolver2& operator=(
        const GridFractionalSinglePhasePressureSolver2&) = delete;

    GridFractionalSinglePhasePressureSolver2& operator=(
        GridFractionalSinglePhasePressureSolver2&&) noexcept = delete;

    void Solve(
        const FaceCenteredGrid2& input, double timeIntervalInSeconds,
        FaceCenteredGrid2* output,
        const ScalarField2& boundarySDF =
            ConstantScalarField2{ std::numeric_limits<double>::max() },
        const VectorField2& boundaryVelocity = ConstantVectorField2{ { 0, 0 } },
        const ScalarField2& fluidSDF =
            ConstantScalarField2{ -std::numeric_limits<double>::max() },
        bool useCompressed = false) override;

    [[nodiscard]] GridBoundaryConditionSolver2Ptr
    SuggestedBoundaryConditionSolver() const override;

    [[nodiscard]] const FDMLinearSystemSolver2Ptr& GetLinearSystemSolver()
        const;

    void SetLinearSystemSolver(const FDMLinearSystemSolver2Ptr& solver);

    [[nodiscard]] const FDMVector2& GetPressure() const;

 private:
    void BuildWeights(const FaceCenteredGrid2& input,
                      const ScalarField2& boundarySDF,
                      const VectorField2& boundaryVelocity,
                      const ScalarField2& fluidSDF);

    void DecompressSolution();

    void BuildSystem(const FaceCenteredGrid2& input, bool useCompressed);

    void ApplyPressureGradient(const FaceCenteredGrid2& input,
                               FaceCenteredGrid2* output);

    FDMLinearSystem2 m_system;
    FDMCompressedLinearSystem2 m_compSystem;
    FDMLinearSystemSolver2Ptr m_systemSolver;

    FDMMGLinearSystem2 m_mgSystem;
    FDMMGSolver2Ptr m_mgSystemSolver;

    std::vector<Array2<double>> m_uWeights;
    std::vector<Array2<double>> m_vWeights;
    std::vector<Array2<double>> m_fluidSDF;

    std::function<Vector2D(const Vector2D&)> m_boundaryVel;
};

using GridFractionalSinglePhasePressureSolver2Ptr =
    std::shared_ptr<GridFractionalSinglePhasePressureSolver2>;
}  // namespace CubbyFlow

#endif