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