pressure_poisson_equation.hpp

Go to the documentation of this file.

#pragma once
 
#include "../particles.hpp"
#include "dirichlet_boundary_condition.hpp"
 
#include <Eigen/Sparse>
#include <vector>
 
namespace PressureCalculator {
 
class PressurePoissonEquation {
public:
    PressurePoissonEquation() = default;
 
    PressurePoissonEquation(
        int dimension,
        double dt,
        double relaxationCoefficient,
        double compressibility,
        double n0_forNumberDensity,
        double n0_forLaplacian,
        double lambda0,
        double reForLaplacian,
        double reForNumberDensity
    );
 
    void setup(const Particles& particles, const DirichletBoundaryCondition& dirichletBoundaryCondition);
 
    std::vector<double> solve();
 
private:
    int dimension;
    double dt;
    double relaxationCoefficient;
    double compressibility;
    double n0_forNumberDensity;
    double n0_forLaplacian;
    double lambda0;
    double reForLaplacian;
    double reForNumberDensity;
    size_t particlesCount;
 
    std::vector<Eigen::Triplet<double>> matrixTriplets; 
    Eigen::SparseMatrix<double, Eigen::RowMajor>
        coefficientMatrix;      
    Eigen::VectorXd sourceTerm; 
 
    void resetEquation();
    void setSourceTerm(const Particles& particles, const DirichletBoundaryCondition& dirichletBoundaryCondition);
    void setMatrixTriplets(const Particles& particles, const DirichletBoundaryCondition& dirichletBoundaryCondition);
};
 
} // namespace PressureCalculator