1af0996ceSBarry Smith #include <petsc/private/dmpleximpl.h> /*I "petscdmplex.h" I*/ 2da97024aSMatthew G. Knepley #include <petscsf.h> 3cb1e1211SMatthew G Knepley 4e8f14785SLisandro Dalcin #include <petsc/private/hashsetij.h> 5af0996ceSBarry Smith #include <petsc/private/petscfeimpl.h> 6af0996ceSBarry Smith #include <petsc/private/petscfvimpl.h> 7a0845e3aSMatthew G. Knepley 82f856554SMatthew G. Knepley static PetscErrorCode DMPlexConvertPlex(DM dm, DM *plex, PetscBool copy) 92f856554SMatthew G. Knepley { 102f856554SMatthew G. Knepley PetscBool isPlex; 112f856554SMatthew G. Knepley PetscErrorCode ierr; 122f856554SMatthew G. Knepley 132f856554SMatthew G. Knepley PetscFunctionBegin; 142f856554SMatthew G. Knepley ierr = PetscObjectTypeCompare((PetscObject) dm, DMPLEX, &isPlex);CHKERRQ(ierr); 152f856554SMatthew G. Knepley if (isPlex) { 162f856554SMatthew G. Knepley *plex = dm; 172f856554SMatthew G. Knepley ierr = PetscObjectReference((PetscObject) dm);CHKERRQ(ierr); 182f856554SMatthew G. Knepley } else { 192f856554SMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "dm_plex", (PetscObject *) plex);CHKERRQ(ierr); 202f856554SMatthew G. Knepley if (!*plex) { 212f856554SMatthew G. Knepley ierr = DMConvert(dm,DMPLEX,plex);CHKERRQ(ierr); 222f856554SMatthew G. Knepley ierr = PetscObjectCompose((PetscObject) dm, "dm_plex", (PetscObject) *plex);CHKERRQ(ierr); 232f856554SMatthew G. Knepley if (copy) { 242f856554SMatthew G. Knepley const char *comps[3] = {"A", "dmAux"}; 252f856554SMatthew G. Knepley PetscObject obj; 262f856554SMatthew G. Knepley PetscInt i; 272f856554SMatthew G. Knepley 282f856554SMatthew G. Knepley { 292f856554SMatthew G. Knepley /* Run the subdomain hook (this will copy the DMSNES/DMTS) */ 302f856554SMatthew G. Knepley DMSubDomainHookLink link; 312f856554SMatthew G. Knepley for (link = dm->subdomainhook; link; link = link->next) { 322f856554SMatthew G. Knepley if (link->ddhook) {ierr = (*link->ddhook)(dm, *plex, link->ctx);CHKERRQ(ierr);} 332f856554SMatthew G. Knepley } 342f856554SMatthew G. Knepley } 352f856554SMatthew G. Knepley for (i = 0; i < 3; i++) { 362f856554SMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, comps[i], &obj);CHKERRQ(ierr); 372f856554SMatthew G. Knepley ierr = PetscObjectCompose((PetscObject) *plex, comps[i], obj);CHKERRQ(ierr); 382f856554SMatthew G. Knepley } 392f856554SMatthew G. Knepley } 402f856554SMatthew G. Knepley } else { 412f856554SMatthew G. Knepley ierr = PetscObjectReference((PetscObject) *plex);CHKERRQ(ierr); 422f856554SMatthew G. Knepley } 432f856554SMatthew G. Knepley } 442f856554SMatthew G. Knepley PetscFunctionReturn(0); 452f856554SMatthew G. Knepley } 462f856554SMatthew G. Knepley 479b6f715bSMatthew G. Knepley static PetscErrorCode PetscContainerUserDestroy_PetscFEGeom (void *ctx) 489b6f715bSMatthew G. Knepley { 499b6f715bSMatthew G. Knepley PetscFEGeom *geom = (PetscFEGeom *) ctx; 509b6f715bSMatthew G. Knepley PetscErrorCode ierr; 519b6f715bSMatthew G. Knepley 529b6f715bSMatthew G. Knepley PetscFunctionBegin; 539b6f715bSMatthew G. Knepley ierr = PetscFEGeomDestroy(&geom);CHKERRQ(ierr); 549b6f715bSMatthew G. Knepley PetscFunctionReturn(0); 559b6f715bSMatthew G. Knepley } 569b6f715bSMatthew G. Knepley 579b6f715bSMatthew G. Knepley static PetscErrorCode DMPlexGetFEGeom(DMField coordField, IS pointIS, PetscQuadrature quad, PetscBool faceData, PetscFEGeom **geom) 589b6f715bSMatthew G. Knepley { 599b6f715bSMatthew G. Knepley char composeStr[33] = {0}; 609b6f715bSMatthew G. Knepley PetscObjectId id; 619b6f715bSMatthew G. Knepley PetscContainer container; 629b6f715bSMatthew G. Knepley PetscErrorCode ierr; 639b6f715bSMatthew G. Knepley 649b6f715bSMatthew G. Knepley PetscFunctionBegin; 659b6f715bSMatthew G. Knepley ierr = PetscObjectGetId((PetscObject)quad,&id);CHKERRQ(ierr); 669b6f715bSMatthew G. Knepley ierr = PetscSNPrintf(composeStr, 32, "DMPlexGetFEGeom_%x\n", id);CHKERRQ(ierr); 679b6f715bSMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) pointIS, composeStr, (PetscObject *) &container);CHKERRQ(ierr); 689b6f715bSMatthew G. Knepley if (container) { 699b6f715bSMatthew G. Knepley ierr = PetscContainerGetPointer(container, (void **) geom);CHKERRQ(ierr); 709b6f715bSMatthew G. Knepley } else { 719b6f715bSMatthew G. Knepley ierr = DMFieldCreateFEGeom(coordField, pointIS, quad, faceData, geom);CHKERRQ(ierr); 729b6f715bSMatthew G. Knepley ierr = PetscContainerCreate(PETSC_COMM_SELF,&container);CHKERRQ(ierr); 739b6f715bSMatthew G. Knepley ierr = PetscContainerSetPointer(container, (void *) *geom);CHKERRQ(ierr); 749b6f715bSMatthew G. Knepley ierr = PetscContainerSetUserDestroy(container, PetscContainerUserDestroy_PetscFEGeom);CHKERRQ(ierr); 759b6f715bSMatthew G. Knepley ierr = PetscObjectCompose((PetscObject) pointIS, composeStr, (PetscObject) container);CHKERRQ(ierr); 769b6f715bSMatthew G. Knepley ierr = PetscContainerDestroy(&container);CHKERRQ(ierr); 779b6f715bSMatthew G. Knepley } 789b6f715bSMatthew G. Knepley PetscFunctionReturn(0); 799b6f715bSMatthew G. Knepley } 809b6f715bSMatthew G. Knepley 819b6f715bSMatthew G. Knepley static PetscErrorCode DMPlexRestoreFEGeom(DMField coordField, IS pointIS, PetscQuadrature quad, PetscBool faceData, PetscFEGeom **geom) 829b6f715bSMatthew G. Knepley { 839b6f715bSMatthew G. Knepley PetscFunctionBegin; 849b6f715bSMatthew G. Knepley *geom = NULL; 859b6f715bSMatthew G. Knepley PetscFunctionReturn(0); 869b6f715bSMatthew G. Knepley } 879b6f715bSMatthew G. Knepley 8846fa42a0SMatthew G. Knepley /*@ 8946fa42a0SMatthew G. Knepley DMPlexGetScale - Get the scale for the specified fundamental unit 9046fa42a0SMatthew G. Knepley 9146fa42a0SMatthew G. Knepley Not collective 9246fa42a0SMatthew G. Knepley 9346fa42a0SMatthew G. Knepley Input Arguments: 9446fa42a0SMatthew G. Knepley + dm - the DM 9546fa42a0SMatthew G. Knepley - unit - The SI unit 9646fa42a0SMatthew G. Knepley 9746fa42a0SMatthew G. Knepley Output Argument: 9846fa42a0SMatthew G. Knepley . scale - The value used to scale all quantities with this unit 9946fa42a0SMatthew G. Knepley 10046fa42a0SMatthew G. Knepley Level: advanced 10146fa42a0SMatthew G. Knepley 10246fa42a0SMatthew G. Knepley .seealso: DMPlexSetScale(), PetscUnit 10346fa42a0SMatthew G. Knepley @*/ 104cb1e1211SMatthew G Knepley PetscErrorCode DMPlexGetScale(DM dm, PetscUnit unit, PetscReal *scale) 105cb1e1211SMatthew G Knepley { 106cb1e1211SMatthew G Knepley DM_Plex *mesh = (DM_Plex*) dm->data; 107cb1e1211SMatthew G Knepley 108cb1e1211SMatthew G Knepley PetscFunctionBegin; 109cb1e1211SMatthew G Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 110cb1e1211SMatthew G Knepley PetscValidPointer(scale, 3); 111cb1e1211SMatthew G Knepley *scale = mesh->scale[unit]; 112cb1e1211SMatthew G Knepley PetscFunctionReturn(0); 113cb1e1211SMatthew G Knepley } 114cb1e1211SMatthew G Knepley 11546fa42a0SMatthew G. Knepley /*@ 11646fa42a0SMatthew G. Knepley DMPlexSetScale - Set the scale for the specified fundamental unit 11746fa42a0SMatthew G. Knepley 11846fa42a0SMatthew G. Knepley Not collective 11946fa42a0SMatthew G. Knepley 12046fa42a0SMatthew G. Knepley Input Arguments: 12146fa42a0SMatthew G. Knepley + dm - the DM 12246fa42a0SMatthew G. Knepley . unit - The SI unit 12346fa42a0SMatthew G. Knepley - scale - The value used to scale all quantities with this unit 12446fa42a0SMatthew G. Knepley 12546fa42a0SMatthew G. Knepley Level: advanced 12646fa42a0SMatthew G. Knepley 12746fa42a0SMatthew G. Knepley .seealso: DMPlexGetScale(), PetscUnit 12846fa42a0SMatthew G. Knepley @*/ 129cb1e1211SMatthew G Knepley PetscErrorCode DMPlexSetScale(DM dm, PetscUnit unit, PetscReal scale) 130cb1e1211SMatthew G Knepley { 131cb1e1211SMatthew G Knepley DM_Plex *mesh = (DM_Plex*) dm->data; 132cb1e1211SMatthew G Knepley 133cb1e1211SMatthew G Knepley PetscFunctionBegin; 134cb1e1211SMatthew G Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 135cb1e1211SMatthew G Knepley mesh->scale[unit] = scale; 136cb1e1211SMatthew G Knepley PetscFunctionReturn(0); 137cb1e1211SMatthew G Knepley } 138cb1e1211SMatthew G Knepley 139d1828a1cSMatthew G. Knepley static PetscErrorCode DMPlexProjectRigidBody_Private(PetscInt dim, PetscReal t, const PetscReal X[], PetscInt Nf, PetscScalar *mode, void *ctx) 140026175e5SToby Isaac { 14112adca46SMatthew G. Knepley const PetscInt eps[3][3][3] = {{{0, 0, 0}, {0, 0, 1}, {0, -1, 0}}, {{0, 0, -1}, {0, 0, 0}, {1, 0, 0}}, {{0, 1, 0}, {-1, 0, 0}, {0, 0, 0}}}; 142026175e5SToby Isaac PetscInt *ctxInt = (PetscInt *) ctx; 143ad917190SMatthew G. Knepley PetscInt dim2 = ctxInt[0]; 144026175e5SToby Isaac PetscInt d = ctxInt[1]; 145026175e5SToby Isaac PetscInt i, j, k = dim > 2 ? d - dim : d; 146026175e5SToby Isaac 147ad917190SMatthew G. Knepley PetscFunctionBegin; 148ff1e0c32SBarry Smith if (dim != dim2) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Input dimension %D does not match context dimension %D", dim, dim2); 149026175e5SToby Isaac for (i = 0; i < dim; i++) mode[i] = 0.; 150026175e5SToby Isaac if (d < dim) { 15112adca46SMatthew G. Knepley mode[d] = 1.; /* Translation along axis d */ 152ad917190SMatthew G. Knepley } else { 153026175e5SToby Isaac for (i = 0; i < dim; i++) { 154026175e5SToby Isaac for (j = 0; j < dim; j++) { 15512adca46SMatthew G. Knepley mode[j] += eps[i][j][k]*X[i]; /* Rotation about axis d */ 156026175e5SToby Isaac } 157026175e5SToby Isaac } 158026175e5SToby Isaac } 159ad917190SMatthew G. Knepley PetscFunctionReturn(0); 160026175e5SToby Isaac } 161026175e5SToby Isaac 162cc4e42d9SMartin Diehl /*@ 16312adca46SMatthew G. Knepley DMPlexCreateRigidBody - For the default global section, create rigid body modes by function space interpolation 164cb1e1211SMatthew G Knepley 165d083f849SBarry Smith Collective on dm 166cb1e1211SMatthew G Knepley 167cb1e1211SMatthew G Knepley Input Arguments: 168026175e5SToby Isaac . dm - the DM 169cb1e1211SMatthew G Knepley 170cb1e1211SMatthew G Knepley Output Argument: 171cb1e1211SMatthew G Knepley . sp - the null space 172cb1e1211SMatthew G Knepley 17312adca46SMatthew G. Knepley Note: This is necessary to provide a suitable coarse space for algebraic multigrid 174cb1e1211SMatthew G Knepley 175cb1e1211SMatthew G Knepley Level: advanced 176cb1e1211SMatthew G Knepley 17712adca46SMatthew G. Knepley .seealso: MatNullSpaceCreate(), PCGAMG 178cb1e1211SMatthew G Knepley @*/ 179026175e5SToby Isaac PetscErrorCode DMPlexCreateRigidBody(DM dm, MatNullSpace *sp) 180cb1e1211SMatthew G Knepley { 181cb1e1211SMatthew G Knepley MPI_Comm comm; 182026175e5SToby Isaac Vec mode[6]; 183026175e5SToby Isaac PetscSection section, globalSection; 184b247467aSMatthew G. Knepley PetscInt dim, dimEmbed, n, m, mmin, d, i, j; 185cb1e1211SMatthew G Knepley PetscErrorCode ierr; 186cb1e1211SMatthew G Knepley 187cb1e1211SMatthew G Knepley PetscFunctionBegin; 188cb1e1211SMatthew G Knepley ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr); 189c73cfb54SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 1909d8fbdeaSMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &dimEmbed);CHKERRQ(ierr); 191cb1e1211SMatthew G Knepley if (dim == 1) { 192cb1e1211SMatthew G Knepley ierr = MatNullSpaceCreate(comm, PETSC_TRUE, 0, NULL, sp);CHKERRQ(ierr); 193cb1e1211SMatthew G Knepley PetscFunctionReturn(0); 194cb1e1211SMatthew G Knepley } 19592fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 196e87a4003SBarry Smith ierr = DMGetGlobalSection(dm, &globalSection);CHKERRQ(ierr); 197cb1e1211SMatthew G Knepley ierr = PetscSectionGetConstrainedStorageSize(globalSection, &n);CHKERRQ(ierr); 198b247467aSMatthew G. Knepley m = (dim*(dim+1))/2; 199cb1e1211SMatthew G Knepley ierr = VecCreate(comm, &mode[0]);CHKERRQ(ierr); 200cb1e1211SMatthew G Knepley ierr = VecSetSizes(mode[0], n, PETSC_DETERMINE);CHKERRQ(ierr); 201cb1e1211SMatthew G Knepley ierr = VecSetUp(mode[0]);CHKERRQ(ierr); 202b247467aSMatthew G. Knepley ierr = VecGetSize(mode[0], &n);CHKERRQ(ierr); 203b247467aSMatthew G. Knepley mmin = PetscMin(m, n); 204cb1e1211SMatthew G Knepley for (i = 1; i < m; ++i) {ierr = VecDuplicate(mode[0], &mode[i]);CHKERRQ(ierr);} 205026175e5SToby Isaac for (d = 0; d < m; d++) { 206330485fdSToby Isaac PetscInt ctx[2]; 207d1828a1cSMatthew G. Knepley PetscErrorCode (*func)(PetscInt, PetscReal, const PetscReal *, PetscInt, PetscScalar *, void *) = DMPlexProjectRigidBody_Private; 208026175e5SToby Isaac void *voidctx = (void *) (&ctx[0]); 209cb1e1211SMatthew G Knepley 2109d8fbdeaSMatthew G. Knepley ctx[0] = dimEmbed; 211330485fdSToby Isaac ctx[1] = d; 2120709b2feSToby Isaac ierr = DMProjectFunction(dm, 0.0, &func, &voidctx, INSERT_VALUES, mode[d]);CHKERRQ(ierr); 213cb1e1211SMatthew G Knepley } 2143b2202bfSJacob Faibussowitsch /* Orthonormalize system */ 215b50a2c0aSJacob Faibussowitsch for (i = 0; i < mmin; ++i) { 216b77f2eeeSJacob Faibussowitsch PetscScalar dots[6]; 217b50a2c0aSJacob Faibussowitsch 2189019a6d7SJacob Faibussowitsch ierr = VecNormalize(mode[i], NULL);CHKERRQ(ierr); 219b77f2eeeSJacob Faibussowitsch ierr = VecMDot(mode[i], mmin-i-1, mode+i+1, dots+i+1);CHKERRQ(ierr); 220b50a2c0aSJacob Faibussowitsch for (j = i+1; j < mmin; ++j) { 221b77f2eeeSJacob Faibussowitsch dots[j] *= -1.0; 222b77f2eeeSJacob Faibussowitsch ierr = VecAXPY(mode[j], dots[j], mode[i]);CHKERRQ(ierr); 223b50a2c0aSJacob Faibussowitsch } 224cb1e1211SMatthew G Knepley } 225b247467aSMatthew G. Knepley ierr = MatNullSpaceCreate(comm, PETSC_FALSE, mmin, mode, sp);CHKERRQ(ierr); 226b247467aSMatthew G. Knepley for (i = 0; i < m; ++i) {ierr = VecDestroy(&mode[i]);CHKERRQ(ierr);} 227cb1e1211SMatthew G Knepley PetscFunctionReturn(0); 228cb1e1211SMatthew G Knepley } 229cb1e1211SMatthew G Knepley 230cc4e42d9SMartin Diehl /*@ 23112adca46SMatthew G. Knepley DMPlexCreateRigidBodies - For the default global section, create rigid body modes by function space interpolation 23212adca46SMatthew G. Knepley 233d083f849SBarry Smith Collective on dm 23412adca46SMatthew G. Knepley 23512adca46SMatthew G. Knepley Input Arguments: 23612adca46SMatthew G. Knepley + dm - the DM 23712adca46SMatthew G. Knepley . nb - The number of bodies 23812adca46SMatthew G. Knepley . label - The DMLabel marking each domain 23912adca46SMatthew G. Knepley . nids - The number of ids per body 24012adca46SMatthew G. Knepley - ids - An array of the label ids in sequence for each domain 24112adca46SMatthew G. Knepley 24212adca46SMatthew G. Knepley Output Argument: 24312adca46SMatthew G. Knepley . sp - the null space 24412adca46SMatthew G. Knepley 24512adca46SMatthew G. Knepley Note: This is necessary to provide a suitable coarse space for algebraic multigrid 24612adca46SMatthew G. Knepley 24712adca46SMatthew G. Knepley Level: advanced 24812adca46SMatthew G. Knepley 24912adca46SMatthew G. Knepley .seealso: MatNullSpaceCreate() 25012adca46SMatthew G. Knepley @*/ 25112adca46SMatthew G. Knepley PetscErrorCode DMPlexCreateRigidBodies(DM dm, PetscInt nb, DMLabel label, const PetscInt nids[], const PetscInt ids[], MatNullSpace *sp) 25212adca46SMatthew G. Knepley { 25312adca46SMatthew G. Knepley MPI_Comm comm; 25412adca46SMatthew G. Knepley PetscSection section, globalSection; 25512adca46SMatthew G. Knepley Vec *mode; 25612adca46SMatthew G. Knepley PetscScalar *dots; 25712adca46SMatthew G. Knepley PetscInt dim, dimEmbed, n, m, b, d, i, j, off; 25812adca46SMatthew G. Knepley PetscErrorCode ierr; 25912adca46SMatthew G. Knepley 26012adca46SMatthew G. Knepley PetscFunctionBegin; 26112adca46SMatthew G. Knepley ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr); 26212adca46SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 26312adca46SMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &dimEmbed);CHKERRQ(ierr); 26492fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 265e87a4003SBarry Smith ierr = DMGetGlobalSection(dm, &globalSection);CHKERRQ(ierr); 26612adca46SMatthew G. Knepley ierr = PetscSectionGetConstrainedStorageSize(globalSection, &n);CHKERRQ(ierr); 26712adca46SMatthew G. Knepley m = nb * (dim*(dim+1))/2; 26812adca46SMatthew G. Knepley ierr = PetscMalloc2(m, &mode, m, &dots);CHKERRQ(ierr); 26912adca46SMatthew G. Knepley ierr = VecCreate(comm, &mode[0]);CHKERRQ(ierr); 27012adca46SMatthew G. Knepley ierr = VecSetSizes(mode[0], n, PETSC_DETERMINE);CHKERRQ(ierr); 27112adca46SMatthew G. Knepley ierr = VecSetUp(mode[0]);CHKERRQ(ierr); 27212adca46SMatthew G. Knepley for (i = 1; i < m; ++i) {ierr = VecDuplicate(mode[0], &mode[i]);CHKERRQ(ierr);} 27312adca46SMatthew G. Knepley for (b = 0, off = 0; b < nb; ++b) { 27412adca46SMatthew G. Knepley for (d = 0; d < m/nb; ++d) { 27512adca46SMatthew G. Knepley PetscInt ctx[2]; 27612adca46SMatthew G. Knepley PetscErrorCode (*func)(PetscInt, PetscReal, const PetscReal *, PetscInt, PetscScalar *, void *) = DMPlexProjectRigidBody_Private; 27712adca46SMatthew G. Knepley void *voidctx = (void *) (&ctx[0]); 27812adca46SMatthew G. Knepley 27912adca46SMatthew G. Knepley ctx[0] = dimEmbed; 28012adca46SMatthew G. Knepley ctx[1] = d; 28112adca46SMatthew G. Knepley ierr = DMProjectFunctionLabel(dm, 0.0, label, nids[b], &ids[off], 0, NULL, &func, &voidctx, INSERT_VALUES, mode[d]);CHKERRQ(ierr); 28212adca46SMatthew G. Knepley off += nids[b]; 28312adca46SMatthew G. Knepley } 28412adca46SMatthew G. Knepley } 2853b2202bfSJacob Faibussowitsch /* Orthonormalize system */ 286606c1a1cSJacob Faibussowitsch for (i = 0; i < m; ++i) { 287b77f2eeeSJacob Faibussowitsch PetscScalar dots[6]; 2885a0e29b9SJacob Faibussowitsch 2899019a6d7SJacob Faibussowitsch ierr = VecNormalize(mode[i], NULL);CHKERRQ(ierr); 290b77f2eeeSJacob Faibussowitsch ierr = VecMDot(mode[i], m-i-1, mode+i+1, dots+i+1);CHKERRQ(ierr); 2915a0e29b9SJacob Faibussowitsch for (j = i+1; j < m; ++j) { 292b77f2eeeSJacob Faibussowitsch dots[j] *= -1.0; 293b77f2eeeSJacob Faibussowitsch ierr = VecAXPY(mode[j], dots[j], mode[i]);CHKERRQ(ierr); 2945a0e29b9SJacob Faibussowitsch } 29512adca46SMatthew G. Knepley } 29612adca46SMatthew G. Knepley ierr = MatNullSpaceCreate(comm, PETSC_FALSE, m, mode, sp);CHKERRQ(ierr); 29712adca46SMatthew G. Knepley for (i = 0; i< m; ++i) {ierr = VecDestroy(&mode[i]);CHKERRQ(ierr);} 29812adca46SMatthew G. Knepley ierr = PetscFree2(mode, dots);CHKERRQ(ierr); 29912adca46SMatthew G. Knepley PetscFunctionReturn(0); 30012adca46SMatthew G. Knepley } 30112adca46SMatthew G. Knepley 302b29cfa1cSToby Isaac /*@ 303b29cfa1cSToby Isaac DMPlexSetMaxProjectionHeight - In DMPlexProjectXXXLocal() functions, the projected values of a basis function's dofs 304b29cfa1cSToby Isaac are computed by associating the basis function with one of the mesh points in its transitively-closed support, and 305b29cfa1cSToby Isaac evaluating the dual space basis of that point. A basis function is associated with the point in its 306b29cfa1cSToby Isaac transitively-closed support whose mesh height is highest (w.r.t. DAG height), but not greater than the maximum 307b29cfa1cSToby Isaac projection height, which is set with this function. By default, the maximum projection height is zero, which means 308b29cfa1cSToby Isaac that only mesh cells are used to project basis functions. A height of one, for example, evaluates a cell-interior 309b29cfa1cSToby Isaac basis functions using its cells dual space basis, but all other basis functions with the dual space basis of a face. 310b29cfa1cSToby Isaac 311b29cfa1cSToby Isaac Input Parameters: 312b29cfa1cSToby Isaac + dm - the DMPlex object 313b29cfa1cSToby Isaac - height - the maximum projection height >= 0 314b29cfa1cSToby Isaac 315b29cfa1cSToby Isaac Level: advanced 316b29cfa1cSToby Isaac 3174d6f44ffSToby Isaac .seealso: DMPlexGetMaxProjectionHeight(), DMProjectFunctionLocal(), DMProjectFunctionLabelLocal() 318b29cfa1cSToby Isaac @*/ 319b29cfa1cSToby Isaac PetscErrorCode DMPlexSetMaxProjectionHeight(DM dm, PetscInt height) 320b29cfa1cSToby Isaac { 321b29cfa1cSToby Isaac DM_Plex *plex = (DM_Plex *) dm->data; 322b29cfa1cSToby Isaac 323b29cfa1cSToby Isaac PetscFunctionBegin; 324b29cfa1cSToby Isaac PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 325b29cfa1cSToby Isaac plex->maxProjectionHeight = height; 326b29cfa1cSToby Isaac PetscFunctionReturn(0); 327b29cfa1cSToby Isaac } 328b29cfa1cSToby Isaac 329b29cfa1cSToby Isaac /*@ 330b29cfa1cSToby Isaac DMPlexGetMaxProjectionHeight - Get the maximum height (w.r.t. DAG) of mesh points used to evaluate dual bases in 331b29cfa1cSToby Isaac DMPlexProjectXXXLocal() functions. 332b29cfa1cSToby Isaac 333b29cfa1cSToby Isaac Input Parameters: 334b29cfa1cSToby Isaac . dm - the DMPlex object 335b29cfa1cSToby Isaac 336b29cfa1cSToby Isaac Output Parameters: 337b29cfa1cSToby Isaac . height - the maximum projection height 338b29cfa1cSToby Isaac 339b29cfa1cSToby Isaac Level: intermediate 340b29cfa1cSToby Isaac 3414d6f44ffSToby Isaac .seealso: DMPlexSetMaxProjectionHeight(), DMProjectFunctionLocal(), DMProjectFunctionLabelLocal() 342b29cfa1cSToby Isaac @*/ 343b29cfa1cSToby Isaac PetscErrorCode DMPlexGetMaxProjectionHeight(DM dm, PetscInt *height) 344b29cfa1cSToby Isaac { 345b29cfa1cSToby Isaac DM_Plex *plex = (DM_Plex *) dm->data; 346b29cfa1cSToby Isaac 347b29cfa1cSToby Isaac PetscFunctionBegin; 348b29cfa1cSToby Isaac PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 349b29cfa1cSToby Isaac *height = plex->maxProjectionHeight; 350b29cfa1cSToby Isaac PetscFunctionReturn(0); 351b29cfa1cSToby Isaac } 352b29cfa1cSToby Isaac 353ca3d3a14SMatthew G. Knepley typedef struct { 354ca3d3a14SMatthew G. Knepley PetscReal alpha; /* The first Euler angle, and in 2D the only one */ 355ca3d3a14SMatthew G. Knepley PetscReal beta; /* The second Euler angle */ 356ca3d3a14SMatthew G. Knepley PetscReal gamma; /* The third Euler angle */ 357ca3d3a14SMatthew G. Knepley PetscInt dim; /* The dimension of R */ 358ca3d3a14SMatthew G. Knepley PetscScalar *R; /* The rotation matrix, transforming a vector in the local basis to the global basis */ 359ca3d3a14SMatthew G. Knepley PetscScalar *RT; /* The transposed rotation matrix, transforming a vector in the global basis to the local basis */ 360ca3d3a14SMatthew G. Knepley } RotCtx; 361ca3d3a14SMatthew G. Knepley 362ca3d3a14SMatthew G. Knepley /* 363ca3d3a14SMatthew G. Knepley Note: Following https://en.wikipedia.org/wiki/Euler_angles, we will specify Euler angles by extrinsic rotations, meaning that 364ca3d3a14SMatthew G. Knepley we rotate with respect to a fixed initial coordinate system, the local basis (x-y-z). The global basis (X-Y-Z) is reached as follows: 365ca3d3a14SMatthew G. Knepley $ The XYZ system rotates about the z axis by alpha. The X axis is now at angle alpha with respect to the x axis. 366ca3d3a14SMatthew G. Knepley $ The XYZ system rotates again about the x axis by beta. The Z axis is now at angle beta with respect to the z axis. 367ca3d3a14SMatthew G. Knepley $ The XYZ system rotates a third time about the z axis by gamma. 368ca3d3a14SMatthew G. Knepley */ 369ca3d3a14SMatthew G. Knepley static PetscErrorCode DMPlexBasisTransformSetUp_Rotation_Internal(DM dm, void *ctx) 370ca3d3a14SMatthew G. Knepley { 371ca3d3a14SMatthew G. Knepley RotCtx *rc = (RotCtx *) ctx; 372ca3d3a14SMatthew G. Knepley PetscInt dim = rc->dim; 373ca3d3a14SMatthew G. Knepley PetscReal c1, s1, c2, s2, c3, s3; 374ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 375ca3d3a14SMatthew G. Knepley 376ca3d3a14SMatthew G. Knepley PetscFunctionBegin; 377ca3d3a14SMatthew G. Knepley ierr = PetscMalloc2(PetscSqr(dim), &rc->R, PetscSqr(dim), &rc->RT);CHKERRQ(ierr); 378ca3d3a14SMatthew G. Knepley switch (dim) { 379ca3d3a14SMatthew G. Knepley case 2: 380ca3d3a14SMatthew G. Knepley c1 = PetscCosReal(rc->alpha);s1 = PetscSinReal(rc->alpha); 381ca3d3a14SMatthew G. Knepley rc->R[0] = c1;rc->R[1] = s1; 382ca3d3a14SMatthew G. Knepley rc->R[2] = -s1;rc->R[3] = c1; 383580bdb30SBarry Smith ierr = PetscArraycpy(rc->RT, rc->R, PetscSqr(dim));CHKERRQ(ierr); 384ca3d3a14SMatthew G. Knepley DMPlex_Transpose2D_Internal(rc->RT);break; 385ca3d3a14SMatthew G. Knepley break; 386ca3d3a14SMatthew G. Knepley case 3: 387ca3d3a14SMatthew G. Knepley c1 = PetscCosReal(rc->alpha);s1 = PetscSinReal(rc->alpha); 388ca3d3a14SMatthew G. Knepley c2 = PetscCosReal(rc->beta); s2 = PetscSinReal(rc->beta); 389ca3d3a14SMatthew G. Knepley c3 = PetscCosReal(rc->gamma);s3 = PetscSinReal(rc->gamma); 390ca3d3a14SMatthew G. Knepley rc->R[0] = c1*c3 - c2*s1*s3;rc->R[1] = c3*s1 + c1*c2*s3;rc->R[2] = s2*s3; 391ca3d3a14SMatthew G. Knepley rc->R[3] = -c1*s3 - c2*c3*s1;rc->R[4] = c1*c2*c3 - s1*s3; rc->R[5] = c3*s2; 392ca3d3a14SMatthew G. Knepley rc->R[6] = s1*s2; rc->R[7] = -c1*s2; rc->R[8] = c2; 393580bdb30SBarry Smith ierr = PetscArraycpy(rc->RT, rc->R, PetscSqr(dim));CHKERRQ(ierr); 394ca3d3a14SMatthew G. Knepley DMPlex_Transpose3D_Internal(rc->RT);break; 395ca3d3a14SMatthew G. Knepley break; 396ca3d3a14SMatthew G. Knepley default: SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_OUTOFRANGE, "Dimension %D not supported", dim); 397ca3d3a14SMatthew G. Knepley } 398ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 399ca3d3a14SMatthew G. Knepley } 400ca3d3a14SMatthew G. Knepley 401ca3d3a14SMatthew G. Knepley static PetscErrorCode DMPlexBasisTransformDestroy_Rotation_Internal(DM dm, void *ctx) 402ca3d3a14SMatthew G. Knepley { 403ca3d3a14SMatthew G. Knepley RotCtx *rc = (RotCtx *) ctx; 404ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 405ca3d3a14SMatthew G. Knepley 406ca3d3a14SMatthew G. Knepley PetscFunctionBegin; 407ca3d3a14SMatthew G. Knepley ierr = PetscFree2(rc->R, rc->RT);CHKERRQ(ierr); 408ca3d3a14SMatthew G. Knepley ierr = PetscFree(rc);CHKERRQ(ierr); 409ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 410ca3d3a14SMatthew G. Knepley } 411ca3d3a14SMatthew G. Knepley 412ca3d3a14SMatthew G. Knepley static PetscErrorCode DMPlexBasisTransformGetMatrix_Rotation_Internal(DM dm, const PetscReal x[], PetscBool l2g, const PetscScalar **A, void *ctx) 413ca3d3a14SMatthew G. Knepley { 414ca3d3a14SMatthew G. Knepley RotCtx *rc = (RotCtx *) ctx; 415ca3d3a14SMatthew G. Knepley 416ca3d3a14SMatthew G. Knepley PetscFunctionBeginHot; 417ca3d3a14SMatthew G. Knepley PetscValidPointer(ctx, 5); 418ca3d3a14SMatthew G. Knepley if (l2g) {*A = rc->R;} 419ca3d3a14SMatthew G. Knepley else {*A = rc->RT;} 420ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 421ca3d3a14SMatthew G. Knepley } 422ca3d3a14SMatthew G. Knepley 423ab6a9622SMatthew G. Knepley PetscErrorCode DMPlexBasisTransformApplyReal_Internal(DM dm, const PetscReal x[], PetscBool l2g, PetscInt dim, const PetscReal *y, PetscReal *z, void *ctx) 424ab6a9622SMatthew G. Knepley { 425ec277c0fSMatthew G. Knepley PetscErrorCode ierr; 426ec277c0fSMatthew G. Knepley 427ec277c0fSMatthew G. Knepley PetscFunctionBegin; 428ab6a9622SMatthew G. Knepley #if defined(PETSC_USE_COMPLEX) 429ab6a9622SMatthew G. Knepley switch (dim) { 430ab6a9622SMatthew G. Knepley case 2: 431ab6a9622SMatthew G. Knepley { 432ab6a9622SMatthew G. Knepley PetscScalar yt[2], zt[2]; 433ab6a9622SMatthew G. Knepley 434ab6a9622SMatthew G. Knepley yt[0] = y[0]; yt[1] = y[1]; 435ec277c0fSMatthew G. Knepley ierr = DMPlexBasisTransformApply_Internal(dm, x, l2g, dim, yt, zt, ctx);CHKERRQ(ierr); 436ab6a9622SMatthew G. Knepley z[0] = PetscRealPart(zt[0]); z[1] = PetscRealPart(zt[1]); 437ab6a9622SMatthew G. Knepley } 4389b4815dcSMatthew G. Knepley break; 439ab6a9622SMatthew G. Knepley case 3: 440ab6a9622SMatthew G. Knepley { 441ab6a9622SMatthew G. Knepley PetscScalar yt[3], zt[3]; 442ab6a9622SMatthew G. Knepley 443ab6a9622SMatthew G. Knepley yt[0] = y[0]; yt[1] = y[1]; yt[2] = y[2]; 444ec277c0fSMatthew G. Knepley ierr = DMPlexBasisTransformApply_Internal(dm, x, l2g, dim, yt, zt, ctx);CHKERRQ(ierr); 445ab6a9622SMatthew G. Knepley z[0] = PetscRealPart(zt[0]); z[1] = PetscRealPart(zt[1]); z[2] = PetscRealPart(zt[2]); 446ab6a9622SMatthew G. Knepley } 4479b4815dcSMatthew G. Knepley break; 448ab6a9622SMatthew G. Knepley } 449ab6a9622SMatthew G. Knepley #else 450ab6a9622SMatthew G. Knepley ierr = DMPlexBasisTransformApply_Internal(dm, x, l2g, dim, y, z, ctx);CHKERRQ(ierr); 451ab6a9622SMatthew G. Knepley #endif 452ec277c0fSMatthew G. Knepley PetscFunctionReturn(0); 453ab6a9622SMatthew G. Knepley } 454ab6a9622SMatthew G. Knepley 455ec277c0fSMatthew G. Knepley PetscErrorCode DMPlexBasisTransformApply_Internal(DM dm, const PetscReal x[], PetscBool l2g, PetscInt dim, const PetscScalar *y, PetscScalar *z, void *ctx) 456ca3d3a14SMatthew G. Knepley { 457ca3d3a14SMatthew G. Knepley const PetscScalar *A; 458ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 459ca3d3a14SMatthew G. Knepley 460ca3d3a14SMatthew G. Knepley PetscFunctionBeginHot; 461ca3d3a14SMatthew G. Knepley ierr = (*dm->transformGetMatrix)(dm, x, l2g, &A, ctx);CHKERRQ(ierr); 462ca3d3a14SMatthew G. Knepley switch (dim) { 4631fba962aSMatthew G. Knepley case 2: DMPlex_Mult2D_Internal(A, 1, y, z);break; 4641fba962aSMatthew G. Knepley case 3: DMPlex_Mult3D_Internal(A, 1, y, z);break; 465ca3d3a14SMatthew G. Knepley } 466ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 467ca3d3a14SMatthew G. Knepley } 468ca3d3a14SMatthew G. Knepley 469ca3d3a14SMatthew G. Knepley static PetscErrorCode DMPlexBasisTransformField_Internal(DM dm, DM tdm, Vec tv, PetscInt p, PetscInt f, PetscBool l2g, PetscScalar *a) 470ca3d3a14SMatthew G. Knepley { 471ca3d3a14SMatthew G. Knepley PetscSection ts; 472ca3d3a14SMatthew G. Knepley const PetscScalar *ta, *tva; 473ca3d3a14SMatthew G. Knepley PetscInt dof; 474ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 475ca3d3a14SMatthew G. Knepley 476ca3d3a14SMatthew G. Knepley PetscFunctionBeginHot; 47792fd8e1eSJed Brown ierr = DMGetLocalSection(tdm, &ts);CHKERRQ(ierr); 478ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetFieldDof(ts, p, f, &dof);CHKERRQ(ierr); 479ca3d3a14SMatthew G. Knepley ierr = VecGetArrayRead(tv, &ta);CHKERRQ(ierr); 480ca3d3a14SMatthew G. Knepley ierr = DMPlexPointLocalFieldRead(tdm, p, f, ta, (void *) &tva);CHKERRQ(ierr); 481ca3d3a14SMatthew G. Knepley if (l2g) { 482ca3d3a14SMatthew G. Knepley switch (dof) { 4831fba962aSMatthew G. Knepley case 4: DMPlex_Mult2D_Internal(tva, 1, a, a);break; 4841fba962aSMatthew G. Knepley case 9: DMPlex_Mult3D_Internal(tva, 1, a, a);break; 485ca3d3a14SMatthew G. Knepley } 486ca3d3a14SMatthew G. Knepley } else { 487ca3d3a14SMatthew G. Knepley switch (dof) { 4881fba962aSMatthew G. Knepley case 4: DMPlex_MultTranspose2D_Internal(tva, 1, a, a);break; 4891fba962aSMatthew G. Knepley case 9: DMPlex_MultTranspose3D_Internal(tva, 1, a, a);break; 490ca3d3a14SMatthew G. Knepley } 491ca3d3a14SMatthew G. Knepley } 492ca3d3a14SMatthew G. Knepley ierr = VecRestoreArrayRead(tv, &ta);CHKERRQ(ierr); 493ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 494ca3d3a14SMatthew G. Knepley } 495ca3d3a14SMatthew G. Knepley 496ca3d3a14SMatthew G. Knepley static PetscErrorCode DMPlexBasisTransformFieldTensor_Internal(DM dm, DM tdm, Vec tv, PetscInt pf, PetscInt f, PetscInt pg, PetscInt g, PetscBool l2g, PetscInt lda, PetscScalar *a) 497ca3d3a14SMatthew G. Knepley { 498ca3d3a14SMatthew G. Knepley PetscSection s, ts; 499ca3d3a14SMatthew G. Knepley const PetscScalar *ta, *tvaf, *tvag; 500ca3d3a14SMatthew G. Knepley PetscInt fdof, gdof, fpdof, gpdof; 501ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 502ca3d3a14SMatthew G. Knepley 503ca3d3a14SMatthew G. Knepley PetscFunctionBeginHot; 50492fd8e1eSJed Brown ierr = DMGetLocalSection(dm, &s);CHKERRQ(ierr); 50592fd8e1eSJed Brown ierr = DMGetLocalSection(tdm, &ts);CHKERRQ(ierr); 506ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetFieldDof(s, pf, f, &fpdof);CHKERRQ(ierr); 507ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetFieldDof(s, pg, g, &gpdof);CHKERRQ(ierr); 508ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetFieldDof(ts, pf, f, &fdof);CHKERRQ(ierr); 509ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetFieldDof(ts, pg, g, &gdof);CHKERRQ(ierr); 510ca3d3a14SMatthew G. Knepley ierr = VecGetArrayRead(tv, &ta);CHKERRQ(ierr); 511ca3d3a14SMatthew G. Knepley ierr = DMPlexPointLocalFieldRead(tdm, pf, f, ta, (void *) &tvaf);CHKERRQ(ierr); 512ca3d3a14SMatthew G. Knepley ierr = DMPlexPointLocalFieldRead(tdm, pg, g, ta, (void *) &tvag);CHKERRQ(ierr); 513ca3d3a14SMatthew G. Knepley if (l2g) { 514ca3d3a14SMatthew G. Knepley switch (fdof) { 515ca3d3a14SMatthew G. Knepley case 4: DMPlex_MatMult2D_Internal(tvaf, gpdof, lda, a, a);break; 516ca3d3a14SMatthew G. Knepley case 9: DMPlex_MatMult3D_Internal(tvaf, gpdof, lda, a, a);break; 517ca3d3a14SMatthew G. Knepley } 518ca3d3a14SMatthew G. Knepley switch (gdof) { 519ca3d3a14SMatthew G. Knepley case 4: DMPlex_MatMultTransposeLeft2D_Internal(tvag, fpdof, lda, a, a);break; 520ca3d3a14SMatthew G. Knepley case 9: DMPlex_MatMultTransposeLeft3D_Internal(tvag, fpdof, lda, a, a);break; 521ca3d3a14SMatthew G. Knepley } 522ca3d3a14SMatthew G. Knepley } else { 523ca3d3a14SMatthew G. Knepley switch (fdof) { 524ca3d3a14SMatthew G. Knepley case 4: DMPlex_MatMultTranspose2D_Internal(tvaf, gpdof, lda, a, a);break; 525ca3d3a14SMatthew G. Knepley case 9: DMPlex_MatMultTranspose3D_Internal(tvaf, gpdof, lda, a, a);break; 526ca3d3a14SMatthew G. Knepley } 527ca3d3a14SMatthew G. Knepley switch (gdof) { 528ca3d3a14SMatthew G. Knepley case 4: DMPlex_MatMultLeft2D_Internal(tvag, fpdof, lda, a, a);break; 529ca3d3a14SMatthew G. Knepley case 9: DMPlex_MatMultLeft3D_Internal(tvag, fpdof, lda, a, a);break; 530ca3d3a14SMatthew G. Knepley } 531ca3d3a14SMatthew G. Knepley } 532ca3d3a14SMatthew G. Knepley ierr = VecRestoreArrayRead(tv, &ta);CHKERRQ(ierr); 533ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 534ca3d3a14SMatthew G. Knepley } 535ca3d3a14SMatthew G. Knepley 536ca3d3a14SMatthew G. Knepley PetscErrorCode DMPlexBasisTransformPoint_Internal(DM dm, DM tdm, Vec tv, PetscInt p, PetscBool fieldActive[], PetscBool l2g, PetscScalar *a) 537ca3d3a14SMatthew G. Knepley { 538ca3d3a14SMatthew G. Knepley PetscSection s; 539ca3d3a14SMatthew G. Knepley PetscSection clSection; 540ca3d3a14SMatthew G. Knepley IS clPoints; 541ca3d3a14SMatthew G. Knepley const PetscInt *clp; 542ca3d3a14SMatthew G. Knepley PetscInt *points = NULL; 543ca3d3a14SMatthew G. Knepley PetscInt Nf, f, Np, cp, dof, d = 0; 544ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 545ca3d3a14SMatthew G. Knepley 546ca3d3a14SMatthew G. Knepley PetscFunctionBegin; 54792fd8e1eSJed Brown ierr = DMGetLocalSection(dm, &s);CHKERRQ(ierr); 548ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetNumFields(s, &Nf);CHKERRQ(ierr); 549ca3d3a14SMatthew G. Knepley ierr = DMPlexGetCompressedClosure(dm, s, p, &Np, &points, &clSection, &clPoints, &clp);CHKERRQ(ierr); 550ca3d3a14SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 551ca3d3a14SMatthew G. Knepley for (cp = 0; cp < Np*2; cp += 2) { 552ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetFieldDof(s, points[cp], f, &dof);CHKERRQ(ierr); 553ca3d3a14SMatthew G. Knepley if (!dof) continue; 554ca3d3a14SMatthew G. Knepley if (fieldActive[f]) {ierr = DMPlexBasisTransformField_Internal(dm, tdm, tv, points[cp], f, l2g, &a[d]);CHKERRQ(ierr);} 555ca3d3a14SMatthew G. Knepley d += dof; 556ca3d3a14SMatthew G. Knepley } 557ca3d3a14SMatthew G. Knepley } 558ca3d3a14SMatthew G. Knepley ierr = DMPlexRestoreCompressedClosure(dm, s, p, &Np, &points, &clSection, &clPoints, &clp);CHKERRQ(ierr); 559ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 560ca3d3a14SMatthew G. Knepley } 561ca3d3a14SMatthew G. Knepley 562ca3d3a14SMatthew G. Knepley PetscErrorCode DMPlexBasisTransformPointTensor_Internal(DM dm, DM tdm, Vec tv, PetscInt p, PetscBool l2g, PetscInt lda, PetscScalar *a) 563ca3d3a14SMatthew G. Knepley { 564ca3d3a14SMatthew G. Knepley PetscSection s; 565ca3d3a14SMatthew G. Knepley PetscSection clSection; 566ca3d3a14SMatthew G. Knepley IS clPoints; 567ca3d3a14SMatthew G. Knepley const PetscInt *clp; 568ca3d3a14SMatthew G. Knepley PetscInt *points = NULL; 5698bdb3c71SMatthew G. Knepley PetscInt Nf, f, g, Np, cpf, cpg, fdof, gdof, r, c = 0; 570ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 571ca3d3a14SMatthew G. Knepley 572ca3d3a14SMatthew G. Knepley PetscFunctionBegin; 57392fd8e1eSJed Brown ierr = DMGetLocalSection(dm, &s);CHKERRQ(ierr); 574ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetNumFields(s, &Nf);CHKERRQ(ierr); 575ca3d3a14SMatthew G. Knepley ierr = DMPlexGetCompressedClosure(dm, s, p, &Np, &points, &clSection, &clPoints, &clp);CHKERRQ(ierr); 576ca3d3a14SMatthew G. Knepley for (f = 0, r = 0; f < Nf; ++f) { 577ca3d3a14SMatthew G. Knepley for (cpf = 0; cpf < Np*2; cpf += 2) { 578ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetFieldDof(s, points[cpf], f, &fdof);CHKERRQ(ierr); 579ca3d3a14SMatthew G. Knepley for (g = 0, c = 0; g < Nf; ++g) { 580ca3d3a14SMatthew G. Knepley for (cpg = 0; cpg < Np*2; cpg += 2) { 581ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetFieldDof(s, points[cpg], g, &gdof);CHKERRQ(ierr); 582ca3d3a14SMatthew G. Knepley ierr = DMPlexBasisTransformFieldTensor_Internal(dm, tdm, tv, points[cpf], f, points[cpg], g, l2g, lda, &a[r*lda+c]);CHKERRQ(ierr); 583ca3d3a14SMatthew G. Knepley c += gdof; 584ca3d3a14SMatthew G. Knepley } 585ca3d3a14SMatthew G. Knepley } 586ca3d3a14SMatthew G. Knepley if (c != lda) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of columns %D should be %D", c, lda); 587ca3d3a14SMatthew G. Knepley r += fdof; 588ca3d3a14SMatthew G. Knepley } 589ca3d3a14SMatthew G. Knepley } 590ca3d3a14SMatthew G. Knepley if (r != lda) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Invalid number of rows %D should be %D", c, lda); 591ca3d3a14SMatthew G. Knepley ierr = DMPlexRestoreCompressedClosure(dm, s, p, &Np, &points, &clSection, &clPoints, &clp);CHKERRQ(ierr); 592ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 593ca3d3a14SMatthew G. Knepley } 594ca3d3a14SMatthew G. Knepley 595ca3d3a14SMatthew G. Knepley static PetscErrorCode DMPlexBasisTransform_Internal(DM dm, Vec lv, PetscBool l2g) 596ca3d3a14SMatthew G. Knepley { 597ca3d3a14SMatthew G. Knepley DM tdm; 598ca3d3a14SMatthew G. Knepley Vec tv; 599ca3d3a14SMatthew G. Knepley PetscSection ts, s; 600ca3d3a14SMatthew G. Knepley const PetscScalar *ta; 601ca3d3a14SMatthew G. Knepley PetscScalar *a, *va; 602ca3d3a14SMatthew G. Knepley PetscInt pStart, pEnd, p, Nf, f; 603ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 604ca3d3a14SMatthew G. Knepley 605ca3d3a14SMatthew G. Knepley PetscFunctionBegin; 606ca3d3a14SMatthew G. Knepley ierr = DMGetBasisTransformDM_Internal(dm, &tdm);CHKERRQ(ierr); 607ca3d3a14SMatthew G. Knepley ierr = DMGetBasisTransformVec_Internal(dm, &tv);CHKERRQ(ierr); 60892fd8e1eSJed Brown ierr = DMGetLocalSection(tdm, &ts);CHKERRQ(ierr); 60992fd8e1eSJed Brown ierr = DMGetLocalSection(dm, &s);CHKERRQ(ierr); 610ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetChart(s, &pStart, &pEnd);CHKERRQ(ierr); 611ca3d3a14SMatthew G. Knepley ierr = PetscSectionGetNumFields(s, &Nf);CHKERRQ(ierr); 612ca3d3a14SMatthew G. Knepley ierr = VecGetArray(lv, &a);CHKERRQ(ierr); 613ca3d3a14SMatthew G. Knepley ierr = VecGetArrayRead(tv, &ta);CHKERRQ(ierr); 614ca3d3a14SMatthew G. Knepley for (p = pStart; p < pEnd; ++p) { 615ca3d3a14SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 616ca3d3a14SMatthew G. Knepley ierr = DMPlexPointLocalFieldRef(dm, p, f, a, (void *) &va);CHKERRQ(ierr); 617ca3d3a14SMatthew G. Knepley ierr = DMPlexBasisTransformField_Internal(dm, tdm, tv, p, f, l2g, va);CHKERRQ(ierr); 618ca3d3a14SMatthew G. Knepley } 619ca3d3a14SMatthew G. Knepley } 620ca3d3a14SMatthew G. Knepley ierr = VecRestoreArray(lv, &a);CHKERRQ(ierr); 621ca3d3a14SMatthew G. Knepley ierr = VecRestoreArrayRead(tv, &ta);CHKERRQ(ierr); 622ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 623ca3d3a14SMatthew G. Knepley } 624ca3d3a14SMatthew G. Knepley 625ca3d3a14SMatthew G. Knepley /*@ 626ca3d3a14SMatthew G. Knepley DMPlexGlobalToLocalBasis - Transform the values in the given local vector from the global basis to the local basis 627ca3d3a14SMatthew G. Knepley 628ca3d3a14SMatthew G. Knepley Input Parameters: 629ca3d3a14SMatthew G. Knepley + dm - The DM 630ca3d3a14SMatthew G. Knepley - lv - A local vector with values in the global basis 631ca3d3a14SMatthew G. Knepley 632ca3d3a14SMatthew G. Knepley Output Parameters: 633ca3d3a14SMatthew G. Knepley . lv - A local vector with values in the local basis 634ca3d3a14SMatthew G. Knepley 635c0f8e1fdSMatthew G. Knepley Note: This method is only intended to be called inside DMGlobalToLocal(). It is unlikely that a user will have a local vector full of coefficients for the global basis unless they are reimplementing GlobalToLocal. 636c0f8e1fdSMatthew G. Knepley 637ca3d3a14SMatthew G. Knepley Level: developer 638ca3d3a14SMatthew G. Knepley 63992fd8e1eSJed Brown .seealso: DMPlexLocalToGlobalBasis(), DMGetLocalSection() 640ca3d3a14SMatthew G. Knepley @*/ 641ca3d3a14SMatthew G. Knepley PetscErrorCode DMPlexGlobalToLocalBasis(DM dm, Vec lv) 642ca3d3a14SMatthew G. Knepley { 643ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 644ca3d3a14SMatthew G. Knepley 645ca3d3a14SMatthew G. Knepley PetscFunctionBegin; 646ca3d3a14SMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 647ca3d3a14SMatthew G. Knepley PetscValidHeaderSpecific(lv, VEC_CLASSID, 2); 648ca3d3a14SMatthew G. Knepley ierr = DMPlexBasisTransform_Internal(dm, lv, PETSC_FALSE);CHKERRQ(ierr); 649ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 650ca3d3a14SMatthew G. Knepley } 651ca3d3a14SMatthew G. Knepley 652ca3d3a14SMatthew G. Knepley /*@ 653ca3d3a14SMatthew G. Knepley DMPlexLocalToGlobalBasis - Transform the values in the given local vector from the local basis to the global basis 654ca3d3a14SMatthew G. Knepley 655ca3d3a14SMatthew G. Knepley Input Parameters: 656ca3d3a14SMatthew G. Knepley + dm - The DM 657ca3d3a14SMatthew G. Knepley - lv - A local vector with values in the local basis 658ca3d3a14SMatthew G. Knepley 659ca3d3a14SMatthew G. Knepley Output Parameters: 660ca3d3a14SMatthew G. Knepley . lv - A local vector with values in the global basis 661ca3d3a14SMatthew G. Knepley 662c0f8e1fdSMatthew G. Knepley Note: This method is only intended to be called inside DMGlobalToLocal(). It is unlikely that a user would want a local vector full of coefficients for the global basis unless they are reimplementing GlobalToLocal. 663c0f8e1fdSMatthew G. Knepley 664ca3d3a14SMatthew G. Knepley Level: developer 665ca3d3a14SMatthew G. Knepley 66692fd8e1eSJed Brown .seealso: DMPlexGlobalToLocalBasis(), DMGetLocalSection() 667ca3d3a14SMatthew G. Knepley @*/ 668ca3d3a14SMatthew G. Knepley PetscErrorCode DMPlexLocalToGlobalBasis(DM dm, Vec lv) 669ca3d3a14SMatthew G. Knepley { 670ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 671ca3d3a14SMatthew G. Knepley 672ca3d3a14SMatthew G. Knepley PetscFunctionBegin; 673ca3d3a14SMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 674ca3d3a14SMatthew G. Knepley PetscValidHeaderSpecific(lv, VEC_CLASSID, 2); 675ca3d3a14SMatthew G. Knepley ierr = DMPlexBasisTransform_Internal(dm, lv, PETSC_TRUE);CHKERRQ(ierr); 676ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 677ca3d3a14SMatthew G. Knepley } 678ca3d3a14SMatthew G. Knepley 679ca3d3a14SMatthew G. Knepley /*@ 680ca3d3a14SMatthew G. Knepley DMPlexCreateBasisRotation - Create an internal transformation from the global basis, used to specify boundary conditions 681ca3d3a14SMatthew G. Knepley and global solutions, to a local basis, appropriate for discretization integrals and assembly. 682ca3d3a14SMatthew G. Knepley 683ca3d3a14SMatthew G. Knepley Input Parameters: 684ca3d3a14SMatthew G. Knepley + dm - The DM 685ca3d3a14SMatthew G. Knepley . alpha - The first Euler angle, and in 2D the only one 686ca3d3a14SMatthew G. Knepley . beta - The second Euler angle 687f0fc11ceSJed Brown - gamma - The third Euler angle 688ca3d3a14SMatthew G. Knepley 689ca3d3a14SMatthew G. Knepley Note: Following https://en.wikipedia.org/wiki/Euler_angles, we will specify Euler angles by extrinsic rotations, meaning that 690ca3d3a14SMatthew G. Knepley we rotate with respect to a fixed initial coordinate system, the local basis (x-y-z). The global basis (X-Y-Z) is reached as follows: 691ca3d3a14SMatthew G. Knepley $ The XYZ system rotates about the z axis by alpha. The X axis is now at angle alpha with respect to the x axis. 692ca3d3a14SMatthew G. Knepley $ The XYZ system rotates again about the x axis by beta. The Z axis is now at angle beta with respect to the z axis. 693ca3d3a14SMatthew G. Knepley $ The XYZ system rotates a third time about the z axis by gamma. 694ca3d3a14SMatthew G. Knepley 695ca3d3a14SMatthew G. Knepley Level: developer 696ca3d3a14SMatthew G. Knepley 697ca3d3a14SMatthew G. Knepley .seealso: DMPlexGlobalToLocalBasis(), DMPlexLocalToGlobalBasis() 698ca3d3a14SMatthew G. Knepley @*/ 699ca3d3a14SMatthew G. Knepley PetscErrorCode DMPlexCreateBasisRotation(DM dm, PetscReal alpha, PetscReal beta, PetscReal gamma) 700ca3d3a14SMatthew G. Knepley { 701ca3d3a14SMatthew G. Knepley RotCtx *rc; 702ca3d3a14SMatthew G. Knepley PetscInt cdim; 703ca3d3a14SMatthew G. Knepley PetscErrorCode ierr; 704ca3d3a14SMatthew G. Knepley 705ca3d3a14SMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &cdim);CHKERRQ(ierr); 706ca3d3a14SMatthew G. Knepley ierr = PetscMalloc1(1, &rc);CHKERRQ(ierr); 707ca3d3a14SMatthew G. Knepley dm->transformCtx = rc; 708ca3d3a14SMatthew G. Knepley dm->transformSetUp = DMPlexBasisTransformSetUp_Rotation_Internal; 709ca3d3a14SMatthew G. Knepley dm->transformDestroy = DMPlexBasisTransformDestroy_Rotation_Internal; 710ca3d3a14SMatthew G. Knepley dm->transformGetMatrix = DMPlexBasisTransformGetMatrix_Rotation_Internal; 711ca3d3a14SMatthew G. Knepley rc->dim = cdim; 712ca3d3a14SMatthew G. Knepley rc->alpha = alpha; 713ca3d3a14SMatthew G. Knepley rc->beta = beta; 714ca3d3a14SMatthew G. Knepley rc->gamma = gamma; 715ca3d3a14SMatthew G. Knepley ierr = (*dm->transformSetUp)(dm, dm->transformCtx);CHKERRQ(ierr); 716ca3d3a14SMatthew G. Knepley ierr = DMConstructBasisTransform_Internal(dm);CHKERRQ(ierr); 717ca3d3a14SMatthew G. Knepley PetscFunctionReturn(0); 718ca3d3a14SMatthew G. Knepley } 719ca3d3a14SMatthew G. Knepley 720b278463cSMatthew G. Knepley /*@C 721*ece3a9fcSMatthew G. Knepley DMPlexInsertBoundaryValuesEssential - Insert boundary values into a local vector using a function of the coordinates 722b278463cSMatthew G. Knepley 723b278463cSMatthew G. Knepley Input Parameters: 724b278463cSMatthew G. Knepley + dm - The DM, with a PetscDS that matches the problem being constrained 725b278463cSMatthew G. Knepley . time - The time 726b278463cSMatthew G. Knepley . field - The field to constrain 7271c531cf8SMatthew G. Knepley . Nc - The number of constrained field components, or 0 for all components 7281c531cf8SMatthew G. Knepley . comps - An array of constrained component numbers, or NULL for all components 729b278463cSMatthew G. Knepley . label - The DMLabel defining constrained points 730b278463cSMatthew G. Knepley . numids - The number of DMLabel ids for constrained points 731b278463cSMatthew G. Knepley . ids - An array of ids for constrained points 732b278463cSMatthew G. Knepley . func - A pointwise function giving boundary values 733b278463cSMatthew G. Knepley - ctx - An optional user context for bcFunc 734b278463cSMatthew G. Knepley 735b278463cSMatthew G. Knepley Output Parameter: 736b278463cSMatthew G. Knepley . locX - A local vector to receives the boundary values 737b278463cSMatthew G. Knepley 738b278463cSMatthew G. Knepley Level: developer 739b278463cSMatthew G. Knepley 740*ece3a9fcSMatthew G. Knepley .seealso: DMPlexInsertBoundaryValuesEssentialField(), DMPlexInsertBoundaryValuesEssentialBdField(), DMAddBoundary() 741b278463cSMatthew G. Knepley @*/ 7421c531cf8SMatthew G. Knepley PetscErrorCode DMPlexInsertBoundaryValuesEssential(DM dm, PetscReal time, PetscInt field, PetscInt Nc, const PetscInt comps[], DMLabel label, PetscInt numids, const PetscInt ids[], PetscErrorCode (*func)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar *, void *), void *ctx, Vec locX) 74355f2e967SMatthew G. Knepley { 7440163fd50SMatthew G. Knepley PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal x[], PetscInt, PetscScalar *u, void *ctx); 74555f2e967SMatthew G. Knepley void **ctxs; 746d7ddef95SMatthew G. Knepley PetscInt numFields; 747d7ddef95SMatthew G. Knepley PetscErrorCode ierr; 748d7ddef95SMatthew G. Knepley 749d7ddef95SMatthew G. Knepley PetscFunctionBegin; 750d7ddef95SMatthew G. Knepley ierr = DMGetNumFields(dm, &numFields);CHKERRQ(ierr); 751d7ddef95SMatthew G. Knepley ierr = PetscCalloc2(numFields,&funcs,numFields,&ctxs);CHKERRQ(ierr); 752d7ddef95SMatthew G. Knepley funcs[field] = func; 753d7ddef95SMatthew G. Knepley ctxs[field] = ctx; 7541c531cf8SMatthew G. Knepley ierr = DMProjectFunctionLabelLocal(dm, time, label, numids, ids, Nc, comps, funcs, ctxs, INSERT_BC_VALUES, locX);CHKERRQ(ierr); 755d7ddef95SMatthew G. Knepley ierr = PetscFree2(funcs,ctxs);CHKERRQ(ierr); 756d7ddef95SMatthew G. Knepley PetscFunctionReturn(0); 757d7ddef95SMatthew G. Knepley } 758d7ddef95SMatthew G. Knepley 759b278463cSMatthew G. Knepley /*@C 760*ece3a9fcSMatthew G. Knepley DMPlexInsertBoundaryValuesEssentialField - Insert boundary values into a local vector using a function of the coordinates and field data 761b278463cSMatthew G. Knepley 762b278463cSMatthew G. Knepley Input Parameters: 763b278463cSMatthew G. Knepley + dm - The DM, with a PetscDS that matches the problem being constrained 764b278463cSMatthew G. Knepley . time - The time 765b278463cSMatthew G. Knepley . locU - A local vector with the input solution values 766b278463cSMatthew G. Knepley . field - The field to constrain 7671c531cf8SMatthew G. Knepley . Nc - The number of constrained field components, or 0 for all components 7681c531cf8SMatthew G. Knepley . comps - An array of constrained component numbers, or NULL for all components 769b278463cSMatthew G. Knepley . label - The DMLabel defining constrained points 770b278463cSMatthew G. Knepley . numids - The number of DMLabel ids for constrained points 771b278463cSMatthew G. Knepley . ids - An array of ids for constrained points 772b278463cSMatthew G. Knepley . func - A pointwise function giving boundary values 773b278463cSMatthew G. Knepley - ctx - An optional user context for bcFunc 774b278463cSMatthew G. Knepley 775b278463cSMatthew G. Knepley Output Parameter: 776b278463cSMatthew G. Knepley . locX - A local vector to receives the boundary values 777b278463cSMatthew G. Knepley 778b278463cSMatthew G. Knepley Level: developer 779b278463cSMatthew G. Knepley 780*ece3a9fcSMatthew G. Knepley .seealso: DMPlexInsertBoundaryValuesEssential(), DMPlexInsertBoundaryValuesEssentialBdField(), DMAddBoundary() 781b278463cSMatthew G. Knepley @*/ 7821c531cf8SMatthew G. Knepley PetscErrorCode DMPlexInsertBoundaryValuesEssentialField(DM dm, PetscReal time, Vec locU, PetscInt field, PetscInt Nc, const PetscInt comps[], DMLabel label, PetscInt numids, const PetscInt ids[], 783c60e475cSMatthew G. Knepley void (*func)(PetscInt, PetscInt, PetscInt, 784c60e475cSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 785c60e475cSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 78697b6e6e8SMatthew G. Knepley PetscReal, const PetscReal[], PetscInt, const PetscScalar[], 787b278463cSMatthew G. Knepley PetscScalar[]), 788b278463cSMatthew G. Knepley void *ctx, Vec locX) 789c60e475cSMatthew G. Knepley { 790c60e475cSMatthew G. Knepley void (**funcs)(PetscInt, PetscInt, PetscInt, 791c60e475cSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 792c60e475cSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 79397b6e6e8SMatthew G. Knepley PetscReal, const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]); 794c60e475cSMatthew G. Knepley void **ctxs; 795c60e475cSMatthew G. Knepley PetscInt numFields; 796c60e475cSMatthew G. Knepley PetscErrorCode ierr; 797c60e475cSMatthew G. Knepley 798c60e475cSMatthew G. Knepley PetscFunctionBegin; 799c60e475cSMatthew G. Knepley ierr = DMGetNumFields(dm, &numFields);CHKERRQ(ierr); 800c60e475cSMatthew G. Knepley ierr = PetscCalloc2(numFields,&funcs,numFields,&ctxs);CHKERRQ(ierr); 801c60e475cSMatthew G. Knepley funcs[field] = func; 802c60e475cSMatthew G. Knepley ctxs[field] = ctx; 8031c531cf8SMatthew G. Knepley ierr = DMProjectFieldLabelLocal(dm, time, label, numids, ids, Nc, comps, locU, funcs, INSERT_BC_VALUES, locX);CHKERRQ(ierr); 804c60e475cSMatthew G. Knepley ierr = PetscFree2(funcs,ctxs);CHKERRQ(ierr); 805c60e475cSMatthew G. Knepley PetscFunctionReturn(0); 806c60e475cSMatthew G. Knepley } 807c60e475cSMatthew G. Knepley 808b278463cSMatthew G. Knepley /*@C 809*ece3a9fcSMatthew G. Knepley DMPlexInsertBoundaryValuesEssentialBdField - Insert boundary values into a local vector using a function of the coodinates and boundary field data 810*ece3a9fcSMatthew G. Knepley 811*ece3a9fcSMatthew G. Knepley Collective on dm 812*ece3a9fcSMatthew G. Knepley 813*ece3a9fcSMatthew G. Knepley Input Parameters: 814*ece3a9fcSMatthew G. Knepley + dm - The DM, with a PetscDS that matches the problem being constrained 815*ece3a9fcSMatthew G. Knepley . time - The time 816*ece3a9fcSMatthew G. Knepley . locU - A local vector with the input solution values 817*ece3a9fcSMatthew G. Knepley . field - The field to constrain 818*ece3a9fcSMatthew G. Knepley . Nc - The number of constrained field components, or 0 for all components 819*ece3a9fcSMatthew G. Knepley . comps - An array of constrained component numbers, or NULL for all components 820*ece3a9fcSMatthew G. Knepley . label - The DMLabel defining constrained points 821*ece3a9fcSMatthew G. Knepley . numids - The number of DMLabel ids for constrained points 822*ece3a9fcSMatthew G. Knepley . ids - An array of ids for constrained points 823*ece3a9fcSMatthew G. Knepley . func - A pointwise function giving boundary values, the calling sequence is given in DMProjectBdFieldLabelLocal() 824*ece3a9fcSMatthew G. Knepley - ctx - An optional user context for bcFunc 825*ece3a9fcSMatthew G. Knepley 826*ece3a9fcSMatthew G. Knepley Output Parameter: 827*ece3a9fcSMatthew G. Knepley . locX - A local vector to receive the boundary values 828*ece3a9fcSMatthew G. Knepley 829*ece3a9fcSMatthew G. Knepley Level: developer 830*ece3a9fcSMatthew G. Knepley 831*ece3a9fcSMatthew G. Knepley .seealso: DMProjectBdFieldLabelLocal(), DMPlexInsertBoundaryValuesEssential(), DMPlexInsertBoundaryValuesEssentialField(), DMAddBoundary() 832*ece3a9fcSMatthew G. Knepley @*/ 833*ece3a9fcSMatthew G. Knepley PetscErrorCode DMPlexInsertBoundaryValuesEssentialBdField(DM dm, PetscReal time, Vec locU, PetscInt field, PetscInt Nc, const PetscInt comps[], DMLabel label, PetscInt numids, const PetscInt ids[], 834*ece3a9fcSMatthew G. Knepley void (*func)(PetscInt, PetscInt, PetscInt, 835*ece3a9fcSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 836*ece3a9fcSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 837*ece3a9fcSMatthew G. Knepley PetscReal, const PetscReal[], const PetscReal[], PetscInt, const PetscScalar[], 838*ece3a9fcSMatthew G. Knepley PetscScalar[]), 839*ece3a9fcSMatthew G. Knepley void *ctx, Vec locX) 840*ece3a9fcSMatthew G. Knepley { 841*ece3a9fcSMatthew G. Knepley void (**funcs)(PetscInt, PetscInt, PetscInt, 842*ece3a9fcSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 843*ece3a9fcSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 844*ece3a9fcSMatthew G. Knepley PetscReal, const PetscReal[], const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]); 845*ece3a9fcSMatthew G. Knepley void **ctxs; 846*ece3a9fcSMatthew G. Knepley PetscInt numFields; 847*ece3a9fcSMatthew G. Knepley PetscErrorCode ierr; 848*ece3a9fcSMatthew G. Knepley 849*ece3a9fcSMatthew G. Knepley PetscFunctionBegin; 850*ece3a9fcSMatthew G. Knepley ierr = DMGetNumFields(dm, &numFields);CHKERRQ(ierr); 851*ece3a9fcSMatthew G. Knepley ierr = PetscCalloc2(numFields,&funcs,numFields,&ctxs);CHKERRQ(ierr); 852*ece3a9fcSMatthew G. Knepley funcs[field] = func; 853*ece3a9fcSMatthew G. Knepley ctxs[field] = ctx; 854*ece3a9fcSMatthew G. Knepley ierr = DMProjectBdFieldLabelLocal(dm, time, label, numids, ids, Nc, comps, locU, funcs, INSERT_BC_VALUES, locX);CHKERRQ(ierr); 855*ece3a9fcSMatthew G. Knepley ierr = PetscFree2(funcs,ctxs);CHKERRQ(ierr); 856*ece3a9fcSMatthew G. Knepley PetscFunctionReturn(0); 857*ece3a9fcSMatthew G. Knepley } 858*ece3a9fcSMatthew G. Knepley 859*ece3a9fcSMatthew G. Knepley /*@C 860b278463cSMatthew G. Knepley DMPlexInsertBoundaryValuesRiemann - Insert boundary values into a local vector 861b278463cSMatthew G. Knepley 862b278463cSMatthew G. Knepley Input Parameters: 863b278463cSMatthew G. Knepley + dm - The DM, with a PetscDS that matches the problem being constrained 864b278463cSMatthew G. Knepley . time - The time 865b278463cSMatthew G. Knepley . faceGeometry - A vector with the FVM face geometry information 866b278463cSMatthew G. Knepley . cellGeometry - A vector with the FVM cell geometry information 867b278463cSMatthew G. Knepley . Grad - A vector with the FVM cell gradient information 868b278463cSMatthew G. Knepley . field - The field to constrain 8691c531cf8SMatthew G. Knepley . Nc - The number of constrained field components, or 0 for all components 8701c531cf8SMatthew G. Knepley . comps - An array of constrained component numbers, or NULL for all components 871b278463cSMatthew G. Knepley . label - The DMLabel defining constrained points 872b278463cSMatthew G. Knepley . numids - The number of DMLabel ids for constrained points 873b278463cSMatthew G. Knepley . ids - An array of ids for constrained points 874b278463cSMatthew G. Knepley . func - A pointwise function giving boundary values 875b278463cSMatthew G. Knepley - ctx - An optional user context for bcFunc 876b278463cSMatthew G. Knepley 877b278463cSMatthew G. Knepley Output Parameter: 878b278463cSMatthew G. Knepley . locX - A local vector to receives the boundary values 879b278463cSMatthew G. Knepley 880b278463cSMatthew G. Knepley Note: This implementation currently ignores the numcomps/comps argument from DMAddBoundary() 881b278463cSMatthew G. Knepley 882b278463cSMatthew G. Knepley Level: developer 883b278463cSMatthew G. Knepley 884b278463cSMatthew G. Knepley .seealso: DMPlexInsertBoundaryValuesEssential(), DMPlexInsertBoundaryValuesEssentialField(), DMAddBoundary() 885b278463cSMatthew G. Knepley @*/ 8861c531cf8SMatthew G. Knepley PetscErrorCode DMPlexInsertBoundaryValuesRiemann(DM dm, PetscReal time, Vec faceGeometry, Vec cellGeometry, Vec Grad, PetscInt field, PetscInt Nc, const PetscInt comps[], DMLabel label, PetscInt numids, const PetscInt ids[], 887b278463cSMatthew G. Knepley PetscErrorCode (*func)(PetscReal,const PetscReal*,const PetscReal*,const PetscScalar*,PetscScalar*,void*), void *ctx, Vec locX) 888d7ddef95SMatthew G. Knepley { 88961f58d28SMatthew G. Knepley PetscDS prob; 890da97024aSMatthew G. Knepley PetscSF sf; 891d7ddef95SMatthew G. Knepley DM dmFace, dmCell, dmGrad; 89220369375SToby Isaac const PetscScalar *facegeom, *cellgeom = NULL, *grad; 893da97024aSMatthew G. Knepley const PetscInt *leaves; 894d7ddef95SMatthew G. Knepley PetscScalar *x, *fx; 895520b3818SMatthew G. Knepley PetscInt dim, nleaves, loc, fStart, fEnd, pdim, i; 896e735a8a9SMatthew G. Knepley PetscErrorCode ierr, ierru = 0; 897d7ddef95SMatthew G. Knepley 898d7ddef95SMatthew G. Knepley PetscFunctionBegin; 899da97024aSMatthew G. Knepley ierr = DMGetPointSF(dm, &sf);CHKERRQ(ierr); 900da97024aSMatthew G. Knepley ierr = PetscSFGetGraph(sf, NULL, &nleaves, &leaves, NULL);CHKERRQ(ierr); 901da97024aSMatthew G. Knepley nleaves = PetscMax(0, nleaves); 902d7ddef95SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 903d7ddef95SMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd);CHKERRQ(ierr); 90461f58d28SMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 905d7ddef95SMatthew G. Knepley ierr = VecGetDM(faceGeometry, &dmFace);CHKERRQ(ierr); 906d7ddef95SMatthew G. Knepley ierr = VecGetArrayRead(faceGeometry, &facegeom);CHKERRQ(ierr); 90720369375SToby Isaac if (cellGeometry) { 908d7ddef95SMatthew G. Knepley ierr = VecGetDM(cellGeometry, &dmCell);CHKERRQ(ierr); 909d7ddef95SMatthew G. Knepley ierr = VecGetArrayRead(cellGeometry, &cellgeom);CHKERRQ(ierr); 91020369375SToby Isaac } 911d7ddef95SMatthew G. Knepley if (Grad) { 912c0a6632aSMatthew G. Knepley PetscFV fv; 913c0a6632aSMatthew G. Knepley 914c0a6632aSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fv);CHKERRQ(ierr); 915d7ddef95SMatthew G. Knepley ierr = VecGetDM(Grad, &dmGrad);CHKERRQ(ierr); 916d7ddef95SMatthew G. Knepley ierr = VecGetArrayRead(Grad, &grad);CHKERRQ(ierr); 917d7ddef95SMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &pdim);CHKERRQ(ierr); 91869291d52SBarry Smith ierr = DMGetWorkArray(dm, pdim, MPIU_SCALAR, &fx);CHKERRQ(ierr); 919d7ddef95SMatthew G. Knepley } 920d7ddef95SMatthew G. Knepley ierr = VecGetArray(locX, &x);CHKERRQ(ierr); 921d7ddef95SMatthew G. Knepley for (i = 0; i < numids; ++i) { 922d7ddef95SMatthew G. Knepley IS faceIS; 923d7ddef95SMatthew G. Knepley const PetscInt *faces; 924d7ddef95SMatthew G. Knepley PetscInt numFaces, f; 925d7ddef95SMatthew G. Knepley 926d7ddef95SMatthew G. Knepley ierr = DMLabelGetStratumIS(label, ids[i], &faceIS);CHKERRQ(ierr); 927d7ddef95SMatthew G. Knepley if (!faceIS) continue; /* No points with that id on this process */ 928d7ddef95SMatthew G. Knepley ierr = ISGetLocalSize(faceIS, &numFaces);CHKERRQ(ierr); 929d7ddef95SMatthew G. Knepley ierr = ISGetIndices(faceIS, &faces);CHKERRQ(ierr); 930d7ddef95SMatthew G. Knepley for (f = 0; f < numFaces; ++f) { 931d7ddef95SMatthew G. Knepley const PetscInt face = faces[f], *cells; 932640bce14SSatish Balay PetscFVFaceGeom *fg; 933d7ddef95SMatthew G. Knepley 934d7ddef95SMatthew G. Knepley if ((face < fStart) || (face >= fEnd)) continue; /* Refinement adds non-faces to labels */ 935da97024aSMatthew G. Knepley ierr = PetscFindInt(face, nleaves, (PetscInt *) leaves, &loc);CHKERRQ(ierr); 936da97024aSMatthew G. Knepley if (loc >= 0) continue; 937d7ddef95SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmFace, face, facegeom, &fg);CHKERRQ(ierr); 938d7ddef95SMatthew G. Knepley ierr = DMPlexGetSupport(dm, face, &cells);CHKERRQ(ierr); 939d7ddef95SMatthew G. Knepley if (Grad) { 940640bce14SSatish Balay PetscFVCellGeom *cg; 941640bce14SSatish Balay PetscScalar *cx, *cgrad; 942d7ddef95SMatthew G. Knepley PetscScalar *xG; 943d7ddef95SMatthew G. Knepley PetscReal dx[3]; 944d7ddef95SMatthew G. Knepley PetscInt d; 945d7ddef95SMatthew G. Knepley 946d7ddef95SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmCell, cells[0], cellgeom, &cg);CHKERRQ(ierr); 947d7ddef95SMatthew G. Knepley ierr = DMPlexPointLocalRead(dm, cells[0], x, &cx);CHKERRQ(ierr); 948d7ddef95SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmGrad, cells[0], grad, &cgrad);CHKERRQ(ierr); 949520b3818SMatthew G. Knepley ierr = DMPlexPointLocalFieldRef(dm, cells[1], field, x, &xG);CHKERRQ(ierr); 950d7ddef95SMatthew G. Knepley DMPlex_WaxpyD_Internal(dim, -1, cg->centroid, fg->centroid, dx); 951d7ddef95SMatthew G. Knepley for (d = 0; d < pdim; ++d) fx[d] = cx[d] + DMPlex_DotD_Internal(dim, &cgrad[d*dim], dx); 952e735a8a9SMatthew G. Knepley ierru = (*func)(time, fg->centroid, fg->normal, fx, xG, ctx); 953e735a8a9SMatthew G. Knepley if (ierru) { 954e735a8a9SMatthew G. Knepley ierr = ISRestoreIndices(faceIS, &faces);CHKERRQ(ierr); 955e735a8a9SMatthew G. Knepley ierr = ISDestroy(&faceIS);CHKERRQ(ierr); 956e735a8a9SMatthew G. Knepley goto cleanup; 957e735a8a9SMatthew G. Knepley } 958d7ddef95SMatthew G. Knepley } else { 959640bce14SSatish Balay PetscScalar *xI; 960d7ddef95SMatthew G. Knepley PetscScalar *xG; 961d7ddef95SMatthew G. Knepley 962d7ddef95SMatthew G. Knepley ierr = DMPlexPointLocalRead(dm, cells[0], x, &xI);CHKERRQ(ierr); 963520b3818SMatthew G. Knepley ierr = DMPlexPointLocalFieldRef(dm, cells[1], field, x, &xG);CHKERRQ(ierr); 964e735a8a9SMatthew G. Knepley ierru = (*func)(time, fg->centroid, fg->normal, xI, xG, ctx); 965e735a8a9SMatthew G. Knepley if (ierru) { 966e735a8a9SMatthew G. Knepley ierr = ISRestoreIndices(faceIS, &faces);CHKERRQ(ierr); 967e735a8a9SMatthew G. Knepley ierr = ISDestroy(&faceIS);CHKERRQ(ierr); 968e735a8a9SMatthew G. Knepley goto cleanup; 969e735a8a9SMatthew G. Knepley } 970d7ddef95SMatthew G. Knepley } 971d7ddef95SMatthew G. Knepley } 972d7ddef95SMatthew G. Knepley ierr = ISRestoreIndices(faceIS, &faces);CHKERRQ(ierr); 973d7ddef95SMatthew G. Knepley ierr = ISDestroy(&faceIS);CHKERRQ(ierr); 974d7ddef95SMatthew G. Knepley } 975e735a8a9SMatthew G. Knepley cleanup: 976d7ddef95SMatthew G. Knepley ierr = VecRestoreArray(locX, &x);CHKERRQ(ierr); 977d7ddef95SMatthew G. Knepley if (Grad) { 97869291d52SBarry Smith ierr = DMRestoreWorkArray(dm, pdim, MPIU_SCALAR, &fx);CHKERRQ(ierr); 979d7ddef95SMatthew G. Knepley ierr = VecRestoreArrayRead(Grad, &grad);CHKERRQ(ierr); 980d7ddef95SMatthew G. Knepley } 981e735a8a9SMatthew G. Knepley if (cellGeometry) {ierr = VecRestoreArrayRead(cellGeometry, &cellgeom);CHKERRQ(ierr);} 982d7ddef95SMatthew G. Knepley ierr = VecRestoreArrayRead(faceGeometry, &facegeom);CHKERRQ(ierr); 983e735a8a9SMatthew G. Knepley CHKERRQ(ierru); 984d7ddef95SMatthew G. Knepley PetscFunctionReturn(0); 985d7ddef95SMatthew G. Knepley } 986d7ddef95SMatthew G. Knepley 987f1d73a7aSMatthew G. Knepley PetscErrorCode DMPlexInsertBoundaryValues_Plex(DM dm, PetscBool insertEssential, Vec locX, PetscReal time, Vec faceGeomFVM, Vec cellGeomFVM, Vec gradFVM) 988d7ddef95SMatthew G. Knepley { 989e5e52638SMatthew G. Knepley PetscDS prob; 990d7ddef95SMatthew G. Knepley PetscInt numBd, b; 99155f2e967SMatthew G. Knepley PetscErrorCode ierr; 99255f2e967SMatthew G. Knepley 99355f2e967SMatthew G. Knepley PetscFunctionBegin; 994e5e52638SMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 995e5e52638SMatthew G. Knepley ierr = PetscDSGetNumBoundary(prob, &numBd);CHKERRQ(ierr); 99655f2e967SMatthew G. Knepley for (b = 0; b < numBd; ++b) { 997f971fd6bSMatthew G. Knepley DMBoundaryConditionType type; 998648eda8cSMatthew G. Knepley const char *name, *labelname; 999d7ddef95SMatthew G. Knepley DMLabel label; 10001c531cf8SMatthew G. Knepley PetscInt field, Nc; 10011c531cf8SMatthew G. Knepley const PetscInt *comps; 1002d7ddef95SMatthew G. Knepley PetscObject obj; 1003d7ddef95SMatthew G. Knepley PetscClassId id; 1004a30ec4eaSSatish Balay void (*func)(void); 1005d7ddef95SMatthew G. Knepley PetscInt numids; 1006d7ddef95SMatthew G. Knepley const PetscInt *ids; 100755f2e967SMatthew G. Knepley void *ctx; 100855f2e967SMatthew G. Knepley 1009648eda8cSMatthew G. Knepley ierr = DMGetBoundary(dm, b, &type, &name, &labelname, &field, &Nc, &comps, &func, &numids, &ids, &ctx);CHKERRQ(ierr); 1010f971fd6bSMatthew G. Knepley if (insertEssential != (type & DM_BC_ESSENTIAL)) continue; 1011c58f1c22SToby Isaac ierr = DMGetLabel(dm, labelname, &label);CHKERRQ(ierr); 1012648eda8cSMatthew G. Knepley if (!label) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONGSTATE, "Label %s for boundary condition %s does not exist in the DM", labelname, name); 101344a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, &obj);CHKERRQ(ierr); 1014d7ddef95SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 1015d7ddef95SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 1016c60e475cSMatthew G. Knepley switch (type) { 1017c60e475cSMatthew G. Knepley /* for FEM, there is no insertion to be done for non-essential boundary conditions */ 1018c60e475cSMatthew G. Knepley case DM_BC_ESSENTIAL: 1019092e5057SToby Isaac ierr = DMPlexLabelAddCells(dm,label);CHKERRQ(ierr); 10201c531cf8SMatthew G. Knepley ierr = DMPlexInsertBoundaryValuesEssential(dm, time, field, Nc, comps, label, numids, ids, (PetscErrorCode (*)(PetscInt, PetscReal, const PetscReal[], PetscInt, PetscScalar *, void *)) func, ctx, locX);CHKERRQ(ierr); 1021092e5057SToby Isaac ierr = DMPlexLabelClearCells(dm,label);CHKERRQ(ierr); 1022c60e475cSMatthew G. Knepley break; 1023c60e475cSMatthew G. Knepley case DM_BC_ESSENTIAL_FIELD: 1024c60e475cSMatthew G. Knepley ierr = DMPlexLabelAddCells(dm,label);CHKERRQ(ierr); 10251c531cf8SMatthew G. Knepley ierr = DMPlexInsertBoundaryValuesEssentialField(dm, time, locX, field, Nc, comps, label, numids, ids, 1026b278463cSMatthew G. Knepley (void (*)(PetscInt, PetscInt, PetscInt, const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 1027c60e475cSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 102897b6e6e8SMatthew G. Knepley PetscReal, const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[])) func, ctx, locX);CHKERRQ(ierr); 1029c60e475cSMatthew G. Knepley ierr = DMPlexLabelClearCells(dm,label);CHKERRQ(ierr); 1030c60e475cSMatthew G. Knepley break; 1031c60e475cSMatthew G. Knepley default: break; 1032c60e475cSMatthew G. Knepley } 1033d7ddef95SMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 103443ea7facSMatthew G. Knepley if (!faceGeomFVM) continue; 10351c531cf8SMatthew G. Knepley ierr = DMPlexInsertBoundaryValuesRiemann(dm, time, faceGeomFVM, cellGeomFVM, gradFVM, field, Nc, comps, label, numids, ids, 1036b278463cSMatthew G. Knepley (PetscErrorCode (*)(PetscReal,const PetscReal*,const PetscReal*,const PetscScalar*,PetscScalar*,void*)) func, ctx, locX);CHKERRQ(ierr); 1037ff1e0c32SBarry Smith } else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 103855f2e967SMatthew G. Knepley } 103955f2e967SMatthew G. Knepley PetscFunctionReturn(0); 104055f2e967SMatthew G. Knepley } 104155f2e967SMatthew G. Knepley 1042f1d73a7aSMatthew G. Knepley /*@ 1043f1d73a7aSMatthew G. Knepley DMPlexInsertBoundaryValues - Puts coefficients which represent boundary values into the local solution vector 1044f1d73a7aSMatthew G. Knepley 1045f1d73a7aSMatthew G. Knepley Input Parameters: 1046f1d73a7aSMatthew G. Knepley + dm - The DM 1047f1d73a7aSMatthew G. Knepley . insertEssential - Should I insert essential (e.g. Dirichlet) or inessential (e.g. Neumann) boundary conditions 1048f1d73a7aSMatthew G. Knepley . time - The time 1049f1d73a7aSMatthew G. Knepley . faceGeomFVM - Face geometry data for FV discretizations 1050f1d73a7aSMatthew G. Knepley . cellGeomFVM - Cell geometry data for FV discretizations 1051f1d73a7aSMatthew G. Knepley - gradFVM - Gradient reconstruction data for FV discretizations 1052f1d73a7aSMatthew G. Knepley 1053f1d73a7aSMatthew G. Knepley Output Parameters: 1054f1d73a7aSMatthew G. Knepley . locX - Solution updated with boundary values 1055f1d73a7aSMatthew G. Knepley 1056f1d73a7aSMatthew G. Knepley Level: developer 1057f1d73a7aSMatthew G. Knepley 1058f1d73a7aSMatthew G. Knepley .seealso: DMProjectFunctionLabelLocal() 1059f1d73a7aSMatthew G. Knepley @*/ 1060f1d73a7aSMatthew G. Knepley PetscErrorCode DMPlexInsertBoundaryValues(DM dm, PetscBool insertEssential, Vec locX, PetscReal time, Vec faceGeomFVM, Vec cellGeomFVM, Vec gradFVM) 1061f1d73a7aSMatthew G. Knepley { 1062f1d73a7aSMatthew G. Knepley PetscErrorCode ierr; 1063f1d73a7aSMatthew G. Knepley 1064f1d73a7aSMatthew G. Knepley PetscFunctionBegin; 1065f1d73a7aSMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 1066f1d73a7aSMatthew G. Knepley PetscValidHeaderSpecific(locX, VEC_CLASSID, 2); 1067f1d73a7aSMatthew G. Knepley if (faceGeomFVM) {PetscValidHeaderSpecific(faceGeomFVM, VEC_CLASSID, 4);} 1068f1d73a7aSMatthew G. Knepley if (cellGeomFVM) {PetscValidHeaderSpecific(cellGeomFVM, VEC_CLASSID, 5);} 1069f1d73a7aSMatthew G. Knepley if (gradFVM) {PetscValidHeaderSpecific(gradFVM, VEC_CLASSID, 6);} 1070f1d73a7aSMatthew G. Knepley ierr = PetscTryMethod(dm,"DMPlexInsertBoundaryValues_C",(DM,PetscBool,Vec,PetscReal,Vec,Vec,Vec),(dm,insertEssential,locX,time,faceGeomFVM,cellGeomFVM,gradFVM));CHKERRQ(ierr); 1071f1d73a7aSMatthew G. Knepley PetscFunctionReturn(0); 1072f1d73a7aSMatthew G. Knepley } 1073f1d73a7aSMatthew G. Knepley 10740709b2feSToby Isaac PetscErrorCode DMComputeL2Diff_Plex(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, PetscReal *diff) 1075cb1e1211SMatthew G Knepley { 1076574a98acSMatthew G. Knepley Vec localX; 1077574a98acSMatthew G. Knepley PetscErrorCode ierr; 1078574a98acSMatthew G. Knepley 1079574a98acSMatthew G. Knepley PetscFunctionBegin; 1080574a98acSMatthew G. Knepley ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr); 10815d42b983SMatthew G. Knepley ierr = DMPlexInsertBoundaryValues(dm, PETSC_TRUE, localX, time, NULL, NULL, NULL);CHKERRQ(ierr); 1082574a98acSMatthew G. Knepley ierr = DMGlobalToLocalBegin(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr); 1083574a98acSMatthew G. Knepley ierr = DMGlobalToLocalEnd(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr); 1084574a98acSMatthew G. Knepley ierr = DMPlexComputeL2DiffLocal(dm, time, funcs, ctxs, localX, diff);CHKERRQ(ierr); 1085574a98acSMatthew G. Knepley ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr); 1086574a98acSMatthew G. Knepley PetscFunctionReturn(0); 1087574a98acSMatthew G. Knepley } 1088574a98acSMatthew G. Knepley 1089574a98acSMatthew G. Knepley /*@C 1090ca3d3a14SMatthew G. Knepley DMComputeL2DiffLocal - This function computes the L_2 difference between a function u and an FEM interpolant solution u_h. 1091574a98acSMatthew G. Knepley 1092d083f849SBarry Smith Collective on dm 1093c0f8e1fdSMatthew G. Knepley 1094574a98acSMatthew G. Knepley Input Parameters: 1095574a98acSMatthew G. Knepley + dm - The DM 1096574a98acSMatthew G. Knepley . time - The time 1097574a98acSMatthew G. Knepley . funcs - The functions to evaluate for each field component 1098574a98acSMatthew G. Knepley . ctxs - Optional array of contexts to pass to each function, or NULL. 1099574a98acSMatthew G. Knepley - localX - The coefficient vector u_h, a local vector 1100574a98acSMatthew G. Knepley 1101574a98acSMatthew G. Knepley Output Parameter: 1102574a98acSMatthew G. Knepley . diff - The diff ||u - u_h||_2 1103574a98acSMatthew G. Knepley 1104574a98acSMatthew G. Knepley Level: developer 1105574a98acSMatthew G. Knepley 1106574a98acSMatthew G. Knepley .seealso: DMProjectFunction(), DMComputeL2FieldDiff(), DMComputeL2GradientDiff() 1107574a98acSMatthew G. Knepley @*/ 1108574a98acSMatthew G. Knepley PetscErrorCode DMPlexComputeL2DiffLocal(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, Vec localX, PetscReal *diff) 1109574a98acSMatthew G. Knepley { 11100f09c10fSMatthew G. Knepley const PetscInt debug = ((DM_Plex*)dm->data)->printL2; 1111ca3d3a14SMatthew G. Knepley DM tdm; 1112ca3d3a14SMatthew G. Knepley Vec tv; 1113cb1e1211SMatthew G Knepley PetscSection section; 1114c5bbbd5bSMatthew G. Knepley PetscQuadrature quad; 11154bee2e38SMatthew G. Knepley PetscFEGeom fegeom; 111615496722SMatthew G. Knepley PetscScalar *funcVal, *interpolant; 11174bee2e38SMatthew G. Knepley PetscReal *coords, *gcoords; 1118cb1e1211SMatthew G Knepley PetscReal localDiff = 0.0; 11197318780aSToby Isaac const PetscReal *quadWeights; 1120412e9a14SMatthew G. Knepley PetscInt dim, coordDim, numFields, numComponents = 0, qNc, Nq, cellHeight, cStart, cEnd, c, field, fieldOffset; 1121ca3d3a14SMatthew G. Knepley PetscBool transform; 1122cb1e1211SMatthew G Knepley PetscErrorCode ierr; 1123cb1e1211SMatthew G Knepley 1124cb1e1211SMatthew G Knepley PetscFunctionBegin; 1125c73cfb54SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 11267318780aSToby Isaac ierr = DMGetCoordinateDim(dm, &coordDim);CHKERRQ(ierr); 11272a4e142eSMatthew G. Knepley fegeom.dimEmbed = coordDim; 112892fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 1129cb1e1211SMatthew G Knepley ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr); 1130ca3d3a14SMatthew G. Knepley ierr = DMGetBasisTransformDM_Internal(dm, &tdm);CHKERRQ(ierr); 1131ca3d3a14SMatthew G. Knepley ierr = DMGetBasisTransformVec_Internal(dm, &tv);CHKERRQ(ierr); 1132ca3d3a14SMatthew G. Knepley ierr = DMHasBasisTransform(dm, &transform);CHKERRQ(ierr); 1133cb1e1211SMatthew G Knepley for (field = 0; field < numFields; ++field) { 113415496722SMatthew G. Knepley PetscObject obj; 113515496722SMatthew G. Knepley PetscClassId id; 1136c5bbbd5bSMatthew G. Knepley PetscInt Nc; 1137c5bbbd5bSMatthew G. Knepley 113844a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, &obj);CHKERRQ(ierr); 113915496722SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 114015496722SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 114115496722SMatthew G. Knepley PetscFE fe = (PetscFE) obj; 114215496722SMatthew G. Knepley 11430f2d7e86SMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &quad);CHKERRQ(ierr); 11440f2d7e86SMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 114515496722SMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 114615496722SMatthew G. Knepley PetscFV fv = (PetscFV) obj; 114715496722SMatthew G. Knepley 114815496722SMatthew G. Knepley ierr = PetscFVGetQuadrature(fv, &quad);CHKERRQ(ierr); 114915496722SMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &Nc);CHKERRQ(ierr); 1150ff1e0c32SBarry Smith } else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 1151c5bbbd5bSMatthew G. Knepley numComponents += Nc; 1152cb1e1211SMatthew G Knepley } 11539c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, &qNc, &Nq, NULL, &quadWeights);CHKERRQ(ierr); 1154beaa55a6SMatthew G. Knepley if ((qNc != 1) && (qNc != numComponents)) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_SIZ, "Quadrature components %D != %D field components", qNc, numComponents); 11552a4e142eSMatthew G. Knepley ierr = PetscMalloc6(numComponents,&funcVal,numComponents,&interpolant,coordDim*Nq,&coords,Nq,&fegeom.detJ,coordDim*coordDim*Nq,&fegeom.J,coordDim*coordDim*Nq,&fegeom.invJ);CHKERRQ(ierr); 1156aed3cbd0SMatthew G. Knepley ierr = DMPlexGetVTKCellHeight(dm, &cellHeight);CHKERRQ(ierr); 1157412e9a14SMatthew G. Knepley ierr = DMPlexGetSimplexOrBoxCells(dm, cellHeight, &cStart, &cEnd);CHKERRQ(ierr); 1158cb1e1211SMatthew G Knepley for (c = cStart; c < cEnd; ++c) { 1159a1e44745SMatthew G. Knepley PetscScalar *x = NULL; 1160cb1e1211SMatthew G Knepley PetscReal elemDiff = 0.0; 11619c3cf19fSMatthew G. Knepley PetscInt qc = 0; 1162cb1e1211SMatthew G Knepley 11632a4e142eSMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dm, c, quad, coords, fegeom.J, fegeom.invJ, fegeom.detJ);CHKERRQ(ierr); 1164cb1e1211SMatthew G Knepley ierr = DMPlexVecGetClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr); 1165cb1e1211SMatthew G Knepley 116615496722SMatthew G. Knepley for (field = 0, fieldOffset = 0; field < numFields; ++field) { 116715496722SMatthew G. Knepley PetscObject obj; 116815496722SMatthew G. Knepley PetscClassId id; 1169c110b1eeSGeoffrey Irving void * const ctx = ctxs ? ctxs[field] : NULL; 117015496722SMatthew G. Knepley PetscInt Nb, Nc, q, fc; 1171cb1e1211SMatthew G Knepley 117244a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, &obj);CHKERRQ(ierr); 117315496722SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 117415496722SMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscFEGetNumComponents((PetscFE) obj, &Nc);CHKERRQ(ierr);ierr = PetscFEGetDimension((PetscFE) obj, &Nb);CHKERRQ(ierr);} 117515496722SMatthew G. Knepley else if (id == PETSCFV_CLASSID) {ierr = PetscFVGetNumComponents((PetscFV) obj, &Nc);CHKERRQ(ierr);Nb = 1;} 1176ff1e0c32SBarry Smith else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 1177cb1e1211SMatthew G Knepley if (debug) { 1178cb1e1211SMatthew G Knepley char title[1024]; 1179ff1e0c32SBarry Smith ierr = PetscSNPrintf(title, 1023, "Solution for Field %D", field);CHKERRQ(ierr); 11804c848028SMatthew G. Knepley ierr = DMPrintCellVector(c, title, Nb, &x[fieldOffset]);CHKERRQ(ierr); 1181cb1e1211SMatthew G Knepley } 11827318780aSToby Isaac for (q = 0; q < Nq; ++q) { 11832a4e142eSMatthew G. Knepley PetscFEGeom qgeom; 11842a4e142eSMatthew G. Knepley 11852a4e142eSMatthew G. Knepley qgeom.dimEmbed = fegeom.dimEmbed; 11862a4e142eSMatthew G. Knepley qgeom.J = &fegeom.J[q*coordDim*coordDim]; 11872a4e142eSMatthew G. Knepley qgeom.invJ = &fegeom.invJ[q*coordDim*coordDim]; 11882a4e142eSMatthew G. Knepley qgeom.detJ = &fegeom.detJ[q]; 1189ff1e0c32SBarry Smith if (fegeom.detJ[q] <= 0.0) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %D, point %D", (double)fegeom.detJ[q], c, q); 1190d3a7d86cSMatthew G. Knepley if (transform) { 1191d3a7d86cSMatthew G. Knepley gcoords = &coords[coordDim*Nq]; 1192d3a7d86cSMatthew G. Knepley ierr = DMPlexBasisTransformApplyReal_Internal(dm, &coords[coordDim*q], PETSC_TRUE, coordDim, &coords[coordDim*q], gcoords, dm->transformCtx);CHKERRQ(ierr); 1193d3a7d86cSMatthew G. Knepley } else { 1194d3a7d86cSMatthew G. Knepley gcoords = &coords[coordDim*q]; 1195d3a7d86cSMatthew G. Knepley } 1196ca3d3a14SMatthew G. Knepley ierr = (*funcs[field])(coordDim, time, gcoords, Nc, funcVal, ctx); 1197e735a8a9SMatthew G. Knepley if (ierr) { 1198e735a8a9SMatthew G. Knepley PetscErrorCode ierr2; 1199e735a8a9SMatthew G. Knepley ierr2 = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr2); 1200e735a8a9SMatthew G. Knepley ierr2 = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr2); 12012a4e142eSMatthew G. Knepley ierr2 = PetscFree6(funcVal,interpolant,coords,fegeom.detJ,fegeom.J,fegeom.invJ);CHKERRQ(ierr2); 1202e735a8a9SMatthew G. Knepley CHKERRQ(ierr); 1203e735a8a9SMatthew G. Knepley } 1204ca3d3a14SMatthew G. Knepley if (transform) {ierr = DMPlexBasisTransformApply_Internal(dm, &coords[coordDim*q], PETSC_FALSE, Nc, funcVal, funcVal, dm->transformCtx);CHKERRQ(ierr);} 12052a4e142eSMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscFEInterpolate_Static((PetscFE) obj, &x[fieldOffset], &qgeom, q, interpolant);CHKERRQ(ierr);} 120615496722SMatthew G. Knepley else if (id == PETSCFV_CLASSID) {ierr = PetscFVInterpolate_Static((PetscFV) obj, &x[fieldOffset], q, interpolant);CHKERRQ(ierr);} 1207ff1e0c32SBarry Smith else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 120815496722SMatthew G. Knepley for (fc = 0; fc < Nc; ++fc) { 1209beaa55a6SMatthew G. Knepley const PetscReal wt = quadWeights[q*qNc+(qNc == 1 ? 0 : qc+fc)]; 1210ff1e0c32SBarry Smith if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, " elem %D field %D,%D diff %g\n", c, field, fc, (double)(PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*wt*fegeom.detJ[q]));CHKERRQ(ierr);} 12114bee2e38SMatthew G. Knepley elemDiff += PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*wt*fegeom.detJ[q]; 1212cb1e1211SMatthew G Knepley } 1213cb1e1211SMatthew G Knepley } 12149c3cf19fSMatthew G. Knepley fieldOffset += Nb; 1215beaa55a6SMatthew G. Knepley qc += Nc; 1216cb1e1211SMatthew G Knepley } 1217cb1e1211SMatthew G Knepley ierr = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr); 1218ff1e0c32SBarry Smith if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, " elem %D diff %g\n", c, (double)elemDiff);CHKERRQ(ierr);} 1219cb1e1211SMatthew G Knepley localDiff += elemDiff; 1220cb1e1211SMatthew G Knepley } 12212a4e142eSMatthew G. Knepley ierr = PetscFree6(funcVal,interpolant,coords,fegeom.detJ,fegeom.J,fegeom.invJ);CHKERRQ(ierr); 1222b2566f29SBarry Smith ierr = MPIU_Allreduce(&localDiff, diff, 1, MPIU_REAL, MPIU_SUM, PetscObjectComm((PetscObject)dm));CHKERRQ(ierr); 1223cb1e1211SMatthew G Knepley *diff = PetscSqrtReal(*diff); 1224cb1e1211SMatthew G Knepley PetscFunctionReturn(0); 1225cb1e1211SMatthew G Knepley } 1226cb1e1211SMatthew G Knepley 1227b698f381SToby Isaac PetscErrorCode DMComputeL2GradientDiff_Plex(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, const PetscReal n[], PetscReal *diff) 1228cb1e1211SMatthew G Knepley { 12290f09c10fSMatthew G. Knepley const PetscInt debug = ((DM_Plex*)dm->data)->printL2; 1230ca3d3a14SMatthew G. Knepley DM tdm; 1231cb1e1211SMatthew G Knepley PetscSection section; 123240e14135SMatthew G. Knepley PetscQuadrature quad; 1233ca3d3a14SMatthew G. Knepley Vec localX, tv; 12349c3cf19fSMatthew G. Knepley PetscScalar *funcVal, *interpolant; 12352a4e142eSMatthew G. Knepley const PetscReal *quadWeights; 12364bee2e38SMatthew G. Knepley PetscFEGeom fegeom; 12374bee2e38SMatthew G. Knepley PetscReal *coords, *gcoords; 123840e14135SMatthew G. Knepley PetscReal localDiff = 0.0; 1239485ad865SMatthew G. Knepley PetscInt dim, coordDim, qNc = 0, Nq = 0, numFields, numComponents = 0, cStart, cEnd, c, field, fieldOffset; 1240ca3d3a14SMatthew G. Knepley PetscBool transform; 1241cb1e1211SMatthew G Knepley PetscErrorCode ierr; 1242cb1e1211SMatthew G Knepley 1243cb1e1211SMatthew G Knepley PetscFunctionBegin; 1244c73cfb54SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 12457318780aSToby Isaac ierr = DMGetCoordinateDim(dm, &coordDim);CHKERRQ(ierr); 12464bee2e38SMatthew G. Knepley fegeom.dimEmbed = coordDim; 124792fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 124840e14135SMatthew G. Knepley ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr); 124940e14135SMatthew G. Knepley ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr); 125040e14135SMatthew G. Knepley ierr = DMGlobalToLocalBegin(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr); 125140e14135SMatthew G. Knepley ierr = DMGlobalToLocalEnd(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr); 1252ca3d3a14SMatthew G. Knepley ierr = DMGetBasisTransformDM_Internal(dm, &tdm);CHKERRQ(ierr); 1253ca3d3a14SMatthew G. Knepley ierr = DMGetBasisTransformVec_Internal(dm, &tv);CHKERRQ(ierr); 1254ca3d3a14SMatthew G. Knepley ierr = DMHasBasisTransform(dm, &transform);CHKERRQ(ierr); 1255652b88e8SMatthew G. Knepley for (field = 0; field < numFields; ++field) { 12560f2d7e86SMatthew G. Knepley PetscFE fe; 125740e14135SMatthew G. Knepley PetscInt Nc; 1258652b88e8SMatthew G. Knepley 125944a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, (PetscObject *) &fe);CHKERRQ(ierr); 12600f2d7e86SMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &quad);CHKERRQ(ierr); 12610f2d7e86SMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 126240e14135SMatthew G. Knepley numComponents += Nc; 1263652b88e8SMatthew G. Knepley } 12642a4e142eSMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, &qNc, &Nq, NULL, &quadWeights);CHKERRQ(ierr); 1265beaa55a6SMatthew G. Knepley if ((qNc != 1) && (qNc != numComponents)) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_SIZ, "Quadrature components %D != %D field components", qNc, numComponents); 12664d6f44ffSToby Isaac /* ierr = DMProjectFunctionLocal(dm, fe, funcs, INSERT_BC_VALUES, localX);CHKERRQ(ierr); */ 12674bee2e38SMatthew G. Knepley ierr = PetscMalloc6(numComponents,&funcVal,coordDim*Nq,&coords,coordDim*coordDim*Nq,&fegeom.J,coordDim*coordDim*Nq,&fegeom.invJ,numComponents*coordDim,&interpolant,Nq,&fegeom.detJ);CHKERRQ(ierr); 1268412e9a14SMatthew G. Knepley ierr = DMPlexGetSimplexOrBoxCells(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 126940e14135SMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 127040e14135SMatthew G. Knepley PetscScalar *x = NULL; 127140e14135SMatthew G. Knepley PetscReal elemDiff = 0.0; 12729c3cf19fSMatthew G. Knepley PetscInt qc = 0; 1273652b88e8SMatthew G. Knepley 12744bee2e38SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dm, c, quad, coords, fegeom.J, fegeom.invJ, fegeom.detJ);CHKERRQ(ierr); 127540e14135SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr); 127640e14135SMatthew G. Knepley 12779c3cf19fSMatthew G. Knepley for (field = 0, fieldOffset = 0; field < numFields; ++field) { 12780f2d7e86SMatthew G. Knepley PetscFE fe; 127951259fa3SMatthew G. Knepley void * const ctx = ctxs ? ctxs[field] : NULL; 12809c3cf19fSMatthew G. Knepley PetscInt Nb, Nc, q, fc; 128140e14135SMatthew G. Knepley 128244a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, (PetscObject *) &fe);CHKERRQ(ierr); 12830f2d7e86SMatthew G. Knepley ierr = PetscFEGetDimension(fe, &Nb);CHKERRQ(ierr); 12849c3cf19fSMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 128540e14135SMatthew G. Knepley if (debug) { 128640e14135SMatthew G. Knepley char title[1024]; 1287ff1e0c32SBarry Smith ierr = PetscSNPrintf(title, 1023, "Solution for Field %D", field);CHKERRQ(ierr); 12889c3cf19fSMatthew G. Knepley ierr = DMPrintCellVector(c, title, Nb, &x[fieldOffset]);CHKERRQ(ierr); 1289652b88e8SMatthew G. Knepley } 12909c3cf19fSMatthew G. Knepley for (q = 0; q < Nq; ++q) { 12912a4e142eSMatthew G. Knepley PetscFEGeom qgeom; 12922a4e142eSMatthew G. Knepley 12932a4e142eSMatthew G. Knepley qgeom.dimEmbed = fegeom.dimEmbed; 12942a4e142eSMatthew G. Knepley qgeom.J = &fegeom.J[q*coordDim*coordDim]; 12952a4e142eSMatthew G. Knepley qgeom.invJ = &fegeom.invJ[q*coordDim*coordDim]; 12962a4e142eSMatthew G. Knepley qgeom.detJ = &fegeom.detJ[q]; 1297ff1e0c32SBarry Smith if (fegeom.detJ[q] <= 0.0) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %D, quadrature points %D", (double)fegeom.detJ[q], c, q); 1298d3a7d86cSMatthew G. Knepley if (transform) { 1299d3a7d86cSMatthew G. Knepley gcoords = &coords[coordDim*Nq]; 1300d3a7d86cSMatthew G. Knepley ierr = DMPlexBasisTransformApplyReal_Internal(dm, &coords[coordDim*q], PETSC_TRUE, coordDim, &coords[coordDim*q], gcoords, dm->transformCtx);CHKERRQ(ierr); 1301d3a7d86cSMatthew G. Knepley } else { 1302d3a7d86cSMatthew G. Knepley gcoords = &coords[coordDim*q]; 1303d3a7d86cSMatthew G. Knepley } 13044bee2e38SMatthew G. Knepley ierr = (*funcs[field])(coordDim, time, gcoords, n, Nc, funcVal, ctx); 1305e735a8a9SMatthew G. Knepley if (ierr) { 1306e735a8a9SMatthew G. Knepley PetscErrorCode ierr2; 1307e735a8a9SMatthew G. Knepley ierr2 = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr2); 1308e735a8a9SMatthew G. Knepley ierr2 = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr2); 13094bee2e38SMatthew G. Knepley ierr2 = PetscFree6(funcVal,coords,fegeom.J,fegeom.invJ,interpolant,fegeom.detJ);CHKERRQ(ierr2); 1310e735a8a9SMatthew G. Knepley CHKERRQ(ierr); 1311e735a8a9SMatthew G. Knepley } 1312ca3d3a14SMatthew G. Knepley if (transform) {ierr = DMPlexBasisTransformApply_Internal(dm, &coords[coordDim*q], PETSC_FALSE, Nc, funcVal, funcVal, dm->transformCtx);CHKERRQ(ierr);} 13132a4e142eSMatthew G. Knepley ierr = PetscFEInterpolateGradient_Static(fe, &x[fieldOffset], &qgeom, q, interpolant);CHKERRQ(ierr); 13144bee2e38SMatthew G. Knepley /* Overwrite with the dot product if the normal is given */ 13154bee2e38SMatthew G. Knepley if (n) { 13164bee2e38SMatthew G. Knepley for (fc = 0; fc < Nc; ++fc) { 13174bee2e38SMatthew G. Knepley PetscScalar sum = 0.0; 13184bee2e38SMatthew G. Knepley PetscInt d; 13194bee2e38SMatthew G. Knepley for (d = 0; d < dim; ++d) sum += interpolant[fc*dim+d]*n[d]; 13204bee2e38SMatthew G. Knepley interpolant[fc] = sum; 13214bee2e38SMatthew G. Knepley } 13224bee2e38SMatthew G. Knepley } 13239c3cf19fSMatthew G. Knepley for (fc = 0; fc < Nc; ++fc) { 1324beaa55a6SMatthew G. Knepley const PetscReal wt = quadWeights[q*qNc+(qNc == 1 ? 0 : qc+fc)]; 1325ff1e0c32SBarry Smith if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, " elem %D fieldDer %D,%D diff %g\n", c, field, fc, (double)(PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*wt*fegeom.detJ[q]));CHKERRQ(ierr);} 13264bee2e38SMatthew G. Knepley elemDiff += PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*wt*fegeom.detJ[q]; 132740e14135SMatthew G. Knepley } 132840e14135SMatthew G. Knepley } 13299c3cf19fSMatthew G. Knepley fieldOffset += Nb; 13309c3cf19fSMatthew G. Knepley qc += Nc; 133140e14135SMatthew G. Knepley } 133240e14135SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr); 1333ff1e0c32SBarry Smith if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, " elem %D diff %g\n", c, (double)elemDiff);CHKERRQ(ierr);} 133440e14135SMatthew G. Knepley localDiff += elemDiff; 133540e14135SMatthew G. Knepley } 13364bee2e38SMatthew G. Knepley ierr = PetscFree6(funcVal,coords,fegeom.J,fegeom.invJ,interpolant,fegeom.detJ);CHKERRQ(ierr); 133740e14135SMatthew G. Knepley ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr); 1338b2566f29SBarry Smith ierr = MPIU_Allreduce(&localDiff, diff, 1, MPIU_REAL, MPIU_SUM, PetscObjectComm((PetscObject)dm));CHKERRQ(ierr); 133940e14135SMatthew G. Knepley *diff = PetscSqrtReal(*diff); 1340cb1e1211SMatthew G Knepley PetscFunctionReturn(0); 1341cb1e1211SMatthew G Knepley } 1342cb1e1211SMatthew G Knepley 1343c6eecec3SToby Isaac PetscErrorCode DMComputeL2FieldDiff_Plex(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, PetscReal *diff) 134473d901b8SMatthew G. Knepley { 13450f09c10fSMatthew G. Knepley const PetscInt debug = ((DM_Plex*)dm->data)->printL2; 1346ca3d3a14SMatthew G. Knepley DM tdm; 134773d901b8SMatthew G. Knepley PetscSection section; 134873d901b8SMatthew G. Knepley PetscQuadrature quad; 1349ca3d3a14SMatthew G. Knepley Vec localX, tv; 13504bee2e38SMatthew G. Knepley PetscFEGeom fegeom; 135115496722SMatthew G. Knepley PetscScalar *funcVal, *interpolant; 13524bee2e38SMatthew G. Knepley PetscReal *coords, *gcoords; 135373d901b8SMatthew G. Knepley PetscReal *localDiff; 135415496722SMatthew G. Knepley const PetscReal *quadPoints, *quadWeights; 1355485ad865SMatthew G. Knepley PetscInt dim, coordDim, numFields, numComponents = 0, qNc, Nq, cStart, cEnd, c, field, fieldOffset; 1356ca3d3a14SMatthew G. Knepley PetscBool transform; 135773d901b8SMatthew G. Knepley PetscErrorCode ierr; 135873d901b8SMatthew G. Knepley 135973d901b8SMatthew G. Knepley PetscFunctionBegin; 1360c73cfb54SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 13617318780aSToby Isaac ierr = DMGetCoordinateDim(dm, &coordDim);CHKERRQ(ierr); 136292fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 136373d901b8SMatthew G. Knepley ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr); 136473d901b8SMatthew G. Knepley ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr); 1365ca3d3a14SMatthew G. Knepley ierr = VecSet(localX, 0.0);CHKERRQ(ierr); 136673d901b8SMatthew G. Knepley ierr = DMGlobalToLocalBegin(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr); 136773d901b8SMatthew G. Knepley ierr = DMGlobalToLocalEnd(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr); 1368ca3d3a14SMatthew G. Knepley ierr = DMProjectFunctionLocal(dm, time, funcs, ctxs, INSERT_BC_VALUES, localX);CHKERRQ(ierr); 1369ca3d3a14SMatthew G. Knepley ierr = DMGetBasisTransformDM_Internal(dm, &tdm);CHKERRQ(ierr); 1370ca3d3a14SMatthew G. Knepley ierr = DMGetBasisTransformVec_Internal(dm, &tv);CHKERRQ(ierr); 1371ca3d3a14SMatthew G. Knepley ierr = DMHasBasisTransform(dm, &transform);CHKERRQ(ierr); 137273d901b8SMatthew G. Knepley for (field = 0; field < numFields; ++field) { 137315496722SMatthew G. Knepley PetscObject obj; 137415496722SMatthew G. Knepley PetscClassId id; 137573d901b8SMatthew G. Knepley PetscInt Nc; 137673d901b8SMatthew G. Knepley 137744a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, &obj);CHKERRQ(ierr); 137815496722SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 137915496722SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 138015496722SMatthew G. Knepley PetscFE fe = (PetscFE) obj; 138115496722SMatthew G. Knepley 13820f2d7e86SMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &quad);CHKERRQ(ierr); 13830f2d7e86SMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 138415496722SMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 138515496722SMatthew G. Knepley PetscFV fv = (PetscFV) obj; 138615496722SMatthew G. Knepley 138715496722SMatthew G. Knepley ierr = PetscFVGetQuadrature(fv, &quad);CHKERRQ(ierr); 138815496722SMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &Nc);CHKERRQ(ierr); 1389ff1e0c32SBarry Smith } else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 139073d901b8SMatthew G. Knepley numComponents += Nc; 139173d901b8SMatthew G. Knepley } 13929c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, &qNc, &Nq, &quadPoints, &quadWeights);CHKERRQ(ierr); 1393beaa55a6SMatthew G. Knepley if ((qNc != 1) && (qNc != numComponents)) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_SIZ, "Quadrature components %D != %D field components", qNc, numComponents); 13942a4e142eSMatthew G. Knepley ierr = PetscCalloc7(numFields,&localDiff,numComponents,&funcVal,numComponents,&interpolant,coordDim*(Nq+1),&coords,Nq,&fegeom.detJ,coordDim*coordDim*Nq,&fegeom.J,coordDim*coordDim*Nq,&fegeom.invJ);CHKERRQ(ierr); 1395412e9a14SMatthew G. Knepley ierr = DMPlexGetSimplexOrBoxCells(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 139673d901b8SMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 139773d901b8SMatthew G. Knepley PetscScalar *x = NULL; 13989c3cf19fSMatthew G. Knepley PetscInt qc = 0; 139973d901b8SMatthew G. Knepley 14002a4e142eSMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dm, c, quad, coords, fegeom.J, fegeom.invJ, fegeom.detJ);CHKERRQ(ierr); 140173d901b8SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr); 140273d901b8SMatthew G. Knepley 140315496722SMatthew G. Knepley for (field = 0, fieldOffset = 0; field < numFields; ++field) { 140415496722SMatthew G. Knepley PetscObject obj; 140515496722SMatthew G. Knepley PetscClassId id; 140673d901b8SMatthew G. Knepley void * const ctx = ctxs ? ctxs[field] : NULL; 140715496722SMatthew G. Knepley PetscInt Nb, Nc, q, fc; 140873d901b8SMatthew G. Knepley 140915496722SMatthew G. Knepley PetscReal elemDiff = 0.0; 141015496722SMatthew G. Knepley 141144a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, &obj);CHKERRQ(ierr); 141215496722SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 141315496722SMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscFEGetNumComponents((PetscFE) obj, &Nc);CHKERRQ(ierr);ierr = PetscFEGetDimension((PetscFE) obj, &Nb);CHKERRQ(ierr);} 141415496722SMatthew G. Knepley else if (id == PETSCFV_CLASSID) {ierr = PetscFVGetNumComponents((PetscFV) obj, &Nc);CHKERRQ(ierr);Nb = 1;} 1415ff1e0c32SBarry Smith else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 141673d901b8SMatthew G. Knepley if (debug) { 141773d901b8SMatthew G. Knepley char title[1024]; 1418ff1e0c32SBarry Smith ierr = PetscSNPrintf(title, 1023, "Solution for Field %D", field);CHKERRQ(ierr); 1419ca3d3a14SMatthew G. Knepley ierr = DMPrintCellVector(c, title, Nb, &x[fieldOffset]);CHKERRQ(ierr); 142073d901b8SMatthew G. Knepley } 14217318780aSToby Isaac for (q = 0; q < Nq; ++q) { 14222a4e142eSMatthew G. Knepley PetscFEGeom qgeom; 14232a4e142eSMatthew G. Knepley 14242a4e142eSMatthew G. Knepley qgeom.dimEmbed = fegeom.dimEmbed; 14252a4e142eSMatthew G. Knepley qgeom.J = &fegeom.J[q*coordDim*coordDim]; 14262a4e142eSMatthew G. Knepley qgeom.invJ = &fegeom.invJ[q*coordDim*coordDim]; 14272a4e142eSMatthew G. Knepley qgeom.detJ = &fegeom.detJ[q]; 1428ff1e0c32SBarry Smith if (fegeom.detJ[q] <= 0.0) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %D, quadrature point %D", (double)fegeom.detJ[q], c, q); 1429d3a7d86cSMatthew G. Knepley if (transform) { 1430d3a7d86cSMatthew G. Knepley gcoords = &coords[coordDim*Nq]; 1431d3a7d86cSMatthew G. Knepley ierr = DMPlexBasisTransformApplyReal_Internal(dm, &coords[coordDim*q], PETSC_TRUE, coordDim, &coords[coordDim*q], gcoords, dm->transformCtx);CHKERRQ(ierr); 1432d3a7d86cSMatthew G. Knepley } else { 1433d3a7d86cSMatthew G. Knepley gcoords = &coords[coordDim*q]; 1434d3a7d86cSMatthew G. Knepley } 14354bee2e38SMatthew G. Knepley ierr = (*funcs[field])(coordDim, time, gcoords, Nc, funcVal, ctx); 1436e735a8a9SMatthew G. Knepley if (ierr) { 1437e735a8a9SMatthew G. Knepley PetscErrorCode ierr2; 1438e735a8a9SMatthew G. Knepley ierr2 = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr2); 1439e735a8a9SMatthew G. Knepley ierr2 = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr2); 14402a4e142eSMatthew G. Knepley ierr2 = PetscFree7(localDiff,funcVal,interpolant,coords,fegeom.detJ,fegeom.J,fegeom.invJ);CHKERRQ(ierr2); 1441e735a8a9SMatthew G. Knepley CHKERRQ(ierr); 1442e735a8a9SMatthew G. Knepley } 1443ca3d3a14SMatthew G. Knepley if (transform) {ierr = DMPlexBasisTransformApply_Internal(dm, &coords[coordDim*q], PETSC_FALSE, Nc, funcVal, funcVal, dm->transformCtx);CHKERRQ(ierr);} 14442a4e142eSMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscFEInterpolate_Static((PetscFE) obj, &x[fieldOffset], &qgeom, q, interpolant);CHKERRQ(ierr);} 144515496722SMatthew G. Knepley else if (id == PETSCFV_CLASSID) {ierr = PetscFVInterpolate_Static((PetscFV) obj, &x[fieldOffset], q, interpolant);CHKERRQ(ierr);} 1446ff1e0c32SBarry Smith else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 144715496722SMatthew G. Knepley for (fc = 0; fc < Nc; ++fc) { 1448beaa55a6SMatthew G. Knepley const PetscReal wt = quadWeights[q*qNc+(qNc == 1 ? 0 : qc+fc)]; 1449ff1e0c32SBarry Smith if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, " elem %D field %D,%D point %g %g %g diff %g\n", c, field, fc, (double)(coordDim > 0 ? coords[coordDim*q] : 0.), (double)(coordDim > 1 ? coords[coordDim*q+1] : 0.),(double)(coordDim > 2 ? coords[coordDim*q+2] : 0.), (double)(PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*wt*fegeom.detJ[q]));CHKERRQ(ierr);} 14504bee2e38SMatthew G. Knepley elemDiff += PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*wt*fegeom.detJ[q]; 145173d901b8SMatthew G. Knepley } 145273d901b8SMatthew G. Knepley } 1453beaa55a6SMatthew G. Knepley fieldOffset += Nb; 14549c3cf19fSMatthew G. Knepley qc += Nc; 145573d901b8SMatthew G. Knepley localDiff[field] += elemDiff; 1456ff1e0c32SBarry Smith if (debug) {ierr = PetscPrintf(PETSC_COMM_SELF, " elem %D field %D cum diff %g\n", c, field, (double)localDiff[field]);CHKERRQ(ierr);} 145773d901b8SMatthew G. Knepley } 145873d901b8SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr); 145973d901b8SMatthew G. Knepley } 146073d901b8SMatthew G. Knepley ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr); 1461b2566f29SBarry Smith ierr = MPIU_Allreduce(localDiff, diff, numFields, MPIU_REAL, MPIU_SUM, PetscObjectComm((PetscObject)dm));CHKERRQ(ierr); 146273d901b8SMatthew G. Knepley for (field = 0; field < numFields; ++field) diff[field] = PetscSqrtReal(diff[field]); 14632a4e142eSMatthew G. Knepley ierr = PetscFree7(localDiff,funcVal,interpolant,coords,fegeom.detJ,fegeom.J,fegeom.invJ);CHKERRQ(ierr); 146473d901b8SMatthew G. Knepley PetscFunctionReturn(0); 146573d901b8SMatthew G. Knepley } 146673d901b8SMatthew G. Knepley 1467e729f68cSMatthew G. Knepley /*@C 1468e729f68cSMatthew G. Knepley DMPlexComputeL2DiffVec - This function computes the cellwise L_2 difference between a function u and an FEM interpolant solution u_h, and stores it in a Vec. 1469e729f68cSMatthew G. Knepley 1470d083f849SBarry Smith Collective on dm 1471c0f8e1fdSMatthew G. Knepley 1472e729f68cSMatthew G. Knepley Input Parameters: 1473e729f68cSMatthew G. Knepley + dm - The DM 14740163fd50SMatthew G. Knepley . time - The time 1475ca3eba1bSToby Isaac . funcs - The functions to evaluate for each field component: NULL means that component does not contribute to error calculation 1476e729f68cSMatthew G. Knepley . ctxs - Optional array of contexts to pass to each function, or NULL. 1477e729f68cSMatthew G. Knepley - X - The coefficient vector u_h 1478e729f68cSMatthew G. Knepley 1479e729f68cSMatthew G. Knepley Output Parameter: 1480e729f68cSMatthew G. Knepley . D - A Vec which holds the difference ||u - u_h||_2 for each cell 1481e729f68cSMatthew G. Knepley 1482e729f68cSMatthew G. Knepley Level: developer 1483e729f68cSMatthew G. Knepley 1484b698f381SToby Isaac .seealso: DMProjectFunction(), DMComputeL2Diff(), DMPlexComputeL2FieldDiff(), DMComputeL2GradientDiff() 1485e729f68cSMatthew G. Knepley @*/ 14860163fd50SMatthew G. Knepley PetscErrorCode DMPlexComputeL2DiffVec(DM dm, PetscReal time, PetscErrorCode (**funcs)(PetscInt, PetscReal, const PetscReal [], PetscInt, PetscScalar *, void *), void **ctxs, Vec X, Vec D) 1487e729f68cSMatthew G. Knepley { 1488e729f68cSMatthew G. Knepley PetscSection section; 1489e729f68cSMatthew G. Knepley PetscQuadrature quad; 1490e729f68cSMatthew G. Knepley Vec localX; 14914bee2e38SMatthew G. Knepley PetscFEGeom fegeom; 1492e729f68cSMatthew G. Knepley PetscScalar *funcVal, *interpolant; 14934bee2e38SMatthew G. Knepley PetscReal *coords; 1494e729f68cSMatthew G. Knepley const PetscReal *quadPoints, *quadWeights; 1495485ad865SMatthew G. Knepley PetscInt dim, coordDim, numFields, numComponents = 0, qNc, Nq, cStart, cEnd, c, field, fieldOffset; 1496e729f68cSMatthew G. Knepley PetscErrorCode ierr; 1497e729f68cSMatthew G. Knepley 1498e729f68cSMatthew G. Knepley PetscFunctionBegin; 1499e729f68cSMatthew G. Knepley ierr = VecSet(D, 0.0);CHKERRQ(ierr); 1500e729f68cSMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 15017318780aSToby Isaac ierr = DMGetCoordinateDim(dm, &coordDim);CHKERRQ(ierr); 150292fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 1503e729f68cSMatthew G. Knepley ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr); 1504e729f68cSMatthew G. Knepley ierr = DMGetLocalVector(dm, &localX);CHKERRQ(ierr); 1505bdd6f66aSToby Isaac ierr = DMProjectFunctionLocal(dm, time, funcs, ctxs, INSERT_BC_VALUES, localX);CHKERRQ(ierr); 1506e729f68cSMatthew G. Knepley ierr = DMGlobalToLocalBegin(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr); 1507e729f68cSMatthew G. Knepley ierr = DMGlobalToLocalEnd(dm, X, INSERT_VALUES, localX);CHKERRQ(ierr); 1508e729f68cSMatthew G. Knepley for (field = 0; field < numFields; ++field) { 1509e729f68cSMatthew G. Knepley PetscObject obj; 1510e729f68cSMatthew G. Knepley PetscClassId id; 1511e729f68cSMatthew G. Knepley PetscInt Nc; 1512e729f68cSMatthew G. Knepley 151344a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, &obj);CHKERRQ(ierr); 1514e729f68cSMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 1515e729f68cSMatthew G. Knepley if (id == PETSCFE_CLASSID) { 1516e729f68cSMatthew G. Knepley PetscFE fe = (PetscFE) obj; 1517e729f68cSMatthew G. Knepley 1518e729f68cSMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &quad);CHKERRQ(ierr); 1519e729f68cSMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 1520e729f68cSMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 1521e729f68cSMatthew G. Knepley PetscFV fv = (PetscFV) obj; 1522e729f68cSMatthew G. Knepley 1523e729f68cSMatthew G. Knepley ierr = PetscFVGetQuadrature(fv, &quad);CHKERRQ(ierr); 1524e729f68cSMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &Nc);CHKERRQ(ierr); 1525ff1e0c32SBarry Smith } else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 1526e729f68cSMatthew G. Knepley numComponents += Nc; 1527e729f68cSMatthew G. Knepley } 15289c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, &qNc, &Nq, &quadPoints, &quadWeights);CHKERRQ(ierr); 1529beaa55a6SMatthew G. Knepley if ((qNc != 1) && (qNc != numComponents)) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_SIZ, "Quadrature components %D != %D field components", qNc, numComponents); 15302a4e142eSMatthew G. Knepley ierr = PetscMalloc6(numComponents,&funcVal,numComponents,&interpolant,coordDim*Nq,&coords,Nq,&fegeom.detJ,coordDim*coordDim*Nq,&fegeom.J,coordDim*coordDim*Nq,&fegeom.invJ);CHKERRQ(ierr); 1531412e9a14SMatthew G. Knepley ierr = DMPlexGetSimplexOrBoxCells(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 1532e729f68cSMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 1533e729f68cSMatthew G. Knepley PetscScalar *x = NULL; 15346f288a59SMatthew G. Knepley PetscScalar elemDiff = 0.0; 15359c3cf19fSMatthew G. Knepley PetscInt qc = 0; 1536e729f68cSMatthew G. Knepley 15372a4e142eSMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dm, c, quad, coords, fegeom.J, fegeom.invJ, fegeom.detJ);CHKERRQ(ierr); 1538e729f68cSMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr); 1539e729f68cSMatthew G. Knepley 1540e729f68cSMatthew G. Knepley for (field = 0, fieldOffset = 0; field < numFields; ++field) { 1541e729f68cSMatthew G. Knepley PetscObject obj; 1542e729f68cSMatthew G. Knepley PetscClassId id; 1543e729f68cSMatthew G. Knepley void * const ctx = ctxs ? ctxs[field] : NULL; 1544e729f68cSMatthew G. Knepley PetscInt Nb, Nc, q, fc; 1545e729f68cSMatthew G. Knepley 154644a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, &obj);CHKERRQ(ierr); 1547e729f68cSMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 1548e729f68cSMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscFEGetNumComponents((PetscFE) obj, &Nc);CHKERRQ(ierr);ierr = PetscFEGetDimension((PetscFE) obj, &Nb);CHKERRQ(ierr);} 1549e729f68cSMatthew G. Knepley else if (id == PETSCFV_CLASSID) {ierr = PetscFVGetNumComponents((PetscFV) obj, &Nc);CHKERRQ(ierr);Nb = 1;} 1550ff1e0c32SBarry Smith else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 155123f34ed2SToby Isaac if (funcs[field]) { 15527318780aSToby Isaac for (q = 0; q < Nq; ++q) { 15532a4e142eSMatthew G. Knepley PetscFEGeom qgeom; 15542a4e142eSMatthew G. Knepley 15552a4e142eSMatthew G. Knepley qgeom.dimEmbed = fegeom.dimEmbed; 15562a4e142eSMatthew G. Knepley qgeom.J = &fegeom.J[q*coordDim*coordDim]; 15572a4e142eSMatthew G. Knepley qgeom.invJ = &fegeom.invJ[q*coordDim*coordDim]; 15582a4e142eSMatthew G. Knepley qgeom.detJ = &fegeom.detJ[q]; 15594bee2e38SMatthew G. Knepley if (fegeom.detJ[q] <= 0.0) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %D, quadrature points %D", (double)fegeom.detJ[q], c, q); 1560274e8aaeSMatthew G. Knepley ierr = (*funcs[field])(coordDim, time, &coords[q*coordDim], Nc, funcVal, ctx); 1561e735a8a9SMatthew G. Knepley if (ierr) { 1562e735a8a9SMatthew G. Knepley PetscErrorCode ierr2; 1563e735a8a9SMatthew G. Knepley ierr2 = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr2); 15642a4e142eSMatthew G. Knepley ierr2 = PetscFree6(funcVal,interpolant,coords,fegeom.detJ,fegeom.J,fegeom.invJ);CHKERRQ(ierr2); 1565e735a8a9SMatthew G. Knepley ierr2 = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr2); 1566e735a8a9SMatthew G. Knepley CHKERRQ(ierr); 1567e735a8a9SMatthew G. Knepley } 15682a4e142eSMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscFEInterpolate_Static((PetscFE) obj, &x[fieldOffset], &qgeom, q, interpolant);CHKERRQ(ierr);} 1569e729f68cSMatthew G. Knepley else if (id == PETSCFV_CLASSID) {ierr = PetscFVInterpolate_Static((PetscFV) obj, &x[fieldOffset], q, interpolant);CHKERRQ(ierr);} 1570ff1e0c32SBarry Smith else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 1571e729f68cSMatthew G. Knepley for (fc = 0; fc < Nc; ++fc) { 1572beaa55a6SMatthew G. Knepley const PetscReal wt = quadWeights[q*qNc+(qNc == 1 ? 0 : qc+fc)]; 15734bee2e38SMatthew G. Knepley elemDiff += PetscSqr(PetscRealPart(interpolant[fc] - funcVal[fc]))*wt*fegeom.detJ[q]; 1574e729f68cSMatthew G. Knepley } 1575e729f68cSMatthew G. Knepley } 157623f34ed2SToby Isaac } 1577beaa55a6SMatthew G. Knepley fieldOffset += Nb; 15789c3cf19fSMatthew G. Knepley qc += Nc; 1579e729f68cSMatthew G. Knepley } 1580e729f68cSMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, NULL, localX, c, NULL, &x);CHKERRQ(ierr); 158123f34ed2SToby Isaac ierr = VecSetValue(D, c - cStart, elemDiff, INSERT_VALUES);CHKERRQ(ierr); 1582e729f68cSMatthew G. Knepley } 15832a4e142eSMatthew G. Knepley ierr = PetscFree6(funcVal,interpolant,coords,fegeom.detJ,fegeom.J,fegeom.invJ);CHKERRQ(ierr); 1584e729f68cSMatthew G. Knepley ierr = DMRestoreLocalVector(dm, &localX);CHKERRQ(ierr); 1585e729f68cSMatthew G. Knepley ierr = VecSqrtAbs(D);CHKERRQ(ierr); 1586e729f68cSMatthew G. Knepley PetscFunctionReturn(0); 1587e729f68cSMatthew G. Knepley } 1588e729f68cSMatthew G. Knepley 15891555c271SMatthew G. Knepley /*@C 15901555c271SMatthew G. Knepley DMPlexComputeGradientClementInterpolant - This function computes the L2 projection of the cellwise gradient of a function u onto P1, and stores it in a Vec. 15911555c271SMatthew G. Knepley 1592d083f849SBarry Smith Collective on dm 1593c0f8e1fdSMatthew G. Knepley 15941555c271SMatthew G. Knepley Input Parameters: 15951555c271SMatthew G. Knepley + dm - The DM 15961555c271SMatthew G. Knepley - LocX - The coefficient vector u_h 15971555c271SMatthew G. Knepley 15981555c271SMatthew G. Knepley Output Parameter: 15991555c271SMatthew G. Knepley . locC - A Vec which holds the Clement interpolant of the gradient 16001555c271SMatthew G. Knepley 160195452b02SPatrick Sanan Notes: 160295452b02SPatrick Sanan Add citation to (Clement, 1975) and definition of the interpolant 16031555c271SMatthew G. Knepley \nabla u_h(v_i) = \sum_{T_i \in support(v_i)} |T_i| \nabla u_h(T_i) / \sum_{T_i \in support(v_i)} |T_i| where |T_i| is the cell volume 16041555c271SMatthew G. Knepley 16051555c271SMatthew G. Knepley Level: developer 16061555c271SMatthew G. Knepley 16071555c271SMatthew G. Knepley .seealso: DMProjectFunction(), DMComputeL2Diff(), DMPlexComputeL2FieldDiff(), DMComputeL2GradientDiff() 16081555c271SMatthew G. Knepley @*/ 16091555c271SMatthew G. Knepley PetscErrorCode DMPlexComputeGradientClementInterpolant(DM dm, Vec locX, Vec locC) 16101555c271SMatthew G. Knepley { 1611db1066baSMatthew G. Knepley DM_Plex *mesh = (DM_Plex *) dm->data; 1612db1066baSMatthew G. Knepley PetscInt debug = mesh->printFEM; 16131555c271SMatthew G. Knepley DM dmC; 16141555c271SMatthew G. Knepley PetscSection section; 16151555c271SMatthew G. Knepley PetscQuadrature quad; 16161555c271SMatthew G. Knepley PetscScalar *interpolant, *gradsum; 16174bee2e38SMatthew G. Knepley PetscFEGeom fegeom; 16184bee2e38SMatthew G. Knepley PetscReal *coords; 16191555c271SMatthew G. Knepley const PetscReal *quadPoints, *quadWeights; 1620485ad865SMatthew G. Knepley PetscInt dim, coordDim, numFields, numComponents = 0, qNc, Nq, cStart, cEnd, vStart, vEnd, v, field, fieldOffset; 16211555c271SMatthew G. Knepley PetscErrorCode ierr; 16221555c271SMatthew G. Knepley 16231555c271SMatthew G. Knepley PetscFunctionBegin; 16241555c271SMatthew G. Knepley ierr = VecGetDM(locC, &dmC);CHKERRQ(ierr); 16251555c271SMatthew G. Knepley ierr = VecSet(locC, 0.0);CHKERRQ(ierr); 16261555c271SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 16271555c271SMatthew G. Knepley ierr = DMGetCoordinateDim(dm, &coordDim);CHKERRQ(ierr); 16284bee2e38SMatthew G. Knepley fegeom.dimEmbed = coordDim; 162992fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 16301555c271SMatthew G. Knepley ierr = PetscSectionGetNumFields(section, &numFields);CHKERRQ(ierr); 16311555c271SMatthew G. Knepley for (field = 0; field < numFields; ++field) { 16321555c271SMatthew G. Knepley PetscObject obj; 16331555c271SMatthew G. Knepley PetscClassId id; 16341555c271SMatthew G. Knepley PetscInt Nc; 16351555c271SMatthew G. Knepley 163644a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, &obj);CHKERRQ(ierr); 16371555c271SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 16381555c271SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 16391555c271SMatthew G. Knepley PetscFE fe = (PetscFE) obj; 16401555c271SMatthew G. Knepley 16411555c271SMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &quad);CHKERRQ(ierr); 16421555c271SMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 16431555c271SMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 16441555c271SMatthew G. Knepley PetscFV fv = (PetscFV) obj; 16451555c271SMatthew G. Knepley 16461555c271SMatthew G. Knepley ierr = PetscFVGetQuadrature(fv, &quad);CHKERRQ(ierr); 16471555c271SMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &Nc);CHKERRQ(ierr); 1648ff1e0c32SBarry Smith } else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 16491555c271SMatthew G. Knepley numComponents += Nc; 16501555c271SMatthew G. Knepley } 16511555c271SMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, &qNc, &Nq, &quadPoints, &quadWeights);CHKERRQ(ierr); 16521555c271SMatthew G. Knepley if ((qNc != 1) && (qNc != numComponents)) SETERRQ2(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_SIZ, "Quadrature components %D != %D field components", qNc, numComponents); 16534bee2e38SMatthew G. Knepley ierr = PetscMalloc6(coordDim*numComponents*2,&gradsum,coordDim*numComponents,&interpolant,coordDim*Nq,&coords,Nq,&fegeom.detJ,coordDim*coordDim*Nq,&fegeom.J,coordDim*coordDim*Nq,&fegeom.invJ);CHKERRQ(ierr); 16541555c271SMatthew G. Knepley ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr); 1655412e9a14SMatthew G. Knepley ierr = DMPlexGetSimplexOrBoxCells(dm, 0, &cStart, &cEnd);CHKERRQ(ierr); 16561555c271SMatthew G. Knepley for (v = vStart; v < vEnd; ++v) { 16571555c271SMatthew G. Knepley PetscScalar volsum = 0.0; 16581555c271SMatthew G. Knepley PetscInt *star = NULL; 16591555c271SMatthew G. Knepley PetscInt starSize, st, d, fc; 16601555c271SMatthew G. Knepley 1661580bdb30SBarry Smith ierr = PetscArrayzero(gradsum, coordDim*numComponents);CHKERRQ(ierr); 16621555c271SMatthew G. Knepley ierr = DMPlexGetTransitiveClosure(dm, v, PETSC_FALSE, &starSize, &star);CHKERRQ(ierr); 16631555c271SMatthew G. Knepley for (st = 0; st < starSize*2; st += 2) { 16641555c271SMatthew G. Knepley const PetscInt cell = star[st]; 16651555c271SMatthew G. Knepley PetscScalar *grad = &gradsum[coordDim*numComponents]; 16661555c271SMatthew G. Knepley PetscScalar *x = NULL; 16671555c271SMatthew G. Knepley PetscReal vol = 0.0; 16681555c271SMatthew G. Knepley 16691555c271SMatthew G. Knepley if ((cell < cStart) || (cell >= cEnd)) continue; 16704bee2e38SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dm, cell, quad, coords, fegeom.J, fegeom.invJ, fegeom.detJ);CHKERRQ(ierr); 16711555c271SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, NULL, locX, cell, NULL, &x);CHKERRQ(ierr); 16721555c271SMatthew G. Knepley for (field = 0, fieldOffset = 0; field < numFields; ++field) { 16731555c271SMatthew G. Knepley PetscObject obj; 16741555c271SMatthew G. Knepley PetscClassId id; 16751555c271SMatthew G. Knepley PetscInt Nb, Nc, q, qc = 0; 16761555c271SMatthew G. Knepley 1677580bdb30SBarry Smith ierr = PetscArrayzero(grad, coordDim*numComponents);CHKERRQ(ierr); 167844a7f3ddSMatthew G. Knepley ierr = DMGetField(dm, field, NULL, &obj);CHKERRQ(ierr); 16791555c271SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 16801555c271SMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscFEGetNumComponents((PetscFE) obj, &Nc);CHKERRQ(ierr);ierr = PetscFEGetDimension((PetscFE) obj, &Nb);CHKERRQ(ierr);} 16811555c271SMatthew G. Knepley else if (id == PETSCFV_CLASSID) {ierr = PetscFVGetNumComponents((PetscFV) obj, &Nc);CHKERRQ(ierr);Nb = 1;} 1682ff1e0c32SBarry Smith else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 16831555c271SMatthew G. Knepley for (q = 0; q < Nq; ++q) { 16842a4e142eSMatthew G. Knepley PetscFEGeom qgeom; 16852a4e142eSMatthew G. Knepley 16862a4e142eSMatthew G. Knepley qgeom.dimEmbed = fegeom.dimEmbed; 16872a4e142eSMatthew G. Knepley qgeom.J = &fegeom.J[q*coordDim*coordDim]; 16882a4e142eSMatthew G. Knepley qgeom.invJ = &fegeom.invJ[q*coordDim*coordDim]; 16892a4e142eSMatthew G. Knepley qgeom.detJ = &fegeom.detJ[q]; 16904bee2e38SMatthew G. Knepley if (fegeom.detJ[q] <= 0.0) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Invalid determinant %g for element %D, quadrature points %D", (double)fegeom.detJ[q], cell, q); 16911555c271SMatthew G. Knepley if (ierr) { 16921555c271SMatthew G. Knepley PetscErrorCode ierr2; 16931555c271SMatthew G. Knepley ierr2 = DMPlexVecRestoreClosure(dm, NULL, locX, cell, NULL, &x);CHKERRQ(ierr2); 16941555c271SMatthew G. Knepley ierr2 = DMPlexRestoreTransitiveClosure(dm, v, PETSC_FALSE, &starSize, &star);CHKERRQ(ierr2); 16954bee2e38SMatthew G. Knepley ierr2 = PetscFree6(gradsum,interpolant,coords,fegeom.detJ,fegeom.J,fegeom.invJ);CHKERRQ(ierr2); 16961555c271SMatthew G. Knepley CHKERRQ(ierr); 16971555c271SMatthew G. Knepley } 16982a4e142eSMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscFEInterpolateGradient_Static((PetscFE) obj, &x[fieldOffset], &qgeom, q, interpolant);CHKERRQ(ierr);} 1699ff1e0c32SBarry Smith else SETERRQ1(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", field); 17001555c271SMatthew G. Knepley for (fc = 0; fc < Nc; ++fc) { 17011555c271SMatthew G. Knepley const PetscReal wt = quadWeights[q*qNc+qc+fc]; 17021555c271SMatthew G. Knepley 17034bee2e38SMatthew G. Knepley for (d = 0; d < coordDim; ++d) grad[fc*coordDim+d] += interpolant[fc*dim+d]*wt*fegeom.detJ[q]; 17041555c271SMatthew G. Knepley } 17054bee2e38SMatthew G. Knepley vol += quadWeights[q*qNc]*fegeom.detJ[q]; 17061555c271SMatthew G. Knepley } 17071555c271SMatthew G. Knepley fieldOffset += Nb; 17081555c271SMatthew G. Knepley qc += Nc; 17091555c271SMatthew G. Knepley } 17101555c271SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, NULL, locX, cell, NULL, &x);CHKERRQ(ierr); 1711f8527842SMatthew G. Knepley for (fc = 0; fc < numComponents; ++fc) { 1712f8527842SMatthew G. Knepley for (d = 0; d < coordDim; ++d) { 1713f8527842SMatthew G. Knepley gradsum[fc*coordDim+d] += grad[fc*coordDim+d]; 1714f8527842SMatthew G. Knepley } 1715f8527842SMatthew G. Knepley } 1716f8527842SMatthew G. Knepley volsum += vol; 1717db1066baSMatthew G. Knepley if (debug) { 17186d20ae03SJed Brown ierr = PetscPrintf(PETSC_COMM_SELF, "Cell %D gradient: [", cell);CHKERRQ(ierr); 17191555c271SMatthew G. Knepley for (fc = 0; fc < numComponents; ++fc) { 17201555c271SMatthew G. Knepley for (d = 0; d < coordDim; ++d) { 17211555c271SMatthew G. Knepley if (fc || d > 0) {ierr = PetscPrintf(PETSC_COMM_SELF, ", ");CHKERRQ(ierr);} 17226d20ae03SJed Brown ierr = PetscPrintf(PETSC_COMM_SELF, "%g", (double)PetscRealPart(grad[fc*coordDim+d]));CHKERRQ(ierr); 17231555c271SMatthew G. Knepley } 17241555c271SMatthew G. Knepley } 17251555c271SMatthew G. Knepley ierr = PetscPrintf(PETSC_COMM_SELF, "]\n");CHKERRQ(ierr); 1726db1066baSMatthew G. Knepley } 17271555c271SMatthew G. Knepley } 17281555c271SMatthew G. Knepley for (fc = 0; fc < numComponents; ++fc) { 17291555c271SMatthew G. Knepley for (d = 0; d < coordDim; ++d) gradsum[fc*coordDim+d] /= volsum; 17301555c271SMatthew G. Knepley } 17311555c271SMatthew G. Knepley ierr = DMPlexRestoreTransitiveClosure(dm, v, PETSC_FALSE, &starSize, &star);CHKERRQ(ierr); 17321555c271SMatthew G. Knepley ierr = DMPlexVecSetClosure(dmC, NULL, locC, v, gradsum, INSERT_VALUES);CHKERRQ(ierr); 17331555c271SMatthew G. Knepley } 17344bee2e38SMatthew G. Knepley ierr = PetscFree6(gradsum,interpolant,coords,fegeom.detJ,fegeom.J,fegeom.invJ);CHKERRQ(ierr); 17351555c271SMatthew G. Knepley PetscFunctionReturn(0); 17361555c271SMatthew G. Knepley } 17371555c271SMatthew G. Knepley 1738338f77d5SMatthew G. Knepley static PetscErrorCode DMPlexComputeIntegral_Internal(DM dm, Vec X, PetscInt cStart, PetscInt cEnd, PetscScalar *cintegral, void *user) 173973d901b8SMatthew G. Knepley { 1740338f77d5SMatthew G. Knepley DM dmAux = NULL; 174161aaff12SToby Isaac PetscDS prob, probAux = NULL; 174273d901b8SMatthew G. Knepley PetscSection section, sectionAux; 1743338f77d5SMatthew G. Knepley Vec locX, locA; 1744c330f8ffSToby Isaac PetscInt dim, numCells = cEnd - cStart, c, f; 1745c330f8ffSToby Isaac PetscBool useFVM = PETSC_FALSE; 1746338f77d5SMatthew G. Knepley /* DS */ 1747338f77d5SMatthew G. Knepley PetscInt Nf, totDim, *uOff, *uOff_x, numConstants; 1748338f77d5SMatthew G. Knepley PetscInt NfAux, totDimAux, *aOff; 1749338f77d5SMatthew G. Knepley PetscScalar *u, *a; 1750338f77d5SMatthew G. Knepley const PetscScalar *constants; 1751338f77d5SMatthew G. Knepley /* Geometry */ 1752c330f8ffSToby Isaac PetscFEGeom *cgeomFEM; 1753338f77d5SMatthew G. Knepley DM dmGrad; 1754c330f8ffSToby Isaac PetscQuadrature affineQuad = NULL; 1755338f77d5SMatthew G. Knepley Vec cellGeometryFVM = NULL, faceGeometryFVM = NULL, locGrad = NULL; 1756b5a3613cSMatthew G. Knepley PetscFVCellGeom *cgeomFVM; 1757338f77d5SMatthew G. Knepley const PetscScalar *lgrad; 1758b7260050SToby Isaac PetscInt maxDegree; 1759c330f8ffSToby Isaac DMField coordField; 1760c330f8ffSToby Isaac IS cellIS; 176173d901b8SMatthew G. Knepley PetscErrorCode ierr; 176273d901b8SMatthew G. Knepley 176373d901b8SMatthew G. Knepley PetscFunctionBegin; 1764338f77d5SMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 1765c73cfb54SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 176692fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 176773d901b8SMatthew G. Knepley ierr = PetscSectionGetNumFields(section, &Nf);CHKERRQ(ierr); 1768338f77d5SMatthew G. Knepley /* Determine which discretizations we have */ 1769b5a3613cSMatthew G. Knepley for (f = 0; f < Nf; ++f) { 1770b5a3613cSMatthew G. Knepley PetscObject obj; 1771b5a3613cSMatthew G. Knepley PetscClassId id; 1772b5a3613cSMatthew G. Knepley 1773b5a3613cSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr); 1774b5a3613cSMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 1775338f77d5SMatthew G. Knepley if (id == PETSCFV_CLASSID) useFVM = PETSC_TRUE; 1776338f77d5SMatthew G. Knepley } 1777338f77d5SMatthew G. Knepley /* Get local solution with boundary values */ 1778338f77d5SMatthew G. Knepley ierr = DMGetLocalVector(dm, &locX);CHKERRQ(ierr); 1779338f77d5SMatthew G. Knepley ierr = DMPlexInsertBoundaryValues(dm, PETSC_TRUE, locX, 0.0, NULL, NULL, NULL);CHKERRQ(ierr); 1780338f77d5SMatthew G. Knepley ierr = DMGlobalToLocalBegin(dm, X, INSERT_VALUES, locX);CHKERRQ(ierr); 1781338f77d5SMatthew G. Knepley ierr = DMGlobalToLocalEnd(dm, X, INSERT_VALUES, locX);CHKERRQ(ierr); 1782338f77d5SMatthew G. Knepley /* Read DS information */ 1783338f77d5SMatthew G. Knepley ierr = PetscDSGetTotalDimension(prob, &totDim);CHKERRQ(ierr); 1784338f77d5SMatthew G. Knepley ierr = PetscDSGetComponentOffsets(prob, &uOff);CHKERRQ(ierr); 1785338f77d5SMatthew G. Knepley ierr = PetscDSGetComponentDerivativeOffsets(prob, &uOff_x);CHKERRQ(ierr); 1786c330f8ffSToby Isaac ierr = ISCreateStride(PETSC_COMM_SELF,numCells,cStart,1,&cellIS);CHKERRQ(ierr); 1787338f77d5SMatthew G. Knepley ierr = PetscDSGetConstants(prob, &numConstants, &constants);CHKERRQ(ierr); 1788338f77d5SMatthew G. Knepley /* Read Auxiliary DS information */ 1789338f77d5SMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "dmAux", (PetscObject *) &dmAux);CHKERRQ(ierr); 1790338f77d5SMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "A", (PetscObject *) &locA);CHKERRQ(ierr); 1791338f77d5SMatthew G. Knepley if (dmAux) { 1792338f77d5SMatthew G. Knepley ierr = DMGetDS(dmAux, &probAux);CHKERRQ(ierr); 1793338f77d5SMatthew G. Knepley ierr = PetscDSGetNumFields(probAux, &NfAux);CHKERRQ(ierr); 179492fd8e1eSJed Brown ierr = DMGetLocalSection(dmAux, §ionAux);CHKERRQ(ierr); 1795338f77d5SMatthew G. Knepley ierr = PetscDSGetTotalDimension(probAux, &totDimAux);CHKERRQ(ierr); 1796338f77d5SMatthew G. Knepley ierr = PetscDSGetComponentOffsets(probAux, &aOff);CHKERRQ(ierr); 1797338f77d5SMatthew G. Knepley } 1798338f77d5SMatthew G. Knepley /* Allocate data arrays */ 1799338f77d5SMatthew G. Knepley ierr = PetscCalloc1(numCells*totDim, &u);CHKERRQ(ierr); 1800338f77d5SMatthew G. Knepley if (dmAux) {ierr = PetscMalloc1(numCells*totDimAux, &a);CHKERRQ(ierr);} 1801338f77d5SMatthew G. Knepley /* Read out geometry */ 1802c330f8ffSToby Isaac ierr = DMGetCoordinateField(dm,&coordField);CHKERRQ(ierr); 1803b7260050SToby Isaac ierr = DMFieldGetDegree(coordField,cellIS,NULL,&maxDegree);CHKERRQ(ierr); 1804b7260050SToby Isaac if (maxDegree <= 1) { 1805c330f8ffSToby Isaac ierr = DMFieldCreateDefaultQuadrature(coordField,cellIS,&affineQuad);CHKERRQ(ierr); 1806c330f8ffSToby Isaac if (affineQuad) { 1807c330f8ffSToby Isaac ierr = DMFieldCreateFEGeom(coordField,cellIS,affineQuad,PETSC_FALSE,&cgeomFEM);CHKERRQ(ierr); 1808338f77d5SMatthew G. Knepley } 1809b5a3613cSMatthew G. Knepley } 1810b5a3613cSMatthew G. Knepley if (useFVM) { 1811338f77d5SMatthew G. Knepley PetscFV fv = NULL; 1812b5a3613cSMatthew G. Knepley Vec grad; 1813b5a3613cSMatthew G. Knepley PetscInt fStart, fEnd; 1814b5a3613cSMatthew G. Knepley PetscBool compGrad; 1815b5a3613cSMatthew G. Knepley 1816338f77d5SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 1817338f77d5SMatthew G. Knepley PetscObject obj; 1818338f77d5SMatthew G. Knepley PetscClassId id; 1819338f77d5SMatthew G. Knepley 1820338f77d5SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr); 1821338f77d5SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 1822338f77d5SMatthew G. Knepley if (id == PETSCFV_CLASSID) {fv = (PetscFV) obj; break;} 1823338f77d5SMatthew G. Knepley } 1824338f77d5SMatthew G. Knepley ierr = PetscFVGetComputeGradients(fv, &compGrad);CHKERRQ(ierr); 1825338f77d5SMatthew G. Knepley ierr = PetscFVSetComputeGradients(fv, PETSC_TRUE);CHKERRQ(ierr); 1826b5a3613cSMatthew G. Knepley ierr = DMPlexComputeGeometryFVM(dm, &cellGeometryFVM, &faceGeometryFVM);CHKERRQ(ierr); 1827338f77d5SMatthew G. Knepley ierr = DMPlexComputeGradientFVM(dm, fv, faceGeometryFVM, cellGeometryFVM, &dmGrad);CHKERRQ(ierr); 1828338f77d5SMatthew G. Knepley ierr = PetscFVSetComputeGradients(fv, compGrad);CHKERRQ(ierr); 1829b5a3613cSMatthew G. Knepley ierr = VecGetArrayRead(cellGeometryFVM, (const PetscScalar **) &cgeomFVM);CHKERRQ(ierr); 1830b5a3613cSMatthew G. Knepley /* Reconstruct and limit cell gradients */ 1831b5a3613cSMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd);CHKERRQ(ierr); 1832b5a3613cSMatthew G. Knepley ierr = DMGetGlobalVector(dmGrad, &grad);CHKERRQ(ierr); 1833338f77d5SMatthew G. Knepley ierr = DMPlexReconstructGradients_Internal(dm, fv, fStart, fEnd, faceGeometryFVM, cellGeometryFVM, locX, grad);CHKERRQ(ierr); 1834b5a3613cSMatthew G. Knepley /* Communicate gradient values */ 1835b5a3613cSMatthew G. Knepley ierr = DMGetLocalVector(dmGrad, &locGrad);CHKERRQ(ierr); 1836b5a3613cSMatthew G. Knepley ierr = DMGlobalToLocalBegin(dmGrad, grad, INSERT_VALUES, locGrad);CHKERRQ(ierr); 1837b5a3613cSMatthew G. Knepley ierr = DMGlobalToLocalEnd(dmGrad, grad, INSERT_VALUES, locGrad);CHKERRQ(ierr); 1838b5a3613cSMatthew G. Knepley ierr = DMRestoreGlobalVector(dmGrad, &grad);CHKERRQ(ierr); 1839b5a3613cSMatthew G. Knepley /* Handle non-essential (e.g. outflow) boundary values */ 1840338f77d5SMatthew G. Knepley ierr = DMPlexInsertBoundaryValues(dm, PETSC_FALSE, locX, 0.0, faceGeometryFVM, cellGeometryFVM, locGrad);CHKERRQ(ierr); 1841b5a3613cSMatthew G. Knepley ierr = VecGetArrayRead(locGrad, &lgrad);CHKERRQ(ierr); 1842b5a3613cSMatthew G. Knepley } 1843338f77d5SMatthew G. Knepley /* Read out data from inputs */ 184473d901b8SMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 184573d901b8SMatthew G. Knepley PetscScalar *x = NULL; 184673d901b8SMatthew G. Knepley PetscInt i; 184773d901b8SMatthew G. Knepley 1848338f77d5SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, section, locX, c, NULL, &x);CHKERRQ(ierr); 18490f2d7e86SMatthew G. Knepley for (i = 0; i < totDim; ++i) u[c*totDim+i] = x[i]; 1850338f77d5SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, section, locX, c, NULL, &x);CHKERRQ(ierr); 185173d901b8SMatthew G. Knepley if (dmAux) { 1852338f77d5SMatthew G. Knepley ierr = DMPlexVecGetClosure(dmAux, sectionAux, locA, c, NULL, &x);CHKERRQ(ierr); 18530f2d7e86SMatthew G. Knepley for (i = 0; i < totDimAux; ++i) a[c*totDimAux+i] = x[i]; 1854338f77d5SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dmAux, sectionAux, locA, c, NULL, &x);CHKERRQ(ierr); 185573d901b8SMatthew G. Knepley } 185673d901b8SMatthew G. Knepley } 1857338f77d5SMatthew G. Knepley /* Do integration for each field */ 185873d901b8SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 1859c1f031eeSMatthew G. Knepley PetscObject obj; 1860c1f031eeSMatthew G. Knepley PetscClassId id; 1861c1f031eeSMatthew G. Knepley PetscInt numChunks, numBatches, batchSize, numBlocks, blockSize, Ne, Nr, offset; 186273d901b8SMatthew G. Knepley 1863c1f031eeSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr); 1864c1f031eeSMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 1865c1f031eeSMatthew G. Knepley if (id == PETSCFE_CLASSID) { 1866c1f031eeSMatthew G. Knepley PetscFE fe = (PetscFE) obj; 1867c1f031eeSMatthew G. Knepley PetscQuadrature q; 1868c330f8ffSToby Isaac PetscFEGeom *chunkGeom = NULL; 1869c1f031eeSMatthew G. Knepley PetscInt Nq, Nb; 1870c1f031eeSMatthew G. Knepley 18710f2d7e86SMatthew G. Knepley ierr = PetscFEGetTileSizes(fe, NULL, &numBlocks, NULL, &numBatches);CHKERRQ(ierr); 1872c1f031eeSMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &q);CHKERRQ(ierr); 18739c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(q, NULL, NULL, &Nq, NULL, NULL);CHKERRQ(ierr); 1874c1f031eeSMatthew G. Knepley ierr = PetscFEGetDimension(fe, &Nb);CHKERRQ(ierr); 1875c1f031eeSMatthew G. Knepley blockSize = Nb*Nq; 187673d901b8SMatthew G. Knepley batchSize = numBlocks * blockSize; 18770f2d7e86SMatthew G. Knepley ierr = PetscFESetTileSizes(fe, blockSize, numBlocks, batchSize, numBatches);CHKERRQ(ierr); 187873d901b8SMatthew G. Knepley numChunks = numCells / (numBatches*batchSize); 187973d901b8SMatthew G. Knepley Ne = numChunks*numBatches*batchSize; 188073d901b8SMatthew G. Knepley Nr = numCells % (numBatches*batchSize); 188173d901b8SMatthew G. Knepley offset = numCells - Nr; 1882c330f8ffSToby Isaac if (!affineQuad) { 1883c330f8ffSToby Isaac ierr = DMFieldCreateFEGeom(coordField,cellIS,q,PETSC_FALSE,&cgeomFEM);CHKERRQ(ierr); 1884c330f8ffSToby Isaac } 1885c330f8ffSToby Isaac ierr = PetscFEGeomGetChunk(cgeomFEM,0,offset,&chunkGeom);CHKERRQ(ierr); 18864bee2e38SMatthew G. Knepley ierr = PetscFEIntegrate(prob, f, Ne, chunkGeom, u, probAux, a, cintegral);CHKERRQ(ierr); 1887c330f8ffSToby Isaac ierr = PetscFEGeomGetChunk(cgeomFEM,offset,numCells,&chunkGeom);CHKERRQ(ierr); 18884bee2e38SMatthew G. Knepley ierr = PetscFEIntegrate(prob, f, Nr, chunkGeom, &u[offset*totDim], probAux, &a[offset*totDimAux], &cintegral[offset*Nf]);CHKERRQ(ierr); 1889c330f8ffSToby Isaac ierr = PetscFEGeomRestoreChunk(cgeomFEM,offset,numCells,&chunkGeom);CHKERRQ(ierr); 1890c330f8ffSToby Isaac if (!affineQuad) { 1891c330f8ffSToby Isaac ierr = PetscFEGeomDestroy(&cgeomFEM);CHKERRQ(ierr); 1892c330f8ffSToby Isaac } 1893c1f031eeSMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 1894c1f031eeSMatthew G. Knepley PetscInt foff; 1895420e96edSMatthew G. Knepley PetscPointFunc obj_func; 1896b69edc29SMatthew G. Knepley PetscScalar lint; 1897c1f031eeSMatthew G. Knepley 1898c1f031eeSMatthew G. Knepley ierr = PetscDSGetObjective(prob, f, &obj_func);CHKERRQ(ierr); 1899c1f031eeSMatthew G. Knepley ierr = PetscDSGetFieldOffset(prob, f, &foff);CHKERRQ(ierr); 1900c1f031eeSMatthew G. Knepley if (obj_func) { 1901c1f031eeSMatthew G. Knepley for (c = 0; c < numCells; ++c) { 1902b5a3613cSMatthew G. Knepley PetscScalar *u_x; 1903b5a3613cSMatthew G. Knepley 1904b5a3613cSMatthew G. Knepley ierr = DMPlexPointLocalRead(dmGrad, c, lgrad, &u_x);CHKERRQ(ierr); 1905338f77d5SMatthew G. Knepley obj_func(dim, Nf, NfAux, uOff, uOff_x, &u[totDim*c+foff], NULL, u_x, aOff, NULL, &a[totDimAux*c], NULL, NULL, 0.0, cgeomFVM[c].centroid, numConstants, constants, &lint); 1906338f77d5SMatthew G. Knepley cintegral[c*Nf+f] += PetscRealPart(lint)*cgeomFVM[c].volume; 190773d901b8SMatthew G. Knepley } 1908c1f031eeSMatthew G. Knepley } 1909ff1e0c32SBarry Smith } else SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", f); 1910c1f031eeSMatthew G. Knepley } 1911338f77d5SMatthew G. Knepley /* Cleanup data arrays */ 1912b5a3613cSMatthew G. Knepley if (useFVM) { 1913b5a3613cSMatthew G. Knepley ierr = VecRestoreArrayRead(locGrad, &lgrad);CHKERRQ(ierr); 1914b5a3613cSMatthew G. Knepley ierr = VecRestoreArrayRead(cellGeometryFVM, (const PetscScalar **) &cgeomFVM);CHKERRQ(ierr); 1915b5a3613cSMatthew G. Knepley ierr = DMRestoreLocalVector(dmGrad, &locGrad);CHKERRQ(ierr); 1916b5a3613cSMatthew G. Knepley ierr = VecDestroy(&faceGeometryFVM);CHKERRQ(ierr); 1917b5a3613cSMatthew G. Knepley ierr = VecDestroy(&cellGeometryFVM);CHKERRQ(ierr); 1918b5a3613cSMatthew G. Knepley ierr = DMDestroy(&dmGrad);CHKERRQ(ierr); 1919b5a3613cSMatthew G. Knepley } 192073d901b8SMatthew G. Knepley if (dmAux) {ierr = PetscFree(a);CHKERRQ(ierr);} 1921338f77d5SMatthew G. Knepley ierr = PetscFree(u);CHKERRQ(ierr); 1922338f77d5SMatthew G. Knepley /* Cleanup */ 1923f99c8401SToby Isaac if (affineQuad) { 1924f99c8401SToby Isaac ierr = PetscFEGeomDestroy(&cgeomFEM);CHKERRQ(ierr); 1925f99c8401SToby Isaac } 1926f99c8401SToby Isaac ierr = PetscQuadratureDestroy(&affineQuad);CHKERRQ(ierr); 1927c330f8ffSToby Isaac ierr = ISDestroy(&cellIS);CHKERRQ(ierr); 1928338f77d5SMatthew G. Knepley ierr = DMRestoreLocalVector(dm, &locX);CHKERRQ(ierr); 1929338f77d5SMatthew G. Knepley PetscFunctionReturn(0); 1930338f77d5SMatthew G. Knepley } 1931338f77d5SMatthew G. Knepley 1932338f77d5SMatthew G. Knepley /*@ 1933338f77d5SMatthew G. Knepley DMPlexComputeIntegralFEM - Form the integral over the domain from the global input X using pointwise functions specified by the user 1934338f77d5SMatthew G. Knepley 1935338f77d5SMatthew G. Knepley Input Parameters: 1936338f77d5SMatthew G. Knepley + dm - The mesh 1937338f77d5SMatthew G. Knepley . X - Global input vector 1938338f77d5SMatthew G. Knepley - user - The user context 1939338f77d5SMatthew G. Knepley 1940338f77d5SMatthew G. Knepley Output Parameter: 1941338f77d5SMatthew G. Knepley . integral - Integral for each field 1942338f77d5SMatthew G. Knepley 1943338f77d5SMatthew G. Knepley Level: developer 1944338f77d5SMatthew G. Knepley 1945338f77d5SMatthew G. Knepley .seealso: DMPlexComputeResidualFEM() 1946338f77d5SMatthew G. Knepley @*/ 1947b8feb594SMatthew G. Knepley PetscErrorCode DMPlexComputeIntegralFEM(DM dm, Vec X, PetscScalar *integral, void *user) 1948338f77d5SMatthew G. Knepley { 1949338f77d5SMatthew G. Knepley DM_Plex *mesh = (DM_Plex *) dm->data; 1950b8feb594SMatthew G. Knepley PetscScalar *cintegral, *lintegral; 1951412e9a14SMatthew G. Knepley PetscInt Nf, f, cellHeight, cStart, cEnd, cell; 1952338f77d5SMatthew G. Knepley PetscErrorCode ierr; 1953338f77d5SMatthew G. Knepley 1954338f77d5SMatthew G. Knepley PetscFunctionBegin; 1955338f77d5SMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 1956338f77d5SMatthew G. Knepley PetscValidHeaderSpecific(X, VEC_CLASSID, 2); 1957338f77d5SMatthew G. Knepley PetscValidPointer(integral, 3); 1958338f77d5SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_IntegralFEM,dm,0,0,0);CHKERRQ(ierr); 1959338f77d5SMatthew G. Knepley ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr); 1960338f77d5SMatthew G. Knepley ierr = DMPlexGetVTKCellHeight(dm, &cellHeight);CHKERRQ(ierr); 1961412e9a14SMatthew G. Knepley ierr = DMPlexGetSimplexOrBoxCells(dm, cellHeight, &cStart, &cEnd);CHKERRQ(ierr); 1962338f77d5SMatthew G. Knepley /* TODO Introduce a loop over large chunks (right now this is a single chunk) */ 1963338f77d5SMatthew G. Knepley ierr = PetscCalloc2(Nf, &lintegral, (cEnd-cStart)*Nf, &cintegral);CHKERRQ(ierr); 1964338f77d5SMatthew G. Knepley ierr = DMPlexComputeIntegral_Internal(dm, X, cStart, cEnd, cintegral, user);CHKERRQ(ierr); 1965338f77d5SMatthew G. Knepley /* Sum up values */ 1966338f77d5SMatthew G. Knepley for (cell = cStart; cell < cEnd; ++cell) { 1967338f77d5SMatthew G. Knepley const PetscInt c = cell - cStart; 1968338f77d5SMatthew G. Knepley 1969338f77d5SMatthew G. Knepley if (mesh->printFEM > 1) {ierr = DMPrintCellVector(cell, "Cell Integral", Nf, &cintegral[c*Nf]);CHKERRQ(ierr);} 1970b8feb594SMatthew G. Knepley for (f = 0; f < Nf; ++f) lintegral[f] += cintegral[c*Nf+f]; 1971338f77d5SMatthew G. Knepley } 1972b8feb594SMatthew G. Knepley ierr = MPIU_Allreduce(lintegral, integral, Nf, MPIU_SCALAR, MPIU_SUM, PetscObjectComm((PetscObject) dm));CHKERRQ(ierr); 197373d901b8SMatthew G. Knepley if (mesh->printFEM) { 1974338f77d5SMatthew G. Knepley ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "Integral:");CHKERRQ(ierr); 1975b8feb594SMatthew G. Knepley for (f = 0; f < Nf; ++f) {ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), " %g", (double) PetscRealPart(integral[f]));CHKERRQ(ierr);} 197673d901b8SMatthew G. Knepley ierr = PetscPrintf(PetscObjectComm((PetscObject) dm), "\n");CHKERRQ(ierr); 197773d901b8SMatthew G. Knepley } 1978338f77d5SMatthew G. Knepley ierr = PetscFree2(lintegral, cintegral);CHKERRQ(ierr); 1979338f77d5SMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_IntegralFEM,dm,0,0,0);CHKERRQ(ierr); 1980338f77d5SMatthew G. Knepley PetscFunctionReturn(0); 1981338f77d5SMatthew G. Knepley } 1982338f77d5SMatthew G. Knepley 1983338f77d5SMatthew G. Knepley /*@ 1984338f77d5SMatthew G. Knepley DMPlexComputeCellwiseIntegralFEM - Form the vector of cellwise integrals F from the global input X using pointwise functions specified by the user 1985338f77d5SMatthew G. Knepley 1986338f77d5SMatthew G. Knepley Input Parameters: 1987338f77d5SMatthew G. Knepley + dm - The mesh 1988338f77d5SMatthew G. Knepley . X - Global input vector 1989338f77d5SMatthew G. Knepley - user - The user context 1990338f77d5SMatthew G. Knepley 1991338f77d5SMatthew G. Knepley Output Parameter: 1992338f77d5SMatthew G. Knepley . integral - Cellwise integrals for each field 1993338f77d5SMatthew G. Knepley 1994338f77d5SMatthew G. Knepley Level: developer 1995338f77d5SMatthew G. Knepley 1996338f77d5SMatthew G. Knepley .seealso: DMPlexComputeResidualFEM() 1997338f77d5SMatthew G. Knepley @*/ 1998338f77d5SMatthew G. Knepley PetscErrorCode DMPlexComputeCellwiseIntegralFEM(DM dm, Vec X, Vec F, void *user) 1999338f77d5SMatthew G. Knepley { 2000338f77d5SMatthew G. Knepley DM_Plex *mesh = (DM_Plex *) dm->data; 2001338f77d5SMatthew G. Knepley DM dmF; 2002338f77d5SMatthew G. Knepley PetscSection sectionF; 2003338f77d5SMatthew G. Knepley PetscScalar *cintegral, *af; 2004412e9a14SMatthew G. Knepley PetscInt Nf, f, cellHeight, cStart, cEnd, cell; 2005338f77d5SMatthew G. Knepley PetscErrorCode ierr; 2006338f77d5SMatthew G. Knepley 2007338f77d5SMatthew G. Knepley PetscFunctionBegin; 2008338f77d5SMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 2009338f77d5SMatthew G. Knepley PetscValidHeaderSpecific(X, VEC_CLASSID, 2); 2010338f77d5SMatthew G. Knepley PetscValidHeaderSpecific(F, VEC_CLASSID, 3); 2011338f77d5SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_IntegralFEM,dm,0,0,0);CHKERRQ(ierr); 2012338f77d5SMatthew G. Knepley ierr = DMGetNumFields(dm, &Nf);CHKERRQ(ierr); 2013338f77d5SMatthew G. Knepley ierr = DMPlexGetVTKCellHeight(dm, &cellHeight);CHKERRQ(ierr); 2014412e9a14SMatthew G. Knepley ierr = DMPlexGetSimplexOrBoxCells(dm, cellHeight, &cStart, &cEnd);CHKERRQ(ierr); 2015338f77d5SMatthew G. Knepley /* TODO Introduce a loop over large chunks (right now this is a single chunk) */ 2016338f77d5SMatthew G. Knepley ierr = PetscCalloc1((cEnd-cStart)*Nf, &cintegral);CHKERRQ(ierr); 2017338f77d5SMatthew G. Knepley ierr = DMPlexComputeIntegral_Internal(dm, X, cStart, cEnd, cintegral, user);CHKERRQ(ierr); 2018338f77d5SMatthew G. Knepley /* Put values in F*/ 2019338f77d5SMatthew G. Knepley ierr = VecGetDM(F, &dmF);CHKERRQ(ierr); 202092fd8e1eSJed Brown ierr = DMGetLocalSection(dmF, §ionF);CHKERRQ(ierr); 2021338f77d5SMatthew G. Knepley ierr = VecGetArray(F, &af);CHKERRQ(ierr); 2022338f77d5SMatthew G. Knepley for (cell = cStart; cell < cEnd; ++cell) { 2023338f77d5SMatthew G. Knepley const PetscInt c = cell - cStart; 2024338f77d5SMatthew G. Knepley PetscInt dof, off; 2025338f77d5SMatthew G. Knepley 2026338f77d5SMatthew G. Knepley if (mesh->printFEM > 1) {ierr = DMPrintCellVector(cell, "Cell Integral", Nf, &cintegral[c*Nf]);CHKERRQ(ierr);} 2027338f77d5SMatthew G. Knepley ierr = PetscSectionGetDof(sectionF, cell, &dof);CHKERRQ(ierr); 2028338f77d5SMatthew G. Knepley ierr = PetscSectionGetOffset(sectionF, cell, &off);CHKERRQ(ierr); 2029338f77d5SMatthew G. Knepley if (dof != Nf) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "The number of cell dofs %D != %D", dof, Nf); 2030338f77d5SMatthew G. Knepley for (f = 0; f < Nf; ++f) af[off+f] = cintegral[c*Nf+f]; 2031338f77d5SMatthew G. Knepley } 2032338f77d5SMatthew G. Knepley ierr = VecRestoreArray(F, &af);CHKERRQ(ierr); 2033338f77d5SMatthew G. Knepley ierr = PetscFree(cintegral);CHKERRQ(ierr); 2034c1f031eeSMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_IntegralFEM,dm,0,0,0);CHKERRQ(ierr); 203573d901b8SMatthew G. Knepley PetscFunctionReturn(0); 203673d901b8SMatthew G. Knepley } 203773d901b8SMatthew G. Knepley 20389b6f715bSMatthew G. Knepley static PetscErrorCode DMPlexComputeBdIntegral_Internal(DM dm, Vec locX, IS pointIS, 20399b6f715bSMatthew G. Knepley void (*func)(PetscInt, PetscInt, PetscInt, 204064c72086SMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 204164c72086SMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 204264c72086SMatthew G. Knepley PetscReal, const PetscReal[], const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), 204364c72086SMatthew G. Knepley PetscScalar *fintegral, void *user) 204464c72086SMatthew G. Knepley { 20459b6f715bSMatthew G. Knepley DM plex = NULL, plexA = NULL; 2046a6e0b375SMatthew G. Knepley DMEnclosureType encAux; 20479b6f715bSMatthew G. Knepley PetscDS prob, probAux = NULL; 20489b6f715bSMatthew G. Knepley PetscSection section, sectionAux = NULL; 20499b6f715bSMatthew G. Knepley Vec locA = NULL; 20509b6f715bSMatthew G. Knepley DMField coordField; 20519b6f715bSMatthew G. Knepley PetscInt Nf, totDim, *uOff, *uOff_x; 20529b6f715bSMatthew G. Knepley PetscInt NfAux = 0, totDimAux = 0, *aOff = NULL; 20539b6f715bSMatthew G. Knepley PetscScalar *u, *a = NULL; 205464c72086SMatthew G. Knepley const PetscScalar *constants; 20559b6f715bSMatthew G. Knepley PetscInt numConstants, f; 205664c72086SMatthew G. Knepley PetscErrorCode ierr; 205764c72086SMatthew G. Knepley 205864c72086SMatthew G. Knepley PetscFunctionBegin; 20599b6f715bSMatthew G. Knepley ierr = DMGetCoordinateField(dm, &coordField);CHKERRQ(ierr); 20609b6f715bSMatthew G. Knepley ierr = DMConvert(dm, DMPLEX, &plex);CHKERRQ(ierr); 206164c72086SMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 206292fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 206364c72086SMatthew G. Knepley ierr = PetscSectionGetNumFields(section, &Nf);CHKERRQ(ierr); 206464c72086SMatthew G. Knepley /* Determine which discretizations we have */ 20659b6f715bSMatthew G. Knepley for (f = 0; f < Nf; ++f) { 206664c72086SMatthew G. Knepley PetscObject obj; 206764c72086SMatthew G. Knepley PetscClassId id; 206864c72086SMatthew G. Knepley 20699b6f715bSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr); 207064c72086SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 20719b6f715bSMatthew G. Knepley if (id == PETSCFV_CLASSID) SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_SUP, "Not supported for FVM (field %D)", f); 207264c72086SMatthew G. Knepley } 207364c72086SMatthew G. Knepley /* Read DS information */ 207464c72086SMatthew G. Knepley ierr = PetscDSGetTotalDimension(prob, &totDim);CHKERRQ(ierr); 207564c72086SMatthew G. Knepley ierr = PetscDSGetComponentOffsets(prob, &uOff);CHKERRQ(ierr); 207664c72086SMatthew G. Knepley ierr = PetscDSGetComponentDerivativeOffsets(prob, &uOff_x);CHKERRQ(ierr); 207764c72086SMatthew G. Knepley ierr = PetscDSGetConstants(prob, &numConstants, &constants);CHKERRQ(ierr); 207864c72086SMatthew G. Knepley /* Read Auxiliary DS information */ 207964c72086SMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "A", (PetscObject *) &locA);CHKERRQ(ierr); 20809b6f715bSMatthew G. Knepley if (locA) { 20819b6f715bSMatthew G. Knepley DM dmAux; 20829b6f715bSMatthew G. Knepley 20839b6f715bSMatthew G. Knepley ierr = VecGetDM(locA, &dmAux);CHKERRQ(ierr); 2084a6e0b375SMatthew G. Knepley ierr = DMGetEnclosureRelation(dmAux, dm, &encAux);CHKERRQ(ierr); 20859b6f715bSMatthew G. Knepley ierr = DMConvert(dmAux, DMPLEX, &plexA);CHKERRQ(ierr); 208664c72086SMatthew G. Knepley ierr = DMGetDS(dmAux, &probAux);CHKERRQ(ierr); 208764c72086SMatthew G. Knepley ierr = PetscDSGetNumFields(probAux, &NfAux);CHKERRQ(ierr); 208892fd8e1eSJed Brown ierr = DMGetLocalSection(dmAux, §ionAux);CHKERRQ(ierr); 208964c72086SMatthew G. Knepley ierr = PetscDSGetTotalDimension(probAux, &totDimAux);CHKERRQ(ierr); 209064c72086SMatthew G. Knepley ierr = PetscDSGetComponentOffsets(probAux, &aOff);CHKERRQ(ierr); 209164c72086SMatthew G. Knepley } 20929b6f715bSMatthew G. Knepley /* Integrate over points */ 20939b6f715bSMatthew G. Knepley { 20949b6f715bSMatthew G. Knepley PetscFEGeom *fgeom, *chunkGeom = NULL; 2095b7260050SToby Isaac PetscInt maxDegree; 20969b6f715bSMatthew G. Knepley PetscQuadrature qGeom = NULL; 20979b6f715bSMatthew G. Knepley const PetscInt *points; 20989b6f715bSMatthew G. Knepley PetscInt numFaces, face, Nq, field; 20999b6f715bSMatthew G. Knepley PetscInt numChunks, chunkSize, chunk, Nr, offset; 210064c72086SMatthew G. Knepley 21019b6f715bSMatthew G. Knepley ierr = ISGetLocalSize(pointIS, &numFaces);CHKERRQ(ierr); 21029b6f715bSMatthew G. Knepley ierr = ISGetIndices(pointIS, &points);CHKERRQ(ierr); 21039b6f715bSMatthew G. Knepley ierr = PetscCalloc2(numFaces*totDim, &u, locA ? numFaces*totDimAux : 0, &a);CHKERRQ(ierr); 2104b7260050SToby Isaac ierr = DMFieldGetDegree(coordField, pointIS, NULL, &maxDegree);CHKERRQ(ierr); 210564c72086SMatthew G. Knepley for (field = 0; field < Nf; ++field) { 210664c72086SMatthew G. Knepley PetscFE fe; 210764c72086SMatthew G. Knepley 210864c72086SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fe);CHKERRQ(ierr); 2109b7260050SToby Isaac if (maxDegree <= 1) {ierr = DMFieldCreateDefaultQuadrature(coordField, pointIS, &qGeom);CHKERRQ(ierr);} 21109b6f715bSMatthew G. Knepley if (!qGeom) { 21119b6f715bSMatthew G. Knepley ierr = PetscFEGetFaceQuadrature(fe, &qGeom);CHKERRQ(ierr); 21129b6f715bSMatthew G. Knepley ierr = PetscObjectReference((PetscObject) qGeom);CHKERRQ(ierr); 21139b6f715bSMatthew G. Knepley } 21149b6f715bSMatthew G. Knepley ierr = PetscQuadratureGetData(qGeom, NULL, NULL, &Nq, NULL, NULL);CHKERRQ(ierr); 21159b6f715bSMatthew G. Knepley ierr = DMPlexGetFEGeom(coordField, pointIS, qGeom, PETSC_TRUE, &fgeom);CHKERRQ(ierr); 21169b6f715bSMatthew G. Knepley for (face = 0; face < numFaces; ++face) { 21179b6f715bSMatthew G. Knepley const PetscInt point = points[face], *support, *cone; 21189b6f715bSMatthew G. Knepley PetscScalar *x = NULL; 21199b6f715bSMatthew G. Knepley PetscInt i, coneSize, faceLoc; 21209b6f715bSMatthew G. Knepley 21219b6f715bSMatthew G. Knepley ierr = DMPlexGetSupport(dm, point, &support);CHKERRQ(ierr); 21229b6f715bSMatthew G. Knepley ierr = DMPlexGetConeSize(dm, support[0], &coneSize);CHKERRQ(ierr); 21239b6f715bSMatthew G. Knepley ierr = DMPlexGetCone(dm, support[0], &cone);CHKERRQ(ierr); 21249b6f715bSMatthew G. Knepley for (faceLoc = 0; faceLoc < coneSize; ++faceLoc) if (cone[faceLoc] == point) break; 21259b6f715bSMatthew G. Knepley if (faceLoc == coneSize) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Could not find face %D in cone of support[0] %D", face, support[0]); 21269b6f715bSMatthew G. Knepley fgeom->face[face][0] = faceLoc; 21279b6f715bSMatthew G. Knepley ierr = DMPlexVecGetClosure(plex, section, locX, support[0], NULL, &x);CHKERRQ(ierr); 21289b6f715bSMatthew G. Knepley for (i = 0; i < totDim; ++i) u[face*totDim+i] = x[i]; 21299b6f715bSMatthew G. Knepley ierr = DMPlexVecRestoreClosure(plex, section, locX, support[0], NULL, &x);CHKERRQ(ierr); 21309b6f715bSMatthew G. Knepley if (locA) { 21319b6f715bSMatthew G. Knepley PetscInt subp; 2132a6e0b375SMatthew G. Knepley ierr = DMGetEnclosurePoint(plexA, dm, encAux, support[0], &subp);CHKERRQ(ierr); 21339b6f715bSMatthew G. Knepley ierr = DMPlexVecGetClosure(plexA, sectionAux, locA, subp, NULL, &x);CHKERRQ(ierr); 21349b6f715bSMatthew G. Knepley for (i = 0; i < totDimAux; ++i) a[f*totDimAux+i] = x[i]; 21359b6f715bSMatthew G. Knepley ierr = DMPlexVecRestoreClosure(plexA, sectionAux, locA, subp, NULL, &x);CHKERRQ(ierr); 21369b6f715bSMatthew G. Knepley } 21379b6f715bSMatthew G. Knepley } 21389b6f715bSMatthew G. Knepley /* Get blocking */ 21399b6f715bSMatthew G. Knepley { 21409b6f715bSMatthew G. Knepley PetscQuadrature q; 21419b6f715bSMatthew G. Knepley PetscInt numBatches, batchSize, numBlocks, blockSize; 21429b6f715bSMatthew G. Knepley PetscInt Nq, Nb; 21439b6f715bSMatthew G. Knepley 214464c72086SMatthew G. Knepley ierr = PetscFEGetTileSizes(fe, NULL, &numBlocks, NULL, &numBatches);CHKERRQ(ierr); 214564c72086SMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &q);CHKERRQ(ierr); 214664c72086SMatthew G. Knepley ierr = PetscQuadratureGetData(q, NULL, NULL, &Nq, NULL, NULL);CHKERRQ(ierr); 214764c72086SMatthew G. Knepley ierr = PetscFEGetDimension(fe, &Nb);CHKERRQ(ierr); 214864c72086SMatthew G. Knepley blockSize = Nb*Nq; 214964c72086SMatthew G. Knepley batchSize = numBlocks * blockSize; 21509b6f715bSMatthew G. Knepley chunkSize = numBatches*batchSize; 215164c72086SMatthew G. Knepley ierr = PetscFESetTileSizes(fe, blockSize, numBlocks, batchSize, numBatches);CHKERRQ(ierr); 21529b6f715bSMatthew G. Knepley numChunks = numFaces / chunkSize; 21539b6f715bSMatthew G. Knepley Nr = numFaces % chunkSize; 215464c72086SMatthew G. Knepley offset = numFaces - Nr; 215564c72086SMatthew G. Knepley } 21569b6f715bSMatthew G. Knepley /* Do integration for each field */ 21579b6f715bSMatthew G. Knepley for (chunk = 0; chunk < numChunks; ++chunk) { 21589b6f715bSMatthew G. Knepley ierr = PetscFEGeomGetChunk(fgeom, chunk*chunkSize, (chunk+1)*chunkSize, &chunkGeom);CHKERRQ(ierr); 21594bee2e38SMatthew G. Knepley ierr = PetscFEIntegrateBd(prob, field, func, chunkSize, chunkGeom, u, probAux, a, fintegral);CHKERRQ(ierr); 21609b6f715bSMatthew G. Knepley ierr = PetscFEGeomRestoreChunk(fgeom, 0, offset, &chunkGeom);CHKERRQ(ierr); 216164c72086SMatthew G. Knepley } 21629b6f715bSMatthew G. Knepley ierr = PetscFEGeomGetChunk(fgeom, offset, numFaces, &chunkGeom);CHKERRQ(ierr); 21634bee2e38SMatthew G. Knepley ierr = PetscFEIntegrateBd(prob, field, func, Nr, chunkGeom, &u[offset*totDim], probAux, a ? &a[offset*totDimAux] : NULL, &fintegral[offset*Nf]);CHKERRQ(ierr); 21649b6f715bSMatthew G. Knepley ierr = PetscFEGeomRestoreChunk(fgeom, offset, numFaces, &chunkGeom);CHKERRQ(ierr); 216564c72086SMatthew G. Knepley /* Cleanup data arrays */ 21669b6f715bSMatthew G. Knepley ierr = DMPlexRestoreFEGeom(coordField, pointIS, qGeom, PETSC_TRUE, &fgeom);CHKERRQ(ierr); 21679b6f715bSMatthew G. Knepley ierr = PetscQuadratureDestroy(&qGeom);CHKERRQ(ierr); 21689b6f715bSMatthew G. Knepley ierr = PetscFree2(u, a);CHKERRQ(ierr); 21699b6f715bSMatthew G. Knepley ierr = ISRestoreIndices(pointIS, &points);CHKERRQ(ierr); 217064c72086SMatthew G. Knepley } 217164c72086SMatthew G. Knepley } 21729b6f715bSMatthew G. Knepley if (plex) {ierr = DMDestroy(&plex);CHKERRQ(ierr);} 21739b6f715bSMatthew G. Knepley if (plexA) {ierr = DMDestroy(&plexA);CHKERRQ(ierr);} 21749b6f715bSMatthew G. Knepley PetscFunctionReturn(0); 21759b6f715bSMatthew G. Knepley } 21769b6f715bSMatthew G. Knepley 21779b6f715bSMatthew G. Knepley /*@ 21789b6f715bSMatthew G. Knepley DMPlexComputeBdIntegral - Form the integral over the specified boundary from the global input X using pointwise functions specified by the user 21799b6f715bSMatthew G. Knepley 21809b6f715bSMatthew G. Knepley Input Parameters: 21819b6f715bSMatthew G. Knepley + dm - The mesh 21829b6f715bSMatthew G. Knepley . X - Global input vector 21839b6f715bSMatthew G. Knepley . label - The boundary DMLabel 21849b6f715bSMatthew G. Knepley . numVals - The number of label values to use, or PETSC_DETERMINE for all values 21859b6f715bSMatthew G. Knepley . vals - The label values to use, or PETSC_NULL for all values 21869b6f715bSMatthew G. Knepley . func = The function to integrate along the boundary 21879b6f715bSMatthew G. Knepley - user - The user context 21889b6f715bSMatthew G. Knepley 21899b6f715bSMatthew G. Knepley Output Parameter: 21909b6f715bSMatthew G. Knepley . integral - Integral for each field 21919b6f715bSMatthew G. Knepley 21929b6f715bSMatthew G. Knepley Level: developer 21939b6f715bSMatthew G. Knepley 21949b6f715bSMatthew G. Knepley .seealso: DMPlexComputeIntegralFEM(), DMPlexComputeBdResidualFEM() 21959b6f715bSMatthew G. Knepley @*/ 21969b6f715bSMatthew G. Knepley PetscErrorCode DMPlexComputeBdIntegral(DM dm, Vec X, DMLabel label, PetscInt numVals, const PetscInt vals[], 21979b6f715bSMatthew G. Knepley void (*func)(PetscInt, PetscInt, PetscInt, 21989b6f715bSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 21999b6f715bSMatthew G. Knepley const PetscInt[], const PetscInt[], const PetscScalar[], const PetscScalar[], const PetscScalar[], 22009b6f715bSMatthew G. Knepley PetscReal, const PetscReal[], const PetscReal[], PetscInt, const PetscScalar[], PetscScalar[]), 22019b6f715bSMatthew G. Knepley PetscScalar *integral, void *user) 22029b6f715bSMatthew G. Knepley { 22039b6f715bSMatthew G. Knepley Vec locX; 22049b6f715bSMatthew G. Knepley PetscSection section; 22059b6f715bSMatthew G. Knepley DMLabel depthLabel; 22069b6f715bSMatthew G. Knepley IS facetIS; 22079b6f715bSMatthew G. Knepley PetscInt dim, Nf, f, v; 22089b6f715bSMatthew G. Knepley PetscErrorCode ierr; 22099b6f715bSMatthew G. Knepley 22109b6f715bSMatthew G. Knepley PetscFunctionBegin; 22119b6f715bSMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 22129b6f715bSMatthew G. Knepley PetscValidHeaderSpecific(X, VEC_CLASSID, 2); 22139b6f715bSMatthew G. Knepley PetscValidPointer(label, 3); 22149b6f715bSMatthew G. Knepley if (vals) PetscValidPointer(vals, 5); 22159b6f715bSMatthew G. Knepley PetscValidPointer(integral, 6); 22169b6f715bSMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_IntegralFEM,dm,0,0,0);CHKERRQ(ierr); 22179b6f715bSMatthew G. Knepley ierr = DMPlexGetDepthLabel(dm, &depthLabel);CHKERRQ(ierr); 22189b6f715bSMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 22199b6f715bSMatthew G. Knepley ierr = DMLabelGetStratumIS(depthLabel, dim-1, &facetIS);CHKERRQ(ierr); 222092fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 22219b6f715bSMatthew G. Knepley ierr = PetscSectionGetNumFields(section, &Nf);CHKERRQ(ierr); 22229b6f715bSMatthew G. Knepley /* Get local solution with boundary values */ 22239b6f715bSMatthew G. Knepley ierr = DMGetLocalVector(dm, &locX);CHKERRQ(ierr); 22249b6f715bSMatthew G. Knepley ierr = DMPlexInsertBoundaryValues(dm, PETSC_TRUE, locX, 0.0, NULL, NULL, NULL);CHKERRQ(ierr); 22259b6f715bSMatthew G. Knepley ierr = DMGlobalToLocalBegin(dm, X, INSERT_VALUES, locX);CHKERRQ(ierr); 22269b6f715bSMatthew G. Knepley ierr = DMGlobalToLocalEnd(dm, X, INSERT_VALUES, locX);CHKERRQ(ierr); 22279b6f715bSMatthew G. Knepley /* Loop over label values */ 2228580bdb30SBarry Smith ierr = PetscArrayzero(integral, Nf);CHKERRQ(ierr); 22299b6f715bSMatthew G. Knepley for (v = 0; v < numVals; ++v) { 22309b6f715bSMatthew G. Knepley IS pointIS; 22319b6f715bSMatthew G. Knepley PetscInt numFaces, face; 22329b6f715bSMatthew G. Knepley PetscScalar *fintegral; 22339b6f715bSMatthew G. Knepley 22349b6f715bSMatthew G. Knepley ierr = DMLabelGetStratumIS(label, vals[v], &pointIS);CHKERRQ(ierr); 22359b6f715bSMatthew G. Knepley if (!pointIS) continue; /* No points with that id on this process */ 22369b6f715bSMatthew G. Knepley { 22379b6f715bSMatthew G. Knepley IS isectIS; 22389b6f715bSMatthew G. Knepley 22399b6f715bSMatthew G. Knepley /* TODO: Special cases of ISIntersect where it is quick to check a priori if one is a superset of the other */ 22409b6f715bSMatthew G. Knepley ierr = ISIntersect_Caching_Internal(facetIS, pointIS, &isectIS);CHKERRQ(ierr); 22419b6f715bSMatthew G. Knepley ierr = ISDestroy(&pointIS);CHKERRQ(ierr); 22429b6f715bSMatthew G. Knepley pointIS = isectIS; 22439b6f715bSMatthew G. Knepley } 22449b6f715bSMatthew G. Knepley ierr = ISGetLocalSize(pointIS, &numFaces);CHKERRQ(ierr); 22459b6f715bSMatthew G. Knepley ierr = PetscCalloc1(numFaces*Nf, &fintegral);CHKERRQ(ierr); 22469b6f715bSMatthew G. Knepley ierr = DMPlexComputeBdIntegral_Internal(dm, locX, pointIS, func, fintegral, user);CHKERRQ(ierr); 22479b6f715bSMatthew G. Knepley /* Sum point contributions into integral */ 22489b6f715bSMatthew G. Knepley for (f = 0; f < Nf; ++f) for (face = 0; face < numFaces; ++face) integral[f] += fintegral[face*Nf+f]; 22499b6f715bSMatthew G. Knepley ierr = PetscFree(fintegral);CHKERRQ(ierr); 22509b6f715bSMatthew G. Knepley ierr = ISDestroy(&pointIS);CHKERRQ(ierr); 22519b6f715bSMatthew G. Knepley } 225264c72086SMatthew G. Knepley ierr = DMRestoreLocalVector(dm, &locX);CHKERRQ(ierr); 22539b6f715bSMatthew G. Knepley ierr = ISDestroy(&facetIS);CHKERRQ(ierr); 22549b6f715bSMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_IntegralFEM,dm,0,0,0);CHKERRQ(ierr); 225564c72086SMatthew G. Knepley PetscFunctionReturn(0); 225664c72086SMatthew G. Knepley } 225764c72086SMatthew G. Knepley 2258d69c5d34SMatthew G. Knepley /*@ 225968132eb9SMatthew G. Knepley DMPlexComputeInterpolatorNested - Form the local portion of the interpolation matrix I from the coarse DM to the uniformly refined DM. 2260d69c5d34SMatthew G. Knepley 2261d69c5d34SMatthew G. Knepley Input Parameters: 2262d69c5d34SMatthew G. Knepley + dmf - The fine mesh 2263d69c5d34SMatthew G. Knepley . dmc - The coarse mesh 2264d69c5d34SMatthew G. Knepley - user - The user context 2265d69c5d34SMatthew G. Knepley 2266d69c5d34SMatthew G. Knepley Output Parameter: 2267934789fcSMatthew G. Knepley . In - The interpolation matrix 2268d69c5d34SMatthew G. Knepley 2269d69c5d34SMatthew G. Knepley Level: developer 2270d69c5d34SMatthew G. Knepley 227168132eb9SMatthew G. Knepley .seealso: DMPlexComputeInterpolatorGeneral(), DMPlexComputeJacobianFEM() 2272d69c5d34SMatthew G. Knepley @*/ 227368132eb9SMatthew G. Knepley PetscErrorCode DMPlexComputeInterpolatorNested(DM dmc, DM dmf, Mat In, void *user) 2274d69c5d34SMatthew G. Knepley { 2275d69c5d34SMatthew G. Knepley DM_Plex *mesh = (DM_Plex *) dmc->data; 2276d69c5d34SMatthew G. Knepley const char *name = "Interpolator"; 22777c2b9c7aSMatthew G. Knepley PetscDS cds, rds; 2278d69c5d34SMatthew G. Knepley PetscFE *feRef; 227997c42addSMatthew G. Knepley PetscFV *fvRef; 2280d69c5d34SMatthew G. Knepley PetscSection fsection, fglobalSection; 2281d69c5d34SMatthew G. Knepley PetscSection csection, cglobalSection; 2282d69c5d34SMatthew G. Knepley PetscScalar *elemMat; 2283485ad865SMatthew G. Knepley PetscInt dim, Nf, f, fieldI, fieldJ, offsetI, offsetJ, cStart, cEnd, c; 22840f2d7e86SMatthew G. Knepley PetscInt cTotDim, rTotDim = 0; 2285d69c5d34SMatthew G. Knepley PetscErrorCode ierr; 2286d69c5d34SMatthew G. Knepley 2287d69c5d34SMatthew G. Knepley PetscFunctionBegin; 2288d69c5d34SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_InterpolatorFEM,dmc,dmf,0,0);CHKERRQ(ierr); 2289c73cfb54SMatthew G. Knepley ierr = DMGetDimension(dmf, &dim);CHKERRQ(ierr); 229092fd8e1eSJed Brown ierr = DMGetLocalSection(dmf, &fsection);CHKERRQ(ierr); 2291e87a4003SBarry Smith ierr = DMGetGlobalSection(dmf, &fglobalSection);CHKERRQ(ierr); 229292fd8e1eSJed Brown ierr = DMGetLocalSection(dmc, &csection);CHKERRQ(ierr); 2293e87a4003SBarry Smith ierr = DMGetGlobalSection(dmc, &cglobalSection);CHKERRQ(ierr); 2294d69c5d34SMatthew G. Knepley ierr = PetscSectionGetNumFields(fsection, &Nf);CHKERRQ(ierr); 2295412e9a14SMatthew G. Knepley ierr = DMPlexGetSimplexOrBoxCells(dmc, 0, &cStart, &cEnd);CHKERRQ(ierr); 22967c2b9c7aSMatthew G. Knepley ierr = DMGetDS(dmc, &cds);CHKERRQ(ierr); 22977c2b9c7aSMatthew G. Knepley ierr = DMGetDS(dmf, &rds);CHKERRQ(ierr); 229897c42addSMatthew G. Knepley ierr = PetscCalloc2(Nf, &feRef, Nf, &fvRef);CHKERRQ(ierr); 2299d69c5d34SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 230097c42addSMatthew G. Knepley PetscObject obj; 230197c42addSMatthew G. Knepley PetscClassId id; 2302aa7890ccSMatthew G. Knepley PetscInt rNb = 0, Nc = 0; 2303d69c5d34SMatthew G. Knepley 23047c2b9c7aSMatthew G. Knepley ierr = PetscDSGetDiscretization(rds, f, &obj);CHKERRQ(ierr); 230597c42addSMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 230697c42addSMatthew G. Knepley if (id == PETSCFE_CLASSID) { 230797c42addSMatthew G. Knepley PetscFE fe = (PetscFE) obj; 230897c42addSMatthew G. Knepley 23090f2d7e86SMatthew G. Knepley ierr = PetscFERefine(fe, &feRef[f]);CHKERRQ(ierr); 2310d69c5d34SMatthew G. Knepley ierr = PetscFEGetDimension(feRef[f], &rNb);CHKERRQ(ierr); 23110f2d7e86SMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 231297c42addSMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 231397c42addSMatthew G. Knepley PetscFV fv = (PetscFV) obj; 231497c42addSMatthew G. Knepley PetscDualSpace Q; 231597c42addSMatthew G. Knepley 231697c42addSMatthew G. Knepley ierr = PetscFVRefine(fv, &fvRef[f]);CHKERRQ(ierr); 231797c42addSMatthew G. Knepley ierr = PetscFVGetDualSpace(fvRef[f], &Q);CHKERRQ(ierr); 231897c42addSMatthew G. Knepley ierr = PetscDualSpaceGetDimension(Q, &rNb);CHKERRQ(ierr); 231997c42addSMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &Nc);CHKERRQ(ierr); 232097c42addSMatthew G. Knepley } 23219c3cf19fSMatthew G. Knepley rTotDim += rNb; 2322d69c5d34SMatthew G. Knepley } 23237c2b9c7aSMatthew G. Knepley ierr = PetscDSGetTotalDimension(cds, &cTotDim);CHKERRQ(ierr); 23240f2d7e86SMatthew G. Knepley ierr = PetscMalloc1(rTotDim*cTotDim,&elemMat);CHKERRQ(ierr); 2325580bdb30SBarry Smith ierr = PetscArrayzero(elemMat, rTotDim*cTotDim);CHKERRQ(ierr); 2326d69c5d34SMatthew G. Knepley for (fieldI = 0, offsetI = 0; fieldI < Nf; ++fieldI) { 2327d69c5d34SMatthew G. Knepley PetscDualSpace Qref; 2328d69c5d34SMatthew G. Knepley PetscQuadrature f; 2329d69c5d34SMatthew G. Knepley const PetscReal *qpoints, *qweights; 2330d69c5d34SMatthew G. Knepley PetscReal *points; 2331d69c5d34SMatthew G. Knepley PetscInt npoints = 0, Nc, Np, fpdim, i, k, p, d; 2332d69c5d34SMatthew G. Knepley 2333d69c5d34SMatthew G. Knepley /* Compose points from all dual basis functionals */ 233497c42addSMatthew G. Knepley if (feRef[fieldI]) { 2335d69c5d34SMatthew G. Knepley ierr = PetscFEGetDualSpace(feRef[fieldI], &Qref);CHKERRQ(ierr); 23360f2d7e86SMatthew G. Knepley ierr = PetscFEGetNumComponents(feRef[fieldI], &Nc);CHKERRQ(ierr); 233797c42addSMatthew G. Knepley } else { 233897c42addSMatthew G. Knepley ierr = PetscFVGetDualSpace(fvRef[fieldI], &Qref);CHKERRQ(ierr); 233997c42addSMatthew G. Knepley ierr = PetscFVGetNumComponents(fvRef[fieldI], &Nc);CHKERRQ(ierr); 234097c42addSMatthew G. Knepley } 2341d69c5d34SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(Qref, &fpdim);CHKERRQ(ierr); 2342d69c5d34SMatthew G. Knepley for (i = 0; i < fpdim; ++i) { 2343d69c5d34SMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(Qref, i, &f);CHKERRQ(ierr); 23449c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(f, NULL, NULL, &Np, NULL, NULL);CHKERRQ(ierr); 2345d69c5d34SMatthew G. Knepley npoints += Np; 2346d69c5d34SMatthew G. Knepley } 2347d69c5d34SMatthew G. Knepley ierr = PetscMalloc1(npoints*dim,&points);CHKERRQ(ierr); 2348d69c5d34SMatthew G. Knepley for (i = 0, k = 0; i < fpdim; ++i) { 2349d69c5d34SMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(Qref, i, &f);CHKERRQ(ierr); 23509c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(f, NULL, NULL, &Np, &qpoints, NULL);CHKERRQ(ierr); 2351d69c5d34SMatthew G. Knepley for (p = 0; p < Np; ++p, ++k) for (d = 0; d < dim; ++d) points[k*dim+d] = qpoints[p*dim+d]; 2352d69c5d34SMatthew G. Knepley } 2353d69c5d34SMatthew G. Knepley 2354d69c5d34SMatthew G. Knepley for (fieldJ = 0, offsetJ = 0; fieldJ < Nf; ++fieldJ) { 235597c42addSMatthew G. Knepley PetscObject obj; 235697c42addSMatthew G. Knepley PetscClassId id; 23579c3cf19fSMatthew G. Knepley PetscInt NcJ = 0, cpdim = 0, j, qNc; 2358d69c5d34SMatthew G. Knepley 23597c2b9c7aSMatthew G. Knepley ierr = PetscDSGetDiscretization(cds, fieldJ, &obj);CHKERRQ(ierr); 236097c42addSMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 236197c42addSMatthew G. Knepley if (id == PETSCFE_CLASSID) { 236297c42addSMatthew G. Knepley PetscFE fe = (PetscFE) obj; 2363ef0bb6c7SMatthew G. Knepley PetscTabulation T = NULL; 2364d69c5d34SMatthew G. Knepley 2365d69c5d34SMatthew G. Knepley /* Evaluate basis at points */ 23660f2d7e86SMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &NcJ);CHKERRQ(ierr); 23670f2d7e86SMatthew G. Knepley ierr = PetscFEGetDimension(fe, &cpdim);CHKERRQ(ierr); 2368ffe73a53SMatthew G. Knepley /* For now, fields only interpolate themselves */ 2369ffe73a53SMatthew G. Knepley if (fieldI == fieldJ) { 23709c3cf19fSMatthew G. Knepley if (Nc != NcJ) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of components in fine space field %D does not match coarse field %D", Nc, NcJ); 2371ef0bb6c7SMatthew G. Knepley ierr = PetscFECreateTabulation(fe, 1, npoints, points, 0, &T);CHKERRQ(ierr); 2372d69c5d34SMatthew G. Knepley for (i = 0, k = 0; i < fpdim; ++i) { 2373d69c5d34SMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(Qref, i, &f);CHKERRQ(ierr); 23749c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(f, NULL, &qNc, &Np, NULL, &qweights);CHKERRQ(ierr); 23759c3cf19fSMatthew G. Knepley if (qNc != NcJ) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of components in quadrature %D does not match coarse field %D", qNc, NcJ); 2376d69c5d34SMatthew G. Knepley for (p = 0; p < Np; ++p, ++k) { 237736a6d9c0SMatthew G. Knepley for (j = 0; j < cpdim; ++j) { 2378d172c84bSMatthew G. Knepley /* 2379d172c84bSMatthew G. Knepley cTotDim: Total columns in element interpolation matrix, sum of number of dual basis functionals in each field 2380d172c84bSMatthew G. Knepley offsetI, offsetJ: Offsets into the larger element interpolation matrix for different fields 2381d172c84bSMatthew G. Knepley fpdim, i, cpdim, j: Dofs for fine and coarse grids, correspond to dual space basis functionals 2382d172c84bSMatthew G. Knepley qNC, Nc, Ncj, c: Number of components in this field 2383d172c84bSMatthew G. Knepley Np, p: Number of quad points in the fine grid functional i 2384d172c84bSMatthew G. Knepley k: i*Np + p, overall point number for the interpolation 2385d172c84bSMatthew G. Knepley */ 2386ef0bb6c7SMatthew G. Knepley for (c = 0; c < Nc; ++c) elemMat[(offsetI + i)*cTotDim + offsetJ + j] += T->T[0][k*cpdim*NcJ+j*Nc+c]*qweights[p*qNc+c]; 238736a6d9c0SMatthew G. Knepley } 2388d69c5d34SMatthew G. Knepley } 2389d69c5d34SMatthew G. Knepley } 2390ef0bb6c7SMatthew G. Knepley ierr = PetscTabulationDestroy(&T);CHKERRQ(ierr);CHKERRQ(ierr); 2391ffe73a53SMatthew G. Knepley } 239297c42addSMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 239397c42addSMatthew G. Knepley PetscFV fv = (PetscFV) obj; 239497c42addSMatthew G. Knepley 239597c42addSMatthew G. Knepley /* Evaluate constant function at points */ 239697c42addSMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &NcJ);CHKERRQ(ierr); 239797c42addSMatthew G. Knepley cpdim = 1; 239897c42addSMatthew G. Knepley /* For now, fields only interpolate themselves */ 239997c42addSMatthew G. Knepley if (fieldI == fieldJ) { 2400ff1e0c32SBarry Smith if (Nc != NcJ) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of components in fine space field %D does not match coarse field %D", Nc, NcJ); 240197c42addSMatthew G. Knepley for (i = 0, k = 0; i < fpdim; ++i) { 240297c42addSMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(Qref, i, &f);CHKERRQ(ierr); 24039c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(f, NULL, &qNc, &Np, NULL, &qweights);CHKERRQ(ierr); 24049c3cf19fSMatthew G. Knepley if (qNc != NcJ) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of components in quadrature %D does not match coarse field %D", qNc, NcJ); 240597c42addSMatthew G. Knepley for (p = 0; p < Np; ++p, ++k) { 240697c42addSMatthew G. Knepley for (j = 0; j < cpdim; ++j) { 2407458eb97cSMatthew G. Knepley for (c = 0; c < Nc; ++c) elemMat[(offsetI + i)*cTotDim + offsetJ + j] += 1.0*qweights[p*qNc+c]; 240897c42addSMatthew G. Knepley } 240997c42addSMatthew G. Knepley } 241097c42addSMatthew G. Knepley } 241197c42addSMatthew G. Knepley } 241297c42addSMatthew G. Knepley } 2413d172c84bSMatthew G. Knepley offsetJ += cpdim; 2414d69c5d34SMatthew G. Knepley } 2415d172c84bSMatthew G. Knepley offsetI += fpdim; 2416549a8adaSMatthew G. Knepley ierr = PetscFree(points);CHKERRQ(ierr); 2417d69c5d34SMatthew G. Knepley } 24180f2d7e86SMatthew G. Knepley if (mesh->printFEM > 1) {ierr = DMPrintCellMatrix(0, name, rTotDim, cTotDim, elemMat);CHKERRQ(ierr);} 24197f5b169aSMatthew G. Knepley /* Preallocate matrix */ 24207f5b169aSMatthew G. Knepley { 2421c094ef40SMatthew G. Knepley Mat preallocator; 2422c094ef40SMatthew G. Knepley PetscScalar *vals; 2423c094ef40SMatthew G. Knepley PetscInt *cellCIndices, *cellFIndices; 2424c094ef40SMatthew G. Knepley PetscInt locRows, locCols, cell; 24257f5b169aSMatthew G. Knepley 2426c094ef40SMatthew G. Knepley ierr = MatGetLocalSize(In, &locRows, &locCols);CHKERRQ(ierr); 2427c094ef40SMatthew G. Knepley ierr = MatCreate(PetscObjectComm((PetscObject) In), &preallocator);CHKERRQ(ierr); 2428c094ef40SMatthew G. Knepley ierr = MatSetType(preallocator, MATPREALLOCATOR);CHKERRQ(ierr); 2429c094ef40SMatthew G. Knepley ierr = MatSetSizes(preallocator, locRows, locCols, PETSC_DETERMINE, PETSC_DETERMINE);CHKERRQ(ierr); 2430c094ef40SMatthew G. Knepley ierr = MatSetUp(preallocator);CHKERRQ(ierr); 2431c094ef40SMatthew G. Knepley ierr = PetscCalloc3(rTotDim*cTotDim, &vals,cTotDim,&cellCIndices,rTotDim,&cellFIndices);CHKERRQ(ierr); 24327f5b169aSMatthew G. Knepley for (cell = cStart; cell < cEnd; ++cell) { 24337f5b169aSMatthew G. Knepley ierr = DMPlexMatGetClosureIndicesRefined(dmf, fsection, fglobalSection, dmc, csection, cglobalSection, cell, cellCIndices, cellFIndices);CHKERRQ(ierr); 2434c094ef40SMatthew G. Knepley ierr = MatSetValues(preallocator, rTotDim, cellFIndices, cTotDim, cellCIndices, vals, INSERT_VALUES);CHKERRQ(ierr); 24357f5b169aSMatthew G. Knepley } 2436c094ef40SMatthew G. Knepley ierr = PetscFree3(vals,cellCIndices,cellFIndices);CHKERRQ(ierr); 2437c094ef40SMatthew G. Knepley ierr = MatAssemblyBegin(preallocator, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2438c094ef40SMatthew G. Knepley ierr = MatAssemblyEnd(preallocator, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2439c094ef40SMatthew G. Knepley ierr = MatPreallocatorPreallocate(preallocator, PETSC_TRUE, In);CHKERRQ(ierr); 2440c094ef40SMatthew G. Knepley ierr = MatDestroy(&preallocator);CHKERRQ(ierr); 24417f5b169aSMatthew G. Knepley } 24427f5b169aSMatthew G. Knepley /* Fill matrix */ 24437f5b169aSMatthew G. Knepley ierr = MatZeroEntries(In);CHKERRQ(ierr); 2444d69c5d34SMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 2445934789fcSMatthew G. Knepley ierr = DMPlexMatSetClosureRefined(dmf, fsection, fglobalSection, dmc, csection, cglobalSection, In, c, elemMat, INSERT_VALUES);CHKERRQ(ierr); 2446d69c5d34SMatthew G. Knepley } 2447549a8adaSMatthew G. Knepley for (f = 0; f < Nf; ++f) {ierr = PetscFEDestroy(&feRef[f]);CHKERRQ(ierr);} 244897c42addSMatthew G. Knepley ierr = PetscFree2(feRef,fvRef);CHKERRQ(ierr); 2449549a8adaSMatthew G. Knepley ierr = PetscFree(elemMat);CHKERRQ(ierr); 2450934789fcSMatthew G. Knepley ierr = MatAssemblyBegin(In, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2451934789fcSMatthew G. Knepley ierr = MatAssemblyEnd(In, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2452d69c5d34SMatthew G. Knepley if (mesh->printFEM) { 2453825f8a23SLisandro Dalcin ierr = PetscPrintf(PetscObjectComm((PetscObject)In), "%s:\n", name);CHKERRQ(ierr); 2454934789fcSMatthew G. Knepley ierr = MatChop(In, 1.0e-10);CHKERRQ(ierr); 2455825f8a23SLisandro Dalcin ierr = MatView(In, NULL);CHKERRQ(ierr); 2456d69c5d34SMatthew G. Knepley } 2457d69c5d34SMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_InterpolatorFEM,dmc,dmf,0,0);CHKERRQ(ierr); 2458d69c5d34SMatthew G. Knepley PetscFunctionReturn(0); 2459d69c5d34SMatthew G. Knepley } 24606c73c22cSMatthew G. Knepley 2461bd041c0cSMatthew G. Knepley PetscErrorCode DMPlexComputeMassMatrixNested(DM dmc, DM dmf, Mat mass, void *user) 2462bd041c0cSMatthew G. Knepley { 2463bd041c0cSMatthew G. Knepley SETERRQ(PetscObjectComm((PetscObject) dmc), PETSC_ERR_SUP, "Laziness"); 2464bd041c0cSMatthew G. Knepley } 2465bd041c0cSMatthew G. Knepley 246668132eb9SMatthew G. Knepley /*@ 246768132eb9SMatthew G. Knepley DMPlexComputeInterpolatorGeneral - Form the local portion of the interpolation matrix I from the coarse DM to a non-nested fine DM. 246868132eb9SMatthew G. Knepley 246968132eb9SMatthew G. Knepley Input Parameters: 247068132eb9SMatthew G. Knepley + dmf - The fine mesh 247168132eb9SMatthew G. Knepley . dmc - The coarse mesh 247268132eb9SMatthew G. Knepley - user - The user context 247368132eb9SMatthew G. Knepley 247468132eb9SMatthew G. Knepley Output Parameter: 247568132eb9SMatthew G. Knepley . In - The interpolation matrix 247668132eb9SMatthew G. Knepley 247768132eb9SMatthew G. Knepley Level: developer 247868132eb9SMatthew G. Knepley 247968132eb9SMatthew G. Knepley .seealso: DMPlexComputeInterpolatorNested(), DMPlexComputeJacobianFEM() 248068132eb9SMatthew G. Knepley @*/ 248168132eb9SMatthew G. Knepley PetscErrorCode DMPlexComputeInterpolatorGeneral(DM dmc, DM dmf, Mat In, void *user) 24824ef9d792SMatthew G. Knepley { 248364e98e1dSMatthew G. Knepley DM_Plex *mesh = (DM_Plex *) dmf->data; 248464e98e1dSMatthew G. Knepley const char *name = "Interpolator"; 24854ef9d792SMatthew G. Knepley PetscDS prob; 24864ef9d792SMatthew G. Knepley PetscSection fsection, csection, globalFSection, globalCSection; 2487e8f14785SLisandro Dalcin PetscHSetIJ ht; 24884ef9d792SMatthew G. Knepley PetscLayout rLayout; 24894ef9d792SMatthew G. Knepley PetscInt *dnz, *onz; 24904ef9d792SMatthew G. Knepley PetscInt locRows, rStart, rEnd; 24914ef9d792SMatthew G. Knepley PetscReal *x, *v0, *J, *invJ, detJ; 24924ef9d792SMatthew G. Knepley PetscReal *v0c, *Jc, *invJc, detJc; 24934ef9d792SMatthew G. Knepley PetscScalar *elemMat; 24944ef9d792SMatthew G. Knepley PetscInt dim, Nf, field, totDim, cStart, cEnd, cell, ccell; 24954ef9d792SMatthew G. Knepley PetscErrorCode ierr; 24964ef9d792SMatthew G. Knepley 24974ef9d792SMatthew G. Knepley PetscFunctionBegin; 249877711781SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_InterpolatorFEM,dmc,dmf,0,0);CHKERRQ(ierr); 24994ef9d792SMatthew G. Knepley ierr = DMGetCoordinateDim(dmc, &dim);CHKERRQ(ierr); 25004ef9d792SMatthew G. Knepley ierr = DMGetDS(dmc, &prob);CHKERRQ(ierr); 25014bee2e38SMatthew G. Knepley ierr = PetscDSGetWorkspace(prob, &x, NULL, NULL, NULL, NULL);CHKERRQ(ierr); 25024ef9d792SMatthew G. Knepley ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr); 25034ef9d792SMatthew G. Knepley ierr = PetscMalloc3(dim,&v0,dim*dim,&J,dim*dim,&invJ);CHKERRQ(ierr); 25044ef9d792SMatthew G. Knepley ierr = PetscMalloc3(dim,&v0c,dim*dim,&Jc,dim*dim,&invJc);CHKERRQ(ierr); 250592fd8e1eSJed Brown ierr = DMGetLocalSection(dmf, &fsection);CHKERRQ(ierr); 2506e87a4003SBarry Smith ierr = DMGetGlobalSection(dmf, &globalFSection);CHKERRQ(ierr); 250792fd8e1eSJed Brown ierr = DMGetLocalSection(dmc, &csection);CHKERRQ(ierr); 2508e87a4003SBarry Smith ierr = DMGetGlobalSection(dmc, &globalCSection);CHKERRQ(ierr); 25094ef9d792SMatthew G. Knepley ierr = DMPlexGetHeightStratum(dmf, 0, &cStart, &cEnd);CHKERRQ(ierr); 25104ef9d792SMatthew G. Knepley ierr = PetscDSGetTotalDimension(prob, &totDim);CHKERRQ(ierr); 25119c3cf19fSMatthew G. Knepley ierr = PetscMalloc1(totDim, &elemMat);CHKERRQ(ierr); 25124ef9d792SMatthew G. Knepley 25134ef9d792SMatthew G. Knepley ierr = MatGetLocalSize(In, &locRows, NULL);CHKERRQ(ierr); 25144ef9d792SMatthew G. Knepley ierr = PetscLayoutCreate(PetscObjectComm((PetscObject) In), &rLayout);CHKERRQ(ierr); 25154ef9d792SMatthew G. Knepley ierr = PetscLayoutSetLocalSize(rLayout, locRows);CHKERRQ(ierr); 25164ef9d792SMatthew G. Knepley ierr = PetscLayoutSetBlockSize(rLayout, 1);CHKERRQ(ierr); 25174ef9d792SMatthew G. Knepley ierr = PetscLayoutSetUp(rLayout);CHKERRQ(ierr); 25184ef9d792SMatthew G. Knepley ierr = PetscLayoutGetRange(rLayout, &rStart, &rEnd);CHKERRQ(ierr); 25194ef9d792SMatthew G. Knepley ierr = PetscLayoutDestroy(&rLayout);CHKERRQ(ierr); 25204ef9d792SMatthew G. Knepley ierr = PetscCalloc2(locRows,&dnz,locRows,&onz);CHKERRQ(ierr); 2521e8f14785SLisandro Dalcin ierr = PetscHSetIJCreate(&ht);CHKERRQ(ierr); 25224ef9d792SMatthew G. Knepley for (field = 0; field < Nf; ++field) { 25234ef9d792SMatthew G. Knepley PetscObject obj; 25244ef9d792SMatthew G. Knepley PetscClassId id; 2525c0d7054bSMatthew G. Knepley PetscDualSpace Q = NULL; 25264ef9d792SMatthew G. Knepley PetscQuadrature f; 252717f047d8SMatthew G. Knepley const PetscReal *qpoints; 252817f047d8SMatthew G. Knepley PetscInt Nc, Np, fpdim, i, d; 25294ef9d792SMatthew G. Knepley 25304ef9d792SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, &obj);CHKERRQ(ierr); 25314ef9d792SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 25324ef9d792SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 25334ef9d792SMatthew G. Knepley PetscFE fe = (PetscFE) obj; 25344ef9d792SMatthew G. Knepley 25354ef9d792SMatthew G. Knepley ierr = PetscFEGetDualSpace(fe, &Q);CHKERRQ(ierr); 25364ef9d792SMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 25374ef9d792SMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 25384ef9d792SMatthew G. Knepley PetscFV fv = (PetscFV) obj; 25394ef9d792SMatthew G. Knepley 25404ef9d792SMatthew G. Knepley ierr = PetscFVGetDualSpace(fv, &Q);CHKERRQ(ierr); 25414ef9d792SMatthew G. Knepley Nc = 1; 25424ef9d792SMatthew G. Knepley } 25434ef9d792SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(Q, &fpdim);CHKERRQ(ierr); 25444ef9d792SMatthew G. Knepley /* For each fine grid cell */ 25454ef9d792SMatthew G. Knepley for (cell = cStart; cell < cEnd; ++cell) { 25464ef9d792SMatthew G. Knepley PetscInt *findices, *cindices; 25474ef9d792SMatthew G. Knepley PetscInt numFIndices, numCIndices; 25484ef9d792SMatthew G. Knepley 25496ecaa68aSToby Isaac ierr = DMPlexGetClosureIndices(dmf, fsection, globalFSection, cell, &numFIndices, &findices, NULL);CHKERRQ(ierr); 25504ef9d792SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dmf, cell, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr); 2551ff1e0c32SBarry Smith if (numFIndices != fpdim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of fine indices %D != %D dual basis vecs", numFIndices, fpdim); 25524ef9d792SMatthew G. Knepley for (i = 0; i < fpdim; ++i) { 25534ef9d792SMatthew G. Knepley Vec pointVec; 25544ef9d792SMatthew G. Knepley PetscScalar *pV; 25553a93e3b7SToby Isaac PetscSF coarseCellSF = NULL; 25563a93e3b7SToby Isaac const PetscSFNode *coarseCells; 25579c3cf19fSMatthew G. Knepley PetscInt numCoarseCells, q, c; 25584ef9d792SMatthew G. Knepley 25594ef9d792SMatthew G. Knepley /* Get points from the dual basis functional quadrature */ 25604ef9d792SMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(Q, i, &f);CHKERRQ(ierr); 25619c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(f, NULL, NULL, &Np, &qpoints, NULL);CHKERRQ(ierr); 25624ef9d792SMatthew G. Knepley ierr = VecCreateSeq(PETSC_COMM_SELF, Np*dim, &pointVec);CHKERRQ(ierr); 25634ef9d792SMatthew G. Knepley ierr = VecSetBlockSize(pointVec, dim);CHKERRQ(ierr); 25644ef9d792SMatthew G. Knepley ierr = VecGetArray(pointVec, &pV);CHKERRQ(ierr); 25654ef9d792SMatthew G. Knepley for (q = 0; q < Np; ++q) { 2566c330f8ffSToby Isaac const PetscReal xi0[3] = {-1., -1., -1.}; 2567c330f8ffSToby Isaac 25684ef9d792SMatthew G. Knepley /* Transform point to real space */ 2569c330f8ffSToby Isaac CoordinatesRefToReal(dim, dim, xi0, v0, J, &qpoints[q*dim], x); 25704ef9d792SMatthew G. Knepley for (d = 0; d < dim; ++d) pV[q*dim+d] = x[d]; 25714ef9d792SMatthew G. Knepley } 25724ef9d792SMatthew G. Knepley ierr = VecRestoreArray(pointVec, &pV);CHKERRQ(ierr); 25734ef9d792SMatthew G. Knepley /* Get set of coarse cells that overlap points (would like to group points by coarse cell) */ 25741555c271SMatthew G. Knepley /* OPT: Pack all quad points from fine cell */ 257562a38674SMatthew G. Knepley ierr = DMLocatePoints(dmc, pointVec, DM_POINTLOCATION_NEAREST, &coarseCellSF);CHKERRQ(ierr); 25763a93e3b7SToby Isaac ierr = PetscSFViewFromOptions(coarseCellSF, NULL, "-interp_sf_view");CHKERRQ(ierr); 25774ef9d792SMatthew G. Knepley /* Update preallocation info */ 25783a93e3b7SToby Isaac ierr = PetscSFGetGraph(coarseCellSF, NULL, &numCoarseCells, NULL, &coarseCells);CHKERRQ(ierr); 25793a93e3b7SToby Isaac if (numCoarseCells != Np) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Not all closure points located"); 25809c3cf19fSMatthew G. Knepley { 2581e8f14785SLisandro Dalcin PetscHashIJKey key; 2582e8f14785SLisandro Dalcin PetscBool missing; 25834ef9d792SMatthew G. Knepley 2584e8f14785SLisandro Dalcin key.i = findices[i]; 2585e8f14785SLisandro Dalcin if (key.i >= 0) { 25864ef9d792SMatthew G. Knepley /* Get indices for coarse elements */ 25874ef9d792SMatthew G. Knepley for (ccell = 0; ccell < numCoarseCells; ++ccell) { 25883a93e3b7SToby Isaac ierr = DMPlexGetClosureIndices(dmc, csection, globalCSection, coarseCells[ccell].index, &numCIndices, &cindices, NULL);CHKERRQ(ierr); 25894ef9d792SMatthew G. Knepley for (c = 0; c < numCIndices; ++c) { 2590e8f14785SLisandro Dalcin key.j = cindices[c]; 2591e8f14785SLisandro Dalcin if (key.j < 0) continue; 2592e8f14785SLisandro Dalcin ierr = PetscHSetIJQueryAdd(ht, key, &missing);CHKERRQ(ierr); 25934ef9d792SMatthew G. Knepley if (missing) { 2594e8f14785SLisandro Dalcin if ((key.j >= rStart) && (key.j < rEnd)) ++dnz[key.i-rStart]; 2595e8f14785SLisandro Dalcin else ++onz[key.i-rStart]; 25964ef9d792SMatthew G. Knepley } 25974ef9d792SMatthew G. Knepley } 25983a93e3b7SToby Isaac ierr = DMPlexRestoreClosureIndices(dmc, csection, globalCSection, coarseCells[ccell].index, &numCIndices, &cindices, NULL);CHKERRQ(ierr); 25994ef9d792SMatthew G. Knepley } 26004ef9d792SMatthew G. Knepley } 26018c543595SMatthew G. Knepley } 26023a93e3b7SToby Isaac ierr = PetscSFDestroy(&coarseCellSF);CHKERRQ(ierr); 26034ef9d792SMatthew G. Knepley ierr = VecDestroy(&pointVec);CHKERRQ(ierr); 26044ef9d792SMatthew G. Knepley } 260546bdb399SToby Isaac ierr = DMPlexRestoreClosureIndices(dmf, fsection, globalFSection, cell, &numFIndices, &findices, NULL);CHKERRQ(ierr); 26064ef9d792SMatthew G. Knepley } 26074ef9d792SMatthew G. Knepley } 2608e8f14785SLisandro Dalcin ierr = PetscHSetIJDestroy(&ht);CHKERRQ(ierr); 26094ef9d792SMatthew G. Knepley ierr = MatXAIJSetPreallocation(In, 1, dnz, onz, NULL, NULL);CHKERRQ(ierr); 26104ef9d792SMatthew G. Knepley ierr = MatSetOption(In, MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 26114ef9d792SMatthew G. Knepley ierr = PetscFree2(dnz,onz);CHKERRQ(ierr); 26124ef9d792SMatthew G. Knepley for (field = 0; field < Nf; ++field) { 26134ef9d792SMatthew G. Knepley PetscObject obj; 26144ef9d792SMatthew G. Knepley PetscClassId id; 2615c0d7054bSMatthew G. Knepley PetscDualSpace Q = NULL; 2616ef0bb6c7SMatthew G. Knepley PetscTabulation T = NULL; 26174ef9d792SMatthew G. Knepley PetscQuadrature f; 26184ef9d792SMatthew G. Knepley const PetscReal *qpoints, *qweights; 26199c3cf19fSMatthew G. Knepley PetscInt Nc, qNc, Np, fpdim, i, d; 26204ef9d792SMatthew G. Knepley 26214ef9d792SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, &obj);CHKERRQ(ierr); 26224ef9d792SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 26234ef9d792SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 26244ef9d792SMatthew G. Knepley PetscFE fe = (PetscFE) obj; 26254ef9d792SMatthew G. Knepley 26264ef9d792SMatthew G. Knepley ierr = PetscFEGetDualSpace(fe, &Q);CHKERRQ(ierr); 26274ef9d792SMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 2628ef0bb6c7SMatthew G. Knepley ierr = PetscFECreateTabulation(fe, 1, 1, x, 0, &T);CHKERRQ(ierr); 26294ef9d792SMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 26304ef9d792SMatthew G. Knepley PetscFV fv = (PetscFV) obj; 26314ef9d792SMatthew G. Knepley 26324ef9d792SMatthew G. Knepley ierr = PetscFVGetDualSpace(fv, &Q);CHKERRQ(ierr); 26334ef9d792SMatthew G. Knepley Nc = 1; 2634ff1e0c32SBarry Smith } else SETERRQ1(PetscObjectComm((PetscObject)dmc),PETSC_ERR_ARG_WRONG,"Unknown discretization type for field %D",field); 26354ef9d792SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(Q, &fpdim);CHKERRQ(ierr); 26364ef9d792SMatthew G. Knepley /* For each fine grid cell */ 26374ef9d792SMatthew G. Knepley for (cell = cStart; cell < cEnd; ++cell) { 26384ef9d792SMatthew G. Knepley PetscInt *findices, *cindices; 26394ef9d792SMatthew G. Knepley PetscInt numFIndices, numCIndices; 26404ef9d792SMatthew G. Knepley 26416ecaa68aSToby Isaac ierr = DMPlexGetClosureIndices(dmf, fsection, globalFSection, cell, &numFIndices, &findices, NULL);CHKERRQ(ierr); 26424ef9d792SMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dmf, cell, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr); 2643ff1e0c32SBarry Smith if (numFIndices != fpdim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of fine indices %D != %D dual basis vecs", numFIndices, fpdim); 26444ef9d792SMatthew G. Knepley for (i = 0; i < fpdim; ++i) { 26454ef9d792SMatthew G. Knepley Vec pointVec; 26464ef9d792SMatthew G. Knepley PetscScalar *pV; 264712111d7cSToby Isaac PetscSF coarseCellSF = NULL; 26483a93e3b7SToby Isaac const PetscSFNode *coarseCells; 264917f047d8SMatthew G. Knepley PetscInt numCoarseCells, cpdim, q, c, j; 26504ef9d792SMatthew G. Knepley 26514ef9d792SMatthew G. Knepley /* Get points from the dual basis functional quadrature */ 26524ef9d792SMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(Q, i, &f);CHKERRQ(ierr); 26539c3cf19fSMatthew G. Knepley ierr = PetscQuadratureGetData(f, NULL, &qNc, &Np, &qpoints, &qweights);CHKERRQ(ierr); 26549c3cf19fSMatthew G. Knepley if (qNc != Nc) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of components in quadrature %D does not match coarse field %D", qNc, Nc); 26554ef9d792SMatthew G. Knepley ierr = VecCreateSeq(PETSC_COMM_SELF, Np*dim, &pointVec);CHKERRQ(ierr); 26564ef9d792SMatthew G. Knepley ierr = VecSetBlockSize(pointVec, dim);CHKERRQ(ierr); 26574ef9d792SMatthew G. Knepley ierr = VecGetArray(pointVec, &pV);CHKERRQ(ierr); 26584ef9d792SMatthew G. Knepley for (q = 0; q < Np; ++q) { 2659c330f8ffSToby Isaac const PetscReal xi0[3] = {-1., -1., -1.}; 2660c330f8ffSToby Isaac 26614ef9d792SMatthew G. Knepley /* Transform point to real space */ 2662c330f8ffSToby Isaac CoordinatesRefToReal(dim, dim, xi0, v0, J, &qpoints[q*dim], x); 26634ef9d792SMatthew G. Knepley for (d = 0; d < dim; ++d) pV[q*dim+d] = x[d]; 26644ef9d792SMatthew G. Knepley } 26654ef9d792SMatthew G. Knepley ierr = VecRestoreArray(pointVec, &pV);CHKERRQ(ierr); 26664ef9d792SMatthew G. Knepley /* Get set of coarse cells that overlap points (would like to group points by coarse cell) */ 26671555c271SMatthew G. Knepley /* OPT: Read this out from preallocation information */ 266862a38674SMatthew G. Knepley ierr = DMLocatePoints(dmc, pointVec, DM_POINTLOCATION_NEAREST, &coarseCellSF);CHKERRQ(ierr); 26694ef9d792SMatthew G. Knepley /* Update preallocation info */ 26703a93e3b7SToby Isaac ierr = PetscSFGetGraph(coarseCellSF, NULL, &numCoarseCells, NULL, &coarseCells);CHKERRQ(ierr); 26713a93e3b7SToby Isaac if (numCoarseCells != Np) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Not all closure points located"); 26724ef9d792SMatthew G. Knepley ierr = VecGetArray(pointVec, &pV);CHKERRQ(ierr); 26734ef9d792SMatthew G. Knepley for (ccell = 0; ccell < numCoarseCells; ++ccell) { 2674826eb36dSMatthew G. Knepley PetscReal pVReal[3]; 2675c330f8ffSToby Isaac const PetscReal xi0[3] = {-1., -1., -1.}; 2676826eb36dSMatthew G. Knepley 26773a93e3b7SToby Isaac ierr = DMPlexGetClosureIndices(dmc, csection, globalCSection, coarseCells[ccell].index, &numCIndices, &cindices, NULL);CHKERRQ(ierr); 26784ef9d792SMatthew G. Knepley /* Transform points from real space to coarse reference space */ 26793a93e3b7SToby Isaac ierr = DMPlexComputeCellGeometryFEM(dmc, coarseCells[ccell].index, NULL, v0c, Jc, invJc, &detJc);CHKERRQ(ierr); 2680e2d86523SMatthew G. Knepley for (d = 0; d < dim; ++d) pVReal[d] = PetscRealPart(pV[ccell*dim+d]); 2681c330f8ffSToby Isaac CoordinatesRealToRef(dim, dim, xi0, v0c, invJc, pVReal, x); 26824ef9d792SMatthew G. Knepley 26834ef9d792SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 26844ef9d792SMatthew G. Knepley PetscFE fe = (PetscFE) obj; 26854ef9d792SMatthew G. Knepley 26864ef9d792SMatthew G. Knepley /* Evaluate coarse basis on contained point */ 26874ef9d792SMatthew G. Knepley ierr = PetscFEGetDimension(fe, &cpdim);CHKERRQ(ierr); 2688ef0bb6c7SMatthew G. Knepley ierr = PetscFEComputeTabulation(fe, 1, x, 0, T);CHKERRQ(ierr); 2689580bdb30SBarry Smith ierr = PetscArrayzero(elemMat, cpdim);CHKERRQ(ierr); 26904ef9d792SMatthew G. Knepley /* Get elemMat entries by multiplying by weight */ 26914ef9d792SMatthew G. Knepley for (j = 0; j < cpdim; ++j) { 2692ef0bb6c7SMatthew G. Knepley for (c = 0; c < Nc; ++c) elemMat[j] += T->T[0][j*Nc + c]*qweights[ccell*qNc + c]; 26934ef9d792SMatthew G. Knepley } 26944ef9d792SMatthew G. Knepley } else { 26954ef9d792SMatthew G. Knepley cpdim = 1; 26964ef9d792SMatthew G. Knepley for (j = 0; j < cpdim; ++j) { 26979c3cf19fSMatthew G. Knepley for (c = 0; c < Nc; ++c) elemMat[j] += 1.0*qweights[ccell*qNc + c]; 26984ef9d792SMatthew G. Knepley } 26994ef9d792SMatthew G. Knepley } 27004ef9d792SMatthew G. Knepley /* Update interpolator */ 27019c3cf19fSMatthew G. Knepley if (mesh->printFEM > 1) {ierr = DMPrintCellMatrix(cell, name, 1, numCIndices, elemMat);CHKERRQ(ierr);} 27029c3cf19fSMatthew G. Knepley if (numCIndices != cpdim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of element matrix columns %D != %D", numCIndices, cpdim); 27039c3cf19fSMatthew G. Knepley ierr = MatSetValues(In, 1, &findices[i], numCIndices, cindices, elemMat, INSERT_VALUES);CHKERRQ(ierr); 27043a93e3b7SToby Isaac ierr = DMPlexRestoreClosureIndices(dmc, csection, globalCSection, coarseCells[ccell].index, &numCIndices, &cindices, NULL);CHKERRQ(ierr); 27054ef9d792SMatthew G. Knepley } 27064ef9d792SMatthew G. Knepley ierr = VecRestoreArray(pointVec, &pV);CHKERRQ(ierr); 27073a93e3b7SToby Isaac ierr = PetscSFDestroy(&coarseCellSF);CHKERRQ(ierr); 27084ef9d792SMatthew G. Knepley ierr = VecDestroy(&pointVec);CHKERRQ(ierr); 27094ef9d792SMatthew G. Knepley } 271046bdb399SToby Isaac ierr = DMPlexRestoreClosureIndices(dmf, fsection, globalFSection, cell, &numFIndices, &findices, NULL);CHKERRQ(ierr); 27114ef9d792SMatthew G. Knepley } 2712ef0bb6c7SMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscTabulationDestroy(&T);CHKERRQ(ierr);} 27134ef9d792SMatthew G. Knepley } 27144ef9d792SMatthew G. Knepley ierr = PetscFree3(v0,J,invJ);CHKERRQ(ierr); 27154ef9d792SMatthew G. Knepley ierr = PetscFree3(v0c,Jc,invJc);CHKERRQ(ierr); 27164ef9d792SMatthew G. Knepley ierr = PetscFree(elemMat);CHKERRQ(ierr); 27174ef9d792SMatthew G. Knepley ierr = MatAssemblyBegin(In, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 27184ef9d792SMatthew G. Knepley ierr = MatAssemblyEnd(In, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 271977711781SMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_InterpolatorFEM,dmc,dmf,0,0);CHKERRQ(ierr); 27204ef9d792SMatthew G. Knepley PetscFunctionReturn(0); 27214ef9d792SMatthew G. Knepley } 27224ef9d792SMatthew G. Knepley 272346fa42a0SMatthew G. Knepley /*@ 2724bd041c0cSMatthew G. Knepley DMPlexComputeMassMatrixGeneral - Form the local portion of the mass matrix M from the coarse DM to a non-nested fine DM. 2725bd041c0cSMatthew G. Knepley 2726bd041c0cSMatthew G. Knepley Input Parameters: 2727bd041c0cSMatthew G. Knepley + dmf - The fine mesh 2728bd041c0cSMatthew G. Knepley . dmc - The coarse mesh 2729bd041c0cSMatthew G. Knepley - user - The user context 2730bd041c0cSMatthew G. Knepley 2731bd041c0cSMatthew G. Knepley Output Parameter: 2732bd041c0cSMatthew G. Knepley . mass - The mass matrix 2733bd041c0cSMatthew G. Knepley 2734bd041c0cSMatthew G. Knepley Level: developer 2735bd041c0cSMatthew G. Knepley 2736bd041c0cSMatthew G. Knepley .seealso: DMPlexComputeMassMatrixNested(), DMPlexComputeInterpolatorNested(), DMPlexComputeInterpolatorGeneral(), DMPlexComputeJacobianFEM() 2737bd041c0cSMatthew G. Knepley @*/ 2738bd041c0cSMatthew G. Knepley PetscErrorCode DMPlexComputeMassMatrixGeneral(DM dmc, DM dmf, Mat mass, void *user) 2739bd041c0cSMatthew G. Knepley { 2740bd041c0cSMatthew G. Knepley DM_Plex *mesh = (DM_Plex *) dmf->data; 2741bd041c0cSMatthew G. Knepley const char *name = "Mass Matrix"; 2742bd041c0cSMatthew G. Knepley PetscDS prob; 2743bd041c0cSMatthew G. Knepley PetscSection fsection, csection, globalFSection, globalCSection; 2744e8f14785SLisandro Dalcin PetscHSetIJ ht; 2745bd041c0cSMatthew G. Knepley PetscLayout rLayout; 2746bd041c0cSMatthew G. Knepley PetscInt *dnz, *onz; 2747bd041c0cSMatthew G. Knepley PetscInt locRows, rStart, rEnd; 2748bd041c0cSMatthew G. Knepley PetscReal *x, *v0, *J, *invJ, detJ; 2749bd041c0cSMatthew G. Knepley PetscReal *v0c, *Jc, *invJc, detJc; 2750bd041c0cSMatthew G. Knepley PetscScalar *elemMat; 2751bd041c0cSMatthew G. Knepley PetscInt dim, Nf, field, totDim, cStart, cEnd, cell, ccell; 2752bd041c0cSMatthew G. Knepley PetscErrorCode ierr; 2753bd041c0cSMatthew G. Knepley 2754bd041c0cSMatthew G. Knepley PetscFunctionBegin; 2755bd041c0cSMatthew G. Knepley ierr = DMGetCoordinateDim(dmc, &dim);CHKERRQ(ierr); 2756bd041c0cSMatthew G. Knepley ierr = DMGetDS(dmc, &prob);CHKERRQ(ierr); 27574bee2e38SMatthew G. Knepley ierr = PetscDSGetWorkspace(prob, &x, NULL, NULL, NULL, NULL);CHKERRQ(ierr); 2758bd041c0cSMatthew G. Knepley ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr); 2759bd041c0cSMatthew G. Knepley ierr = PetscMalloc3(dim,&v0,dim*dim,&J,dim*dim,&invJ);CHKERRQ(ierr); 2760bd041c0cSMatthew G. Knepley ierr = PetscMalloc3(dim,&v0c,dim*dim,&Jc,dim*dim,&invJc);CHKERRQ(ierr); 276192fd8e1eSJed Brown ierr = DMGetLocalSection(dmf, &fsection);CHKERRQ(ierr); 2762e87a4003SBarry Smith ierr = DMGetGlobalSection(dmf, &globalFSection);CHKERRQ(ierr); 276392fd8e1eSJed Brown ierr = DMGetLocalSection(dmc, &csection);CHKERRQ(ierr); 2764e87a4003SBarry Smith ierr = DMGetGlobalSection(dmc, &globalCSection);CHKERRQ(ierr); 2765bd041c0cSMatthew G. Knepley ierr = DMPlexGetHeightStratum(dmf, 0, &cStart, &cEnd);CHKERRQ(ierr); 2766bd041c0cSMatthew G. Knepley ierr = PetscDSGetTotalDimension(prob, &totDim);CHKERRQ(ierr); 2767bd041c0cSMatthew G. Knepley ierr = PetscMalloc1(totDim, &elemMat);CHKERRQ(ierr); 2768bd041c0cSMatthew G. Knepley 2769bd041c0cSMatthew G. Knepley ierr = MatGetLocalSize(mass, &locRows, NULL);CHKERRQ(ierr); 2770bd041c0cSMatthew G. Knepley ierr = PetscLayoutCreate(PetscObjectComm((PetscObject) mass), &rLayout);CHKERRQ(ierr); 2771bd041c0cSMatthew G. Knepley ierr = PetscLayoutSetLocalSize(rLayout, locRows);CHKERRQ(ierr); 2772bd041c0cSMatthew G. Knepley ierr = PetscLayoutSetBlockSize(rLayout, 1);CHKERRQ(ierr); 2773bd041c0cSMatthew G. Knepley ierr = PetscLayoutSetUp(rLayout);CHKERRQ(ierr); 2774bd041c0cSMatthew G. Knepley ierr = PetscLayoutGetRange(rLayout, &rStart, &rEnd);CHKERRQ(ierr); 2775bd041c0cSMatthew G. Knepley ierr = PetscLayoutDestroy(&rLayout);CHKERRQ(ierr); 2776bd041c0cSMatthew G. Knepley ierr = PetscCalloc2(locRows,&dnz,locRows,&onz);CHKERRQ(ierr); 2777e8f14785SLisandro Dalcin ierr = PetscHSetIJCreate(&ht);CHKERRQ(ierr); 2778bd041c0cSMatthew G. Knepley for (field = 0; field < Nf; ++field) { 2779bd041c0cSMatthew G. Knepley PetscObject obj; 2780bd041c0cSMatthew G. Knepley PetscClassId id; 2781bd041c0cSMatthew G. Knepley PetscQuadrature quad; 2782bd041c0cSMatthew G. Knepley const PetscReal *qpoints; 2783bd041c0cSMatthew G. Knepley PetscInt Nq, Nc, i, d; 2784bd041c0cSMatthew G. Knepley 2785bd041c0cSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, &obj);CHKERRQ(ierr); 2786bd041c0cSMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 2787bd041c0cSMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscFEGetQuadrature((PetscFE) obj, &quad);CHKERRQ(ierr);} 2788bd041c0cSMatthew G. Knepley else {ierr = PetscFVGetQuadrature((PetscFV) obj, &quad);CHKERRQ(ierr);} 2789bd041c0cSMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, &Nc, &Nq, &qpoints, NULL);CHKERRQ(ierr); 2790bd041c0cSMatthew G. Knepley /* For each fine grid cell */ 2791bd041c0cSMatthew G. Knepley for (cell = cStart; cell < cEnd; ++cell) { 2792bd041c0cSMatthew G. Knepley Vec pointVec; 2793bd041c0cSMatthew G. Knepley PetscScalar *pV; 2794bd041c0cSMatthew G. Knepley PetscSF coarseCellSF = NULL; 2795bd041c0cSMatthew G. Knepley const PetscSFNode *coarseCells; 2796bd041c0cSMatthew G. Knepley PetscInt numCoarseCells, q, c; 2797bd041c0cSMatthew G. Knepley PetscInt *findices, *cindices; 2798bd041c0cSMatthew G. Knepley PetscInt numFIndices, numCIndices; 2799bd041c0cSMatthew G. Knepley 2800bd041c0cSMatthew G. Knepley ierr = DMPlexGetClosureIndices(dmf, fsection, globalFSection, cell, &numFIndices, &findices, NULL);CHKERRQ(ierr); 2801bd041c0cSMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dmf, cell, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr); 2802bd041c0cSMatthew G. Knepley /* Get points from the quadrature */ 2803bd041c0cSMatthew G. Knepley ierr = VecCreateSeq(PETSC_COMM_SELF, Nq*dim, &pointVec);CHKERRQ(ierr); 2804bd041c0cSMatthew G. Knepley ierr = VecSetBlockSize(pointVec, dim);CHKERRQ(ierr); 2805bd041c0cSMatthew G. Knepley ierr = VecGetArray(pointVec, &pV);CHKERRQ(ierr); 2806bd041c0cSMatthew G. Knepley for (q = 0; q < Nq; ++q) { 2807c330f8ffSToby Isaac const PetscReal xi0[3] = {-1., -1., -1.}; 2808c330f8ffSToby Isaac 2809bd041c0cSMatthew G. Knepley /* Transform point to real space */ 2810c330f8ffSToby Isaac CoordinatesRefToReal(dim, dim, xi0, v0, J, &qpoints[q*dim], x); 2811bd041c0cSMatthew G. Knepley for (d = 0; d < dim; ++d) pV[q*dim+d] = x[d]; 2812bd041c0cSMatthew G. Knepley } 2813bd041c0cSMatthew G. Knepley ierr = VecRestoreArray(pointVec, &pV);CHKERRQ(ierr); 2814bd041c0cSMatthew G. Knepley /* Get set of coarse cells that overlap points (would like to group points by coarse cell) */ 2815bd041c0cSMatthew G. Knepley ierr = DMLocatePoints(dmc, pointVec, DM_POINTLOCATION_NEAREST, &coarseCellSF);CHKERRQ(ierr); 2816bd041c0cSMatthew G. Knepley ierr = PetscSFViewFromOptions(coarseCellSF, NULL, "-interp_sf_view");CHKERRQ(ierr); 2817bd041c0cSMatthew G. Knepley /* Update preallocation info */ 2818bd041c0cSMatthew G. Knepley ierr = PetscSFGetGraph(coarseCellSF, NULL, &numCoarseCells, NULL, &coarseCells);CHKERRQ(ierr); 2819bd041c0cSMatthew G. Knepley if (numCoarseCells != Nq) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Not all closure points located"); 2820bd041c0cSMatthew G. Knepley { 2821e8f14785SLisandro Dalcin PetscHashIJKey key; 2822e8f14785SLisandro Dalcin PetscBool missing; 2823bd041c0cSMatthew G. Knepley 2824bd041c0cSMatthew G. Knepley for (i = 0; i < numFIndices; ++i) { 2825e8f14785SLisandro Dalcin key.i = findices[i]; 2826e8f14785SLisandro Dalcin if (key.i >= 0) { 2827bd041c0cSMatthew G. Knepley /* Get indices for coarse elements */ 2828bd041c0cSMatthew G. Knepley for (ccell = 0; ccell < numCoarseCells; ++ccell) { 2829bd041c0cSMatthew G. Knepley ierr = DMPlexGetClosureIndices(dmc, csection, globalCSection, coarseCells[ccell].index, &numCIndices, &cindices, NULL);CHKERRQ(ierr); 2830bd041c0cSMatthew G. Knepley for (c = 0; c < numCIndices; ++c) { 2831e8f14785SLisandro Dalcin key.j = cindices[c]; 2832e8f14785SLisandro Dalcin if (key.j < 0) continue; 2833e8f14785SLisandro Dalcin ierr = PetscHSetIJQueryAdd(ht, key, &missing);CHKERRQ(ierr); 2834bd041c0cSMatthew G. Knepley if (missing) { 2835e8f14785SLisandro Dalcin if ((key.j >= rStart) && (key.j < rEnd)) ++dnz[key.i-rStart]; 2836e8f14785SLisandro Dalcin else ++onz[key.i-rStart]; 2837bd041c0cSMatthew G. Knepley } 2838bd041c0cSMatthew G. Knepley } 2839bd041c0cSMatthew G. Knepley ierr = DMPlexRestoreClosureIndices(dmc, csection, globalCSection, coarseCells[ccell].index, &numCIndices, &cindices, NULL);CHKERRQ(ierr); 2840bd041c0cSMatthew G. Knepley } 2841bd041c0cSMatthew G. Knepley } 2842bd041c0cSMatthew G. Knepley } 2843bd041c0cSMatthew G. Knepley } 2844bd041c0cSMatthew G. Knepley ierr = PetscSFDestroy(&coarseCellSF);CHKERRQ(ierr); 2845bd041c0cSMatthew G. Knepley ierr = VecDestroy(&pointVec);CHKERRQ(ierr); 2846bd041c0cSMatthew G. Knepley ierr = DMPlexRestoreClosureIndices(dmf, fsection, globalFSection, cell, &numFIndices, &findices, NULL);CHKERRQ(ierr); 2847bd041c0cSMatthew G. Knepley } 2848bd041c0cSMatthew G. Knepley } 2849e8f14785SLisandro Dalcin ierr = PetscHSetIJDestroy(&ht);CHKERRQ(ierr); 2850bd041c0cSMatthew G. Knepley ierr = MatXAIJSetPreallocation(mass, 1, dnz, onz, NULL, NULL);CHKERRQ(ierr); 2851bd041c0cSMatthew G. Knepley ierr = MatSetOption(mass, MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 2852bd041c0cSMatthew G. Knepley ierr = PetscFree2(dnz,onz);CHKERRQ(ierr); 2853bd041c0cSMatthew G. Knepley for (field = 0; field < Nf; ++field) { 2854bd041c0cSMatthew G. Knepley PetscObject obj; 2855bd041c0cSMatthew G. Knepley PetscClassId id; 2856ef0bb6c7SMatthew G. Knepley PetscTabulation T, Tfine; 2857bd041c0cSMatthew G. Knepley PetscQuadrature quad; 2858bd041c0cSMatthew G. Knepley const PetscReal *qpoints, *qweights; 2859bd041c0cSMatthew G. Knepley PetscInt Nq, Nc, i, d; 2860bd041c0cSMatthew G. Knepley 2861bd041c0cSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, &obj);CHKERRQ(ierr); 2862bd041c0cSMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 2863ef0bb6c7SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 2864ef0bb6c7SMatthew G. Knepley ierr = PetscFEGetQuadrature((PetscFE) obj, &quad);CHKERRQ(ierr); 2865ef0bb6c7SMatthew G. Knepley ierr = PetscFEGetCellTabulation((PetscFE) obj, &Tfine);CHKERRQ(ierr); 2866ef0bb6c7SMatthew G. Knepley ierr = PetscFECreateTabulation((PetscFE) obj, 1, 1, x, 0, &T);CHKERRQ(ierr); 2867ef0bb6c7SMatthew G. Knepley } else { 2868ef0bb6c7SMatthew G. Knepley ierr = PetscFVGetQuadrature((PetscFV) obj, &quad);CHKERRQ(ierr); 2869ef0bb6c7SMatthew G. Knepley } 2870bd041c0cSMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, &Nc, &Nq, &qpoints, &qweights);CHKERRQ(ierr); 2871bd041c0cSMatthew G. Knepley /* For each fine grid cell */ 2872bd041c0cSMatthew G. Knepley for (cell = cStart; cell < cEnd; ++cell) { 2873bd041c0cSMatthew G. Knepley Vec pointVec; 2874bd041c0cSMatthew G. Knepley PetscScalar *pV; 2875bd041c0cSMatthew G. Knepley PetscSF coarseCellSF = NULL; 2876bd041c0cSMatthew G. Knepley const PetscSFNode *coarseCells; 2877bd041c0cSMatthew G. Knepley PetscInt numCoarseCells, cpdim, q, c, j; 2878bd041c0cSMatthew G. Knepley PetscInt *findices, *cindices; 2879bd041c0cSMatthew G. Knepley PetscInt numFIndices, numCIndices; 2880bd041c0cSMatthew G. Knepley 2881bd041c0cSMatthew G. Knepley ierr = DMPlexGetClosureIndices(dmf, fsection, globalFSection, cell, &numFIndices, &findices, NULL);CHKERRQ(ierr); 2882bd041c0cSMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dmf, cell, NULL, v0, J, invJ, &detJ);CHKERRQ(ierr); 2883bd041c0cSMatthew G. Knepley /* Get points from the quadrature */ 2884bd041c0cSMatthew G. Knepley ierr = VecCreateSeq(PETSC_COMM_SELF, Nq*dim, &pointVec);CHKERRQ(ierr); 2885bd041c0cSMatthew G. Knepley ierr = VecSetBlockSize(pointVec, dim);CHKERRQ(ierr); 2886bd041c0cSMatthew G. Knepley ierr = VecGetArray(pointVec, &pV);CHKERRQ(ierr); 2887bd041c0cSMatthew G. Knepley for (q = 0; q < Nq; ++q) { 2888c330f8ffSToby Isaac const PetscReal xi0[3] = {-1., -1., -1.}; 2889c330f8ffSToby Isaac 2890bd041c0cSMatthew G. Knepley /* Transform point to real space */ 2891c330f8ffSToby Isaac CoordinatesRefToReal(dim, dim, xi0, v0, J, &qpoints[q*dim], x); 2892bd041c0cSMatthew G. Knepley for (d = 0; d < dim; ++d) pV[q*dim+d] = x[d]; 2893bd041c0cSMatthew G. Knepley } 2894bd041c0cSMatthew G. Knepley ierr = VecRestoreArray(pointVec, &pV);CHKERRQ(ierr); 2895bd041c0cSMatthew G. Knepley /* Get set of coarse cells that overlap points (would like to group points by coarse cell) */ 2896bd041c0cSMatthew G. Knepley ierr = DMLocatePoints(dmc, pointVec, DM_POINTLOCATION_NEAREST, &coarseCellSF);CHKERRQ(ierr); 2897bd041c0cSMatthew G. Knepley /* Update matrix */ 2898bd041c0cSMatthew G. Knepley ierr = PetscSFGetGraph(coarseCellSF, NULL, &numCoarseCells, NULL, &coarseCells);CHKERRQ(ierr); 2899bd041c0cSMatthew G. Knepley if (numCoarseCells != Nq) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Not all closure points located"); 2900bd041c0cSMatthew G. Knepley ierr = VecGetArray(pointVec, &pV);CHKERRQ(ierr); 2901bd041c0cSMatthew G. Knepley for (ccell = 0; ccell < numCoarseCells; ++ccell) { 2902bd041c0cSMatthew G. Knepley PetscReal pVReal[3]; 2903c330f8ffSToby Isaac const PetscReal xi0[3] = {-1., -1., -1.}; 2904c330f8ffSToby Isaac 2905bd041c0cSMatthew G. Knepley 2906bd041c0cSMatthew G. Knepley ierr = DMPlexGetClosureIndices(dmc, csection, globalCSection, coarseCells[ccell].index, &numCIndices, &cindices, NULL);CHKERRQ(ierr); 2907bd041c0cSMatthew G. Knepley /* Transform points from real space to coarse reference space */ 2908bd041c0cSMatthew G. Knepley ierr = DMPlexComputeCellGeometryFEM(dmc, coarseCells[ccell].index, NULL, v0c, Jc, invJc, &detJc);CHKERRQ(ierr); 2909bd041c0cSMatthew G. Knepley for (d = 0; d < dim; ++d) pVReal[d] = PetscRealPart(pV[ccell*dim+d]); 2910c330f8ffSToby Isaac CoordinatesRealToRef(dim, dim, xi0, v0c, invJc, pVReal, x); 2911bd041c0cSMatthew G. Knepley 2912bd041c0cSMatthew G. Knepley if (id == PETSCFE_CLASSID) { 2913bd041c0cSMatthew G. Knepley PetscFE fe = (PetscFE) obj; 2914bd041c0cSMatthew G. Knepley 2915bd041c0cSMatthew G. Knepley /* Evaluate coarse basis on contained point */ 2916bd041c0cSMatthew G. Knepley ierr = PetscFEGetDimension(fe, &cpdim);CHKERRQ(ierr); 2917ef0bb6c7SMatthew G. Knepley ierr = PetscFEComputeTabulation(fe, 1, x, 0, T);CHKERRQ(ierr); 2918bd041c0cSMatthew G. Knepley /* Get elemMat entries by multiplying by weight */ 2919bd041c0cSMatthew G. Knepley for (i = 0; i < numFIndices; ++i) { 2920580bdb30SBarry Smith ierr = PetscArrayzero(elemMat, cpdim);CHKERRQ(ierr); 2921bd041c0cSMatthew G. Knepley for (j = 0; j < cpdim; ++j) { 2922ef0bb6c7SMatthew G. Knepley for (c = 0; c < Nc; ++c) elemMat[j] += T->T[0][j*Nc + c]*Tfine->T[0][(ccell*numFIndices + i)*Nc + c]*qweights[ccell*Nc + c]*detJ; 2923bd041c0cSMatthew G. Knepley } 2924bd041c0cSMatthew G. Knepley /* Update interpolator */ 2925bd041c0cSMatthew G. Knepley if (mesh->printFEM > 1) {ierr = DMPrintCellMatrix(cell, name, 1, numCIndices, elemMat);CHKERRQ(ierr);} 2926bd041c0cSMatthew G. Knepley if (numCIndices != cpdim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of element matrix columns %D != %D", numCIndices, cpdim); 2927bd041c0cSMatthew G. Knepley ierr = MatSetValues(mass, 1, &findices[i], numCIndices, cindices, elemMat, ADD_VALUES);CHKERRQ(ierr); 2928bd041c0cSMatthew G. Knepley } 2929bd041c0cSMatthew G. Knepley } else { 2930bd041c0cSMatthew G. Knepley cpdim = 1; 2931bd041c0cSMatthew G. Knepley for (i = 0; i < numFIndices; ++i) { 2932580bdb30SBarry Smith ierr = PetscArrayzero(elemMat, cpdim);CHKERRQ(ierr); 2933bd041c0cSMatthew G. Knepley for (j = 0; j < cpdim; ++j) { 2934bd041c0cSMatthew G. Knepley for (c = 0; c < Nc; ++c) elemMat[j] += 1.0*1.0*qweights[ccell*Nc + c]*detJ; 2935bd041c0cSMatthew G. Knepley } 2936bd041c0cSMatthew G. Knepley /* Update interpolator */ 2937bd041c0cSMatthew G. Knepley if (mesh->printFEM > 1) {ierr = DMPrintCellMatrix(cell, name, 1, numCIndices, elemMat);CHKERRQ(ierr);} 2938ff1e0c32SBarry Smith ierr = PetscPrintf(PETSC_COMM_SELF, "Nq: %D %D Nf: %D %D Nc: %D %D\n", ccell, Nq, i, numFIndices, j, numCIndices);CHKERRQ(ierr); 2939bd041c0cSMatthew G. Knepley if (numCIndices != cpdim) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of element matrix columns %D != %D", numCIndices, cpdim); 2940bd041c0cSMatthew G. Knepley ierr = MatSetValues(mass, 1, &findices[i], numCIndices, cindices, elemMat, ADD_VALUES);CHKERRQ(ierr); 2941bd041c0cSMatthew G. Knepley } 2942bd041c0cSMatthew G. Knepley } 2943bd041c0cSMatthew G. Knepley ierr = DMPlexRestoreClosureIndices(dmc, csection, globalCSection, coarseCells[ccell].index, &numCIndices, &cindices, NULL);CHKERRQ(ierr); 2944bd041c0cSMatthew G. Knepley } 2945bd041c0cSMatthew G. Knepley ierr = VecRestoreArray(pointVec, &pV);CHKERRQ(ierr); 2946bd041c0cSMatthew G. Knepley ierr = PetscSFDestroy(&coarseCellSF);CHKERRQ(ierr); 2947bd041c0cSMatthew G. Knepley ierr = VecDestroy(&pointVec);CHKERRQ(ierr); 2948bd041c0cSMatthew G. Knepley ierr = DMPlexRestoreClosureIndices(dmf, fsection, globalFSection, cell, &numFIndices, &findices, NULL);CHKERRQ(ierr); 2949bd041c0cSMatthew G. Knepley } 2950ef0bb6c7SMatthew G. Knepley if (id == PETSCFE_CLASSID) {ierr = PetscTabulationDestroy(&T);CHKERRQ(ierr);} 2951bd041c0cSMatthew G. Knepley } 2952bd041c0cSMatthew G. Knepley ierr = PetscFree3(v0,J,invJ);CHKERRQ(ierr); 2953bd041c0cSMatthew G. Knepley ierr = PetscFree3(v0c,Jc,invJc);CHKERRQ(ierr); 2954bd041c0cSMatthew G. Knepley ierr = PetscFree(elemMat);CHKERRQ(ierr); 2955bd041c0cSMatthew G. Knepley ierr = MatAssemblyBegin(mass, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2956bd041c0cSMatthew G. Knepley ierr = MatAssemblyEnd(mass, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2957bd041c0cSMatthew G. Knepley PetscFunctionReturn(0); 2958bd041c0cSMatthew G. Knepley } 2959bd041c0cSMatthew G. Knepley 2960bd041c0cSMatthew G. Knepley /*@ 296146fa42a0SMatthew G. Knepley DMPlexComputeInjectorFEM - Compute a mapping from coarse unknowns to fine unknowns 296246fa42a0SMatthew G. Knepley 296346fa42a0SMatthew G. Knepley Input Parameters: 296446fa42a0SMatthew G. Knepley + dmc - The coarse mesh 296546fa42a0SMatthew G. Knepley - dmf - The fine mesh 296646fa42a0SMatthew G. Knepley - user - The user context 296746fa42a0SMatthew G. Knepley 296846fa42a0SMatthew G. Knepley Output Parameter: 296946fa42a0SMatthew G. Knepley . sc - The mapping 297046fa42a0SMatthew G. Knepley 297146fa42a0SMatthew G. Knepley Level: developer 297246fa42a0SMatthew G. Knepley 297346fa42a0SMatthew G. Knepley .seealso: DMPlexComputeInterpolatorNested(), DMPlexComputeJacobianFEM() 297446fa42a0SMatthew G. Knepley @*/ 29757c927364SMatthew G. Knepley PetscErrorCode DMPlexComputeInjectorFEM(DM dmc, DM dmf, VecScatter *sc, void *user) 29767c927364SMatthew G. Knepley { 2977e9d4ef1bSMatthew G. Knepley PetscDS prob; 29787c927364SMatthew G. Knepley PetscFE *feRef; 297997c42addSMatthew G. Knepley PetscFV *fvRef; 29807c927364SMatthew G. Knepley Vec fv, cv; 29817c927364SMatthew G. Knepley IS fis, cis; 29827c927364SMatthew G. Knepley PetscSection fsection, fglobalSection, csection, cglobalSection; 29837c927364SMatthew G. Knepley PetscInt *cmap, *cellCIndices, *cellFIndices, *cindices, *findices; 2984485ad865SMatthew G. Knepley PetscInt cTotDim, fTotDim = 0, Nf, f, field, cStart, cEnd, c, dim, d, startC, endC, offsetC, offsetF, m; 29856f3d3cbcSMatthew G. Knepley PetscBool *needAvg; 29867c927364SMatthew G. Knepley PetscErrorCode ierr; 29877c927364SMatthew G. Knepley 29887c927364SMatthew G. Knepley PetscFunctionBegin; 298975a69067SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_InjectorFEM,dmc,dmf,0,0);CHKERRQ(ierr); 2990c73cfb54SMatthew G. Knepley ierr = DMGetDimension(dmf, &dim);CHKERRQ(ierr); 299192fd8e1eSJed Brown ierr = DMGetLocalSection(dmf, &fsection);CHKERRQ(ierr); 2992e87a4003SBarry Smith ierr = DMGetGlobalSection(dmf, &fglobalSection);CHKERRQ(ierr); 299392fd8e1eSJed Brown ierr = DMGetLocalSection(dmc, &csection);CHKERRQ(ierr); 2994e87a4003SBarry Smith ierr = DMGetGlobalSection(dmc, &cglobalSection);CHKERRQ(ierr); 29957c927364SMatthew G. Knepley ierr = PetscSectionGetNumFields(fsection, &Nf);CHKERRQ(ierr); 2996412e9a14SMatthew G. Knepley ierr = DMPlexGetSimplexOrBoxCells(dmc, 0, &cStart, &cEnd);CHKERRQ(ierr); 2997e9d4ef1bSMatthew G. Knepley ierr = DMGetDS(dmc, &prob);CHKERRQ(ierr); 29986f3d3cbcSMatthew G. Knepley ierr = PetscCalloc3(Nf,&feRef,Nf,&fvRef,Nf,&needAvg);CHKERRQ(ierr); 29997c927364SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 300097c42addSMatthew G. Knepley PetscObject obj; 300197c42addSMatthew G. Knepley PetscClassId id; 3002aa7890ccSMatthew G. Knepley PetscInt fNb = 0, Nc = 0; 30037c927364SMatthew G. Knepley 300497c42addSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr); 300597c42addSMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 300697c42addSMatthew G. Knepley if (id == PETSCFE_CLASSID) { 300797c42addSMatthew G. Knepley PetscFE fe = (PetscFE) obj; 30086f3d3cbcSMatthew G. Knepley PetscSpace sp; 30099b2fc754SMatthew G. Knepley PetscInt maxDegree; 301097c42addSMatthew G. Knepley 30117c927364SMatthew G. Knepley ierr = PetscFERefine(fe, &feRef[f]);CHKERRQ(ierr); 30127c927364SMatthew G. Knepley ierr = PetscFEGetDimension(feRef[f], &fNb);CHKERRQ(ierr); 30137c927364SMatthew G. Knepley ierr = PetscFEGetNumComponents(fe, &Nc);CHKERRQ(ierr); 30146f3d3cbcSMatthew G. Knepley ierr = PetscFEGetBasisSpace(fe, &sp);CHKERRQ(ierr); 30159b2fc754SMatthew G. Knepley ierr = PetscSpaceGetDegree(sp, NULL, &maxDegree);CHKERRQ(ierr); 30169b2fc754SMatthew G. Knepley if (!maxDegree) needAvg[f] = PETSC_TRUE; 301797c42addSMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 301897c42addSMatthew G. Knepley PetscFV fv = (PetscFV) obj; 301997c42addSMatthew G. Knepley PetscDualSpace Q; 302097c42addSMatthew G. Knepley 302197c42addSMatthew G. Knepley ierr = PetscFVRefine(fv, &fvRef[f]);CHKERRQ(ierr); 302297c42addSMatthew G. Knepley ierr = PetscFVGetDualSpace(fvRef[f], &Q);CHKERRQ(ierr); 302397c42addSMatthew G. Knepley ierr = PetscDualSpaceGetDimension(Q, &fNb);CHKERRQ(ierr); 302497c42addSMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &Nc);CHKERRQ(ierr); 30256f3d3cbcSMatthew G. Knepley needAvg[f] = PETSC_TRUE; 302697c42addSMatthew G. Knepley } 3027d172c84bSMatthew G. Knepley fTotDim += fNb; 30287c927364SMatthew G. Knepley } 3029e9d4ef1bSMatthew G. Knepley ierr = PetscDSGetTotalDimension(prob, &cTotDim);CHKERRQ(ierr); 30307c927364SMatthew G. Knepley ierr = PetscMalloc1(cTotDim,&cmap);CHKERRQ(ierr); 30317c927364SMatthew G. Knepley for (field = 0, offsetC = 0, offsetF = 0; field < Nf; ++field) { 30327c927364SMatthew G. Knepley PetscFE feC; 303397c42addSMatthew G. Knepley PetscFV fvC; 30347c927364SMatthew G. Knepley PetscDualSpace QF, QC; 3035d172c84bSMatthew G. Knepley PetscInt order = -1, NcF, NcC, fpdim, cpdim; 30367c927364SMatthew G. Knepley 303797c42addSMatthew G. Knepley if (feRef[field]) { 3038e9d4ef1bSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &feC);CHKERRQ(ierr); 30397c927364SMatthew G. Knepley ierr = PetscFEGetNumComponents(feC, &NcC);CHKERRQ(ierr); 30407c927364SMatthew G. Knepley ierr = PetscFEGetNumComponents(feRef[field], &NcF);CHKERRQ(ierr); 30417c927364SMatthew G. Knepley ierr = PetscFEGetDualSpace(feRef[field], &QF);CHKERRQ(ierr); 3042d172c84bSMatthew G. Knepley ierr = PetscDualSpaceGetOrder(QF, &order);CHKERRQ(ierr); 30437c927364SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(QF, &fpdim);CHKERRQ(ierr); 30447c927364SMatthew G. Knepley ierr = PetscFEGetDualSpace(feC, &QC);CHKERRQ(ierr); 30457c927364SMatthew G. Knepley ierr = PetscDualSpaceGetDimension(QC, &cpdim);CHKERRQ(ierr); 304697c42addSMatthew G. Knepley } else { 304797c42addSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, field, (PetscObject *) &fvC);CHKERRQ(ierr); 304897c42addSMatthew G. Knepley ierr = PetscFVGetNumComponents(fvC, &NcC);CHKERRQ(ierr); 304997c42addSMatthew G. Knepley ierr = PetscFVGetNumComponents(fvRef[field], &NcF);CHKERRQ(ierr); 305097c42addSMatthew G. Knepley ierr = PetscFVGetDualSpace(fvRef[field], &QF);CHKERRQ(ierr); 305197c42addSMatthew G. Knepley ierr = PetscDualSpaceGetDimension(QF, &fpdim);CHKERRQ(ierr); 305297c42addSMatthew G. Knepley ierr = PetscFVGetDualSpace(fvC, &QC);CHKERRQ(ierr); 305397c42addSMatthew G. Knepley ierr = PetscDualSpaceGetDimension(QC, &cpdim);CHKERRQ(ierr); 305497c42addSMatthew G. Knepley } 3055ff1e0c32SBarry Smith if (NcF != NcC) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of components in fine space field %D does not match coarse field %D", NcF, NcC); 30567c927364SMatthew G. Knepley for (c = 0; c < cpdim; ++c) { 30577c927364SMatthew G. Knepley PetscQuadrature cfunc; 3058d172c84bSMatthew G. Knepley const PetscReal *cqpoints, *cqweights; 3059d172c84bSMatthew G. Knepley PetscInt NqcC, NpC; 306097c42addSMatthew G. Knepley PetscBool found = PETSC_FALSE; 30617c927364SMatthew G. Knepley 30627c927364SMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(QC, c, &cfunc);CHKERRQ(ierr); 3063d172c84bSMatthew G. Knepley ierr = PetscQuadratureGetData(cfunc, NULL, &NqcC, &NpC, &cqpoints, &cqweights);CHKERRQ(ierr); 3064ff1e0c32SBarry Smith if (NqcC != NcC) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of quadrature components %D must match number of field components %D", NqcC, NcC); 306597c42addSMatthew G. Knepley if (NpC != 1 && feRef[field]) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Do not know how to do injection for moments"); 30667c927364SMatthew G. Knepley for (f = 0; f < fpdim; ++f) { 30677c927364SMatthew G. Knepley PetscQuadrature ffunc; 3068d172c84bSMatthew G. Knepley const PetscReal *fqpoints, *fqweights; 30697c927364SMatthew G. Knepley PetscReal sum = 0.0; 3070d172c84bSMatthew G. Knepley PetscInt NqcF, NpF; 30717c927364SMatthew G. Knepley 30727c927364SMatthew G. Knepley ierr = PetscDualSpaceGetFunctional(QF, f, &ffunc);CHKERRQ(ierr); 3073d172c84bSMatthew G. Knepley ierr = PetscQuadratureGetData(ffunc, NULL, &NqcF, &NpF, &fqpoints, &fqweights);CHKERRQ(ierr); 3074ff1e0c32SBarry Smith if (NqcF != NcF) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Number of quadrature components %D must match number of field components %D", NqcF, NcF); 30757c927364SMatthew G. Knepley if (NpC != NpF) continue; 30767c927364SMatthew G. Knepley for (d = 0; d < dim; ++d) sum += PetscAbsReal(cqpoints[d] - fqpoints[d]); 30777c927364SMatthew G. Knepley if (sum > 1.0e-9) continue; 3078d172c84bSMatthew G. Knepley for (d = 0; d < NcC; ++d) sum += PetscAbsReal(cqweights[d]*fqweights[d]); 3079d172c84bSMatthew G. Knepley if (sum < 1.0e-9) continue; 3080d172c84bSMatthew G. Knepley cmap[offsetC+c] = offsetF+f; 308197c42addSMatthew G. Knepley found = PETSC_TRUE; 30827c927364SMatthew G. Knepley break; 30837c927364SMatthew G. Knepley } 308497c42addSMatthew G. Knepley if (!found) { 308597c42addSMatthew G. Knepley /* TODO We really want the average here, but some asshole put VecScatter in the interface */ 3086d172c84bSMatthew G. Knepley if (fvRef[field] || (feRef[field] && order == 0)) { 3087d172c84bSMatthew G. Knepley cmap[offsetC+c] = offsetF+0; 308897c42addSMatthew G. Knepley } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Could not locate matching functional for injection"); 308997c42addSMatthew G. Knepley } 30907c927364SMatthew G. Knepley } 3091d172c84bSMatthew G. Knepley offsetC += cpdim; 3092d172c84bSMatthew G. Knepley offsetF += fpdim; 30937c927364SMatthew G. Knepley } 309497c42addSMatthew G. Knepley for (f = 0; f < Nf; ++f) {ierr = PetscFEDestroy(&feRef[f]);CHKERRQ(ierr);ierr = PetscFVDestroy(&fvRef[f]);CHKERRQ(ierr);} 30956f3d3cbcSMatthew G. Knepley ierr = PetscFree3(feRef,fvRef,needAvg);CHKERRQ(ierr); 30967c927364SMatthew G. Knepley 30977c927364SMatthew G. Knepley ierr = DMGetGlobalVector(dmf, &fv);CHKERRQ(ierr); 30987c927364SMatthew G. Knepley ierr = DMGetGlobalVector(dmc, &cv);CHKERRQ(ierr); 30990bd915a7SMatthew G. Knepley ierr = VecGetOwnershipRange(cv, &startC, &endC);CHKERRQ(ierr); 31007c927364SMatthew G. Knepley ierr = PetscSectionGetConstrainedStorageSize(cglobalSection, &m);CHKERRQ(ierr); 31017c927364SMatthew G. Knepley ierr = PetscMalloc2(cTotDim,&cellCIndices,fTotDim,&cellFIndices);CHKERRQ(ierr); 3102aa7da3c4SMatthew G. Knepley ierr = PetscMalloc1(m,&cindices);CHKERRQ(ierr); 3103aa7da3c4SMatthew G. Knepley ierr = PetscMalloc1(m,&findices);CHKERRQ(ierr); 31047c927364SMatthew G. Knepley for (d = 0; d < m; ++d) cindices[d] = findices[d] = -1; 31057c927364SMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 31067c927364SMatthew G. Knepley ierr = DMPlexMatGetClosureIndicesRefined(dmf, fsection, fglobalSection, dmc, csection, cglobalSection, c, cellCIndices, cellFIndices);CHKERRQ(ierr); 31077c927364SMatthew G. Knepley for (d = 0; d < cTotDim; ++d) { 31080bd915a7SMatthew G. Knepley if ((cellCIndices[d] < startC) || (cellCIndices[d] >= endC)) continue; 3109ff1e0c32SBarry Smith if ((findices[cellCIndices[d]-startC] >= 0) && (findices[cellCIndices[d]-startC] != cellFIndices[cmap[d]])) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Coarse dof %D maps to both %D and %D", cindices[cellCIndices[d]-startC], findices[cellCIndices[d]-startC], cellFIndices[cmap[d]]); 31107c927364SMatthew G. Knepley cindices[cellCIndices[d]-startC] = cellCIndices[d]; 31117c927364SMatthew G. Knepley findices[cellCIndices[d]-startC] = cellFIndices[cmap[d]]; 31127c927364SMatthew G. Knepley } 31137c927364SMatthew G. Knepley } 31147c927364SMatthew G. Knepley ierr = PetscFree(cmap);CHKERRQ(ierr); 31157c927364SMatthew G. Knepley ierr = PetscFree2(cellCIndices,cellFIndices);CHKERRQ(ierr); 31167c927364SMatthew G. Knepley 31177c927364SMatthew G. Knepley ierr = ISCreateGeneral(PETSC_COMM_SELF, m, cindices, PETSC_OWN_POINTER, &cis);CHKERRQ(ierr); 31187c927364SMatthew G. Knepley ierr = ISCreateGeneral(PETSC_COMM_SELF, m, findices, PETSC_OWN_POINTER, &fis);CHKERRQ(ierr); 31199448b7f1SJunchao Zhang ierr = VecScatterCreate(cv, cis, fv, fis, sc);CHKERRQ(ierr); 31207c927364SMatthew G. Knepley ierr = ISDestroy(&cis);CHKERRQ(ierr); 31217c927364SMatthew G. Knepley ierr = ISDestroy(&fis);CHKERRQ(ierr); 31227c927364SMatthew G. Knepley ierr = DMRestoreGlobalVector(dmf, &fv);CHKERRQ(ierr); 31237c927364SMatthew G. Knepley ierr = DMRestoreGlobalVector(dmc, &cv);CHKERRQ(ierr); 312475a69067SMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_InjectorFEM,dmc,dmf,0,0);CHKERRQ(ierr); 3125cb1e1211SMatthew G Knepley PetscFunctionReturn(0); 3126cb1e1211SMatthew G Knepley } 3127a1cf66bbSMatthew G. Knepley 31282f856554SMatthew G. Knepley /*@C 31292f856554SMatthew G. Knepley DMPlexGetCellFields - Retrieve the field values values for a chunk of cells 31302f856554SMatthew G. Knepley 31312f856554SMatthew G. Knepley Input Parameters: 31322f856554SMatthew G. Knepley + dm - The DM 31332f856554SMatthew G. Knepley . cellIS - The cells to include 31342f856554SMatthew G. Knepley . locX - A local vector with the solution fields 31352f856554SMatthew G. Knepley . locX_t - A local vector with solution field time derivatives, or NULL 31362f856554SMatthew G. Knepley - locA - A local vector with auxiliary fields, or NULL 31372f856554SMatthew G. Knepley 31382f856554SMatthew G. Knepley Output Parameters: 31392f856554SMatthew G. Knepley + u - The field coefficients 31402f856554SMatthew G. Knepley . u_t - The fields derivative coefficients 31412f856554SMatthew G. Knepley - a - The auxiliary field coefficients 31422f856554SMatthew G. Knepley 31432f856554SMatthew G. Knepley Level: developer 31442f856554SMatthew G. Knepley 31452f856554SMatthew G. Knepley .seealso: DMPlexGetFaceFields() 31462f856554SMatthew G. Knepley @*/ 31472f856554SMatthew G. Knepley PetscErrorCode DMPlexGetCellFields(DM dm, IS cellIS, Vec locX, Vec locX_t, Vec locA, PetscScalar **u, PetscScalar **u_t, PetscScalar **a) 31482f856554SMatthew G. Knepley { 31492f856554SMatthew G. Knepley DM plex, plexA = NULL; 3150a6e0b375SMatthew G. Knepley DMEnclosureType encAux; 31512f856554SMatthew G. Knepley PetscSection section, sectionAux; 31522f856554SMatthew G. Knepley PetscDS prob; 31532f856554SMatthew G. Knepley const PetscInt *cells; 31542f856554SMatthew G. Knepley PetscInt cStart, cEnd, numCells, totDim, totDimAux, c; 31552f856554SMatthew G. Knepley PetscErrorCode ierr; 31562f856554SMatthew G. Knepley 31572f856554SMatthew G. Knepley PetscFunctionBegin; 31582f856554SMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 31592f856554SMatthew G. Knepley PetscValidHeaderSpecific(locX, VEC_CLASSID, 4); 31602f856554SMatthew G. Knepley if (locX_t) {PetscValidHeaderSpecific(locX_t, VEC_CLASSID, 5);} 31612f856554SMatthew G. Knepley if (locA) {PetscValidHeaderSpecific(locA, VEC_CLASSID, 6);} 31622f856554SMatthew G. Knepley PetscValidPointer(u, 7); 31632f856554SMatthew G. Knepley PetscValidPointer(u_t, 8); 31642f856554SMatthew G. Knepley PetscValidPointer(a, 9); 31652f856554SMatthew G. Knepley ierr = DMPlexConvertPlex(dm, &plex, PETSC_FALSE);CHKERRQ(ierr); 31662f856554SMatthew G. Knepley ierr = ISGetPointRange(cellIS, &cStart, &cEnd, &cells);CHKERRQ(ierr); 316792fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 31682f856554SMatthew G. Knepley ierr = DMGetCellDS(dm, cStart, &prob);CHKERRQ(ierr); 31692f856554SMatthew G. Knepley ierr = PetscDSGetTotalDimension(prob, &totDim);CHKERRQ(ierr); 31702f856554SMatthew G. Knepley if (locA) { 31712f856554SMatthew G. Knepley DM dmAux; 31722f856554SMatthew G. Knepley PetscDS probAux; 31732f856554SMatthew G. Knepley 31742f856554SMatthew G. Knepley ierr = VecGetDM(locA, &dmAux);CHKERRQ(ierr); 3175a6e0b375SMatthew G. Knepley ierr = DMGetEnclosureRelation(dmAux, dm, &encAux);CHKERRQ(ierr); 31762f856554SMatthew G. Knepley ierr = DMPlexConvertPlex(dmAux, &plexA, PETSC_FALSE);CHKERRQ(ierr); 317792fd8e1eSJed Brown ierr = DMGetLocalSection(dmAux, §ionAux);CHKERRQ(ierr); 31782f856554SMatthew G. Knepley ierr = DMGetDS(dmAux, &probAux);CHKERRQ(ierr); 31792f856554SMatthew G. Knepley ierr = PetscDSGetTotalDimension(probAux, &totDimAux);CHKERRQ(ierr); 31802f856554SMatthew G. Knepley } 31812f856554SMatthew G. Knepley numCells = cEnd - cStart; 31822f856554SMatthew G. Knepley ierr = DMGetWorkArray(dm, numCells*totDim, MPIU_SCALAR, u);CHKERRQ(ierr); 31832f856554SMatthew G. Knepley if (locX_t) {ierr = DMGetWorkArray(dm, numCells*totDim, MPIU_SCALAR, u_t);CHKERRQ(ierr);} else {*u_t = NULL;} 31842f856554SMatthew G. Knepley if (locA) {ierr = DMGetWorkArray(dm, numCells*totDimAux, MPIU_SCALAR, a);CHKERRQ(ierr);} else {*a = NULL;} 31852f856554SMatthew G. Knepley for (c = cStart; c < cEnd; ++c) { 31862f856554SMatthew G. Knepley const PetscInt cell = cells ? cells[c] : c; 31872f856554SMatthew G. Knepley const PetscInt cind = c - cStart; 31882f856554SMatthew G. Knepley PetscScalar *x = NULL, *x_t = NULL, *ul = *u, *ul_t = *u_t, *al = *a; 31892f856554SMatthew G. Knepley PetscInt i; 31902f856554SMatthew G. Knepley 31912f856554SMatthew G. Knepley ierr = DMPlexVecGetClosure(plex, section, locX, cell, NULL, &x);CHKERRQ(ierr); 31922f856554SMatthew G. Knepley for (i = 0; i < totDim; ++i) ul[cind*totDim+i] = x[i]; 31932f856554SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(plex, section, locX, cell, NULL, &x);CHKERRQ(ierr); 31942f856554SMatthew G. Knepley if (locX_t) { 31952f856554SMatthew G. Knepley ierr = DMPlexVecGetClosure(plex, section, locX_t, cell, NULL, &x_t);CHKERRQ(ierr); 31962f856554SMatthew G. Knepley for (i = 0; i < totDim; ++i) ul_t[cind*totDim+i] = x_t[i]; 31972f856554SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(plex, section, locX_t, cell, NULL, &x_t);CHKERRQ(ierr); 31982f856554SMatthew G. Knepley } 31992f856554SMatthew G. Knepley if (locA) { 32002f856554SMatthew G. Knepley PetscInt subcell; 3201a6e0b375SMatthew G. Knepley ierr = DMGetEnclosurePoint(plexA, dm, encAux, cell, &subcell);CHKERRQ(ierr); 32022f856554SMatthew G. Knepley ierr = DMPlexVecGetClosure(plexA, sectionAux, locA, subcell, NULL, &x);CHKERRQ(ierr); 32032f856554SMatthew G. Knepley for (i = 0; i < totDimAux; ++i) al[cind*totDimAux+i] = x[i]; 32042f856554SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(plexA, sectionAux, locA, subcell, NULL, &x);CHKERRQ(ierr); 32052f856554SMatthew G. Knepley } 32062f856554SMatthew G. Knepley } 32072f856554SMatthew G. Knepley ierr = DMDestroy(&plex);CHKERRQ(ierr); 32082f856554SMatthew G. Knepley if (locA) {ierr = DMDestroy(&plexA);CHKERRQ(ierr);} 32092f856554SMatthew G. Knepley ierr = ISRestorePointRange(cellIS, &cStart, &cEnd, &cells);CHKERRQ(ierr); 32102f856554SMatthew G. Knepley PetscFunctionReturn(0); 32112f856554SMatthew G. Knepley } 32122f856554SMatthew G. Knepley 32132f856554SMatthew G. Knepley /*@C 32142f856554SMatthew G. Knepley DMPlexRestoreCellFields - Restore the field values values for a chunk of cells 32152f856554SMatthew G. Knepley 32162f856554SMatthew G. Knepley Input Parameters: 32172f856554SMatthew G. Knepley + dm - The DM 32182f856554SMatthew G. Knepley . cellIS - The cells to include 32192f856554SMatthew G. Knepley . locX - A local vector with the solution fields 32202f856554SMatthew G. Knepley . locX_t - A local vector with solution field time derivatives, or NULL 32212f856554SMatthew G. Knepley - locA - A local vector with auxiliary fields, or NULL 32222f856554SMatthew G. Knepley 32232f856554SMatthew G. Knepley Output Parameters: 32242f856554SMatthew G. Knepley + u - The field coefficients 32252f856554SMatthew G. Knepley . u_t - The fields derivative coefficients 32262f856554SMatthew G. Knepley - a - The auxiliary field coefficients 32272f856554SMatthew G. Knepley 32282f856554SMatthew G. Knepley Level: developer 32292f856554SMatthew G. Knepley 32302f856554SMatthew G. Knepley .seealso: DMPlexGetFaceFields() 32312f856554SMatthew G. Knepley @*/ 32322f856554SMatthew G. Knepley PetscErrorCode DMPlexRestoreCellFields(DM dm, IS cellIS, Vec locX, Vec locX_t, Vec locA, PetscScalar **u, PetscScalar **u_t, PetscScalar **a) 32332f856554SMatthew G. Knepley { 32342f856554SMatthew G. Knepley PetscErrorCode ierr; 32352f856554SMatthew G. Knepley 32362f856554SMatthew G. Knepley PetscFunctionBegin; 32372f856554SMatthew G. Knepley ierr = DMRestoreWorkArray(dm, 0, MPIU_SCALAR, u);CHKERRQ(ierr); 32382f856554SMatthew G. Knepley if (locX_t) {ierr = DMRestoreWorkArray(dm, 0, MPIU_SCALAR, u_t);CHKERRQ(ierr);} 32392f856554SMatthew G. Knepley if (locA) {ierr = DMRestoreWorkArray(dm, 0, MPIU_SCALAR, a);CHKERRQ(ierr);} 32402f856554SMatthew G. Knepley PetscFunctionReturn(0); 32412f856554SMatthew G. Knepley } 32422f856554SMatthew G. Knepley 32432f856554SMatthew G. Knepley /*@C 32442f856554SMatthew G. Knepley DMPlexGetFaceFields - Retrieve the field values values for a chunk of faces 32452f856554SMatthew G. Knepley 32462f856554SMatthew G. Knepley Input Parameters: 32472f856554SMatthew G. Knepley + dm - The DM 32482f856554SMatthew G. Knepley . fStart - The first face to include 32492f856554SMatthew G. Knepley . fEnd - The first face to exclude 32502f856554SMatthew G. Knepley . locX - A local vector with the solution fields 32512f856554SMatthew G. Knepley . locX_t - A local vector with solution field time derivatives, or NULL 32522f856554SMatthew G. Knepley . faceGeometry - A local vector with face geometry 32532f856554SMatthew G. Knepley . cellGeometry - A local vector with cell geometry 32542f856554SMatthew G. Knepley - locaGrad - A local vector with field gradients, or NULL 32552f856554SMatthew G. Knepley 32562f856554SMatthew G. Knepley Output Parameters: 32572f856554SMatthew G. Knepley + Nface - The number of faces with field values 32582f856554SMatthew G. Knepley . uL - The field values at the left side of the face 32592f856554SMatthew G. Knepley - uR - The field values at the right side of the face 32602f856554SMatthew G. Knepley 32612f856554SMatthew G. Knepley Level: developer 32622f856554SMatthew G. Knepley 32632f856554SMatthew G. Knepley .seealso: DMPlexGetCellFields() 32642f856554SMatthew G. Knepley @*/ 32652f856554SMatthew G. Knepley PetscErrorCode DMPlexGetFaceFields(DM dm, PetscInt fStart, PetscInt fEnd, Vec locX, Vec locX_t, Vec faceGeometry, Vec cellGeometry, Vec locGrad, PetscInt *Nface, PetscScalar **uL, PetscScalar **uR) 32662f856554SMatthew G. Knepley { 32672f856554SMatthew G. Knepley DM dmFace, dmCell, dmGrad = NULL; 32682f856554SMatthew G. Knepley PetscSection section; 32692f856554SMatthew G. Knepley PetscDS prob; 32702f856554SMatthew G. Knepley DMLabel ghostLabel; 32712f856554SMatthew G. Knepley const PetscScalar *facegeom, *cellgeom, *x, *lgrad; 32722f856554SMatthew G. Knepley PetscBool *isFE; 32732f856554SMatthew G. Knepley PetscInt dim, Nf, f, Nc, numFaces = fEnd - fStart, iface, face; 32742f856554SMatthew G. Knepley PetscErrorCode ierr; 32752f856554SMatthew G. Knepley 32762f856554SMatthew G. Knepley PetscFunctionBegin; 32772f856554SMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 32782f856554SMatthew G. Knepley PetscValidHeaderSpecific(locX, VEC_CLASSID, 4); 32792f856554SMatthew G. Knepley if (locX_t) {PetscValidHeaderSpecific(locX_t, VEC_CLASSID, 5);} 32802f856554SMatthew G. Knepley PetscValidHeaderSpecific(faceGeometry, VEC_CLASSID, 6); 32812f856554SMatthew G. Knepley PetscValidHeaderSpecific(cellGeometry, VEC_CLASSID, 7); 32822f856554SMatthew G. Knepley if (locGrad) {PetscValidHeaderSpecific(locGrad, VEC_CLASSID, 8);} 32832f856554SMatthew G. Knepley PetscValidPointer(uL, 9); 32842f856554SMatthew G. Knepley PetscValidPointer(uR, 10); 32852f856554SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 32862f856554SMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 328792fd8e1eSJed Brown ierr = DMGetLocalSection(dm, §ion);CHKERRQ(ierr); 32882f856554SMatthew G. Knepley ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr); 32892f856554SMatthew G. Knepley ierr = PetscDSGetTotalComponents(prob, &Nc);CHKERRQ(ierr); 32902f856554SMatthew G. Knepley ierr = PetscMalloc1(Nf, &isFE);CHKERRQ(ierr); 32912f856554SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 32922f856554SMatthew G. Knepley PetscObject obj; 32932f856554SMatthew G. Knepley PetscClassId id; 32942f856554SMatthew G. Knepley 32952f856554SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr); 32962f856554SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 32972f856554SMatthew G. Knepley if (id == PETSCFE_CLASSID) {isFE[f] = PETSC_TRUE;} 32982f856554SMatthew G. Knepley else if (id == PETSCFV_CLASSID) {isFE[f] = PETSC_FALSE;} 32992f856554SMatthew G. Knepley else {isFE[f] = PETSC_FALSE;} 33002f856554SMatthew G. Knepley } 33012f856554SMatthew G. Knepley ierr = DMGetLabel(dm, "ghost", &ghostLabel);CHKERRQ(ierr); 33022f856554SMatthew G. Knepley ierr = VecGetArrayRead(locX, &x);CHKERRQ(ierr); 33032f856554SMatthew G. Knepley ierr = VecGetDM(faceGeometry, &dmFace);CHKERRQ(ierr); 33042f856554SMatthew G. Knepley ierr = VecGetArrayRead(faceGeometry, &facegeom);CHKERRQ(ierr); 33052f856554SMatthew G. Knepley ierr = VecGetDM(cellGeometry, &dmCell);CHKERRQ(ierr); 33062f856554SMatthew G. Knepley ierr = VecGetArrayRead(cellGeometry, &cellgeom);CHKERRQ(ierr); 33072f856554SMatthew G. Knepley if (locGrad) { 33082f856554SMatthew G. Knepley ierr = VecGetDM(locGrad, &dmGrad);CHKERRQ(ierr); 33092f856554SMatthew G. Knepley ierr = VecGetArrayRead(locGrad, &lgrad);CHKERRQ(ierr); 33102f856554SMatthew G. Knepley } 33112f856554SMatthew G. Knepley ierr = DMGetWorkArray(dm, numFaces*Nc, MPIU_SCALAR, uL);CHKERRQ(ierr); 33122f856554SMatthew G. Knepley ierr = DMGetWorkArray(dm, numFaces*Nc, MPIU_SCALAR, uR);CHKERRQ(ierr); 33132f856554SMatthew G. Knepley /* Right now just eat the extra work for FE (could make a cell loop) */ 33142f856554SMatthew G. Knepley for (face = fStart, iface = 0; face < fEnd; ++face) { 33152f856554SMatthew G. Knepley const PetscInt *cells; 33162f856554SMatthew G. Knepley PetscFVFaceGeom *fg; 33172f856554SMatthew G. Knepley PetscFVCellGeom *cgL, *cgR; 33182f856554SMatthew G. Knepley PetscScalar *xL, *xR, *gL, *gR; 33192f856554SMatthew G. Knepley PetscScalar *uLl = *uL, *uRl = *uR; 33202f856554SMatthew G. Knepley PetscInt ghost, nsupp, nchild; 33212f856554SMatthew G. Knepley 33222f856554SMatthew G. Knepley ierr = DMLabelGetValue(ghostLabel, face, &ghost);CHKERRQ(ierr); 33232f856554SMatthew G. Knepley ierr = DMPlexGetSupportSize(dm, face, &nsupp);CHKERRQ(ierr); 33242f856554SMatthew G. Knepley ierr = DMPlexGetTreeChildren(dm, face, &nchild, NULL);CHKERRQ(ierr); 33252f856554SMatthew G. Knepley if (ghost >= 0 || nsupp > 2 || nchild > 0) continue; 33262f856554SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmFace, face, facegeom, &fg);CHKERRQ(ierr); 33272f856554SMatthew G. Knepley ierr = DMPlexGetSupport(dm, face, &cells);CHKERRQ(ierr); 33282f856554SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmCell, cells[0], cellgeom, &cgL);CHKERRQ(ierr); 33292f856554SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmCell, cells[1], cellgeom, &cgR);CHKERRQ(ierr); 33302f856554SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 33312f856554SMatthew G. Knepley PetscInt off; 33322f856554SMatthew G. Knepley 33332f856554SMatthew G. Knepley ierr = PetscDSGetComponentOffset(prob, f, &off);CHKERRQ(ierr); 33342f856554SMatthew G. Knepley if (isFE[f]) { 33352f856554SMatthew G. Knepley const PetscInt *cone; 33362f856554SMatthew G. Knepley PetscInt comp, coneSizeL, coneSizeR, faceLocL, faceLocR, ldof, rdof, d; 33372f856554SMatthew G. Knepley 33382f856554SMatthew G. Knepley xL = xR = NULL; 33392f856554SMatthew G. Knepley ierr = PetscSectionGetFieldComponents(section, f, &comp);CHKERRQ(ierr); 33402f856554SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, section, locX, cells[0], &ldof, (PetscScalar **) &xL);CHKERRQ(ierr); 33412f856554SMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, section, locX, cells[1], &rdof, (PetscScalar **) &xR);CHKERRQ(ierr); 33422f856554SMatthew G. Knepley ierr = DMPlexGetCone(dm, cells[0], &cone);CHKERRQ(ierr); 33432f856554SMatthew G. Knepley ierr = DMPlexGetConeSize(dm, cells[0], &coneSizeL);CHKERRQ(ierr); 33442f856554SMatthew G. Knepley for (faceLocL = 0; faceLocL < coneSizeL; ++faceLocL) if (cone[faceLocL] == face) break; 33452f856554SMatthew G. Knepley ierr = DMPlexGetCone(dm, cells[1], &cone);CHKERRQ(ierr); 33462f856554SMatthew G. Knepley ierr = DMPlexGetConeSize(dm, cells[1], &coneSizeR);CHKERRQ(ierr); 33472f856554SMatthew G. Knepley for (faceLocR = 0; faceLocR < coneSizeR; ++faceLocR) if (cone[faceLocR] == face) break; 3348ff1e0c32SBarry Smith if (faceLocL == coneSizeL && faceLocR == coneSizeR) SETERRQ3(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Could not find face %D in cone of cell %D or cell %D", face, cells[0], cells[1]); 33492f856554SMatthew G. Knepley /* Check that FEM field has values in the right cell (sometimes its an FV ghost cell) */ 33502f856554SMatthew G. Knepley /* TODO: this is a hack that might not be right for nonconforming */ 33512f856554SMatthew G. Knepley if (faceLocL < coneSizeL) { 3352a8f1f9e5SMatthew G. Knepley ierr = PetscFEEvaluateFaceFields_Internal(prob, f, faceLocL, xL, &uLl[iface*Nc+off]);CHKERRQ(ierr); 3353a8f1f9e5SMatthew G. Knepley if (rdof == ldof && faceLocR < coneSizeR) {ierr = PetscFEEvaluateFaceFields_Internal(prob, f, faceLocR, xR, &uRl[iface*Nc+off]);CHKERRQ(ierr);} 33542f856554SMatthew G. Knepley else {for(d = 0; d < comp; ++d) uRl[iface*Nc+off+d] = uLl[iface*Nc+off+d];} 33552f856554SMatthew G. Knepley } 33562f856554SMatthew G. Knepley else { 3357a8f1f9e5SMatthew G. Knepley ierr = PetscFEEvaluateFaceFields_Internal(prob, f, faceLocR, xR, &uRl[iface*Nc+off]);CHKERRQ(ierr); 33582f856554SMatthew G. Knepley ierr = PetscSectionGetFieldComponents(section, f, &comp);CHKERRQ(ierr); 33592f856554SMatthew G. Knepley for(d = 0; d < comp; ++d) uLl[iface*Nc+off+d] = uRl[iface*Nc+off+d]; 33602f856554SMatthew G. Knepley } 33612f856554SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, section, locX, cells[0], &ldof, (PetscScalar **) &xL);CHKERRQ(ierr); 33622f856554SMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, section, locX, cells[1], &rdof, (PetscScalar **) &xR);CHKERRQ(ierr); 33632f856554SMatthew G. Knepley } else { 33642f856554SMatthew G. Knepley PetscFV fv; 33652f856554SMatthew G. Knepley PetscInt numComp, c; 33662f856554SMatthew G. Knepley 33672f856554SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, (PetscObject *) &fv);CHKERRQ(ierr); 33682f856554SMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &numComp);CHKERRQ(ierr); 33692f856554SMatthew G. Knepley ierr = DMPlexPointLocalFieldRead(dm, cells[0], f, x, &xL);CHKERRQ(ierr); 33702f856554SMatthew G. Knepley ierr = DMPlexPointLocalFieldRead(dm, cells[1], f, x, &xR);CHKERRQ(ierr); 33712f856554SMatthew G. Knepley if (dmGrad) { 33722f856554SMatthew G. Knepley PetscReal dxL[3], dxR[3]; 33732f856554SMatthew G. Knepley 33742f856554SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmGrad, cells[0], lgrad, &gL);CHKERRQ(ierr); 33752f856554SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmGrad, cells[1], lgrad, &gR);CHKERRQ(ierr); 33762f856554SMatthew G. Knepley DMPlex_WaxpyD_Internal(dim, -1, cgL->centroid, fg->centroid, dxL); 33772f856554SMatthew G. Knepley DMPlex_WaxpyD_Internal(dim, -1, cgR->centroid, fg->centroid, dxR); 33782f856554SMatthew G. Knepley for (c = 0; c < numComp; ++c) { 33792f856554SMatthew G. Knepley uLl[iface*Nc+off+c] = xL[c] + DMPlex_DotD_Internal(dim, &gL[c*dim], dxL); 33802f856554SMatthew G. Knepley uRl[iface*Nc+off+c] = xR[c] + DMPlex_DotD_Internal(dim, &gR[c*dim], dxR); 33812f856554SMatthew G. Knepley } 33822f856554SMatthew G. Knepley } else { 33832f856554SMatthew G. Knepley for (c = 0; c < numComp; ++c) { 33842f856554SMatthew G. Knepley uLl[iface*Nc+off+c] = xL[c]; 33852f856554SMatthew G. Knepley uRl[iface*Nc+off+c] = xR[c]; 33862f856554SMatthew G. Knepley } 33872f856554SMatthew G. Knepley } 33882f856554SMatthew G. Knepley } 33892f856554SMatthew G. Knepley } 33902f856554SMatthew G. Knepley ++iface; 33912f856554SMatthew G. Knepley } 33922f856554SMatthew G. Knepley *Nface = iface; 33932f856554SMatthew G. Knepley ierr = VecRestoreArrayRead(locX, &x);CHKERRQ(ierr); 33942f856554SMatthew G. Knepley ierr = VecRestoreArrayRead(faceGeometry, &facegeom);CHKERRQ(ierr); 33952f856554SMatthew G. Knepley ierr = VecRestoreArrayRead(cellGeometry, &cellgeom);CHKERRQ(ierr); 33962f856554SMatthew G. Knepley if (locGrad) { 33972f856554SMatthew G. Knepley ierr = VecRestoreArrayRead(locGrad, &lgrad);CHKERRQ(ierr); 33982f856554SMatthew G. Knepley } 33992f856554SMatthew G. Knepley ierr = PetscFree(isFE);CHKERRQ(ierr); 34002f856554SMatthew G. Knepley PetscFunctionReturn(0); 34012f856554SMatthew G. Knepley } 34022f856554SMatthew G. Knepley 34032f856554SMatthew G. Knepley /*@C 34042f856554SMatthew G. Knepley DMPlexRestoreFaceFields - Restore the field values values for a chunk of faces 34052f856554SMatthew G. Knepley 34062f856554SMatthew G. Knepley Input Parameters: 34072f856554SMatthew G. Knepley + dm - The DM 34082f856554SMatthew G. Knepley . fStart - The first face to include 34092f856554SMatthew G. Knepley . fEnd - The first face to exclude 34102f856554SMatthew G. Knepley . locX - A local vector with the solution fields 34112f856554SMatthew G. Knepley . locX_t - A local vector with solution field time derivatives, or NULL 34122f856554SMatthew G. Knepley . faceGeometry - A local vector with face geometry 34132f856554SMatthew G. Knepley . cellGeometry - A local vector with cell geometry 34142f856554SMatthew G. Knepley - locaGrad - A local vector with field gradients, or NULL 34152f856554SMatthew G. Knepley 34162f856554SMatthew G. Knepley Output Parameters: 34172f856554SMatthew G. Knepley + Nface - The number of faces with field values 34182f856554SMatthew G. Knepley . uL - The field values at the left side of the face 34192f856554SMatthew G. Knepley - uR - The field values at the right side of the face 34202f856554SMatthew G. Knepley 34212f856554SMatthew G. Knepley Level: developer 34222f856554SMatthew G. Knepley 34232f856554SMatthew G. Knepley .seealso: DMPlexGetFaceFields() 34242f856554SMatthew G. Knepley @*/ 34252f856554SMatthew G. Knepley PetscErrorCode DMPlexRestoreFaceFields(DM dm, PetscInt fStart, PetscInt fEnd, Vec locX, Vec locX_t, Vec faceGeometry, Vec cellGeometry, Vec locGrad, PetscInt *Nface, PetscScalar **uL, PetscScalar **uR) 34262f856554SMatthew G. Knepley { 34272f856554SMatthew G. Knepley PetscErrorCode ierr; 34282f856554SMatthew G. Knepley 34292f856554SMatthew G. Knepley PetscFunctionBegin; 34302f856554SMatthew G. Knepley ierr = DMRestoreWorkArray(dm, 0, MPIU_SCALAR, uL);CHKERRQ(ierr); 34312f856554SMatthew G. Knepley ierr = DMRestoreWorkArray(dm, 0, MPIU_SCALAR, uR);CHKERRQ(ierr); 34322f856554SMatthew G. Knepley PetscFunctionReturn(0); 34332f856554SMatthew G. Knepley } 34342f856554SMatthew G. Knepley 34352f856554SMatthew G. Knepley /*@C 34362f856554SMatthew G. Knepley DMPlexGetFaceGeometry - Retrieve the geometric values for a chunk of faces 34372f856554SMatthew G. Knepley 34382f856554SMatthew G. Knepley Input Parameters: 34392f856554SMatthew G. Knepley + dm - The DM 34402f856554SMatthew G. Knepley . fStart - The first face to include 34412f856554SMatthew G. Knepley . fEnd - The first face to exclude 34422f856554SMatthew G. Knepley . faceGeometry - A local vector with face geometry 34432f856554SMatthew G. Knepley - cellGeometry - A local vector with cell geometry 34442f856554SMatthew G. Knepley 34452f856554SMatthew G. Knepley Output Parameters: 34462f856554SMatthew G. Knepley + Nface - The number of faces with field values 34472f856554SMatthew G. Knepley . fgeom - The extract the face centroid and normal 34482f856554SMatthew G. Knepley - vol - The cell volume 34492f856554SMatthew G. Knepley 34502f856554SMatthew G. Knepley Level: developer 34512f856554SMatthew G. Knepley 34522f856554SMatthew G. Knepley .seealso: DMPlexGetCellFields() 34532f856554SMatthew G. Knepley @*/ 34542f856554SMatthew G. Knepley PetscErrorCode DMPlexGetFaceGeometry(DM dm, PetscInt fStart, PetscInt fEnd, Vec faceGeometry, Vec cellGeometry, PetscInt *Nface, PetscFVFaceGeom **fgeom, PetscReal **vol) 34552f856554SMatthew G. Knepley { 34562f856554SMatthew G. Knepley DM dmFace, dmCell; 34572f856554SMatthew G. Knepley DMLabel ghostLabel; 34582f856554SMatthew G. Knepley const PetscScalar *facegeom, *cellgeom; 34592f856554SMatthew G. Knepley PetscInt dim, numFaces = fEnd - fStart, iface, face; 34602f856554SMatthew G. Knepley PetscErrorCode ierr; 34612f856554SMatthew G. Knepley 34622f856554SMatthew G. Knepley PetscFunctionBegin; 34632f856554SMatthew G. Knepley PetscValidHeaderSpecific(dm, DM_CLASSID, 1); 34642f856554SMatthew G. Knepley PetscValidHeaderSpecific(faceGeometry, VEC_CLASSID, 4); 34652f856554SMatthew G. Knepley PetscValidHeaderSpecific(cellGeometry, VEC_CLASSID, 5); 34662f856554SMatthew G. Knepley PetscValidPointer(fgeom, 6); 34672f856554SMatthew G. Knepley PetscValidPointer(vol, 7); 34682f856554SMatthew G. Knepley ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr); 34692f856554SMatthew G. Knepley ierr = DMGetLabel(dm, "ghost", &ghostLabel);CHKERRQ(ierr); 34702f856554SMatthew G. Knepley ierr = VecGetDM(faceGeometry, &dmFace);CHKERRQ(ierr); 34712f856554SMatthew G. Knepley ierr = VecGetArrayRead(faceGeometry, &facegeom);CHKERRQ(ierr); 34722f856554SMatthew G. Knepley ierr = VecGetDM(cellGeometry, &dmCell);CHKERRQ(ierr); 34732f856554SMatthew G. Knepley ierr = VecGetArrayRead(cellGeometry, &cellgeom);CHKERRQ(ierr); 34742f856554SMatthew G. Knepley ierr = PetscMalloc1(numFaces, fgeom);CHKERRQ(ierr); 34752f856554SMatthew G. Knepley ierr = DMGetWorkArray(dm, numFaces*2, MPIU_SCALAR, vol);CHKERRQ(ierr); 34762f856554SMatthew G. Knepley for (face = fStart, iface = 0; face < fEnd; ++face) { 34772f856554SMatthew G. Knepley const PetscInt *cells; 34782f856554SMatthew G. Knepley PetscFVFaceGeom *fg; 34792f856554SMatthew G. Knepley PetscFVCellGeom *cgL, *cgR; 34802f856554SMatthew G. Knepley PetscFVFaceGeom *fgeoml = *fgeom; 34812f856554SMatthew G. Knepley PetscReal *voll = *vol; 34822f856554SMatthew G. Knepley PetscInt ghost, d, nchild, nsupp; 34832f856554SMatthew G. Knepley 34842f856554SMatthew G. Knepley ierr = DMLabelGetValue(ghostLabel, face, &ghost);CHKERRQ(ierr); 34852f856554SMatthew G. Knepley ierr = DMPlexGetSupportSize(dm, face, &nsupp);CHKERRQ(ierr); 34862f856554SMatthew G. Knepley ierr = DMPlexGetTreeChildren(dm, face, &nchild, NULL);CHKERRQ(ierr); 34872f856554SMatthew G. Knepley if (ghost >= 0 || nsupp > 2 || nchild > 0) continue; 34882f856554SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmFace, face, facegeom, &fg);CHKERRQ(ierr); 34892f856554SMatthew G. Knepley ierr = DMPlexGetSupport(dm, face, &cells);CHKERRQ(ierr); 34902f856554SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmCell, cells[0], cellgeom, &cgL);CHKERRQ(ierr); 34912f856554SMatthew G. Knepley ierr = DMPlexPointLocalRead(dmCell, cells[1], cellgeom, &cgR);CHKERRQ(ierr); 34922f856554SMatthew G. Knepley for (d = 0; d < dim; ++d) { 34932f856554SMatthew G. Knepley fgeoml[iface].centroid[d] = fg->centroid[d]; 34942f856554SMatthew G. Knepley fgeoml[iface].normal[d] = fg->normal[d]; 34952f856554SMatthew G. Knepley } 34962f856554SMatthew G. Knepley voll[iface*2+0] = cgL->volume; 34972f856554SMatthew G. Knepley voll[iface*2+1] = cgR->volume; 34982f856554SMatthew G. Knepley ++iface; 34992f856554SMatthew G. Knepley } 35002f856554SMatthew G. Knepley *Nface = iface; 35012f856554SMatthew G. Knepley ierr = VecRestoreArrayRead(faceGeometry, &facegeom);CHKERRQ(ierr); 35022f856554SMatthew G. Knepley ierr = VecRestoreArrayRead(cellGeometry, &cellgeom);CHKERRQ(ierr); 35032f856554SMatthew G. Knepley PetscFunctionReturn(0); 35042f856554SMatthew G. Knepley } 35052f856554SMatthew G. Knepley 35062f856554SMatthew G. Knepley /*@C 35072f856554SMatthew G. Knepley DMPlexRestoreFaceGeometry - Restore the field values values for a chunk of faces 35082f856554SMatthew G. Knepley 35092f856554SMatthew G. Knepley Input Parameters: 35102f856554SMatthew G. Knepley + dm - The DM 35112f856554SMatthew G. Knepley . fStart - The first face to include 35122f856554SMatthew G. Knepley . fEnd - The first face to exclude 35132f856554SMatthew G. Knepley . faceGeometry - A local vector with face geometry 35142f856554SMatthew G. Knepley - cellGeometry - A local vector with cell geometry 35152f856554SMatthew G. Knepley 35162f856554SMatthew G. Knepley Output Parameters: 35172f856554SMatthew G. Knepley + Nface - The number of faces with field values 35182f856554SMatthew G. Knepley . fgeom - The extract the face centroid and normal 35192f856554SMatthew G. Knepley - vol - The cell volume 35202f856554SMatthew G. Knepley 35212f856554SMatthew G. Knepley Level: developer 35222f856554SMatthew G. Knepley 35232f856554SMatthew G. Knepley .seealso: DMPlexGetFaceFields() 35242f856554SMatthew G. Knepley @*/ 35252f856554SMatthew G. Knepley PetscErrorCode DMPlexRestoreFaceGeometry(DM dm, PetscInt fStart, PetscInt fEnd, Vec faceGeometry, Vec cellGeometry, PetscInt *Nface, PetscFVFaceGeom **fgeom, PetscReal **vol) 35262f856554SMatthew G. Knepley { 35272f856554SMatthew G. Knepley PetscErrorCode ierr; 35282f856554SMatthew G. Knepley 35292f856554SMatthew G. Knepley PetscFunctionBegin; 35302f856554SMatthew G. Knepley ierr = PetscFree(*fgeom);CHKERRQ(ierr); 35312f856554SMatthew G. Knepley ierr = DMRestoreWorkArray(dm, 0, MPIU_REAL, vol);CHKERRQ(ierr); 35322f856554SMatthew G. Knepley PetscFunctionReturn(0); 35332f856554SMatthew G. Knepley } 35342f856554SMatthew G. Knepley 3535a1cf66bbSMatthew G. Knepley PetscErrorCode DMSNESGetFEGeom(DMField coordField, IS pointIS, PetscQuadrature quad, PetscBool faceData, PetscFEGeom **geom) 3536a1cf66bbSMatthew G. Knepley { 3537a1cf66bbSMatthew G. Knepley char composeStr[33] = {0}; 3538a1cf66bbSMatthew G. Knepley PetscObjectId id; 3539a1cf66bbSMatthew G. Knepley PetscContainer container; 3540a1cf66bbSMatthew G. Knepley PetscErrorCode ierr; 3541a1cf66bbSMatthew G. Knepley 3542a1cf66bbSMatthew G. Knepley PetscFunctionBegin; 3543a1cf66bbSMatthew G. Knepley ierr = PetscObjectGetId((PetscObject)quad,&id);CHKERRQ(ierr); 3544a1cf66bbSMatthew G. Knepley ierr = PetscSNPrintf(composeStr, 32, "DMSNESGetFEGeom_%x\n", id);CHKERRQ(ierr); 3545a1cf66bbSMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) pointIS, composeStr, (PetscObject *) &container);CHKERRQ(ierr); 3546a1cf66bbSMatthew G. Knepley if (container) { 3547a1cf66bbSMatthew G. Knepley ierr = PetscContainerGetPointer(container, (void **) geom);CHKERRQ(ierr); 3548a1cf66bbSMatthew G. Knepley } else { 3549a1cf66bbSMatthew G. Knepley ierr = DMFieldCreateFEGeom(coordField, pointIS, quad, faceData, geom);CHKERRQ(ierr); 3550a1cf66bbSMatthew G. Knepley ierr = PetscContainerCreate(PETSC_COMM_SELF,&container);CHKERRQ(ierr); 3551a1cf66bbSMatthew G. Knepley ierr = PetscContainerSetPointer(container, (void *) *geom);CHKERRQ(ierr); 3552a1cf66bbSMatthew G. Knepley ierr = PetscContainerSetUserDestroy(container, PetscContainerUserDestroy_PetscFEGeom);CHKERRQ(ierr); 3553a1cf66bbSMatthew G. Knepley ierr = PetscObjectCompose((PetscObject) pointIS, composeStr, (PetscObject) container);CHKERRQ(ierr); 3554a1cf66bbSMatthew G. Knepley ierr = PetscContainerDestroy(&container);CHKERRQ(ierr); 3555a1cf66bbSMatthew G. Knepley } 3556a1cf66bbSMatthew G. Knepley PetscFunctionReturn(0); 3557a1cf66bbSMatthew G. Knepley } 3558a1cf66bbSMatthew G. Knepley 3559a1cf66bbSMatthew G. Knepley PetscErrorCode DMSNESRestoreFEGeom(DMField coordField, IS pointIS, PetscQuadrature quad, PetscBool faceData, PetscFEGeom **geom) 3560a1cf66bbSMatthew G. Knepley { 3561a1cf66bbSMatthew G. Knepley PetscFunctionBegin; 3562a1cf66bbSMatthew G. Knepley *geom = NULL; 3563a1cf66bbSMatthew G. Knepley PetscFunctionReturn(0); 3564a1cf66bbSMatthew G. Knepley } 3565a1cf66bbSMatthew G. Knepley 356692d50984SMatthew G. Knepley PetscErrorCode DMPlexComputeResidual_Patch_Internal(DM dm, PetscSection section, IS cellIS, PetscReal t, Vec locX, Vec locX_t, Vec locF, void *user) 356792d50984SMatthew G. Knepley { 356892d50984SMatthew G. Knepley DM_Plex *mesh = (DM_Plex *) dm->data; 356992d50984SMatthew G. Knepley const char *name = "Residual"; 357092d50984SMatthew G. Knepley DM dmAux = NULL; 357192d50984SMatthew G. Knepley DMLabel ghostLabel = NULL; 357292d50984SMatthew G. Knepley PetscDS prob = NULL; 357392d50984SMatthew G. Knepley PetscDS probAux = NULL; 357492d50984SMatthew G. Knepley PetscBool useFEM = PETSC_FALSE; 357592d50984SMatthew G. Knepley PetscBool isImplicit = (locX_t || t == PETSC_MIN_REAL) ? PETSC_TRUE : PETSC_FALSE; 357692d50984SMatthew G. Knepley DMField coordField = NULL; 3577c0006e53SPatrick Farrell Vec locA; 3578c0006e53SPatrick Farrell PetscScalar *u = NULL, *u_t, *a, *uL = NULL, *uR = NULL; 357992d50984SMatthew G. Knepley IS chunkIS; 358092d50984SMatthew G. Knepley const PetscInt *cells; 358192d50984SMatthew G. Knepley PetscInt cStart, cEnd, numCells; 3582364207b6SKarl Rupp PetscInt Nf, f, totDim, totDimAux, numChunks, cellChunkSize, chunk, fStart, fEnd; 358392d50984SMatthew G. Knepley PetscInt maxDegree = PETSC_MAX_INT; 358492d50984SMatthew G. Knepley PetscQuadrature affineQuad = NULL, *quads = NULL; 358592d50984SMatthew G. Knepley PetscFEGeom *affineGeom = NULL, **geoms = NULL; 358692d50984SMatthew G. Knepley PetscErrorCode ierr; 358792d50984SMatthew G. Knepley 358892d50984SMatthew G. Knepley PetscFunctionBegin; 358992d50984SMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_ResidualFEM,dm,0,0,0);CHKERRQ(ierr); 359092d50984SMatthew G. Knepley /* FEM+FVM */ 359192d50984SMatthew G. Knepley /* 1: Get sizes from dm and dmAux */ 359292d50984SMatthew G. Knepley ierr = DMGetLabel(dm, "ghost", &ghostLabel);CHKERRQ(ierr); 359392d50984SMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 359492d50984SMatthew G. Knepley ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr); 359592d50984SMatthew G. Knepley ierr = PetscDSGetTotalDimension(prob, &totDim);CHKERRQ(ierr); 359692d50984SMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "A", (PetscObject *) &locA);CHKERRQ(ierr); 359792d50984SMatthew G. Knepley if (locA) { 359892d50984SMatthew G. Knepley ierr = VecGetDM(locA, &dmAux);CHKERRQ(ierr); 359992d50984SMatthew G. Knepley ierr = DMGetDS(dmAux, &probAux);CHKERRQ(ierr); 360092d50984SMatthew G. Knepley ierr = PetscDSGetTotalDimension(probAux, &totDimAux);CHKERRQ(ierr); 360192d50984SMatthew G. Knepley } 360292d50984SMatthew G. Knepley /* 2: Get geometric data */ 360392d50984SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 360492d50984SMatthew G. Knepley PetscObject obj; 360592d50984SMatthew G. Knepley PetscClassId id; 360692d50984SMatthew G. Knepley PetscBool fimp; 360792d50984SMatthew G. Knepley 360892d50984SMatthew G. Knepley ierr = PetscDSGetImplicit(prob, f, &fimp);CHKERRQ(ierr); 360992d50984SMatthew G. Knepley if (isImplicit != fimp) continue; 361092d50984SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr); 361192d50984SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 361292d50984SMatthew G. Knepley if (id == PETSCFE_CLASSID) {useFEM = PETSC_TRUE;} 3613364207b6SKarl Rupp if (id == PETSCFV_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Use of FVM with PCPATCH not yet implemented"); 361492d50984SMatthew G. Knepley } 361592d50984SMatthew G. Knepley if (useFEM) { 361692d50984SMatthew G. Knepley ierr = DMGetCoordinateField(dm, &coordField);CHKERRQ(ierr); 361792d50984SMatthew G. Knepley ierr = DMFieldGetDegree(coordField,cellIS,NULL,&maxDegree);CHKERRQ(ierr); 361892d50984SMatthew G. Knepley if (maxDegree <= 1) { 361992d50984SMatthew G. Knepley ierr = DMFieldCreateDefaultQuadrature(coordField,cellIS,&affineQuad);CHKERRQ(ierr); 362092d50984SMatthew G. Knepley if (affineQuad) { 362192d50984SMatthew G. Knepley ierr = DMSNESGetFEGeom(coordField,cellIS,affineQuad,PETSC_FALSE,&affineGeom);CHKERRQ(ierr); 362292d50984SMatthew G. Knepley } 362392d50984SMatthew G. Knepley } else { 362492d50984SMatthew G. Knepley ierr = PetscCalloc2(Nf,&quads,Nf,&geoms);CHKERRQ(ierr); 362592d50984SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 362692d50984SMatthew G. Knepley PetscObject obj; 362792d50984SMatthew G. Knepley PetscClassId id; 362892d50984SMatthew G. Knepley PetscBool fimp; 362992d50984SMatthew G. Knepley 363092d50984SMatthew G. Knepley ierr = PetscDSGetImplicit(prob, f, &fimp);CHKERRQ(ierr); 363192d50984SMatthew G. Knepley if (isImplicit != fimp) continue; 363292d50984SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr); 363392d50984SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 363492d50984SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 363592d50984SMatthew G. Knepley PetscFE fe = (PetscFE) obj; 363692d50984SMatthew G. Knepley 363792d50984SMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &quads[f]);CHKERRQ(ierr); 363892d50984SMatthew G. Knepley ierr = PetscObjectReference((PetscObject)quads[f]);CHKERRQ(ierr); 363992d50984SMatthew G. Knepley ierr = DMSNESGetFEGeom(coordField,cellIS,quads[f],PETSC_FALSE,&geoms[f]);CHKERRQ(ierr); 364092d50984SMatthew G. Knepley } 364192d50984SMatthew G. Knepley } 364292d50984SMatthew G. Knepley } 364392d50984SMatthew G. Knepley } 364492d50984SMatthew G. Knepley /* Loop over chunks */ 364592d50984SMatthew G. Knepley ierr = ISGetPointRange(cellIS, &cStart, &cEnd, &cells);CHKERRQ(ierr); 364692d50984SMatthew G. Knepley ierr = DMPlexGetHeightStratum(dm, 1, &fStart, &fEnd);CHKERRQ(ierr); 364792d50984SMatthew G. Knepley if (useFEM) {ierr = ISCreate(PETSC_COMM_SELF, &chunkIS);CHKERRQ(ierr);} 364892d50984SMatthew G. Knepley numCells = cEnd - cStart; 364992d50984SMatthew G. Knepley numChunks = 1; 365092d50984SMatthew G. Knepley cellChunkSize = numCells/numChunks; 365192d50984SMatthew G. Knepley numChunks = PetscMin(1,numCells); 365292d50984SMatthew G. Knepley for (chunk = 0; chunk < numChunks; ++chunk) { 3653c0006e53SPatrick Farrell PetscScalar *elemVec, *fluxL = NULL, *fluxR = NULL; 3654c0006e53SPatrick Farrell PetscReal *vol = NULL; 3655c0006e53SPatrick Farrell PetscFVFaceGeom *fgeom = NULL; 365692d50984SMatthew G. Knepley PetscInt cS = cStart+chunk*cellChunkSize, cE = PetscMin(cS+cellChunkSize, cEnd), numCells = cE - cS, c; 3657c0006e53SPatrick Farrell PetscInt numFaces = 0; 365892d50984SMatthew G. Knepley 365992d50984SMatthew G. Knepley /* Extract field coefficients */ 366092d50984SMatthew G. Knepley if (useFEM) { 366192d50984SMatthew G. Knepley ierr = ISGetPointSubrange(chunkIS, cS, cE, cells);CHKERRQ(ierr); 366292d50984SMatthew G. Knepley ierr = DMPlexGetCellFields(dm, chunkIS, locX, locX_t, locA, &u, &u_t, &a);CHKERRQ(ierr); 366392d50984SMatthew G. Knepley ierr = DMGetWorkArray(dm, numCells*totDim, MPIU_SCALAR, &elemVec);CHKERRQ(ierr); 3664580bdb30SBarry Smith ierr = PetscArrayzero(elemVec, numCells*totDim);CHKERRQ(ierr); 366592d50984SMatthew G. Knepley } 366692d50984SMatthew G. Knepley /* TODO We will interlace both our field coefficients (u, u_t, uL, uR, etc.) and our output (elemVec, fL, fR). I think this works */ 366792d50984SMatthew G. Knepley /* Loop over fields */ 366892d50984SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 366992d50984SMatthew G. Knepley PetscObject obj; 367092d50984SMatthew G. Knepley PetscClassId id; 367192d50984SMatthew G. Knepley PetscBool fimp; 367292d50984SMatthew G. Knepley PetscInt numChunks, numBatches, batchSize, numBlocks, blockSize, Ne, Nr, offset; 367392d50984SMatthew G. Knepley 367492d50984SMatthew G. Knepley ierr = PetscDSGetImplicit(prob, f, &fimp);CHKERRQ(ierr); 367592d50984SMatthew G. Knepley if (isImplicit != fimp) continue; 367692d50984SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr); 367792d50984SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 367892d50984SMatthew G. Knepley if (id == PETSCFE_CLASSID) { 367992d50984SMatthew G. Knepley PetscFE fe = (PetscFE) obj; 368092d50984SMatthew G. Knepley PetscFEGeom *geom = affineGeom ? affineGeom : geoms[f]; 368192d50984SMatthew G. Knepley PetscFEGeom *chunkGeom = NULL; 368292d50984SMatthew G. Knepley PetscQuadrature quad = affineQuad ? affineQuad : quads[f]; 368392d50984SMatthew G. Knepley PetscInt Nq, Nb; 368492d50984SMatthew G. Knepley 368592d50984SMatthew G. Knepley ierr = PetscFEGetTileSizes(fe, NULL, &numBlocks, NULL, &numBatches);CHKERRQ(ierr); 368692d50984SMatthew G. Knepley ierr = PetscQuadratureGetData(quad, NULL, NULL, &Nq, NULL, NULL);CHKERRQ(ierr); 368792d50984SMatthew G. Knepley ierr = PetscFEGetDimension(fe, &Nb);CHKERRQ(ierr); 368892d50984SMatthew G. Knepley blockSize = Nb; 368992d50984SMatthew G. Knepley batchSize = numBlocks * blockSize; 369092d50984SMatthew G. Knepley ierr = PetscFESetTileSizes(fe, blockSize, numBlocks, batchSize, numBatches);CHKERRQ(ierr); 369192d50984SMatthew G. Knepley numChunks = numCells / (numBatches*batchSize); 369292d50984SMatthew G. Knepley Ne = numChunks*numBatches*batchSize; 369392d50984SMatthew G. Knepley Nr = numCells % (numBatches*batchSize); 369492d50984SMatthew G. Knepley offset = numCells - Nr; 369592d50984SMatthew G. Knepley /* Integrate FE residual to get elemVec (need fields at quadrature points) */ 369692d50984SMatthew G. Knepley /* For FV, I think we use a P0 basis and the cell coefficients (for subdivided cells, we can tweak the basis tabulation to be the indicator function) */ 369792d50984SMatthew G. Knepley ierr = PetscFEGeomGetChunk(geom,0,offset,&chunkGeom);CHKERRQ(ierr); 36984bee2e38SMatthew G. Knepley ierr = PetscFEIntegrateResidual(prob, f, Ne, chunkGeom, u, u_t, probAux, a, t, elemVec);CHKERRQ(ierr); 369992d50984SMatthew G. Knepley ierr = PetscFEGeomGetChunk(geom,offset,numCells,&chunkGeom);CHKERRQ(ierr); 37004bee2e38SMatthew G. Knepley ierr = PetscFEIntegrateResidual(prob, f, Nr, chunkGeom, &u[offset*totDim], u_t ? &u_t[offset*totDim] : NULL, probAux, &a[offset*totDimAux], t, &elemVec[offset*totDim]);CHKERRQ(ierr); 370192d50984SMatthew G. Knepley ierr = PetscFEGeomRestoreChunk(geom,offset,numCells,&chunkGeom);CHKERRQ(ierr); 370292d50984SMatthew G. Knepley } else if (id == PETSCFV_CLASSID) { 370392d50984SMatthew G. Knepley PetscFV fv = (PetscFV) obj; 370492d50984SMatthew G. Knepley 370592d50984SMatthew G. Knepley Ne = numFaces; 370692d50984SMatthew G. Knepley /* Riemann solve over faces (need fields at face centroids) */ 370792d50984SMatthew G. Knepley /* We need to evaluate FE fields at those coordinates */ 370892d50984SMatthew G. Knepley ierr = PetscFVIntegrateRHSFunction(fv, prob, f, Ne, fgeom, vol, uL, uR, fluxL, fluxR);CHKERRQ(ierr); 3709ff1e0c32SBarry Smith } else SETERRQ1(PetscObjectComm((PetscObject) dm), PETSC_ERR_ARG_WRONG, "Unknown discretization type for field %D", f); 371092d50984SMatthew G. Knepley } 371192d50984SMatthew G. Knepley /* Loop over domain */ 371292d50984SMatthew G. Knepley if (useFEM) { 371392d50984SMatthew G. Knepley /* Add elemVec to locX */ 371492d50984SMatthew G. Knepley for (c = cS; c < cE; ++c) { 371592d50984SMatthew G. Knepley const PetscInt cell = cells ? cells[c] : c; 371692d50984SMatthew G. Knepley const PetscInt cind = c - cStart; 371792d50984SMatthew G. Knepley 371892d50984SMatthew G. Knepley if (mesh->printFEM > 1) {ierr = DMPrintCellVector(cell, name, totDim, &elemVec[cind*totDim]);CHKERRQ(ierr);} 371992d50984SMatthew G. Knepley if (ghostLabel) { 372092d50984SMatthew G. Knepley PetscInt ghostVal; 372192d50984SMatthew G. Knepley 372292d50984SMatthew G. Knepley ierr = DMLabelGetValue(ghostLabel,cell,&ghostVal);CHKERRQ(ierr); 372392d50984SMatthew G. Knepley if (ghostVal > 0) continue; 372492d50984SMatthew G. Knepley } 372592d50984SMatthew G. Knepley ierr = DMPlexVecSetClosure(dm, section, locF, cell, &elemVec[cind*totDim], ADD_ALL_VALUES);CHKERRQ(ierr); 372692d50984SMatthew G. Knepley } 372792d50984SMatthew G. Knepley } 372892d50984SMatthew G. Knepley /* Handle time derivative */ 372992d50984SMatthew G. Knepley if (locX_t) { 373092d50984SMatthew G. Knepley PetscScalar *x_t, *fa; 373192d50984SMatthew G. Knepley 373292d50984SMatthew G. Knepley ierr = VecGetArray(locF, &fa);CHKERRQ(ierr); 373392d50984SMatthew G. Knepley ierr = VecGetArray(locX_t, &x_t);CHKERRQ(ierr); 373492d50984SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 373592d50984SMatthew G. Knepley PetscFV fv; 373692d50984SMatthew G. Knepley PetscObject obj; 373792d50984SMatthew G. Knepley PetscClassId id; 373892d50984SMatthew G. Knepley PetscInt pdim, d; 373992d50984SMatthew G. Knepley 374092d50984SMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, f, &obj);CHKERRQ(ierr); 374192d50984SMatthew G. Knepley ierr = PetscObjectGetClassId(obj, &id);CHKERRQ(ierr); 374292d50984SMatthew G. Knepley if (id != PETSCFV_CLASSID) continue; 374392d50984SMatthew G. Knepley fv = (PetscFV) obj; 374492d50984SMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &pdim);CHKERRQ(ierr); 374592d50984SMatthew G. Knepley for (c = cS; c < cE; ++c) { 374692d50984SMatthew G. Knepley const PetscInt cell = cells ? cells[c] : c; 374792d50984SMatthew G. Knepley PetscScalar *u_t, *r; 374892d50984SMatthew G. Knepley 374992d50984SMatthew G. Knepley if (ghostLabel) { 375092d50984SMatthew G. Knepley PetscInt ghostVal; 375192d50984SMatthew G. Knepley 375292d50984SMatthew G. Knepley ierr = DMLabelGetValue(ghostLabel, cell, &ghostVal);CHKERRQ(ierr); 375392d50984SMatthew G. Knepley if (ghostVal > 0) continue; 375492d50984SMatthew G. Knepley } 375592d50984SMatthew G. Knepley ierr = DMPlexPointLocalFieldRead(dm, cell, f, x_t, &u_t);CHKERRQ(ierr); 375692d50984SMatthew G. Knepley ierr = DMPlexPointLocalFieldRef(dm, cell, f, fa, &r);CHKERRQ(ierr); 375792d50984SMatthew G. Knepley for (d = 0; d < pdim; ++d) r[d] += u_t[d]; 375892d50984SMatthew G. Knepley } 375992d50984SMatthew G. Knepley } 376092d50984SMatthew G. Knepley ierr = VecRestoreArray(locX_t, &x_t);CHKERRQ(ierr); 376192d50984SMatthew G. Knepley ierr = VecRestoreArray(locF, &fa);CHKERRQ(ierr); 376292d50984SMatthew G. Knepley } 376392d50984SMatthew G. Knepley if (useFEM) { 376492d50984SMatthew G. Knepley ierr = DMPlexRestoreCellFields(dm, chunkIS, locX, locX_t, locA, &u, &u_t, &a);CHKERRQ(ierr); 376592d50984SMatthew G. Knepley ierr = DMRestoreWorkArray(dm, numCells*totDim, MPIU_SCALAR, &elemVec);CHKERRQ(ierr); 376692d50984SMatthew G. Knepley } 376792d50984SMatthew G. Knepley } 376892d50984SMatthew G. Knepley if (useFEM) {ierr = ISDestroy(&chunkIS);CHKERRQ(ierr);} 376992d50984SMatthew G. Knepley ierr = ISRestorePointRange(cellIS, &cStart, &cEnd, &cells);CHKERRQ(ierr); 377092d50984SMatthew G. Knepley /* TODO Could include boundary residual here (see DMPlexComputeResidual_Internal) */ 377192d50984SMatthew G. Knepley if (useFEM) { 377292d50984SMatthew G. Knepley if (maxDegree <= 1) { 377392d50984SMatthew G. Knepley ierr = DMSNESRestoreFEGeom(coordField,cellIS,affineQuad,PETSC_FALSE,&affineGeom);CHKERRQ(ierr); 377492d50984SMatthew G. Knepley ierr = PetscQuadratureDestroy(&affineQuad);CHKERRQ(ierr); 377592d50984SMatthew G. Knepley } else { 377692d50984SMatthew G. Knepley for (f = 0; f < Nf; ++f) { 377792d50984SMatthew G. Knepley ierr = DMSNESRestoreFEGeom(coordField,cellIS,quads[f],PETSC_FALSE,&geoms[f]);CHKERRQ(ierr); 377892d50984SMatthew G. Knepley ierr = PetscQuadratureDestroy(&quads[f]);CHKERRQ(ierr); 377992d50984SMatthew G. Knepley } 378092d50984SMatthew G. Knepley ierr = PetscFree2(quads,geoms);CHKERRQ(ierr); 378192d50984SMatthew G. Knepley } 378292d50984SMatthew G. Knepley } 378392d50984SMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_ResidualFEM,dm,0,0,0);CHKERRQ(ierr); 378492d50984SMatthew G. Knepley PetscFunctionReturn(0); 378592d50984SMatthew G. Knepley } 378692d50984SMatthew G. Knepley 3787a1cf66bbSMatthew G. Knepley /* 3788a1cf66bbSMatthew G. Knepley We always assemble JacP, and if the matrix is different from Jac and two different sets of point functions are provided, we also assemble Jac 3789a1cf66bbSMatthew G. Knepley 3790a1cf66bbSMatthew G. Knepley X - The local solution vector 3791a1cf66bbSMatthew G. Knepley X_t - The local solution time derviative vector, or NULL 3792a1cf66bbSMatthew G. Knepley */ 3793a1cf66bbSMatthew G. Knepley PetscErrorCode DMPlexComputeJacobian_Patch_Internal(DM dm, PetscSection section, PetscSection globalSection, IS cellIS, 3794a1cf66bbSMatthew G. Knepley PetscReal t, PetscReal X_tShift, Vec X, Vec X_t, Mat Jac, Mat JacP, void *ctx) 3795a1cf66bbSMatthew G. Knepley { 3796a1cf66bbSMatthew G. Knepley DM_Plex *mesh = (DM_Plex *) dm->data; 3797a1cf66bbSMatthew G. Knepley const char *name = "Jacobian", *nameP = "JacobianPre"; 3798a1cf66bbSMatthew G. Knepley DM dmAux = NULL; 3799a1cf66bbSMatthew G. Knepley PetscDS prob, probAux = NULL; 3800a1cf66bbSMatthew G. Knepley PetscSection sectionAux = NULL; 3801a1cf66bbSMatthew G. Knepley Vec A; 3802a1cf66bbSMatthew G. Knepley DMField coordField; 3803a1cf66bbSMatthew G. Knepley PetscFEGeom *cgeomFEM; 3804a1cf66bbSMatthew G. Knepley PetscQuadrature qGeom = NULL; 3805a1cf66bbSMatthew G. Knepley Mat J = Jac, JP = JacP; 3806a1cf66bbSMatthew G. Knepley PetscScalar *work, *u = NULL, *u_t = NULL, *a = NULL, *elemMat = NULL, *elemMatP = NULL, *elemMatD = NULL; 38079b2fc754SMatthew G. Knepley PetscBool hasJac, hasPrec, hasDyn, assembleJac, isMatIS, isMatISP, *isFE, hasFV = PETSC_FALSE; 3808a1cf66bbSMatthew G. Knepley const PetscInt *cells; 38099b2fc754SMatthew G. Knepley PetscInt Nf, fieldI, fieldJ, maxDegree, numCells, cStart, cEnd, numChunks, chunkSize, chunk, totDim, totDimAux = 0, sz, wsz, off = 0, offCell = 0; 3810a1cf66bbSMatthew G. Knepley PetscErrorCode ierr; 3811a1cf66bbSMatthew G. Knepley 3812a1cf66bbSMatthew G. Knepley PetscFunctionBegin; 3813a1cf66bbSMatthew G. Knepley CHKMEMQ; 3814a1cf66bbSMatthew G. Knepley ierr = ISGetLocalSize(cellIS, &numCells);CHKERRQ(ierr); 3815a1cf66bbSMatthew G. Knepley ierr = ISGetPointRange(cellIS, &cStart, &cEnd, &cells);CHKERRQ(ierr); 3816a1cf66bbSMatthew G. Knepley ierr = PetscLogEventBegin(DMPLEX_JacobianFEM,dm,0,0,0);CHKERRQ(ierr); 3817a1cf66bbSMatthew G. Knepley ierr = DMGetDS(dm, &prob);CHKERRQ(ierr); 3818a1cf66bbSMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "dmAux", (PetscObject *) &dmAux);CHKERRQ(ierr); 3819a1cf66bbSMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "A", (PetscObject *) &A);CHKERRQ(ierr); 3820a1cf66bbSMatthew G. Knepley if (dmAux) { 382192fd8e1eSJed Brown ierr = DMGetLocalSection(dmAux, §ionAux);CHKERRQ(ierr); 3822a1cf66bbSMatthew G. Knepley ierr = DMGetDS(dmAux, &probAux);CHKERRQ(ierr); 3823a1cf66bbSMatthew G. Knepley } 3824a1cf66bbSMatthew G. Knepley /* Get flags */ 3825a1cf66bbSMatthew G. Knepley ierr = PetscDSGetNumFields(prob, &Nf);CHKERRQ(ierr); 3826a1cf66bbSMatthew G. Knepley ierr = DMGetWorkArray(dm, Nf, MPIU_BOOL, &isFE);CHKERRQ(ierr); 3827a1cf66bbSMatthew G. Knepley for (fieldI = 0; fieldI < Nf; ++fieldI) { 3828a1cf66bbSMatthew G. Knepley PetscObject disc; 3829a1cf66bbSMatthew G. Knepley PetscClassId id; 3830a1cf66bbSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, fieldI, &disc);CHKERRQ(ierr); 3831a1cf66bbSMatthew G. Knepley ierr = PetscObjectGetClassId(disc, &id);CHKERRQ(ierr); 3832a1cf66bbSMatthew G. Knepley if (id == PETSCFE_CLASSID) {isFE[fieldI] = PETSC_TRUE;} 3833a1cf66bbSMatthew G. Knepley else if (id == PETSCFV_CLASSID) {hasFV = PETSC_TRUE; isFE[fieldI] = PETSC_FALSE;} 3834a1cf66bbSMatthew G. Knepley } 3835a1cf66bbSMatthew G. Knepley ierr = PetscDSHasJacobian(prob, &hasJac);CHKERRQ(ierr); 3836a1cf66bbSMatthew G. Knepley ierr = PetscDSHasJacobianPreconditioner(prob, &hasPrec);CHKERRQ(ierr); 3837a1cf66bbSMatthew G. Knepley ierr = PetscDSHasDynamicJacobian(prob, &hasDyn);CHKERRQ(ierr); 3838a1cf66bbSMatthew G. Knepley assembleJac = hasJac && hasPrec && (Jac != JacP) ? PETSC_TRUE : PETSC_FALSE; 3839a1cf66bbSMatthew G. Knepley hasDyn = hasDyn && (X_tShift != 0.0) ? PETSC_TRUE : PETSC_FALSE; 3840a1cf66bbSMatthew G. Knepley ierr = PetscObjectTypeCompare((PetscObject) Jac, MATIS, &isMatIS);CHKERRQ(ierr); 3841a1cf66bbSMatthew G. Knepley ierr = PetscObjectTypeCompare((PetscObject) JacP, MATIS, &isMatISP);CHKERRQ(ierr); 3842a1cf66bbSMatthew G. Knepley /* Setup input data and temp arrays (should be DMGetWorkArray) */ 3843a1cf66bbSMatthew G. Knepley if (isMatISP || isMatISP) {ierr = DMPlexGetSubdomainSection(dm, &globalSection);CHKERRQ(ierr);} 3844a1cf66bbSMatthew G. Knepley if (isMatIS) {ierr = MatISGetLocalMat(Jac, &J);CHKERRQ(ierr);} 3845a1cf66bbSMatthew G. Knepley if (isMatISP) {ierr = MatISGetLocalMat(JacP, &JP);CHKERRQ(ierr);} 3846a1cf66bbSMatthew G. Knepley if (hasFV) {ierr = MatSetOption(JP, MAT_IGNORE_ZERO_ENTRIES, PETSC_TRUE);CHKERRQ(ierr);} /* No allocated space for FV stuff, so ignore the zero entries */ 3847a1cf66bbSMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "dmAux", (PetscObject *) &dmAux);CHKERRQ(ierr); 3848a1cf66bbSMatthew G. Knepley ierr = PetscObjectQuery((PetscObject) dm, "A", (PetscObject *) &A);CHKERRQ(ierr); 3849a1cf66bbSMatthew G. Knepley ierr = PetscDSGetTotalDimension(prob, &totDim);CHKERRQ(ierr); 3850a1cf66bbSMatthew G. Knepley if (probAux) {ierr = PetscDSGetTotalDimension(probAux, &totDimAux);CHKERRQ(ierr);} 3851a1cf66bbSMatthew G. Knepley CHKMEMQ; 3852a1cf66bbSMatthew G. Knepley /* Compute batch sizes */ 3853a1cf66bbSMatthew G. Knepley if (isFE[0]) { 3854a1cf66bbSMatthew G. Knepley PetscFE fe; 3855a1cf66bbSMatthew G. Knepley PetscQuadrature q; 3856a1cf66bbSMatthew G. Knepley PetscInt numQuadPoints, numBatches, batchSize, numBlocks, blockSize, Nb; 3857a1cf66bbSMatthew G. Knepley 3858a1cf66bbSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, 0, (PetscObject *) &fe);CHKERRQ(ierr); 3859a1cf66bbSMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &q);CHKERRQ(ierr); 3860a1cf66bbSMatthew G. Knepley ierr = PetscQuadratureGetData(q, NULL, NULL, &numQuadPoints, NULL, NULL);CHKERRQ(ierr); 3861a1cf66bbSMatthew G. Knepley ierr = PetscFEGetDimension(fe, &Nb);CHKERRQ(ierr); 3862a1cf66bbSMatthew G. Knepley ierr = PetscFEGetTileSizes(fe, NULL, &numBlocks, NULL, &numBatches);CHKERRQ(ierr); 3863a1cf66bbSMatthew G. Knepley blockSize = Nb*numQuadPoints; 3864a1cf66bbSMatthew G. Knepley batchSize = numBlocks * blockSize; 3865a1cf66bbSMatthew G. Knepley chunkSize = numBatches * batchSize; 3866a1cf66bbSMatthew G. Knepley numChunks = numCells / chunkSize + numCells % chunkSize; 3867a1cf66bbSMatthew G. Knepley ierr = PetscFESetTileSizes(fe, blockSize, numBlocks, batchSize, numBatches);CHKERRQ(ierr); 3868a1cf66bbSMatthew G. Knepley } else { 3869a1cf66bbSMatthew G. Knepley chunkSize = numCells; 3870a1cf66bbSMatthew G. Knepley numChunks = 1; 3871a1cf66bbSMatthew G. Knepley } 3872a1cf66bbSMatthew G. Knepley /* Get work space */ 3873a1cf66bbSMatthew G. Knepley wsz = (((X?1:0) + (X_t?1:0) + (dmAux?1:0))*totDim + ((hasJac?1:0) + (hasPrec?1:0) + (hasDyn?1:0))*totDim*totDim)*chunkSize; 3874a1cf66bbSMatthew G. Knepley ierr = DMGetWorkArray(dm, wsz, MPIU_SCALAR, &work);CHKERRQ(ierr); 3875580bdb30SBarry Smith ierr = PetscArrayzero(work, wsz);CHKERRQ(ierr); 3876a1cf66bbSMatthew G. Knepley off = 0; 3877a1cf66bbSMatthew G. Knepley u = X ? (sz = chunkSize*totDim, off += sz, work+off-sz) : NULL; 3878a1cf66bbSMatthew G. Knepley u_t = X_t ? (sz = chunkSize*totDim, off += sz, work+off-sz) : NULL; 3879a1cf66bbSMatthew G. Knepley a = dmAux ? (sz = chunkSize*totDimAux, off += sz, work+off-sz) : NULL; 3880a1cf66bbSMatthew G. Knepley elemMat = hasJac ? (sz = chunkSize*totDim*totDim, off += sz, work+off-sz) : NULL; 3881a1cf66bbSMatthew G. Knepley elemMatP = hasPrec ? (sz = chunkSize*totDim*totDim, off += sz, work+off-sz) : NULL; 3882a1cf66bbSMatthew G. Knepley elemMatD = hasDyn ? (sz = chunkSize*totDim*totDim, off += sz, work+off-sz) : NULL; 3883a1cf66bbSMatthew G. Knepley if (off != wsz) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Error is workspace size %D should be %D", off, wsz); 3884a1cf66bbSMatthew G. Knepley /* Setup geometry */ 3885a1cf66bbSMatthew G. Knepley ierr = DMGetCoordinateField(dm, &coordField);CHKERRQ(ierr); 38869b2fc754SMatthew G. Knepley ierr = DMFieldGetDegree(coordField, cellIS, NULL, &maxDegree);CHKERRQ(ierr); 38879b2fc754SMatthew G. Knepley if (maxDegree <= 1) {ierr = DMFieldCreateDefaultQuadrature(coordField, cellIS, &qGeom);CHKERRQ(ierr);} 3888a1cf66bbSMatthew G. Knepley if (!qGeom) { 3889a1cf66bbSMatthew G. Knepley PetscFE fe; 3890a1cf66bbSMatthew G. Knepley 3891a1cf66bbSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, 0, (PetscObject *) &fe);CHKERRQ(ierr); 3892a1cf66bbSMatthew G. Knepley ierr = PetscFEGetQuadrature(fe, &qGeom);CHKERRQ(ierr); 3893a1cf66bbSMatthew G. Knepley ierr = PetscObjectReference((PetscObject) qGeom);CHKERRQ(ierr); 3894a1cf66bbSMatthew G. Knepley } 3895a1cf66bbSMatthew G. Knepley ierr = DMSNESGetFEGeom(coordField, cellIS, qGeom, PETSC_FALSE, &cgeomFEM);CHKERRQ(ierr); 3896a1cf66bbSMatthew G. Knepley /* Compute volume integrals */ 3897a1cf66bbSMatthew G. Knepley if (assembleJac) {ierr = MatZeroEntries(J);CHKERRQ(ierr);} 3898a1cf66bbSMatthew G. Knepley ierr = MatZeroEntries(JP);CHKERRQ(ierr); 3899a1cf66bbSMatthew G. Knepley for (chunk = 0; chunk < numChunks; ++chunk, offCell += chunkSize) { 3900a1cf66bbSMatthew G. Knepley const PetscInt Ncell = PetscMin(chunkSize, numCells - offCell); 3901a1cf66bbSMatthew G. Knepley PetscInt c; 3902a1cf66bbSMatthew G. Knepley 3903a1cf66bbSMatthew G. Knepley /* Extract values */ 3904a1cf66bbSMatthew G. Knepley for (c = 0; c < Ncell; ++c) { 3905a1cf66bbSMatthew G. Knepley const PetscInt cell = cells ? cells[c+offCell] : c+offCell; 3906a1cf66bbSMatthew G. Knepley PetscScalar *x = NULL, *x_t = NULL; 3907a1cf66bbSMatthew G. Knepley PetscInt i; 3908a1cf66bbSMatthew G. Knepley 3909a1cf66bbSMatthew G. Knepley if (X) { 3910a1cf66bbSMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, section, X, cell, NULL, &x);CHKERRQ(ierr); 3911a1cf66bbSMatthew G. Knepley for (i = 0; i < totDim; ++i) u[c*totDim+i] = x[i]; 3912a1cf66bbSMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, section, X, cell, NULL, &x);CHKERRQ(ierr); 3913a1cf66bbSMatthew G. Knepley } 3914a1cf66bbSMatthew G. Knepley if (X_t) { 3915a1cf66bbSMatthew G. Knepley ierr = DMPlexVecGetClosure(dm, section, X_t, cell, NULL, &x_t);CHKERRQ(ierr); 3916a1cf66bbSMatthew G. Knepley for (i = 0; i < totDim; ++i) u_t[c*totDim+i] = x_t[i]; 3917a1cf66bbSMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dm, section, X_t, cell, NULL, &x_t);CHKERRQ(ierr); 3918a1cf66bbSMatthew G. Knepley } 3919a1cf66bbSMatthew G. Knepley if (dmAux) { 3920a1cf66bbSMatthew G. Knepley ierr = DMPlexVecGetClosure(dmAux, sectionAux, A, cell, NULL, &x);CHKERRQ(ierr); 3921a1cf66bbSMatthew G. Knepley for (i = 0; i < totDimAux; ++i) a[c*totDimAux+i] = x[i]; 3922a1cf66bbSMatthew G. Knepley ierr = DMPlexVecRestoreClosure(dmAux, sectionAux, A, cell, NULL, &x);CHKERRQ(ierr); 3923a1cf66bbSMatthew G. Knepley } 3924a1cf66bbSMatthew G. Knepley } 3925a1cf66bbSMatthew G. Knepley CHKMEMQ; 3926a1cf66bbSMatthew G. Knepley for (fieldI = 0; fieldI < Nf; ++fieldI) { 3927a1cf66bbSMatthew G. Knepley PetscFE fe; 3928a1cf66bbSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, fieldI, (PetscObject *) &fe);CHKERRQ(ierr); 3929a1cf66bbSMatthew G. Knepley for (fieldJ = 0; fieldJ < Nf; ++fieldJ) { 39304bee2e38SMatthew G. Knepley if (hasJac) {ierr = PetscFEIntegrateJacobian(prob, PETSCFE_JACOBIAN, fieldI, fieldJ, Ncell, cgeomFEM, u, u_t, probAux, a, t, X_tShift, elemMat);CHKERRQ(ierr);} 39314bee2e38SMatthew G. Knepley if (hasPrec) {ierr = PetscFEIntegrateJacobian(prob, PETSCFE_JACOBIAN_PRE, fieldI, fieldJ, Ncell, cgeomFEM, u, u_t, probAux, a, t, X_tShift, elemMatP);CHKERRQ(ierr);} 39324bee2e38SMatthew G. Knepley if (hasDyn) {ierr = PetscFEIntegrateJacobian(prob, PETSCFE_JACOBIAN_DYN, fieldI, fieldJ, Ncell, cgeomFEM, u, u_t, probAux, a, t, X_tShift, elemMatD);CHKERRQ(ierr);} 3933a1cf66bbSMatthew G. Knepley } 3934a1cf66bbSMatthew G. Knepley /* For finite volume, add the identity */ 3935a1cf66bbSMatthew G. Knepley if (!isFE[fieldI]) { 3936a1cf66bbSMatthew G. Knepley PetscFV fv; 3937a1cf66bbSMatthew G. Knepley PetscInt eOffset = 0, Nc, fc, foff; 3938a1cf66bbSMatthew G. Knepley 3939a1cf66bbSMatthew G. Knepley ierr = PetscDSGetFieldOffset(prob, fieldI, &foff);CHKERRQ(ierr); 3940a1cf66bbSMatthew G. Knepley ierr = PetscDSGetDiscretization(prob, fieldI, (PetscObject *) &fv);CHKERRQ(ierr); 3941a1cf66bbSMatthew G. Knepley ierr = PetscFVGetNumComponents(fv, &Nc);CHKERRQ(ierr); 3942a1cf66bbSMatthew G. Knepley for (c = 0; c < chunkSize; ++c, eOffset += totDim*totDim) { 3943a1cf66bbSMatthew G. Knepley for (fc = 0; fc < Nc; ++fc) { 3944a1cf66bbSMatthew G. Knepley const PetscInt i = foff + fc; 3945a1cf66bbSMatthew G. Knepley if (hasJac) {elemMat [eOffset+i*totDim+i] = 1.0;} 3946a1cf66bbSMatthew G. Knepley if (hasPrec) {elemMatP[eOffset+i*totDim+i] = 1.0;} 3947a1cf66bbSMatthew G. Knepley } 3948a1cf66bbSMatthew G. Knepley } 3949a1cf66bbSMatthew G. Knepley } 3950a1cf66bbSMatthew G. Knepley } 3951a1cf66bbSMatthew G. Knepley CHKMEMQ; 3952a1cf66bbSMatthew G. Knepley /* Add contribution from X_t */ 3953a1cf66bbSMatthew G. Knepley if (hasDyn) {for (c = 0; c < chunkSize*totDim*totDim; ++c) elemMat[c] += X_tShift*elemMatD[c];} 3954a1cf66bbSMatthew G. Knepley /* Insert values into matrix */ 3955a1cf66bbSMatthew G. Knepley for (c = 0; c < Ncell; ++c) { 3956a1cf66bbSMatthew G. Knepley const PetscInt cell = cells ? cells[c+offCell] : c+offCell; 3957a1cf66bbSMatthew G. Knepley if (mesh->printFEM > 1) { 3958a1cf66bbSMatthew G. Knepley if (hasJac) {ierr = DMPrintCellMatrix(cell, name, totDim, totDim, &elemMat[(c-cStart)*totDim*totDim]);CHKERRQ(ierr);} 3959a1cf66bbSMatthew G. Knepley if (hasPrec) {ierr = DMPrintCellMatrix(cell, nameP, totDim, totDim, &elemMatP[(c-cStart)*totDim*totDim]);CHKERRQ(ierr);} 3960a1cf66bbSMatthew G. Knepley } 3961a1cf66bbSMatthew G. Knepley if (assembleJac) {ierr = DMPlexMatSetClosure(dm, section, globalSection, Jac, cell, &elemMat[(c-cStart)*totDim*totDim], ADD_VALUES);CHKERRQ(ierr);} 3962a1cf66bbSMatthew G. Knepley ierr = DMPlexMatSetClosure(dm, section, globalSection, JP, cell, &elemMat[(c-cStart)*totDim*totDim], ADD_VALUES);CHKERRQ(ierr); 3963a1cf66bbSMatthew G. Knepley } 3964a1cf66bbSMatthew G. Knepley CHKMEMQ; 3965a1cf66bbSMatthew G. Knepley } 3966a1cf66bbSMatthew G. Knepley /* Cleanup */ 3967a1cf66bbSMatthew G. Knepley ierr = DMSNESRestoreFEGeom(coordField, cellIS, qGeom, PETSC_FALSE, &cgeomFEM);CHKERRQ(ierr); 3968a1cf66bbSMatthew G. Knepley ierr = PetscQuadratureDestroy(&qGeom);CHKERRQ(ierr); 3969a1cf66bbSMatthew G. Knepley if (hasFV) {ierr = MatSetOption(JacP, MAT_IGNORE_ZERO_ENTRIES, PETSC_FALSE);CHKERRQ(ierr);} 3970a1cf66bbSMatthew G. Knepley ierr = DMRestoreWorkArray(dm, Nf, MPIU_BOOL, &isFE);CHKERRQ(ierr); 3971a1cf66bbSMatthew G. Knepley ierr = DMRestoreWorkArray(dm, ((1 + (X_t?1:0) + (dmAux?1:0))*totDim + ((hasJac?1:0) + (hasPrec?1:0) + (hasDyn?1:0))*totDim*totDim)*chunkSize, MPIU_SCALAR, &work);CHKERRQ(ierr); 3972a1cf66bbSMatthew G. Knepley /* Compute boundary integrals */ 3973a1cf66bbSMatthew G. Knepley /* ierr = DMPlexComputeBdJacobian_Internal(dm, X, X_t, t, X_tShift, Jac, JacP, ctx);CHKERRQ(ierr); */ 3974a1cf66bbSMatthew G. Knepley /* Assemble matrix */ 3975a1cf66bbSMatthew G. Knepley if (assembleJac) {ierr = MatAssemblyBegin(Jac, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);ierr = MatAssemblyEnd(Jac, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);} 3976a1cf66bbSMatthew G. Knepley ierr = MatAssemblyBegin(JacP, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);ierr = MatAssemblyEnd(JacP, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 3977a1cf66bbSMatthew G. Knepley ierr = PetscLogEventEnd(DMPLEX_JacobianFEM,dm,0,0,0);CHKERRQ(ierr); 3978a1cf66bbSMatthew G. Knepley CHKMEMQ; 3979a1cf66bbSMatthew G. Knepley PetscFunctionReturn(0); 3980a1cf66bbSMatthew G. Knepley } 3981