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 2949d66b22SVijay Mahadevan PetscInt DMMoab_SetTensorElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, PetscInt offset, PetscInt corner, std::vector<PetscInt>& subent_conn, moab::EntityHandle *connectivity) 3051d15aeeSVijay Mahadevan { 31ce27a4eeSVijay Mahadevan switch (genCtx.dim) { 32e5136372SVijay Mahadevan case 1: 33ce27a4eeSVijay Mahadevan subent_conn.resize(2); 34ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBEDGE, 1, 0, subent_conn.data()); 35ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[0]] = corner; 36ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[1]] = corner + 1; 37e5136372SVijay Mahadevan break; 38e5136372SVijay Mahadevan case 2: 39ce27a4eeSVijay Mahadevan subent_conn.resize(4); 40ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBQUAD, 2, 0, subent_conn.data()); 41ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[0]] = corner; 42ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[1]] = corner + 1; 43ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[2]] = corner + 1 + genCtx.ystride; 44ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[3]] = corner + genCtx.ystride; 45e5136372SVijay Mahadevan break; 46e5136372SVijay Mahadevan case 3: 47e5136372SVijay Mahadevan default: 48ce27a4eeSVijay Mahadevan subent_conn.resize(8); 49ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBHEX, 3, 0, subent_conn.data()); 50ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[0]] = corner; 51ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[1]] = corner + 1; 52ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[2]] = corner + 1 + genCtx.ystride; 53ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[3]] = corner + genCtx.ystride; 54ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[4]] = corner + genCtx.zstride; 55ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[5]] = corner + 1 + genCtx.zstride; 56ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[6]] = corner + 1 + genCtx.ystride + genCtx.zstride; 57ce27a4eeSVijay Mahadevan connectivity[offset + subent_conn[7]] = corner + genCtx.ystride + genCtx.zstride; 58e5136372SVijay Mahadevan break; 59e5136372SVijay Mahadevan } 60ce27a4eeSVijay Mahadevan return subent_conn.size(); 61e5136372SVijay Mahadevan } 6251d15aeeSVijay Mahadevan 63ce27a4eeSVijay Mahadevan 6449d66b22SVijay Mahadevan PetscInt DMMoab_SetSimplexElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, PetscInt subelem, PetscInt offset, PetscInt corner, std::vector<PetscInt>& subent_conn, moab::EntityHandle *connectivity) 65c3dd00c7SVijay Mahadevan { 66755f3dfbSVijay Mahadevan PetscInt A, B, C, D, E, F, G, H, M; 67755f3dfbSVijay Mahadevan const PetscInt trigen_opts = 1; /* 1 - Aligned diagonally to right, 2 - Aligned diagonally to left, 3 - 4 elements per quad */ 6849d66b22SVijay Mahadevan A = corner; 6949d66b22SVijay Mahadevan B = corner + 1; 70ce27a4eeSVijay Mahadevan switch (genCtx.dim) { 71c3dd00c7SVijay Mahadevan case 1: 72ce27a4eeSVijay Mahadevan subent_conn.resize(2); /* only linear EDGE supported now */ 73ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBEDGE, 1, 0, subent_conn.data()); 7449d66b22SVijay Mahadevan connectivity[offset + subent_conn[0]] = A; 7549d66b22SVijay Mahadevan connectivity[offset + subent_conn[1]] = B; 76c3dd00c7SVijay Mahadevan break; 77c3dd00c7SVijay Mahadevan case 2: 7849d66b22SVijay Mahadevan C = corner + 1 + genCtx.ystride; 7949d66b22SVijay Mahadevan D = corner + genCtx.ystride; 8063d025dbSVijay Mahadevan M = corner + 0.5; /* technically -- need to modify vertex generation */ 81ce27a4eeSVijay Mahadevan subent_conn.resize(3); /* only linear TRI supported */ 82ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBTRI, 2, 0, subent_conn.data()); 83755f3dfbSVijay Mahadevan if (trigen_opts == 1) { 84ce27a4eeSVijay Mahadevan if (subelem) { /* 0 1 2 of a QUAD */ 8577a8c067SVijay Mahadevan connectivity[offset + subent_conn[0]] = B; 86755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[1]] = C; 8777a8c067SVijay Mahadevan connectivity[offset + subent_conn[2]] = A; 88c3dd00c7SVijay Mahadevan } 89ce27a4eeSVijay Mahadevan else { /* 2 3 0 of a QUAD */ 9077a8c067SVijay Mahadevan connectivity[offset + subent_conn[0]] = D; 9177a8c067SVijay Mahadevan connectivity[offset + subent_conn[1]] = A; 92755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[2]] = C; 93755f3dfbSVijay Mahadevan } 94755f3dfbSVijay Mahadevan } 95755f3dfbSVijay Mahadevan else if (trigen_opts == 2) { 96755f3dfbSVijay Mahadevan if (subelem) { /* 0 1 2 of a QUAD */ 97755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[0]] = A; 9877a8c067SVijay Mahadevan connectivity[offset + subent_conn[1]] = B; 9977a8c067SVijay Mahadevan connectivity[offset + subent_conn[2]] = D; 100755f3dfbSVijay Mahadevan } 101755f3dfbSVijay Mahadevan else { /* 2 3 0 of a QUAD */ 10277a8c067SVijay Mahadevan connectivity[offset + subent_conn[0]] = C; 10377a8c067SVijay Mahadevan connectivity[offset + subent_conn[1]] = D; 104755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[2]] = B; 105755f3dfbSVijay Mahadevan } 106755f3dfbSVijay Mahadevan } 107755f3dfbSVijay Mahadevan else { 108755f3dfbSVijay Mahadevan switch (subelem) { /* 0 1 2 of a QUAD */ 109755f3dfbSVijay Mahadevan case 0: 110755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[0]] = A; 111755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[1]] = B; 112755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[2]] = M; 113755f3dfbSVijay Mahadevan break; 114755f3dfbSVijay Mahadevan case 1: 115755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[0]] = B; 116755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[1]] = C; 117755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[2]] = M; 118755f3dfbSVijay Mahadevan break; 119755f3dfbSVijay Mahadevan case 2: 12049d66b22SVijay Mahadevan connectivity[offset + subent_conn[0]] = C; 12149d66b22SVijay Mahadevan connectivity[offset + subent_conn[1]] = D; 122755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[2]] = M; 123755f3dfbSVijay Mahadevan break; 124755f3dfbSVijay Mahadevan case 3: 125755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[0]] = D; 126755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[1]] = A; 127755f3dfbSVijay Mahadevan connectivity[offset + subent_conn[2]] = M; 128755f3dfbSVijay Mahadevan break; 129755f3dfbSVijay Mahadevan } 130c3dd00c7SVijay Mahadevan } 131c3dd00c7SVijay Mahadevan break; 132c3dd00c7SVijay Mahadevan case 3: 133c3dd00c7SVijay Mahadevan default: 13449d66b22SVijay Mahadevan C = corner + 1 + genCtx.ystride; 13549d66b22SVijay Mahadevan D = corner + genCtx.ystride; 13649d66b22SVijay Mahadevan E = corner + genCtx.zstride; 13749d66b22SVijay Mahadevan F = corner + 1 + genCtx.zstride; 13849d66b22SVijay Mahadevan G = corner + 1 + genCtx.ystride + genCtx.zstride; 13949d66b22SVijay Mahadevan H = corner + genCtx.ystride + genCtx.zstride; 140ce27a4eeSVijay Mahadevan subent_conn.resize(4); /* only linear TET supported */ 141ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBTET, 3, 0, subent_conn.data()); 142c3dd00c7SVijay Mahadevan switch (subelem) { 14349d66b22SVijay Mahadevan case 0: /* 4 3 7 6 of a HEX */ 14449d66b22SVijay Mahadevan connectivity[offset + subent_conn[0]] = E; 14549d66b22SVijay Mahadevan connectivity[offset + subent_conn[1]] = D; 14649d66b22SVijay Mahadevan connectivity[offset + subent_conn[2]] = H; 14749d66b22SVijay Mahadevan connectivity[offset + subent_conn[3]] = G; 148c3dd00c7SVijay Mahadevan break; 14949d66b22SVijay Mahadevan case 1: /* 0 1 2 5 of a HEX */ 15049d66b22SVijay Mahadevan connectivity[offset + subent_conn[0]] = A; 15149d66b22SVijay Mahadevan connectivity[offset + subent_conn[1]] = B; 15249d66b22SVijay Mahadevan connectivity[offset + subent_conn[2]] = C; 15349d66b22SVijay Mahadevan connectivity[offset + subent_conn[3]] = F; 154c3dd00c7SVijay Mahadevan break; 15549d66b22SVijay Mahadevan case 2: /* 0 3 4 5 of a HEX */ 15649d66b22SVijay Mahadevan connectivity[offset + subent_conn[0]] = A; 15749d66b22SVijay Mahadevan connectivity[offset + subent_conn[1]] = D; 15849d66b22SVijay Mahadevan connectivity[offset + subent_conn[2]] = E; 15949d66b22SVijay Mahadevan connectivity[offset + subent_conn[3]] = F; 160c3dd00c7SVijay Mahadevan break; 16149d66b22SVijay Mahadevan case 3: /* 2 6 3 5 of a HEX */ 16249d66b22SVijay Mahadevan connectivity[offset + subent_conn[0]] = C; 16349d66b22SVijay Mahadevan connectivity[offset + subent_conn[1]] = G; 16449d66b22SVijay Mahadevan connectivity[offset + subent_conn[2]] = D; 16549d66b22SVijay Mahadevan connectivity[offset + subent_conn[3]] = F; 166c3dd00c7SVijay Mahadevan break; 16749d66b22SVijay Mahadevan case 4: /* 0 2 3 5 of a HEX */ 16849d66b22SVijay Mahadevan connectivity[offset + subent_conn[0]] = A; 16949d66b22SVijay Mahadevan connectivity[offset + subent_conn[1]] = C; 17049d66b22SVijay Mahadevan connectivity[offset + subent_conn[2]] = D; 17149d66b22SVijay Mahadevan connectivity[offset + subent_conn[3]] = F; 17249d66b22SVijay Mahadevan break; 17349d66b22SVijay Mahadevan case 5: /* 3 6 4 5 of a HEX */ 17449d66b22SVijay Mahadevan connectivity[offset + subent_conn[0]] = D; 17549d66b22SVijay Mahadevan connectivity[offset + subent_conn[1]] = G; 17649d66b22SVijay Mahadevan connectivity[offset + subent_conn[2]] = E; 17749d66b22SVijay Mahadevan connectivity[offset + subent_conn[3]] = F; 178c3dd00c7SVijay Mahadevan break; 179c3dd00c7SVijay Mahadevan } 180c3dd00c7SVijay Mahadevan break; 181c3dd00c7SVijay Mahadevan } 182ce27a4eeSVijay Mahadevan return subent_conn.size(); 183c3dd00c7SVijay Mahadevan } 184c3dd00c7SVijay Mahadevan 185ce27a4eeSVijay Mahadevan 18649d66b22SVijay Mahadevan std::pair<PetscInt, PetscInt> DMMoab_SetElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, PetscInt offset, PetscInt corner, moab::EntityHandle *connectivity) 187c3dd00c7SVijay Mahadevan { 18849d66b22SVijay Mahadevan PetscInt vcount = 0; 18949d66b22SVijay Mahadevan PetscInt simplices_per_tensor[4] = {0, 1, 2, 6}; 19049d66b22SVijay Mahadevan std::vector<PetscInt> subent_conn; /* only linear edge, tri, tet supported now */ 191ce27a4eeSVijay Mahadevan subent_conn.reserve(27); 19249d66b22SVijay Mahadevan PetscInt m, subelem; 193ce27a4eeSVijay Mahadevan if (genCtx.simplex) { 19449d66b22SVijay Mahadevan subelem = simplices_per_tensor[genCtx.dim]; 195ce27a4eeSVijay Mahadevan for (m = 0; m < subelem; m++) { 196ce27a4eeSVijay Mahadevan vcount = DMMoab_SetSimplexElementConnectivity_Private(genCtx, m, offset, corner, subent_conn, connectivity); 197ce27a4eeSVijay Mahadevan offset += vcount; 198ce27a4eeSVijay Mahadevan } 199c3dd00c7SVijay Mahadevan } 200c3dd00c7SVijay Mahadevan else { 201c3dd00c7SVijay Mahadevan subelem = 1; 202ce27a4eeSVijay Mahadevan vcount = DMMoab_SetTensorElementConnectivity_Private(genCtx, offset, corner, subent_conn, connectivity); 203c3dd00c7SVijay Mahadevan } 20449d66b22SVijay Mahadevan return std::pair<PetscInt, PetscInt>(vcount * subelem, subelem); 205c3dd00c7SVijay Mahadevan } 206c3dd00c7SVijay Mahadevan 207c3dd00c7SVijay Mahadevan 20849d66b22SVijay Mahadevan PetscErrorCode DMMoab_GenerateVertices_Private(moab::Interface *mbImpl, moab::ReadUtilIface *iface, DMMoabMeshGeneratorCtx& genCtx, PetscInt m, PetscInt n, PetscInt k, 20949d66b22SVijay Mahadevan PetscInt a, PetscInt b, PetscInt c, moab::Tag& global_id_tag, moab::EntityHandle& startv, moab::Range& uverts) 210ce27a4eeSVijay Mahadevan { 21149d66b22SVijay Mahadevan PetscInt x, y, z, ix, nnodes; 212755f3dfbSVijay Mahadevan PetscErrorCode ierr; 21349d66b22SVijay Mahadevan PetscInt ii, jj, kk; 21449d66b22SVijay Mahadevan std::vector<PetscReal*> arrays; 215755f3dfbSVijay Mahadevan PetscInt* gids; 216ce27a4eeSVijay Mahadevan moab::ErrorCode merr; 217ce27a4eeSVijay Mahadevan 218ce27a4eeSVijay Mahadevan PetscFunctionBegin; 219ce27a4eeSVijay Mahadevan /* we will generate (q*block+1)^3 vertices, and block^3 hexas; q is 1 for linear, 2 for quadratic 220ce27a4eeSVijay Mahadevan * the global id of the vertices will come from m, n, k, a, b, c 221ce27a4eeSVijay Mahadevan * x will vary from m*A*q*block + a*q*block to m*A*q*block+(a+1)*q*block etc. 222ce27a4eeSVijay Mahadevan */ 223ce27a4eeSVijay Mahadevan nnodes = genCtx.blockSizeVertexXYZ[0] * (genCtx.dim > 1 ? genCtx.blockSizeVertexXYZ[1] * (genCtx.dim > 2 ? genCtx.blockSizeVertexXYZ[2] : 1) : 1); 224755f3dfbSVijay Mahadevan ierr = PetscMalloc1(nnodes, &gids);CHKERRQ(ierr); 225ce27a4eeSVijay Mahadevan 226ce27a4eeSVijay Mahadevan merr = iface->get_node_coords(3, nnodes, 0, startv, arrays);MBERR("Can't get node coords.", merr); 227ce27a4eeSVijay Mahadevan 228ce27a4eeSVijay Mahadevan /* will start with the lower corner: */ 22963d025dbSVijay Mahadevan /* x = ( m * genCtx.A + a ) * genCtx.q * genCtx.blockSizeElementXYZ[0]; */ 23063d025dbSVijay Mahadevan /* y = ( n * genCtx.B + b ) * genCtx.q * genCtx.blockSizeElementXYZ[1]; */ 23163d025dbSVijay Mahadevan /* z = ( k * genCtx.C + c ) * genCtx.q * genCtx.blockSizeElementXYZ[2]; */ 232755f3dfbSVijay Mahadevan 233755f3dfbSVijay Mahadevan x = ( m * genCtx.A + a ) * genCtx.q; 234755f3dfbSVijay Mahadevan y = ( n * genCtx.B + b ) * genCtx.q; 235755f3dfbSVijay Mahadevan z = ( k * genCtx.C + c ) * genCtx.q; 236755f3dfbSVijay Mahadevan PetscInfo3(NULL, "Starting offset for coordinates := %d, %d, %d\n", x, y, z); 237ce27a4eeSVijay Mahadevan ix = 0; 238ce27a4eeSVijay Mahadevan moab::Range verts(startv, startv + nnodes - 1); 239ce27a4eeSVijay Mahadevan for (kk = 0; kk < (genCtx.dim > 2 ? genCtx.blockSizeVertexXYZ[2] : 1); kk++) { 240ce27a4eeSVijay Mahadevan for (jj = 0; jj < (genCtx.dim > 1 ? genCtx.blockSizeVertexXYZ[1] : 1); jj++) { 241ce27a4eeSVijay Mahadevan for (ii = 0; ii < genCtx.blockSizeVertexXYZ[0]; ii++, ix++) { 242ce27a4eeSVijay Mahadevan /* set coordinates for the vertices */ 243ce27a4eeSVijay Mahadevan arrays[0][ix] = (x + ii) * genCtx.dx + genCtx.xyzbounds[0]; 244ce27a4eeSVijay Mahadevan arrays[1][ix] = (y + jj) * genCtx.dy + genCtx.xyzbounds[2]; 245ce27a4eeSVijay Mahadevan arrays[2][ix] = (z + kk) * genCtx.dz + genCtx.xyzbounds[4]; 246755f3dfbSVijay Mahadevan PetscInfo3(NULL, "Creating vertex with coordinates := %f, %f, %f\n", arrays[0][ix], arrays[1][ix], arrays[2][ix]); 247ce27a4eeSVijay Mahadevan 248ce27a4eeSVijay Mahadevan /* If we want to set some tags on the vertices -> use the following entity handle definition: 249ce27a4eeSVijay Mahadevan moab::EntityHandle v = startv + ix; 250ce27a4eeSVijay Mahadevan */ 251ce27a4eeSVijay Mahadevan /* compute the global ID for vertex */ 252ce27a4eeSVijay Mahadevan gids[ix] = 1 + (x + ii) + (y + jj) * genCtx.NX + (z + kk) * (genCtx.NX * genCtx.NY); 253ce27a4eeSVijay Mahadevan } 254ce27a4eeSVijay Mahadevan } 255ce27a4eeSVijay Mahadevan } 256ce27a4eeSVijay Mahadevan /* set global ID data on vertices */ 257ce27a4eeSVijay Mahadevan mbImpl->tag_set_data(global_id_tag, verts, &gids[0]); 258ce27a4eeSVijay Mahadevan verts.swap(uverts); 259755f3dfbSVijay Mahadevan ierr = PetscFree(gids);CHKERRQ(ierr); 260ce27a4eeSVijay Mahadevan PetscFunctionReturn(0); 261ce27a4eeSVijay Mahadevan } 262ce27a4eeSVijay Mahadevan 26349d66b22SVijay Mahadevan PetscErrorCode DMMoab_GenerateElements_Private(moab::Interface* mbImpl, moab::ReadUtilIface* iface, DMMoabMeshGeneratorCtx& genCtx, PetscInt m, PetscInt n, PetscInt k, 26449d66b22SVijay Mahadevan PetscInt a, PetscInt b, PetscInt c, moab::Tag& global_id_tag, moab::EntityHandle startv, moab::Range& cells) 265ce27a4eeSVijay Mahadevan { 266ce27a4eeSVijay Mahadevan moab::ErrorCode merr; 267ce27a4eeSVijay Mahadevan PetscInt ix, ie, xe, ye, ze; 268ce27a4eeSVijay Mahadevan PetscInt ii, jj, kk, nvperelem; 26949d66b22SVijay Mahadevan PetscInt simplices_per_tensor[4] = {0, 1, 2, 6}; 270ce27a4eeSVijay 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);*/ 271ce27a4eeSVijay Mahadevan PetscInt nelems = ntensorelems; 27263d025dbSVijay Mahadevan moab::EntityHandle starte; /* connectivity */ 273ce27a4eeSVijay Mahadevan moab::EntityHandle* conn; 274ce27a4eeSVijay Mahadevan 275ce27a4eeSVijay Mahadevan PetscFunctionBegin; 276ce27a4eeSVijay Mahadevan switch (genCtx.dim) { 277ce27a4eeSVijay Mahadevan case 1: 278ce27a4eeSVijay Mahadevan nvperelem = 2; 279ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 2, moab::MBEDGE, 0, starte, conn);MBERR("Can't get EDGE2 element connectivity.", merr); 280ce27a4eeSVijay Mahadevan break; 281ce27a4eeSVijay Mahadevan case 2: 282ce27a4eeSVijay Mahadevan if (genCtx.simplex) { 283ce27a4eeSVijay Mahadevan nvperelem = 3; 28449d66b22SVijay Mahadevan nelems = ntensorelems * simplices_per_tensor[genCtx.dim]; 285ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 3, moab::MBTRI, 0, starte, conn);MBERR("Can't get TRI3 element connectivity.", merr); 286ce27a4eeSVijay Mahadevan } 287ce27a4eeSVijay Mahadevan else { 288ce27a4eeSVijay Mahadevan nvperelem = 4; 289ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 4, moab::MBQUAD, 0, starte, conn);MBERR("Can't get QUAD4 element connectivity.", merr); 290ce27a4eeSVijay Mahadevan } 291ce27a4eeSVijay Mahadevan break; 292ce27a4eeSVijay Mahadevan case 3: 293ce27a4eeSVijay Mahadevan default: 294ce27a4eeSVijay Mahadevan if (genCtx.simplex) { 295ce27a4eeSVijay Mahadevan nvperelem = 4; 29649d66b22SVijay Mahadevan nelems = ntensorelems * simplices_per_tensor[genCtx.dim]; 297ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 4, moab::MBTET, 0, starte, conn);MBERR("Can't get TET4 element connectivity.", merr); 298ce27a4eeSVijay Mahadevan } 299ce27a4eeSVijay Mahadevan else { 300ce27a4eeSVijay Mahadevan nvperelem = 8; 301ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 8, moab::MBHEX, 0, starte, conn);MBERR("Can't get HEX8 element connectivity.", merr); 302ce27a4eeSVijay Mahadevan } 303ce27a4eeSVijay Mahadevan break; 304ce27a4eeSVijay Mahadevan } 305ce27a4eeSVijay Mahadevan 30663d025dbSVijay Mahadevan ix = ie = 0; /* index now in the elements, for global ids */ 307ce27a4eeSVijay Mahadevan 308ce27a4eeSVijay Mahadevan /* create a temporary range to store local element handles */ 309ce27a4eeSVijay Mahadevan moab::Range tmp(starte, starte + nelems - 1); 31049d66b22SVijay Mahadevan std::vector<PetscInt> gids(nelems); 311ce27a4eeSVijay Mahadevan 312ce27a4eeSVijay Mahadevan /* identify the elements at the lower corner, for their global ids */ 313ce27a4eeSVijay Mahadevan xe = m * genCtx.A * genCtx.blockSizeElementXYZ[0] + a * genCtx.blockSizeElementXYZ[0]; 314ce27a4eeSVijay Mahadevan ye = (genCtx.dim > 1 ? n * genCtx.B * genCtx.blockSizeElementXYZ[1] + b * genCtx.blockSizeElementXYZ[1] : 0); 315ce27a4eeSVijay Mahadevan ze = (genCtx.dim > 2 ? k * genCtx.C * genCtx.blockSizeElementXYZ[2] + c * genCtx.blockSizeElementXYZ[2] : 0); 316ce27a4eeSVijay Mahadevan 317ce27a4eeSVijay Mahadevan /* create owned elements requested by genCtx */ 318ce27a4eeSVijay Mahadevan for (kk = 0; kk < (genCtx.dim > 2 ? genCtx.blockSizeElementXYZ[2] : 1); kk++) { 319ce27a4eeSVijay Mahadevan for (jj = 0; jj < (genCtx.dim > 1 ? genCtx.blockSizeElementXYZ[1] : 1); jj++) { 320ce27a4eeSVijay Mahadevan for (ii = 0; ii < genCtx.blockSizeElementXYZ[0]; ii++) { 321ce27a4eeSVijay Mahadevan 322ce27a4eeSVijay Mahadevan moab::EntityHandle corner = startv + genCtx.q * ii + genCtx.q * jj * genCtx.ystride + genCtx.q * kk * genCtx.zstride; 323ce27a4eeSVijay Mahadevan 32449d66b22SVijay Mahadevan std::pair<PetscInt, PetscInt> entoffset = DMMoab_SetElementConnectivity_Private(genCtx, ix, corner, conn); 325ce27a4eeSVijay Mahadevan 32649d66b22SVijay Mahadevan for (PetscInt j = 0; j < entoffset.second; j++) { 327ce27a4eeSVijay Mahadevan /* The entity handle for the particular element -> if we want to set some tags is 328ce27a4eeSVijay Mahadevan moab::EntityHandle eh = starte + ie + j; 329ce27a4eeSVijay Mahadevan */ 330ce27a4eeSVijay Mahadevan gids[ie + j] = 1 + ((xe + ii) + (ye + jj) * genCtx.nex + (ze + kk) * (genCtx.nex * genCtx.ney)); 33163d025dbSVijay Mahadevan /* gids[ie+j] = ie + j + ((xe + ii) + (ye + jj) * genCtx.nex + (ze + kk) * (genCtx.nex * genCtx.ney)); */ 33263d025dbSVijay Mahadevan /* gids[ie+j] = 1 + ie; */ 33363d025dbSVijay Mahadevan /* ie++; */ 334ce27a4eeSVijay Mahadevan } 335ce27a4eeSVijay Mahadevan 336ce27a4eeSVijay Mahadevan ix += entoffset.first; 33749d66b22SVijay Mahadevan ie += entoffset.second; 338ce27a4eeSVijay Mahadevan } 339ce27a4eeSVijay Mahadevan } 340ce27a4eeSVijay Mahadevan } 341ce27a4eeSVijay Mahadevan if (genCtx.adjEnts) { /* we need to update adjacencies now, because some elements are new */ 342ce27a4eeSVijay Mahadevan merr = iface->update_adjacencies(starte, nelems, nvperelem, conn);MBERR("Can't update adjacencies", merr); 343ce27a4eeSVijay Mahadevan } 344ce27a4eeSVijay Mahadevan tmp.swap(cells); 345ce27a4eeSVijay Mahadevan merr = mbImpl->tag_set_data(global_id_tag, cells, &gids[0]);MBERR("Can't set global ids to elements.", merr); 346ce27a4eeSVijay Mahadevan PetscFunctionReturn(0); 347ce27a4eeSVijay Mahadevan } 348ce27a4eeSVijay Mahadevan 349755f3dfbSVijay Mahadevan PetscErrorCode DMMBUtil_InitializeOptions(DMMoabMeshGeneratorCtx& genCtx, PetscInt dim, PetscBool simplex, PetscInt rank, PetscInt nprocs, const PetscReal* bounds, PetscInt nelems) 350ce27a4eeSVijay Mahadevan { 351ce27a4eeSVijay Mahadevan PetscFunctionBegin; 352ce27a4eeSVijay Mahadevan /* Initialize all genCtx data */ 353ce27a4eeSVijay Mahadevan genCtx.dim = dim; 354ce27a4eeSVijay Mahadevan genCtx.simplex = simplex; 35549d66b22SVijay Mahadevan genCtx.newMergeMethod = genCtx.keep_skins = genCtx.adjEnts = true; 356ce27a4eeSVijay Mahadevan /* determine other global quantities for the mesh used for nodes increments */ 357ce27a4eeSVijay Mahadevan genCtx.q = 1; 358755f3dfbSVijay Mahadevan genCtx.fraction = genCtx.remainder = genCtx.cumfraction = 0; 359ce27a4eeSVijay Mahadevan 360ce27a4eeSVijay Mahadevan if (!genCtx.usrxyzgrid) { /* not overridden by genCtx - assume nele equally and that genCtx wants a uniform cube mesh */ 361755f3dfbSVijay Mahadevan 362755f3dfbSVijay Mahadevan genCtx.fraction = nelems / nprocs; /* partition only by the largest dimension */ 363755f3dfbSVijay Mahadevan genCtx.remainder = nelems % nprocs; /* remainder after partition which gets evenly distributed by round-robin */ 364755f3dfbSVijay Mahadevan genCtx.cumfraction = (rank > 0 ? (genCtx.fraction) * (rank) + (rank - 1 < genCtx.remainder ? rank : genCtx.remainder ) : 0); 365755f3dfbSVijay Mahadevan if (rank < genCtx.remainder) /* This process gets "fraction+1" elements */ 366755f3dfbSVijay Mahadevan genCtx.fraction++; 367755f3dfbSVijay Mahadevan 3689daf19fdSVijay Mahadevan PetscInfo3(NULL, "Fraction = %D, Remainder = %D, Cumulative fraction = %D\n", genCtx.fraction, genCtx.remainder, genCtx.cumfraction); 369755f3dfbSVijay Mahadevan switch (genCtx.dim) { 370755f3dfbSVijay Mahadevan case 1: 371755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[0] = genCtx.fraction; 372755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[1] = 1; 373755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[2] = 1; 374755f3dfbSVijay Mahadevan break; 375755f3dfbSVijay Mahadevan case 2: 376755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[0] = nelems; 377755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[1] = genCtx.fraction; 378755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[2] = 1; 379755f3dfbSVijay Mahadevan break; 380755f3dfbSVijay Mahadevan case 3: 381755f3dfbSVijay Mahadevan default: 382755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[0] = nelems; 383755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[1] = nelems; 384755f3dfbSVijay Mahadevan genCtx.blockSizeElementXYZ[2] = genCtx.fraction; 385755f3dfbSVijay Mahadevan break; 38649d66b22SVijay Mahadevan } 387ce27a4eeSVijay Mahadevan } 388ce27a4eeSVijay Mahadevan 389ce27a4eeSVijay Mahadevan /* partition only by the largest dimension */ 390ce27a4eeSVijay Mahadevan /* Total number of local elements := genCtx.blockSizeElementXYZ[0]*(genCtx.dim>1? genCtx.blockSizeElementXYZ[1]*(genCtx.dim>2 ? genCtx.blockSizeElementXYZ[2]:1) :1); */ 391ce27a4eeSVijay Mahadevan if (bounds) { 39249d66b22SVijay Mahadevan for (PetscInt i = 0; i < 6; i++) 393ce27a4eeSVijay Mahadevan genCtx.xyzbounds[i] = bounds[i]; 394ce27a4eeSVijay Mahadevan } 395ce27a4eeSVijay Mahadevan else { 396ce27a4eeSVijay Mahadevan genCtx.xyzbounds[0] = genCtx.xyzbounds[2] = genCtx.xyzbounds[4] = 0.0; 397ce27a4eeSVijay Mahadevan genCtx.xyzbounds[1] = genCtx.xyzbounds[3] = genCtx.xyzbounds[5] = 1.0; 398ce27a4eeSVijay Mahadevan } 399ce27a4eeSVijay Mahadevan 400ce27a4eeSVijay Mahadevan if (!genCtx.usrprocgrid) { 401ce27a4eeSVijay Mahadevan switch (genCtx.dim) { 402ce27a4eeSVijay Mahadevan case 1: 403ce27a4eeSVijay Mahadevan genCtx.M = nprocs; 404ce27a4eeSVijay Mahadevan genCtx.N = genCtx.K = 1; 405ce27a4eeSVijay Mahadevan break; 406ce27a4eeSVijay Mahadevan case 2: 407755f3dfbSVijay Mahadevan genCtx.N = nprocs; 40863d025dbSVijay Mahadevan genCtx.M = genCtx.K = 1; 409ce27a4eeSVijay Mahadevan break; 410ce27a4eeSVijay Mahadevan default: 411755f3dfbSVijay Mahadevan genCtx.K = nprocs; 41263d025dbSVijay Mahadevan genCtx.M = genCtx.N = 1; 413ce27a4eeSVijay Mahadevan break; 414ce27a4eeSVijay Mahadevan } 415ce27a4eeSVijay Mahadevan } 416ce27a4eeSVijay Mahadevan 417ce27a4eeSVijay Mahadevan if (!genCtx.usrrefgrid) { 418ce27a4eeSVijay Mahadevan genCtx.A = genCtx.B = genCtx.C = 1; 419ce27a4eeSVijay Mahadevan } 420ce27a4eeSVijay Mahadevan 421ce27a4eeSVijay Mahadevan /* more default values */ 422ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.ney = genCtx.nez = 0; 423ce27a4eeSVijay Mahadevan genCtx.xstride = genCtx.ystride = genCtx.zstride = 0; 424ce27a4eeSVijay Mahadevan genCtx.NX = genCtx.NY = genCtx.NZ = 0; 425ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.ney = genCtx.nez = 0; 426ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0] = genCtx.blockSizeVertexXYZ[1] = genCtx.blockSizeVertexXYZ[2] = 1; 427ce27a4eeSVijay Mahadevan 428ce27a4eeSVijay Mahadevan switch (genCtx.dim) { 429ce27a4eeSVijay Mahadevan case 3: 430ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0] = genCtx.q * genCtx.blockSizeElementXYZ[0] + 1; 431ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[1] = genCtx.q * genCtx.blockSizeElementXYZ[1] + 1; 432ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[2] = genCtx.q * genCtx.blockSizeElementXYZ[2] + 1; 433ce27a4eeSVijay Mahadevan 434ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0]; /* number of elements in x direction, used for global id on element */ 435755f3dfbSVijay Mahadevan genCtx.dx = (genCtx.xyzbounds[1] - genCtx.xyzbounds[0]) / (nelems * genCtx.q); /* distance between 2 nodes in x direction */ 436ce27a4eeSVijay Mahadevan genCtx.NX = (genCtx.q * genCtx.nex + 1); 437ce27a4eeSVijay Mahadevan genCtx.xstride = 1; 438ce27a4eeSVijay Mahadevan genCtx.ney = genCtx.N * genCtx.B * genCtx.blockSizeElementXYZ[1]; /* number of elements in y direction .... */ 439755f3dfbSVijay Mahadevan genCtx.dy = (genCtx.xyzbounds[3] - genCtx.xyzbounds[2]) / (nelems * genCtx.q); /* distance between 2 nodes in y direction */ 440ce27a4eeSVijay Mahadevan genCtx.NY = (genCtx.q * genCtx.ney + 1); 441ce27a4eeSVijay Mahadevan genCtx.ystride = genCtx.blockSizeVertexXYZ[0]; 442ce27a4eeSVijay Mahadevan genCtx.nez = genCtx.K * genCtx.C * genCtx.blockSizeElementXYZ[2]; /* number of elements in z direction .... */ 443755f3dfbSVijay Mahadevan genCtx.dz = (genCtx.xyzbounds[5] - genCtx.xyzbounds[4]) / (nelems * genCtx.q); /* distance between 2 nodes in z direction */ 444ce27a4eeSVijay Mahadevan genCtx.NZ = (genCtx.q * genCtx.nez + 1); 445ce27a4eeSVijay Mahadevan genCtx.zstride = genCtx.blockSizeVertexXYZ[0] * genCtx.blockSizeVertexXYZ[1]; 446ce27a4eeSVijay Mahadevan break; 447ce27a4eeSVijay Mahadevan case 2: 448ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0] = genCtx.q * genCtx.blockSizeElementXYZ[0] + 1; 449ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[1] = genCtx.q * genCtx.blockSizeElementXYZ[1] + 1; 4509daf19fdSVijay Mahadevan genCtx.blockSizeVertexXYZ[2] = 0; 451ce27a4eeSVijay Mahadevan 452ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0]; /* number of elements in x direction, used for global id on element */ 453ce27a4eeSVijay Mahadevan genCtx.dx = (genCtx.xyzbounds[1] - genCtx.xyzbounds[0]) / (genCtx.nex * genCtx.q); /* distance between 2 nodes in x direction */ 454ce27a4eeSVijay Mahadevan genCtx.NX = (genCtx.q * genCtx.nex + 1); 455ce27a4eeSVijay Mahadevan genCtx.xstride = 1; 456ce27a4eeSVijay Mahadevan genCtx.ney = genCtx.N * genCtx.B * genCtx.blockSizeElementXYZ[1]; /* number of elements in y direction .... */ 457755f3dfbSVijay Mahadevan genCtx.dy = (genCtx.xyzbounds[3] - genCtx.xyzbounds[2]) / (nelems * genCtx.q); /* distance between 2 nodes in y direction */ 458ce27a4eeSVijay Mahadevan genCtx.NY = (genCtx.q * genCtx.ney + 1); 459ce27a4eeSVijay Mahadevan genCtx.ystride = genCtx.blockSizeVertexXYZ[0]; 460ce27a4eeSVijay Mahadevan break; 461ce27a4eeSVijay Mahadevan case 1: 4629daf19fdSVijay Mahadevan genCtx.blockSizeVertexXYZ[1] = genCtx.blockSizeVertexXYZ[2] = 0; 463ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0] = genCtx.q * genCtx.blockSizeElementXYZ[0] + 1; 464ce27a4eeSVijay Mahadevan 465ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0]; /* number of elements in x direction, used for global id on element */ 466755f3dfbSVijay Mahadevan genCtx.dx = (genCtx.xyzbounds[1] - genCtx.xyzbounds[0]) / (nelems * genCtx.q); /* distance between 2 nodes in x direction */ 467ce27a4eeSVijay Mahadevan genCtx.NX = (genCtx.q * genCtx.nex + 1); 468ce27a4eeSVijay Mahadevan genCtx.xstride = 1; 469ce27a4eeSVijay Mahadevan break; 470ce27a4eeSVijay Mahadevan } 471ce27a4eeSVijay Mahadevan 472755f3dfbSVijay Mahadevan /* Lets check for some valid input */ 473755f3dfbSVijay Mahadevan if (genCtx.dim < 1 || genCtx.dim > 3) SETERRQ1(PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Invalid topological dimension specified: %d.\n", genCtx.dim); 474755f3dfbSVijay 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); 475755f3dfbSVijay Mahadevan /* validate the bounds data */ 476755f3dfbSVijay 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]); 477755f3dfbSVijay 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]); 478755f3dfbSVijay 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]); 479755f3dfbSVijay Mahadevan 480ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local elements:= %d, %d, %d\n", genCtx.blockSizeElementXYZ[0], genCtx.blockSizeElementXYZ[1], genCtx.blockSizeElementXYZ[2]); 481ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local vertices:= %d, %d, %d\n", genCtx.blockSizeVertexXYZ[0], genCtx.blockSizeVertexXYZ[1], genCtx.blockSizeVertexXYZ[2]); 482755f3dfbSVijay Mahadevan PetscInfo3(NULL, "Local blocks/processors := %d, %d, %d\n", genCtx.A, genCtx.B, genCtx.C); 483755f3dfbSVijay Mahadevan PetscInfo3(NULL, "Local processors := %d, %d, %d\n", genCtx.M, genCtx.N, genCtx.K); 484ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local nexyz:= %d, %d, %d\n", genCtx.nex, genCtx.ney, genCtx.nez); 485ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local delxyz:= %g, %g, %g\n", genCtx.dx, genCtx.dy, genCtx.dz); 486ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local strides:= %d, %d, %d\n", genCtx.xstride, genCtx.ystride, genCtx.zstride); 487ce27a4eeSVijay Mahadevan PetscFunctionReturn(0); 488ce27a4eeSVijay Mahadevan } 489ce27a4eeSVijay Mahadevan 490aa859218SVijay Mahadevan /*@ 491ce27a4eeSVijay Mahadevan DMMoabCreateBoxMesh - Creates a mesh on the tensor product (box) of intervals with genCtx specified bounds. 492aa859218SVijay Mahadevan 493aa859218SVijay Mahadevan Collective on MPI_Comm 494aa859218SVijay Mahadevan 495aa859218SVijay Mahadevan Input Parameters: 496aa859218SVijay Mahadevan + comm - The communicator for the DM object 497aa859218SVijay Mahadevan . dim - The spatial dimension 498b8ecf6d3SVijay 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 499b8ecf6d3SVijay Mahadevan . nele - The number of discrete elements in each direction 500*a2b725a8SWilliam Gropp - user_nghost - The number of ghosted layers needed in the partitioned mesh 501aa859218SVijay Mahadevan 502aa859218SVijay Mahadevan Output Parameter: 503aa859218SVijay Mahadevan . dm - The DM object 504aa859218SVijay Mahadevan 505aa859218SVijay Mahadevan Level: beginner 506aa859218SVijay Mahadevan 507aa859218SVijay Mahadevan .keywords: DM, create 508aa859218SVijay Mahadevan .seealso: DMSetType(), DMCreate(), DMMoabLoadFromFile() 509aa859218SVijay Mahadevan @*/ 510ce27a4eeSVijay Mahadevan PetscErrorCode DMMoabCreateBoxMesh(MPI_Comm comm, PetscInt dim, PetscBool useSimplex, const PetscReal* bounds, PetscInt nele, PetscInt nghost, DM *dm) 51151d15aeeSVijay Mahadevan { 51251d15aeeSVijay Mahadevan PetscErrorCode ierr; 51351d15aeeSVijay Mahadevan moab::ErrorCode merr; 5149daf19fdSVijay Mahadevan PetscInt a, b, c, n, global_size, global_rank; 51551d15aeeSVijay Mahadevan DM_Moab *dmmoab; 516ce27a4eeSVijay Mahadevan moab::Interface *mbImpl; 5179daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 51851d15aeeSVijay Mahadevan moab::ParallelComm *pcomm; 5199daf19fdSVijay Mahadevan #endif 520a4717116SVijay Mahadevan moab::ReadUtilIface *readMeshIface; 521ce27a4eeSVijay Mahadevan moab::Range verts, cells, edges, faces, adj, dim3, dim2; 522ce27a4eeSVijay Mahadevan DMMoabMeshGeneratorCtx genCtx; 523a4717116SVijay Mahadevan const PetscInt npts = nele + 1; /* Number of points in every dimension */ 524ce27a4eeSVijay Mahadevan 5256ea892caSVijay Mahadevan moab::Tag global_id_tag, part_tag, geom_tag, mat_tag, dir_tag, neu_tag; 526ce27a4eeSVijay Mahadevan moab::Range ownedvtx, ownedelms, localvtxs, localelms; 527ce27a4eeSVijay Mahadevan moab::EntityHandle regionset; 528755f3dfbSVijay Mahadevan PetscInt ml = 0, nl = 0, kl = 0; 52951d15aeeSVijay Mahadevan 53051d15aeeSVijay Mahadevan PetscFunctionBegin; 531e5136372SVijay Mahadevan if (dim < 1 || dim > 3) SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Invalid dimension argument for mesh: dim=[1,3].\n"); 532e5136372SVijay Mahadevan 533755f3dfbSVijay Mahadevan ierr = PetscLogEventRegister("GenerateMesh", DM_CLASSID, &genCtx.generateMesh);CHKERRQ(ierr); 534755f3dfbSVijay Mahadevan ierr = PetscLogEventRegister("AddVertices", DM_CLASSID, &genCtx.generateVertices);CHKERRQ(ierr); 535755f3dfbSVijay Mahadevan ierr = PetscLogEventRegister("AddElements", DM_CLASSID, &genCtx.generateElements);CHKERRQ(ierr); 536755f3dfbSVijay Mahadevan ierr = PetscLogEventRegister("ParResolve", DM_CLASSID, &genCtx.parResolve);CHKERRQ(ierr); 537755f3dfbSVijay Mahadevan ierr = PetscLogEventBegin(genCtx.generateMesh, 0, 0, 0, 0);CHKERRQ(ierr); 538ce27a4eeSVijay Mahadevan ierr = MPI_Comm_size(comm, &global_size);CHKERRQ(ierr); 539e5136372SVijay Mahadevan /* total number of vertices in all dimensions */ 540e5136372SVijay Mahadevan n = pow(npts, dim); 541e5136372SVijay Mahadevan 542e5136372SVijay Mahadevan /* do some error checking */ 543e5136372SVijay Mahadevan if (n < 2) SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Number of points must be >= 2.\n"); 544ce27a4eeSVijay 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"); 545e5136372SVijay Mahadevan if (nghost < 0) SETERRQ(PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Number of ghost layers cannot be negative.\n"); 546e5136372SVijay Mahadevan 547a4717116SVijay Mahadevan /* Create the basic DMMoab object and keep the default parameters created by DM impls */ 5489daf19fdSVijay Mahadevan ierr = DMMoabCreateMoab(comm, NULL, NULL, NULL, dm);CHKERRQ(ierr); 54951d15aeeSVijay Mahadevan 550a4717116SVijay Mahadevan /* get all the necessary handles from the private DM object */ 55151d15aeeSVijay Mahadevan dmmoab = (DM_Moab*)(*dm)->data; 552ce27a4eeSVijay Mahadevan mbImpl = dmmoab->mbiface; 5539daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 55451d15aeeSVijay Mahadevan pcomm = dmmoab->pcomm; 555ce27a4eeSVijay Mahadevan global_rank = pcomm->rank(); 5569daf19fdSVijay Mahadevan #else 5579daf19fdSVijay Mahadevan global_rank = 0; 5589daf19fdSVijay Mahadevan global_size = 1; 5599daf19fdSVijay Mahadevan #endif 5609daf19fdSVijay Mahadevan global_id_tag = dmmoab->ltog_tag; 561c8d5365dSVijay Mahadevan dmmoab->dim = dim; 56249d66b22SVijay Mahadevan dmmoab->nghostrings = nghost; 563304006b3SVijay Mahadevan dmmoab->refct = 1; 56451d15aeeSVijay Mahadevan 565b5410836SVijay Mahadevan /* create a file set to associate all entities in current mesh */ 566ce27a4eeSVijay Mahadevan merr = mbImpl->create_meshset(moab::MESHSET_SET, dmmoab->fileset);MBERR("Creating file set failed", merr); 567b5410836SVijay Mahadevan 568a4717116SVijay Mahadevan /* No errors yet; proceed with building the mesh */ 569ce27a4eeSVijay Mahadevan merr = mbImpl->query_interface(readMeshIface);MBERRNM(merr); 57051d15aeeSVijay Mahadevan 571755f3dfbSVijay Mahadevan genCtx.M = genCtx.N = genCtx.K = 1; 572755f3dfbSVijay Mahadevan genCtx.A = genCtx.B = genCtx.C = 1; 573755f3dfbSVijay Mahadevan 574755f3dfbSVijay Mahadevan ierr = PetscOptionsBegin(comm, "", "DMMoab Creation Options", "DMMOAB");CHKERRQ(ierr); 575ce27a4eeSVijay Mahadevan /* Handle DMMoab spatial resolution */ 576ce27a4eeSVijay Mahadevan PetscOptionsInt("-dmb_grid_x", "Number of grid points in x direction", "DMMoabSetSizes", genCtx.blockSizeElementXYZ[0], &genCtx.blockSizeElementXYZ[0], &genCtx.usrxyzgrid); 577ce27a4eeSVijay Mahadevan if (dim > 1) {PetscOptionsInt("-dmb_grid_y", "Number of grid points in y direction", "DMMoabSetSizes", genCtx.blockSizeElementXYZ[1], &genCtx.blockSizeElementXYZ[1], &genCtx.usrxyzgrid);} 578ce27a4eeSVijay Mahadevan if (dim > 2) {PetscOptionsInt("-dmb_grid_z", "Number of grid points in z direction", "DMMoabSetSizes", genCtx.blockSizeElementXYZ[2], &genCtx.blockSizeElementXYZ[2], &genCtx.usrxyzgrid);} 579a4717116SVijay Mahadevan 580ce27a4eeSVijay Mahadevan /* Handle DMMoab parallel distibution */ 581ce27a4eeSVijay Mahadevan PetscOptionsInt("-dmb_processors_x", "Number of processors in x direction", "DMMoabSetNumProcs", genCtx.M, &genCtx.M, &genCtx.usrprocgrid); 582ce27a4eeSVijay Mahadevan if (dim > 1) {PetscOptionsInt("-dmb_processors_y", "Number of processors in y direction", "DMMoabSetNumProcs", genCtx.N, &genCtx.N, &genCtx.usrprocgrid);} 583ce27a4eeSVijay Mahadevan if (dim > 2) {PetscOptionsInt("-dmb_processors_z", "Number of processors in z direction", "DMMoabSetNumProcs", genCtx.K, &genCtx.K, &genCtx.usrprocgrid);} 58451d15aeeSVijay Mahadevan 585ce27a4eeSVijay Mahadevan /* Handle DMMoab block refinement */ 586ce27a4eeSVijay Mahadevan PetscOptionsInt("-dmb_refine_x", "Number of refinement blocks in x direction", "DMMoabSetRefinement", genCtx.A, &genCtx.A, &genCtx.usrrefgrid); 587ce27a4eeSVijay Mahadevan if (dim > 1) {PetscOptionsInt("-dmb_refine_y", "Number of refinement blocks in y direction", "DMMoabSetRefinement", genCtx.B, &genCtx.B, &genCtx.usrrefgrid);} 588ce27a4eeSVijay Mahadevan if (dim > 2) {PetscOptionsInt("-dmb_refine_z", "Number of refinement blocks in z direction", "DMMoabSetRefinement", genCtx.C, &genCtx.C, &genCtx.usrrefgrid);} 589ce27a4eeSVijay Mahadevan PetscOptionsEnd(); 59051d15aeeSVijay Mahadevan 591ce27a4eeSVijay Mahadevan ierr = DMMBUtil_InitializeOptions(genCtx, dim, useSimplex, global_rank, global_size, bounds, nele);CHKERRQ(ierr); 59251d15aeeSVijay Mahadevan 59364e1c140SVijay 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); 59464e1c140SVijay Mahadevan 595ce27a4eeSVijay Mahadevan if (genCtx.adjEnts) genCtx.keep_skins = true; /* do not delete anything - consumes more memory */ 59666f88a78SVijay Mahadevan 597ce27a4eeSVijay Mahadevan /* determine m, n, k for processor rank */ 598755f3dfbSVijay Mahadevan ml = nl = kl = 0; 599755f3dfbSVijay Mahadevan switch (genCtx.dim) { 600755f3dfbSVijay Mahadevan case 1: 601755f3dfbSVijay Mahadevan ml = (genCtx.cumfraction); 602755f3dfbSVijay Mahadevan break; 603755f3dfbSVijay Mahadevan case 2: 604755f3dfbSVijay Mahadevan nl = (genCtx.cumfraction); 605755f3dfbSVijay Mahadevan break; 606755f3dfbSVijay Mahadevan default: 607755f3dfbSVijay Mahadevan kl = (genCtx.cumfraction) / genCtx.q / genCtx.blockSizeElementXYZ[2] / genCtx.C; //genCtx.K 608755f3dfbSVijay Mahadevan break; 609755f3dfbSVijay Mahadevan } 610e5136372SVijay Mahadevan 611ce27a4eeSVijay Mahadevan /* 612ce27a4eeSVijay Mahadevan * so there are a total of M * A * blockSizeElement elements in x direction (so M * A * blockSizeElement + 1 verts in x direction) 613ce27a4eeSVijay Mahadevan * so there are a total of N * B * blockSizeElement elements in y direction (so N * B * blockSizeElement + 1 verts in y direction) 614ce27a4eeSVijay Mahadevan * so there are a total of K * C * blockSizeElement elements in z direction (so K * C * blockSizeElement + 1 verts in z direction) 615c8d5365dSVijay Mahadevan 616ce27a4eeSVijay Mahadevan * there are ( M * A blockSizeElement ) * ( N * B * blockSizeElement) * (K * C * blockSizeElement ) hexas 617ce27a4eeSVijay Mahadevan * there are ( M * A * blockSizeElement + 1) * ( N * B * blockSizeElement + 1 ) * (K * C * blockSizeElement + 1) vertices 618ce27a4eeSVijay Mahadevan * x is the first dimension that varies 619ce27a4eeSVijay Mahadevan */ 620e5136372SVijay Mahadevan 621ce27a4eeSVijay Mahadevan /* generate the block at (a, b, c); it will represent a partition , it will get a partition tag */ 62249d66b22SVijay Mahadevan PetscInt dum_id = -1; 6236ea892caSVijay Mahadevan merr = mbImpl->tag_get_handle("GLOBAL_ID", 1, moab::MB_TYPE_INTEGER, global_id_tag);MBERR("Getting Global_ID Tag handle failed", merr); 62451d15aeeSVijay Mahadevan 6256ea892caSVijay Mahadevan merr = mbImpl->tag_get_handle(MATERIAL_SET_TAG_NAME, 1, moab::MB_TYPE_INTEGER, mat_tag);MBERR("Getting Material set Tag handle failed", merr); 6266ea892caSVijay Mahadevan merr = mbImpl->tag_get_handle(DIRICHLET_SET_TAG_NAME, 1, moab::MB_TYPE_INTEGER, dir_tag);MBERR("Getting Dirichlet set Tag handle failed", merr); 6276ea892caSVijay Mahadevan merr = mbImpl->tag_get_handle(NEUMANN_SET_TAG_NAME, 1, moab::MB_TYPE_INTEGER, neu_tag);MBERR("Getting Neumann set Tag handle failed", merr); 6286ea892caSVijay Mahadevan 6296ea892caSVijay 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 Partition Tag handle failed", merr); 63051d15aeeSVijay Mahadevan 6313a4aeca1SVijay Mahadevan /* lets create some sets */ 632ce27a4eeSVijay 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); 63349d66b22SVijay Mahadevan merr = mbImpl->create_meshset(moab::MESHSET_SET, regionset);MBERRNM(merr); 634755f3dfbSVijay Mahadevan ierr = PetscLogEventEnd(genCtx.generateMesh, 0, 0, 0, 0);CHKERRQ(ierr); 6353a4aeca1SVijay Mahadevan 636ce27a4eeSVijay Mahadevan for (a = 0; a < (genCtx.dim > 0 ? genCtx.A : genCtx.A); a++) { 637ce27a4eeSVijay Mahadevan for (b = 0; b < (genCtx.dim > 1 ? genCtx.B : 1); b++) { 638ce27a4eeSVijay Mahadevan for (c = 0; c < (genCtx.dim > 2 ? genCtx.C : 1); c++) { 63949d66b22SVijay Mahadevan 64049d66b22SVijay Mahadevan moab::EntityHandle startv; 64149d66b22SVijay Mahadevan 642755f3dfbSVijay Mahadevan ierr = PetscLogEventBegin(genCtx.generateVertices, 0, 0, 0, 0);CHKERRQ(ierr); 643ce27a4eeSVijay Mahadevan ierr = DMMoab_GenerateVertices_Private(mbImpl, readMeshIface, genCtx, ml, nl, kl, a, b, c, global_id_tag, startv, verts);CHKERRQ(ierr); 644755f3dfbSVijay Mahadevan ierr = PetscLogEventEnd(genCtx.generateVertices, 0, 0, 0, 0);CHKERRQ(ierr); 6453a4aeca1SVijay Mahadevan 646755f3dfbSVijay Mahadevan ierr = PetscLogEventBegin(genCtx.generateElements, 0, 0, 0, 0);CHKERRQ(ierr); 647ce27a4eeSVijay Mahadevan ierr = DMMoab_GenerateElements_Private(mbImpl, readMeshIface, genCtx, ml, nl, kl, a, b, c, global_id_tag, startv, cells);CHKERRQ(ierr); 648755f3dfbSVijay Mahadevan ierr = PetscLogEventEnd(genCtx.generateElements, 0, 0, 0, 0);CHKERRQ(ierr); 6493a4aeca1SVijay Mahadevan 65049d66b22SVijay Mahadevan PetscInt part_num = 0; 65149d66b22SVijay Mahadevan switch (genCtx.dim) { 65249d66b22SVijay Mahadevan case 3: 65349d66b22SVijay Mahadevan part_num += (c + kl * genCtx.C) * (genCtx.M * genCtx.A * genCtx.N * genCtx.B); 65449d66b22SVijay Mahadevan case 2: 65549d66b22SVijay Mahadevan part_num += (b + nl * genCtx.B) * (genCtx.M * genCtx.A); 65649d66b22SVijay Mahadevan case 1: 65749d66b22SVijay Mahadevan part_num += (a + ml * genCtx.A); 65849d66b22SVijay Mahadevan break; 65949d66b22SVijay Mahadevan } 66049d66b22SVijay Mahadevan 66149d66b22SVijay Mahadevan moab::EntityHandle part_set; 662ce27a4eeSVijay Mahadevan merr = mbImpl->create_meshset(moab::MESHSET_SET, part_set);MBERR("Can't create mesh set.", merr); 6633a4aeca1SVijay Mahadevan 664ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(part_set, verts);MBERR("Can't add vertices to set.", merr); 66549d66b22SVijay Mahadevan merr = mbImpl->add_entities(part_set, cells);MBERR("Can't add entities to set.", merr); 666ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(regionset, cells);MBERR("Can't add entities to set.", merr); 667ce27a4eeSVijay Mahadevan 668ce27a4eeSVijay Mahadevan /* if needed, add all edges and faces */ 669ce27a4eeSVijay Mahadevan if (genCtx.adjEnts) 670ce27a4eeSVijay Mahadevan { 671ce27a4eeSVijay Mahadevan if (genCtx.dim > 1) { 672ce27a4eeSVijay Mahadevan merr = mbImpl->get_adjacencies(cells, 1, true, edges, moab::Interface::UNION);MBERR("Can't get edges", merr); 673ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(part_set, edges);MBERR("Can't add edges to partition set.", merr); 674ce27a4eeSVijay Mahadevan } 675ce27a4eeSVijay Mahadevan if (genCtx.dim > 2) { 676ce27a4eeSVijay Mahadevan merr = mbImpl->get_adjacencies(cells, 2, true, faces, moab::Interface::UNION);MBERR("Can't get faces", merr); 677ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(part_set, faces);MBERR("Can't add faces to partition set.", merr); 678ce27a4eeSVijay Mahadevan } 679ce27a4eeSVijay Mahadevan edges.clear(); 680ce27a4eeSVijay Mahadevan faces.clear(); 681ce27a4eeSVijay Mahadevan } 68249d66b22SVijay Mahadevan verts.clear(); cells.clear(); 683ce27a4eeSVijay Mahadevan 684ce27a4eeSVijay Mahadevan merr = mbImpl->tag_set_data(part_tag, &part_set, 1, &part_num);MBERR("Can't set part tag on set", merr); 68549d66b22SVijay Mahadevan if (dmmoab->fileset) { 686ce27a4eeSVijay Mahadevan merr = mbImpl->add_parent_child(dmmoab->fileset, part_set);MBERR("Can't add part set to file set.", merr); 687ce27a4eeSVijay Mahadevan merr = mbImpl->unite_meshset(dmmoab->fileset, part_set);MBERRNM(merr); 68849d66b22SVijay Mahadevan } 68949d66b22SVijay Mahadevan merr = mbImpl->add_entities(dmmoab->fileset, &part_set, 1);MBERRNM(merr); 690ce27a4eeSVijay Mahadevan } 691ce27a4eeSVijay Mahadevan } 692ce27a4eeSVijay Mahadevan } 693ce27a4eeSVijay Mahadevan 69449d66b22SVijay Mahadevan merr = mbImpl->add_parent_child(dmmoab->fileset, regionset);MBERRNM(merr); 695ce27a4eeSVijay Mahadevan 69649d66b22SVijay Mahadevan /* Only in parallel: resolve shared entities between processors and exchange ghost layers */ 697ce27a4eeSVijay Mahadevan if (global_size > 1) { 698ce27a4eeSVijay Mahadevan 69977a8c067SVijay Mahadevan ierr = PetscLogEventBegin(genCtx.parResolve, 0, 0, 0, 0);CHKERRQ(ierr); 70077a8c067SVijay Mahadevan 701ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, genCtx.dim, cells);MBERR("Can't get all d-dimensional elements.", merr); 702ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 0, verts);MBERR("Can't get all vertices.", merr); 703ce27a4eeSVijay Mahadevan 704ce27a4eeSVijay Mahadevan if (genCtx.A * genCtx.B * genCtx.C != 1) { // merge needed 70549d66b22SVijay Mahadevan moab::MergeMesh mm(mbImpl); 706ce27a4eeSVijay Mahadevan if (genCtx.newMergeMethod) { 707ce27a4eeSVijay Mahadevan merr = mm.merge_using_integer_tag(verts, global_id_tag);MBERR("Can't merge with GLOBAL_ID tag", merr); 708ce27a4eeSVijay Mahadevan } 709ce27a4eeSVijay Mahadevan else { 710ce27a4eeSVijay Mahadevan merr = mm.merge_entities(cells, 0.0001);MBERR("Can't merge with coordinates", merr); 7113a4aeca1SVijay Mahadevan } 7123a4aeca1SVijay Mahadevan } 7133a4aeca1SVijay Mahadevan 7149daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 715a4717116SVijay Mahadevan /* check the handles */ 716ce27a4eeSVijay Mahadevan merr = pcomm->check_all_shared_handles();MBERRV(mbImpl, merr); 71751d15aeeSVijay Mahadevan 718a4717116SVijay Mahadevan /* resolve the shared entities by exchanging information to adjacent processors */ 719ce27a4eeSVijay Mahadevan merr = pcomm->resolve_shared_ents(dmmoab->fileset, cells, dim, dim - 1, NULL, &global_id_tag);MBERRV(mbImpl, merr); 72049d66b22SVijay Mahadevan if (dmmoab->fileset) { 721ce27a4eeSVijay Mahadevan merr = pcomm->exchange_ghost_cells(dim, 0, nghost, dim, true, false, &dmmoab->fileset);MBERRV(mbImpl, merr); 72249d66b22SVijay Mahadevan } 72349d66b22SVijay Mahadevan else { 72449d66b22SVijay Mahadevan merr = pcomm->exchange_ghost_cells(dim, 0, nghost, dim, true, false);MBERRV(mbImpl, merr); 72549d66b22SVijay Mahadevan } 726c8d5365dSVijay Mahadevan 727e427d9c9SVijay Mahadevan /* Reassign global IDs on all entities. */ 728e427d9c9SVijay Mahadevan merr = pcomm->assign_global_ids(dmmoab->fileset, dim, 1, false, true, false);MBERRNM(merr); 7299daf19fdSVijay Mahadevan #endif 73077a8c067SVijay Mahadevan 73177a8c067SVijay Mahadevan ierr = PetscLogEventEnd(genCtx.parResolve, 0, 0, 0, 0);CHKERRQ(ierr); 73249d66b22SVijay Mahadevan } 7333f1c6e43SVijay Mahadevan 734ce27a4eeSVijay Mahadevan if (!genCtx.keep_skins) { // default is to delete the 1- and 2-dimensional entities 735ce27a4eeSVijay Mahadevan // delete all quads and edges 736ce27a4eeSVijay Mahadevan moab::Range toDelete; 737ce27a4eeSVijay Mahadevan if (genCtx.dim > 1) { 738ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 1, toDelete);MBERR("Can't get edges", merr); 739ce27a4eeSVijay Mahadevan } 7403f1c6e43SVijay Mahadevan 741ce27a4eeSVijay Mahadevan if (genCtx.dim > 2) { 742ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 2, toDelete);MBERR("Can't get faces", merr); 743ce27a4eeSVijay Mahadevan } 744c8d5365dSVijay Mahadevan 7459daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 746ce27a4eeSVijay Mahadevan merr = dmmoab->pcomm->delete_entities(toDelete) ;MBERR("Can't delete entities", merr); 7479daf19fdSVijay Mahadevan #endif 748ce27a4eeSVijay Mahadevan } 7496ea892caSVijay Mahadevan 7506ea892caSVijay Mahadevan /* set geometric dimension tag for regions */ 7516ea892caSVijay Mahadevan merr = mbImpl->tag_set_data(geom_tag, ®ionset, 1, &dmmoab->dim);MBERRNM(merr); 7526ea892caSVijay Mahadevan /* set default material ID for regions */ 7536ea892caSVijay Mahadevan int default_material = 1; 7546ea892caSVijay Mahadevan merr = mbImpl->tag_set_data(mat_tag, ®ionset, 1, &default_material);MBERRNM(merr); 7556ea892caSVijay Mahadevan /* 7566ea892caSVijay Mahadevan int default_dbc = 0; 7576ea892caSVijay Mahadevan merr = mbImpl->tag_set_data(dir_tag, &vertexset, 1, &default_dbc);MBERRNM(merr); 7586ea892caSVijay Mahadevan */ 75951d15aeeSVijay Mahadevan PetscFunctionReturn(0); 76051d15aeeSVijay Mahadevan } 76151d15aeeSVijay Mahadevan 76251d15aeeSVijay Mahadevan 76349d66b22SVijay 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) 76451d15aeeSVijay Mahadevan { 7656acfe860SVijay Mahadevan char *ropts; 76649d66b22SVijay Mahadevan char ropts_par[PETSC_MAX_PATH_LEN], ropts_pargh[PETSC_MAX_PATH_LEN]; 76761eb6e27SVijay Mahadevan char ropts_dbg[PETSC_MAX_PATH_LEN]; 768f90c3b0eSVijay Mahadevan PetscErrorCode ierr; 76951d15aeeSVijay Mahadevan 770a4717116SVijay Mahadevan PetscFunctionBegin; 77195dccacaSBarry Smith ierr = PetscMalloc1(PETSC_MAX_PATH_LEN, &ropts);CHKERRQ(ierr); 77261eb6e27SVijay Mahadevan ierr = PetscMemzero(&ropts_par, PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 77349d66b22SVijay Mahadevan ierr = PetscMemzero(&ropts_pargh, PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 77461eb6e27SVijay Mahadevan ierr = PetscMemzero(&ropts_dbg, PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 77561eb6e27SVijay Mahadevan 776e23c60ebSVijay Mahadevan /* do parallel read unless using only one processor */ 777a4717116SVijay Mahadevan if (numproc > 1) { 77849d66b22SVijay 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); 77949d66b22SVijay 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); 78049d66b22SVijay Mahadevan if (nghost) { 78149d66b22SVijay Mahadevan ierr = PetscSNPrintf(ropts_pargh, PETSC_MAX_PATH_LEN, "PARALLEL_GHOSTS=%d.0.%d;", dim, nghost);CHKERRQ(ierr); 78249d66b22SVijay Mahadevan } 7832e4e7c01SVijay Mahadevan } 7842e4e7c01SVijay Mahadevan 785c8d5365dSVijay Mahadevan if (dbglevel) { 786f90c3b0eSVijay Mahadevan if (numproc > 1) { 78749d66b22SVijay Mahadevan ierr = PetscSNPrintf(ropts_dbg, PETSC_MAX_PATH_LEN, "CPUTIME;DEBUG_IO=%d;DEBUG_PIO=%d;", dbglevel, dbglevel);CHKERRQ(ierr); 78861eb6e27SVijay Mahadevan } 78961eb6e27SVijay Mahadevan else { 79049d66b22SVijay Mahadevan ierr = PetscSNPrintf(ropts_dbg, PETSC_MAX_PATH_LEN, "CPUTIME;DEBUG_IO=%d;", dbglevel);CHKERRQ(ierr); 791f90c3b0eSVijay Mahadevan } 792c8d5365dSVijay Mahadevan } 79351d15aeeSVijay Mahadevan 79449d66b22SVijay 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); 795f90c3b0eSVijay Mahadevan *read_opts = ropts; 79651d15aeeSVijay Mahadevan PetscFunctionReturn(0); 79751d15aeeSVijay Mahadevan } 79851d15aeeSVijay Mahadevan 79951d15aeeSVijay Mahadevan 800aa859218SVijay Mahadevan /*@ 801aa859218SVijay Mahadevan DMMoabLoadFromFile - Creates a DM object by loading the mesh from a user specified file. 802aa859218SVijay Mahadevan 803aa859218SVijay Mahadevan Collective on MPI_Comm 804aa859218SVijay Mahadevan 805aa859218SVijay Mahadevan Input Parameters: 806aa859218SVijay Mahadevan + comm - The communicator for the DM object 807aa859218SVijay Mahadevan . dim - The spatial dimension 808b8ecf6d3SVijay Mahadevan . filename - The name of the mesh file to be loaded 809*a2b725a8SWilliam Gropp - usrreadopts - The options string to read a MOAB mesh. 810*a2b725a8SWilliam Gropp 811aa859218SVijay Mahadevan Reference (Parallel Mesh Initialization: http://www.mcs.anl.gov/~fathom/moab-docs/html/contents.html#fivetwo) 812aa859218SVijay Mahadevan 813aa859218SVijay Mahadevan Output Parameter: 814aa859218SVijay Mahadevan . dm - The DM object 815aa859218SVijay Mahadevan 816aa859218SVijay Mahadevan Level: beginner 817aa859218SVijay Mahadevan 818aa859218SVijay Mahadevan .keywords: DM, create 819b8ecf6d3SVijay Mahadevan 820aa859218SVijay Mahadevan .seealso: DMSetType(), DMCreate(), DMMoabCreateBoxMesh() 821aa859218SVijay Mahadevan @*/ 82249d66b22SVijay Mahadevan PetscErrorCode DMMoabLoadFromFile(MPI_Comm comm, PetscInt dim, PetscInt nghost, const char* filename, const char* usrreadopts, DM *dm) 82351d15aeeSVijay Mahadevan { 824a4717116SVijay Mahadevan moab::ErrorCode merr; 8252e4e7c01SVijay Mahadevan PetscInt nprocs; 826a4717116SVijay Mahadevan DM_Moab *dmmoab; 827a4717116SVijay Mahadevan moab::Interface *mbiface; 8289daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 829a4717116SVijay Mahadevan moab::ParallelComm *pcomm; 8309daf19fdSVijay Mahadevan #endif 831a4717116SVijay Mahadevan moab::Range verts, elems; 8327023aa44SVijay Mahadevan const char *readopts; 833a4717116SVijay Mahadevan PetscErrorCode ierr; 83451d15aeeSVijay Mahadevan 83551d15aeeSVijay Mahadevan PetscFunctionBegin; 83649d66b22SVijay Mahadevan PetscValidPointer(dm, 6); 83751d15aeeSVijay Mahadevan 838a4717116SVijay Mahadevan /* Create the basic DMMoab object and keep the default parameters created by DM impls */ 8399daf19fdSVijay Mahadevan ierr = DMMoabCreateMoab(comm, NULL, NULL, NULL, dm);CHKERRQ(ierr); 84051d15aeeSVijay Mahadevan 841a4717116SVijay Mahadevan /* get all the necessary handles from the private DM object */ 842a4717116SVijay Mahadevan dmmoab = (DM_Moab*)(*dm)->data; 843a4717116SVijay Mahadevan mbiface = dmmoab->mbiface; 8449daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 845a4717116SVijay Mahadevan pcomm = dmmoab->pcomm; 846a4717116SVijay Mahadevan nprocs = pcomm->size(); 8479daf19fdSVijay Mahadevan #else 8489daf19fdSVijay Mahadevan nprocs = 1; 8499daf19fdSVijay Mahadevan #endif 850aa859218SVijay Mahadevan /* TODO: Decipher dimension based on the loaded mesh instead of getting from user */ 851c8d5365dSVijay Mahadevan dmmoab->dim = dim; 85249d66b22SVijay Mahadevan dmmoab->nghostrings = nghost; 853304006b3SVijay Mahadevan dmmoab->refct = 1; 854a4717116SVijay Mahadevan 855b5410836SVijay Mahadevan /* create a file set to associate all entities in current mesh */ 856b5410836SVijay Mahadevan merr = dmmoab->mbiface->create_meshset(moab::MESHSET_SET, dmmoab->fileset);MBERR("Creating file set failed", merr); 857b5410836SVijay Mahadevan 858a4717116SVijay Mahadevan /* add mesh loading options specific to the DM */ 85949d66b22SVijay Mahadevan ierr = DMMoab_GetReadOptions_Private(dmmoab->partition_by_rank, nprocs, dim, nghost, dmmoab->read_mode, 8602e4e7c01SVijay Mahadevan dmmoab->rw_dbglevel, dmmoab->extra_read_options, usrreadopts, &readopts);CHKERRQ(ierr); 8617023aa44SVijay Mahadevan 862e5136372SVijay Mahadevan PetscInfo2(*dm, "Reading file %s with options: %s\n", filename, readopts); 863a4717116SVijay Mahadevan 864a4717116SVijay Mahadevan /* Load the mesh from a file. */ 86549d66b22SVijay Mahadevan if (dmmoab->fileset) { 8667023aa44SVijay Mahadevan merr = mbiface->load_file(filename, &dmmoab->fileset, readopts);MBERRVM(mbiface, "Reading MOAB file failed.", merr); 86749d66b22SVijay Mahadevan } 86849d66b22SVijay Mahadevan else { 86949d66b22SVijay Mahadevan merr = mbiface->load_file(filename, 0, readopts);MBERRVM(mbiface, "Reading MOAB file failed.", merr); 87049d66b22SVijay Mahadevan } 871a4717116SVijay Mahadevan 8729daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 8736e40195eSVijay Mahadevan /* Reassign global IDs on all entities. */ 8746ea892caSVijay Mahadevan /* merr = pcomm->assign_global_ids(dmmoab->fileset, dim, 1, true, true, true);MBERRNM(merr); */ 8759daf19fdSVijay Mahadevan #endif 876e5136372SVijay Mahadevan 877e5136372SVijay Mahadevan /* load the local vertices */ 878e5136372SVijay Mahadevan merr = mbiface->get_entities_by_type(dmmoab->fileset, moab::MBVERTEX, verts, true);MBERRNM(merr); 879e5136372SVijay Mahadevan /* load the local elements */ 880e5136372SVijay Mahadevan merr = mbiface->get_entities_by_dimension(dmmoab->fileset, dim, elems, true);MBERRNM(merr); 881e5136372SVijay Mahadevan 8829daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 883e5136372SVijay Mahadevan /* Everything is set up, now just do a tag exchange to update tags 884e5136372SVijay Mahadevan on all of the ghost vertexes */ 885e5136372SVijay Mahadevan merr = pcomm->exchange_tags(dmmoab->ltog_tag, verts);MBERRV(mbiface, merr); 886e5136372SVijay Mahadevan merr = pcomm->exchange_tags(dmmoab->ltog_tag, elems);MBERRV(mbiface, merr); 887a4717116SVijay Mahadevan merr = pcomm->collective_sync_partition();MBERR("Collective sync failed", merr); 8889daf19fdSVijay Mahadevan #endif 889a4717116SVijay Mahadevan 890e5136372SVijay Mahadevan PetscInfo3(*dm, "MOAB file '%s' was successfully loaded. Found %D vertices and %D elements.\n", filename, verts.size(), elems.size()); 8916acfe860SVijay Mahadevan ierr = PetscFree(readopts);CHKERRQ(ierr); 89251d15aeeSVijay Mahadevan PetscFunctionReturn(0); 89351d15aeeSVijay Mahadevan } 89451d15aeeSVijay Mahadevan 895304006b3SVijay Mahadevan 896304006b3SVijay Mahadevan /*@ 897304006b3SVijay Mahadevan DMMoabRenumberMeshEntities - Order and number all entities (vertices->elements) to be contiguously ordered 898304006b3SVijay Mahadevan in parallel 899304006b3SVijay Mahadevan 900304006b3SVijay Mahadevan Collective on MPI_Comm 901304006b3SVijay Mahadevan 902304006b3SVijay Mahadevan Input Parameters: 903*a2b725a8SWilliam Gropp . dm - The DM object 904304006b3SVijay Mahadevan 905304006b3SVijay Mahadevan Level: advanced 906304006b3SVijay Mahadevan 907304006b3SVijay Mahadevan .keywords: DM, reorder 908304006b3SVijay Mahadevan 909304006b3SVijay Mahadevan .seealso: DMSetUp(), DMCreate() 910304006b3SVijay Mahadevan @*/ 911304006b3SVijay Mahadevan PetscErrorCode DMMoabRenumberMeshEntities(DM dm) 912304006b3SVijay Mahadevan { 913304006b3SVijay Mahadevan moab::Range verts; 914304006b3SVijay Mahadevan 915304006b3SVijay Mahadevan PetscFunctionBegin; 916304006b3SVijay Mahadevan PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 917304006b3SVijay Mahadevan 9189daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI 919304006b3SVijay Mahadevan /* Insert new points */ 9209daf19fdSVijay Mahadevan moab::ErrorCode merr; 9219daf19fdSVijay Mahadevan merr = ((DM_Moab*) dm->data)->pcomm->assign_global_ids(((DM_Moab*) dm->data)->fileset, 3, 0, false, true, false);MBERRNM(merr); 9229daf19fdSVijay Mahadevan #endif 923304006b3SVijay Mahadevan PetscFunctionReturn(0); 924304006b3SVijay Mahadevan } 925304006b3SVijay Mahadevan 926