Bug fixes to make things work in parallel

Author: vdweide

Common/src/fem/fem_geometry_structure.cpp

@@ -223,6 +223,7 @@ CMeshFEM::CMeshFEM(CGeometry* geometry, CConfig* config) {
   blasFunctions = new CBlasStructure;
 
   /*--- Define the linear partitioning of the elements. ---*/
+  Global_nElem = geometry->GetGlobal_nElem();
   CLinearPartitioner elemPartitioner(Global_nElem, 0);
 
   /*--- The new FEM mesh class has the same problem dimension/zone. ---*/
@@ -671,9 +672,8 @@ CMeshFEM::CMeshFEM(CGeometry* geometry, CConfig* config) {
   /* Determine the number of elements per rank of the originally partitioned grid
      stored in cumulative storage format. */
   vector<unsigned long> nElemPerRankOr(size + 1);
-
-  for (int i = 0; i < size; ++i) nElemPerRankOr[i] = elemPartitioner.GetFirstIndexOnRank(i);
-  nElemPerRankOr[size] = Global_nElem;
+  for (int i = 0; i <= size; ++i)
+    nElemPerRankOr[i] = elemPartitioner.GetCumulativeSizeBeforeRank(i);
 
   /* Determine to which ranks I have to send messages to find out the information
      of the halos stored on this rank. */

Common/src/fem/geometry_structure_fem_part.cpp

@@ -1162,8 +1162,8 @@ void CPhysicalGeometry::SetColorFEMGrid_Parallel(CConfig* config) {
     /*--- Determine the array, which stores the distribution of the graph nodes
           over the ranks.     ---*/
     vector<idx_t> vtxdist(size + 1);
-    vtxdist[0] = 0;
-    for (int i = 0; i < size; ++i) vtxdist[i + 1] = (idx_t)end_node[i];
+    for (int i = 0; i <= size; ++i)
+      vtxdist[i] = static_cast<idx_t>(elemPartitioner.GetCumulativeSizeBeforeRank(i));
 
     /* Create the array xadjPar, which contains the number of edges for each
        vertex of the graph in ParMETIS format. */