Actual source code: ex35.c


  2: static char help[] = "Test VecGetArray4d()\n";

  4: #include <petscvec.h>

  6: int main(int argc, char **argv)
  7: {
  8:   PetscInt        cnt, i, j, k, l, m = 2, n = 3, p = 4, q = 5;
  9:   Vec             x;
 10:   PetscScalar ****xx;

 12:   PetscFunctionBeginUser;
 13:   PetscCall(PetscInitialize(&argc, &argv, (char *)0, help));
 14:   PetscCall(VecCreateSeq(PETSC_COMM_WORLD, m * n * p * q, &x));
 15:   PetscCall(VecGetArray4d(x, m, n, p, q, 0, 0, 0, 0, &xx));
 16:   cnt = 0;
 17:   for (i = 0; i < m; i++)
 18:     for (j = 0; j < n; j++)
 19:       for (k = 0; k < p; k++)
 20:         for (l = 0; l < q; l++) xx[i][j][k][l] = cnt++;

 22:   PetscCall(VecRestoreArray4d(x, m, n, p, q, 0, 0, 0, 0, &xx));
 23:   PetscCall(VecView(x, PETSC_VIEWER_STDOUT_WORLD));
 24:   PetscCall(VecDestroy(&x));
 25:   PetscCall(PetscFinalize());
 26:   return 0;
 27: }

 29: /*TEST

 31:    test:

 33: TEST*/