1 static char help[] = "Tests various 2-dimensional DMDA routines.\n\n"; 2 3 #include <petscdmda.h> 4 #include <petscdraw.h> 5 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]; 15 16 PetscFunctionBeginUser; 17 PetscCall(PetscInitialize(&argc, &argv, NULL, 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)); 22 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)); 30 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 } 40 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)); 46 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)); 60 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)); 66 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 } 75 76 /*TEST 77 78 test: 79 args: -dof 2 80 filter: grep -v -i Object 81 requires: x 82 83 test: 84 suffix: 2 85 nsize: 2 86 args: -dof 2 87 filter: grep -v -i Object 88 requires: x 89 90 test: 91 suffix: 3 92 nsize: 3 93 args: -dof 2 94 filter: grep -v -i Object 95 requires: x 96 97 TEST*/ 98