Lines Matching refs:bases
213 PetscInt *bases, *ldims, base, x_t, y_t, z_t, s_t, count, s_x, s_y, s_z; in DMSetUp_DA_3D() local
461 PetscCall(PetscMalloc2(size + 1, &bases, size, &ldims)); in DMSetUp_DA_3D()
463 bases[0] = 0; in DMSetUp_DA_3D()
464 for (i = 1; i <= size; i++) bases[i] = ldims[i - 1]; in DMSetUp_DA_3D()
465 for (i = 1; i <= size; i++) bases[i] += bases[i - 1]; in DMSetUp_DA_3D()
466 base = bases[rank] * dof; in DMSetUp_DA_3D()
818 s_t = bases[n0] + x_t * y_t * z_t - (s_y - i) * x_t - s_x - (s_z - k - 1) * x_t * y_t; in DMSetUp_DA_3D()
819 … if (twod && (s_t < 0)) s_t = bases[n0] + x_t * y_t * z_t - (s_y - i) * x_t - s_x; /* 2D case */ in DMSetUp_DA_3D()
826 s_t = bases[n1] + x_t * y_t * z_t - (s_y + 1 - i) * x_t - (s_z - k - 1) * x_t * y_t; in DMSetUp_DA_3D()
827 … if (twod && (s_t < 0)) s_t = bases[n1] + x_t * y_t * z_t - (s_y + 1 - i) * x_t; /* 2D case */ in DMSetUp_DA_3D()
834 s_t = bases[n2] + x_t * y_t * z_t - (s_y + 1 - i) * x_t - (s_z - k - 1) * x_t * y_t; in DMSetUp_DA_3D()
835 … if (twod && (s_t < 0)) s_t = bases[n2] + x_t * y_t * z_t - (s_y + 1 - i) * x_t; /* 2D case */ in DMSetUp_DA_3D()
845 s_t = bases[n3] + (i + 1) * x_t - s_x + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
846 …if (twod && (s_t < 0)) s_t = bases[n3] + (i + 1) * x_t - s_x + x_t * y_t * z_t - x_t * y_t; /* 2D … in DMSetUp_DA_3D()
854 s_t = bases[n4] + i * x_t + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
855 … if (twod && (s_t < 0)) s_t = bases[n4] + i * x_t + x_t * y_t * z_t - x_t * y_t; /* 2D case */ in DMSetUp_DA_3D()
858 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + j + x * i + (s_z - k - 1) * x * y; in DMSetUp_DA_3D()
865 s_t = bases[n5] + i * x_t + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
866 … if (twod && (s_t < 0)) s_t = bases[n5] + i * x_t + x_t * y_t * z_t - x_t * y_t; /* 2D case */ in DMSetUp_DA_3D()
876 s_t = bases[n6] + i * x_t - s_x + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
877 …if (twod && (s_t < 0)) s_t = bases[n6] + i * x_t - s_x + x_t * y_t * z_t - x_t * y_t; /* 2D case */ in DMSetUp_DA_3D()
884 s_t = bases[n7] + (i - 1) * x_t + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
885 …if (twod && (s_t < 0)) s_t = bases[n7] + (i - 1) * x_t + x_t * y_t * z_t - x_t * y_t; /* 2D case */ in DMSetUp_DA_3D()
892 s_t = bases[n8] + (i - 1) * x_t + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
893 …if (twod && (s_t < 0)) s_t = bases[n8] + (i - 1) * x_t + x_t * y_t * z_t - x_t * y_t; /* 2D case */ in DMSetUp_DA_3D()
906 s_t = bases[n9] - (s_y - i) * x_t - s_x + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
913 s_t = bases[n10] - (s_y + 1 - i) * x_t + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
916 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + j + k * x * y + (s_y - i) * x; in DMSetUp_DA_3D()
922 s_t = bases[n11] - (s_y + 1 - i) * x_t + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
932 s_t = bases[n12] + (i + 1) * x_t - s_x + k * x_t * y_t; in DMSetUp_DA_3D()
935 for (j = 0; j < s_x; j++) idx[nn++] = bases[rank] + s_x - j - 1 + k * x * y + i * x; in DMSetUp_DA_3D()
939 s_t = bases[rank] + i * x + k * x * y; in DMSetUp_DA_3D()
946 s_t = bases[n14] + i * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
949 for (j = 0; j < s_x; j++) idx[nn++] = bases[rank] + x - j - 1 + k * x * y + i * x; in DMSetUp_DA_3D()
958 s_t = bases[n15] + i * x_t - s_x + k * x_t * y_t; in DMSetUp_DA_3D()
965 s_t = bases[n16] + (i - 1) * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
968 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + j + k * x * y + (y - i) * x; in DMSetUp_DA_3D()
974 s_t = bases[n17] + (i - 1) * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
987 s_t = bases[n18] - (s_y - i) * x_t - s_x + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
988 …if (twod && (s_t >= M * N * P)) s_t = bases[n18] - (s_y - i) * x_t - s_x + x_t * y_t; /* 2d case */ in DMSetUp_DA_3D()
995 s_t = bases[n19] - (s_y + 1 - i) * x_t + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
996 … if (twod && (s_t >= M * N * P)) s_t = bases[n19] - (s_y + 1 - i) * x_t + x_t * y_t; /* 2d case */ in DMSetUp_DA_3D()
1003 s_t = bases[n20] - (s_y + 1 - i) * x_t + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
1004 … if (twod && (s_t >= M * N * P)) s_t = bases[n20] - (s_y + 1 - i) * x_t + x_t * y_t; /* 2d case */ in DMSetUp_DA_3D()
1014 s_t = bases[n21] + (i + 1) * x_t - s_x + k * x_t * y_t; in DMSetUp_DA_3D()
1015 if (twod && (s_t >= M * N * P)) s_t = bases[n21] + (i + 1) * x_t - s_x; /* 2d case */ in DMSetUp_DA_3D()
1023 s_t = bases[n22] + i * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1024 if (twod && (s_t >= M * N * P)) s_t = bases[n22] + i * x_t; /* 2d case */ in DMSetUp_DA_3D()
1027 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + j + (z - k - 1) * x * y + i * x; in DMSetUp_DA_3D()
1034 s_t = bases[n23] + i * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1035 if (twod && (s_t >= M * N * P)) s_t = bases[n23] + i * x_t; /* 2d case */ in DMSetUp_DA_3D()
1045 s_t = bases[n24] + i * x_t - s_x + k * x_t * y_t; in DMSetUp_DA_3D()
1046 if (twod && (s_t >= M * N * P)) s_t = bases[n24] + i * x_t - s_x; /* 2d case */ in DMSetUp_DA_3D()
1053 s_t = bases[n25] + (i - 1) * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1054 if (twod && (s_t >= M * N * P)) s_t = bases[n25] + (i - 1) * x_t; /* 2d case */ in DMSetUp_DA_3D()
1061 s_t = bases[n26] + (i - 1) * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1062 if (twod && (s_t >= M * N * P)) s_t = bases[n26] + (i - 1) * x_t; /* 2d case */ in DMSetUp_DA_3D()
1109 s_t = bases[n0] + x_t * y_t * z_t - (s_y - i) * x_t - s_x - (s_z - k - 1) * x_t * y_t; in DMSetUp_DA_3D()
1118 s_t = bases[n1] + x_t * y_t * z_t - (s_y + 1 - i) * x_t - (s_z - k - 1) * x_t * y_t; in DMSetUp_DA_3D()
1127 s_t = bases[n2] + x_t * y_t * z_t - (s_y + 1 - i) * x_t - (s_z - k - 1) * x_t * y_t; in DMSetUp_DA_3D()
1139 s_t = bases[n3] + (i + 1) * x_t - s_x + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
1149 s_t = bases[n4] + i * x_t + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
1153 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + j + x * i + (s_z - k) * x * y; in DMSetUp_DA_3D()
1163 s_t = bases[n5] + i * x_t + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
1175 s_t = bases[n6] + i * x_t - s_x + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
1184 s_t = bases[n7] + (i - 1) * x_t + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
1193 s_t = bases[n8] + (i - 1) * x_t + x_t * y_t * z_t - (s_z - k) * x_t * y_t; in DMSetUp_DA_3D()
1208 s_t = bases[n9] - (s_y - i) * x_t - s_x + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
1217 s_t = bases[n10] - (s_y + 1 - i) * x_t + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
1221 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + j + k * x * y + (s_y - i + 1) * x; in DMSetUp_DA_3D()
1230 s_t = bases[n11] - (s_y + 1 - i) * x_t + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
1242 s_t = bases[n12] + (i + 1) * x_t - s_x + k * x_t * y_t; in DMSetUp_DA_3D()
1246 for (j = 0; j < s_x; j++) idx[nn++] = bases[rank] + s_x - j - 1 + k * x * y + i * x + 1; in DMSetUp_DA_3D()
1253 s_t = bases[rank] + i * x + k * x * y; in DMSetUp_DA_3D()
1260 s_t = bases[n14] + i * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1264 for (j = 0; j < s_x; j++) idx[nn++] = bases[rank] + x - j - 1 + k * x * y + i * x - 1; in DMSetUp_DA_3D()
1276 s_t = bases[n15] + i * x_t - s_x + k * x_t * y_t; in DMSetUp_DA_3D()
1285 s_t = bases[n16] + (i - 1) * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1289 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + j + k * x * y + (y - i - 1) * x; in DMSetUp_DA_3D()
1298 s_t = bases[n17] + (i - 1) * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1313 s_t = bases[n18] - (s_y - i) * x_t - s_x + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
1322 s_t = bases[n19] - (s_y + 1 - i) * x_t + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
1331 s_t = bases[n20] - (s_y + 1 - i) * x_t + (k + 1) * x_t * y_t; in DMSetUp_DA_3D()
1343 s_t = bases[n21] + (i + 1) * x_t - s_x + k * x_t * y_t; in DMSetUp_DA_3D()
1353 s_t = bases[n22] + i * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1357 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + j + (z - k - 2) * x * y + i * x; in DMSetUp_DA_3D()
1367 s_t = bases[n23] + i * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1379 s_t = bases[n24] + i * x_t - s_x + k * x_t * y_t; in DMSetUp_DA_3D()
1388 s_t = bases[n25] + (i - 1) * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1397 s_t = bases[n26] + (i - 1) * x_t + k * x_t * y_t; in DMSetUp_DA_3D()
1411 PetscCall(PetscFree2(bases, ldims)); in DMSetUp_DA_3D()