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> 7*ce27a4eeSVijay Mahadevan #include <moab/MergeMesh.hpp> 851d15aeeSVijay Mahadevan #include <moab/CN.hpp> 951d15aeeSVijay Mahadevan 10*ce27a4eeSVijay Mahadevan typedef struct { 11*ce27a4eeSVijay Mahadevan // options 12*ce27a4eeSVijay Mahadevan PetscInt A,B,C,M,N,K,dim; 13*ce27a4eeSVijay Mahadevan PetscInt blockSizeVertexXYZ[3]; // Number of element blocks per partition 14*ce27a4eeSVijay Mahadevan PetscInt blockSizeElementXYZ[3]; 15*ce27a4eeSVijay Mahadevan PetscReal xyzbounds[6]; // the physical size of the domain 16*ce27a4eeSVijay Mahadevan bool newMergeMethod,keep_skins,simplex,adjEnts; 1751d15aeeSVijay Mahadevan 18*ce27a4eeSVijay Mahadevan // compute params 19*ce27a4eeSVijay Mahadevan PetscReal dx,dy,dz; 20*ce27a4eeSVijay Mahadevan PetscInt NX,NY,NZ,nex,ney,nez; 21*ce27a4eeSVijay Mahadevan PetscInt q,xstride,ystride,zstride; 22*ce27a4eeSVijay Mahadevan PetscBool usrxyzgrid,usrprocgrid,usrrefgrid; 23*ce27a4eeSVijay Mahadevan } DMMoabMeshGeneratorCtx; 24*ce27a4eeSVijay Mahadevan 25*ce27a4eeSVijay Mahadevan 26*ce27a4eeSVijay Mahadevan #undef __FUNCT__ 27*ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMoab_SetTensorElementConnectivity_Private" 28*ce27a4eeSVijay Mahadevan int DMMoab_SetTensorElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, int offset, int corner, std::vector<int>& subent_conn, moab::EntityHandle *connectivity) 2951d15aeeSVijay Mahadevan { 30*ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 31e5136372SVijay Mahadevan case 1: 32*ce27a4eeSVijay Mahadevan subent_conn.resize(2); 33*ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBEDGE, 1, 0, subent_conn.data()); 34*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 35*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1; 36e5136372SVijay Mahadevan break; 37e5136372SVijay Mahadevan case 2: 38*ce27a4eeSVijay Mahadevan subent_conn.resize(4); 39*ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBQUAD, 2, 0, subent_conn.data()); 40*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 41*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1; 42*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner + 1 + genCtx.ystride; 43*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[3]] = corner + genCtx.ystride; 44e5136372SVijay Mahadevan break; 45e5136372SVijay Mahadevan case 3: 46e5136372SVijay Mahadevan default: 47*ce27a4eeSVijay Mahadevan subent_conn.resize(8); 48*ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBHEX, 3, 0, subent_conn.data()); 49*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 50*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1; 51*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner + 1 + genCtx.ystride; 52*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[3]] = corner + genCtx.ystride; 53*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[4]] = corner + genCtx.zstride; 54*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[5]] = corner + 1 + genCtx.zstride; 55*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[6]] = corner + 1 + genCtx.ystride + genCtx.zstride; 56*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[7]] = corner + genCtx.ystride + genCtx.zstride; 57e5136372SVijay Mahadevan break; 58e5136372SVijay Mahadevan } 59*ce27a4eeSVijay Mahadevan return subent_conn.size(); 60e5136372SVijay Mahadevan } 6151d15aeeSVijay Mahadevan 62*ce27a4eeSVijay Mahadevan 63*ce27a4eeSVijay Mahadevan #undef __FUNCT__ 64*ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMoab_SetSimplexElementConnectivity_Private" 65*ce27a4eeSVijay Mahadevan int DMMoab_SetSimplexElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, int subelem, int offset, int corner, std::vector<int>& subent_conn, moab::EntityHandle *connectivity) 66c3dd00c7SVijay Mahadevan { 67*ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 68c3dd00c7SVijay Mahadevan case 1: 69*ce27a4eeSVijay Mahadevan subent_conn.resize(2); /* only linear EDGE supported now */ 70*ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBEDGE, 1, 0, subent_conn.data()); 71*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 72*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner+1; 73c3dd00c7SVijay Mahadevan break; 74c3dd00c7SVijay Mahadevan case 2: 75*ce27a4eeSVijay Mahadevan subent_conn.resize(3); /* only linear TRI supported */ 76*ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBTRI, 2, 0, subent_conn.data()); 77*ce27a4eeSVijay Mahadevan if (subelem) { /* 0 1 2 of a QUAD */ 78*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 79*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1; 80*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner + 1 + genCtx.ystride; 81c3dd00c7SVijay Mahadevan } 82*ce27a4eeSVijay Mahadevan else { /* 2 3 0 of a QUAD */ 83*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner + 1 + genCtx.ystride; 84*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + genCtx.ystride; 85*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner; 86c3dd00c7SVijay Mahadevan } 87c3dd00c7SVijay Mahadevan break; 88c3dd00c7SVijay Mahadevan case 3: 89c3dd00c7SVijay Mahadevan default: 90*ce27a4eeSVijay Mahadevan subent_conn.resize(4); /* only linear TET supported */ 91*ce27a4eeSVijay Mahadevan moab::CN::SubEntityVertexIndices(moab::MBTET, 3, 0, subent_conn.data()); 92c3dd00c7SVijay Mahadevan switch(subelem) { 93c3dd00c7SVijay Mahadevan case 0: /* 0 1 2 5 of a HEX */ 94*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 95*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1; 96*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner + 1 + genCtx.ystride; 97*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[3]] = corner + 1 + genCtx.zstride; 98c3dd00c7SVijay Mahadevan break; 99c3dd00c7SVijay Mahadevan case 1: /* 0 2 7 5 of a HEX */ 100*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 101*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1 + genCtx.ystride; 102*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner + genCtx.ystride + genCtx.zstride; 103*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[3]] = corner + 1 + genCtx.zstride; 104c3dd00c7SVijay Mahadevan break; 105c3dd00c7SVijay Mahadevan case 2: /* 0 2 3 7 of a HEX */ 106*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 107*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1 + genCtx.ystride; 108*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner + genCtx.ystride; 109*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[3]] = corner + genCtx.ystride + genCtx.zstride; 110c3dd00c7SVijay Mahadevan break; 111c3dd00c7SVijay Mahadevan case 3: /* 0 5 7 4 of a HEX */ 112*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner; 113*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + 1 + genCtx.zstride; 114*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner + genCtx.ystride + genCtx.zstride; 115*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[3]] = corner + genCtx.zstride; 116c3dd00c7SVijay Mahadevan break; 117c3dd00c7SVijay Mahadevan case 4: /* 2 7 5 6 of a HEX */ 118*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[0]] = corner + 1 + genCtx.ystride; 119*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[1]] = corner + genCtx.ystride + genCtx.zstride; 120*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[2]] = corner + 1 + genCtx.zstride; 121*ce27a4eeSVijay Mahadevan connectivity[offset+subent_conn[3]] = corner + 1 + genCtx.ystride + genCtx.zstride; 122c3dd00c7SVijay Mahadevan break; 123c3dd00c7SVijay Mahadevan } 124c3dd00c7SVijay Mahadevan break; 125c3dd00c7SVijay Mahadevan } 126*ce27a4eeSVijay Mahadevan return subent_conn.size(); 127c3dd00c7SVijay Mahadevan } 128c3dd00c7SVijay Mahadevan 129*ce27a4eeSVijay Mahadevan 130*ce27a4eeSVijay Mahadevan #undef __FUNCT__ 131*ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMoab_SetElementConnectivity_Private" 132*ce27a4eeSVijay Mahadevan std::pair<int,int> DMMoab_SetElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, int offset, int corner, moab::EntityHandle *connectivity) 133c3dd00c7SVijay Mahadevan { 134*ce27a4eeSVijay Mahadevan int vcount=0; 135*ce27a4eeSVijay Mahadevan std::vector<int> subent_conn; /* only linear edge, tri, tet supported now */ 136*ce27a4eeSVijay Mahadevan subent_conn.reserve(27); 137*ce27a4eeSVijay Mahadevan int m,subelem; 138*ce27a4eeSVijay Mahadevan if (genCtx.simplex) { 139*ce27a4eeSVijay Mahadevan subelem=(genCtx.dim==1 ? 1 : (genCtx.dim==2 ? 2 : 5)); 140*ce27a4eeSVijay Mahadevan for (m=0; m<subelem; m++) { 141*ce27a4eeSVijay Mahadevan vcount=DMMoab_SetSimplexElementConnectivity_Private(genCtx, m, offset, corner, subent_conn, connectivity); 142*ce27a4eeSVijay Mahadevan offset+=vcount; 143*ce27a4eeSVijay Mahadevan } 144c3dd00c7SVijay Mahadevan } 145c3dd00c7SVijay Mahadevan else { 146c3dd00c7SVijay Mahadevan subelem=1; 147*ce27a4eeSVijay Mahadevan vcount=DMMoab_SetTensorElementConnectivity_Private(genCtx, offset, corner, subent_conn, connectivity); 148c3dd00c7SVijay Mahadevan } 149*ce27a4eeSVijay Mahadevan return std::pair<int,int>(vcount*subelem,subelem); 150c3dd00c7SVijay Mahadevan } 151c3dd00c7SVijay Mahadevan 152c3dd00c7SVijay Mahadevan 153*ce27a4eeSVijay Mahadevan #undef __FUNCT__ 154*ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMoab_GenerateVertices_Private" 155*ce27a4eeSVijay Mahadevan PetscErrorCode DMMoab_GenerateVertices_Private(moab::Interface *mbImpl, moab::ReadUtilIface *iface, DMMoabMeshGeneratorCtx& genCtx, int m, int n, int k, 156*ce27a4eeSVijay Mahadevan int a, int b, int c, moab::Tag& global_id_tag, moab::EntityHandle& startv, moab::Range& uverts) 157*ce27a4eeSVijay Mahadevan { 158*ce27a4eeSVijay Mahadevan int x,y,z,ix,nnodes; 159*ce27a4eeSVijay Mahadevan int ii,jj,kk; 160*ce27a4eeSVijay Mahadevan std::vector<double*> arrays; 161*ce27a4eeSVijay Mahadevan std::vector<int> gids; 162*ce27a4eeSVijay Mahadevan moab::ErrorCode merr; 163*ce27a4eeSVijay Mahadevan 164*ce27a4eeSVijay Mahadevan PetscFunctionBegin; 165*ce27a4eeSVijay Mahadevan /* we will generate (q*block+1)^3 vertices, and block^3 hexas; q is 1 for linear, 2 for quadratic 166*ce27a4eeSVijay Mahadevan * the global id of the vertices will come from m, n, k, a, b, c 167*ce27a4eeSVijay Mahadevan * x will vary from m*A*q*block + a*q*block to m*A*q*block+(a+1)*q*block etc. 168*ce27a4eeSVijay Mahadevan */ 169*ce27a4eeSVijay Mahadevan nnodes = genCtx.blockSizeVertexXYZ[0]*(genCtx.dim>1? genCtx.blockSizeVertexXYZ[1]*(genCtx.dim>2 ? genCtx.blockSizeVertexXYZ[2]:1) :1); 170*ce27a4eeSVijay Mahadevan 171*ce27a4eeSVijay Mahadevan merr = iface->get_node_coords(3, nnodes, 0, startv, arrays);MBERR("Can't get node coords.", merr); 172*ce27a4eeSVijay Mahadevan 173*ce27a4eeSVijay Mahadevan /* will start with the lower corner: */ 174*ce27a4eeSVijay Mahadevan x = m * genCtx.A * genCtx.q * genCtx.blockSizeElementXYZ[0] + a * genCtx.q * genCtx.blockSizeElementXYZ[0]; 175*ce27a4eeSVijay Mahadevan y = n * genCtx.B * genCtx.q * genCtx.blockSizeElementXYZ[1] + b * genCtx.q * genCtx.blockSizeElementXYZ[1]; 176*ce27a4eeSVijay Mahadevan z = k * genCtx.C * genCtx.q * genCtx.blockSizeElementXYZ[2] + c * genCtx.q * genCtx.blockSizeElementXYZ[2]; 177*ce27a4eeSVijay Mahadevan ix = 0; 178*ce27a4eeSVijay Mahadevan gids.resize(nnodes); 179*ce27a4eeSVijay Mahadevan moab::Range verts(startv, startv + nnodes - 1); 180*ce27a4eeSVijay Mahadevan for (kk = 0; kk < (genCtx.dim>2?genCtx.blockSizeVertexXYZ[2]:1); kk++) { 181*ce27a4eeSVijay Mahadevan for (jj = 0; jj < (genCtx.dim>1?genCtx.blockSizeVertexXYZ[1]:1); jj++) { 182*ce27a4eeSVijay Mahadevan for (ii = 0; ii < genCtx.blockSizeVertexXYZ[0]; ii++,ix++) { 183*ce27a4eeSVijay Mahadevan /* set coordinates for the vertices */ 184*ce27a4eeSVijay Mahadevan arrays[0][ix] = (x + ii)*genCtx.dx + genCtx.xyzbounds[0]; 185*ce27a4eeSVijay Mahadevan arrays[1][ix] = (y + jj)*genCtx.dy + genCtx.xyzbounds[2]; 186*ce27a4eeSVijay Mahadevan arrays[2][ix] = (z + kk)*genCtx.dz + genCtx.xyzbounds[4]; 187*ce27a4eeSVijay Mahadevan 188*ce27a4eeSVijay Mahadevan /* If we want to set some tags on the vertices -> use the following entity handle definition: 189*ce27a4eeSVijay Mahadevan moab::EntityHandle v = startv + ix; 190*ce27a4eeSVijay Mahadevan */ 191*ce27a4eeSVijay Mahadevan /* compute the global ID for vertex */ 192*ce27a4eeSVijay Mahadevan gids[ix] = 1 + (x + ii) + (y + jj) * genCtx.NX + (z + kk) * (genCtx.NX * genCtx.NY); 193*ce27a4eeSVijay Mahadevan } 194*ce27a4eeSVijay Mahadevan } 195*ce27a4eeSVijay Mahadevan } 196*ce27a4eeSVijay Mahadevan /* set global ID data on vertices */ 197*ce27a4eeSVijay Mahadevan mbImpl->tag_set_data(global_id_tag, verts, &gids[0]); 198*ce27a4eeSVijay Mahadevan verts.swap(uverts); 199*ce27a4eeSVijay Mahadevan PetscFunctionReturn(0); 200*ce27a4eeSVijay Mahadevan } 201*ce27a4eeSVijay Mahadevan 202*ce27a4eeSVijay Mahadevan #undef __FUNCT__ 203*ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMoab_GenerateElements_Private" 204*ce27a4eeSVijay Mahadevan PetscErrorCode DMMoab_GenerateElements_Private(moab::Interface* mbImpl, moab::ReadUtilIface* iface, DMMoabMeshGeneratorCtx& genCtx, int m, int n, int k, 205*ce27a4eeSVijay Mahadevan int a, int b, int c, moab::Tag& global_id_tag, moab::EntityHandle startv, moab::Range& cells) 206*ce27a4eeSVijay Mahadevan { 207*ce27a4eeSVijay Mahadevan moab::ErrorCode merr; 208*ce27a4eeSVijay Mahadevan PetscInt ix,ie,xe,ye,ze; 209*ce27a4eeSVijay Mahadevan PetscInt ii,jj,kk,nvperelem; 210*ce27a4eeSVijay 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);*/ 211*ce27a4eeSVijay Mahadevan PetscInt nelems = ntensorelems; 212*ce27a4eeSVijay Mahadevan moab::EntityHandle starte; // connectivity 213*ce27a4eeSVijay Mahadevan moab::EntityHandle* conn; 214*ce27a4eeSVijay Mahadevan 215*ce27a4eeSVijay Mahadevan PetscFunctionBegin; 216*ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 217*ce27a4eeSVijay Mahadevan case 1: 218*ce27a4eeSVijay Mahadevan nvperelem = 2; 219*ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 2, moab::MBEDGE, 0, starte, conn);MBERR("Can't get EDGE2 element connectivity.", merr); 220*ce27a4eeSVijay Mahadevan break; 221*ce27a4eeSVijay Mahadevan case 2: 222*ce27a4eeSVijay Mahadevan if (genCtx.simplex) { 223*ce27a4eeSVijay Mahadevan nvperelem = 3; 224*ce27a4eeSVijay Mahadevan nelems = ntensorelems*2; 225*ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 3, moab::MBTRI, 0, starte, conn);MBERR("Can't get TRI3 element connectivity.", merr); 226*ce27a4eeSVijay Mahadevan } 227*ce27a4eeSVijay Mahadevan else { 228*ce27a4eeSVijay Mahadevan nvperelem = 4; 229*ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 4, moab::MBQUAD, 0, starte, conn);MBERR("Can't get QUAD4 element connectivity.", merr); 230*ce27a4eeSVijay Mahadevan } 231*ce27a4eeSVijay Mahadevan break; 232*ce27a4eeSVijay Mahadevan case 3: 233*ce27a4eeSVijay Mahadevan default: 234*ce27a4eeSVijay Mahadevan if (genCtx.simplex) { 235*ce27a4eeSVijay Mahadevan nvperelem = 4; 236*ce27a4eeSVijay Mahadevan nelems = ntensorelems*5; 237*ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 4, moab::MBTET, 0, starte, conn);MBERR("Can't get TET4 element connectivity.", merr); 238*ce27a4eeSVijay Mahadevan } 239*ce27a4eeSVijay Mahadevan else { 240*ce27a4eeSVijay Mahadevan nvperelem = 8; 241*ce27a4eeSVijay Mahadevan merr = iface->get_element_connect(nelems, 8, moab::MBHEX, 0, starte, conn);MBERR("Can't get HEX8 element connectivity.", merr); 242*ce27a4eeSVijay Mahadevan } 243*ce27a4eeSVijay Mahadevan break; 244*ce27a4eeSVijay Mahadevan } 245*ce27a4eeSVijay Mahadevan 246*ce27a4eeSVijay Mahadevan ix = ie = 0; // index now in the elements, for global ids 247*ce27a4eeSVijay Mahadevan 248*ce27a4eeSVijay Mahadevan /* create a temporary range to store local element handles */ 249*ce27a4eeSVijay Mahadevan moab::Range tmp(starte, starte + nelems - 1); 250*ce27a4eeSVijay Mahadevan std::vector<int> gids(nelems); 251*ce27a4eeSVijay Mahadevan 252*ce27a4eeSVijay Mahadevan /* identify the elements at the lower corner, for their global ids */ 253*ce27a4eeSVijay Mahadevan xe = m * genCtx.A * genCtx.blockSizeElementXYZ[0] + a * genCtx.blockSizeElementXYZ[0]; 254*ce27a4eeSVijay Mahadevan ye = (genCtx.dim > 1 ? n * genCtx.B * genCtx.blockSizeElementXYZ[1] + b * genCtx.blockSizeElementXYZ[1] : 0); 255*ce27a4eeSVijay Mahadevan ze = (genCtx.dim > 2 ? k * genCtx.C * genCtx.blockSizeElementXYZ[2] + c * genCtx.blockSizeElementXYZ[2] : 0); 256*ce27a4eeSVijay Mahadevan 257*ce27a4eeSVijay Mahadevan /* create owned elements requested by genCtx */ 258*ce27a4eeSVijay Mahadevan for (kk = 0; kk < (genCtx.dim>2?genCtx.blockSizeElementXYZ[2]:1); kk++) { 259*ce27a4eeSVijay Mahadevan for (jj = 0; jj < (genCtx.dim>1?genCtx.blockSizeElementXYZ[1]:1); jj++) { 260*ce27a4eeSVijay Mahadevan for (ii = 0; ii < genCtx.blockSizeElementXYZ[0]; ii++) { 261*ce27a4eeSVijay Mahadevan 262*ce27a4eeSVijay Mahadevan moab::EntityHandle corner = startv + genCtx.q * ii + genCtx.q * jj * genCtx.ystride + genCtx.q * kk * genCtx.zstride; 263*ce27a4eeSVijay Mahadevan 264*ce27a4eeSVijay Mahadevan std::pair<int,int> entoffset = DMMoab_SetElementConnectivity_Private(genCtx, ix, corner, conn); 265*ce27a4eeSVijay Mahadevan 266*ce27a4eeSVijay Mahadevan for (int j = 0; j < entoffset.second; j++) { 267*ce27a4eeSVijay Mahadevan /* The entity handle for the particular element -> if we want to set some tags is 268*ce27a4eeSVijay Mahadevan moab::EntityHandle eh = starte + ie + j; 269*ce27a4eeSVijay Mahadevan */ 270*ce27a4eeSVijay Mahadevan gids[ie+j] = 1 + ((xe + ii) + (ye + jj) * genCtx.nex + (ze + kk) * (genCtx.nex * genCtx.ney)); 271*ce27a4eeSVijay Mahadevan //gids[ie+j] = ie + j + ((xe + ii) + (ye + jj) * genCtx.nex + (ze + kk) * (genCtx.nex * genCtx.ney)) ; 272*ce27a4eeSVijay Mahadevan //gids[ie+j] = 1 + ie; 273*ce27a4eeSVijay Mahadevan ie++; 274*ce27a4eeSVijay Mahadevan } 275*ce27a4eeSVijay Mahadevan 276*ce27a4eeSVijay Mahadevan ix += entoffset.first; 277*ce27a4eeSVijay Mahadevan //ie += entoffset.second; 278*ce27a4eeSVijay Mahadevan } 279*ce27a4eeSVijay Mahadevan } 280*ce27a4eeSVijay Mahadevan } 281*ce27a4eeSVijay Mahadevan if (genCtx.adjEnts) { /* we need to update adjacencies now, because some elements are new */ 282*ce27a4eeSVijay Mahadevan merr = iface->update_adjacencies(starte, nelems, nvperelem, conn);MBERR("Can't update adjacencies", merr); 283*ce27a4eeSVijay Mahadevan } 284*ce27a4eeSVijay Mahadevan tmp.swap(cells); 285*ce27a4eeSVijay Mahadevan merr = mbImpl->tag_set_data(global_id_tag, cells, &gids[0]);MBERR("Can't set global ids to elements.", merr); 286*ce27a4eeSVijay Mahadevan PetscFunctionReturn(0); 287*ce27a4eeSVijay Mahadevan } 288*ce27a4eeSVijay Mahadevan 289*ce27a4eeSVijay Mahadevan #undef __FUNCT__ 290*ce27a4eeSVijay Mahadevan #define __FUNCT__ "DMMBUtil_InitializeOptions" 291*ce27a4eeSVijay Mahadevan PetscErrorCode DMMBUtil_InitializeOptions(DMMoabMeshGeneratorCtx& genCtx, PetscInt dim, PetscBool simplex, PetscInt rank, PetscInt nprocs, const PetscReal* bounds, PetscInt nele) 292*ce27a4eeSVijay Mahadevan { 293*ce27a4eeSVijay Mahadevan PetscFunctionBegin; 294*ce27a4eeSVijay 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); 295*ce27a4eeSVijay Mahadevan /* Initialize all genCtx data */ 296*ce27a4eeSVijay Mahadevan genCtx.dim=dim; 297*ce27a4eeSVijay Mahadevan genCtx.simplex=simplex; 298*ce27a4eeSVijay Mahadevan genCtx.newMergeMethod=genCtx.keep_skins=genCtx.adjEnts=true; 299*ce27a4eeSVijay Mahadevan /* determine other global quantities for the mesh used for nodes increments */ 300*ce27a4eeSVijay Mahadevan genCtx.q = 1; 301*ce27a4eeSVijay Mahadevan 302*ce27a4eeSVijay Mahadevan if (!genCtx.usrxyzgrid) { /* not overridden by genCtx - assume nele equally and that genCtx wants a uniform cube mesh */ 303*ce27a4eeSVijay Mahadevan //genCtx.blockSizeElementXYZ[0]=genCtx.blockSizeElementXYZ[1]=genCtx.blockSizeElementXYZ[2]=std::pow(nlocalele,1.0/dim); 304*ce27a4eeSVijay Mahadevan genCtx.blockSizeElementXYZ[0]=nele; 305*ce27a4eeSVijay Mahadevan genCtx.blockSizeElementXYZ[1]=(dim>2?nele:nele/nprocs); 306*ce27a4eeSVijay Mahadevan genCtx.blockSizeElementXYZ[2]=(dim>2?nele/nprocs:1); 307*ce27a4eeSVijay Mahadevan } 308*ce27a4eeSVijay Mahadevan 309*ce27a4eeSVijay Mahadevan /* partition only by the largest dimension */ 310*ce27a4eeSVijay Mahadevan /* Total number of local elements := genCtx.blockSizeElementXYZ[0]*(genCtx.dim>1? genCtx.blockSizeElementXYZ[1]*(genCtx.dim>2 ? genCtx.blockSizeElementXYZ[2]:1) :1); */ 311*ce27a4eeSVijay Mahadevan if (bounds) { 312*ce27a4eeSVijay Mahadevan for (int i=0; i<6; i++) 313*ce27a4eeSVijay Mahadevan genCtx.xyzbounds[i]=bounds[i]; 314*ce27a4eeSVijay Mahadevan } 315*ce27a4eeSVijay Mahadevan else { 316*ce27a4eeSVijay Mahadevan genCtx.xyzbounds[0]=genCtx.xyzbounds[2]=genCtx.xyzbounds[4]=0.0; 317*ce27a4eeSVijay Mahadevan genCtx.xyzbounds[1]=genCtx.xyzbounds[3]=genCtx.xyzbounds[5]=1.0; 318*ce27a4eeSVijay Mahadevan } 319*ce27a4eeSVijay Mahadevan 320*ce27a4eeSVijay Mahadevan if (!genCtx.usrprocgrid) { 321*ce27a4eeSVijay Mahadevan genCtx.M=genCtx.N=genCtx.K=std::pow(nprocs,1.0/dim); 322*ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 323*ce27a4eeSVijay Mahadevan case 1: 324*ce27a4eeSVijay Mahadevan genCtx.M=nprocs; 325*ce27a4eeSVijay Mahadevan genCtx.N=genCtx.K=1; 326*ce27a4eeSVijay Mahadevan break; 327*ce27a4eeSVijay Mahadevan case 2: 328*ce27a4eeSVijay Mahadevan genCtx.K=1; 329*ce27a4eeSVijay Mahadevan genCtx.N=nprocs/(genCtx.M); 330*ce27a4eeSVijay Mahadevan break; 331*ce27a4eeSVijay Mahadevan default: 332*ce27a4eeSVijay Mahadevan genCtx.K=nprocs/(genCtx.M*genCtx.N); 333*ce27a4eeSVijay Mahadevan break; 334*ce27a4eeSVijay Mahadevan } 335*ce27a4eeSVijay Mahadevan } 336*ce27a4eeSVijay Mahadevan 337*ce27a4eeSVijay Mahadevan if (!genCtx.usrrefgrid) { 338*ce27a4eeSVijay Mahadevan genCtx.A=genCtx.B=genCtx.C=1; 339*ce27a4eeSVijay Mahadevan } 340*ce27a4eeSVijay Mahadevan 341*ce27a4eeSVijay Mahadevan /* more default values */ 342*ce27a4eeSVijay Mahadevan genCtx.nex=genCtx.ney=genCtx.nez=0; 343*ce27a4eeSVijay Mahadevan genCtx.xstride=genCtx.ystride=genCtx.zstride=0; 344*ce27a4eeSVijay Mahadevan genCtx.NX=genCtx.NY=genCtx.NZ=0; 345*ce27a4eeSVijay Mahadevan genCtx.nex=genCtx.ney=genCtx.nez=0; 346*ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0]=genCtx.blockSizeVertexXYZ[1]=genCtx.blockSizeVertexXYZ[2]=1; 347*ce27a4eeSVijay Mahadevan 348*ce27a4eeSVijay Mahadevan //genCtx.blockSizeElementXYZ[0]=genCtx.blockSizeElementXYZ[1]=genCtx.blockSizeElementXYZ[2]=nele; 349*ce27a4eeSVijay Mahadevan //genCtx.blockSizeVertexXYZ[0]=genCtx.blockSizeVertexXYZ[1]=genCtx.blockSizeVertexXYZ[2]=nele+1; 350*ce27a4eeSVijay Mahadevan 351*ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 352*ce27a4eeSVijay Mahadevan case 3: 353*ce27a4eeSVijay Mahadevan //genCtx.blockSizeElementXYZ[2]=std::max(1,genCtx.blockSizeElementXYZ[2]/nprocs); 354*ce27a4eeSVijay Mahadevan //genCtx.blockSizeElementXYZ[2]=std::max(1,nele/nprocs); 355*ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0] = genCtx.q*genCtx.blockSizeElementXYZ[0] + 1; 356*ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[1] = genCtx.q*genCtx.blockSizeElementXYZ[1] + 1; 357*ce27a4eeSVijay Mahadevan //genCtx.blockSizeElementXYZ[2]=fraction/(genCtx.blockSizeElementXYZ[0]*genCtx.blockSizeElementXYZ[1]); 358*ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[2] = genCtx.q*genCtx.blockSizeElementXYZ[2] + 1; 359*ce27a4eeSVijay Mahadevan 360*ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0]; /* number of elements in x direction, used for global id on element */ 361*ce27a4eeSVijay Mahadevan genCtx.dx = (genCtx.xyzbounds[1]-genCtx.xyzbounds[0])/(genCtx.nex*genCtx.q); /* distance between 2 nodes in x direction */ 362*ce27a4eeSVijay Mahadevan genCtx.NX = (genCtx.q * genCtx.nex + 1); 363*ce27a4eeSVijay Mahadevan genCtx.xstride = 1; 364*ce27a4eeSVijay Mahadevan genCtx.ney = genCtx.N * genCtx.B * genCtx.blockSizeElementXYZ[1]; /* number of elements in y direction .... */ 365*ce27a4eeSVijay Mahadevan genCtx.dy = (genCtx.xyzbounds[3]-genCtx.xyzbounds[2])/(genCtx.ney*genCtx.q); /* distance between 2 nodes in y direction */ 366*ce27a4eeSVijay Mahadevan genCtx.NY = (genCtx.q * genCtx.ney + 1); 367*ce27a4eeSVijay Mahadevan genCtx.ystride = genCtx.blockSizeVertexXYZ[0]; 368*ce27a4eeSVijay Mahadevan genCtx.nez = genCtx.K * genCtx.C * genCtx.blockSizeElementXYZ[2]; /* number of elements in z direction .... */ 369*ce27a4eeSVijay Mahadevan genCtx.dz = (genCtx.xyzbounds[5]-genCtx.xyzbounds[4])/(genCtx.nez*genCtx.q); /* distance between 2 nodes in z direction */ 370*ce27a4eeSVijay Mahadevan genCtx.NZ = (genCtx.q * genCtx.nez + 1); 371*ce27a4eeSVijay Mahadevan genCtx.zstride = genCtx.blockSizeVertexXYZ[0] * genCtx.blockSizeVertexXYZ[1]; 372*ce27a4eeSVijay Mahadevan break; 373*ce27a4eeSVijay Mahadevan case 2: 374*ce27a4eeSVijay Mahadevan //genCtx.blockSizeElementXYZ[1]=genCtx.blockSizeElementXYZ[2]=std::max(1,nele/nprocs); 375*ce27a4eeSVijay Mahadevan //genCtx.blockSizeElementXYZ[1]=fraction/genCtx.blockSizeElementXYZ[0]; 376*ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0] = genCtx.q*genCtx.blockSizeElementXYZ[0] + 1; 377*ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[1] = genCtx.q*genCtx.blockSizeElementXYZ[1] + 1; 378*ce27a4eeSVijay Mahadevan genCtx.blockSizeElementXYZ[2]=0; 379*ce27a4eeSVijay Mahadevan 380*ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0]; /* number of elements in x direction, used for global id on element */ 381*ce27a4eeSVijay Mahadevan genCtx.dx = (genCtx.xyzbounds[1]-genCtx.xyzbounds[0])/(genCtx.nex*genCtx.q); /* distance between 2 nodes in x direction */ 382*ce27a4eeSVijay Mahadevan genCtx.NX = (genCtx.q * genCtx.nex + 1); 383*ce27a4eeSVijay Mahadevan genCtx.xstride = 1; 384*ce27a4eeSVijay Mahadevan genCtx.ney = genCtx.N * genCtx.B * genCtx.blockSizeElementXYZ[1]; /* number of elements in y direction .... */ 385*ce27a4eeSVijay Mahadevan genCtx.dy = (genCtx.xyzbounds[3]-genCtx.xyzbounds[2])/(genCtx.ney*genCtx.q); /* distance between 2 nodes in y direction */ 386*ce27a4eeSVijay Mahadevan genCtx.NY = (genCtx.q * genCtx.ney + 1); 387*ce27a4eeSVijay Mahadevan genCtx.ystride = genCtx.blockSizeVertexXYZ[0]; 388*ce27a4eeSVijay Mahadevan break; 389*ce27a4eeSVijay Mahadevan case 1: 390*ce27a4eeSVijay Mahadevan genCtx.blockSizeElementXYZ[1]=genCtx.blockSizeElementXYZ[2]=0; 391*ce27a4eeSVijay Mahadevan genCtx.blockSizeElementXYZ[0]=nele; 392*ce27a4eeSVijay Mahadevan genCtx.blockSizeVertexXYZ[0] = genCtx.q*genCtx.blockSizeElementXYZ[0] + 1; 393*ce27a4eeSVijay Mahadevan 394*ce27a4eeSVijay Mahadevan genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0]; /* number of elements in x direction, used for global id on element */ 395*ce27a4eeSVijay Mahadevan genCtx.dx = (genCtx.xyzbounds[1]-genCtx.xyzbounds[0])/(genCtx.nex*genCtx.q); /* distance between 2 nodes in x direction */ 396*ce27a4eeSVijay Mahadevan genCtx.NX = (genCtx.q * genCtx.nex + 1); 397*ce27a4eeSVijay Mahadevan genCtx.xstride = 1; 398*ce27a4eeSVijay Mahadevan break; 399*ce27a4eeSVijay Mahadevan } 400*ce27a4eeSVijay Mahadevan 401*ce27a4eeSVijay Mahadevan /* 402*ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local elements:= %d, %d, %d\n",genCtx.blockSizeElementXYZ[0],genCtx.blockSizeElementXYZ[1],genCtx.blockSizeElementXYZ[2]); 403*ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local vertices:= %d, %d, %d\n",genCtx.blockSizeVertexXYZ[0],genCtx.blockSizeVertexXYZ[1],genCtx.blockSizeVertexXYZ[2]); 404*ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local nexyz:= %d, %d, %d\n",genCtx.nex,genCtx.ney,genCtx.nez); 405*ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local delxyz:= %g, %g, %g\n",genCtx.dx,genCtx.dy,genCtx.dz); 406*ce27a4eeSVijay Mahadevan PetscInfo3(NULL, "Local strides:= %d, %d, %d\n",genCtx.xstride,genCtx.ystride,genCtx.zstride); 407*ce27a4eeSVijay Mahadevan */ 408*ce27a4eeSVijay Mahadevan PetscFunctionReturn(0); 409*ce27a4eeSVijay Mahadevan } 410*ce27a4eeSVijay Mahadevan 411aa859218SVijay Mahadevan /*@ 412*ce27a4eeSVijay Mahadevan DMMoabCreateBoxMesh - Creates a mesh on the tensor product (box) of intervals with genCtx specified bounds. 413aa859218SVijay Mahadevan 414aa859218SVijay Mahadevan Collective on MPI_Comm 415aa859218SVijay Mahadevan 416aa859218SVijay Mahadevan Input Parameters: 417aa859218SVijay Mahadevan + comm - The communicator for the DM object 418aa859218SVijay Mahadevan . dim - The spatial dimension 419b8ecf6d3SVijay 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 420b8ecf6d3SVijay Mahadevan . nele - The number of discrete elements in each direction 421b8ecf6d3SVijay Mahadevan . user_nghost - The number of ghosted layers needed in the partitioned mesh 422aa859218SVijay Mahadevan 423aa859218SVijay Mahadevan Output Parameter: 424aa859218SVijay Mahadevan . dm - The DM object 425aa859218SVijay Mahadevan 426aa859218SVijay Mahadevan Level: beginner 427aa859218SVijay Mahadevan 428aa859218SVijay Mahadevan .keywords: DM, create 429aa859218SVijay Mahadevan .seealso: DMSetType(), DMCreate(), DMMoabLoadFromFile() 430aa859218SVijay Mahadevan @*/ 431*ce27a4eeSVijay Mahadevan PetscErrorCode DMMoabCreateBoxMesh(MPI_Comm comm, PetscInt dim, PetscBool useSimplex, const PetscReal* bounds, PetscInt nele, PetscInt nghost, DM *dm) 43251d15aeeSVijay Mahadevan { 43351d15aeeSVijay Mahadevan PetscErrorCode ierr; 43451d15aeeSVijay Mahadevan moab::ErrorCode merr; 435*ce27a4eeSVijay Mahadevan PetscInt a,b,c,n,global_size,global_rank=-1; 43651d15aeeSVijay Mahadevan DM_Moab *dmmoab; 437*ce27a4eeSVijay Mahadevan moab::Interface *mbImpl; 43851d15aeeSVijay Mahadevan moab::ParallelComm *pcomm; 439a4717116SVijay Mahadevan moab::ReadUtilIface *readMeshIface; 440*ce27a4eeSVijay Mahadevan moab::Range verts,cells,edges,faces,adj,dim3,dim2; 441*ce27a4eeSVijay Mahadevan DMMoabMeshGeneratorCtx genCtx; 442a4717116SVijay Mahadevan const PetscInt npts=nele+1; /* Number of points in every dimension */ 443*ce27a4eeSVijay Mahadevan 444*ce27a4eeSVijay Mahadevan moab::Tag global_id_tag,part_tag,geom_tag; 445*ce27a4eeSVijay Mahadevan moab::Range ownedvtx,ownedelms,localvtxs,localelms; 446*ce27a4eeSVijay Mahadevan moab::EntityHandle regionset; 447*ce27a4eeSVijay Mahadevan PetscInt ml=0,nl=0,kl=0,leftover; 44851d15aeeSVijay Mahadevan 44951d15aeeSVijay Mahadevan PetscFunctionBegin; 450e5136372SVijay Mahadevan if(dim < 1 || dim > 3) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Invalid dimension argument for mesh: dim=[1,3].\n"); 451e5136372SVijay Mahadevan 452*ce27a4eeSVijay Mahadevan ierr = MPI_Comm_size(comm, &global_size);CHKERRQ(ierr); 453e5136372SVijay Mahadevan /* total number of vertices in all dimensions */ 454e5136372SVijay Mahadevan n=pow(npts,dim); 455e5136372SVijay Mahadevan 456e5136372SVijay Mahadevan /* do some error checking */ 457e5136372SVijay Mahadevan if(n < 2) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Number of points must be >= 2.\n"); 458*ce27a4eeSVijay 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"); 459e5136372SVijay Mahadevan if(nghost < 0) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Number of ghost layers cannot be negative.\n"); 460e5136372SVijay Mahadevan 461a4717116SVijay Mahadevan /* Create the basic DMMoab object and keep the default parameters created by DM impls */ 462c528d872SBarry Smith ierr = DMMoabCreateMoab(comm, NULL, NULL, NULL, NULL, dm);CHKERRQ(ierr); 46351d15aeeSVijay Mahadevan 464a4717116SVijay Mahadevan /* get all the necessary handles from the private DM object */ 46551d15aeeSVijay Mahadevan dmmoab = (DM_Moab*)(*dm)->data; 466*ce27a4eeSVijay Mahadevan mbImpl = dmmoab->mbiface; 46751d15aeeSVijay Mahadevan pcomm = dmmoab->pcomm; 468*ce27a4eeSVijay Mahadevan global_id_tag = dmmoab->ltog_tag; 469*ce27a4eeSVijay Mahadevan global_rank = pcomm->rank(); 470c8d5365dSVijay Mahadevan dmmoab->dim = dim; 47151d15aeeSVijay Mahadevan 472b5410836SVijay Mahadevan /* create a file set to associate all entities in current mesh */ 473*ce27a4eeSVijay Mahadevan merr = mbImpl->create_meshset(moab::MESHSET_SET, dmmoab->fileset);MBERR("Creating file set failed", merr); 474b5410836SVijay Mahadevan 475a4717116SVijay Mahadevan /* No errors yet; proceed with building the mesh */ 476*ce27a4eeSVijay Mahadevan merr = mbImpl->query_interface(readMeshIface);MBERRNM(merr); 47751d15aeeSVijay Mahadevan 478*ce27a4eeSVijay Mahadevan //PetscObjectOptionsBegin(dm); 479*ce27a4eeSVijay Mahadevan PetscOptionsBegin(comm,"","DMMoab Creation Options","DMMOAB"); 480*ce27a4eeSVijay Mahadevan /* Handle DMMoab spatial resolution */ 481*ce27a4eeSVijay Mahadevan PetscOptionsInt("-dmb_grid_x","Number of grid points in x direction","DMMoabSetSizes",genCtx.blockSizeElementXYZ[0],&genCtx.blockSizeElementXYZ[0],&genCtx.usrxyzgrid); 482*ce27a4eeSVijay Mahadevan if (dim > 1) {PetscOptionsInt("-dmb_grid_y","Number of grid points in y direction","DMMoabSetSizes",genCtx.blockSizeElementXYZ[1],&genCtx.blockSizeElementXYZ[1],&genCtx.usrxyzgrid);} 483*ce27a4eeSVijay Mahadevan if (dim > 2) {PetscOptionsInt("-dmb_grid_z","Number of grid points in z direction","DMMoabSetSizes",genCtx.blockSizeElementXYZ[2],&genCtx.blockSizeElementXYZ[2],&genCtx.usrxyzgrid);} 484a4717116SVijay Mahadevan 485*ce27a4eeSVijay Mahadevan /* Handle DMMoab parallel distibution */ 486*ce27a4eeSVijay Mahadevan PetscOptionsInt("-dmb_processors_x","Number of processors in x direction","DMMoabSetNumProcs",genCtx.M,&genCtx.M,&genCtx.usrprocgrid); 487*ce27a4eeSVijay Mahadevan if (dim > 1) {PetscOptionsInt("-dmb_processors_y","Number of processors in y direction","DMMoabSetNumProcs",genCtx.N,&genCtx.N,&genCtx.usrprocgrid);} 488*ce27a4eeSVijay Mahadevan if (dim > 2) {PetscOptionsInt("-dmb_processors_z","Number of processors in z direction","DMMoabSetNumProcs",genCtx.K,&genCtx.K,&genCtx.usrprocgrid);} 48951d15aeeSVijay Mahadevan 490*ce27a4eeSVijay Mahadevan /* Handle DMMoab block refinement */ 491*ce27a4eeSVijay Mahadevan PetscOptionsInt("-dmb_refine_x","Number of refinement blocks in x direction","DMMoabSetRefinement",genCtx.A,&genCtx.A,&genCtx.usrrefgrid); 492*ce27a4eeSVijay Mahadevan if (dim > 1) {PetscOptionsInt("-dmb_refine_y","Number of refinement blocks in y direction","DMMoabSetRefinement",genCtx.B,&genCtx.B,&genCtx.usrrefgrid);} 493*ce27a4eeSVijay Mahadevan if (dim > 2) {PetscOptionsInt("-dmb_refine_z","Number of refinement blocks in z direction","DMMoabSetRefinement",genCtx.C,&genCtx.C,&genCtx.usrrefgrid);} 494*ce27a4eeSVijay Mahadevan PetscOptionsEnd(); 49551d15aeeSVijay Mahadevan 496*ce27a4eeSVijay Mahadevan ierr = DMMBUtil_InitializeOptions(genCtx, dim, useSimplex, global_rank, global_size, bounds, nele);CHKERRQ(ierr); 49751d15aeeSVijay Mahadevan 498*ce27a4eeSVijay Mahadevan /* Lets check for some valid input */ 499*ce27a4eeSVijay Mahadevan if (genCtx.dim < 1 || genCtx.dim > 3) SETERRQ1(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Invalid topological dimension specified: %d.\n",genCtx.dim); 500*ce27a4eeSVijay Mahadevan if (genCtx.M * genCtx.N * genCtx.K != global_size) 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,global_size); 501*ce27a4eeSVijay Mahadevan if (genCtx.xyzbounds) { 50266f88a78SVijay Mahadevan /* validate the bounds data */ 503*ce27a4eeSVijay 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]); 504*ce27a4eeSVijay 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]); 505*ce27a4eeSVijay 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]); 50666f88a78SVijay Mahadevan } 507*ce27a4eeSVijay Mahadevan if (genCtx.adjEnts) genCtx.keep_skins = true; /* do not delete anything - consumes more memory */ 50866f88a78SVijay Mahadevan 509*ce27a4eeSVijay Mahadevan /* determine m, n, k for processor rank */ 510*ce27a4eeSVijay Mahadevan kl = (genCtx.dim>2?global_rank/(genCtx.M*genCtx.N):0); 511*ce27a4eeSVijay Mahadevan leftover = global_rank%(genCtx.M*genCtx.N); 512*ce27a4eeSVijay Mahadevan nl = leftover/genCtx.M; 513*ce27a4eeSVijay Mahadevan ml = leftover%genCtx.M; 514*ce27a4eeSVijay Mahadevan //if (global_rank==global_size-1) cout << "m, n, k for last rank:" << m << " " << n << " " << k << "\n"; 515e5136372SVijay Mahadevan 516*ce27a4eeSVijay Mahadevan /* 517*ce27a4eeSVijay Mahadevan * so there are a total of M * A * blockSizeElement elements in x direction (so M * A * blockSizeElement + 1 verts in x direction) 518*ce27a4eeSVijay Mahadevan * so there are a total of N * B * blockSizeElement elements in y direction (so N * B * blockSizeElement + 1 verts in y direction) 519*ce27a4eeSVijay Mahadevan * so there are a total of K * C * blockSizeElement elements in z direction (so K * C * blockSizeElement + 1 verts in z direction) 520c8d5365dSVijay Mahadevan 521*ce27a4eeSVijay Mahadevan * there are ( M * A blockSizeElement ) * ( N * B * blockSizeElement) * (K * C * blockSizeElement ) hexas 522*ce27a4eeSVijay Mahadevan * there are ( M * A * blockSizeElement + 1) * ( N * B * blockSizeElement + 1 ) * (K * C * blockSizeElement + 1) vertices 523*ce27a4eeSVijay Mahadevan * x is the first dimension that varies 524*ce27a4eeSVijay Mahadevan */ 525e5136372SVijay Mahadevan 526*ce27a4eeSVijay Mahadevan /* generate the block at (a, b, c); it will represent a partition , it will get a partition tag */ 527*ce27a4eeSVijay Mahadevan int dum_id=-1; 528*ce27a4eeSVijay Mahadevan merr = mbImpl->tag_get_handle("GLOBAL_ID", 1, moab::MB_TYPE_INTEGER, global_id_tag);MBERR("Getting Tag handle failed", merr); 52951d15aeeSVijay Mahadevan 530*ce27a4eeSVijay 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); 53151d15aeeSVijay Mahadevan 5323a4aeca1SVijay Mahadevan /* lets create some sets */ 533*ce27a4eeSVijay 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); 5343a4aeca1SVijay Mahadevan 535*ce27a4eeSVijay Mahadevan for (a=0; a<(genCtx.dim>0?genCtx.A:genCtx.A); a++) { 536*ce27a4eeSVijay Mahadevan for (b=0; b<(genCtx.dim>1?genCtx.B:1); b++) { 537*ce27a4eeSVijay Mahadevan for (c=0; c<(genCtx.dim>2?genCtx.C:1); c++) { 538*ce27a4eeSVijay Mahadevan moab::EntityHandle startv,part_set; 539*ce27a4eeSVijay Mahadevan ierr = DMMoab_GenerateVertices_Private(mbImpl, readMeshIface, genCtx, ml, nl, kl, a, b, c, global_id_tag, startv, verts);CHKERRQ(ierr); 5403a4aeca1SVijay Mahadevan 541*ce27a4eeSVijay Mahadevan ierr = DMMoab_GenerateElements_Private(mbImpl, readMeshIface, genCtx, ml, nl, kl, a, b, c, global_id_tag, startv, cells);CHKERRQ(ierr); 5423a4aeca1SVijay Mahadevan 543*ce27a4eeSVijay Mahadevan merr = mbImpl->create_meshset(moab::MESHSET_SET, part_set);MBERR("Can't create mesh set.", merr); 5443a4aeca1SVijay Mahadevan 545*ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(part_set, verts);MBERR("Can't add vertices to set.", merr); 546*ce27a4eeSVijay Mahadevan //merr = mbImpl->add_entities(part_set, cells);MBERR("Can't add entities to set.", merr); 5473a4aeca1SVijay Mahadevan 548*ce27a4eeSVijay Mahadevan merr = mbImpl->create_meshset(moab::MESHSET_SET, regionset);MBERRNM(merr); 549*ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(regionset, cells);MBERR("Can't add entities to set.", merr); 550*ce27a4eeSVijay Mahadevan merr = mbImpl->tag_set_data(geom_tag, ®ionset, 1, &dmmoab->dim);MBERRNM(merr); 551*ce27a4eeSVijay Mahadevan merr = mbImpl->add_parent_child(part_set,regionset);MBERRNM(merr); 552*ce27a4eeSVijay Mahadevan merr = mbImpl->unite_meshset(part_set, regionset);MBERRNM(merr); 553*ce27a4eeSVijay Mahadevan 554*ce27a4eeSVijay Mahadevan /* if needed, add all edges and faces */ 555*ce27a4eeSVijay Mahadevan if (genCtx.adjEnts) 556*ce27a4eeSVijay Mahadevan { 557*ce27a4eeSVijay Mahadevan if (genCtx.dim > 1) { 558*ce27a4eeSVijay Mahadevan merr = mbImpl->get_adjacencies(cells, 1, true, edges, moab::Interface::UNION);MBERR("Can't get edges", merr); 559*ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(part_set, edges);MBERR("Can't add edges to partition set.", merr); 560*ce27a4eeSVijay Mahadevan } 561*ce27a4eeSVijay Mahadevan if (genCtx.dim > 2) { 562*ce27a4eeSVijay Mahadevan merr = mbImpl->get_adjacencies(cells, 2, true, faces, moab::Interface::UNION);MBERR("Can't get faces", merr); 563*ce27a4eeSVijay Mahadevan merr = mbImpl->add_entities(part_set, faces);MBERR("Can't add faces to partition set.", merr); 564*ce27a4eeSVijay Mahadevan } 565*ce27a4eeSVijay Mahadevan edges.clear(); 566*ce27a4eeSVijay Mahadevan faces.clear(); 567*ce27a4eeSVijay Mahadevan } 568*ce27a4eeSVijay Mahadevan 569*ce27a4eeSVijay Mahadevan int part_num=0; 570*ce27a4eeSVijay Mahadevan switch(genCtx.dim) { 571*ce27a4eeSVijay Mahadevan case 3: 572*ce27a4eeSVijay Mahadevan part_num += (c+kl*genCtx.C)*(genCtx.M*genCtx.A * genCtx.N*genCtx.B); 573*ce27a4eeSVijay Mahadevan case 2: 574*ce27a4eeSVijay Mahadevan part_num += (b+nl*genCtx.B)*(genCtx.M*genCtx.A); 575*ce27a4eeSVijay Mahadevan case 1: 576*ce27a4eeSVijay Mahadevan part_num += (a+ml*genCtx.A); 577*ce27a4eeSVijay Mahadevan break; 578*ce27a4eeSVijay Mahadevan } 579*ce27a4eeSVijay Mahadevan 580*ce27a4eeSVijay Mahadevan merr = mbImpl->tag_set_data(part_tag, &part_set, 1, &part_num);MBERR("Can't set part tag on set", merr); 581*ce27a4eeSVijay Mahadevan merr = mbImpl->add_parent_child(dmmoab->fileset, part_set);MBERR("Can't add part set to file set.", merr); 582*ce27a4eeSVijay Mahadevan merr = mbImpl->unite_meshset(dmmoab->fileset, part_set);MBERRNM(merr); 583*ce27a4eeSVijay Mahadevan 584*ce27a4eeSVijay Mahadevan cells.clear(); 585*ce27a4eeSVijay Mahadevan verts.clear(); 586*ce27a4eeSVijay Mahadevan } 587*ce27a4eeSVijay Mahadevan } 588*ce27a4eeSVijay Mahadevan } 589*ce27a4eeSVijay Mahadevan 590*ce27a4eeSVijay Mahadevan PetscInfo2(NULL, "Generated local mesh: %D vertices and %D elements.\n", verts.size(), cells.size()); 591*ce27a4eeSVijay Mahadevan 592*ce27a4eeSVijay Mahadevan /* resolving shared entities between processors */ 593*ce27a4eeSVijay Mahadevan if (global_size>1) { 594*ce27a4eeSVijay Mahadevan 595*ce27a4eeSVijay Mahadevan moab::MergeMesh mm(mbImpl); 596*ce27a4eeSVijay Mahadevan 597*ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, genCtx.dim, cells);MBERR("Can't get all d-dimensional elements.", merr); 598*ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 0, verts);MBERR("Can't get all vertices.", merr); 599*ce27a4eeSVijay Mahadevan 600*ce27a4eeSVijay Mahadevan if (genCtx.A*genCtx.B*genCtx.C!=1) { // merge needed 601*ce27a4eeSVijay Mahadevan if (genCtx.newMergeMethod) { 602*ce27a4eeSVijay Mahadevan merr = mm.merge_using_integer_tag(verts, global_id_tag);MBERR("Can't merge with GLOBAL_ID tag", merr); 603*ce27a4eeSVijay Mahadevan } 604*ce27a4eeSVijay Mahadevan else { 605*ce27a4eeSVijay Mahadevan merr = mm.merge_entities(cells, 0.0001);MBERR("Can't merge with coordinates", merr); 6063a4aeca1SVijay Mahadevan } 6073a4aeca1SVijay Mahadevan } 6083a4aeca1SVijay Mahadevan 609a4717116SVijay Mahadevan /* check the handles */ 610*ce27a4eeSVijay Mahadevan merr = pcomm->check_all_shared_handles();MBERRV(mbImpl,merr); 61151d15aeeSVijay Mahadevan 612a4717116SVijay Mahadevan /* resolve the shared entities by exchanging information to adjacent processors */ 613*ce27a4eeSVijay Mahadevan merr = pcomm->resolve_shared_ents(dmmoab->fileset,cells,dim,dim-1,NULL,&global_id_tag);MBERRV(mbImpl,merr); 614*ce27a4eeSVijay Mahadevan merr = pcomm->exchange_ghost_cells(dim,0,nghost,dim,true,false,&dmmoab->fileset);MBERRV(mbImpl,merr); 615c8d5365dSVijay Mahadevan 616e427d9c9SVijay Mahadevan /* Reassign global IDs on all entities. */ 617e427d9c9SVijay Mahadevan merr = pcomm->assign_global_ids(dmmoab->fileset,dim,1,false,true,false);MBERRNM(merr); 618*ce27a4eeSVijay Mahadevan merr = pcomm->exchange_tags(global_id_tag,verts);MBERRV(mbImpl,merr); 619*ce27a4eeSVijay Mahadevan merr = pcomm->exchange_tags(global_id_tag,cells);MBERRV(mbImpl,merr); 620e427d9c9SVijay Mahadevan 621*ce27a4eeSVijay Mahadevan /* now resolve the shared entities */ 622*ce27a4eeSVijay Mahadevan //merr = dmmoab->pcomm->resolve_shared_ents( dmmoab->fileset, cells, genCtx.dim, 0 );MBERR("Can't resolve shared entities", merr); 623*ce27a4eeSVijay Mahadevan //PetscInfo(NULL, "Resolving shared entities.\n"); 6243f1c6e43SVijay Mahadevan 625*ce27a4eeSVijay Mahadevan if (!genCtx.keep_skins) { // default is to delete the 1- and 2-dimensional entities 626*ce27a4eeSVijay Mahadevan // delete all quads and edges 627*ce27a4eeSVijay Mahadevan moab::Range toDelete; 628*ce27a4eeSVijay Mahadevan if (genCtx.dim > 1) { 629*ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 1, toDelete);MBERR("Can't get edges", merr); 630*ce27a4eeSVijay Mahadevan } 6313f1c6e43SVijay Mahadevan 632*ce27a4eeSVijay Mahadevan if (genCtx.dim > 2) { 633*ce27a4eeSVijay Mahadevan merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 2, toDelete);MBERR("Can't get faces", merr); 634*ce27a4eeSVijay Mahadevan } 635c8d5365dSVijay Mahadevan 636*ce27a4eeSVijay Mahadevan merr = dmmoab->pcomm->delete_entities(toDelete) ;MBERR("Can't delete entities", merr); 637*ce27a4eeSVijay Mahadevan PetscInfo(NULL, "Deleted edges and faces, and correct sharedEnts.\n"); 638*ce27a4eeSVijay Mahadevan } 639*ce27a4eeSVijay Mahadevan 640*ce27a4eeSVijay Mahadevan } 64151d15aeeSVijay Mahadevan PetscFunctionReturn(0); 64251d15aeeSVijay Mahadevan } 64351d15aeeSVijay Mahadevan 64451d15aeeSVijay Mahadevan 6452e4e7c01SVijay Mahadevan PetscErrorCode DMMoab_GetReadOptions_Private(PetscBool by_rank, PetscInt numproc, PetscInt dim, MoabReadMode mode, PetscInt dbglevel, const char* dm_opts, const char* extra_opts, const char** read_opts) 64651d15aeeSVijay Mahadevan { 6476acfe860SVijay Mahadevan char *ropts; 64861eb6e27SVijay Mahadevan char ropts_par[PETSC_MAX_PATH_LEN]; 64961eb6e27SVijay Mahadevan char ropts_dbg[PETSC_MAX_PATH_LEN]; 650f90c3b0eSVijay Mahadevan PetscErrorCode ierr; 65151d15aeeSVijay Mahadevan 652a4717116SVijay Mahadevan PetscFunctionBegin; 65395dccacaSBarry Smith ierr = PetscMalloc1(PETSC_MAX_PATH_LEN,&ropts);CHKERRQ(ierr); 65461eb6e27SVijay Mahadevan ierr = PetscMemzero(&ropts_par,PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 65561eb6e27SVijay Mahadevan ierr = PetscMemzero(&ropts_dbg,PETSC_MAX_PATH_LEN);CHKERRQ(ierr); 65661eb6e27SVijay Mahadevan 657e23c60ebSVijay Mahadevan /* do parallel read unless using only one processor */ 658a4717116SVijay Mahadevan if (numproc > 1) { 6596acfe860SVijay 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); 6602e4e7c01SVijay Mahadevan } 6612e4e7c01SVijay Mahadevan 662c8d5365dSVijay Mahadevan if (dbglevel) { 663f90c3b0eSVijay Mahadevan if (numproc>1) { 6646acfe860SVijay Mahadevan ierr = PetscSNPrintf(ropts_dbg, PETSC_MAX_PATH_LEN, "%sCPUTIME;DEBUG_IO=%d;DEBUG_PIO=%d;",dbglevel,dbglevel);CHKERRQ(ierr); 66561eb6e27SVijay Mahadevan } 66661eb6e27SVijay Mahadevan else { 6676acfe860SVijay Mahadevan ierr = PetscSNPrintf(ropts_dbg, PETSC_MAX_PATH_LEN, "%sCPUTIME;DEBUG_IO=%d;",dbglevel);CHKERRQ(ierr); 668f90c3b0eSVijay Mahadevan } 669c8d5365dSVijay Mahadevan } 67051d15aeeSVijay Mahadevan 6716acfe860SVijay Mahadevan ierr = PetscSNPrintf(ropts, PETSC_MAX_PATH_LEN, "%s%s%s%s",ropts_par,ropts_dbg,(extra_opts?extra_opts:""),(dm_opts?dm_opts:""));CHKERRQ(ierr); 672f90c3b0eSVijay Mahadevan *read_opts = ropts; 67351d15aeeSVijay Mahadevan PetscFunctionReturn(0); 67451d15aeeSVijay Mahadevan } 67551d15aeeSVijay Mahadevan 67651d15aeeSVijay Mahadevan 677aa859218SVijay Mahadevan /*@ 678aa859218SVijay Mahadevan DMMoabLoadFromFile - Creates a DM object by loading the mesh from a user specified file. 679aa859218SVijay Mahadevan 680aa859218SVijay Mahadevan Collective on MPI_Comm 681aa859218SVijay Mahadevan 682aa859218SVijay Mahadevan Input Parameters: 683aa859218SVijay Mahadevan + comm - The communicator for the DM object 684aa859218SVijay Mahadevan . dim - The spatial dimension 685b8ecf6d3SVijay Mahadevan . filename - The name of the mesh file to be loaded 686b8ecf6d3SVijay Mahadevan . usrreadopts - The options string to read a MOAB mesh. 687aa859218SVijay Mahadevan Reference (Parallel Mesh Initialization: http://www.mcs.anl.gov/~fathom/moab-docs/html/contents.html#fivetwo) 688aa859218SVijay Mahadevan 689aa859218SVijay Mahadevan Output Parameter: 690aa859218SVijay Mahadevan . dm - The DM object 691aa859218SVijay Mahadevan 692aa859218SVijay Mahadevan Level: beginner 693aa859218SVijay Mahadevan 694aa859218SVijay Mahadevan .keywords: DM, create 695b8ecf6d3SVijay Mahadevan 696aa859218SVijay Mahadevan .seealso: DMSetType(), DMCreate(), DMMoabCreateBoxMesh() 697aa859218SVijay Mahadevan @*/ 698a4717116SVijay Mahadevan PetscErrorCode DMMoabLoadFromFile(MPI_Comm comm,PetscInt dim,const char* filename, const char* usrreadopts, DM *dm) 69951d15aeeSVijay Mahadevan { 700a4717116SVijay Mahadevan moab::ErrorCode merr; 7012e4e7c01SVijay Mahadevan PetscInt nprocs; 702a4717116SVijay Mahadevan DM_Moab *dmmoab; 703a4717116SVijay Mahadevan moab::Interface *mbiface; 704a4717116SVijay Mahadevan moab::ParallelComm *pcomm; 705a4717116SVijay Mahadevan moab::Range verts,elems; 7067023aa44SVijay Mahadevan const char *readopts; 707a4717116SVijay Mahadevan PetscErrorCode ierr; 70851d15aeeSVijay Mahadevan 70951d15aeeSVijay Mahadevan PetscFunctionBegin; 710a4717116SVijay Mahadevan PetscValidPointer(dm,5); 71151d15aeeSVijay Mahadevan 712a4717116SVijay Mahadevan /* Create the basic DMMoab object and keep the default parameters created by DM impls */ 713c528d872SBarry Smith ierr = DMMoabCreateMoab(comm, NULL, NULL, NULL, NULL, dm);CHKERRQ(ierr); 71451d15aeeSVijay Mahadevan 715a4717116SVijay Mahadevan /* get all the necessary handles from the private DM object */ 716a4717116SVijay Mahadevan dmmoab = (DM_Moab*)(*dm)->data; 717a4717116SVijay Mahadevan mbiface = dmmoab->mbiface; 718a4717116SVijay Mahadevan pcomm = dmmoab->pcomm; 719a4717116SVijay Mahadevan nprocs = pcomm->size(); 720aa859218SVijay Mahadevan /* TODO: Decipher dimension based on the loaded mesh instead of getting from user */ 721c8d5365dSVijay Mahadevan dmmoab->dim = dim; 722a4717116SVijay Mahadevan 723b5410836SVijay Mahadevan /* create a file set to associate all entities in current mesh */ 724b5410836SVijay Mahadevan merr = dmmoab->mbiface->create_meshset(moab::MESHSET_SET, dmmoab->fileset);MBERR("Creating file set failed", merr); 725b5410836SVijay Mahadevan 726a4717116SVijay Mahadevan /* add mesh loading options specific to the DM */ 7272e4e7c01SVijay Mahadevan ierr = DMMoab_GetReadOptions_Private(dmmoab->partition_by_rank, nprocs, dim, dmmoab->read_mode, 7282e4e7c01SVijay Mahadevan dmmoab->rw_dbglevel, dmmoab->extra_read_options, usrreadopts, &readopts);CHKERRQ(ierr); 7297023aa44SVijay Mahadevan 730e5136372SVijay Mahadevan PetscInfo2(*dm, "Reading file %s with options: %s\n",filename,readopts); 731a4717116SVijay Mahadevan 732a4717116SVijay Mahadevan /* Load the mesh from a file. */ 7337023aa44SVijay Mahadevan merr = mbiface->load_file(filename, &dmmoab->fileset, readopts);MBERRVM(mbiface,"Reading MOAB file failed.", merr); 734a4717116SVijay Mahadevan 7356e40195eSVijay Mahadevan /* Reassign global IDs on all entities. */ 736e5136372SVijay Mahadevan merr = pcomm->assign_global_ids(dmmoab->fileset,dim,1,true,true,true);MBERRNM(merr); 737e5136372SVijay Mahadevan 738e5136372SVijay Mahadevan /* load the local vertices */ 739e5136372SVijay Mahadevan merr = mbiface->get_entities_by_type(dmmoab->fileset, moab::MBVERTEX, verts, true);MBERRNM(merr); 740e5136372SVijay Mahadevan /* load the local elements */ 741e5136372SVijay Mahadevan merr = mbiface->get_entities_by_dimension(dmmoab->fileset, dim, elems, true);MBERRNM(merr); 742e5136372SVijay Mahadevan 743e5136372SVijay Mahadevan /* Everything is set up, now just do a tag exchange to update tags 744e5136372SVijay Mahadevan on all of the ghost vertexes */ 745e5136372SVijay Mahadevan merr = pcomm->exchange_tags(dmmoab->ltog_tag,verts);MBERRV(mbiface,merr); 746e5136372SVijay Mahadevan merr = pcomm->exchange_tags(dmmoab->ltog_tag,elems);MBERRV(mbiface,merr); 747e5136372SVijay Mahadevan 748e5136372SVijay Mahadevan merr = pcomm->exchange_ghost_cells(dim,0,1,0,true,true,&dmmoab->fileset);MBERRV(mbiface,merr); 7496e40195eSVijay Mahadevan 750a4717116SVijay Mahadevan merr = pcomm->collective_sync_partition();MBERR("Collective sync failed", merr); 751a4717116SVijay Mahadevan 752e5136372SVijay Mahadevan PetscInfo3(*dm, "MOAB file '%s' was successfully loaded. Found %D vertices and %D elements.\n", filename, verts.size(), elems.size()); 7536acfe860SVijay Mahadevan ierr = PetscFree(readopts);CHKERRQ(ierr); 75451d15aeeSVijay Mahadevan PetscFunctionReturn(0); 75551d15aeeSVijay Mahadevan } 75651d15aeeSVijay Mahadevan 757