102 lines
3.2 KiB
C++
102 lines
3.2 KiB
C++
#include <exception>
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <stdexcept>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <sys/stat.h>
|
|
|
|
#include "common/Utilities.h"
|
|
#include "models/FreeLeeModel.h"
|
|
|
|
//*******************************************************************
|
|
// Implementation of Free-Energy Two-Phase LBM (Lee model)
|
|
//*******************************************************************
|
|
|
|
int main( int argc, char **argv )
|
|
{
|
|
|
|
// Initialize
|
|
Utilities::startup( argc, argv );
|
|
|
|
// Load the input database
|
|
auto db = std::make_shared<Database>( argv[1] );
|
|
|
|
{ // Limit scope so variables that contain communicators will free before MPI_Finialize
|
|
|
|
Utilities::MPI comm( MPI_COMM_WORLD );
|
|
int rank = comm.getRank();
|
|
int nprocs = comm.getSize();
|
|
|
|
if (rank == 0){
|
|
printf("********************************************************\n");
|
|
printf("Running Free Energy Lee LBM \n");
|
|
printf("********************************************************\n");
|
|
}
|
|
// Initialize compute device
|
|
int device=ScaLBL_SetDevice(rank);
|
|
NULL_USE( device );
|
|
ScaLBL_DeviceBarrier();
|
|
comm.barrier();
|
|
|
|
PROFILE_ENABLE(1);
|
|
//PROFILE_ENABLE_TRACE();
|
|
//PROFILE_ENABLE_MEMORY();
|
|
PROFILE_SYNCHRONIZE();
|
|
PROFILE_START("Main");
|
|
Utilities::setErrorHandlers();
|
|
|
|
auto filename = argv[1];
|
|
ScaLBL_FreeLeeModel LeeModel( rank,nprocs,comm );
|
|
LeeModel.ReadParams( filename );
|
|
LeeModel.SetDomain();
|
|
LeeModel.ReadInput();
|
|
LeeModel.Create_TwoFluid();
|
|
LeeModel.Initialize_TwoFluid();
|
|
/* check neighbors */
|
|
|
|
|
|
/* Copy the initial density to test that global mass is conserved */
|
|
int Nx = LeeModel.Dm->Nx;
|
|
int Ny = LeeModel.Dm->Ny;
|
|
int Nz = LeeModel.Dm->Nz;
|
|
DoubleArray DensityInit(Nx,Ny,Nz);
|
|
LeeModel.ScaLBL_Comm->RegularLayout(LeeModel.Map,LeeModel.Den,DensityInit);
|
|
|
|
double MLUPS = LeeModel.Run_TwoFluid(LeeModel.timestepMax);
|
|
|
|
DoubleArray DensityFinal(Nx,Ny,Nz);
|
|
LeeModel.ScaLBL_Comm->RegularLayout(LeeModel.Map,LeeModel.Den,DensityFinal);
|
|
|
|
DoubleArray DensityChange(Nx,Ny,Nz);
|
|
double totalChange=0.0;
|
|
for (int k=1; k<Nz-1; k++){
|
|
for (int j=1; j<Ny-1; j++){
|
|
for (int i=1; i<Nx-1; i++){
|
|
double change = DensityFinal(i,j,k)-DensityInit(i,j,k);
|
|
DensityChange(i,j,k) = change;
|
|
totalChange += change;
|
|
}
|
|
}
|
|
}
|
|
printf("Density change, %f\n", totalChange);
|
|
|
|
FILE *OUTFILE;
|
|
char LocalRankFilename[40];
|
|
sprintf(LocalRankFilename,"DensityChange.%05i.raw",rank);
|
|
OUTFILE = fopen(LocalRankFilename,"wb");
|
|
fwrite(DensityChange.data(),8,Nx*Ny*Nz,OUTFILE);
|
|
fclose(OUTFILE);
|
|
|
|
//LeeModel.WriteDebug_TwoFluid();
|
|
|
|
PROFILE_STOP("Main");
|
|
// ****************************************************
|
|
|
|
|
|
} // Limit scope so variables that contain communicators will free before MPI_Finialize
|
|
|
|
Utilities::shutdown();
|
|
return 0;
|
|
}
|