grid_fractional_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_FRACTIONAL_SINGLE_PHASE_PRESSURE_SOLVER2_H_
#define INCLUDE_JET_GRID_FRACTIONAL_SINGLE_PHASE_PRESSURE_SOLVER2_H_
 
#include <jet/cell_centered_scalar_grid2.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 <jet/vertex_centered_scalar_grid2.h>
 
#include <memory>
 
namespace jet {
 
class GridFractionalSinglePhasePressureSolver2 final
    : public GridPressureSolver2 {
 public:
    GridFractionalSinglePhasePressureSolver2();
 
    virtual ~GridFractionalSinglePhasePressureSolver2();
 
    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<float>> _uWeights;
    std::vector<Array2<float>> _vWeights;
    std::vector<Array2<float>> _fluidSdf;
 
    std::function<Vector2D(const Vector2D&)> _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);
 
    virtual void applyPressureGradient(const FaceCenteredGrid2& input,
                                       FaceCenteredGrid2* output);
};
 
typedef std::shared_ptr<GridFractionalSinglePhasePressureSolver2>
    GridFractionalSinglePhasePressureSolver2Ptr;
 
}  // namespace jet
 
#endif  // INCLUDE_JET_GRID_FRACTIONAL_SINGLE_PHASE_PRESSURE_SOLVER2_H_