1: static char help[] = "Create a Plex sphere from quads and create a P1 section\n\n";

  3: #include <petscdmplex.h>

  5: static PetscErrorCode SetupSection(DM dm)
  6: {
  7:   PetscSection   s;
  8:   PetscInt       vStart, vEnd, v;

 11:   DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);
 12:   PetscSectionCreate(PetscObjectComm((PetscObject) dm), &s);
 13:   PetscSectionSetNumFields(s, 1);
 14:   PetscSectionSetFieldComponents(s, 0, 1);
 15:   PetscSectionSetChart(s, vStart, vEnd);
 16:   for (v = vStart; v < vEnd; ++v) {
 17:     PetscSectionSetDof(s, v, 1);
 18:     PetscSectionSetFieldDof(s, v, 0, 1);
 19:   }
 20:   PetscSectionSetUp(s);
 21:   DMSetLocalSection(dm, s);
 22:   PetscSectionDestroy(&s);
 23:   return 0;
 24: }

 26: int main(int argc, char **argv)
 27: {
 28:   DM             dm;
 29:   Vec            u;

 31:   PetscInitialize(&argc, &argv, NULL,help);
 32:   DMCreate(PETSC_COMM_WORLD, &dm);
 33:   DMSetType(dm, DMPLEX);
 34:   DMSetFromOptions(dm);
 35:   PetscObjectSetName((PetscObject) dm, "Sphere");
 36:   DMViewFromOptions(dm, NULL, "-dm_view");

 38:   SetupSection(dm);
 39:   DMGetGlobalVector(dm, &u);
 40:   VecSet(u, 2);
 41:   VecViewFromOptions(u, NULL, "-vec_view");
 42:   DMRestoreGlobalVector(dm, &u);
 43:   DMDestroy(&dm);
 44:   PetscFinalize();
 45:   return 0;
 46: }

 48: /*TEST

 50:   testset:
 51:     requires: !__float128
 52:     args: -dm_plex_shape sphere -dm_view

 54:     test:
 55:       suffix: 2d_quad
 56:       args: -dm_plex_simplex 0

 58:     test:
 59:       suffix: 2d_tri
 60:       args:

 62:     test:
 63:       suffix: 3d_tri
 64:       args: -dm_plex_dim 3

 66:   testset:
 67:     requires: !__float128
 68:     args: -dm_plex_shape sphere -petscpartitioner_type simple -dm_view

 70:     test:
 71:       suffix: 2d_quad_parallel
 72:       nsize: 2
 73:       args: -dm_plex_simplex 0

 75:     test:
 76:       suffix: 2d_tri_parallel
 77:       nsize: 2

 79:     test:
 80:       suffix: 3d_tri_parallel
 81:       nsize: 2
 82:       args: -dm_plex_dim 3

 84: TEST*/