test/parallel.cpp

#include "DAPTA.hpp"

#include <set>
#include <fstream>
#include <iostream>
#include <iomanip>
#include <sstream>
#undef SEEK_SET
#undef SEEK_CUR
#undef SEEK_END
#include "mpi.h"
using namespace DAPTA;

typedef Mesh<Tri3MeshTraits> TriMesh;
typedef MeshParalleliser<Tri3MeshTraits> TriMeshParalleliser;

int main(int argc, char **argv) {
   // Initialize MPI
   MPI_Init(&argc, &argv);
   MPI_Status status;
   
   // Load in the complete mesh
   TriMesh *mesh = new TriMesh();
   mesh->ClearMesh();
   std::ifstream meshfile;
   meshfile.open(argv[1]);
   if(!meshfile) {
      std::cout << "ERROR READING FILE!" << std::endl;
      exit(1);
   }
   mesh->ReadFile(meshfile);
   meshfile.close();
   
   TriMeshParalleliser *mesh_paralleliser = new TriMeshParalleliser(MPI_COMM_WORLD, mesh);
   mesh_paralleliser->initialiseCommunication();
   
   // ********************
   // * BEGIN FEM CODE
   // ********************
   
   int myRank, numProcs;
   MPI_Comm_rank(MPI_COMM_WORLD, &myRank);
   MPI_Comm_size(MPI_COMM_WORLD, &numProcs);
   
   int DebugWait=1;
   //while(DebugWait);
   
   //if(myRank == 0) {
   // mesh->FractureFace(mesh->FindElementFromIndex(2),1);
   //}
   
   MPI_Barrier(MPI_COMM_WORLD);
   mesh_paralleliser->exchange();
   MPI_Barrier(MPI_COMM_WORLD);
   
   sleep(myRank*2);
   std::cout << "==============================================" << std::endl;
   std::cout << " client(" << myRank << ")" << std::endl;
   std::cout << "==============================================" << std::endl;
   TriMesh::EdgeMapIterator e_iter;
   for(e_iter=mesh->EdgeMapBegin(); e_iter!=mesh->EdgeMapEnd(); ++e_iter) {
      std::cout << " " << e_iter->first.vertices.at(0)->globalID << "\t";
      std::cout << e_iter->second->isClosed() << "\t";
      std::cout << e_iter->second->needSplit() << "\t";
      for(TriMesh::ElementConListType::iterator list_iter=e_iter->second->begin(); list_iter!=e_iter->second->end(); ++list_iter) {
         std::cout << (*list_iter).elem_globalID << "(" << (*list_iter).link_state << ")" << "->";
      }
      std::cout << std::endl;
   }
   std::cout << "==============================================" << std::endl;
   
   MPI_Barrier(MPI_COMM_WORLD);
   mesh->UpdateEdges();
   MPI_Barrier(MPI_COMM_WORLD);
   
   sleep(myRank*2);
   std::cout << "==============================================" << std::endl;
   std::cout << " client(" << myRank << ")" << std::endl;
   std::cout << "==============================================" << std::endl;
   std::cout << mesh->ExportFile() << std::endl;
   std::cout << "==============================================" << std::endl;
   
   std::stringstream filename_ss;
   filename_ss << "export-parallel-" << myRank << ".txt";
   
   std::string filename = filename_ss.str();
   
   std::ofstream outfile(filename.c_str(), std::ios::out);
   outfile << mesh->ExportFile();
   outfile.close();
   
   // ********************
   // * END FEM CODE
   // ********************
   
   // Shut Down MPI
   MPI_Finalize();
   
   return 0;
}

---