OilfieldToolsNet/LBPM
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

code clean up

Browse Source
This commit is contained in:
Rex Zhe Li 2021-02-01 00:39:21 -05:00
parent 2ed34592d0
commit 7cdc358af2
1 changed files with 51 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

113
tests/lbpm_freelee_simulator.cpp
View File

@ -1,81 +1,70 @@
#include <exception>
#include <fstream>
#include <iostream>
#include <stdexcept>
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <iostream>
#include <exception>
#include <stdexcept>
#include <fstream>
#include "models/FreeLeeModel.h"
#include "common/Utilities.h" #include "common/Utilities.h"
#include "models/FreeLeeModel.h"
//#define WRE_SURFACES //*******************************************************************
// Implementation of Free-Energy Two-Phase LBM (Lee model)
//*******************************************************************
/* int main( int argc, char **argv )
* Simulator for two-phase flow in porous media
* James E. McClure 2013-2014
*/
//*************************************************************************
// Implementation of Two-Phase Immiscible LBM using CUDA
//*************************************************************************
int main(int argc, char **argv)
{ {
// Initialize MPI // Initialize
Utilities::startup( argc, argv ); Utilities::startup( argc, argv );
Utilities::MPI comm( MPI_COMM_WORLD );
int rank = comm.getRank(); // Load the input database
int nprocs = comm.getSize(); auto db = std::make_shared<Database>( argv[1] );
// Load the input database { // Limit scope so variables that contain communicators will free before MPI_Finialize
auto db = std::make_shared<Database>( argv[1] );
// Initialize MPI and error handlers Utilities::MPI comm( MPI_COMM_WORLD );
auto multiple = db->getWithDefault<bool>( "MPI_THREAD_MULTIPLE", true ); int rank = comm.getRank();
//Utilities::startup( argc, argv, multiple ); int nprocs = comm.getSize();
//Utilities::MPI::changeProfileLevel( 1 );
{ // Limit scope so variables that contain communicators will free before MPI_Finialize 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();
if (rank == 0){ PROFILE_ENABLE(1);
printf("********************************************************\n"); //PROFILE_ENABLE_TRACE();
printf("Running Free Energy Lee LBM \n"); //PROFILE_ENABLE_MEMORY();
printf("********************************************************\n"); PROFILE_SYNCHRONIZE();
} PROFILE_START("Main");
// Initialize compute device Utilities::setErrorHandlers();
int device=ScaLBL_SetDevice(rank);
NULL_USE( device );
ScaLBL_DeviceBarrier();
comm.barrier();
PROFILE_ENABLE(1); auto filename = argv[1];
//PROFILE_ENABLE_TRACE(); ScaLBL_FreeLeeModel LeeModel( rank,nprocs,comm );
//PROFILE_ENABLE_MEMORY(); LeeModel.ReadParams( filename );
PROFILE_SYNCHRONIZE(); LeeModel.SetDomain();
PROFILE_START("Main"); LeeModel.ReadInput();
Utilities::setErrorHandlers(); LeeModel.Create();
LeeModel.Initialize();
LeeModel.Run();
LeeModel.WriteDebug();
auto filename = argv[1]; PROFILE_STOP("Main");
ScaLBL_FreeLeeModel LeeModel(rank,nprocs,comm); auto file = db->getWithDefault<std::string>( "TimerFile", "lbpm_freelee_simulator" );
LeeModel.ReadParams(filename); auto level = db->getWithDefault<int>( "TimerLevel", 1 );
LeeModel.SetDomain(); PROFILE_SAVE( file,level );
LeeModel.ReadInput(); // ****************************************************
LeeModel.Create(); // creating the model will create data structure to match the pore structure and allocate variables
LeeModel.Initialize(); // initializing the model will set initial conditions for variables
LeeModel.Run();
LeeModel.WriteDebug();
PROFILE_STOP("Main");
auto file = db->getWithDefault<std::string>( "TimerFile", "lbpm_freelee_simulator" );
auto level = db->getWithDefault<int>( "TimerLevel", 1 );
PROFILE_SAVE(file,level);
// ****************************************************
} // Limit scope so variables that contain communicators will free before MPI_Finialize } // Limit scope so variables that contain communicators will free before MPI_Finialize
Utilities::shutdown(); Utilities::shutdown();
return 0;
} }