Actual source code: ex35.c
1: static char help[] = "Test VecGetArray4d()\n";
3: #include <petscvec.h>
5: int main(int argc, char **argv)
6: {
7: PetscInt cnt, i, j, k, l, m = 2, n = 3, p = 4, q = 5;
8: Vec x;
9: PetscScalar ****xx;
11: PetscFunctionBeginUser;
12: PetscCall(PetscInitialize(&argc, &argv, (char *)0, help));
13: PetscCall(VecCreateSeq(PETSC_COMM_WORLD, m * n * p * q, &x));
14: PetscCall(VecGetArray4d(x, m, n, p, q, 0, 0, 0, 0, &xx));
15: cnt = 0;
16: for (i = 0; i < m; i++)
17: for (j = 0; j < n; j++)
18: for (k = 0; k < p; k++)
19: for (l = 0; l < q; l++) xx[i][j][k][l] = cnt++;
21: PetscCall(VecRestoreArray4d(x, m, n, p, q, 0, 0, 0, 0, &xx));
22: PetscCall(VecView(x, PETSC_VIEWER_STDOUT_WORLD));
23: PetscCall(VecDestroy(&x));
24: PetscCall(PetscFinalize());
25: return 0;
26: }
28: /*TEST
30: test:
32: TEST*/