navierstokes.h

Go to the documentation of this file.

 
#pragma once
 
#include <yaml-cpp/yaml.h>
 
#include <petibm/boundary.h>
#include <petibm/linsolver.h>
#include <petibm/mesh.h>
#include <petibm/operators.h>
#include <petibm/probes.h>
#include <petibm/solution.h>
#include <petibm/timeintegration.h>
 
class NavierStokesSolver
{
public:
    NavierStokesSolver() = default;
 
    NavierStokesSolver(const MPI_Comm &world,
                       const YAML::Node &node);
 
    ~NavierStokesSolver();
 
    PetscErrorCode destroy();
 
    PetscErrorCode init(const MPI_Comm &world, const YAML::Node &node);
 
    PetscErrorCode ioInitialData();
 
    PetscErrorCode advance();
 
    PetscErrorCode write();
 
    bool finished();
 
protected:
    MPI_Comm comm;
 
    PetscMPIInt commSize;
 
    PetscMPIInt commRank;
 
    YAML::Node config;
 
    petibm::type::Mesh mesh;
 
    petibm::type::Boundary bc;
 
    petibm::type::Solution solution;
 
    petibm::type::TimeIntegration convCoeffs;
 
    petibm::type::TimeIntegration diffCoeffs;
 
    petibm::type::LinSolver vSolver;
 
    petibm::type::LinSolver pSolver;
 
    std::vector<petibm::type::Probe> probes;
 
    PetscReal dt;
 
    PetscInt ite;
 
    PetscReal t;
 
    PetscInt nstart;
 
    PetscInt nt;
 
    PetscInt nsave;
 
    PetscInt nrestart;
 
    PetscReal nu;
 
    Mat L;
 
    Mat LCorrection;
 
    Mat G;
 
    Mat D;
 
    Mat DCorrection;
 
    Mat N;
 
    Mat A;
 
    Mat BNG;
 
    Mat DBNG;
 
    Vec dP;
 
    Vec bc1;
 
    Vec rhs1;
 
    Vec rhs2;
 
    std::vector<Vec> conv;
 
    std::vector<Vec> diff;
 
    PetscBool isRefP;
 
    PetscLogStage stageInitialize;
 
    PetscLogStage stageRHSVelocity;
 
    PetscLogStage stageSolveVelocity;
 
    PetscLogStage stageRHSPoisson;
 
    PetscLogStage stageSolvePoisson;
 
    PetscLogStage stageUpdate;
 
    PetscLogStage stageWrite;
 
    PetscLogStage stageMonitor;
 
    PetscViewer solversViewer;
 
    virtual PetscErrorCode assembleRHSVelocity();
 
    virtual PetscErrorCode solveVelocity();
 
    virtual PetscErrorCode assembleRHSPoisson();
 
    virtual PetscErrorCode solvePoisson();
 
    virtual PetscErrorCode applyDivergenceFreeVelocity();
 
    virtual PetscErrorCode updatePressure();
 
    virtual PetscErrorCode createOperators();
 
    virtual PetscErrorCode createVectors();
 
    virtual PetscErrorCode setNullSpace();
 
    virtual PetscErrorCode createPetscViewerASCII(const std::string &filePath,
                                                  const PetscFileMode &mode,
                                                  PetscViewer &viewer);
 
    virtual PetscErrorCode writeSolutionHDF5(const std::string &filePath);
 
    virtual PetscErrorCode writeRestartDataHDF5(const std::string &filePath);
 
    virtual PetscErrorCode readRestartDataHDF5(const std::string &filePath);
 
    virtual PetscErrorCode writeLinSolversInfo();
 
    virtual PetscErrorCode writeTimeHDF5(const PetscReal &t,
                                         const std::string &filePath);
 
    virtual PetscErrorCode readTimeHDF5(const std::string &filePath,
                                        PetscReal &t);
 
    virtual PetscErrorCode monitorProbes();
 
};  // NavierStokesSolver