1 #include <petsc-private/dmmbimpl.h> /*I "petscdm.h" I*/ 2 #include <petsc-private/vecimpl.h> /*I "petscdm.h" I*/ 3 4 #include <petscdmmoab.h> 5 #include <MBTagConventions.hpp> 6 #include <moab/ReadUtilIface.hpp> 7 #include <moab/ScdInterface.hpp> 8 #include <moab/CN.hpp> 9 10 11 #undef __FUNCT__ 12 #define __FUNCT__ "DMMoabComputeDomainBounds_Private" 13 PetscErrorCode DMMoabComputeDomainBounds_Private(moab::ParallelComm* pcomm, PetscInt dim, PetscInt neleglob, PetscInt *ise) 14 { 15 PetscInt size,rank; 16 PetscInt fraction,remainder; 17 PetscInt starts[3],sizes[3]; 18 19 PetscFunctionBegin; 20 if(dim<1 && dim>3) SETERRQ1(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"The problem dimension is invalid: %D",dim); 21 22 size=pcomm->size(); 23 rank=pcomm->rank(); 24 fraction=neleglob/size; /* partition only by the largest dimension */ 25 remainder=neleglob%size; /* remainder after partition which gets evenly distributed by round-robin */ 26 27 if(fraction==0) SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"The leading dimension size should be greater than number of processors: %D < %D",neleglob,size); 28 29 starts[0]=starts[1]=starts[2]=0; /* default dimensional element = 1 */ 30 sizes[0]=sizes[1]=sizes[2]=neleglob; /* default dimensional element = 1 */ 31 32 if(rank < remainder) { 33 /* This process gets "fraction+1" elements */ 34 sizes[dim-1] = (fraction + 1); 35 starts[dim-1] = rank * (fraction+1); 36 } else { 37 /* This process gets "fraction" elements */ 38 sizes[dim-1] = fraction; 39 starts[dim-1] = (remainder*(fraction+1) + fraction*(rank-remainder)); 40 } 41 42 for(int i=dim-1; i>=0; --i) { 43 ise[2*i]=starts[i];ise[2*i+1]=starts[i]+sizes[i]; 44 } 45 46 PetscFunctionReturn(0); 47 } 48 49 static void set_structured_coordinates(PetscInt i, PetscInt j, PetscInt k, PetscReal hxyz, PetscInt vcount, std::vector<double*>& vcoords) 50 { 51 vcoords[0][vcount] = i*hxyz; 52 vcoords[1][vcount] = j*hxyz; 53 vcoords[2][vcount] = k*hxyz; 54 } 55 56 static void set_element_connectivity(PetscInt dim, moab::EntityType etype, PetscInt offset, PetscInt nele, PetscInt i, PetscInt j, PetscInt k, PetscInt vfirst, moab::EntityHandle *connectivity) 57 { 58 std::vector<int> subent_conn(pow(2,dim)); /* only linear edge, quad, hex supported now */ 59 60 moab::CN::SubEntityVertexIndices(etype, dim, 0, subent_conn.data()); 61 62 switch(dim) { 63 case 1: 64 connectivity[offset+subent_conn[0]] = vfirst+i; 65 connectivity[offset+subent_conn[1]] = vfirst+(i+1); 66 break; 67 case 2: 68 connectivity[offset+subent_conn[0]] = vfirst+i+j*(nele+1); 69 connectivity[offset+subent_conn[1]] = vfirst+(i+1)+j*(nele+1); 70 connectivity[offset+subent_conn[2]] = vfirst+(i+1)+(j+1)*(nele+1); 71 connectivity[offset+subent_conn[3]] = vfirst+i+(j+1)*(nele+1); 72 break; 73 case 3: 74 default: 75 connectivity[offset+subent_conn[0]] = vfirst+i+(nele+1)*(j+(nele+1)*k); 76 connectivity[offset+subent_conn[1]] = vfirst+(i+1)+(nele+1)*(j+(nele+1)*k); 77 connectivity[offset+subent_conn[2]] = vfirst+(i+1)+(nele+1)*((j+1)+(nele+1)*k); 78 connectivity[offset+subent_conn[3]] = vfirst+i+(nele+1)*((j+1)+(nele+1)*k); 79 connectivity[offset+subent_conn[4]] = vfirst+i+(nele+1)*(j+(nele+1)*(k+1)); 80 connectivity[offset+subent_conn[5]] = vfirst+(i+1)+(nele+1)*(j+(nele+1)*(k+1)); 81 connectivity[offset+subent_conn[6]] = vfirst+(i+1)+(nele+1)*((j+1)+(nele+1)*(k+1)); 82 connectivity[offset+subent_conn[7]] = vfirst+i+(nele+1)*((j+1)+(nele+1)*(k+1)); 83 break; 84 } 85 } 86 87 #undef __FUNCT__ 88 #define __FUNCT__ "DMMoabCreateBoxMesh" 89 PetscErrorCode DMMoabCreateBoxMesh(MPI_Comm comm, PetscInt dim, PetscInt nele, PetscInt user_nghost, DM *dm) 90 { 91 PetscErrorCode ierr; 92 moab::ErrorCode merr; 93 PetscInt i,j,k,n,nprocs; 94 DM_Moab *dmmoab; 95 moab::Interface *mbiface; 96 moab::ParallelComm *pcomm; 97 moab::ReadUtilIface* readMeshIface; 98 99 moab::Tag id_tag=PETSC_NULL; 100 moab::Range ownedvtx,ownedelms; 101 moab::EntityHandle vfirst,efirst,regionset,faceset,edgeset,vtxset; 102 std::vector<double*> vcoords; 103 moab::EntityHandle *connectivity = 0; 104 moab::EntityType etype; 105 PetscInt ise[6]; 106 PetscReal xse[6]; 107 /* TODO: Fix nghost > 0 - now relying on exchange_ghost_cells */ 108 const PetscInt nghost=0; 109 110 moab::Tag geom_tag; 111 112 moab::Range adj,dim3,dim2; 113 bool build_adjacencies=false; 114 115 const PetscInt npts=nele+1; /* Number of points in every dimension */ 116 PetscInt vpere,locnele,locnpts,ghnele,ghnpts; /* Number of verts/element, vertices, elements owned by this process */ 117 118 PetscFunctionBegin; 119 if(dim < 1 || dim > 3) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Invalid dimension argument for mesh: dim=[1,3].\n"); 120 121 ierr = MPI_Comm_size(comm, &nprocs);CHKERRQ(ierr); 122 /* total number of vertices in all dimensions */ 123 n=pow(npts,dim); 124 125 /* do some error checking */ 126 if(n < 2) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Number of points must be >= 2.\n"); 127 if(nprocs > n) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Number of processors must be less than or equal to number of elements.\n"); 128 if(nghost < 0) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Number of ghost layers cannot be negative.\n"); 129 130 /* Create the basic DMMoab object and keep the default parameters created by DM impls */ 131 ierr = DMMoabCreateMoab(comm, PETSC_NULL, PETSC_NULL, PETSC_NULL, PETSC_NULL, dm);CHKERRQ(ierr); 132 133 /* get all the necessary handles from the private DM object */ 134 dmmoab = (DM_Moab*)(*dm)->data; 135 mbiface = dmmoab->mbiface; 136 pcomm = dmmoab->pcomm; 137 id_tag = dmmoab->ltog_tag; 138 nprocs = pcomm->size(); 139 dmmoab->dim = dim; 140 141 /* No errors yet; proceed with building the mesh */ 142 merr = mbiface->query_interface(readMeshIface);MBERRNM(merr); 143 144 ierr = PetscMemzero(ise,sizeof(PetscInt)*6);CHKERRQ(ierr); 145 146 /* call the collective routine that computes the domain bounds for a structured mesh using MOAB */ 147 ierr = DMMoabComputeDomainBounds_Private(pcomm, dim, nele, ise);CHKERRQ(ierr); 148 149 /* set some variables based on dimension */ 150 switch(dim) { 151 case 1: 152 vpere = 2; 153 locnele = (ise[1]-ise[0]); 154 locnpts = (ise[1]-ise[0]+1); 155 ghnele = (nghost > 0 ? (ise[0] > nghost ? 1 : 0) + (ise[1] < nele - nghost ? 1 : 0) : 0 ); 156 ghnpts = (nghost > 0 ? (ise[0] > 0 ? 1 : 0) + (ise[1] < nele ? 1 : 0) : 0); 157 etype = moab::MBEDGE; 158 break; 159 case 2: 160 vpere = 4; 161 locnele = (ise[1]-ise[0])*(ise[3]-ise[2]); 162 locnpts = (ise[1]-ise[0]+1)*(ise[3]-ise[2]+1); 163 ghnele = (nghost > 0 ? (ise[2] > 0 ? nele : 0) + (ise[3] < nele ? nele : 0) : 0); 164 ghnpts = (nghost > 0 ? (ise[2] > 0 ? npts : 0) + (ise[3] < nele ? npts : 0) : 0); 165 etype = moab::MBQUAD; 166 break; 167 case 3: 168 vpere = 8; 169 locnele = (ise[1]-ise[0])*(ise[3]-ise[2])*(ise[5]-ise[4]); 170 locnpts = (ise[1]-ise[0]+1)*(ise[3]-ise[2]+1)*(ise[5]-ise[4]+1); 171 ghnele = (nghost > 0 ? (ise[4] > 0 ? nele*nele : 0) + (ise[5] < nele ? nele*nele : 0) : 0); 172 ghnpts = (nghost > 0 ? (ise[4] > 0 ? npts*npts : 0) + (ise[5] < nele ? npts*npts : 0) : 0); 173 etype = moab::MBHEX; 174 break; 175 } 176 177 /* we have a domain of size [1,1,1] - now compute local co-ordinate box */ 178 ierr = PetscMemzero(xse,sizeof(PetscReal)*6);CHKERRQ(ierr); 179 for(i=0; i<6; ++i) { 180 xse[i]=(PetscReal)ise[i]/nele; 181 } 182 183 /* Create vertexes and set the coodinate of each vertex */ 184 merr = readMeshIface->get_node_coords(3,locnpts+ghnpts,0,vfirst,vcoords,n);MBERRNM(merr); 185 186 /* Compute the co-ordinates of vertices and global IDs */ 187 std::vector<int> vgid(locnpts+ghnpts); 188 int vcount=0; 189 double hxyz=1.0/nele; 190 191 /* create all the owned vertices */ 192 for (k = ise[4]; k <= ise[5]; k++) { 193 for (j = ise[2]; j <= ise[3]; j++) { 194 for (i = ise[0]; i <= ise[1]; i++, vcount++) { 195 set_structured_coordinates(i,j,k,hxyz,vcount,vcoords); 196 vgid[vcount] = (k*npts+j)*npts+i+1; 197 } 198 } 199 } 200 201 /* create ghosted vertices requested by user - below the current plane */ 202 if (ise[2*dim-2] > 0) { 203 for (k = (dim==3?ise[4]-nghost:ise[4]); k <= (dim==3?ise[4]-1:ise[5]); k++) { 204 for (j = (dim==2?ise[2]-nghost:ise[2]); j <= (dim==2?ise[2]-1:ise[3]); j++) { 205 for (i = (dim>1?ise[0]:ise[0]-nghost); i <= (dim>1?ise[1]:ise[0]-1); i++, vcount++) { 206 set_structured_coordinates(i,j,k,hxyz,vcount,vcoords); 207 vgid[vcount] = (k*npts+j)*npts+i+1; 208 } 209 } 210 } 211 } 212 213 /* create ghosted vertices requested by user - above the current plane */ 214 if (ise[2*dim-1] < nele) { 215 for (k = (dim==3?ise[5]+1:ise[4]); k <= (dim==3?ise[5]+nghost:ise[5]); k++) { 216 for (j = (dim==2?ise[3]+1:ise[2]); j <= (dim==2?ise[3]+nghost:ise[3]); j++) { 217 for (i = (dim>1?ise[0]:ise[1]+1); i <= (dim>1?ise[1]:ise[1]+nghost); i++, vcount++) { 218 set_structured_coordinates(i,j,k,hxyz,vcount,vcoords); 219 vgid[vcount] = (k*npts+j)*npts+i+1; 220 } 221 } 222 } 223 } 224 225 if (locnpts+ghnpts != vcount) SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Created the wrong number of vertices! (%D!=%D)",locnpts+ghnpts,vcount); 226 227 merr = mbiface->get_entities_by_type(0,moab::MBVERTEX,ownedvtx,true);MBERRNM(merr); 228 229 /* The global ID tag is applied to each owned 230 vertex. It acts as an global identifier which MOAB uses to 231 assemble the individual pieces of the mesh: 232 Set the global ID indices */ 233 merr = mbiface->tag_set_data(id_tag,ownedvtx,vgid.data());MBERRNM(merr); 234 235 /* Create elements between mesh points using the ReadUtilInterface 236 get the reference to element connectivities for all local elements from the ReadUtilInterface */ 237 merr = readMeshIface->get_element_connect (locnele+ghnele,vpere,etype,1,efirst,connectivity);MBERRNM(merr); 238 239 /* offset appropriately so that only local ID and not global ID numbers are set for connectivity array */ 240 // PetscPrintf(PETSC_COMM_SELF, "[%D] first local handle %D\n", rank, vfirst); 241 vfirst-=vgid[0]-1; 242 243 /* 3. Loop over elements in 3 nested loops over i, j, k; for each (i,j,k): 244 and then set the connectivity for each element appropriately */ 245 int ecount=0; 246 247 /* create ghosted elements requested by user - below the current plane */ 248 if (ise[2*dim-2] >= nghost) { 249 for (k = (dim==3?ise[4]-nghost:ise[4]); k < (dim==3?ise[4]:std::max(ise[5],1)); k++) { 250 for (j = (dim==2?ise[2]-nghost:ise[2]); j < (dim==2?ise[2]:std::max(ise[3],1)); j++) { 251 for (i = (dim>1?ise[0]:ise[0]-nghost); i < (dim>1?std::max(ise[1],1):ise[0]); i++, ecount++) { 252 set_element_connectivity(dim, etype, ecount*vpere, nele, i, j, k, vfirst, connectivity); 253 } 254 } 255 } 256 } 257 258 /* create owned elements requested by user */ 259 for (k = ise[4]; k < std::max(ise[5],1); k++) { 260 for (j = ise[2]; j < std::max(ise[3],1); j++) { 261 for (i = ise[0]; i < std::max(ise[1],1); i++,ecount++) { 262 set_element_connectivity(dim, etype, ecount*vpere, nele, i, j, k, vfirst, connectivity); 263 } 264 } 265 } 266 267 /* create ghosted elements requested by user - above the current plane */ 268 if (ise[2*dim-1] <= nele-nghost) { 269 for (k = (dim==3?ise[5]:ise[4]); k < (dim==3?ise[5]+nghost:std::max(ise[5],1)); k++) { 270 for (j = (dim==2?ise[3]:ise[2]); j < (dim==2?ise[3]+nghost:std::max(ise[3],1)); j++) { 271 for (i = (dim>1?ise[0]:ise[1]); i < (dim>1?std::max(ise[1],1):ise[1]+nghost); i++, ecount++) { 272 set_element_connectivity(dim, etype, ecount*vpere, nele, i, j, k, vfirst, connectivity); 273 } 274 } 275 } 276 } 277 278 merr = readMeshIface->update_adjacencies(efirst,locnele+ghnele,vpere,connectivity);MBERRNM(merr); 279 280 /* 2. Get the vertices and hexes from moab and check their numbers against I*J*K and (I-1)*(J-1)*(K-1), resp. 281 first '0' specifies "root set", or entire MOAB instance, second the entity dimension being requested */ 282 merr = mbiface->get_entities_by_dimension(0, dim, ownedelms);MBERRNM(merr); 283 284 if (locnele+ghnele != (int) ownedelms.size()) 285 SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Created the wrong number of elements! (%D!=%D)",locnele+ghnele,ownedelms.size()); 286 else if(locnpts+ghnpts != (int) ownedvtx.size()) 287 SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Created the wrong number of vertices! (%D!=%D)",locnpts+ghnpts,ownedvtx.size()); 288 else 289 PetscInfo2(*dm, "Created %D elements and %D vertices.\n", ownedelms.size(), ownedvtx.size()); 290 291 /* lets create some sets */ 292 merr = mbiface->tag_get_handle(GEOM_DIMENSION_TAG_NAME, 1, moab::MB_TYPE_INTEGER, geom_tag, moab::MB_TAG_SPARSE|moab::MB_TAG_CREAT);MBERRNM(merr); 293 294 merr = mbiface->create_meshset(moab::MESHSET_SET, regionset);MBERRNM(merr); 295 merr = mbiface->add_entities(regionset, ownedelms);MBERRNM(merr); 296 merr = mbiface->add_parent_child(dmmoab->fileset,regionset);MBERRNM(merr); 297 merr = mbiface->tag_set_data(geom_tag, ®ionset, 1, &dmmoab->dim);MBERRNM(merr); 298 merr = mbiface->unite_meshset(dmmoab->fileset, regionset);MBERRNM(merr); 299 300 merr = mbiface->create_meshset(moab::MESHSET_SET, vtxset);MBERRNM(merr); 301 merr = mbiface->add_entities(vtxset, ownedvtx);MBERRNM(merr); 302 merr = mbiface->add_parent_child(dmmoab->fileset,vtxset);MBERRNM(merr); 303 merr = mbiface->unite_meshset(dmmoab->fileset, vtxset);MBERRNM(merr); 304 305 if (build_adjacencies) { 306 // generate all lower dimensional adjacencies 307 merr = mbiface->get_adjacencies( ownedelms, dim-1, true, adj, moab::Interface::UNION );MBERRNM(merr); 308 merr = dmmoab->pcomm->get_part_entities(dim2, dim-1);MBERRNM(merr); 309 adj.merge(dim2); 310 311 /* create face sets */ 312 merr = mbiface->create_meshset(moab::MESHSET_SET, faceset);MBERRNM(merr); 313 merr = mbiface->add_entities(faceset, adj);MBERRNM(merr); 314 merr = mbiface->add_parent_child(dmmoab->fileset,faceset);MBERRNM(merr); 315 i=dim-1; 316 merr = mbiface->tag_set_data(geom_tag, &faceset, 1, &i);MBERRNM(merr); 317 merr = mbiface->unite_meshset(dmmoab->fileset, faceset);MBERRNM(merr); 318 PetscInfo2(dm, "Found %d %d-Dim quantities.\n", adj.size(), dim-1); 319 320 if (dim > 2) { 321 dim2.clear(); 322 /* create edge sets, if appropriate i.e., if dim=3 */ 323 merr = mbiface->create_meshset(moab::MESHSET_SET, edgeset);MBERRNM(merr); 324 merr = mbiface->get_adjacencies(adj, dim-1, true, dim2, moab::Interface::UNION );MBERRNM(merr); 325 merr = mbiface->add_entities(edgeset, dim2);MBERRNM(merr); 326 merr = mbiface->add_parent_child(dmmoab->fileset,edgeset);MBERRNM(merr); 327 i=dim-2; 328 merr = mbiface->tag_set_data(geom_tag, &edgeset, 1, &i);MBERRNM(merr); 329 merr = mbiface->unite_meshset(dmmoab->fileset, edgeset);MBERRNM(merr); 330 PetscInfo2(dm, "Found %d %d-Dim quantities.\n", adj.size(), dim-2); 331 } 332 } 333 334 /* check the handles */ 335 merr = pcomm->check_all_shared_handles();MBERRV(mbiface,merr); 336 337 #if 0 338 std::stringstream sstr; 339 sstr << "test_" << pcomm->rank() << ".vtk"; 340 mbiface->write_mesh(sstr.str().c_str()); 341 #endif 342 343 /* resolve the shared entities by exchanging information to adjacent processors */ 344 merr = mbiface->get_entities_by_type(dmmoab->fileset,etype,ownedelms,true);MBERRNM(merr); 345 merr = pcomm->resolve_shared_ents(dmmoab->fileset,ownedelms,dim,dim-1,&id_tag);MBERRV(mbiface,merr); 346 347 /* Reassign global IDs on all entities. */ 348 merr = pcomm->assign_global_ids(dmmoab->fileset,dim,1,true,true,true);MBERRNM(merr); 349 350 merr = pcomm->exchange_ghost_cells(dim,0,user_nghost,dim,true,false,&dmmoab->fileset);MBERRV(mbiface,merr); 351 352 /* Everything is set up, now just do a tag exchange to update tags 353 on all of the ghost vertexes */ 354 merr = mbiface->get_entities_by_type(dmmoab->fileset,moab::MBVERTEX,ownedvtx,true);MBERRNM(merr); 355 merr = mbiface->get_entities_by_dimension(dmmoab->fileset, dim, ownedelms);MBERRNM(merr); 356 357 merr = pcomm->exchange_tags(id_tag,ownedvtx);MBERRV(mbiface,merr); 358 merr = pcomm->exchange_tags(id_tag,ownedelms);MBERRV(mbiface,merr); 359 PetscFunctionReturn(0); 360 } 361 362 363 #undef __FUNCT__ 364 #define __FUNCT__ "DMMoab_GetReadOptions_Private" 365 PetscErrorCode DMMoab_GetReadOptions_Private(PetscBool by_rank, PetscInt numproc, PetscInt dim, MoabReadMode mode, PetscInt dbglevel, const char* extra_opts, const char** read_opts) 366 { 367 std::ostringstream str; 368 369 PetscFunctionBegin; 370 /* do parallel read unless using only one processor */ 371 if (numproc > 1) { 372 str << "PARALLEL=" << mode << ";PARTITION=PARALLEL_PARTITION;PARTITION_DISTRIBUTE;"; 373 str << "PARALLEL_RESOLVE_SHARED_ENTS;PARALLEL_GHOSTS=" << dim << ".0.1;"; 374 if (by_rank) 375 str << "PARTITION_BY_RANK;"; 376 } 377 378 if (dbglevel) { 379 str << "CPUTIME;DEBUG_IO=" << dbglevel << ";"; 380 if (numproc>1) 381 str << "DEBUG_PIO=" << dbglevel << ";"; 382 } 383 384 if (extra_opts) 385 str << extra_opts; 386 387 *read_opts = str.str().c_str(); 388 PetscFunctionReturn(0); 389 } 390 391 392 #undef __FUNCT__ 393 #define __FUNCT__ "DMMoabLoadFromFile" 394 PetscErrorCode DMMoabLoadFromFile(MPI_Comm comm,PetscInt dim,const char* filename, const char* usrreadopts, DM *dm) 395 { 396 moab::ErrorCode merr; 397 PetscInt nprocs,dbglevel=0; 398 PetscBool part_by_rank=PETSC_FALSE; 399 DM_Moab *dmmoab; 400 moab::Interface *mbiface; 401 moab::ParallelComm *pcomm; 402 moab::Range verts,elems; 403 const char *readopts; 404 PetscErrorCode ierr; 405 406 PetscFunctionBegin; 407 PetscValidPointer(dm,5); 408 409 /* Create the basic DMMoab object and keep the default parameters created by DM impls */ 410 ierr = DMMoabCreateMoab(comm, PETSC_NULL, PETSC_NULL, PETSC_NULL, PETSC_NULL, dm);CHKERRQ(ierr); 411 412 /* get all the necessary handles from the private DM object */ 413 dmmoab = (DM_Moab*)(*dm)->data; 414 mbiface = dmmoab->mbiface; 415 pcomm = dmmoab->pcomm; 416 nprocs = pcomm->size(); 417 dmmoab->dim = dim; 418 419 /* TODO: Use command-line options to control by_rank, verbosity, MoabReadMode and extra options */ 420 ierr = PetscOptionsBegin(PETSC_COMM_WORLD, "", "Options for reading/writing MOAB based meshes from file", "DMMoab"); 421 ierr = PetscOptionsInt("-dmmb_rw_dbg", "The verbosity level for reading and writing MOAB meshes", "dmmbutil.cxx", dbglevel, &dbglevel, NULL);CHKERRQ(ierr); 422 ierr = PetscOptionsBool("-dmmb_part_by_rank", "Use partition by rank when reading MOAB meshes from file", "dmmbutil.cxx", part_by_rank, &part_by_rank, NULL);CHKERRQ(ierr); 423 ierr = PetscOptionsEnd(); 424 425 /* add mesh loading options specific to the DM */ 426 ierr = DMMoab_GetReadOptions_Private(part_by_rank, nprocs, dim, MOAB_PARROPTS_READ_PART, dbglevel, usrreadopts, &readopts);CHKERRQ(ierr); 427 428 PetscInfo2(*dm, "Reading file %s with options: %s\n",filename,readopts); 429 430 /* Load the mesh from a file. */ 431 merr = mbiface->load_file(filename, &dmmoab->fileset, readopts);MBERRVM(mbiface,"Reading MOAB file failed.", merr); 432 433 /* Reassign global IDs on all entities. */ 434 merr = pcomm->assign_global_ids(dmmoab->fileset,dim,1,true,true,true);MBERRNM(merr); 435 436 /* load the local vertices */ 437 merr = mbiface->get_entities_by_type(dmmoab->fileset, moab::MBVERTEX, verts, true);MBERRNM(merr); 438 /* load the local elements */ 439 merr = mbiface->get_entities_by_dimension(dmmoab->fileset, dim, elems, true);MBERRNM(merr); 440 441 /* Everything is set up, now just do a tag exchange to update tags 442 on all of the ghost vertexes */ 443 merr = pcomm->exchange_tags(dmmoab->ltog_tag,verts);MBERRV(mbiface,merr); 444 merr = pcomm->exchange_tags(dmmoab->ltog_tag,elems);MBERRV(mbiface,merr); 445 446 merr = pcomm->exchange_ghost_cells(dim,0,1,0,true,true,&dmmoab->fileset);MBERRV(mbiface,merr); 447 448 merr = pcomm->collective_sync_partition();MBERR("Collective sync failed", merr); 449 450 PetscInfo3(*dm, "MOAB file '%s' was successfully loaded. Found %D vertices and %D elements.\n", filename, verts.size(), elems.size()); 451 452 #if 1 453 std::stringstream sstr; 454 sstr << "test_" << pcomm->rank() << ".vtk"; 455 mbiface->write_mesh(sstr.str().c_str()); 456 #endif 457 PetscFunctionReturn(0); 458 } 459 460