GridSinglePhasePressureSolver2.h

Go to the documentation of this file.

/*************************************************************************
> File Name: GridSinglePhasePressureSolver2.h
> Project Name: CubbyFlow
> Author: Chan-Ho Chris Ohk
> Purpose: 2-D single-phase pressure solver.
> Created Time: 2017/08/14
> Copyright (c) 2018, Chan-Ho Chris Ohk
*************************************************************************/
#ifndef CUBBYFLOW_SINGLE_PHASE_PRESSURE_SOLVER2_H
#define CUBBYFLOW_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 GridSinglePhasePressureSolver2 : public GridPressureSolver2
    {
    public:
        GridSinglePhasePressureSolver2();

        virtual ~GridSinglePhasePressureSolver2();

        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<char>> m_markers;

        void BuildMarkers(
            const Size2& size,
            const std::function<Vector2D(size_t, size_t)>& pos,
            const ScalarField2& boundarySDF,
            const ScalarField2& fluidSDF);

        void DecompressSolution();

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

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

    using GridSinglePhasePressureSolver2Ptr = std::shared_ptr<GridSinglePhasePressureSolver2>;
}

#endif