Actual source code: ex1.c

  1: static char help[] = "Make a 2D grid of patches and view them\n\n";

  3: /*
  4: Serial Test
  5: Parallel Test where all zooms are serials
  6: Parallel Test where zooms are parallel

  8: Return DMPatch from Zoom
  9: Override refine from DMPatch to split cells
 10:  */
 11: #include <petscdmpatch.h>

 13: typedef struct {
 14:   PetscInt   debug;     /* The debugging level */
 15:   PetscInt   dim;       /* The spatial dimension */
 16:   MatStencil patchSize; /* Size of patches */
 17:   MatStencil gridSize;  /* Size of patch grid */
 18:   MatStencil commSize;  /* Size of patch comm */
 19: } AppCtx;

 21: PetscErrorCode ProcessOptions(MPI_Comm comm, AppCtx *options)
 22: {
 23:   PetscInt patchSize, commSize, gridSize;

 25:   PetscFunctionBegin;
 26:   options->debug = 0;
 27:   options->dim   = 2;
 28:   patchSize      = 0;
 29:   commSize       = 0;
 30:   gridSize       = 1;

 32:   PetscOptionsBegin(comm, "", "Patch Test Options", "DMPATCH");
 33:   PetscCall(PetscOptionsBoundedInt("-debug", "The debugging level", "ex1.c", options->debug, &options->debug, NULL, 0));
 34:   PetscCall(PetscOptionsRangeInt("-dim", "The spatial dimension", "ex1.c", options->dim, &options->dim, NULL, 1, 3));
 35:   PetscCall(PetscOptionsBoundedInt("-patch_size", "The patch size in each dimension", "ex1.c", patchSize, &patchSize, NULL, 0));
 36:   PetscCall(PetscOptionsBoundedInt("-comm_size", "The comm size in each dimension", "ex1.c", commSize, &commSize, NULL, 0));
 37:   PetscCall(PetscOptionsBoundedInt("-grid_size", "The grid size in each dimension", "ex1.c", gridSize, &gridSize, NULL, 1));
 38:   PetscOptionsEnd();

 40:   options->patchSize.i = options->patchSize.j = options->patchSize.k = 1;
 41:   options->commSize.i = options->commSize.j = options->commSize.k = 1;
 42:   options->gridSize.i = options->gridSize.j = options->gridSize.k = 1;
 43:   if (options->dim > 0) {
 44:     options->patchSize.i = patchSize;
 45:     options->commSize.i  = commSize;
 46:     options->gridSize.i  = gridSize;
 47:   }
 48:   if (options->dim > 1) {
 49:     options->patchSize.j = patchSize;
 50:     options->commSize.j  = commSize;
 51:     options->gridSize.j  = gridSize;
 52:   }
 53:   if (options->dim > 2) {
 54:     options->patchSize.k = patchSize;
 55:     options->commSize.k  = commSize;
 56:     options->gridSize.k  = gridSize;
 57:   }
 58:   PetscFunctionReturn(PETSC_SUCCESS);
 59: }

 61: int main(int argc, char **argv)
 62: {
 63:   DM     dm;
 64:   AppCtx user; /* user-defined work context */

 66:   PetscFunctionBeginUser;
 67:   PetscCall(PetscInitialize(&argc, &argv, NULL, help));
 68:   PetscCall(ProcessOptions(PETSC_COMM_WORLD, &user));
 69:   PetscCall(DMPatchCreateGrid(PETSC_COMM_WORLD, user.dim, user.patchSize, user.commSize, user.gridSize, &dm));
 70:   PetscCall(PetscObjectSetName((PetscObject)dm, "Patch Mesh"));
 71:   PetscCall(DMSetFromOptions(dm));
 72:   PetscCall(DMSetUp(dm));
 73:   PetscCall(DMView(dm, PETSC_VIEWER_STDOUT_WORLD));
 74:   PetscCall(DMPatchSolve(dm));
 75:   PetscCall(DMDestroy(&dm));
 76:   PetscCall(PetscFinalize());
 77:   return 0;
 78: }

 80: /*TEST

 82:    test:

 84: TEST*/