GPU version is available now

2020-09-20 11:00:36 -04:00
parent 11e9af54b6
commit 039978cc81
12 changed files with 1858 additions and 1013 deletions
--- a/tests/lbpm_electrokinetic_SingleFluid_simulator.cpp
+++ b/tests/lbpm_electrokinetic_SingleFluid_simulator.cpp
@@ -0,0 +1,120 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/stat.h>
+#include <iostream>
+#include <exception>
+#include <stdexcept>
+#include <fstream>
+#include <math.h>
+
+#include "models/IonModel.h"
+#include "models/StokesModel.h"
+#include "models/PoissonSolver.h"
+#include "models/MultiPhysController.h"
+
+using namespace std;
+
+//***************************************************************************
+// Test lattice-Boltzmann Ion Model coupled with Poisson equation
+//***************************************************************************
+
+int main(int argc, char **argv)
+{
+    // Initialize MPI
+    int provided_thread_support = -1;
+    MPI_Init_thread(&argc,&argv,MPI_THREAD_MULTIPLE,&provided_thread_support);
+    MPI_Comm comm;
+    MPI_Comm_dup(MPI_COMM_WORLD,&comm);
+    int rank = comm_rank(comm);
+    int nprocs = comm_size(comm);
+    if ( rank==0 && provided_thread_support<MPI_THREAD_MULTIPLE ){
+      std::cerr << "Warning: Failed to start MPI with necessary thread support, thread support will be disabled" << std::endl;
+    }
+
+    // Limit scope so variables that contain communicators will free before MPI_Finialize
+    { 
+        if (rank == 0){
+            printf("********************************************************\n");
+            printf("Running LBPM electrokinetic single-fluid solver \n");
+            printf("********************************************************\n");
+        }
+        //PROFILE_ENABLE_TRACE();
+        //PROFILE_ENABLE_MEMORY();
+        PROFILE_SYNCHRONIZE();
+        PROFILE_START("Main");
+        Utilities::setErrorHandlers();
+
+        auto filename = argv[1];
+        ScaLBL_StokesModel StokesModel(rank,nprocs,comm);
+        ScaLBL_IonModel IonModel(rank,nprocs,comm);
+        ScaLBL_Poisson PoissonSolver(rank,nprocs,comm); 
+        ScaLBL_Multiphys_Controller Study(rank,nprocs,comm);//multiphysics controller coordinating multi-model coupling
+
+        // Load controller information
+        Study.ReadParams(filename);
+
+        // Load user input database files for Navier-Stokes and Ion solvers
+        StokesModel.ReadParams(filename);
+        IonModel.ReadParams(filename);
+
+        // Setup other model specific structures
+        StokesModel.SetDomain();    
+        StokesModel.ReadInput();    
+        StokesModel.Create();       // creating the model will create data structure to match the pore structure and allocate variables
+
+        IonModel.SetDomain();    
+        IonModel.ReadInput();    
+        IonModel.Create();       
+
+        // Get internal iteration number
+        StokesModel.timestepMax = Study.getStokesNumIter_PNP_coupling(StokesModel.time_conv,IonModel.time_conv);
+        StokesModel.Initialize();   // initializing the model will set initial conditions for variables
+
+        IonModel.timestepMax = Study.getIonNumIter_PNP_coupling(StokesModel.time_conv,IonModel.time_conv);
+        IonModel.Initialize();   
+
+        // Initialize LB-Poisson model
+        PoissonSolver.ReadParams(filename);
+        PoissonSolver.SetDomain();    
+        PoissonSolver.ReadInput();    
+        PoissonSolver.Create();       
+        PoissonSolver.Initialize();   
+
+
+        int timestep=0;
+        while (timestep < Study.timestepMax){
+            
+            timestep++;
+            PoissonSolver.Run(IonModel.ChargeDensity);//solve Poisson equtaion to get steady-state electrical potental
+            StokesModel.Run_Lite(IonModel.ChargeDensity, PoissonSolver.ElectricField);// Solve the N-S equations to get velocity
+            IonModel.Run(StokesModel.Velocity,PoissonSolver.ElectricField); //solve for ion transport and electric potential
+            
+            timestep++;//AA operations
+
+            if (timestep%Study.visualization_interval==0){
+                PoissonSolver.getElectricPotential(timestep);
+                PoissonSolver.getElectricField(timestep);
+                IonModel.getIonConcentration(timestep);
+                StokesModel.getVelocity(timestep);
+            }
+        }
+
+        if (rank==0) printf("Save simulation raw data at maximum timestep\n");
+        PoissonSolver.getElectricPotential(timestep);
+        PoissonSolver.getElectricField(timestep);
+        IonModel.getIonConcentration(timestep);
+        StokesModel.getVelocity(timestep);
+
+        if (rank==0) printf("Maximum timestep is reached and the simulation is completed\n");
+        if (rank==0) printf("*************************************************************\n");
+
+        PROFILE_STOP("Main");
+        PROFILE_SAVE("lbpm_electrokinetic_SingleFluid_simulator",1);
+        // ****************************************************
+        MPI_Barrier(comm);
+    } // Limit scope so variables that contain communicators will free before MPI_Finialize
+    MPI_Comm_free(&comm);
+    MPI_Finalize();
+}
+
+