1af0996ceSBarry Smith #include <petsc/private/dmmbimpl.h> /*I "petscdmmoab.h" I*/ 2af0996ceSBarry Smith #include <petsc/private/vecimpl.h> 351d15aeeSVijay Mahadevan 451d15aeeSVijay Mahadevan #include <petscdmmoab.h> 551d15aeeSVijay Mahadevan #include <MBTagConventions.hpp> 651d15aeeSVijay Mahadevan #include <moab/ReadUtilIface.hpp> 7ce27a4eeSVijay Mahadevan #include <moab/MergeMesh.hpp> 851d15aeeSVijay Mahadevan #include <moab/CN.hpp> 951d15aeeSVijay Mahadevan 10ce27a4eeSVijay Mahadevan typedef struct { 11ce27a4eeSVijay Mahadevan // options 1249d66b22SVijay Mahadevan PetscInt A,B,C,M,N,K,dim; 13ce27a4eeSVijay Mahadevan PetscInt blockSizeVertexXYZ[3]; // Number of element blocks per partition 14ce27a4eeSVijay Mahadevan PetscInt blockSizeElementXYZ[3]; 15ce27a4eeSVijay Mahadevan PetscReal xyzbounds[6]; // the physical size of the domain 16ce27a4eeSVijay Mahadevan bool newMergeMethod,keep_skins,simplex,adjEnts; 1751d15aeeSVijay Mahadevan 18ce27a4eeSVijay Mahadevan // compute params 19ce27a4eeSVijay Mahadevan PetscReal dx,dy,dz; 20ce27a4eeSVijay Mahadevan PetscInt NX,NY,NZ,nex,ney,nez; 21ce27a4eeSVijay Mahadevan PetscInt q,xstride,ystride,zstride; 22ce27a4eeSVijay Mahadevan PetscBool usrxyzgrid,usrprocgrid,usrrefgrid; 23755f3dfbSVijay Mahadevan PetscInt fraction,remainder,cumfraction; 24755f3dfbSVijay Mahadevan PetscLogEvent generateMesh, generateElements, generateVertices, parResolve; 25755f3dfbSVijay Mahadevan 26ce27a4eeSVijay Mahadevan } DMMoabMeshGeneratorCtx; 27ce27a4eeSVijay Mahadevan 28ce27a4eeSVijay Mahadevan 29ce27a4eeSVijay Mahadevan #undef __FUNCT__ 30ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMoab_SetTensorElementConnectivity_Private" 3149d66b22SVijay Mahadevan PetscInt DMMoab_SetTensorElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, PetscInt offset, PetscInt corner, std::vector<PetscInt>& subent_conn, moab::EntityHandle *connectivity) 3251d15aeeSVijay Mahadevan { 33ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 34e5136372SVijay Mahadevan case 1: 35ce27a4eeSVijay Mahadevan subent_conn.resize(2); 36ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBEDGE, 1, 0, subent_conn.data()); 37ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 38ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1; 39e5136372SVijay Mahadevan break; 40e5136372SVijay Mahadevan case 2: 41ce27a4eeSVijay Mahadevan subent_conn.resize(4); 42ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBQUAD, 2, 0, subent_conn.data()); 43ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 44ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1; 45ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner + 1 + genCtx.ystride; 46ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[3]] = corner + genCtx.ystride; 47e5136372SVijay Mahadevan break; 48e5136372SVijay Mahadevan case 3: 49e5136372SVijay Mahadevan default: 50ce27a4eeSVijay Mahadevan subent_conn.resize(8); 51ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBHEX, 3, 0, subent_conn.data()); 52ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 53ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1; 54ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner + 1 + genCtx.ystride; 55ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[3]] = corner + genCtx.ystride; 56ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[4]] = corner + genCtx.zstride; 57ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[5]] = corner + 1 + genCtx.zstride; 58ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[6]] = corner + 1 + genCtx.ystride + genCtx.zstride; 59ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[7]] = corner + genCtx.ystride + genCtx.zstride; 60e5136372SVijay Mahadevan break; 61e5136372SVijay Mahadevan } 62ce27a4eeSVijay Mahadevan return subent_conn.size(); 63e5136372SVijay Mahadevan } 6451d15aeeSVijay Mahadevan 65ce27a4eeSVijay Mahadevan 66ce27a4eeSVijay Mahadevan #undef __FUNCT__ 67ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMoab_SetSimplexElementConnectivity_Private" 6849d66b22SVijay Mahadevan PetscInt DMMoab_SetSimplexElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, PetscInt subelem, PetscInt offset, PetscInt corner, std::vector<PetscInt>& subent_conn, moab::EntityHandle *connectivity) 69c3dd00c7SVijay Mahadevan { 70755f3dfbSVijay Mahadevan PetscInt A, B, C, D, E, F, G, H, M; 71755f3dfbSVijay Mahadevan const PetscInt trigen_opts=1; /* 1 - Aligned diagonally to right, 2 - Aligned diagonally to left, 3 - 4 elements per quad */ 7249d66b22SVijay Mahadevan A = corner; 7349d66b22SVijay Mahadevan B = corner + 1; 74ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 75c3dd00c7SVijay Mahadevan case 1: 76ce27a4eeSVijay Mahadevan subent_conn.resize(2); /* only linear EDGE supported now */ 77ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBEDGE, 1, 0, subent_conn.data()); 7849d66b22SVijay Mahadevan connectivity[offset+subent_conn[0]] = A; 7949d66b22SVijay Mahadevan connectivity[offset+subent_conn[1]] = B; 80c3dd00c7SVijay Mahadevan break; 81c3dd00c7SVijay Mahadevan case 2: 8249d66b22SVijay Mahadevan C = corner + 1 + genCtx.ystride; 8349d66b22SVijay Mahadevan D = corner + genCtx.ystride; 84755f3dfbSVijay Mahadevan M = corner + 0.5; // technically -- need to modify vertex generation 85ce27a4eeSVijay Mahadevan subent_conn.resize(3); /* only linear TRI supported */ 86ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBTRI, 2, 0, subent_conn.data()); 87755f3dfbSVijay Mahadevan if (trigen_opts == 1) { 88ce27a4eeSVijay Mahadevan if (subelem) { /* 0 1 2 of a QUAD */ 8977a8c067SVijay Mahadevan connectivity[offset+subent_conn[0]] = B; 90755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[1]] = C; 9177a8c067SVijay Mahadevan connectivity[offset+subent_conn[2]] = A; 92c3dd00c7SVijay Mahadevan } 93ce27a4eeSVijay Mahadevan else { /* 2 3 0 of a QUAD */ 9477a8c067SVijay Mahadevan connectivity[offset+subent_conn[0]] = D; 9577a8c067SVijay Mahadevan connectivity[offset+subent_conn[1]] = A; 96755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[2]] = C; 97755f3dfbSVijay Mahadevan } 98755f3dfbSVijay Mahadevan } 99755f3dfbSVijay Mahadevan else if (trigen_opts == 2) { 100755f3dfbSVijay Mahadevan if (subelem) { /* 0 1 2 of a QUAD */ 101755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[0]] = A; 10277a8c067SVijay Mahadevan connectivity[offset+subent_conn[1]] = B; 10377a8c067SVijay Mahadevan connectivity[offset+subent_conn[2]] = D; 104755f3dfbSVijay Mahadevan } 105755f3dfbSVijay Mahadevan else { /* 2 3 0 of a QUAD */ 10677a8c067SVijay Mahadevan connectivity[offset+subent_conn[0]] = C; 10777a8c067SVijay Mahadevan connectivity[offset+subent_conn[1]] = D; 108755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[2]] = B; 109755f3dfbSVijay Mahadevan } 110755f3dfbSVijay Mahadevan } 111755f3dfbSVijay Mahadevan else { 112755f3dfbSVijay Mahadevan switch (subelem) { /* 0 1 2 of a QUAD */ 113755f3dfbSVijay Mahadevan case 0: 114755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[0]] = A; 115755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[1]] = B; 116755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[2]] = M; 117755f3dfbSVijay Mahadevan break; 118755f3dfbSVijay Mahadevan case 1: 119755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[0]] = B; 120755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[1]] = C; 121755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[2]] = M; 122755f3dfbSVijay Mahadevan break; 123755f3dfbSVijay Mahadevan case 2: 12449d66b22SVijay Mahadevan connectivity[offset+subent_conn[0]] = C; 12549d66b22SVijay Mahadevan connectivity[offset+subent_conn[1]] = D; 126755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[2]] = M; 127755f3dfbSVijay Mahadevan break; 128755f3dfbSVijay Mahadevan case 3: 129755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[0]] = D; 130755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[1]] = A; 131755f3dfbSVijay Mahadevan connectivity[offset+subent_conn[2]] = M; 132755f3dfbSVijay Mahadevan break; 133755f3dfbSVijay Mahadevan } 134c3dd00c7SVijay Mahadevan } 135c3dd00c7SVijay Mahadevan break; 136c3dd00c7SVijay Mahadevan case 3: 137c3dd00c7SVijay Mahadevan default: 13849d66b22SVijay Mahadevan C = corner + 1 + genCtx.ystride; 13949d66b22SVijay Mahadevan D = corner + genCtx.ystride; 14049d66b22SVijay Mahadevan E = corner + genCtx.zstride; 14149d66b22SVijay Mahadevan F = corner + 1 + genCtx.zstride; 14249d66b22SVijay Mahadevan G = corner + 1 + genCtx.ystride + genCtx.zstride; 14349d66b22SVijay Mahadevan H = corner + genCtx.ystride + genCtx.zstride; 144ce27a4eeSVijay Mahadevan subent_conn.resize(4); /* only linear TET supported */ 145ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBTET, 3, 0, subent_conn.data()); 146c3dd00c7SVijay Mahadevan switch(subelem) { 14749d66b22SVijay Mahadevan case 0: /* 4 3 7 6 of a HEX */ 14849d66b22SVijay Mahadevan connectivity[offset+subent_conn[0]] = E; 14949d66b22SVijay Mahadevan connectivity[offset+subent_conn[1]] = D; 15049d66b22SVijay Mahadevan connectivity[offset+subent_conn[2]] = H; 15149d66b22SVijay Mahadevan connectivity[offset+subent_conn[3]] = G; 152c3dd00c7SVijay Mahadevan break; 15349d66b22SVijay Mahadevan case 1: /* 0 1 2 5 of a HEX */ 15449d66b22SVijay Mahadevan connectivity[offset+subent_conn[0]] = A; 15549d66b22SVijay Mahadevan connectivity[offset+subent_conn[1]] = B; 15649d66b22SVijay Mahadevan connectivity[offset+subent_conn[2]] = C; 15749d66b22SVijay Mahadevan connectivity[offset+subent_conn[3]] = F; 158c3dd00c7SVijay Mahadevan break; 15949d66b22SVijay Mahadevan case 2: /* 0 3 4 5 of a HEX */ 16049d66b22SVijay Mahadevan connectivity[offset+subent_conn[0]] = A; 16149d66b22SVijay Mahadevan connectivity[offset+subent_conn[1]] = D; 16249d66b22SVijay Mahadevan connectivity[offset+subent_conn[2]] = E; 16349d66b22SVijay Mahadevan connectivity[offset+subent_conn[3]] = F; 164c3dd00c7SVijay Mahadevan break; 16549d66b22SVijay Mahadevan case 3: /* 2 6 3 5 of a HEX */ 16649d66b22SVijay Mahadevan connectivity[offset+subent_conn[0]] = C; 16749d66b22SVijay Mahadevan connectivity[offset+subent_conn[1]] = G; 16849d66b22SVijay Mahadevan connectivity[offset+subent_conn[2]] = D; 16949d66b22SVijay Mahadevan connectivity[offset+subent_conn[3]] = F; 170c3dd00c7SVijay Mahadevan break; 17149d66b22SVijay Mahadevan case 4: /* 0 2 3 5 of a HEX */ 17249d66b22SVijay Mahadevan connectivity[offset+subent_conn[0]] = A; 17349d66b22SVijay Mahadevan connectivity[offset+subent_conn[1]] = C; 17449d66b22SVijay Mahadevan connectivity[offset+subent_conn[2]] = D; 17549d66b22SVijay Mahadevan connectivity[offset+subent_conn[3]] = F; 17649d66b22SVijay Mahadevan break; 17749d66b22SVijay Mahadevan case 5: /* 3 6 4 5 of a HEX */ 17849d66b22SVijay Mahadevan connectivity[offset+subent_conn[0]] = D; 17949d66b22SVijay Mahadevan connectivity[offset+subent_conn[1]] = G; 18049d66b22SVijay Mahadevan connectivity[offset+subent_conn[2]] = E; 18149d66b22SVijay Mahadevan connectivity[offset+subent_conn[3]] = F; 182c3dd00c7SVijay Mahadevan break; 183c3dd00c7SVijay Mahadevan } 184c3dd00c7SVijay Mahadevan break; 185c3dd00c7SVijay Mahadevan } 186ce27a4eeSVijay Mahadevan return subent_conn.size(); 187c3dd00c7SVijay Mahadevan } 188c3dd00c7SVijay Mahadevan 189ce27a4eeSVijay Mahadevan 190ce27a4eeSVijay Mahadevan #undef __FUNCT__ 191ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMoab_SetElementConnectivity_Private" 19249d66b22SVijay Mahadevan std::pair<PetscInt,PetscInt> DMMoab_SetElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, PetscInt offset, PetscInt corner, moab::EntityHandle *connectivity) 193c3dd00c7SVijay Mahadevan { 19449d66b22SVijay Mahadevan PetscInt vcount=0; 19549d66b22SVijay Mahadevan PetscInt simplices_per_tensor[4] = {0,1,2,6}; 19649d66b22SVijay Mahadevan std::vector<PetscInt> subent_conn; /* only linear edge, tri, tet supported now */ 197ce27a4eeSVijay Mahadevan subent_conn.reserve(27); 19849d66b22SVijay Mahadevan PetscInt m,subelem; 199ce27a4eeSVijay Mahadevan if (genCtx.simplex) { 20049d66b22SVijay Mahadevan subelem=simplices_per_tensor[genCtx.dim]; 201ce27a4eeSVijay Mahadevan for (m=0; m<subelem; m++) { 202ce27a4eeSVijay Mahadevan vcount=DMMoab_SetSimplexElementConnectivity_Private(genCtx, m, offset, corner, subent_conn, connectivity); 203ce27a4eeSVijay Mahadevan offset+=vcount; 204ce27a4eeSVijay Mahadevan } 205c3dd00c7SVijay Mahadevan } 206c3dd00c7SVijay Mahadevan else { 207c3dd00c7SVijay Mahadevan subelem=1; 208ce27a4eeSVijay Mahadevan vcount=DMMoab_SetTensorElementConnectivity_Private(genCtx, offset, corner, subent_conn, connectivity); 209c3dd00c7SVijay Mahadevan } 21049d66b22SVijay Mahadevan return std::pair<PetscInt,PetscInt>(vcount*subelem,subelem); 211c3dd00c7SVijay Mahadevan } 212c3dd00c7SVijay Mahadevan 213c3dd00c7SVijay Mahadevan 214ce27a4eeSVijay Mahadevan #undef __FUNCT__ 215ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMoab_GenerateVertices_Private" 21649d66b22SVijay Mahadevan PetscErrorCode DMMoab_GenerateVertices_Private(moab::Interface *mbImpl, moab::ReadUtilIface *iface, DMMoabMeshGeneratorCtx& genCtx, PetscInt m, PetscInt n, PetscInt k, 21749d66b22SVijay Mahadevan PetscInt a, PetscInt b, PetscInt c, moab::Tag& global_id_tag, moab::EntityHandle& startv, moab::Range& uverts) 218ce27a4eeSVijay Mahadevan { 21949d66b22SVijay Mahadevan PetscInt x,y,z,ix,nnodes; 220755f3dfbSVijay Mahadevan PetscErrorCode ierr; 22149d66b22SVijay Mahadevan PetscInt ii,jj,kk; 22249d66b22SVijay Mahadevan std::vector<PetscReal*> arrays; 223755f3dfbSVijay Mahadevan PetscInt* gids; 224ce27a4eeSVijay Mahadevan moab::ErrorCode merr; 225ce27a4eeSVijay Mahadevan 226ce27a4eeSVijay Mahadevan PetscFunctionBegin; 227ce27a4eeSVijay Mahadevan /* we will generate (q*block+1)^3 vertices, and block^3 hexas; q is 1 for linear, 2 for quadratic 228ce27a4eeSVijay Mahadevan * the global id of the vertices will come from m, n, k, a, b, c 229ce27a4eeSVijay Mahadevan * x will vary from m*A*q*block + a*q*block to m*A*q*block+(a+1)*q*block etc. 230ce27a4eeSVijay Mahadevan */ 231ce27a4eeSVijay Mahadevan nnodes = genCtx.blockSizeVertexXYZ[0]*(genCtx.dim>1? genCtx.blockSizeVertexXYZ[1]*(genCtx.dim>2 ? genCtx.blockSizeVertexXYZ[2]:1) :1); 232755f3dfbSVijay Mahadevan ierr = PetscMalloc1(nnodes,&gids);CHKERRQ(ierr); 233ce27a4eeSVijay Mahadevan 234ce27a4eeSVijay Mahadevan merr = iface->get_node_coords(3, nnodes, 0, startv, arrays);MBERR("Can't get node coords.", merr); 235ce27a4eeSVijay Mahadevan 236ce27a4eeSVijay Mahadevan /* will start with the lower corner: */ 237755f3dfbSVijay Mahadevan //x = ( m * genCtx.A + a ) * genCtx.q * genCtx.blockSizeElementXYZ[0]; 238755f3dfbSVijay Mahadevan //y = ( n * genCtx.B + b ) * genCtx.q * genCtx.blockSizeElementXYZ[1]; 239755f3dfbSVijay Mahadevan //z = ( k * genCtx.C + c ) * genCtx.q * genCtx.blockSizeElementXYZ[2]; 240755f3dfbSVijay Mahadevan 241755f3dfbSVijay Mahadevan x = ( m * genCtx.A + a ) * genCtx.q; 242755f3dfbSVijay Mahadevan y = ( n * genCtx.B + b ) * genCtx.q; 243755f3dfbSVijay Mahadevan z = ( k * genCtx.C + c ) * genCtx.q; 244755f3dfbSVijay Mahadevan PetscInfo3(NULL, "Starting offset for coordinates := %d, %d, %d\n",x,y,z); 245ce27a4eeSVijay Mahadevan ix = 0; 246ce27a4eeSVijay Mahadevan moab::Range verts(startv, startv + nnodes - 1); 247ce27a4eeSVijay Mahadevan for (kk = 0; kk < (genCtx.dim>2?genCtx.blockSizeVertexXYZ[2]:1); kk++) { 248ce27a4eeSVijay Mahadevan for (jj = 0; jj < (genCtx.dim>1?genCtx.blockSizeVertexXYZ[1]:1); jj++) { 249ce27a4eeSVijay Mahadevan for (ii = 0; ii < genCtx.blockSizeVertexXYZ[0]; ii++,ix++) { 250ce27a4eeSVijay Mahadevan /* set coordinates for the vertices */ 251ce27a4eeSVijay Mahadevan arrays[0][ix] = (x + ii)*genCtx.dx + genCtx.xyzbounds[0]; 252ce27a4eeSVijay Mahadevan arrays[1][ix] = (y + jj)*genCtx.dy + genCtx.xyzbounds[2]; 253ce27a4eeSVijay Mahadevan arrays[2][ix] = (z + kk)*genCtx.dz + genCtx.xyzbounds[4]; 254755f3dfbSVijay Mahadevan PetscInfo3(NULL, "Creating vertex with coordinates := %f, %f, %f\n",arrays[0][ix],arrays[1][ix],arrays[2][ix]); 255ce27a4eeSVijay Mahadevan 256ce27a4eeSVijay Mahadevan /* If we want to set some tags on the vertices -> use the following entity handle definition: 257ce27a4eeSVijay Mahadevan moab::EntityHandle v = startv + ix; 258ce27a4eeSVijay Mahadevan */ 259ce27a4eeSVijay Mahadevan /* compute the global ID for vertex */ 260ce27a4eeSVijay Mahadevan gids[ix] = 1 + (x + ii) + (y + jj) * genCtx.NX + (z + kk) * (genCtx.NX * genCtx.NY); 261ce27a4eeSVijay Mahadevan } 262ce27a4eeSVijay Mahadevan } 263ce27a4eeSVijay Mahadevan } 264ce27a4eeSVijay Mahadevan /* set global ID data on vertices */ 265ce27a4eeSVijay Mahadevan mbImpl->tag_set_data(global_id_tag, verts, &gids[0]); 266ce27a4eeSVijay Mahadevan verts.swap(uverts); 267755f3dfbSVijay Mahadevan ierr = PetscFree(gids);CHKERRQ(ierr); 268ce27a4eeSVijay Mahadevan PetscFunctionReturn(0); 269ce27a4eeSVijay Mahadevan } 270ce27a4eeSVijay Mahadevan 271ce27a4eeSVijay Mahadevan #undef __FUNCT__ 272ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMoab_GenerateElements_Private" 27349d66b22SVijay Mahadevan PetscErrorCode DMMoab_GenerateElements_Private(moab::Interface* mbImpl, moab::ReadUtilIface* iface, DMMoabMeshGeneratorCtx& genCtx, PetscInt m, PetscInt n, PetscInt k, 27449d66b22SVijay Mahadevan PetscInt a, PetscInt b, PetscInt c, moab::Tag& global_id_tag, moab::EntityHandle startv, moab::Range& cells) 275ce27a4eeSVijay Mahadevan { 276ce27a4eeSVijay Mahadevan moab::ErrorCode merr; 277ce27a4eeSVijay Mahadevan PetscInt ix,ie,xe,ye,ze; 278ce27a4eeSVijay Mahadevan PetscInt ii,jj,kk,nvperelem; 27949d66b22SVijay Mahadevan PetscInt simplices_per_tensor[4] = {0,1,2,6}; 280ce27a4eeSVijay Mahadevan PetscInt ntensorelems = genCtx.blockSizeElementXYZ[0]*(genCtx.dim>1? genCtx.blockSizeElementXYZ[1]*(genCtx.dim>2 ? genCtx.blockSizeElementXYZ[2]:1) :1); /*pow(genCtx.blockSizeElement,genCtx.dim);*/ 281ce27a4eeSVijay Mahadevan PetscInt nelems = ntensorelems; 282ce27a4eeSVijay Mahadevan moab::EntityHandle starte; // connectivity 283ce27a4eeSVijay Mahadevan moab::EntityHandle* conn; 284ce27a4eeSVijay Mahadevan 285ce27a4eeSVijay Mahadevan PetscFunctionBegin; 286ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 287ce27a4eeSVijay Mahadevan case 1: 288ce27a4eeSVijay Mahadevan nvperelem = 2; 289ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 2, moab::MBEDGE, 0, starte, conn);MBERR("Can't get EDGE2 element connectivity.", merr); 290ce27a4eeSVijay Mahadevan break; 291ce27a4eeSVijay Mahadevan case 2: 292ce27a4eeSVijay Mahadevan if (genCtx.simplex) { 293ce27a4eeSVijay Mahadevan nvperelem = 3; 29449d66b22SVijay Mahadevan nelems = ntensorelems*simplices_per_tensor[genCtx.dim]; 295ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 3, moab::MBTRI, 0, starte, conn);MBERR("Can't get TRI3 element connectivity.", merr); 296ce27a4eeSVijay Mahadevan } 297ce27a4eeSVijay Mahadevan else { 298ce27a4eeSVijay Mahadevan nvperelem = 4; 299ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 4, moab::MBQUAD, 0, starte, conn);MBERR("Can't get QUAD4 element connectivity.", merr); 300ce27a4eeSVijay Mahadevan } 301ce27a4eeSVijay Mahadevan break; 302ce27a4eeSVijay Mahadevan case 3: 303ce27a4eeSVijay Mahadevan default: 304ce27a4eeSVijay Mahadevan if (genCtx.simplex) { 305ce27a4eeSVijay Mahadevan nvperelem = 4; 30649d66b22SVijay Mahadevan nelems = ntensorelems*simplices_per_tensor[genCtx.dim]; 307ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 4, moab::MBTET, 0, starte, conn);MBERR("Can't get TET4 element connectivity.", merr); 308ce27a4eeSVijay Mahadevan } 309ce27a4eeSVijay Mahadevan else { 310ce27a4eeSVijay Mahadevan nvperelem = 8; 311ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 8, moab::MBHEX, 0, starte, conn);MBERR("Can't get HEX8 element connectivity.", merr); 312ce27a4eeSVijay Mahadevan } 313ce27a4eeSVijay Mahadevan break; 314ce27a4eeSVijay Mahadevan } 315ce27a4eeSVijay Mahadevan 316ce27a4eeSVijay Mahadevan ix = ie = 0; // index now in the elements, for global ids 317ce27a4eeSVijay Mahadevan 318ce27a4eeSVijay Mahadevan /* create a temporary range to store local element handles */ 319ce27a4eeSVijay Mahadevan moab::Range tmp(starte, starte + nelems - 1); 32049d66b22SVijay Mahadevan std::vector<PetscInt> gids(nelems); 321ce27a4eeSVijay Mahadevan 322ce27a4eeSVijay Mahadevan /* identify the elements at the lower corner, for their global ids */ 323ce27a4eeSVijay Mahadevan xe = m * genCtx.A * genCtx.blockSizeElementXYZ[0] + a * genCtx.blockSizeElementXYZ[0]; 324ce27a4eeSVijay Mahadevan ye = (genCtx.dim > 1 ? n * genCtx.B * genCtx.blockSizeElementXYZ[1] + b * genCtx.blockSizeElementXYZ[1] : 0); 325ce27a4eeSVijay Mahadevan ze = (genCtx.dim > 2 ? k * genCtx.C * genCtx.blockSizeElementXYZ[2] + c * genCtx.blockSizeElementXYZ[2] : 0); 326ce27a4eeSVijay Mahadevan 327ce27a4eeSVijay Mahadevan /* create owned elements requested by genCtx */ 328ce27a4eeSVijay Mahadevan for (kk = 0; kk < (genCtx.dim>2?genCtx.blockSizeElementXYZ[2]:1); kk++) { 329ce27a4eeSVijay Mahadevan for (jj = 0; jj < (genCtx.dim>1?genCtx.blockSizeElementXYZ[1]:1); jj++) { 330ce27a4eeSVijay Mahadevan for (ii = 0; ii < genCtx.blockSizeElementXYZ[0]; ii++) { 331ce27a4eeSVijay Mahadevan 332ce27a4eeSVijay Mahadevan moab::EntityHandle corner = startv + genCtx.q * ii + genCtx.q * jj * genCtx.ystride + genCtx.q * kk * genCtx.zstride; 333ce27a4eeSVijay Mahadevan 33449d66b22SVijay Mahadevan std::pair<PetscInt,PetscInt> entoffset = DMMoab_SetElementConnectivity_Private(genCtx, ix, corner, conn); 335ce27a4eeSVijay Mahadevan 33649d66b22SVijay Mahadevan for (PetscInt j = 0; j < entoffset.second; j++) { 337ce27a4eeSVijay Mahadevan /* The entity handle for the particular element -> if we want to set some tags is 338ce27a4eeSVijay Mahadevan moab::EntityHandle eh = starte + ie + j; 339ce27a4eeSVijay Mahadevan */ 340ce27a4eeSVijay Mahadevan gids[ie+j] = 1 + ((xe + ii) + (ye + jj) * genCtx.nex + (ze + kk) * (genCtx.nex * genCtx.ney)); 341ce27a4eeSVijay Mahadevan //gids[ie+j] = ie + j + ((xe + ii) + (ye + jj) * genCtx.nex + (ze + kk) * (genCtx.nex * genCtx.ney)) ; 342ce27a4eeSVijay Mahadevan //gids[ie+j] = 1 + ie; 34349d66b22SVijay Mahadevan //ie++; 344ce27a4eeSVijay Mahadevan } 345ce27a4eeSVijay Mahadevan 346ce27a4eeSVijay Mahadevan ix += entoffset.first; 34749d66b22SVijay Mahadevan ie += entoffset.second; 348ce27a4eeSVijay Mahadevan } 349ce27a4eeSVijay Mahadevan } 350ce27a4eeSVijay Mahadevan } 351ce27a4eeSVijay Mahadevan if (genCtx.adjEnts) { /* we need to update adjacencies now, because some elements are new */ 352ce27a4eeSVijay Mahadevan merr = iface->update_adjacencies(starte, nelems, nvperelem, conn);MBERR("Can't update adjacencies", merr); 353ce27a4eeSVijay Mahadevan } 354ce27a4eeSVijay Mahadevan tmp.swap(cells); 355ce27a4eeSVijay Mahadevan merr = mbImpl->tag_set_data(global_id_tag, cells, &gids[0]);MBERR("Can't set global ids to elements.", merr); 356ce27a4eeSVijay Mahadevan PetscFunctionReturn(0); 357ce27a4eeSVijay Mahadevan } 358ce27a4eeSVijay Mahadevan 359ce27a4eeSVijay Mahadevan #undef __FUNCT__ 360ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMBUtil_InitializeOptions" 361755f3dfbSVijay Mahadevan PetscErrorCode DMMBUtil_InitializeOptions(DMMoabMeshGeneratorCtx& genCtx, PetscInt dim, PetscBool simplex, PetscInt rank, PetscInt nprocs, const PetscReal* bounds, PetscInt nelems) 362ce27a4eeSVijay Mahadevan { 363ce27a4eeSVijay Mahadevan PetscFunctionBegin; 364ce27a4eeSVijay Mahadevan /* Initialize all genCtx data */ 365ce27a4eeSVijay Mahadevan genCtx.dim=dim; 366ce27a4eeSVijay Mahadevan genCtx.simplex=simplex; 36749d66b22SVijay Mahadevan genCtx.newMergeMethod=genCtx.keep_skins=genCtx.adjEnts=true; 368ce27a4eeSVijay Mahadevan /* determine other global quantities for the mesh used for nodes increments */ 369ce27a4eeSVijay Mahadevan genCtx.q = 1; 370755f3dfbSVijay Mahadevan genCtx.fraction=genCtx.remainder=genCtx.cumfraction=0; 371ce27a4eeSVijay Mahadevan 372ce27a4eeSVijay Mahadevan if (!genCtx.usrxyzgrid) { /* not overridden by genCtx - assume nele equally and that genCtx wants a uniform cube mesh */ 373755f3dfbSVijay Mahadevan 374755f3dfbSVijay Mahadevan genCtx.fraction=nelems/nprocs; /* partition only by the largest dimension */ 375755f3dfbSVijay Mahadevan genCtx.remainder=nelems%nprocs; /* remainder after partition which gets evenly distributed by round-robin */ 376755f3dfbSVijay Mahadevan genCtx.cumfraction = (rank > 0 ? (genCtx.fraction) * (rank) + (rank-1 < genCtx.remainder ? rank : genCtx.remainder ) : 0); 377755f3dfbSVijay Mahadevan if(rank < genCtx.remainder) /* This process gets "fraction+1" elements */ 378755f3dfbSVijay Mahadevan genCtx.fraction++; 379755f3dfbSVijay Mahadevan 380*9daf19fdSVijay Mahadevan PetscInfo3(NULL,"Fraction = %D, Remainder = %D, Cumulative fraction = %D\n",genCtx.fraction,genCtx.remainder,genCtx.cumfraction); 381755f3dfbSVijay Mahadevan switch(genCtx.dim) { 382755f3dfbSVijay Mahadevan case 1: 383755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[0]=genCtx.fraction; 384755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[1]=1; 385755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[2]=1; 386755f3dfbSVijay Mahadevan break; 387755f3dfbSVijay Mahadevan case 2: 388755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[0]=nelems; 389755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[1]=genCtx.fraction; 390755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[2]=1; 391755f3dfbSVijay Mahadevan break; 392755f3dfbSVijay Mahadevan case 3: 393755f3dfbSVijay Mahadevan default: 394755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[0]=nelems; 395755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[1]=nelems; 396755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[2]=genCtx.fraction; 397755f3dfbSVijay Mahadevan break; 39849d66b22SVijay Mahadevan } 399ce27a4eeSVijay Mahadevan } 400ce27a4eeSVijay Mahadevan 401ce27a4eeSVijay Mahadevan /* partition only by the largest dimension */ 402ce27a4eeSVijay Mahadevan /* Total number of local elements := genCtx.blockSizeElementXYZ[0]*(genCtx.dim>1? genCtx.blockSizeElementXYZ[1]*(genCtx.dim>2 ? genCtx.blockSizeElementXYZ[2]:1) :1); */ 403ce27a4eeSVijay Mahadevan if (bounds) { 40449d66b22SVijay Mahadevan for (PetscInt i=0; i<6; i++) 405ce27a4eeSVijay Mahadevan genCtx.xyzbounds[i]=bounds[i]; 406ce27a4eeSVijay Mahadevan } 407ce27a4eeSVijay Mahadevan else { 408ce27a4eeSVijay Mahadevan genCtx.xyzbounds[0]=genCtx.xyzbounds[2]=genCtx.xyzbounds[4]=0.0; 409ce27a4eeSVijay Mahadevan genCtx.xyzbounds[1]=genCtx.xyzbounds[3]=genCtx.xyzbounds[5]=1.0; 410ce27a4eeSVijay Mahadevan } 411ce27a4eeSVijay Mahadevan 412ce27a4eeSVijay Mahadevan if (!genCtx.usrprocgrid) { 413ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 414ce27a4eeSVijay Mahadevan case 1: 415ce27a4eeSVijay Mahadevan genCtx.M=nprocs; 416ce27a4eeSVijay Mahadevan genCtx.N=genCtx.K=1; 417ce27a4eeSVijay Mahadevan break; 418ce27a4eeSVijay Mahadevan case 2: 419755f3dfbSVijay Mahadevan genCtx.M=1; 420755f3dfbSVijay Mahadevan genCtx.N=nprocs; 421ce27a4eeSVijay Mahadevan genCtx.K=1; 422ce27a4eeSVijay Mahadevan break; 423ce27a4eeSVijay Mahadevan default: 424755f3dfbSVijay Mahadevan genCtx.M=1; 425755f3dfbSVijay Mahadevan genCtx.N=1; 426755f3dfbSVijay Mahadevan // genCtx.M=genCtx.N=1; 427755f3dfbSVijay Mahadevan genCtx.K=nprocs; 428ce27a4eeSVijay Mahadevan break; 429ce27a4eeSVijay Mahadevan } 430ce27a4eeSVijay Mahadevan } 431ce27a4eeSVijay Mahadevan 432ce27a4eeSVijay Mahadevan if (!genCtx.usrrefgrid) { 433ce27a4eeSVijay Mahadevan genCtx.A=genCtx.B=genCtx.C=1; 434ce27a4eeSVijay Mahadevan } 435ce27a4eeSVijay Mahadevan 436ce27a4eeSVijay Mahadevan /* more default values */ 437ce27a4eeSVijay Mahadevan genCtx.nex=genCtx.ney=genCtx.nez=0; 438ce27a4eeSVijay Mahadevan genCtx.xstride=genCtx.ystride=genCtx.zstride=0; 439ce27a4eeSVijay Mahadevan genCtx.NX=genCtx.NY=genCtx.NZ=0; 440ce27a4eeSVijay Mahadevan genCtx.nex=genCtx.ney=genCtx.nez=0; 441ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0]=genCtx.blockSizeVertexXYZ[1]=genCtx.blockSizeVertexXYZ[2]=1; 442ce27a4eeSVijay Mahadevan 443ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 444ce27a4eeSVijay Mahadevan case 3: 445ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0] = genCtx.q*genCtx.blockSizeElementXYZ[0] + 1; 446ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[1] = genCtx.q*genCtx.blockSizeElementXYZ[1] + 1; 447ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[2] = genCtx.q*genCtx.blockSizeElementXYZ[2] + 1; 448ce27a4eeSVijay Mahadevan 449ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0]; /* number of elements in x direction, used for global id on element */ 450755f3dfbSVijay Mahadevan //genCtx.dx = (genCtx.xyzbounds[1]-genCtx.xyzbounds[0])/(genCtx.nex*genCtx.q); /* distance between 2 nodes in x direction */ 451755f3dfbSVijay Mahadevan genCtx.dx = (genCtx.xyzbounds[1]-genCtx.xyzbounds[0])/(nelems*genCtx.q); /* distance between 2 nodes in x direction */ 452ce27a4eeSVijay Mahadevan genCtx.NX = (genCtx.q * genCtx.nex + 1); 453ce27a4eeSVijay Mahadevan genCtx.xstride = 1; 454ce27a4eeSVijay Mahadevan genCtx.ney = genCtx.N * genCtx.B * genCtx.blockSizeElementXYZ[1]; /* number of elements in y direction .... */ 455755f3dfbSVijay Mahadevan genCtx.dy = (genCtx.xyzbounds[3]-genCtx.xyzbounds[2])/(nelems*genCtx.q); /* distance between 2 nodes in y direction */ 456ce27a4eeSVijay Mahadevan genCtx.NY = (genCtx.q * genCtx.ney + 1); 457ce27a4eeSVijay Mahadevan genCtx.ystride = genCtx.blockSizeVertexXYZ[0]; 458ce27a4eeSVijay Mahadevan genCtx.nez = genCtx.K * genCtx.C * genCtx.blockSizeElementXYZ[2]; /* number of elements in z direction .... */ 459755f3dfbSVijay Mahadevan //genCtx.dz = (genCtx.xyzbounds[5]-genCtx.xyzbounds[4])/(genCtx.nez*genCtx.q); /* distance between 2 nodes in z direction */ 460755f3dfbSVijay Mahadevan genCtx.dz = (genCtx.xyzbounds[5]-genCtx.xyzbounds[4])/(nelems*genCtx.q); /* distance between 2 nodes in z direction */ 461ce27a4eeSVijay Mahadevan genCtx.NZ = (genCtx.q * genCtx.nez + 1); 462ce27a4eeSVijay Mahadevan genCtx.zstride = genCtx.blockSizeVertexXYZ[0] * genCtx.blockSizeVertexXYZ[1]; 463ce27a4eeSVijay Mahadevan break; 464ce27a4eeSVijay Mahadevan case 2: 465ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0] = genCtx.q*genCtx.blockSizeElementXYZ[0] + 1; 466ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[1] = genCtx.q*genCtx.blockSizeElementXYZ[1] + 1; 467*9daf19fdSVijay Mahadevan genCtx.blockSizeVertexXYZ[2]=0; 468ce27a4eeSVijay Mahadevan 469ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0]; /* number of elements in x direction, used for global id on element */ 470755f3dfbSVijay Mahadevan // genCtx.dx = (genCtx.xyzbounds[1]-genCtx.xyzbounds[0])/(genCtx.nex*genCtx.q); /* distance between 2 nodes in x direction */ 471ce27a4eeSVijay Mahadevan genCtx.dx = (genCtx.xyzbounds[1]-genCtx.xyzbounds[0])/(genCtx.nex*genCtx.q); /* distance between 2 nodes in x direction */ 472ce27a4eeSVijay Mahadevan genCtx.NX = (genCtx.q * genCtx.nex + 1); 473ce27a4eeSVijay Mahadevan genCtx.xstride = 1; 474ce27a4eeSVijay Mahadevan genCtx.ney = genCtx.N * genCtx.B * genCtx.blockSizeElementXYZ[1]; /* number of elements in y direction .... */ 475755f3dfbSVijay Mahadevan // genCtx.dy = (genCtx.xyzbounds[3]-genCtx.xyzbounds[2])/(genCtx.ney*genCtx.q); /* distance between 2 nodes in y direction */ 476755f3dfbSVijay Mahadevan genCtx.dy = (genCtx.xyzbounds[3]-genCtx.xyzbounds[2])/(nelems*genCtx.q); /* distance between 2 nodes in y direction */ 477ce27a4eeSVijay Mahadevan genCtx.NY = (genCtx.q * genCtx.ney + 1); 478ce27a4eeSVijay Mahadevan genCtx.ystride = genCtx.blockSizeVertexXYZ[0]; 479ce27a4eeSVijay Mahadevan break; 480ce27a4eeSVijay Mahadevan case 1: 481*9daf19fdSVijay Mahadevan genCtx.blockSizeVertexXYZ[1]=genCtx.blockSizeVertexXYZ[2]=0; 482ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0] = genCtx.q*genCtx.blockSizeElementXYZ[0] + 1; 483ce27a4eeSVijay Mahadevan 484ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0]; /* number of elements in x direction, used for global id on element */ 485755f3dfbSVijay Mahadevan //genCtx.dx = (genCtx.xyzbounds[1]-genCtx.xyzbounds[0])/(genCtx.nex*genCtx.q); /* distance between 2 nodes in x direction */ 486755f3dfbSVijay Mahadevan genCtx.dx = (genCtx.xyzbounds[1]-genCtx.xyzbounds[0])/(nelems*genCtx.q); /* distance between 2 nodes in x direction */ 487ce27a4eeSVijay Mahadevan genCtx.NX = (genCtx.q * genCtx.nex + 1); 488ce27a4eeSVijay Mahadevan genCtx.xstride = 1; 489ce27a4eeSVijay Mahadevan break; 490ce27a4eeSVijay Mahadevan } 491ce27a4eeSVijay Mahadevan 492755f3dfbSVijay Mahadevan /* Lets check for some valid input */ 493755f3dfbSVijay Mahadevan if (genCtx.dim < 1 || genCtx.dim > 3) SETERRQ1(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Invalid topological dimension specified: %d.\n",genCtx.dim); 494755f3dfbSVijay Mahadevan if (genCtx.M * genCtx.N * genCtx.K != nprocs) SETERRQ4(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Invalid [m, n, k] data: %d, %d, %d. Product must be equal to global size = %d.\n",genCtx.M,genCtx.N,genCtx.K,nprocs); 495755f3dfbSVijay Mahadevan if (genCtx.xyzbounds) { 496755f3dfbSVijay Mahadevan /* validate the bounds data */ 497755f3dfbSVijay Mahadevan if(genCtx.xyzbounds[0] >= genCtx.xyzbounds[1]) SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"X-dim: Left boundary cannot be greater than right. [%G >= %G]\n",genCtx.xyzbounds[0],genCtx.xyzbounds[1]); 498755f3dfbSVijay Mahadevan if(genCtx.dim > 1 && (genCtx.xyzbounds[2] >= genCtx.xyzbounds[3])) SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Y-dim: Left boundary cannot be greater than right. [%G >= %G]\n",genCtx.xyzbounds[2],genCtx.xyzbounds[3]); 499755f3dfbSVijay Mahadevan if(genCtx.dim > 2 && (genCtx.xyzbounds[4] >= genCtx.xyzbounds[5])) SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Z-dim: Left boundary cannot be greater than right. [%G >= %G]\n",genCtx.xyzbounds[4],genCtx.xyzbounds[5]); 500755f3dfbSVijay Mahadevan } 501755f3dfbSVijay Mahadevan 502ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local elements:= %d, %d, %d\n",genCtx.blockSizeElementXYZ[0],genCtx.blockSizeElementXYZ[1],genCtx.blockSizeElementXYZ[2]); 503ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local vertices:= %d, %d, %d\n",genCtx.blockSizeVertexXYZ[0],genCtx.blockSizeVertexXYZ[1],genCtx.blockSizeVertexXYZ[2]); 504755f3dfbSVijay Mahadevan PetscInfo3(NULL, "Local blocks/processors := %d, %d, %d\n",genCtx.A,genCtx.B,genCtx.C); 505755f3dfbSVijay Mahadevan PetscInfo3(NULL, "Local processors := %d, %d, %d\n",genCtx.M,genCtx.N,genCtx.K); 506ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local nexyz:= %d, %d, %d\n",genCtx.nex,genCtx.ney,genCtx.nez); 507ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local delxyz:= %g, %g, %g\n",genCtx.dx,genCtx.dy,genCtx.dz); 508ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local strides:= %d, %d, %d\n",genCtx.xstride,genCtx.ystride,genCtx.zstride); 509755f3dfbSVijay Mahadevan 510ce27a4eeSVijay Mahadevan PetscFunctionReturn(0); 511ce27a4eeSVijay Mahadevan } 512ce27a4eeSVijay Mahadevan 513aa859218SVijay Mahadevan /*@ 514ce27a4eeSVijay Mahadevan DMMoabCreateBoxMesh - Creates a mesh on the tensor product (box) of intervals with genCtx specified bounds. 515aa859218SVijay Mahadevan 516aa859218SVijay Mahadevan Collective on MPI_Comm 517aa859218SVijay Mahadevan 518aa859218SVijay Mahadevan Input Parameters: 519aa859218SVijay Mahadevan + comm - The communicator for the DM object 520aa859218SVijay Mahadevan . dim - The spatial dimension 521b8ecf6d3SVijay Mahadevan . bounds - The bounds of the box specified with [x-left, x-right, y-bottom, y-top, z-bottom, z-top] depending on the spatial dimension 522b8ecf6d3SVijay Mahadevan . nele - The number of discrete elements in each direction 523b8ecf6d3SVijay Mahadevan . user_nghost - The number of ghosted layers needed in the partitioned mesh 524aa859218SVijay Mahadevan 525aa859218SVijay Mahadevan Output Parameter: 526aa859218SVijay Mahadevan . dm - The DM object 527aa859218SVijay Mahadevan 528aa859218SVijay Mahadevan Level: beginner 529aa859218SVijay Mahadevan 530aa859218SVijay Mahadevan .keywords: DM, create 531aa859218SVijay Mahadevan .seealso: DMSetType(), DMCreate(), DMMoabLoadFromFile() 532aa859218SVijay Mahadevan @*/ 533ce27a4eeSVijay Mahadevan PetscErrorCode DMMoabCreateBoxMesh(MPI_Comm comm, PetscInt dim, PetscBool useSimplex, const PetscReal* bounds, PetscInt nele, PetscInt nghost, DM *dm) 53451d15aeeSVijay Mahadevan { 53551d15aeeSVijay Mahadevan PetscErrorCode ierr; 53651d15aeeSVijay Mahadevan moab::ErrorCode merr; 537*9daf19fdSVijay Mahadevan PetscInt a,b,c,n,global_size,global_rank; 53851d15aeeSVijay Mahadevan DM_Moab *dmmoab; 539ce27a4eeSVijay Mahadevan moab::Interface *mbImpl; 540*9daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 54151d15aeeSVijay Mahadevan moab::ParallelComm *pcomm; 542*9daf19fdSVijay Mahadevan #endif 543a4717116SVijay Mahadevan moab::ReadUtilIface *readMeshIface; 544ce27a4eeSVijay Mahadevan moab::Range verts,cells,edges,faces,adj,dim3,dim2; 545ce27a4eeSVijay Mahadevan DMMoabMeshGeneratorCtx genCtx; 546a4717116SVijay Mahadevan const PetscInt npts=nele+1; /* Number of points in every dimension */ 547ce27a4eeSVijay Mahadevan 548ce27a4eeSVijay Mahadevan moab::Tag global_id_tag,part_tag,geom_tag; 549ce27a4eeSVijay Mahadevan moab::Range ownedvtx,ownedelms,localvtxs,localelms; 550ce27a4eeSVijay Mahadevan moab::EntityHandle regionset; 551755f3dfbSVijay Mahadevan PetscInt ml=0,nl=0,kl=0; 55251d15aeeSVijay Mahadevan 55351d15aeeSVijay Mahadevan PetscFunctionBegin; 554e5136372SVijay Mahadevan if(dim < 1 || dim > 3) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Invalid dimension argument for mesh: dim=[1,3].\n"); 555e5136372SVijay Mahadevan 556755f3dfbSVijay Mahadevan ierr = PetscLogEventRegister("GenerateMesh", DM_CLASSID, &genCtx.generateMesh);CHKERRQ(ierr); 557755f3dfbSVijay Mahadevan ierr = PetscLogEventRegister("AddVertices", DM_CLASSID, &genCtx.generateVertices);CHKERRQ(ierr); 558755f3dfbSVijay Mahadevan ierr = PetscLogEventRegister("AddElements", DM_CLASSID, &genCtx.generateElements);CHKERRQ(ierr); 559755f3dfbSVijay Mahadevan ierr = PetscLogEventRegister("ParResolve", DM_CLASSID, &genCtx.parResolve);CHKERRQ(ierr); 560755f3dfbSVijay Mahadevan ierr = PetscLogEventBegin(genCtx.generateMesh,0,0,0,0);CHKERRQ(ierr); 561ce27a4eeSVijay Mahadevan ierr = MPI_Comm_size(comm, &global_size);CHKERRQ(ierr); 562e5136372SVijay Mahadevan /* total number of vertices in all dimensions */ 563e5136372SVijay Mahadevan n=pow(npts,dim); 564e5136372SVijay Mahadevan 565e5136372SVijay Mahadevan /* do some error checking */ 566e5136372SVijay Mahadevan if(n < 2) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Number of points must be >= 2.\n"); 567ce27a4eeSVijay Mahadevan if(global_size > n) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Number of processors must be less than or equal to number of elements.\n"); 568e5136372SVijay Mahadevan if(nghost < 0) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Number of ghost layers cannot be negative.\n"); 569e5136372SVijay Mahadevan 570a4717116SVijay Mahadevan /* Create the basic DMMoab object and keep the default parameters created by DM impls */ 571*9daf19fdSVijay Mahadevan ierr = DMMoabCreateMoab(comm, NULL, NULL, NULL, dm);CHKERRQ(ierr); 57251d15aeeSVijay Mahadevan 573a4717116SVijay Mahadevan /* get all the necessary handles from the private DM object */ 57451d15aeeSVijay Mahadevan dmmoab = (DM_Moab*)(*dm)->data; 575ce27a4eeSVijay Mahadevan mbImpl = dmmoab->mbiface; 576*9daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 57751d15aeeSVijay Mahadevan pcomm = dmmoab->pcomm; 578ce27a4eeSVijay Mahadevan global_rank = pcomm->rank(); 579*9daf19fdSVijay Mahadevan #else 580*9daf19fdSVijay Mahadevan global_rank = 0; 581*9daf19fdSVijay Mahadevan global_size = 1; 582*9daf19fdSVijay Mahadevan #endif 583*9daf19fdSVijay Mahadevan global_id_tag = dmmoab->ltog_tag; 584c8d5365dSVijay Mahadevan dmmoab->dim = dim; 58549d66b22SVijay Mahadevan dmmoab->nghostrings=nghost; 586304006b3SVijay Mahadevan dmmoab->refct = 1; 58751d15aeeSVijay Mahadevan 588b5410836SVijay Mahadevan /* create a file set to associate all entities in current mesh */ 589ce27a4eeSVijay Mahadevan merr = mbImpl->create_meshset(moab::MESHSET_SET, dmmoab->fileset);MBERR("Creating file set failed", merr); 590b5410836SVijay Mahadevan 591a4717116SVijay Mahadevan /* No errors yet; proceed with building the mesh */ 592ce27a4eeSVijay Mahadevan merr = mbImpl->query_interface(readMeshIface);MBERRNM(merr); 59351d15aeeSVijay Mahadevan 594755f3dfbSVijay Mahadevan genCtx.M=genCtx.N=genCtx.K=1; 595755f3dfbSVijay Mahadevan genCtx.A=genCtx.B=genCtx.C=1; 596755f3dfbSVijay Mahadevan 597755f3dfbSVijay Mahadevan ierr = PetscOptionsBegin(comm,"","DMMoab Creation Options","DMMOAB");CHKERRQ(ierr); 598ce27a4eeSVijay Mahadevan /* Handle DMMoab spatial resolution */ 599ce27a4eeSVijay Mahadevan PetscOptionsInt("-dmb_grid_x","Number of grid points in x direction","DMMoabSetSizes",genCtx.blockSizeElementXYZ[0],&genCtx.blockSizeElementXYZ[0],&genCtx.usrxyzgrid); 600ce27a4eeSVijay Mahadevan if (dim > 1) {PetscOptionsInt("-dmb_grid_y","Number of grid points in y direction","DMMoabSetSizes",genCtx.blockSizeElementXYZ[1],&genCtx.blockSizeElementXYZ[1],&genCtx.usrxyzgrid);} 601ce27a4eeSVijay Mahadevan if (dim > 2) {PetscOptionsInt("-dmb_grid_z","Number of grid points in z direction","DMMoabSetSizes",genCtx.blockSizeElementXYZ[2],&genCtx.blockSizeElementXYZ[2],&genCtx.usrxyzgrid);} 602a4717116SVijay Mahadevan 603ce27a4eeSVijay Mahadevan /* Handle DMMoab parallel distibution */ 604ce27a4eeSVijay Mahadevan PetscOptionsInt("-dmb_processors_x","Number of processors in x direction","DMMoabSetNumProcs",genCtx.M,&genCtx.M,&genCtx.usrprocgrid); 605ce27a4eeSVijay Mahadevan if (dim > 1) {PetscOptionsInt("-dmb_processors_y","Number of processors in y direction","DMMoabSetNumProcs",genCtx.N,&genCtx.N,&genCtx.usrprocgrid);} 606ce27a4eeSVijay Mahadevan if (dim > 2) {PetscOptionsInt("-dmb_processors_z","Number of processors in z direction","DMMoabSetNumProcs",genCtx.K,&genCtx.K,&genCtx.usrprocgrid);} 60751d15aeeSVijay Mahadevan 608ce27a4eeSVijay Mahadevan /* Handle DMMoab block refinement */ 609ce27a4eeSVijay Mahadevan PetscOptionsInt("-dmb_refine_x","Number of refinement blocks in x direction","DMMoabSetRefinement",genCtx.A,&genCtx.A,&genCtx.usrrefgrid); 610ce27a4eeSVijay Mahadevan if (dim > 1) {PetscOptionsInt("-dmb_refine_y","Number of refinement blocks in y direction","DMMoabSetRefinement",genCtx.B,&genCtx.B,&genCtx.usrrefgrid);} 611ce27a4eeSVijay Mahadevan if (dim > 2) {PetscOptionsInt("-dmb_refine_z","Number of refinement blocks in z direction","DMMoabSetRefinement",genCtx.C,&genCtx.C,&genCtx.usrrefgrid);} 612ce27a4eeSVijay Mahadevan PetscOptionsEnd(); 61351d15aeeSVijay Mahadevan 614ce27a4eeSVijay Mahadevan ierr = DMMBUtil_InitializeOptions(genCtx, dim, useSimplex, global_rank, global_size, bounds, nele);CHKERRQ(ierr); 61551d15aeeSVijay Mahadevan 61664e1c140SVijay Mahadevan //if(nele<nprocs) SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"The dimensional discretization size should be greater or equal to number of processors: %D < %D",nele,nprocs); 61764e1c140SVijay Mahadevan 618ce27a4eeSVijay Mahadevan if (genCtx.adjEnts) genCtx.keep_skins = true; /* do not delete anything - consumes more memory */ 61966f88a78SVijay Mahadevan 620ce27a4eeSVijay Mahadevan /* determine m, n, k for processor rank */ 621755f3dfbSVijay Mahadevan ml=nl=kl=0; 622755f3dfbSVijay Mahadevan switch(genCtx.dim) { 623755f3dfbSVijay Mahadevan case 1: 624755f3dfbSVijay Mahadevan ml=(genCtx.cumfraction); 625755f3dfbSVijay Mahadevan break; 626755f3dfbSVijay Mahadevan case 2: 627755f3dfbSVijay Mahadevan nl=(genCtx.cumfraction); 628755f3dfbSVijay Mahadevan break; 629755f3dfbSVijay Mahadevan default: 630755f3dfbSVijay Mahadevan kl=(genCtx.cumfraction)/genCtx.q / genCtx.blockSizeElementXYZ[2] / genCtx.C;//genCtx.K 631755f3dfbSVijay Mahadevan break; 632755f3dfbSVijay Mahadevan } 633e5136372SVijay Mahadevan 634ce27a4eeSVijay Mahadevan /* 635ce27a4eeSVijay Mahadevan * so there are a total of M * A * blockSizeElement elements in x direction (so M * A * blockSizeElement + 1 verts in x direction) 636ce27a4eeSVijay Mahadevan * so there are a total of N * B * blockSizeElement elements in y direction (so N * B * blockSizeElement + 1 verts in y direction) 637ce27a4eeSVijay Mahadevan * so there are a total of K * C * blockSizeElement elements in z direction (so K * C * blockSizeElement + 1 verts in z direction) 638c8d5365dSVijay Mahadevan 639ce27a4eeSVijay Mahadevan * there are ( M * A blockSizeElement ) * ( N * B * blockSizeElement) * (K * C * blockSizeElement ) hexas 640ce27a4eeSVijay Mahadevan * there are ( M * A * blockSizeElement + 1) * ( N * B * blockSizeElement + 1 ) * (K * C * blockSizeElement + 1) vertices 641ce27a4eeSVijay Mahadevan * x is the first dimension that varies 642ce27a4eeSVijay Mahadevan */ 643e5136372SVijay Mahadevan 644ce27a4eeSVijay Mahadevan /* generate the block at (a, b, c); it will represent a partition , it will get a partition tag */ 64549d66b22SVijay Mahadevan PetscInt dum_id=-1; 646ce27a4eeSVijay Mahadevan merr = mbImpl->tag_get_handle("GLOBAL_ID", 1, moab::MB_TYPE_INTEGER, global_id_tag);MBERR("Getting Tag handle failed", merr); 64751d15aeeSVijay Mahadevan 648ce27a4eeSVijay Mahadevan merr = mbImpl->tag_get_handle("PARALLEL_PARTITION", 1, moab::MB_TYPE_INTEGER, part_tag, moab::MB_TAG_CREAT|moab::MB_TAG_SPARSE, &dum_id);MBERR("Getting Tag handle failed", merr); 64951d15aeeSVijay Mahadevan 6503a4aeca1SVijay Mahadevan /* lets create some sets */ 651ce27a4eeSVijay Mahadevan merr = mbImpl->tag_get_handle(GEOM_DIMENSION_TAG_NAME, 1, moab::MB_TYPE_INTEGER, geom_tag, moab::MB_TAG_CREAT|moab::MB_TAG_SPARSE, &dum_id);MBERRNM(merr); 65249d66b22SVijay Mahadevan merr = mbImpl->create_meshset(moab::MESHSET_SET, regionset);MBERRNM(merr); 653755f3dfbSVijay Mahadevan ierr = PetscLogEventEnd(genCtx.generateMesh,0,0,0,0);CHKERRQ(ierr); 6543a4aeca1SVijay Mahadevan 655ce27a4eeSVijay Mahadevan for (a=0; a<(genCtx.dim>0?genCtx.A:genCtx.A); a++) { 656ce27a4eeSVijay Mahadevan for (b=0; b<(genCtx.dim>1?genCtx.B:1); b++) { 657ce27a4eeSVijay Mahadevan for (c=0; c<(genCtx.dim>2?genCtx.C:1); c++) { 65849d66b22SVijay Mahadevan 65949d66b22SVijay Mahadevan moab::EntityHandle startv; 66049d66b22SVijay Mahadevan 661755f3dfbSVijay Mahadevan ierr = PetscLogEventBegin(genCtx.generateVertices,0,0,0,0);CHKERRQ(ierr); 662ce27a4eeSVijay Mahadevan ierr = DMMoab_GenerateVertices_Private(mbImpl, readMeshIface, genCtx, ml, nl, kl, a, b, c, global_id_tag, startv, verts);CHKERRQ(ierr); 663755f3dfbSVijay Mahadevan ierr = PetscLogEventEnd(genCtx.generateVertices,0,0,0,0);CHKERRQ(ierr); 6643a4aeca1SVijay Mahadevan 665755f3dfbSVijay Mahadevan ierr = PetscLogEventBegin(genCtx.generateElements,0,0,0,0);CHKERRQ(ierr); 666ce27a4eeSVijay Mahadevan ierr = DMMoab_GenerateElements_Private(mbImpl, readMeshIface, genCtx, ml, nl, kl, a, b, c, global_id_tag, startv, cells);CHKERRQ(ierr); 667755f3dfbSVijay Mahadevan ierr = PetscLogEventEnd(genCtx.generateElements,0,0,0,0);CHKERRQ(ierr); 6683a4aeca1SVijay Mahadevan 66949d66b22SVijay Mahadevan PetscInt part_num=0; 67049d66b22SVijay Mahadevan switch(genCtx.dim) { 67149d66b22SVijay Mahadevan case 3: 67249d66b22SVijay Mahadevan part_num += (c+kl*genCtx.C)*(genCtx.M*genCtx.A * genCtx.N*genCtx.B); 67349d66b22SVijay Mahadevan case 2: 67449d66b22SVijay Mahadevan part_num += (b+nl*genCtx.B)*(genCtx.M*genCtx.A); 67549d66b22SVijay Mahadevan case 1: 67649d66b22SVijay Mahadevan part_num += (a+ml*genCtx.A); 67749d66b22SVijay Mahadevan break; 67849d66b22SVijay Mahadevan } 67949d66b22SVijay Mahadevan 68049d66b22SVijay Mahadevan moab::EntityHandle part_set; 681ce27a4eeSVijay Mahadevan merr = mbImpl->create_meshset(moab::MESHSET_SET, part_set);MBERR("Can't create mesh set.", merr); 6823a4aeca1SVijay Mahadevan 683ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(part_set, verts);MBERR("Can't add vertices to set.", merr); 68449d66b22SVijay Mahadevan merr = mbImpl->add_entities(part_set, cells);MBERR("Can't add entities to set.", merr); 68549d66b22SVijay Mahadevan // PetscInfo2(NULL, "Generated local mesh: %D vertices and %D elements.\n", verts.size(), cells.size()); 6863a4aeca1SVijay Mahadevan 687ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(regionset, cells);MBERR("Can't add entities to set.", merr); 688ce27a4eeSVijay Mahadevan 689ce27a4eeSVijay Mahadevan /* if needed, add all edges and faces */ 690ce27a4eeSVijay Mahadevan if (genCtx.adjEnts) 691ce27a4eeSVijay Mahadevan { 692ce27a4eeSVijay Mahadevan if (genCtx.dim > 1) { 693ce27a4eeSVijay Mahadevan merr = mbImpl->get_adjacencies(cells, 1, true, edges, moab::Interface::UNION);MBERR("Can't get edges", merr); 694ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(part_set, edges);MBERR("Can't add edges to partition set.", merr); 695ce27a4eeSVijay Mahadevan } 696ce27a4eeSVijay Mahadevan if (genCtx.dim > 2) { 697ce27a4eeSVijay Mahadevan merr = mbImpl->get_adjacencies(cells, 2, true, faces, moab::Interface::UNION);MBERR("Can't get faces", merr); 698ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(part_set, faces);MBERR("Can't add faces to partition set.", merr); 699ce27a4eeSVijay Mahadevan } 700ce27a4eeSVijay Mahadevan edges.clear(); 701ce27a4eeSVijay Mahadevan faces.clear(); 702ce27a4eeSVijay Mahadevan } 70349d66b22SVijay Mahadevan verts.clear(); cells.clear(); 704ce27a4eeSVijay Mahadevan 705ce27a4eeSVijay Mahadevan merr = mbImpl->tag_set_data(part_tag, &part_set, 1, &part_num);MBERR("Can't set part tag on set", merr); 70649d66b22SVijay Mahadevan if (dmmoab->fileset) { 707ce27a4eeSVijay Mahadevan merr = mbImpl->add_parent_child(dmmoab->fileset, part_set);MBERR("Can't add part set to file set.", merr); 708ce27a4eeSVijay Mahadevan merr = mbImpl->unite_meshset(dmmoab->fileset, part_set);MBERRNM(merr); 70949d66b22SVijay Mahadevan } 71049d66b22SVijay Mahadevan merr = mbImpl->add_entities(dmmoab->fileset, &part_set, 1);MBERRNM(merr); 711ce27a4eeSVijay Mahadevan } 712ce27a4eeSVijay Mahadevan } 713ce27a4eeSVijay Mahadevan } 714ce27a4eeSVijay Mahadevan 71549d66b22SVijay Mahadevan /* set geometric dimension tag for regions */ 71649d66b22SVijay Mahadevan merr = mbImpl->tag_set_data(geom_tag, ®ionset, 1, &dmmoab->dim);MBERRNM(merr); 71749d66b22SVijay Mahadevan merr = mbImpl->add_parent_child(dmmoab->fileset,regionset);MBERRNM(merr); 718ce27a4eeSVijay Mahadevan 71949d66b22SVijay Mahadevan /* Only in parallel: resolve shared entities between processors and exchange ghost layers */ 720ce27a4eeSVijay Mahadevan if (global_size>1) { 721ce27a4eeSVijay Mahadevan 72277a8c067SVijay Mahadevan ierr = PetscLogEventBegin(genCtx.parResolve,0,0,0,0);CHKERRQ(ierr); 72377a8c067SVijay Mahadevan 724ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, genCtx.dim, cells);MBERR("Can't get all d-dimensional elements.", merr); 725ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 0, verts);MBERR("Can't get all vertices.", merr); 726ce27a4eeSVijay Mahadevan 727ce27a4eeSVijay Mahadevan if (genCtx.A*genCtx.B*genCtx.C!=1) { // merge needed 72849d66b22SVijay Mahadevan moab::MergeMesh mm(mbImpl); 729ce27a4eeSVijay Mahadevan if (genCtx.newMergeMethod) { 730ce27a4eeSVijay Mahadevan merr = mm.merge_using_integer_tag(verts, global_id_tag);MBERR("Can't merge with GLOBAL_ID tag", merr); 731ce27a4eeSVijay Mahadevan } 732ce27a4eeSVijay Mahadevan else { 733ce27a4eeSVijay Mahadevan merr = mm.merge_entities(cells, 0.0001);MBERR("Can't merge with coordinates", merr); 7343a4aeca1SVijay Mahadevan } 7353a4aeca1SVijay Mahadevan } 7363a4aeca1SVijay Mahadevan 737*9daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 738a4717116SVijay Mahadevan /* check the handles */ 739ce27a4eeSVijay Mahadevan merr = pcomm->check_all_shared_handles();MBERRV(mbImpl,merr); 74051d15aeeSVijay Mahadevan 741a4717116SVijay Mahadevan /* resolve the shared entities by exchanging information to adjacent processors */ 742ce27a4eeSVijay Mahadevan merr = pcomm->resolve_shared_ents(dmmoab->fileset,cells,dim,dim-1,NULL,&global_id_tag);MBERRV(mbImpl,merr); 74349d66b22SVijay Mahadevan if (dmmoab->fileset) { 744ce27a4eeSVijay Mahadevan merr = pcomm->exchange_ghost_cells(dim,0,nghost,dim,true,false,&dmmoab->fileset);MBERRV(mbImpl,merr); 74549d66b22SVijay Mahadevan } 74649d66b22SVijay Mahadevan else { 74749d66b22SVijay Mahadevan merr = pcomm->exchange_ghost_cells(dim,0,nghost,dim,true,false);MBERRV(mbImpl,merr); 74849d66b22SVijay Mahadevan } 749c8d5365dSVijay Mahadevan 750e427d9c9SVijay Mahadevan /* Reassign global IDs on all entities. */ 751e427d9c9SVijay Mahadevan merr = pcomm->assign_global_ids(dmmoab->fileset,dim,1,false,true,false);MBERRNM(merr); 752*9daf19fdSVijay Mahadevan #endif 75377a8c067SVijay Mahadevan 75477a8c067SVijay Mahadevan ierr = PetscLogEventEnd(genCtx.parResolve,0,0,0,0);CHKERRQ(ierr); 75549d66b22SVijay Mahadevan } 7563f1c6e43SVijay Mahadevan 757ce27a4eeSVijay Mahadevan if (!genCtx.keep_skins) { // default is to delete the 1- and 2-dimensional entities 758ce27a4eeSVijay Mahadevan // delete all quads and edges 759ce27a4eeSVijay Mahadevan moab::Range toDelete; 760ce27a4eeSVijay Mahadevan if (genCtx.dim > 1) { 761ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 1, toDelete);MBERR("Can't get edges", merr); 762ce27a4eeSVijay Mahadevan } 7633f1c6e43SVijay Mahadevan 764ce27a4eeSVijay Mahadevan if (genCtx.dim > 2) { 765ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 2, toDelete);MBERR("Can't get faces", merr); 766ce27a4eeSVijay Mahadevan } 767c8d5365dSVijay Mahadevan 768*9daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 769ce27a4eeSVijay Mahadevan merr = dmmoab->pcomm->delete_entities(toDelete) ;MBERR("Can't delete entities", merr); 770*9daf19fdSVijay Mahadevan #endif 771ce27a4eeSVijay Mahadevan } 77251d15aeeSVijay Mahadevan PetscFunctionReturn(0); 77351d15aeeSVijay Mahadevan } 77451d15aeeSVijay Mahadevan 77551d15aeeSVijay Mahadevan 77649d66b22SVijay Mahadevan #undef __FUNCT__ 77749d66b22SVijay Mahadevan #define __FUNCT__ "DMMoab_GetReadOptions_Private" 77849d66b22SVijay Mahadevan PetscErrorCode DMMoab_GetReadOptions_Private(PetscBool by_rank, PetscInt numproc, PetscInt dim, PetscInt nghost, MoabReadMode mode, PetscInt dbglevel, const char* dm_opts, const char* extra_opts, const char** read_opts) 77951d15aeeSVijay Mahadevan { 7806acfe860SVijay Mahadevan char *ropts; 78149d66b22SVijay Mahadevan char ropts_par[PETSC_MAX_PATH_LEN],ropts_pargh[PETSC_MAX_PATH_LEN]; 78261eb6e27SVijay Mahadevan char ropts_dbg[PETSC_MAX_PATH_LEN]; 783f90c3b0eSVijay Mahadevan PetscErrorCode ierr; 78451d15aeeSVijay Mahadevan 785a4717116SVijay Mahadevan PetscFunctionBegin; 78695dccacaSBarry Smith ierr = PetscMalloc1(PETSC_MAX_PATH_LEN,&ropts);CHKERRQ(ierr); 78761eb6e27SVijay Mahadevan ierr = PetscMemzero(&ropts_par,PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 78849d66b22SVijay Mahadevan ierr = PetscMemzero(&ropts_pargh,PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 78961eb6e27SVijay Mahadevan ierr = PetscMemzero(&ropts_dbg,PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 79061eb6e27SVijay Mahadevan 791e23c60ebSVijay Mahadevan /* do parallel read unless using only one processor */ 792a4717116SVijay Mahadevan if (numproc > 1) { 79349d66b22SVijay Mahadevan // ierr = PetscSNPrintf(ropts_par, PETSC_MAX_PATH_LEN, "PARALLEL=%s;PARTITION=PARALLEL_PARTITION;PARTITION_DISTRIBUTE;PARALLEL_RESOLVE_SHARED_ENTS;PARALLEL_GHOSTS=%d.0.1%s;",MoabReadModes[mode],dim,(by_rank ? ";PARTITION_BY_RANK":""));CHKERRQ(ierr); 79449d66b22SVijay Mahadevan ierr = PetscSNPrintf(ropts_par, PETSC_MAX_PATH_LEN, "PARALLEL=%s;PARTITION=PARALLEL_PARTITION;PARTITION_DISTRIBUTE;PARALLEL_RESOLVE_SHARED_ENTS;%s",MoabReadModes[mode],(by_rank ? "PARTITION_BY_RANK;":""));CHKERRQ(ierr); 79549d66b22SVijay Mahadevan if (nghost) { 79649d66b22SVijay Mahadevan ierr = PetscSNPrintf(ropts_pargh, PETSC_MAX_PATH_LEN, "PARALLEL_GHOSTS=%d.0.%d;",dim,nghost);CHKERRQ(ierr); 79749d66b22SVijay Mahadevan } 7982e4e7c01SVijay Mahadevan } 7992e4e7c01SVijay Mahadevan 800c8d5365dSVijay Mahadevan if (dbglevel) { 801f90c3b0eSVijay Mahadevan if (numproc>1) { 80249d66b22SVijay Mahadevan ierr = PetscSNPrintf(ropts_dbg, PETSC_MAX_PATH_LEN, "CPUTIME;DEBUG_IO=%d;DEBUG_PIO=%d;",dbglevel,dbglevel);CHKERRQ(ierr); 80361eb6e27SVijay Mahadevan } 80461eb6e27SVijay Mahadevan else { 80549d66b22SVijay Mahadevan ierr = PetscSNPrintf(ropts_dbg, PETSC_MAX_PATH_LEN, "CPUTIME;DEBUG_IO=%d;",dbglevel);CHKERRQ(ierr); 806f90c3b0eSVijay Mahadevan } 807c8d5365dSVijay Mahadevan } 80851d15aeeSVijay Mahadevan 80949d66b22SVijay Mahadevan ierr = PetscSNPrintf(ropts, PETSC_MAX_PATH_LEN, "%s%s%s%s%s",ropts_par,(nghost?ropts_pargh:""),ropts_dbg,(extra_opts?extra_opts:""),(dm_opts?dm_opts:""));CHKERRQ(ierr); 810f90c3b0eSVijay Mahadevan *read_opts = ropts; 81151d15aeeSVijay Mahadevan PetscFunctionReturn(0); 81251d15aeeSVijay Mahadevan } 81351d15aeeSVijay Mahadevan 81451d15aeeSVijay Mahadevan 815aa859218SVijay Mahadevan /*@ 816aa859218SVijay Mahadevan DMMoabLoadFromFile - Creates a DM object by loading the mesh from a user specified file. 817aa859218SVijay Mahadevan 818aa859218SVijay Mahadevan Collective on MPI_Comm 819aa859218SVijay Mahadevan 820aa859218SVijay Mahadevan Input Parameters: 821aa859218SVijay Mahadevan + comm - The communicator for the DM object 822aa859218SVijay Mahadevan . dim - The spatial dimension 823b8ecf6d3SVijay Mahadevan . filename - The name of the mesh file to be loaded 824b8ecf6d3SVijay Mahadevan . usrreadopts - The options string to read a MOAB mesh. 825aa859218SVijay Mahadevan Reference (Parallel Mesh Initialization: http://www.mcs.anl.gov/~fathom/moab-docs/html/contents.html#fivetwo) 826aa859218SVijay Mahadevan 827aa859218SVijay Mahadevan Output Parameter: 828aa859218SVijay Mahadevan . dm - The DM object 829aa859218SVijay Mahadevan 830aa859218SVijay Mahadevan Level: beginner 831aa859218SVijay Mahadevan 832aa859218SVijay Mahadevan .keywords: DM, create 833b8ecf6d3SVijay Mahadevan 834aa859218SVijay Mahadevan .seealso: DMSetType(), DMCreate(), DMMoabCreateBoxMesh() 835aa859218SVijay Mahadevan @*/ 83649d66b22SVijay Mahadevan PetscErrorCode DMMoabLoadFromFile(MPI_Comm comm,PetscInt dim,PetscInt nghost,const char* filename,const char* usrreadopts,DM *dm) 83751d15aeeSVijay Mahadevan { 838a4717116SVijay Mahadevan moab::ErrorCode merr; 8392e4e7c01SVijay Mahadevan PetscInt nprocs; 840a4717116SVijay Mahadevan DM_Moab *dmmoab; 841a4717116SVijay Mahadevan moab::Interface *mbiface; 842*9daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 843a4717116SVijay Mahadevan moab::ParallelComm *pcomm; 844*9daf19fdSVijay Mahadevan #endif 845a4717116SVijay Mahadevan moab::Range verts,elems; 8467023aa44SVijay Mahadevan const char *readopts; 847a4717116SVijay Mahadevan PetscErrorCode ierr; 84851d15aeeSVijay Mahadevan 84951d15aeeSVijay Mahadevan PetscFunctionBegin; 85049d66b22SVijay Mahadevan PetscValidPointer(dm,6); 85151d15aeeSVijay Mahadevan 852a4717116SVijay Mahadevan /* Create the basic DMMoab object and keep the default parameters created by DM impls */ 853*9daf19fdSVijay Mahadevan ierr = DMMoabCreateMoab(comm, NULL, NULL, NULL, dm);CHKERRQ(ierr); 85451d15aeeSVijay Mahadevan 855a4717116SVijay Mahadevan /* get all the necessary handles from the private DM object */ 856a4717116SVijay Mahadevan dmmoab = (DM_Moab*)(*dm)->data; 857a4717116SVijay Mahadevan mbiface = dmmoab->mbiface; 858*9daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 859a4717116SVijay Mahadevan pcomm = dmmoab->pcomm; 860a4717116SVijay Mahadevan nprocs = pcomm->size(); 861*9daf19fdSVijay Mahadevan #else 862*9daf19fdSVijay Mahadevan nprocs = 1; 863*9daf19fdSVijay Mahadevan #endif 864aa859218SVijay Mahadevan /* TODO: Decipher dimension based on the loaded mesh instead of getting from user */ 865c8d5365dSVijay Mahadevan dmmoab->dim = dim; 86649d66b22SVijay Mahadevan dmmoab->nghostrings=nghost; 867304006b3SVijay Mahadevan dmmoab->refct = 1; 868a4717116SVijay Mahadevan 869b5410836SVijay Mahadevan /* create a file set to associate all entities in current mesh */ 870b5410836SVijay Mahadevan merr = dmmoab->mbiface->create_meshset(moab::MESHSET_SET, dmmoab->fileset);MBERR("Creating file set failed", merr); 871b5410836SVijay Mahadevan 872a4717116SVijay Mahadevan /* add mesh loading options specific to the DM */ 87349d66b22SVijay Mahadevan ierr = DMMoab_GetReadOptions_Private(dmmoab->partition_by_rank, nprocs, dim, nghost, dmmoab->read_mode, 8742e4e7c01SVijay Mahadevan dmmoab->rw_dbglevel, dmmoab->extra_read_options, usrreadopts, &readopts);CHKERRQ(ierr); 8757023aa44SVijay Mahadevan 876e5136372SVijay Mahadevan PetscInfo2(*dm, "Reading file %s with options: %s\n",filename,readopts); 877a4717116SVijay Mahadevan 878a4717116SVijay Mahadevan /* Load the mesh from a file. */ 87949d66b22SVijay Mahadevan if (dmmoab->fileset) { 8807023aa44SVijay Mahadevan merr = mbiface->load_file(filename, &dmmoab->fileset, readopts);MBERRVM(mbiface,"Reading MOAB file failed.", merr); 88149d66b22SVijay Mahadevan } 88249d66b22SVijay Mahadevan else { 88349d66b22SVijay Mahadevan merr = mbiface->load_file(filename, 0, readopts);MBERRVM(mbiface,"Reading MOAB file failed.", merr); 88449d66b22SVijay Mahadevan } 885a4717116SVijay Mahadevan 886*9daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 8876e40195eSVijay Mahadevan /* Reassign global IDs on all entities. */ 888e5136372SVijay Mahadevan merr = pcomm->assign_global_ids(dmmoab->fileset,dim,1,true,true,true);MBERRNM(merr); 889*9daf19fdSVijay Mahadevan #endif 890e5136372SVijay Mahadevan 891e5136372SVijay Mahadevan /* load the local vertices */ 892e5136372SVijay Mahadevan merr = mbiface->get_entities_by_type(dmmoab->fileset, moab::MBVERTEX, verts, true);MBERRNM(merr); 893e5136372SVijay Mahadevan /* load the local elements */ 894e5136372SVijay Mahadevan merr = mbiface->get_entities_by_dimension(dmmoab->fileset, dim, elems, true);MBERRNM(merr); 895e5136372SVijay Mahadevan 896*9daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 897e5136372SVijay Mahadevan /* Everything is set up, now just do a tag exchange to update tags 898e5136372SVijay Mahadevan on all of the ghost vertexes */ 899e5136372SVijay Mahadevan merr = pcomm->exchange_tags(dmmoab->ltog_tag,verts);MBERRV(mbiface,merr); 900e5136372SVijay Mahadevan merr = pcomm->exchange_tags(dmmoab->ltog_tag,elems);MBERRV(mbiface,merr); 901a4717116SVijay Mahadevan merr = pcomm->collective_sync_partition();MBERR("Collective sync failed", merr); 902*9daf19fdSVijay Mahadevan #endif 903a4717116SVijay Mahadevan 904e5136372SVijay Mahadevan PetscInfo3(*dm, "MOAB file '%s' was successfully loaded. Found %D vertices and %D elements.\n", filename, verts.size(), elems.size()); 9056acfe860SVijay Mahadevan ierr = PetscFree(readopts);CHKERRQ(ierr); 90651d15aeeSVijay Mahadevan PetscFunctionReturn(0); 90751d15aeeSVijay Mahadevan } 90851d15aeeSVijay Mahadevan 909304006b3SVijay Mahadevan 910304006b3SVijay Mahadevan #undef __FUNCT__ 911304006b3SVijay Mahadevan #define __FUNCT__ "DMMoabRenumberMeshEntities" 912304006b3SVijay Mahadevan /*@ 913304006b3SVijay Mahadevan DMMoabRenumberMeshEntities - Order and number all entities (vertices->elements) to be contiguously ordered 914304006b3SVijay Mahadevan in parallel 915304006b3SVijay Mahadevan 916304006b3SVijay Mahadevan Collective on MPI_Comm 917304006b3SVijay Mahadevan 918304006b3SVijay Mahadevan Input Parameters: 919304006b3SVijay Mahadevan + dm - The DM object 920304006b3SVijay Mahadevan 921304006b3SVijay Mahadevan Level: advanced 922304006b3SVijay Mahadevan 923304006b3SVijay Mahadevan .keywords: DM, reorder 924304006b3SVijay Mahadevan 925304006b3SVijay Mahadevan .seealso: DMSetUp(), DMCreate() 926304006b3SVijay Mahadevan @*/ 927304006b3SVijay Mahadevan PetscErrorCode DMMoabRenumberMeshEntities(DM dm) 928304006b3SVijay Mahadevan { 929304006b3SVijay Mahadevan moab::Range verts; 930304006b3SVijay Mahadevan 931304006b3SVijay Mahadevan PetscFunctionBegin; 932304006b3SVijay Mahadevan PetscValidHeaderSpecific(dm,DM_CLASSID,1); 933304006b3SVijay Mahadevan 934*9daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 935304006b3SVijay Mahadevan /* Insert new points */ 936*9daf19fdSVijay Mahadevan moab::ErrorCode merr; 937*9daf19fdSVijay Mahadevan merr = ((DM_Moab*) dm->data)->pcomm->assign_global_ids(((DM_Moab*) dm->data)->fileset,3,0,false,true,false);MBERRNM(merr); 938*9daf19fdSVijay Mahadevan #endif 939304006b3SVijay Mahadevan PetscFunctionReturn(0); 940304006b3SVijay Mahadevan } 941304006b3SVijay Mahadevan 942