def get_local_mesh(self): # Create the mesh X = np.mgrid[self.rank*self.Np[0]:(self.rank+1)*self.Np[0], :self.N[1]].astype(self.float) X[0] *= self.L[0]/self.N[0] X[1] *= self.L[1]/self.N[1] return X