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 /* create a file set to associate all entities in current mesh */ 142 merr = dmmoab->mbiface->create_meshset(moab::MESHSET_SET, dmmoab->fileset);MBERR("Creating file set failed", merr); 143 144 /* No errors yet; proceed with building the mesh */ 145 merr = mbiface->query_interface(readMeshIface);MBERRNM(merr); 146 147 ierr = PetscMemzero(ise,sizeof(PetscInt)*6);CHKERRQ(ierr); 148 149 /* call the collective routine that computes the domain bounds for a structured mesh using MOAB */ 150 ierr = DMMoabComputeDomainBounds_Private(pcomm, dim, nele, ise);CHKERRQ(ierr); 151 152 /* set some variables based on dimension */ 153 switch(dim) { 154 case 1: 155 vpere = 2; 156 locnele = (ise[1]-ise[0]); 157 locnpts = (ise[1]-ise[0]+1); 158 ghnele = (nghost > 0 ? (ise[0] > nghost ? 1 : 0) + (ise[1] < nele - nghost ? 1 : 0) : 0 ); 159 ghnpts = (nghost > 0 ? (ise[0] > 0 ? 1 : 0) + (ise[1] < nele ? 1 : 0) : 0); 160 etype = moab::MBEDGE; 161 break; 162 case 2: 163 vpere = 4; 164 locnele = (ise[1]-ise[0])*(ise[3]-ise[2]); 165 locnpts = (ise[1]-ise[0]+1)*(ise[3]-ise[2]+1); 166 ghnele = (nghost > 0 ? (ise[2] > 0 ? nele : 0) + (ise[3] < nele ? nele : 0) : 0); 167 ghnpts = (nghost > 0 ? (ise[2] > 0 ? npts : 0) + (ise[3] < nele ? npts : 0) : 0); 168 etype = moab::MBQUAD; 169 break; 170 case 3: 171 vpere = 8; 172 locnele = (ise[1]-ise[0])*(ise[3]-ise[2])*(ise[5]-ise[4]); 173 locnpts = (ise[1]-ise[0]+1)*(ise[3]-ise[2]+1)*(ise[5]-ise[4]+1); 174 ghnele = (nghost > 0 ? (ise[4] > 0 ? nele*nele : 0) + (ise[5] < nele ? nele*nele : 0) : 0); 175 ghnpts = (nghost > 0 ? (ise[4] > 0 ? npts*npts : 0) + (ise[5] < nele ? npts*npts : 0) : 0); 176 etype = moab::MBHEX; 177 break; 178 } 179 180 /* we have a domain of size [1,1,1] - now compute local co-ordinate box */ 181 ierr = PetscMemzero(xse,sizeof(PetscReal)*6);CHKERRQ(ierr); 182 for(i=0; i<6; ++i) { 183 xse[i]=(PetscReal)ise[i]/nele; 184 } 185 186 /* Create vertexes and set the coodinate of each vertex */ 187 merr = readMeshIface->get_node_coords(3,locnpts+ghnpts,0,vfirst,vcoords,n);MBERRNM(merr); 188 189 /* Compute the co-ordinates of vertices and global IDs */ 190 std::vector<int> vgid(locnpts+ghnpts); 191 int vcount=0; 192 double hxyz=1.0/nele; 193 194 /* create all the owned vertices */ 195 for (k = ise[4]; k <= ise[5]; k++) { 196 for (j = ise[2]; j <= ise[3]; j++) { 197 for (i = ise[0]; i <= ise[1]; i++, vcount++) { 198 set_structured_coordinates(i,j,k,hxyz,vcount,vcoords); 199 vgid[vcount] = (k*npts+j)*npts+i+1; 200 } 201 } 202 } 203 204 /* create ghosted vertices requested by user - below the current plane */ 205 if (ise[2*dim-2] > 0) { 206 for (k = (dim==3?ise[4]-nghost:ise[4]); k <= (dim==3?ise[4]-1:ise[5]); k++) { 207 for (j = (dim==2?ise[2]-nghost:ise[2]); j <= (dim==2?ise[2]-1:ise[3]); j++) { 208 for (i = (dim>1?ise[0]:ise[0]-nghost); i <= (dim>1?ise[1]:ise[0]-1); i++, vcount++) { 209 set_structured_coordinates(i,j,k,hxyz,vcount,vcoords); 210 vgid[vcount] = (k*npts+j)*npts+i+1; 211 } 212 } 213 } 214 } 215 216 /* create ghosted vertices requested by user - above the current plane */ 217 if (ise[2*dim-1] < nele) { 218 for (k = (dim==3?ise[5]+1:ise[4]); k <= (dim==3?ise[5]+nghost:ise[5]); k++) { 219 for (j = (dim==2?ise[3]+1:ise[2]); j <= (dim==2?ise[3]+nghost:ise[3]); j++) { 220 for (i = (dim>1?ise[0]:ise[1]+1); i <= (dim>1?ise[1]:ise[1]+nghost); i++, vcount++) { 221 set_structured_coordinates(i,j,k,hxyz,vcount,vcoords); 222 vgid[vcount] = (k*npts+j)*npts+i+1; 223 } 224 } 225 } 226 } 227 228 if (locnpts+ghnpts != vcount) SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Created the wrong number of vertices! (%D!=%D)",locnpts+ghnpts,vcount); 229 230 merr = mbiface->get_entities_by_type(0,moab::MBVERTEX,ownedvtx,true);MBERRNM(merr); 231 232 /* The global ID tag is applied to each owned 233 vertex. It acts as an global identifier which MOAB uses to 234 assemble the individual pieces of the mesh: 235 Set the global ID indices */ 236 merr = mbiface->tag_set_data(id_tag,ownedvtx,vgid.data());MBERRNM(merr); 237 238 /* Create elements between mesh points using the ReadUtilInterface 239 get the reference to element connectivities for all local elements from the ReadUtilInterface */ 240 merr = readMeshIface->get_element_connect (locnele+ghnele,vpere,etype,1,efirst,connectivity);MBERRNM(merr); 241 242 /* offset appropriately so that only local ID and not global ID numbers are set for connectivity array */ 243 // PetscPrintf(PETSC_COMM_SELF, "[%D] first local handle %D\n", rank, vfirst); 244 vfirst-=vgid[0]-1; 245 246 /* 3. Loop over elements in 3 nested loops over i, j, k; for each (i,j,k): 247 and then set the connectivity for each element appropriately */ 248 int ecount=0; 249 250 /* create ghosted elements requested by user - below the current plane */ 251 if (ise[2*dim-2] >= nghost) { 252 for (k = (dim==3?ise[4]-nghost:ise[4]); k < (dim==3?ise[4]:std::max(ise[5],1)); k++) { 253 for (j = (dim==2?ise[2]-nghost:ise[2]); j < (dim==2?ise[2]:std::max(ise[3],1)); j++) { 254 for (i = (dim>1?ise[0]:ise[0]-nghost); i < (dim>1?std::max(ise[1],1):ise[0]); i++, ecount++) { 255 set_element_connectivity(dim, etype, ecount*vpere, nele, i, j, k, vfirst, connectivity); 256 } 257 } 258 } 259 } 260 261 /* create owned elements requested by user */ 262 for (k = ise[4]; k < std::max(ise[5],1); k++) { 263 for (j = ise[2]; j < std::max(ise[3],1); j++) { 264 for (i = ise[0]; i < std::max(ise[1],1); i++,ecount++) { 265 set_element_connectivity(dim, etype, ecount*vpere, nele, i, j, k, vfirst, connectivity); 266 } 267 } 268 } 269 270 /* create ghosted elements requested by user - above the current plane */ 271 if (ise[2*dim-1] <= nele-nghost) { 272 for (k = (dim==3?ise[5]:ise[4]); k < (dim==3?ise[5]+nghost:std::max(ise[5],1)); k++) { 273 for (j = (dim==2?ise[3]:ise[2]); j < (dim==2?ise[3]+nghost:std::max(ise[3],1)); j++) { 274 for (i = (dim>1?ise[0]:ise[1]); i < (dim>1?std::max(ise[1],1):ise[1]+nghost); i++, ecount++) { 275 set_element_connectivity(dim, etype, ecount*vpere, nele, i, j, k, vfirst, connectivity); 276 } 277 } 278 } 279 } 280 281 merr = readMeshIface->update_adjacencies(efirst,locnele+ghnele,vpere,connectivity);MBERRNM(merr); 282 283 /* 2. Get the vertices and hexes from moab and check their numbers against I*J*K and (I-1)*(J-1)*(K-1), resp. 284 first '0' specifies "root set", or entire MOAB instance, second the entity dimension being requested */ 285 merr = mbiface->get_entities_by_dimension(0, dim, ownedelms);MBERRNM(merr); 286 287 if (locnele+ghnele != (int) ownedelms.size()) 288 SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Created the wrong number of elements! (%D!=%D)",locnele+ghnele,ownedelms.size()); 289 else if(locnpts+ghnpts != (int) ownedvtx.size()) 290 SETERRQ2(PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"Created the wrong number of vertices! (%D!=%D)",locnpts+ghnpts,ownedvtx.size()); 291 else 292 PetscInfo2(*dm, "Created %D elements and %D vertices.\n", ownedelms.size(), ownedvtx.size()); 293 294 /* lets create some sets */ 295 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); 296 297 merr = mbiface->create_meshset(moab::MESHSET_SET, regionset);MBERRNM(merr); 298 merr = mbiface->add_entities(regionset, ownedelms);MBERRNM(merr); 299 merr = mbiface->tag_set_data(geom_tag, ®ionset, 1, &dmmoab->dim);MBERRNM(merr); 300 merr = mbiface->add_parent_child(dmmoab->fileset,regionset);MBERRNM(merr); 301 merr = mbiface->unite_meshset(dmmoab->fileset, regionset);MBERRNM(merr); 302 303 merr = mbiface->create_meshset(moab::MESHSET_SET, vtxset);MBERRNM(merr); 304 merr = mbiface->add_entities(vtxset, ownedvtx);MBERRNM(merr); 305 merr = mbiface->add_parent_child(dmmoab->fileset,vtxset);MBERRNM(merr); 306 merr = mbiface->unite_meshset(dmmoab->fileset, vtxset);MBERRNM(merr); 307 308 if (build_adjacencies) { 309 // generate all lower dimensional adjacencies 310 merr = mbiface->get_adjacencies( ownedelms, dim-1, true, adj, moab::Interface::UNION );MBERRNM(merr); 311 merr = dmmoab->pcomm->get_part_entities(dim2, dim-1);MBERRNM(merr); 312 adj.merge(dim2); 313 314 /* create face sets */ 315 merr = mbiface->create_meshset(moab::MESHSET_SET, faceset);MBERRNM(merr); 316 merr = mbiface->add_entities(faceset, adj);MBERRNM(merr); 317 merr = mbiface->add_parent_child(dmmoab->fileset,faceset);MBERRNM(merr); 318 i=dim-1; 319 merr = mbiface->tag_set_data(geom_tag, &faceset, 1, &i);MBERRNM(merr); 320 merr = mbiface->unite_meshset(dmmoab->fileset, faceset);MBERRNM(merr); 321 PetscInfo2(dm, "Found %d %d-Dim quantities.\n", adj.size(), dim-1); 322 323 if (dim > 2) { 324 dim2.clear(); 325 /* create edge sets, if appropriate i.e., if dim=3 */ 326 merr = mbiface->create_meshset(moab::MESHSET_SET, edgeset);MBERRNM(merr); 327 merr = mbiface->get_adjacencies(adj, dim-1, true, dim2, moab::Interface::UNION );MBERRNM(merr); 328 merr = mbiface->add_entities(edgeset, dim2);MBERRNM(merr); 329 merr = mbiface->add_parent_child(dmmoab->fileset,edgeset);MBERRNM(merr); 330 i=dim-2; 331 merr = mbiface->tag_set_data(geom_tag, &edgeset, 1, &i);MBERRNM(merr); 332 merr = mbiface->unite_meshset(dmmoab->fileset, edgeset);MBERRNM(merr); 333 PetscInfo2(dm, "Found %d %d-Dim quantities.\n", adj.size(), dim-2); 334 } 335 } 336 337 /* check the handles */ 338 merr = pcomm->check_all_shared_handles();MBERRV(mbiface,merr); 339 340 #if 0 341 std::stringstream sstr; 342 sstr << "test_" << pcomm->rank() << ".vtk"; 343 mbiface->write_mesh(sstr.str().c_str()); 344 #endif 345 346 /* resolve the shared entities by exchanging information to adjacent processors */ 347 merr = mbiface->get_entities_by_type(dmmoab->fileset,etype,ownedelms,true);MBERRNM(merr); 348 merr = pcomm->resolve_shared_ents(dmmoab->fileset,ownedelms,dim,dim-1,NULL,&id_tag);MBERRV(mbiface,merr); 349 350 /* Reassign global IDs on all entities. */ 351 merr = pcomm->assign_global_ids(dmmoab->fileset,dim,1,true,true,true);MBERRNM(merr); 352 353 merr = pcomm->exchange_ghost_cells(dim,0,user_nghost,dim,true,false,&dmmoab->fileset);MBERRV(mbiface,merr); 354 355 /* Everything is set up, now just do a tag exchange to update tags 356 on all of the ghost vertexes */ 357 merr = mbiface->get_entities_by_type(dmmoab->fileset,moab::MBVERTEX,ownedvtx,true);MBERRNM(merr); 358 merr = mbiface->get_entities_by_dimension(dmmoab->fileset, dim, ownedelms);MBERRNM(merr); 359 360 merr = pcomm->exchange_tags(id_tag,ownedvtx);MBERRV(mbiface,merr); 361 merr = pcomm->exchange_tags(id_tag,ownedelms);MBERRV(mbiface,merr); 362 PetscFunctionReturn(0); 363 } 364 365 366 #undef __FUNCT__ 367 #define __FUNCT__ "DMMoab_GetReadOptions_Private" 368 PetscErrorCode DMMoab_GetReadOptions_Private(PetscBool by_rank, PetscInt numproc, PetscInt dim, MoabReadMode mode, PetscInt dbglevel, const char* extra_opts, const char** read_opts) 369 { 370 std::ostringstream str; 371 372 PetscFunctionBegin; 373 /* do parallel read unless using only one processor */ 374 if (numproc > 1) { 375 str << "PARALLEL=" << mode << ";PARTITION=PARALLEL_PARTITION;PARTITION_DISTRIBUTE;"; 376 str << "PARALLEL_RESOLVE_SHARED_ENTS;PARALLEL_GHOSTS=" << dim << ".0.1;"; 377 if (by_rank) 378 str << "PARTITION_BY_RANK;"; 379 } 380 381 if (dbglevel) { 382 str << "CPUTIME;DEBUG_IO=" << dbglevel << ";"; 383 if (numproc>1) 384 str << "DEBUG_PIO=" << dbglevel << ";"; 385 } 386 387 if (extra_opts) 388 str << extra_opts; 389 390 *read_opts = str.str().c_str(); 391 PetscFunctionReturn(0); 392 } 393 394 395 #undef __FUNCT__ 396 #define __FUNCT__ "DMMoabLoadFromFile" 397 PetscErrorCode DMMoabLoadFromFile(MPI_Comm comm,PetscInt dim,const char* filename, const char* usrreadopts, DM *dm) 398 { 399 moab::ErrorCode merr; 400 PetscInt nprocs,dbglevel=0; 401 PetscBool part_by_rank=PETSC_FALSE; 402 DM_Moab *dmmoab; 403 moab::Interface *mbiface; 404 moab::ParallelComm *pcomm; 405 moab::Range verts,elems; 406 const char *readopts; 407 PetscErrorCode ierr; 408 409 PetscFunctionBegin; 410 PetscValidPointer(dm,5); 411 412 /* Create the basic DMMoab object and keep the default parameters created by DM impls */ 413 ierr = DMMoabCreateMoab(comm, PETSC_NULL, PETSC_NULL, PETSC_NULL, PETSC_NULL, dm);CHKERRQ(ierr); 414 415 /* get all the necessary handles from the private DM object */ 416 dmmoab = (DM_Moab*)(*dm)->data; 417 mbiface = dmmoab->mbiface; 418 pcomm = dmmoab->pcomm; 419 nprocs = pcomm->size(); 420 dmmoab->dim = dim; 421 422 /* create a file set to associate all entities in current mesh */ 423 merr = dmmoab->mbiface->create_meshset(moab::MESHSET_SET, dmmoab->fileset);MBERR("Creating file set failed", merr); 424 425 /* TODO: Use command-line options to control by_rank, verbosity, MoabReadMode and extra options */ 426 ierr = PetscOptionsBegin(PETSC_COMM_WORLD, "", "Options for reading/writing MOAB based meshes from file", "DMMoab"); 427 ierr = PetscOptionsInt("-dmmb_rw_dbg", "The verbosity level for reading and writing MOAB meshes", "dmmbutil.cxx", dbglevel, &dbglevel, NULL);CHKERRQ(ierr); 428 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); 429 ierr = PetscOptionsEnd(); 430 431 /* add mesh loading options specific to the DM */ 432 ierr = DMMoab_GetReadOptions_Private(part_by_rank, nprocs, dim, MOAB_PARROPTS_READ_PART, dbglevel, usrreadopts, &readopts);CHKERRQ(ierr); 433 434 PetscInfo2(*dm, "Reading file %s with options: %s\n",filename,readopts); 435 436 /* Load the mesh from a file. */ 437 merr = mbiface->load_file(filename, &dmmoab->fileset, readopts);MBERRVM(mbiface,"Reading MOAB file failed.", merr); 438 439 /* Reassign global IDs on all entities. */ 440 merr = pcomm->assign_global_ids(dmmoab->fileset,dim,1,true,true,true);MBERRNM(merr); 441 442 /* load the local vertices */ 443 merr = mbiface->get_entities_by_type(dmmoab->fileset, moab::MBVERTEX, verts, true);MBERRNM(merr); 444 /* load the local elements */ 445 merr = mbiface->get_entities_by_dimension(dmmoab->fileset, dim, elems, true);MBERRNM(merr); 446 447 /* Everything is set up, now just do a tag exchange to update tags 448 on all of the ghost vertexes */ 449 merr = pcomm->exchange_tags(dmmoab->ltog_tag,verts);MBERRV(mbiface,merr); 450 merr = pcomm->exchange_tags(dmmoab->ltog_tag,elems);MBERRV(mbiface,merr); 451 452 merr = pcomm->exchange_ghost_cells(dim,0,1,0,true,true,&dmmoab->fileset);MBERRV(mbiface,merr); 453 454 merr = pcomm->collective_sync_partition();MBERR("Collective sync failed", merr); 455 456 PetscInfo3(*dm, "MOAB file '%s' was successfully loaded. Found %D vertices and %D elements.\n", filename, verts.size(), elems.size()); 457 458 #if 1 459 std::stringstream sstr; 460 sstr << "test_" << pcomm->rank() << ".vtk"; 461 mbiface->write_mesh(sstr.str().c_str()); 462 #endif 463 PetscFunctionReturn(0); 464 } 465 466