147923291SMatthew G. Knepley #include <petsc/private/dmpleximpl.h> /*I "petscdmplex.h" I*/ 247923291SMatthew G. Knepley 38c6c5593SMatthew G. Knepley #include <petsc/private/petscfeimpl.h> 48c6c5593SMatthew G. Knepley 5496733e2SMatthew G. Knepley static PetscErrorCode DMProjectPoint_Func_Private(DM dm, PetscReal time, PetscFECellGeom *fegeom, PetscFVCellGeom *fvgeom, PetscBool isFE[], PetscDualSpace sp[], 68c6c5593SMatthew G. Knepley PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, 78c6c5593SMatthew G. Knepley PetscScalar values[]) 847923291SMatthew G. Knepley { 98c6c5593SMatthew G. Knepley PetscDS prob; 10496733e2SMatthew G. Knepley PetscInt Nf, *Nc, f, spDim, d, c, v; 118c6c5593SMatthew G. Knepley PetscErrorCode ierr; 128c6c5593SMatthew G. Knepley 138c6c5593SMatthew G. Knepley PetscFunctionBeginHot; 148c6c5593SMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 158c6c5593SMatthew G. Knepley ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr); 16496733e2SMatthew G. Knepley ierr = PetscDSGetComponents(prob, &Nc);CHKERRQ(ierr); 178c6c5593SMatthew G. Knepley /* Get values for closure */ 188c6c5593SMatthew G. Knepley for (f = 0, v = 0; f < Nf; ++f) { 198c6c5593SMatthew G. Knepley void * const ctx = ctxs ? ctxs[f] : NULL; 208c6c5593SMatthew G. Knepley 218c6c5593SMatthew G. Knepley if (!sp[f]) continue; 228c6c5593SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(sp[f], &spDim);CHKERRQ(ierr); 238c6c5593SMatthew G. Knepley for (d = 0; d < spDim; ++d) { 248c6c5593SMatthew G. Knepley if (funcs[f]) { 258c6c5593SMatthew G. Knepley if (isFE[f]) {ierr = PetscDualSpaceApply(sp[f], d, time, fegeom, Nc[f], funcs[f], ctx, &values[v]);CHKERRQ(ierr);} 268c6c5593SMatthew G. Knepley else {ierr = PetscDualSpaceApplyFVM(sp[f], d, time, fvgeom, Nc[f], funcs[f], ctx, &values[v]);CHKERRQ(ierr);} 278c6c5593SMatthew G. Knepley } else { 288c6c5593SMatthew G. Knepley for (c = 0; c < Nc[f]; ++c) values[v+c] = 0.0; 298c6c5593SMatthew G. Knepley } 308c6c5593SMatthew G. Knepley v += Nc[f]; 318c6c5593SMatthew G. Knepley } 328c6c5593SMatthew G. Knepley } 338c6c5593SMatthew G. Knepley PetscFunctionReturn(0); 348c6c5593SMatthew G. Knepley } 358c6c5593SMatthew G. Knepley 36496733e2SMatthew G. Knepley static PetscErrorCode DMProjectPoint_Field_Private(DM dm, DM dmAux, PetscReal time, Vec localU, Vec localA, PetscFECellGeom *fegeom, PetscDualSpace sp[], PetscInt p, 370c898c18SMatthew G. Knepley PetscReal **basisTab, PetscReal **basisDerTab, PetscReal **basisTabAux, PetscReal **basisDerTabAux, 388c6c5593SMatthew G. Knepley void (**funcs)(PetscInt, PetscInt, PetscInt, 398c6c5593SMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 408c6c5593SMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 418c6c5593SMatthew G. Knepley PetscReal, const PetscReal[], PetscScalar[]), void **ctxs, 428c6c5593SMatthew G. Knepley PetscScalar values[]) 438c6c5593SMatthew G. Knepley { 448c6c5593SMatthew G. Knepley PetscDS prob, probAux = NULL; 458c6c5593SMatthew G. Knepley PetscSection section, sectionAux = NULL; 46496733e2SMatthew G. Knepley PetscScalar *u, *u_t = NULL, *u_x, *a = NULL, *a_t = NULL, *a_x = NULL, *refSpaceDer, *refSpaceDerAux = NULL; 478c6c5593SMatthew G. Knepley PetscScalar *coefficients = NULL, *coefficientsAux = NULL; 488c6c5593SMatthew G. Knepley PetscScalar *coefficients_t = NULL, *coefficientsAux_t = NULL; 498c6c5593SMatthew G. Knepley PetscReal *x; 50496733e2SMatthew G. Knepley PetscInt *uOff, *uOff_x, *aOff = NULL, *aOff_x = NULL, *Nb, *Nc, *NbAux = NULL, *NcAux = NULL; 511ba34526SMatthew G. Knepley PetscInt dim, dimAux = 0, Nf, NfAux = 0, f, spDim, d, c, v, tp = 0; 528c6c5593SMatthew G. Knepley PetscErrorCode ierr; 538c6c5593SMatthew G. Knepley 548c6c5593SMatthew G. Knepley PetscFunctionBeginHot; 558c6c5593SMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 568c6c5593SMatthew G. Knepley ierr = PetscDSGetSpatialDimension(prob, &dim);CHKERRQ(ierr); 578c6c5593SMatthew G. Knepley ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr); 58496733e2SMatthew G. Knepley ierr = PetscDSGetDimensions(prob, &Nb);CHKERRQ(ierr); 59496733e2SMatthew G. Knepley ierr = PetscDSGetComponents(prob, &Nc);CHKERRQ(ierr); 608c6c5593SMatthew G. Knepley ierr = PetscDSGetComponentOffsets(prob, &uOff);CHKERRQ(ierr); 618c6c5593SMatthew G. Knepley ierr = PetscDSGetComponentDerivativeOffsets(prob, &uOff_x);CHKERRQ(ierr); 6211d189daSMatthew G. Knepley ierr = PetscDSGetEvaluationArrays(prob, &u, NULL /*&u_t*/, &u_x);CHKERRQ(ierr); 63496733e2SMatthew G. Knepley ierr = PetscDSGetRefCoordArrays(prob, &x, &refSpaceDer);CHKERRQ(ierr); 648c6c5593SMatthew G. Knepley ierr = DMGetDefaultSection(dm, §ion);CHKERRQ(ierr); 65496733e2SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, section, localU, p, NULL, &coefficients);CHKERRQ(ierr); 668c6c5593SMatthew G. Knepley if (dmAux) { 671ba34526SMatthew G. Knepley DMLabel spmap; 681ba34526SMatthew G. Knepley PetscInt subp; 691ba34526SMatthew G. Knepley 701ba34526SMatthew G. Knepley ierr = DMPlexGetSubpointMap(dmAux, &spmap);CHKERRQ(ierr); 711ba34526SMatthew G. Knepley if (spmap) { 721ba34526SMatthew G. Knepley IS subpointIS; 731ba34526SMatthew G. Knepley const PetscInt *subpoints; 741ba34526SMatthew G. Knepley PetscInt numSubpoints; 751ba34526SMatthew G. Knepley 761ba34526SMatthew G. Knepley ierr = DMPlexCreateSubpointIS(dmAux, &subpointIS);CHKERRQ(ierr); 771ba34526SMatthew G. Knepley ierr = ISGetLocalSize(subpointIS, &numSubpoints);CHKERRQ(ierr); 781ba34526SMatthew G. Knepley ierr = ISGetIndices(subpointIS, &subpoints);CHKERRQ(ierr); 791ba34526SMatthew G. Knepley ierr = PetscFindInt(p, numSubpoints, subpoints, &subp);CHKERRQ(ierr); 801ba34526SMatthew G. Knepley if (subp < 0) SETERRQ1(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Point %d not found in submesh", p); 811ba34526SMatthew G. Knepley ierr = ISRestoreIndices(subpointIS, &subpoints);CHKERRQ(ierr); 821ba34526SMatthew G. Knepley ierr = ISDestroy(&subpointIS);CHKERRQ(ierr); 831ba34526SMatthew G. Knepley } else subp = p; 848c6c5593SMatthew G. Knepley ierr = DMGetDS(dmAux, &probAux);CHKERRQ(ierr); 851ba34526SMatthew G. Knepley ierr = PetscDSGetSpatialDimension(probAux, &dimAux);CHKERRQ(ierr); 868c6c5593SMatthew G. Knepley ierr = PetscDSGetNumFields(probAux, &NfAux);CHKERRQ(ierr); 87496733e2SMatthew G. Knepley ierr = PetscDSGetDimensions(probAux, &NbAux);CHKERRQ(ierr); 88496733e2SMatthew G. Knepley ierr = PetscDSGetComponents(probAux, &NcAux);CHKERRQ(ierr); 898c6c5593SMatthew G. Knepley ierr = DMGetDefaultSection(dmAux, §ionAux);CHKERRQ(ierr); 908c6c5593SMatthew G. Knepley ierr = PetscDSGetComponentOffsets(probAux, &aOff);CHKERRQ(ierr); 918c6c5593SMatthew G. Knepley ierr = PetscDSGetComponentDerivativeOffsets(probAux, &aOff_x);CHKERRQ(ierr); 9211d189daSMatthew G. Knepley ierr = PetscDSGetEvaluationArrays(probAux, &a, NULL /*&a_t*/, &a_x);CHKERRQ(ierr); 93496733e2SMatthew G. Knepley ierr = PetscDSGetRefCoordArrays(probAux, NULL, &refSpaceDerAux);CHKERRQ(ierr); 941ba34526SMatthew G. Knepley ierr = DMPlexVecGetClosure(dmAux, sectionAux, localA, subp, NULL, &coefficientsAux);CHKERRQ(ierr); 958c6c5593SMatthew G. Knepley } 968c6c5593SMatthew G. Knepley /* Get values for closure */ 978c6c5593SMatthew G. Knepley for (f = 0, v = 0; f < Nf; ++f) { 988c6c5593SMatthew G. Knepley if (!sp[f]) continue; 998c6c5593SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(sp[f], &spDim);CHKERRQ(ierr); 1008c6c5593SMatthew G. Knepley for (d = 0; d < spDim; ++d) { 1018c6c5593SMatthew G. Knepley if (funcs[f]) { 1028c6c5593SMatthew G. Knepley PetscQuadrature quad; 1030c898c18SMatthew G. Knepley const PetscReal *points; 1048c6c5593SMatthew G. Knepley PetscInt numPoints, q; 1058c6c5593SMatthew G. Knepley 1068c6c5593SMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(sp[f], d, &quad);CHKERRQ(ierr); 1070c898c18SMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, &numPoints, &points, NULL);CHKERRQ(ierr); 1080c898c18SMatthew G. Knepley for (q = 0; q < numPoints; ++q, ++tp) { 1098c6c5593SMatthew G. Knepley CoordinatesRefToReal(dim, dim, fegeom->v0, fegeom->J, &points[q*dim], x); 110496733e2SMatthew G. Knepley EvaluateFieldJets(dim, Nf, Nb, Nc, tp, basisTab, basisDerTab, refSpaceDer, fegeom->invJ, coefficients, coefficients_t, u, u_x, u_t); 1111ba34526SMatthew G. Knepley if (probAux) {EvaluateFieldJets(dimAux, NfAux, NbAux, NcAux, tp, basisTabAux, basisDerTabAux, refSpaceDerAux, fegeom->invJ, coefficientsAux, coefficientsAux_t, a, a_x, a_t);} 1128c6c5593SMatthew G. Knepley (*funcs[f])(dim, Nf, NfAux, uOff, uOff_x, u, u_t, u_x, aOff, aOff_x, a, a_t, a_x, time, x, &values[v]); 1138c6c5593SMatthew G. Knepley } 1148c6c5593SMatthew G. Knepley } else { 1158c6c5593SMatthew G. Knepley for (c = 0; c < Nc[f]; ++c) values[v+c] = 0.0; 1168c6c5593SMatthew G. Knepley } 1178c6c5593SMatthew G. Knepley v += Nc[f]; 1188c6c5593SMatthew G. Knepley } 1198c6c5593SMatthew G. Knepley } 1208c6c5593SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, section, localU, p, NULL, &coefficients);CHKERRQ(ierr); 1218c6c5593SMatthew G. Knepley if (dmAux) {ierr = DMPlexVecRestoreClosure(dmAux, sectionAux, localA, p, NULL, &coefficientsAux);CHKERRQ(ierr);} 1228c6c5593SMatthew G. Knepley PetscFunctionReturn(0); 1238c6c5593SMatthew G. Knepley } 1248c6c5593SMatthew G. Knepley 1250c898c18SMatthew G. Knepley static PetscErrorCode DMProjectPoint_Private(DM dm, DM dmAux, PetscInt h, PetscReal time, Vec localU, Vec localA, PetscBool hasFE, PetscBool hasFV, PetscBool isFE[], 126496733e2SMatthew G. Knepley PetscDualSpace sp[], PetscInt p, PetscReal **basisTab, PetscReal **basisDerTab, PetscReal **basisTabAux, PetscReal **basisDerTabAux, 1278c6c5593SMatthew G. Knepley DMBoundaryConditionType type, void (**funcs)(void), void **ctxs, PetscBool fieldActive[], PetscScalar values[]) 1288c6c5593SMatthew G. Knepley { 1298c6c5593SMatthew G. Knepley PetscFECellGeom fegeom; 1308c6c5593SMatthew G. Knepley PetscFVCellGeom fvgeom; 1318c6c5593SMatthew G. Knepley PetscInt dim, dimEmbed; 1328c6c5593SMatthew G. Knepley PetscErrorCode ierr; 1338c6c5593SMatthew G. Knepley 1348c6c5593SMatthew G. Knepley PetscFunctionBeginHot; 1358c6c5593SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 1368c6c5593SMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &dimEmbed);CHKERRQ(ierr); 1378c6c5593SMatthew G. Knepley if (hasFE) { 13843048b12SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dm, p, NULL, fegeom.v0, fegeom.J, fegeom.invJ, &fegeom.detJ);CHKERRQ(ierr); 1398c6c5593SMatthew G. Knepley fegeom.dim = dim - h; 1408c6c5593SMatthew G. Knepley fegeom.dimEmbed = dimEmbed; 1418c6c5593SMatthew G. Knepley } 1428c6c5593SMatthew G. Knepley if (hasFV) {ierr = DMPlexComputeCellGeometryFVM(dm, p, &fvgeom.volume, fvgeom.centroid, NULL);CHKERRQ(ierr);} 1438c6c5593SMatthew G. Knepley switch (type) { 1448c6c5593SMatthew G. Knepley case DM_BC_ESSENTIAL: 1458c6c5593SMatthew G. Knepley case DM_BC_NATURAL: 146496733e2SMatthew G. Knepley ierr = DMProjectPoint_Func_Private(dm, time, &fegeom, &fvgeom, isFE, sp, (PetscErrorCode (**)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *)) funcs, ctxs, values);CHKERRQ(ierr);break; 1478c6c5593SMatthew G. Knepley case DM_BC_ESSENTIAL_FIELD: 1488c6c5593SMatthew G. Knepley case DM_BC_NATURAL_FIELD: 149496733e2SMatthew G. Knepley ierr = DMProjectPoint_Field_Private(dm, dmAux, time, localU, localA, &fegeom, sp, p, 1500c898c18SMatthew G. Knepley basisTab, basisDerTab, basisTabAux, basisDerTabAux, 1510c898c18SMatthew G. Knepley (void (**)(PetscInt, PetscInt, PetscInt, 1520c898c18SMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 1530c898c18SMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 1540c898c18SMatthew G. Knepley PetscReal, const PetscReal[], PetscScalar[])) funcs, ctxs, values);CHKERRQ(ierr);break; 1558c6c5593SMatthew G. Knepley default: SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Unknown boundary condition type: %d", (int) type); 1568c6c5593SMatthew G. Knepley } 1578c6c5593SMatthew G. Knepley PetscFunctionReturn(0); 1588c6c5593SMatthew G. Knepley } 1598c6c5593SMatthew G. Knepley 1608c6c5593SMatthew G. Knepley PetscErrorCode DMProjectLocal_Generic_Plex(DM dm, PetscReal time, Vec localU, 1618c6c5593SMatthew G. Knepley DMLabel label, PetscInt numIds, const PetscInt ids[], 1628c6c5593SMatthew G. Knepley DMBoundaryConditionType type, void (**funcs)(void), void **ctxs, 1638c6c5593SMatthew G. Knepley InsertMode mode, Vec localX) 1648c6c5593SMatthew G. Knepley { 1658c6c5593SMatthew G. Knepley DM dmAux = NULL; 1660c898c18SMatthew G. Knepley PetscDS prob, probAux = NULL; 1678c6c5593SMatthew G. Knepley Vec localA = NULL; 16847923291SMatthew G. Knepley PetscSection section; 1698c6c5593SMatthew G. Knepley PetscDualSpace *sp, *cellsp; 1700c898c18SMatthew G. Knepley PetscReal **basisTab = NULL, **basisDerTab = NULL, **basisTabAux = NULL, **basisDerTabAux = NULL; 1718c6c5593SMatthew G. Knepley PetscInt *Nc; 1722af58022SMatthew G. Knepley PetscInt dim, dimEmbed, minHeight, maxHeight, h, Nf, NfAux = 0, f; 173*88aca1feSMatthew G. Knepley PetscBool *isFE, hasFE = PETSC_FALSE, hasFV = PETSC_FALSE, auxBd = PETSC_FALSE; 17447923291SMatthew G. Knepley PetscErrorCode ierr; 17547923291SMatthew G. Knepley 17647923291SMatthew G. Knepley PetscFunctionBegin; 177*88aca1feSMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "dmAux", (PetscObject *) &dmAux);CHKERRQ(ierr); 178*88aca1feSMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "A", (PetscObject *) &localA);CHKERRQ(ierr); 1798c6c5593SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 1802af58022SMatthew G. Knepley ierr = DMPlexGetVTKCellHeight(dm, &minHeight);CHKERRQ(ierr); 181*88aca1feSMatthew G. Knepley if (!minHeight && dmAux) { 182*88aca1feSMatthew G. Knepley DMLabel spmap; 183*88aca1feSMatthew G. Knepley 184*88aca1feSMatthew G. Knepley /* If dmAux is a surface, then force the projection to take place over a surface */ 185*88aca1feSMatthew G. Knepley ierr = DMPlexGetSubpointMap(dmAux, &spmap);CHKERRQ(ierr); 186*88aca1feSMatthew G. Knepley if (spmap) {minHeight = 1; auxBd = PETSC_TRUE;} 187*88aca1feSMatthew G. Knepley } 1888c6c5593SMatthew G. Knepley ierr = DMPlexGetMaxProjectionHeight(dm, &maxHeight);CHKERRQ(ierr); 1892af58022SMatthew G. Knepley maxHeight = PetscMax(maxHeight, minHeight); 1908c6c5593SMatthew G. Knepley if (maxHeight < 0 || maxHeight > dim) {SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Maximum projection height %d not in [0, %d)\n", maxHeight, dim);} 1910c898c18SMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 1928c6c5593SMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &dimEmbed);CHKERRQ(ierr); 19347923291SMatthew G. Knepley ierr = DMGetDefaultSection(dm, §ion);CHKERRQ(ierr); 19447923291SMatthew G. Knepley ierr = PetscSectionGetNumFields(section, &Nf);CHKERRQ(ierr); 1950c898c18SMatthew G. Knepley if (dmAux) { 1960c898c18SMatthew G. Knepley ierr = DMGetDS(dmAux, &probAux);CHKERRQ(ierr); 1970c898c18SMatthew G. Knepley ierr = PetscDSGetNumFields(probAux, &NfAux);CHKERRQ(ierr); 1980c898c18SMatthew G. Knepley } 199496733e2SMatthew G. Knepley ierr = PetscDSGetComponents(prob, &Nc);CHKERRQ(ierr); 200496733e2SMatthew G. Knepley ierr = PetscMalloc2(Nf, &isFE, Nf, &sp);CHKERRQ(ierr); 2018c6c5593SMatthew G. Knepley if (maxHeight > 0) {ierr = PetscMalloc1(Nf, &cellsp);CHKERRQ(ierr);} 2028c6c5593SMatthew G. Knepley else {cellsp = sp;} 2038c6c5593SMatthew G. Knepley if (localU && localU != localX) {ierr = DMPlexInsertBoundaryValues(dm, PETSC_TRUE, localU, time, NULL, NULL, NULL);CHKERRQ(ierr);} 2048c6c5593SMatthew G. Knepley /* Get cell dual spaces */ 20547923291SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 20647923291SMatthew G. Knepley PetscObject obj; 20747923291SMatthew G. Knepley PetscClassId id; 20847923291SMatthew G. Knepley 20947923291SMatthew G. Knepley ierr = DMGetField(dm, f, &obj);CHKERRQ(ierr); 21047923291SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 21147923291SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 21247923291SMatthew G. Knepley PetscFE fe = (PetscFE) obj; 21347923291SMatthew G. Knepley 21447923291SMatthew G. Knepley hasFE = PETSC_TRUE; 21547923291SMatthew G. Knepley isFE[f] = PETSC_TRUE; 21647923291SMatthew G. Knepley ierr = PetscFEGetDualSpace(fe, &cellsp[f]);CHKERRQ(ierr); 21747923291SMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 21847923291SMatthew G. Knepley PetscFV fv = (PetscFV) obj; 21947923291SMatthew G. Knepley 22047923291SMatthew G. Knepley hasFV = PETSC_TRUE; 22147923291SMatthew G. Knepley isFE[f] = PETSC_FALSE; 22247923291SMatthew G. Knepley ierr = PetscFVGetDualSpace(fv, &cellsp[f]);CHKERRQ(ierr); 22347923291SMatthew G. Knepley } else SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %d", f); 22447923291SMatthew G. Knepley } 2250c898c18SMatthew G. Knepley if (type == DM_BC_ESSENTIAL_FIELD || type == DM_BC_NATURAL_FIELD) { 2260c898c18SMatthew G. Knepley PetscFE fem; 2270c898c18SMatthew G. Knepley PetscReal *points; 2280c898c18SMatthew G. Knepley PetscInt numPoints, spDim, d; 2290c898c18SMatthew G. Knepley 2302af58022SMatthew G. Knepley if (maxHeight > minHeight) SETERRQ(PetscObjectComm((PetscObject) dm), PETSC_ERR_SUP, "Field projection not supported for face interpolation"); 2310c898c18SMatthew G. Knepley numPoints = 0; 2320c898c18SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 2330c898c18SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(cellsp[f], &spDim);CHKERRQ(ierr); 2340c898c18SMatthew G. Knepley for (d = 0; d < spDim; ++d) { 2350c898c18SMatthew G. Knepley if (funcs[f]) { 2360c898c18SMatthew G. Knepley PetscQuadrature quad; 2370c898c18SMatthew G. Knepley PetscInt Nq; 2380c898c18SMatthew G. Knepley 2390c898c18SMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(cellsp[f], d, &quad);CHKERRQ(ierr); 2400c898c18SMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, &Nq, NULL, NULL);CHKERRQ(ierr); 2410c898c18SMatthew G. Knepley numPoints += Nq; 2420c898c18SMatthew G. Knepley } 2430c898c18SMatthew G. Knepley } 2440c898c18SMatthew G. Knepley } 2450c898c18SMatthew G. Knepley ierr = PetscMalloc1(numPoints*dim, &points);CHKERRQ(ierr); 2460c898c18SMatthew G. Knepley numPoints = 0; 2470c898c18SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 2480c898c18SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(cellsp[f], &spDim);CHKERRQ(ierr); 2490c898c18SMatthew G. Knepley for (d = 0; d < spDim; ++d) { 2500c898c18SMatthew G. Knepley if (funcs[f]) { 2510c898c18SMatthew G. Knepley PetscQuadrature quad; 2520c898c18SMatthew G. Knepley const PetscReal *qpoints; 2530c898c18SMatthew G. Knepley PetscInt Nq, q; 2540c898c18SMatthew G. Knepley 2550c898c18SMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(cellsp[f], d, &quad);CHKERRQ(ierr); 2560c898c18SMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, &Nq, &qpoints, NULL);CHKERRQ(ierr); 2570c898c18SMatthew G. Knepley for (q = 0; q < Nq*dim; ++q) points[numPoints*dim+q] = qpoints[q]; 2580c898c18SMatthew G. Knepley numPoints += Nq; 2590c898c18SMatthew G. Knepley } 2600c898c18SMatthew G. Knepley } 2610c898c18SMatthew G. Knepley } 2620c898c18SMatthew G. Knepley ierr = PetscMalloc4(Nf, &basisTab, Nf, &basisDerTab, NfAux, &basisTabAux, NfAux, &basisDerTabAux);CHKERRQ(ierr); 2630c898c18SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 2640c898c18SMatthew G. Knepley if (!isFE[f]) continue; 2650c898c18SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, (PetscObject *) &fem);CHKERRQ(ierr); 2660c898c18SMatthew G. Knepley ierr = PetscFEGetTabulation(fem, numPoints, points, &basisTab[f], &basisDerTab[f], NULL);CHKERRQ(ierr); 2670c898c18SMatthew G. Knepley } 2680c898c18SMatthew G. Knepley for (f = 0; f < NfAux; ++f) { 2690c898c18SMatthew G. Knepley ierr = PetscDSGetDiscretization(probAux, f, (PetscObject *) &fem);CHKERRQ(ierr); 2700c898c18SMatthew G. Knepley ierr = PetscFEGetTabulation(fem, numPoints, points, &basisTabAux[f], &basisDerTabAux[f], NULL);CHKERRQ(ierr); 2710c898c18SMatthew G. Knepley } 2720c898c18SMatthew G. Knepley ierr = PetscFree(points);CHKERRQ(ierr); 2730c898c18SMatthew G. Knepley } 27447923291SMatthew G. Knepley /* Note: We make no attempt to optimize for height. Higher height things just overwrite the lower height results. */ 2752af58022SMatthew G. Knepley for (h = minHeight; h <= maxHeight; h++) { 276*88aca1feSMatthew G. Knepley PetscInt effectiveHeight = h - (auxBd ? 0 : minHeight); 2778c6c5593SMatthew G. Knepley PetscScalar *values; 2788c6c5593SMatthew G. Knepley PetscBool *fieldActive; 27984f7fa7aSMatthew G. Knepley PetscInt pStart, pEnd, p, spDim, totDim, numValues; 2808c6c5593SMatthew G. Knepley 28147923291SMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, h, &pStart, &pEnd);CHKERRQ(ierr); 2828c6c5593SMatthew G. Knepley if (!h) { 2838c6c5593SMatthew G. Knepley PetscInt cEndInterior; 2848c6c5593SMatthew G. Knepley 2858c6c5593SMatthew G. Knepley ierr = DMPlexGetHybridBounds(dm, &cEndInterior, NULL, NULL, NULL);CHKERRQ(ierr); 2868c6c5593SMatthew G. Knepley pEnd = cEndInterior < 0 ? pEnd : cEndInterior; 2878c6c5593SMatthew G. Knepley } 28847923291SMatthew G. Knepley if (pEnd <= pStart) continue; 28947923291SMatthew G. Knepley /* Compute totDim, the number of dofs in the closure of a point at this height */ 29047923291SMatthew G. Knepley totDim = 0; 29147923291SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 292bec1f4aaSMatthew G. Knepley if (!effectiveHeight) { 29347923291SMatthew G. Knepley sp[f] = cellsp[f]; 29447923291SMatthew G. Knepley } else { 295bec1f4aaSMatthew G. Knepley ierr = PetscDualSpaceGetHeightSubspace(cellsp[f], effectiveHeight, &sp[f]);CHKERRQ(ierr); 29647923291SMatthew G. Knepley if (!sp[f]) continue; 29747923291SMatthew G. Knepley } 29847923291SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(sp[f], &spDim);CHKERRQ(ierr); 2998c6c5593SMatthew G. Knepley totDim += spDim*Nc[f]; 30047923291SMatthew G. Knepley } 30147923291SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, section, localX, pStart, &numValues, NULL);CHKERRQ(ierr); 3028c6c5593SMatthew G. Knepley if (numValues != totDim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "The section point closure size %d != dual space dimension %d", numValues, totDim); 30347923291SMatthew G. Knepley if (!totDim) continue; 30447923291SMatthew G. Knepley /* Loop over points at this height */ 30547923291SMatthew G. Knepley ierr = DMGetWorkArray(dm, numValues, PETSC_SCALAR, &values);CHKERRQ(ierr); 3068c6c5593SMatthew G. Knepley ierr = DMGetWorkArray(dm, Nf, PETSC_BOOL, &fieldActive);CHKERRQ(ierr); 3078c6c5593SMatthew G. Knepley for (f = 0; f < Nf; ++f) fieldActive[f] = (funcs[f] && sp[f]) ? PETSC_TRUE : PETSC_FALSE; 3088c6c5593SMatthew G. Knepley if (label) { 3098c6c5593SMatthew G. Knepley PetscInt i; 31047923291SMatthew G. Knepley 31147923291SMatthew G. Knepley for (i = 0; i < numIds; ++i) { 31247923291SMatthew G. Knepley IS pointIS; 31347923291SMatthew G. Knepley const PetscInt *points; 3148c6c5593SMatthew G. Knepley PetscInt n; 31547923291SMatthew G. Knepley 31647923291SMatthew G. Knepley ierr = DMLabelGetStratumIS(label, ids[i], &pointIS);CHKERRQ(ierr); 31747923291SMatthew G. Knepley if (!pointIS) continue; /* No points with that id on this process */ 31847923291SMatthew G. Knepley ierr = ISGetLocalSize(pointIS, &n);CHKERRQ(ierr); 31947923291SMatthew G. Knepley ierr = ISGetIndices(pointIS, &points);CHKERRQ(ierr); 32047923291SMatthew G. Knepley for (p = 0; p < n; ++p) { 32147923291SMatthew G. Knepley const PetscInt point = points[p]; 32247923291SMatthew G. Knepley 32347923291SMatthew G. Knepley if ((point < pStart) || (point >= pEnd)) continue; 324bd1e2261SMatthew G. Knepley ierr = DMProjectPoint_Private(dm, dmAux, effectiveHeight, time, localU, localA, hasFE, hasFV, isFE, sp, point, basisTab, basisDerTab, basisTabAux, basisDerTabAux, type, funcs, ctxs, fieldActive, values); 32547923291SMatthew G. Knepley if (ierr) { 32647923291SMatthew G. Knepley PetscErrorCode ierr2; 32747923291SMatthew G. Knepley ierr2 = DMRestoreWorkArray(dm, numValues, PETSC_SCALAR, &values);CHKERRQ(ierr2); 3288c6c5593SMatthew G. Knepley ierr2 = DMRestoreWorkArray(dm, Nf, PETSC_BOOL, &fieldActive);CHKERRQ(ierr2); 32947923291SMatthew G. Knepley CHKERRQ(ierr); 33047923291SMatthew G. Knepley } 33147923291SMatthew G. Knepley ierr = DMPlexVecSetFieldClosure_Internal(dm, section, localX, fieldActive, point, values, mode);CHKERRQ(ierr); 33247923291SMatthew G. Knepley } 33347923291SMatthew G. Knepley ierr = ISRestoreIndices(pointIS, &points);CHKERRQ(ierr); 33447923291SMatthew G. Knepley ierr = ISDestroy(&pointIS);CHKERRQ(ierr); 33547923291SMatthew G. Knepley } 3368c6c5593SMatthew G. Knepley } else { 3378c6c5593SMatthew G. Knepley for (p = pStart; p < pEnd; ++p) { 338bd1e2261SMatthew G. Knepley ierr = DMProjectPoint_Private(dm, dmAux, effectiveHeight, time, localU, localA, hasFE, hasFV, isFE, sp, p, basisTab, basisDerTab, basisTabAux, basisDerTabAux, type, funcs, ctxs, fieldActive, values); 3398c6c5593SMatthew G. Knepley if (ierr) { 3408c6c5593SMatthew G. Knepley PetscErrorCode ierr2; 3418c6c5593SMatthew G. Knepley ierr2 = DMRestoreWorkArray(dm, numValues, PETSC_SCALAR, &values);CHKERRQ(ierr2); 3428c6c5593SMatthew G. Knepley ierr2 = DMRestoreWorkArray(dm, Nf, PETSC_BOOL, &fieldActive);CHKERRQ(ierr2); 3438c6c5593SMatthew G. Knepley CHKERRQ(ierr); 3448c6c5593SMatthew G. Knepley } 3458c6c5593SMatthew G. Knepley ierr = DMPlexVecSetFieldClosure_Internal(dm, section, localX, fieldActive, p, values, mode);CHKERRQ(ierr); 3468c6c5593SMatthew G. Knepley } 3478c6c5593SMatthew G. Knepley } 34847923291SMatthew G. Knepley ierr = DMRestoreWorkArray(dm, numValues, PETSC_SCALAR, &values);CHKERRQ(ierr); 3498c6c5593SMatthew G. Knepley ierr = DMRestoreWorkArray(dm, Nf, PETSC_BOOL, &fieldActive);CHKERRQ(ierr); 35047923291SMatthew G. Knepley } 3518c6c5593SMatthew G. Knepley /* Cleanup */ 3520c898c18SMatthew G. Knepley if (type == DM_BC_ESSENTIAL_FIELD || type == DM_BC_NATURAL_FIELD) { 3530c898c18SMatthew G. Knepley PetscFE fem; 3540c898c18SMatthew G. Knepley 3550c898c18SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 3560c898c18SMatthew G. Knepley if (!isFE[f]) continue; 3570c898c18SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, (PetscObject *) &fem);CHKERRQ(ierr); 3580c898c18SMatthew G. Knepley ierr = PetscFERestoreTabulation(fem, 0, NULL, &basisTab[f], &basisDerTab[f], NULL);CHKERRQ(ierr); 3590c898c18SMatthew G. Knepley } 3600c898c18SMatthew G. Knepley for (f = 0; f < NfAux; ++f) { 3610c898c18SMatthew G. Knepley ierr = PetscDSGetDiscretization(probAux, f, (PetscObject *) &fem);CHKERRQ(ierr); 3620c898c18SMatthew G. Knepley ierr = PetscFERestoreTabulation(fem, 0, NULL, &basisTabAux[f], &basisDerTabAux[f], NULL);CHKERRQ(ierr); 3630c898c18SMatthew G. Knepley } 3640c898c18SMatthew G. Knepley ierr = PetscFree4(basisTab, basisDerTab, basisTabAux, basisDerTabAux);CHKERRQ(ierr); 3650c898c18SMatthew G. Knepley } 366496733e2SMatthew G. Knepley ierr = PetscFree2(isFE, sp);CHKERRQ(ierr); 3678c6c5593SMatthew G. Knepley if (maxHeight > 0) {ierr = PetscFree(cellsp);CHKERRQ(ierr);} 3688c6c5593SMatthew G. Knepley PetscFunctionReturn(0); 36947923291SMatthew G. Knepley } 3708c6c5593SMatthew G. Knepley 3718c6c5593SMatthew G. Knepley PetscErrorCode DMProjectFunctionLocal_Plex(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, InsertMode mode, Vec localX) 3728c6c5593SMatthew G. Knepley { 3738c6c5593SMatthew G. Knepley PetscErrorCode ierr; 3748c6c5593SMatthew G. Knepley 3758c6c5593SMatthew G. Knepley PetscFunctionBegin; 3768c6c5593SMatthew G. Knepley ierr = DMProjectLocal_Generic_Plex(dm, time, localX, NULL, 0, NULL, DM_BC_ESSENTIAL, (void (**)(void)) funcs, ctxs, mode, localX);CHKERRQ(ierr); 3778c6c5593SMatthew G. Knepley PetscFunctionReturn(0); 3788c6c5593SMatthew G. Knepley } 3798c6c5593SMatthew G. Knepley 3808c6c5593SMatthew G. Knepley PetscErrorCode DMProjectFunctionLabelLocal_Plex(DM dm, PetscReal time, DMLabel label, PetscInt numIds, const PetscInt ids[], PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, InsertMode mode, Vec localX) 3818c6c5593SMatthew G. Knepley { 3828c6c5593SMatthew G. Knepley PetscErrorCode ierr; 3838c6c5593SMatthew G. Knepley 3848c6c5593SMatthew G. Knepley PetscFunctionBegin; 3858c6c5593SMatthew G. Knepley ierr = DMProjectLocal_Generic_Plex(dm, time, localX, label, numIds, ids, DM_BC_ESSENTIAL, (void (**)(void)) funcs, ctxs, mode, localX);CHKERRQ(ierr); 38647923291SMatthew G. Knepley PetscFunctionReturn(0); 38747923291SMatthew G. Knepley } 38847923291SMatthew G. Knepley 3898c6c5593SMatthew G. Knepley PetscErrorCode DMProjectFieldLocal_Plex(DM dm, PetscReal time, Vec localU, 39047923291SMatthew G. Knepley void (**funcs)(PetscInt, PetscInt, PetscInt, 39147923291SMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 39247923291SMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 39347923291SMatthew G. Knepley PetscReal, const PetscReal[], PetscScalar[]), 39447923291SMatthew G. Knepley InsertMode mode, Vec localX) 39547923291SMatthew G. Knepley { 39647923291SMatthew G. Knepley PetscErrorCode ierr; 39747923291SMatthew G. Knepley 39847923291SMatthew G. Knepley PetscFunctionBegin; 3998c6c5593SMatthew G. Knepley ierr = DMProjectLocal_Generic_Plex(dm, time, localU, NULL, 0, NULL, DM_BC_ESSENTIAL_FIELD, (void (**)(void)) funcs, NULL, mode, localX);CHKERRQ(ierr); 4008c6c5593SMatthew G. Knepley PetscFunctionReturn(0); 40147923291SMatthew G. Knepley } 40247923291SMatthew G. Knepley 4038c6c5593SMatthew G. Knepley PetscErrorCode DMProjectFieldLabelLocal_Plex(DM dm, PetscReal time, DMLabel label, PetscInt numIds, const PetscInt ids[], Vec localU, 4048c6c5593SMatthew G. Knepley void (**funcs)(PetscInt, PetscInt, PetscInt, 4058c6c5593SMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 4068c6c5593SMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 4078c6c5593SMatthew G. Knepley PetscReal, const PetscReal[], PetscScalar[]), 4088c6c5593SMatthew G. Knepley InsertMode mode, Vec localX) 4098c6c5593SMatthew G. Knepley { 4108c6c5593SMatthew G. Knepley PetscErrorCode ierr; 41147923291SMatthew G. Knepley 4128c6c5593SMatthew G. Knepley PetscFunctionBegin; 4138c6c5593SMatthew G. Knepley ierr = DMProjectLocal_Generic_Plex(dm, time, localU, label, numIds, ids, DM_BC_ESSENTIAL_FIELD, (void (**)(void)) funcs, NULL, mode, localX);CHKERRQ(ierr); 41447923291SMatthew G. Knepley PetscFunctionReturn(0); 41547923291SMatthew G. Knepley } 416