Lines Matching refs:nx

55   PetscInt  nx, ny;                        /* nb of cells in x- and y-direction */  member
132 for (i = 0; i < s->nx; i++) { in StokesWriteSolution()
133 n = j * s->nx + i; in StokesWriteSolution()
134 …rt(array[n]), (double)PetscRealPart(array[n + s->nx * s->ny]), (double)PetscRealPart(array[n + 2 *… in StokesWriteSolution()
156 PetscCall(VecSetSizes(s->x, PETSC_DECIDE, 3 * s->nx * s->ny)); in StokesSetupVectors()
175 n = row % (s->nx * s->ny); in StokesGetPosition()
176 *i = n % s->nx; in StokesGetPosition()
177 *j = (n - (*i)) / s->nx; in StokesGetPosition()
193 if (row < s->nx * s->ny) { in StokesExactSolution()
230 if (row < s->nx * s->ny) { in StokesRhs()
262 …PetscCall(MatSetSizes(s->subA[0], PETSC_DECIDE, PETSC_DECIDE, 2 * s->nx * s->ny, 2 * s->nx * s->ny… in StokesSetupMatBlock00()
272 if (row >= s->nx * s->ny) { in StokesSetupMatBlock00()
273 for (i = 0; i < sz; i++) cols[i] += s->nx * s->ny; in StokesSetupMatBlock00()
293 PetscCall(MatSetSizes(s->subA[1], PETSC_DECIDE, PETSC_DECIDE, 2 * s->nx * s->ny, s->nx * s->ny)); in StokesSetupMatBlock01()
303 if (row < s->nx * s->ny) { in StokesSetupMatBlock01()
331 PetscCall(MatSetSizes(s->subA[3], PETSC_DECIDE, PETSC_DECIDE, s->nx * s->ny, s->nx * s->ny)); in StokesSetupMatBlock11()
351 PetscCall(VecSetSizes(diag, PETSC_DECIDE, 2 * s->nx * s->ny)); in StokesSetupApproxSchur()
382 PetscInt p = j * s->nx + i, w = p - 1, e = p + 1, s2 = p - s->nx, n = p + s->nx; in StokesStencilLaplacian()
405 } else if (i == s->nx - 1 && j == 0) { /* south-east corner */ in StokesStencilLaplacian()
413 } else if (i == s->nx - 1 && j == s->ny - 1) { /* north-east corner */ in StokesStencilLaplacian()
431 } else if (i == s->nx - 1) { /* east boundary */ in StokesStencilLaplacian()
479 PetscInt p = j * s->nx + i, w = p - 1, e = p + 1; in StokesStencilGradientX()
496 } else if (i == s->nx - 1 && j == 0) { /* south-east corner */ in StokesStencilGradientX()
502 } else if (i == s->nx - 1 && j == s->ny - 1) { /* north-east corner */ in StokesStencilGradientX()
514 } else if (i == s->nx - 1) { /* east boundary */ in StokesStencilGradientX()
550 PetscInt p = j * s->nx + i, s2 = p - s->nx, n = p + s->nx; in StokesStencilGradientY()
567 } else if (i == s->nx - 1 && j == 0) { /* south-east corner */ in StokesStencilGradientY()
573 } else if (i == s->nx - 1 && j == s->ny - 1) { /* north-east corner */ in StokesStencilGradientY()
587 } else if (i == s->nx - 1) { /* east boundary */ in StokesStencilGradientY()
684 PetscScalar scale = PetscSqrtReal((double)s->nx * s->ny); in StokesCalcError()
717 s.nx = 4; in main()
719 PetscCall(PetscOptionsGetInt(NULL, NULL, "-nx", &s.nx, NULL)); in main()
721 s.hx = 2.0 / s.nx; in main()