impls/tensor/spacetensor.c

31   PetscInt           Ns, Nc, i, Nv, deg;  in PetscSpaceSetFromOptions_Tensor()  local
35   PetscCall(PetscSpaceGetNumVariables(sp, &Nv));  in PetscSpaceSetFromOptions_Tensor()
36   if (!Nv) PetscFunctionReturn(PETSC_SUCCESS);  in PetscSpaceSetFromOptions_Tensor()
54   Ns = (Ns == PETSC_DEFAULT) ? PetscMax(Nv, 1) : Ns;  in PetscSpaceSetFromOptions_Tensor()
59 …PetscCheck(Ns >= 0 && (Nv <= 0 || Ns != 0), PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_OUTOFR…  in PetscSpaceSetFromOptions_Tensor()
60 …Nv <= 0 || Ns <= Nv, PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_OUTOFRANGE, "Cannot have a te…  in PetscSpaceSetFromOptions_Tensor()
63     PetscInt   Nvs = Nv / Ns;  in PetscSpaceSetFromOptions_Tensor()
67 …Nv % Ns == 0, PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_WRONG, "Cannot use %" PetscInt_FMT "…  in PetscSpaceSetFromOptions_Tensor()
133   PetscInt           Nc, Nv, Ns;  in PetscSpaceSetUp_Tensor()  local
140   PetscCall(PetscSpaceGetNumVariables(sp, &Nv));  in PetscSpaceSetUp_Tensor()
144     Ns = Nv;  in PetscSpaceSetUp_Tensor()
152 …Nv <= 0 || Ns <= Nv, PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_OUTOFRANGE, "Cannot have a te…  in PetscSpaceSetUp_Tensor()
164       PetscInt Nvs = Nv / Ns;  in PetscSpaceSetUp_Tensor()
167 …Nv % Ns == 0, PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_WRONG, "Cannot use %" PetscInt_FMT "…  in PetscSpaceSetUp_Tensor()
197 …Nv, PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_WRONG, "Sum of subspace variables %" PetscInt_…  in PetscSpaceSetUp_Tensor()
202       PetscInt    Nv      = sp->Nv;  in PetscSpaceSetUp_Tensor()  local
218       PetscCall(PetscSpaceSetNumVariables(subsp, Nv));  in PetscSpaceSetUp_Tensor()
266     for (d = 0; d < sp->Nv; ++d) PetscCall(PetscSpaceDestroy(&tens->heightsubspaces[d]));  in PetscSpaceDestroy_Tensor()
303   PetscInt           Nv   = sp->Nv;  in PetscSpaceEvaluate_Tensor()  local
316   PetscCall(DMGetWorkArray(dm, npoints * Nv, MPIU_REAL, &lpoints));  in PetscSpaceEvaluate_Tensor()
318   if (D || H) PetscCall(DMGetWorkArray(dm, npoints * pdim * Nc * Nv, MPIU_REAL, &sD));  in PetscSpaceEvaluate_Tensor()
319   if (H) PetscCall(DMGetWorkArray(dm, npoints * pdim * Nc * Nv * Nv, MPIU_REAL, &sH));  in PetscSpaceEvaluate_Tensor()
324     for (PetscInt i = 0; i < npoints * pdim * Nc * Nv; i++) D[i] = 1.;  in PetscSpaceEvaluate_Tensor()
327     for (PetscInt i = 0; i < npoints * pdim * Nc * Nv * Nv; i++) H[i] = 1.;  in PetscSpaceEvaluate_Tensor()
336 …MT ", s %" PetscInt_FMT ", vstep %" PetscInt_FMT ", spdim %" PetscInt_FMT, Nv, Ns, pdim, s, vstep,…  in PetscSpaceEvaluate_Tensor()
337 …_FMT ", s %" PetscInt_FMT ", cstep %" PetscInt_FMT ", sNc %" PetscInt_FMT, Nv, Ns, Nc, s, cstep, s…  in PetscSpaceEvaluate_Tensor()
341       for (PetscInt i = 0; i < sNv; i++) lpoints[p * sNv + i] = points[p * Nv + d + i];  in PetscSpaceEvaluate_Tensor()
374                   for (PetscInt der = 0; der < Nv; der++) {  in PetscSpaceEvaluate_Tensor()
376 …for (PetscInt p = 0; p < npoints; p++) D[((pdim * p + i) * Nc + c) * Nv + der] *= sD[((spdim * p +…  in PetscSpaceEvaluate_Tensor()
378 …for (PetscInt p = 0; p < npoints; p++) D[((pdim * p + i) * Nc + c) * Nv + der] *= sB[(spdim * p + …  in PetscSpaceEvaluate_Tensor()
399                   for (PetscInt der = 0; der < Nv; der++) {  in PetscSpaceEvaluate_Tensor()
400                     for (PetscInt der2 = 0; der2 < Nv; der2++) {  in PetscSpaceEvaluate_Tensor()
402 …or (PetscInt p = 0; p < npoints; p++) H[(((pdim * p + i) * Nc + c) * Nv + der) * Nv + der2] *= sH[…  in PetscSpaceEvaluate_Tensor()
404 …or (PetscInt p = 0; p < npoints; p++) H[(((pdim * p + i) * Nc + c) * Nv + der) * Nv + der2] *= sD[…  in PetscSpaceEvaluate_Tensor()
406 …or (PetscInt p = 0; p < npoints; p++) H[(((pdim * p + i) * Nc + c) * Nv + der) * Nv + der2] *= sD[…  in PetscSpaceEvaluate_Tensor()
408 …or (PetscInt p = 0; p < npoints; p++) H[(((pdim * p + i) * Nc + c) * Nv + der) * Nv + der2] *= sB[…  in PetscSpaceEvaluate_Tensor()
423   if (H) PetscCall(DMRestoreWorkArray(dm, npoints * pdim * Nc * Nv * Nv, MPIU_REAL, &sH));  in PetscSpaceEvaluate_Tensor()
424   if (D || H) PetscCall(DMRestoreWorkArray(dm, npoints * pdim * Nc * Nv, MPIU_REAL, &sD));  in PetscSpaceEvaluate_Tensor()
426   PetscCall(DMRestoreWorkArray(dm, npoints * Nv, MPIU_REAL, &lpoints));  in PetscSpaceEvaluate_Tensor()