LBPM/models/GreyscaleModel.h

/*
  Copyright 2020 Equinor ASA                                                          
  Copyright Equnior ASA
                                                                                      
  This file is part of the Open Porous Media project (OPM).                           
  OPM is free software: you can redistribute it and/or modify                         
  it under the terms of the GNU General Public License as published by                
  the Free Software Foundation, either version 3 of the License, or                   
  (at your option) any later version.                                                 
  OPM is distributed in the hope that it will be useful,                              
  but WITHOUT ANY WARRANTY; without even the implied warranty of                      
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the                       
  GNU General Public License for more details.                                        
  You should have received a copy of the GNU General Public License                   
  along with OPM.  If not, see <http://www.gnu.org/licenses/>.       

 greyscale lattice boltzmann model
 */
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <iostream>
#include <exception>
#include <stdexcept>
#include <fstream>

#include "common/Communication.h"
#include "common/MPI.h"
#include "common/Database.h"
#include "common/ScaLBL.h"
#include "ProfilerApp.h"
#include "threadpool/thread_pool.h"

class ScaLBL_GreyscaleModel {
public:
    ScaLBL_GreyscaleModel(int RANK, int NP, const Utilities::MPI &COMM);
    ~ScaLBL_GreyscaleModel();

    // functions in they should be run
    void ReadParams(string filename);
    void ReadParams(std::shared_ptr<Database> db0);
    void SetDomain();
    void ReadInput();
    void Create();
    void Initialize();
    void Run();
    void WriteDebug();
    void VelocityField();

    bool Restart, pBC;
    int timestep, timestepMax;
    int BoundaryCondition;
    int CollisionType;
    double tau;
    double tau_eff;
    double Den; //constant density
    double tolerance;
    double Fx, Fy, Fz, flux;
    double din, dout;
    double dp; //solid particle diameter, unit in voxel
    double GreyPorosity;

    int Nx, Ny, Nz, N, Np;
    int rank, nprocx, nprocy, nprocz, nprocs;
    double Lx, Ly, Lz;

    std::shared_ptr<Domain> Dm;   // this domain is for analysis
    std::shared_ptr<Domain> Mask; // this domain is for lbm
    std::shared_ptr<ScaLBL_Communicator> ScaLBL_Comm;

    // input database
    std::shared_ptr<Database> db;
    std::shared_ptr<Database> domain_db;
    std::shared_ptr<Database> greyscale_db;
    std::shared_ptr<Database> analysis_db;
    std::shared_ptr<Database> vis_db;

    signed char *id;
    int *NeighborList;
    double *fq;
    double *Permeability; //grey voxel permeability
    double *Porosity;
    double *Velocity;
    double *Pressure_dvc;
    IntArray Map;
    DoubleArray SignDist;
    DoubleArray Velocity_x;
    DoubleArray Velocity_y;
    DoubleArray Velocity_z;
    DoubleArray PorosityMap;
    DoubleArray Pressure;

private:
    Utilities::MPI comm;

    size_t dist_mem_size;
    size_t neighborSize;
    // filenames
    char LocalRankString[8];
    char LocalRankFilename[40];
    char LocalRestartFile[40];

    void AssignComponentLabels(double *Porosity, double *Permeablity);
    void AssignComponentLabels(double *Porosity, double *Permeability,
                               const vector<std::string> &File_poro,
                               const vector<std::string> &File_perm);
};