Actual source code: ex7.c


  2: static char help[] = "Tests DMLocalToLocalxxx() for DMDA.\n\n";

  4: #include <petscdmda.h>

  6: int main(int argc, char **argv)
  7: {
  8:   PetscMPIInt     rank;
  9:   PetscInt        M = 8, dof = 1, stencil_width = 1, i, start, end, P = 5, N = 6, m = PETSC_DECIDE, n = PETSC_DECIDE, p = PETSC_DECIDE, pt = 0, st = 0;
 10:   PetscBool       flg = PETSC_FALSE, flg2, flg3;
 11:   DMBoundaryType  periodic;
 12:   DMDAStencilType stencil_type;
 13:   DM              da;
 14:   Vec             local, global, local_copy;
 15:   PetscScalar     value;
 16:   PetscReal       norm, work;
 17:   PetscViewer     viewer;
 18:   char            filename[64];
 19:   FILE           *file;

 22:   PetscInitialize(&argc, &argv, (char *)0, help);
 23:   PetscOptionsGetInt(NULL, NULL, "-M", &M, NULL);
 24:   PetscOptionsGetInt(NULL, NULL, "-N", &N, NULL);
 25:   PetscOptionsGetInt(NULL, NULL, "-dof", &dof, NULL);
 26:   PetscOptionsGetInt(NULL, NULL, "-stencil_width", &stencil_width, NULL);
 27:   PetscOptionsGetInt(NULL, NULL, "-periodic", &pt, NULL);

 29:   periodic = (DMBoundaryType)pt;

 31:   PetscOptionsGetInt(NULL, NULL, "-stencil_type", &st, NULL);

 33:   stencil_type = (DMDAStencilType)st;

 35:   PetscOptionsHasName(NULL, NULL, "-grid2d", &flg2);
 36:   PetscOptionsHasName(NULL, NULL, "-grid3d", &flg3);
 37:   if (flg2) {
 38:     DMDACreate2d(PETSC_COMM_WORLD, periodic, periodic, stencil_type, M, N, m, n, dof, stencil_width, NULL, NULL, &da);
 39:   } else if (flg3) {
 40:     DMDACreate3d(PETSC_COMM_WORLD, periodic, periodic, periodic, stencil_type, M, N, P, m, n, p, dof, stencil_width, NULL, NULL, NULL, &da);
 41:   } else {
 42:     DMDACreate1d(PETSC_COMM_WORLD, periodic, M, dof, stencil_width, NULL, &da);
 43:   }
 44:   DMSetFromOptions(da);
 45:   DMSetUp(da);

 47:   DMCreateGlobalVector(da, &global);
 48:   DMCreateLocalVector(da, &local);
 49:   VecDuplicate(local, &local_copy);

 51:   /* zero out vectors so that ghostpoints are zero */
 52:   value = 0;
 53:   VecSet(local, value);
 54:   VecSet(local_copy, value);

 56:   VecGetOwnershipRange(global, &start, &end);
 57:   for (i = start; i < end; i++) {
 58:     value = i + 1;
 59:     VecSetValues(global, 1, &i, &value, INSERT_VALUES);
 60:   }
 61:   VecAssemblyBegin(global);
 62:   VecAssemblyEnd(global);

 64:   DMGlobalToLocalBegin(da, global, INSERT_VALUES, local);
 65:   DMGlobalToLocalEnd(da, global, INSERT_VALUES, local);

 67:   DMLocalToLocalBegin(da, local, INSERT_VALUES, local_copy);
 68:   DMLocalToLocalEnd(da, local, INSERT_VALUES, local_copy);

 70:   PetscOptionsGetBool(NULL, NULL, "-save", &flg, NULL);
 71:   if (flg) {
 72:     MPI_Comm_rank(PETSC_COMM_WORLD, &rank);
 73:     sprintf(filename, "local.%d", rank);
 74:     PetscViewerASCIIOpen(PETSC_COMM_SELF, filename, &viewer);
 75:     PetscViewerASCIIGetPointer(viewer, &file);
 76:     VecView(local, viewer);
 77:     fprintf(file, "Vector with correct ghost points\n");
 78:     VecView(local_copy, viewer);
 79:     PetscViewerDestroy(&viewer);
 80:   }

 82:   VecAXPY(local_copy, -1.0, local);
 83:   VecNorm(local_copy, NORM_MAX, &work);
 84:   MPI_Allreduce(&work, &norm, 1, MPIU_REAL, MPIU_MAX, PETSC_COMM_WORLD);
 85:   PetscPrintf(PETSC_COMM_WORLD, "Norm of difference %g should be zero\n", (double)norm);

 87:   VecDestroy(&local_copy);
 88:   VecDestroy(&local);
 89:   VecDestroy(&global);
 90:   DMDestroy(&da);
 91:   PetscFinalize();
 92:   return 0;
 93: }

 95: /*TEST

 97:    test:
 98:       nsize: 8
 99:       args: -dof 3 -stencil_width 2 -M 50 -N 50 -periodic

101:    test:
102:       suffix: 2
103:       nsize: 8
104:       args: -dof 3 -stencil_width 2 -M 50 -N 50 -periodic -grid2d
105:       output_file: output/ex7_1.out

107:    test:
108:       suffix: 3
109:       nsize: 8
110:       args: -dof 3 -stencil_width 2 -M 50 -N 50 -periodic -grid3d
111:       output_file: output/ex7_1.out

113: TEST*/