test/parallel-old.cpp

#include "DAPTA.hpp"

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

#define LENGTH_EXPORT_FILE 0
#define STRING_EXPORT_FILE 1

typedef Mesh<Tri3MeshTraits> TriMesh;
TriMesh mesh;
RCBDecomposer<TriMesh::VertexType> rcb;
std::vector<TriMesh::VertexHandle> points;

int main(int argc, char **argv) {
   // Setup the mesh
   int          myrank, nprocs;
   MPI_Status   status;
   
   // Initialize MPI
   MPI_Init(&argc, &argv);

   // Get my rank id and number of MPI processes
   MPI_Comm_rank(MPI_COMM_WORLD, &myrank);
   MPI_Comm_size(MPI_COMM_WORLD, &nprocs);
   
   // Load in the complete mesh
   mesh.ClearMesh();
   std::ifstream meshfile;
   int num;
   
   meshfile.open(argv[1]);
   if(!meshfile) {
      std::cout << "ERROR READING FILE!" << std::endl;
      exit(1);
   }
   
   mesh.ReadFile(meshfile);
   meshfile.close();
   points = mesh.GetCentres();
   
   rcb.loadVertices(points);
   
   std::cout << "client(" << myrank << "): LOADING IN THE MESH FILE..." << std::endl;
   std::vector<int> myElems = rcb.decompose(nprocs, myrank);
   std::set<TriMesh::VertexHandle> part_vert;
   std::set<TriMesh::ElementHandle> part_elem;
   std::string export_file;
   
   for(std::vector<int>::iterator j=myElems.begin(); j!=myElems.end(); ++j) {
   TriMesh::ElementHandle this_elem = mesh.FindElementFromIndex(*j);
      part_elem.insert(this_elem);
      for(int k=0; k<TriMesh::ElementType::num_vertices; k++) {
         part_vert.insert(this_elem->vertices.at(k));
      }
   }
   
   myElems.clear();
   
   // Now load in the domain mesh
   std::cout << "client(" << myrank << "): READING MY DOMAIN..." << std::endl;
   export_file = mesh.ExportDomainFile(part_vert, part_elem);
   part_vert.clear();
   part_elem.clear();
   std::stringstream export_file_stream;
   export_file_stream.str(export_file);
   
   mesh.ClearMesh();
   mesh.ReadFile(export_file_stream);
   
   std::cout << "client(" << myrank << "): ================================================" << std::endl;
   for(TriMesh::VertexMapIterator v_iter=mesh.VertexMapBegin(); v_iter!=mesh.VertexMapEnd(); ++v_iter) {
      std::cout << std::setprecision(4);
      std::cout << "client(" << myrank << "): " << (*v_iter)->globalID << "\t";
      for(int i=0; i<TriMesh::VertexType::dimension; i++) {
         std::cout << (*v_iter)->coords.at(i) << "\t";
      }
      std::cout << (*v_iter) << std::endl;
   }
   std::cout << "client(" << myrank << "): " << std::endl;
   for(TriMesh::ElementMapIterator e_iter=mesh.ElementMapBegin(); e_iter!=mesh.ElementMapEnd(); ++e_iter) {
      std::cout << "client(" << myrank << "): " << (*e_iter)->globalID << "\t";
      for(int i=0; i<TriMesh::ElementType::num_vertices; i++) {
         std::cout << (*e_iter)->vertices.at(i)->globalID << "\t";
      }
      std::cout << (*e_iter) << std::endl;
   }
   std::cout << "client(" << myrank << "): ================================================" << std::endl;
   
   std::cout << "client(" << myrank << "): WAITING FOR ALL PROCESSES..." << std::endl;
   MPI_Barrier(MPI_COMM_WORLD);
   
   // ********************
   // * BEGIN FEM CODE
   // ********************
   
   
   
   // ********************
   // * END FEM CODE
   // ********************
   
   // Shut Down MPI
   MPI_Finalize();
   
   return 0;
}

---