1 #define PETSC_DESIRE_FEATURE_TEST_MACROS /* for fileno() */
2 #include <petsc/private/dmpleximpl.h> /*I "petscdmplex.h" I*/
3
4 /* Utility struct to store the contents of a Fluent file in memory */
5 typedef struct {
6 int index; /* Type of section */
7 unsigned int zoneID;
8 unsigned int first;
9 unsigned int last;
10 int type;
11 int nd; /* Either ND or element-type */
12 void *data;
13 } FluentSection;
14
15 /*@
16 DMPlexCreateFluentFromFile - Create a `DMPLEX` mesh from a Fluent mesh file
17
18 Collective
19
20 Input Parameters:
21 + comm - The MPI communicator
22 . filename - Name of the Fluent mesh file
23 - interpolate - Create faces and edges in the mesh
24
25 Output Parameter:
26 . dm - The `DM` object representing the mesh
27
28 Level: beginner
29
30 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMPlexCreateFromFile()`, `DMPlexCreateFluent()`, `DMPlexCreate()`
31 @*/
DMPlexCreateFluentFromFile(MPI_Comm comm,const char filename[],PetscBool interpolate,DM * dm)32 PetscErrorCode DMPlexCreateFluentFromFile(MPI_Comm comm, const char filename[], PetscBool interpolate, DM *dm)
33 {
34 PetscViewer viewer;
35
36 PetscFunctionBegin;
37 /* Create file viewer and build plex */
38 PetscCall(PetscViewerCreate(comm, &viewer));
39 PetscCall(PetscViewerSetType(viewer, PETSCVIEWERASCII));
40 PetscCall(PetscViewerFileSetMode(viewer, FILE_MODE_READ));
41 PetscCall(PetscViewerFileSetName(viewer, filename));
42 PetscCall(DMPlexCreateFluent(comm, viewer, interpolate, dm));
43 PetscCall(PetscViewerDestroy(&viewer));
44 PetscFunctionReturn(PETSC_SUCCESS);
45 }
46
DMPlexCreateFluent_ReadString(PetscViewer viewer,char * buffer,char delim)47 static PetscErrorCode DMPlexCreateFluent_ReadString(PetscViewer viewer, char *buffer, char delim)
48 {
49 PetscInt ret, i = 0;
50
51 PetscFunctionBegin;
52 do PetscCall(PetscViewerRead(viewer, &buffer[i++], 1, &ret, PETSC_CHAR));
53 while (ret > 0 && buffer[i - 1] != '\0' && buffer[i - 1] != delim && i < PETSC_MAX_PATH_LEN - 1);
54 if (!ret) buffer[i - 1] = '\0';
55 else buffer[i] = '\0';
56 PetscCheck(i < PETSC_MAX_PATH_LEN - 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Buffer overflow! This is not a valid Fluent file.");
57 PetscFunctionReturn(PETSC_SUCCESS);
58 }
59
DMPlexCreateFluent_ReadValues(PetscViewer viewer,void * data,PetscInt count,PetscDataType dtype,PetscBool binary,PetscInt * numClosingParens)60 static PetscErrorCode DMPlexCreateFluent_ReadValues(PetscViewer viewer, void *data, PetscInt count, PetscDataType dtype, PetscBool binary, PetscInt *numClosingParens)
61 {
62 int fdes = 0;
63 FILE *file;
64 PetscInt i;
65
66 PetscFunctionBegin;
67 *numClosingParens = 0;
68 if (binary) {
69 /* Extract raw file descriptor to read binary block */
70 PetscCall(PetscViewerASCIIGetPointer(viewer, &file));
71 PetscCall(PetscFFlush(file));
72 fdes = fileno(file);
73 }
74
75 if (!binary && dtype == PETSC_INT) {
76 char cbuf[256];
77 unsigned int ibuf;
78 int snum;
79 /* Parse hexadecimal ascii integers */
80 for (i = 0; i < count; i++) {
81 size_t len;
82
83 PetscCall(PetscViewerRead(viewer, cbuf, 1, NULL, PETSC_STRING));
84 snum = sscanf(cbuf, "%x", &ibuf);
85 PetscCheck(snum == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
86 ((PetscInt *)data)[i] = (PetscInt)ibuf;
87 // Check for trailing parentheses
88 PetscCall(PetscStrlen(cbuf, &len));
89 while (cbuf[len - 1] == ')' && len > 0) {
90 ++(*numClosingParens);
91 --len;
92 }
93 }
94 } else if (binary && dtype == PETSC_INT) {
95 /* Always read 32-bit ints and cast to PetscInt */
96 int *ibuf;
97 PetscCall(PetscMalloc1(count, &ibuf));
98 PetscCall(PetscBinaryRead(fdes, ibuf, count, NULL, PETSC_ENUM));
99 PetscCall(PetscByteSwap(ibuf, PETSC_ENUM, count));
100 for (i = 0; i < count; i++) ((PetscInt *)data)[i] = ibuf[i];
101 PetscCall(PetscFree(ibuf));
102
103 } else if (binary && dtype == PETSC_SCALAR) {
104 float *fbuf;
105 /* Always read 32-bit floats and cast to PetscScalar */
106 PetscCall(PetscMalloc1(count, &fbuf));
107 PetscCall(PetscBinaryRead(fdes, fbuf, count, NULL, PETSC_FLOAT));
108 PetscCall(PetscByteSwap(fbuf, PETSC_FLOAT, count));
109 for (i = 0; i < count; i++) ((PetscScalar *)data)[i] = fbuf[i];
110 PetscCall(PetscFree(fbuf));
111 } else {
112 PetscCall(PetscViewerASCIIRead(viewer, data, count, NULL, dtype));
113 }
114 PetscFunctionReturn(PETSC_SUCCESS);
115 }
116
DMPlexCreateFluent_ReadSection(PetscViewer viewer,FluentSection * s)117 static PetscErrorCode DMPlexCreateFluent_ReadSection(PetscViewer viewer, FluentSection *s)
118 {
119 char buffer[PETSC_MAX_PATH_LEN];
120 int snum;
121
122 PetscFunctionBegin;
123 /* Fast-forward to next section and derive its index */
124 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, '('));
125 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ' '));
126 snum = sscanf(buffer, "%d", &s->index);
127 /* If we can't match an index return -1 to signal end-of-file */
128 if (snum < 1) {
129 s->index = -1;
130 PetscFunctionReturn(PETSC_SUCCESS);
131 }
132
133 if (s->index == 0) { /* Comment */
134 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
135
136 } else if (s->index == 2) { /* Dimension */
137 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
138 snum = sscanf(buffer, "%d", &s->nd);
139 PetscCheck(snum == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
140
141 } else if (s->index == 10 || s->index == 2010) { /* Vertices */
142 PetscInt numClosingParens = 0;
143
144 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
145 snum = sscanf(buffer, "(%x %x %x %d %d)", &s->zoneID, &s->first, &s->last, &s->type, &s->nd);
146 PetscCheck(snum == 5, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
147 if (s->zoneID > 0) {
148 PetscInt numCoords = s->last - s->first + 1;
149 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, '('));
150 PetscCall(PetscMalloc1(s->nd * numCoords, (PetscScalar **)&s->data));
151 PetscCall(DMPlexCreateFluent_ReadValues(viewer, s->data, s->nd * numCoords, PETSC_SCALAR, s->index == 2010 ? PETSC_TRUE : PETSC_FALSE, &numClosingParens));
152 if (!numClosingParens) PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
153 else --numClosingParens;
154 }
155 if (!numClosingParens) PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
156 else --numClosingParens;
157 PetscCheck(!numClosingParens, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
158 } else if (s->index == 12 || s->index == 2012) { /* Cells */
159 PetscInt numClosingParens = 0;
160
161 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
162 snum = sscanf(buffer, "(%x", &s->zoneID);
163 PetscCheck(snum == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
164 if (s->zoneID == 0) { /* Header section */
165 snum = sscanf(buffer, "(%x %x %x %d)", &s->zoneID, &s->first, &s->last, &s->nd);
166 PetscCheck(snum == 4, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
167 } else { /* Data section */
168 snum = sscanf(buffer, "(%x %x %x %d %d)", &s->zoneID, &s->first, &s->last, &s->type, &s->nd);
169 PetscCheck(snum == 5, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
170 if (s->nd == 0) {
171 /* Read cell type definitions for mixed cells */
172 PetscInt numCells = s->last - s->first + 1;
173 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, '('));
174 PetscCall(PetscMalloc1(numCells, (PetscInt **)&s->data));
175 PetscCall(DMPlexCreateFluent_ReadValues(viewer, s->data, numCells, PETSC_INT, s->index == 2012 ? PETSC_TRUE : PETSC_FALSE, &numClosingParens));
176 if (!numClosingParens) PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
177 else --numClosingParens;
178 }
179 }
180 if (!numClosingParens) PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
181 else --numClosingParens;
182 PetscCheck(!numClosingParens, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
183 } else if (s->index == 13 || s->index == 2013) { /* Faces */
184 PetscInt numClosingParens = 0;
185
186 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
187 snum = sscanf(buffer, "(%x", &s->zoneID);
188 PetscCheck(snum == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
189 if (s->zoneID == 0) { /* Header section */
190 snum = sscanf(buffer, "(%x %x %x %d)", &s->zoneID, &s->first, &s->last, &s->nd);
191 PetscCheck(snum == 4, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
192 } else { /* Data section */
193 PetscInt numEntries, numFaces, maxsize = 0, offset = 0;
194
195 snum = sscanf(buffer, "(%x %x %x %d %d)", &s->zoneID, &s->first, &s->last, &s->type, &s->nd);
196 PetscCheck(snum == 5, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
197 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, '('));
198 switch (s->nd) {
199 case 0:
200 numEntries = PETSC_DETERMINE;
201 break;
202 case 2:
203 numEntries = 2 + 2;
204 break; /* linear */
205 case 3:
206 numEntries = 2 + 3;
207 break; /* triangular */
208 case 4:
209 numEntries = 2 + 4;
210 break; /* quadrilateral */
211 default:
212 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unknown face type in Fluent file");
213 }
214 numFaces = s->last - s->first + 1;
215 if (numEntries != PETSC_DETERMINE) {
216 /* Allocate space only if we already know the size of the block */
217 PetscCall(PetscMalloc1(numEntries * numFaces, (PetscInt **)&s->data));
218 }
219 for (PetscInt f = 0; f < numFaces; f++) {
220 if (s->nd == 0) {
221 /* Determine the size of the block for "mixed" facets */
222 PetscInt numFaceVert = 0;
223 PetscCall(DMPlexCreateFluent_ReadValues(viewer, &numFaceVert, 1, PETSC_INT, s->index == 2013 ? PETSC_TRUE : PETSC_FALSE, &numClosingParens));
224 if (!f) {
225 maxsize = (numFaceVert + 3) * numFaces;
226 PetscCall(PetscMalloc1(maxsize, (PetscInt **)&s->data));
227 } else {
228 if (offset + numFaceVert + 3 >= maxsize) {
229 PetscInt *tmp;
230
231 PetscCall(PetscMalloc1(maxsize * 2, &tmp));
232 PetscCall(PetscArraycpy(tmp, (PetscInt *)s->data, maxsize));
233 PetscCall(PetscFree(s->data));
234 maxsize *= 2;
235 s->data = tmp;
236 }
237 }
238 ((PetscInt *)s->data)[offset] = numFaceVert;
239 ++offset;
240 numEntries = numFaceVert + 2;
241 }
242 PetscCall(DMPlexCreateFluent_ReadValues(viewer, &(((PetscInt *)s->data)[offset]), numEntries, PETSC_INT, s->index == 2013 ? PETSC_TRUE : PETSC_FALSE, &numClosingParens));
243 offset += numEntries;
244 }
245 if (s->nd != 0) PetscCall(PetscMPIIntCast(numEntries - 2, &s->nd));
246 if (!numClosingParens) PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
247 else --numClosingParens;
248 }
249 if (!numClosingParens) PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
250 else --numClosingParens;
251 PetscCheck(!numClosingParens, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file");
252 } else if (s->index == 39) { /* Label information */
253 char labelName[PETSC_MAX_PATH_LEN];
254 char caseName[PETSC_MAX_PATH_LEN];
255
256 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, ')'));
257 snum = sscanf(buffer, "(%u %s %s %d)", &s->zoneID, caseName, labelName, &s->nd);
258 PetscCheck(snum == 4, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "File is not a valid Fluent file: %d", snum);
259 PetscInt depth = 1;
260 do {
261 /* Match parentheses when parsing unknown sections */
262 do PetscCall(PetscViewerRead(viewer, &buffer[0], 1, NULL, PETSC_CHAR));
263 while (buffer[0] != '(' && buffer[0] != ')');
264 if (buffer[0] == '(') depth++;
265 if (buffer[0] == ')') depth--;
266 } while (depth > 0);
267 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, '\n'));
268 PetscCall(PetscStrallocpy(labelName, (char **)&s->data));
269 PetscCall(PetscInfo((PetscObject)viewer, "CASE: Zone ID %u is label %s\n", s->zoneID, labelName));
270 } else { /* Unknown section type */
271 PetscInt depth = 1;
272 do {
273 /* Match parentheses when parsing unknown sections */
274 do PetscCall(PetscViewerRead(viewer, &buffer[0], 1, NULL, PETSC_CHAR));
275 while (buffer[0] != '(' && buffer[0] != ')');
276 if (buffer[0] == '(') depth++;
277 if (buffer[0] == ')') depth--;
278 } while (depth > 0);
279 PetscCall(DMPlexCreateFluent_ReadString(viewer, buffer, '\n'));
280 }
281 PetscFunctionReturn(PETSC_SUCCESS);
282 }
283
284 // Inserts point `face` with orientation `ornt` into the cone of point `cell` at position `c`, which is the first empty slot
InsertFace(DM dm,PetscInt cell,PetscInt face,PetscInt ornt)285 static PetscErrorCode InsertFace(DM dm, PetscInt cell, PetscInt face, PetscInt ornt)
286 {
287 const PetscInt *cone;
288 PetscInt coneSize, c;
289
290 PetscFunctionBegin;
291 PetscCall(DMPlexGetCone(dm, cell, &cone));
292 PetscCall(DMPlexGetConeSize(dm, cell, &coneSize));
293 for (c = 0; c < coneSize; ++c)
294 if (cone[c] < 0) break;
295 PetscCheck(c < coneSize, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Face %" PetscInt_FMT " could not be inserted in cone of cell %" PetscInt_FMT " with size %" PetscInt_FMT, face, cell, coneSize);
296 PetscCall(DMPlexInsertCone(dm, cell, c, face));
297 PetscCall(DMPlexInsertConeOrientation(dm, cell, c, ornt));
298 PetscFunctionReturn(PETSC_SUCCESS);
299 }
300
ReorderPolygon(DM dm,PetscInt cell)301 static PetscErrorCode ReorderPolygon(DM dm, PetscInt cell)
302 {
303 const PetscInt *cone, *ornt;
304 PetscInt coneSize, newCone[16], newOrnt[16];
305
306 PetscFunctionBegin;
307 PetscCall(DMPlexGetOrientedCone(dm, cell, &cone, &ornt));
308 PetscCall(DMPlexGetConeSize(dm, cell, &coneSize));
309 newCone[0] = cone[0];
310 newOrnt[0] = ornt[0];
311 for (PetscInt c = 1; c < coneSize; ++c) {
312 const PetscInt *fcone;
313 PetscInt firstVertex, lastVertex, c2;
314
315 PetscCall(DMPlexGetCone(dm, newCone[c - 1], &fcone));
316 lastVertex = newOrnt[c - 1] ? fcone[0] : fcone[1];
317 for (c2 = 0; c2 < coneSize; ++c2) {
318 const PetscInt *fcone2;
319
320 PetscCall(DMPlexGetCone(dm, cone[c2], &fcone2));
321 firstVertex = ornt[c2] ? fcone2[1] : fcone2[0];
322 if (lastVertex == firstVertex) {
323 // Point `cell` matched point `lastVertex` on face `cone[c2]` with orientation `ornt[c2]`
324 break;
325 }
326 }
327 PetscCheck(c2 < coneSize, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " could not find a face match as position %" PetscInt_FMT, cell, c);
328 newCone[c] = cone[c2];
329 newOrnt[c] = ornt[c2];
330 }
331 {
332 const PetscInt *fcone, *fcone2;
333 PetscInt vertex, vertex2;
334
335 PetscCall(DMPlexGetCone(dm, newCone[coneSize - 1], &fcone));
336 PetscCall(DMPlexGetCone(dm, newCone[0], &fcone2));
337 vertex = newOrnt[coneSize - 1] ? fcone[0] : fcone[1];
338 vertex2 = newOrnt[0] ? fcone2[1] : fcone2[0];
339 PetscCheck(vertex == vertex2, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " did not match at the endpoint", cell);
340 }
341 PetscCall(DMPlexSetCone(dm, cell, newCone));
342 PetscCall(DMPlexSetConeOrientation(dm, cell, newOrnt));
343 PetscCall(DMPlexRestoreOrientedCone(dm, cell, &cone, &ornt));
344 PetscFunctionReturn(PETSC_SUCCESS);
345 }
346
ReorderTetrahedron(PetscViewer viewer,DM dm,PetscInt cell)347 static PetscErrorCode ReorderTetrahedron(PetscViewer viewer, DM dm, PetscInt cell)
348 {
349 const PetscInt *cone, *ornt, *fcone, *fornt, *farr, faces[4] = {0, 1, 3, 2};
350 PetscInt newCone[16], newOrnt[16];
351
352 PetscFunctionBegin;
353 PetscCall(DMPlexGetOrientedCone(dm, cell, &cone, &ornt));
354 newCone[0] = cone[0];
355 newOrnt[0] = ornt[0];
356 PetscCall(DMPlexGetOrientedCone(dm, newCone[0], &fcone, &fornt));
357 farr = DMPolytopeTypeGetArrangement(DM_POLYTOPE_TRIANGLE, newOrnt[0]);
358 // Loop over each edge in the initial triangle
359 for (PetscInt e = 0; e < 3; ++e) {
360 const PetscInt edge = fcone[farr[e * 2 + 0]], eornt = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_SEGMENT, farr[e * 2 + 1], fornt[farr[e * 2 + 0]]);
361 PetscInt c;
362
363 // Loop over each remaining face in the tetrahedron
364 // On face `newCone[0]`, trying to match edge `edge` with final orientation `eornt` to an edge on another face
365 for (c = 1; c < 4; ++c) {
366 const PetscInt *fcone2, *fornt2, *farr2;
367 PetscInt c2;
368 PetscBool flip = PETSC_FALSE;
369
370 // Checking face `cone[c]` with orientation `ornt[c]`
371 PetscCall(DMPlexGetOrientedCone(dm, cone[c], &fcone2, &fornt2));
372 farr2 = DMPolytopeTypeGetArrangement(DM_POLYTOPE_TRIANGLE, ornt[c]);
373 // Check for edge
374 for (c2 = 0; c2 < 3; ++c2) {
375 const PetscInt edge2 = fcone2[farr2[c2 * 2 + 0]], eornt2 = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_SEGMENT, farr2[c2 * 2 + 1], fornt2[farr2[c2 * 2 + 0]]);
376 // Trying to match edge `edge2` with final orientation `eornt2`
377 if (edge == edge2) {
378 //PetscCheck(eornt == -(eornt2 + 1), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell % " PetscInt_FMT " edge %" PetscInt_FMT " (%" PetscInt_FMT ") found twice with the same orientation in face %" PetscInt_FMT " edge %" PetscInt_FMT, cell, edge, e, c, c2);
379 // Matched face `newCone[0]` with orientation `newOrnt[0]` to face `cone[c]` with orientation `ornt[c]` along edge `edge`
380 PetscCall(PetscInfo((PetscObject)viewer, "CASE: Matched cell %" PetscInt_FMT " edge %" PetscInt_FMT "/%" PetscInt_FMT " (%" PetscInt_FMT ") to face %" PetscInt_FMT "/%" PetscInt_FMT " edge %" PetscInt_FMT " (%" PetscInt_FMT ")\n", cell, edge, e, eornt, cone[c], c, c2, eornt2));
381 flip = eornt != -(eornt2 + 1) ? PETSC_TRUE : PETSC_FALSE;
382 break;
383 }
384 }
385 if (c2 < 3) {
386 newCone[faces[e + 1]] = cone[c];
387 // Compute new orientation of face based on which edge was matched (only the first edge matches a side different from 0)
388 // Face 1 should match its edge 2
389 // Face 2 should match its edge 0
390 // Face 3 should match its edge 0
391 if (flip) {
392 newOrnt[faces[e + 1]] = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_TRIANGLE, -((c2 + (!e ? 1 : 2)) % 3 + 1), ornt[c]);
393 } else {
394 newOrnt[faces[e + 1]] = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_TRIANGLE, !e ? (c2 + 1) % 3 : c2, ornt[c]);
395 }
396 break;
397 }
398 }
399 PetscCheck(c < 4, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " could not find a face match for edge %" PetscInt_FMT, cell, e);
400 }
401 PetscCall(DMPlexRestoreOrientedCone(dm, newCone[0], &fcone, &fornt));
402 PetscCall(DMPlexSetCone(dm, cell, newCone));
403 PetscCall(DMPlexSetConeOrientation(dm, cell, newOrnt));
404 PetscCall(DMPlexRestoreOrientedCone(dm, cell, &cone, &ornt));
405 PetscFunctionReturn(PETSC_SUCCESS);
406 }
407
ReorderHexahedron(DM dm,PetscInt cell)408 static PetscErrorCode ReorderHexahedron(DM dm, PetscInt cell)
409 {
410 const PetscInt *cone, *ornt, *fcone, *fornt, *farr;
411 const PetscInt faces[6] = {0, 5, 3, 4, 2, 1};
412 PetscInt used[6] = {1, 0, 0, 0, 0, 0};
413 PetscInt newCone[16], newOrnt[16];
414
415 PetscFunctionBegin;
416 PetscCall(DMPlexGetOrientedCone(dm, cell, &cone, &ornt));
417 newCone[0] = cone[0];
418 newOrnt[0] = ornt[0];
419 PetscCall(DMPlexGetOrientedCone(dm, newCone[0], &fcone, &fornt));
420 farr = DMPolytopeTypeGetArrangement(DM_POLYTOPE_QUADRILATERAL, newOrnt[0]);
421 // Loop over each edge in the initial quadrilateral
422 for (PetscInt e = 0; e < 4; ++e) {
423 const PetscInt edge = fcone[farr[e * 2 + 0]], eornt = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_SEGMENT, farr[e * 2 + 1], fornt[farr[e * 2 + 0]]);
424 PetscInt c;
425
426 // Loop over each remaining face in the hexahedron
427 // On face `newCone[0]`, trying to match edge `edge` with final orientation `eornt` to an edge on another face
428 for (c = 1; c < 6; ++c) {
429 const PetscInt *fcone2, *fornt2, *farr2;
430 PetscInt c2;
431
432 // Checking face `cone[c]` with orientation `ornt[c]`
433 PetscCall(DMPlexGetOrientedCone(dm, cone[c], &fcone2, &fornt2));
434 farr2 = DMPolytopeTypeGetArrangement(DM_POLYTOPE_QUADRILATERAL, ornt[c]);
435 // Check for edge
436 for (c2 = 0; c2 < 4; ++c2) {
437 const PetscInt edge2 = fcone2[farr2[c2 * 2 + 0]], eornt2 = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_SEGMENT, farr2[c2 * 2 + 1], fornt2[farr2[c2 * 2 + 0]]);
438 // Trying to match edge `edge2` with final orientation `eornt2`
439 if (edge == edge2) {
440 PetscCheck(eornt == -(eornt2 + 1), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Edge %" PetscInt_FMT " found twice with the same orientation", edge);
441 // Matched face `newCone[0]` with orientation `newOrnt[0]` to face `cone[c]` with orientation `ornt[c]` along edge `edge`
442 break;
443 }
444 }
445 if (c2 < 4) {
446 used[c] = 1;
447 newCone[faces[e + 1]] = cone[c];
448 // Compute new orientation of face based on which edge was matched (only the first edge matches a side different from 0)
449 newOrnt[faces[e + 1]] = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_QUADRILATERAL, !e ? (c2 + 1) % 4 : c2, ornt[c]);
450 break;
451 }
452 }
453 PetscCheck(c < 6, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " could not find a face match for edge %" PetscInt_FMT, cell, e);
454 }
455 PetscCall(DMPlexRestoreOrientedCone(dm, newCone[0], &fcone, &fornt));
456 // Add last face
457 {
458 PetscInt c, c2;
459
460 for (c = 1; c < 6; ++c)
461 if (!used[c]) break;
462 PetscCheck(c < 6, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " could not find an available face", cell);
463 // Match first edge to 3rd edge in newCone[2]
464 {
465 const PetscInt *fcone2, *fornt2, *farr2;
466
467 PetscCall(DMPlexGetOrientedCone(dm, newCone[2], &fcone, &fornt));
468 farr = DMPolytopeTypeGetArrangement(DM_POLYTOPE_QUADRILATERAL, newOrnt[2]);
469 PetscCall(DMPlexGetOrientedCone(dm, cone[c], &fcone2, &fornt2));
470 farr2 = DMPolytopeTypeGetArrangement(DM_POLYTOPE_QUADRILATERAL, ornt[c]);
471
472 const PetscInt e = 2;
473 const PetscInt edge = fcone[farr[e * 2 + 0]], eornt = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_SEGMENT, farr[e * 2 + 1], fornt[farr[e * 2 + 0]]);
474 // Trying to match edge `edge` with final orientation `eornt` of face `newCone[2]` to some edge of face `cone[c]` with orientation `ornt[c]`
475 for (c2 = 0; c2 < 4; ++c2) {
476 const PetscInt edge2 = fcone2[farr2[c2 * 2 + 0]], eornt2 = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_SEGMENT, farr2[c2 * 2 + 1], fornt2[farr2[c2 * 2 + 0]]);
477 // Trying to match edge `edge2` with final orientation `eornt2`
478 if (edge == edge2) {
479 PetscCheck(eornt == -(eornt2 + 1), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Edge %" PetscInt_FMT " found twice with the same orientation", edge);
480 // Matched face `newCone[2]` with orientation `newOrnt[2]` to face `cone[c]` with orientation `ornt[c]` along edge `edge`
481 break;
482 }
483 }
484 PetscCheck(c2 < 4, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Could not fit last face in");
485 }
486 newCone[faces[5]] = cone[c];
487 // Compute new orientation of face based on which edge was matched
488 newOrnt[faces[5]] = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_QUADRILATERAL, c2, ornt[c]);
489 PetscCall(DMPlexRestoreOrientedCone(dm, newCone[0], &fcone, &fornt));
490 }
491 PetscCall(DMPlexSetCone(dm, cell, newCone));
492 PetscCall(DMPlexSetConeOrientation(dm, cell, newOrnt));
493 PetscCall(DMPlexRestoreOrientedCone(dm, cell, &cone, &ornt));
494 PetscFunctionReturn(PETSC_SUCCESS);
495 }
496
497 // {0, 1, 2}, {3, 4, 5}, {0, 2, 4, 3}, {2, 1, 5, 4}, {1, 0, 3, 5}
ReorderWedge(DM dm,PetscInt cell)498 static PetscErrorCode ReorderWedge(DM dm, PetscInt cell)
499 {
500 const PetscInt *cone, *ornt, *fcone, *fornt, *farr;
501 const PetscInt faces[5] = {0, 4, 3, 2, 1};
502 PetscInt used[5] = {0, 0, 0, 0, 0};
503 PetscInt newCone[16], newOrnt[16], cS, bottom = 0;
504
505 PetscFunctionBegin;
506 PetscCall(DMPlexGetConeSize(dm, cell, &cS));
507 PetscCall(DMPlexGetOrientedCone(dm, cell, &cone, &ornt));
508 for (PetscInt c = 0; c < cS; ++c) {
509 DMPolytopeType ct;
510
511 PetscCall(DMPlexGetCellType(dm, cone[c], &ct));
512 if (ct == DM_POLYTOPE_TRIANGLE) {
513 bottom = c;
514 break;
515 }
516 }
517 used[bottom] = 1;
518 newCone[0] = cone[bottom];
519 newOrnt[0] = ornt[bottom];
520 PetscCall(DMPlexGetOrientedCone(dm, newCone[0], &fcone, &fornt));
521 farr = DMPolytopeTypeGetArrangement(DM_POLYTOPE_TRIANGLE, newOrnt[0]);
522 // Loop over each edge in the initial triangle
523 for (PetscInt e = 0; e < 3; ++e) {
524 const PetscInt edge = fcone[farr[e * 2 + 0]], eornt = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_SEGMENT, farr[e * 2 + 1], fornt[farr[e * 2 + 0]]);
525 PetscInt c;
526
527 // Loop over each remaining face in the prism
528 // On face `newCone[0]`, trying to match edge `edge` with final orientation `eornt` to an edge on another face
529 for (c = 0; c < 5; ++c) {
530 const PetscInt *fcone2, *fornt2, *farr2;
531 DMPolytopeType ct;
532 PetscInt c2;
533
534 if (c == bottom) continue;
535 PetscCall(DMPlexGetCellType(dm, cone[c], &ct));
536 if (ct != DM_POLYTOPE_QUADRILATERAL) continue;
537 // Checking face `cone[c]` with orientation `ornt[c]`
538 PetscCall(DMPlexGetOrientedCone(dm, cone[c], &fcone2, &fornt2));
539 farr2 = DMPolytopeTypeGetArrangement(DM_POLYTOPE_QUADRILATERAL, ornt[c]);
540 // Check for edge
541 for (c2 = 0; c2 < 4; ++c2) {
542 const PetscInt edge2 = fcone2[farr2[c2 * 2 + 0]], eornt2 = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_SEGMENT, farr2[c2 * 2 + 1], fornt2[farr2[c2 * 2 + 0]]);
543 // Trying to match edge `edge2` with final orientation `eornt2`
544 if (edge == edge2) {
545 PetscCheck(eornt == -(eornt2 + 1), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Edge %" PetscInt_FMT " found twice with the same orientation", edge);
546 // Matched face `newCone[0]` with orientation `newOrnt[0]` to face `cone[c]` with orientation `ornt[c]` along edge `edge`
547 break;
548 }
549 }
550 if (c2 < 4) {
551 used[c] = 1;
552 newCone[faces[e + 1]] = cone[c];
553 // Compute new orientation of face based on which edge was matched, edge 0 should always match the bottom
554 newOrnt[faces[e + 1]] = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_QUADRILATERAL, c2, ornt[c]);
555 break;
556 }
557 }
558 PetscCheck(c < 5, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " could not find a face match for edge %" PetscInt_FMT, cell, e);
559 }
560 PetscCall(DMPlexRestoreOrientedCone(dm, newCone[0], &fcone, &fornt));
561 // Add last face
562 {
563 PetscInt c, c2;
564
565 for (c = 0; c < 5; ++c)
566 if (!used[c]) break;
567 PetscCheck(c < 5, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Cell %" PetscInt_FMT " could not find an available face", cell);
568 // Match first edge to 3rd edge in newCone[2]
569 {
570 const PetscInt *fcone2, *fornt2, *farr2;
571
572 PetscCall(DMPlexGetOrientedCone(dm, newCone[2], &fcone, &fornt));
573 farr = DMPolytopeTypeGetArrangement(DM_POLYTOPE_QUADRILATERAL, newOrnt[2]);
574 PetscCall(DMPlexGetOrientedCone(dm, cone[c], &fcone2, &fornt2));
575 farr2 = DMPolytopeTypeGetArrangement(DM_POLYTOPE_TRIANGLE, ornt[c]);
576
577 const PetscInt e = 2;
578 const PetscInt edge = fcone[farr[e * 2 + 0]], eornt = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_SEGMENT, farr[e * 2 + 1], fornt[farr[e * 2 + 0]]);
579 // Trying to match edge `edge` with final orientation `eornt` of face `newCone[2]` to some edge of face `cone[c]` with orientation `ornt[c]`
580 for (c2 = 0; c2 < 3; ++c2) {
581 const PetscInt edge2 = fcone2[farr2[c2 * 2 + 0]], eornt2 = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_SEGMENT, farr2[c2 * 2 + 1], fornt2[farr2[c2 * 2 + 0]]);
582 // Trying to match edge `edge2` with final orientation `eornt2`
583 if (edge == edge2) {
584 PetscCheck(eornt == -(eornt2 + 1), PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Edge %" PetscInt_FMT " found twice with the same orientation", edge);
585 // Matched face `newCone[2]` with orientation `newOrnt[2]` to face `cone[c]` with orientation `ornt[c]` along edge `edge`
586 break;
587 }
588 }
589 PetscCheck(c2 < 3, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Could not fit last face in");
590 }
591 newCone[faces[4]] = cone[c];
592 // Compute new orientation of face based on which edge was matched
593 newOrnt[faces[4]] = DMPolytopeTypeComposeOrientation(DM_POLYTOPE_TRIANGLE, c2, ornt[c]);
594 PetscCall(DMPlexRestoreOrientedCone(dm, newCone[0], &fcone, &fornt));
595 }
596 PetscCall(DMPlexSetCone(dm, cell, newCone));
597 PetscCall(DMPlexSetConeOrientation(dm, cell, newOrnt));
598 PetscCall(DMPlexRestoreOrientedCone(dm, cell, &cone, &ornt));
599 PetscFunctionReturn(PETSC_SUCCESS);
600 }
601
ReorderCell(PetscViewer viewer,DM dm,PetscInt cell,DMPolytopeType ct)602 static PetscErrorCode ReorderCell(PetscViewer viewer, DM dm, PetscInt cell, DMPolytopeType ct)
603 {
604 PetscFunctionBegin;
605 switch (ct) {
606 case DM_POLYTOPE_TRIANGLE:
607 case DM_POLYTOPE_QUADRILATERAL:
608 PetscCall(ReorderPolygon(dm, cell));
609 break;
610 case DM_POLYTOPE_TETRAHEDRON:
611 PetscCall(ReorderTetrahedron(viewer, dm, cell));
612 break;
613 case DM_POLYTOPE_HEXAHEDRON:
614 PetscCall(ReorderHexahedron(dm, cell));
615 break;
616 case DM_POLYTOPE_TRI_PRISM:
617 PetscCall(ReorderWedge(dm, cell));
618 break;
619 default:
620 PetscCheck(0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Celltype %s is unsupported", DMPolytopeTypes[ct]);
621 break;
622 }
623 PetscFunctionReturn(PETSC_SUCCESS);
624 }
625
GetNumCellFaces(int nd,PetscInt * numCellFaces,DMPolytopeType * ct)626 static PetscErrorCode GetNumCellFaces(int nd, PetscInt *numCellFaces, DMPolytopeType *ct)
627 {
628 PetscFunctionBegin;
629 *ct = DM_POLYTOPE_POINT;
630 switch (nd) {
631 case 0:
632 *numCellFaces = PETSC_DETERMINE;
633 break;
634 case 1:
635 *numCellFaces = 3;
636 *ct = DM_POLYTOPE_TRIANGLE;
637 break;
638 case 2:
639 *numCellFaces = 4;
640 *ct = DM_POLYTOPE_TETRAHEDRON;
641 break;
642 case 3:
643 *numCellFaces = 4;
644 *ct = DM_POLYTOPE_QUADRILATERAL;
645 break;
646 case 4:
647 *numCellFaces = 6;
648 *ct = DM_POLYTOPE_HEXAHEDRON;
649 break;
650 case 5:
651 *numCellFaces = 5;
652 *ct = DM_POLYTOPE_PYRAMID;
653 break;
654 case 6:
655 *numCellFaces = 5;
656 *ct = DM_POLYTOPE_TRI_PRISM;
657 break;
658 default:
659 *numCellFaces = PETSC_DETERMINE;
660 }
661 PetscFunctionReturn(PETSC_SUCCESS);
662 }
663
664 /*@C
665 DMPlexCreateFluent - Create a `DMPLEX` mesh from a Fluent mesh file <http://aerojet.engr.ucdavis.edu/fluenthelp/html/ug/node1490.htm>.
666
667 Collective
668
669 Input Parameters:
670 + comm - The MPI communicator
671 . viewer - The `PetscViewer` associated with a Fluent mesh file
672 - interpolate - Create faces and edges in the mesh
673
674 Output Parameter:
675 . dm - The `DM` object representing the mesh
676
677 Level: beginner
678
679 .seealso: [](ch_unstructured), `DM`, `DMPLEX`, `DMCreate()`
680 @*/
DMPlexCreateFluent(MPI_Comm comm,PetscViewer viewer,PetscBool interpolate,DM * dm)681 PetscErrorCode DMPlexCreateFluent(MPI_Comm comm, PetscViewer viewer, PetscBool interpolate, DM *dm)
682 {
683 PetscInt dim = PETSC_DETERMINE;
684 PetscInt numCells = 0;
685 PetscInt numVertices = 0;
686 PetscInt *cellSizes = NULL;
687 DMPolytopeType *cellTypes = NULL;
688 PetscInt numFaces = 0;
689 PetscInt *faces = NULL;
690 PetscInt *faceSizes = NULL;
691 PetscInt *faceAdjCell = NULL;
692 PetscInt *cellVertices = NULL;
693 unsigned int *faceZoneIDs = NULL;
694 DMLabel faceSets = NULL;
695 DMLabel *zoneLabels = NULL;
696 const char **zoneNames = NULL;
697 unsigned int maxZoneID = 0;
698 PetscScalar *coordsIn = NULL;
699 PetscScalar *coords;
700 PetscSection coordSection;
701 Vec coordinates;
702 PetscInt coordSize, maxFaceSize = 0, totFaceVert = 0, f;
703 PetscMPIInt rank;
704
705 PetscFunctionBegin;
706 PetscCallMPI(MPI_Comm_rank(comm, &rank));
707
708 if (rank == 0) {
709 FluentSection s;
710
711 s.data = NULL;
712 numFaces = PETSC_DETERMINE;
713 do {
714 PetscCall(DMPlexCreateFluent_ReadSection(viewer, &s));
715 if (s.index == 2) { /* Dimension */
716 dim = s.nd;
717 PetscCall(PetscInfo((PetscObject)viewer, "CASE: Found dimension: %" PetscInt_FMT "\n", dim));
718 } else if (s.index == 10 || s.index == 2010) { /* Vertices */
719 if (s.zoneID == 0) {
720 numVertices = s.last;
721 PetscCall(PetscInfo((PetscObject)viewer, "CASE: Found number of vertices: %" PetscInt_FMT "\n", numVertices));
722 } else {
723 PetscCall(PetscInfo((PetscObject)viewer, "CASE: Found vertex coordinates\n"));
724 PetscCheck(!coordsIn, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Currently no support for multiple coordinate sets in Fluent files");
725 coordsIn = (PetscScalar *)s.data;
726 }
727
728 } else if (s.index == 12 || s.index == 2012) { /* Cells */
729 if (s.zoneID == 0) {
730 numCells = s.last;
731 PetscCall(PetscInfo((PetscObject)viewer, "CASE: Found number of cells %" PetscInt_FMT "\n", numCells));
732 } else {
733 PetscCall(PetscMalloc2(numCells, &cellSizes, numCells, &cellTypes));
734 for (PetscInt c = 0; c < numCells; ++c) PetscCall(GetNumCellFaces(s.nd ? s.nd : (int)((PetscInt *)s.data)[c], &cellSizes[c], &cellTypes[c]));
735 PetscCall(PetscFree(s.data));
736 PetscCall(PetscInfo((PetscObject)viewer, "CASE: Found number of cell faces %" PetscInt_FMT "\n", numCells && s.nd ? cellSizes[0] : 0));
737 }
738 } else if (s.index == 13 || s.index == 2013) { /* Facets */
739 if (s.zoneID == 0) { /* Header section */
740 numFaces = (PetscInt)(s.last - s.first + 1);
741 PetscCall(PetscInfo((PetscObject)viewer, "CASE: Found number of faces %" PetscInt_FMT " face vertices: %d\n", numFaces, s.nd));
742 } else { /* Data section */
743 PetscInt *tmp;
744 PetscInt totSize = 0, offset = 0, doffset;
745
746 PetscCheck(numFaces >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "No header section for facets in Fluent file");
747 if (!faceZoneIDs) PetscCall(PetscMalloc3(numFaces, &faceSizes, numFaces * 2, &faceAdjCell, numFaces, &faceZoneIDs));
748 // Record the zoneID and face size for each face set
749 for (unsigned int z = s.first - 1; z < s.last; z++) {
750 faceZoneIDs[z] = s.zoneID;
751 if (s.nd) {
752 faceSizes[z] = s.nd;
753 } else {
754 faceSizes[z] = ((PetscInt *)s.data)[offset];
755 offset += faceSizes[z] + 3;
756 }
757 totSize += faceSizes[z];
758 maxFaceSize = PetscMax(maxFaceSize, faceSizes[z]);
759 }
760
761 offset = totFaceVert;
762 doffset = s.nd ? 0 : 1;
763 PetscCall(PetscMalloc1(totFaceVert + totSize, &tmp));
764 if (faces) PetscCall(PetscArraycpy(tmp, faces, totFaceVert));
765 PetscCall(PetscFree(faces));
766 totFaceVert += totSize;
767 faces = tmp;
768
769 // Record face vertices and adjacent faces
770 const PetscInt Nfz = s.last - s.first + 1;
771 for (PetscInt f = 0; f < Nfz; ++f) {
772 const PetscInt face = f + s.first - 1;
773 const PetscInt faceSize = faceSizes[face];
774
775 for (PetscInt v = 0; v < faceSize; ++v) faces[offset + v] = ((PetscInt *)s.data)[doffset + v];
776 faceAdjCell[face * 2 + 0] = ((PetscInt *)s.data)[doffset + faceSize + 0];
777 faceAdjCell[face * 2 + 1] = ((PetscInt *)s.data)[doffset + faceSize + 1];
778 offset += faceSize;
779 doffset += faceSize + (s.nd ? 2 : 3);
780 }
781 PetscCall(PetscFree(s.data));
782 }
783 } else if (s.index == 39) { /* Label information */
784 if (s.zoneID >= maxZoneID) {
785 DMLabel *tmpL;
786 const char **tmp;
787 unsigned int newmax = maxZoneID + 1;
788
789 while (newmax < s.zoneID + 1) newmax *= 2;
790 PetscCall(PetscCalloc2(newmax, &tmp, newmax, &tmpL));
791 for (PetscInt i = 0; i < (PetscInt)maxZoneID; ++i) {
792 tmp[i] = zoneNames[i];
793 tmpL[i] = zoneLabels[i];
794 }
795 maxZoneID = newmax;
796 PetscCall(PetscFree2(zoneNames, zoneLabels));
797 zoneNames = tmp;
798 zoneLabels = tmpL;
799 }
800 zoneNames[s.zoneID] = (const char *)s.data;
801 }
802 } while (s.index >= 0);
803 }
804 PetscCallMPI(MPI_Bcast(&dim, 1, MPIU_INT, 0, comm));
805 PetscCheck(dim >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Fluent file does not include dimension");
806
807 /* Allocate cell-vertex mesh */
808 PetscCall(DMCreate(comm, dm));
809 PetscCall(DMSetType(*dm, DMPLEX));
810 PetscCall(DMSetDimension(*dm, dim));
811 // We do not want this label automatically computed, instead we fill it here
812 PetscCall(DMCreateLabel(*dm, "celltype"));
813 PetscCall(DMPlexSetChart(*dm, 0, numCells + numFaces + numVertices));
814 if (rank == 0) {
815 PetscCheck(numCells >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unknown number of cells in Fluent file");
816 for (PetscInt c = 0; c < numCells; ++c) {
817 PetscCall(DMPlexSetConeSize(*dm, c, cellSizes[c]));
818 PetscCall(DMPlexSetCellType(*dm, c, cellTypes[c]));
819 }
820 for (PetscInt v = numCells; v < numCells + numVertices; ++v) PetscCall(DMPlexSetCellType(*dm, v, DM_POLYTOPE_POINT));
821 for (PetscInt f = 0; f < numFaces; ++f) {
822 DMPolytopeType ct;
823
824 switch (faceSizes[f]) {
825 case 2:
826 ct = DM_POLYTOPE_SEGMENT;
827 break;
828 case 3:
829 ct = DM_POLYTOPE_TRIANGLE;
830 break;
831 case 4:
832 ct = DM_POLYTOPE_QUADRILATERAL;
833 break;
834 default:
835 SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Unknown face type in Fluent file with cone size %" PetscInt_FMT, faceSizes[f]);
836 }
837 PetscCall(DMPlexSetConeSize(*dm, f + numCells + numVertices, faceSizes[f]));
838 PetscCall(DMPlexSetCellType(*dm, f + numCells + numVertices, ct));
839 }
840 }
841 PetscCall(DMSetUp(*dm));
842
843 if (rank == 0 && faces) {
844 PetscSection s;
845 PetscInt *cones, csize, foffset = 0;
846
847 PetscCall(DMPlexGetCones(*dm, &cones));
848 PetscCall(DMPlexGetConeSection(*dm, &s));
849 PetscCall(PetscSectionGetConstrainedStorageSize(s, &csize));
850 for (PetscInt c = 0; c < csize; ++c) cones[c] = -1;
851 for (PetscInt f = 0; f < numFaces; f++) {
852 const PetscInt cl = faceAdjCell[f * 2 + 0] - 1;
853 const PetscInt cr = faceAdjCell[f * 2 + 1] - 1;
854 const PetscInt face = f + numCells + numVertices;
855 PetscInt fcone[16];
856
857 // How could Fluent define the outward normal differently? Is there no end to the pain?
858 if (dim == 3) {
859 if (cl >= 0) PetscCall(InsertFace(*dm, cl, face, -1));
860 if (cr >= 0) PetscCall(InsertFace(*dm, cr, face, 0));
861 } else {
862 if (cl >= 0) PetscCall(InsertFace(*dm, cl, face, 0));
863 if (cr >= 0) PetscCall(InsertFace(*dm, cr, face, -1));
864 }
865 PetscCheck(faceSizes[f] < 16, PETSC_COMM_SELF, PETSC_ERR_SUP, "Number of face vertices %" PetscInt_FMT " exceeds temporary storage", faceSizes[f]);
866 for (PetscInt v = 0; v < faceSizes[f]; ++v) fcone[v] = faces[foffset + v] + numCells - 1;
867 foffset += faceSizes[f];
868 PetscCall(DMPlexSetCone(*dm, face, fcone));
869 }
870 }
871 PetscCall(DMPlexSymmetrize(*dm));
872 PetscCall(DMPlexStratify(*dm));
873 if (dim == 3) {
874 DM idm;
875
876 PetscCall(DMCreate(PetscObjectComm((PetscObject)*dm), &idm));
877 PetscCall(DMSetType(idm, DMPLEX));
878 PetscCall(DMSetDimension(idm, dim));
879 PetscCall(DMPlexInterpolateFaces_Internal(*dm, 1, idm));
880 PetscCall(DMDestroy(dm));
881 *dm = idm;
882 }
883 PetscCall(DMViewFromOptions(*dm, NULL, "-cas_dm_view"));
884 if (rank == 0 && faces) {
885 for (PetscInt c = 0; c < numCells; ++c) PetscCall(ReorderCell(viewer, *dm, c, cellTypes[c]));
886 }
887
888 if (rank == 0 && faces) {
889 PetscInt joinSize, meetSize, *fverts, cells[2];
890 const PetscInt *join, *meet;
891 PetscInt foffset = 0;
892
893 PetscCall(PetscMalloc1(maxFaceSize, &fverts));
894 /* Mark facets by finding the full join of all adjacent vertices */
895 for (f = 0; f < numFaces; f++) {
896 const PetscInt cl = faceAdjCell[f * 2 + 0] - 1;
897 const PetscInt cr = faceAdjCell[f * 2 + 1] - 1;
898 const PetscInt id = (PetscInt)faceZoneIDs[f];
899
900 if (cl > 0 && cr > 0) {
901 /* If we know both adjoining cells we can use a single-level meet */
902 cells[0] = cl;
903 cells[1] = cr;
904 PetscCall(DMPlexGetMeet(*dm, 2, cells, &meetSize, &meet));
905 PetscCheck(meetSize == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Could not determine Plex facet for Fluent face %" PetscInt_FMT " cells: %" PetscInt_FMT ", %" PetscInt_FMT, f, cl, cr);
906 PetscCall(DMSetLabelValue_Fast(*dm, &faceSets, "Face Sets", meet[0], id));
907 if (zoneNames && zoneNames[id]) PetscCall(DMSetLabelValue_Fast(*dm, &zoneLabels[id], zoneNames[id], meet[0], 1));
908 PetscCall(DMPlexRestoreMeet(*dm, meetSize, fverts, &meetSize, &meet));
909 } else {
910 for (PetscInt fi = 0; fi < faceSizes[f]; fi++) fverts[fi] = faces[foffset + fi] + numCells - 1;
911 PetscCall(DMPlexGetFullJoin(*dm, faceSizes[f], fverts, &joinSize, &join));
912 PetscCheck(joinSize == 1, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Could not determine Plex facet for Fluent face %" PetscInt_FMT, f);
913 PetscCall(DMSetLabelValue_Fast(*dm, &faceSets, "Face Sets", join[0], id));
914 if (zoneNames && zoneNames[id]) PetscCall(DMSetLabelValue_Fast(*dm, &zoneLabels[id], zoneNames[id], join[0], 1));
915 PetscCall(DMPlexRestoreJoin(*dm, joinSize, fverts, &joinSize, &join));
916 }
917 foffset += faceSizes[f];
918 }
919 PetscCall(PetscFree(fverts));
920 }
921
922 { /* Create Face Sets label at all processes */
923 enum {
924 n = 1
925 };
926 PetscBool flag[n];
927
928 flag[0] = faceSets ? PETSC_TRUE : PETSC_FALSE;
929 PetscCallMPI(MPI_Bcast(flag, n, MPI_C_BOOL, 0, comm));
930 if (flag[0]) PetscCall(DMCreateLabel(*dm, "Face Sets"));
931 // TODO Code to create all the zone labels on each process
932 }
933
934 if (!interpolate) {
935 DM udm;
936
937 PetscCall(DMPlexUninterpolate(*dm, &udm));
938 PetscCall(DMDestroy(dm));
939 *dm = udm;
940 }
941
942 /* Read coordinates */
943 PetscCall(DMGetCoordinateSection(*dm, &coordSection));
944 PetscCall(PetscSectionSetNumFields(coordSection, 1));
945 PetscCall(PetscSectionSetFieldComponents(coordSection, 0, dim));
946 PetscCall(PetscSectionSetChart(coordSection, numCells, numCells + numVertices));
947 for (PetscInt v = numCells; v < numCells + numVertices; ++v) {
948 PetscCall(PetscSectionSetDof(coordSection, v, dim));
949 PetscCall(PetscSectionSetFieldDof(coordSection, v, 0, dim));
950 }
951 PetscCall(PetscSectionSetUp(coordSection));
952 PetscCall(PetscSectionGetStorageSize(coordSection, &coordSize));
953 PetscCall(VecCreate(PETSC_COMM_SELF, &coordinates));
954 PetscCall(PetscObjectSetName((PetscObject)coordinates, "coordinates"));
955 PetscCall(VecSetSizes(coordinates, coordSize, PETSC_DETERMINE));
956 PetscCall(VecSetType(coordinates, VECSTANDARD));
957 PetscCall(VecGetArray(coordinates, &coords));
958 if (rank == 0 && coordsIn) {
959 for (PetscInt v = 0; v < numVertices; ++v) {
960 for (PetscInt d = 0; d < dim; ++d) coords[v * dim + d] = coordsIn[v * dim + d];
961 }
962 }
963 PetscCall(VecRestoreArray(coordinates, &coords));
964 PetscCall(DMSetCoordinatesLocal(*dm, coordinates));
965 PetscCall(VecDestroy(&coordinates));
966
967 if (rank == 0) {
968 PetscCall(PetscFree(cellVertices));
969 PetscCall(PetscFree2(cellSizes, cellTypes));
970 PetscCall(PetscFree(faces));
971 PetscCall(PetscFree3(faceSizes, faceAdjCell, faceZoneIDs));
972 PetscCall(PetscFree(coordsIn));
973 if (zoneNames)
974 for (PetscInt i = 0; i < (PetscInt)maxZoneID; ++i) PetscCall(PetscFree(zoneNames[i]));
975 PetscCall(PetscFree2(zoneNames, zoneLabels));
976 }
977 PetscFunctionReturn(PETSC_SUCCESS);
978 }
979