xref: /petsc/src/dm/impls/plex/plexegads.c (revision 9566063d113dddea24716c546802770db7481bc0)
1a8ededdfSMatthew G. Knepley #include <petsc/private/dmpleximpl.h>   /*I      "petscdmplex.h"   I*/
27bee2925SMatthew Knepley #include <petsc/private/hashmapi.h>
3a8ededdfSMatthew G. Knepley 
4a8ededdfSMatthew G. Knepley #ifdef PETSC_HAVE_EGADS
5a8ededdfSMatthew G. Knepley #include <egads.h>
6a8ededdfSMatthew G. Knepley #endif
7a8ededdfSMatthew G. Knepley 
8a8ededdfSMatthew G. Knepley /* We need to understand how to natively parse STEP files. There seems to be only one open source implementation of
9a8ededdfSMatthew G. Knepley    the STEP parser contained in the OpenCASCADE package. It is enough to make a strong man weep:
10a8ededdfSMatthew G. Knepley 
11a8ededdfSMatthew G. Knepley      https://github.com/tpaviot/oce/tree/master/src/STEPControl
12a8ededdfSMatthew G. Knepley 
13a8ededdfSMatthew G. Knepley    The STEP, and inner EXPRESS, formats are ISO standards, so they are documented
14a8ededdfSMatthew G. Knepley 
15a8ededdfSMatthew G. Knepley      https://stackoverflow.com/questions/26774037/documentation-or-specification-for-step-and-stp-files
16a8ededdfSMatthew G. Knepley      http://stepmod.sourceforge.net/express_model_spec/
17a8ededdfSMatthew G. Knepley 
18a8ededdfSMatthew G. Knepley    but again it seems that there has been a deliberate effort at obfuscation, probably to raise the bar for entrants.
19a8ededdfSMatthew G. Knepley */
20a8ededdfSMatthew G. Knepley 
21c1cad2e7SMatthew G. Knepley #ifdef PETSC_HAVE_EGADS
22c1cad2e7SMatthew G. Knepley PETSC_INTERN PetscErrorCode DMPlexSnapToGeomModel_EGADS_Internal(DM, PetscInt, ego, PetscInt, PetscInt, PetscInt, const PetscScalar[], PetscScalar[]);
23c1cad2e7SMatthew G. Knepley PETSC_INTERN PetscErrorCode DMPlexSnapToGeomModel_EGADSLite_Internal(DM, PetscInt, ego, PetscInt, PetscInt, PetscInt, const PetscScalar[], PetscScalar[]);
24c1cad2e7SMatthew G. Knepley 
25c1cad2e7SMatthew G. Knepley PetscErrorCode DMPlexSnapToGeomModel_EGADS_Internal(DM dm, PetscInt p, ego model, PetscInt bodyID, PetscInt faceID, PetscInt edgeID, const PetscScalar mcoords[], PetscScalar gcoords[])
26c1cad2e7SMatthew G. Knepley {
27c1cad2e7SMatthew G. Knepley   DM             cdm;
28c1cad2e7SMatthew G. Knepley   ego           *bodies;
29c1cad2e7SMatthew G. Knepley   ego            geom, body, obj;
30a5b23f4aSJose E. Roman   /* result has to hold derivatives, along with the value */
31c1cad2e7SMatthew G. Knepley   double         params[3], result[18], paramsV[16*3], resultV[16*3], range[4];
32c1cad2e7SMatthew G. Knepley   int            Nb, oclass, mtype, *senses, peri;
33c1cad2e7SMatthew G. Knepley   Vec            coordinatesLocal;
34c1cad2e7SMatthew G. Knepley   PetscScalar   *coords = NULL;
35c1cad2e7SMatthew G. Knepley   PetscInt       Nv, v, Np = 0, pm;
36c1cad2e7SMatthew G. Knepley   PetscInt       dE, d;
37c1cad2e7SMatthew G. Knepley 
38c1cad2e7SMatthew G. Knepley   PetscFunctionBeginHot;
39*9566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dm, &cdm));
40*9566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &dE));
41*9566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dm, &coordinatesLocal));
42*9566063dSJacob Faibussowitsch   PetscCall(EG_getTopology(model, &geom, &oclass, &mtype, NULL, &Nb, &bodies, &senses));
432c71b3e2SJacob Faibussowitsch   PetscCheckFalse(bodyID >= Nb,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Body %D is not in [0, %d)", bodyID, Nb);
44c1cad2e7SMatthew G. Knepley   body = bodies[bodyID];
45c1cad2e7SMatthew G. Knepley 
46*9566063dSJacob Faibussowitsch   if (edgeID >= 0)      {PetscCall(EG_objectBodyTopo(body, EDGE, edgeID, &obj)); Np = 1;}
47*9566063dSJacob Faibussowitsch   else if (faceID >= 0) {PetscCall(EG_objectBodyTopo(body, FACE, faceID, &obj)); Np = 2;}
48c1cad2e7SMatthew G. Knepley   else {
49c1cad2e7SMatthew G. Knepley     for (d = 0; d < dE; ++d) gcoords[d] = mcoords[d];
50c1cad2e7SMatthew G. Knepley     PetscFunctionReturn(0);
51c1cad2e7SMatthew G. Knepley   }
52c1cad2e7SMatthew G. Knepley 
53c1cad2e7SMatthew G. Knepley   /* Calculate parameters (t or u,v) for vertices */
54*9566063dSJacob Faibussowitsch   PetscCall(DMPlexVecGetClosure(cdm, NULL, coordinatesLocal, p, &Nv, &coords));
55c1cad2e7SMatthew G. Knepley   Nv  /= dE;
56c1cad2e7SMatthew G. Knepley   if (Nv == 1) {
57*9566063dSJacob Faibussowitsch     PetscCall(DMPlexVecRestoreClosure(cdm, NULL, coordinatesLocal, p, &Nv, &coords));
58c1cad2e7SMatthew G. Knepley     for (d = 0; d < dE; ++d) gcoords[d] = mcoords[d];
59c1cad2e7SMatthew G. Knepley     PetscFunctionReturn(0);
60c1cad2e7SMatthew G. Knepley   }
612c71b3e2SJacob Faibussowitsch   PetscCheckFalse(Nv > 16,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot handle %D coordinates associated to point %D", Nv, p);
62c1cad2e7SMatthew G. Knepley 
63c1cad2e7SMatthew G. Knepley   /* Correct EGADSlite 2pi bug when calculating nearest point on Periodic Surfaces */
64*9566063dSJacob Faibussowitsch   PetscCall(EG_getRange(obj, range, &peri));
65c1cad2e7SMatthew G. Knepley   for (v = 0; v < Nv; ++v) {
66*9566063dSJacob Faibussowitsch     PetscCall(EG_invEvaluate(obj, &coords[v*dE], &paramsV[v*3], &resultV[v*3]));
67c1cad2e7SMatthew G. Knepley #if 1
68c1cad2e7SMatthew G. Knepley     if (peri > 0) {
69c1cad2e7SMatthew G. Knepley       if      (paramsV[v*3+0] + 1.e-4 < range[0]) {paramsV[v*3+0] += 2. * PETSC_PI;}
70c1cad2e7SMatthew G. Knepley       else if (paramsV[v*3+0] - 1.e-4 > range[1]) {paramsV[v*3+0] -= 2. * PETSC_PI;}
71c1cad2e7SMatthew G. Knepley     }
72c1cad2e7SMatthew G. Knepley     if (peri > 1) {
73c1cad2e7SMatthew G. Knepley       if      (paramsV[v*3+1] + 1.e-4 < range[2]) {paramsV[v*3+1] += 2. * PETSC_PI;}
74c1cad2e7SMatthew G. Knepley       else if (paramsV[v*3+1] - 1.e-4 > range[3]) {paramsV[v*3+1] -= 2. * PETSC_PI;}
75c1cad2e7SMatthew G. Knepley     }
76c1cad2e7SMatthew G. Knepley #endif
77c1cad2e7SMatthew G. Knepley   }
78*9566063dSJacob Faibussowitsch   PetscCall(DMPlexVecRestoreClosure(cdm, NULL, coordinatesLocal, p, &Nv, &coords));
79c1cad2e7SMatthew G. Knepley   /* Calculate parameters (t or u,v) for new vertex at edge midpoint */
80c1cad2e7SMatthew G. Knepley   for (pm = 0; pm < Np; ++pm) {
81c1cad2e7SMatthew G. Knepley     params[pm] = 0.;
82c1cad2e7SMatthew G. Knepley     for (v = 0; v < Nv; ++v) params[pm] += paramsV[v*3+pm];
83c1cad2e7SMatthew G. Knepley     params[pm] /= Nv;
84c1cad2e7SMatthew G. Knepley   }
852c71b3e2SJacob Faibussowitsch   PetscCheckFalse((params[0] < range[0]) || (params[0] > range[1]),PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Point %D had bad interpolation", p);
862c71b3e2SJacob Faibussowitsch   PetscCheckFalse(Np > 1 && ((params[1] < range[2]) || (params[1] > range[3])),PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Point %D had bad interpolation", p);
87c1cad2e7SMatthew G. Knepley   /* Put coordinates for new vertex in result[] */
88*9566063dSJacob Faibussowitsch   PetscCall(EG_evaluate(obj, params, result));
89c1cad2e7SMatthew G. Knepley   for (d = 0; d < dE; ++d) gcoords[d] = result[d];
90c1cad2e7SMatthew G. Knepley   PetscFunctionReturn(0);
91c1cad2e7SMatthew G. Knepley }
92c1cad2e7SMatthew G. Knepley #endif
93c1cad2e7SMatthew G. Knepley 
94a8ededdfSMatthew G. Knepley /*@
95a8ededdfSMatthew G. Knepley   DMPlexSnapToGeomModel - Given a coordinate point 'mcoords' on the mesh point 'p', return the closest coordinate point 'gcoords' on the geometry model associated with that point.
96a8ededdfSMatthew G. Knepley 
97a8ededdfSMatthew G. Knepley   Not collective
98a8ededdfSMatthew G. Knepley 
99a8ededdfSMatthew G. Knepley   Input Parameters:
100a8ededdfSMatthew G. Knepley + dm      - The DMPlex object
101a8ededdfSMatthew G. Knepley . p       - The mesh point
102d410b0cfSMatthew G. Knepley . dE      - The coordinate dimension
103a8ededdfSMatthew G. Knepley - mcoords - A coordinate point lying on the mesh point
104a8ededdfSMatthew G. Knepley 
105a8ededdfSMatthew G. Knepley   Output Parameter:
106a8ededdfSMatthew G. Knepley . gcoords - The closest coordinate point on the geometry model associated with 'p' to the given point
107a8ededdfSMatthew G. Knepley 
108d410b0cfSMatthew G. Knepley   Note: Returns the original coordinates if no geometry model is found. Right now the only supported geometry model is EGADS. The coordinate dimension may be different from the coordinate dimension of the dm, for example if the transformation is extrusion.
109a8ededdfSMatthew G. Knepley 
110a8ededdfSMatthew G. Knepley   Level: intermediate
111a8ededdfSMatthew G. Knepley 
112a8ededdfSMatthew G. Knepley .seealso: DMRefine(), DMPlexCreate(), DMPlexSetRefinementUniform()
113a8ededdfSMatthew G. Knepley @*/
114d410b0cfSMatthew G. Knepley PetscErrorCode DMPlexSnapToGeomModel(DM dm, PetscInt p, PetscInt dE, const PetscScalar mcoords[], PetscScalar gcoords[])
115a8ededdfSMatthew G. Knepley {
116d410b0cfSMatthew G. Knepley   PetscInt d;
117a8ededdfSMatthew G. Knepley 
118c1cad2e7SMatthew G. Knepley   PetscFunctionBeginHot;
119a8ededdfSMatthew G. Knepley #ifdef PETSC_HAVE_EGADS
120c1cad2e7SMatthew G. Knepley   {
121c1cad2e7SMatthew G. Knepley     DM_Plex       *plex = (DM_Plex *) dm->data;
122c1cad2e7SMatthew G. Knepley     DMLabel        bodyLabel, faceLabel, edgeLabel;
123c1cad2e7SMatthew G. Knepley     PetscInt       bodyID, faceID, edgeID;
124c1cad2e7SMatthew G. Knepley     PetscContainer modelObj;
125c1cad2e7SMatthew G. Knepley     ego            model;
126c1cad2e7SMatthew G. Knepley     PetscBool      islite = PETSC_FALSE;
127c1cad2e7SMatthew G. Knepley 
128*9566063dSJacob Faibussowitsch     PetscCall(DMGetLabel(dm, "EGADS Body ID", &bodyLabel));
129*9566063dSJacob Faibussowitsch     PetscCall(DMGetLabel(dm, "EGADS Face ID", &faceLabel));
130*9566063dSJacob Faibussowitsch     PetscCall(DMGetLabel(dm, "EGADS Edge ID", &edgeLabel));
131c1cad2e7SMatthew G. Knepley     if (!bodyLabel || !faceLabel || !edgeLabel || plex->ignoreModel) {
132f0fcf0adSMatthew G. Knepley       for (d = 0; d < dE; ++d) gcoords[d] = mcoords[d];
133a8ededdfSMatthew G. Knepley       PetscFunctionReturn(0);
134a8ededdfSMatthew G. Knepley     }
135*9566063dSJacob Faibussowitsch     PetscCall(PetscObjectQuery((PetscObject) dm, "EGADS Model", (PetscObject *) &modelObj));
136c1cad2e7SMatthew G. Knepley     if (!modelObj) {
137*9566063dSJacob Faibussowitsch       PetscCall(PetscObjectQuery((PetscObject) dm, "EGADSLite Model", (PetscObject *) &modelObj));
138c1cad2e7SMatthew G. Knepley       islite = PETSC_TRUE;
139c1cad2e7SMatthew G. Knepley     }
140c1cad2e7SMatthew G. Knepley     if (!modelObj) {
141c1cad2e7SMatthew G. Knepley       for (d = 0; d < dE; ++d) gcoords[d] = mcoords[d];
142c1cad2e7SMatthew G. Knepley       PetscFunctionReturn(0);
143c1cad2e7SMatthew G. Knepley     }
144*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerGetPointer(modelObj, (void **) &model));
145*9566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(bodyLabel, p, &bodyID));
146*9566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(faceLabel, p, &faceID));
147*9566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(edgeLabel, p, &edgeID));
148c1cad2e7SMatthew G. Knepley     /* Allows for "Connective" Plex Edges present in models with multiple non-touching Entities */
149c1cad2e7SMatthew G. Knepley     if (bodyID < 0) {
150f0fcf0adSMatthew G. Knepley       for (d = 0; d < dE; ++d) gcoords[d] = mcoords[d];
151f0fcf0adSMatthew G. Knepley       PetscFunctionReturn(0);
152a8ededdfSMatthew G. Knepley     }
153*9566063dSJacob Faibussowitsch     if (islite) PetscCall(DMPlexSnapToGeomModel_EGADSLite_Internal(dm, p, model, bodyID, faceID, edgeID, mcoords, gcoords));
154*9566063dSJacob Faibussowitsch     else        PetscCall(DMPlexSnapToGeomModel_EGADS_Internal(dm, p, model, bodyID, faceID, edgeID, mcoords, gcoords));
155f0fcf0adSMatthew G. Knepley   }
156a8ededdfSMatthew G. Knepley #else
157f0fcf0adSMatthew G. Knepley   for (d = 0; d < dE; ++d) gcoords[d] = mcoords[d];
158a8ededdfSMatthew G. Knepley #endif
159a8ededdfSMatthew G. Knepley   PetscFunctionReturn(0);
160a8ededdfSMatthew G. Knepley }
1617bee2925SMatthew Knepley 
1627bee2925SMatthew Knepley #if defined(PETSC_HAVE_EGADS)
163c1cad2e7SMatthew G. Knepley static PetscErrorCode DMPlexEGADSPrintModel_Internal(ego model)
1647bee2925SMatthew Knepley {
1657bee2925SMatthew Knepley   ego geom, *bodies, *objs, *nobjs, *mobjs, *lobjs;
1667bee2925SMatthew Knepley   int oclass, mtype, *senses;
1677bee2925SMatthew Knepley   int Nb, b;
1687bee2925SMatthew Knepley 
1697bee2925SMatthew Knepley   PetscFunctionBeginUser;
1707bee2925SMatthew Knepley   /* test bodyTopo functions */
171*9566063dSJacob Faibussowitsch   PetscCall(EG_getTopology(model, &geom, &oclass, &mtype, NULL, &Nb, &bodies, &senses));
172*9566063dSJacob Faibussowitsch   PetscCall(PetscPrintf(PETSC_COMM_SELF, " Number of BODIES (nbodies): %d \n", Nb));
1737bee2925SMatthew Knepley 
1747bee2925SMatthew Knepley   for (b = 0; b < Nb; ++b) {
1757bee2925SMatthew Knepley     ego body = bodies[b];
1767bee2925SMatthew Knepley     int id, Nsh, Nf, Nl, l, Ne, e, Nv, v;
1777bee2925SMatthew Knepley 
1787bee2925SMatthew Knepley     /* Output Basic Model Topology */
179*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, SHELL, &Nsh, &objs));
180*9566063dSJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "   Number of SHELLS: %d \n", Nsh));
1817bee2925SMatthew Knepley     EG_free(objs);
1827bee2925SMatthew Knepley 
183*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, FACE,  &Nf, &objs));
184*9566063dSJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "   Number of FACES: %d \n", Nf));
1857bee2925SMatthew Knepley     EG_free(objs);
1867bee2925SMatthew Knepley 
187*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, LOOP,  &Nl, &lobjs));
188*9566063dSJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "   Number of LOOPS: %d \n", Nl));
1897bee2925SMatthew Knepley 
190*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, EDGE,  &Ne, &objs));
191*9566063dSJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "   Number of EDGES: %d \n", Ne));
1927bee2925SMatthew Knepley     EG_free(objs);
1937bee2925SMatthew Knepley 
194*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, NODE,  &Nv, &objs));
195*9566063dSJacob Faibussowitsch     PetscCall(PetscPrintf(PETSC_COMM_SELF, "   Number of NODES: %d \n", Nv));
1967bee2925SMatthew Knepley     EG_free(objs);
1977bee2925SMatthew Knepley 
1987bee2925SMatthew Knepley     for (l = 0; l < Nl; ++l) {
1997bee2925SMatthew Knepley       ego loop = lobjs[l];
2007bee2925SMatthew Knepley 
2017bee2925SMatthew Knepley       id   = EG_indexBodyTopo(body, loop);
202*9566063dSJacob Faibussowitsch       PetscCall(PetscPrintf(PETSC_COMM_SELF, "          LOOP ID: %d\n", id));
2037bee2925SMatthew Knepley 
2047bee2925SMatthew Knepley       /* Get EDGE info which associated with the current LOOP */
205*9566063dSJacob Faibussowitsch       PetscCall(EG_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &objs, &senses));
2067bee2925SMatthew Knepley 
2077bee2925SMatthew Knepley       for (e = 0; e < Ne; ++e) {
2087bee2925SMatthew Knepley         ego    edge      = objs[e];
2097bee2925SMatthew Knepley         double range[4]  = {0., 0., 0., 0.};
2107bee2925SMatthew Knepley         double point[3]  = {0., 0., 0.};
211266cfabeSMatthew G. Knepley         double params[3] = {0., 0., 0.};
212266cfabeSMatthew G. Knepley         double result[18];
2137bee2925SMatthew Knepley         int    peri;
2147bee2925SMatthew Knepley 
2155f80ce2aSJacob Faibussowitsch         id   = EG_indexBodyTopo(body, edge);
216*9566063dSJacob Faibussowitsch         PetscCall(PetscPrintf(PETSC_COMM_SELF, "            EDGE ID: %d (%d)\n", id, e));
2177bee2925SMatthew Knepley 
218*9566063dSJacob Faibussowitsch         PetscCall(EG_getRange(edge, range, &peri));
219*9566063dSJacob Faibussowitsch         PetscCall(PetscPrintf(PETSC_COMM_SELF, "  Range = %lf, %lf, %lf, %lf \n", range[0], range[1], range[2], range[3]));
2207bee2925SMatthew Knepley 
2217bee2925SMatthew Knepley         /* Get NODE info which associated with the current EDGE */
222*9566063dSJacob Faibussowitsch         PetscCall(EG_getTopology(edge, &geom, &oclass, &mtype, NULL, &Nv, &nobjs, &senses));
223266cfabeSMatthew G. Knepley         if (mtype == DEGENERATE) {
224*9566063dSJacob Faibussowitsch           PetscCall(PetscPrintf(PETSC_COMM_SELF, "  EDGE %d is DEGENERATE \n", id));
225266cfabeSMatthew G. Knepley         } else {
226266cfabeSMatthew G. Knepley           params[0] = range[0];
227*9566063dSJacob Faibussowitsch           PetscCall(EG_evaluate(edge, params, result));
228*9566063dSJacob Faibussowitsch           PetscCall(PetscPrintf(PETSC_COMM_SELF, "   between (%lf, %lf, %lf)", result[0], result[1], result[2]));
229266cfabeSMatthew G. Knepley           params[0] = range[1];
230*9566063dSJacob Faibussowitsch           PetscCall(EG_evaluate(edge, params, result));
231*9566063dSJacob Faibussowitsch           PetscCall(PetscPrintf(PETSC_COMM_SELF, " and (%lf, %lf, %lf)\n", result[0], result[1], result[2]));
232266cfabeSMatthew G. Knepley         }
2337bee2925SMatthew Knepley 
2347bee2925SMatthew Knepley         for (v = 0; v < Nv; ++v) {
2357bee2925SMatthew Knepley           ego    vertex = nobjs[v];
2367bee2925SMatthew Knepley           double limits[4];
2377bee2925SMatthew Knepley           int    dummy;
2387bee2925SMatthew Knepley 
239*9566063dSJacob Faibussowitsch           PetscCall(EG_getTopology(vertex, &geom, &oclass, &mtype, limits, &dummy, &mobjs, &senses));
2407bee2925SMatthew Knepley           id   = EG_indexBodyTopo(body, vertex);
241*9566063dSJacob Faibussowitsch           PetscCall(PetscPrintf(PETSC_COMM_SELF, "              NODE ID: %d \n", id));
242*9566063dSJacob Faibussowitsch           PetscCall(PetscPrintf(PETSC_COMM_SELF, "                 (x, y, z) = (%lf, %lf, %lf) \n", limits[0], limits[1], limits[2]));
2437bee2925SMatthew Knepley 
2447bee2925SMatthew Knepley           point[0] = point[0] + limits[0];
2457bee2925SMatthew Knepley           point[1] = point[1] + limits[1];
2467bee2925SMatthew Knepley           point[2] = point[2] + limits[2];
2477bee2925SMatthew Knepley         }
2487bee2925SMatthew Knepley       }
2497bee2925SMatthew Knepley     }
2507bee2925SMatthew Knepley     EG_free(lobjs);
2517bee2925SMatthew Knepley   }
2527bee2925SMatthew Knepley   PetscFunctionReturn(0);
2537bee2925SMatthew Knepley }
2547bee2925SMatthew Knepley 
2557bee2925SMatthew Knepley static PetscErrorCode DMPlexEGADSDestroy_Private(void *context)
2567bee2925SMatthew Knepley {
2577bee2925SMatthew Knepley   if (context) EG_close((ego) context);
2587bee2925SMatthew Knepley   return 0;
2597bee2925SMatthew Knepley }
2607bee2925SMatthew Knepley 
261c1cad2e7SMatthew G. Knepley static PetscErrorCode DMPlexCreateEGADS_Internal(MPI_Comm comm, ego context, ego model, DM *newdm)
2627bee2925SMatthew Knepley {
263c1cad2e7SMatthew G. Knepley   DMLabel        bodyLabel, faceLabel, edgeLabel, vertexLabel;
2647bee2925SMatthew Knepley   PetscInt       cStart, cEnd, c;
2657bee2925SMatthew Knepley   /* EGADSLite variables */
2667bee2925SMatthew Knepley   ego            geom, *bodies, *objs, *nobjs, *mobjs, *lobjs;
2677bee2925SMatthew Knepley   int            oclass, mtype, nbodies, *senses;
2687bee2925SMatthew Knepley   int            b;
2697bee2925SMatthew Knepley   /* PETSc variables */
2707bee2925SMatthew Knepley   DM             dm;
2717bee2925SMatthew Knepley   PetscHMapI     edgeMap = NULL;
2727bee2925SMatthew Knepley   PetscInt       dim = -1, cdim = -1, numCorners = 0, maxCorners = 0, numVertices = 0, newVertices = 0, numEdges = 0, numCells = 0, newCells = 0, numQuads = 0, cOff = 0, fOff = 0;
2737bee2925SMatthew Knepley   PetscInt      *cells  = NULL, *cone = NULL;
2747bee2925SMatthew Knepley   PetscReal     *coords = NULL;
2757bee2925SMatthew Knepley   PetscMPIInt    rank;
2767bee2925SMatthew Knepley 
2777bee2925SMatthew Knepley   PetscFunctionBegin;
278*9566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
279dd400576SPatrick Sanan   if (rank == 0) {
280266cfabeSMatthew G. Knepley     const PetscInt debug = 0;
281266cfabeSMatthew G. Knepley 
2827bee2925SMatthew Knepley     /* ---------------------------------------------------------------------------------------------------
2837bee2925SMatthew Knepley     Generate Petsc Plex
2847bee2925SMatthew Knepley       Get all Nodes in model, record coordinates in a correctly formatted array
2857bee2925SMatthew Knepley       Cycle through bodies, cycle through loops, recorde NODE IDs in a correctly formatted array
2867bee2925SMatthew Knepley       We need to uniformly refine the initial geometry to guarantee a valid mesh
2877bee2925SMatthew Knepley     */
2887bee2925SMatthew Knepley 
2897bee2925SMatthew Knepley     /* Calculate cell and vertex sizes */
290*9566063dSJacob Faibussowitsch     PetscCall(EG_getTopology(model, &geom, &oclass, &mtype, NULL, &nbodies, &bodies, &senses));
291*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapICreate(&edgeMap));
2927bee2925SMatthew Knepley     numEdges = 0;
2937bee2925SMatthew Knepley     for (b = 0; b < nbodies; ++b) {
2947bee2925SMatthew Knepley       ego body = bodies[b];
2957bee2925SMatthew Knepley       int id, Nl, l, Nv, v;
2967bee2925SMatthew Knepley 
297*9566063dSJacob Faibussowitsch       PetscCall(EG_getBodyTopos(body, NULL, LOOP, &Nl, &lobjs));
2987bee2925SMatthew Knepley       for (l = 0; l < Nl; ++l) {
2997bee2925SMatthew Knepley         ego loop = lobjs[l];
3007bee2925SMatthew Knepley         int Ner  = 0, Ne, e, Nc;
3017bee2925SMatthew Knepley 
302*9566063dSJacob Faibussowitsch         PetscCall(EG_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &objs, &senses));
3037bee2925SMatthew Knepley         for (e = 0; e < Ne; ++e) {
3047bee2925SMatthew Knepley           ego edge = objs[e];
3057bee2925SMatthew Knepley           int Nv, id;
3067bee2925SMatthew Knepley           PetscHashIter iter;
3077bee2925SMatthew Knepley           PetscBool     found;
3087bee2925SMatthew Knepley 
309*9566063dSJacob Faibussowitsch           PetscCall(EG_getTopology(edge, &geom, &oclass, &mtype, NULL, &Nv, &nobjs, &senses));
3107bee2925SMatthew Knepley           if (mtype == DEGENERATE) continue;
3115f80ce2aSJacob Faibussowitsch           id = EG_indexBodyTopo(body, edge);
312*9566063dSJacob Faibussowitsch           PetscCall(PetscHMapIFind(edgeMap, id-1, &iter, &found));
313*9566063dSJacob Faibussowitsch           if (!found) PetscCall(PetscHMapISet(edgeMap, id-1, numEdges++));
3147bee2925SMatthew Knepley           ++Ner;
3157bee2925SMatthew Knepley         }
3167bee2925SMatthew Knepley         if (Ner == 2)      {Nc = 2;}
3177bee2925SMatthew Knepley         else if (Ner == 3) {Nc = 4;}
3187bee2925SMatthew Knepley         else if (Ner == 4) {Nc = 8; ++numQuads;}
31998921bdaSJacob Faibussowitsch         else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Cannot support loop with %d edges", Ner);
3207bee2925SMatthew Knepley         numCells += Nc;
3217bee2925SMatthew Knepley         newCells += Nc-1;
3227bee2925SMatthew Knepley         maxCorners = PetscMax(Ner*2+1, maxCorners);
3237bee2925SMatthew Knepley       }
324*9566063dSJacob Faibussowitsch       PetscCall(EG_getBodyTopos(body, NULL, NODE, &Nv, &nobjs));
3257bee2925SMatthew Knepley       for (v = 0; v < Nv; ++v) {
3267bee2925SMatthew Knepley         ego vertex = nobjs[v];
3277bee2925SMatthew Knepley 
3287bee2925SMatthew Knepley         id = EG_indexBodyTopo(body, vertex);
3297bee2925SMatthew Knepley         /* TODO: Instead of assuming contiguous ids, we could use a hash table */
3307bee2925SMatthew Knepley         numVertices = PetscMax(id, numVertices);
3317bee2925SMatthew Knepley       }
3327bee2925SMatthew Knepley       EG_free(lobjs);
3337bee2925SMatthew Knepley       EG_free(nobjs);
3347bee2925SMatthew Knepley     }
335*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIGetSize(edgeMap, &numEdges));
3367bee2925SMatthew Knepley     newVertices  = numEdges + numQuads;
3377bee2925SMatthew Knepley     numVertices += newVertices;
3387bee2925SMatthew Knepley 
3397bee2925SMatthew Knepley     dim        = 2; /* Assume 3D Models :: Need to update to handle 2D Models in the future */
3407bee2925SMatthew Knepley     cdim       = 3; /* Assume 3D Models :: Need to update to handle 2D Models in the future */
3417bee2925SMatthew Knepley     numCorners = 3; /* Split cells into triangles */
342*9566063dSJacob Faibussowitsch     PetscCall(PetscMalloc3(numVertices*cdim, &coords, numCells*numCorners, &cells, maxCorners, &cone));
3437bee2925SMatthew Knepley 
3447bee2925SMatthew Knepley     /* Get vertex coordinates */
3457bee2925SMatthew Knepley     for (b = 0; b < nbodies; ++b) {
3467bee2925SMatthew Knepley       ego body = bodies[b];
3477bee2925SMatthew Knepley       int id, Nv, v;
3487bee2925SMatthew Knepley 
349*9566063dSJacob Faibussowitsch       PetscCall(EG_getBodyTopos(body, NULL, NODE, &Nv, &nobjs));
3507bee2925SMatthew Knepley       for (v = 0; v < Nv; ++v) {
3517bee2925SMatthew Knepley         ego    vertex = nobjs[v];
3527bee2925SMatthew Knepley         double limits[4];
3537bee2925SMatthew Knepley         int    dummy;
3547bee2925SMatthew Knepley 
355*9566063dSJacob Faibussowitsch         PetscCall(EG_getTopology(vertex, &geom, &oclass, &mtype, limits, &dummy, &mobjs, &senses));
3565f80ce2aSJacob Faibussowitsch         id   = EG_indexBodyTopo(body, vertex);
3577bee2925SMatthew Knepley         coords[(id-1)*cdim+0] = limits[0];
3587bee2925SMatthew Knepley         coords[(id-1)*cdim+1] = limits[1];
3597bee2925SMatthew Knepley         coords[(id-1)*cdim+2] = limits[2];
3607bee2925SMatthew Knepley       }
3617bee2925SMatthew Knepley       EG_free(nobjs);
3627bee2925SMatthew Knepley     }
363*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIClear(edgeMap));
3647bee2925SMatthew Knepley     fOff     = numVertices - newVertices + numEdges;
3657bee2925SMatthew Knepley     numEdges = 0;
3667bee2925SMatthew Knepley     numQuads = 0;
3677bee2925SMatthew Knepley     for (b = 0; b < nbodies; ++b) {
3687bee2925SMatthew Knepley       ego body = bodies[b];
3697bee2925SMatthew Knepley       int Nl, l;
3707bee2925SMatthew Knepley 
371*9566063dSJacob Faibussowitsch       PetscCall(EG_getBodyTopos(body, NULL, LOOP, &Nl, &lobjs));
3727bee2925SMatthew Knepley       for (l = 0; l < Nl; ++l) {
3737bee2925SMatthew Knepley         ego loop = lobjs[l];
3747bee2925SMatthew Knepley         int lid, Ner = 0, Ne, e;
3757bee2925SMatthew Knepley 
3765f80ce2aSJacob Faibussowitsch         lid  = EG_indexBodyTopo(body, loop);
377*9566063dSJacob Faibussowitsch         PetscCall(EG_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &objs, &senses));
3787bee2925SMatthew Knepley         for (e = 0; e < Ne; ++e) {
3797bee2925SMatthew Knepley           ego       edge = objs[e];
3807bee2925SMatthew Knepley           int       eid, Nv;
3817bee2925SMatthew Knepley           PetscHashIter iter;
3827bee2925SMatthew Knepley           PetscBool     found;
3837bee2925SMatthew Knepley 
384*9566063dSJacob Faibussowitsch           PetscCall(EG_getTopology(edge, &geom, &oclass, &mtype, NULL, &Nv, &nobjs, &senses));
3857bee2925SMatthew Knepley           if (mtype == DEGENERATE) continue;
3867bee2925SMatthew Knepley           ++Ner;
3875f80ce2aSJacob Faibussowitsch           eid  = EG_indexBodyTopo(body, edge);
388*9566063dSJacob Faibussowitsch           PetscCall(PetscHMapIFind(edgeMap, eid-1, &iter, &found));
3897bee2925SMatthew Knepley           if (!found) {
3907bee2925SMatthew Knepley             PetscInt v = numVertices - newVertices + numEdges;
3917bee2925SMatthew Knepley             double range[4], params[3] = {0., 0., 0.}, result[18];
3927bee2925SMatthew Knepley             int    periodic[2];
3937bee2925SMatthew Knepley 
394*9566063dSJacob Faibussowitsch             PetscCall(PetscHMapISet(edgeMap, eid-1, numEdges++));
395*9566063dSJacob Faibussowitsch             PetscCall(EG_getRange(edge, range, periodic));
3967bee2925SMatthew Knepley             params[0] = 0.5*(range[0] + range[1]);
397*9566063dSJacob Faibussowitsch             PetscCall(EG_evaluate(edge, params, result));
3987bee2925SMatthew Knepley             coords[v*cdim+0] = result[0];
3997bee2925SMatthew Knepley             coords[v*cdim+1] = result[1];
4007bee2925SMatthew Knepley             coords[v*cdim+2] = result[2];
4017bee2925SMatthew Knepley           }
4027bee2925SMatthew Knepley         }
4037bee2925SMatthew Knepley         if (Ner == 4) {
4047bee2925SMatthew Knepley           PetscInt v = fOff + numQuads++;
405266cfabeSMatthew G. Knepley           ego     *fobjs, face;
4067bee2925SMatthew Knepley           double   range[4], params[3] = {0., 0., 0.}, result[18];
407266cfabeSMatthew G. Knepley           int      Nf, fid, periodic[2];
4087bee2925SMatthew Knepley 
409*9566063dSJacob Faibussowitsch           PetscCall(EG_getBodyTopos(body, loop, FACE, &Nf, &fobjs));
410266cfabeSMatthew G. Knepley           face = fobjs[0];
4115f80ce2aSJacob Faibussowitsch           fid  = EG_indexBodyTopo(body, face);
4122c71b3e2SJacob Faibussowitsch           PetscCheckFalse(Nf != 1,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Loop %d has %d faces, instead of 1 (%d)", lid-1, Nf, fid);
413*9566063dSJacob Faibussowitsch           PetscCall(EG_getRange(face, range, periodic));
4147bee2925SMatthew Knepley           params[0] = 0.5*(range[0] + range[1]);
4157bee2925SMatthew Knepley           params[1] = 0.5*(range[2] + range[3]);
416*9566063dSJacob Faibussowitsch           PetscCall(EG_evaluate(face, params, result));
4177bee2925SMatthew Knepley           coords[v*cdim+0] = result[0];
4187bee2925SMatthew Knepley           coords[v*cdim+1] = result[1];
4197bee2925SMatthew Knepley           coords[v*cdim+2] = result[2];
4207bee2925SMatthew Knepley         }
4217bee2925SMatthew Knepley       }
4227bee2925SMatthew Knepley     }
4232c71b3e2SJacob Faibussowitsch     PetscCheckFalse(numEdges + numQuads != newVertices,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of new vertices %D != %D previous count", numEdges + numQuads, newVertices);
4247bee2925SMatthew Knepley 
4257bee2925SMatthew Knepley     /* Get cell vertices by traversing loops */
4267bee2925SMatthew Knepley     numQuads = 0;
4277bee2925SMatthew Knepley     cOff     = 0;
4287bee2925SMatthew Knepley     for (b = 0; b < nbodies; ++b) {
4297bee2925SMatthew Knepley       ego body = bodies[b];
4307bee2925SMatthew Knepley       int id, Nl, l;
4317bee2925SMatthew Knepley 
432*9566063dSJacob Faibussowitsch       PetscCall(EG_getBodyTopos(body, NULL, LOOP, &Nl, &lobjs));
4337bee2925SMatthew Knepley       for (l = 0; l < Nl; ++l) {
4347bee2925SMatthew Knepley         ego loop = lobjs[l];
4357bee2925SMatthew Knepley         int lid, Ner = 0, Ne, e, nc = 0, c, Nt, t;
4367bee2925SMatthew Knepley 
4375f80ce2aSJacob Faibussowitsch         lid  = EG_indexBodyTopo(body, loop);
438*9566063dSJacob Faibussowitsch         PetscCall(EG_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &objs, &senses));
4397bee2925SMatthew Knepley 
4407bee2925SMatthew Knepley         for (e = 0; e < Ne; ++e) {
4417bee2925SMatthew Knepley           ego edge = objs[e];
4427bee2925SMatthew Knepley           int points[3];
4437bee2925SMatthew Knepley           int eid, Nv, v, tmp;
4447bee2925SMatthew Knepley 
4457bee2925SMatthew Knepley           eid = EG_indexBodyTopo(body, edge);
446*9566063dSJacob Faibussowitsch           PetscCall(EG_getTopology(edge, &geom, &oclass, &mtype, NULL, &Nv, &nobjs, &senses));
447266cfabeSMatthew G. Knepley           if (mtype == DEGENERATE) continue;
448266cfabeSMatthew G. Knepley           else                     ++Ner;
4492c71b3e2SJacob Faibussowitsch           PetscCheckFalse(Nv != 2,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Edge %d has %d vertices != 2", eid, Nv);
4507bee2925SMatthew Knepley 
4517bee2925SMatthew Knepley           for (v = 0; v < Nv; ++v) {
4527bee2925SMatthew Knepley             ego vertex = nobjs[v];
4537bee2925SMatthew Knepley 
4547bee2925SMatthew Knepley             id = EG_indexBodyTopo(body, vertex);
4557bee2925SMatthew Knepley             points[v*2] = id-1;
4567bee2925SMatthew Knepley           }
4577bee2925SMatthew Knepley           {
4587bee2925SMatthew Knepley             PetscInt edgeNum;
4597bee2925SMatthew Knepley 
460*9566063dSJacob Faibussowitsch             PetscCall(PetscHMapIGet(edgeMap, eid-1, &edgeNum));
4617bee2925SMatthew Knepley             points[1] = numVertices - newVertices + edgeNum;
4627bee2925SMatthew Knepley           }
4637bee2925SMatthew Knepley           /* EGADS loops are not oriented, but seem to be in order, so we must piece them together */
4647bee2925SMatthew Knepley           if (!nc) {
4657bee2925SMatthew Knepley             for (v = 0; v < Nv+1; ++v) cone[nc++] = points[v];
4667bee2925SMatthew Knepley           } else {
4677bee2925SMatthew Knepley             if (cone[nc-1] == points[0])      {cone[nc++] = points[1]; if (cone[0] != points[2]) cone[nc++] = points[2];}
4687bee2925SMatthew Knepley             else if (cone[nc-1] == points[2]) {cone[nc++] = points[1]; if (cone[0] != points[0]) cone[nc++] = points[0];}
4697bee2925SMatthew Knepley             else if (cone[nc-3] == points[0]) {tmp = cone[nc-3]; cone[nc-3] = cone[nc-1]; cone[nc-1] = tmp; cone[nc++] = points[1]; if (cone[0] != points[2]) cone[nc++] = points[2];}
4707bee2925SMatthew Knepley             else if (cone[nc-3] == points[2]) {tmp = cone[nc-3]; cone[nc-3] = cone[nc-1]; cone[nc-1] = tmp; cone[nc++] = points[1]; if (cone[0] != points[0]) cone[nc++] = points[0];}
47198921bdaSJacob Faibussowitsch             else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Edge %d does not match its predecessor", eid);
4727bee2925SMatthew Knepley           }
4737bee2925SMatthew Knepley         }
4742c71b3e2SJacob Faibussowitsch         PetscCheckFalse(nc != 2*Ner,PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Number of corners %D != %D", nc, 2*Ner);
4757bee2925SMatthew Knepley         if (Ner == 4) {cone[nc++] = numVertices - newVertices + numEdges + numQuads++;}
4762c71b3e2SJacob Faibussowitsch         PetscCheckFalse(nc > maxCorners,PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Number of corners %D > %D max", nc, maxCorners);
4777bee2925SMatthew Knepley         /* Triangulate the loop */
4787bee2925SMatthew Knepley         switch (Ner) {
4797bee2925SMatthew Knepley           case 2: /* Bi-Segment -> 2 triangles */
4807bee2925SMatthew Knepley             Nt = 2;
4817bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[0];
4827bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[1];
4837bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[2];
4847bee2925SMatthew Knepley             ++cOff;
4857bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[0];
4867bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[2];
4877bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[3];
4887bee2925SMatthew Knepley             ++cOff;
4897bee2925SMatthew Knepley             break;
4907bee2925SMatthew Knepley           case 3: /* Triangle   -> 4 triangles */
4917bee2925SMatthew Knepley             Nt = 4;
4927bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[0];
4937bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[1];
4947bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[5];
4957bee2925SMatthew Knepley             ++cOff;
4967bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[1];
4977bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[2];
4987bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[3];
4997bee2925SMatthew Knepley             ++cOff;
5007bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[5];
5017bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[3];
5027bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[4];
5037bee2925SMatthew Knepley             ++cOff;
5047bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[1];
5057bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[3];
5067bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[5];
5077bee2925SMatthew Knepley             ++cOff;
5087bee2925SMatthew Knepley             break;
5097bee2925SMatthew Knepley           case 4: /* Quad       -> 8 triangles */
5107bee2925SMatthew Knepley             Nt = 8;
5117bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[0];
5127bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[1];
5137bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[7];
5147bee2925SMatthew Knepley             ++cOff;
5157bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[1];
5167bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[2];
5177bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[3];
5187bee2925SMatthew Knepley             ++cOff;
5197bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[3];
5207bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[4];
5217bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[5];
5227bee2925SMatthew Knepley             ++cOff;
5237bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[5];
5247bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[6];
5257bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[7];
5267bee2925SMatthew Knepley             ++cOff;
5277bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[8];
5287bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[1];
5297bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[3];
5307bee2925SMatthew Knepley             ++cOff;
5317bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[8];
5327bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[3];
5337bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[5];
5347bee2925SMatthew Knepley             ++cOff;
5357bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[8];
5367bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[5];
5377bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[7];
5387bee2925SMatthew Knepley             ++cOff;
5397bee2925SMatthew Knepley             cells[cOff*numCorners+0] = cone[8];
5407bee2925SMatthew Knepley             cells[cOff*numCorners+1] = cone[7];
5417bee2925SMatthew Knepley             cells[cOff*numCorners+2] = cone[1];
5427bee2925SMatthew Knepley             ++cOff;
5437bee2925SMatthew Knepley             break;
54498921bdaSJacob Faibussowitsch           default: SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Loop %d has %d edges, which we do not support", lid, Ner);
5457bee2925SMatthew Knepley         }
546266cfabeSMatthew G. Knepley         if (debug) {
5477bee2925SMatthew Knepley           for (t = 0; t < Nt; ++t) {
548*9566063dSJacob Faibussowitsch             PetscCall(PetscPrintf(PETSC_COMM_SELF, "  LOOP Corner NODEs Triangle %D (", t));
5497bee2925SMatthew Knepley             for (c = 0; c < numCorners; ++c) {
550*9566063dSJacob Faibussowitsch               if (c > 0) PetscCall(PetscPrintf(PETSC_COMM_SELF, ", "));
551*9566063dSJacob Faibussowitsch               PetscCall(PetscPrintf(PETSC_COMM_SELF, "%D", cells[(cOff-Nt+t)*numCorners+c]));
5527bee2925SMatthew Knepley             }
553*9566063dSJacob Faibussowitsch             PetscCall(PetscPrintf(PETSC_COMM_SELF, ")\n"));
5547bee2925SMatthew Knepley           }
5557bee2925SMatthew Knepley         }
556266cfabeSMatthew G. Knepley       }
5577bee2925SMatthew Knepley       EG_free(lobjs);
5587bee2925SMatthew Knepley     }
5597bee2925SMatthew Knepley   }
5602c71b3e2SJacob Faibussowitsch   PetscCheckFalse(cOff != numCells,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Count of total cells %D != %D previous count", cOff, numCells);
561*9566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateFromCellListPetsc(PETSC_COMM_WORLD, dim, numCells, numVertices, numCorners, PETSC_TRUE, cells, cdim, coords, &dm));
562*9566063dSJacob Faibussowitsch   PetscCall(PetscFree3(coords, cells, cone));
563*9566063dSJacob Faibussowitsch   PetscCall(PetscInfo(dm, " Total Number of Unique Cells    = %D (%D)\n", numCells, newCells));
564*9566063dSJacob Faibussowitsch   PetscCall(PetscInfo(dm, " Total Number of Unique Vertices = %D (%D)\n", numVertices, newVertices));
5657bee2925SMatthew Knepley   /* Embed EGADS model in DM */
5667bee2925SMatthew Knepley   {
5677bee2925SMatthew Knepley     PetscContainer modelObj, contextObj;
5687bee2925SMatthew Knepley 
569*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &modelObj));
570*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerSetPointer(modelObj, model));
571*9566063dSJacob Faibussowitsch     PetscCall(PetscObjectCompose((PetscObject) dm, "EGADS Model", (PetscObject) modelObj));
572*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerDestroy(&modelObj));
5737bee2925SMatthew Knepley 
574*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &contextObj));
575*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerSetPointer(contextObj, context));
576*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerSetUserDestroy(contextObj, DMPlexEGADSDestroy_Private));
577*9566063dSJacob Faibussowitsch     PetscCall(PetscObjectCompose((PetscObject) dm, "EGADS Context", (PetscObject) contextObj));
578*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerDestroy(&contextObj));
5797bee2925SMatthew Knepley   }
5807bee2925SMatthew Knepley   /* Label points */
581*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Body ID"));
582*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Body ID", &bodyLabel));
583*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Face ID"));
584*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Face ID", &faceLabel));
585*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Edge ID"));
586*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Edge ID", &edgeLabel));
587*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Vertex ID"));
588*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Vertex ID", &vertexLabel));
5897bee2925SMatthew Knepley   cOff = 0;
5907bee2925SMatthew Knepley   for (b = 0; b < nbodies; ++b) {
5917bee2925SMatthew Knepley     ego body = bodies[b];
5927bee2925SMatthew Knepley     int id, Nl, l;
5937bee2925SMatthew Knepley 
594*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, LOOP, &Nl, &lobjs));
5957bee2925SMatthew Knepley     for (l = 0; l < Nl; ++l) {
5967bee2925SMatthew Knepley       ego  loop = lobjs[l];
5977bee2925SMatthew Knepley       ego *fobjs;
5987bee2925SMatthew Knepley       int  lid, Nf, fid, Ner = 0, Ne, e, Nt = 0, t;
5997bee2925SMatthew Knepley 
6005f80ce2aSJacob Faibussowitsch       lid  = EG_indexBodyTopo(body, loop);
601*9566063dSJacob Faibussowitsch       PetscCall(EG_getBodyTopos(body, loop, FACE, &Nf, &fobjs));
6022c71b3e2SJacob Faibussowitsch       PetscCheckFalse(Nf > 1,PETSC_COMM_SELF, PETSC_ERR_SUP, "Loop %d has %d > 1 faces, which is not supported", lid, Nf);
6035f80ce2aSJacob Faibussowitsch       fid  = EG_indexBodyTopo(body, fobjs[0]);
6047bee2925SMatthew Knepley       EG_free(fobjs);
605*9566063dSJacob Faibussowitsch       PetscCall(EG_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &objs, &senses));
6067bee2925SMatthew Knepley       for (e = 0; e < Ne; ++e) {
6077bee2925SMatthew Knepley         ego             edge = objs[e];
6087bee2925SMatthew Knepley         int             eid, Nv, v;
6097bee2925SMatthew Knepley         PetscInt        points[3], support[2], numEdges, edgeNum;
6107bee2925SMatthew Knepley         const PetscInt *edges;
6117bee2925SMatthew Knepley 
6127bee2925SMatthew Knepley         eid = EG_indexBodyTopo(body, edge);
613*9566063dSJacob Faibussowitsch         PetscCall(EG_getTopology(edge, &geom, &oclass, &mtype, NULL, &Nv, &nobjs, &senses));
6147bee2925SMatthew Knepley         if (mtype == DEGENERATE) continue;
6157bee2925SMatthew Knepley         else                     ++Ner;
6167bee2925SMatthew Knepley         for (v = 0; v < Nv; ++v) {
6177bee2925SMatthew Knepley           ego vertex = nobjs[v];
6187bee2925SMatthew Knepley 
6197bee2925SMatthew Knepley           id   = EG_indexBodyTopo(body, vertex);
620*9566063dSJacob Faibussowitsch           PetscCall(DMLabelSetValue(edgeLabel, numCells + id-1, eid));
6217bee2925SMatthew Knepley           points[v*2] = numCells + id-1;
6227bee2925SMatthew Knepley         }
623*9566063dSJacob Faibussowitsch         PetscCall(PetscHMapIGet(edgeMap, eid-1, &edgeNum));
6247bee2925SMatthew Knepley         points[1] = numCells + numVertices - newVertices + edgeNum;
6257bee2925SMatthew Knepley 
626*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(edgeLabel, points[1], eid));
6277bee2925SMatthew Knepley         support[0] = points[0];
6287bee2925SMatthew Knepley         support[1] = points[1];
629*9566063dSJacob Faibussowitsch         PetscCall(DMPlexGetJoin(dm, 2, support, &numEdges, &edges));
6302c71b3e2SJacob Faibussowitsch         PetscCheckFalse(numEdges != 1,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Vertices (%D, %D) should only bound 1 edge, not %D", support[0], support[1], numEdges);
631*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(edgeLabel, edges[0], eid));
632*9566063dSJacob Faibussowitsch         PetscCall(DMPlexRestoreJoin(dm, 2, support, &numEdges, &edges));
6337bee2925SMatthew Knepley         support[0] = points[1];
6347bee2925SMatthew Knepley         support[1] = points[2];
635*9566063dSJacob Faibussowitsch         PetscCall(DMPlexGetJoin(dm, 2, support, &numEdges, &edges));
6362c71b3e2SJacob Faibussowitsch         PetscCheckFalse(numEdges != 1,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Vertices (%D, %D) should only bound 1 edge, not %D", support[0], support[1], numEdges);
637*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(edgeLabel, edges[0], eid));
638*9566063dSJacob Faibussowitsch         PetscCall(DMPlexRestoreJoin(dm, 2, support, &numEdges, &edges));
6397bee2925SMatthew Knepley       }
6407bee2925SMatthew Knepley       switch (Ner) {
6417bee2925SMatthew Knepley         case 2: Nt = 2;break;
6427bee2925SMatthew Knepley         case 3: Nt = 4;break;
6437bee2925SMatthew Knepley         case 4: Nt = 8;break;
64498921bdaSJacob Faibussowitsch         default: SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Loop with %d edges is unsupported", Ner);
6457bee2925SMatthew Knepley       }
6467bee2925SMatthew Knepley       for (t = 0; t < Nt; ++t) {
647*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(bodyLabel, cOff+t, b));
648*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(faceLabel, cOff+t, fid));
6497bee2925SMatthew Knepley       }
6507bee2925SMatthew Knepley       cOff += Nt;
6517bee2925SMatthew Knepley     }
6527bee2925SMatthew Knepley     EG_free(lobjs);
6537bee2925SMatthew Knepley   }
654*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIDestroy(&edgeMap));
655*9566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd));
6567bee2925SMatthew Knepley   for (c = cStart; c < cEnd; ++c) {
6577bee2925SMatthew Knepley     PetscInt *closure = NULL;
6587bee2925SMatthew Knepley     PetscInt  clSize, cl, bval, fval;
6597bee2925SMatthew Knepley 
660*9566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &clSize, &closure));
661*9566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(bodyLabel, c, &bval));
662*9566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(faceLabel, c, &fval));
6637bee2925SMatthew Knepley     for (cl = 0; cl < clSize*2; cl += 2) {
664*9566063dSJacob Faibussowitsch       PetscCall(DMLabelSetValue(bodyLabel, closure[cl], bval));
665*9566063dSJacob Faibussowitsch       PetscCall(DMLabelSetValue(faceLabel, closure[cl], fval));
6667bee2925SMatthew Knepley     }
667*9566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &clSize, &closure));
6687bee2925SMatthew Knepley   }
6697bee2925SMatthew Knepley   *newdm = dm;
6707bee2925SMatthew Knepley   PetscFunctionReturn(0);
6717bee2925SMatthew Knepley }
672c1cad2e7SMatthew G. Knepley 
673c1cad2e7SMatthew G. Knepley static PetscErrorCode DMPlexCreateEGADS(MPI_Comm comm, ego context, ego model, DM *newdm)
674c1cad2e7SMatthew G. Knepley {
675c1cad2e7SMatthew G. Knepley   DMLabel         bodyLabel, faceLabel, edgeLabel, vertexLabel;
676c1cad2e7SMatthew G. Knepley   // EGADS/EGADSLite variables
677c1cad2e7SMatthew G. Knepley   ego             geom, *bodies, *mobjs, *fobjs, *lobjs, *eobjs, *nobjs;
678c1cad2e7SMatthew G. Knepley   ego             topRef, prev, next;
679c1cad2e7SMatthew G. Knepley   int             oclass, mtype, nbodies, *senses, *lSenses, *eSenses;
680c1cad2e7SMatthew G. Knepley   int             b;
681c1cad2e7SMatthew G. Knepley   // PETSc variables
682c1cad2e7SMatthew G. Knepley   DM              dm;
683c1cad2e7SMatthew G. Knepley   PetscHMapI      edgeMap = NULL, bodyIndexMap = NULL, bodyVertexMap = NULL, bodyEdgeMap = NULL, bodyFaceMap = NULL, bodyEdgeGlobalMap = NULL;
684c1cad2e7SMatthew G. Knepley   PetscInt        dim = -1, cdim = -1, numCorners = 0, numVertices = 0, numEdges = 0, numFaces = 0, numCells = 0, edgeCntr = 0;
685c1cad2e7SMatthew G. Knepley   PetscInt        cellCntr = 0, numPoints = 0;
686c1cad2e7SMatthew G. Knepley   PetscInt        *cells  = NULL;
687c1cad2e7SMatthew G. Knepley   const PetscInt  *cone = NULL;
688c1cad2e7SMatthew G. Knepley   PetscReal       *coords = NULL;
689c1cad2e7SMatthew G. Knepley   PetscMPIInt      rank;
690c1cad2e7SMatthew G. Knepley 
691c1cad2e7SMatthew G. Knepley   PetscFunctionBeginUser;
692*9566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
693c1cad2e7SMatthew G. Knepley   if (!rank) {
694c1cad2e7SMatthew G. Knepley     // ---------------------------------------------------------------------------------------------------
695c1cad2e7SMatthew G. Knepley     // Generate Petsc Plex
696c1cad2e7SMatthew G. Knepley     //  Get all Nodes in model, record coordinates in a correctly formatted array
697c1cad2e7SMatthew G. Knepley     //  Cycle through bodies, cycle through loops, recorde NODE IDs in a correctly formatted array
698c1cad2e7SMatthew G. Knepley     //  We need to uniformly refine the initial geometry to guarantee a valid mesh
699c1cad2e7SMatthew G. Knepley 
700c1cad2e7SMatthew G. Knepley   // Caluculate cell and vertex sizes
701*9566063dSJacob Faibussowitsch   PetscCall(EG_getTopology(model, &geom, &oclass, &mtype, NULL, &nbodies, &bodies, &senses));
702c1cad2e7SMatthew G. Knepley 
703*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapICreate(&edgeMap));
704*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&bodyIndexMap));
705*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&bodyVertexMap));
706*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&bodyEdgeMap));
707*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&bodyEdgeGlobalMap));
708*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&bodyFaceMap));
709c1cad2e7SMatthew G. Knepley 
710c1cad2e7SMatthew G. Knepley   for (b = 0; b < nbodies; ++b) {
711c1cad2e7SMatthew G. Knepley       ego             body = bodies[b];
712c1cad2e7SMatthew G. Knepley     int             Nf, Ne, Nv;
713c1cad2e7SMatthew G. Knepley     PetscHashIter   BIiter, BViter, BEiter, BEGiter, BFiter, EMiter;
714c1cad2e7SMatthew G. Knepley     PetscBool       BIfound, BVfound, BEfound, BEGfound, BFfound, EMfound;
715c1cad2e7SMatthew G. Knepley 
716*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyIndexMap, b, &BIiter, &BIfound));
717*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyVertexMap, b, &BViter, &BVfound));
718*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyEdgeMap, b, &BEiter, &BEfound));
719*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyEdgeGlobalMap, b, &BEGiter, &BEGfound));
720*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyFaceMap, b, &BFiter, &BFfound));
721c1cad2e7SMatthew G. Knepley 
722*9566063dSJacob Faibussowitsch     if (!BIfound)  PetscCall(PetscHMapISet(bodyIndexMap, b, numFaces + numEdges + numVertices));
723*9566063dSJacob Faibussowitsch     if (!BVfound)  PetscCall(PetscHMapISet(bodyVertexMap, b, numVertices));
724*9566063dSJacob Faibussowitsch     if (!BEfound)  PetscCall(PetscHMapISet(bodyEdgeMap, b, numEdges));
725*9566063dSJacob Faibussowitsch     if (!BEGfound) PetscCall(PetscHMapISet(bodyEdgeGlobalMap, b, edgeCntr));
726*9566063dSJacob Faibussowitsch     if (!BFfound)  PetscCall(PetscHMapISet(bodyFaceMap, b, numFaces));
727c1cad2e7SMatthew G. Knepley 
728*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, FACE, &Nf, &fobjs));
729*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, EDGE, &Ne, &eobjs));
730*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, NODE, &Nv, &nobjs));
731c1cad2e7SMatthew G. Knepley     EG_free(fobjs);
732c1cad2e7SMatthew G. Knepley     EG_free(eobjs);
733c1cad2e7SMatthew G. Knepley     EG_free(nobjs);
734c1cad2e7SMatthew G. Knepley 
735c1cad2e7SMatthew G. Knepley     // Remove DEGENERATE EDGES from Edge count
736*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, EDGE, &Ne, &eobjs));
737c1cad2e7SMatthew G. Knepley     int Netemp = 0;
738c1cad2e7SMatthew G. Knepley     for (int e = 0; e < Ne; ++e) {
739c1cad2e7SMatthew G. Knepley       ego     edge = eobjs[e];
740c1cad2e7SMatthew G. Knepley       int     eid;
741c1cad2e7SMatthew G. Knepley 
742*9566063dSJacob Faibussowitsch       PetscCall(EG_getInfo(edge, &oclass, &mtype, &topRef, &prev, &next));
7435f80ce2aSJacob Faibussowitsch       eid = EG_indexBodyTopo(body, edge);
744c1cad2e7SMatthew G. Knepley 
745*9566063dSJacob Faibussowitsch       PetscCall(PetscHMapIFind(edgeMap, edgeCntr + eid - 1, &EMiter, &EMfound));
746c1cad2e7SMatthew G. Knepley       if (mtype == DEGENERATE) {
747*9566063dSJacob Faibussowitsch         if (!EMfound) PetscCall(PetscHMapISet(edgeMap, edgeCntr + eid - 1, -1));
748c1cad2e7SMatthew G. Knepley       }
749c1cad2e7SMatthew G. Knepley       else {
750c1cad2e7SMatthew G. Knepley       ++Netemp;
751*9566063dSJacob Faibussowitsch         if (!EMfound) PetscCall(PetscHMapISet(edgeMap, edgeCntr + eid - 1, Netemp));
752c1cad2e7SMatthew G. Knepley       }
753c1cad2e7SMatthew G. Knepley     }
754c1cad2e7SMatthew G. Knepley     EG_free(eobjs);
755c1cad2e7SMatthew G. Knepley 
756c1cad2e7SMatthew G. Knepley     // Determine Number of Cells
757*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, FACE, &Nf, &fobjs));
758c1cad2e7SMatthew G. Knepley     for (int f = 0; f < Nf; ++f) {
759c1cad2e7SMatthew G. Knepley         ego     face = fobjs[f];
760c1cad2e7SMatthew G. Knepley     int     edgeTemp = 0;
761c1cad2e7SMatthew G. Knepley 
762*9566063dSJacob Faibussowitsch       PetscCall(EG_getBodyTopos(body, face, EDGE, &Ne, &eobjs));
763c1cad2e7SMatthew G. Knepley       for (int e = 0; e < Ne; ++e) {
764c1cad2e7SMatthew G. Knepley         ego     edge = eobjs[e];
765c1cad2e7SMatthew G. Knepley 
766*9566063dSJacob Faibussowitsch         PetscCall(EG_getInfo(edge, &oclass, &mtype, &topRef, &prev, &next));
767c1cad2e7SMatthew G. Knepley         if (mtype != DEGENERATE) {++edgeTemp;}
768c1cad2e7SMatthew G. Knepley       }
769c1cad2e7SMatthew G. Knepley       numCells += (2 * edgeTemp);
770c1cad2e7SMatthew G. Knepley       EG_free(eobjs);
771c1cad2e7SMatthew G. Knepley     }
772c1cad2e7SMatthew G. Knepley     EG_free(fobjs);
773c1cad2e7SMatthew G. Knepley 
774c1cad2e7SMatthew G. Knepley     numFaces    += Nf;
775c1cad2e7SMatthew G. Knepley     numEdges    += Netemp;
776c1cad2e7SMatthew G. Knepley     numVertices += Nv;
777c1cad2e7SMatthew G. Knepley     edgeCntr    += Ne;
778c1cad2e7SMatthew G. Knepley   }
779c1cad2e7SMatthew G. Knepley 
780c1cad2e7SMatthew G. Knepley   // Set up basic DMPlex parameters
781c1cad2e7SMatthew G. Knepley   dim        = 2;    // Assumes 3D Models :: Need to handle 2D modles in the future
782c1cad2e7SMatthew G. Knepley   cdim       = 3;     // Assumes 3D Models :: Need to update to handle 2D modles in future
783c1cad2e7SMatthew G. Knepley   numCorners = 3;     // Split Faces into triangles
784c1cad2e7SMatthew G. Knepley     numPoints  = numVertices + numEdges + numFaces;   // total number of coordinate points
785c1cad2e7SMatthew G. Knepley 
786*9566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(numPoints*cdim, &coords, numCells*numCorners, &cells));
787c1cad2e7SMatthew G. Knepley 
788c1cad2e7SMatthew G. Knepley   // Get Vertex Coordinates and Set up Cells
789c1cad2e7SMatthew G. Knepley   for (b = 0; b < nbodies; ++b) {
790c1cad2e7SMatthew G. Knepley     ego             body = bodies[b];
791c1cad2e7SMatthew G. Knepley     int             Nf, Ne, Nv;
792c1cad2e7SMatthew G. Knepley     PetscInt        bodyVertexIndexStart, bodyEdgeIndexStart, bodyEdgeGlobalIndexStart, bodyFaceIndexStart;
793c1cad2e7SMatthew G. Knepley     PetscHashIter   BViter, BEiter, BEGiter, BFiter, EMiter;
794c1cad2e7SMatthew G. Knepley     PetscBool       BVfound, BEfound, BEGfound, BFfound, EMfound;
795c1cad2e7SMatthew G. Knepley 
796c1cad2e7SMatthew G. Knepley     // Vertices on Current Body
797*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, NODE, &Nv, &nobjs));
798c1cad2e7SMatthew G. Knepley 
799*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyVertexMap, b, &BViter, &BVfound));
80028b400f6SJacob Faibussowitsch     PetscCheck(BVfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Body %d not found in bodyVertexMap", b);
801*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIGet(bodyVertexMap, b, &bodyVertexIndexStart));
802c1cad2e7SMatthew G. Knepley 
803c1cad2e7SMatthew G. Knepley     for (int v = 0; v < Nv; ++v) {
804c1cad2e7SMatthew G. Knepley       ego    vertex = nobjs[v];
805c1cad2e7SMatthew G. Knepley     double limits[4];
806c1cad2e7SMatthew G. Knepley     int    id, dummy;
807c1cad2e7SMatthew G. Knepley 
808*9566063dSJacob Faibussowitsch     PetscCall(EG_getTopology(vertex, &geom, &oclass, &mtype, limits, &dummy, &mobjs, &senses));
8095f80ce2aSJacob Faibussowitsch     id = EG_indexBodyTopo(body, vertex);
810c1cad2e7SMatthew G. Knepley 
811c1cad2e7SMatthew G. Knepley     coords[(bodyVertexIndexStart + id - 1)*cdim + 0] = limits[0];
812c1cad2e7SMatthew G. Knepley     coords[(bodyVertexIndexStart + id - 1)*cdim + 1] = limits[1];
813c1cad2e7SMatthew G. Knepley     coords[(bodyVertexIndexStart + id - 1)*cdim + 2] = limits[2];
814c1cad2e7SMatthew G. Knepley     }
815c1cad2e7SMatthew G. Knepley     EG_free(nobjs);
816c1cad2e7SMatthew G. Knepley 
817c1cad2e7SMatthew G. Knepley     // Edge Midpoint Vertices on Current Body
818*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, EDGE, &Ne, &eobjs));
819c1cad2e7SMatthew G. Knepley 
820*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyEdgeMap, b, &BEiter, &BEfound));
82128b400f6SJacob Faibussowitsch     PetscCheck(BEfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Body %d not found in bodyEdgeMap", b);
822*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIGet(bodyEdgeMap, b, &bodyEdgeIndexStart));
823c1cad2e7SMatthew G. Knepley 
824*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyEdgeGlobalMap, b, &BEGiter, &BEGfound));
82528b400f6SJacob Faibussowitsch     PetscCheck(BEGfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Body %d not found in bodyEdgeGlobalMap", b);
826*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIGet(bodyEdgeGlobalMap, b, &bodyEdgeGlobalIndexStart));
827c1cad2e7SMatthew G. Knepley 
828c1cad2e7SMatthew G. Knepley     for (int e = 0; e < Ne; ++e) {
829c1cad2e7SMatthew G. Knepley       ego          edge = eobjs[e];
830c1cad2e7SMatthew G. Knepley     double       range[2], avgt[1], cntrPnt[9];
831c1cad2e7SMatthew G. Knepley     int          eid, eOffset;
832c1cad2e7SMatthew G. Knepley     int          periodic;
833c1cad2e7SMatthew G. Knepley 
834*9566063dSJacob Faibussowitsch     PetscCall(EG_getInfo(edge, &oclass, &mtype, &topRef, &prev, &next));
835c1cad2e7SMatthew G. Knepley     if (mtype == DEGENERATE) {continue;}
836c1cad2e7SMatthew G. Knepley 
8375f80ce2aSJacob Faibussowitsch     eid = EG_indexBodyTopo(body, edge);
838c1cad2e7SMatthew G. Knepley 
839c1cad2e7SMatthew G. Knepley     // get relative offset from globalEdgeID Vector
840*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(edgeMap, bodyEdgeGlobalIndexStart + eid - 1, &EMiter, &EMfound));
84128b400f6SJacob Faibussowitsch       PetscCheck(EMfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Edge %d not found in edgeMap", bodyEdgeGlobalIndexStart + eid - 1);
842*9566063dSJacob Faibussowitsch       PetscCall(PetscHMapIGet(edgeMap, bodyEdgeGlobalIndexStart + eid - 1, &eOffset));
843c1cad2e7SMatthew G. Knepley 
844*9566063dSJacob Faibussowitsch     PetscCall(EG_getRange(edge, range, &periodic));
845c1cad2e7SMatthew G. Knepley     avgt[0] = (range[0] + range[1]) /  2.;
846c1cad2e7SMatthew G. Knepley 
847*9566063dSJacob Faibussowitsch     PetscCall(EG_evaluate(edge, avgt, cntrPnt));
848c1cad2e7SMatthew G. Knepley     coords[(numVertices + bodyEdgeIndexStart + eOffset - 1)*cdim + 0] = cntrPnt[0];
849c1cad2e7SMatthew G. Knepley         coords[(numVertices + bodyEdgeIndexStart + eOffset - 1)*cdim + 1] = cntrPnt[1];
850c1cad2e7SMatthew G. Knepley     coords[(numVertices + bodyEdgeIndexStart + eOffset - 1)*cdim + 2] = cntrPnt[2];
851c1cad2e7SMatthew G. Knepley     }
852c1cad2e7SMatthew G. Knepley     EG_free(eobjs);
853c1cad2e7SMatthew G. Knepley 
854c1cad2e7SMatthew G. Knepley     // Face Midpoint Vertices on Current Body
855*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, FACE, &Nf, &fobjs));
856c1cad2e7SMatthew G. Knepley 
857*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyFaceMap, b, &BFiter, &BFfound));
85828b400f6SJacob Faibussowitsch     PetscCheck(BFfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Body %d not found in bodyFaceMap", b);
859*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIGet(bodyFaceMap, b, &bodyFaceIndexStart));
860c1cad2e7SMatthew G. Knepley 
861c1cad2e7SMatthew G. Knepley     for (int f = 0; f < Nf; ++f) {
862c1cad2e7SMatthew G. Knepley     ego       face = fobjs[f];
863c1cad2e7SMatthew G. Knepley     double    range[4], avgUV[2], cntrPnt[18];
864c1cad2e7SMatthew G. Knepley     int       peri, id;
865c1cad2e7SMatthew G. Knepley 
866c1cad2e7SMatthew G. Knepley     id = EG_indexBodyTopo(body, face);
867*9566063dSJacob Faibussowitsch     PetscCall(EG_getRange(face, range, &peri));
868c1cad2e7SMatthew G. Knepley 
869c1cad2e7SMatthew G. Knepley     avgUV[0] = (range[0] + range[1]) / 2.;
870c1cad2e7SMatthew G. Knepley     avgUV[1] = (range[2] + range[3]) / 2.;
871*9566063dSJacob Faibussowitsch     PetscCall(EG_evaluate(face, avgUV, cntrPnt));
872c1cad2e7SMatthew G. Knepley 
873c1cad2e7SMatthew G. Knepley     coords[(numVertices + numEdges + bodyFaceIndexStart + id - 1)*cdim + 0] = cntrPnt[0];
874c1cad2e7SMatthew G. Knepley     coords[(numVertices + numEdges + bodyFaceIndexStart + id - 1)*cdim + 1] = cntrPnt[1];
875c1cad2e7SMatthew G. Knepley     coords[(numVertices + numEdges + bodyFaceIndexStart + id - 1)*cdim + 2] = cntrPnt[2];
876c1cad2e7SMatthew G. Knepley     }
877c1cad2e7SMatthew G. Knepley     EG_free(fobjs);
878c1cad2e7SMatthew G. Knepley 
879c1cad2e7SMatthew G. Knepley     // Define Cells :: Note - This could be incorporated in the Face Midpoint Vertices Loop but was kept separate for clarity
880*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, FACE, &Nf, &fobjs));
881c1cad2e7SMatthew G. Knepley     for (int f = 0; f < Nf; ++f) {
882c1cad2e7SMatthew G. Knepley     ego      face = fobjs[f];
883c1cad2e7SMatthew G. Knepley     int      fID, midFaceID, midPntID, startID, endID, Nl;
884c1cad2e7SMatthew G. Knepley 
8855f80ce2aSJacob Faibussowitsch     fID = EG_indexBodyTopo(body, face);
886c1cad2e7SMatthew G. Knepley     midFaceID = numVertices + numEdges + bodyFaceIndexStart + fID - 1;
887c1cad2e7SMatthew G. Knepley     // Must Traverse Loop to ensure we have all necessary information like the sense (+/- 1) of the edges.
888c1cad2e7SMatthew G. Knepley     // TODO :: Only handles single loop faces (No holes). The choices for handling multiloop faces are:
889c1cad2e7SMatthew G. Knepley     //            1) Use the DMPlexCreateEGADSFromFile() with the -dm_plex_egads_with_tess = 1 option.
890c1cad2e7SMatthew G. Knepley     //               This will use a default EGADS tessellation as an initial surface mesh.
891c1cad2e7SMatthew G. Knepley     //            2) Create the initial surface mesh via a 2D mesher :: Currently not availble (?future?)
892c1cad2e7SMatthew G. Knepley     //               May I suggest the XXXX as a starting point?
893c1cad2e7SMatthew G. Knepley 
894*9566063dSJacob Faibussowitsch     PetscCall(EG_getTopology(face, &geom, &oclass, &mtype, NULL, &Nl, &lobjs, &lSenses));
895c1cad2e7SMatthew G. Knepley 
8962c71b3e2SJacob Faibussowitsch       PetscCheckFalse(Nl > 1,PETSC_COMM_SELF, PETSC_ERR_PLIB, "Face has %d Loops. Can only handle Faces with 1 Loop. Please use --dm_plex_egads_with_tess = 1 Option", Nl);
897c1cad2e7SMatthew G. Knepley     for (int l = 0; l < Nl; ++l) {
898c1cad2e7SMatthew G. Knepley           ego      loop = lobjs[l];
899c1cad2e7SMatthew G. Knepley 
900*9566063dSJacob Faibussowitsch           PetscCall(EG_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &eobjs, &eSenses));
901c1cad2e7SMatthew G. Knepley       for (int e = 0; e < Ne; ++e) {
902c1cad2e7SMatthew G. Knepley         ego     edge = eobjs[e];
903c1cad2e7SMatthew G. Knepley         int     eid, eOffset;
904c1cad2e7SMatthew G. Knepley 
905*9566063dSJacob Faibussowitsch         PetscCall(EG_getInfo(edge, &oclass, &mtype, &topRef, &prev, &next));
906c1cad2e7SMatthew G. Knepley       eid = EG_indexBodyTopo(body, edge);
907c1cad2e7SMatthew G. Knepley         if (mtype == DEGENERATE) { continue; }
908c1cad2e7SMatthew G. Knepley 
909c1cad2e7SMatthew G. Knepley         // get relative offset from globalEdgeID Vector
910*9566063dSJacob Faibussowitsch         PetscCall(PetscHMapIFind(edgeMap, bodyEdgeGlobalIndexStart + eid - 1, &EMiter, &EMfound));
91128b400f6SJacob Faibussowitsch           PetscCheck(EMfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Edge %d of Body %d not found in edgeMap. Global Edge ID :: %d", eid, b, bodyEdgeGlobalIndexStart + eid - 1);
912*9566063dSJacob Faibussowitsch           PetscCall(PetscHMapIGet(edgeMap, bodyEdgeGlobalIndexStart + eid - 1, &eOffset));
913c1cad2e7SMatthew G. Knepley 
914c1cad2e7SMatthew G. Knepley       midPntID = numVertices + bodyEdgeIndexStart + eOffset - 1;
915c1cad2e7SMatthew G. Knepley 
916*9566063dSJacob Faibussowitsch         PetscCall(EG_getTopology(edge, &geom, &oclass, &mtype, NULL, &Nv, &nobjs, &senses));
917c1cad2e7SMatthew G. Knepley 
918c1cad2e7SMatthew G. Knepley         if (eSenses[e] > 0) { startID = EG_indexBodyTopo(body, nobjs[0]); endID = EG_indexBodyTopo(body, nobjs[1]); }
919c1cad2e7SMatthew G. Knepley         else { startID = EG_indexBodyTopo(body, nobjs[1]); endID = EG_indexBodyTopo(body, nobjs[0]); }
920c1cad2e7SMatthew G. Knepley 
921c1cad2e7SMatthew G. Knepley       // Define 2 Cells per Edge with correct orientation
922c1cad2e7SMatthew G. Knepley       cells[cellCntr*numCorners + 0] = midFaceID;
923c1cad2e7SMatthew G. Knepley       cells[cellCntr*numCorners + 1] = bodyVertexIndexStart + startID - 1;
924c1cad2e7SMatthew G. Knepley       cells[cellCntr*numCorners + 2] = midPntID;
925c1cad2e7SMatthew G. Knepley 
926c1cad2e7SMatthew G. Knepley       cells[cellCntr*numCorners + 3] = midFaceID;
927c1cad2e7SMatthew G. Knepley       cells[cellCntr*numCorners + 4] = midPntID;
928c1cad2e7SMatthew G. Knepley       cells[cellCntr*numCorners + 5] = bodyVertexIndexStart + endID - 1;
929c1cad2e7SMatthew G. Knepley 
930c1cad2e7SMatthew G. Knepley       cellCntr = cellCntr + 2;
931c1cad2e7SMatthew G. Knepley       }
932c1cad2e7SMatthew G. Knepley     }
933c1cad2e7SMatthew G. Knepley     }
934c1cad2e7SMatthew G. Knepley     EG_free(fobjs);
935c1cad2e7SMatthew G. Knepley   }
936c1cad2e7SMatthew G. Knepley   }
937c1cad2e7SMatthew G. Knepley 
938c1cad2e7SMatthew G. Knepley   // Generate DMPlex
939*9566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateFromCellListPetsc(PETSC_COMM_WORLD, dim, numCells, numPoints, numCorners, PETSC_TRUE, cells, cdim, coords, &dm));
940*9566063dSJacob Faibussowitsch   PetscCall(PetscFree2(coords, cells));
941*9566063dSJacob Faibussowitsch   PetscCall(PetscInfo(dm, " Total Number of Unique Cells    = %D \n", numCells));
942*9566063dSJacob Faibussowitsch   PetscCall(PetscInfo(dm, " Total Number of Unique Vertices = %D \n", numVertices));
943c1cad2e7SMatthew G. Knepley 
944c1cad2e7SMatthew G. Knepley   // Embed EGADS model in DM
945c1cad2e7SMatthew G. Knepley   {
946c1cad2e7SMatthew G. Knepley     PetscContainer modelObj, contextObj;
947c1cad2e7SMatthew G. Knepley 
948*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &modelObj));
949*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerSetPointer(modelObj, model));
950*9566063dSJacob Faibussowitsch     PetscCall(PetscObjectCompose((PetscObject) dm, "EGADS Model", (PetscObject) modelObj));
951*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerDestroy(&modelObj));
952c1cad2e7SMatthew G. Knepley 
953*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &contextObj));
954*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerSetPointer(contextObj, context));
955*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerSetUserDestroy(contextObj, DMPlexEGADSDestroy_Private));
956*9566063dSJacob Faibussowitsch     PetscCall(PetscObjectCompose((PetscObject) dm, "EGADS Context", (PetscObject) contextObj));
957*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerDestroy(&contextObj));
958c1cad2e7SMatthew G. Knepley   }
959c1cad2e7SMatthew G. Knepley   // Label points
960c1cad2e7SMatthew G. Knepley   PetscInt   nStart, nEnd;
961c1cad2e7SMatthew G. Knepley 
962*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Body ID"));
963*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Body ID", &bodyLabel));
964*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Face ID"));
965*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Face ID", &faceLabel));
966*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Edge ID"));
967*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Edge ID", &edgeLabel));
968*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Vertex ID"));
969*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Vertex ID", &vertexLabel));
970c1cad2e7SMatthew G. Knepley 
971*9566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm, 2, &nStart, &nEnd));
972c1cad2e7SMatthew G. Knepley 
973c1cad2e7SMatthew G. Knepley   cellCntr = 0;
974c1cad2e7SMatthew G. Knepley   for (b = 0; b < nbodies; ++b) {
975c1cad2e7SMatthew G. Knepley     ego             body = bodies[b];
976c1cad2e7SMatthew G. Knepley   int             Nv, Ne, Nf;
977c1cad2e7SMatthew G. Knepley   PetscInt        bodyVertexIndexStart, bodyEdgeIndexStart, bodyEdgeGlobalIndexStart, bodyFaceIndexStart;
978c1cad2e7SMatthew G. Knepley   PetscHashIter   BViter, BEiter, BEGiter, BFiter, EMiter;
979c1cad2e7SMatthew G. Knepley   PetscBool       BVfound, BEfound, BEGfound, BFfound, EMfound;
980c1cad2e7SMatthew G. Knepley 
981*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIFind(bodyVertexMap, b, &BViter, &BVfound));
98228b400f6SJacob Faibussowitsch   PetscCheck(BVfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Body %d not found in bodyVertexMap", b);
983*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIGet(bodyVertexMap, b, &bodyVertexIndexStart));
984c1cad2e7SMatthew G. Knepley 
985*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIFind(bodyEdgeMap, b, &BEiter, &BEfound));
98628b400f6SJacob Faibussowitsch   PetscCheck(BEfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Body %d not found in bodyEdgeMap", b);
987*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIGet(bodyEdgeMap, b, &bodyEdgeIndexStart));
988c1cad2e7SMatthew G. Knepley 
989*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyFaceMap, b, &BFiter, &BFfound));
99028b400f6SJacob Faibussowitsch   PetscCheck(BFfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Body %d not found in bodyFaceMap", b);
991*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIGet(bodyFaceMap, b, &bodyFaceIndexStart));
992c1cad2e7SMatthew G. Knepley 
993*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(bodyEdgeGlobalMap, b, &BEGiter, &BEGfound));
99428b400f6SJacob Faibussowitsch     PetscCheck(BEGfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Body %d not found in bodyEdgeGlobalMap", b);
995*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIGet(bodyEdgeGlobalMap, b, &bodyEdgeGlobalIndexStart));
996c1cad2e7SMatthew G. Knepley 
997*9566063dSJacob Faibussowitsch   PetscCall(EG_getBodyTopos(body, NULL, FACE,  &Nf, &fobjs));
998c1cad2e7SMatthew G. Knepley   for (int f = 0; f < Nf; ++f) {
999c1cad2e7SMatthew G. Knepley     ego   face = fobjs[f];
1000c1cad2e7SMatthew G. Knepley       int   fID, Nl;
1001c1cad2e7SMatthew G. Knepley 
10025f80ce2aSJacob Faibussowitsch     fID  = EG_indexBodyTopo(body, face);
1003c1cad2e7SMatthew G. Knepley 
1004*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, face, LOOP, &Nl, &lobjs));
1005c1cad2e7SMatthew G. Knepley     for (int l = 0; l < Nl; ++l) {
1006c1cad2e7SMatthew G. Knepley         ego  loop = lobjs[l];
1007c1cad2e7SMatthew G. Knepley     int  lid;
1008c1cad2e7SMatthew G. Knepley 
10095f80ce2aSJacob Faibussowitsch     lid  = EG_indexBodyTopo(body, loop);
10102c71b3e2SJacob Faibussowitsch       PetscCheckFalse(Nl > 1,PETSC_COMM_SELF, PETSC_ERR_SUP, "Loop %d has %d > 1 faces, which is not supported", lid, Nf);
1011c1cad2e7SMatthew G. Knepley 
1012*9566063dSJacob Faibussowitsch     PetscCall(EG_getTopology(loop, &geom, &oclass, &mtype, NULL, &Ne, &eobjs, &eSenses));
1013c1cad2e7SMatthew G. Knepley     for (int e = 0; e < Ne; ++e) {
1014c1cad2e7SMatthew G. Knepley       ego     edge = eobjs[e];
1015c1cad2e7SMatthew G. Knepley       int     eid, eOffset;
1016c1cad2e7SMatthew G. Knepley 
1017c1cad2e7SMatthew G. Knepley       // Skip DEGENERATE Edges
1018*9566063dSJacob Faibussowitsch       PetscCall(EG_getInfo(edge, &oclass, &mtype, &topRef, &prev, &next));
1019c1cad2e7SMatthew G. Knepley       if (mtype == DEGENERATE) {continue;}
10205f80ce2aSJacob Faibussowitsch       eid = EG_indexBodyTopo(body, edge);
1021c1cad2e7SMatthew G. Knepley 
1022c1cad2e7SMatthew G. Knepley       // get relative offset from globalEdgeID Vector
1023*9566063dSJacob Faibussowitsch       PetscCall(PetscHMapIFind(edgeMap, bodyEdgeGlobalIndexStart + eid - 1, &EMiter, &EMfound));
102428b400f6SJacob Faibussowitsch       PetscCheck(EMfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Edge %d of Body %d not found in edgeMap. Global Edge ID :: %d", eid, b, bodyEdgeGlobalIndexStart + eid - 1);
1025*9566063dSJacob Faibussowitsch       PetscCall(PetscHMapIGet(edgeMap, bodyEdgeGlobalIndexStart + eid - 1, &eOffset));
1026c1cad2e7SMatthew G. Knepley 
1027*9566063dSJacob Faibussowitsch       PetscCall(EG_getBodyTopos(body, edge, NODE, &Nv, &nobjs));
1028c1cad2e7SMatthew G. Knepley       for (int v = 0; v < Nv; ++v){
1029c1cad2e7SMatthew G. Knepley         ego vertex = nobjs[v];
1030c1cad2e7SMatthew G. Knepley         int vID;
1031c1cad2e7SMatthew G. Knepley 
10325f80ce2aSJacob Faibussowitsch         vID = EG_indexBodyTopo(body, vertex);
1033*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(bodyLabel, nStart + bodyVertexIndexStart + vID - 1, b));
1034*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(vertexLabel, nStart + bodyVertexIndexStart + vID - 1, vID));
1035c1cad2e7SMatthew G. Knepley       }
1036c1cad2e7SMatthew G. Knepley       EG_free(nobjs);
1037c1cad2e7SMatthew G. Knepley 
1038*9566063dSJacob Faibussowitsch       PetscCall(DMLabelSetValue(bodyLabel, nStart + numVertices + bodyEdgeIndexStart + eOffset - 1, b));
1039*9566063dSJacob Faibussowitsch       PetscCall(DMLabelSetValue(edgeLabel, nStart + numVertices + bodyEdgeIndexStart + eOffset - 1, eid));
1040c1cad2e7SMatthew G. Knepley 
1041c1cad2e7SMatthew G. Knepley       // Define Cell faces
1042c1cad2e7SMatthew G. Knepley       for (int jj = 0; jj < 2; ++jj){
1043*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(bodyLabel, cellCntr, b));
1044*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(faceLabel, cellCntr, fID));
1045*9566063dSJacob Faibussowitsch         PetscCall(DMPlexGetCone(dm, cellCntr, &cone));
1046c1cad2e7SMatthew G. Knepley 
1047*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(bodyLabel, cone[0], b));
1048*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(faceLabel, cone[0], fID));
1049c1cad2e7SMatthew G. Knepley 
1050*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(bodyLabel, cone[1], b));
1051*9566063dSJacob Faibussowitsch         PetscCall(DMLabelSetValue(edgeLabel, cone[1], eid));
1052c1cad2e7SMatthew G. Knepley 
1053*9566063dSJacob Faibussowitsch        PetscCall(DMLabelSetValue(bodyLabel, cone[2], b));
1054*9566063dSJacob Faibussowitsch        PetscCall(DMLabelSetValue(faceLabel, cone[2], fID));
1055c1cad2e7SMatthew G. Knepley 
1056c1cad2e7SMatthew G. Knepley        cellCntr = cellCntr + 1;
1057c1cad2e7SMatthew G. Knepley       }
1058c1cad2e7SMatthew G. Knepley     }
1059c1cad2e7SMatthew G. Knepley     }
1060c1cad2e7SMatthew G. Knepley     EG_free(lobjs);
1061c1cad2e7SMatthew G. Knepley 
1062*9566063dSJacob Faibussowitsch     PetscCall(DMLabelSetValue(bodyLabel, nStart + numVertices + numEdges + bodyFaceIndexStart + fID - 1, b));
1063*9566063dSJacob Faibussowitsch     PetscCall(DMLabelSetValue(faceLabel, nStart + numVertices + numEdges + bodyFaceIndexStart + fID - 1, fID));
1064c1cad2e7SMatthew G. Knepley   }
1065c1cad2e7SMatthew G. Knepley   EG_free(fobjs);
1066c1cad2e7SMatthew G. Knepley   }
1067c1cad2e7SMatthew G. Knepley 
1068*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIDestroy(&edgeMap));
1069*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIDestroy(&bodyIndexMap));
1070*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIDestroy(&bodyVertexMap));
1071*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIDestroy(&bodyEdgeMap));
1072*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIDestroy(&bodyEdgeGlobalMap));
1073*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIDestroy(&bodyFaceMap));
1074c1cad2e7SMatthew G. Knepley 
1075c1cad2e7SMatthew G. Knepley   *newdm = dm;
1076c1cad2e7SMatthew G. Knepley   PetscFunctionReturn(0);
1077c1cad2e7SMatthew G. Knepley }
1078c1cad2e7SMatthew G. Knepley 
1079c1cad2e7SMatthew G. Knepley static PetscErrorCode DMPlexCreateEGADS_Tess_Internal(MPI_Comm comm, ego context, ego model, DM *newdm)
1080c1cad2e7SMatthew G. Knepley {
1081c1cad2e7SMatthew G. Knepley   DMLabel              bodyLabel, faceLabel, edgeLabel, vertexLabel;
1082c1cad2e7SMatthew G. Knepley   /* EGADSLite variables */
1083c1cad2e7SMatthew G. Knepley   ego                  geom, *bodies, *fobjs;
1084c1cad2e7SMatthew G. Knepley   int                  b, oclass, mtype, nbodies, *senses;
1085c1cad2e7SMatthew G. Knepley   int                  totalNumTris = 0, totalNumPoints = 0;
1086c1cad2e7SMatthew G. Knepley   double               boundBox[6] = {0., 0., 0., 0., 0., 0.}, tessSize;
1087c1cad2e7SMatthew G. Knepley   /* PETSc variables */
1088c1cad2e7SMatthew G. Knepley   DM                   dm;
1089c1cad2e7SMatthew G. Knepley   PetscHMapI           pointIndexStartMap = NULL, triIndexStartMap = NULL, pTypeLabelMap = NULL, pIndexLabelMap = NULL;
1090c1cad2e7SMatthew G. Knepley   PetscHMapI           pBodyIndexLabelMap = NULL, triFaceIDLabelMap = NULL, triBodyIDLabelMap = NULL;
1091c1cad2e7SMatthew G. Knepley   PetscInt             dim = -1, cdim = -1, numCorners = 0, counter = 0;
1092c1cad2e7SMatthew G. Knepley   PetscInt            *cells  = NULL;
1093c1cad2e7SMatthew G. Knepley   const PetscInt      *cone = NULL;
1094c1cad2e7SMatthew G. Knepley   PetscReal           *coords = NULL;
1095c1cad2e7SMatthew G. Knepley   PetscMPIInt          rank;
1096c1cad2e7SMatthew G. Knepley 
1097c1cad2e7SMatthew G. Knepley   PetscFunctionBeginUser;
1098*9566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
1099c1cad2e7SMatthew G. Knepley   if (!rank) {
1100c1cad2e7SMatthew G. Knepley     // ---------------------------------------------------------------------------------------------------
1101c1cad2e7SMatthew G. Knepley     // Generate Petsc Plex from EGADSlite created Tessellation of geometry
1102c1cad2e7SMatthew G. Knepley     // ---------------------------------------------------------------------------------------------------
1103c1cad2e7SMatthew G. Knepley 
1104c1cad2e7SMatthew G. Knepley   // Caluculate cell and vertex sizes
1105*9566063dSJacob Faibussowitsch   PetscCall(EG_getTopology(model, &geom, &oclass, &mtype, NULL, &nbodies, &bodies, &senses));
1106c1cad2e7SMatthew G. Knepley 
1107*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&pointIndexStartMap));
1108*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&triIndexStartMap));
1109*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&pTypeLabelMap));
1110*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&pIndexLabelMap));
1111*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&pBodyIndexLabelMap));
1112*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&triFaceIDLabelMap));
1113*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapICreate(&triBodyIDLabelMap));
1114c1cad2e7SMatthew G. Knepley 
1115c1cad2e7SMatthew G. Knepley   /* Create Tessellation of Bodies */
1116c1cad2e7SMatthew G. Knepley   ego tessArray[nbodies];
1117c1cad2e7SMatthew G. Knepley 
1118c1cad2e7SMatthew G. Knepley   for (b = 0; b < nbodies; ++b) {
1119c1cad2e7SMatthew G. Knepley     ego             body = bodies[b];
1120c1cad2e7SMatthew G. Knepley     double          params[3] = {0.0, 0.0, 0.0};    // Parameters for Tessellation
1121c1cad2e7SMatthew G. Knepley     int             Nf, bodyNumPoints = 0, bodyNumTris = 0;
1122c1cad2e7SMatthew G. Knepley     PetscHashIter   PISiter, TISiter;
1123c1cad2e7SMatthew G. Knepley     PetscBool       PISfound, TISfound;
1124c1cad2e7SMatthew G. Knepley 
1125c1cad2e7SMatthew G. Knepley     /* Store Start Index for each Body's Point and Tris */
1126*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(pointIndexStartMap, b, &PISiter, &PISfound));
1127*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(triIndexStartMap, b, &TISiter, &TISfound));
1128c1cad2e7SMatthew G. Knepley 
1129*9566063dSJacob Faibussowitsch     if (!PISfound)  PetscCall(PetscHMapISet(pointIndexStartMap, b, totalNumPoints));
1130*9566063dSJacob Faibussowitsch     if (!TISfound)  PetscCall(PetscHMapISet(triIndexStartMap, b, totalNumTris));
1131c1cad2e7SMatthew G. Knepley 
1132c1cad2e7SMatthew G. Knepley     /* Calculate Tessellation parameters based on Bounding Box */
1133c1cad2e7SMatthew G. Knepley     /* Get Bounding Box Dimensions of the BODY */
1134*9566063dSJacob Faibussowitsch     PetscCall(EG_getBoundingBox(body, boundBox));
1135c1cad2e7SMatthew G. Knepley     tessSize = boundBox[3] - boundBox[0];
1136c1cad2e7SMatthew G. Knepley     if (tessSize < boundBox[4] - boundBox[1]) tessSize = boundBox[4] - boundBox[1];
1137c1cad2e7SMatthew G. Knepley     if (tessSize < boundBox[5] - boundBox[2]) tessSize = boundBox[5] - boundBox[2];
1138c1cad2e7SMatthew G. Knepley 
1139c1cad2e7SMatthew G. Knepley     // TODO :: May want to give users tessellation parameter options //
1140c1cad2e7SMatthew G. Knepley     params[0] = 0.0250 * tessSize;
1141c1cad2e7SMatthew G. Knepley     params[1] = 0.0075 * tessSize;
1142c1cad2e7SMatthew G. Knepley     params[2] = 15.0;
1143c1cad2e7SMatthew G. Knepley 
1144*9566063dSJacob Faibussowitsch     PetscCall(EG_makeTessBody(body, params, &tessArray[b]));
1145c1cad2e7SMatthew G. Knepley 
1146*9566063dSJacob Faibussowitsch     PetscCall(EG_getBodyTopos(body, NULL, FACE,  &Nf, &fobjs));
1147c1cad2e7SMatthew G. Knepley 
1148c1cad2e7SMatthew G. Knepley     for (int f = 0; f < Nf; ++f) {
1149c1cad2e7SMatthew G. Knepley       ego             face = fobjs[f];
1150c1cad2e7SMatthew G. Knepley     int             len, fID, ntris;
1151c1cad2e7SMatthew G. Knepley     const int      *ptype, *pindex, *ptris, *ptric;
1152c1cad2e7SMatthew G. Knepley     const double   *pxyz, *puv;
1153c1cad2e7SMatthew G. Knepley 
1154c1cad2e7SMatthew G. Knepley     // Get Face ID //
1155c1cad2e7SMatthew G. Knepley     fID = EG_indexBodyTopo(body, face);
1156c1cad2e7SMatthew G. Knepley 
1157c1cad2e7SMatthew G. Knepley     // Checkout the Surface Tessellation //
1158*9566063dSJacob Faibussowitsch     PetscCall(EG_getTessFace(tessArray[b], fID, &len, &pxyz, &puv, &ptype, &pindex, &ntris, &ptris, &ptric));
1159c1cad2e7SMatthew G. Knepley 
1160c1cad2e7SMatthew G. Knepley     // Determine total number of triangle cells in the tessellation //
1161c1cad2e7SMatthew G. Knepley     bodyNumTris += (int) ntris;
1162c1cad2e7SMatthew G. Knepley 
1163c1cad2e7SMatthew G. Knepley     // Check out the point index and coordinate //
1164c1cad2e7SMatthew G. Knepley     for (int p = 0; p < len; ++p) {
1165c1cad2e7SMatthew G. Knepley         int global;
1166c1cad2e7SMatthew G. Knepley 
1167*9566063dSJacob Faibussowitsch         PetscCall(EG_localToGlobal(tessArray[b], fID, p+1, &global));
1168c1cad2e7SMatthew G. Knepley 
1169c1cad2e7SMatthew G. Knepley       // Determine the total number of points in the tessellation //
1170c1cad2e7SMatthew G. Knepley         bodyNumPoints = PetscMax(bodyNumPoints, global);
1171c1cad2e7SMatthew G. Knepley     }
1172c1cad2e7SMatthew G. Knepley     }
1173c1cad2e7SMatthew G. Knepley     EG_free(fobjs);
1174c1cad2e7SMatthew G. Knepley 
1175c1cad2e7SMatthew G. Knepley     totalNumPoints += bodyNumPoints;
1176c1cad2e7SMatthew G. Knepley     totalNumTris += bodyNumTris;
1177c1cad2e7SMatthew G. Knepley     }
1178c1cad2e7SMatthew G. Knepley   //}  - Original End of (!rank)
1179c1cad2e7SMatthew G. Knepley 
1180c1cad2e7SMatthew G. Knepley   dim = 2;
1181c1cad2e7SMatthew G. Knepley   cdim = 3;
1182c1cad2e7SMatthew G. Knepley   numCorners = 3;
1183c1cad2e7SMatthew G. Knepley   //PetscInt counter = 0;
1184c1cad2e7SMatthew G. Knepley 
1185c1cad2e7SMatthew G. Knepley   /* NEED TO DEFINE MATRICES/VECTORS TO STORE GEOM REFERENCE DATA   */
1186c1cad2e7SMatthew G. Knepley   /* Fill in below and use to define DMLabels after DMPlex creation */
1187*9566063dSJacob Faibussowitsch   PetscCall(PetscMalloc2(totalNumPoints*cdim, &coords, totalNumTris*numCorners, &cells));
1188c1cad2e7SMatthew G. Knepley 
1189c1cad2e7SMatthew G. Knepley   for (b = 0; b < nbodies; ++b) {
1190c1cad2e7SMatthew G. Knepley   ego             body = bodies[b];
1191c1cad2e7SMatthew G. Knepley   int             Nf;
1192c1cad2e7SMatthew G. Knepley   PetscInt        pointIndexStart;
1193c1cad2e7SMatthew G. Knepley   PetscHashIter   PISiter;
1194c1cad2e7SMatthew G. Knepley   PetscBool       PISfound;
1195c1cad2e7SMatthew G. Knepley 
1196*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIFind(pointIndexStartMap, b, &PISiter, &PISfound));
119728b400f6SJacob Faibussowitsch   PetscCheck(PISfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "Body %d not found in pointIndexStartMap", b);
1198*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIGet(pointIndexStartMap, b, &pointIndexStart));
1199c1cad2e7SMatthew G. Knepley 
1200*9566063dSJacob Faibussowitsch   PetscCall(EG_getBodyTopos(body, NULL, FACE,  &Nf, &fobjs));
1201c1cad2e7SMatthew G. Knepley 
1202c1cad2e7SMatthew G. Knepley   for (int f = 0; f < Nf; ++f) {
1203c1cad2e7SMatthew G. Knepley     /* Get Face Object */
1204c1cad2e7SMatthew G. Knepley     ego              face = fobjs[f];
1205c1cad2e7SMatthew G. Knepley     int              len, fID, ntris;
1206c1cad2e7SMatthew G. Knepley     const int       *ptype, *pindex, *ptris, *ptric;
1207c1cad2e7SMatthew G. Knepley     const double    *pxyz, *puv;
1208c1cad2e7SMatthew G. Knepley 
1209c1cad2e7SMatthew G. Knepley     /* Get Face ID */
1210c1cad2e7SMatthew G. Knepley     fID = EG_indexBodyTopo(body, face);
1211c1cad2e7SMatthew G. Knepley 
1212c1cad2e7SMatthew G. Knepley     /* Checkout the Surface Tessellation */
1213*9566063dSJacob Faibussowitsch     PetscCall(EG_getTessFace(tessArray[b], fID, &len, &pxyz, &puv, &ptype, &pindex, &ntris, &ptris, &ptric));
1214c1cad2e7SMatthew G. Knepley 
1215c1cad2e7SMatthew G. Knepley     /* Check out the point index and coordinate */
1216c1cad2e7SMatthew G. Knepley     for (int p = 0; p < len; ++p) {
1217c1cad2e7SMatthew G. Knepley     int              global;
1218c1cad2e7SMatthew G. Knepley     PetscHashIter    PTLiter, PILiter, PBLiter;
1219c1cad2e7SMatthew G. Knepley     PetscBool        PTLfound, PILfound, PBLfound;
1220c1cad2e7SMatthew G. Knepley 
1221*9566063dSJacob Faibussowitsch     PetscCall(EG_localToGlobal(tessArray[b], fID, p+1, &global));
1222c1cad2e7SMatthew G. Knepley 
1223c1cad2e7SMatthew G. Knepley     /* Set the coordinates array for DAG */
1224c1cad2e7SMatthew G. Knepley     coords[((global-1+pointIndexStart)*3) + 0] = pxyz[(p*3)+0];
1225c1cad2e7SMatthew G. Knepley     coords[((global-1+pointIndexStart)*3) + 1] = pxyz[(p*3)+1];
1226c1cad2e7SMatthew G. Knepley     coords[((global-1+pointIndexStart)*3) + 2] = pxyz[(p*3)+2];
1227c1cad2e7SMatthew G. Knepley 
1228c1cad2e7SMatthew G. Knepley     /* Store Geometry Label Information for DMLabel assignment later */
1229*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(pTypeLabelMap, global-1+pointIndexStart, &PTLiter, &PTLfound));
1230*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(pIndexLabelMap, global-1+pointIndexStart, &PILiter, &PILfound));
1231*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(pBodyIndexLabelMap, global-1+pointIndexStart, &PBLiter, &PBLfound));
1232c1cad2e7SMatthew G. Knepley 
1233*9566063dSJacob Faibussowitsch     if (!PTLfound) PetscCall(PetscHMapISet(pTypeLabelMap, global-1+pointIndexStart, ptype[p]));
1234*9566063dSJacob Faibussowitsch     if (!PILfound) PetscCall(PetscHMapISet(pIndexLabelMap, global-1+pointIndexStart, pindex[p]));
1235*9566063dSJacob Faibussowitsch     if (!PBLfound) PetscCall(PetscHMapISet(pBodyIndexLabelMap, global-1+pointIndexStart, b));
1236c1cad2e7SMatthew G. Knepley 
1237*9566063dSJacob Faibussowitsch     if (ptype[p] < 0) PetscCall(PetscHMapISet(pIndexLabelMap, global-1+pointIndexStart, fID));
1238c1cad2e7SMatthew G. Knepley     }
1239c1cad2e7SMatthew G. Knepley 
1240c1cad2e7SMatthew G. Knepley     for (int t = 0; t < (int) ntris; ++t){
1241c1cad2e7SMatthew G. Knepley     int           global, globalA, globalB;
1242c1cad2e7SMatthew G. Knepley     PetscHashIter TFLiter, TBLiter;
1243c1cad2e7SMatthew G. Knepley     PetscBool     TFLfound, TBLfound;
1244c1cad2e7SMatthew G. Knepley 
1245*9566063dSJacob Faibussowitsch     PetscCall(EG_localToGlobal(tessArray[b], fID, ptris[(t*3) + 0], &global));
1246c1cad2e7SMatthew G. Knepley     cells[(counter*3) +0] = global-1+pointIndexStart;
1247c1cad2e7SMatthew G. Knepley 
1248*9566063dSJacob Faibussowitsch     PetscCall(EG_localToGlobal(tessArray[b], fID, ptris[(t*3) + 1], &globalA));
1249c1cad2e7SMatthew G. Knepley     cells[(counter*3) +1] = globalA-1+pointIndexStart;
1250c1cad2e7SMatthew G. Knepley 
1251*9566063dSJacob Faibussowitsch     PetscCall(EG_localToGlobal(tessArray[b], fID, ptris[(t*3) + 2], &globalB));
1252c1cad2e7SMatthew G. Knepley     cells[(counter*3) +2] = globalB-1+pointIndexStart;
1253c1cad2e7SMatthew G. Knepley 
1254*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(triFaceIDLabelMap, counter, &TFLiter, &TFLfound));
1255*9566063dSJacob Faibussowitsch         PetscCall(PetscHMapIFind(triBodyIDLabelMap, counter, &TBLiter, &TBLfound));
1256c1cad2e7SMatthew G. Knepley 
1257*9566063dSJacob Faibussowitsch     if (!TFLfound)  PetscCall(PetscHMapISet(triFaceIDLabelMap, counter, fID));
1258*9566063dSJacob Faibussowitsch         if (!TBLfound)  PetscCall(PetscHMapISet(triBodyIDLabelMap, counter, b));
1259c1cad2e7SMatthew G. Knepley 
1260c1cad2e7SMatthew G. Knepley     counter += 1;
1261c1cad2e7SMatthew G. Knepley     }
1262c1cad2e7SMatthew G. Knepley   }
1263c1cad2e7SMatthew G. Knepley   EG_free(fobjs);
1264c1cad2e7SMatthew G. Knepley   }
1265c1cad2e7SMatthew G. Knepley }
1266c1cad2e7SMatthew G. Knepley 
1267c1cad2e7SMatthew G. Knepley   //Build DMPlex
1268*9566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateFromCellListPetsc(PETSC_COMM_WORLD, dim, totalNumTris, totalNumPoints, numCorners, PETSC_TRUE, cells, cdim, coords, &dm));
1269*9566063dSJacob Faibussowitsch   PetscCall(PetscFree2(coords, cells));
1270c1cad2e7SMatthew G. Knepley 
1271c1cad2e7SMatthew G. Knepley   // Embed EGADS model in DM
1272c1cad2e7SMatthew G. Knepley   {
1273c1cad2e7SMatthew G. Knepley     PetscContainer modelObj, contextObj;
1274c1cad2e7SMatthew G. Knepley 
1275*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &modelObj));
1276*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerSetPointer(modelObj, model));
1277*9566063dSJacob Faibussowitsch     PetscCall(PetscObjectCompose((PetscObject) dm, "EGADS Model", (PetscObject) modelObj));
1278*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerDestroy(&modelObj));
1279c1cad2e7SMatthew G. Knepley 
1280*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerCreate(PETSC_COMM_SELF, &contextObj));
1281*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerSetPointer(contextObj, context));
1282*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerSetUserDestroy(contextObj, DMPlexEGADSDestroy_Private));
1283*9566063dSJacob Faibussowitsch     PetscCall(PetscObjectCompose((PetscObject) dm, "EGADS Context", (PetscObject) contextObj));
1284*9566063dSJacob Faibussowitsch     PetscCall(PetscContainerDestroy(&contextObj));
1285c1cad2e7SMatthew G. Knepley   }
1286c1cad2e7SMatthew G. Knepley 
1287c1cad2e7SMatthew G. Knepley   // Label Points
1288*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Body ID"));
1289*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Body ID", &bodyLabel));
1290*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Face ID"));
1291*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Face ID", &faceLabel));
1292*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Edge ID"));
1293*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Edge ID", &edgeLabel));
1294*9566063dSJacob Faibussowitsch   PetscCall(DMCreateLabel(dm, "EGADS Vertex ID"));
1295*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Vertex ID", &vertexLabel));
1296c1cad2e7SMatthew G. Knepley 
1297c1cad2e7SMatthew G. Knepley    /* Get Number of DAG Nodes at each level */
1298c1cad2e7SMatthew G. Knepley   int   fStart, fEnd, eStart, eEnd, nStart, nEnd;
1299c1cad2e7SMatthew G. Knepley 
1300*9566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm, 0, &fStart, &fEnd));
1301*9566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm, 1, &eStart, &eEnd));
1302*9566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm, 2, &nStart, &nEnd));
1303c1cad2e7SMatthew G. Knepley 
1304c1cad2e7SMatthew G. Knepley   /* Set DMLabels for NODES */
1305c1cad2e7SMatthew G. Knepley   for (int n = nStart; n < nEnd; ++n) {
1306c1cad2e7SMatthew G. Knepley     int             pTypeVal, pIndexVal, pBodyVal;
1307c1cad2e7SMatthew G. Knepley     PetscHashIter   PTLiter, PILiter, PBLiter;
1308c1cad2e7SMatthew G. Knepley     PetscBool       PTLfound, PILfound, PBLfound;
1309c1cad2e7SMatthew G. Knepley 
1310c1cad2e7SMatthew G. Knepley     //Converted to Hash Tables
1311*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(pTypeLabelMap, n - nStart, &PTLiter, &PTLfound));
131228b400f6SJacob Faibussowitsch     PetscCheck(PTLfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "DAG Point %d not found in pTypeLabelMap", n);
1313*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIGet(pTypeLabelMap, n - nStart, &pTypeVal));
1314c1cad2e7SMatthew G. Knepley 
1315*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(pIndexLabelMap, n - nStart, &PILiter, &PILfound));
131628b400f6SJacob Faibussowitsch     PetscCheck(PILfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "DAG Point %d not found in pIndexLabelMap", n);
1317*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIGet(pIndexLabelMap, n - nStart, &pIndexVal));
1318c1cad2e7SMatthew G. Knepley 
1319*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIFind(pBodyIndexLabelMap, n - nStart, &PBLiter, &PBLfound));
132028b400f6SJacob Faibussowitsch     PetscCheck(PBLfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "DAG Point %d not found in pBodyLabelMap", n);
1321*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIGet(pBodyIndexLabelMap, n - nStart, &pBodyVal));
1322c1cad2e7SMatthew G. Knepley 
1323*9566063dSJacob Faibussowitsch     PetscCall(DMLabelSetValue(bodyLabel, n, pBodyVal));
1324*9566063dSJacob Faibussowitsch     if (pTypeVal == 0) PetscCall(DMLabelSetValue(vertexLabel, n, pIndexVal));
1325*9566063dSJacob Faibussowitsch     if (pTypeVal >  0) PetscCall(DMLabelSetValue(edgeLabel, n, pIndexVal));
1326*9566063dSJacob Faibussowitsch     if (pTypeVal <  0) PetscCall(DMLabelSetValue(faceLabel, n, pIndexVal));
1327c1cad2e7SMatthew G. Knepley   }
1328c1cad2e7SMatthew G. Knepley 
1329c1cad2e7SMatthew G. Knepley   /* Set DMLabels for Edges - Based on the DMLabels of the EDGE's NODES */
1330c1cad2e7SMatthew G. Knepley   for (int e = eStart; e < eEnd; ++e) {
1331c1cad2e7SMatthew G. Knepley   int    bodyID_0, vertexID_0, vertexID_1, edgeID_0, edgeID_1, faceID_0, faceID_1;
1332c1cad2e7SMatthew G. Knepley 
1333*9566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCone(dm, e, &cone));
1334*9566063dSJacob Faibussowitsch   PetscCall(DMLabelGetValue(bodyLabel, cone[0], &bodyID_0));    // Do I need to check the other end?
1335*9566063dSJacob Faibussowitsch   PetscCall(DMLabelGetValue(vertexLabel, cone[0], &vertexID_0));
1336*9566063dSJacob Faibussowitsch   PetscCall(DMLabelGetValue(vertexLabel, cone[1], &vertexID_1));
1337*9566063dSJacob Faibussowitsch   PetscCall(DMLabelGetValue(edgeLabel, cone[0], &edgeID_0));
1338*9566063dSJacob Faibussowitsch   PetscCall(DMLabelGetValue(edgeLabel, cone[1], &edgeID_1));
1339*9566063dSJacob Faibussowitsch   PetscCall(DMLabelGetValue(faceLabel, cone[0], &faceID_0));
1340*9566063dSJacob Faibussowitsch   PetscCall(DMLabelGetValue(faceLabel, cone[1], &faceID_1));
1341c1cad2e7SMatthew G. Knepley 
1342*9566063dSJacob Faibussowitsch   PetscCall(DMLabelSetValue(bodyLabel, e, bodyID_0));
1343c1cad2e7SMatthew G. Knepley 
1344*9566063dSJacob Faibussowitsch   if (edgeID_0 == edgeID_1) PetscCall(DMLabelSetValue(edgeLabel, e, edgeID_0));
1345*9566063dSJacob Faibussowitsch   else if (vertexID_0 > 0 && edgeID_1 > 0) PetscCall(DMLabelSetValue(edgeLabel, e, edgeID_1));
1346*9566063dSJacob Faibussowitsch   else if (vertexID_1 > 0 && edgeID_0 > 0) PetscCall(DMLabelSetValue(edgeLabel, e, edgeID_0));
1347c1cad2e7SMatthew G. Knepley   else { /* Do Nothing */ }
1348c1cad2e7SMatthew G. Knepley   }
1349c1cad2e7SMatthew G. Knepley 
1350c1cad2e7SMatthew G. Knepley   /* Set DMLabels for Cells */
1351c1cad2e7SMatthew G. Knepley   for (int f = fStart; f < fEnd; ++f){
1352c1cad2e7SMatthew G. Knepley   int             edgeID_0;
1353c1cad2e7SMatthew G. Knepley   PetscInt        triBodyVal, triFaceVal;
1354c1cad2e7SMatthew G. Knepley   PetscHashIter   TFLiter, TBLiter;
1355c1cad2e7SMatthew G. Knepley   PetscBool       TFLfound, TBLfound;
1356c1cad2e7SMatthew G. Knepley 
1357c1cad2e7SMatthew G. Knepley     // Convert to Hash Table
1358*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIFind(triFaceIDLabelMap, f - fStart, &TFLiter, &TFLfound));
135928b400f6SJacob Faibussowitsch   PetscCheck(TFLfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "DAG Point %d not found in triFaceIDLabelMap", f);
1360*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIGet(triFaceIDLabelMap, f - fStart, &triFaceVal));
1361c1cad2e7SMatthew G. Knepley 
1362*9566063dSJacob Faibussowitsch   PetscCall(PetscHMapIFind(triBodyIDLabelMap, f - fStart, &TBLiter, &TBLfound));
136328b400f6SJacob Faibussowitsch   PetscCheck(TBLfound,PETSC_COMM_SELF, PETSC_ERR_SUP, "DAG Point %d not found in triBodyIDLabelMap", f);
1364*9566063dSJacob Faibussowitsch     PetscCall(PetscHMapIGet(triBodyIDLabelMap, f - fStart, &triBodyVal));
1365c1cad2e7SMatthew G. Knepley 
1366*9566063dSJacob Faibussowitsch   PetscCall(DMLabelSetValue(bodyLabel, f, triBodyVal));
1367*9566063dSJacob Faibussowitsch   PetscCall(DMLabelSetValue(faceLabel, f, triFaceVal));
1368c1cad2e7SMatthew G. Knepley 
1369c1cad2e7SMatthew G. Knepley   /* Finish Labeling previously unlabeled DMPlex Edges - Assumes Triangular Cell (3 Edges Max) */
1370*9566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCone(dm, f, &cone));
1371c1cad2e7SMatthew G. Knepley 
1372c1cad2e7SMatthew G. Knepley   for (int jj = 0; jj < 3; ++jj) {
1373*9566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(edgeLabel, cone[jj], &edgeID_0));
1374c1cad2e7SMatthew G. Knepley 
1375c1cad2e7SMatthew G. Knepley     if (edgeID_0 < 0) {
1376*9566063dSJacob Faibussowitsch     PetscCall(DMLabelSetValue(bodyLabel, cone[jj], triBodyVal));
1377*9566063dSJacob Faibussowitsch       PetscCall(DMLabelSetValue(faceLabel, cone[jj], triFaceVal));
1378c1cad2e7SMatthew G. Knepley     }
1379c1cad2e7SMatthew G. Knepley   }
1380c1cad2e7SMatthew G. Knepley   }
1381c1cad2e7SMatthew G. Knepley 
1382c1cad2e7SMatthew G. Knepley   *newdm = dm;
1383c1cad2e7SMatthew G. Knepley   PetscFunctionReturn(0);
1384c1cad2e7SMatthew G. Knepley }
13857bee2925SMatthew Knepley #endif
13867bee2925SMatthew Knepley 
1387c1cad2e7SMatthew G. Knepley /*@
1388c1cad2e7SMatthew G. Knepley   DMPlexInflateToGeomModel - Snaps the vertex coordinates of a DMPlex object representing the mesh to its geometry if some vertices depart from the model. This usually happens with non-conforming refinement.
1389c1cad2e7SMatthew G. Knepley 
1390c1cad2e7SMatthew G. Knepley   Collective on dm
1391c1cad2e7SMatthew G. Knepley 
1392c1cad2e7SMatthew G. Knepley   Input Parameter:
1393c1cad2e7SMatthew G. Knepley . dm - The uninflated DM object representing the mesh
1394c1cad2e7SMatthew G. Knepley 
1395c1cad2e7SMatthew G. Knepley   Output Parameter:
1396c1cad2e7SMatthew G. Knepley . dm - The inflated DM object representing the mesh
1397c1cad2e7SMatthew G. Knepley 
1398c1cad2e7SMatthew G. Knepley   Level: intermediate
1399c1cad2e7SMatthew G. Knepley 
1400c1cad2e7SMatthew G. Knepley .seealso: DMPLEX, DMCreate(), DMPlexCreateEGADS()
1401c1cad2e7SMatthew G. Knepley @*/
1402c1cad2e7SMatthew G. Knepley PetscErrorCode DMPlexInflateToGeomModel(DM dm)
1403c1cad2e7SMatthew G. Knepley {
1404c1cad2e7SMatthew G. Knepley #if defined(PETSC_HAVE_EGADS)
1405c1cad2e7SMatthew G. Knepley   /* EGADS Variables */
1406c1cad2e7SMatthew G. Knepley   ego            model, geom, body, face, edge;
1407c1cad2e7SMatthew G. Knepley   ego           *bodies;
1408c1cad2e7SMatthew G. Knepley   int            Nb, oclass, mtype, *senses;
1409c1cad2e7SMatthew G. Knepley   double         result[3];
1410c1cad2e7SMatthew G. Knepley   /* PETSc Variables */
1411c1cad2e7SMatthew G. Knepley   DM             cdm;
1412c1cad2e7SMatthew G. Knepley   PetscContainer modelObj;
1413c1cad2e7SMatthew G. Knepley   DMLabel        bodyLabel, faceLabel, edgeLabel, vertexLabel;
1414c1cad2e7SMatthew G. Knepley   Vec            coordinates;
1415c1cad2e7SMatthew G. Knepley   PetscScalar   *coords;
1416c1cad2e7SMatthew G. Knepley   PetscInt       bodyID, faceID, edgeID, vertexID;
1417c1cad2e7SMatthew G. Knepley   PetscInt       cdim, d, vStart, vEnd, v;
1418c1cad2e7SMatthew G. Knepley #endif
1419c1cad2e7SMatthew G. Knepley 
1420c1cad2e7SMatthew G. Knepley   PetscFunctionBegin;
1421c1cad2e7SMatthew G. Knepley #if defined(PETSC_HAVE_EGADS)
1422*9566063dSJacob Faibussowitsch   PetscCall(PetscObjectQuery((PetscObject) dm, "EGADS Model", (PetscObject *) &modelObj));
1423c1cad2e7SMatthew G. Knepley   if (!modelObj) PetscFunctionReturn(0);
1424*9566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm, &cdim));
1425*9566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDM(dm, &cdm));
1426*9566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dm, &coordinates));
1427*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Body ID", &bodyLabel));
1428*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Face ID", &faceLabel));
1429*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Edge ID", &edgeLabel));
1430*9566063dSJacob Faibussowitsch   PetscCall(DMGetLabel(dm, "EGADS Vertex ID", &vertexLabel));
1431c1cad2e7SMatthew G. Knepley 
1432*9566063dSJacob Faibussowitsch   PetscCall(PetscContainerGetPointer(modelObj, (void **) &model));
1433*9566063dSJacob Faibussowitsch   PetscCall(EG_getTopology(model, &geom, &oclass, &mtype, NULL, &Nb, &bodies, &senses));
1434c1cad2e7SMatthew G. Knepley 
1435*9566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd));
1436*9566063dSJacob Faibussowitsch   PetscCall(VecGetArrayWrite(coordinates, &coords));
1437c1cad2e7SMatthew G. Knepley   for (v = vStart; v < vEnd; ++v) {
1438c1cad2e7SMatthew G. Knepley     PetscScalar *vcoords;
1439c1cad2e7SMatthew G. Knepley 
1440*9566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(bodyLabel, v, &bodyID));
1441*9566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(faceLabel, v, &faceID));
1442*9566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(edgeLabel, v, &edgeID));
1443*9566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(vertexLabel, v, &vertexID));
1444c1cad2e7SMatthew G. Knepley 
14452c71b3e2SJacob Faibussowitsch     PetscCheckFalse(bodyID >= Nb,PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Body %D is not in [0, %d)", bodyID, Nb);
1446c1cad2e7SMatthew G. Knepley     body = bodies[bodyID];
1447c1cad2e7SMatthew G. Knepley 
1448*9566063dSJacob Faibussowitsch     PetscCall(DMPlexPointLocalRef(cdm, v, coords, (void *) &vcoords));
1449c1cad2e7SMatthew G. Knepley     if (edgeID > 0) {
1450c1cad2e7SMatthew G. Knepley       /* Snap to EDGE at nearest location */
1451c1cad2e7SMatthew G. Knepley       double params[1];
1452*9566063dSJacob Faibussowitsch       PetscCall(EG_objectBodyTopo(body, EDGE, edgeID, &edge));
1453*9566063dSJacob Faibussowitsch       PetscCall(EG_invEvaluate(edge, vcoords, params, result)); // Get (x,y,z) of nearest point on EDGE
1454c1cad2e7SMatthew G. Knepley       for (d = 0; d < cdim; ++d) vcoords[d] = result[d];
1455c1cad2e7SMatthew G. Knepley     } else if (faceID > 0) {
1456c1cad2e7SMatthew G. Knepley       /* Snap to FACE at nearest location */
1457c1cad2e7SMatthew G. Knepley       double params[2];
1458*9566063dSJacob Faibussowitsch       PetscCall(EG_objectBodyTopo(body, FACE, faceID, &face));
1459*9566063dSJacob Faibussowitsch       PetscCall(EG_invEvaluate(face, vcoords, params, result)); // Get (x,y,z) of nearest point on FACE
1460c1cad2e7SMatthew G. Knepley       for (d = 0; d < cdim; ++d) vcoords[d] = result[d];
1461c1cad2e7SMatthew G. Knepley     }
1462c1cad2e7SMatthew G. Knepley   }
1463*9566063dSJacob Faibussowitsch   PetscCall(VecRestoreArrayWrite(coordinates, &coords));
1464c1cad2e7SMatthew G. Knepley   /* Clear out global coordinates */
1465*9566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&dm->coordinates));
1466c1cad2e7SMatthew G. Knepley #endif
1467c1cad2e7SMatthew G. Knepley   PetscFunctionReturn(0);
1468c1cad2e7SMatthew G. Knepley }
1469c1cad2e7SMatthew G. Knepley 
14707bee2925SMatthew Knepley /*@C
1471c1cad2e7SMatthew G. Knepley   DMPlexCreateEGADSFromFile - Create a DMPlex mesh from an EGADS, IGES, or STEP file.
14727bee2925SMatthew Knepley 
14737bee2925SMatthew Knepley   Collective
14747bee2925SMatthew Knepley 
14757bee2925SMatthew Knepley   Input Parameters:
14767bee2925SMatthew Knepley + comm     - The MPI communicator
1477c1cad2e7SMatthew G. Knepley - filename - The name of the EGADS, IGES, or STEP file
14787bee2925SMatthew Knepley 
14797bee2925SMatthew Knepley   Output Parameter:
14807bee2925SMatthew Knepley . dm       - The DM object representing the mesh
14817bee2925SMatthew Knepley 
14827bee2925SMatthew Knepley   Level: beginner
14837bee2925SMatthew Knepley 
1484c1cad2e7SMatthew G. Knepley .seealso: DMPLEX, DMCreate(), DMPlexCreateEGADS(), DMPlexCreateEGADSLiteFromFile()
14857bee2925SMatthew Knepley @*/
14867bee2925SMatthew Knepley PetscErrorCode DMPlexCreateEGADSFromFile(MPI_Comm comm, const char filename[], DM *dm)
14877bee2925SMatthew Knepley {
14887bee2925SMatthew Knepley   PetscMPIInt    rank;
14897bee2925SMatthew Knepley #if defined(PETSC_HAVE_EGADS)
14907bee2925SMatthew Knepley   ego            context= NULL, model = NULL;
14917bee2925SMatthew Knepley #endif
1492c1cad2e7SMatthew G. Knepley   PetscBool      printModel = PETSC_FALSE, tessModel = PETSC_FALSE, newModel = PETSC_FALSE;
14937bee2925SMatthew Knepley 
14947bee2925SMatthew Knepley   PetscFunctionBegin;
14957bee2925SMatthew Knepley   PetscValidCharPointer(filename, 2);
1496*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(NULL, NULL, "-dm_plex_egads_print_model", &printModel, NULL));
1497*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(NULL, NULL, "-dm_plex_egads_tess_model", &tessModel, NULL));
1498*9566063dSJacob Faibussowitsch   PetscCall(PetscOptionsGetBool(NULL, NULL, "-dm_plex_egads_new_model", &newModel, NULL));
1499*9566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_rank(comm, &rank));
15007bee2925SMatthew Knepley #if defined(PETSC_HAVE_EGADS)
1501dd400576SPatrick Sanan   if (rank == 0) {
15027bee2925SMatthew Knepley 
1503*9566063dSJacob Faibussowitsch     PetscCall(EG_open(&context));
1504*9566063dSJacob Faibussowitsch     PetscCall(EG_loadModel(context, 0, filename, &model));
1505*9566063dSJacob Faibussowitsch     if (printModel) PetscCall(DMPlexEGADSPrintModel_Internal(model));
15067bee2925SMatthew Knepley 
15077bee2925SMatthew Knepley   }
1508*9566063dSJacob Faibussowitsch   if (tessModel)     PetscCall(DMPlexCreateEGADS_Tess_Internal(comm, context, model, dm));
1509*9566063dSJacob Faibussowitsch   else if (newModel) PetscCall(DMPlexCreateEGADS(comm, context, model, dm));
1510*9566063dSJacob Faibussowitsch   else               PetscCall(DMPlexCreateEGADS_Internal(comm, context, model, dm));
1511c03d1177SJose E. Roman   PetscFunctionReturn(0);
15127bee2925SMatthew Knepley #else
1513c1cad2e7SMatthew G. Knepley   SETERRQ(comm, PETSC_ERR_SUP, "This method requires EGADS support. Reconfigure using --download-egads");
15147bee2925SMatthew Knepley #endif
15157bee2925SMatthew Knepley }
1516