| /petsc/src/dm/impls/stag/tests/ |
| H A D | ex30.c | 66 PetscInt startx, starty, startz, N[3], nx, ny, nz, ex, ey, ez, d; in CreateMat() local
97 for (ez = startz; ez < startz + nz; ++ez) { in CreateMat()
106 row.k = ez; in CreateMat()
117 row.k = ez; in CreateMat()
122 if (ez == N[2] - 1) { in CreateMat()
128 row.k = ez; in CreateMat()
141 row.k = ez; in CreateMat()
153 row.k = ez; in CreateMat()
157 if (ez == 0) { in CreateMat()
161 col[0].k = ez; in CreateMat()
[all …]
|
| H A D | ex8.c | 20 PetscInt startx, starty, startz, nx, ny, nz, ex, ey, ez, nExtrax, nExtray, nExtraz; in main() local
52 for (ez = startz; ez < startz + nz + nExtraz; ++ez) { in main()
55 arrSol[ez][ey][ex][iux] = 2 * PRESSURE_CONST; in main()
56 arrRHS[ez][ey][ex][iux] = 0.0; in main()
57 arrSol[ez][ey][ex][iuy] = 2 * PRESSURE_CONST; in main()
58 arrRHS[ez][ey][ex][iuy] = 0.0; in main()
59 arrSol[ez][ey][ex][iuz] = 2 * PRESSURE_CONST; in main()
60 arrRHS[ez][ey][ex][iuz] = 0.0; in main()
61 if (ex < startx + nx && ey < starty + ny && ez < startz + nz) { in main()
62 arrSol[ez][ey][ex][ip] = PRESSURE_CONST; in main()
[all …]
|
| H A D | ex23.c | 9 PetscInt ex, ey, ez, n[3], start[3], nExtra[3], iNext, iPrev, iCenter, d, round; in main() local
39 for (ez = start[2]; ez < start[2] + n[2]; ++ez) { in main()
40 cArrZ[ez][iPrev] *= 1.1; in main()
41 cArrZ[ez][iNext] = cArrZ[ez][iPrev] + 0.1; in main()
42 cArrZ[ez][iCenter] = 0.5 * (cArrZ[ez][iPrev] + cArrZ[ez][iNext]); in main()
56 for (ez = start[2]; ez < start[2] + n[2]; ++ez) { in main()
57 cArrZ[ez][iPrev] = 2.0; in main()
58 cArrZ[ez][iNext] = 2.0; in main()
59 cArrZ[ez][iCenter] = 2.0; in main()
|
| H A D | ex40.c | 842 for (PetscInt ez = start[2]; ez < start[2] + n[2] + n_extra[2]; ++ez) { in FormFunction3DNoCoupling() local
851 row.k = ez; in FormFunction3DNoCoupling()
858 if (ez < N[2]) { in FormFunction3DNoCoupling()
865 row.k = ez; in FormFunction3DNoCoupling()
880 row.k = ez; in FormFunction3DNoCoupling()
895 row.k = ez; in FormFunction3DNoCoupling()
903 if (ex < N[0] && ez < N[2]) { in FormFunction3DNoCoupling()
910 row.k = ez; in FormFunction3DNoCoupling()
918 if (ey < N[1] && ez < N[2]) { in FormFunction3DNoCoupling()
925 row.k = ez; in FormFunction3DNoCoupling()
[all …]
|
| /petsc/src/dm/impls/stag/tutorials/ |
| H A D | ex3.c | 170 PetscInt startx, starty, startz, N[3], nx, ny, nz, ex, ey, ez, d; in CreateSystem() local
203 …for (ez = startz; ez < startz + nz; ++ez) { /* With DMStag, always iterate x fastest, y second fas… in CreateSystem()
213 row.k = ez; in CreateSystem()
217 …valRhs = uxRef(arrCoord[ez][ey][ex][icux_right[0]], arrCoord[ez][ey][ex][icux_right[1]], arrCoord[… in CreateSystem()
227 row.k = ez; in CreateSystem()
231 …valRhs = uyRef(arrCoord[ez][ey][ex][icuy_up[0]], arrCoord[ez][ey][ex][icuy_up[1]], arrCoord[ez][ey… in CreateSystem()
234 if (ez == N[2] - 1) { in CreateSystem()
241 row.k = ez; in CreateSystem()
245 …valRhs = uzRef(arrCoord[ez][ey][ex][icuz_front[0]], arrCoord[ez][ey][ex][icuz_front[1]], arrCoord[… in CreateSystem()
257 row.k = ez; in CreateSystem()
[all …]
|
| H A D | ex6.c | 228 PetscInt N[3], ex, ey, ez, startx, starty, startz, nx, ny, nz, extrax, extray, extraz; in CreateLame() local
266 for (ez = startz; ez < startz + nz; ++ez) { in CreateLame()
273 pos.k = ez; in CreateLame()
279 pos.k = ez; in CreateLame()
287 for (ez = startz; ez < startz + nz + extraz; ++ez) { in CreateLame()
295 pos.k = ez; in CreateLame()
303 pos.k = ez; in CreateLame()
308 if (ez < N[2]) { in CreateLame()
311 pos.k = ez; in CreateLame()
450 PetscInt ex, ey, ez, startx, starty, startz, nx, ny, nz; in UpdateVelocity_3d() local
[all …]
|
| H A D | ex4.c | 994 PetscInt ex, ey, ez, startx, starty, startz, nx, ny, nz; in CreateSystem3d() local
1045 for (ez = startz; ez < startz + nz; ++ez) { in CreateSystem3d()
1052 const PetscBool back_boundary = (PetscBool)(ez == 0); in CreateSystem3d()
1053 const PetscBool front_boundary = (PetscBool)(ez == N[2] - 1); in CreateSystem3d()
1068 row.k = ez; in CreateSystem3d()
1081 row.k = ez; in CreateSystem3d()
1104 eta_point[0].k = ez; in CreateSystem3d()
1109 eta_point[1].k = ez; in CreateSystem3d()
1114 eta_point[2].k = ez; in CreateSystem3d()
1119 eta_point[3].k = ez; in CreateSystem3d()
[all …]
|
| /petsc/src/dm/impls/stag/ |
| H A D | stagda.c | 375 PetscInt ex, ey, ez; in DMStagTransferCoordinatesToDMDA() local
383 for (ez = start[2]; ez < start[2] + n[2] + extraPoint[2]; ++ez) { in DMStagTransferCoordinatesToDMDA()
386 for (d = 0; d < 3; ++d) cArrDa[ez][ey][ex][d] = cArrStag[ez][ey][ex][slot + d]; in DMStagTransferCoordinatesToDMDA()
394 for (ez = start[2]; ez < start[2] + n[2] + extraPoint[2]; ++ez) { in DMStagTransferCoordinatesToDMDA()
397 cArrDa[ez][ey][ex][0] = cArrX[ex][0]; in DMStagTransferCoordinatesToDMDA()
398 cArrDa[ez][ey][ex][1] = cArrY[ey][0]; in DMStagTransferCoordinatesToDMDA()
399 cArrDa[ez][ey][ex][2] = cArrZ[ez][0]; in DMStagTransferCoordinatesToDMDA()
|
| H A D | stag3d.c | 3462 for (PetscInt ez = start[2]; ez < start[2] + n[2] + n_extra[2]; ++ez) { in DMCreateMatrix_Stag_3D_AIJ_Assemble() local
3468 row_vertex[c].k = ez; in DMCreateMatrix_Stag_3D_AIJ_Assemble()
3472 if (ez < N[2]) { in DMCreateMatrix_Stag_3D_AIJ_Assemble()
3476 row_edge_down_left[c].k = ez; in DMCreateMatrix_Stag_3D_AIJ_Assemble()
3485 row_edge_back_left[c].k = ez; in DMCreateMatrix_Stag_3D_AIJ_Assemble()
3494 row_edge_back_down[c].k = ez; in DMCreateMatrix_Stag_3D_AIJ_Assemble()
3499 if (ey < N[1] && ez < N[2]) { in DMCreateMatrix_Stag_3D_AIJ_Assemble()
3503 row_face_left[c].k = ez; in DMCreateMatrix_Stag_3D_AIJ_Assemble()
3508 if (ex < N[0] && ez < N[2]) { in DMCreateMatrix_Stag_3D_AIJ_Assemble()
3512 row_face_down[c].k = ez; in DMCreateMatrix_Stag_3D_AIJ_Assemble()
[all …]
|
| /petsc/src/dm/impls/plex/ |
| H A D | plexcreate.c | 1321 const PetscInt ez = firstZEdge + 4 * (ey * numXEdges + ex); in DMPlexCreateSquareMesh_Simplex_CrissCross() local
1332 … const PetscInt cones[] = {ez, e0, ez + 1, ez + 1, e1, ez + 2, ez + 2, e2, ez + 3, ez + 3, e3, ez}; in DMPlexCreateSquareMesh_Simplex_CrissCross()
1338 PetscInt edge = ez + c; in DMPlexCreateSquareMesh_Simplex_CrissCross()
1570 PetscInt c, f, fx, fy, fz, e, ex, ey, ez; in DMPlexCreateCubeMesh_Internal() local
1720 for (ez = 0; ez < numZEdges; ez++) { in DMPlexCreateCubeMesh_Internal()
1721 const PetscInt edge = firstZEdge + (vy * numXVertices + vx) * numZEdges + ez; in DMPlexCreateCubeMesh_Internal()
1722 const PetscInt vertexB = firstVertex + (ez * numYVertices + vy) * numXVertices + vx; in DMPlexCreateCubeMesh_Internal()
1723 …const PetscInt vertexT = firstVertex + (((ez + 1) % numZVertices) * numYVertices + vy) * numXVerti… in DMPlexCreateCubeMesh_Internal()
1734 … if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerRight)); in DMPlexCreateCubeMesh_Internal()
1738 … if (ez == numZEdges - 1) PetscCall(DMSetLabelValue(dm, "marker", cone[1], markerLeft)); in DMPlexCreateCubeMesh_Internal()
[all …]
|