Actual source code: ex7.c

petsc-master 2020-11-24
Report Typos and Errors
  1: static char help[] = "Create a Plex sphere from quads and create a P1 section\n\n";

  3: #include <petscdmplex.h>

  5: typedef struct {
  6:   PetscInt  dim;     /* Topological problem dimension */
  7:   PetscBool simplex; /* Mesh with simplices */
  8: } AppCtx;

 10: static PetscErrorCode ProcessOptions(MPI_Comm comm, AppCtx *options)
 11: {

 15:   options->dim     = 2;
 16:   options->simplex = PETSC_FALSE;

 18:   PetscOptionsBegin(comm, "", "Sphere Mesh Options", "DMPLEX");
 19:   PetscOptionsRangeInt("-dim", "Problem dimension", "ex7.c", options->dim, &options->dim, NULL,1,3);
 20:   PetscOptionsBool("-simplex", "Use simplices, or tensor product cells", "ex7.c", options->simplex, &options->simplex, NULL);
 21:   PetscOptionsEnd();
 22:   return(0);
 23: }

 25: static PetscErrorCode SetupSection(DM dm)
 26: {
 27:   PetscSection   s;
 28:   PetscInt       vStart, vEnd, v;

 32:   DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);
 33:   PetscSectionCreate(PetscObjectComm((PetscObject) dm), &s);
 34:   PetscSectionSetNumFields(s, 1);
 35:   PetscSectionSetFieldComponents(s, 0, 1);
 36:   PetscSectionSetChart(s, vStart, vEnd);
 37:   for (v = vStart; v < vEnd; ++v) {
 38:     PetscSectionSetDof(s, v, 1);
 39:     PetscSectionSetFieldDof(s, v, 0, 1);
 40:   }
 41:   PetscSectionSetUp(s);
 42:   DMSetLocalSection(dm, s);
 43:   PetscSectionDestroy(&s);
 44:   return(0);
 45: }

 47: int main(int argc, char **argv)
 48: {
 49:   DM             dm;
 50:   Vec            u;
 51:   AppCtx         ctx;

 54:   PetscInitialize(&argc, &argv, NULL,help);if (ierr) return ierr;
 55:   ProcessOptions(PETSC_COMM_WORLD, &ctx);
 56:   DMPlexCreateSphereMesh(PETSC_COMM_WORLD, ctx.dim, ctx.simplex, 1.0, &dm);
 57:   PetscObjectSetName((PetscObject) dm, "Sphere");
 58:   /* Distribute mesh over processes */
 59:   {
 60:      DM dmDist = NULL;
 61:      PetscPartitioner part;
 62:      DMPlexGetPartitioner(dm, &part);
 63:      PetscPartitionerSetFromOptions(part);
 64:      DMPlexDistribute(dm, 0, NULL, &dmDist);
 65:      if (dmDist) {
 66:        DMDestroy(&dm);
 67:        dm  = dmDist;
 68:      }
 69:   }
 70:   DMSetFromOptions(dm);
 71:   DMViewFromOptions(dm, NULL, "-dm_view");
 72:   SetupSection(dm);
 73:   DMGetGlobalVector(dm, &u);
 74:   VecSet(u, 2);
 75:   VecViewFromOptions(u, NULL, "-vec_view");
 76:   DMRestoreGlobalVector(dm, &u);
 77:   DMDestroy(&dm);
 78:   PetscFinalize();
 79:   return ierr;
 80: }

 82: /*TEST

 84:   test:
 85:     suffix: 2d_quad
 86:     requires: !__float128
 87:     args: -dm_view

 89:   test:
 90:     suffix: 2d_quad_parallel
 91:     requires: !__float128
 92:     args: -dm_view -petscpartitioner_type simple
 93:     nsize: 2

 95:   test:
 96:     suffix: 2d_tri
 97:     requires: !__float128
 98:     args: -simplex -dm_view

100:   test:
101:     suffix: 2d_tri_parallel
102:     requires: !__float128
103:     args: -simplex -dm_view -petscpartitioner_type simple
104:     nsize: 2

106:   test:
107:     suffix: 3d_tri
108:     requires: !__float128
109:     args: -dim 3 -simplex -dm_view

111:   test:
112:     suffix: 3d_tri_parallel
113:     requires: !__float128
114:     args: -dim 3 -simplex -dm_view -petscpartitioner_type simple
115:     nsize: 2

117: TEST*/