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, NULL, 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*/