GridFractionalSinglePhasePressureSolver3.h

Go to the documentation of this file.

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

#include <Core/FDM/FDMMGLinearSystem3.h>
#include <Core/Solver/FDM/FDMLinearSystemSolver3.h>
#include <Core/Solver/FDM/FDMMGSolver3.h>
#include <Core/Solver/Grid/GridPressureSolver3.h>

namespace CubbyFlow
{
    class GridFractionalSinglePhasePressureSolver3 : public GridPressureSolver3
    {
    public:
        GridFractionalSinglePhasePressureSolver3();

        virtual ~GridFractionalSinglePhasePressureSolver3();

        void Solve(
            const FaceCenteredGrid3& input,
            double timeIntervalInSeconds,
            FaceCenteredGrid3* output,
            const ScalarField3& boundarySDF = ConstantScalarField3(std::numeric_limits<double>::max()),
            const VectorField3& boundaryVelocity = ConstantVectorField3({ 0, 0, 0 }),
            const ScalarField3& fluidSDF = ConstantScalarField3(-std::numeric_limits<double>::max()),
            bool useCompressed = false) override;

        GridBoundaryConditionSolver3Ptr SuggestedBoundaryConditionSolver() const override;

        const FDMLinearSystemSolver3Ptr& GetLinearSystemSolver() const;

        void SetLinearSystemSolver(const FDMLinearSystemSolver3Ptr& solver);

        const FDMVector3& GetPressure() const;

    private:
        FDMLinearSystem3 m_system;
        FDMCompressedLinearSystem3 m_compSystem;
        FDMLinearSystemSolver3Ptr m_systemSolver;

        FDMMGLinearSystem3 m_mgSystem;
        FDMMGSolver3Ptr m_mgSystemSolver;

        std::vector<Array3<float>> m_uWeights;
        std::vector<Array3<float>> m_vWeights;
        std::vector<Array3<float>> m_wWeights;
        std::vector<Array3<float>> m_fluidSDF;

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

        void BuildWeights(
            const FaceCenteredGrid3& input,
            const ScalarField3& boundarySDF,
            const VectorField3& boundaryVelocity,
            const ScalarField3& fluidSDF);

        void DecompressSolution();

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

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

    using GridFractionalSinglePhasePressureSolver3Ptr = std::shared_ptr<GridFractionalSinglePhasePressureSolver3>;
}

#endif