grid_single_phase_pressure_solver2.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_SOLVER2_H_
#define INCLUDE_JET_GRID_SINGLE_PHASE_PRESSURE_SOLVER2_H_
 
#include <jet/fdm_linear_system_solver2.h>
#include <jet/fdm_mg_linear_system2.h>
#include <jet/fdm_mg_solver2.h>
#include <jet/grid_boundary_condition_solver2.h>
#include <jet/grid_pressure_solver2.h>
 
#include <memory>
 
namespace jet {
 
class GridSinglePhasePressureSolver2 : public GridPressureSolver2 {
 public:
    GridSinglePhasePressureSolver2();
 
    virtual ~GridSinglePhasePressureSolver2();
 
    void solve(const FaceCenteredGrid2& input, double timeIntervalInSeconds,
               FaceCenteredGrid2* output,
               const ScalarField2& boundarySdf = ConstantScalarField2(kMaxD),
               const VectorField2& boundaryVelocity = ConstantVectorField2({0,
                                                                            0}),
               const ScalarField2& fluidSdf = ConstantScalarField2(-kMaxD),
               bool useCompressed = false) override;
 
    GridBoundaryConditionSolver2Ptr suggestedBoundaryConditionSolver()
        const override;
 
    const FdmLinearSystemSolver2Ptr& linearSystemSolver() const;
 
    void setLinearSystemSolver(const FdmLinearSystemSolver2Ptr& solver);
 
    const FdmVector2& pressure() const;
 
 private:
    FdmLinearSystem2 _system;
    FdmCompressedLinearSystem2 _compSystem;
    FdmLinearSystemSolver2Ptr _systemSolver;
 
    FdmMgLinearSystem2 _mgSystem;
    FdmMgSolver2Ptr _mgSystemSolver;
 
    std::vector<Array2<char>> _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);
};
 
typedef std::shared_ptr<GridSinglePhasePressureSolver2>
    GridSinglePhasePressureSolver2Ptr;
 
}  // namespace jet
 
#endif  // INCLUDE_JET_GRID_SINGLE_PHASE_PRESSURE_SOLVER2_H_