PetscErrorCode CreateSimplex_3D(MPI_Comm comm, DM dm)
{
PetscInt depth = 1, testNum = 0, p;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
switch (testNum) {
case 0:
{
PetscInt numPoints[2] = {5, 2};
PetscInt coneSize[23] = {4, 4, 0, 0, 0, 0, 0};
PetscInt cones[8] = {2, 4, 3, 5, 3, 4, 6, 5};
PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[15] = {0.0, 0.0, -0.5, 0.0, -0.5, 0.0, 1.0, 0.0, 0.0, 0.0, 0.5, 0.0, 0.0, 0.0, 0.5};
PetscInt markerPoints[8] = {2, 1, 3, 1, 4, 1, 5, 1};
ierr = DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
for (p = 0; p < 4; ++p) {
ierr = DMSetLabelValue(dm, "marker", markerPoints[p*2], markerPoints[p*2+1]);CHKERRQ(ierr);
}
}
break;
default:
SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "No test mesh %d", testNum);
}
} else {
PetscInt numPoints[2] = {0, 0};
ierr = DMPlexCreateFromDAG(dm, depth, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
PetscErrorCode CreateTensorProduct_3D(MPI_Comm comm, PetscInt testNum, DM *dm)
{
DM idm = NULL;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
switch (testNum) {
case 0:
{
PetscInt numPoints[2] = {12, 2};
PetscInt coneSize[14] = {8, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
PetscInt cones[16] = {2, 3, 4, 5, 6, 7, 8, 9, 5, 4, 10, 11, 7, 12, 13, 8};
PetscInt coneOrientations[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[36] = {-1.0, -0.5, -0.5, -1.0, 0.5, -0.5, 0.0, 0.5, -0.5, 0.0, -0.5, -0.5,
-1.0, -0.5, 0.5, 0.0, -0.5, 0.5, 0.0, 0.5, 0.5, -1.0, 0.5, 0.5,
1.0, 0.5, -0.5, 1.0, -0.5, -0.5, 1.0, -0.5, 0.5, 1.0, 0.5, 0.5};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
}
break;
case 1:
{
PetscInt numPoints[2] = {8, 1};
PetscInt coneSize[9] = {8, 0, 0, 0, 0, 0, 0, 0, 0};
PetscInt cones[8] = {1, 2, 3, 4, 5, 6, 7, 8};
PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[24] = {-1.0, -1.0, -1.0, -1.0, 1.0, -1.0, 1.0, 1.0, -1.0, 1.0, -1.0, -1.0,
-1.0, -1.0, 1.0, 1.0, -1.0, 1.0, 1.0, 1.0, 1.0, -1.0, 1.0, 1.0};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
}
break;
default: SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "No test mesh %d", testNum);
}
} else {
PetscInt numPoints[4] = {0, 0, 0, 0};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
}
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = DMPlexCopyCoordinates(*dm, idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-in_dm_view");CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = idm;
PetscFunctionReturn(0);
}
static PetscErrorCode CreateHex_3D(MPI_Comm comm, DM *newdm)
{
DM dm;
PetscInt numPoints[2] = {12, 2};
PetscInt coneSize[14] = {8, 8, 0,0,0,0,0,0,0,0,0,0,0,0};
PetscInt cones[16] = {2,5,4,3,6,7,8,9, 3,4,11,10,7,12,13,8};
PetscInt coneOrientations[16] = {0,0,0,0,0,0,0,0, 0,0, 0, 0,0, 0, 0,0};
PetscScalar vertexCoords[36] = {-0.5,0.0,0.0, 0.0,0.0,0.0, 0.0,1.0,0.0, -0.5,1.0,0.0,
-0.5,0.0,1.0, 0.0,0.0,1.0, 0.0,1.0,1.0, -0.5,1.0,1.0,
0.5,0.0,0.0, 0.5,1.0,0.0, 0.5,0.0,1.0, 0.5,1.0,1.0};
PetscInt markerPoints[24] = {2,1,3,1,4,1,5,1,6,1,7,1,8,1,9,1,10,1,11,1,12,1,13,1};
PetscInt dim = 3, depth = 1, p;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = DMCreate(comm, &dm);CHKERRQ(ierr);
ierr = PetscObjectSetName((PetscObject) dm, "hexahedral");CHKERRQ(ierr);
ierr = DMSetType(dm, DMPLEX);CHKERRQ(ierr);
ierr = DMSetDimension(dm, dim);CHKERRQ(ierr);
ierr = DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
for(p = 0; p < 12; ++p) {
ierr = DMSetLabelValue(dm, "marker", markerPoints[p*2], markerPoints[p*2+1]);CHKERRQ(ierr);
}
*newdm = dm;
PetscFunctionReturn(0);
}
PetscErrorCode CreateTensorProductHybrid_2D(MPI_Comm comm, PetscInt testNum, DM *dm)
{
DM idm, hdm = NULL;
DMLabel faultLabel, hybridLabel;
PetscInt p;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
PetscInt numPoints[2] = {6, 2};
PetscInt coneSize[8] = {4, 4, 0, 0, 0, 0, 0, 0};
PetscInt cones[8] = {2, 3, 4, 5, 3, 6, 7, 4,};
PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[12] = {-1.0, -0.5, 0.0, -0.5, 0.0, 0.5, -1.0, 0.5, 1.0, -0.5, 1.0, 0.5};
PetscInt faultPoints[2] = {3, 4};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
for(p = 0; p < 2; ++p) {ierr = DMSetLabelValue(*dm, "fault", faultPoints[p], 1);CHKERRQ(ierr);}
ierr = PetscObjectSetOptionsPrefix((PetscObject) *dm, "orig_");CHKERRQ(ierr);
ierr = DMSetFromOptions(*dm);CHKERRQ(ierr);
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = PetscObjectSetOptionsPrefix((PetscObject) idm, "in_");CHKERRQ(ierr);
ierr = DMSetFromOptions(idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-dm_view");CHKERRQ(ierr);
ierr = DMGetLabel(*dm, "fault", &faultLabel);CHKERRQ(ierr);
ierr = DMPlexCreateHybridMesh(idm, faultLabel, NULL, &hybridLabel, NULL, NULL, &hdm);CHKERRQ(ierr);
ierr = DMLabelDestroy(&hybridLabel);CHKERRQ(ierr);
} else {
PetscInt numPoints[3] = {0, 0, 0};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-in_dm_view");CHKERRQ(ierr);
ierr = DMPlexCreateHybridMesh(idm, NULL, NULL, NULL, NULL, NULL, &hdm);CHKERRQ(ierr);
}
ierr = DMDestroy(&idm);CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = hdm;
PetscFunctionReturn(0);
}
/* Two segments
2-------0-------3-------1-------4
become
4---0---7---1---5---2---8---3---6
*/
PetscErrorCode CreateSimplex_1D(MPI_Comm comm, DM *dm)
{
PetscInt depth = 1;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
PetscInt numPoints[2] = {3, 2};
PetscInt coneSize[5] = {2, 2, 0, 0, 0};
PetscInt cones[4] = {2, 3, 3, 4};
PetscInt coneOrientations[16] = {0, 0, 0, 0};
PetscScalar vertexCoords[3] = {-1.0, 0.0, 1.0};
ierr = DMPlexCreateFromDAG(*dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
} else {
PetscInt numPoints[2] = {0, 0};
ierr = DMPlexCreateFromDAG(*dm, depth, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
/* Two quadrilaterals
5----10-----4----14-----7
| | |
| | |
| | |
11 0 9 1 13
| | |
| | |
| | |
2-----8-----3----12-----6
*/
PetscErrorCode CreateTensorProduct_2D(MPI_Comm comm, PetscInt testNum, DM *dm)
{
PetscInt depth = 2;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
PetscInt numPoints[3] = {6, 7, 2};
PetscInt coneSize[15] = {4, 4, 0, 0, 0, 0, 0, 0, 2, 2, 2, 2, 2, 2, 2};
PetscInt cones[22] = {8, 9, 10, 11, 12, 13, 14, 9, 2, 3, 3, 4, 4, 5, 5, 2, 3, 6, 6, 7, 7, 4};
PetscInt coneOrientations[22] = {0, 0, 0, 0, 0, 0, 0, -2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[12] = {-1.0, -0.5, 0.0, -0.5, 0.0, 0.5, -1.0, 0.5, 1.0, -0.5, 1.0, 0.5};
ierr = DMPlexCreateFromDAG(*dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
} else {
PetscInt numPoints[3] = {0, 0, 0};
ierr = DMPlexCreateFromDAG(*dm, depth, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
/* Two triangles
4
/ | \
8 | 10
/ | \
2 0 7 1 5
\ | /
6 | 9
\ | /
3
Becomes
10
/ | \
21 | 26
/ | \
14 2 20 4 16
/|\ | /|\
22 | 28 | 32 | 25
/ | \ | / | 6\
8 29 3 13 7 31 11
\0 | / | \ | /
17 | 27 | 30 | 24
\|/ | \|/
12 1 19 5 15
\ | /
18 | 23
\ | /
9
*/
PetscErrorCode CreateSimplex_2D(MPI_Comm comm, DM *dm)
{
PetscInt depth = 2;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
PetscInt numPoints[3] = {4, 5, 2};
PetscInt coneSize[11] = {3, 3, 0, 0, 0, 0, 2, 2, 2, 2, 2};
PetscInt cones[16] = {6, 7, 8, 7, 9, 10, 2, 3, 3, 4, 4, 2, 3, 5, 5, 4};
PetscInt coneOrientations[16] = {0, 0, 0, -2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[8] = {-0.5, 0.0, 0.0, -0.5, 0.0, 0.5, 0.5, 0.0};
ierr = DMPlexCreateFromDAG(*dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
} else {
PetscInt numPoints[3] = {0, 0, 0};
ierr = DMPlexCreateFromDAG(*dm, depth, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
}
PetscFunctionReturn(0);
}
static PetscErrorCode CreateSimplex_2D(MPI_Comm comm, DM *newdm)
{
DM dm;
PetscInt numPoints[2] = {4, 2};
PetscInt coneSize[6] = {3, 3, 0, 0, 0, 0};
PetscInt cones[6] = {2, 3, 4, 5, 4, 3};
PetscInt coneOrientations[6] = {0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[8] = {-0.5, 0.5, 0.0, 0.0, 0.0, 1.0, 0.5, 0.5};
PetscInt markerPoints[8] = {2, 1, 3, 1, 4, 1, 5, 1};
PetscInt dim = 2, depth = 1, p;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = DMCreate(comm, &dm);CHKERRQ(ierr);
ierr = PetscObjectSetName((PetscObject) dm, "triangular");CHKERRQ(ierr);
ierr = DMSetType(dm, DMPLEX);CHKERRQ(ierr);
ierr = DMSetDimension(dm, dim);CHKERRQ(ierr);
ierr = DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
for (p = 0; p < 4; ++p) {
ierr = DMSetLabelValue(dm, "marker", markerPoints[p*2], markerPoints[p*2+1]);CHKERRQ(ierr);
}
*newdm = dm;
PetscFunctionReturn(0);
}
PetscErrorCode CreateTensorProductHybrid_3D(MPI_Comm comm, PetscInt testNum, DM *dm)
{
DM idm, hdm = NULL;
DMLabel faultLabel;
PetscInt p;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
switch (testNum) {
case 0:
{
PetscInt numPoints[2] = {12, 2};
PetscInt coneSize[14] = {8, 8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
PetscInt cones[16] = {2, 3, 4, 5, 6, 7, 8, 9, 5, 4, 10, 11, 7, 12, 13, 8};
PetscInt coneOrientations[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[36] = {-1.0, -0.5, -0.5, -1.0, 0.5, -0.5, 0.0, 0.5, -0.5, 0.0, -0.5, -0.5,
-1.0, -0.5, 0.5, 0.0, -0.5, 0.5, 0.0, 0.5, 0.5, -1.0, 0.5, 0.5,
1.0, 0.5, -0.5, 1.0, -0.5, -0.5, 1.0, -0.5, 0.5, 1.0, 0.5, 0.5};
PetscInt faultPoints[4] = {2, 3, 5, 6};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
for(p = 0; p < 4; ++p) {ierr = DMSetLabelValue(*dm, "fault", faultPoints[p], 1);CHKERRQ(ierr);}
}
break;
case 1:
{
PetscInt numPoints[2] = {30, 7};
PetscInt coneSize[37] = {8,8,8,8,8,8,8, 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
PetscInt cones[56] = { 8, 21, 20, 7, 13, 12, 23, 24,
14, 15, 10, 9, 13, 8, 21, 24,
15, 16, 11, 10, 24, 21, 22, 25,
30, 29, 28, 21, 35, 24, 33, 34,
24, 21, 30, 35, 25, 36, 31, 22,
27, 20, 21, 28, 32, 33, 24, 23,
15, 24, 13, 14, 19, 18, 17, 26};
PetscInt coneOrientations[56] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
PetscScalar vertexCoords[90] = {-2.0, -2.0, -2.0, -2.0, -1.0, -2.0, -3.0, 0.0, -2.0, -2.0, 1.0, -2.0, -2.0, 2.0, -2.0, -2.0, -2.0, 0.0,
-2.0, -1.0, 0.0, -3.0, 0.0, 0.0, -2.0, 1.0, 0.0, -2.0, 2.0, 0.0, -2.0, -1.0, 2.0, -3.0, 0.0, 2.0,
-2.0, 1.0, 2.0, 0.0, -2.0, -2.0, 0.0, 0.0, -2.0, 0.0, 2.0, -2.0, 0.0, -2.0, 0.0, 0.0, 0.0, 0.0,
0.0, 2.0, 0.0, 0.0, 0.0, 2.0, 2.0, -2.0, -2.0, 2.0, -1.0, -2.0, 3.0, 0.0, -2.0, 2.0, 1.0, -2.0,
2.0, 2.0, -2.0, 2.0, -2.0, 0.0, 2.0, -1.0, 0.0, 3.0, 0.0, 0.0, 2.0, 1.0, 0.0, 2.0, 2.0, 0.0};
PetscInt faultPoints[6] = {20, 21, 22, 23, 24, 25};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
for(p = 0; p < 6; ++p) {ierr = DMSetLabelValue(*dm, "fault", faultPoints[p], 1);CHKERRQ(ierr);}
}
break;
default: SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "No test mesh %d", testNum);
}
ierr = PetscObjectSetOptionsPrefix((PetscObject) *dm, "orig_");CHKERRQ(ierr);
ierr = DMSetFromOptions(*dm);CHKERRQ(ierr);
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = PetscObjectSetOptionsPrefix((PetscObject) idm, "in_");CHKERRQ(ierr);
ierr = DMSetFromOptions(idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-dm_view");CHKERRQ(ierr);
ierr = DMGetLabel(*dm, "fault", &faultLabel);CHKERRQ(ierr);
ierr = DMPlexCreateHybridMesh(idm, faultLabel, NULL, NULL, NULL, NULL, &hdm);CHKERRQ(ierr);
ierr = DMDestroy(&idm);CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = hdm;
} else {
PetscInt numPoints[4] = {0, 0, 0, 0};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-in_dm_view");CHKERRQ(ierr);
ierr = DMPlexCreateHybridMesh(idm, NULL, NULL, NULL, NULL, NULL, &hdm);CHKERRQ(ierr);
ierr = DMDestroy(&idm);CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = hdm;
}
PetscFunctionReturn(0);
}
/* Two tetrahedrons separated by a zero-volume cell with 6 vertices
cell 6 ___33___10______ cell
0 / | \ |\ \ 1
21 | 23 | 29 27
/12 24 14\ 30 \ \
3-20-|----5--32-|---9--26--7
\ 13| 11/ |18 / /
19 | 22 | 28 25
\ | / |/ /
4----31----8------
cell 2
*/
PetscErrorCode CreateSimplexHybrid_3D(MPI_Comm comm, PetscInt testNum, DM *dm)
{
DM idm, hdm = NULL;
DMLabel faultLabel, hybridLabel;
PetscInt p;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
switch (testNum) {
case 0:
{
PetscInt numPoints[2] = {5, 2};
PetscInt coneSize[7] = {4, 4, 0, 0, 0, 0, 0};
PetscInt cones[8] = {4, 3, 5, 2, 5, 3, 4, 6};
PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[15] = {-1.0, 0.0, 0.0, 0.0, -1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0, 0.0, 0.0};
PetscInt faultPoints[3] = {3, 4, 5};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
for(p = 0; p < 3; ++p) {ierr = DMSetLabelValue(*dm, "fault", faultPoints[p], 1);CHKERRQ(ierr);}
}
break;
case 1:
{
/* Tets 0,3,5 and 1,2,4 */
PetscInt numPoints[2] = {9, 6};
PetscInt coneSize[15] = {4, 4, 4, 4, 4, 4, 0, 0, 0, 0, 0, 0, 0, 0, 0};
PetscInt cones[24] = { 7, 9, 8, 6, 11, 9, 13, 14, 10, 13, 11, 9,
10, 9, 11, 7, 9, 13, 14, 12, 7, 11, 8, 9};
PetscInt coneOrientations[24] = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[27] = {-2.0, -1.0, 0.0, -2.0, 0.0, 0.0, -2.0, 0.0, 1.0,
0.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0,
2.0, -1.0, 0.0, 2.0, 0.0, 0.0, 2.0, 0.0, 1.0};
PetscInt faultPoints[3] = {9, 10, 11};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
for(p = 0; p < 3; ++p) {ierr = DMSetLabelValue(*dm, "fault", faultPoints[p], 1);CHKERRQ(ierr);}
}
break;
default: SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "No test mesh %d", testNum);
}
ierr = PetscObjectSetOptionsPrefix((PetscObject) *dm, "orig_");CHKERRQ(ierr);
ierr = DMSetFromOptions(*dm);CHKERRQ(ierr);
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = PetscObjectSetOptionsPrefix((PetscObject) idm, "in_");CHKERRQ(ierr);
ierr = DMSetFromOptions(idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-dm_view");CHKERRQ(ierr);
ierr = DMGetLabel(*dm, "fault", &faultLabel);CHKERRQ(ierr);
ierr = DMPlexCreateHybridMesh(idm, faultLabel, NULL, &hybridLabel, NULL, NULL, &hdm);CHKERRQ(ierr);
ierr = DMLabelDestroy(&hybridLabel);CHKERRQ(ierr);
ierr = DMDestroy(&idm);CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = hdm;
} else {
PetscInt numPoints[4] = {0, 0, 0, 0};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-in_dm_view");CHKERRQ(ierr);
ierr = DMPlexCreateHybridMesh(idm, NULL, NULL, NULL, NULL, NULL, &hdm);CHKERRQ(ierr);
ierr = DMDestroy(&idm);CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = hdm;
}
PetscFunctionReturn(0);
}
/* Two tetrahedrons
cell 5 5______ cell
0 / | \ |\ \ 1
17 | 18 | 18 13 21
/8 19 10\ 19 \ \
2-14-|----4 | 4--22--6
\ 9 | 7 / |10 / /
16 | 15 | 15 12 20
\ | / |/ /
3 3------
*/
PetscErrorCode CreateSimplex_3D(MPI_Comm comm, PetscInt testNum, DM *dm)
{
DM idm;
PetscInt depth = 3;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
switch (testNum) {
case 0:
{
PetscInt numPoints[4] = {5, 9, 7, 2};
PetscInt coneSize[23] = {4, 4, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 2, 2, 2, 2, 2, 2, 2, 2, 2};
PetscInt cones[47] = { 7, 8, 9, 10, 10, 11, 12, 13, 14, 15, 16, 17, 18, 14, 16, 19, 17, 15, 18, 19, 20, 21, 19, 15, 22, 20, 18, 21, 22, 2, 4, 4, 3, 3, 2, 2, 5, 5, 4, 3, 5, 3, 6, 6, 5, 4, 6};
PetscInt coneOrientations[47] = { 0, 0, 0, 0, -3, 0, 0, 0, 0, 0, 0, 0, 0, -2, -2, 0, -2, -2, -2, -2, 0, 0, -2, -2, 0, -2, -2, -2, -2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[15] = {0.0, 0.0, -0.5, 0.0, -0.5, 0.0, 1.0, 0.0, 0.0, 0.0, 0.5, 0.0, 0.0, 0.0, 0.5};
ierr = DMPlexCreateFromDAG(*dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
}
break;
case 1:
{
PetscInt numPoints[2] = {5, 2};
PetscInt coneSize[7] = {4, 4, 0, 0, 0, 0, 0};
PetscInt cones[8] = {4, 3, 5, 2, 5, 3, 4, 6};
PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[15] = {-1.0, 0.0, 0.0, 0.0, -1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 1.0, 0.0, 1.0, 0.0, 0.0};
depth = 1;
ierr = DMPlexCreateFromDAG(*dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-in_dm_view");CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = idm;
}
break;
case 2:
{
PetscInt numPoints[2] = {4, 1};
PetscInt coneSize[5] = {4, 0, 0, 0, 0};
PetscInt cones[4] = {2, 3, 4, 1};
PetscInt coneOrientations[4] = {0, 0, 0, 0};
PetscScalar vertexCoords[12] = {0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0};
depth = 1;
ierr = DMPlexCreateFromDAG(*dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-in_dm_view");CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = idm;
}
break;
default: SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "No test mesh %d", testNum);
}
} else {
PetscInt numPoints[4] = {0, 0, 0, 0};
ierr = DMPlexCreateFromDAG(*dm, depth, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
switch (testNum) {
case 1:
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-in_dm_view");CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = idm;
break;
}
}
PetscFunctionReturn(0);
}
PetscErrorCode TestTriangle(MPI_Comm comm, PetscBool interpolate, PetscBool transform)
{
DM dm;
PetscRandom r, ang, ang2;
PetscInt dim, t;
PetscErrorCode ierr;
PetscFunctionBegin;
/* Create reference triangle */
dim = 2;
ierr = DMCreate(comm, &dm);CHKERRQ(ierr);
ierr = PetscObjectSetName((PetscObject) dm, "triangle");CHKERRQ(ierr);
ierr = DMSetType(dm, DMPLEX);CHKERRQ(ierr);
ierr = DMPlexSetDimension(dm, dim);CHKERRQ(ierr);
{
PetscInt numPoints[2] = {3, 1};
PetscInt coneSize[4] = {3, 0, 0, 0};
PetscInt cones[3] = {1, 2, 3};
PetscInt coneOrientations[3] = {0, 0, 0};
PetscScalar vertexCoords[6] = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0};
ierr = DMPlexCreateFromDAG(dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
if (interpolate) {
DM idm;
ierr = DMPlexInterpolate(dm, &idm);CHKERRQ(ierr);
ierr = PetscObjectSetName((PetscObject) idm, "triangle");CHKERRQ(ierr);
ierr = DMPlexCopyCoordinates(dm, idm);CHKERRQ(ierr);
ierr = DMDestroy(&dm);CHKERRQ(ierr);
dm = idm;
}
ierr = DMSetFromOptions(dm);CHKERRQ(ierr);
}
/* Check reference geometry: determinant is scaled by reference volume (2.0) */
{
PetscReal v0Ex[2] = {-1.0, -1.0};
PetscReal JEx[4] = {1.0, 0.0, 0.0, 1.0};
PetscReal invJEx[4] = {1.0, 0.0, 0.0, 1.0};
PetscReal detJEx = 1.0;
PetscReal centroidEx[2] = {-0.333333333333, -0.333333333333};
PetscReal normalEx[2] = {0.0, 0.0};
PetscReal volEx = 2.0;
ierr = CheckFEMGeometry(dm, 0, dim, v0Ex, JEx, invJEx, detJEx);CHKERRQ(ierr);
if (interpolate) {ierr = CheckFVMGeometry(dm, 0, dim, centroidEx, normalEx, volEx);CHKERRQ(ierr);}
}
/* Check random triangles: rotate, scale, then translate */
if (transform) {
ierr = PetscRandomCreate(PETSC_COMM_SELF, &r);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(r);CHKERRQ(ierr);
ierr = PetscRandomSetInterval(r, 0.0, 10.0);CHKERRQ(ierr);
ierr = PetscRandomCreate(PETSC_COMM_SELF, &ang);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(ang);CHKERRQ(ierr);
ierr = PetscRandomSetInterval(ang, 0.0, 2*PETSC_PI);CHKERRQ(ierr);
for (t = 0; t < 100; ++t) {
PetscScalar vertexCoords[6] = {-1.0, -1.0, 1.0, -1.0, -1.0, 1.0}, trans[2];
PetscReal v0Ex[2] = {-1.0, -1.0};
PetscReal JEx[4] = {1.0, 0.0, 0.0, 1.0}, R[4], rot[2], rotM[4];
PetscReal invJEx[4] = {1.0, 0.0, 0.0, 1.0};
PetscReal detJEx = 1.0, scale, phi;
PetscReal centroidEx[2] = {-0.333333333333, -0.333333333333};
PetscReal normalEx[2] = {0.0, 0.0};
PetscReal volEx = 2.0;
PetscInt d, e, f, p;
ierr = PetscRandomGetValueReal(r, &scale);CHKERRQ(ierr);
ierr = PetscRandomGetValueReal(ang, &phi);CHKERRQ(ierr);
R[0] = cos(phi); R[1] = -sin(phi);
R[2] = sin(phi); R[3] = cos(phi);
for (p = 0; p < 3; ++p) {
for (d = 0; d < dim; ++d) {
for (e = 0, rot[d] = 0.0; e < dim; ++e) {
rot[d] += R[d*dim+e] * vertexCoords[p*dim+e];
}
}
for (d = 0; d < dim; ++d) vertexCoords[p*dim+d] = rot[d];
}
for (d = 0; d < dim; ++d) {
for (e = 0, rot[d] = 0.0; e < dim; ++e) {
rot[d] += R[d*dim+e] * centroidEx[e];
}
}
for (d = 0; d < dim; ++d) centroidEx[d] = rot[d];
for (d = 0; d < dim; ++d) {
for (e = 0; e < dim; ++e) {
for (f = 0, rotM[d*dim+e] = 0.0; f < dim; ++f) {
rotM[d*dim+e] += R[d*dim+f] * JEx[f*dim+e];
}
}
}
for (d = 0; d < dim; ++d) {
for (e = 0; e < dim; ++e) {
JEx[d*dim+e] = rotM[d*dim+e];
}
}
for (d = 0; d < dim; ++d) {
for (e = 0; e < dim; ++e) {
for (f = 0, rotM[d*dim+e] = 0.0; f < dim; ++f) {
rotM[d*dim+e] += invJEx[d*dim+f] * R[e*dim+f];
}
}
}
for (d = 0; d < dim; ++d) {
for (e = 0; e < dim; ++e) {
invJEx[d*dim+e] = rotM[d*dim+e];
}
}
for (d = 0; d < dim; ++d) {
ierr = PetscRandomGetValueReal(r, &trans[d]);CHKERRQ(ierr);
for (p = 0; p < 3; ++p) {
vertexCoords[p*dim+d] *= scale;
vertexCoords[p*dim+d] += trans[d];
}
v0Ex[d] = vertexCoords[d];
for (e = 0; e < dim; ++e) {
JEx[d*dim+e] *= scale;
invJEx[d*dim+e] /= scale;
}
detJEx *= scale;
centroidEx[d] *= scale;
centroidEx[d] += trans[d];
volEx *= scale;
}
ierr = ChangeCoordinates(dm, dim, vertexCoords);CHKERRQ(ierr);
ierr = CheckFEMGeometry(dm, 0, dim, v0Ex, JEx, invJEx, detJEx);CHKERRQ(ierr);
if (interpolate) {ierr = CheckFVMGeometry(dm, 0, dim, centroidEx, normalEx, volEx);CHKERRQ(ierr);}
}
ierr = PetscRandomDestroy(&r);CHKERRQ(ierr);
ierr = PetscRandomDestroy(&ang);CHKERRQ(ierr);
}
/* Move to 3D: Check reference geometry: determinant is scaled by reference volume (2.0) */
dim = 3;
{
PetscScalar vertexCoords[9] = {-1.0, -1.0, 0.0, 1.0, -1.0, 0.0, -1.0, 1.0, 0.0};
PetscReal v0Ex[3] = {-1.0, -1.0, 0.0};
PetscReal JEx[9] = {1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0};
PetscReal invJEx[9] = {1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0};
PetscReal detJEx = 1.0;
PetscReal centroidEx[3] = {-0.333333333333, -0.333333333333, 0.0};
PetscReal normalEx[3] = {0.0, 0.0, 1.0};
PetscReal volEx = 2.0;
ierr = ChangeCoordinates(dm, dim, vertexCoords);CHKERRQ(ierr);
ierr = CheckFEMGeometry(dm, 0, dim, v0Ex, JEx, invJEx, detJEx);CHKERRQ(ierr);
if (interpolate) {ierr = CheckFVMGeometry(dm, 0, dim, centroidEx, normalEx, volEx);CHKERRQ(ierr);}
}
/* Rotated reference element */
{
PetscScalar vertexCoords[9] = {0.0, -1.0, -1.0, 0.0, 1.0, -1.0, 0.0, -1.0, 1.0};
PetscReal v0Ex[3] = {0.0, -1.0, -1.0};
PetscReal JEx[9] = {0.0, 0.0, 1.0, 1.0, 0.0, 0.0, 0.0, 1.0, 0.0};
PetscReal invJEx[9] = {0.0, 1.0, 0.0, 0.0, 0.0, 1.0, 1.0, 0.0, 0.0};
PetscReal detJEx = 1.0;
PetscReal centroidEx[3] = {0.0, -0.333333333333, -0.333333333333};
PetscReal normalEx[3] = {1.0, 0.0, 0.0};
PetscReal volEx = 2.0;
ierr = ChangeCoordinates(dm, dim, vertexCoords);CHKERRQ(ierr);
ierr = CheckFEMGeometry(dm, 0, dim, v0Ex, JEx, invJEx, detJEx);CHKERRQ(ierr);
if (interpolate) {ierr = CheckFVMGeometry(dm, 0, dim, centroidEx, normalEx, volEx);CHKERRQ(ierr);}
}
/* Check random triangles: scale, translate, then rotate */
if (transform) {
ierr = PetscRandomCreate(PETSC_COMM_SELF, &r);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(r);CHKERRQ(ierr);
ierr = PetscRandomSetInterval(r, 0.0, 10.0);CHKERRQ(ierr);
ierr = PetscRandomCreate(PETSC_COMM_SELF, &ang);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(ang);CHKERRQ(ierr);
ierr = PetscRandomSetInterval(ang, 0.0, 2*PETSC_PI);CHKERRQ(ierr);
ierr = PetscRandomCreate(PETSC_COMM_SELF, &ang2);CHKERRQ(ierr);
ierr = PetscRandomSetFromOptions(ang2);CHKERRQ(ierr);
ierr = PetscRandomSetInterval(ang2, 0.0, PETSC_PI);CHKERRQ(ierr);
for (t = 0; t < 100; ++t) {
PetscScalar vertexCoords[9] = {-1.0, -1.0, 0.0, 1.0, -1.0, 0.0, -1.0, 1.0, 0.0}, trans[3];
PetscReal v0Ex[3] = {-1.0, -1.0, 0.0};
PetscReal JEx[9] = {1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0}, R[9], rot[3], rotM[9];
PetscReal invJEx[9] = {1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0};
PetscReal detJEx = 1.0, scale, phi, theta, psi = 0.0;
PetscReal centroidEx[3] = {-0.333333333333, -0.333333333333, 0.0};
PetscReal normalEx[3] = {0.0, 0.0, 1.0};
PetscReal volEx = 2.0;
PetscInt d, e, f, p;
ierr = PetscRandomGetValueReal(r, &scale);CHKERRQ(ierr);
ierr = PetscRandomGetValueReal(ang, &phi);CHKERRQ(ierr);
ierr = PetscRandomGetValueReal(ang2, &theta);CHKERRQ(ierr);
for (d = 0; d < dim; ++d) {
ierr = PetscRandomGetValueReal(r, &trans[d]);CHKERRQ(ierr);
for (p = 0; p < 3; ++p) {
vertexCoords[p*dim+d] *= scale;
vertexCoords[p*dim+d] += trans[d];
}
centroidEx[d] *= scale;
centroidEx[d] += trans[d];
for (e = 0; e < dim-1; ++e) {
JEx[d*dim+e] *= scale;
invJEx[d*dim+e] /= scale;
}
if (d < dim-1) {
detJEx *= scale;
volEx *= scale;
}
}
R[0] = cos(theta)*cos(psi); R[1] = sin(phi)*sin(theta)*cos(psi) - cos(phi)*sin(psi); R[2] = sin(phi)*sin(psi) + cos(phi)*sin(theta)*cos(psi);
R[3] = cos(theta)*sin(psi); R[4] = cos(phi)*cos(psi) + sin(phi)*sin(theta)*sin(psi); R[5] = cos(phi)*sin(theta)*sin(psi) - sin(phi)*cos(psi);
R[6] = -sin(theta); R[7] = sin(phi)*cos(theta); R[8] = cos(phi)*cos(theta);
for (p = 0; p < 3; ++p) {
for (d = 0; d < dim; ++d) {
for (e = 0, rot[d] = 0.0; e < dim; ++e) {
rot[d] += R[d*dim+e] * vertexCoords[p*dim+e];
}
}
for (d = 0; d < dim; ++d) vertexCoords[p*dim+d] = rot[d];
}
for (d = 0; d < dim; ++d) {
for (e = 0, rot[d] = 0.0; e < dim; ++e) {
rot[d] += R[d*dim+e] * centroidEx[e];
}
}
for (d = 0; d < dim; ++d) centroidEx[d] = rot[d];
for (d = 0; d < dim; ++d) {
for (e = 0, rot[d] = 0.0; e < dim; ++e) {
rot[d] += R[d*dim+e] * normalEx[e];
}
}
for (d = 0; d < dim; ++d) normalEx[d] = rot[d];
for (d = 0; d < dim; ++d) {
v0Ex[d] = vertexCoords[d];
for (e = 0; e < dim; ++e) {
for (f = 0, rotM[d*dim+e] = 0.0; f < dim; ++f) {
rotM[d*dim+e] += R[d*dim+f] * JEx[f*dim+e];
}
}
}
for (d = 0; d < dim; ++d) {
for (e = 0; e < dim; ++e) {
JEx[d*dim+e] = rotM[d*dim+e];
}
}
for (d = 0; d < dim; ++d) {
for (e = 0; e < dim; ++e) {
for (f = 0, rotM[d*dim+e] = 0.0; f < dim; ++f) {
rotM[d*dim+e] += invJEx[d*dim+f] * R[e*dim+f];
}
}
}
for (d = 0; d < dim; ++d) {
for (e = 0; e < dim; ++e) {
invJEx[d*dim+e] = rotM[d*dim+e];
}
}
ierr = ChangeCoordinates(dm, dim, vertexCoords);CHKERRQ(ierr);
ierr = CheckFEMGeometry(dm, 0, dim, v0Ex, JEx, invJEx, detJEx);CHKERRQ(ierr);
if (interpolate) {ierr = CheckFVMGeometry(dm, 0, dim, centroidEx, normalEx, volEx);CHKERRQ(ierr);}
}
ierr = PetscRandomDestroy(&r);CHKERRQ(ierr);
ierr = PetscRandomDestroy(&ang);CHKERRQ(ierr);
ierr = PetscRandomDestroy(&ang2);CHKERRQ(ierr);
}
/* Cleanup */
ierr = DMDestroy(&dm);CHKERRQ(ierr);
PetscFunctionReturn(0);
}
int
main (int argc, char **argv)
{
sc_MPI_Comm mpicomm;
int mpiret;
int mpisize, mpirank;
p4est_t *p4est;
p4est_connectivity_t *conn;
sc_array_t *points_per_dim, *cone_sizes, *cones,
*cone_orientations, *coords,
*children, *parents, *childids, *leaves, *remotes;
p4est_locidx_t first_local_quad = -1;
/* initialize MPI */
mpiret = sc_MPI_Init (&argc, &argv);
SC_CHECK_MPI (mpiret);
mpicomm = sc_MPI_COMM_WORLD;
mpiret = sc_MPI_Comm_size (mpicomm, &mpisize);
SC_CHECK_MPI (mpiret);
mpiret = sc_MPI_Comm_rank (mpicomm, &mpirank);
SC_CHECK_MPI (mpiret);
sc_init (mpicomm, 1, 1, NULL, SC_LP_DEFAULT);
p4est_init (NULL, SC_LP_DEFAULT);
#ifndef P4_TO_P8
conn = p4est_connectivity_new_moebius ();
#else
conn = p8est_connectivity_new_rotcubes ();
#endif
p4est = p4est_new_ext (mpicomm, conn, 0, 1, 1, 0, NULL, NULL);
p4est_refine (p4est, 1, refine_fn, NULL);
p4est_balance (p4est, P4EST_CONNECT_FULL, NULL);
p4est_partition (p4est, 0, NULL);
points_per_dim = sc_array_new (sizeof (p4est_locidx_t));
cone_sizes = sc_array_new (sizeof (p4est_locidx_t));
cones = sc_array_new (sizeof (p4est_locidx_t));
cone_orientations = sc_array_new (sizeof (p4est_locidx_t));
coords = sc_array_new (3 * sizeof (double));
children = sc_array_new (sizeof (p4est_locidx_t));
parents = sc_array_new (sizeof (p4est_locidx_t));
childids = sc_array_new (sizeof (p4est_locidx_t));
leaves = sc_array_new (sizeof (p4est_locidx_t));
remotes = sc_array_new (2 * sizeof (p4est_locidx_t));
p4est_get_plex_data (p4est, P4EST_CONNECT_FULL, (mpisize > 1) ? 2 : 0,
&first_local_quad, points_per_dim, cone_sizes, cones,
cone_orientations, coords, children, parents, childids,
leaves, remotes);
#ifdef P4EST_WITH_PETSC
{
PetscErrorCode ierr;
DM plex, refTree;
PetscInt pStart, pEnd;
PetscSection parentSection;
PetscSF pointSF;
size_t zz, count;
locidx_to_PetscInt (points_per_dim);
locidx_to_PetscInt (cone_sizes);
locidx_to_PetscInt (cones);
locidx_to_PetscInt (cone_orientations);
coords_double_to_PetscScalar (coords);
locidx_to_PetscInt (children);
locidx_to_PetscInt (parents);
locidx_to_PetscInt (childids);
locidx_to_PetscInt (leaves);
locidx_pair_to_PetscSFNode (remotes);
P4EST_GLOBAL_PRODUCTION ("Begin PETSc routines\n");
ierr = PetscInitialize (&argc, &argv, 0, help);
CHKERRQ (ierr);
ierr = DMPlexCreate (mpicomm, &plex);
CHKERRQ (ierr);
ierr = DMSetDimension (plex, P4EST_DIM);
CHKERRQ (ierr);
ierr = DMSetCoordinateDim (plex, 3);
CHKERRQ (ierr);
ierr = DMPlexCreateFromDAG (plex, P4EST_DIM,
(PetscInt *) points_per_dim->array,
(PetscInt *) cone_sizes->array,
(PetscInt *) cones->array,
(PetscInt *) cone_orientations->array,
(PetscScalar *) coords->array);
CHKERRQ (ierr);
ierr = PetscSFCreate (mpicomm, &pointSF);
CHKERRQ (ierr);
ierr =
DMPlexCreateDefaultReferenceTree (mpicomm, P4EST_DIM, PETSC_FALSE,
&refTree);
CHKERRQ (ierr);
ierr = DMPlexSetReferenceTree (plex, refTree);
CHKERRQ (ierr);
ierr = DMDestroy (&refTree);
CHKERRQ (ierr);
ierr = PetscSectionCreate (mpicomm, &parentSection);
CHKERRQ (ierr);
ierr = DMPlexGetChart (plex, &pStart, &pEnd);
CHKERRQ (ierr);
ierr = PetscSectionSetChart (parentSection, pStart, pEnd);
CHKERRQ (ierr);
count = children->elem_count;
for (zz = 0; zz < count; zz++) {
PetscInt child =
*((PetscInt *) sc_array_index (children, zz));
ierr = PetscSectionSetDof (parentSection, child, 1);
CHKERRQ (ierr);
}
ierr = PetscSectionSetUp (parentSection);
CHKERRQ (ierr);
ierr =
DMPlexSetTree (plex, parentSection, (PetscInt *) parents->array,
(PetscInt *) childids->array);
CHKERRQ (ierr);
ierr = PetscSectionDestroy (&parentSection);
CHKERRQ (ierr);
ierr =
PetscSFSetGraph (pointSF, pEnd - pStart, (PetscInt) leaves->elem_count,
(PetscInt *) leaves->array, PETSC_COPY_VALUES,
(PetscSFNode *) remotes->array, PETSC_COPY_VALUES);
CHKERRQ (ierr);
ierr = DMViewFromOptions (plex, NULL, "-dm_view");
CHKERRQ (ierr);
/* TODO: test with rigid body modes as in plex ex3 */
ierr = DMDestroy (&plex);
CHKERRQ (ierr);
ierr = PetscFinalize ();
P4EST_GLOBAL_PRODUCTION ("End PETSc routines\n");
}
#endif
sc_array_destroy (points_per_dim);
sc_array_destroy (cone_sizes);
sc_array_destroy (cones);
sc_array_destroy (cone_orientations);
sc_array_destroy (coords);
sc_array_destroy (children);
sc_array_destroy (parents);
sc_array_destroy (childids);
sc_array_destroy (leaves);
sc_array_destroy (remotes);
p4est_destroy (p4est);
p4est_connectivity_destroy (conn);
sc_finalize ();
mpiret = sc_MPI_Finalize ();
SC_CHECK_MPI (mpiret);
return 0;
}
PETSC_EXTERN void PETSC_STDCALL dmplexcreatefromdag_(DM dm,PetscInt *depth, PetscInt numPoints[], PetscInt coneSize[], PetscInt cones[], PetscInt coneOrientations[], PetscScalar vertexCoords[], int *__ierr ){
*__ierr = DMPlexCreateFromDAG(
(DM)PetscToPointer((dm) ),*depth,numPoints,coneSize,cones,coneOrientations,vertexCoords);
}
/* Two triangles separated by a zero-volume cell with 4 vertices/2 edges
5--16--8
/ | | \
11 | | 12
/ | | \
3 0 10 2 14 1 6
\ | | /
9 | | 13
\ | | /
4--15--7
*/
PetscErrorCode CreateSimplexHybrid_2D(MPI_Comm comm, PetscInt testNum, DM *dm)
{
DM idm = NULL, hdm = NULL;
DMLabel faultLabel, hybridLabel;
PetscInt p;
PetscMPIInt rank;
PetscErrorCode ierr;
PetscFunctionBegin;
ierr = MPI_Comm_rank(comm, &rank);CHKERRQ(ierr);
if (!rank) {
switch (testNum) {
case 0:
{
PetscInt numPoints[2] = {4, 2};
PetscInt coneSize[6] = {3, 3, 0, 0, 0, 0};
PetscInt cones[6] = {2, 3, 4, 5, 4, 3};
PetscInt coneOrientations[6] = {0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[8] = {-1.0, -0.5, 0.0, -0.5, 0.0, 0.5, 1.0, 0.5};
PetscInt faultPoints[2] = {3, 4};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
for(p = 0; p < 2; ++p) {ierr = DMSetLabelValue(*dm, "fault", faultPoints[p], 1);CHKERRQ(ierr);}
}
break;
case 1:
{
PetscInt numPoints[2] = {5, 4};
PetscInt coneSize[9] = {3, 3, 3, 3, 0, 0, 0, 0, 0};
PetscInt cones[12] = {4, 5, 6, 6, 7, 4, 6, 5, 8, 6, 8, 7};
PetscInt coneOrientations[12] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
PetscScalar vertexCoords[10] = {-1.0, 0.0, 0.0, -1.0, 0.0, 0.0, 0.0, 1.0, 1.0, 0.0};
PetscInt faultPoints[3] = {5, 6, 7};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr);
for(p = 0; p < 3; ++p) {ierr = DMSetLabelValue(*dm, "fault", faultPoints[p], 1);CHKERRQ(ierr);}
}
break;
default: SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "No test mesh %d", testNum);
}
ierr = DMPlexCheckSymmetry(*dm);CHKERRQ(ierr);
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = DMPlexCopyCoordinates(*dm, idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-in_dm_view");CHKERRQ(ierr);
ierr = DMPlexCheckSymmetry(idm);CHKERRQ(ierr);
ierr = DMGetLabel(*dm, "fault", &faultLabel);CHKERRQ(ierr);
ierr = DMPlexCreateHybridMesh(idm, faultLabel, &hybridLabel, &hdm);CHKERRQ(ierr);
ierr = DMLabelDestroy(&hybridLabel);CHKERRQ(ierr);
ierr = DMDestroy(&idm);CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = hdm;
} else {
PetscInt numPoints[2] = {0, 0};
ierr = DMPlexCreateFromDAG(*dm, 1, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr);
ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr);
ierr = DMPlexCopyCoordinates(*dm, idm);CHKERRQ(ierr);
ierr = DMViewFromOptions(idm, NULL, "-in_dm_view");CHKERRQ(ierr);
ierr = DMPlexCreateHybridMesh(idm, NULL, NULL, &hdm);CHKERRQ(ierr);
ierr = DMDestroy(&idm);CHKERRQ(ierr);
ierr = DMDestroy(dm);CHKERRQ(ierr);
*dm = hdm;
}
PetscFunctionReturn(0);
}
