Date: 2026-05-01T06:11:48+00:00

Actual source code: stag3d.c

      1: /* Functions specific to the 3-dimensional implementation of DMStag */
      2: #include <petsc/private/dmstagimpl.h>

      4: /*@
      5:   DMStagCreate3d - Create an object to manage data living on the elements, faces, edges, and vertices of a parallelized regular 3D grid.

      7:   Collective

      9:   Input Parameters:
     10: + comm         - MPI communicator
     11: . bndx         - x boundary type, `DM_BOUNDARY_NONE`, `DM_BOUNDARY_PERIODIC`, or `DM_BOUNDARY_GHOSTED`
     12: . bndy         - y boundary type, `DM_BOUNDARY_NONE`, `DM_BOUNDARY_PERIODIC`, or `DM_BOUNDARY_GHOSTED`
     13: . bndz         - z boundary type, `DM_BOUNDARY_NONE`, `DM_BOUNDARY_PERIODIC`, or `DM_BOUNDARY_GHOSTED`
     14: . M            - global number of elements in x direction
     15: . N            - global number of elements in y direction
     16: . P            - global number of elements in z direction
     17: . m            - number of ranks in the x direction (may be `PETSC_DECIDE`)
     18: . n            - number of ranks in the y direction (may be `PETSC_DECIDE`)
     19: . p            - number of ranks in the z direction (may be `PETSC_DECIDE`)
     20: . dof0         - number of degrees of freedom per vertex/0-cell
     21: . dof1         - number of degrees of freedom per edge/1-cell
     22: . dof2         - number of degrees of freedom per face/2-cell
     23: . dof3         - number of degrees of freedom per element/3-cell
     24: . stencilType  - ghost/halo region type: `DMSTAG_STENCIL_NONE`, `DMSTAG_STENCIL_BOX`, or `DMSTAG_STENCIL_STAR`
     25: . stencilWidth - width, in elements, of halo/ghost region
     26: . lx           - array of local x  element counts, of length equal to `m`, summing to `M`, or `NULL`
     27: . ly           - arrays of local y element counts, of length equal to `n`, summing to `N`, or `NULL`
     28: - lz           - arrays of local z element counts, of length equal to `p`, summing to `P`, or `NULL`

     30:   Output Parameter:
     31: . dm - the new `DMSTAG` object

     33:   Options Database Keys:
     34: + -dm_view                                      - calls `DMViewFromOptions()` at the conclusion of `DMSetUp()`
     35: . -stag_grid_x nx                               - number of elements in the x direction
     36: . -stag_grid_y ny                               - number of elements in the y direction
     37: . -stag_grid_z nz                               - number of elements in the z direction
     38: . -stag_ranks_x rx                              - number of ranks in the x direction
     39: . -stag_ranks_y ry                              - number of ranks in the y direction
     40: . -stag_ranks_z rz                              - number of ranks in the z direction
     41: . -stag_ghost_stencil_width                     - width of ghost region, in elements
     42: . -stag_boundary_type x (none|ghosted|periodic) - `DMBoundaryType` value
     43: . -stag_boundary_type y (none|ghosted|periodic) - `DMBoundaryType` value
     44: - -stag_boundary_type z (none|ghosted|periodic) - `DMBoundaryType` value

     46:   Level: beginner

     48:   Notes:
     49:   You must call `DMSetUp()` after this call before using the `DM`.
     50:   If you wish to use the options database (see the keys above) to change values in the `DMSTAG`, you must call
     51:   `DMSetFromOptions()` after this function but before `DMSetUp()`.

     53: .seealso: [](ch_stag), `DMSTAG`, `DMStagCreate1d()`, `DMStagCreate2d()`, `DMDestroy()`, `DMView()`, `DMCreateGlobalVector()`, `DMCreateLocalVector()`, `DMLocalToGlobalBegin()`, `DMDACreate3d()`
     54: @*/
     55: PetscErrorCode DMStagCreate3d(MPI_Comm comm, DMBoundaryType bndx, DMBoundaryType bndy, DMBoundaryType bndz, PetscInt M, PetscInt N, PetscInt P, PetscInt m, PetscInt n, PetscInt p, PetscInt dof0, PetscInt dof1, PetscInt dof2, PetscInt dof3, DMStagStencilType stencilType, PetscInt stencilWidth, const PetscInt lx[], const PetscInt ly[], const PetscInt lz[], DM *dm)
     56: {
     57:   PetscFunctionBegin;
     58:   PetscCall(DMCreate(comm, dm));
     59:   PetscCall(DMSetDimension(*dm, 3));
     60:   PetscCall(DMStagInitialize(bndx, bndy, bndz, M, N, P, m, n, p, dof0, dof1, dof2, dof3, stencilType, stencilWidth, lx, ly, lz, *dm));
     61:   PetscFunctionReturn(PETSC_SUCCESS);
     62: }

     64: PETSC_INTERN PetscErrorCode DMStagRestrictSimple_3d(DM dmf, Vec xf_local, DM dmc, Vec xc_local)
     65: {
     66:   PetscInt              Mf, Nf, Pf, Mc, Nc, Pc, factorx, factory, factorz, dof[4];
     67:   PetscInt              xc, yc, zc, mc, nc, pc, nExtraxc, nExtrayc, nExtrazc, i, j, k, d;
     68:   PetscInt              ibackdownleftf, ibackdownf, ibackleftf, ibackf, idownleftf, idownf, ileftf, ielemf;
     69:   PetscInt              ibackdownleftc, ibackdownc, ibackleftc, ibackc, idownleftc, idownc, ileftc, ielemc;
     70:   const PetscScalar ****arrf;
     71:   PetscScalar       ****arrc;

     73:   PetscFunctionBegin;
     74:   PetscCall(DMStagGetGlobalSizes(dmf, &Mf, &Nf, &Pf));
     75:   PetscCall(DMStagGetGlobalSizes(dmc, &Mc, &Nc, &Pc));
     76:   factorx = Mf / Mc;
     77:   factory = Nf / Nc;
     78:   factorz = Pf / Pc;
     79:   PetscCall(DMStagGetDOF(dmc, &dof[0], &dof[1], &dof[2], &dof[3]));

     81:   PetscCall(DMStagGetCorners(dmc, &xc, &yc, &zc, &mc, &nc, &pc, &nExtraxc, &nExtrayc, &nExtrazc));
     82:   PetscCall(VecZeroEntries(xc_local));
     83:   PetscCall(DMStagVecGetArray(dmf, xf_local, &arrf));
     84:   PetscCall(DMStagVecGetArray(dmc, xc_local, &arrc));
     85:   PetscCall(DMStagGetLocationSlot(dmf, DMSTAG_BACK_DOWN_LEFT, 0, &ibackdownleftf));
     86:   PetscCall(DMStagGetLocationSlot(dmf, DMSTAG_BACK_DOWN, 0, &ibackdownf));
     87:   PetscCall(DMStagGetLocationSlot(dmf, DMSTAG_BACK_LEFT, 0, &ibackleftf));
     88:   PetscCall(DMStagGetLocationSlot(dmf, DMSTAG_BACK, 0, &ibackf));
     89:   PetscCall(DMStagGetLocationSlot(dmf, DMSTAG_DOWN_LEFT, 0, &idownleftf));
     90:   PetscCall(DMStagGetLocationSlot(dmf, DMSTAG_DOWN, 0, &idownf));
     91:   PetscCall(DMStagGetLocationSlot(dmf, DMSTAG_LEFT, 0, &ileftf));
     92:   PetscCall(DMStagGetLocationSlot(dmf, DMSTAG_ELEMENT, 0, &ielemf));
     93:   PetscCall(DMStagGetLocationSlot(dmc, DMSTAG_BACK_DOWN_LEFT, 0, &ibackdownleftc));
     94:   PetscCall(DMStagGetLocationSlot(dmc, DMSTAG_BACK_DOWN, 0, &ibackdownc));
     95:   PetscCall(DMStagGetLocationSlot(dmc, DMSTAG_BACK_LEFT, 0, &ibackleftc));
     96:   PetscCall(DMStagGetLocationSlot(dmc, DMSTAG_BACK, 0, &ibackc));
     97:   PetscCall(DMStagGetLocationSlot(dmc, DMSTAG_DOWN_LEFT, 0, &idownleftc));
     98:   PetscCall(DMStagGetLocationSlot(dmc, DMSTAG_DOWN, 0, &idownc));
     99:   PetscCall(DMStagGetLocationSlot(dmc, DMSTAG_LEFT, 0, &ileftc));
    100:   PetscCall(DMStagGetLocationSlot(dmc, DMSTAG_ELEMENT, 0, &ielemc));

    102:   for (d = 0; d < dof[0]; ++d)
    103:     for (k = zc; k < zc + pc + nExtrazc; ++k)
    104:       for (j = yc; j < yc + nc + nExtrayc; ++j)
    105:         for (i = xc; i < xc + mc + nExtraxc; ++i) {
    106:           const PetscInt ii = i * factorx, jj = j * factory, kk = k * factorz;

    108:           arrc[k][j][i][ibackdownleftc + d] = arrf[kk][jj][ii][ibackdownleftf + d];
    109:         }

    111:   for (d = 0; d < dof[1]; ++d)
    112:     for (k = zc; k < zc + pc + nExtrazc; ++k)
    113:       for (j = yc; j < yc + nc + nExtrayc; ++j)
    114:         for (i = xc; i < xc + mc; ++i) {
    115:           const PetscInt ii = i * factorx + factorx / 2, jj = j * factory, kk = k * factorz;

    117:           if (factorx % 2 == 0) arrc[k][j][i][ibackdownc + d] = 0.5 * (arrf[kk][jj][ii - 1][ibackdownf + d] + arrf[kk][jj][ii][ibackdownf + d]);
    118:           else arrc[k][j][i][ibackdownc + d] = arrf[kk][jj][ii][ibackdownf + d];
    119:         }

    121:   for (d = 0; d < dof[1]; ++d)
    122:     for (k = zc; k < zc + pc + nExtrazc; ++k)
    123:       for (j = yc; j < yc + nc; ++j)
    124:         for (i = xc; i < xc + mc + nExtraxc; ++i) {
    125:           const PetscInt ii = i * factorx, jj = j * factory + factory / 2, kk = k * factorz;

    127:           if (factory % 2 == 0) arrc[k][j][i][ibackleftc + d] = 0.5 * (arrf[kk][jj - 1][ii][ibackleftf + d] + arrf[kk][jj][ii][ibackleftf + d]);
    128:           else arrc[k][j][i][ibackleftc + d] = arrf[kk][jj][ii][ibackleftf + d];
    129:         }

    131:   for (d = 0; d < dof[1]; ++d)
    132:     for (k = zc; k < zc + pc; ++k)
    133:       for (j = yc; j < yc + nc + nExtrayc; ++j)
    134:         for (i = xc; i < xc + mc + nExtraxc; ++i) {
    135:           const PetscInt ii = i * factorx, jj = j * factory, kk = k * factorz + factorz / 2;

    137:           if (factorz % 2 == 0) arrc[k][j][i][idownleftc + d] = 0.5 * (arrf[kk - 1][jj][ii][idownleftf + d] + arrf[kk][jj][ii][idownleftf + d]);
    138:           else arrc[k][j][i][idownleftc + d] = arrf[kk][jj][ii][idownleftf + d];
    139:         }

    141:   for (d = 0; d < dof[2]; ++d)
    142:     for (k = zc; k < zc + pc + nExtrazc; ++k)
    143:       for (j = yc; j < yc + nc; ++j)
    144:         for (i = xc; i < xc + mc; ++i) {
    145:           const PetscInt ii = i * factorx + factorx / 2, jj = j * factory + factory / 2, kk = k * factorz;

    147:           if (factorx % 2 == 0 && factory % 2 == 0) arrc[k][j][i][ibackc + d] = 0.25 * (arrf[kk][jj - 1][ii - 1][ibackf + d] + arrf[kk][jj - 1][ii][ibackf + d] + arrf[kk][jj][ii - 1][ibackf + d] + arrf[kk][jj][ii][ibackf + d]);
    148:           else if (factorx % 2 == 0) arrc[k][j][i][ibackc + d] = 0.5 * (arrf[kk][jj][ii - 1][ibackf + d] + arrf[kk][jj][ii][ibackf + d]);
    149:           else if (factory % 2 == 0) arrc[k][j][i][ibackc + d] = 0.5 * (arrf[kk][jj - 1][ii][ibackf + d] + arrf[kk][jj][ii][ibackf + d]);
    150:           else arrc[k][j][i][ibackc + d] = arrf[kk][jj][ii][ibackf + d];
    151:         }

    153:   for (d = 0; d < dof[2]; ++d)
    154:     for (k = zc; k < zc + pc; ++k)
    155:       for (j = yc; j < yc + nc + nExtrayc; ++j)
    156:         for (i = xc; i < xc + mc; ++i) {
    157:           const PetscInt ii = i * factorx + factorx / 2, jj = j * factory, kk = k * factorz + factorz / 2;

    159:           if (factorx % 2 == 0 && factorz % 2 == 0) arrc[k][j][i][idownc + d] = 0.25 * (arrf[kk - 1][jj][ii - 1][idownf + d] + arrf[kk - 1][jj][ii][idownf + d] + arrf[kk][jj][ii - 1][idownf + d] + arrf[kk][jj][ii][idownf + d]);
    160:           else if (factorx % 2 == 0) arrc[k][j][i][idownc + d] = 0.5 * (arrf[kk][jj][ii - 1][idownf + d] + arrf[kk][jj][ii][idownf + d]);
    161:           else if (factorz % 2 == 0) arrc[k][j][i][idownc + d] = 0.5 * (arrf[kk - 1][jj][ii][idownf + d] + arrf[kk][jj][ii][idownf + d]);
    162:           else arrc[k][j][i][idownc + d] = arrf[kk][jj][ii][idownf + d];
    163:         }

    165:   for (d = 0; d < dof[2]; ++d)
    166:     for (k = zc; k < zc + pc; ++k)
    167:       for (j = yc; j < yc + nc; ++j)
    168:         for (i = xc; i < xc + mc + nExtraxc; ++i) {
    169:           const PetscInt ii = i * factorx, jj = j * factory + factory / 2, kk = k * factorz + factorz / 2;

    171:           if (factory % 2 == 0 && factorz % 2 == 0) arrc[k][j][i][ileftc + d] = 0.25 * (arrf[kk - 1][jj - 1][ii][ileftf + d] + arrf[kk - 1][jj][ii][ileftf + d] + arrf[kk][jj - 1][ii][ileftf + d] + arrf[kk][jj][ii][ileftf + d]);
    172:           else if (factory % 2 == 0) arrc[k][j][i][ileftc + d] = 0.5 * (arrf[kk][jj - 1][ii][ileftf + d] + arrf[kk][jj][ii][ileftf + d]);
    173:           else if (factorz % 2 == 0) arrc[k][j][i][ileftc + d] = 0.5 * (arrf[kk - 1][jj][ii][ileftf + d] + arrf[kk][jj][ii][ileftf + d]);
    174:           else arrc[k][j][i][ileftc + d] = arrf[kk][jj][ii][ileftf + d];
    175:         }

    177:   for (d = 0; d < dof[3]; ++d)
    178:     for (k = zc; k < zc + pc; ++k)
    179:       for (j = yc; j < yc + nc; ++j)
    180:         for (i = xc; i < xc + mc; ++i) {
    181:           const PetscInt ii = i * factorx + factorx / 2, jj = j * factory + factory / 2, kk = k * factorz + factorz / 2;

    183:           if (factorx % 2 == 0 && factory % 2 == 0 && factorz % 2 == 0)
    184:             arrc[k][j][i][ielemc + d] = 0.125 * (arrf[kk - 1][jj - 1][ii - 1][ielemf + d] + arrf[kk - 1][jj - 1][ii][ielemf + d] + arrf[kk - 1][jj][ii - 1][ielemf + d] + arrf[kk - 1][jj][ii][ielemf + d] + arrf[kk][jj - 1][ii - 1][ielemf + d] + arrf[kk][jj - 1][ii][ielemf + d] + arrf[kk][jj][ii - 1][ielemf + d] + arrf[kk][jj][ii][ielemf + d]);
    185:           else if (factorx % 2 == 0 && factory % 2 == 0) arrc[k][j][i][ielemc + d] = 0.25 * (arrf[kk][jj - 1][ii - 1][ielemf + d] + arrf[kk][jj - 1][ii][ielemf + d] + arrf[kk][jj][ii - 1][ielemf + d] + arrf[kk][jj][ii][ielemf + d]);
    186:           else if (factorx % 2 == 0 && factorz % 2 == 0) arrc[k][j][i][ielemc + d] = 0.25 * (arrf[kk - 1][jj][ii - 1][ielemf + d] + arrf[kk - 1][jj][ii][ielemf + d] + arrf[kk][jj][ii - 1][ielemf + d] + arrf[kk][jj][ii][ielemf + d]);
    187:           else if (factory % 2 == 0 && factorz % 2 == 0) arrc[k][j][i][ielemc + d] = 0.25 * (arrf[kk - 1][jj - 1][ii][ielemf + d] + arrf[kk - 1][jj][ii][ielemf + d] + arrf[kk][jj - 1][ii][ielemf + d] + arrf[kk][jj][ii][ielemf + d]);
    188:           else if (factorx % 2 == 0) arrc[k][j][i][ielemc + d] = 0.5 * (arrf[kk][jj][ii - 1][ielemf + d] + arrf[kk][jj][ii][ielemf + d]);
    189:           else if (factory % 2 == 0) arrc[k][j][i][ielemc + d] = 0.5 * (arrf[kk][jj - 1][ii][ielemf + d] + arrf[kk][jj][ii][ielemf + d]);
    190:           else if (factorz % 2 == 0) arrc[k][j][i][ielemc + d] = 0.5 * (arrf[kk - 1][jj][ii][ielemf + d] + arrf[kk][jj][ii][ielemf + d]);
    191:           else arrc[k][j][i][ielemc + d] = arrf[kk][jj][ii][ielemf + d];
    192:         }

    194:   PetscCall(DMStagVecRestoreArray(dmf, xf_local, &arrf));
    195:   PetscCall(DMStagVecRestoreArray(dmc, xc_local, &arrc));
    196:   PetscFunctionReturn(PETSC_SUCCESS);
    197: }

    199: PETSC_INTERN PetscErrorCode DMStagSetUniformCoordinatesExplicit_3d(DM dm, PetscReal xmin, PetscReal xmax, PetscReal ymin, PetscReal ymax, PetscReal zmin, PetscReal zmax)
    200: {
    201:   DM_Stag        *stagCoord;
    202:   DM              dmCoord;
    203:   Vec             coordLocal;
    204:   PetscReal       h[3], min[3];
    205:   PetscScalar ****arr;
    206:   PetscInt        ind[3], start_ghost[3], n_ghost[3], s, c;
    207:   PetscInt        ibackdownleft, ibackdown, ibackleft, iback, idownleft, idown, ileft, ielement;

    209:   PetscFunctionBegin;
    210:   PetscCall(DMGetCoordinateDM(dm, &dmCoord));
    211:   stagCoord = (DM_Stag *)dmCoord->data;
    212:   for (s = 0; s < 4; ++s) {
    213:     PetscCheck(stagCoord->dof[s] == 0 || stagCoord->dof[s] == 3, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Coordinate DM in 3 dimensions must have 0 or 3 dof on each stratum, but stratum %" PetscInt_FMT " has %" PetscInt_FMT " dof", s,
    214:                stagCoord->dof[s]);
    215:   }
    216:   PetscCall(DMCreateLocalVector(dmCoord, &coordLocal));
    217:   PetscCall(DMStagVecGetArray(dmCoord, coordLocal, &arr));
    218:   if (stagCoord->dof[0]) PetscCall(DMStagGetLocationSlot(dmCoord, DMSTAG_BACK_DOWN_LEFT, 0, &ibackdownleft));
    219:   if (stagCoord->dof[1]) {
    220:     PetscCall(DMStagGetLocationSlot(dmCoord, DMSTAG_BACK_DOWN, 0, &ibackdown));
    221:     PetscCall(DMStagGetLocationSlot(dmCoord, DMSTAG_BACK_LEFT, 0, &ibackleft));
    222:     PetscCall(DMStagGetLocationSlot(dmCoord, DMSTAG_DOWN_LEFT, 0, &idownleft));
    223:   }
    224:   if (stagCoord->dof[2]) {
    225:     PetscCall(DMStagGetLocationSlot(dmCoord, DMSTAG_BACK, 0, &iback));
    226:     PetscCall(DMStagGetLocationSlot(dmCoord, DMSTAG_DOWN, 0, &idown));
    227:     PetscCall(DMStagGetLocationSlot(dmCoord, DMSTAG_LEFT, 0, &ileft));
    228:   }
    229:   if (stagCoord->dof[3]) PetscCall(DMStagGetLocationSlot(dmCoord, DMSTAG_ELEMENT, 0, &ielement));
    230:   PetscCall(DMStagGetGhostCorners(dmCoord, &start_ghost[0], &start_ghost[1], &start_ghost[2], &n_ghost[0], &n_ghost[1], &n_ghost[2]));
    231:   min[0] = xmin;
    232:   min[1] = ymin;
    233:   min[2] = zmin;
    234:   h[0]   = (xmax - xmin) / stagCoord->N[0];
    235:   h[1]   = (ymax - ymin) / stagCoord->N[1];
    236:   h[2]   = (zmax - zmin) / stagCoord->N[2];

    238:   for (ind[2] = start_ghost[2]; ind[2] < start_ghost[2] + n_ghost[2]; ++ind[2]) {
    239:     for (ind[1] = start_ghost[1]; ind[1] < start_ghost[1] + n_ghost[1]; ++ind[1]) {
    240:       for (ind[0] = start_ghost[0]; ind[0] < start_ghost[0] + n_ghost[0]; ++ind[0]) {
    241:         if (stagCoord->dof[0]) {
    242:           const PetscReal offs[3] = {0.0, 0.0, 0.0};
    243:           for (c = 0; c < 3; ++c) arr[ind[2]][ind[1]][ind[0]][ibackdownleft + c] = min[c] + ((PetscReal)ind[c] + offs[c]) * h[c];
    244:         }
    245:         if (stagCoord->dof[1]) {
    246:           const PetscReal offs[3] = {0.5, 0.0, 0.0};
    247:           for (c = 0; c < 3; ++c) arr[ind[2]][ind[1]][ind[0]][ibackdown + c] = min[c] + ((PetscReal)ind[c] + offs[c]) * h[c];
    248:         }
    249:         if (stagCoord->dof[1]) {
    250:           const PetscReal offs[3] = {0.0, 0.5, 0.0};
    251:           for (c = 0; c < 3; ++c) arr[ind[2]][ind[1]][ind[0]][ibackleft + c] = min[c] + ((PetscReal)ind[c] + offs[c]) * h[c];
    252:         }
    253:         if (stagCoord->dof[2]) {
    254:           const PetscReal offs[3] = {0.5, 0.5, 0.0};
    255:           for (c = 0; c < 3; ++c) arr[ind[2]][ind[1]][ind[0]][iback + c] = min[c] + ((PetscReal)ind[c] + offs[c]) * h[c];
    256:         }
    257:         if (stagCoord->dof[1]) {
    258:           const PetscReal offs[3] = {0.0, 0.0, 0.5};
    259:           for (c = 0; c < 3; ++c) arr[ind[2]][ind[1]][ind[0]][idownleft + c] = min[c] + ((PetscReal)ind[c] + offs[c]) * h[c];
    260:         }
    261:         if (stagCoord->dof[2]) {
    262:           const PetscReal offs[3] = {0.5, 0.0, 0.5};
    263:           for (c = 0; c < 3; ++c) arr[ind[2]][ind[1]][ind[0]][idown + c] = min[c] + ((PetscReal)ind[c] + offs[c]) * h[c];
    264:         }
    265:         if (stagCoord->dof[2]) {
    266:           const PetscReal offs[3] = {0.0, 0.5, 0.5};
    267:           for (c = 0; c < 3; ++c) arr[ind[2]][ind[1]][ind[0]][ileft + c] = min[c] + ((PetscReal)ind[c] + offs[c]) * h[c];
    268:         }
    269:         if (stagCoord->dof[3]) {
    270:           const PetscReal offs[3] = {0.5, 0.5, 0.5};
    271:           for (c = 0; c < 3; ++c) arr[ind[2]][ind[1]][ind[0]][ielement + c] = min[c] + ((PetscReal)ind[c] + offs[c]) * h[c];
    272:         }
    273:       }
    274:     }
    275:   }
    276:   PetscCall(DMStagVecRestoreArray(dmCoord, coordLocal, &arr));
    277:   PetscCall(DMSetCoordinatesLocal(dm, coordLocal));
    278:   PetscCall(VecDestroy(&coordLocal));
    279:   PetscFunctionReturn(PETSC_SUCCESS);
    280: }

    282: /* Helper functions used in DMSetUp_Stag() */
    283: static PetscErrorCode DMStagSetUpBuildRankGrid_3d(DM);
    284: static PetscErrorCode DMStagSetUpBuildNeighbors_3d(DM);
    285: static PetscErrorCode DMStagSetUpBuildGlobalOffsets_3d(DM, PetscInt **);
    286: static PetscErrorCode DMStagSetUpBuildScatter_3d(DM, const PetscInt *);
    287: static PetscErrorCode DMStagSetUpBuildL2G_3d(DM, const PetscInt *);
    288: static PetscErrorCode DMStagComputeLocationOffsets_3d(DM);

    290: PETSC_INTERN PetscErrorCode DMSetUp_Stag_3d(DM dm)
    291: {
    292:   DM_Stag *const stag = (DM_Stag *)dm->data;
    293:   PetscMPIInt    rank;
    294:   PetscInt       i, j, d;
    295:   PetscInt      *globalOffsets;
    296:   const PetscInt dim = 3;

    298:   PetscFunctionBegin;
    299:   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));

    301:   /* Rank grid sizes (populates stag->nRanks) */
    302:   PetscCall(DMStagSetUpBuildRankGrid_3d(dm));

    304:   /* Determine location of rank in grid */
    305:   stag->rank[0] = rank % stag->nRanks[0];
    306:   stag->rank[1] = rank % (stag->nRanks[0] * stag->nRanks[1]) / stag->nRanks[0];
    307:   stag->rank[2] = rank / (stag->nRanks[0] * stag->nRanks[1]);
    308:   for (d = 0; d < dim; ++d) {
    309:     stag->firstRank[d] = PetscNot(stag->rank[d]);
    310:     stag->lastRank[d]  = (PetscBool)(stag->rank[d] == stag->nRanks[d] - 1);
    311:   }

    313:   /* Determine locally owned region (if not already prescribed).
    314:    Divide equally, giving lower ranks in each dimension and extra element if needbe.
    315:    Note that this uses O(P) storage. If this ever becomes an issue, this could
    316:    be refactored to not keep this data around.  */
    317:   for (i = 0; i < dim; ++i) {
    318:     if (!stag->l[i]) {
    319:       const PetscInt Ni = stag->N[i], nRanksi = stag->nRanks[i];
    320:       PetscCall(PetscMalloc1(stag->nRanks[i], &stag->l[i]));
    321:       for (j = 0; j < stag->nRanks[i]; ++j) stag->l[i][j] = Ni / nRanksi + ((Ni % nRanksi) > j);
    322:     }
    323:   }

    325:   /* Retrieve local size in stag->n */
    326:   for (i = 0; i < dim; ++i) stag->n[i] = stag->l[i][stag->rank[i]];
    327:   if (PetscDefined(USE_DEBUG)) {
    328:     for (i = 0; i < dim; ++i) {
    329:       PetscInt Ncheck, j;
    330:       Ncheck = 0;
    331:       for (j = 0; j < stag->nRanks[i]; ++j) Ncheck += stag->l[i][j];
    332:       PetscCheck(Ncheck == stag->N[i], PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Local sizes in dimension %" PetscInt_FMT " don't add up. %" PetscInt_FMT " != %" PetscInt_FMT, i, Ncheck, stag->N[i]);
    333:     }
    334:   }

    336:   /* Compute starting elements */
    337:   for (i = 0; i < dim; ++i) {
    338:     stag->start[i] = 0;
    339:     for (j = 0; j < stag->rank[i]; ++j) stag->start[i] += stag->l[i][j];
    340:   }

    342:   /* Determine ranks of neighbors */
    343:   PetscCall(DMStagSetUpBuildNeighbors_3d(dm));

    345:   /* Define useful sizes */
    346:   {
    347:     PetscInt  entriesPerEdge, entriesPerFace, entriesPerCorner, entriesPerElementRow, entriesPerElementLayer;
    348:     PetscBool dummyEnd[3];
    349:     for (d = 0; d < 3; ++d) dummyEnd[d] = (PetscBool)(stag->lastRank[d] && stag->boundaryType[d] != DM_BOUNDARY_PERIODIC);
    350:     stag->entriesPerElement = stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2] + stag->dof[3];
    351:     entriesPerFace          = stag->dof[0] + 2 * stag->dof[1] + stag->dof[2];
    352:     entriesPerEdge          = stag->dof[0] + stag->dof[1];
    353:     entriesPerCorner        = stag->dof[0];
    354:     entriesPerElementRow    = stag->n[0] * stag->entriesPerElement + (dummyEnd[0] ? entriesPerFace : 0);
    355:     entriesPerElementLayer  = stag->n[1] * entriesPerElementRow + (dummyEnd[1] ? stag->n[0] * entriesPerFace : 0) + (dummyEnd[1] && dummyEnd[0] ? entriesPerEdge : 0);
    356:     stag->entries = stag->n[2] * entriesPerElementLayer + (dummyEnd[2] ? stag->n[0] * stag->n[1] * entriesPerFace : 0) + (dummyEnd[2] && dummyEnd[0] ? stag->n[1] * entriesPerEdge : 0) + (dummyEnd[2] && dummyEnd[1] ? stag->n[0] * entriesPerEdge : 0) + (dummyEnd[2] && dummyEnd[1] && dummyEnd[0] ? entriesPerCorner : 0);
    357:   }

    359:   /* Check that we will not overflow 32-bit indices (slightly overconservative) */
    360: #if !defined(PETSC_USE_64BIT_INDICES)
    361:   PetscCheck(((PetscInt64)stag->n[0]) * ((PetscInt64)stag->n[1]) * ((PetscInt64)stag->n[2]) * ((PetscInt64)stag->entriesPerElement) <= (PetscInt64)PETSC_MPI_INT_MAX, PetscObjectComm((PetscObject)dm), PETSC_ERR_INT_OVERFLOW, "Mesh of %" PetscInt_FMT " x %" PetscInt_FMT " x %" PetscInt_FMT " with %" PetscInt_FMT " entries per (interior) element is likely too large for 32-bit indices",
    362:              stag->n[0], stag->n[1], stag->n[2], stag->entriesPerElement);
    363: #endif

    365:   /* Compute offsets for each rank into global vectors

    367:     This again requires O(P) storage, which could be replaced with some global
    368:     communication.
    369:   */
    370:   PetscCall(DMStagSetUpBuildGlobalOffsets_3d(dm, &globalOffsets));

    372:   for (d = 0; d < dim; ++d)
    373:     PetscCheck(stag->boundaryType[d] == DM_BOUNDARY_NONE || stag->boundaryType[d] == DM_BOUNDARY_PERIODIC || stag->boundaryType[d] == DM_BOUNDARY_GHOSTED, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported boundary type");

    375:   /* Define ghosted/local sizes */
    376:   for (d = 0; d < dim; ++d) {
    377:     switch (stag->boundaryType[d]) {
    378:     case DM_BOUNDARY_NONE:
    379:       /* Note: for a elements-only DMStag, the extra elements on the edges aren't necessary but we include them anyway */
    380:       switch (stag->stencilType) {
    381:       case DMSTAG_STENCIL_NONE: /* only the extra one on the right/top edges */
    382:         stag->nGhost[d]     = stag->n[d];
    383:         stag->startGhost[d] = stag->start[d];
    384:         if (stag->lastRank[d]) stag->nGhost[d] += 1;
    385:         break;
    386:       case DMSTAG_STENCIL_STAR: /* allocate the corners but don't use them */
    387:       case DMSTAG_STENCIL_BOX:
    388:         stag->nGhost[d]     = stag->n[d];
    389:         stag->startGhost[d] = stag->start[d];
    390:         if (!stag->firstRank[d]) {
    391:           stag->nGhost[d] += stag->stencilWidth; /* add interior ghost elements */
    392:           stag->startGhost[d] -= stag->stencilWidth;
    393:         }
    394:         if (!stag->lastRank[d]) {
    395:           stag->nGhost[d] += stag->stencilWidth; /* add interior ghost elements */
    396:         } else {
    397:           stag->nGhost[d] += 1; /* one element on the boundary to complete blocking */
    398:         }
    399:         break;
    400:       default:
    401:         SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unrecognized ghost stencil type %d", stag->stencilType);
    402:       }
    403:       break;
    404:     case DM_BOUNDARY_GHOSTED:
    405:       switch (stag->stencilType) {
    406:       case DMSTAG_STENCIL_NONE:
    407:         stag->startGhost[d] = stag->start[d];
    408:         stag->nGhost[d]     = stag->n[d] + (stag->lastRank[d] ? 1 : 0);
    409:         break;
    410:       case DMSTAG_STENCIL_STAR:
    411:       case DMSTAG_STENCIL_BOX:
    412:         stag->startGhost[d] = stag->start[d] - stag->stencilWidth; /* This value may be negative */
    413:         stag->nGhost[d]     = stag->n[d] + 2 * stag->stencilWidth + (stag->lastRank[d] && stag->stencilWidth == 0 ? 1 : 0);
    414:         break;
    415:       default:
    416:         SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unrecognized ghost stencil type %d", stag->stencilType);
    417:       }
    418:       break;
    419:     case DM_BOUNDARY_PERIODIC:
    420:       switch (stag->stencilType) {
    421:       case DMSTAG_STENCIL_NONE: /* only the extra one on the right/top edges */
    422:         stag->nGhost[d]     = stag->n[d];
    423:         stag->startGhost[d] = stag->start[d];
    424:         break;
    425:       case DMSTAG_STENCIL_STAR:
    426:       case DMSTAG_STENCIL_BOX:
    427:         stag->nGhost[d]     = stag->n[d] + 2 * stag->stencilWidth;
    428:         stag->startGhost[d] = stag->start[d] - stag->stencilWidth;
    429:         break;
    430:       default:
    431:         SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unrecognized ghost stencil type %d", stag->stencilType);
    432:       }
    433:       break;
    434:     default:
    435:       SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported boundary type in dimension %" PetscInt_FMT, d);
    436:     }
    437:   }
    438:   stag->entriesGhost = stag->nGhost[0] * stag->nGhost[1] * stag->nGhost[2] * stag->entriesPerElement;

    440:   /* Create global-->local VecScatter and local->global ISLocalToGlobalMapping */
    441:   PetscCall(DMStagSetUpBuildScatter_3d(dm, globalOffsets));
    442:   PetscCall(DMStagSetUpBuildL2G_3d(dm, globalOffsets));

    444:   /* In special cases, create a dedicated injective local-to-global map */
    445:   if ((stag->boundaryType[0] == DM_BOUNDARY_PERIODIC && stag->nRanks[0] == 1) || (stag->boundaryType[1] == DM_BOUNDARY_PERIODIC && stag->nRanks[1] == 1) || (stag->boundaryType[2] == DM_BOUNDARY_PERIODIC && stag->nRanks[2] == 1)) {
    446:     PetscCall(DMStagPopulateLocalToGlobalInjective(dm));
    447:   }

    449:   /* Free global offsets */
    450:   PetscCall(PetscFree(globalOffsets));

    452:   /* Precompute location offsets */
    453:   PetscCall(DMStagComputeLocationOffsets_3d(dm));

    455:   /* View from Options */
    456:   PetscCall(DMViewFromOptions(dm, NULL, "-dm_view"));
    457:   PetscFunctionReturn(PETSC_SUCCESS);
    458: }

    460: /* adapted from da3.c */
    461: static PetscErrorCode DMStagSetUpBuildRankGrid_3d(DM dm)
    462: {
    463:   PetscMPIInt    rank, size;
    464:   PetscMPIInt    m, n, p, pm;
    465:   DM_Stag *const stag = (DM_Stag *)dm->data;
    466:   const PetscInt M    = stag->N[0];
    467:   const PetscInt N    = stag->N[1];
    468:   const PetscInt P    = stag->N[2];

    470:   PetscFunctionBegin;
    471:   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)dm), &size));
    472:   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));

    474:   m = stag->nRanks[0];
    475:   n = stag->nRanks[1];
    476:   p = stag->nRanks[2];

    478:   if (m != PETSC_DECIDE) {
    479:     PetscCheck(m >= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Non-positive number of processors in X direction: %d", m);
    480:     PetscCheck(m <= size, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Too many processors in X direction: %d %d", m, size);
    481:   }
    482:   if (n != PETSC_DECIDE) {
    483:     PetscCheck(n >= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Non-positive number of processors in Y direction: %d", n);
    484:     PetscCheck(n <= size, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Too many processors in Y direction: %d %d", n, size);
    485:   }
    486:   if (p != PETSC_DECIDE) {
    487:     PetscCheck(p >= 1, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Non-positive number of processors in Z direction: %d", p);
    488:     PetscCheck(p <= size, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Too many processors in Z direction: %d %d", p, size);
    489:   }
    490:   PetscCheck(m <= 0 || n <= 0 || p <= 0 || m * n * p == size, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "m %d * n %d * p %d != size %d", m, n, p, size);

    492:   /* Partition the array among the processors */
    493:   if (m == PETSC_DECIDE && n != PETSC_DECIDE && p != PETSC_DECIDE) {
    494:     m = size / (n * p);
    495:   } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) {
    496:     n = size / (m * p);
    497:   } else if (m != PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) {
    498:     p = size / (m * n);
    499:   } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) {
    500:     /* try for squarish distribution */
    501:     m = (PetscMPIInt)(0.5 + PetscSqrtReal(((PetscReal)M) * ((PetscReal)size) / ((PetscReal)N * p)));
    502:     if (!m) m = 1;
    503:     while (m > 0) {
    504:       n = size / (m * p);
    505:       if (m * n * p == size) break;
    506:       m--;
    507:     }
    508:     PetscCheck(m, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "bad p value: p = %d", p);
    509:     if (M > N && m < n) {
    510:       PetscMPIInt _m = m;
    511:       m              = n;
    512:       n              = _m;
    513:     }
    514:   } else if (m == PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) {
    515:     /* try for squarish distribution */
    516:     m = (PetscMPIInt)(0.5 + PetscSqrtReal(((PetscReal)M) * ((PetscReal)size) / ((PetscReal)P * n)));
    517:     if (!m) m = 1;
    518:     while (m > 0) {
    519:       p = size / (m * n);
    520:       if (m * n * p == size) break;
    521:       m--;
    522:     }
    523:     PetscCheck(m, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "bad n value: n = %d", n);
    524:     if (M > P && m < p) {
    525:       PetscMPIInt _m = m;
    526:       m              = p;
    527:       p              = _m;
    528:     }
    529:   } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) {
    530:     /* try for squarish distribution */
    531:     n = (PetscMPIInt)(0.5 + PetscSqrtReal(((PetscReal)N) * ((PetscReal)size) / ((PetscReal)P * m)));
    532:     if (!n) n = 1;
    533:     while (n > 0) {
    534:       p = size / (m * n);
    535:       if (m * n * p == size) break;
    536:       n--;
    537:     }
    538:     PetscCheck(n, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "bad m value: m = %d", n);
    539:     if (N > P && n < p) {
    540:       PetscMPIInt _n = n;
    541:       n              = p;
    542:       p              = _n;
    543:     }
    544:   } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) {
    545:     /* try for squarish distribution */
    546:     n = (PetscMPIInt)(0.5 + PetscPowReal(((PetscReal)N * N) * ((PetscReal)size) / ((PetscReal)P * M), (PetscReal)(1. / 3.)));
    547:     if (!n) n = 1;
    548:     while (n > 0) {
    549:       pm = size / n;
    550:       if (n * pm == size) break;
    551:       n--;
    552:     }
    553:     if (!n) n = 1;
    554:     m = (PetscMPIInt)(0.5 + PetscSqrtReal(((PetscReal)M) * ((PetscReal)size) / ((PetscReal)P * n)));
    555:     if (!m) m = 1;
    556:     while (m > 0) {
    557:       p = size / (m * n);
    558:       if (m * n * p == size) break;
    559:       m--;
    560:     }
    561:     if (M > P && m < p) {
    562:       PetscMPIInt _m = m;
    563:       m              = p;
    564:       p              = _m;
    565:     }
    566:   } else PetscCheck(m * n * p == size, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Given Bad partition");

    568:   PetscCheck(m * n * p == size, PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "Could not find good partition");
    569:   PetscCheck(M >= m, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Partition in x direction is too fine! %" PetscInt_FMT " %d", M, m);
    570:   PetscCheck(N >= n, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Partition in y direction is too fine! %" PetscInt_FMT " %d", N, n);
    571:   PetscCheck(P >= p, PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Partition in z direction is too fine! %" PetscInt_FMT " %d", P, p);

    573:   stag->nRanks[0] = m;
    574:   stag->nRanks[1] = n;
    575:   stag->nRanks[2] = p;
    576:   PetscFunctionReturn(PETSC_SUCCESS);
    577: }

    579: /* Determine ranks of neighbors, using DMDA's convention

    581:         n24 n25 n26
    582:         n21 n22 n23
    583:         n18 n19 n20 (Front, bigger z)

    585:         n15 n16 n17
    586:         n12     n14   ^ y
    587:         n9  n10 n11   |
    588:                       +--> x
    589:         n6  n7  n8
    590:         n3  n4  n5
    591:         n0  n1  n2 (Back, smaller z) */
    592: static PetscErrorCode DMStagSetUpBuildNeighbors_3d(DM dm)
    593: {
    594:   DM_Stag *const stag = (DM_Stag *)dm->data;
    595:   PetscInt       d, i;
    596:   PetscBool      per[3], first[3], last[3];
    597:   PetscMPIInt    neighborRank[27][3], r[3], n[3];
    598:   const PetscInt dim = 3;

    600:   PetscFunctionBegin;
    601:   for (d = 0; d < dim; ++d)
    602:     PetscCheck(stag->boundaryType[d] == DM_BOUNDARY_NONE || stag->boundaryType[d] == DM_BOUNDARY_PERIODIC || stag->boundaryType[d] == DM_BOUNDARY_GHOSTED, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Neighbor determination not implemented for %s",
    603:                DMBoundaryTypes[stag->boundaryType[d]]);

    605:   /* Assemble some convenience variables */
    606:   for (d = 0; d < dim; ++d) {
    607:     per[d]   = (PetscBool)(stag->boundaryType[d] == DM_BOUNDARY_PERIODIC);
    608:     first[d] = stag->firstRank[d];
    609:     last[d]  = stag->lastRank[d];
    610:     r[d]     = stag->rank[d];
    611:     n[d]     = stag->nRanks[d];
    612:   }

    614:   /* First, compute the position in the rank grid for all neighbors */

    616:   neighborRank[0][0] = first[0] ? (per[0] ? n[0] - 1 : -1) : r[0] - 1; /* left  down back  */
    617:   neighborRank[0][1] = first[1] ? (per[1] ? n[1] - 1 : -1) : r[1] - 1;
    618:   neighborRank[0][2] = first[2] ? (per[2] ? n[2] - 1 : -1) : r[2] - 1;

    620:   neighborRank[1][0] = r[0]; /*       down back  */
    621:   neighborRank[1][1] = first[1] ? (per[1] ? n[1] - 1 : -1) : r[1] - 1;
    622:   neighborRank[1][2] = first[2] ? (per[2] ? n[2] - 1 : -1) : r[2] - 1;

    624:   neighborRank[2][0] = last[0] ? (per[0] ? 0 : -1) : r[0] + 1; /* right down back  */
    625:   neighborRank[2][1] = first[1] ? (per[1] ? n[1] - 1 : -1) : r[1] - 1;
    626:   neighborRank[2][2] = first[2] ? (per[2] ? n[2] - 1 : -1) : r[2] - 1;

    628:   neighborRank[3][0] = first[0] ? (per[0] ? n[0] - 1 : -1) : r[0] - 1; /* left       back  */
    629:   neighborRank[3][1] = r[1];
    630:   neighborRank[3][2] = first[2] ? (per[2] ? n[2] - 1 : -1) : r[2] - 1;

    632:   neighborRank[4][0] = r[0]; /*            back  */
    633:   neighborRank[4][1] = r[1];
    634:   neighborRank[4][2] = first[2] ? (per[2] ? n[2] - 1 : -1) : r[2] - 1;

    636:   neighborRank[5][0] = last[0] ? (per[0] ? 0 : -1) : r[0] + 1; /* right      back  */
    637:   neighborRank[5][1] = r[1];
    638:   neighborRank[5][2] = first[2] ? (per[2] ? n[2] - 1 : -1) : r[2] - 1;

    640:   neighborRank[6][0] = first[0] ? (per[0] ? n[0] - 1 : -1) : r[0] - 1; /* left  up   back  */
    641:   neighborRank[6][1] = last[1] ? (per[1] ? 0 : -1) : r[1] + 1;
    642:   neighborRank[6][2] = first[2] ? (per[2] ? n[2] - 1 : -1) : r[2] - 1;

    644:   neighborRank[7][0] = r[0]; /*       up   back  */
    645:   neighborRank[7][1] = last[1] ? (per[1] ? 0 : -1) : r[1] + 1;
    646:   neighborRank[7][2] = first[2] ? (per[2] ? n[2] - 1 : -1) : r[2] - 1;

    648:   neighborRank[8][0] = last[0] ? (per[0] ? 0 : -1) : r[0] + 1; /* right up   back  */
    649:   neighborRank[8][1] = last[1] ? (per[1] ? 0 : -1) : r[1] + 1;
    650:   neighborRank[8][2] = first[2] ? (per[2] ? n[2] - 1 : -1) : r[2] - 1;

    652:   neighborRank[9][0] = first[0] ? (per[0] ? n[0] - 1 : -1) : r[0] - 1; /* left  down       */
    653:   neighborRank[9][1] = first[1] ? (per[1] ? n[1] - 1 : -1) : r[1] - 1;
    654:   neighborRank[9][2] = r[2];

    656:   neighborRank[10][0] = r[0]; /*       down       */
    657:   neighborRank[10][1] = first[1] ? (per[1] ? n[1] - 1 : -1) : r[1] - 1;
    658:   neighborRank[10][2] = r[2];

    660:   neighborRank[11][0] = last[0] ? (per[0] ? 0 : -1) : r[0] + 1; /* right down       */
    661:   neighborRank[11][1] = first[1] ? (per[1] ? n[1] - 1 : -1) : r[1] - 1;
    662:   neighborRank[11][2] = r[2];

    664:   neighborRank[12][0] = first[0] ? (per[0] ? n[0] - 1 : -1) : r[0] - 1; /* left             */
    665:   neighborRank[12][1] = r[1];
    666:   neighborRank[12][2] = r[2];

    668:   neighborRank[13][0] = r[0];
    669:   neighborRank[13][1] = r[1];
    670:   neighborRank[13][2] = r[2];

    672:   neighborRank[14][0] = last[0] ? (per[0] ? 0 : -1) : r[0] + 1; /* right            */
    673:   neighborRank[14][1] = r[1];
    674:   neighborRank[14][2] = r[2];

    676:   neighborRank[15][0] = first[0] ? (per[0] ? n[0] - 1 : -1) : r[0] - 1; /* left  up         */
    677:   neighborRank[15][1] = last[1] ? (per[1] ? 0 : -1) : r[1] + 1;
    678:   neighborRank[15][2] = r[2];

    680:   neighborRank[16][0] = r[0]; /*       up         */
    681:   neighborRank[16][1] = last[1] ? (per[1] ? 0 : -1) : r[1] + 1;
    682:   neighborRank[16][2] = r[2];

    684:   neighborRank[17][0] = last[0] ? (per[0] ? 0 : -1) : r[0] + 1; /* right up         */
    685:   neighborRank[17][1] = last[1] ? (per[1] ? 0 : -1) : r[1] + 1;
    686:   neighborRank[17][2] = r[2];

    688:   neighborRank[18][0] = first[0] ? (per[0] ? n[0] - 1 : -1) : r[0] - 1; /* left  down front */
    689:   neighborRank[18][1] = first[1] ? (per[1] ? n[1] - 1 : -1) : r[1] - 1;
    690:   neighborRank[18][2] = last[2] ? (per[2] ? 0 : -1) : r[2] + 1;

    692:   neighborRank[19][0] = r[0]; /*       down front */
    693:   neighborRank[19][1] = first[1] ? (per[1] ? n[1] - 1 : -1) : r[1] - 1;
    694:   neighborRank[19][2] = last[2] ? (per[2] ? 0 : -1) : r[2] + 1;

    696:   neighborRank[20][0] = last[0] ? (per[0] ? 0 : -1) : r[0] + 1; /* right down front */
    697:   neighborRank[20][1] = first[1] ? (per[1] ? n[1] - 1 : -1) : r[1] - 1;
    698:   neighborRank[20][2] = last[2] ? (per[2] ? 0 : -1) : r[2] + 1;

    700:   neighborRank[21][0] = first[0] ? (per[0] ? n[0] - 1 : -1) : r[0] - 1; /* left       front */
    701:   neighborRank[21][1] = r[1];
    702:   neighborRank[21][2] = last[2] ? (per[2] ? 0 : -1) : r[2] + 1;

    704:   neighborRank[22][0] = r[0]; /*            front */
    705:   neighborRank[22][1] = r[1];
    706:   neighborRank[22][2] = last[2] ? (per[2] ? 0 : -1) : r[2] + 1;

    708:   neighborRank[23][0] = last[0] ? (per[0] ? 0 : -1) : r[0] + 1; /* right      front */
    709:   neighborRank[23][1] = r[1];
    710:   neighborRank[23][2] = last[2] ? (per[2] ? 0 : -1) : r[2] + 1;

    712:   neighborRank[24][0] = first[0] ? (per[0] ? n[0] - 1 : -1) : r[0] - 1; /* left  up   front */
    713:   neighborRank[24][1] = last[1] ? (per[1] ? 0 : -1) : r[1] + 1;
    714:   neighborRank[24][2] = last[2] ? (per[2] ? 0 : -1) : r[2] + 1;

    716:   neighborRank[25][0] = r[0]; /*       up   front */
    717:   neighborRank[25][1] = last[1] ? (per[1] ? 0 : -1) : r[1] + 1;
    718:   neighborRank[25][2] = last[2] ? (per[2] ? 0 : -1) : r[2] + 1;

    720:   neighborRank[26][0] = last[0] ? (per[0] ? 0 : -1) : r[0] + 1; /* right up   front */
    721:   neighborRank[26][1] = last[1] ? (per[1] ? 0 : -1) : r[1] + 1;
    722:   neighborRank[26][2] = last[2] ? (per[2] ? 0 : -1) : r[2] + 1;

    724:   /* Then, compute the rank of each in the linear ordering */
    725:   PetscCall(PetscMalloc1(27, &stag->neighbors));
    726:   for (i = 0; i < 27; ++i) {
    727:     if (neighborRank[i][0] >= 0 && neighborRank[i][1] >= 0 && neighborRank[i][2] >= 0) {
    728:       stag->neighbors[i] = neighborRank[i][0] + n[0] * neighborRank[i][1] + n[0] * n[1] * neighborRank[i][2];
    729:     } else {
    730:       stag->neighbors[i] = -1;
    731:     }
    732:   }
    733:   PetscFunctionReturn(PETSC_SUCCESS);
    734: }

    736: static PetscErrorCode DMStagSetUpBuildGlobalOffsets_3d(DM dm, PetscInt **pGlobalOffsets)
    737: {
    738:   const DM_Stag *const stag = (DM_Stag *)dm->data;
    739:   PetscInt            *globalOffsets;
    740:   PetscInt             i, j, k, d, entriesPerEdge, entriesPerFace, count;
    741:   PetscMPIInt          size;
    742:   PetscBool            extra[3];

    744:   PetscFunctionBegin;
    745:   for (d = 0; d < 3; ++d) extra[d] = (PetscBool)(stag->boundaryType[d] != DM_BOUNDARY_PERIODIC); /* Extra points in global rep */
    746:   entriesPerFace = stag->dof[0] + 2 * stag->dof[1] + stag->dof[2];
    747:   entriesPerEdge = stag->dof[0] + stag->dof[1];
    748:   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)dm), &size));
    749:   PetscCall(PetscMalloc1(size, pGlobalOffsets));
    750:   globalOffsets    = *pGlobalOffsets;
    751:   globalOffsets[0] = 0;
    752:   count            = 1; /* note the count is offset by 1 here. We add the size of the previous rank */
    753:   for (k = 0; k < stag->nRanks[2] - 1; ++k) {
    754:     const PetscInt nnk = stag->l[2][k];
    755:     for (j = 0; j < stag->nRanks[1] - 1; ++j) {
    756:       const PetscInt nnj = stag->l[1][j];
    757:       for (i = 0; i < stag->nRanks[0] - 1; ++i) {
    758:         const PetscInt nni = stag->l[0][i];
    759:         /* Interior : No right/top/front boundaries */
    760:         globalOffsets[count] = globalOffsets[count - 1] + nni * nnj * nnk * stag->entriesPerElement;
    761:         ++count;
    762:       }
    763:       {
    764:         /* Right boundary - extra faces */
    765:         /* i = stag->nRanks[0]-1; */
    766:         const PetscInt nni   = stag->l[0][i];
    767:         globalOffsets[count] = globalOffsets[count - 1] + nnj * nni * nnk * stag->entriesPerElement + (extra[0] ? nnj * nnk * entriesPerFace : 0);
    768:         ++count;
    769:       }
    770:     }
    771:     {
    772:       /* j = stag->nRanks[1]-1; */
    773:       const PetscInt nnj = stag->l[1][j];
    774:       for (i = 0; i < stag->nRanks[0] - 1; ++i) {
    775:         const PetscInt nni = stag->l[0][i];
    776:         /* Up boundary - extra faces */
    777:         globalOffsets[count] = globalOffsets[count - 1] + nnj * nni * nnk * stag->entriesPerElement + (extra[1] ? nni * nnk * entriesPerFace : 0);
    778:         ++count;
    779:       }
    780:       {
    781:         /* i = stag->nRanks[0]-1; */
    782:         const PetscInt nni = stag->l[0][i];
    783:         /* Up right boundary - 2x extra faces and extra edges */
    784:         globalOffsets[count] = globalOffsets[count - 1] + nnj * nni * nnk * stag->entriesPerElement + (extra[0] ? nnj * nnk * entriesPerFace : 0) + (extra[1] ? nni * nnk * entriesPerFace : 0) + (extra[0] && extra[1] ? nnk * entriesPerEdge : 0);
    785:         ++count;
    786:       }
    787:     }
    788:   }
    789:   {
    790:     /* k = stag->nRanks[2]-1; */
    791:     const PetscInt nnk = stag->l[2][k];
    792:     for (j = 0; j < stag->nRanks[1] - 1; ++j) {
    793:       const PetscInt nnj = stag->l[1][j];
    794:       for (i = 0; i < stag->nRanks[0] - 1; ++i) {
    795:         const PetscInt nni = stag->l[0][i];
    796:         /* Front boundary - extra faces */
    797:         globalOffsets[count] = globalOffsets[count - 1] + nnj * nni * nnk * stag->entriesPerElement + (extra[2] ? nni * nnj * entriesPerFace : 0);
    798:         ++count;
    799:       }
    800:       {
    801:         /* i = stag->nRanks[0]-1; */
    802:         const PetscInt nni = stag->l[0][i];
    803:         /* Front right boundary - 2x extra faces and extra edges */
    804:         globalOffsets[count] = globalOffsets[count - 1] + nnj * nni * nnk * stag->entriesPerElement + (extra[0] ? nnk * nnj * entriesPerFace : 0) + (extra[2] ? nni * nnj * entriesPerFace : 0) + (extra[0] && extra[2] ? nnj * entriesPerEdge : 0);
    805:         ++count;
    806:       }
    807:     }
    808:     {
    809:       /* j = stag->nRanks[1]-1; */
    810:       const PetscInt nnj = stag->l[1][j];
    811:       for (i = 0; i < stag->nRanks[0] - 1; ++i) {
    812:         const PetscInt nni = stag->l[0][i];
    813:         /* Front up boundary - 2x extra faces and extra edges */
    814:         globalOffsets[count] = globalOffsets[count - 1] + nnj * nni * nnk * stag->entriesPerElement + (extra[1] ? nnk * nni * entriesPerFace : 0) + (extra[2] ? nnj * nni * entriesPerFace : 0) + (extra[1] && extra[2] ? nni * entriesPerEdge : 0);
    815:         ++count;
    816:       }
    817:       /* Don't need to compute entries in last element */
    818:     }
    819:   }
    820:   PetscFunctionReturn(PETSC_SUCCESS);
    821: }

    823: /* A helper function to reduce code duplication as we loop over various ranges.
    824:    i,j,k refer to element numbers on the rank where an element lives in the global
    825:    representation (without ghosts) to be offset by a global offset per rank.
    826:    ig,jg,kg refer to element numbers in the local (this rank) ghosted numbering.
    827:    Note that this function could easily be converted to a macro (it does not compute
    828:    anything except loop indices and the values of idxGlobal and idxLocal).  */
    829: static PetscErrorCode DMStagSetUpBuildScatterPopulateIdx_3d(DM_Stag *stag, PetscInt *count, PetscInt *idxLocal, PetscInt *idxGlobal, PetscInt entriesPerEdge, PetscInt entriesPerFace, PetscInt eprNeighbor, PetscInt eplNeighbor, PetscInt eprGhost, PetscInt eplGhost, PetscInt epFaceRow, PetscInt globalOffset, PetscInt startx, PetscInt starty, PetscInt startz, PetscInt startGhostx, PetscInt startGhosty, PetscInt startGhostz, PetscInt endGhostx, PetscInt endGhosty, PetscInt endGhostz, PetscBool extrax, PetscBool extray, PetscBool extraz)
    830: {
    831:   PetscInt ig, jg, kg, d, c;

    833:   PetscFunctionBegin;
    834:   c = *count;
    835:   for (kg = startGhostz; kg < endGhostz; ++kg) {
    836:     const PetscInt k = kg - startGhostz + startz;
    837:     for (jg = startGhosty; jg < endGhosty; ++jg) {
    838:       const PetscInt j = jg - startGhosty + starty;
    839:       for (ig = startGhostx; ig < endGhostx; ++ig) {
    840:         const PetscInt i = ig - startGhostx + startx;
    841:         for (d = 0; d < stag->entriesPerElement; ++d, ++c) {
    842:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * eprNeighbor + i * stag->entriesPerElement + d;
    843:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + d;
    844:         }
    845:       }
    846:       if (extrax) {
    847:         PetscInt       dLocal;
    848:         const PetscInt i = endGhostx - startGhostx + startx;
    849:         ig               = endGhostx;
    850:         for (d = 0, dLocal = 0; d < stag->dof[0]; ++d, ++dLocal, ++c) { /* Vertex */
    851:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * eprNeighbor + i * stag->entriesPerElement + d;
    852:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + dLocal;
    853:         }
    854:         dLocal += stag->dof[1];                                                /* Skip back down edge */
    855:         for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++c) { /* Back left edge */
    856:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * eprNeighbor + i * stag->entriesPerElement + d;
    857:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + dLocal;
    858:         }
    859:         dLocal += stag->dof[2];                                                               /* Skip back face */
    860:         for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++c) { /* Down left edge */
    861:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * eprNeighbor + i * stag->entriesPerElement + d;
    862:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + dLocal;
    863:         }
    864:         dLocal += stag->dof[2];                                                                   /* Skip down face */
    865:         for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2]; ++d, ++dLocal, ++c) { /* Left face */
    866:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * eprNeighbor + i * stag->entriesPerElement + d;
    867:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + dLocal;
    868:         }
    869:         /* Skip element */
    870:       }
    871:     }
    872:     if (extray) {
    873:       const PetscInt j = endGhosty - startGhosty + starty;
    874:       jg               = endGhosty;
    875:       for (ig = startGhostx; ig < endGhostx; ++ig) {
    876:         const PetscInt i = ig - startGhostx + startx;
    877:         /* Vertex and Back down edge */
    878:         PetscInt dLocal;
    879:         for (d = 0, dLocal = 0; d < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++c) {              /* Vertex */
    880:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * eprNeighbor + i * entriesPerFace + d; /* Note face increment in  x */
    881:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + dLocal;
    882:         }
    883:         /* Skip back left edge and back face */
    884:         dLocal += stag->dof[1] + stag->dof[2];
    885:         /* Down face and down left edge */
    886:         for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++d, ++dLocal, ++c) {   /* Back left edge */
    887:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * eprNeighbor + i * entriesPerFace + d; /* Note face increment in x */
    888:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + dLocal;
    889:         }
    890:         /* Skip left face and element */
    891:       }
    892:       if (extrax) {
    893:         PetscInt       dLocal;
    894:         const PetscInt i = endGhostx - startGhostx + startx;
    895:         ig               = endGhostx;
    896:         for (d = 0, dLocal = 0; d < stag->dof[0]; ++d, ++dLocal, ++c) {                             /* Vertex */
    897:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * eprNeighbor + i * entriesPerFace + d; /* Note face increment in x */
    898:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + dLocal;
    899:         }
    900:         dLocal += 2 * stag->dof[1] + stag->dof[2];                                                  /* Skip back down edge, back face, and back left edge */
    901:         for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++c) {       /* Down left edge */
    902:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * eprNeighbor + i * entriesPerFace + d; /* Note face increment in x */
    903:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + dLocal;
    904:         }
    905:         /* Skip remaining entries */
    906:       }
    907:     }
    908:   }
    909:   if (extraz) {
    910:     const PetscInt k = endGhostz - startGhostz + startz;
    911:     kg               = endGhostz;
    912:     for (jg = startGhosty; jg < endGhosty; ++jg) {
    913:       const PetscInt j = jg - startGhosty + starty;
    914:       for (ig = startGhostx; ig < endGhostx; ++ig) {
    915:         const PetscInt i = ig - startGhostx + startx;
    916:         for (d = 0; d < entriesPerFace; ++d, ++c) {
    917:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * epFaceRow + i * entriesPerFace + d; /* Note face-based x and y increments */
    918:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + d;
    919:         }
    920:       }
    921:       if (extrax) {
    922:         PetscInt       dLocal;
    923:         const PetscInt i = endGhostx - startGhostx + startx;
    924:         ig               = endGhostx;
    925:         for (d = 0, dLocal = 0; d < stag->dof[0]; ++d, ++dLocal, ++c) {                           /* Vertex */
    926:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * epFaceRow + i * entriesPerFace + d; /* Note face-based x and y increments */
    927:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + dLocal;
    928:         }
    929:         dLocal += stag->dof[1];                                                                   /* Skip back down edge */
    930:         for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++c) {                    /* Back left edge */
    931:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * epFaceRow + i * entriesPerFace + d; /* Note face-based x and y increments */
    932:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + dLocal;
    933:         }
    934:         /* Skip the rest */
    935:       }
    936:     }
    937:     if (extray) {
    938:       const PetscInt j = endGhosty - startGhosty + starty;
    939:       jg               = endGhosty;
    940:       for (ig = startGhostx; ig < endGhostx; ++ig) {
    941:         const PetscInt i = ig - startGhostx + startx;
    942:         for (d = 0; d < entriesPerEdge; ++d, ++c) {
    943:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * epFaceRow + i * entriesPerEdge + d; /* Note face-based y increment and edge-based x increment */
    944:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + d;
    945:         }
    946:       }
    947:       if (extrax) {
    948:         const PetscInt i = endGhostx - startGhostx + startx;
    949:         ig               = endGhostx;
    950:         for (d = 0; d < stag->dof[0]; ++d, ++c) {                                                 /* Vertex (only) */
    951:           idxGlobal[c] = globalOffset + k * eplNeighbor + j * epFaceRow + i * entriesPerEdge + d; /* Note face-based y increment and edge-based x increment */
    952:           idxLocal[c]  = kg * eplGhost + jg * eprGhost + ig * stag->entriesPerElement + d;
    953:         }
    954:       }
    955:     }
    956:   }
    957:   *count = c;
    958:   PetscFunctionReturn(PETSC_SUCCESS);
    959: }

    961: static PetscErrorCode DMStagSetUpBuildScatter_3d(DM dm, const PetscInt *globalOffsets)
    962: {
    963:   DM_Stag *const stag = (DM_Stag *)dm->data;
    964:   PetscInt       d, ghostOffsetStart[3], ghostOffsetEnd[3], entriesPerCorner, entriesPerEdge, entriesPerFace, entriesToTransferTotal, count, eprGhost, eplGhost;
    965:   PetscInt      *idxLocal, *idxGlobal;
    966:   PetscMPIInt    rank;
    967:   PetscInt       nNeighbors[27][3];
    968:   PetscBool      star, nextToDummyEnd[3], dummyStart[3], dummyEnd[3];

    970:   PetscFunctionBegin;
    971:   PetscCallMPI(MPI_Comm_rank(PetscObjectComm((PetscObject)dm), &rank));
    972:   if (stag->stencilType != DMSTAG_STENCIL_NONE && (stag->n[0] < stag->stencilWidth || stag->n[1] < stag->stencilWidth || stag->n[2] < stag->stencilWidth)) {
    973:     SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "DMStag 3d setup does not support local sizes (%" PetscInt_FMT " x %" PetscInt_FMT " x %" PetscInt_FMT ") smaller than the elementwise stencil width (%" PetscInt_FMT ")", stag->n[0], stag->n[1], stag->n[2],
    974:             stag->stencilWidth);
    975:   }

    977:   /* Check stencil type */
    978:   PetscCheck(stag->stencilType == DMSTAG_STENCIL_NONE || stag->stencilType == DMSTAG_STENCIL_BOX || stag->stencilType == DMSTAG_STENCIL_STAR, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported stencil type %s", DMStagStencilTypes[stag->stencilType]);
    979:   star = (PetscBool)(stag->stencilType == DMSTAG_STENCIL_STAR || stag->stencilType == DMSTAG_STENCIL_NONE);

    981:   /* Compute numbers of elements on each neighbor */
    982:   {
    983:     PetscInt i;
    984:     for (i = 0; i < 27; ++i) {
    985:       const PetscInt neighborRank = stag->neighbors[i];
    986:       if (neighborRank >= 0) { /* note we copy the values for our own rank (neighbor 13) */
    987:         nNeighbors[i][0] = stag->l[0][neighborRank % stag->nRanks[0]];
    988:         nNeighbors[i][1] = stag->l[1][neighborRank % (stag->nRanks[0] * stag->nRanks[1]) / stag->nRanks[0]];
    989:         nNeighbors[i][2] = stag->l[2][neighborRank / (stag->nRanks[0] * stag->nRanks[1])];
    990:       } /* else leave uninitialized - error if accessed */
    991:     }
    992:   }

    994:   /* These offsets should always be non-negative, and describe how many
    995:      ghost elements exist at each boundary. These are not always equal to the stencil width,
    996:      because we may have different numbers of ghost elements at the boundaries. In particular,
    997:      in the non-periodic casewe always have at least one ghost (dummy) element at the right/top/front. */
    998:   for (d = 0; d < 3; ++d) ghostOffsetStart[d] = stag->start[d] - stag->startGhost[d];
    999:   for (d = 0; d < 3; ++d) ghostOffsetEnd[d] = stag->startGhost[d] + stag->nGhost[d] - (stag->start[d] + stag->n[d]);

    1001:   /* Determine whether the ghost region includes dummies or not. This is currently
    1002:      equivalent to having a non-periodic boundary. If not, then
    1003:      ghostOffset{Start,End}[d] elements correspond to elements on the neighbor.
    1004:      If true, then
    1005:      - at the start, there are ghostOffsetStart[d] ghost elements
    1006:      - at the end, there is a layer of extra "physical" points inside a layer of
    1007:        ghostOffsetEnd[d] ghost elements
    1008:      Note that this computation should be updated if any boundary types besides
    1009:      NONE, GHOSTED, and PERIODIC are supported.  */
    1010:   for (d = 0; d < 3; ++d) dummyStart[d] = (PetscBool)(stag->firstRank[d] && stag->boundaryType[d] != DM_BOUNDARY_PERIODIC);
    1011:   for (d = 0; d < 3; ++d) dummyEnd[d] = (PetscBool)(stag->lastRank[d] && stag->boundaryType[d] != DM_BOUNDARY_PERIODIC);

    1013:   /* Convenience variables */
    1014:   entriesPerFace   = stag->dof[0] + 2 * stag->dof[1] + stag->dof[2];
    1015:   entriesPerEdge   = stag->dof[0] + stag->dof[1];
    1016:   entriesPerCorner = stag->dof[0];
    1017:   for (d = 0; d < 3; ++d) nextToDummyEnd[d] = (PetscBool)(stag->boundaryType[d] != DM_BOUNDARY_PERIODIC && stag->rank[d] == stag->nRanks[d] - 2);
    1018:   eprGhost = stag->nGhost[0] * stag->entriesPerElement; /* epr = entries per (element) row   */
    1019:   eplGhost = stag->nGhost[1] * eprGhost;                /* epl = entries per (element) layer */

    1021:   /* Compute the number of local entries which correspond to any global entry */
    1022:   {
    1023:     PetscInt nNonDummyGhost[3];
    1024:     for (d = 0; d < 3; ++d) nNonDummyGhost[d] = stag->nGhost[d] - (dummyStart[d] ? ghostOffsetStart[d] : 0) - (dummyEnd[d] ? ghostOffsetEnd[d] : 0);
    1025:     if (star) {
    1026:       entriesToTransferTotal = (nNonDummyGhost[0] * stag->n[1] * stag->n[2] + stag->n[0] * nNonDummyGhost[1] * stag->n[2] + stag->n[0] * stag->n[1] * nNonDummyGhost[2] - 2 * (stag->n[0] * stag->n[1] * stag->n[2])) * stag->entriesPerElement +
    1027:                                (dummyEnd[0] ? (nNonDummyGhost[1] * stag->n[2] + stag->n[1] * nNonDummyGhost[2] - stag->n[1] * stag->n[2]) * entriesPerFace : 0) +
    1028:                                (dummyEnd[1] ? (nNonDummyGhost[0] * stag->n[2] + stag->n[0] * nNonDummyGhost[2] - stag->n[0] * stag->n[2]) * entriesPerFace : 0) +
    1029:                                (dummyEnd[2] ? (nNonDummyGhost[0] * stag->n[1] + stag->n[0] * nNonDummyGhost[1] - stag->n[0] * stag->n[1]) * entriesPerFace : 0) + (dummyEnd[0] && dummyEnd[1] ? nNonDummyGhost[2] * entriesPerEdge : 0) + (dummyEnd[2] && dummyEnd[0] ? nNonDummyGhost[1] * entriesPerEdge : 0) + (dummyEnd[1] && dummyEnd[2] ? nNonDummyGhost[0] * entriesPerEdge : 0) + (dummyEnd[0] && dummyEnd[1] && dummyEnd[2] ? entriesPerCorner : 0);
    1030:     } else {
    1031:       entriesToTransferTotal = nNonDummyGhost[0] * nNonDummyGhost[1] * nNonDummyGhost[2] * stag->entriesPerElement + (dummyEnd[0] ? nNonDummyGhost[1] * nNonDummyGhost[2] * entriesPerFace : 0) + (dummyEnd[1] ? nNonDummyGhost[2] * nNonDummyGhost[0] * entriesPerFace : 0) + (dummyEnd[2] ? nNonDummyGhost[0] * nNonDummyGhost[1] * entriesPerFace : 0) + (dummyEnd[0] && dummyEnd[1] ? nNonDummyGhost[2] * entriesPerEdge : 0) + (dummyEnd[2] && dummyEnd[0] ? nNonDummyGhost[1] * entriesPerEdge : 0) + (dummyEnd[1] && dummyEnd[2] ? nNonDummyGhost[0] * entriesPerEdge : 0) + (dummyEnd[0] && dummyEnd[1] && dummyEnd[2] ? entriesPerCorner : 0);
    1032:     }
    1033:   }

    1035:   /* Allocate arrays to populate */
    1036:   PetscCall(PetscMalloc1(entriesToTransferTotal, &idxLocal));
    1037:   PetscCall(PetscMalloc1(entriesToTransferTotal, &idxGlobal));

    1039:   /* Counts into idxLocal/idxGlobal */
    1040:   count = 0;

    1042:   /*  Loop over each of the 27 neighbor, populating idxLocal and idxGlobal. These
    1043:       cases are principally distinguished by

    1045:       1. The loop bounds (i/ighost,j/jghost,k/kghost)
    1046:       2. the strides used in the loop body, which depend on whether the current
    1047:       rank and/or its neighbor are a non-periodic right/top/front boundary, which has additional
    1048:       points in the global representation.
    1049:       - If the neighboring rank is a right/top/front boundary,
    1050:       then eprNeighbor (entries per element row on the neighbor) and/or eplNeighbor (entries per element layer on the neighbor)
    1051:       are different.
    1052:       - If this rank is a non-periodic right/top/front boundary (checking entries of stag->lastRank),
    1053:       there is an extra loop over 1-3 boundary faces)

    1055:       Here, we do not include "dummy" dof (points in the local representation which
    1056:       do not correspond to any global dof). This, and the fact that we iterate over points in terms of
    1057:       increasing global ordering, are the main two differences from the construction of
    1058:       the Local-to-global map, which iterates over points in local order, and does include dummy points. */

    1060:   /* LEFT DOWN BACK */
    1061:   if (!star && !dummyStart[0] && !dummyStart[1] && !dummyStart[2]) {
    1062:     const PetscInt  neighbor    = 0;
    1063:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0];
    1064:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1];
    1065:     const PetscInt  epFaceRow   = -1;
    1066:     const PetscInt  start0      = nNeighbors[neighbor][0] - ghostOffsetStart[0];
    1067:     const PetscInt  start1      = nNeighbors[neighbor][1] - ghostOffsetStart[1];
    1068:     const PetscInt  start2      = nNeighbors[neighbor][2] - ghostOffsetStart[2];
    1069:     const PetscInt  startGhost0 = 0;
    1070:     const PetscInt  startGhost1 = 0;
    1071:     const PetscInt  startGhost2 = 0;
    1072:     const PetscInt  endGhost0   = ghostOffsetStart[0];
    1073:     const PetscInt  endGhost1   = ghostOffsetStart[1];
    1074:     const PetscInt  endGhost2   = ghostOffsetStart[2];
    1075:     const PetscBool extra0      = PETSC_FALSE;
    1076:     const PetscBool extra1      = PETSC_FALSE;
    1077:     const PetscBool extra2      = PETSC_FALSE;
    1078:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1079:   }

    1081:   /* DOWN BACK */
    1082:   if (!star && !dummyStart[1] && !dummyStart[2]) {
    1083:     const PetscInt  neighbor    = 1;
    1084:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerFace : 0); /* We+neighbor may be a right boundary */
    1085:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1];
    1086:     const PetscInt  epFaceRow   = -1;
    1087:     const PetscInt  start0      = 0;
    1088:     const PetscInt  start1      = nNeighbors[neighbor][1] - ghostOffsetStart[1];
    1089:     const PetscInt  start2      = nNeighbors[neighbor][2] - ghostOffsetStart[2];
    1090:     const PetscInt  startGhost0 = ghostOffsetStart[0];
    1091:     const PetscInt  startGhost1 = 0;
    1092:     const PetscInt  startGhost2 = 0;
    1093:     const PetscInt  endGhost0   = stag->nGhost[0] - ghostOffsetEnd[0];
    1094:     const PetscInt  endGhost1   = ghostOffsetStart[1];
    1095:     const PetscInt  endGhost2   = ghostOffsetStart[2];
    1096:     const PetscBool extra0      = dummyEnd[0];
    1097:     const PetscBool extra1      = PETSC_FALSE;
    1098:     const PetscBool extra2      = PETSC_FALSE;
    1099:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1100:   }

    1102:   /* RIGHT DOWN BACK */
    1103:   if (!star && !dummyEnd[0] && !dummyStart[1] && !dummyStart[2]) {
    1104:     const PetscInt  neighbor    = 2;
    1105:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerFace : 0); /* Neighbor may be a right boundary */
    1106:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1];
    1107:     const PetscInt  epFaceRow   = -1;
    1108:     const PetscInt  start0      = 0;
    1109:     const PetscInt  start1      = nNeighbors[neighbor][1] - ghostOffsetStart[1];
    1110:     const PetscInt  start2      = nNeighbors[neighbor][2] - ghostOffsetStart[2];
    1111:     const PetscInt  startGhost0 = stag->nGhost[0] - ghostOffsetEnd[0];
    1112:     const PetscInt  startGhost1 = 0;
    1113:     const PetscInt  startGhost2 = 0;
    1114:     const PetscInt  endGhost0   = stag->nGhost[0];
    1115:     const PetscInt  endGhost1   = ghostOffsetStart[1];
    1116:     const PetscInt  endGhost2   = ghostOffsetStart[2];
    1117:     const PetscBool extra0      = PETSC_FALSE;
    1118:     const PetscBool extra1      = PETSC_FALSE;
    1119:     const PetscBool extra2      = PETSC_FALSE;
    1120:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1121:   }

    1123:   /* LEFT BACK */
    1124:   if (!star && !dummyStart[0] && !dummyStart[2]) {
    1125:     const PetscInt  neighbor    = 3;
    1126:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0];
    1127:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (dummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace : 0); /* May be a top boundary */
    1128:     const PetscInt  epFaceRow   = -1;
    1129:     const PetscInt  start0      = nNeighbors[neighbor][0] - ghostOffsetStart[0];
    1130:     const PetscInt  start1      = 0;
    1131:     const PetscInt  start2      = nNeighbors[neighbor][2] - ghostOffsetStart[2];
    1132:     const PetscInt  startGhost0 = 0;
    1133:     const PetscInt  startGhost1 = ghostOffsetStart[1];
    1134:     const PetscInt  startGhost2 = 0;
    1135:     const PetscInt  endGhost0   = ghostOffsetStart[0];
    1136:     const PetscInt  endGhost1   = stag->nGhost[1] - ghostOffsetEnd[1];
    1137:     const PetscInt  endGhost2   = ghostOffsetStart[2];
    1138:     const PetscBool extra0      = PETSC_FALSE;
    1139:     const PetscBool extra1      = dummyEnd[1];
    1140:     const PetscBool extra2      = PETSC_FALSE;
    1141:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1142:   }

    1144:   /* BACK */
    1145:   if (!dummyStart[2]) {
    1146:     const PetscInt  neighbor    = 4;
    1147:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                    /* We+neighbor may  be a right boundary */
    1148:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (dummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* We+neighbor may be a top boundary */
    1149:     const PetscInt  epFaceRow   = -1;
    1150:     const PetscInt  start0      = 0;
    1151:     const PetscInt  start1      = 0;
    1152:     const PetscInt  start2      = nNeighbors[neighbor][2] - ghostOffsetStart[2];
    1153:     const PetscInt  startGhost0 = ghostOffsetStart[0];
    1154:     const PetscInt  startGhost1 = ghostOffsetStart[1];
    1155:     const PetscInt  startGhost2 = 0;
    1156:     const PetscInt  endGhost0   = stag->nGhost[0] - ghostOffsetEnd[0];
    1157:     const PetscInt  endGhost1   = stag->nGhost[1] - ghostOffsetEnd[1];
    1158:     const PetscInt  endGhost2   = ghostOffsetStart[2];
    1159:     const PetscBool extra0      = dummyEnd[0];
    1160:     const PetscBool extra1      = dummyEnd[1];
    1161:     const PetscBool extra2      = PETSC_FALSE;
    1162:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1163:   }

    1165:   /* RIGHT BACK */
    1166:   if (!star && !dummyEnd[0] && !dummyStart[2]) {
    1167:     const PetscInt  neighbor    = 5;
    1168:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                    /* Neighbor may be a right boundary */
    1169:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (dummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* We and neighbor may be a top boundary */
    1170:     const PetscInt  epFaceRow   = -1;
    1171:     const PetscInt  start0      = 0;
    1172:     const PetscInt  start1      = 0;
    1173:     const PetscInt  start2      = nNeighbors[neighbor][2] - ghostOffsetStart[2];
    1174:     const PetscInt  startGhost0 = stag->nGhost[0] - ghostOffsetEnd[0];
    1175:     const PetscInt  startGhost1 = ghostOffsetStart[1];
    1176:     const PetscInt  startGhost2 = 0;
    1177:     const PetscInt  endGhost0   = stag->nGhost[0];
    1178:     const PetscInt  endGhost1   = stag->nGhost[1] - ghostOffsetEnd[1];
    1179:     const PetscInt  endGhost2   = ghostOffsetStart[2];
    1180:     const PetscBool extra0      = PETSC_FALSE;
    1181:     const PetscBool extra1      = dummyEnd[1];
    1182:     const PetscBool extra2      = PETSC_FALSE;
    1183:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1184:   }

    1186:   /* LEFT UP BACK */
    1187:   if (!star && !dummyStart[0] && !dummyEnd[1] && !dummyStart[2]) {
    1188:     const PetscInt  neighbor    = 6;
    1189:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0];
    1190:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (nextToDummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace : 0); /* Neighbor may be a top boundary */
    1191:     const PetscInt  epFaceRow   = -1;
    1192:     const PetscInt  start0      = nNeighbors[neighbor][0] - ghostOffsetStart[0];
    1193:     const PetscInt  start1      = 0;
    1194:     const PetscInt  start2      = nNeighbors[neighbor][2] - ghostOffsetStart[2];
    1195:     const PetscInt  startGhost0 = 0;
    1196:     const PetscInt  startGhost1 = stag->nGhost[1] - ghostOffsetEnd[1];
    1197:     const PetscInt  startGhost2 = 0;
    1198:     const PetscInt  endGhost0   = ghostOffsetStart[0];
    1199:     const PetscInt  endGhost1   = stag->nGhost[1];
    1200:     const PetscInt  endGhost2   = ghostOffsetStart[2];
    1201:     const PetscBool extra0      = PETSC_FALSE;
    1202:     const PetscBool extra1      = PETSC_FALSE;
    1203:     const PetscBool extra2      = PETSC_FALSE;
    1204:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1205:   }

    1207:   /* UP BACK */
    1208:   if (!star && !dummyEnd[1] && !dummyStart[2]) {
    1209:     const PetscInt  neighbor    = 7;
    1210:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                          /* We+neighbor may be a right boundary */
    1211:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (nextToDummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1212:     const PetscInt  epFaceRow   = -1;
    1213:     const PetscInt  start0      = 0;
    1214:     const PetscInt  start1      = 0;
    1215:     const PetscInt  start2      = nNeighbors[neighbor][2] - ghostOffsetStart[2];
    1216:     const PetscInt  startGhost0 = ghostOffsetStart[0];
    1217:     const PetscInt  startGhost1 = stag->nGhost[1] - ghostOffsetEnd[1];
    1218:     const PetscInt  startGhost2 = 0;
    1219:     const PetscInt  endGhost0   = stag->nGhost[0] - ghostOffsetEnd[0];
    1220:     const PetscInt  endGhost1   = stag->nGhost[1];
    1221:     const PetscInt  endGhost2   = ghostOffsetStart[2];
    1222:     const PetscBool extra0      = dummyEnd[0];
    1223:     const PetscBool extra1      = PETSC_FALSE;
    1224:     const PetscBool extra2      = PETSC_FALSE;
    1225:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1226:   }

    1228:   /* RIGHT UP BACK */
    1229:   if (!star && !dummyEnd[0] && !dummyEnd[1] && !dummyStart[2]) {
    1230:     const PetscInt  neighbor    = 8;
    1231:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                          /* Neighbor may be a right boundary */
    1232:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (nextToDummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1233:     const PetscInt  epFaceRow   = -1;
    1234:     const PetscInt  start0      = 0;
    1235:     const PetscInt  start1      = 0;
    1236:     const PetscInt  start2      = nNeighbors[neighbor][2] - ghostOffsetStart[2];
    1237:     const PetscInt  startGhost0 = stag->nGhost[0] - ghostOffsetEnd[0];
    1238:     const PetscInt  startGhost1 = stag->nGhost[1] - ghostOffsetEnd[1];
    1239:     const PetscInt  startGhost2 = 0;
    1240:     const PetscInt  endGhost0   = stag->nGhost[0];
    1241:     const PetscInt  endGhost1   = stag->nGhost[1];
    1242:     const PetscInt  endGhost2   = ghostOffsetStart[2];
    1243:     const PetscBool extra0      = PETSC_FALSE;
    1244:     const PetscBool extra1      = PETSC_FALSE;
    1245:     const PetscBool extra2      = PETSC_FALSE;
    1246:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1247:   }

    1249:   /* LEFT DOWN */
    1250:   if (!star && !dummyStart[0] && !dummyStart[1]) {
    1251:     const PetscInt  neighbor    = 9;
    1252:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0];
    1253:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1];
    1254:     const PetscInt  epFaceRow   = entriesPerFace * nNeighbors[neighbor][0]; /* Note that we can't be a right boundary */
    1255:     const PetscInt  start0      = nNeighbors[neighbor][0] - ghostOffsetStart[0];
    1256:     const PetscInt  start1      = nNeighbors[neighbor][1] - ghostOffsetStart[1];
    1257:     const PetscInt  start2      = 0;
    1258:     const PetscInt  startGhost0 = 0;
    1259:     const PetscInt  startGhost1 = 0;
    1260:     const PetscInt  startGhost2 = ghostOffsetStart[2];
    1261:     const PetscInt  endGhost0   = ghostOffsetStart[0];
    1262:     const PetscInt  endGhost1   = ghostOffsetStart[1];
    1263:     const PetscInt  endGhost2   = stag->nGhost[2] - ghostOffsetEnd[2];
    1264:     const PetscBool extra0      = PETSC_FALSE;
    1265:     const PetscBool extra1      = PETSC_FALSE;
    1266:     const PetscBool extra2      = dummyEnd[2];
    1267:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1268:   }

    1270:   /* DOWN */
    1271:   if (!dummyStart[1]) {
    1272:     const PetscInt  neighbor    = 10;
    1273:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerFace : 0); /* We+neighbor may be a right boundary */
    1274:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1];
    1275:     const PetscInt  epFaceRow   = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0); /* We+neighbor may be a right boundary */
    1276:     const PetscInt  start0      = 0;
    1277:     const PetscInt  start1      = nNeighbors[neighbor][1] - ghostOffsetStart[1];
    1278:     const PetscInt  start2      = 0;
    1279:     const PetscInt  startGhost0 = ghostOffsetStart[0];
    1280:     const PetscInt  startGhost1 = 0;
    1281:     const PetscInt  startGhost2 = ghostOffsetStart[2];
    1282:     const PetscInt  endGhost0   = stag->nGhost[0] - ghostOffsetEnd[0];
    1283:     const PetscInt  endGhost1   = ghostOffsetStart[1];
    1284:     const PetscInt  endGhost2   = stag->nGhost[2] - ghostOffsetEnd[2];
    1285:     const PetscBool extra0      = dummyEnd[0];
    1286:     const PetscBool extra1      = PETSC_FALSE;
    1287:     const PetscBool extra2      = dummyEnd[2];
    1288:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1289:   }

    1291:   /* RIGHT DOWN */
    1292:   if (!star && !dummyEnd[0] && !dummyStart[1]) {
    1293:     const PetscInt  neighbor    = 11;
    1294:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerFace : 0); /* Neighbor may be a right boundary */
    1295:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1];
    1296:     const PetscInt  epFaceRow   = entriesPerFace * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerEdge : 0); /* Neighbor may be a right boundary */
    1297:     const PetscInt  start0      = 0;
    1298:     const PetscInt  start1      = nNeighbors[neighbor][1] - ghostOffsetStart[1];
    1299:     const PetscInt  start2      = 0;
    1300:     const PetscInt  startGhost0 = stag->nGhost[0] - ghostOffsetEnd[0];
    1301:     const PetscInt  startGhost1 = 0;
    1302:     const PetscInt  startGhost2 = ghostOffsetStart[2];
    1303:     const PetscInt  endGhost0   = stag->nGhost[0];
    1304:     const PetscInt  endGhost1   = ghostOffsetStart[1];
    1305:     const PetscInt  endGhost2   = stag->nGhost[2] - ghostOffsetEnd[2];
    1306:     const PetscBool extra0      = PETSC_FALSE;
    1307:     const PetscBool extra1      = PETSC_FALSE;
    1308:     const PetscBool extra2      = dummyEnd[2];
    1309:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1310:   }

    1312:   /* LEFT */
    1313:   if (!dummyStart[0]) {
    1314:     const PetscInt  neighbor    = 12;
    1315:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0];
    1316:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (dummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace : 0); /* We+neighbor may be a top boundary */
    1317:     const PetscInt  epFaceRow   = entriesPerFace * nNeighbors[neighbor][0];
    1318:     const PetscInt  start0      = nNeighbors[neighbor][0] - ghostOffsetStart[0];
    1319:     const PetscInt  start1      = 0;
    1320:     const PetscInt  start2      = 0;
    1321:     const PetscInt  startGhost0 = 0;
    1322:     const PetscInt  startGhost1 = ghostOffsetStart[1];
    1323:     const PetscInt  startGhost2 = ghostOffsetStart[2];
    1324:     const PetscInt  endGhost0   = ghostOffsetStart[0];
    1325:     const PetscInt  endGhost1   = stag->nGhost[1] - ghostOffsetEnd[1];
    1326:     const PetscInt  endGhost2   = stag->nGhost[2] - ghostOffsetEnd[2];
    1327:     const PetscBool extra0      = PETSC_FALSE;
    1328:     const PetscBool extra1      = dummyEnd[1];
    1329:     const PetscBool extra2      = dummyEnd[2];
    1330:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1331:   }

    1333:   /* (HERE) */
    1334:   {
    1335:     const PetscInt  neighbor    = 13;
    1336:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                    /* We may be a right boundary */
    1337:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (dummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* We may be a top boundary */
    1338:     const PetscInt  epFaceRow   = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0);                                                             /* We may be a right boundary */
    1339:     const PetscInt  start0      = 0;
    1340:     const PetscInt  start1      = 0;
    1341:     const PetscInt  start2      = 0;
    1342:     const PetscInt  startGhost0 = ghostOffsetStart[0];
    1343:     const PetscInt  startGhost1 = ghostOffsetStart[1];
    1344:     const PetscInt  startGhost2 = ghostOffsetStart[2];
    1345:     const PetscInt  endGhost0   = stag->nGhost[0] - ghostOffsetEnd[0];
    1346:     const PetscInt  endGhost1   = stag->nGhost[1] - ghostOffsetEnd[1];
    1347:     const PetscInt  endGhost2   = stag->nGhost[2] - ghostOffsetEnd[2];
    1348:     const PetscBool extra0      = dummyEnd[0];
    1349:     const PetscBool extra1      = dummyEnd[1];
    1350:     const PetscBool extra2      = dummyEnd[2];
    1351:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1352:   }

    1354:   /* RIGHT */
    1355:   if (!dummyEnd[0]) {
    1356:     const PetscInt  neighbor    = 14;
    1357:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                    /* Neighbor may be a right boundary */
    1358:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (dummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* We and neighbor may be a top boundary */
    1359:     const PetscInt  epFaceRow   = entriesPerFace * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerEdge : 0);                                                             /* Neighbor may be a right boundary */
    1360:     const PetscInt  start0      = 0;
    1361:     const PetscInt  start1      = 0;
    1362:     const PetscInt  start2      = 0;
    1363:     const PetscInt  startGhost0 = stag->nGhost[0] - ghostOffsetEnd[0];
    1364:     const PetscInt  startGhost1 = ghostOffsetStart[1];
    1365:     const PetscInt  startGhost2 = ghostOffsetStart[2];
    1366:     const PetscInt  endGhost0   = stag->nGhost[0];
    1367:     const PetscInt  endGhost1   = stag->nGhost[1] - ghostOffsetEnd[1];
    1368:     const PetscInt  endGhost2   = stag->nGhost[2] - ghostOffsetEnd[2];
    1369:     const PetscBool extra0      = PETSC_FALSE;
    1370:     const PetscBool extra1      = dummyEnd[1];
    1371:     const PetscBool extra2      = dummyEnd[2];
    1372:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1373:   }

    1375:   /* LEFT UP */
    1376:   if (!star && !dummyStart[0] && !dummyEnd[1]) {
    1377:     const PetscInt  neighbor    = 15;
    1378:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0];
    1379:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (nextToDummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace : 0); /* Neighbor may be a top boundary */
    1380:     const PetscInt  epFaceRow   = entriesPerFace * nNeighbors[neighbor][0];
    1381:     const PetscInt  start0      = nNeighbors[neighbor][0] - ghostOffsetStart[0];
    1382:     const PetscInt  start1      = 0;
    1383:     const PetscInt  start2      = 0;
    1384:     const PetscInt  startGhost0 = 0;
    1385:     const PetscInt  startGhost1 = stag->nGhost[1] - ghostOffsetEnd[1];
    1386:     const PetscInt  startGhost2 = ghostOffsetStart[2];
    1387:     const PetscInt  endGhost0   = ghostOffsetStart[0];
    1388:     const PetscInt  endGhost1   = stag->nGhost[1];
    1389:     const PetscInt  endGhost2   = stag->nGhost[2] - ghostOffsetEnd[2];
    1390:     const PetscBool extra0      = PETSC_FALSE;
    1391:     const PetscBool extra1      = PETSC_FALSE;
    1392:     const PetscBool extra2      = dummyEnd[2];
    1393:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1394:   }

    1396:   /* UP */
    1397:   if (!dummyEnd[1]) {
    1398:     const PetscInt  neighbor    = 16;
    1399:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                          /* We+neighbor may be a right boundary */
    1400:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (nextToDummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1401:     const PetscInt  epFaceRow   = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0);                                                                   /* We+neighbor may be a right boundary */
    1402:     const PetscInt  start0      = 0;
    1403:     const PetscInt  start1      = 0;
    1404:     const PetscInt  start2      = 0;
    1405:     const PetscInt  startGhost0 = ghostOffsetStart[0];
    1406:     const PetscInt  startGhost1 = stag->nGhost[1] - ghostOffsetEnd[1];
    1407:     const PetscInt  startGhost2 = ghostOffsetStart[2];
    1408:     const PetscInt  endGhost0   = stag->nGhost[0] - ghostOffsetEnd[0];
    1409:     const PetscInt  endGhost1   = stag->nGhost[1];
    1410:     const PetscInt  endGhost2   = stag->nGhost[2] - ghostOffsetEnd[2];
    1411:     const PetscBool extra0      = dummyEnd[0];
    1412:     const PetscBool extra1      = PETSC_FALSE;
    1413:     const PetscBool extra2      = dummyEnd[2];
    1414:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1415:   }

    1417:   /* RIGHT UP */
    1418:   if (!star && !dummyEnd[0] && !dummyEnd[1]) {
    1419:     const PetscInt  neighbor    = 17;
    1420:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                          /* Neighbor may be a right boundary */
    1421:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (nextToDummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1422:     const PetscInt  epFaceRow   = entriesPerFace * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerEdge : 0);                                                                   /* Neighbor may be a right boundary */
    1423:     const PetscInt  start0      = 0;
    1424:     const PetscInt  start1      = 0;
    1425:     const PetscInt  start2      = 0;
    1426:     const PetscInt  startGhost0 = stag->nGhost[0] - ghostOffsetEnd[0];
    1427:     const PetscInt  startGhost1 = stag->nGhost[1] - ghostOffsetEnd[1];
    1428:     const PetscInt  startGhost2 = ghostOffsetStart[2];
    1429:     const PetscInt  endGhost0   = stag->nGhost[0];
    1430:     const PetscInt  endGhost1   = stag->nGhost[1];
    1431:     const PetscInt  endGhost2   = stag->nGhost[2] - ghostOffsetEnd[2];
    1432:     const PetscBool extra0      = PETSC_FALSE;
    1433:     const PetscBool extra1      = PETSC_FALSE;
    1434:     const PetscBool extra2      = dummyEnd[2];
    1435:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1436:   }

    1438:   /* LEFT DOWN FRONT */
    1439:   if (!star && !dummyStart[0] && !dummyStart[1] && !dummyEnd[2]) {
    1440:     const PetscInt  neighbor    = 18;
    1441:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0];
    1442:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1];
    1443:     const PetscInt  epFaceRow   = -1;
    1444:     const PetscInt  start0      = nNeighbors[neighbor][0] - ghostOffsetStart[0];
    1445:     const PetscInt  start1      = nNeighbors[neighbor][1] - ghostOffsetStart[1];
    1446:     const PetscInt  start2      = 0;
    1447:     const PetscInt  startGhost0 = 0;
    1448:     const PetscInt  startGhost1 = 0;
    1449:     const PetscInt  startGhost2 = stag->nGhost[2] - ghostOffsetEnd[2];
    1450:     const PetscInt  endGhost0   = ghostOffsetStart[0];
    1451:     const PetscInt  endGhost1   = ghostOffsetStart[1];
    1452:     const PetscInt  endGhost2   = stag->nGhost[2];
    1453:     const PetscBool extra0      = PETSC_FALSE;
    1454:     const PetscBool extra1      = PETSC_FALSE;
    1455:     const PetscBool extra2      = PETSC_FALSE;
    1456:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1457:   }

    1459:   /* DOWN FRONT */
    1460:   if (!star && !dummyStart[1] && !dummyEnd[2]) {
    1461:     const PetscInt  neighbor    = 19;
    1462:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerFace : 0); /* We+neighbor may be a right boundary */
    1463:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1];
    1464:     const PetscInt  epFaceRow   = -1;
    1465:     const PetscInt  start0      = 0;
    1466:     const PetscInt  start1      = nNeighbors[neighbor][1] - ghostOffsetStart[1];
    1467:     const PetscInt  start2      = 0;
    1468:     const PetscInt  startGhost0 = ghostOffsetStart[0];
    1469:     const PetscInt  startGhost1 = 0;
    1470:     const PetscInt  startGhost2 = stag->nGhost[2] - ghostOffsetEnd[2];
    1471:     const PetscInt  endGhost0   = stag->nGhost[0] - ghostOffsetEnd[0];
    1472:     const PetscInt  endGhost1   = ghostOffsetStart[1];
    1473:     const PetscInt  endGhost2   = stag->nGhost[2];
    1474:     const PetscBool extra0      = dummyEnd[0];
    1475:     const PetscBool extra1      = PETSC_FALSE;
    1476:     const PetscBool extra2      = PETSC_FALSE;
    1477:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1478:   }

    1480:   /* RIGHT DOWN FRONT */
    1481:   if (!star && !dummyEnd[0] && !dummyStart[1] && !dummyEnd[2]) {
    1482:     const PetscInt  neighbor    = 20;
    1483:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerFace : 0); /* Neighbor may be a right boundary */
    1484:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1];
    1485:     const PetscInt  epFaceRow   = -1;
    1486:     const PetscInt  start0      = 0;
    1487:     const PetscInt  start1      = nNeighbors[neighbor][1] - ghostOffsetStart[1];
    1488:     const PetscInt  start2      = 0;
    1489:     const PetscInt  startGhost0 = stag->nGhost[0] - ghostOffsetEnd[0];
    1490:     const PetscInt  startGhost1 = 0;
    1491:     const PetscInt  startGhost2 = stag->nGhost[2] - ghostOffsetEnd[2];
    1492:     const PetscInt  endGhost0   = stag->nGhost[0];
    1493:     const PetscInt  endGhost1   = ghostOffsetStart[1];
    1494:     const PetscInt  endGhost2   = stag->nGhost[2];
    1495:     const PetscBool extra0      = PETSC_FALSE;
    1496:     const PetscBool extra1      = PETSC_FALSE;
    1497:     const PetscBool extra2      = PETSC_FALSE;
    1498:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1499:   }

    1501:   /* LEFT FRONT */
    1502:   if (!star && !dummyStart[0] && !dummyEnd[2]) {
    1503:     const PetscInt  neighbor    = 21;
    1504:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0];
    1505:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (dummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace : 0); /* We+neighbor may be a top boundary */
    1506:     const PetscInt  epFaceRow   = -1;
    1507:     const PetscInt  start0      = nNeighbors[neighbor][0] - ghostOffsetStart[0];
    1508:     const PetscInt  start1      = 0;
    1509:     const PetscInt  start2      = 0;
    1510:     const PetscInt  startGhost0 = 0;
    1511:     const PetscInt  startGhost1 = ghostOffsetStart[1];
    1512:     const PetscInt  startGhost2 = stag->nGhost[2] - ghostOffsetEnd[2];
    1513:     const PetscInt  endGhost0   = ghostOffsetStart[0];
    1514:     const PetscInt  endGhost1   = stag->nGhost[1] - ghostOffsetEnd[1];
    1515:     const PetscInt  endGhost2   = stag->nGhost[2];
    1516:     const PetscBool extra0      = PETSC_FALSE;
    1517:     const PetscBool extra1      = dummyEnd[1];
    1518:     const PetscBool extra2      = PETSC_FALSE;
    1519:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1520:   }

    1522:   /* FRONT */
    1523:   if (!dummyEnd[2]) {
    1524:     const PetscInt  neighbor    = 22;
    1525:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                    /* neighbor is a right boundary if we are*/
    1526:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (dummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* May be a top boundary */
    1527:     const PetscInt  epFaceRow   = -1;
    1528:     const PetscInt  start0      = 0;
    1529:     const PetscInt  start1      = 0;
    1530:     const PetscInt  start2      = 0;
    1531:     const PetscInt  startGhost0 = ghostOffsetStart[0];
    1532:     const PetscInt  startGhost1 = ghostOffsetStart[1];
    1533:     const PetscInt  startGhost2 = stag->nGhost[2] - ghostOffsetEnd[2];
    1534:     const PetscInt  endGhost0   = stag->nGhost[0] - ghostOffsetEnd[0];
    1535:     const PetscInt  endGhost1   = stag->nGhost[1] - ghostOffsetEnd[1];
    1536:     const PetscInt  endGhost2   = stag->nGhost[2];
    1537:     const PetscBool extra0      = dummyEnd[0];
    1538:     const PetscBool extra1      = dummyEnd[1];
    1539:     const PetscBool extra2      = PETSC_FALSE;
    1540:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1541:   }

    1543:   /* RIGHT FRONT */
    1544:   if (!star && !dummyEnd[0] && !dummyEnd[2]) {
    1545:     const PetscInt  neighbor    = 23;
    1546:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                    /* Neighbor may be a right boundary */
    1547:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (dummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* We and neighbor may be a top boundary */
    1548:     const PetscInt  epFaceRow   = -1;
    1549:     const PetscInt  start0      = 0;
    1550:     const PetscInt  start1      = 0;
    1551:     const PetscInt  start2      = 0;
    1552:     const PetscInt  startGhost0 = stag->nGhost[0] - ghostOffsetEnd[0];
    1553:     const PetscInt  startGhost1 = ghostOffsetStart[1];
    1554:     const PetscInt  startGhost2 = stag->nGhost[2] - ghostOffsetEnd[2];
    1555:     const PetscInt  endGhost0   = stag->nGhost[0];
    1556:     const PetscInt  endGhost1   = stag->nGhost[1] - ghostOffsetEnd[1];
    1557:     const PetscInt  endGhost2   = stag->nGhost[2];
    1558:     const PetscBool extra0      = PETSC_FALSE;
    1559:     const PetscBool extra1      = dummyEnd[1];
    1560:     const PetscBool extra2      = PETSC_FALSE;
    1561:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1562:   }

    1564:   /* LEFT UP FRONT */
    1565:   if (!star && !dummyStart[0] && !dummyEnd[1] && !dummyEnd[2]) {
    1566:     const PetscInt  neighbor    = 24;
    1567:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0];
    1568:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (nextToDummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace : 0); /* Neighbor may be a top boundary */
    1569:     const PetscInt  epFaceRow   = -1;
    1570:     const PetscInt  start0      = nNeighbors[neighbor][0] - ghostOffsetStart[0];
    1571:     const PetscInt  start1      = 0;
    1572:     const PetscInt  start2      = 0;
    1573:     const PetscInt  startGhost0 = 0;
    1574:     const PetscInt  startGhost1 = stag->nGhost[1] - ghostOffsetEnd[1];
    1575:     const PetscInt  startGhost2 = stag->nGhost[2] - ghostOffsetEnd[2];
    1576:     const PetscInt  endGhost0   = ghostOffsetStart[0];
    1577:     const PetscInt  endGhost1   = stag->nGhost[1];
    1578:     const PetscInt  endGhost2   = stag->nGhost[2];
    1579:     const PetscBool extra0      = PETSC_FALSE;
    1580:     const PetscBool extra1      = PETSC_FALSE;
    1581:     const PetscBool extra2      = PETSC_FALSE;
    1582:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1583:   }

    1585:   /* UP FRONT */
    1586:   if (!star && !dummyEnd[1] && !dummyEnd[2]) {
    1587:     const PetscInt  neighbor    = 25;
    1588:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                          /* We+neighbor may be a right boundary */
    1589:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (nextToDummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1590:     const PetscInt  epFaceRow   = -1;
    1591:     const PetscInt  start0      = 0;
    1592:     const PetscInt  start1      = 0;
    1593:     const PetscInt  start2      = 0;
    1594:     const PetscInt  startGhost0 = ghostOffsetStart[0];
    1595:     const PetscInt  startGhost1 = stag->nGhost[1] - ghostOffsetEnd[1];
    1596:     const PetscInt  startGhost2 = stag->nGhost[2] - ghostOffsetEnd[2];
    1597:     const PetscInt  endGhost0   = stag->nGhost[0] - ghostOffsetEnd[0];
    1598:     const PetscInt  endGhost1   = stag->nGhost[1];
    1599:     const PetscInt  endGhost2   = stag->nGhost[2];
    1600:     const PetscBool extra0      = dummyEnd[0];
    1601:     const PetscBool extra1      = PETSC_FALSE;
    1602:     const PetscBool extra2      = PETSC_FALSE;
    1603:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1604:   }

    1606:   /* RIGHT UP FRONT */
    1607:   if (!star && !dummyEnd[0] && !dummyEnd[1] && !dummyEnd[2]) {
    1608:     const PetscInt  neighbor    = 26;
    1609:     const PetscInt  eprNeighbor = stag->entriesPerElement * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                          /* Neighbor may be a right boundary */
    1610:     const PetscInt  eplNeighbor = eprNeighbor * nNeighbors[neighbor][1] + (nextToDummyEnd[1] ? nNeighbors[neighbor][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1611:     const PetscInt  epFaceRow   = -1;
    1612:     const PetscInt  start0      = 0;
    1613:     const PetscInt  start1      = 0;
    1614:     const PetscInt  start2      = 0;
    1615:     const PetscInt  startGhost0 = stag->nGhost[0] - ghostOffsetEnd[0];
    1616:     const PetscInt  startGhost1 = stag->nGhost[1] - ghostOffsetEnd[1];
    1617:     const PetscInt  startGhost2 = stag->nGhost[2] - ghostOffsetEnd[2];
    1618:     const PetscInt  endGhost0   = stag->nGhost[0];
    1619:     const PetscInt  endGhost1   = stag->nGhost[1];
    1620:     const PetscInt  endGhost2   = stag->nGhost[2];
    1621:     const PetscBool extra0      = PETSC_FALSE;
    1622:     const PetscBool extra1      = PETSC_FALSE;
    1623:     const PetscBool extra2      = PETSC_FALSE;
    1624:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, eprNeighbor, eplNeighbor, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    1625:   }

    1627:   PetscCheck(count == entriesToTransferTotal, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of entries computed in gtol (%" PetscInt_FMT ") is not as expected (%" PetscInt_FMT ")", count, entriesToTransferTotal);

    1629:   /* Create stag->gtol. The order is computed as PETSc ordering, and doesn't include dummy entries */
    1630:   {
    1631:     Vec local, global;
    1632:     IS  isLocal, isGlobal;
    1633:     PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)dm), entriesToTransferTotal, idxLocal, PETSC_OWN_POINTER, &isLocal));
    1634:     PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)dm), entriesToTransferTotal, idxGlobal, PETSC_OWN_POINTER, &isGlobal));
    1635:     PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)dm), 1, stag->entries, PETSC_DECIDE, NULL, &global));
    1636:     PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, PetscMax(stag->entriesPerElement, 1), stag->entriesGhost, NULL, &local));
    1637:     PetscCall(VecScatterCreate(global, isGlobal, local, isLocal, &stag->gtol));
    1638:     PetscCall(VecDestroy(&global));
    1639:     PetscCall(VecDestroy(&local));
    1640:     PetscCall(ISDestroy(&isLocal));  /* frees idxLocal */
    1641:     PetscCall(ISDestroy(&isGlobal)); /* free idxGlobal */
    1642:   }
    1643:   PetscFunctionReturn(PETSC_SUCCESS);
    1644: }

    1646: /* Note: this function assumes that DMBoundary types of none, ghosted, and periodic are the only ones of interest.
    1647: Adding support for others should be done very carefully.  */
    1648: static PetscErrorCode DMStagSetUpBuildL2G_3d(DM dm, const PetscInt *globalOffsets)
    1649: {
    1650:   const DM_Stag *const stag = (DM_Stag *)dm->data;
    1651:   PetscInt            *idxGlobalAll;
    1652:   PetscInt             d, count, ighost, jghost, kghost, ghostOffsetStart[3], ghostOffsetEnd[3], entriesPerFace, entriesPerEdge;
    1653:   PetscInt             nNeighbors[27][3], eprNeighbor[27], eplNeighbor[27], globalOffsetNeighbor[27];
    1654:   PetscBool            nextToDummyEnd[3], dummyStart[3], dummyEnd[3], star;

    1656:   PetscFunctionBegin;
    1657:   /* Check stencil type */
    1658:   PetscCheck(stag->stencilType == DMSTAG_STENCIL_NONE || stag->stencilType == DMSTAG_STENCIL_BOX || stag->stencilType == DMSTAG_STENCIL_STAR, PetscObjectComm((PetscObject)dm), PETSC_ERR_SUP, "Unsupported stencil type %s", DMStagStencilTypes[stag->stencilType]);
    1659:   star = (PetscBool)(stag->stencilType == DMSTAG_STENCIL_STAR || stag->stencilType == DMSTAG_STENCIL_NONE);

    1661:   /* Convenience variables */
    1662:   entriesPerFace = stag->dof[0] + 2 * stag->dof[1] + stag->dof[2];
    1663:   entriesPerEdge = stag->dof[0] + stag->dof[1];
    1664:   for (d = 0; d < 3; ++d) nextToDummyEnd[d] = (PetscBool)(stag->boundaryType[d] != DM_BOUNDARY_PERIODIC && stag->rank[d] == stag->nRanks[d] - 2);

    1666:   /* Ghost offsets (may not be the ghost width, since we always have at least one ghost element on the right/top/front) */
    1667:   for (d = 0; d < 3; ++d) ghostOffsetStart[d] = stag->start[d] - stag->startGhost[d];
    1668:   for (d = 0; d < 3; ++d) ghostOffsetEnd[d] = stag->startGhost[d] + stag->nGhost[d] - (stag->start[d] + stag->n[d]);

    1670:   /* Whether the ghost region includes dummies. Currently equivalent to being a non-periodic boundary. */
    1671:   for (d = 0; d < 3; ++d) dummyStart[d] = (PetscBool)(stag->firstRank[d] && stag->boundaryType[d] != DM_BOUNDARY_PERIODIC);
    1672:   for (d = 0; d < 3; ++d) dummyEnd[d] = (PetscBool)(stag->lastRank[d] && stag->boundaryType[d] != DM_BOUNDARY_PERIODIC);

    1674:   /* Compute numbers of elements on each neighbor  and offset*/
    1675:   {
    1676:     PetscInt i;
    1677:     for (i = 0; i < 27; ++i) {
    1678:       const PetscInt neighborRank = stag->neighbors[i];
    1679:       if (neighborRank >= 0) { /* note we copy the values for our own rank (neighbor 13) */
    1680:         nNeighbors[i][0]        = stag->l[0][neighborRank % stag->nRanks[0]];
    1681:         nNeighbors[i][1]        = stag->l[1][neighborRank % (stag->nRanks[0] * stag->nRanks[1]) / stag->nRanks[0]];
    1682:         nNeighbors[i][2]        = stag->l[2][neighborRank / (stag->nRanks[0] * stag->nRanks[1])];
    1683:         globalOffsetNeighbor[i] = globalOffsets[stag->neighbors[i]];
    1684:       } /* else leave uninitialized - error if accessed */
    1685:     }
    1686:   }

    1688:   /* Precompute elements per row and layer on neighbor (zero unused) */
    1689:   PetscCall(PetscMemzero(eprNeighbor, sizeof(eprNeighbor)));
    1690:   PetscCall(PetscMemzero(eplNeighbor, sizeof(eplNeighbor)));
    1691:   if (stag->neighbors[0] >= 0) {
    1692:     eprNeighbor[0] = stag->entriesPerElement * nNeighbors[0][0];
    1693:     eplNeighbor[0] = eprNeighbor[0] * nNeighbors[0][1];
    1694:   }
    1695:   if (stag->neighbors[1] >= 0) {
    1696:     eprNeighbor[1] = stag->entriesPerElement * nNeighbors[1][0] + (dummyEnd[0] ? entriesPerFace : 0); /* We+neighbor may be a right boundary */
    1697:     eplNeighbor[1] = eprNeighbor[1] * nNeighbors[1][1];
    1698:   }
    1699:   if (stag->neighbors[2] >= 0) {
    1700:     eprNeighbor[2] = stag->entriesPerElement * nNeighbors[2][0] + (nextToDummyEnd[0] ? entriesPerFace : 0); /* Neighbor may be a right boundary */
    1701:     eplNeighbor[2] = eprNeighbor[2] * nNeighbors[2][1];
    1702:   }
    1703:   if (stag->neighbors[3] >= 0) {
    1704:     eprNeighbor[3] = stag->entriesPerElement * nNeighbors[3][0];
    1705:     eplNeighbor[3] = eprNeighbor[3] * nNeighbors[3][1] + (dummyEnd[1] ? nNeighbors[3][0] * entriesPerFace : 0); /* May be a top boundary */
    1706:   }
    1707:   if (stag->neighbors[4] >= 0) {
    1708:     eprNeighbor[4] = stag->entriesPerElement * nNeighbors[4][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                /* We+neighbor may  be a right boundary */
    1709:     eplNeighbor[4] = eprNeighbor[4] * nNeighbors[4][1] + (dummyEnd[1] ? nNeighbors[4][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* We+neighbor may be a top boundary */
    1710:   }
    1711:   if (stag->neighbors[5] >= 0) {
    1712:     eprNeighbor[5] = stag->entriesPerElement * nNeighbors[5][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                /* Neighbor may be a right boundary */
    1713:     eplNeighbor[5] = eprNeighbor[5] * nNeighbors[5][1] + (dummyEnd[1] ? nNeighbors[5][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* We and neighbor may be a top boundary */
    1714:   }
    1715:   if (stag->neighbors[6] >= 0) {
    1716:     eprNeighbor[6] = stag->entriesPerElement * nNeighbors[6][0];
    1717:     eplNeighbor[6] = eprNeighbor[6] * nNeighbors[6][1] + (nextToDummyEnd[1] ? nNeighbors[6][0] * entriesPerFace : 0); /* Neighbor may be a top boundary */
    1718:   }
    1719:   if (stag->neighbors[7] >= 0) {
    1720:     eprNeighbor[7] = stag->entriesPerElement * nNeighbors[7][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                      /* We+neighbor may be a right boundary */
    1721:     eplNeighbor[7] = eprNeighbor[7] * nNeighbors[7][1] + (nextToDummyEnd[1] ? nNeighbors[7][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1722:   }
    1723:   if (stag->neighbors[8] >= 0) {
    1724:     eprNeighbor[8] = stag->entriesPerElement * nNeighbors[8][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                      /* Neighbor may be a right boundary */
    1725:     eplNeighbor[8] = eprNeighbor[8] * nNeighbors[8][1] + (nextToDummyEnd[1] ? nNeighbors[8][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1726:   }
    1727:   if (stag->neighbors[9] >= 0) {
    1728:     eprNeighbor[9] = stag->entriesPerElement * nNeighbors[9][0];
    1729:     eplNeighbor[9] = eprNeighbor[9] * nNeighbors[9][1];
    1730:   }
    1731:   if (stag->neighbors[10] >= 0) {
    1732:     eprNeighbor[10] = stag->entriesPerElement * nNeighbors[10][0] + (dummyEnd[0] ? entriesPerFace : 0); /* We+neighbor may be a right boundary */
    1733:     eplNeighbor[10] = eprNeighbor[10] * nNeighbors[10][1];
    1734:   }
    1735:   if (stag->neighbors[11] >= 0) {
    1736:     eprNeighbor[11] = stag->entriesPerElement * nNeighbors[11][0] + (nextToDummyEnd[0] ? entriesPerFace : 0); /* Neighbor may be a right boundary */
    1737:     eplNeighbor[11] = eprNeighbor[11] * nNeighbors[11][1];
    1738:   }
    1739:   if (stag->neighbors[12] >= 0) {
    1740:     eprNeighbor[12] = stag->entriesPerElement * nNeighbors[12][0];
    1741:     eplNeighbor[12] = eprNeighbor[12] * nNeighbors[12][1] + (dummyEnd[1] ? nNeighbors[12][0] * entriesPerFace : 0); /* We+neighbor may be a top boundary */
    1742:   }
    1743:   if (stag->neighbors[13] >= 0) {
    1744:     eprNeighbor[13] = stag->entriesPerElement * nNeighbors[13][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                  /* We may be a right boundary */
    1745:     eplNeighbor[13] = eprNeighbor[13] * nNeighbors[13][1] + (dummyEnd[1] ? nNeighbors[13][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* We may be a top boundary */
    1746:   }
    1747:   if (stag->neighbors[14] >= 0) {
    1748:     eprNeighbor[14] = stag->entriesPerElement * nNeighbors[14][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                  /* Neighbor may be a right boundary */
    1749:     eplNeighbor[14] = eprNeighbor[14] * nNeighbors[14][1] + (dummyEnd[1] ? nNeighbors[14][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* We and neighbor may be a top boundary */
    1750:   }
    1751:   if (stag->neighbors[15] >= 0) {
    1752:     eprNeighbor[15] = stag->entriesPerElement * nNeighbors[15][0];
    1753:     eplNeighbor[15] = eprNeighbor[15] * nNeighbors[15][1] + (nextToDummyEnd[1] ? nNeighbors[15][0] * entriesPerFace : 0); /* Neighbor may be a top boundary */
    1754:   }
    1755:   if (stag->neighbors[16] >= 0) {
    1756:     eprNeighbor[16] = stag->entriesPerElement * nNeighbors[16][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                        /* We+neighbor may be a right boundary */
    1757:     eplNeighbor[16] = eprNeighbor[16] * nNeighbors[16][1] + (nextToDummyEnd[1] ? nNeighbors[16][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1758:   }
    1759:   if (stag->neighbors[17] >= 0) {
    1760:     eprNeighbor[17] = stag->entriesPerElement * nNeighbors[17][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                        /* Neighbor may be a right boundary */
    1761:     eplNeighbor[17] = eprNeighbor[17] * nNeighbors[17][1] + (nextToDummyEnd[1] ? nNeighbors[17][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1762:   }
    1763:   if (stag->neighbors[18] >= 0) {
    1764:     eprNeighbor[18] = stag->entriesPerElement * nNeighbors[18][0];
    1765:     eplNeighbor[18] = eprNeighbor[18] * nNeighbors[18][1];
    1766:   }
    1767:   if (stag->neighbors[19] >= 0) {
    1768:     eprNeighbor[19] = stag->entriesPerElement * nNeighbors[19][0] + (dummyEnd[0] ? entriesPerFace : 0); /* We+neighbor may be a right boundary */
    1769:     eplNeighbor[19] = eprNeighbor[19] * nNeighbors[19][1];
    1770:   }
    1771:   if (stag->neighbors[20] >= 0) {
    1772:     eprNeighbor[20] = stag->entriesPerElement * nNeighbors[20][0] + (nextToDummyEnd[0] ? entriesPerFace : 0); /* Neighbor may be a right boundary */
    1773:     eplNeighbor[20] = eprNeighbor[20] * nNeighbors[20][1];
    1774:   }
    1775:   if (stag->neighbors[21] >= 0) {
    1776:     eprNeighbor[21] = stag->entriesPerElement * nNeighbors[21][0];
    1777:     eplNeighbor[21] = eprNeighbor[21] * nNeighbors[21][1] + (dummyEnd[1] ? nNeighbors[21][0] * entriesPerFace : 0); /* We+neighbor may be a top boundary */
    1778:   }
    1779:   if (stag->neighbors[22] >= 0) {
    1780:     eprNeighbor[22] = stag->entriesPerElement * nNeighbors[22][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                  /* neighbor is a right boundary if we are*/
    1781:     eplNeighbor[22] = eprNeighbor[22] * nNeighbors[22][1] + (dummyEnd[1] ? nNeighbors[22][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* May be a top boundary */
    1782:   }
    1783:   if (stag->neighbors[23] >= 0) {
    1784:     eprNeighbor[23] = stag->entriesPerElement * nNeighbors[23][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                  /* Neighbor may be a right boundary */
    1785:     eplNeighbor[23] = eprNeighbor[23] * nNeighbors[23][1] + (dummyEnd[1] ? nNeighbors[23][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* We and neighbor may be a top boundary */
    1786:   }
    1787:   if (stag->neighbors[24] >= 0) {
    1788:     eprNeighbor[24] = stag->entriesPerElement * nNeighbors[24][0];
    1789:     eplNeighbor[24] = eprNeighbor[24] * nNeighbors[24][1] + (nextToDummyEnd[1] ? nNeighbors[24][0] * entriesPerFace : 0); /* Neighbor may be a top boundary */
    1790:   }
    1791:   if (stag->neighbors[25] >= 0) {
    1792:     eprNeighbor[25] = stag->entriesPerElement * nNeighbors[25][0] + (dummyEnd[0] ? entriesPerFace : 0);                                                        /* We+neighbor may be a right boundary */
    1793:     eplNeighbor[25] = eprNeighbor[25] * nNeighbors[25][1] + (nextToDummyEnd[1] ? nNeighbors[25][0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1794:   }
    1795:   if (stag->neighbors[26] >= 0) {
    1796:     eprNeighbor[26] = stag->entriesPerElement * nNeighbors[26][0] + (nextToDummyEnd[0] ? entriesPerFace : 0);                                                        /* Neighbor may be a right boundary */
    1797:     eplNeighbor[26] = eprNeighbor[26] * nNeighbors[26][1] + (nextToDummyEnd[1] ? nNeighbors[26][0] * entriesPerFace + (nextToDummyEnd[0] ? entriesPerEdge : 0) : 0); /* Neighbor may be a top boundary */
    1798:   }

    1800:   /* Populate idxGlobalAll */
    1801:   PetscCall(PetscMalloc1(stag->entriesGhost, &idxGlobalAll));
    1802:   count = 0;

    1804:   /* Loop over layers 1/3 : Back */
    1805:   if (!dummyStart[2]) {
    1806:     for (kghost = 0; kghost < ghostOffsetStart[2]; ++kghost) {
    1807:       const PetscInt k = nNeighbors[4][2] - ghostOffsetStart[2] + kghost; /* Note: this is the same value for all neighbors in this layer (use neighbor 4 which will always exist if we're lookng at this layer) */

    1809:       /* Loop over rows 1/3: Down Back*/
    1810:       if (!star && !dummyStart[1]) {
    1811:         for (jghost = 0; jghost < ghostOffsetStart[1]; ++jghost) {
    1812:           const PetscInt j = nNeighbors[1][1] - ghostOffsetStart[1] + jghost; /* Note: this is the same value for all neighbors in this row (use neighbor 1, down back)*/

    1814:           /* Loop over columns 1/3: Left Back Down*/
    1815:           if (!dummyStart[0]) {
    1816:             const PetscInt neighbor = 0;
    1817:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    1818:               const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    1819:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1820:             }
    1821:           } else {
    1822:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    1823:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    1824:             }
    1825:           }

    1827:           /* Loop over columns 2/3: (Middle) Down Back */
    1828:           {
    1829:             const PetscInt neighbor = 1;
    1830:             for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    1831:               const PetscInt i = ighost - ghostOffsetStart[0];
    1832:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1833:             }
    1834:           }

    1836:           /* Loop over columns 3/3: Right Down Back */
    1837:           if (!dummyEnd[0]) {
    1838:             const PetscInt neighbor = 2;
    1839:             PetscInt       i;
    1840:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    1841:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1842:             }
    1843:           } else {
    1844:             /* Partial dummy entries on (Middle) Down Back neighbor */
    1845:             const PetscInt neighbor = 1;
    1846:             PetscInt       i, dLocal;
    1847:             i = stag->n[0];
    1848:             for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) { /* Vertex */
    1849:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1850:             }
    1851:             for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    1852:               idxGlobalAll[count] = -1;
    1853:             }
    1854:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) { /* Back left edge */
    1855:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1856:             }
    1857:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face */
    1858:               idxGlobalAll[count] = -1;
    1859:             }
    1860:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) { /* Down left edge */
    1861:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1862:             }
    1863:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++dLocal, ++count) { /* Dummies on down face */
    1864:               idxGlobalAll[count] = -1;
    1865:             }
    1866:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2]; ++d, ++dLocal, ++count) { /* Left face */
    1867:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1868:             }
    1869:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on element */
    1870:               idxGlobalAll[count] = -1;
    1871:             }
    1872:             ++i;
    1873:             for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    1874:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    1875:             }
    1876:           }
    1877:         }
    1878:       } else {
    1879:         for (jghost = 0; jghost < ghostOffsetStart[1]; ++jghost) {
    1880:           for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    1881:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    1882:           }
    1883:         }
    1884:       }

    1886:       /* Loop over rows 2/3: (Middle) Back */
    1887:       {
    1888:         for (jghost = ghostOffsetStart[1]; jghost < stag->nGhost[1] - ghostOffsetEnd[1]; ++jghost) {
    1889:           const PetscInt j = jghost - ghostOffsetStart[1];

    1891:           /* Loop over columns 1/3: Left (Middle) Back */
    1892:           if (!star && !dummyStart[0]) {
    1893:             const PetscInt neighbor = 3;
    1894:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    1895:               const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    1896:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1897:             }
    1898:           } else {
    1899:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    1900:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    1901:             }
    1902:           }

    1904:           /* Loop over columns 2/3: (Middle) (Middle) Back */
    1905:           {
    1906:             const PetscInt neighbor = 4;
    1907:             for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    1908:               const PetscInt i = ighost - ghostOffsetStart[0];
    1909:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1910:             }
    1911:           }

    1913:           /* Loop over columns 3/3: Right (Middle) Back */
    1914:           if (!star && !dummyEnd[0]) {
    1915:             const PetscInt neighbor = 5;
    1916:             PetscInt       i;
    1917:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    1918:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1919:             }
    1920:           } else if (dummyEnd[0]) {
    1921:             /* Partial dummy entries on (Middle) (Middle) Back rank */
    1922:             const PetscInt neighbor = 4;
    1923:             PetscInt       i, dLocal;
    1924:             i = stag->n[0];
    1925:             for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) { /* Vertex */
    1926:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1927:             }
    1928:             for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    1929:               idxGlobalAll[count] = -1;
    1930:             }
    1931:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) { /* Back left edge */
    1932:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1933:             }
    1934:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face */
    1935:               idxGlobalAll[count] = -1;
    1936:             }
    1937:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) { /* Down left edge */
    1938:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1939:             }
    1940:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++dLocal, ++count) { /* Dummies on down face */
    1941:               idxGlobalAll[count] = -1;
    1942:             }
    1943:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2]; ++d, ++dLocal, ++count) { /* Left face */
    1944:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1945:             }
    1946:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on element */
    1947:               idxGlobalAll[count] = -1;
    1948:             }
    1949:             ++i;
    1950:             for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    1951:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    1952:             }
    1953:           } else {
    1954:             /* Right (Middle) Back dummies */
    1955:             PetscInt i;
    1956:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    1957:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    1958:             }
    1959:           }
    1960:         }
    1961:       }

    1963:       /* Loop over rows 3/3: Up Back */
    1964:       if (!star && !dummyEnd[1]) {
    1965:         PetscInt j;
    1966:         for (j = 0; j < ghostOffsetEnd[1]; ++j) {
    1967:           /* Loop over columns 1/3: Left Up Back*/
    1968:           if (!dummyStart[0]) {
    1969:             const PetscInt neighbor = 6;
    1970:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    1971:               const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    1972:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1973:             }
    1974:           } else {
    1975:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    1976:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    1977:             }
    1978:           }

    1980:           /* Loop over columns 2/3: (Middle) Up Back*/
    1981:           {
    1982:             const PetscInt neighbor = 7;
    1983:             for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    1984:               const PetscInt i = ighost - ghostOffsetStart[0];
    1985:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1986:             }
    1987:           }

    1989:           /* Loop over columns 3/3: Right Up Back*/
    1990:           if (!dummyEnd[0]) {
    1991:             const PetscInt neighbor = 8;
    1992:             PetscInt       i;
    1993:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    1994:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    1995:             }
    1996:           } else {
    1997:             /* Partial dummies on (Middle) Up Back */
    1998:             const PetscInt neighbor = 7;
    1999:             PetscInt       i, dLocal;
    2000:             i = stag->n[0];
    2001:             for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) { /* Vertex */
    2002:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2003:             }
    2004:             for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    2005:               idxGlobalAll[count] = -1;
    2006:             }
    2007:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) { /* Back left edge */
    2008:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2009:             }
    2010:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face */
    2011:               idxGlobalAll[count] = -1;
    2012:             }
    2013:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) { /* Down left edge */
    2014:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2015:             }
    2016:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++dLocal, ++count) { /* Dummies on down face */
    2017:               idxGlobalAll[count] = -1;
    2018:             }
    2019:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2]; ++d, ++dLocal, ++count) { /* Left face */
    2020:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2021:             }
    2022:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on element */
    2023:               idxGlobalAll[count] = -1;
    2024:             }
    2025:             ++i;
    2026:             for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2027:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2028:             }
    2029:           }
    2030:         }
    2031:       } else if (dummyEnd[1]) {
    2032:         /* Up Back partial dummy row */
    2033:         PetscInt j = stag->n[1];

    2035:         if (!star && !dummyStart[0]) {
    2036:           /* Up Back partial dummy row: Loop over columns 1/3: Left Up Back, on Left (Middle) Back rank */
    2037:           const PetscInt neighbor = 3;
    2038:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2039:             const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2040:             PetscInt       dLocal;
    2041:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                                  /* Vertex and back down edge */
    2042:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2043:             }

    2045:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face and back left edge */
    2046:               idxGlobalAll[count] = -1;
    2047:             }
    2048:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++d, ++dLocal, ++count) {                                            /* Down left edge and down face */
    2049:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2050:             }
    2051:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies left face and element */
    2052:               idxGlobalAll[count] = -1;
    2053:             }
    2054:           }
    2055:         } else {
    2056:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2057:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2058:           }
    2059:         }

    2061:         /* Up Back partial dummy row: Loop over columns 2/3: (Middle) Up Back, on (Middle) (Middle) Back rank */
    2062:         {
    2063:           const PetscInt neighbor = 4;
    2064:           for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2065:             const PetscInt i = ighost - ghostOffsetStart[0];
    2066:             PetscInt       dLocal;
    2067:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                                  /* Vertex and back down edge */
    2068:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2069:             }
    2070:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face and back left edge */
    2071:               idxGlobalAll[count] = -1;
    2072:             }
    2073:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++d, ++dLocal, ++count) {                                            /* Down left edge and down face */
    2074:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2075:             }
    2076:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies left face and element */
    2077:               idxGlobalAll[count] = -1;
    2078:             }
    2079:           }
    2080:         }

    2082:         /* Up Back partial dummy row: Loop over columns 3/3: Right Up Back, on Right (Middle) Back rank */
    2083:         if (!star && !dummyEnd[0]) {
    2084:           const PetscInt neighbor = 5;
    2085:           PetscInt       i;
    2086:           for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2087:             PetscInt dLocal;
    2088:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                                  /* Vertex and back down edge */
    2089:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2090:             }
    2091:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face and back left edge */
    2092:               idxGlobalAll[count] = -1;
    2093:             }
    2094:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++d, ++dLocal, ++count) {                                            /* Down left edge and down face */
    2095:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2096:             }
    2097:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies left face and element */
    2098:               idxGlobalAll[count] = -1;
    2099:             }
    2100:           }
    2101:         } else if (dummyEnd[0]) {
    2102:           /* Up Right Back partial dummy element, on (Middle) (Middle) Back rank */
    2103:           const PetscInt neighbor = 4;
    2104:           PetscInt       i, dLocal;
    2105:           i = stag->n[0];
    2106:           for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) {                                                                 /* Vertex */
    2107:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2108:           }
    2109:           for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back down edge, back face and back left edge */
    2110:             idxGlobalAll[count] = -1;
    2111:           }
    2112:           for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                                                /* Down left edge */
    2113:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2114:           }
    2115:           for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies down face, left face and element */
    2116:             idxGlobalAll[count] = -1;
    2117:           }
    2118:           ++i;
    2119:           for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2120:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2121:           }
    2122:         } else {
    2123:           /* Up Right Back dummies */
    2124:           PetscInt i;
    2125:           for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2126:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2127:           }
    2128:         }
    2129:         ++j;
    2130:         /* Up Back additional dummy rows */
    2131:         for (; j < stag->n[1] + ghostOffsetEnd[1]; ++j) {
    2132:           for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    2133:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2134:           }
    2135:         }
    2136:       } else {
    2137:         /* Up Back dummy rows */
    2138:         PetscInt j;
    2139:         for (j = 0; j < ghostOffsetEnd[1]; ++j) {
    2140:           for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    2141:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2142:           }
    2143:         }
    2144:       }
    2145:     }
    2146:   } else {
    2147:     for (kghost = 0; kghost < ghostOffsetStart[2]; ++kghost) {
    2148:       for (jghost = 0; jghost < stag->nGhost[1]; ++jghost) {
    2149:         for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    2150:           for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2151:         }
    2152:       }
    2153:     }
    2154:   }

    2156:   /* Loop over layers 2/3 : (Middle)  */
    2157:   {
    2158:     for (kghost = ghostOffsetStart[2]; kghost < stag->nGhost[2] - ghostOffsetEnd[2]; ++kghost) {
    2159:       const PetscInt k = kghost - ghostOffsetStart[2];

    2161:       /* Loop over rows 1/3: Down (Middle) */
    2162:       if (!dummyStart[1]) {
    2163:         for (jghost = 0; jghost < ghostOffsetStart[1]; ++jghost) {
    2164:           const PetscInt j = nNeighbors[10][1] - ghostOffsetStart[1] + jghost; /* down neighbor (10) always exists */

    2166:           /* Loop over columns 1/3: Left Down (Middle) */
    2167:           if (!star && !dummyStart[0]) {
    2168:             const PetscInt neighbor = 9;
    2169:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2170:               const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2171:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2172:             }
    2173:           } else {
    2174:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2175:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2176:             }
    2177:           }

    2179:           /* Loop over columns 2/3: (Middle) Down (Middle) */
    2180:           {
    2181:             const PetscInt neighbor = 10;
    2182:             for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2183:               const PetscInt i = ighost - ghostOffsetStart[0];
    2184:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2185:             }
    2186:           }

    2188:           /* Loop over columns 3/3: Right Down (Middle) */
    2189:           if (!star && !dummyEnd[0]) {
    2190:             const PetscInt neighbor = 11;
    2191:             PetscInt       i;
    2192:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2193:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2194:             }
    2195:           } else if (dummyEnd[0]) {
    2196:             /* Partial dummies on (Middle) Down (Middle) neighbor */
    2197:             const PetscInt neighbor = 10;
    2198:             PetscInt       i, dLocal;
    2199:             i = stag->n[0];
    2200:             for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) { /* Vertex */
    2201:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2202:             }
    2203:             for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    2204:               idxGlobalAll[count] = -1;
    2205:             }
    2206:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) { /* Back left edge */
    2207:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2208:             }
    2209:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face */
    2210:               idxGlobalAll[count] = -1;
    2211:             }
    2212:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) { /* Down left edge */
    2213:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2214:             }
    2215:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++dLocal, ++count) { /* Dummies on down face */
    2216:               idxGlobalAll[count] = -1;
    2217:             }
    2218:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2]; ++d, ++dLocal, ++count) { /* Left face */
    2219:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2220:             }
    2221:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on element */
    2222:               idxGlobalAll[count] = -1;
    2223:             }
    2224:             ++i;
    2225:             for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2226:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2227:             }
    2228:           } else {
    2229:             /* Right Down (Middle) dummies */
    2230:             PetscInt i;
    2231:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2232:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2233:             }
    2234:           }
    2235:         }
    2236:       } else {
    2237:         for (jghost = 0; jghost < ghostOffsetStart[1]; ++jghost) {
    2238:           for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    2239:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2240:           }
    2241:         }
    2242:       }

    2244:       /* Loop over rows 2/3: (Middle) (Middle) */
    2245:       {
    2246:         for (jghost = ghostOffsetStart[1]; jghost < stag->nGhost[1] - ghostOffsetEnd[1]; ++jghost) {
    2247:           const PetscInt j = jghost - ghostOffsetStart[1];

    2249:           /* Loop over columns 1/3: Left (Middle) (Middle) */
    2250:           if (!dummyStart[0]) {
    2251:             const PetscInt neighbor = 12;
    2252:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2253:               const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2254:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2255:             }
    2256:           } else {
    2257:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2258:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2259:             }
    2260:           }

    2262:           /* Loop over columns 2/3: (Middle) (Middle) (Middle) aka (Here) */
    2263:           {
    2264:             const PetscInt neighbor = 13;
    2265:             for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2266:               const PetscInt i = ighost - ghostOffsetStart[0];
    2267:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2268:             }
    2269:           }

    2271:           /* Loop over columns 3/3: Right (Middle) (Middle) */
    2272:           if (!dummyEnd[0]) {
    2273:             const PetscInt neighbor = 14;
    2274:             PetscInt       i;
    2275:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2276:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2277:             }
    2278:           } else {
    2279:             /* Partial dummies on this rank */
    2280:             const PetscInt neighbor = 13;
    2281:             PetscInt       i, dLocal;
    2282:             i = stag->n[0];
    2283:             for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) { /* Vertex */
    2284:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2285:             }
    2286:             for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    2287:               idxGlobalAll[count] = -1;
    2288:             }
    2289:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) { /* Back left edge */
    2290:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2291:             }
    2292:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face */
    2293:               idxGlobalAll[count] = -1;
    2294:             }
    2295:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) { /* Down left edge */
    2296:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2297:             }
    2298:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++dLocal, ++count) { /* Dummies on down face */
    2299:               idxGlobalAll[count] = -1;
    2300:             }
    2301:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2]; ++d, ++dLocal, ++count) { /* Left face */
    2302:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2303:             }
    2304:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on element */
    2305:               idxGlobalAll[count] = -1;
    2306:             }
    2307:             ++i;
    2308:             for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2309:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2310:             }
    2311:           }
    2312:         }
    2313:       }

    2315:       /* Loop over rows 3/3: Up (Middle) */
    2316:       if (!dummyEnd[1]) {
    2317:         PetscInt j;
    2318:         for (j = 0; j < ghostOffsetEnd[1]; ++j) {
    2319:           /* Loop over columns 1/3: Left Up (Middle) */
    2320:           if (!star && !dummyStart[0]) {
    2321:             const PetscInt neighbor = 15;
    2322:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2323:               const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2324:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2325:             }
    2326:           } else {
    2327:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2328:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2329:             }
    2330:           }

    2332:           /* Loop over columns 2/3: (Middle) Up (Middle) */
    2333:           {
    2334:             const PetscInt neighbor = 16;
    2335:             for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2336:               const PetscInt i = ighost - ghostOffsetStart[0];
    2337:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2338:             }
    2339:           }

    2341:           /* Loop over columns 3/3: Right Up (Middle) */
    2342:           if (!star && !dummyEnd[0]) {
    2343:             const PetscInt neighbor = 17;
    2344:             PetscInt       i;
    2345:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2346:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2347:             }
    2348:           } else if (dummyEnd[0]) {
    2349:             /* Partial dummies on (Middle) Up (Middle) rank */
    2350:             const PetscInt neighbor = 16;
    2351:             PetscInt       i, dLocal;
    2352:             i = stag->n[0];
    2353:             for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) { /* Vertex */
    2354:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2355:             }
    2356:             for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    2357:               idxGlobalAll[count] = -1;
    2358:             }
    2359:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) { /* Back left edge */
    2360:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2361:             }
    2362:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face */
    2363:               idxGlobalAll[count] = -1;
    2364:             }
    2365:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) { /* Down left edge */
    2366:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2367:             }
    2368:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++dLocal, ++count) { /* Dummies on down face */
    2369:               idxGlobalAll[count] = -1;
    2370:             }
    2371:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2]; ++d, ++dLocal, ++count) { /* Left face */
    2372:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2373:             }
    2374:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on element */
    2375:               idxGlobalAll[count] = -1;
    2376:             }
    2377:             ++i;
    2378:             for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2379:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2380:             }
    2381:           } else {
    2382:             /* Right Up (Middle) dumies */
    2383:             PetscInt i;
    2384:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2385:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2386:             }
    2387:           }
    2388:         }
    2389:       } else {
    2390:         /* Up (Middle) partial dummy row */
    2391:         PetscInt j = stag->n[1];

    2393:         /* Up (Middle) partial dummy row: columns 1/3: Left Up (Middle), on Left (Middle) (Middle) rank */
    2394:         if (!dummyStart[0]) {
    2395:           const PetscInt neighbor = 12;
    2396:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2397:             const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2398:             PetscInt       dLocal;
    2399:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                                  /* Vertex and back down edge */
    2400:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2401:             }
    2402:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face and back left edge */
    2403:               idxGlobalAll[count] = -1;
    2404:             }
    2405:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++d, ++dLocal, ++count) {                                            /* Down left edge and down face */
    2406:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2407:             }
    2408:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies left face and element */
    2409:               idxGlobalAll[count] = -1;
    2410:             }
    2411:           }
    2412:         } else {
    2413:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2414:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2415:           }
    2416:         }

    2418:         /* Up (Middle) partial dummy row: columns 2/3: (Middle) Up (Middle), on (Middle) (Middle) (Middle) = here rank */
    2419:         {
    2420:           const PetscInt neighbor = 13;
    2421:           for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2422:             const PetscInt i = ighost - ghostOffsetStart[0];
    2423:             PetscInt       dLocal;
    2424:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                                  /* Vertex and back down edge */
    2425:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2426:             }
    2427:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face and back left edge */
    2428:               idxGlobalAll[count] = -1;
    2429:             }
    2430:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++d, ++dLocal, ++count) {                                            /* Down left edge and down face */
    2431:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2432:             }
    2433:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies left face and element */
    2434:               idxGlobalAll[count] = -1;
    2435:             }
    2436:           }
    2437:         }

    2439:         if (!dummyEnd[0]) {
    2440:           /* Up (Middle) partial dummy row: columns 3/3: Right Up (Middle), on Right (Middle) (Middle) rank */
    2441:           const PetscInt neighbor = 14;
    2442:           PetscInt       i;
    2443:           for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2444:             PetscInt dLocal;
    2445:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                                  /* Vertex and back down edge */
    2446:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2447:             }
    2448:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face and back left edge */
    2449:               idxGlobalAll[count] = -1;
    2450:             }
    2451:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++d, ++dLocal, ++count) {                                            /* Down left edge and down face */
    2452:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2453:             }
    2454:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies left face and element */
    2455:               idxGlobalAll[count] = -1;
    2456:             }
    2457:           }
    2458:         } else {
    2459:           /* Up (Middle) Right partial dummy element, on (Middle) (Middle) (Middle) = here rank */
    2460:           const PetscInt neighbor = 13;
    2461:           PetscInt       i, dLocal;
    2462:           i = stag->n[0];
    2463:           for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) {                                                                 /* Vertex */
    2464:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2465:           }
    2466:           for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back down edge, back face and back left edge */
    2467:             idxGlobalAll[count] = -1;
    2468:           }
    2469:           for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                                                /* Down left edge */
    2470:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2471:           }
    2472:           for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies down face, left face and element */
    2473:             idxGlobalAll[count] = -1;
    2474:           }
    2475:           ++i;
    2476:           for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2477:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2478:           }
    2479:         }
    2480:         ++j;
    2481:         /* Up (Middle) additional dummy rows */
    2482:         for (; j < stag->n[1] + ghostOffsetEnd[1]; ++j) {
    2483:           for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    2484:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2485:           }
    2486:         }
    2487:       }
    2488:     }
    2489:   }

    2491:   /* Loop over layers 3/3 : Front */
    2492:   if (!dummyEnd[2]) {
    2493:     PetscInt k;
    2494:     for (k = 0; k < ghostOffsetEnd[2]; ++k) {
    2495:       /* Loop over rows 1/3: Down Front */
    2496:       if (!star && !dummyStart[1]) {
    2497:         for (jghost = 0; jghost < ghostOffsetStart[1]; ++jghost) {
    2498:           const PetscInt j = nNeighbors[19][1] - ghostOffsetStart[1] + jghost; /* Constant for whole row (use down front neighbor) */

    2500:           /* Loop over columns 1/3: Left Down Front */
    2501:           if (!dummyStart[0]) {
    2502:             const PetscInt neighbor = 18;
    2503:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2504:               const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2505:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2506:             }
    2507:           } else {
    2508:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2509:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2510:             }
    2511:           }

    2513:           /* Loop over columns 2/3: (Middle) Down Front */
    2514:           {
    2515:             const PetscInt neighbor = 19;
    2516:             for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2517:               const PetscInt i = ighost - ghostOffsetStart[0];
    2518:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) { /* vertex, 2 edges, and face associated with back face */
    2519:                 idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2520:               }
    2521:             }
    2522:           }

    2524:           /* Loop over columns 3/3: Right Down Front */
    2525:           if (!dummyEnd[0]) {
    2526:             const PetscInt neighbor = 20;
    2527:             PetscInt       i;
    2528:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2529:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2530:             }
    2531:           } else {
    2532:             /* Partial dummy element on (Middle) Down Front rank */
    2533:             const PetscInt neighbor = 19;
    2534:             PetscInt       i, dLocal;
    2535:             i = stag->n[0];
    2536:             for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) { /* Vertex */
    2537:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2538:             }
    2539:             for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    2540:               idxGlobalAll[count] = -1;
    2541:             }
    2542:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) { /* Back left edge */
    2543:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2544:             }
    2545:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face */
    2546:               idxGlobalAll[count] = -1;
    2547:             }
    2548:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) { /* Down left edge */
    2549:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2550:             }
    2551:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++dLocal, ++count) { /* Dummies on down face */
    2552:               idxGlobalAll[count] = -1;
    2553:             }
    2554:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2]; ++d, ++dLocal, ++count) { /* Left face */
    2555:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2556:             }
    2557:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on element */
    2558:               idxGlobalAll[count] = -1;
    2559:             }
    2560:             ++i;
    2561:             for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2562:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2563:             }
    2564:           }
    2565:         }
    2566:       } else {
    2567:         for (jghost = 0; jghost < ghostOffsetStart[1]; ++jghost) {
    2568:           for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    2569:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2570:           }
    2571:         }
    2572:       }

    2574:       /* Loop over rows 2/3: (Middle) Front */
    2575:       {
    2576:         for (jghost = ghostOffsetStart[1]; jghost < stag->nGhost[1] - ghostOffsetEnd[1]; ++jghost) {
    2577:           const PetscInt j = jghost - ghostOffsetStart[1];

    2579:           /* Loop over columns 1/3: Left (Middle) Front*/
    2580:           if (!star && !dummyStart[0]) {
    2581:             const PetscInt neighbor = 21;
    2582:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2583:               const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2584:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2585:             }
    2586:           } else {
    2587:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2588:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2589:             }
    2590:           }

    2592:           /* Loop over columns 2/3: (Middle) (Middle) Front*/
    2593:           {
    2594:             const PetscInt neighbor = 22;
    2595:             for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2596:               const PetscInt i = ighost - ghostOffsetStart[0];
    2597:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2598:             }
    2599:           }

    2601:           /* Loop over columns 3/3: Right (Middle) Front*/
    2602:           if (!star && !dummyEnd[0]) {
    2603:             const PetscInt neighbor = 23;
    2604:             PetscInt       i;
    2605:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2606:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2607:             }
    2608:           } else if (dummyEnd[0]) {
    2609:             /* Partial dummy element on (Middle) (Middle) Front element */
    2610:             const PetscInt neighbor = 22;
    2611:             PetscInt       i, dLocal;
    2612:             i = stag->n[0];
    2613:             for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) { /* Vertex */
    2614:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2615:             }
    2616:             for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    2617:               idxGlobalAll[count] = -1;
    2618:             }
    2619:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) { /* Back left edge */
    2620:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2621:             }
    2622:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face */
    2623:               idxGlobalAll[count] = -1;
    2624:             }
    2625:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) { /* Down left edge */
    2626:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2627:             }
    2628:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++dLocal, ++count) { /* Dummies on down face */
    2629:               idxGlobalAll[count] = -1;
    2630:             }
    2631:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2]; ++d, ++dLocal, ++count) { /* Left face */
    2632:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2633:             }
    2634:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on element */
    2635:               idxGlobalAll[count] = -1;
    2636:             }
    2637:             ++i;
    2638:             for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2639:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2640:             }
    2641:           } else {
    2642:             /* Right (Middle) Front dummies */
    2643:             PetscInt i;
    2644:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2645:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2646:             }
    2647:           }
    2648:         }
    2649:       }

    2651:       /* Loop over rows 3/3: Up Front */
    2652:       if (!star && !dummyEnd[1]) {
    2653:         PetscInt j;
    2654:         for (j = 0; j < ghostOffsetEnd[1]; ++j) {
    2655:           /* Loop over columns 1/3: Left Up Front */
    2656:           if (!dummyStart[0]) {
    2657:             const PetscInt neighbor = 24;
    2658:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2659:               const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2660:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2661:             }
    2662:           } else {
    2663:             for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2664:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2665:             }
    2666:           }

    2668:           /* Loop over columns 2/3: (Middle) Up Front */
    2669:           {
    2670:             const PetscInt neighbor = 25;
    2671:             for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2672:               const PetscInt i = ighost - ghostOffsetStart[0];
    2673:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2674:             }
    2675:           }

    2677:           /* Loop over columns 3/3: Right Up Front */
    2678:           if (!dummyEnd[0]) {
    2679:             const PetscInt neighbor = 26;
    2680:             PetscInt       i;
    2681:             for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2682:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2683:             }
    2684:           } else {
    2685:             /* Partial dummy element on (Middle) Up Front rank */
    2686:             const PetscInt neighbor = 25;
    2687:             PetscInt       i, dLocal;
    2688:             i = stag->n[0];
    2689:             for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) { /* Vertex */
    2690:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2691:             }
    2692:             for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    2693:               idxGlobalAll[count] = -1;
    2694:             }
    2695:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) { /* Back left edge */
    2696:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2697:             }
    2698:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face */
    2699:               idxGlobalAll[count] = -1;
    2700:             }
    2701:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) { /* Down left edge */
    2702:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2703:             }
    2704:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++dLocal, ++count) { /* Dummies on down face */
    2705:               idxGlobalAll[count] = -1;
    2706:             }
    2707:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 3 * stag->dof[2]; ++d, ++dLocal, ++count) { /* Left face */
    2708:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * stag->entriesPerElement + d;
    2709:             }
    2710:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on element */
    2711:               idxGlobalAll[count] = -1;
    2712:             }
    2713:             ++i;
    2714:             for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2715:               for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2716:             }
    2717:           }
    2718:         }
    2719:       } else if (dummyEnd[1]) {
    2720:         /* Up Front partial dummy row */
    2721:         PetscInt j = stag->n[1];

    2723:         /* Up Front partial dummy row: columns 1/3: Left Up Front, on Left (Middle) Front rank */
    2724:         if (!star && !dummyStart[0]) {
    2725:           const PetscInt neighbor = 21;
    2726:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2727:             const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2728:             PetscInt       dLocal;
    2729:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                                  /* Vertex and back down edge */
    2730:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2731:             }
    2732:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face and back left edge */
    2733:               idxGlobalAll[count] = -1;
    2734:             }
    2735:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++d, ++dLocal, ++count) {                                            /* Down left edge and down face */
    2736:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2737:             }
    2738:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies left face and element */
    2739:               idxGlobalAll[count] = -1;
    2740:             }
    2741:           }
    2742:         } else {
    2743:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2744:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2745:           }
    2746:         }

    2748:         /* Up Front partial dummy row: columns 2/3: (Middle) Up Front, on (Middle) (Middle) Front rank */
    2749:         {
    2750:           const PetscInt neighbor = 22;
    2751:           for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2752:             const PetscInt i = ighost - ghostOffsetStart[0];
    2753:             PetscInt       dLocal;
    2754:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                                  /* Vertex and back down edge */
    2755:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2756:             }
    2757:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face and back left edge */
    2758:               idxGlobalAll[count] = -1;
    2759:             }
    2760:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++d, ++dLocal, ++count) {                                            /* Down left edge and down face */
    2761:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2762:             }
    2763:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies left face and element */
    2764:               idxGlobalAll[count] = -1;
    2765:             }
    2766:           }
    2767:         }

    2769:         if (!star && !dummyEnd[0]) {
    2770:           /* Up Front partial dummy row: columns 3/3: Right Up Front, on Right (Middle) Front rank */
    2771:           const PetscInt neighbor = 23;
    2772:           PetscInt       i;
    2773:           for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2774:             PetscInt dLocal;
    2775:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                                  /* Vertex and back down edge */
    2776:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2777:             }
    2778:             for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back face and back left edge */
    2779:               idxGlobalAll[count] = -1;
    2780:             }
    2781:             for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + 2 * stag->dof[2]; ++d, ++dLocal, ++count) {                                            /* Down left edge and down face */
    2782:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2783:             }
    2784:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies left face and element */
    2785:               idxGlobalAll[count] = -1;
    2786:             }
    2787:           }

    2789:         } else if (dummyEnd[0]) {
    2790:           /* Partial Right Up Front dummy element, on (Middle) (Middle) Front rank */
    2791:           const PetscInt neighbor = 22;
    2792:           PetscInt       i, dLocal;
    2793:           i = stag->n[0];
    2794:           for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) {                                                                 /* Vertex */
    2795:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2796:           }
    2797:           for (; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++dLocal, ++count) { /* Dummies on back down edge, back face and back left edge */
    2798:             idxGlobalAll[count] = -1;
    2799:           }
    2800:           for (; dLocal < stag->dof[0] + 3 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                                                /* Down left edge */
    2801:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * eprNeighbor[neighbor] + i * entriesPerFace + d; /* Note i increment by face */
    2802:           }
    2803:           for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies down face, left face and element */
    2804:             idxGlobalAll[count] = -1;
    2805:           }
    2806:           ++i;
    2807:           for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2808:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2809:           }
    2810:         } else {
    2811:           /* Right Up Front dummies */
    2812:           PetscInt i;
    2813:           for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2814:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2815:           }
    2816:         }
    2817:         ++j;
    2818:         /* Up Front additional dummy rows */
    2819:         for (; j < stag->n[1] + ghostOffsetEnd[1]; ++j) {
    2820:           for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    2821:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2822:           }
    2823:         }
    2824:       } else {
    2825:         /* Up Front dummies */
    2826:         PetscInt j;
    2827:         for (j = 0; j < ghostOffsetEnd[1]; ++j) {
    2828:           for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    2829:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2830:           }
    2831:         }
    2832:       }
    2833:     }
    2834:   } else {
    2835:     PetscInt k = stag->n[2];

    2837:     /* Front partial ghost layer: rows 1/3: Down Front, on Down (Middle) */
    2838:     if (!dummyStart[1]) {
    2839:       for (jghost = 0; jghost < ghostOffsetStart[1]; ++jghost) {
    2840:         const PetscInt j = nNeighbors[10][1] - ghostOffsetStart[1] + jghost; /* wrt down neighbor (10) */

    2842:         /* Down Front partial ghost row: columns 1/3: Left Down Front, on  Left Down (Middle) */
    2843:         if (!star && !dummyStart[0]) {
    2844:           const PetscInt neighbor  = 9;
    2845:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0]; /* Note that we can't be a right boundary */
    2846:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2847:             const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2848:             PetscInt       dLocal;
    2849:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                   /* Vertex, back down edge, back face, back left edge */
    2850:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note j increment by face */
    2851:             }
    2852:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining points */
    2853:               idxGlobalAll[count] = -1;
    2854:             }
    2855:           }
    2856:         } else {
    2857:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2858:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2859:           }
    2860:         }

    2862:         /* Down Front partial ghost row: columns 2/3: (Middle) Down Front, on (Middle) Down (Middle) */
    2863:         {
    2864:           const PetscInt neighbor  = 10;
    2865:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0); /* We+neighbor may be a right boundary */
    2866:           for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2867:             const PetscInt i = ighost - ghostOffsetStart[0];
    2868:             PetscInt       dLocal;
    2869:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                   /* Vertex, back down edge, back face, back left edge */
    2870:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note j increment by face */
    2871:             }
    2872:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining points */
    2873:               idxGlobalAll[count] = -1;
    2874:             }
    2875:           }
    2876:         }

    2878:         if (!star && !dummyEnd[0]) {
    2879:           /* Down Front partial dummy row: columns 3/3: Right Down Front, on Right Down (Middle) */
    2880:           const PetscInt neighbor  = 11;
    2881:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerEdge : 0); /* Neighbor may be a right boundary */
    2882:           PetscInt       i;
    2883:           for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2884:             PetscInt dLocal;
    2885:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                   /* Vertex, back down edge, back face, back left edge */
    2886:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note j increment by face */
    2887:             }
    2888:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining points */
    2889:               idxGlobalAll[count] = -1;
    2890:             }
    2891:           }
    2892:         } else if (dummyEnd[0]) {
    2893:           /* Right Down Front partial dummy element, living on (Middle) Down (Middle) rank */
    2894:           const PetscInt neighbor  = 10;
    2895:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0); /* We+neighbor may be a right boundary */
    2896:           PetscInt       i, dLocal;
    2897:           i = stag->n[0];
    2898:           for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) {                                                     /* Vertex */
    2899:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note i increment by face */
    2900:           }
    2901:           for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    2902:             idxGlobalAll[count] = -1;
    2903:           }
    2904:           for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) {                                                   /* left back edge */
    2905:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note i increment by face */
    2906:           }
    2907:           for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining */
    2908:             idxGlobalAll[count] = -1;
    2909:           }
    2910:           ++i;
    2911:           for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    2912:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2913:           }
    2914:         } else {
    2915:           PetscInt i;
    2916:           /* Right Down Front dummies */
    2917:           for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2918:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2919:           }
    2920:         }
    2921:       }
    2922:     } else {
    2923:       for (jghost = 0; jghost < ghostOffsetStart[1]; ++jghost) {
    2924:         for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    2925:           for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2926:         }
    2927:       }
    2928:     }

    2930:     /* Front partial ghost layer: rows 2/3: (Middle) Front, on (Middle) (Middle) */
    2931:     {
    2932:       for (jghost = ghostOffsetStart[1]; jghost < stag->nGhost[1] - ghostOffsetEnd[1]; ++jghost) {
    2933:         const PetscInt j = jghost - ghostOffsetStart[1];

    2935:         /* (Middle) Front partial dummy row: columns 1/3: Left (Middle) Front, on Left (Middle) (Middle) */
    2936:         if (!dummyStart[0]) {
    2937:           const PetscInt neighbor  = 12;
    2938:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0];
    2939:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2940:             const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    2941:             PetscInt       dLocal;
    2942:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                   /* Vertex, back down edge, back face, back left edge */
    2943:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note j increment by face */
    2944:             }
    2945:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining points */
    2946:               idxGlobalAll[count] = -1;
    2947:             }
    2948:           }
    2949:         } else {
    2950:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    2951:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    2952:           }
    2953:         }

    2955:         /* (Middle) Front partial dummy row: columns 2/3: (Middle) (Middle) Front, on (Middle) (Middle) (Middle) */
    2956:         {
    2957:           const PetscInt neighbor  = 13;
    2958:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0); /* We may be a right boundary */
    2959:           for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    2960:             const PetscInt i = ighost - ghostOffsetStart[0];
    2961:             PetscInt       dLocal;
    2962:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                   /* Vertex, back down edge, back face, back left edge */
    2963:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note j increment by face */
    2964:             }
    2965:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining points */
    2966:               idxGlobalAll[count] = -1;
    2967:             }
    2968:           }
    2969:         }

    2971:         if (!dummyEnd[0]) {
    2972:           /* (Middle) Front partial dummy row: columns 3/3: Right (Middle) Front, on Right (Middle) (Middle) */
    2973:           const PetscInt neighbor  = 14;
    2974:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerEdge : 0); /* Neighbor may be a right boundary */
    2975:           PetscInt       i;
    2976:           for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    2977:             PetscInt dLocal;
    2978:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                   /* Vertex, back down edge, back face, back left edge */
    2979:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note j increment by face */
    2980:             }
    2981:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining points */
    2982:               idxGlobalAll[count] = -1;
    2983:             }
    2984:           }
    2985:         } else {
    2986:           /* Right (Middle) Front partial dummy element, on (Middle) (Middle) (Middle) rank */
    2987:           const PetscInt neighbor  = 13;
    2988:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0); /* We may be a right boundary */
    2989:           PetscInt       i, dLocal;
    2990:           i = stag->n[0];
    2991:           for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) {                                                     /* Vertex */
    2992:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note i increment by face */
    2993:           }
    2994:           for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    2995:             idxGlobalAll[count] = -1;
    2996:           }
    2997:           for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) {                                                   /* left back edge */
    2998:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note i increment by face */
    2999:           }
    3000:           for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining */
    3001:             idxGlobalAll[count] = -1;
    3002:           }
    3003:           ++i;
    3004:           for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    3005:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    3006:           }
    3007:         }
    3008:       }
    3009:     }

    3011:     /* Front partial ghost layer: rows 3/3: Up Front, on Up (Middle) */
    3012:     if (!dummyEnd[1]) {
    3013:       PetscInt j;
    3014:       for (j = 0; j < ghostOffsetEnd[1]; ++j) {
    3015:         /* Up Front partial dummy row: columns 1/3: Left Up Front, on Left Up (Middle) */
    3016:         if (!star && !dummyStart[0]) {
    3017:           const PetscInt neighbor  = 15;
    3018:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0];
    3019:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    3020:             const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    3021:             PetscInt       dLocal;
    3022:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                   /* Vertex, back down edge, back face, back left edge */
    3023:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note j increment by face */
    3024:             }
    3025:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining points */
    3026:               idxGlobalAll[count] = -1;
    3027:             }
    3028:           }
    3029:         } else {
    3030:           for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    3031:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    3032:           }
    3033:         }

    3035:         /* Up Front partial dummy row: columns 2/3: (Middle) Up Front, on (Middle) Up (Middle) */
    3036:         {
    3037:           const PetscInt neighbor  = 16;
    3038:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0); /* We+neighbor may be a right boundary */
    3039:           for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    3040:             const PetscInt i = ighost - ghostOffsetStart[0];
    3041:             PetscInt       dLocal;
    3042:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                   /* Vertex, back down edge, back face, back left edge */
    3043:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note j increment by face */
    3044:             }
    3045:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining points */
    3046:               idxGlobalAll[count] = -1;
    3047:             }
    3048:           }
    3049:         }

    3051:         if (!star && !dummyEnd[0]) {
    3052:           /* Up Front partial dummy row: columns 3/3: Right Up Front, on Right Up (Middle) */
    3053:           const PetscInt neighbor  = 17;
    3054:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerEdge : 0); /* Neighbor may be a right boundary */
    3055:           PetscInt       i;
    3056:           for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    3057:             PetscInt dLocal;
    3058:             for (d = 0, dLocal = 0; dLocal < stag->dof[0] + 2 * stag->dof[1] + stag->dof[2]; ++d, ++dLocal, ++count) {                   /* Vertex, back down edge, back face, back left edge */
    3059:               idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note j increment by face */
    3060:             }
    3061:             for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining points */
    3062:               idxGlobalAll[count] = -1;
    3063:             }
    3064:           }
    3065:         } else if (dummyEnd[0]) {
    3066:           /* Right Up Front partial dummy element, on (Middle) Up (Middle) */
    3067:           const PetscInt neighbor  = 16;
    3068:           const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0); /* We+neighbor may be a right boundary */
    3069:           PetscInt       i, dLocal;
    3070:           i = stag->n[0];
    3071:           for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) {                                                     /* Vertex */
    3072:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note i increment by face */
    3073:           }
    3074:           for (; dLocal < stag->dof[0] + stag->dof[1]; ++dLocal, ++count) { /* Dummies on back down edge */
    3075:             idxGlobalAll[count] = -1;
    3076:           }
    3077:           for (; dLocal < stag->dof[0] + 2 * stag->dof[1]; ++d, ++dLocal, ++count) {                                                   /* left back edge */
    3078:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerFace + d; /* Note i increment by face */
    3079:           }
    3080:           for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining */
    3081:             idxGlobalAll[count] = -1;
    3082:           }
    3083:           ++i;
    3084:           for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    3085:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    3086:           }
    3087:         } else {
    3088:           /* Right Up Front dummies */
    3089:           PetscInt i;
    3090:           /* Right Down Front dummies */
    3091:           for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    3092:             for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    3093:           }
    3094:         }
    3095:       }
    3096:     } else {
    3097:       /* Up Front partial dummy row */
    3098:       PetscInt j = stag->n[1];

    3100:       /* Up Front partial dummy row: columns 1/3: Left Up Front, on Left (Middle) (Middle) */
    3101:       if (!dummyStart[0]) {
    3102:         const PetscInt neighbor  = 12;
    3103:         const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0];
    3104:         for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    3105:           const PetscInt i = nNeighbors[neighbor][0] - ghostOffsetStart[0] + ighost;
    3106:           PetscInt       dLocal;
    3107:           for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                      /* Vertex  and down back edge */
    3108:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerEdge + d; /* Note increments */
    3109:           }
    3110:           for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining */
    3111:             idxGlobalAll[count] = -1;
    3112:           }
    3113:         }
    3114:       } else {
    3115:         for (ighost = 0; ighost < ghostOffsetStart[0]; ++ighost) {
    3116:           for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    3117:         }
    3118:       }

    3120:       /* Up Front partial dummy row: columns 2/3: (Middle) Up Front, on (Middle) (Middle) (Middle) */
    3121:       {
    3122:         const PetscInt neighbor  = 13;
    3123:         const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0); /* We may be a right boundary */
    3124:         for (ighost = ghostOffsetStart[0]; ighost < stag->nGhost[0] - ghostOffsetEnd[0]; ++ighost) {
    3125:           const PetscInt i = ighost - ghostOffsetStart[0];
    3126:           PetscInt       dLocal;
    3127:           for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                      /* Vertex  and down back edge */
    3128:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerEdge + d; /* Note increments */
    3129:           }
    3130:           for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining */
    3131:             idxGlobalAll[count] = -1;
    3132:           }
    3133:         }
    3134:       }

    3136:       if (!dummyEnd[0]) {
    3137:         /* Up Front partial dummy row: columns 3/3: Right Up Front, on Right (Middle) (Middle) */
    3138:         const PetscInt neighbor  = 14;
    3139:         const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (nextToDummyEnd[0] ? entriesPerEdge : 0); /* Neighbor may be a right boundary */
    3140:         PetscInt       i;
    3141:         for (i = 0; i < ghostOffsetEnd[0]; ++i) {
    3142:           PetscInt dLocal;
    3143:           for (d = 0, dLocal = 0; dLocal < stag->dof[0] + stag->dof[1]; ++d, ++dLocal, ++count) {                                      /* Vertex  and down back edge */
    3144:             idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerEdge + d; /* Note increments */
    3145:           }
    3146:           for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining */
    3147:             idxGlobalAll[count] = -1;
    3148:           }
    3149:         }
    3150:       } else {
    3151:         /* Right Up Front partial dummy element, on this rank */
    3152:         const PetscInt neighbor  = 13;
    3153:         const PetscInt epFaceRow = entriesPerFace * nNeighbors[neighbor][0] + (dummyEnd[0] ? entriesPerEdge : 0); /* We may be a right boundary */
    3154:         PetscInt       i, dLocal;
    3155:         i = stag->n[0];
    3156:         for (d = 0, dLocal = 0; dLocal < stag->dof[0]; ++d, ++dLocal, ++count) {                                                     /* Vertex */
    3157:           idxGlobalAll[count] = globalOffsetNeighbor[neighbor] + k * eplNeighbor[neighbor] + j * epFaceRow + i * entriesPerEdge + d; /* Note increments */
    3158:         }
    3159:         for (; dLocal < stag->entriesPerElement; ++dLocal, ++count) { /* Dummies on remaining */
    3160:           idxGlobalAll[count] = -1;
    3161:         }
    3162:         ++i;
    3163:         for (; i < stag->n[0] + ghostOffsetEnd[0]; ++i) {
    3164:           for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    3165:         }
    3166:       }
    3167:       ++j;
    3168:       /* Up Front additional dummy rows */
    3169:       for (; j < stag->n[1] + ghostOffsetEnd[1]; ++j) {
    3170:         for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    3171:           for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    3172:         }
    3173:       }
    3174:     }
    3175:     /* Front additional dummy layers */
    3176:     ++k;
    3177:     for (; k < stag->n[2] + ghostOffsetEnd[2]; ++k) {
    3178:       for (jghost = 0; jghost < stag->nGhost[1]; ++jghost) {
    3179:         for (ighost = 0; ighost < stag->nGhost[0]; ++ighost) {
    3180:           for (d = 0; d < stag->entriesPerElement; ++d, ++count) idxGlobalAll[count] = -1;
    3181:         }
    3182:       }
    3183:     }
    3184:   }

    3186:   /* Create local-to-global map (in local ordering, includes maps to -1 for dummy points) */
    3187:   PetscCall(ISLocalToGlobalMappingCreate(PetscObjectComm((PetscObject)dm), 1, stag->entriesGhost, idxGlobalAll, PETSC_OWN_POINTER, &dm->ltogmap));
    3188:   PetscFunctionReturn(PETSC_SUCCESS);
    3189: }

    3191: static PetscErrorCode DMStagComputeLocationOffsets_3d(DM dm)
    3192: {
    3193:   DM_Stag *const stag = (DM_Stag *)dm->data;
    3194:   const PetscInt epe  = stag->entriesPerElement;
    3195:   const PetscInt epr  = stag->nGhost[0] * epe;
    3196:   const PetscInt epl  = stag->nGhost[1] * epr;

    3198:   PetscFunctionBegin;
    3199:   PetscCall(PetscMalloc1(DMSTAG_NUMBER_LOCATIONS, &stag->locationOffsets));
    3200:   stag->locationOffsets[DMSTAG_BACK_DOWN_LEFT]   = 0;
    3201:   stag->locationOffsets[DMSTAG_BACK_DOWN]        = stag->locationOffsets[DMSTAG_BACK_DOWN_LEFT] + stag->dof[0];
    3202:   stag->locationOffsets[DMSTAG_BACK_DOWN_RIGHT]  = stag->locationOffsets[DMSTAG_BACK_DOWN_LEFT] + epe;
    3203:   stag->locationOffsets[DMSTAG_BACK_LEFT]        = stag->locationOffsets[DMSTAG_BACK_DOWN] + stag->dof[1];
    3204:   stag->locationOffsets[DMSTAG_BACK]             = stag->locationOffsets[DMSTAG_BACK_LEFT] + stag->dof[1];
    3205:   stag->locationOffsets[DMSTAG_BACK_RIGHT]       = stag->locationOffsets[DMSTAG_BACK_LEFT] + epe;
    3206:   stag->locationOffsets[DMSTAG_BACK_UP_LEFT]     = stag->locationOffsets[DMSTAG_BACK_DOWN_LEFT] + epr;
    3207:   stag->locationOffsets[DMSTAG_BACK_UP]          = stag->locationOffsets[DMSTAG_BACK_DOWN] + epr;
    3208:   stag->locationOffsets[DMSTAG_BACK_UP_RIGHT]    = stag->locationOffsets[DMSTAG_BACK_UP_LEFT] + epe;
    3209:   stag->locationOffsets[DMSTAG_DOWN_LEFT]        = stag->locationOffsets[DMSTAG_BACK] + stag->dof[2];
    3210:   stag->locationOffsets[DMSTAG_DOWN]             = stag->locationOffsets[DMSTAG_DOWN_LEFT] + stag->dof[1];
    3211:   stag->locationOffsets[DMSTAG_DOWN_RIGHT]       = stag->locationOffsets[DMSTAG_DOWN_LEFT] + epe;
    3212:   stag->locationOffsets[DMSTAG_LEFT]             = stag->locationOffsets[DMSTAG_DOWN] + stag->dof[2];
    3213:   stag->locationOffsets[DMSTAG_ELEMENT]          = stag->locationOffsets[DMSTAG_LEFT] + stag->dof[2];
    3214:   stag->locationOffsets[DMSTAG_RIGHT]            = stag->locationOffsets[DMSTAG_LEFT] + epe;
    3215:   stag->locationOffsets[DMSTAG_UP_LEFT]          = stag->locationOffsets[DMSTAG_DOWN_LEFT] + epr;
    3216:   stag->locationOffsets[DMSTAG_UP]               = stag->locationOffsets[DMSTAG_DOWN] + epr;
    3217:   stag->locationOffsets[DMSTAG_UP_RIGHT]         = stag->locationOffsets[DMSTAG_UP_LEFT] + epe;
    3218:   stag->locationOffsets[DMSTAG_FRONT_DOWN_LEFT]  = stag->locationOffsets[DMSTAG_BACK_DOWN_LEFT] + epl;
    3219:   stag->locationOffsets[DMSTAG_FRONT_DOWN]       = stag->locationOffsets[DMSTAG_BACK_DOWN] + epl;
    3220:   stag->locationOffsets[DMSTAG_FRONT_DOWN_RIGHT] = stag->locationOffsets[DMSTAG_FRONT_DOWN_LEFT] + epe;
    3221:   stag->locationOffsets[DMSTAG_FRONT_LEFT]       = stag->locationOffsets[DMSTAG_BACK_LEFT] + epl;
    3222:   stag->locationOffsets[DMSTAG_FRONT]            = stag->locationOffsets[DMSTAG_BACK] + epl;
    3223:   stag->locationOffsets[DMSTAG_FRONT_RIGHT]      = stag->locationOffsets[DMSTAG_FRONT_LEFT] + epe;
    3224:   stag->locationOffsets[DMSTAG_FRONT_UP_LEFT]    = stag->locationOffsets[DMSTAG_FRONT_DOWN_LEFT] + epr;
    3225:   stag->locationOffsets[DMSTAG_FRONT_UP]         = stag->locationOffsets[DMSTAG_FRONT_DOWN] + epr;
    3226:   stag->locationOffsets[DMSTAG_FRONT_UP_RIGHT]   = stag->locationOffsets[DMSTAG_FRONT_UP_LEFT] + epe;
    3227:   PetscFunctionReturn(PETSC_SUCCESS);
    3228: }

    3230: PETSC_INTERN PetscErrorCode DMStagPopulateLocalToGlobalInjective_3d(DM dm)
    3231: {
    3232:   DM_Stag *const  stag = (DM_Stag *)dm->data;
    3233:   PetscInt       *idxLocal, *idxGlobal, *globalOffsetsRecomputed;
    3234:   const PetscInt *globalOffsets;
    3235:   PetscInt        count, d, entriesPerEdge, entriesPerFace, eprGhost, eplGhost, ghostOffsetStart[3], ghostOffsetEnd[3];
    3236:   IS              isLocal, isGlobal;
    3237:   PetscBool       dummyEnd[3];

    3239:   PetscFunctionBegin;
    3240:   PetscCall(DMStagSetUpBuildGlobalOffsets_3d(dm, &globalOffsetsRecomputed)); /* note that we don't actually use all of these. An available optimization is to pass them, when available */
    3241:   globalOffsets = globalOffsetsRecomputed;
    3242:   PetscCall(PetscMalloc1(stag->entries, &idxLocal));
    3243:   PetscCall(PetscMalloc1(stag->entries, &idxGlobal));
    3244:   for (d = 0; d < 3; ++d) dummyEnd[d] = (PetscBool)(stag->lastRank[d] && stag->boundaryType[d] != DM_BOUNDARY_PERIODIC);
    3245:   entriesPerFace = stag->dof[0] + 2 * stag->dof[1] + stag->dof[2];
    3246:   entriesPerEdge = stag->dof[0] + stag->dof[1];
    3247:   eprGhost       = stag->nGhost[0] * stag->entriesPerElement; /* epr = entries per (element) row   */
    3248:   eplGhost       = stag->nGhost[1] * eprGhost;                /* epl = entries per (element) layer */
    3249:   count          = 0;
    3250:   for (d = 0; d < 3; ++d) ghostOffsetStart[d] = stag->start[d] - stag->startGhost[d];
    3251:   for (d = 0; d < 3; ++d) ghostOffsetEnd[d] = stag->startGhost[d] + stag->nGhost[d] - (stag->start[d] + stag->n[d]);
    3252:   {
    3253:     const PetscInt  neighbor    = 13;
    3254:     const PetscInt  epr         = stag->entriesPerElement * stag->n[0] + (dummyEnd[0] ? entriesPerFace : 0);                               /* We may be a right boundary */
    3255:     const PetscInt  epl         = epr * stag->n[1] + (dummyEnd[1] ? stag->n[0] * entriesPerFace + (dummyEnd[0] ? entriesPerEdge : 0) : 0); /* We may be a top boundary */
    3256:     const PetscInt  epFaceRow   = entriesPerFace * stag->n[0] + (dummyEnd[0] ? entriesPerEdge : 0);                                        /* We may be a right boundary */
    3257:     const PetscInt  start0      = 0;
    3258:     const PetscInt  start1      = 0;
    3259:     const PetscInt  start2      = 0;
    3260:     const PetscInt  startGhost0 = ghostOffsetStart[0];
    3261:     const PetscInt  startGhost1 = ghostOffsetStart[1];
    3262:     const PetscInt  startGhost2 = ghostOffsetStart[2];
    3263:     const PetscInt  endGhost0   = stag->nGhost[0] - ghostOffsetEnd[0];
    3264:     const PetscInt  endGhost1   = stag->nGhost[1] - ghostOffsetEnd[1];
    3265:     const PetscInt  endGhost2   = stag->nGhost[2] - ghostOffsetEnd[2];
    3266:     const PetscBool extra0      = dummyEnd[0];
    3267:     const PetscBool extra1      = dummyEnd[1];
    3268:     const PetscBool extra2      = dummyEnd[2];
    3269:     PetscCall(DMStagSetUpBuildScatterPopulateIdx_3d(stag, &count, idxLocal, idxGlobal, entriesPerEdge, entriesPerFace, epr, epl, eprGhost, eplGhost, epFaceRow, globalOffsets[stag->neighbors[neighbor]], start0, start1, start2, startGhost0, startGhost1, startGhost2, endGhost0, endGhost1, endGhost2, extra0, extra1, extra2));
    3270:   }
    3271:   PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)dm), stag->entries, idxLocal, PETSC_OWN_POINTER, &isLocal));
    3272:   PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)dm), stag->entries, idxGlobal, PETSC_OWN_POINTER, &isGlobal));
    3273:   {
    3274:     Vec local, global;
    3275:     PetscCall(VecCreateMPIWithArray(PetscObjectComm((PetscObject)dm), 1, stag->entries, PETSC_DECIDE, NULL, &global));
    3276:     PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, PetscMax(stag->entriesPerElement, 1), stag->entriesGhost, NULL, &local));
    3277:     PetscCall(VecScatterCreate(local, isLocal, global, isGlobal, &stag->ltog_injective));
    3278:     PetscCall(VecDestroy(&global));
    3279:     PetscCall(VecDestroy(&local));
    3280:   }
    3281:   PetscCall(ISDestroy(&isLocal));
    3282:   PetscCall(ISDestroy(&isGlobal));
    3283:   PetscCall(PetscFree(globalOffsetsRecomputed));
    3284:   PetscFunctionReturn(PETSC_SUCCESS);
    3285: }

    3287: PETSC_INTERN PetscErrorCode DMStagPopulateLocalToLocal3d_Internal(DM dm)
    3288: {
    3289:   DM_Stag *const stag = (DM_Stag *)dm->data;
    3290:   PetscInt      *idxRemap;
    3291:   PetscBool      dummyEnd[3];
    3292:   PetscInt       i, j, k, d, count, leftGhostElements, downGhostElements, backGhostElements, iOffset, jOffset, kOffset;
    3293:   PetscInt       entriesPerRowGhost, entriesPerRowColGhost;
    3294:   PetscInt       dOffset[8] = {0};

    3296:   PetscFunctionBegin;
    3297:   PetscCall(VecScatterCopy(stag->gtol, &stag->ltol));
    3298:   PetscCall(PetscMalloc1(stag->entries, &idxRemap));

    3300:   for (d = 0; d < 3; ++d) dummyEnd[d] = (PetscBool)(stag->lastRank[d] && stag->boundaryType[d] != DM_BOUNDARY_PERIODIC);
    3301:   leftGhostElements     = stag->start[0] - stag->startGhost[0];
    3302:   downGhostElements     = stag->start[1] - stag->startGhost[1];
    3303:   backGhostElements     = stag->start[2] - stag->startGhost[2];
    3304:   entriesPerRowGhost    = stag->nGhost[0] * stag->entriesPerElement;
    3305:   entriesPerRowColGhost = stag->nGhost[0] * stag->nGhost[1] * stag->entriesPerElement;
    3306:   dOffset[1]            = dOffset[0] + stag->dof[0];
    3307:   dOffset[2]            = dOffset[1] + stag->dof[1];
    3308:   dOffset[3]            = dOffset[2] + stag->dof[1];
    3309:   dOffset[4]            = dOffset[3] + stag->dof[2];
    3310:   dOffset[5]            = dOffset[4] + stag->dof[1];
    3311:   dOffset[6]            = dOffset[5] + stag->dof[2];
    3312:   dOffset[7]            = dOffset[6] + stag->dof[2];

    3314:   count = 0;
    3315:   for (k = 0; k < stag->n[2]; ++k) {
    3316:     kOffset = entriesPerRowColGhost * (backGhostElements + k);
    3317:     for (j = 0; j < stag->n[1]; ++j) {
    3318:       jOffset = entriesPerRowGhost * (downGhostElements + j);
    3319:       for (i = 0; i < stag->n[0]; ++i) {
    3320:         iOffset = stag->entriesPerElement * (leftGhostElements + i);
    3321:         // all
    3322:         for (d = 0; d < stag->entriesPerElement; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + d;
    3323:       }
    3324:       if (dummyEnd[0]) {
    3325:         iOffset = stag->entriesPerElement * (leftGhostElements + stag->n[0]);
    3326:         // back down left, back left, down left, left
    3327:         for (d = 0; d < stag->dof[0]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[0] + d;
    3328:         for (d = 0; d < stag->dof[1]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[2] + d;
    3329:         for (d = 0; d < stag->dof[1]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[4] + d;
    3330:         for (d = 0; d < stag->dof[2]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[6] + d;
    3331:       }
    3332:     }
    3333:     if (dummyEnd[1]) {
    3334:       jOffset = entriesPerRowGhost * (downGhostElements + stag->n[1]);
    3335:       for (i = 0; i < stag->n[0]; ++i) {
    3336:         iOffset = stag->entriesPerElement * (leftGhostElements + i);
    3337:         // back down left, back down, down left, down
    3338:         for (d = 0; d < stag->dof[0]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[0] + d;
    3339:         for (d = 0; d < stag->dof[1]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[1] + d;
    3340:         for (d = 0; d < stag->dof[1]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[4] + d;
    3341:         for (d = 0; d < stag->dof[2]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[5] + d;
    3342:       }
    3343:       if (dummyEnd[0]) {
    3344:         iOffset = stag->entriesPerElement * (leftGhostElements + stag->n[0]);
    3345:         // back down left, down left
    3346:         for (d = 0; d < stag->dof[0]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[0] + d;
    3347:         for (d = 0; d < stag->dof[1]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[4] + d;
    3348:       }
    3349:     }
    3350:   }
    3351:   if (dummyEnd[2]) {
    3352:     kOffset = entriesPerRowColGhost * (backGhostElements + stag->n[2]);
    3353:     for (j = 0; j < stag->n[1]; ++j) {
    3354:       jOffset = entriesPerRowGhost * (downGhostElements + j);
    3355:       for (i = 0; i < stag->n[0]; ++i) {
    3356:         iOffset = stag->entriesPerElement * (leftGhostElements + i);
    3357:         // back down left, back down, back left, back
    3358:         for (d = 0; d < stag->dof[0]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[0] + d;
    3359:         for (d = 0; d < stag->dof[1]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[1] + d;
    3360:         for (d = 0; d < stag->dof[1]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[2] + d;
    3361:         for (d = 0; d < stag->dof[2]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[3] + d;
    3362:       }
    3363:       if (dummyEnd[0]) {
    3364:         iOffset = stag->entriesPerElement * (leftGhostElements + stag->n[0]);
    3365:         // back down left, back left
    3366:         for (d = 0; d < stag->dof[0]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[0] + d;
    3367:         for (d = 0; d < stag->dof[1]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[2] + d;
    3368:       }
    3369:     }
    3370:     if (dummyEnd[1]) {
    3371:       jOffset = entriesPerRowGhost * (downGhostElements + stag->n[1]);
    3372:       for (i = 0; i < stag->n[0]; ++i) {
    3373:         iOffset = stag->entriesPerElement * (leftGhostElements + i);
    3374:         // back down left, back down
    3375:         for (d = 0; d < stag->dof[0]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[0] + d;
    3376:         for (d = 0; d < stag->dof[1]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[1] + d;
    3377:       }
    3378:       if (dummyEnd[0]) {
    3379:         iOffset = stag->entriesPerElement * (leftGhostElements + stag->n[0]);
    3380:         // back down left
    3381:         for (d = 0; d < stag->dof[0]; ++d) idxRemap[count++] = kOffset + jOffset + iOffset + dOffset[0] + d;
    3382:       }
    3383:     }
    3384:   }

    3386:   PetscCheck(count == stag->entries, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of entries computed in ltol (%" PetscInt_FMT ") is not as expected (%" PetscInt_FMT ")", count, stag->entries);

    3388:   PetscCall(VecScatterRemap(stag->ltol, idxRemap, NULL));
    3389:   PetscCall(PetscFree(idxRemap));
    3390:   PetscFunctionReturn(PETSC_SUCCESS);
    3391: }

    3393: PETSC_INTERN PetscErrorCode DMCreateMatrix_Stag_3D_AIJ_Assemble(DM dm, Mat A)
    3394: {
    3395:   PetscInt          dof[DMSTAG_MAX_STRATA], epe, stencil_width, N[3], start[3], n[3], n_extra[3];
    3396:   DMStagStencilType stencil_type;
    3397:   DMBoundaryType    boundary_type[3];

    3399:   /* This implementation gives a very dense stencil, which is likely unsuitable for
    3400:      (typical) applications which have fewer couplings */
    3401:   PetscFunctionBegin;
    3402:   PetscCall(DMStagGetDOF(dm, &dof[0], &dof[1], &dof[2], &dof[3]));
    3403:   PetscCall(DMStagGetStencilType(dm, &stencil_type));
    3404:   PetscCall(DMStagGetStencilWidth(dm, &stencil_width));
    3405:   PetscCall(DMStagGetEntriesPerElement(dm, &epe));
    3406:   PetscCall(DMStagGetCorners(dm, &start[0], &start[1], &start[2], &n[0], &n[1], &n[2], &n_extra[0], &n_extra[1], &n_extra[2]));
    3407:   PetscCall(DMStagGetGlobalSizes(dm, &N[0], &N[1], &N[2]));
    3408:   PetscCall(DMStagGetBoundaryTypes(dm, &boundary_type[0], &boundary_type[1], &boundary_type[2]));

    3410:   if (stencil_type == DMSTAG_STENCIL_NONE) {
    3411:     /* Couple all DOF at each location to each other */
    3412:     DMStagStencil *row_vertex, *row_edge_down_left, *row_edge_back_down, *row_edge_back_left, *row_face_down, *row_face_left, *row_face_back, *row_element;

    3414:     PetscCall(PetscMalloc1(dof[0], &row_vertex));
    3415:     for (PetscInt c = 0; c < dof[0]; ++c) {
    3416:       row_vertex[c].loc = DMSTAG_BACK_DOWN_LEFT;
    3417:       row_vertex[c].c   = c;
    3418:     }

    3420:     PetscCall(PetscMalloc1(dof[1], &row_edge_down_left));
    3421:     for (PetscInt c = 0; c < dof[1]; ++c) {
    3422:       row_edge_down_left[c].loc = DMSTAG_DOWN_LEFT;
    3423:       row_edge_down_left[c].c   = c;
    3424:     }

    3426:     PetscCall(PetscMalloc1(dof[1], &row_edge_back_left));
    3427:     for (PetscInt c = 0; c < dof[1]; ++c) {
    3428:       row_edge_back_left[c].loc = DMSTAG_BACK_LEFT;
    3429:       row_edge_back_left[c].c   = c;
    3430:     }

    3432:     PetscCall(PetscMalloc1(dof[1], &row_edge_back_down));
    3433:     for (PetscInt c = 0; c < dof[1]; ++c) {
    3434:       row_edge_back_down[c].loc = DMSTAG_BACK_DOWN;
    3435:       row_edge_back_down[c].c   = c;
    3436:     }

    3438:     PetscCall(PetscMalloc1(dof[2], &row_face_left));
    3439:     for (PetscInt c = 0; c < dof[2]; ++c) {
    3440:       row_face_left[c].loc = DMSTAG_LEFT;
    3441:       row_face_left[c].c   = c;
    3442:     }

    3444:     PetscCall(PetscMalloc1(dof[2], &row_face_down));
    3445:     for (PetscInt c = 0; c < dof[2]; ++c) {
    3446:       row_face_down[c].loc = DMSTAG_DOWN;
    3447:       row_face_down[c].c   = c;
    3448:     }

    3450:     PetscCall(PetscMalloc1(dof[2], &row_face_back));
    3451:     for (PetscInt c = 0; c < dof[2]; ++c) {
    3452:       row_face_back[c].loc = DMSTAG_BACK;
    3453:       row_face_back[c].c   = c;
    3454:     }

    3456:     PetscCall(PetscMalloc1(dof[3], &row_element));
    3457:     for (PetscInt c = 0; c < dof[3]; ++c) {
    3458:       row_element[c].loc = DMSTAG_ELEMENT;
    3459:       row_element[c].c   = c;
    3460:     }

    3462:     for (PetscInt ez = start[2]; ez < start[2] + n[2] + n_extra[2]; ++ez) {
    3463:       for (PetscInt ey = start[1]; ey < start[1] + n[1] + n_extra[1]; ++ey) {
    3464:         for (PetscInt ex = start[0]; ex < start[0] + n[0] + n_extra[0]; ++ex) {
    3465:           for (PetscInt c = 0; c < dof[0]; ++c) {
    3466:             row_vertex[c].i = ex;
    3467:             row_vertex[c].j = ey;
    3468:             row_vertex[c].k = ez;
    3469:           }
    3470:           PetscCall(DMStagMatSetValuesStencil(dm, A, dof[0], row_vertex, dof[0], row_vertex, NULL, INSERT_VALUES));

    3472:           if (ez < N[2]) {
    3473:             for (PetscInt c = 0; c < dof[1]; ++c) {
    3474:               row_edge_down_left[c].i = ex;
    3475:               row_edge_down_left[c].j = ey;
    3476:               row_edge_down_left[c].k = ez;
    3477:             }
    3478:             PetscCall(DMStagMatSetValuesStencil(dm, A, dof[1], row_edge_down_left, dof[1], row_edge_down_left, NULL, INSERT_VALUES));
    3479:           }

    3481:           if (ey < N[1]) {
    3482:             for (PetscInt c = 0; c < dof[1]; ++c) {
    3483:               row_edge_back_left[c].i = ex;
    3484:               row_edge_back_left[c].j = ey;
    3485:               row_edge_back_left[c].k = ez;
    3486:             }
    3487:             PetscCall(DMStagMatSetValuesStencil(dm, A, dof[1], row_edge_back_left, dof[1], row_edge_back_left, NULL, INSERT_VALUES));
    3488:           }

    3490:           if (ey < N[0]) {
    3491:             for (PetscInt c = 0; c < dof[1]; ++c) {
    3492:               row_edge_back_down[c].i = ex;
    3493:               row_edge_back_down[c].j = ey;
    3494:               row_edge_back_down[c].k = ez;
    3495:             }
    3496:             PetscCall(DMStagMatSetValuesStencil(dm, A, dof[1], row_edge_back_down, dof[1], row_edge_back_down, NULL, INSERT_VALUES));
    3497:           }

    3499:           if (ey < N[1] && ez < N[2]) {
    3500:             for (PetscInt c = 0; c < dof[2]; ++c) {
    3501:               row_face_left[c].i = ex;
    3502:               row_face_left[c].j = ey;
    3503:               row_face_left[c].k = ez;
    3504:             }
    3505:             PetscCall(DMStagMatSetValuesStencil(dm, A, dof[2], row_face_left, dof[2], row_face_left, NULL, INSERT_VALUES));
    3506:           }

    3508:           if (ex < N[0] && ez < N[2]) {
    3509:             for (PetscInt c = 0; c < dof[2]; ++c) {
    3510:               row_face_down[c].i = ex;
    3511:               row_face_down[c].j = ey;
    3512:               row_face_down[c].k = ez;
    3513:             }
    3514:             PetscCall(DMStagMatSetValuesStencil(dm, A, dof[2], row_face_down, dof[2], row_face_down, NULL, INSERT_VALUES));
    3515:           }

    3517:           if (ex < N[0] && ey < N[1]) {
    3518:             for (PetscInt c = 0; c < dof[2]; ++c) {
    3519:               row_face_back[c].i = ex;
    3520:               row_face_back[c].j = ey;
    3521:               row_face_back[c].k = ez;
    3522:             }
    3523:             PetscCall(DMStagMatSetValuesStencil(dm, A, dof[2], row_face_back, dof[2], row_face_back, NULL, INSERT_VALUES));
    3524:           }

    3526:           if (ex < N[0] && ey < N[1] && ez < N[2]) {
    3527:             for (PetscInt c = 0; c < dof[3]; ++c) {
    3528:               row_element[c].i = ex;
    3529:               row_element[c].j = ey;
    3530:               row_element[c].k = ez;
    3531:             }
    3532:             PetscCall(DMStagMatSetValuesStencil(dm, A, dof[3], row_element, dof[3], row_element, NULL, INSERT_VALUES));
    3533:           }
    3534:         }
    3535:       }
    3536:     }
    3537:     PetscCall(PetscFree(row_vertex));
    3538:     PetscCall(PetscFree(row_edge_back_left));
    3539:     PetscCall(PetscFree(row_edge_back_down));
    3540:     PetscCall(PetscFree(row_edge_down_left));
    3541:     PetscCall(PetscFree(row_face_left));
    3542:     PetscCall(PetscFree(row_face_back));
    3543:     PetscCall(PetscFree(row_face_down));
    3544:     PetscCall(PetscFree(row_element));
    3545:   } else if (stencil_type == DMSTAG_STENCIL_STAR || stencil_type == DMSTAG_STENCIL_BOX) {
    3546:     DMStagStencil *col, *row;

    3548:     PetscCall(PetscMalloc1(epe, &row));
    3549:     {
    3550:       PetscInt nrows = 0;

    3552:       for (PetscInt c = 0; c < dof[0]; ++c) {
    3553:         row[nrows].c   = c;
    3554:         row[nrows].loc = DMSTAG_BACK_DOWN_LEFT;
    3555:         ++nrows;
    3556:       }
    3557:       for (PetscInt c = 0; c < dof[1]; ++c) {
    3558:         row[nrows].c   = c;
    3559:         row[nrows].loc = DMSTAG_DOWN_LEFT;
    3560:         ++nrows;
    3561:       }
    3562:       for (PetscInt c = 0; c < dof[1]; ++c) {
    3563:         row[nrows].c   = c;
    3564:         row[nrows].loc = DMSTAG_BACK_LEFT;
    3565:         ++nrows;
    3566:       }
    3567:       for (PetscInt c = 0; c < dof[1]; ++c) {
    3568:         row[nrows].c   = c;
    3569:         row[nrows].loc = DMSTAG_BACK_DOWN;
    3570:         ++nrows;
    3571:       }
    3572:       for (PetscInt c = 0; c < dof[2]; ++c) {
    3573:         row[nrows].c   = c;
    3574:         row[nrows].loc = DMSTAG_LEFT;
    3575:         ++nrows;
    3576:       }
    3577:       for (PetscInt c = 0; c < dof[2]; ++c) {
    3578:         row[nrows].c   = c;
    3579:         row[nrows].loc = DMSTAG_DOWN;
    3580:         ++nrows;
    3581:       }
    3582:       for (PetscInt c = 0; c < dof[2]; ++c) {
    3583:         row[nrows].c   = c;
    3584:         row[nrows].loc = DMSTAG_BACK;
    3585:         ++nrows;
    3586:       }
    3587:       for (PetscInt c = 0; c < dof[3]; ++c) {
    3588:         row[nrows].c   = c;
    3589:         row[nrows].loc = DMSTAG_ELEMENT;
    3590:         ++nrows;
    3591:       }
    3592:     }

    3594:     PetscCall(PetscMalloc1(epe, &col));
    3595:     {
    3596:       PetscInt ncols = 0;

    3598:       for (PetscInt c = 0; c < dof[0]; ++c) {
    3599:         col[ncols].c   = c;
    3600:         col[ncols].loc = DMSTAG_BACK_DOWN_LEFT;
    3601:         ++ncols;
    3602:       }
    3603:       for (PetscInt c = 0; c < dof[1]; ++c) {
    3604:         col[ncols].c   = c;
    3605:         col[ncols].loc = DMSTAG_DOWN_LEFT;
    3606:         ++ncols;
    3607:       }
    3608:       for (PetscInt c = 0; c < dof[1]; ++c) {
    3609:         col[ncols].c   = c;
    3610:         col[ncols].loc = DMSTAG_BACK_LEFT;
    3611:         ++ncols;
    3612:       }
    3613:       for (PetscInt c = 0; c < dof[1]; ++c) {
    3614:         col[ncols].c   = c;
    3615:         col[ncols].loc = DMSTAG_BACK_DOWN;
    3616:         ++ncols;
    3617:       }
    3618:       for (PetscInt c = 0; c < dof[2]; ++c) {
    3619:         col[ncols].c   = c;
    3620:         col[ncols].loc = DMSTAG_LEFT;
    3621:         ++ncols;
    3622:       }
    3623:       for (PetscInt c = 0; c < dof[2]; ++c) {
    3624:         col[ncols].c   = c;
    3625:         col[ncols].loc = DMSTAG_DOWN;
    3626:         ++ncols;
    3627:       }
    3628:       for (PetscInt c = 0; c < dof[2]; ++c) {
    3629:         col[ncols].c   = c;
    3630:         col[ncols].loc = DMSTAG_BACK;
    3631:         ++ncols;
    3632:       }
    3633:       for (PetscInt c = 0; c < dof[3]; ++c) {
    3634:         col[ncols].c   = c;
    3635:         col[ncols].loc = DMSTAG_ELEMENT;
    3636:         ++ncols;
    3637:       }
    3638:     }

    3640:     for (PetscInt ez = start[2]; ez < start[2] + n[2] + n_extra[2]; ++ez) {
    3641:       for (PetscInt ey = start[1]; ey < start[1] + n[1] + n_extra[1]; ++ey) {
    3642:         for (PetscInt ex = start[0]; ex < start[0] + n[0] + n_extra[0]; ++ex) {
    3643:           for (PetscInt i = 0; i < epe; ++i) {
    3644:             row[i].i = ex;
    3645:             row[i].j = ey;
    3646:             row[i].k = ez;
    3647:           }
    3648:           for (PetscInt offset_z = -stencil_width; offset_z <= stencil_width; ++offset_z) {
    3649:             const PetscInt ez_offset = ez + offset_z;
    3650:             for (PetscInt offset_y = -stencil_width; offset_y <= stencil_width; ++offset_y) {
    3651:               const PetscInt ey_offset = ey + offset_y;
    3652:               for (PetscInt offset_x = -stencil_width; offset_x <= stencil_width; ++offset_x) {
    3653:                 const PetscInt  ex_offset     = ex + offset_x;
    3654:                 const PetscBool is_star_point = (PetscBool)(((offset_x == 0) && (offset_y == 0 || offset_z == 0)) || (offset_y == 0 && offset_z == 0));
    3655:                 /* Only set values corresponding to elements which can have non-dummy entries,
    3656:                    meaning those that map to unknowns in the global representation. In the periodic
    3657:                    case, this is the entire stencil, but in all other cases, only includes a single
    3658:                    "extra" element which is partially outside the physical domain (those points in the
    3659:                    global representation */
    3660:                 if ((stencil_type == DMSTAG_STENCIL_BOX || is_star_point) && (boundary_type[0] == DM_BOUNDARY_PERIODIC || (ex_offset < N[0] + 1 && ex_offset >= 0)) && (boundary_type[1] == DM_BOUNDARY_PERIODIC || (ey_offset < N[1] + 1 && ey_offset >= 0)) && (boundary_type[2] == DM_BOUNDARY_PERIODIC || (ez_offset < N[2] + 1 && ez_offset >= 0))) {
    3661:                   for (PetscInt i = 0; i < epe; ++i) {
    3662:                     col[i].i = ex_offset;
    3663:                     col[i].j = ey_offset;
    3664:                     col[i].k = ez_offset;
    3665:                   }
    3666:                   PetscCall(DMStagMatSetValuesStencil(dm, A, epe, row, epe, col, NULL, INSERT_VALUES));
    3667:                 }
    3668:               }
    3669:             }
    3670:           }
    3671:         }
    3672:       }
    3673:     }
    3674:     PetscCall(PetscFree(row));
    3675:     PetscCall(PetscFree(col));
    3676:   } else SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_OUTOFRANGE, "Unsupported stencil type %s", DMStagStencilTypes[stencil_type]);
    3677:   PetscCall(MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY));
    3678:   PetscCall(MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY));
    3679:   PetscFunctionReturn(PETSC_SUCCESS);
    3680: }