static char help[] = "Tests DMLocalToGlobal() for dof > 1\n\n"; #include #include int main(int argc, char **argv) { PetscInt M = 6, N = 5, P = 4, m = PETSC_DECIDE, n = PETSC_DECIDE, p = PETSC_DECIDE, i, j, k, is, js, ks, in, jen, kn; DM da; Vec local, global; PetscScalar ****l; PetscFunctionBeginUser; PetscCall(PetscInitialize(&argc, &argv, NULL, help)); /* Create distributed array and get vectors */ PetscCall(DMDACreate3d(PETSC_COMM_WORLD, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DM_BOUNDARY_NONE, DMDA_STENCIL_BOX, M, N, P, m, n, p, 2, 1, NULL, NULL, NULL, &da)); PetscCall(DMSetFromOptions(da)); PetscCall(DMSetUp(da)); PetscCall(DMCreateGlobalVector(da, &global)); PetscCall(DMCreateLocalVector(da, &local)); PetscCall(DMDAGetCorners(da, &is, &js, &ks, &in, &jen, &kn)); PetscCall(DMDAVecGetArrayDOF(da, local, &l)); for (i = is; i < is + in; i++) { for (j = js; j < js + jen; j++) { for (k = ks; k < ks + kn; k++) { l[k][j][i][0] = 2 * (i + j * M + k * M * N); l[k][j][i][1] = 2 * (i + j * M + k * M * N) + 1; } } } PetscCall(DMDAVecRestoreArrayDOF(da, local, &l)); PetscCall(DMLocalToGlobalBegin(da, local, ADD_VALUES, global)); PetscCall(DMLocalToGlobalEnd(da, local, ADD_VALUES, global)); PetscCall(VecView(global, PETSC_VIEWER_STDOUT_WORLD)); /* Free memory */ PetscCall(VecDestroy(&local)); PetscCall(VecDestroy(&global)); PetscCall(DMDestroy(&da)); PetscCall(PetscFinalize()); return 0; } /*TEST test: filter: grep -v -i Process output_file: output/ex25_1.out test: suffix: 2 nsize: 2 filter: grep -v -i Process output_file: output/ex25_2.out test: suffix: 3 nsize: 3 filter: grep -v -i Process output_file: output/ex25_2.out test: suffix: 4 nsize: 4 filter: grep -v -i Process output_file: output/ex25_2.out test: suffix: 5 nsize: 5 filter: grep -v -i Process output_file: output/ex25_2.out test: suffix: 6 nsize: 6 filter: grep -v -i Process output_file: output/ex25_2.out TEST*/