GridFractionalSinglePhasePressureSolver2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: GridFractionalSinglePhasePressureSolver2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 2-D fractional single-phase pressure solver.
> Created Time: 2017/08/14
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_FRACTIONAL_SINGLE_PHASE_PRESSURE_SOLVER2_H
#define CUBBYFLOW_FRACTIONAL_SINGLE_PHASE_PRESSURE_SOLVER2_H

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

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

        virtual ~GridFractionalSinglePhasePressureSolver2();

        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;

        GridBoundaryConditionSolver2Ptr SuggestedBoundaryConditionSolver() const override;

        const FDMLinearSystemSolver2Ptr& GetLinearSystemSolver() const;

        void SetLinearSystemSolver(const FDMLinearSystemSolver2Ptr& solver);

        const FDMVector2& GetPressure() const;

    private:
        FDMLinearSystem2 m_system;
        FDMCompressedLinearSystem2 m_compSystem;
        FDMLinearSystemSolver2Ptr m_systemSolver;
        
        FDMMGLinearSystem2 m_mgSystem;
        FDMMGSolver2Ptr m_mgSystemSolver;
        
        std::vector<Array2<float>> m_uWeights;
        std::vector<Array2<float>> m_vWeights;
        std::vector<Array2<float>> m_fluidSDF;

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

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

        void DecompressSolution();

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

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

    using GridFractionalSinglePhasePressureSolver2Ptr = std::shared_ptr<GridFractionalSinglePhasePressureSolver2>;
}

#endif