96 lines
3.3 KiB
C++
96 lines
3.3 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"
|
|
#include "analysis/FreeEnergy.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();
|
|
|
|
FreeEnergyAnalyzer Analysis(LeeModel.Dm);
|
|
|
|
LeeModel.Initialize_TwoFluid();
|
|
|
|
/*** RUN MAIN TIMESTEPS HERE ************/
|
|
double MLUPS=0.0;
|
|
int timestep = 0;
|
|
int visualization_time = LeeModel.timestepMax;
|
|
if (LeeModel.vis_db->keyExists( "visualization_interval" )){
|
|
visualization_time = LeeModel.vis_db->getScalar<int>( "visualization_interval" );
|
|
timestep += visualization_time;
|
|
}
|
|
while (LeeModel.timestep < LeeModel.timestepMax){
|
|
MLUPS = LeeModel.Run_TwoFluid(timestep);
|
|
if (rank==0) printf("Lattice update rate (per MPI process)= %f MLUPS \n", MLUPS);
|
|
Analysis.WriteVis(LeeModel,LeeModel.db, timestep);
|
|
timestep += visualization_time;
|
|
}
|
|
LeeModel.WriteDebug_TwoFluid();
|
|
if (rank==0) printf("********************************************************\n");
|
|
if (rank==0) printf("Lattice update rate (per core)= %f MLUPS \n", MLUPS);
|
|
MLUPS *= nprocs;
|
|
if (rank==0) printf("Lattice update rate (total)= %f MLUPS \n", MLUPS);
|
|
if (rank==0) printf("********************************************************\n");
|
|
// ************************************************************************
|
|
|
|
PROFILE_STOP("Main");
|
|
auto file = db->getWithDefault<std::string>( "TimerFile", "lbpm_freelee_simulator" );
|
|
auto level = db->getWithDefault<int>( "TimerLevel", 1 );
|
|
NULL_USE(level);
|
|
PROFILE_SAVE( file,level );
|
|
// ****************************************************
|
|
|
|
|
|
} // Limit scope so variables that contain communicators will free before MPI_Finialize
|
|
|
|
Utilities::shutdown();
|
|
return 0;
|
|
}
|