def getranks(mesh): element_rank = Function(FunctionSpace(mesh,'DG',0)) element_rank.dat.data[:] = mesh.comm.rank element_rank.rename('element_rank') vertex_rank = Function(FunctionSpace(mesh,'CG',1)) vertex_rank.dat.data[:] = mesh.comm.rank vertex_rank.rename('vertex_rank') return (element_rank,vertex_rank)