CPU only;finish preliminary work on testing Poisson and Ion models and their coupling

2020-09-11 22:56:00 -04:00
parent 20c8cc9c3b
commit 11e9af54b6
18 changed files with 2453 additions and 226 deletions
--- a/tests/TestIonModel.cpp
+++ b/tests/TestIonModel.cpp
@@ -0,0 +1,89 @@
+#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/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 Test for Ion Transport \n");
+            printf("**************************************\n");
+        }
+        //PROFILE_ENABLE_TRACE();
+        //PROFILE_ENABLE_MEMORY();
+        PROFILE_SYNCHRONIZE();
+        PROFILE_START("Main");
+        Utilities::setErrorHandlers();
+
+        auto filename = argv[1];
+        ScaLBL_IonModel IonModel(rank,nprocs,comm);
+        ScaLBL_Multiphys_Controller Study(rank,nprocs,comm);//multiphysics controller coordinating multi-model coupling
+
+        // Load controller information
+        Study.ReadParams(filename);
+
+        // Initialize LB-Ion model
+        IonModel.ReadParams(filename,Study.num_iter_Ion);
+        IonModel.SetDomain();    
+        IonModel.ReadInput();    
+        IonModel.Create();       
+        IonModel.Initialize();   
+        IonModel.DummyFluidVelocity();
+        IonModel.DummyElectricField();
+
+
+        int timestep=0;
+        double error = 1.0;
+        vector<double>ci_avg_previous{0.0,0.0};//assuming 1:1 solution
+        while (timestep < Study.timestepMax && error > Study.tolerance){
+            
+            timestep++;
+            IonModel.Run(IonModel.FluidVelocityDummy,IonModel.ElectricFieldDummy); //solve for ion transport and electric potential
+            timestep++;//AA operations
+
+            if (timestep%Study.analysis_interval==0){
+                error = IonModel.CalIonDenConvergence(ci_avg_previous);
+            }
+        }
+        IonModel.getIonConcentration(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("TestIonModel",1);
+        // ****************************************************
+        MPI_Barrier(comm);
+    } // Limit scope so variables that contain communicators will free before MPI_Finialize
+    MPI_Comm_free(&comm);
+    MPI_Finalize();
+}
+
+