1c4762a1bSJed Brown static char help[] = "Tests for mesh interpolation\n\n"; 2c4762a1bSJed Brown 3c4762a1bSJed Brown #include <petscdmplex.h> 4c4762a1bSJed Brown 5c4762a1bSJed Brown /* List of test meshes 6c4762a1bSJed Brown 7c4762a1bSJed Brown Triangle 8c4762a1bSJed Brown -------- 9c4762a1bSJed Brown Test 0: 10c4762a1bSJed Brown Two triangles sharing a face 11c4762a1bSJed Brown 12c4762a1bSJed Brown 4 13c4762a1bSJed Brown / | \ 14c4762a1bSJed Brown / | \ 15c4762a1bSJed Brown / | \ 16c4762a1bSJed Brown 2 0 | 1 5 17c4762a1bSJed Brown \ | / 18c4762a1bSJed Brown \ | / 19c4762a1bSJed Brown \ | / 20c4762a1bSJed Brown 3 21c4762a1bSJed Brown 22c4762a1bSJed Brown should become 23c4762a1bSJed Brown 24c4762a1bSJed Brown 4 25c4762a1bSJed Brown / | \ 26c4762a1bSJed Brown 8 | 9 27c4762a1bSJed Brown / | \ 28c4762a1bSJed Brown 2 0 7 1 5 29c4762a1bSJed Brown \ | / 30c4762a1bSJed Brown 6 | 10 31c4762a1bSJed Brown \ | / 32c4762a1bSJed Brown 3 33c4762a1bSJed Brown 34c4762a1bSJed Brown Tetrahedron 35c4762a1bSJed Brown ----------- 36c4762a1bSJed Brown Test 0: 37c4762a1bSJed Brown Two tets sharing a face 38c4762a1bSJed Brown 39c4762a1bSJed Brown cell 5 _______ cell 40c4762a1bSJed Brown 0 / | \ \ 1 41c4762a1bSJed Brown / | \ \ 42c4762a1bSJed Brown / | \ \ 43c4762a1bSJed Brown 2----|----4-----6 44c4762a1bSJed Brown \ | / / 45c4762a1bSJed Brown \ | / / 46c4762a1bSJed Brown \ | / / 47c4762a1bSJed Brown 3------- 48c4762a1bSJed Brown 49c4762a1bSJed Brown should become 50c4762a1bSJed Brown 51c4762a1bSJed Brown cell 5 _______ cell 52c4762a1bSJed Brown 0 / | \ \ 1 53c4762a1bSJed Brown / | \ \ 54c4762a1bSJed Brown 17 | 18 13 22 55c4762a1bSJed Brown / 8 19 10 \ \ 56c4762a1bSJed Brown / | \ \ 57c4762a1bSJed Brown 2---14-|------4--20--6 58c4762a1bSJed Brown \ | / / 59c4762a1bSJed Brown \ 9 | 7 / / 60c4762a1bSJed Brown 16 | 15 11 21 61c4762a1bSJed Brown \ | / / 62c4762a1bSJed Brown \ | / / 63c4762a1bSJed Brown 3------- 64c4762a1bSJed Brown 65c4762a1bSJed Brown Quadrilateral 66c4762a1bSJed Brown ------------- 67c4762a1bSJed Brown Test 0: 68c4762a1bSJed Brown Two quads sharing a face 69c4762a1bSJed Brown 70c4762a1bSJed Brown 5-------4-------7 71c4762a1bSJed Brown | | | 72c4762a1bSJed Brown | 0 | 1 | 73c4762a1bSJed Brown | | | 74c4762a1bSJed Brown 2-------3-------6 75c4762a1bSJed Brown 76c4762a1bSJed Brown should become 77c4762a1bSJed Brown 78c4762a1bSJed Brown 5--10---4--14---7 79c4762a1bSJed Brown | | | 80c4762a1bSJed Brown 11 0 9 1 13 81c4762a1bSJed Brown | | | 82c4762a1bSJed Brown 2---8---3--12---6 83c4762a1bSJed Brown 84c4762a1bSJed Brown Test 1: 85c4762a1bSJed Brown A quad and a triangle sharing a face 86c4762a1bSJed Brown 87c4762a1bSJed Brown 5-------4 88c4762a1bSJed Brown | | \ 89c4762a1bSJed Brown | 0 | \ 90c4762a1bSJed Brown | | 1 \ 91c4762a1bSJed Brown 2-------3----6 92c4762a1bSJed Brown 93c4762a1bSJed Brown should become 94c4762a1bSJed Brown 95c4762a1bSJed Brown 5---9---4 96c4762a1bSJed Brown | | \ 97c4762a1bSJed Brown 10 0 8 12 98c4762a1bSJed Brown | | 1 \ 99c4762a1bSJed Brown 2---7---3-11-6 100c4762a1bSJed Brown 101c4762a1bSJed Brown Hexahedron 102c4762a1bSJed Brown ---------- 103c4762a1bSJed Brown Test 0: 104c4762a1bSJed Brown Two hexes sharing a face 105c4762a1bSJed Brown 106c4762a1bSJed Brown cell 9-------------8-------------13 cell 107c4762a1bSJed Brown 0 /| /| /| 1 108c4762a1bSJed Brown / | 15 / | 21 / | 109c4762a1bSJed Brown / | / | / | 110c4762a1bSJed Brown 6-------------7-------------12 | 111c4762a1bSJed Brown | | 18 | | 24 | | 112c4762a1bSJed Brown | | | | | | 113c4762a1bSJed Brown |19 | |17 | |23 | 114c4762a1bSJed Brown | | 16 | | 22 | | 115c4762a1bSJed Brown | 5---------|---4---------|---11 116c4762a1bSJed Brown | / | / | / 117c4762a1bSJed Brown | / 14 | / 20 | / 118c4762a1bSJed Brown |/ |/ |/ 119c4762a1bSJed Brown 2-------------3-------------10 120c4762a1bSJed Brown 121c4762a1bSJed Brown should become 122c4762a1bSJed Brown 123c4762a1bSJed Brown cell 9-----31------8-----42------13 cell 124c4762a1bSJed Brown 0 /| /| /| 1 125c4762a1bSJed Brown 32 | 15 30| 21 41| 126c4762a1bSJed Brown / | / | / | 127c4762a1bSJed Brown 6-----29------7-----40------12 | 128c4762a1bSJed Brown | | 17 | | 23 | | 129c4762a1bSJed Brown | 35 | 36 | 44 130c4762a1bSJed Brown |19 | |18 | |24 | 131c4762a1bSJed Brown 34 | 16 33 | 22 43 | 132c4762a1bSJed Brown | 5-----26--|---4-----37--|---11 133c4762a1bSJed Brown | / | / | / 134c4762a1bSJed Brown | 25 14 | 27 20 | 38 135c4762a1bSJed Brown |/ |/ |/ 136c4762a1bSJed Brown 2-----28------3-----39------10 137c4762a1bSJed Brown 138c4762a1bSJed Brown */ 139c4762a1bSJed Brown 140c4762a1bSJed Brown typedef struct { 141c4762a1bSJed Brown PetscInt testNum; /* Indicates the mesh to create */ 142c4762a1bSJed Brown PetscInt dim; /* The topological mesh dimension */ 14330602db0SMatthew G. Knepley PetscBool simplex; /* Use simplices or hexes */ 144c4762a1bSJed Brown PetscBool useGenerator; /* Construct mesh with a mesh generator */ 145da9060c4SMatthew G. Knepley PetscBool femGeometry; /* Flag to compute FEM geometry */ 146c4762a1bSJed Brown } AppCtx; 147c4762a1bSJed Brown 148c4762a1bSJed Brown PetscErrorCode ProcessOptions(MPI_Comm comm, AppCtx *options) 149c4762a1bSJed Brown { 150c4762a1bSJed Brown PetscErrorCode ierr; 151c4762a1bSJed Brown 152c4762a1bSJed Brown PetscFunctionBegin; 153c4762a1bSJed Brown options->testNum = 0; 154c4762a1bSJed Brown options->dim = 2; 15530602db0SMatthew G. Knepley options->simplex = PETSC_TRUE; 156c4762a1bSJed Brown options->useGenerator = PETSC_FALSE; 157da9060c4SMatthew G. Knepley options->femGeometry = PETSC_TRUE; 158c4762a1bSJed Brown 159c4762a1bSJed Brown ierr = PetscOptionsBegin(comm, "", "Meshing Interpolation Test Options", "DMPLEX");CHKERRQ(ierr); 160c4762a1bSJed Brown ierr = PetscOptionsBoundedInt("-testnum", "The mesh to create", "ex7.c", options->testNum, &options->testNum, NULL,0);CHKERRQ(ierr); 161c4762a1bSJed Brown ierr = PetscOptionsRangeInt("-dim", "The topological mesh dimension", "ex7.c", options->dim, &options->dim, NULL,1,3);CHKERRQ(ierr); 16230602db0SMatthew G. Knepley ierr = PetscOptionsBool("-simplex", "Use simplices if true, otherwise hexes", "ex7.c", options->simplex, &options->simplex, NULL);CHKERRQ(ierr); 163c4762a1bSJed Brown ierr = PetscOptionsBool("-use_generator", "Use a mesh generator to build the mesh", "ex7.c", options->useGenerator, &options->useGenerator, NULL);CHKERRQ(ierr); 164da9060c4SMatthew G. Knepley ierr = PetscOptionsBool("-fem_geometry", "Flag to compute FEM geometry", "ex7.c", options->femGeometry, &options->femGeometry, NULL);CHKERRQ(ierr); 165*1e1ea65dSPierre Jolivet ierr = PetscOptionsEnd();CHKERRQ(ierr); 166c4762a1bSJed Brown PetscFunctionReturn(0); 167c4762a1bSJed Brown } 168c4762a1bSJed Brown 169c4762a1bSJed Brown PetscErrorCode CreateSimplex_2D(MPI_Comm comm, DM dm) 170c4762a1bSJed Brown { 171c4762a1bSJed Brown PetscInt depth = 1, testNum = 0, p; 172c4762a1bSJed Brown PetscMPIInt rank; 173c4762a1bSJed Brown PetscErrorCode ierr; 174c4762a1bSJed Brown 175c4762a1bSJed Brown PetscFunctionBegin; 176ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm, &rank);CHKERRMPI(ierr); 177c4762a1bSJed Brown if (!rank) { 178c4762a1bSJed Brown switch (testNum) { 179c4762a1bSJed Brown case 0: 180c4762a1bSJed Brown { 181c4762a1bSJed Brown PetscInt numPoints[2] = {4, 2}; 182c4762a1bSJed Brown PetscInt coneSize[6] = {3, 3, 0, 0, 0, 0}; 183c4762a1bSJed Brown PetscInt cones[6] = {2, 3, 4, 5, 4, 3}; 184c4762a1bSJed Brown PetscInt coneOrientations[6] = {0, 0, 0, 0, 0, 0}; 185c4762a1bSJed Brown PetscScalar vertexCoords[8] = {-0.5, 0.5, 0.0, 0.0, 0.0, 1.0, 0.5, 0.5}; 186c4762a1bSJed Brown PetscInt markerPoints[8] = {2, 1, 3, 1, 4, 1, 5, 1}; 187c4762a1bSJed Brown 188c4762a1bSJed Brown ierr = DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr); 189c4762a1bSJed Brown for (p = 0; p < 4; ++p) { 190c4762a1bSJed Brown ierr = DMSetLabelValue(dm, "marker", markerPoints[p*2], markerPoints[p*2+1]);CHKERRQ(ierr); 191c4762a1bSJed Brown } 192c4762a1bSJed Brown } 193c4762a1bSJed Brown break; 194c4762a1bSJed Brown default: 195c4762a1bSJed Brown SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "No test mesh %d", testNum); 196c4762a1bSJed Brown } 197c4762a1bSJed Brown } else { 198c4762a1bSJed Brown PetscInt numPoints[2] = {0, 0}; 199c4762a1bSJed Brown 200c4762a1bSJed Brown ierr = DMPlexCreateFromDAG(dm, depth, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr); 201c4762a1bSJed Brown } 202c4762a1bSJed Brown PetscFunctionReturn(0); 203c4762a1bSJed Brown } 204c4762a1bSJed Brown 205c4762a1bSJed Brown PetscErrorCode CreateSimplex_3D(MPI_Comm comm, DM dm) 206c4762a1bSJed Brown { 207c4762a1bSJed Brown PetscInt depth = 1, testNum = 0, p; 208c4762a1bSJed Brown PetscMPIInt rank; 209c4762a1bSJed Brown PetscErrorCode ierr; 210c4762a1bSJed Brown 211c4762a1bSJed Brown PetscFunctionBegin; 212ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm, &rank);CHKERRMPI(ierr); 213c4762a1bSJed Brown if (!rank) { 214c4762a1bSJed Brown switch (testNum) { 215c4762a1bSJed Brown case 0: 216c4762a1bSJed Brown { 217c4762a1bSJed Brown PetscInt numPoints[2] = {5, 2}; 218c4762a1bSJed Brown PetscInt coneSize[23] = {4, 4, 0, 0, 0, 0, 0}; 219c4762a1bSJed Brown PetscInt cones[8] = {2, 4, 3, 5, 3, 4, 6, 5}; 220c4762a1bSJed Brown PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0}; 221c4762a1bSJed Brown 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}; 222c4762a1bSJed Brown PetscInt markerPoints[8] = {2, 1, 3, 1, 4, 1, 5, 1}; 223c4762a1bSJed Brown 224c4762a1bSJed Brown ierr = DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr); 225c4762a1bSJed Brown for (p = 0; p < 4; ++p) { 226c4762a1bSJed Brown ierr = DMSetLabelValue(dm, "marker", markerPoints[p*2], markerPoints[p*2+1]);CHKERRQ(ierr); 227c4762a1bSJed Brown } 228c4762a1bSJed Brown } 229c4762a1bSJed Brown break; 230c4762a1bSJed Brown default: 231c4762a1bSJed Brown SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "No test mesh %d", testNum); 232c4762a1bSJed Brown } 233c4762a1bSJed Brown } else { 234c4762a1bSJed Brown PetscInt numPoints[2] = {0, 0}; 235c4762a1bSJed Brown 236c4762a1bSJed Brown ierr = DMPlexCreateFromDAG(dm, depth, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr); 237c4762a1bSJed Brown } 238c4762a1bSJed Brown PetscFunctionReturn(0); 239c4762a1bSJed Brown } 240c4762a1bSJed Brown 241c4762a1bSJed Brown PetscErrorCode CreateQuad_2D(MPI_Comm comm, PetscInt testNum, DM dm) 242c4762a1bSJed Brown { 243c4762a1bSJed Brown PetscInt depth = 1, p; 244c4762a1bSJed Brown PetscMPIInt rank; 245c4762a1bSJed Brown PetscErrorCode ierr; 246c4762a1bSJed Brown 247c4762a1bSJed Brown PetscFunctionBegin; 248ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm, &rank);CHKERRMPI(ierr); 249c4762a1bSJed Brown if (!rank) { 250c4762a1bSJed Brown switch (testNum) { 251c4762a1bSJed Brown case 0: 252c4762a1bSJed Brown { 253c4762a1bSJed Brown PetscInt numPoints[2] = {6, 2}; 254c4762a1bSJed Brown PetscInt coneSize[8] = {4, 4, 0, 0, 0, 0, 0, 0}; 255c4762a1bSJed Brown PetscInt cones[8] = {2, 3, 4, 5, 3, 6, 7, 4}; 256c4762a1bSJed Brown PetscInt coneOrientations[8] = {0, 0, 0, 0, 0, 0, 0, 0}; 257c4762a1bSJed Brown PetscScalar vertexCoords[12] = {-0.5, 0.0, 0.0, 0.0, 0.0, 1.0, -0.5, 1.0, 0.5, 0.0, 0.5, 1.0}; 258c4762a1bSJed Brown PetscInt markerPoints[12] = {2, 1, 3, 1, 4, 1, 5, 1, 6, 1, 7, 1}; 259c4762a1bSJed Brown 260c4762a1bSJed Brown ierr = DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr); 261c4762a1bSJed Brown for (p = 0; p < 6; ++p) { 262c4762a1bSJed Brown ierr = DMSetLabelValue(dm, "marker", markerPoints[p*2], markerPoints[p*2+1]);CHKERRQ(ierr); 263c4762a1bSJed Brown } 264c4762a1bSJed Brown } 265c4762a1bSJed Brown break; 266c4762a1bSJed Brown case 1: 267c4762a1bSJed Brown { 268c4762a1bSJed Brown PetscInt numPoints[2] = {5, 2}; 269c4762a1bSJed Brown PetscInt coneSize[7] = {4, 3, 0, 0, 0, 0, 0}; 270c4762a1bSJed Brown PetscInt cones[7] = {2, 3, 4, 5, 3, 6, 4}; 271c4762a1bSJed Brown PetscInt coneOrientations[7] = {0, 0, 0, 0, 0, 0, 0}; 272c4762a1bSJed Brown PetscScalar vertexCoords[10] = {-0.5, 0.0, 0.0, 0.0, 0.0, 1.0, -0.5, 1.0, 0.5, 0.0}; 273c4762a1bSJed Brown PetscInt markerPoints[10] = {2, 1, 3, 1, 4, 1, 5, 1, 6, 1}; 274c4762a1bSJed Brown 275c4762a1bSJed Brown ierr = DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr); 276c4762a1bSJed Brown for (p = 0; p < 5; ++p) { 277c4762a1bSJed Brown ierr = DMSetLabelValue(dm, "marker", markerPoints[p*2], markerPoints[p*2+1]);CHKERRQ(ierr); 278c4762a1bSJed Brown } 279c4762a1bSJed Brown } 280c4762a1bSJed Brown break; 281c4762a1bSJed Brown default: 282c4762a1bSJed Brown SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "No test mesh %d", testNum); 283c4762a1bSJed Brown } 284c4762a1bSJed Brown } else { 285c4762a1bSJed Brown PetscInt numPoints[2] = {0, 0}; 286c4762a1bSJed Brown 287c4762a1bSJed Brown ierr = DMPlexCreateFromDAG(dm, depth, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr); 288c4762a1bSJed Brown } 289c4762a1bSJed Brown PetscFunctionReturn(0); 290c4762a1bSJed Brown } 291c4762a1bSJed Brown 292c4762a1bSJed Brown PetscErrorCode CreateHex_3D(MPI_Comm comm, DM dm) 293c4762a1bSJed Brown { 294c4762a1bSJed Brown PetscInt depth = 1, testNum = 0, p; 295c4762a1bSJed Brown PetscMPIInt rank; 296c4762a1bSJed Brown PetscErrorCode ierr; 297c4762a1bSJed Brown 298c4762a1bSJed Brown PetscFunctionBegin; 299ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm, &rank);CHKERRMPI(ierr); 300c4762a1bSJed Brown if (!rank) { 301c4762a1bSJed Brown switch (testNum) { 302c4762a1bSJed Brown case 0: 303c4762a1bSJed Brown { 304c4762a1bSJed Brown PetscInt numPoints[2] = {12, 2}; 305c4762a1bSJed Brown PetscInt coneSize[14] = {8, 8, 0,0,0,0,0,0,0,0,0,0,0,0}; 306c4762a1bSJed Brown PetscInt cones[16] = {2,5,4,3,6,7,8,9, 3,4,11,10,7,12,13,8}; 307c4762a1bSJed Brown PetscInt coneOrientations[16] = {0,0,0,0,0,0,0,0, 0,0, 0, 0,0, 0, 0,0}; 308c4762a1bSJed Brown 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, 309c4762a1bSJed Brown -0.5,0.0,1.0, 0.0,0.0,1.0, 0.0,1.0,1.0, -0.5,1.0,1.0, 310c4762a1bSJed Brown 0.5,0.0,0.0, 0.5,1.0,0.0, 0.5,0.0,1.0, 0.5,1.0,1.0}; 311c4762a1bSJed Brown PetscInt markerPoints[16] = {2,1,3,1,4,1,5,1,6,1,7,1,8,1,9,1}; 312c4762a1bSJed Brown 313c4762a1bSJed Brown ierr = DMPlexCreateFromDAG(dm, depth, numPoints, coneSize, cones, coneOrientations, vertexCoords);CHKERRQ(ierr); 314c4762a1bSJed Brown for (p = 0; p < 8; ++p) { 315c4762a1bSJed Brown ierr = DMSetLabelValue(dm, "marker", markerPoints[p*2], markerPoints[p*2+1]);CHKERRQ(ierr); 316c4762a1bSJed Brown } 317c4762a1bSJed Brown } 318c4762a1bSJed Brown break; 319c4762a1bSJed Brown default: 320c4762a1bSJed Brown SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "No test mesh %d", testNum); 321c4762a1bSJed Brown } 322c4762a1bSJed Brown } else { 323c4762a1bSJed Brown PetscInt numPoints[2] = {0, 0}; 324c4762a1bSJed Brown 325c4762a1bSJed Brown ierr = DMPlexCreateFromDAG(dm, depth, numPoints, NULL, NULL, NULL, NULL);CHKERRQ(ierr); 326c4762a1bSJed Brown } 327c4762a1bSJed Brown PetscFunctionReturn(0); 328c4762a1bSJed Brown } 329c4762a1bSJed Brown 330c4762a1bSJed Brown PetscErrorCode CheckMesh(DM dm, AppCtx *user) 331c4762a1bSJed Brown { 332c4762a1bSJed Brown PetscReal detJ, J[9], refVol = 1.0; 333c4762a1bSJed Brown PetscReal vol; 334c4762a1bSJed Brown PetscInt dim, depth, d, cStart, cEnd, c; 335c4762a1bSJed Brown PetscErrorCode ierr; 336c4762a1bSJed Brown 337c4762a1bSJed Brown PetscFunctionBegin; 338c4762a1bSJed Brown ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 339c4762a1bSJed Brown ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr); 340c4762a1bSJed Brown for (d = 0; d < dim; ++d) { 341c4762a1bSJed Brown refVol *= 2.0; 342c4762a1bSJed Brown } 343c4762a1bSJed Brown ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 344c4762a1bSJed Brown for (c = cStart; c < cEnd; ++c) { 345da9060c4SMatthew G. Knepley if (user->femGeometry) { 346c4762a1bSJed Brown ierr = DMPlexComputeCellGeometryFEM(dm, c, NULL, NULL, J, NULL, &detJ);CHKERRQ(ierr); 347da9060c4SMatthew G. Knepley if (detJ <= 0.0) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mesh cell %D is inverted, |J| = %g", c, (double) detJ); 348da9060c4SMatthew G. Knepley ierr = PetscInfo1(dm, "FEM Volume: %g\n", (double) detJ*refVol);CHKERRQ(ierr); 349da9060c4SMatthew G. Knepley } 350c4762a1bSJed Brown if (depth > 1) { 351c4762a1bSJed Brown ierr = DMPlexComputeCellGeometryFVM(dm, c, &vol, NULL, NULL);CHKERRQ(ierr); 352da9060c4SMatthew G. Knepley if (vol <= 0.0) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Mesh cell %D is inverted, vol = %g", c, (double) vol); 353da9060c4SMatthew G. Knepley ierr = PetscInfo1(dm, "FVM Volume: %g\n", (double) vol);CHKERRQ(ierr); 354c4762a1bSJed Brown } 355c4762a1bSJed Brown } 356c4762a1bSJed Brown PetscFunctionReturn(0); 357c4762a1bSJed Brown } 358c4762a1bSJed Brown 359c4762a1bSJed Brown PetscErrorCode CompareCones(DM dm, DM idm) 360c4762a1bSJed Brown { 361c4762a1bSJed Brown PetscInt cStart, cEnd, c, vStart, vEnd, v; 362c4762a1bSJed Brown PetscErrorCode ierr; 363c4762a1bSJed Brown 364c4762a1bSJed Brown PetscFunctionBegin; 365c4762a1bSJed Brown ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 366c4762a1bSJed Brown ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr); 367c4762a1bSJed Brown for (c = cStart; c < cEnd; ++c) { 368c4762a1bSJed Brown const PetscInt *cone; 369c4762a1bSJed Brown PetscInt *points = NULL, numPoints, p, numVertices = 0, coneSize; 370c4762a1bSJed Brown 371c4762a1bSJed Brown ierr = DMPlexGetCone(dm, c, &cone);CHKERRQ(ierr); 372c4762a1bSJed Brown ierr = DMPlexGetConeSize(dm, c, &coneSize);CHKERRQ(ierr); 373c4762a1bSJed Brown ierr = DMPlexGetTransitiveClosure(idm, c, PETSC_TRUE, &numPoints, &points);CHKERRQ(ierr); 374c4762a1bSJed Brown for (p = 0; p < numPoints*2; p += 2) { 375c4762a1bSJed Brown const PetscInt point = points[p]; 376c4762a1bSJed Brown if ((point >= vStart) && (point < vEnd)) points[numVertices++] = point; 377c4762a1bSJed Brown } 378c4762a1bSJed Brown if (numVertices != coneSize) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "In cell %d, cone size %d != %d vertices in closure", c, coneSize, numVertices); 379c4762a1bSJed Brown for (v = 0; v < numVertices; ++v) { 380c4762a1bSJed Brown if (cone[v] != points[v]) SETERRQ4(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "In cell %d, cone point %d is %d != %d vertex in closure", c, v, cone[v], points[v]); 381c4762a1bSJed Brown } 382c4762a1bSJed Brown ierr = DMPlexRestoreTransitiveClosure(idm, c, PETSC_TRUE, &numPoints, &points);CHKERRQ(ierr); 383c4762a1bSJed Brown } 384c4762a1bSJed Brown PetscFunctionReturn(0); 385c4762a1bSJed Brown } 386c4762a1bSJed Brown 387c4762a1bSJed Brown PetscErrorCode CreateMesh(MPI_Comm comm, PetscInt testNum, AppCtx *user, DM *dm) 388c4762a1bSJed Brown { 389c4762a1bSJed Brown PetscBool useGenerator = user->useGenerator; 390c4762a1bSJed Brown PetscErrorCode ierr; 391c4762a1bSJed Brown 392c4762a1bSJed Brown PetscFunctionBegin; 393c4762a1bSJed Brown ierr = DMCreate(comm, dm);CHKERRQ(ierr); 394c4762a1bSJed Brown ierr = DMSetType(*dm, DMPLEX);CHKERRQ(ierr); 39530602db0SMatthew G. Knepley if (useGenerator) { 39630602db0SMatthew G. Knepley ierr = DMSetFromOptions(*dm);CHKERRQ(ierr); 39730602db0SMatthew G. Knepley } else { 39830602db0SMatthew G. Knepley PetscInt dim = user->dim; 39930602db0SMatthew G. Knepley PetscBool simplex = user->simplex; 40030602db0SMatthew G. Knepley 401c4762a1bSJed Brown ierr = DMSetDimension(*dm, dim);CHKERRQ(ierr); 402c4762a1bSJed Brown switch (dim) { 403c4762a1bSJed Brown case 2: 40430602db0SMatthew G. Knepley if (simplex) { 405c4762a1bSJed Brown ierr = CreateSimplex_2D(comm, *dm);CHKERRQ(ierr); 406c4762a1bSJed Brown } else { 407c4762a1bSJed Brown ierr = CreateQuad_2D(comm, testNum, *dm);CHKERRQ(ierr); 408c4762a1bSJed Brown } 409c4762a1bSJed Brown break; 410c4762a1bSJed Brown case 3: 41130602db0SMatthew G. Knepley if (simplex) { 412c4762a1bSJed Brown ierr = CreateSimplex_3D(comm, *dm);CHKERRQ(ierr); 413c4762a1bSJed Brown } else { 414c4762a1bSJed Brown ierr = CreateHex_3D(comm, *dm);CHKERRQ(ierr); 415c4762a1bSJed Brown } 416c4762a1bSJed Brown break; 417c4762a1bSJed Brown default: 418c4762a1bSJed Brown SETERRQ1(comm, PETSC_ERR_ARG_OUTOFRANGE, "Cannot make meshes for dimension %d", dim); 419c4762a1bSJed Brown } 420c4762a1bSJed Brown } 421c4762a1bSJed Brown { 422da9060c4SMatthew G. Knepley DMPlexInterpolatedFlag interpolated; 423c4762a1bSJed Brown 424c4762a1bSJed Brown ierr = CheckMesh(*dm, user);CHKERRQ(ierr); 425da9060c4SMatthew G. Knepley ierr = DMPlexIsInterpolated(*dm, &interpolated);CHKERRQ(ierr); 426da9060c4SMatthew G. Knepley if (interpolated == DMPLEX_INTERPOLATED_FULL) { 427da9060c4SMatthew G. Knepley DM udm; 428da9060c4SMatthew G. Knepley 429da9060c4SMatthew G. Knepley ierr = DMPlexUninterpolate(*dm, &udm);CHKERRQ(ierr); 430da9060c4SMatthew G. Knepley ierr = CompareCones(udm, *dm);CHKERRQ(ierr); 431da9060c4SMatthew G. Knepley ierr = DMDestroy(&udm);CHKERRQ(ierr); 432da9060c4SMatthew G. Knepley } else { 433da9060c4SMatthew G. Knepley DM idm; 434da9060c4SMatthew G. Knepley 435c4762a1bSJed Brown ierr = DMPlexInterpolate(*dm, &idm);CHKERRQ(ierr); 436c4762a1bSJed Brown ierr = CompareCones(*dm, idm);CHKERRQ(ierr); 437c4762a1bSJed Brown ierr = DMDestroy(dm);CHKERRQ(ierr); 438c4762a1bSJed Brown *dm = idm; 439c4762a1bSJed Brown } 440da9060c4SMatthew G. Knepley } 441c4762a1bSJed Brown { 442c4762a1bSJed Brown DM distributedMesh = NULL; 443c4762a1bSJed Brown PetscPartitioner part; 444c4762a1bSJed Brown 445c4762a1bSJed Brown ierr = DMPlexGetPartitioner(*dm, &part);CHKERRQ(ierr); 446c4762a1bSJed Brown ierr = PetscPartitionerSetFromOptions(part);CHKERRQ(ierr); 447c4762a1bSJed Brown /* Distribute mesh over processes */ 448c4762a1bSJed Brown ierr = DMPlexDistribute(*dm, 0, NULL, &distributedMesh);CHKERRQ(ierr); 449c4762a1bSJed Brown if (distributedMesh) { 450c4762a1bSJed Brown ierr = DMViewFromOptions(distributedMesh, NULL, "-dm_view");CHKERRQ(ierr); 451c4762a1bSJed Brown ierr = DMDestroy(dm);CHKERRQ(ierr); 452c4762a1bSJed Brown *dm = distributedMesh; 453c4762a1bSJed Brown } 454c4762a1bSJed Brown } 455c4762a1bSJed Brown ierr = PetscObjectSetName((PetscObject) *dm, "Interpolated Mesh");CHKERRQ(ierr); 456da9060c4SMatthew G. Knepley ierr = DMSetFromOptions(*dm);CHKERRQ(ierr); 457c4762a1bSJed Brown ierr = DMViewFromOptions(*dm, NULL, "-dm_view");CHKERRQ(ierr); 458c4762a1bSJed Brown PetscFunctionReturn(0); 459c4762a1bSJed Brown } 460c4762a1bSJed Brown 461c4762a1bSJed Brown int main(int argc, char **argv) 462c4762a1bSJed Brown { 46330602db0SMatthew G. Knepley DM dm; 464c4762a1bSJed Brown AppCtx user; /* user-defined work context */ 465c4762a1bSJed Brown PetscErrorCode ierr; 466c4762a1bSJed Brown 467c4762a1bSJed Brown ierr = PetscInitialize(&argc, &argv, NULL,help);if (ierr) return ierr; 468c4762a1bSJed Brown ierr = ProcessOptions(PETSC_COMM_WORLD, &user);CHKERRQ(ierr); 46930602db0SMatthew G. Knepley ierr = CreateMesh(PETSC_COMM_WORLD, user.testNum, &user, &dm);CHKERRQ(ierr); 47030602db0SMatthew G. Knepley ierr = CheckMesh(dm, &user);CHKERRQ(ierr); 47130602db0SMatthew G. Knepley ierr = DMDestroy(&dm);CHKERRQ(ierr); 472c4762a1bSJed Brown ierr = PetscFinalize(); 473c4762a1bSJed Brown return ierr; 474c4762a1bSJed Brown } 475c4762a1bSJed Brown 476c4762a1bSJed Brown /*TEST 477c4762a1bSJed Brown 478c4762a1bSJed Brown # Two cell test meshes 0-7 479c4762a1bSJed Brown test: 480c4762a1bSJed Brown suffix: 0 481c4762a1bSJed Brown args: -dim 2 -dm_view ascii::ascii_info_detail 482c4762a1bSJed Brown test: 483c4762a1bSJed Brown suffix: 1 484c4762a1bSJed Brown nsize: 2 485c4762a1bSJed Brown args: -petscpartitioner_type simple -dim 2 -dm_view ascii::ascii_info_detail 486c4762a1bSJed Brown test: 487c4762a1bSJed Brown suffix: 2 48830602db0SMatthew G. Knepley args: -dim 2 -simplex 0 -dm_view ascii::ascii_info_detail 489c4762a1bSJed Brown test: 490c4762a1bSJed Brown suffix: 3 491c4762a1bSJed Brown nsize: 2 49230602db0SMatthew G. Knepley args: -petscpartitioner_type simple -dim 2 -simplex 0 -dm_view ascii::ascii_info_detail 493c4762a1bSJed Brown test: 494c4762a1bSJed Brown suffix: 4 495c4762a1bSJed Brown args: -dim 3 -dm_view ascii::ascii_info_detail 496c4762a1bSJed Brown test: 497c4762a1bSJed Brown suffix: 5 498c4762a1bSJed Brown nsize: 2 499c4762a1bSJed Brown args: -petscpartitioner_type simple -dim 3 -dm_view ascii::ascii_info_detail 500c4762a1bSJed Brown test: 501c4762a1bSJed Brown suffix: 6 50230602db0SMatthew G. Knepley args: -dim 3 -simplex 0 -dm_view ascii::ascii_info_detail 503c4762a1bSJed Brown test: 504c4762a1bSJed Brown suffix: 7 505c4762a1bSJed Brown nsize: 2 50630602db0SMatthew G. Knepley args: -petscpartitioner_type simple -dim 3 -simplex 0 -dm_view ascii::ascii_info_detail 507c4762a1bSJed Brown # 2D Hybrid Mesh 8 508c4762a1bSJed Brown test: 509c4762a1bSJed Brown suffix: 8 51030602db0SMatthew G. Knepley args: -dim 2 -simplex 0 -testnum 1 -dm_view ascii::ascii_info_detail 511da9060c4SMatthew G. Knepley # Reference cells 512da9060c4SMatthew G. Knepley test: 513da9060c4SMatthew G. Knepley suffix: 12 51430602db0SMatthew G. Knepley args: -use_generator -dm_plex_reference_cell_domain -dm_plex_cell pyramid -fem_geometry 0 -dm_plex_check_all 515c4762a1bSJed Brown # TetGen meshes 9-10 516c4762a1bSJed Brown test: 517c4762a1bSJed Brown suffix: 9 518c4762a1bSJed Brown requires: triangle 51930602db0SMatthew G. Knepley args: -use_generator -dm_view ascii::ascii_info_detail -dm_coord_space 0 520c4762a1bSJed Brown test: 521c4762a1bSJed Brown suffix: 10 522c4762a1bSJed Brown requires: ctetgen 52330602db0SMatthew G. Knepley args: -use_generator -dm_plex_dim 3 -dm_view ascii::ascii_info_detail -dm_coord_space 0 524c4762a1bSJed Brown # Cubit meshes 11 525c4762a1bSJed Brown test: 526c4762a1bSJed Brown suffix: 11 527c4762a1bSJed Brown requires: exodusii 52830602db0SMatthew G. Knepley args: -use_generator -dm_plex_filename ${wPETSC_DIR}/share/petsc/datafiles/meshes/blockcylinder-50.exo -dm_view ascii::ascii_info_detail -dm_coord_space 0 529c4762a1bSJed Brown 530c4762a1bSJed Brown TEST*/ 531