106 lines
2.9 KiB
C++
106 lines
2.9 KiB
C++
/*
|
|
Implementation of Lee et al JCP 2016 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 "ProfilerApp.h"
|
|
#include "threadpool/thread_pool.h"
|
|
#include "common/ScaLBL.h"
|
|
#include "common/WideHalo.h"
|
|
|
|
#ifndef ScaLBL_FreeLeeModel_INC
|
|
#define ScaLBL_FreeLeeModel_INC
|
|
|
|
class ScaLBL_FreeLeeModel {
|
|
public:
|
|
ScaLBL_FreeLeeModel(int RANK, int NP, const Utilities::MPI &COMM);
|
|
~ScaLBL_FreeLeeModel();
|
|
|
|
// functions in they should be run
|
|
void ReadParams(string filename);
|
|
void ReadParams(std::shared_ptr<Database> db0);
|
|
void SetDomain();
|
|
void ReadInput();
|
|
void Create_TwoFluid();
|
|
void Initialize_TwoFluid();
|
|
double Run_TwoFluid(int returntime);
|
|
|
|
void WriteDebug_TwoFluid();
|
|
void Create_SingleFluid();
|
|
void Initialize_SingleFluid();
|
|
void Run_SingleFluid();
|
|
|
|
void WriteDebug_SingleFluid();
|
|
// test utilities
|
|
void Create_DummyPhase_MGTest();
|
|
void MGTest();
|
|
|
|
bool Restart, pBC;
|
|
int timestep, timestepMax;
|
|
int BoundaryCondition;
|
|
double tauA, tauB, rhoA, rhoB;
|
|
double tau, rho0; //only for single-fluid Lee model
|
|
double tauM; //relaxation time for phase field (or mass)
|
|
double W, gamma, kappa, beta;
|
|
double Fx, Fy, Fz, flux;
|
|
double din, dout, inletA, inletB, outletA, outletB;
|
|
|
|
int Nx, Ny, Nz, N, Np;
|
|
int Nxh, Nyh, Nzh, Nh; // extra halo width
|
|
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;
|
|
std::shared_ptr<ScaLBL_Communicator> ScaLBL_Comm_Regular;
|
|
std::shared_ptr<ScaLBLWideHalo_Communicator> ScaLBL_Comm_WideHalo;
|
|
|
|
// input database
|
|
std::shared_ptr<Database> db;
|
|
std::shared_ptr<Database> domain_db;
|
|
std::shared_ptr<Database> freelee_db;
|
|
std::shared_ptr<Database> analysis_db;
|
|
std::shared_ptr<Database> vis_db;
|
|
|
|
IntArray Map;
|
|
signed char *id;
|
|
int *NeighborList;
|
|
int *dvcMap;
|
|
double *gqbar, *hq;
|
|
double *mu_phi, *Den, *Phi;
|
|
double *ColorGrad;
|
|
double *Velocity;
|
|
double *Pressure;
|
|
|
|
void getPhase(DoubleArray &PhaseValues);
|
|
void getPotential(DoubleArray &PressureValues, DoubleArray &MuValues);
|
|
void getVelocity(DoubleArray &Vx, DoubleArray &Vy, DoubleArray &Vz);
|
|
void getData_RegularLayout(const double *data, DoubleArray ®data);
|
|
|
|
DoubleArray SignDist;
|
|
|
|
private:
|
|
Utilities::MPI comm;
|
|
|
|
int dist_mem_size;
|
|
int neighborSize;
|
|
// filenames
|
|
char LocalRankString[8];
|
|
char LocalRankFilename[40];
|
|
char LocalRestartFile[40];
|
|
|
|
//int rank,nprocs;
|
|
void LoadParams(std::shared_ptr<Database> db0);
|
|
void AssignComponentLabels_ChemPotential_ColorGrad();
|
|
};
|
|
#endif
|