xref: /petsc/src/dm/impls/moab/dmmbutil.cxx (revision 5f80ce2ab25dff0f4601e710601cbbcecf323266)
1af0996ceSBarry Smith #include <petsc/private/dmmbimpl.h> /*I  "petscdmmoab.h"   I*/
2af0996ceSBarry Smith #include <petsc/private/vecimpl.h>
351d15aeeSVijay Mahadevan 
451d15aeeSVijay Mahadevan #include <petscdmmoab.h>
551d15aeeSVijay Mahadevan #include <MBTagConventions.hpp>
651d15aeeSVijay Mahadevan #include <moab/ReadUtilIface.hpp>
7ce27a4eeSVijay Mahadevan #include <moab/MergeMesh.hpp>
851d15aeeSVijay Mahadevan #include <moab/CN.hpp>
951d15aeeSVijay Mahadevan 
10ce27a4eeSVijay Mahadevan typedef struct {
11ce27a4eeSVijay Mahadevan   // options
1249d66b22SVijay Mahadevan   PetscInt  A, B, C, M, N, K, dim;
13ce27a4eeSVijay Mahadevan   PetscInt  blockSizeVertexXYZ[3];              // Number of element blocks per partition
14ce27a4eeSVijay Mahadevan   PetscInt  blockSizeElementXYZ[3];
15ce27a4eeSVijay Mahadevan   PetscReal xyzbounds[6]; // the physical size of the domain
16ce27a4eeSVijay Mahadevan   bool      newMergeMethod, keep_skins, simplex, adjEnts;
1751d15aeeSVijay Mahadevan 
18ce27a4eeSVijay Mahadevan   // compute params
19ce27a4eeSVijay Mahadevan   PetscReal dx, dy, dz;
20ce27a4eeSVijay Mahadevan   PetscInt  NX, NY, NZ, nex, ney, nez;
21ce27a4eeSVijay Mahadevan   PetscInt  q, xstride, ystride, zstride;
22ce27a4eeSVijay Mahadevan   PetscBool usrxyzgrid, usrprocgrid, usrrefgrid;
23755f3dfbSVijay Mahadevan   PetscInt  fraction, remainder, cumfraction;
24755f3dfbSVijay Mahadevan   PetscLogEvent generateMesh, generateElements, generateVertices, parResolve;
25ce27a4eeSVijay Mahadevan } DMMoabMeshGeneratorCtx;
26ce27a4eeSVijay Mahadevan 
2749d66b22SVijay Mahadevan PetscInt DMMoab_SetTensorElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, PetscInt offset, PetscInt corner, std::vector<PetscInt>& subent_conn, moab::EntityHandle *connectivity)
2851d15aeeSVijay Mahadevan {
29ce27a4eeSVijay Mahadevan   switch (genCtx.dim) {
30e5136372SVijay Mahadevan   case 1:
31ce27a4eeSVijay Mahadevan     subent_conn.resize(2);
32ce27a4eeSVijay Mahadevan     moab::CN::SubEntityVertexIndices(moab::MBEDGE, 1, 0, subent_conn.data());
33ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[0]] = corner;
34ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[1]] = corner + 1;
35e5136372SVijay Mahadevan     break;
36e5136372SVijay Mahadevan   case 2:
37ce27a4eeSVijay Mahadevan     subent_conn.resize(4);
38ce27a4eeSVijay Mahadevan     moab::CN::SubEntityVertexIndices(moab::MBQUAD, 2, 0, subent_conn.data());
39ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[0]] = corner;
40ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[1]] = corner + 1;
41ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[2]] = corner + 1 + genCtx.ystride;
42ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[3]] = corner + genCtx.ystride;
43e5136372SVijay Mahadevan     break;
44e5136372SVijay Mahadevan   case 3:
45e5136372SVijay Mahadevan   default:
46ce27a4eeSVijay Mahadevan     subent_conn.resize(8);
47ce27a4eeSVijay Mahadevan     moab::CN::SubEntityVertexIndices(moab::MBHEX, 3, 0, subent_conn.data());
48ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[0]] = corner;
49ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[1]] = corner + 1;
50ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[2]] = corner + 1 + genCtx.ystride;
51ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[3]] = corner + genCtx.ystride;
52ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[4]] = corner + genCtx.zstride;
53ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[5]] = corner + 1 + genCtx.zstride;
54ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[6]] = corner + 1 + genCtx.ystride + genCtx.zstride;
55ce27a4eeSVijay Mahadevan     connectivity[offset + subent_conn[7]] = corner + genCtx.ystride + genCtx.zstride;
56e5136372SVijay Mahadevan     break;
57e5136372SVijay Mahadevan   }
58ce27a4eeSVijay Mahadevan   return subent_conn.size();
59e5136372SVijay Mahadevan }
6051d15aeeSVijay Mahadevan 
6149d66b22SVijay Mahadevan PetscInt DMMoab_SetSimplexElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, PetscInt subelem, PetscInt offset, PetscInt corner, std::vector<PetscInt>& subent_conn, moab::EntityHandle *connectivity)
62c3dd00c7SVijay Mahadevan {
63755f3dfbSVijay Mahadevan   PetscInt A, B, C, D, E, F, G, H, M;
64755f3dfbSVijay Mahadevan   const PetscInt trigen_opts = 1; /* 1 - Aligned diagonally to right, 2 - Aligned diagonally to left, 3 - 4 elements per quad */
6549d66b22SVijay Mahadevan   A = corner;
6649d66b22SVijay Mahadevan   B = corner + 1;
67ce27a4eeSVijay Mahadevan   switch (genCtx.dim) {
68c3dd00c7SVijay Mahadevan   case 1:
69ce27a4eeSVijay Mahadevan     subent_conn.resize(2);  /* only linear EDGE supported now */
70ce27a4eeSVijay Mahadevan     moab::CN::SubEntityVertexIndices(moab::MBEDGE, 1, 0, subent_conn.data());
7149d66b22SVijay Mahadevan     connectivity[offset + subent_conn[0]] = A;
7249d66b22SVijay Mahadevan     connectivity[offset + subent_conn[1]] = B;
73c3dd00c7SVijay Mahadevan     break;
74c3dd00c7SVijay Mahadevan   case 2:
7549d66b22SVijay Mahadevan     C = corner + 1 + genCtx.ystride;
7649d66b22SVijay Mahadevan     D = corner +     genCtx.ystride;
7763d025dbSVijay Mahadevan     M = corner + 0.5; /* technically -- need to modify vertex generation */
78ce27a4eeSVijay Mahadevan     subent_conn.resize(3);  /* only linear TRI supported */
79ce27a4eeSVijay Mahadevan     moab::CN::SubEntityVertexIndices(moab::MBTRI, 2, 0, subent_conn.data());
80755f3dfbSVijay Mahadevan     if (trigen_opts == 1) {
81ce27a4eeSVijay Mahadevan       if (subelem) { /* 0 1 2 of a QUAD */
8277a8c067SVijay Mahadevan         connectivity[offset + subent_conn[0]] = B;
83755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[1]] = C;
8477a8c067SVijay Mahadevan         connectivity[offset + subent_conn[2]] = A;
85c3dd00c7SVijay Mahadevan       }
86ce27a4eeSVijay Mahadevan       else {        /* 2 3 0 of a QUAD */
8777a8c067SVijay Mahadevan         connectivity[offset + subent_conn[0]] = D;
8877a8c067SVijay Mahadevan         connectivity[offset + subent_conn[1]] = A;
89755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[2]] = C;
90755f3dfbSVijay Mahadevan       }
91755f3dfbSVijay Mahadevan     }
92755f3dfbSVijay Mahadevan     else if (trigen_opts == 2) {
93755f3dfbSVijay Mahadevan       if (subelem) { /* 0 1 2 of a QUAD */
94755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[0]] = A;
9577a8c067SVijay Mahadevan         connectivity[offset + subent_conn[1]] = B;
9677a8c067SVijay Mahadevan         connectivity[offset + subent_conn[2]] = D;
97755f3dfbSVijay Mahadevan       }
98755f3dfbSVijay Mahadevan       else {        /* 2 3 0 of a QUAD */
9977a8c067SVijay Mahadevan         connectivity[offset + subent_conn[0]] = C;
10077a8c067SVijay Mahadevan         connectivity[offset + subent_conn[1]] = D;
101755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[2]] = B;
102755f3dfbSVijay Mahadevan       }
103755f3dfbSVijay Mahadevan     }
104755f3dfbSVijay Mahadevan     else {
105755f3dfbSVijay Mahadevan       switch (subelem) { /* 0 1 2 of a QUAD */
106755f3dfbSVijay Mahadevan       case 0:
107755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[0]] = A;
108755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[1]] = B;
109755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[2]] = M;
110755f3dfbSVijay Mahadevan         break;
111755f3dfbSVijay Mahadevan       case 1:
112755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[0]] = B;
113755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[1]] = C;
114755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[2]] = M;
115755f3dfbSVijay Mahadevan         break;
116755f3dfbSVijay Mahadevan       case 2:
11749d66b22SVijay Mahadevan         connectivity[offset + subent_conn[0]] = C;
11849d66b22SVijay Mahadevan         connectivity[offset + subent_conn[1]] = D;
119755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[2]] = M;
120755f3dfbSVijay Mahadevan         break;
121755f3dfbSVijay Mahadevan       case 3:
122755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[0]] = D;
123755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[1]] = A;
124755f3dfbSVijay Mahadevan         connectivity[offset + subent_conn[2]] = M;
125755f3dfbSVijay Mahadevan         break;
126755f3dfbSVijay Mahadevan       }
127c3dd00c7SVijay Mahadevan     }
128c3dd00c7SVijay Mahadevan     break;
129c3dd00c7SVijay Mahadevan   case 3:
130c3dd00c7SVijay Mahadevan   default:
13149d66b22SVijay Mahadevan     C = corner + 1 + genCtx.ystride;
13249d66b22SVijay Mahadevan     D = corner +     genCtx.ystride;
13349d66b22SVijay Mahadevan     E = corner +                      genCtx.zstride;
13449d66b22SVijay Mahadevan     F = corner + 1 +                  genCtx.zstride;
13549d66b22SVijay Mahadevan     G = corner + 1 + genCtx.ystride + genCtx.zstride;
13649d66b22SVijay Mahadevan     H = corner +     genCtx.ystride + genCtx.zstride;
137ce27a4eeSVijay Mahadevan     subent_conn.resize(4);  /* only linear TET supported */
138ce27a4eeSVijay Mahadevan     moab::CN::SubEntityVertexIndices(moab::MBTET, 3, 0, subent_conn.data());
139c3dd00c7SVijay Mahadevan     switch (subelem) {
14049d66b22SVijay Mahadevan     case 0: /* 4 3 7 6 of a HEX */
14149d66b22SVijay Mahadevan       connectivity[offset + subent_conn[0]] = E;
14249d66b22SVijay Mahadevan       connectivity[offset + subent_conn[1]] = D;
14349d66b22SVijay Mahadevan       connectivity[offset + subent_conn[2]] = H;
14449d66b22SVijay Mahadevan       connectivity[offset + subent_conn[3]] = G;
145c3dd00c7SVijay Mahadevan       break;
14649d66b22SVijay Mahadevan     case 1: /* 0 1 2 5 of a HEX */
14749d66b22SVijay Mahadevan       connectivity[offset + subent_conn[0]] = A;
14849d66b22SVijay Mahadevan       connectivity[offset + subent_conn[1]] = B;
14949d66b22SVijay Mahadevan       connectivity[offset + subent_conn[2]] = C;
15049d66b22SVijay Mahadevan       connectivity[offset + subent_conn[3]] = F;
151c3dd00c7SVijay Mahadevan       break;
15249d66b22SVijay Mahadevan     case 2: /* 0 3 4 5 of a HEX */
15349d66b22SVijay Mahadevan       connectivity[offset + subent_conn[0]] = A;
15449d66b22SVijay Mahadevan       connectivity[offset + subent_conn[1]] = D;
15549d66b22SVijay Mahadevan       connectivity[offset + subent_conn[2]] = E;
15649d66b22SVijay Mahadevan       connectivity[offset + subent_conn[3]] = F;
157c3dd00c7SVijay Mahadevan       break;
15849d66b22SVijay Mahadevan     case 3: /* 2 6 3 5 of a HEX */
15949d66b22SVijay Mahadevan       connectivity[offset + subent_conn[0]] = C;
16049d66b22SVijay Mahadevan       connectivity[offset + subent_conn[1]] = G;
16149d66b22SVijay Mahadevan       connectivity[offset + subent_conn[2]] = D;
16249d66b22SVijay Mahadevan       connectivity[offset + subent_conn[3]] = F;
163c3dd00c7SVijay Mahadevan       break;
16449d66b22SVijay Mahadevan     case 4: /* 0 2 3 5 of a HEX */
16549d66b22SVijay Mahadevan       connectivity[offset + subent_conn[0]] = A;
16649d66b22SVijay Mahadevan       connectivity[offset + subent_conn[1]] = C;
16749d66b22SVijay Mahadevan       connectivity[offset + subent_conn[2]] = D;
16849d66b22SVijay Mahadevan       connectivity[offset + subent_conn[3]] = F;
16949d66b22SVijay Mahadevan       break;
17049d66b22SVijay Mahadevan     case 5: /* 3 6 4 5 of a HEX */
17149d66b22SVijay Mahadevan       connectivity[offset + subent_conn[0]] = D;
17249d66b22SVijay Mahadevan       connectivity[offset + subent_conn[1]] = G;
17349d66b22SVijay Mahadevan       connectivity[offset + subent_conn[2]] = E;
17449d66b22SVijay Mahadevan       connectivity[offset + subent_conn[3]] = F;
175c3dd00c7SVijay Mahadevan       break;
176c3dd00c7SVijay Mahadevan     }
177c3dd00c7SVijay Mahadevan     break;
178c3dd00c7SVijay Mahadevan   }
179ce27a4eeSVijay Mahadevan   return subent_conn.size();
180c3dd00c7SVijay Mahadevan }
181c3dd00c7SVijay Mahadevan 
18249d66b22SVijay Mahadevan std::pair<PetscInt, PetscInt> DMMoab_SetElementConnectivity_Private(DMMoabMeshGeneratorCtx& genCtx, PetscInt offset, PetscInt corner, moab::EntityHandle *connectivity)
183c3dd00c7SVijay Mahadevan {
18449d66b22SVijay Mahadevan   PetscInt vcount = 0;
18549d66b22SVijay Mahadevan   PetscInt simplices_per_tensor[4] = {0, 1, 2, 6};
18649d66b22SVijay Mahadevan   std::vector<PetscInt> subent_conn;  /* only linear edge, tri, tet supported now */
187ce27a4eeSVijay Mahadevan   subent_conn.reserve(27);
18849d66b22SVijay Mahadevan   PetscInt m, subelem;
189ce27a4eeSVijay Mahadevan   if (genCtx.simplex) {
19049d66b22SVijay Mahadevan     subelem = simplices_per_tensor[genCtx.dim];
191ce27a4eeSVijay Mahadevan     for (m = 0; m < subelem; m++) {
192ce27a4eeSVijay Mahadevan       vcount = DMMoab_SetSimplexElementConnectivity_Private(genCtx, m, offset, corner, subent_conn, connectivity);
193ce27a4eeSVijay Mahadevan       offset += vcount;
194ce27a4eeSVijay Mahadevan     }
195c3dd00c7SVijay Mahadevan   }
196c3dd00c7SVijay Mahadevan   else {
197c3dd00c7SVijay Mahadevan     subelem = 1;
198ce27a4eeSVijay Mahadevan     vcount = DMMoab_SetTensorElementConnectivity_Private(genCtx, offset, corner, subent_conn, connectivity);
199c3dd00c7SVijay Mahadevan   }
20049d66b22SVijay Mahadevan   return std::pair<PetscInt, PetscInt>(vcount * subelem, subelem);
201c3dd00c7SVijay Mahadevan }
202c3dd00c7SVijay Mahadevan 
20349d66b22SVijay Mahadevan PetscErrorCode DMMoab_GenerateVertices_Private(moab::Interface *mbImpl, moab::ReadUtilIface *iface, DMMoabMeshGeneratorCtx& genCtx, PetscInt m, PetscInt n, PetscInt k,
20449d66b22SVijay Mahadevan     PetscInt a, PetscInt b, PetscInt c, moab::Tag& global_id_tag, moab::EntityHandle& startv, moab::Range& uverts)
205ce27a4eeSVijay Mahadevan {
20649d66b22SVijay Mahadevan   PetscInt x, y, z, ix, nnodes;
20749d66b22SVijay Mahadevan   PetscInt ii, jj, kk;
20849d66b22SVijay Mahadevan   std::vector<PetscReal*> arrays;
209755f3dfbSVijay Mahadevan   PetscInt* gids;
210ce27a4eeSVijay Mahadevan   moab::ErrorCode merr;
211ce27a4eeSVijay Mahadevan 
212ce27a4eeSVijay Mahadevan   PetscFunctionBegin;
213ce27a4eeSVijay Mahadevan   /* we will generate (q*block+1)^3 vertices, and block^3 hexas; q is 1 for linear, 2 for quadratic
214ce27a4eeSVijay Mahadevan    * the global id of the vertices will come from m, n, k, a, b, c
215ce27a4eeSVijay Mahadevan    * x will vary from  m*A*q*block + a*q*block to m*A*q*block+(a+1)*q*block etc.
216ce27a4eeSVijay Mahadevan    */
217ce27a4eeSVijay Mahadevan   nnodes = genCtx.blockSizeVertexXYZ[0] * (genCtx.dim > 1 ? genCtx.blockSizeVertexXYZ[1] * (genCtx.dim > 2 ? genCtx.blockSizeVertexXYZ[2] : 1) : 1);
218*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(nnodes, &gids));
219ce27a4eeSVijay Mahadevan 
220ce27a4eeSVijay Mahadevan   merr = iface->get_node_coords(3, nnodes, 0, startv, arrays);MBERR("Can't get node coords.", merr);
221ce27a4eeSVijay Mahadevan 
222ce27a4eeSVijay Mahadevan   /* will start with the lower corner: */
22363d025dbSVijay Mahadevan   /* x = ( m * genCtx.A + a) * genCtx.q * genCtx.blockSizeElementXYZ[0]; */
22463d025dbSVijay Mahadevan   /* y = ( n * genCtx.B + b) * genCtx.q * genCtx.blockSizeElementXYZ[1]; */
22563d025dbSVijay Mahadevan   /* z = ( k * genCtx.C + c) * genCtx.q * genCtx.blockSizeElementXYZ[2]; */
226755f3dfbSVijay Mahadevan 
227755f3dfbSVijay Mahadevan   x = ( m * genCtx.A + a) * genCtx.q;
228755f3dfbSVijay Mahadevan   y = ( n * genCtx.B + b) * genCtx.q;
229755f3dfbSVijay Mahadevan   z = ( k * genCtx.C + c) * genCtx.q;
230*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(NULL, "Starting offset for coordinates := %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT "\n", x, y, z));
231ce27a4eeSVijay Mahadevan   ix = 0;
232ce27a4eeSVijay Mahadevan   moab::Range verts(startv, startv + nnodes - 1);
233ce27a4eeSVijay Mahadevan   for (kk = 0; kk < (genCtx.dim > 2 ? genCtx.blockSizeVertexXYZ[2] : 1); kk++) {
234ce27a4eeSVijay Mahadevan     for (jj = 0; jj < (genCtx.dim > 1 ? genCtx.blockSizeVertexXYZ[1] : 1); jj++) {
235ce27a4eeSVijay Mahadevan       for (ii = 0; ii < genCtx.blockSizeVertexXYZ[0]; ii++, ix++) {
236ce27a4eeSVijay Mahadevan         /* set coordinates for the vertices */
237ce27a4eeSVijay Mahadevan         arrays[0][ix] = (x + ii) * genCtx.dx + genCtx.xyzbounds[0];
238ce27a4eeSVijay Mahadevan         arrays[1][ix] = (y + jj) * genCtx.dy + genCtx.xyzbounds[2];
239ce27a4eeSVijay Mahadevan         arrays[2][ix] = (z + kk) * genCtx.dz + genCtx.xyzbounds[4];
240*5f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscInfo(NULL, "Creating vertex with coordinates := %f, %f, %f\n", arrays[0][ix], arrays[1][ix], arrays[2][ix]));
241ce27a4eeSVijay Mahadevan 
242ce27a4eeSVijay Mahadevan         /* If we want to set some tags on the vertices -> use the following entity handle definition:
243ce27a4eeSVijay Mahadevan            moab::EntityHandle v = startv + ix;
244ce27a4eeSVijay Mahadevan         */
245ce27a4eeSVijay Mahadevan         /* compute the global ID for vertex */
246ce27a4eeSVijay Mahadevan         gids[ix] = 1 + (x + ii) + (y + jj) * genCtx.NX + (z + kk) * (genCtx.NX * genCtx.NY);
247ce27a4eeSVijay Mahadevan       }
248ce27a4eeSVijay Mahadevan     }
249ce27a4eeSVijay Mahadevan   }
250ce27a4eeSVijay Mahadevan   /* set global ID data on vertices */
251ce27a4eeSVijay Mahadevan   mbImpl->tag_set_data(global_id_tag, verts, &gids[0]);
252ce27a4eeSVijay Mahadevan   verts.swap(uverts);
253*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(gids));
254ce27a4eeSVijay Mahadevan   PetscFunctionReturn(0);
255ce27a4eeSVijay Mahadevan }
256ce27a4eeSVijay Mahadevan 
25749d66b22SVijay Mahadevan PetscErrorCode DMMoab_GenerateElements_Private(moab::Interface* mbImpl, moab::ReadUtilIface* iface, DMMoabMeshGeneratorCtx& genCtx, PetscInt m, PetscInt n, PetscInt k,
25849d66b22SVijay Mahadevan     PetscInt a, PetscInt b, PetscInt c, moab::Tag& global_id_tag, moab::EntityHandle startv, moab::Range& cells)
259ce27a4eeSVijay Mahadevan {
260ce27a4eeSVijay Mahadevan   moab::ErrorCode merr;
261ce27a4eeSVijay Mahadevan   PetscInt ix, ie, xe, ye, ze;
262ce27a4eeSVijay Mahadevan   PetscInt ii, jj, kk, nvperelem;
26349d66b22SVijay Mahadevan   PetscInt simplices_per_tensor[4] = {0, 1, 2, 6};
264ce27a4eeSVijay 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);*/
265ce27a4eeSVijay Mahadevan   PetscInt nelems = ntensorelems;
26663d025dbSVijay Mahadevan   moab::EntityHandle starte; /* connectivity */
267ce27a4eeSVijay Mahadevan   moab::EntityHandle* conn;
268ce27a4eeSVijay Mahadevan 
269ce27a4eeSVijay Mahadevan   PetscFunctionBegin;
270ce27a4eeSVijay Mahadevan   switch (genCtx.dim) {
271ce27a4eeSVijay Mahadevan   case 1:
272ce27a4eeSVijay Mahadevan     nvperelem = 2;
273ce27a4eeSVijay Mahadevan     merr = iface->get_element_connect(nelems, 2, moab::MBEDGE, 0, starte, conn);MBERR("Can't get EDGE2 element connectivity.", merr);
274ce27a4eeSVijay Mahadevan     break;
275ce27a4eeSVijay Mahadevan   case 2:
276ce27a4eeSVijay Mahadevan     if (genCtx.simplex) {
277ce27a4eeSVijay Mahadevan       nvperelem = 3;
27849d66b22SVijay Mahadevan       nelems = ntensorelems * simplices_per_tensor[genCtx.dim];
279ce27a4eeSVijay Mahadevan       merr = iface->get_element_connect(nelems, 3, moab::MBTRI, 0, starte, conn);MBERR("Can't get TRI3 element connectivity.", merr);
280ce27a4eeSVijay Mahadevan     }
281ce27a4eeSVijay Mahadevan     else {
282ce27a4eeSVijay Mahadevan       nvperelem = 4;
283ce27a4eeSVijay Mahadevan       merr = iface->get_element_connect(nelems, 4, moab::MBQUAD, 0, starte, conn);MBERR("Can't get QUAD4 element connectivity.", merr);
284ce27a4eeSVijay Mahadevan     }
285ce27a4eeSVijay Mahadevan     break;
286ce27a4eeSVijay Mahadevan   case 3:
287ce27a4eeSVijay Mahadevan   default:
288ce27a4eeSVijay Mahadevan     if (genCtx.simplex) {
289ce27a4eeSVijay Mahadevan       nvperelem = 4;
29049d66b22SVijay Mahadevan       nelems = ntensorelems * simplices_per_tensor[genCtx.dim];
291ce27a4eeSVijay Mahadevan       merr = iface->get_element_connect(nelems, 4, moab::MBTET, 0, starte, conn);MBERR("Can't get TET4 element connectivity.", merr);
292ce27a4eeSVijay Mahadevan     }
293ce27a4eeSVijay Mahadevan     else {
294ce27a4eeSVijay Mahadevan       nvperelem = 8;
295ce27a4eeSVijay Mahadevan       merr = iface->get_element_connect(nelems, 8, moab::MBHEX, 0, starte, conn);MBERR("Can't get HEX8 element connectivity.", merr);
296ce27a4eeSVijay Mahadevan     }
297ce27a4eeSVijay Mahadevan     break;
298ce27a4eeSVijay Mahadevan   }
299ce27a4eeSVijay Mahadevan 
30063d025dbSVijay Mahadevan   ix = ie = 0; /* index now in the elements, for global ids */
301ce27a4eeSVijay Mahadevan 
302ce27a4eeSVijay Mahadevan   /* create a temporary range to store local element handles */
303ce27a4eeSVijay Mahadevan   moab::Range tmp(starte, starte + nelems - 1);
30449d66b22SVijay Mahadevan   std::vector<PetscInt> gids(nelems);
305ce27a4eeSVijay Mahadevan 
306ce27a4eeSVijay Mahadevan   /* identify the elements at the lower corner, for their global ids */
307ce27a4eeSVijay Mahadevan   xe = m * genCtx.A * genCtx.blockSizeElementXYZ[0] + a * genCtx.blockSizeElementXYZ[0];
308ce27a4eeSVijay Mahadevan   ye = (genCtx.dim > 1 ? n * genCtx.B * genCtx.blockSizeElementXYZ[1] + b * genCtx.blockSizeElementXYZ[1] : 0);
309ce27a4eeSVijay Mahadevan   ze = (genCtx.dim > 2 ? k * genCtx.C * genCtx.blockSizeElementXYZ[2] + c * genCtx.blockSizeElementXYZ[2] : 0);
310ce27a4eeSVijay Mahadevan 
311ce27a4eeSVijay Mahadevan   /* create owned elements requested by genCtx */
312ce27a4eeSVijay Mahadevan   for (kk = 0; kk < (genCtx.dim > 2 ? genCtx.blockSizeElementXYZ[2] : 1); kk++) {
313ce27a4eeSVijay Mahadevan     for (jj = 0; jj < (genCtx.dim > 1 ? genCtx.blockSizeElementXYZ[1] : 1); jj++) {
314ce27a4eeSVijay Mahadevan       for (ii = 0; ii < genCtx.blockSizeElementXYZ[0]; ii++) {
315ce27a4eeSVijay Mahadevan 
316ce27a4eeSVijay Mahadevan         moab::EntityHandle corner = startv + genCtx.q * ii + genCtx.q * jj * genCtx.ystride + genCtx.q * kk * genCtx.zstride;
317ce27a4eeSVijay Mahadevan 
31849d66b22SVijay Mahadevan         std::pair<PetscInt, PetscInt> entoffset = DMMoab_SetElementConnectivity_Private(genCtx, ix, corner, conn);
319ce27a4eeSVijay Mahadevan 
32049d66b22SVijay Mahadevan         for (PetscInt j = 0; j < entoffset.second; j++) {
321ce27a4eeSVijay Mahadevan           /* The entity handle for the particular element -> if we want to set some tags is
322ce27a4eeSVijay Mahadevan              moab::EntityHandle eh = starte + ie + j;
323ce27a4eeSVijay Mahadevan           */
324ce27a4eeSVijay Mahadevan           gids[ie + j] = 1 + ((xe + ii) + (ye + jj) * genCtx.nex + (ze + kk) * (genCtx.nex * genCtx.ney));
32563d025dbSVijay Mahadevan           /* gids[ie+j] = ie + j + ((xe + ii) + (ye + jj) * genCtx.nex + (ze + kk) * (genCtx.nex * genCtx.ney)); */
32663d025dbSVijay Mahadevan           /* gids[ie+j] = 1 + ie; */
32763d025dbSVijay Mahadevan           /* ie++; */
328ce27a4eeSVijay Mahadevan         }
329ce27a4eeSVijay Mahadevan 
330ce27a4eeSVijay Mahadevan         ix += entoffset.first;
33149d66b22SVijay Mahadevan         ie += entoffset.second;
332ce27a4eeSVijay Mahadevan       }
333ce27a4eeSVijay Mahadevan     }
334ce27a4eeSVijay Mahadevan   }
335ce27a4eeSVijay Mahadevan   if (genCtx.adjEnts) { /* we need to update adjacencies now, because some elements are new */
336ce27a4eeSVijay Mahadevan     merr = iface->update_adjacencies(starte, nelems, nvperelem, conn);MBERR("Can't update adjacencies", merr);
337ce27a4eeSVijay Mahadevan   }
338ce27a4eeSVijay Mahadevan   tmp.swap(cells);
339ce27a4eeSVijay Mahadevan   merr = mbImpl->tag_set_data(global_id_tag, cells, &gids[0]);MBERR("Can't set global ids to elements.", merr);
340ce27a4eeSVijay Mahadevan   PetscFunctionReturn(0);
341ce27a4eeSVijay Mahadevan }
342ce27a4eeSVijay Mahadevan 
343755f3dfbSVijay Mahadevan PetscErrorCode DMMBUtil_InitializeOptions(DMMoabMeshGeneratorCtx& genCtx, PetscInt dim, PetscBool simplex, PetscInt rank, PetscInt nprocs, const PetscReal* bounds, PetscInt nelems)
344ce27a4eeSVijay Mahadevan {
345ce27a4eeSVijay Mahadevan   PetscFunctionBegin;
346ce27a4eeSVijay Mahadevan   /* Initialize all genCtx data */
347ce27a4eeSVijay Mahadevan   genCtx.dim = dim;
348ce27a4eeSVijay Mahadevan   genCtx.simplex = simplex;
34949d66b22SVijay Mahadevan   genCtx.newMergeMethod = genCtx.keep_skins = genCtx.adjEnts = true;
350ce27a4eeSVijay Mahadevan   /* determine other global quantities for the mesh used for nodes increments */
351ce27a4eeSVijay Mahadevan   genCtx.q = 1;
352755f3dfbSVijay Mahadevan   genCtx.fraction = genCtx.remainder = genCtx.cumfraction = 0;
353ce27a4eeSVijay Mahadevan 
354ce27a4eeSVijay Mahadevan   if (!genCtx.usrxyzgrid) { /* not overridden by genCtx - assume nele equally and that genCtx wants a uniform cube mesh */
355755f3dfbSVijay Mahadevan 
356755f3dfbSVijay Mahadevan     genCtx.fraction = nelems / nprocs; /* partition only by the largest dimension */
357755f3dfbSVijay Mahadevan     genCtx.remainder = nelems % nprocs; /* remainder after partition which gets evenly distributed by round-robin */
358755f3dfbSVijay Mahadevan     genCtx.cumfraction = (rank > 0 ? (genCtx.fraction) * (rank) + (rank - 1 < genCtx.remainder ? rank : genCtx.remainder) : 0);
359755f3dfbSVijay Mahadevan     if (rank < genCtx.remainder)    /* This process gets "fraction+1" elements */
360755f3dfbSVijay Mahadevan       genCtx.fraction++;
361755f3dfbSVijay Mahadevan 
362*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscInfo(NULL, "Fraction = %" PetscInt_FMT ", Remainder = %" PetscInt_FMT ", Cumulative fraction = %" PetscInt_FMT "\n", genCtx.fraction, genCtx.remainder, genCtx.cumfraction));
363755f3dfbSVijay Mahadevan     switch (genCtx.dim) {
364755f3dfbSVijay Mahadevan     case 1:
365755f3dfbSVijay Mahadevan       genCtx.blockSizeElementXYZ[0] = genCtx.fraction;
366755f3dfbSVijay Mahadevan       genCtx.blockSizeElementXYZ[1] = 1;
367755f3dfbSVijay Mahadevan       genCtx.blockSizeElementXYZ[2] = 1;
368755f3dfbSVijay Mahadevan       break;
369755f3dfbSVijay Mahadevan     case 2:
370755f3dfbSVijay Mahadevan       genCtx.blockSizeElementXYZ[0] = nelems;
371755f3dfbSVijay Mahadevan       genCtx.blockSizeElementXYZ[1] = genCtx.fraction;
372755f3dfbSVijay Mahadevan       genCtx.blockSizeElementXYZ[2] = 1;
373755f3dfbSVijay Mahadevan       break;
374755f3dfbSVijay Mahadevan     case 3:
375755f3dfbSVijay Mahadevan     default:
376755f3dfbSVijay Mahadevan       genCtx.blockSizeElementXYZ[0] = nelems;
377755f3dfbSVijay Mahadevan       genCtx.blockSizeElementXYZ[1] = nelems;
378755f3dfbSVijay Mahadevan       genCtx.blockSizeElementXYZ[2] = genCtx.fraction;
379755f3dfbSVijay Mahadevan       break;
38049d66b22SVijay Mahadevan     }
381ce27a4eeSVijay Mahadevan   }
382ce27a4eeSVijay Mahadevan 
383ce27a4eeSVijay Mahadevan   /* partition only by the largest dimension */
384ce27a4eeSVijay Mahadevan   /* Total number of local elements := genCtx.blockSizeElementXYZ[0]*(genCtx.dim>1? genCtx.blockSizeElementXYZ[1]*(genCtx.dim>2 ? genCtx.blockSizeElementXYZ[2]:1) :1); */
385ce27a4eeSVijay Mahadevan   if (bounds) {
386*5f80ce2aSJacob Faibussowitsch     for (PetscInt i = 0; i < 6; i++) genCtx.xyzbounds[i] = bounds[i];
387ce27a4eeSVijay Mahadevan   }
388ce27a4eeSVijay Mahadevan   else {
389ce27a4eeSVijay Mahadevan     genCtx.xyzbounds[0] = genCtx.xyzbounds[2] = genCtx.xyzbounds[4] = 0.0;
390ce27a4eeSVijay Mahadevan     genCtx.xyzbounds[1] = genCtx.xyzbounds[3] = genCtx.xyzbounds[5] = 1.0;
391ce27a4eeSVijay Mahadevan   }
392ce27a4eeSVijay Mahadevan 
393ce27a4eeSVijay Mahadevan   if (!genCtx.usrprocgrid) {
394ce27a4eeSVijay Mahadevan     switch (genCtx.dim) {
395ce27a4eeSVijay Mahadevan     case 1:
396ce27a4eeSVijay Mahadevan       genCtx.M = nprocs;
397ce27a4eeSVijay Mahadevan       genCtx.N = genCtx.K = 1;
398ce27a4eeSVijay Mahadevan       break;
399ce27a4eeSVijay Mahadevan     case 2:
400755f3dfbSVijay Mahadevan       genCtx.N = nprocs;
40163d025dbSVijay Mahadevan       genCtx.M = genCtx.K = 1;
402ce27a4eeSVijay Mahadevan       break;
403ce27a4eeSVijay Mahadevan     default:
404755f3dfbSVijay Mahadevan       genCtx.K = nprocs;
40563d025dbSVijay Mahadevan       genCtx.M = genCtx.N = 1;
406ce27a4eeSVijay Mahadevan       break;
407ce27a4eeSVijay Mahadevan     }
408ce27a4eeSVijay Mahadevan   }
409ce27a4eeSVijay Mahadevan 
410ce27a4eeSVijay Mahadevan   if (!genCtx.usrrefgrid) {
411ce27a4eeSVijay Mahadevan     genCtx.A = genCtx.B = genCtx.C = 1;
412ce27a4eeSVijay Mahadevan   }
413ce27a4eeSVijay Mahadevan 
414ce27a4eeSVijay Mahadevan   /* more default values */
415ce27a4eeSVijay Mahadevan   genCtx.nex = genCtx.ney = genCtx.nez = 0;
416ce27a4eeSVijay Mahadevan   genCtx.xstride = genCtx.ystride = genCtx.zstride = 0;
417ce27a4eeSVijay Mahadevan   genCtx.NX = genCtx.NY = genCtx.NZ = 0;
418ce27a4eeSVijay Mahadevan   genCtx.nex = genCtx.ney = genCtx.nez = 0;
419ce27a4eeSVijay Mahadevan   genCtx.blockSizeVertexXYZ[0] = genCtx.blockSizeVertexXYZ[1] = genCtx.blockSizeVertexXYZ[2] = 1;
420ce27a4eeSVijay Mahadevan 
421ce27a4eeSVijay Mahadevan   switch (genCtx.dim) {
422ce27a4eeSVijay Mahadevan   case 3:
423ce27a4eeSVijay Mahadevan     genCtx.blockSizeVertexXYZ[0] = genCtx.q * genCtx.blockSizeElementXYZ[0] + 1;
424ce27a4eeSVijay Mahadevan     genCtx.blockSizeVertexXYZ[1] = genCtx.q * genCtx.blockSizeElementXYZ[1] + 1;
425ce27a4eeSVijay Mahadevan     genCtx.blockSizeVertexXYZ[2] = genCtx.q * genCtx.blockSizeElementXYZ[2] + 1;
426ce27a4eeSVijay Mahadevan 
427ce27a4eeSVijay Mahadevan     genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0];   /* number of elements in x direction, used for global id on element */
428755f3dfbSVijay Mahadevan     genCtx.dx = (genCtx.xyzbounds[1] - genCtx.xyzbounds[0]) / (nelems * genCtx.q); /* distance between 2 nodes in x direction */
429ce27a4eeSVijay Mahadevan     genCtx.NX = (genCtx.q * genCtx.nex + 1);
430ce27a4eeSVijay Mahadevan     genCtx.xstride = 1;
431ce27a4eeSVijay Mahadevan     genCtx.ney = genCtx.N * genCtx.B * genCtx.blockSizeElementXYZ[1];  /* number of elements in y direction  .... */
432755f3dfbSVijay Mahadevan     genCtx.dy = (genCtx.xyzbounds[3] - genCtx.xyzbounds[2]) / (nelems * genCtx.q); /* distance between 2 nodes in y direction */
433ce27a4eeSVijay Mahadevan     genCtx.NY = (genCtx.q * genCtx.ney + 1);
434ce27a4eeSVijay Mahadevan     genCtx.ystride = genCtx.blockSizeVertexXYZ[0];
435ce27a4eeSVijay Mahadevan     genCtx.nez = genCtx.K * genCtx.C * genCtx.blockSizeElementXYZ[2];  /* number of elements in z direction  .... */
436755f3dfbSVijay Mahadevan     genCtx.dz = (genCtx.xyzbounds[5] - genCtx.xyzbounds[4]) / (nelems * genCtx.q); /* distance between 2 nodes in z direction */
437ce27a4eeSVijay Mahadevan     genCtx.NZ = (genCtx.q * genCtx.nez + 1);
438ce27a4eeSVijay Mahadevan     genCtx.zstride = genCtx.blockSizeVertexXYZ[0] * genCtx.blockSizeVertexXYZ[1];
439ce27a4eeSVijay Mahadevan     break;
440ce27a4eeSVijay Mahadevan   case 2:
441ce27a4eeSVijay Mahadevan     genCtx.blockSizeVertexXYZ[0] = genCtx.q * genCtx.blockSizeElementXYZ[0] + 1;
442ce27a4eeSVijay Mahadevan     genCtx.blockSizeVertexXYZ[1] = genCtx.q * genCtx.blockSizeElementXYZ[1] + 1;
4439daf19fdSVijay Mahadevan     genCtx.blockSizeVertexXYZ[2] = 0;
444ce27a4eeSVijay Mahadevan 
445ce27a4eeSVijay Mahadevan     genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0];   /* number of elements in x direction, used for global id on element */
446ce27a4eeSVijay Mahadevan     genCtx.dx = (genCtx.xyzbounds[1] - genCtx.xyzbounds[0]) / (genCtx.nex * genCtx.q); /* distance between 2 nodes in x direction */
447ce27a4eeSVijay Mahadevan     genCtx.NX = (genCtx.q * genCtx.nex + 1);
448ce27a4eeSVijay Mahadevan     genCtx.xstride = 1;
449ce27a4eeSVijay Mahadevan     genCtx.ney = genCtx.N * genCtx.B * genCtx.blockSizeElementXYZ[1];  /* number of elements in y direction  .... */
450755f3dfbSVijay Mahadevan     genCtx.dy = (genCtx.xyzbounds[3] - genCtx.xyzbounds[2]) / (nelems * genCtx.q); /* distance between 2 nodes in y direction */
451ce27a4eeSVijay Mahadevan     genCtx.NY = (genCtx.q * genCtx.ney + 1);
452ce27a4eeSVijay Mahadevan     genCtx.ystride = genCtx.blockSizeVertexXYZ[0];
453ce27a4eeSVijay Mahadevan     break;
454ce27a4eeSVijay Mahadevan   case 1:
4559daf19fdSVijay Mahadevan     genCtx.blockSizeVertexXYZ[1] = genCtx.blockSizeVertexXYZ[2] = 0;
456ce27a4eeSVijay Mahadevan     genCtx.blockSizeVertexXYZ[0] = genCtx.q * genCtx.blockSizeElementXYZ[0] + 1;
457ce27a4eeSVijay Mahadevan 
458ce27a4eeSVijay Mahadevan     genCtx.nex = genCtx.M * genCtx.A * genCtx.blockSizeElementXYZ[0];   /* number of elements in x direction, used for global id on element */
459755f3dfbSVijay Mahadevan     genCtx.dx = (genCtx.xyzbounds[1] - genCtx.xyzbounds[0]) / (nelems * genCtx.q); /* distance between 2 nodes in x direction */
460ce27a4eeSVijay Mahadevan     genCtx.NX = (genCtx.q * genCtx.nex + 1);
461ce27a4eeSVijay Mahadevan     genCtx.xstride = 1;
462ce27a4eeSVijay Mahadevan     break;
463ce27a4eeSVijay Mahadevan   }
464ce27a4eeSVijay Mahadevan 
465755f3dfbSVijay Mahadevan   /* Lets check for some valid input */
466*5f80ce2aSJacob Faibussowitsch   PetscCheck(genCtx.dim >= 1 && genCtx.dim <= 3,PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Invalid topological dimension specified: %" PetscInt_FMT ".", genCtx.dim);
467*5f80ce2aSJacob Faibussowitsch   PetscCheck(genCtx.M * genCtx.N * genCtx.K == nprocs,PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Invalid [m, n, k] data: %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT ". Product must be equal to global size = %" PetscInt_FMT ".", genCtx.M, genCtx.N, genCtx.K, nprocs);
468755f3dfbSVijay Mahadevan   /* validate the bounds data */
469*5f80ce2aSJacob Faibussowitsch   PetscCheck(genCtx.xyzbounds[0] < genCtx.xyzbounds[1],PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "X-dim: Left boundary cannot be greater than right. [%G >= %G]", genCtx.xyzbounds[0], genCtx.xyzbounds[1]);
4702c71b3e2SJacob Faibussowitsch   PetscCheckFalse(genCtx.dim > 1 && (genCtx.xyzbounds[2] >= genCtx.xyzbounds[3]),PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Y-dim: Left boundary cannot be greater than right. [%G >= %G]", genCtx.xyzbounds[2], genCtx.xyzbounds[3]);
4712c71b3e2SJacob Faibussowitsch   PetscCheckFalse(genCtx.dim > 2 && (genCtx.xyzbounds[4] >= genCtx.xyzbounds[5]),PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Z-dim: Left boundary cannot be greater than right. [%G >= %G]", genCtx.xyzbounds[4], genCtx.xyzbounds[5]);
472755f3dfbSVijay Mahadevan 
473*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(NULL, "Local elements:= %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT "\n", genCtx.blockSizeElementXYZ[0], genCtx.blockSizeElementXYZ[1], genCtx.blockSizeElementXYZ[2]));
474*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(NULL, "Local vertices:= %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT "\n", genCtx.blockSizeVertexXYZ[0], genCtx.blockSizeVertexXYZ[1], genCtx.blockSizeVertexXYZ[2]));
475*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(NULL, "Local blocks/processors := %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT "\n", genCtx.A, genCtx.B, genCtx.C));
476*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(NULL, "Local processors := %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT "\n", genCtx.M, genCtx.N, genCtx.K));
477*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(NULL, "Local nexyz:= %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT "\n", genCtx.nex, genCtx.ney, genCtx.nez));
478*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(NULL, "Local delxyz:= %g, %g, %g\n", genCtx.dx, genCtx.dy, genCtx.dz));
479*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(NULL, "Local strides:= %" PetscInt_FMT ", %" PetscInt_FMT ", %" PetscInt_FMT "\n", genCtx.xstride, genCtx.ystride, genCtx.zstride));
480ce27a4eeSVijay Mahadevan   PetscFunctionReturn(0);
481ce27a4eeSVijay Mahadevan }
482ce27a4eeSVijay Mahadevan 
483cab5ea25SPierre Jolivet /*@C
484ce27a4eeSVijay Mahadevan   DMMoabCreateBoxMesh - Creates a mesh on the tensor product (box) of intervals with genCtx specified bounds.
485aa859218SVijay Mahadevan 
486d083f849SBarry Smith   Collective
487aa859218SVijay Mahadevan 
488aa859218SVijay Mahadevan   Input Parameters:
489aa859218SVijay Mahadevan + comm - The communicator for the DM object
490aa859218SVijay Mahadevan . dim - The spatial dimension
491b8ecf6d3SVijay 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
492b8ecf6d3SVijay Mahadevan . nele - The number of discrete elements in each direction
493a2b725a8SWilliam Gropp - user_nghost - The number of ghosted layers needed in the partitioned mesh
494aa859218SVijay Mahadevan 
495aa859218SVijay Mahadevan   Output Parameter:
496aa859218SVijay Mahadevan . dm  - The DM object
497aa859218SVijay Mahadevan 
498aa859218SVijay Mahadevan   Level: beginner
499aa859218SVijay Mahadevan 
500aa859218SVijay Mahadevan .seealso: DMSetType(), DMCreate(), DMMoabLoadFromFile()
501aa859218SVijay Mahadevan @*/
502ce27a4eeSVijay Mahadevan PetscErrorCode DMMoabCreateBoxMesh(MPI_Comm comm, PetscInt dim, PetscBool useSimplex, const PetscReal* bounds, PetscInt nele, PetscInt nghost, DM *dm)
50351d15aeeSVijay Mahadevan {
50451d15aeeSVijay Mahadevan   PetscErrorCode         ierr;
50551d15aeeSVijay Mahadevan   moab::ErrorCode        merr;
5069daf19fdSVijay Mahadevan   PetscInt               a, b, c, n, global_size, global_rank;
50751d15aeeSVijay Mahadevan   DM_Moab               *dmmoab;
508ce27a4eeSVijay Mahadevan   moab::Interface       *mbImpl;
5099daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
51051d15aeeSVijay Mahadevan   moab::ParallelComm    *pcomm;
5119daf19fdSVijay Mahadevan #endif
512a4717116SVijay Mahadevan   moab::ReadUtilIface   *readMeshIface;
513ce27a4eeSVijay Mahadevan   moab::Range            verts, cells, edges, faces, adj, dim3, dim2;
514ce27a4eeSVijay Mahadevan   DMMoabMeshGeneratorCtx genCtx;
515a4717116SVijay Mahadevan   const PetscInt         npts = nele + 1;    /* Number of points in every dimension */
516ce27a4eeSVijay Mahadevan 
5176ea892caSVijay Mahadevan   moab::Tag              global_id_tag, part_tag, geom_tag, mat_tag, dir_tag, neu_tag;
518ce27a4eeSVijay Mahadevan   moab::Range            ownedvtx, ownedelms, localvtxs, localelms;
519ce27a4eeSVijay Mahadevan   moab::EntityHandle     regionset;
520755f3dfbSVijay Mahadevan   PetscInt               ml = 0, nl = 0, kl = 0;
52151d15aeeSVijay Mahadevan 
52251d15aeeSVijay Mahadevan   PetscFunctionBegin;
5232c71b3e2SJacob Faibussowitsch   PetscCheckFalse(dim < 1 || dim > 3,PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Invalid dimension argument for mesh: dim=[1,3].");
524e5136372SVijay Mahadevan 
525*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogEventRegister("GenerateMesh", DM_CLASSID,   &genCtx.generateMesh));
526*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogEventRegister("AddVertices", DM_CLASSID,   &genCtx.generateVertices));
527*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogEventRegister("AddElements", DM_CLASSID,   &genCtx.generateElements));
528*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogEventRegister("ParResolve", DM_CLASSID,   &genCtx.parResolve));
529*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogEventBegin(genCtx.generateMesh, 0, 0, 0, 0));
530*5f80ce2aSJacob Faibussowitsch   CHKERRMPI(MPI_Comm_size(comm, &global_size));
531e5136372SVijay Mahadevan   /* total number of vertices in all dimensions */
532e5136372SVijay Mahadevan   n = pow(npts, dim);
533e5136372SVijay Mahadevan 
534e5136372SVijay Mahadevan   /* do some error checking */
5352c71b3e2SJacob Faibussowitsch   PetscCheckFalse(n < 2,PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Number of points must be >= 2.");
5362c71b3e2SJacob Faibussowitsch   PetscCheckFalse(global_size > n,PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Number of processors must be less than or equal to number of elements.");
5372c71b3e2SJacob Faibussowitsch   PetscCheckFalse(nghost < 0,PETSC_COMM_WORLD, PETSC_ERR_ARG_OUTOFRANGE, "Number of ghost layers cannot be negative.");
538e5136372SVijay Mahadevan 
539a4717116SVijay Mahadevan   /* Create the basic DMMoab object and keep the default parameters created by DM impls */
540*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMMoabCreateMoab(comm, NULL, NULL, NULL, dm));
54151d15aeeSVijay Mahadevan 
542a4717116SVijay Mahadevan   /* get all the necessary handles from the private DM object */
54351d15aeeSVijay Mahadevan   dmmoab = (DM_Moab*)(*dm)->data;
544ce27a4eeSVijay Mahadevan   mbImpl = dmmoab->mbiface;
5459daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
54651d15aeeSVijay Mahadevan   pcomm = dmmoab->pcomm;
547ce27a4eeSVijay Mahadevan   global_rank = pcomm->rank();
5489daf19fdSVijay Mahadevan #else
5499daf19fdSVijay Mahadevan   global_rank = 0;
5509daf19fdSVijay Mahadevan   global_size = 1;
5519daf19fdSVijay Mahadevan #endif
5529daf19fdSVijay Mahadevan   global_id_tag = dmmoab->ltog_tag;
553c8d5365dSVijay Mahadevan   dmmoab->dim = dim;
55449d66b22SVijay Mahadevan   dmmoab->nghostrings = nghost;
555304006b3SVijay Mahadevan   dmmoab->refct = 1;
55651d15aeeSVijay Mahadevan 
557b5410836SVijay Mahadevan   /* create a file set to associate all entities in current mesh */
558ce27a4eeSVijay Mahadevan   merr = mbImpl->create_meshset(moab::MESHSET_SET, dmmoab->fileset);MBERR("Creating file set failed", merr);
559b5410836SVijay Mahadevan 
560a4717116SVijay Mahadevan   /* No errors yet; proceed with building the mesh */
561ce27a4eeSVijay Mahadevan   merr = mbImpl->query_interface(readMeshIface);MBERRNM(merr);
56251d15aeeSVijay Mahadevan 
563755f3dfbSVijay Mahadevan   genCtx.M = genCtx.N = genCtx.K = 1;
564755f3dfbSVijay Mahadevan   genCtx.A = genCtx.B = genCtx.C = 1;
565da8b1a3eSBarry Smith   genCtx.blockSizeElementXYZ[0] = 0;
566da8b1a3eSBarry Smith   genCtx.blockSizeElementXYZ[1] = 0;
567da8b1a3eSBarry Smith   genCtx.blockSizeElementXYZ[2] = 0;
568755f3dfbSVijay Mahadevan 
569755f3dfbSVijay Mahadevan   ierr = PetscOptionsBegin(comm, "", "DMMoab Creation Options", "DMMOAB");CHKERRQ(ierr);
570ce27a4eeSVijay Mahadevan   /* Handle DMMoab spatial resolution */
571*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsInt("-dmb_grid_x", "Number of grid points in x direction", "DMMoabSetSizes", genCtx.blockSizeElementXYZ[0], &genCtx.blockSizeElementXYZ[0], &genCtx.usrxyzgrid));
572*5f80ce2aSJacob Faibussowitsch   if (dim > 1) CHKERRQ(PetscOptionsInt("-dmb_grid_y", "Number of grid points in y direction", "DMMoabSetSizes", genCtx.blockSizeElementXYZ[1], &genCtx.blockSizeElementXYZ[1], &genCtx.usrxyzgrid));
573*5f80ce2aSJacob Faibussowitsch   if (dim > 2) CHKERRQ(PetscOptionsInt("-dmb_grid_z", "Number of grid points in z direction", "DMMoabSetSizes", genCtx.blockSizeElementXYZ[2], &genCtx.blockSizeElementXYZ[2], &genCtx.usrxyzgrid));
574a4717116SVijay Mahadevan 
575ce27a4eeSVijay Mahadevan   /* Handle DMMoab parallel distibution */
576*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsInt("-dmb_processors_x", "Number of processors in x direction", "DMMoabSetNumProcs", genCtx.M, &genCtx.M, &genCtx.usrprocgrid));
577*5f80ce2aSJacob Faibussowitsch   if (dim > 1) CHKERRQ(PetscOptionsInt("-dmb_processors_y", "Number of processors in y direction", "DMMoabSetNumProcs", genCtx.N, &genCtx.N, &genCtx.usrprocgrid));
578*5f80ce2aSJacob Faibussowitsch   if (dim > 2) CHKERRQ(PetscOptionsInt("-dmb_processors_z", "Number of processors in z direction", "DMMoabSetNumProcs", genCtx.K, &genCtx.K, &genCtx.usrprocgrid));
57951d15aeeSVijay Mahadevan 
580ce27a4eeSVijay Mahadevan   /* Handle DMMoab block refinement */
581*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscOptionsInt("-dmb_refine_x", "Number of refinement blocks in x direction", "DMMoabSetRefinement", genCtx.A, &genCtx.A, &genCtx.usrrefgrid));
582*5f80ce2aSJacob Faibussowitsch   if (dim > 1) CHKERRQ(PetscOptionsInt("-dmb_refine_y", "Number of refinement blocks in y direction", "DMMoabSetRefinement", genCtx.B, &genCtx.B, &genCtx.usrrefgrid));
583*5f80ce2aSJacob Faibussowitsch   if (dim > 2) CHKERRQ(PetscOptionsInt("-dmb_refine_z", "Number of refinement blocks in z direction", "DMMoabSetRefinement", genCtx.C, &genCtx.C, &genCtx.usrrefgrid));
584*5f80ce2aSJacob Faibussowitsch   ierr = PetscOptionsEnd();CHKERRQ(ierr);
58551d15aeeSVijay Mahadevan 
586*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMMBUtil_InitializeOptions(genCtx, dim, useSimplex, global_rank, global_size, bounds, nele));
58751d15aeeSVijay Mahadevan 
5882c71b3e2SJacob Faibussowitsch   //PetscCheckFalse(nele<nprocs,PETSC_COMM_WORLD,PETSC_ERR_ARG_OUTOFRANGE,"The dimensional discretization size should be greater or equal to number of processors: %D < %D",nele,nprocs);
58964e1c140SVijay Mahadevan 
590ce27a4eeSVijay Mahadevan   if (genCtx.adjEnts) genCtx.keep_skins = true; /* do not delete anything - consumes more memory */
59166f88a78SVijay Mahadevan 
592ce27a4eeSVijay Mahadevan   /* determine m, n, k for processor rank */
593755f3dfbSVijay Mahadevan   ml = nl = kl = 0;
594755f3dfbSVijay Mahadevan   switch (genCtx.dim) {
595755f3dfbSVijay Mahadevan   case 1:
596755f3dfbSVijay Mahadevan     ml = (genCtx.cumfraction);
597755f3dfbSVijay Mahadevan     break;
598755f3dfbSVijay Mahadevan   case 2:
599755f3dfbSVijay Mahadevan     nl = (genCtx.cumfraction);
600755f3dfbSVijay Mahadevan     break;
601755f3dfbSVijay Mahadevan   default:
602755f3dfbSVijay Mahadevan     kl = (genCtx.cumfraction) / genCtx.q / genCtx.blockSizeElementXYZ[2] / genCtx.C; //genCtx.K
603755f3dfbSVijay Mahadevan     break;
604755f3dfbSVijay Mahadevan   }
605e5136372SVijay Mahadevan 
606ce27a4eeSVijay Mahadevan   /*
607ce27a4eeSVijay Mahadevan    * so there are a total of M * A * blockSizeElement elements in x direction (so M * A * blockSizeElement + 1 verts in x direction)
608ce27a4eeSVijay Mahadevan    * so there are a total of N * B * blockSizeElement elements in y direction (so N * B * blockSizeElement + 1 verts in y direction)
609ce27a4eeSVijay Mahadevan    * so there are a total of K * C * blockSizeElement elements in z direction (so K * C * blockSizeElement + 1 verts in z direction)
610c8d5365dSVijay Mahadevan 
611ce27a4eeSVijay Mahadevan    * there are ( M * A blockSizeElement)      *  ( N * B * blockSizeElement)      * (K * C * blockSizeElement)    hexas
612ce27a4eeSVijay Mahadevan    * there are ( M * A * blockSizeElement + 1) *  ( N * B * blockSizeElement + 1) * (K * C * blockSizeElement + 1) vertices
613ce27a4eeSVijay Mahadevan    * x is the first dimension that varies
614ce27a4eeSVijay Mahadevan    */
615e5136372SVijay Mahadevan 
616ce27a4eeSVijay Mahadevan   /* generate the block at (a, b, c); it will represent a partition , it will get a partition tag */
61749d66b22SVijay Mahadevan   PetscInt dum_id = -1;
6186ea892caSVijay Mahadevan   merr = mbImpl->tag_get_handle("GLOBAL_ID", 1, moab::MB_TYPE_INTEGER, global_id_tag);MBERR("Getting Global_ID Tag handle failed", merr);
61951d15aeeSVijay Mahadevan 
6206ea892caSVijay Mahadevan   merr = mbImpl->tag_get_handle(MATERIAL_SET_TAG_NAME, 1, moab::MB_TYPE_INTEGER, mat_tag);MBERR("Getting Material set Tag handle failed", merr);
6216ea892caSVijay Mahadevan   merr = mbImpl->tag_get_handle(DIRICHLET_SET_TAG_NAME, 1, moab::MB_TYPE_INTEGER, dir_tag);MBERR("Getting Dirichlet set Tag handle failed", merr);
6226ea892caSVijay Mahadevan   merr = mbImpl->tag_get_handle(NEUMANN_SET_TAG_NAME, 1, moab::MB_TYPE_INTEGER, neu_tag);MBERR("Getting Neumann set Tag handle failed", merr);
6236ea892caSVijay Mahadevan 
6246ea892caSVijay Mahadevan   merr = mbImpl->tag_get_handle("PARALLEL_PARTITION", 1, moab::MB_TYPE_INTEGER, part_tag, moab::MB_TAG_CREAT | moab::MB_TAG_SPARSE, &dum_id);MBERR("Getting Partition Tag handle failed", merr);
62551d15aeeSVijay Mahadevan 
6263a4aeca1SVijay Mahadevan   /* lets create some sets */
627ce27a4eeSVijay 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);
62849d66b22SVijay Mahadevan   merr = mbImpl->create_meshset(moab::MESHSET_SET, regionset);MBERRNM(merr);
629*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscLogEventEnd(genCtx.generateMesh, 0, 0, 0, 0));
6303a4aeca1SVijay Mahadevan 
631ce27a4eeSVijay Mahadevan   for (a = 0; a < (genCtx.dim > 0 ? genCtx.A : genCtx.A); a++) {
632ce27a4eeSVijay Mahadevan     for (b = 0; b < (genCtx.dim > 1 ? genCtx.B : 1); b++) {
633ce27a4eeSVijay Mahadevan       for (c = 0; c < (genCtx.dim > 2 ? genCtx.C : 1); c++) {
63449d66b22SVijay Mahadevan 
63549d66b22SVijay Mahadevan         moab::EntityHandle startv;
63649d66b22SVijay Mahadevan 
637*5f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscLogEventBegin(genCtx.generateVertices, 0, 0, 0, 0));
638*5f80ce2aSJacob Faibussowitsch         CHKERRQ(DMMoab_GenerateVertices_Private(mbImpl, readMeshIface, genCtx, ml, nl, kl, a, b, c, global_id_tag, startv, verts));
639*5f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscLogEventEnd(genCtx.generateVertices, 0, 0, 0, 0));
6403a4aeca1SVijay Mahadevan 
641*5f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscLogEventBegin(genCtx.generateElements, 0, 0, 0, 0));
642*5f80ce2aSJacob Faibussowitsch         CHKERRQ(DMMoab_GenerateElements_Private(mbImpl, readMeshIface, genCtx, ml, nl, kl, a, b, c, global_id_tag, startv, cells));
643*5f80ce2aSJacob Faibussowitsch         CHKERRQ(PetscLogEventEnd(genCtx.generateElements, 0, 0, 0, 0));
6443a4aeca1SVijay Mahadevan 
64549d66b22SVijay Mahadevan         PetscInt part_num = 0;
64649d66b22SVijay Mahadevan         switch (genCtx.dim) {
64749d66b22SVijay Mahadevan         case 3:
64849d66b22SVijay Mahadevan           part_num += (c + kl * genCtx.C) * (genCtx.M * genCtx.A * genCtx.N * genCtx.B);
64949d66b22SVijay Mahadevan         case 2:
65049d66b22SVijay Mahadevan           part_num += (b + nl * genCtx.B) * (genCtx.M * genCtx.A);
65149d66b22SVijay Mahadevan         case 1:
65249d66b22SVijay Mahadevan           part_num += (a + ml * genCtx.A);
65349d66b22SVijay Mahadevan           break;
65449d66b22SVijay Mahadevan         }
65549d66b22SVijay Mahadevan 
65649d66b22SVijay Mahadevan         moab::EntityHandle part_set;
657ce27a4eeSVijay Mahadevan         merr = mbImpl->create_meshset(moab::MESHSET_SET, part_set);MBERR("Can't create mesh set.", merr);
6583a4aeca1SVijay Mahadevan 
659ce27a4eeSVijay Mahadevan         merr = mbImpl->add_entities(part_set, verts);MBERR("Can't add vertices to set.", merr);
66049d66b22SVijay Mahadevan         merr = mbImpl->add_entities(part_set, cells);MBERR("Can't add entities to set.", merr);
661ce27a4eeSVijay Mahadevan         merr = mbImpl->add_entities(regionset, cells);MBERR("Can't add entities to set.", merr);
662ce27a4eeSVijay Mahadevan 
663ce27a4eeSVijay Mahadevan         /* if needed, add all edges and faces */
664ce27a4eeSVijay Mahadevan         if (genCtx.adjEnts)
665ce27a4eeSVijay Mahadevan         {
666ce27a4eeSVijay Mahadevan           if (genCtx.dim > 1) {
667ce27a4eeSVijay Mahadevan             merr = mbImpl->get_adjacencies(cells, 1, true, edges, moab::Interface::UNION);MBERR("Can't get edges", merr);
668ce27a4eeSVijay Mahadevan             merr = mbImpl->add_entities(part_set, edges);MBERR("Can't add edges to partition set.", merr);
669ce27a4eeSVijay Mahadevan           }
670ce27a4eeSVijay Mahadevan           if (genCtx.dim > 2) {
671ce27a4eeSVijay Mahadevan             merr = mbImpl->get_adjacencies(cells, 2, true, faces, moab::Interface::UNION);MBERR("Can't get faces", merr);
672ce27a4eeSVijay Mahadevan             merr = mbImpl->add_entities(part_set, faces);MBERR("Can't add faces to partition set.", merr);
673ce27a4eeSVijay Mahadevan           }
674ce27a4eeSVijay Mahadevan           edges.clear();
675ce27a4eeSVijay Mahadevan           faces.clear();
676ce27a4eeSVijay Mahadevan         }
67749d66b22SVijay Mahadevan         verts.clear(); cells.clear();
678ce27a4eeSVijay Mahadevan 
679ce27a4eeSVijay Mahadevan         merr = mbImpl->tag_set_data(part_tag, &part_set, 1, &part_num);MBERR("Can't set part tag on set", merr);
68049d66b22SVijay Mahadevan         if (dmmoab->fileset) {
681ce27a4eeSVijay Mahadevan           merr = mbImpl->add_parent_child(dmmoab->fileset, part_set);MBERR("Can't add part set to file set.", merr);
682ce27a4eeSVijay Mahadevan           merr = mbImpl->unite_meshset(dmmoab->fileset, part_set);MBERRNM(merr);
68349d66b22SVijay Mahadevan         }
68449d66b22SVijay Mahadevan         merr = mbImpl->add_entities(dmmoab->fileset, &part_set, 1);MBERRNM(merr);
685ce27a4eeSVijay Mahadevan       }
686ce27a4eeSVijay Mahadevan     }
687ce27a4eeSVijay Mahadevan   }
688ce27a4eeSVijay Mahadevan 
68949d66b22SVijay Mahadevan   merr = mbImpl->add_parent_child(dmmoab->fileset, regionset);MBERRNM(merr);
690ce27a4eeSVijay Mahadevan 
69149d66b22SVijay Mahadevan   /* Only in parallel: resolve shared entities between processors and exchange ghost layers */
692ce27a4eeSVijay Mahadevan   if (global_size > 1) {
693ce27a4eeSVijay Mahadevan 
694*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogEventBegin(genCtx.parResolve, 0, 0, 0, 0));
69577a8c067SVijay Mahadevan 
696ce27a4eeSVijay Mahadevan     merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, genCtx.dim, cells);MBERR("Can't get all d-dimensional elements.", merr);
697ce27a4eeSVijay Mahadevan     merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 0, verts);MBERR("Can't get all vertices.", merr);
698ce27a4eeSVijay Mahadevan 
699ce27a4eeSVijay Mahadevan     if (genCtx.A * genCtx.B * genCtx.C != 1) { //  merge needed
70049d66b22SVijay Mahadevan       moab::MergeMesh mm(mbImpl);
701ce27a4eeSVijay Mahadevan       if (genCtx.newMergeMethod) {
702ce27a4eeSVijay Mahadevan         merr = mm.merge_using_integer_tag(verts, global_id_tag);MBERR("Can't merge with GLOBAL_ID tag", merr);
703ce27a4eeSVijay Mahadevan       }
704ce27a4eeSVijay Mahadevan       else {
705ce27a4eeSVijay Mahadevan         merr = mm.merge_entities(cells, 0.0001);MBERR("Can't merge with coordinates", merr);
7063a4aeca1SVijay Mahadevan       }
7073a4aeca1SVijay Mahadevan     }
7083a4aeca1SVijay Mahadevan 
7099daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
710a4717116SVijay Mahadevan     /* check the handles */
711ce27a4eeSVijay Mahadevan     merr = pcomm->check_all_shared_handles();MBERRV(mbImpl, merr);
71251d15aeeSVijay Mahadevan 
713a4717116SVijay Mahadevan     /* resolve the shared entities by exchanging information to adjacent processors */
714ce27a4eeSVijay Mahadevan     merr = pcomm->resolve_shared_ents(dmmoab->fileset, cells, dim, dim - 1, NULL, &global_id_tag);MBERRV(mbImpl, merr);
71549d66b22SVijay Mahadevan     if (dmmoab->fileset) {
716ce27a4eeSVijay Mahadevan       merr = pcomm->exchange_ghost_cells(dim, 0, nghost, dim, true, false, &dmmoab->fileset);MBERRV(mbImpl, merr);
71749d66b22SVijay Mahadevan     }
71849d66b22SVijay Mahadevan     else {
71949d66b22SVijay Mahadevan       merr = pcomm->exchange_ghost_cells(dim, 0, nghost, dim, true, false);MBERRV(mbImpl, merr);
72049d66b22SVijay Mahadevan     }
721c8d5365dSVijay Mahadevan 
722e427d9c9SVijay Mahadevan     /* Reassign global IDs on all entities. */
723e427d9c9SVijay Mahadevan     merr = pcomm->assign_global_ids(dmmoab->fileset, dim, 1, false, true, false);MBERRNM(merr);
7249daf19fdSVijay Mahadevan #endif
72577a8c067SVijay Mahadevan 
726*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscLogEventEnd(genCtx.parResolve, 0, 0, 0, 0));
72749d66b22SVijay Mahadevan   }
7283f1c6e43SVijay Mahadevan 
729ce27a4eeSVijay Mahadevan   if (!genCtx.keep_skins) { // default is to delete the 1- and 2-dimensional entities
730ce27a4eeSVijay Mahadevan     // delete all quads and edges
731ce27a4eeSVijay Mahadevan     moab::Range toDelete;
732ce27a4eeSVijay Mahadevan     if (genCtx.dim > 1) {
733ce27a4eeSVijay Mahadevan       merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 1, toDelete);MBERR("Can't get edges", merr);
734ce27a4eeSVijay Mahadevan     }
7353f1c6e43SVijay Mahadevan 
736ce27a4eeSVijay Mahadevan     if (genCtx.dim > 2) {
737ce27a4eeSVijay Mahadevan       merr = mbImpl->get_entities_by_dimension(dmmoab->fileset, 2, toDelete);MBERR("Can't get faces", merr);
738ce27a4eeSVijay Mahadevan     }
739c8d5365dSVijay Mahadevan 
7409daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
741ce27a4eeSVijay Mahadevan     merr = dmmoab->pcomm->delete_entities(toDelete) ;MBERR("Can't delete entities", merr);
7429daf19fdSVijay Mahadevan #endif
743ce27a4eeSVijay Mahadevan   }
7446ea892caSVijay Mahadevan 
7456ea892caSVijay Mahadevan   /* set geometric dimension tag for regions */
7466ea892caSVijay Mahadevan   merr = mbImpl->tag_set_data(geom_tag, &regionset, 1, &dmmoab->dim);MBERRNM(merr);
7476ea892caSVijay Mahadevan   /* set default material ID for regions */
7486ea892caSVijay Mahadevan   int default_material = 1;
7496ea892caSVijay Mahadevan   merr = mbImpl->tag_set_data(mat_tag, &regionset, 1, &default_material);MBERRNM(merr);
7506ea892caSVijay Mahadevan   /*
7516ea892caSVijay Mahadevan     int default_dbc = 0;
7526ea892caSVijay Mahadevan     merr = mbImpl->tag_set_data(dir_tag, &vertexset, 1, &default_dbc);MBERRNM(merr);
7536ea892caSVijay Mahadevan   */
75451d15aeeSVijay Mahadevan   PetscFunctionReturn(0);
75551d15aeeSVijay Mahadevan }
75651d15aeeSVijay Mahadevan 
75749d66b22SVijay Mahadevan PetscErrorCode DMMoab_GetReadOptions_Private(PetscBool by_rank, PetscInt numproc, PetscInt dim, PetscInt nghost, MoabReadMode mode, PetscInt dbglevel, const char* dm_opts, const char* extra_opts, const char** read_opts)
75851d15aeeSVijay Mahadevan {
7596acfe860SVijay Mahadevan   char *ropts;
76049d66b22SVijay Mahadevan   char  ropts_par[PETSC_MAX_PATH_LEN], ropts_pargh[PETSC_MAX_PATH_LEN];
76161eb6e27SVijay Mahadevan   char  ropts_dbg[PETSC_MAX_PATH_LEN];
76251d15aeeSVijay Mahadevan 
763a4717116SVijay Mahadevan   PetscFunctionBegin;
764*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMalloc1(PETSC_MAX_PATH_LEN, &ropts));
765*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemzero(&ropts_par, PETSC_MAX_PATH_LEN));
766*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemzero(&ropts_pargh, PETSC_MAX_PATH_LEN));
767*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscMemzero(&ropts_dbg, PETSC_MAX_PATH_LEN));
76861eb6e27SVijay Mahadevan 
769e23c60ebSVijay Mahadevan   /* do parallel read unless using only one processor */
770a4717116SVijay Mahadevan   if (numproc > 1) {
771*5f80ce2aSJacob Faibussowitsch     // CHKERRQ(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":"")));
772*5f80ce2aSJacob Faibussowitsch     CHKERRQ(PetscSNPrintf(ropts_par, PETSC_MAX_PATH_LEN, "PARALLEL=%s;PARTITION=PARALLEL_PARTITION;PARTITION_DISTRIBUTE;PARALLEL_RESOLVE_SHARED_ENTS;%s", MoabReadModes[mode], (by_rank ? "PARTITION_BY_RANK;" : "")));
77349d66b22SVijay Mahadevan     if (nghost) {
774*5f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscSNPrintf(ropts_pargh, PETSC_MAX_PATH_LEN, "PARALLEL_GHOSTS=%" PetscInt_FMT ".0.%" PetscInt_FMT ";", dim, nghost));
77549d66b22SVijay Mahadevan     }
7762e4e7c01SVijay Mahadevan   }
7772e4e7c01SVijay Mahadevan 
778c8d5365dSVijay Mahadevan   if (dbglevel) {
779f90c3b0eSVijay Mahadevan     if (numproc > 1) {
780*5f80ce2aSJacob Faibussowitsch       CHKERRQ(PetscSNPrintf(ropts_dbg, PETSC_MAX_PATH_LEN, "CPUTIME;DEBUG_IO=%" PetscInt_FMT ";DEBUG_PIO=%" PetscInt_FMT ";", dbglevel, dbglevel));
78161eb6e27SVijay Mahadevan     }
782*5f80ce2aSJacob Faibussowitsch     else CHKERRQ(PetscSNPrintf(ropts_dbg, PETSC_MAX_PATH_LEN, "CPUTIME;DEBUG_IO=%" PetscInt_FMT ";", dbglevel));
783c8d5365dSVijay Mahadevan   }
78451d15aeeSVijay Mahadevan 
785*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscSNPrintf(ropts, PETSC_MAX_PATH_LEN, "%s%s%s%s%s", ropts_par, (nghost ? ropts_pargh : ""), ropts_dbg, (extra_opts ? extra_opts : ""), (dm_opts ? dm_opts : "")));
786f90c3b0eSVijay Mahadevan   *read_opts = ropts;
78751d15aeeSVijay Mahadevan   PetscFunctionReturn(0);
78851d15aeeSVijay Mahadevan }
78951d15aeeSVijay Mahadevan 
790cab5ea25SPierre Jolivet /*@C
791aa859218SVijay Mahadevan   DMMoabLoadFromFile - Creates a DM object by loading the mesh from a user specified file.
792aa859218SVijay Mahadevan 
793d083f849SBarry Smith   Collective
794aa859218SVijay Mahadevan 
795aa859218SVijay Mahadevan   Input Parameters:
796aa859218SVijay Mahadevan + comm - The communicator for the DM object
797aa859218SVijay Mahadevan . dim - The spatial dimension
798b8ecf6d3SVijay Mahadevan . filename - The name of the mesh file to be loaded
799a2b725a8SWilliam Gropp - usrreadopts - The options string to read a MOAB mesh.
800a2b725a8SWilliam Gropp 
801a8d69d7bSBarry Smith   Reference (Parallel Mesh Initialization: https://www.mcs.anl.gov/~fathom/moab-docs/html/contents.html#fivetwo)
802aa859218SVijay Mahadevan 
803aa859218SVijay Mahadevan   Output Parameter:
804aa859218SVijay Mahadevan . dm  - The DM object
805aa859218SVijay Mahadevan 
806aa859218SVijay Mahadevan   Level: beginner
807aa859218SVijay Mahadevan 
808aa859218SVijay Mahadevan .seealso: DMSetType(), DMCreate(), DMMoabCreateBoxMesh()
809aa859218SVijay Mahadevan @*/
81049d66b22SVijay Mahadevan PetscErrorCode DMMoabLoadFromFile(MPI_Comm comm, PetscInt dim, PetscInt nghost, const char* filename, const char* usrreadopts, DM *dm)
81151d15aeeSVijay Mahadevan {
812a4717116SVijay Mahadevan   moab::ErrorCode     merr;
8132e4e7c01SVijay Mahadevan   PetscInt            nprocs;
814a4717116SVijay Mahadevan   DM_Moab            *dmmoab;
815a4717116SVijay Mahadevan   moab::Interface    *mbiface;
8169daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
817a4717116SVijay Mahadevan   moab::ParallelComm *pcomm;
8189daf19fdSVijay Mahadevan #endif
819a4717116SVijay Mahadevan   moab::Range         verts, elems;
8207023aa44SVijay Mahadevan   const char         *readopts;
82151d15aeeSVijay Mahadevan 
82251d15aeeSVijay Mahadevan   PetscFunctionBegin;
82349d66b22SVijay Mahadevan   PetscValidPointer(dm, 6);
82451d15aeeSVijay Mahadevan 
825a4717116SVijay Mahadevan   /* Create the basic DMMoab object and keep the default parameters created by DM impls */
826*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMMoabCreateMoab(comm, NULL, NULL, NULL, dm));
82751d15aeeSVijay Mahadevan 
828a4717116SVijay Mahadevan   /* get all the necessary handles from the private DM object */
829a4717116SVijay Mahadevan   dmmoab = (DM_Moab*)(*dm)->data;
830a4717116SVijay Mahadevan   mbiface = dmmoab->mbiface;
8319daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
832a4717116SVijay Mahadevan   pcomm = dmmoab->pcomm;
833a4717116SVijay Mahadevan   nprocs = pcomm->size();
8349daf19fdSVijay Mahadevan #else
8359daf19fdSVijay Mahadevan   nprocs = 1;
8369daf19fdSVijay Mahadevan #endif
837aa859218SVijay Mahadevan   /* TODO: Decipher dimension based on the loaded mesh instead of getting from user */
838c8d5365dSVijay Mahadevan   dmmoab->dim = dim;
83949d66b22SVijay Mahadevan   dmmoab->nghostrings = nghost;
840304006b3SVijay Mahadevan   dmmoab->refct = 1;
841a4717116SVijay Mahadevan 
842b5410836SVijay Mahadevan   /* create a file set to associate all entities in current mesh */
843b5410836SVijay Mahadevan   merr = dmmoab->mbiface->create_meshset(moab::MESHSET_SET, dmmoab->fileset);MBERR("Creating file set failed", merr);
844b5410836SVijay Mahadevan 
845a4717116SVijay Mahadevan   /* add mesh loading options specific to the DM */
846*5f80ce2aSJacob Faibussowitsch   CHKERRQ(DMMoab_GetReadOptions_Private(dmmoab->partition_by_rank, nprocs, dim, nghost, dmmoab->read_mode,
847*5f80ce2aSJacob Faibussowitsch                                         dmmoab->rw_dbglevel, dmmoab->extra_read_options, usrreadopts, &readopts));
8487023aa44SVijay Mahadevan 
849*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(*dm, "Reading file %s with options: %s\n", filename, readopts));
850a4717116SVijay Mahadevan 
851a4717116SVijay Mahadevan   /* Load the mesh from a file. */
85249d66b22SVijay Mahadevan   if (dmmoab->fileset) {
8537023aa44SVijay Mahadevan     merr = mbiface->load_file(filename, &dmmoab->fileset, readopts);MBERRVM(mbiface, "Reading MOAB file failed.", merr);
85449d66b22SVijay Mahadevan   }
85549d66b22SVijay Mahadevan   else {
85649d66b22SVijay Mahadevan     merr = mbiface->load_file(filename, 0, readopts);MBERRVM(mbiface, "Reading MOAB file failed.", merr);
85749d66b22SVijay Mahadevan   }
858a4717116SVijay Mahadevan 
8599daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
8606e40195eSVijay Mahadevan   /* Reassign global IDs on all entities. */
8616ea892caSVijay Mahadevan   /* merr = pcomm->assign_global_ids(dmmoab->fileset, dim, 1, true, true, true);MBERRNM(merr); */
8629daf19fdSVijay Mahadevan #endif
863e5136372SVijay Mahadevan 
864e5136372SVijay Mahadevan   /* load the local vertices */
865e5136372SVijay Mahadevan   merr = mbiface->get_entities_by_type(dmmoab->fileset, moab::MBVERTEX, verts, true);MBERRNM(merr);
866e5136372SVijay Mahadevan   /* load the local elements */
867e5136372SVijay Mahadevan   merr = mbiface->get_entities_by_dimension(dmmoab->fileset, dim, elems, true);MBERRNM(merr);
868e5136372SVijay Mahadevan 
8699daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
870e5136372SVijay Mahadevan   /* Everything is set up, now just do a tag exchange to update tags
871e5136372SVijay Mahadevan      on all of the ghost vertexes */
872e5136372SVijay Mahadevan   merr = pcomm->exchange_tags(dmmoab->ltog_tag, verts);MBERRV(mbiface, merr);
873e5136372SVijay Mahadevan   merr = pcomm->exchange_tags(dmmoab->ltog_tag, elems);MBERRV(mbiface, merr);
874a4717116SVijay Mahadevan   merr = pcomm->collective_sync_partition();MBERR("Collective sync failed", merr);
8759daf19fdSVijay Mahadevan #endif
876a4717116SVijay Mahadevan 
877*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscInfo(*dm, "MOAB file '%s' was successfully loaded. Found %D vertices and %D elements.\n", filename, verts.size(), elems.size()));
878*5f80ce2aSJacob Faibussowitsch   CHKERRQ(PetscFree(readopts));
87951d15aeeSVijay Mahadevan   PetscFunctionReturn(0);
88051d15aeeSVijay Mahadevan }
88151d15aeeSVijay Mahadevan 
882cab5ea25SPierre Jolivet /*@C
883304006b3SVijay Mahadevan   DMMoabRenumberMeshEntities - Order and number all entities (vertices->elements) to be contiguously ordered
884304006b3SVijay Mahadevan   in parallel
885304006b3SVijay Mahadevan 
886d083f849SBarry Smith   Collective
887304006b3SVijay Mahadevan 
888304006b3SVijay Mahadevan   Input Parameters:
889a2b725a8SWilliam Gropp . dm  - The DM object
890304006b3SVijay Mahadevan 
891304006b3SVijay Mahadevan   Level: advanced
892304006b3SVijay Mahadevan 
893304006b3SVijay Mahadevan .seealso: DMSetUp(), DMCreate()
894304006b3SVijay Mahadevan @*/
895304006b3SVijay Mahadevan PetscErrorCode DMMoabRenumberMeshEntities(DM dm)
896304006b3SVijay Mahadevan {
897304006b3SVijay Mahadevan   moab::Range         verts;
898304006b3SVijay Mahadevan 
899304006b3SVijay Mahadevan   PetscFunctionBegin;
900304006b3SVijay Mahadevan   PetscValidHeaderSpecific(dm, DM_CLASSID, 1);
901304006b3SVijay Mahadevan 
9029daf19fdSVijay Mahadevan #ifdef MOAB_HAVE_MPI
903304006b3SVijay Mahadevan   /* Insert new points */
9049daf19fdSVijay Mahadevan   moab::ErrorCode     merr;
9059daf19fdSVijay Mahadevan   merr = ((DM_Moab*) dm->data)->pcomm->assign_global_ids(((DM_Moab*) dm->data)->fileset, 3, 0, false, true, false);MBERRNM(merr);
9069daf19fdSVijay Mahadevan #endif
907304006b3SVijay Mahadevan   PetscFunctionReturn(0);
908304006b3SVijay Mahadevan }
909