implicit.cpp

Go to the documentation of this file.

#include "implicit.hpp"
 
#include "../refvalues.hpp"
#include "../weight.hpp"
 
#include <iostream>
#include <queue>
 
using PressureCalculator::Implicit;
using std::cerr;
using std::endl;
 
Implicit::Implicit(
    int dimension,
    double particleDistance,
    double reForNumberDensity,
    double reForLaplacian,
    double dt,
    double compressibility,
    double relaxationCoefficient,
    std::unique_ptr<DirichletBoundaryConditionGenerator::Interface>&& dirichletBoundaryConditionGenerator
) {
    auto refValuesForNumberDensity =
        RefValues(dimension, particleDistance, reForNumberDensity, WeightType::SemiImplicit);
    auto refValuesForLaplacian = RefValues(dimension, particleDistance, reForLaplacian, WeightType::SemiImplicit);
    this->dirichletBoundaryConditionGenerator = std::move(dirichletBoundaryConditionGenerator);
    this->pressurePoissonEquation             = PressurePoissonEquation(
        dimension,
        dt,
        relaxationCoefficient,
        compressibility,
        refValuesForNumberDensity.n0,
        refValuesForLaplacian.n0,
        refValuesForLaplacian.lambda,
        reForLaplacian,
        reForNumberDensity
    );
}
 
std::vector<double> Implicit::calc(Particles& particles) {
    auto dirichletBoundaryCondition = dirichletBoundaryConditionGenerator->generate(particles);
    this->pressurePoissonEquation.setup(particles, dirichletBoundaryCondition);
    this->pressure = this->pressurePoissonEquation.solve();
    removeNegativePressure();
 
    return this->pressure;
}
 
Implicit::~Implicit() {
}
 
void Implicit::removeNegativePressure() {
#pragma omp parallel for
    for (auto& p : pressure) {
        if (p < 0) {
            p = 0;
        }
    }
}