impls/ptrimmed/spaceptrimmed.c

130   p in [0, npoints), i in [0, pdim), c in [0, Nc)
134 static PetscErrorCode PetscSpaceEvaluate_Ptrimmed(PetscSpace sp, PetscInt npoints, const PetscReal points[], PetscReal B[], PetscReal D[], PetscReal H[])
148     PetscCall(PetscSpaceEvaluate(sp, npoints, points, B, D, H));
165   PetscCall(DMGetWorkArray(dm, Nb * Nf * Njet * npoints, MPIU_REAL, &eval));
166   PetscCall(PetscDTPTrimmedEvalJet(dim, npoints, points, degree, f, jet, eval));
172     PetscInt b_strr = Nf * Njet * npoints;
173     PetscInt v_strr = Njet * npoints;
178         for (PetscInt p = 0; p < npoints; p++) B[p * p_strl + b * b_strl + v * v_strl] = eval[b * b_strr + v * v_strr + p * p_strr];
188     PetscInt b_strr = Nf * Njet * npoints;
189     PetscInt v_strr = Njet * npoints;
190     PetscInt d_strr = npoints;
196           for (PetscInt p = 0; p < npoints; p++) D[p * p_strl + b * b_strl + v * v_strl + d * d_strl] = eval[b * b_strr + v * v_strr + (1 + d) * d_strr + p * p_strr];
208     PetscInt b_strr = Nf * Njet * npoints;
209     PetscInt v_strr = Njet * npoints;
210     PetscInt j_strr = npoints;
225             for (PetscInt p = 0; p < npoints; p++) H[p * p_strl + b * b_strl + v * v_strl + d1 * d1_strl + d2 * d2_strl] = eval[b * b_strr + v * v_strr + j * j_strr + p * p_strr];
232   PetscCall(DMRestoreWorkArray(dm, Nb * Nf * Njet * npoints, MPIU_REAL, &eval));