LevelSetLiquidSolver2.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_LEVEL_SET_LIQUID_SOLVER2_HPP
#define CUBBYFLOW_LEVEL_SET_LIQUID_SOLVER2_HPP
 
#include <Core/Solver/Grid/GridFluidSolver2.hpp>
#include <Core/Solver/LevelSet/LevelSetSolver2.hpp>
 
namespace CubbyFlow
{
class LevelSetLiquidSolver2 : public GridFluidSolver2
{
 public:
    class Builder;
 
    LevelSetLiquidSolver2();
 
    LevelSetLiquidSolver2(const Vector2UZ& resolution,
                          const Vector2D& gridSpacing,
                          const Vector2D& gridOrigin);
 
    LevelSetLiquidSolver2(const LevelSetLiquidSolver2&) = delete;
 
    LevelSetLiquidSolver2(LevelSetLiquidSolver2&&) noexcept = delete;
 
    ~LevelSetLiquidSolver2() override = default;
 
    LevelSetLiquidSolver2& operator=(const LevelSetLiquidSolver2&) = delete;
 
    LevelSetLiquidSolver2& operator=(LevelSetLiquidSolver2&&) noexcept = delete;
 
    [[nodiscard]] ScalarGrid2Ptr GetSignedDistanceField() const;
 
    [[nodiscard]] LevelSetSolver2Ptr GetLevelSetSolver() const;
 
    void SetLevelSetSolver(const LevelSetSolver2Ptr& newSolver);
 
    void SetMinReinitializeDistance(double distance);
 
    void SetIsGlobalCompensationEnabled(bool isEnabled);
 
    [[nodiscard]] double ComputeVolume() const;
 
    [[nodiscard]] static Builder GetBuilder();
 
 protected:
    void OnBeginAdvanceTimeStep(double timeIntervalInSeconds) override;
 
    void OnEndAdvanceTimeStep(double timeIntervalInSeconds) override;
 
    void ComputeAdvection(double timeIntervalInSeconds) override;
 
    [[nodiscard]] ScalarField2Ptr GetFluidSDF() const override;
 
 private:
    void Reinitialize(double currentCFL);
 
    void ExtrapolateVelocityToAir(double currentCFL);
 
    void AddVolume(double volDiff);
 
    size_t m_signedDistanceFieldId;
    LevelSetSolver2Ptr m_levelSetSolver;
    double m_minReinitializeDistance = 10.0;
    bool m_isGlobalCompensationEnabled = false;
    double m_lastKnownVolume = 0.0;
};
 
using LevelSetLiquidSolver2Ptr = std::shared_ptr<LevelSetLiquidSolver2>;
 
class LevelSetLiquidSolver2::Builder final
    : public GridFluidSolverBuilderBase2<Builder>
{
 public:
    [[nodiscard]] LevelSetLiquidSolver2 Build() const;
 
    [[nodiscard]] LevelSetLiquidSolver2Ptr MakeShared() const;
};
}  // namespace CubbyFlow
 
#endif