grid_single_phase_pressure_solver3.h

Go to the documentation of this file.

// Copyright (c) 2018 Doyub Kim
//
// I am making my contributions/submissions to this project solely in my
// personal capacity and am not conveying any rights to any intellectual
// property of any third parties.
 
#ifndef INCLUDE_JET_GRID_SINGLE_PHASE_PRESSURE_SOLVER3_H_
#define INCLUDE_JET_GRID_SINGLE_PHASE_PRESSURE_SOLVER3_H_
 
#include <jet/fdm_linear_system_solver3.h>
#include <jet/fdm_mg_linear_system3.h>
#include <jet/fdm_mg_solver3.h>
#include <jet/grid_boundary_condition_solver3.h>
#include <jet/grid_pressure_solver3.h>
 
#include <memory>
 
namespace jet {
 
class GridSinglePhasePressureSolver3 : public GridPressureSolver3 {
 public:
    GridSinglePhasePressureSolver3();
 
    virtual ~GridSinglePhasePressureSolver3();
 
    void solve(
        const FaceCenteredGrid3& input, double timeIntervalInSeconds,
        FaceCenteredGrid3* output,
        const ScalarField3& boundarySdf = ConstantScalarField3(kMaxD),
        const VectorField3& boundaryVelocity = ConstantVectorField3({0, 0, 0}),
        const ScalarField3& fluidSdf = ConstantScalarField3(-kMaxD),
        bool useCompressed = false) override;
 
    GridBoundaryConditionSolver3Ptr suggestedBoundaryConditionSolver()
        const override;
 
    const FdmLinearSystemSolver3Ptr& linearSystemSolver() const;
 
    void setLinearSystemSolver(const FdmLinearSystemSolver3Ptr& solver);
 
    const FdmVector3& pressure() const;
 
 private:
    FdmLinearSystem3 _system;
    FdmCompressedLinearSystem3 _compSystem;
    FdmLinearSystemSolver3Ptr _systemSolver;
 
    FdmMgLinearSystem3 _mgSystem;
    FdmMgSolver3Ptr _mgSystemSolver;
 
    std::vector<Array3<char>> _markers;
 
    void buildMarkers(
        const Size3& size,
        const std::function<Vector3D(size_t, size_t, size_t)>& pos,
        const ScalarField3& boundarySdf, const ScalarField3& fluidSdf);
 
    void decompressSolution();
 
    virtual void buildSystem(const FaceCenteredGrid3& input,
                             bool useCompressed);
 
    virtual void applyPressureGradient(const FaceCenteredGrid3& input,
                                       FaceCenteredGrid3* output);
};
 
typedef std::shared_ptr<GridSinglePhasePressureSolver3>
    GridSinglePhasePressureSolver3Ptr;
 
}  // namespace jet
 
#endif  // INCLUDE_JET_GRID_SINGLE_PHASE_PRESSURE_SOLVER3_H_