set procGridLoc, procGridDim for cartesian communicator

include/ALL.hpp

@@ -204,6 +204,15 @@ public:
     if ( comm_type == MPI_CART) {
       communicator = comm;
       balancer->setCommunicator(communicator);
+      // set procGridLoc and procGridDim
+      const int dimension = balancer->getDimension();
+      int location[dimension];
+      int size[dimension];
+      int periodicity[dimension];
+      MPI_Cart_get(communicator, dimension, size, periodicity, location);
+      procGridLoc.assign(location, location+dimension);
+      procGridDim.assign(size, size+dimension);
+      procGridSet = true;
     } else {
       int rank;
       MPI_Comm_rank(comm, &rank);