Actual source code: ex11.c

  1: static char help[] = "Tests various 2-dimensional DMDA routines.\n\n";

  3: #include <petscdmda.h>
  4: #include <petscdraw.h>

  6: int main(int argc, char **argv)
  7: {
  8:   PetscInt       M = 10, N = 8, dof = 1, s = 1, bx = 0, by = 0, i, n, j, k, m, wrap, xs, ys;
  9:   DM             da, dac;
 10:   PetscViewer    viewer;
 11:   Vec            local, global, coors;
 12:   PetscScalar ***xy, ***aglobal;
 13:   PetscDraw      draw;
 14:   char           fname[32];

 16:   PetscFunctionBeginUser;
 17:   PetscCall(PetscInitialize(&argc, &argv, (char *)0, help));
 18:   /* Create viewers */
 19:   PetscCall(PetscViewerDrawOpen(PETSC_COMM_WORLD, 0, "", PETSC_DECIDE, PETSC_DECIDE, 600, 200, &viewer));
 20:   PetscCall(PetscViewerDrawGetDraw(viewer, 0, &draw));
 21:   PetscCall(PetscDrawSetDoubleBuffer(draw));

 23:   /* Read options */
 24:   PetscCall(PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL));
 25:   PetscCall(PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL));
 26:   PetscCall(PetscOptionsGetInt(NULL, NULL, "-dof", &dof, NULL));
 27:   PetscCall(PetscOptionsGetInt(NULL, NULL, "-s", &s, NULL));
 28:   PetscCall(PetscOptionsGetInt(NULL, NULL, "-periodic_x", &wrap, NULL));
 29:   PetscCall(PetscOptionsGetInt(NULL, NULL, "-periodic_y", &wrap, NULL));

 31:   /* Create distributed array and get vectors */
 32:   PetscCall(DMDACreate2d(PETSC_COMM_WORLD, (DMBoundaryType)bx, (DMBoundaryType)by, DMDA_STENCIL_BOX, M, N, PETSC_DECIDE, PETSC_DECIDE, dof, s, NULL, NULL, &da));
 33:   PetscCall(DMSetFromOptions(da));
 34:   PetscCall(DMSetUp(da));
 35:   PetscCall(DMDASetUniformCoordinates(da, 0.0, 1.0, 0.0, 1.0, 0.0, 0.0));
 36:   for (i = 0; i < dof; i++) {
 37:     PetscCall(PetscSNPrintf(fname, PETSC_STATIC_ARRAY_LENGTH(fname), "Field %" PetscInt_FMT, i));
 38:     PetscCall(DMDASetFieldName(da, i, fname));
 39:   }

 41:   PetscCall(DMView(da, viewer));
 42:   PetscCall(DMCreateGlobalVector(da, &global));
 43:   PetscCall(DMCreateLocalVector(da, &local));
 44:   PetscCall(DMGetCoordinates(da, &coors));
 45:   PetscCall(DMGetCoordinateDM(da, &dac));

 47:   /* Set values into global vectors */
 48:   PetscCall(DMDAVecGetArrayDOFRead(dac, coors, &xy));
 49:   PetscCall(DMDAVecGetArrayDOF(da, global, &aglobal));
 50:   PetscCall(DMDAGetCorners(da, &xs, &ys, 0, &m, &n, 0));
 51:   for (k = 0; k < dof; k++) {
 52:     for (j = ys; j < ys + n; j++) {
 53:       for (i = xs; i < xs + m; i++) aglobal[j][i][k] = PetscSinScalar(2.0 * PETSC_PI * (k + 1) * xy[j][i][0]);
 54:     }
 55:   }
 56:   PetscCall(DMDAVecRestoreArrayDOF(da, global, &aglobal));
 57:   PetscCall(DMDAVecRestoreArrayDOFRead(dac, coors, &xy));
 58:   PetscCall(DMGlobalToLocalBegin(da, global, INSERT_VALUES, local));
 59:   PetscCall(DMGlobalToLocalEnd(da, global, INSERT_VALUES, local));

 61:   PetscCall(VecSet(global, 0.0));
 62:   PetscCall(DMLocalToGlobalBegin(da, local, INSERT_VALUES, global));
 63:   PetscCall(DMLocalToGlobalEnd(da, local, INSERT_VALUES, global));
 64:   PetscCall(VecView(global, PETSC_VIEWER_STDOUT_WORLD));
 65:   PetscCall(VecView(global, viewer));

 67:   /* Free memory */
 68:   PetscCall(PetscViewerDestroy(&viewer));
 69:   PetscCall(VecDestroy(&global));
 70:   PetscCall(VecDestroy(&local));
 71:   PetscCall(DMDestroy(&da));
 72:   PetscCall(PetscFinalize());
 73:   return 0;
 74: }

 76: /*TEST

 78:    test:
 79:       args: -dof 2
 80:       filter: grep -v -i Object
 81:       requires: x

 83:    test:
 84:       suffix: 2
 85:       nsize: 2
 86:       args: -dof 2
 87:       filter: grep -v -i Object
 88:       requires: x

 90:    test:
 91:       suffix: 3
 92:       nsize: 3
 93:       args: -dof 2
 94:       filter: grep -v -i Object
 95:       requires: x

 97: TEST*/