/*
This takes as input the raw Hasse Diagram data
*/
PetscErrorCode DMPlexCreateFromDAG(DM dm, PetscInt depth, const PetscInt numPoints[], const PetscInt coneSize[], const PetscInt cones[], const PetscInt coneOrientations[], const PetscScalar vertexCoords[])
{
Vec coordinates;
PetscSection coordSection;
PetscScalar *coords;
PetscInt coordSize, firstVertex = numPoints[depth], pStart = 0, pEnd = 0, p, v, dim, d, off;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = DMPlexGetDimension(dm, &dim);CHKERRQ(ierr);
for (d = 0; d <= depth; ++d) pEnd += numPoints[d];
ierr = DMPlexSetChart(dm, pStart, pEnd);CHKERRQ(ierr);
for (p = pStart; p < pEnd; ++p) {
ierr = DMPlexSetConeSize(dm, p, coneSize[p-pStart]);CHKERRQ(ierr);
}
ierr = DMSetUp(dm);CHKERRQ(ierr); /* Allocate space for cones */
for (p = pStart, off = 0; p < pEnd; off += coneSize[p-pStart], ++p) {
ierr = DMPlexSetCone(dm, p, &cones[off]);CHKERRQ(ierr);
ierr = DMPlexSetConeOrientation(dm, p, &coneOrientations[off]);CHKERRQ(ierr);
}
ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
ierr = DMPlexStratify(dm);CHKERRQ(ierr);
/* Build coordinates */
ierr = DMPlexGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
ierr = PetscSectionSetNumFields(coordSection, 1);CHKERRQ(ierr);
ierr = PetscSectionSetFieldComponents(coordSection, 0, dim);CHKERRQ(ierr);
ierr = PetscSectionSetChart(coordSection, firstVertex, firstVertex+numPoints[0]);CHKERRQ(ierr);
for (v = firstVertex; v < firstVertex+numPoints[0]; ++v) {
ierr = PetscSectionSetDof(coordSection, v, dim);CHKERRQ(ierr);
ierr = PetscSectionSetFieldDof(coordSection, v, 0, dim);CHKERRQ(ierr);
}
ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
ierr = VecCreate(PetscObjectComm((PetscObject)dm), &coordinates);CHKERRQ(ierr);
ierr = PetscObjectSetName((PetscObject) coordinates, "coordinates");CHKERRQ(ierr);
ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
ierr = VecSetFromOptions(coordinates);CHKERRQ(ierr);
ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
for (v = 0; v < numPoints[0]; ++v) {
PetscInt off;
ierr = PetscSectionGetOffset(coordSection, v+firstVertex, &off);CHKERRQ(ierr);
for (d = 0; d < dim; ++d) {
coords[off+d] = vertexCoords[v*dim+d];
}
}
ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/*@
DMPlexReverseCell - Give a mesh cell the opposite orientation
Input Parameters:
+ dm - The DM
- cell - The cell number
Note: The modification of the DM is done in-place.
Level: advanced
.seealso: DMPlexOrient(), DMCreate(), DMPLEX
@*/
PetscErrorCode DMPlexReverseCell(DM dm, PetscInt cell)
{
/* Note that the reverse orientation ro of a face with orientation o is:
ro = o >= 0 ? -(faceSize - o) : faceSize + o
where faceSize is the size of the cone for the face.
*/
const PetscInt *cone, *coneO, *support;
PetscInt *revcone, *revconeO;
PetscInt maxConeSize, coneSize, supportSize, faceSize, cp, sp;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = DMPlexGetMaxSizes(dm, &maxConeSize, NULL);CHKERRQ(ierr);
ierr = DMGetWorkArray(dm, maxConeSize, MPIU_INT, &revcone);CHKERRQ(ierr);
ierr = DMGetWorkArray(dm, maxConeSize, MPIU_INT, &revconeO);CHKERRQ(ierr);
/* Reverse cone, and reverse orientations of faces */
ierr = DMPlexGetConeSize(dm, cell, &coneSize);CHKERRQ(ierr);
ierr = DMPlexGetCone(dm, cell, &cone);CHKERRQ(ierr);
ierr = DMPlexGetConeOrientation(dm, cell, &coneO);CHKERRQ(ierr);
for (cp = 0; cp < coneSize; ++cp) {
const PetscInt rcp = coneSize-cp-1;
ierr = DMPlexGetConeSize(dm, cone[rcp], &faceSize);CHKERRQ(ierr);
revcone[cp] = cone[rcp];
revconeO[cp] = coneO[rcp] >= 0 ? -(faceSize-coneO[rcp]) : faceSize+coneO[rcp];
}
ierr = DMPlexSetCone(dm, cell, revcone);CHKERRQ(ierr);
ierr = DMPlexSetConeOrientation(dm, cell, revconeO);CHKERRQ(ierr);
/* Reverse orientation of this cell in the support hypercells */
faceSize = coneSize;
ierr = DMPlexGetSupportSize(dm, cell, &supportSize);CHKERRQ(ierr);
ierr = DMPlexGetSupport(dm, cell, &support);CHKERRQ(ierr);
for (sp = 0; sp < supportSize; ++sp) {
ierr = DMPlexGetConeSize(dm, support[sp], &coneSize);CHKERRQ(ierr);
ierr = DMPlexGetCone(dm, support[sp], &cone);CHKERRQ(ierr);
ierr = DMPlexGetConeOrientation(dm, support[sp], &coneO);CHKERRQ(ierr);
for (cp = 0; cp < coneSize; ++cp) {
if (cone[cp] != cell) continue;
ierr = DMPlexInsertConeOrientation(dm, support[sp], cp, coneO[cp] >= 0 ? -(faceSize-coneO[cp]) : faceSize+coneO[cp]);CHKERRQ(ierr);
}
}
ierr = DMRestoreWorkArray(dm, maxConeSize, MPIU_INT, &revcone);CHKERRQ(ierr);
ierr = DMRestoreWorkArray(dm, maxConeSize, MPIU_INT, &revconeO);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
/* This interpolates faces for cells at some stratum */
static PetscErrorCode DMPlexInterpolateFaces_Internal(DM dm, PetscInt cellDepth, DM idm)
{
PetscHashIJKL faceTable;
PetscInt *pStart, *pEnd;
PetscInt cellDim, depth, faceDepth = cellDepth, numPoints = 0, faceSizeAll = 0, face, c, d;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = DMPlexGetDimension(dm, &cellDim);CHKERRQ(ierr);
ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
++depth;
++cellDepth;
cellDim -= depth - cellDepth;
ierr = PetscMalloc2(depth+1,PetscInt,&pStart,depth+1,PetscInt,&pEnd);CHKERRQ(ierr);
for (d = depth-1; d >= faceDepth; --d) {
ierr = DMPlexGetDepthStratum(dm, d, &pStart[d+1], &pEnd[d+1]);CHKERRQ(ierr);
}
ierr = DMPlexGetDepthStratum(dm, -1, NULL, &pStart[faceDepth]);CHKERRQ(ierr);
pEnd[faceDepth] = pStart[faceDepth];
for (d = faceDepth-1; d >= 0; --d) {
ierr = DMPlexGetDepthStratum(dm, d, &pStart[d], &pEnd[d]);CHKERRQ(ierr);
}
if (pEnd[cellDepth] > pStart[cellDepth]) {ierr = DMPlexGetFaces_Internal(dm, cellDim, pStart[cellDepth], NULL, &faceSizeAll, NULL);CHKERRQ(ierr);}
if (faceSizeAll > 4) SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Do not support interpolation of meshes with faces of %D vertices", faceSizeAll);
ierr = PetscHashIJKLCreate(&faceTable);CHKERRQ(ierr);
ierr = PetscHashIJKLSetMultivalued(faceTable, PETSC_FALSE);CHKERRQ(ierr);
for (c = pStart[cellDepth], face = pStart[faceDepth]; c < pEnd[cellDepth]; ++c) {
const PetscInt *cellFaces;
PetscInt numCellFaces, faceSize, cf, f;
ierr = DMPlexGetFaces_Internal(dm, cellDim, c, &numCellFaces, &faceSize, &cellFaces);CHKERRQ(ierr);
if (faceSize != faceSizeAll) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent face for cell %D of size %D != %D", c, faceSize, faceSizeAll);
for (cf = 0; cf < numCellFaces; ++cf) {
const PetscInt *cellFace = &cellFaces[cf*faceSize];
PetscHashIJKLKey key;
if (faceSize == 2) {
key.i = PetscMin(cellFace[0], cellFace[1]);
key.j = PetscMax(cellFace[0], cellFace[1]);
} else {
key.i = cellFace[0]; key.j = cellFace[1]; key.k = cellFace[2]; key.l = faceSize > 3 ? cellFace[3] : 0;
ierr = PetscSortInt(faceSize, (PetscInt *) &key);
}
ierr = PetscHashIJKLGet(faceTable, key, &f);CHKERRQ(ierr);
if (f < 0) {
ierr = PetscHashIJKLAdd(faceTable, key, face);CHKERRQ(ierr);
f = face++;
}
}
}
pEnd[faceDepth] = face;
ierr = PetscHashIJKLDestroy(&faceTable);CHKERRQ(ierr);
/* Count new points */
for (d = 0; d <= depth; ++d) {
numPoints += pEnd[d]-pStart[d];
}
ierr = DMPlexSetChart(idm, 0, numPoints);CHKERRQ(ierr);
/* Set cone sizes */
for (d = 0; d <= depth; ++d) {
PetscInt coneSize, p;
if (d == faceDepth) {
for (p = pStart[d]; p < pEnd[d]; ++p) {
/* I see no way to do this if we admit faces of different shapes */
ierr = DMPlexSetConeSize(idm, p, faceSizeAll);CHKERRQ(ierr);
}
} else if (d == cellDepth) {
for (p = pStart[d]; p < pEnd[d]; ++p) {
/* Number of cell faces may be different from number of cell vertices*/
ierr = DMPlexGetFaces_Internal(dm, cellDim, p, &coneSize, NULL, NULL);CHKERRQ(ierr);
ierr = DMPlexSetConeSize(idm, p, coneSize);CHKERRQ(ierr);
}
} else {
for (p = pStart[d]; p < pEnd[d]; ++p) {
ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr);
ierr = DMPlexSetConeSize(idm, p, coneSize);CHKERRQ(ierr);
}
}
}
ierr = DMSetUp(idm);CHKERRQ(ierr);
/* Get face cones from subsets of cell vertices */
if (faceSizeAll > 4) SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Do not support interpolation of meshes with faces of %D vertices", faceSizeAll);
ierr = PetscHashIJKLCreate(&faceTable);CHKERRQ(ierr);
ierr = PetscHashIJKLSetMultivalued(faceTable, PETSC_FALSE);CHKERRQ(ierr);
for (d = depth; d > cellDepth; --d) {
const PetscInt *cone;
PetscInt p;
for (p = pStart[d]; p < pEnd[d]; ++p) {
ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr);
ierr = DMPlexSetCone(idm, p, cone);CHKERRQ(ierr);
ierr = DMPlexGetConeOrientation(dm, p, &cone);CHKERRQ(ierr);
ierr = DMPlexSetConeOrientation(idm, p, cone);CHKERRQ(ierr);
}
}
for (c = pStart[cellDepth], face = pStart[faceDepth]; c < pEnd[cellDepth]; ++c) {
const PetscInt *cellFaces;
PetscInt numCellFaces, faceSize, cf, f;
ierr = DMPlexGetFaces_Internal(dm, cellDim, c, &numCellFaces, &faceSize, &cellFaces);CHKERRQ(ierr);
if (faceSize != faceSizeAll) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Inconsistent face for cell %D of size %D != %D", c, faceSize, faceSizeAll);
for (cf = 0; cf < numCellFaces; ++cf) {
const PetscInt *cellFace = &cellFaces[cf*faceSize];
PetscHashIJKLKey key;
if (faceSize == 2) {
key.i = PetscMin(cellFace[0], cellFace[1]);
key.j = PetscMax(cellFace[0], cellFace[1]);
} else {
key.i = cellFace[0]; key.j = cellFace[1]; key.k = cellFace[2]; key.l = faceSize > 3 ? cellFace[3] : 0;
ierr = PetscSortInt(faceSize, (PetscInt *) &key);
}
ierr = PetscHashIJKLGet(faceTable, key, &f);CHKERRQ(ierr);
if (f < 0) {
ierr = DMPlexSetCone(idm, face, cellFace);CHKERRQ(ierr);
ierr = PetscHashIJKLAdd(faceTable, key, face);CHKERRQ(ierr);
f = face++;
ierr = DMPlexInsertCone(idm, c, cf, f);CHKERRQ(ierr);
} else {
const PetscInt *cone;
PetscInt coneSize, ornt, i, j;
ierr = DMPlexInsertCone(idm, c, cf, f);CHKERRQ(ierr);
/* Orient face */
ierr = DMPlexGetConeSize(idm, f, &coneSize);CHKERRQ(ierr);
ierr = DMPlexGetCone(idm, f, &cone);CHKERRQ(ierr);
if (coneSize != faceSize) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of face vertices %D for face %D should be %D", coneSize, f, faceSize);
/* - First find the initial vertex */
for (i = 0; i < faceSize; ++i) if (cellFace[0] == cone[i]) break;
/* - Try forward comparison */
for (j = 0; j < faceSize; ++j) if (cellFace[j] != cone[(i+j)%faceSize]) break;
if (j == faceSize) {
if ((faceSize == 2) && (i == 1)) ornt = -2;
else ornt = i;
} else {
/* - Try backward comparison */
for (j = 0; j < faceSize; ++j) if (cellFace[j] != cone[(i+faceSize-j)%faceSize]) break;
if (j == faceSize) ornt = -(i+1);
else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Could not determine face orientation");
}
ierr = DMPlexInsertConeOrientation(idm, c, cf, ornt);CHKERRQ(ierr);
}
}
}
if (face != pEnd[faceDepth]) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_PLIB, "Invalid number of faces %D should be %D", face-pStart[faceDepth], pEnd[faceDepth]-pStart[faceDepth]);
ierr = PetscFree2(pStart,pEnd);CHKERRQ(ierr);
ierr = PetscHashIJKLDestroy(&faceTable);CHKERRQ(ierr);
ierr = DMPlexSymmetrize(idm);CHKERRQ(ierr);
ierr = DMPlexStratify(idm);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
void PETSC_STDCALL dmplexsetconeorientation_(DM dm,PetscInt *p, PetscInt coneOrientation[], int *__ierr ){
*__ierr = DMPlexSetConeOrientation(
(DM)PetscToPointer((dm) ),*p,coneOrientation);
}
/*
Simple square mesh:
22--8-23--9--24
| | |
13 2 14 3 15
| | |
19--6-20--7--21
| | |
10 0 11 1 12
| | |
16--4-17--5--18
*/
PetscErrorCode DMPlexCreateSquareMesh(DM dm, const PetscReal lower[], const PetscReal upper[], const PetscInt edges[])
{
PetscInt markerTop = 1;
PetscInt markerBottom = 1;
PetscInt markerRight = 1;
PetscInt markerLeft = 1;
PetscBool markerSeparate = PETSC_FALSE;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank);CHKERRQ(ierr);
ierr = PetscOptionsGetBool(((PetscObject) dm)->prefix, "-dm_plex_separate_marker", &markerSeparate, NULL);CHKERRQ(ierr);
if (markerSeparate) {
markerTop = 3;
markerBottom = 1;
markerRight = 2;
markerLeft = 4;
}
{
const PetscInt numXEdges = !rank ? edges[0] : 0;
const PetscInt numYEdges = !rank ? edges[1] : 0;
const PetscInt numXVertices = !rank ? edges[0]+1 : 0;
const PetscInt numYVertices = !rank ? edges[1]+1 : 0;
const PetscInt numTotXEdges = numXEdges*numYVertices;
const PetscInt numTotYEdges = numYEdges*numXVertices;
const PetscInt numVertices = numXVertices*numYVertices;
const PetscInt numEdges = numTotXEdges + numTotYEdges;
const PetscInt numFaces = numXEdges*numYEdges;
const PetscInt firstVertex = numFaces;
const PetscInt firstXEdge = numFaces + numVertices;
const PetscInt firstYEdge = numFaces + numVertices + numTotXEdges;
Vec coordinates;
PetscSection coordSection;
PetscScalar *coords;
PetscInt coordSize;
PetscInt v, vx, vy;
PetscInt f, fx, fy, e, ex, ey;
ierr = DMPlexSetChart(dm, 0, numFaces+numEdges+numVertices);CHKERRQ(ierr);
for (f = 0; f < numFaces; ++f) {
ierr = DMPlexSetConeSize(dm, f, 4);CHKERRQ(ierr);
}
for (e = firstXEdge; e < firstXEdge+numEdges; ++e) {
ierr = DMPlexSetConeSize(dm, e, 2);CHKERRQ(ierr);
}
ierr = DMSetUp(dm);CHKERRQ(ierr); /* Allocate space for cones */
/* Build faces */
for (fy = 0; fy < numYEdges; fy++) {
for (fx = 0; fx < numXEdges; fx++) {
const PetscInt face = fy*numXEdges + fx;
const PetscInt edgeL = firstYEdge + fx*numYEdges + fy;
const PetscInt edgeB = firstXEdge + fy*numXEdges + fx;
const PetscInt ornt[4] = {0, 0, -2, -2};
PetscInt cone[4];
cone[0] = edgeB; cone[1] = edgeL+numYEdges; cone[2] = edgeB+numXEdges; cone[3] = edgeL;
ierr = DMPlexSetCone(dm, face, cone);CHKERRQ(ierr);
ierr = DMPlexSetConeOrientation(dm, face, ornt);CHKERRQ(ierr);
}
}
/* Build Y edges*/
for (vx = 0; vx < numXVertices; vx++) {
for (ey = 0; ey < numYEdges; ey++) {
const PetscInt edge = firstYEdge + vx*numYEdges + ey;
const PetscInt vertex = firstVertex + ey*numXVertices + vx;
PetscInt cone[2];
cone[0] = vertex; cone[1] = vertex+numXVertices;
ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
if (vx == numXVertices-1) {
ierr = DMPlexSetLabelValue(dm, "marker", edge, markerRight);CHKERRQ(ierr);
ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerRight);CHKERRQ(ierr);
if (ey == numYEdges-1) {
ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerRight);CHKERRQ(ierr);
}
} else if (vx == 0) {
ierr = DMPlexSetLabelValue(dm, "marker", edge, markerLeft);CHKERRQ(ierr);
ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerLeft);CHKERRQ(ierr);
if (ey == numYEdges-1) {
ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerLeft);CHKERRQ(ierr);
}
}
}
}
/* Build X edges*/
for (vy = 0; vy < numYVertices; vy++) {
for (ex = 0; ex < numXEdges; ex++) {
const PetscInt edge = firstXEdge + vy*numXEdges + ex;
const PetscInt vertex = firstVertex + vy*numXVertices + ex;
PetscInt cone[2];
cone[0] = vertex; cone[1] = vertex+1;
ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
if (vy == numYVertices-1) {
ierr = DMPlexSetLabelValue(dm, "marker", edge, markerTop);CHKERRQ(ierr);
ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerTop);CHKERRQ(ierr);
if (ex == numXEdges-1) {
ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerTop);CHKERRQ(ierr);
}
} else if (vy == 0) {
ierr = DMPlexSetLabelValue(dm, "marker", edge, markerBottom);CHKERRQ(ierr);
ierr = DMPlexSetLabelValue(dm, "marker", cone[0], markerBottom);CHKERRQ(ierr);
if (ex == numXEdges-1) {
ierr = DMPlexSetLabelValue(dm, "marker", cone[1], markerBottom);CHKERRQ(ierr);
}
}
}
}
ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
ierr = DMPlexStratify(dm);CHKERRQ(ierr);
/* Build coordinates */
ierr = DMPlexGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
ierr = PetscSectionSetChart(coordSection, firstVertex, firstVertex+numVertices);CHKERRQ(ierr);
for (v = firstVertex; v < firstVertex+numVertices; ++v) {
ierr = PetscSectionSetDof(coordSection, v, 2);CHKERRQ(ierr);
}
ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
ierr = VecCreate(PetscObjectComm((PetscObject)dm), &coordinates);CHKERRQ(ierr);
ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
ierr = VecSetFromOptions(coordinates);CHKERRQ(ierr);
ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
for (vy = 0; vy < numYVertices; ++vy) {
for (vx = 0; vx < numXVertices; ++vx) {
coords[(vy*numXVertices+vx)*2+0] = lower[0] + ((upper[0] - lower[0])/numXEdges)*vx;
coords[(vy*numXVertices+vx)*2+1] = lower[1] + ((upper[1] - lower[1])/numYEdges)*vy;
}
}
ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
/* Two triangles separated by a zero-volume cell with 6 vertices
5--16--8
/ | | \
11 | | 12
/ | | \
3 0 10 2 14 1 6
\ | | /
9 | | 13
\ | | /
4--15--7
*/
PetscErrorCode CreateSimplexHybrid_2D(MPI_Comm comm, DM dm)
{
Vec coordinates;
PetscSection coordSection;
PetscScalar *coords;
PetscInt numVertices = 0, numEdges = 0, numCells = 0, cMax = PETSC_DETERMINE, fMax = PETSC_DETERMINE;
PetscInt firstVertex, firstEdge, coordSize;
PetscInt v, e;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
numVertices = 3 + 3;
numEdges = 6 + 2;
numCells = 3;
cMax = 2;
fMax = 15;
}
firstVertex = numCells;
firstEdge = numCells + numVertices;
ierr = DMPlexSetChart(dm, 0, numCells+numEdges+numVertices);CHKERRQ(ierr);
if (numCells) {
ierr = DMPlexSetConeSize(dm, 0, 3);CHKERRQ(ierr);
ierr = DMPlexSetConeSize(dm, 1, 3);CHKERRQ(ierr);
ierr = DMPlexSetConeSize(dm, 2, 4);CHKERRQ(ierr);
}
for(e = firstEdge; e < firstEdge+numEdges; ++e) {
ierr = DMPlexSetConeSize(dm, e, 2);CHKERRQ(ierr);
}
ierr = DMSetUp(dm);CHKERRQ(ierr); /* Allocate space for cones */
ierr = DMPlexSetHybridBounds(dm, cMax, PETSC_DETERMINE, fMax, PETSC_DETERMINE);CHKERRQ(ierr); /* Indicate a hybrid mesh */
/* Build cells */
if (numCells > 0) {
const PetscInt cone[3] = {9, 10, 11};
const PetscInt ornt[3] = {0, 0, 0};
const PetscInt cell = 0;
ierr = DMPlexSetCone(dm, cell, cone);CHKERRQ(ierr);
ierr = DMPlexSetConeOrientation(dm, cell, ornt);CHKERRQ(ierr);
}
if (numCells > 1) {
const PetscInt cone[3] = {12, 14, 13};
const PetscInt ornt[3] = { 0, -2, 0};
const PetscInt cell = 1;
ierr = DMPlexSetCone(dm, cell, cone);CHKERRQ(ierr);
ierr = DMPlexSetConeOrientation(dm, cell, ornt);CHKERRQ(ierr);
}
if (numCells > 2) {
const PetscInt cone[4] = {10, 14, 15, 16};
const PetscInt ornt[4] = { 0, 0, 0, 0};
const PetscInt cell = 2;
ierr = DMPlexSetCone(dm, cell, cone);CHKERRQ(ierr);
ierr = DMPlexSetConeOrientation(dm, cell, ornt);CHKERRQ(ierr);
}
/* Build edges*/
if (numEdges > 0) {
const PetscInt cone[2] = {3, 4};
const PetscInt edge = 9;
ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
ierr = DMPlexSetLabelValue(dm, "marker", edge, 1);CHKERRQ(ierr);
}
if (numEdges > 1) {
const PetscInt cone[2] = {4, 5};
const PetscInt edge = 10;
ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
}
if (numEdges > 2) {
const PetscInt cone[2] = {5, 3};
const PetscInt edge = 11;
ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
ierr = DMPlexSetLabelValue(dm, "marker", edge, 1);CHKERRQ(ierr);
}
if (numEdges > 3) {
const PetscInt cone[2] = {6, 8};
const PetscInt edge = 12;
ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
ierr = DMPlexSetLabelValue(dm, "marker", edge, 1);CHKERRQ(ierr);
}
if (numEdges > 4) {
const PetscInt cone[2] = {7, 6};
const PetscInt edge = 13;
ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
ierr = DMPlexSetLabelValue(dm, "marker", edge, 1);CHKERRQ(ierr);
}
if (numEdges > 5) {
const PetscInt cone[2] = {7, 8};
const PetscInt edge = 14;
ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
}
if (numEdges > 6) {
const PetscInt cone[2] = {4, 7};
const PetscInt edge = 15;
ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
}
if (numEdges > 7) {
const PetscInt cone[2] = {5, 8};
const PetscInt edge = 16;
ierr = DMPlexSetCone(dm, edge, cone);CHKERRQ(ierr);
}
ierr = DMPlexSymmetrize(dm);CHKERRQ(ierr);
ierr = DMPlexStratify(dm);CHKERRQ(ierr);
/* Build coordinates */
ierr = DMPlexGetCoordinateSection(dm, &coordSection);CHKERRQ(ierr);
ierr = PetscSectionSetChart(coordSection, firstVertex, firstVertex+numVertices);CHKERRQ(ierr);
for(v = firstVertex; v < firstVertex+numVertices; ++v) {
ierr = PetscSectionSetDof(coordSection, v, 2);CHKERRQ(ierr);
}
ierr = PetscSectionSetUp(coordSection);CHKERRQ(ierr);
ierr = PetscSectionGetStorageSize(coordSection, &coordSize);CHKERRQ(ierr);
ierr = VecCreate(((PetscObject) dm)->comm, &coordinates);CHKERRQ(ierr);
ierr = VecSetSizes(coordinates, coordSize, PETSC_DETERMINE);CHKERRQ(ierr);
ierr = VecSetFromOptions(coordinates);CHKERRQ(ierr);
ierr = VecGetArray(coordinates, &coords);CHKERRQ(ierr);
if (numVertices) {
coords[0] = -0.5; coords[1] = 0.5;
coords[2] = -0.2; coords[3] = 0.0;
coords[4] = -0.2; coords[5] = 1.0;
coords[6] = 0.5; coords[7] = 0.5;
coords[8] = 0.2; coords[9] = 0.0;
coords[10] = 0.2; coords[11] = 1.0;
}
ierr = VecRestoreArray(coordinates, &coords);CHKERRQ(ierr);
ierr = DMSetCoordinatesLocal(dm, coordinates);CHKERRQ(ierr);
ierr = VecDestroy(&coordinates);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
