Lines Matching refs:origDim

51     PetscInt origDim, subDim, origNc, subNc, o, s;  in PetscSpaceView_Subspace()  local
53 PetscCall(PetscSpaceGetNumVariables(subsp->origSpace, &origDim)); in PetscSpaceView_Subspace()
59 …for (o = 0; o < origDim; o++) PetscCall(PetscViewerASCIIPrintf(viewer, " %g\n", (double)subsp->x[o… in PetscSpaceView_Subspace()
63 for (o = 0; o < origDim; o++) { in PetscSpaceView_Subspace()
96 PetscInt origDim, subDim, origNc, subNc, subNb, origNb, i, j, k, l, m, n, o; in PetscSpaceEvaluate_Subspace() local
102 PetscCall(PetscSpaceGetNumVariables(origsp, &origDim)); in PetscSpaceEvaluate_Subspace()
107 PetscCall(DMGetWorkArray(sp->dm, npoints * origDim, MPIU_REAL, &inpoints)); in PetscSpaceEvaluate_Subspace()
110 for (j = 0; j < origDim; j++) inpoints[i * origDim + j] = subsp->x[j]; in PetscSpaceEvaluate_Subspace()
112 for (j = 0; j < origDim; j++) inpoints[i * origDim + j] = 0.0; in PetscSpaceEvaluate_Subspace()
115 for (j = 0; j < origDim; j++) { in PetscSpaceEvaluate_Subspace()
116 …for (k = 0; k < subDim; k++) inpoints[i * origDim + j] += subsp->Jx[j * subDim + k] * points[i * s… in PetscSpaceEvaluate_Subspace()
119 …for (j = 0; j < PetscMin(subDim, origDim); j++) inpoints[i * origDim + j] += points[i * subDim + j… in PetscSpaceEvaluate_Subspace()
123 if (D) PetscCall(DMGetWorkArray(sp->dm, npoints * origNb * origNc * origDim, MPIU_REAL, &inD)); in PetscSpaceEvaluate_Subspace()
124 …if (H) PetscCall(DMGetWorkArray(sp->dm, npoints * origNb * origNc * origDim * origDim, MPIU_REAL, … in PetscSpaceEvaluate_Subspace()
129 PetscCall(DMGetWorkArray(sp->dm, origNc * origDim * origDim, MPIU_REAL, &phi)); in PetscSpaceEvaluate_Subspace()
136 for (k = 0; k < origNc * origDim; k++) phi[k] = 0.; in PetscSpaceEvaluate_Subspace()
139 …for (l = 0; l < origNc * origDim * origDim; l++) phi[l] += inH[(j * origNb + k) * origNc * origDim in PetscSpaceEvaluate_Subspace()
144 for (m = 0; m < origDim; m++) { in PetscSpaceEvaluate_Subspace()
146 …0; o < origDim; o++) psi[(k * subDim + l) * subDim + n] += subsp->Jx[m * subDim + l] * subsp->Jx[o… in PetscSpaceEvaluate_Subspace()
153 for (l = 0; l < PetscMin(subDim, origDim); l++) { in PetscSpaceEvaluate_Subspace()
154 …for (m = 0; m < PetscMin(subDim, origDim); m++) psi[(k * subDim + l) * subDim + m] += phi[(k * ori… in PetscSpaceEvaluate_Subspace()
171 PetscCall(DMRestoreWorkArray(sp->dm, subNc * origDim, MPIU_REAL, &psi)); in PetscSpaceEvaluate_Subspace()
172 PetscCall(DMRestoreWorkArray(sp->dm, origNc * origDim, MPIU_REAL, &phi)); in PetscSpaceEvaluate_Subspace()
173 PetscCall(DMRestoreWorkArray(sp->dm, npoints * origNb * origNc * origDim, MPIU_REAL, &inH)); in PetscSpaceEvaluate_Subspace()
178 PetscCall(DMGetWorkArray(sp->dm, origNc * origDim, MPIU_REAL, &phi)); in PetscSpaceEvaluate_Subspace()
185 for (k = 0; k < origNc * origDim; k++) phi[k] = 0.; in PetscSpaceEvaluate_Subspace()
188 …for (l = 0; l < origNc * origDim; l++) phi[l] += inD[(j * origNb + k) * origNc * origDim + l] * su… in PetscSpaceEvaluate_Subspace()
193 …for (m = 0; m < origDim; m++) psi[k * subDim + l] += subsp->Jx[m * subDim + l] * phi[k * origDim +… in PetscSpaceEvaluate_Subspace()
198 … for (l = 0; l < PetscMin(subDim, origDim); l++) psi[k * subDim + l] += phi[k * origDim + l]; in PetscSpaceEvaluate_Subspace()
214 PetscCall(DMRestoreWorkArray(sp->dm, subNc * origDim, MPIU_REAL, &psi)); in PetscSpaceEvaluate_Subspace()
215 PetscCall(DMRestoreWorkArray(sp->dm, origNc * origDim, MPIU_REAL, &phi)); in PetscSpaceEvaluate_Subspace()
216 PetscCall(DMRestoreWorkArray(sp->dm, npoints * origNb * origNc * origDim, MPIU_REAL, &inD)); in PetscSpaceEvaluate_Subspace()
249 PetscCall(DMRestoreWorkArray(sp->dm, npoints * origDim, MPIU_REAL, &inpoints)); in PetscSpaceEvaluate_Subspace()
281 PetscInt origDim, subDim, origNc, subNc, origNb, subNb, f, i, j, numPoints, offset; in PetscSpaceSetUp_Subspace() local
296 PetscCall(PetscSpaceGetNumVariables(origSpace, &origDim)); in PetscSpaceSetUp_Subspace()
312 …PetscCall(PetscMalloc3(numPoints * origDim, &allPoints, numPoints * origNc, &allWeights, numPoints… in PetscSpaceSetUp_Subspace()
323 for (i = 0; i < origDim; i++) allPoints[origDim * offset + i] = x[i]; in PetscSpaceSetUp_Subspace()
325 for (i = 0; i < origDim; i++) allPoints[origDim * offset + i] = 0.0; in PetscSpaceSetUp_Subspace()
328 for (i = 0; i < origDim; i++) { in PetscSpaceSetUp_Subspace()
329 … for (j = 0; j < subDim; j++) allPoints[origDim * offset + i] += Jx[i * subDim + j] * qp[j]; in PetscSpaceSetUp_Subspace()
332 for (i = 0; i < PetscMin(subDim, origDim); i++) allPoints[origDim * offset + i] += qp[i]; in PetscSpaceSetUp_Subspace()
393 PetscInt subDim, origDim, i, j; in PetscSpacePolynomialGetTensor_Subspace() local
396 PetscCall(PetscSpaceGetNumVariables(subsp->origSpace, &origDim)); in PetscSpacePolynomialGetTensor_Subspace()
399 for (i = 0; i < origDim; i++) { in PetscSpacePolynomialGetTensor_Subspace()
408 for (i = 0; i < origDim; i++) nnz += (subsp->Jx[i * subDim + j] != 0.); in PetscSpacePolynomialGetTensor_Subspace()
450 PetscInt origDim, subDim, origNc, subNc, subNb; in PetscSpaceCreateSubspace() local
463 PetscCall(PetscSpaceGetNumVariables(origSpace, &origDim)); in PetscSpaceCreateSubspace()
491 PetscCall(PetscMalloc1(origDim, &subsp->x_alloc)); in PetscSpaceCreateSubspace()
492 PetscCall(PetscArraycpy(subsp->x_alloc, x, origDim)); in PetscSpaceCreateSubspace()
496 PetscCall(PetscMalloc1(origDim * subDim, &subsp->Jx_alloc)); in PetscSpaceCreateSubspace()
497 PetscCall(PetscArraycpy(subsp->Jx_alloc, Jx, origDim * subDim)); in PetscSpaceCreateSubspace()