xref: /petsc/src/dm/dt/dualspace/impls/lagrange/dspacelagrange.c (revision 6ff1568834b9ee5da48e4fa02ac7a4227caec5c3)
120cf1dd8SToby Isaac #include <petsc/private/petscfeimpl.h> /*I "petscfe.h" I*/
220cf1dd8SToby Isaac #include <petscdmplex.h>
33f27d899SToby Isaac #include <petscblaslapack.h>
43f27d899SToby Isaac 
53f27d899SToby Isaac PetscErrorCode DMPlexGetTransitiveClosure_Internal(DM, PetscInt, PetscInt, PetscBool, PetscInt *, PetscInt *[]);
63f27d899SToby Isaac 
73f27d899SToby Isaac struct _n_Petsc1DNodeFamily
83f27d899SToby Isaac {
93f27d899SToby Isaac   PetscInt         refct;
103f27d899SToby Isaac   PetscDTNodeType  nodeFamily;
113f27d899SToby Isaac   PetscReal        gaussJacobiExp;
123f27d899SToby Isaac   PetscInt         nComputed;
133f27d899SToby Isaac   PetscReal      **nodesets;
143f27d899SToby Isaac   PetscBool        endpoints;
153f27d899SToby Isaac };
163f27d899SToby Isaac 
173f27d899SToby Isaac static PetscErrorCode Petsc1DNodeFamilyCreate(PetscDTNodeType family, PetscReal gaussJacobiExp, PetscBool endpoints, Petsc1DNodeFamily *nf)
183f27d899SToby Isaac {
193f27d899SToby Isaac   Petsc1DNodeFamily f;
203f27d899SToby Isaac   PetscErrorCode ierr;
213f27d899SToby Isaac 
223f27d899SToby Isaac   PetscFunctionBegin;
233f27d899SToby Isaac   ierr = PetscNew(&f);CHKERRQ(ierr);
243f27d899SToby Isaac   switch (family) {
253f27d899SToby Isaac   case PETSCDTNODES_GAUSSJACOBI:
263f27d899SToby Isaac   case PETSCDTNODES_EQUISPACED:
273f27d899SToby Isaac     f->nodeFamily = family;
283f27d899SToby Isaac     break;
293f27d899SToby Isaac   default: SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unknown 1D node family");
303f27d899SToby Isaac   }
313f27d899SToby Isaac   f->endpoints = endpoints;
323f27d899SToby Isaac   f->gaussJacobiExp = 0.;
333f27d899SToby Isaac   if (family == PETSCDTNODES_GAUSSJACOBI) {
343f27d899SToby Isaac     if (gaussJacobiExp <= -1.) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Gauss-Jacobi exponent must be > -1.\n");
353f27d899SToby Isaac     f->gaussJacobiExp = gaussJacobiExp;
363f27d899SToby Isaac   }
373f27d899SToby Isaac   f->refct = 1;
383f27d899SToby Isaac   *nf = f;
393f27d899SToby Isaac   PetscFunctionReturn(0);
403f27d899SToby Isaac }
413f27d899SToby Isaac 
423f27d899SToby Isaac static PetscErrorCode Petsc1DNodeFamilyReference(Petsc1DNodeFamily nf)
433f27d899SToby Isaac {
443f27d899SToby Isaac   PetscFunctionBegin;
453f27d899SToby Isaac   if (nf) nf->refct++;
463f27d899SToby Isaac   PetscFunctionReturn(0);
473f27d899SToby Isaac }
483f27d899SToby Isaac 
493f27d899SToby Isaac static PetscErrorCode Petsc1DNodeFamilyDestroy(Petsc1DNodeFamily *nf) {
503f27d899SToby Isaac   PetscInt       i, nc;
513f27d899SToby Isaac   PetscErrorCode ierr;
523f27d899SToby Isaac 
533f27d899SToby Isaac   PetscFunctionBegin;
543f27d899SToby Isaac   if (!(*nf)) PetscFunctionReturn(0);
553f27d899SToby Isaac   if (--(*nf)->refct > 0) {
563f27d899SToby Isaac     *nf = NULL;
573f27d899SToby Isaac     PetscFunctionReturn(0);
583f27d899SToby Isaac   }
593f27d899SToby Isaac   nc = (*nf)->nComputed;
603f27d899SToby Isaac   for (i = 0; i < nc; i++) {
613f27d899SToby Isaac     ierr = PetscFree((*nf)->nodesets[i]);CHKERRQ(ierr);
623f27d899SToby Isaac   }
633f27d899SToby Isaac   ierr = PetscFree((*nf)->nodesets);CHKERRQ(ierr);
643f27d899SToby Isaac   ierr = PetscFree(*nf);CHKERRQ(ierr);
653f27d899SToby Isaac   *nf = NULL;
663f27d899SToby Isaac   PetscFunctionReturn(0);
673f27d899SToby Isaac }
683f27d899SToby Isaac 
693f27d899SToby Isaac static PetscErrorCode Petsc1DNodeFamilyGetNodeSets(Petsc1DNodeFamily f, PetscInt degree, PetscReal ***nodesets)
703f27d899SToby Isaac {
713f27d899SToby Isaac   PetscInt       nc;
723f27d899SToby Isaac   PetscErrorCode ierr;
733f27d899SToby Isaac 
743f27d899SToby Isaac   PetscFunctionBegin;
753f27d899SToby Isaac   nc = f->nComputed;
763f27d899SToby Isaac   if (degree >= nc) {
773f27d899SToby Isaac     PetscInt    i, j;
783f27d899SToby Isaac     PetscReal **new_nodesets;
793f27d899SToby Isaac     PetscReal  *w;
803f27d899SToby Isaac 
813f27d899SToby Isaac     ierr = PetscMalloc1(degree + 1, &new_nodesets);CHKERRQ(ierr);
823f27d899SToby Isaac     ierr = PetscArraycpy(new_nodesets, f->nodesets, nc);CHKERRQ(ierr);
833f27d899SToby Isaac     ierr = PetscFree(f->nodesets);CHKERRQ(ierr);
843f27d899SToby Isaac     f->nodesets = new_nodesets;
853f27d899SToby Isaac     ierr = PetscMalloc1(degree + 1, &w);CHKERRQ(ierr);
863f27d899SToby Isaac     for (i = nc; i < degree + 1; i++) {
873f27d899SToby Isaac       ierr = PetscMalloc1(i + 1, &(f->nodesets[i]));CHKERRQ(ierr);
883f27d899SToby Isaac       if (!i) {
893f27d899SToby Isaac         f->nodesets[i][0] = 0.5;
903f27d899SToby Isaac       } else {
913f27d899SToby Isaac         switch (f->nodeFamily) {
923f27d899SToby Isaac         case PETSCDTNODES_EQUISPACED:
933f27d899SToby Isaac           if (f->endpoints) {
943f27d899SToby Isaac             for (j = 0; j <= i; j++) f->nodesets[i][j] = (PetscReal) j / (PetscReal) i;
953f27d899SToby Isaac           } else {
963f27d899SToby Isaac             for (j = 0; j <= i; j++) f->nodesets[i][j] = ((PetscReal) j + 0.5) / ((PetscReal) i + 1.);
973f27d899SToby Isaac           }
983f27d899SToby Isaac           break;
993f27d899SToby Isaac         case PETSCDTNODES_GAUSSJACOBI:
1003f27d899SToby Isaac           if (f->endpoints) {
1013f27d899SToby Isaac             ierr = PetscDTGaussLobattoJacobiQuadrature(i + 1, 0., 1., f->gaussJacobiExp, f->gaussJacobiExp, f->nodesets[i], w);CHKERRQ(ierr);
1023f27d899SToby Isaac           } else {
1033f27d899SToby Isaac             ierr = PetscDTGaussJacobiQuadrature(i + 1, 0., 1., f->gaussJacobiExp, f->gaussJacobiExp, f->nodesets[i], w);CHKERRQ(ierr);
1043f27d899SToby Isaac           }
1053f27d899SToby Isaac           break;
1063f27d899SToby Isaac         default: SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Unknown 1D node family");
1073f27d899SToby Isaac         }
1083f27d899SToby Isaac       }
1093f27d899SToby Isaac     }
1103f27d899SToby Isaac     ierr = PetscFree(w);CHKERRQ(ierr);
1113f27d899SToby Isaac     f->nComputed = degree + 1;
1123f27d899SToby Isaac   }
1133f27d899SToby Isaac   *nodesets = f->nodesets;
1143f27d899SToby Isaac   PetscFunctionReturn(0);
1153f27d899SToby Isaac }
1163f27d899SToby Isaac 
1173f27d899SToby Isaac static PetscErrorCode PetscNodeRecursive_Internal(PetscInt dim, PetscInt degree, PetscReal **nodesets, PetscInt tup[], PetscReal node[])
1183f27d899SToby Isaac {
1193f27d899SToby Isaac   PetscReal w;
1203f27d899SToby Isaac   PetscInt i, j;
1213f27d899SToby Isaac   PetscErrorCode ierr;
1223f27d899SToby Isaac 
1233f27d899SToby Isaac   PetscFunctionBeginHot;
1243f27d899SToby Isaac   w = 0.;
1253f27d899SToby Isaac   if (dim == 1) {
1263f27d899SToby Isaac     node[0] = nodesets[degree][tup[0]];
1273f27d899SToby Isaac     node[1] = nodesets[degree][tup[1]];
1283f27d899SToby Isaac   } else {
1293f27d899SToby Isaac     for (i = 0; i < dim + 1; i++) node[i] = 0.;
1303f27d899SToby Isaac     for (i = 0; i < dim + 1; i++) {
1313f27d899SToby Isaac       PetscReal wi = nodesets[degree][degree-tup[i]];
1323f27d899SToby Isaac 
1333f27d899SToby Isaac       for (j = 0; j < dim+1; j++) tup[dim+1+j] = tup[j+(j>=i)];
1343f27d899SToby Isaac       ierr = PetscNodeRecursive_Internal(dim-1,degree-tup[i],nodesets,&tup[dim+1],&node[dim+1]);CHKERRQ(ierr);
1353f27d899SToby Isaac       for (j = 0; j < dim+1; j++) node[j+(j>=i)] += wi * node[dim+1+j];
1363f27d899SToby Isaac       w += wi;
1373f27d899SToby Isaac     }
1383f27d899SToby Isaac     for (i = 0; i < dim+1; i++) node[i] /= w;
1393f27d899SToby Isaac   }
1403f27d899SToby Isaac   PetscFunctionReturn(0);
1413f27d899SToby Isaac }
1423f27d899SToby Isaac 
1433f27d899SToby Isaac /* compute simplex nodes for the biunit simplex from the 1D node family */
1443f27d899SToby Isaac static PetscErrorCode Petsc1DNodeFamilyComputeSimplexNodes(Petsc1DNodeFamily f, PetscInt dim, PetscInt degree, PetscReal points[])
1453f27d899SToby Isaac {
1463f27d899SToby Isaac   PetscInt      *tup;
1473f27d899SToby Isaac   PetscInt       k;
1483f27d899SToby Isaac   PetscInt       npoints;
1493f27d899SToby Isaac   PetscReal    **nodesets = NULL;
1503f27d899SToby Isaac   PetscInt       worksize;
1513f27d899SToby Isaac   PetscReal     *nodework;
1523f27d899SToby Isaac   PetscInt      *tupwork;
1533f27d899SToby Isaac   PetscErrorCode ierr;
1543f27d899SToby Isaac 
1553f27d899SToby Isaac   PetscFunctionBegin;
1563f27d899SToby Isaac   if (dim < 0) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must have non-negative dimension\n");
1573f27d899SToby Isaac   if (degree < 0) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Must have non-negative degree\n");
1583f27d899SToby Isaac   if (!dim) PetscFunctionReturn(0);
1593f27d899SToby Isaac   ierr = PetscCalloc1(dim+2, &tup);CHKERRQ(ierr);
1603f27d899SToby Isaac   k = 0;
1613f27d899SToby Isaac   ierr = PetscDTBinomialInt(degree + dim, dim, &npoints);CHKERRQ(ierr);
1623f27d899SToby Isaac   ierr = Petsc1DNodeFamilyGetNodeSets(f, degree, &nodesets);CHKERRQ(ierr);
1633f27d899SToby Isaac   worksize = ((dim + 2) * (dim + 3)) / 2;
1643f27d899SToby Isaac   ierr = PetscMalloc2(worksize, &nodework, worksize, &tupwork);CHKERRQ(ierr);
1653f27d899SToby Isaac   for (k = 0; k < npoints; k++) {
1663f27d899SToby Isaac     PetscInt i;
1673f27d899SToby Isaac 
1683f27d899SToby Isaac     tup[0] = degree;
1693f27d899SToby Isaac     for (i = 0; i < dim; i++) {
1703f27d899SToby Isaac       tup[0] -= tup[i+1];
1713f27d899SToby Isaac     }
1723f27d899SToby Isaac     switch(f->nodeFamily) {
1733f27d899SToby Isaac     case PETSCDTNODES_EQUISPACED:
1743f27d899SToby Isaac       if (f->endpoints) {
1753f27d899SToby Isaac         for (i = 0; i < dim; i++) {
1763f27d899SToby Isaac           points[dim*k + i] = (PetscReal) tup[i+1] / (PetscReal) degree;
1773f27d899SToby Isaac         }
1783f27d899SToby Isaac       } else {
1793f27d899SToby Isaac         for (i = 0; i < dim; i++) {
1803f27d899SToby Isaac           points[dim*k + i] = ((PetscReal) tup[i+1] + 1./(dim+1.)) / (PetscReal) (degree + 1.);
1813f27d899SToby Isaac         }
1823f27d899SToby Isaac       }
1833f27d899SToby Isaac       break;
1843f27d899SToby Isaac     default:
1853f27d899SToby Isaac       for (i = 0; i < dim + 1; i++) tupwork[i] = tup[i];
1863f27d899SToby Isaac       ierr = PetscNodeRecursive_Internal(dim, degree, nodesets, tupwork, nodework);CHKERRQ(ierr);
1873f27d899SToby Isaac       for (i = 0; i < dim; i++) points[dim*k + i] = nodework[i + 1];
1883f27d899SToby Isaac       break;
1893f27d899SToby Isaac     }
1903f27d899SToby Isaac     ierr = PetscDualSpaceLatticePointLexicographic_Internal(dim, degree, &tup[1]);CHKERRQ(ierr);
1913f27d899SToby Isaac   }
1923f27d899SToby Isaac   /* map from unit simplex to biunit simplex */
1933f27d899SToby Isaac   for (k = 0; k < npoints * dim; k++) points[k] = points[k] * 2. - 1.;
1943f27d899SToby Isaac   ierr = PetscFree2(nodework, tupwork);CHKERRQ(ierr);
1953f27d899SToby Isaac   ierr = PetscFree(tup);
1963f27d899SToby Isaac   PetscFunctionReturn(0);
1973f27d899SToby Isaac }
1983f27d899SToby Isaac 
1993f27d899SToby Isaac struct _n_PetscLagNodeIndices
2003f27d899SToby Isaac {
2013f27d899SToby Isaac   PetscInt   refct;
2023f27d899SToby Isaac   PetscInt   nodeIdxDim;
2033f27d899SToby Isaac   PetscInt   nodeVecDim;
2043f27d899SToby Isaac   PetscInt   nNodes;
2053f27d899SToby Isaac   PetscInt  *nodeIdx;      /* for each node an index of size nodeIdxDim */
2063f27d899SToby Isaac   PetscReal *nodeVec;      /* for each node a vector of size nodeVecDim */
2073f27d899SToby Isaac   PetscInt  *perm;         /* if these are vertices, perm takes DMPlex point index to closure order;
2083f27d899SToby Isaac                               if these are nodes, perm lists nodes in index revlex order */
2093f27d899SToby Isaac };
2103f27d899SToby Isaac 
2113f27d899SToby Isaac PetscErrorCode PetscLagNodeIndicesGetData_Internal(PetscLagNodeIndices ni, PetscInt *nodeIdxDim, PetscInt *nodeVecDim, PetscInt *nNodes, const PetscInt *nodeIdx[], const PetscReal *nodeVec[])
2123f27d899SToby Isaac {
2133f27d899SToby Isaac   PetscFunctionBegin;
2143f27d899SToby Isaac   *nodeIdxDim = ni->nodeIdxDim;
2153f27d899SToby Isaac   *nodeVecDim = ni->nodeVecDim;
2163f27d899SToby Isaac   *nNodes = ni->nNodes;
2173f27d899SToby Isaac   *nodeIdx = ni->nodeIdx;
2183f27d899SToby Isaac   *nodeVec = ni->nodeVec;
2193f27d899SToby Isaac   PetscFunctionReturn(0);
2203f27d899SToby Isaac }
2213f27d899SToby Isaac 
2223f27d899SToby Isaac static PetscErrorCode PetscLagNodeIndicesReference(PetscLagNodeIndices ni)
2233f27d899SToby Isaac {
2243f27d899SToby Isaac   PetscFunctionBegin;
2253f27d899SToby Isaac   if (ni) ni->refct++;
2263f27d899SToby Isaac   PetscFunctionReturn(0);
2273f27d899SToby Isaac }
2283f27d899SToby Isaac 
2293f27d899SToby Isaac static PetscErrorCode PetscLagNodeIndicesDestroy(PetscLagNodeIndices *ni) {
2303f27d899SToby Isaac   PetscErrorCode ierr;
2313f27d899SToby Isaac 
2323f27d899SToby Isaac   PetscFunctionBegin;
2333f27d899SToby Isaac   if (!(*ni)) PetscFunctionReturn(0);
2343f27d899SToby Isaac   if (--(*ni)->refct > 0) {
2353f27d899SToby Isaac     *ni = NULL;
2363f27d899SToby Isaac     PetscFunctionReturn(0);
2373f27d899SToby Isaac   }
2383f27d899SToby Isaac   ierr = PetscFree((*ni)->nodeIdx);CHKERRQ(ierr);
2393f27d899SToby Isaac   ierr = PetscFree((*ni)->nodeVec);CHKERRQ(ierr);
2403f27d899SToby Isaac   ierr = PetscFree((*ni)->perm);CHKERRQ(ierr);
2413f27d899SToby Isaac   ierr = PetscFree(*ni);CHKERRQ(ierr);
2423f27d899SToby Isaac   *ni = NULL;
2433f27d899SToby Isaac   PetscFunctionReturn(0);
2443f27d899SToby Isaac }
2453f27d899SToby Isaac 
2463f27d899SToby Isaac /* The vertex indices were written as though the vertices were in revlex order
2473f27d899SToby Isaac  * wrt coordinates.  To understand the effect of different symmetries, we need
2483f27d899SToby Isaac  * them to be in closure order.  We also need a permutation that takes point index
2493f27d899SToby Isaac  * to closure number */
2503f27d899SToby Isaac static PetscErrorCode PetscLagNodeIndicesComputeVertexOrder(DM dm, PetscLagNodeIndices ni, PetscBool sortIdx)
2513f27d899SToby Isaac {
2523f27d899SToby Isaac   PetscInt        v, w, vStart, vEnd, c, d;
2533f27d899SToby Isaac   PetscInt        nVerts;
2543f27d899SToby Isaac   PetscInt        closureSize = 0;
2553f27d899SToby Isaac   PetscInt       *closure = NULL;
2563f27d899SToby Isaac   PetscInt       *closureOrder;
2573f27d899SToby Isaac   PetscInt       *invClosureOrder;
2583f27d899SToby Isaac   PetscInt       *revlexOrder;
2593f27d899SToby Isaac   PetscInt       *newNodeIdx;
2603f27d899SToby Isaac   PetscInt        dim;
2613f27d899SToby Isaac   Vec             coordVec;
2623f27d899SToby Isaac   const PetscScalar *coords;
2633f27d899SToby Isaac   PetscErrorCode  ierr;
2643f27d899SToby Isaac 
2653f27d899SToby Isaac   PetscFunctionBegin;
2663f27d899SToby Isaac   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
2673f27d899SToby Isaac   ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
2683f27d899SToby Isaac   nVerts = vEnd - vStart;
2693f27d899SToby Isaac   ierr = PetscMalloc1(nVerts, &closureOrder);CHKERRQ(ierr);
2703f27d899SToby Isaac   ierr = PetscMalloc1(nVerts, &invClosureOrder);CHKERRQ(ierr);
2713f27d899SToby Isaac   ierr = PetscMalloc1(nVerts, &revlexOrder);CHKERRQ(ierr);
2723f27d899SToby Isaac   if (sortIdx) {
2733f27d899SToby Isaac     PetscInt nodeIdxDim = ni->nodeIdxDim;
2743f27d899SToby Isaac     PetscInt *idxOrder;
2753f27d899SToby Isaac 
2763f27d899SToby Isaac     ierr = PetscMalloc1(nVerts * nodeIdxDim, &newNodeIdx);CHKERRQ(ierr);
2773f27d899SToby Isaac     ierr = PetscMalloc1(nVerts, &idxOrder);CHKERRQ(ierr);
2783f27d899SToby Isaac     for (v = 0; v < nVerts; v++) idxOrder[v] = v;
2793f27d899SToby Isaac     for (v = 0; v < nVerts; v++) {
2803f27d899SToby Isaac       for (w = v + 1; w < nVerts; w++) {
2813f27d899SToby Isaac         const PetscInt *iv = &(ni->nodeIdx[idxOrder[v] * nodeIdxDim]);
2823f27d899SToby Isaac         const PetscInt *iw = &(ni->nodeIdx[idxOrder[w] * nodeIdxDim]);
2833f27d899SToby Isaac         PetscInt diff = 0;
2843f27d899SToby Isaac 
2853f27d899SToby Isaac         for (d = nodeIdxDim - 1; d >= 0; d--) if ((diff = (iv[d] - iw[d]))) break;
2863f27d899SToby Isaac         if (diff > 0) {
2873f27d899SToby Isaac           PetscInt swap = idxOrder[v];
2883f27d899SToby Isaac 
2893f27d899SToby Isaac           idxOrder[v] = idxOrder[w];
2903f27d899SToby Isaac           idxOrder[w] = swap;
2913f27d899SToby Isaac         }
2923f27d899SToby Isaac       }
2933f27d899SToby Isaac     }
2943f27d899SToby Isaac     for (v = 0; v < nVerts; v++) {
2953f27d899SToby Isaac       for (d = 0; d < nodeIdxDim; d++) {
2963f27d899SToby Isaac         newNodeIdx[v * ni->nodeIdxDim + d] = ni->nodeIdx[idxOrder[v] * nodeIdxDim + d];
2973f27d899SToby Isaac       }
2983f27d899SToby Isaac     }
2993f27d899SToby Isaac     ierr = PetscFree(ni->nodeIdx);CHKERRQ(ierr);
3003f27d899SToby Isaac     ni->nodeIdx = newNodeIdx;
3013f27d899SToby Isaac     newNodeIdx = NULL;
3023f27d899SToby Isaac     ierr = PetscFree(idxOrder);CHKERRQ(ierr);
3033f27d899SToby Isaac   }
3043f27d899SToby Isaac   ierr = DMPlexGetTransitiveClosure(dm, 0, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
3053f27d899SToby Isaac   c = closureSize - nVerts;
3063f27d899SToby Isaac   for (v = 0; v < nVerts; v++) closureOrder[v] = closure[2 * (c + v)] - vStart;
3073f27d899SToby Isaac   for (v = 0; v < nVerts; v++) invClosureOrder[closureOrder[v]] = v;
3083f27d899SToby Isaac   ierr = DMPlexRestoreTransitiveClosure(dm, 0, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
3093f27d899SToby Isaac   ierr = DMGetCoordinatesLocal(dm, &coordVec);CHKERRQ(ierr);
3103f27d899SToby Isaac   ierr = VecGetArrayRead(coordVec, &coords);CHKERRQ(ierr);
3113f27d899SToby Isaac   /* bubble sort closure vertices by coordinates in revlex order */
3123f27d899SToby Isaac   for (v = 0; v < nVerts; v++) revlexOrder[v] = v;
3133f27d899SToby Isaac   for (v = 0; v < nVerts; v++) {
3143f27d899SToby Isaac     for (w = v + 1; w < nVerts; w++) {
3153f27d899SToby Isaac       const PetscScalar *cv = &coords[closureOrder[revlexOrder[v]] * dim];
3163f27d899SToby Isaac       const PetscScalar *cw = &coords[closureOrder[revlexOrder[w]] * dim];
3173f27d899SToby Isaac       PetscReal diff = 0;
3183f27d899SToby Isaac 
3193f27d899SToby Isaac       for (d = dim - 1; d >= 0; d--) if ((diff = PetscRealPart(cv[d] - cw[d])) != 0.) break;
3203f27d899SToby Isaac       if (diff > 0.) {
3213f27d899SToby Isaac         PetscInt swap = revlexOrder[v];
3223f27d899SToby Isaac 
3233f27d899SToby Isaac         revlexOrder[v] = revlexOrder[w];
3243f27d899SToby Isaac         revlexOrder[w] = swap;
3253f27d899SToby Isaac       }
3263f27d899SToby Isaac     }
3273f27d899SToby Isaac   }
3283f27d899SToby Isaac   ierr = VecRestoreArrayRead(coordVec, &coords);CHKERRQ(ierr);
3293f27d899SToby Isaac   ierr = PetscMalloc1(ni->nodeIdxDim * nVerts, &newNodeIdx);CHKERRQ(ierr);
3303f27d899SToby Isaac   /* reorder nodeIdx to be in closure order */
3313f27d899SToby Isaac   for (v = 0; v < nVerts; v++) {
3323f27d899SToby Isaac     for (d = 0; d < ni->nodeIdxDim; d++) {
3333f27d899SToby Isaac       newNodeIdx[revlexOrder[v] * ni->nodeIdxDim + d] = ni->nodeIdx[v * ni->nodeIdxDim + d];
3343f27d899SToby Isaac     }
3353f27d899SToby Isaac   }
3363f27d899SToby Isaac   ierr = PetscFree(ni->nodeIdx);CHKERRQ(ierr);
3373f27d899SToby Isaac   ni->nodeIdx = newNodeIdx;
3383f27d899SToby Isaac   ni->perm = invClosureOrder;
3393f27d899SToby Isaac   ierr = PetscFree(revlexOrder);CHKERRQ(ierr);
3403f27d899SToby Isaac   ierr = PetscFree(closureOrder);CHKERRQ(ierr);
3413f27d899SToby Isaac   PetscFunctionReturn(0);
3423f27d899SToby Isaac }
3433f27d899SToby Isaac 
3443f27d899SToby Isaac static PetscErrorCode PetscLagNodeIndicesCreateSimplexVertices(DM dm, PetscLagNodeIndices *nodeIndices)
3453f27d899SToby Isaac {
3463f27d899SToby Isaac   PetscLagNodeIndices ni;
3473f27d899SToby Isaac   PetscInt       dim, d;
3483f27d899SToby Isaac 
3493f27d899SToby Isaac   PetscErrorCode ierr;
3503f27d899SToby Isaac 
3513f27d899SToby Isaac   PetscFunctionBegin;
3523f27d899SToby Isaac   ierr = PetscNew(&ni);CHKERRQ(ierr);
3533f27d899SToby Isaac   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
3543f27d899SToby Isaac   ni->nodeIdxDim = dim + 1;
3553f27d899SToby Isaac   ni->nodeVecDim = 0;
3563f27d899SToby Isaac   ni->nNodes = dim + 1;
3573f27d899SToby Isaac   ni->refct = 1;
3583f27d899SToby Isaac   ierr = PetscCalloc1((dim + 1)*(dim + 1), &(ni->nodeIdx));CHKERRQ(ierr);
3593f27d899SToby Isaac   for (d = 0; d < dim + 1; d++) ni->nodeIdx[d*(dim + 2)] = 1;
3603f27d899SToby Isaac   ierr = PetscLagNodeIndicesComputeVertexOrder(dm, ni, PETSC_FALSE);CHKERRQ(ierr);
3613f27d899SToby Isaac   *nodeIndices = ni;
3623f27d899SToby Isaac   PetscFunctionReturn(0);
3633f27d899SToby Isaac }
3643f27d899SToby Isaac 
3653f27d899SToby Isaac static PetscErrorCode PetscLagNodeIndicesCreateTensorVertices(DM dm, PetscLagNodeIndices facetni, PetscLagNodeIndices *nodeIndices)
3663f27d899SToby Isaac {
3673f27d899SToby Isaac   PetscLagNodeIndices ni;
3683f27d899SToby Isaac   PetscInt       nodeIdxDim, subNodeIdxDim = facetni->nodeIdxDim;
3693f27d899SToby Isaac   PetscInt       nVerts, nSubVerts = facetni->nNodes;
3703f27d899SToby Isaac   PetscInt       dim, d, e, f, g;
3713f27d899SToby Isaac 
3723f27d899SToby Isaac   PetscErrorCode ierr;
3733f27d899SToby Isaac 
3743f27d899SToby Isaac   PetscFunctionBegin;
3753f27d899SToby Isaac   ierr = PetscNew(&ni);CHKERRQ(ierr);
3763f27d899SToby Isaac   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
3773f27d899SToby Isaac   ni->nodeIdxDim = nodeIdxDim = subNodeIdxDim + 2;
3783f27d899SToby Isaac   ni->nodeVecDim = 0;
3793f27d899SToby Isaac   ni->nNodes = nVerts = 2 * nSubVerts;
3803f27d899SToby Isaac   ni->refct = 1;
3813f27d899SToby Isaac   ierr = PetscCalloc1(nodeIdxDim * nVerts, &(ni->nodeIdx));CHKERRQ(ierr);
3823f27d899SToby Isaac   for (f = 0, d = 0; d < 2; d++) {
3833f27d899SToby Isaac     for (e = 0; e < nSubVerts; e++, f++) {
3843f27d899SToby Isaac       for (g = 0; g < subNodeIdxDim; g++) {
3853f27d899SToby Isaac         ni->nodeIdx[f * nodeIdxDim + g] = facetni->nodeIdx[e * subNodeIdxDim + g];
3863f27d899SToby Isaac       }
3873f27d899SToby Isaac       ni->nodeIdx[f * nodeIdxDim + subNodeIdxDim] = (1 - d);
3883f27d899SToby Isaac       ni->nodeIdx[f * nodeIdxDim + subNodeIdxDim + 1] = d;
3893f27d899SToby Isaac     }
3903f27d899SToby Isaac   }
3913f27d899SToby Isaac   ierr = PetscLagNodeIndicesComputeVertexOrder(dm, ni, PETSC_TRUE);CHKERRQ(ierr);
3923f27d899SToby Isaac   *nodeIndices = ni;
3933f27d899SToby Isaac   PetscFunctionReturn(0);
3943f27d899SToby Isaac }
3953f27d899SToby Isaac 
3963f27d899SToby Isaac static PetscErrorCode PetscLagNodeIndicesPushForward(DM dm, PetscLagNodeIndices vert, PetscInt p, PetscLagNodeIndices vertp, PetscLagNodeIndices nodep, PetscInt ornt, PetscInt formDegree, PetscInt pfNodeIdx[], PetscReal pfNodeVec[])
3973f27d899SToby Isaac {
3983f27d899SToby Isaac   PetscInt       *closureVerts;
3993f27d899SToby Isaac   PetscInt        closureSize = 0;
4003f27d899SToby Isaac   PetscInt       *closure = NULL;
4013f27d899SToby Isaac   PetscInt        dim, pdim, c, i, j, k, n, v, vStart, vEnd;
4023f27d899SToby Isaac   PetscInt        nSubVert = vertp->nNodes;
4033f27d899SToby Isaac   PetscInt        nodeIdxDim = vert->nodeIdxDim;
4043f27d899SToby Isaac   PetscInt        subNodeIdxDim = vertp->nodeIdxDim;
4053f27d899SToby Isaac   PetscInt        nNodes = nodep->nNodes;
4063f27d899SToby Isaac   const PetscInt  *vertIdx = vert->nodeIdx;
4073f27d899SToby Isaac   const PetscInt  *subVertIdx = vertp->nodeIdx;
4083f27d899SToby Isaac   const PetscInt  *nodeIdx = nodep->nodeIdx;
4093f27d899SToby Isaac   const PetscReal *nodeVec = nodep->nodeVec;
4103f27d899SToby Isaac   PetscReal       *J, *Jstar;
4113f27d899SToby Isaac   PetscReal       detJ;
4123f27d899SToby Isaac   PetscInt        depth, pdepth, Nk, pNk;
4133f27d899SToby Isaac   Vec             coordVec;
4143f27d899SToby Isaac   PetscScalar      *newCoords = NULL;
4153f27d899SToby Isaac   const PetscScalar *oldCoords = NULL;
4163f27d899SToby Isaac   PetscErrorCode  ierr;
4173f27d899SToby Isaac 
4183f27d899SToby Isaac   PetscFunctionBegin;
4193f27d899SToby Isaac   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
4203f27d899SToby Isaac   ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
4213f27d899SToby Isaac   ierr = DMGetCoordinatesLocal(dm, &coordVec);CHKERRQ(ierr);
4223f27d899SToby Isaac   ierr = DMPlexGetPointDepth(dm, p, &pdepth);CHKERRQ(ierr);
4233f27d899SToby Isaac   pdim = pdepth != depth ? pdepth != 0 ? pdepth : 0 : dim;
4243f27d899SToby Isaac   ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
4253f27d899SToby Isaac   ierr = DMGetWorkArray(dm, nSubVert, MPIU_INT, &closureVerts);CHKERRQ(ierr);
4263f27d899SToby Isaac   ierr = DMPlexGetTransitiveClosure_Internal(dm, p, ornt, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
4273f27d899SToby Isaac   c = closureSize - nSubVert;
4283f27d899SToby Isaac   /* we want which cell closure indices the closure of this point corresponds to */
4293f27d899SToby Isaac   for (v = 0; v < nSubVert; v++) closureVerts[v] = vert->perm[closure[2 * (c + v)] - vStart];
4303f27d899SToby Isaac   ierr = DMPlexRestoreTransitiveClosure(dm, p, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
4313f27d899SToby Isaac   /* push forward indices */
4323f27d899SToby Isaac   for (i = 0; i < nodeIdxDim; i++) { /* for every component of the target index space */
4333f27d899SToby Isaac     /* check if this is a component that all vertices around this point have in common */
4343f27d899SToby Isaac     for (j = 1; j < nSubVert; j++) {
4353f27d899SToby Isaac       if (vertIdx[closureVerts[j] * nodeIdxDim + i] != vertIdx[closureVerts[0] * nodeIdxDim + i]) break;
4363f27d899SToby Isaac     }
4373f27d899SToby Isaac     if (j == nSubVert) { /* all vertices have this component in common, directly copy to output */
4383f27d899SToby Isaac       PetscInt val = vertIdx[closureVerts[0] * nodeIdxDim + i];
4393f27d899SToby Isaac       for (n = 0; n < nNodes; n++) pfNodeIdx[n * nodeIdxDim + i] = val;
4403f27d899SToby Isaac     } else {
4413f27d899SToby Isaac       PetscInt subi = -1;
4423f27d899SToby Isaac       /* there must be a component in vertp that looks the same */
4433f27d899SToby Isaac       for (k = 0; k < subNodeIdxDim; k++) {
4443f27d899SToby Isaac         for (j = 0; j < nSubVert; j++) {
4453f27d899SToby Isaac           if (vertIdx[closureVerts[j] * nodeIdxDim + i] != subVertIdx[j * subNodeIdxDim + k]) break;
4463f27d899SToby Isaac         }
4473f27d899SToby Isaac         if (j == nSubVert) {
4483f27d899SToby Isaac           subi = k;
4493f27d899SToby Isaac           break;
4503f27d899SToby Isaac         }
4513f27d899SToby Isaac       }
4523f27d899SToby Isaac       if (subi < 0) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Did not find matching coordinate\n");
4533f27d899SToby Isaac       for (n = 0; n < nNodes; n++) pfNodeIdx[n * nodeIdxDim + i] = nodeIdx[n * subNodeIdxDim + subi];
4543f27d899SToby Isaac     }
4553f27d899SToby Isaac   }
4563f27d899SToby Isaac   /* push forward vectors */
4573f27d899SToby Isaac   ierr = DMGetWorkArray(dm, dim * dim, MPIU_REAL, &J);CHKERRQ(ierr);
4583f27d899SToby Isaac   if (ornt != 0) {
4593f27d899SToby Isaac     PetscInt        closureSize2 = 0;
4603f27d899SToby Isaac     PetscInt       *closure2 = NULL;
4613f27d899SToby Isaac 
4623f27d899SToby Isaac     ierr = DMPlexGetTransitiveClosure_Internal(dm, p, 0, PETSC_TRUE, &closureSize2, &closure2);CHKERRQ(ierr);
4633f27d899SToby Isaac     ierr = PetscMalloc1(dim * nSubVert, &newCoords);CHKERRQ(ierr);
4643f27d899SToby Isaac     ierr = VecGetArrayRead(coordVec, &oldCoords);CHKERRQ(ierr);
4653f27d899SToby Isaac     for (v = 0; v < nSubVert; v++) {
4663f27d899SToby Isaac       PetscInt d;
4673f27d899SToby Isaac       for (d = 0; d < dim; d++) {
4683f27d899SToby Isaac         newCoords[(closure2[2 * (c + v)] - vStart) * dim + d] = oldCoords[closureVerts[v] * dim + d];
4693f27d899SToby Isaac       }
4703f27d899SToby Isaac     }
4713f27d899SToby Isaac     ierr = VecRestoreArrayRead(coordVec, &oldCoords);CHKERRQ(ierr);
4723f27d899SToby Isaac     ierr = DMPlexRestoreTransitiveClosure(dm, p, PETSC_TRUE, &closureSize2, &closure2);CHKERRQ(ierr);
4733f27d899SToby Isaac     ierr = VecPlaceArray(coordVec, newCoords);CHKERRQ(ierr);
4743f27d899SToby Isaac   }
4753f27d899SToby Isaac   ierr = DMPlexComputeCellGeometryAffineFEM(dm, p, NULL, J, NULL, &detJ);CHKERRQ(ierr);
4763f27d899SToby Isaac   if (ornt != 0) {
4773f27d899SToby Isaac     ierr = VecResetArray(coordVec);CHKERRQ(ierr);
4783f27d899SToby Isaac     ierr = PetscFree(newCoords);CHKERRQ(ierr);
4793f27d899SToby Isaac   }
4803f27d899SToby Isaac   ierr = DMRestoreWorkArray(dm, nSubVert, MPIU_INT, &closureVerts);CHKERRQ(ierr);
4813f27d899SToby Isaac   /* compactify */
4823f27d899SToby Isaac   for (i = 0; i < dim; i++) for (j = 0; j < pdim; j++) J[i * pdim + j] = J[i * dim + j];
4833f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(formDegree), &Nk);CHKERRQ(ierr);
4843f27d899SToby Isaac   ierr = PetscDTBinomialInt(pdim, PetscAbsInt(formDegree), &pNk);CHKERRQ(ierr);
4853f27d899SToby Isaac   ierr = DMGetWorkArray(dm, pNk * Nk, MPIU_REAL, &Jstar);CHKERRQ(ierr);
4863f27d899SToby Isaac   ierr = PetscDTAltVPullbackMatrix(pdim, dim, J, formDegree, Jstar);CHKERRQ(ierr);
4873f27d899SToby Isaac   for (n = 0; n < nNodes; n++) {
4883f27d899SToby Isaac     for (i = 0; i < Nk; i++) {
4893f27d899SToby Isaac       PetscReal val = 0.;
4903f27d899SToby Isaac       for (j = 0; j < pNk; j++) val += nodeVec[n * pNk + j] * Jstar[j * pNk + i];
4913f27d899SToby Isaac       pfNodeVec[n * Nk + i] = val;
4923f27d899SToby Isaac     }
4933f27d899SToby Isaac   }
4943f27d899SToby Isaac   ierr = DMRestoreWorkArray(dm, pNk * Nk, MPIU_REAL, &Jstar);CHKERRQ(ierr);
4953f27d899SToby Isaac   ierr = DMRestoreWorkArray(dm, dim * dim, MPIU_REAL, &J);CHKERRQ(ierr);
4963f27d899SToby Isaac   PetscFunctionReturn(0);
4973f27d899SToby Isaac }
4983f27d899SToby Isaac 
4993f27d899SToby Isaac static PetscErrorCode PetscLagNodeIndicesTensor(PetscLagNodeIndices tracei, PetscInt dimT, PetscInt kT, PetscLagNodeIndices fiberi, PetscInt dimF, PetscInt kF, PetscLagNodeIndices *nodeIndices)
5003f27d899SToby Isaac {
5013f27d899SToby Isaac   PetscInt       dim = dimT + dimF;
5023f27d899SToby Isaac   PetscInt       nodeIdxDim, nNodes;
5033f27d899SToby Isaac   PetscInt       formDegree = kT + kF;
5043f27d899SToby Isaac   PetscInt       Nk, NkT, NkF;
5053f27d899SToby Isaac   PetscInt       MkT, MkF;
5063f27d899SToby Isaac   PetscLagNodeIndices ni;
5073f27d899SToby Isaac   PetscInt       i, j, l;
5083f27d899SToby Isaac   PetscReal      *projF, *projT;
5093f27d899SToby Isaac   PetscReal      *projFstar, *projTstar;
5103f27d899SToby Isaac   PetscReal      *workF, *workF2, *workT, *workT2, *work, *work2;
5113f27d899SToby Isaac   PetscReal      *wedgeMat;
5123f27d899SToby Isaac   PetscReal      sign;
5133f27d899SToby Isaac   PetscErrorCode ierr;
5143f27d899SToby Isaac 
5153f27d899SToby Isaac   PetscFunctionBegin;
5163f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(formDegree), &Nk);CHKERRQ(ierr);
5173f27d899SToby Isaac   ierr = PetscDTBinomialInt(dimT, PetscAbsInt(kT), &NkT);CHKERRQ(ierr);
5183f27d899SToby Isaac   ierr = PetscDTBinomialInt(dimF, PetscAbsInt(kF), &NkF);CHKERRQ(ierr);
5193f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(kT), &MkT);CHKERRQ(ierr);
5203f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(kF), &MkF);CHKERRQ(ierr);
5213f27d899SToby Isaac   ierr = PetscNew(&ni);CHKERRQ(ierr);
5223f27d899SToby Isaac   ni->nodeIdxDim = nodeIdxDim = tracei->nodeIdxDim + fiberi->nodeIdxDim;
5233f27d899SToby Isaac   ni->nodeVecDim = Nk;
5243f27d899SToby Isaac   ni->nNodes = nNodes = tracei->nNodes * fiberi->nNodes;
5253f27d899SToby Isaac   ni->refct = 1;
5263f27d899SToby Isaac   ierr = PetscMalloc1(nNodes * nodeIdxDim, &(ni->nodeIdx));CHKERRQ(ierr);
5273f27d899SToby Isaac   /* first concatenate the indices */
5283f27d899SToby Isaac   for (l = 0, j = 0; j < fiberi->nNodes; j++) {
5293f27d899SToby Isaac     for (i = 0; i < tracei->nNodes; i++, l++) {
5303f27d899SToby Isaac       PetscInt m, n = 0;
5313f27d899SToby Isaac 
5323f27d899SToby Isaac       for (m = 0; m < tracei->nodeIdxDim; m++) ni->nodeIdx[l * nodeIdxDim + n++] = tracei->nodeIdx[i * tracei->nodeIdxDim + m];
5333f27d899SToby Isaac       for (m = 0; m < fiberi->nodeIdxDim; m++) ni->nodeIdx[l * nodeIdxDim + n++] = fiberi->nodeIdx[j * fiberi->nodeIdxDim + m];
5343f27d899SToby Isaac     }
5353f27d899SToby Isaac   }
5363f27d899SToby Isaac 
5373f27d899SToby Isaac   /* now wedge together the push-forward vectors */
5383f27d899SToby Isaac   ierr = PetscMalloc1(nNodes * Nk, &(ni->nodeVec));CHKERRQ(ierr);
5393f27d899SToby Isaac   ierr = PetscCalloc2(dimT*dim, &projT, dimF*dim, &projF);CHKERRQ(ierr);
5403f27d899SToby Isaac   for (i = 0; i < dimT; i++) projT[i * (dim + 1)] = 1.;
5413f27d899SToby Isaac   for (i = 0; i < dimF; i++) projF[i * (dim + dimT + 1) + dimT] = 1.;
5423f27d899SToby Isaac   ierr = PetscMalloc2(MkT*NkT, &projTstar, MkF*NkF, &projFstar);CHKERRQ(ierr);
5433f27d899SToby Isaac   ierr = PetscDTAltVPullbackMatrix(dim, dimT, projT, kT, projTstar);CHKERRQ(ierr);
5443f27d899SToby Isaac   ierr = PetscDTAltVPullbackMatrix(dim, dimF, projF, kF, projFstar);CHKERRQ(ierr);
5453f27d899SToby Isaac   ierr = PetscMalloc6(MkT, &workT, MkT, &workT2, MkF, &workF, MkF, &workF2, Nk, &work, Nk, &work2);CHKERRQ(ierr);
5463f27d899SToby Isaac   ierr = PetscMalloc1(Nk * MkT, &wedgeMat);CHKERRQ(ierr);
5473f27d899SToby Isaac   sign = (PetscAbsInt(kT * kF) & 1) ? -1. : 1.;
5483f27d899SToby Isaac   for (l = 0, j = 0; j < fiberi->nNodes; j++) {
5493f27d899SToby Isaac     PetscInt d, e;
5503f27d899SToby Isaac 
5513f27d899SToby Isaac     /* push forward fiber k-form */
5523f27d899SToby Isaac     for (d = 0; d < MkF; d++) {
5533f27d899SToby Isaac       PetscReal val = 0.;
5543f27d899SToby Isaac       for (e = 0; e < NkF; e++) val += projFstar[d * NkF + e] * fiberi->nodeVec[j * NkF + e];
5553f27d899SToby Isaac       workF[d] = val;
5563f27d899SToby Isaac     }
5573f27d899SToby Isaac     /* Hodge star to proper form if necessary */
5583f27d899SToby Isaac     if (kF < 0) {
5593f27d899SToby Isaac       for (d = 0; d < MkF; d++) workF2[d] = workF[d];
5603f27d899SToby Isaac       ierr = PetscDTAltVStar(dim, PetscAbsInt(kF), 1, workF2, workF);CHKERRQ(ierr);
5613f27d899SToby Isaac     }
5623f27d899SToby Isaac     /* Compute the matrix that wedges this form with one of the trace k-form */
5633f27d899SToby Isaac     ierr = PetscDTAltVWedgeMatrix(dim, PetscAbsInt(kF), PetscAbsInt(kT), workF, wedgeMat);CHKERRQ(ierr);
5643f27d899SToby Isaac     for (i = 0; i < tracei->nNodes; i++, l++) {
5653f27d899SToby Isaac       /* push forward trace k-form */
5663f27d899SToby Isaac       for (d = 0; d < MkT; d++) {
5673f27d899SToby Isaac         PetscReal val = 0.;
5683f27d899SToby Isaac         for (e = 0; e < NkT; e++) val += projTstar[d * NkT + e] * tracei->nodeVec[i * NkT + e];
5693f27d899SToby Isaac         workT[d] = val;
5703f27d899SToby Isaac       }
5713f27d899SToby Isaac       /* Hodge star to proper form if necessary */
5723f27d899SToby Isaac       if (kT < 0) {
5733f27d899SToby Isaac         for (d = 0; d < MkT; d++) workT2[d] = workT[d];
5743f27d899SToby Isaac         ierr = PetscDTAltVStar(dim, PetscAbsInt(kT), 1, workT2, workT);CHKERRQ(ierr);
5753f27d899SToby Isaac       }
5763f27d899SToby Isaac       /* compute the wedge product of the push-forward trace form and firer forms */
5773f27d899SToby Isaac       for (d = 0; d < Nk; d++) {
5783f27d899SToby Isaac         PetscReal val = 0.;
5793f27d899SToby Isaac         for (e = 0; e < MkT; e++) val += wedgeMat[d * MkT + e] * workT[e];
5803f27d899SToby Isaac         work[d] = val;
5813f27d899SToby Isaac       }
5823f27d899SToby Isaac       /* inverse Hodge star from proper form if necessary */
5833f27d899SToby Isaac       if (formDegree < 0) {
5843f27d899SToby Isaac         for (d = 0; d < Nk; d++) work2[d] = work[d];
5853f27d899SToby Isaac         ierr = PetscDTAltVStar(dim, PetscAbsInt(formDegree), -1, work2, work);CHKERRQ(ierr);
5863f27d899SToby Isaac       }
5873f27d899SToby Isaac       /* insert into the array (adjusting for sign) */
5883f27d899SToby Isaac       for (d = 0; d < Nk; d++) ni->nodeVec[l * Nk + d] = sign * work[d];
5893f27d899SToby Isaac     }
5903f27d899SToby Isaac   }
5913f27d899SToby Isaac   ierr = PetscFree(wedgeMat);CHKERRQ(ierr);
5923f27d899SToby Isaac   ierr = PetscFree6(workT, workT2, workF, workF2, work, work2);CHKERRQ(ierr);
5933f27d899SToby Isaac   ierr = PetscFree2(projTstar, projFstar);CHKERRQ(ierr);
5943f27d899SToby Isaac   ierr = PetscFree2(projT, projF);CHKERRQ(ierr);
5953f27d899SToby Isaac   *nodeIndices = ni;
5963f27d899SToby Isaac   PetscFunctionReturn(0);
5973f27d899SToby Isaac }
5983f27d899SToby Isaac 
5993f27d899SToby Isaac static PetscErrorCode PetscLagNodeIndicesMerge(PetscLagNodeIndices niA, PetscLagNodeIndices niB, PetscLagNodeIndices *nodeIndices)
6003f27d899SToby Isaac {
6013f27d899SToby Isaac   PetscLagNodeIndices ni;
6023f27d899SToby Isaac   PetscInt            nodeIdxDim, nodeVecDim, nNodes;
6033f27d899SToby Isaac   PetscErrorCode      ierr;
6043f27d899SToby Isaac 
6053f27d899SToby Isaac   PetscFunctionBegin;
6063f27d899SToby Isaac   ierr = PetscNew(&ni);CHKERRQ(ierr);
6073f27d899SToby Isaac   ni->nodeIdxDim = nodeIdxDim = niA->nodeIdxDim;
6083f27d899SToby Isaac   if (niB->nodeIdxDim != nodeIdxDim) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Cannot merge PetscLagNodeIndices with different nodeIdxDim");
6093f27d899SToby Isaac   ni->nodeVecDim = nodeVecDim = niA->nodeVecDim;
6103f27d899SToby Isaac   if (niB->nodeVecDim != nodeVecDim) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Cannot merge PetscLagNodeIndices with different nodeVecDim");
6113f27d899SToby Isaac   ni->nNodes = nNodes = niA->nNodes + niB->nNodes;
6123f27d899SToby Isaac   ni->refct = 1;
6133f27d899SToby Isaac   ierr = PetscMalloc1(nNodes * nodeIdxDim, &(ni->nodeIdx));CHKERRQ(ierr);
6143f27d899SToby Isaac   ierr = PetscMalloc1(nNodes * nodeVecDim, &(ni->nodeVec));CHKERRQ(ierr);
6153f27d899SToby Isaac   ierr = PetscArraycpy(ni->nodeIdx, niA->nodeIdx, niA->nNodes * nodeIdxDim);CHKERRQ(ierr);
6163f27d899SToby Isaac   ierr = PetscArraycpy(ni->nodeVec, niA->nodeVec, niA->nNodes * nodeVecDim);CHKERRQ(ierr);
6173f27d899SToby Isaac   ierr = PetscArraycpy(&(ni->nodeIdx[niA->nNodes * nodeIdxDim]), niB->nodeIdx, niB->nNodes * nodeIdxDim);CHKERRQ(ierr);
6183f27d899SToby Isaac   ierr = PetscArraycpy(&(ni->nodeVec[niA->nNodes * nodeVecDim]), niB->nodeVec, niB->nNodes * nodeVecDim);CHKERRQ(ierr);
6193f27d899SToby Isaac   *nodeIndices = ni;
6203f27d899SToby Isaac   PetscFunctionReturn(0);
6213f27d899SToby Isaac }
6223f27d899SToby Isaac 
6233f27d899SToby Isaac #define PETSCTUPINTCOMPREVLEX(N)                                   \
6243f27d899SToby Isaac static int PetscTupIntCompRevlex_##N(const void *a, const void *b) \
6253f27d899SToby Isaac {                                                                  \
6263f27d899SToby Isaac   const PetscInt *A = (const PetscInt *) a;                        \
6273f27d899SToby Isaac   const PetscInt *B = (const PetscInt *) b;                        \
6283f27d899SToby Isaac   int i;                                                           \
6293f27d899SToby Isaac   PetscInt diff = 0;                                               \
6303f27d899SToby Isaac   for (i = 0; i < N; i++) {                                        \
6313f27d899SToby Isaac     diff = A[N - i] - B[N - i];                                    \
6323f27d899SToby Isaac     if (diff) break;                                               \
6333f27d899SToby Isaac   }                                                                \
6343f27d899SToby Isaac   return (diff <= 0) ? (diff < 0) ? -1 : 0 : 1;                    \
6353f27d899SToby Isaac }
6363f27d899SToby Isaac 
6373f27d899SToby Isaac PETSCTUPINTCOMPREVLEX(3)
6383f27d899SToby Isaac PETSCTUPINTCOMPREVLEX(4)
6393f27d899SToby Isaac PETSCTUPINTCOMPREVLEX(5)
6403f27d899SToby Isaac PETSCTUPINTCOMPREVLEX(6)
6413f27d899SToby Isaac PETSCTUPINTCOMPREVLEX(7)
6423f27d899SToby Isaac 
6433f27d899SToby Isaac static int PetscTupIntCompRevlex_N(const void *a, const void *b)
6443f27d899SToby Isaac {
6453f27d899SToby Isaac   const PetscInt *A = (const PetscInt *) a;
6463f27d899SToby Isaac   const PetscInt *B = (const PetscInt *) b;
6473f27d899SToby Isaac   int i;
6483f27d899SToby Isaac   int N = A[0];
6493f27d899SToby Isaac   PetscInt diff = 0;
6503f27d899SToby Isaac   for (i = 0; i < N; i++) {
6513f27d899SToby Isaac     diff = A[N - i] - B[N - i];
6523f27d899SToby Isaac     if (diff) break;
6533f27d899SToby Isaac   }
6543f27d899SToby Isaac   return (diff <= 0) ? (diff < 0) ? -1 : 0 : 1;
6553f27d899SToby Isaac }
6563f27d899SToby Isaac 
6573f27d899SToby Isaac static PetscErrorCode PetscLagNodeIndicesGetPermutation(PetscLagNodeIndices ni, PetscInt *perm[])
6583f27d899SToby Isaac {
6593f27d899SToby Isaac   PetscErrorCode ierr;
6603f27d899SToby Isaac 
6613f27d899SToby Isaac   PetscFunctionBegin;
6623f27d899SToby Isaac   if (!(ni->perm)) {
6633f27d899SToby Isaac     PetscInt *sorter;
6643f27d899SToby Isaac     PetscInt m = ni->nNodes;
6653f27d899SToby Isaac     PetscInt nodeIdxDim = ni->nodeIdxDim;
6663f27d899SToby Isaac     PetscInt i, j, k, l;
6673f27d899SToby Isaac     PetscInt *prm;
6683f27d899SToby Isaac     int (*comp) (const void *, const void *);
6693f27d899SToby Isaac 
6703f27d899SToby Isaac     ierr = PetscMalloc1((nodeIdxDim + 2) * m, &sorter);CHKERRQ(ierr);
6713f27d899SToby Isaac     for (k = 0, l = 0, i = 0; i < m; i++) {
6723f27d899SToby Isaac       sorter[k++] = nodeIdxDim + 1;
6733f27d899SToby Isaac       sorter[k++] = i;
6743f27d899SToby Isaac       for (j = 0; j < nodeIdxDim; j++) sorter[k++] = ni->nodeIdx[l++];
6753f27d899SToby Isaac     }
6763f27d899SToby Isaac     switch (nodeIdxDim) {
6773f27d899SToby Isaac     case 2:
6783f27d899SToby Isaac       comp = PetscTupIntCompRevlex_3;
6793f27d899SToby Isaac       break;
6803f27d899SToby Isaac     case 3:
6813f27d899SToby Isaac       comp = PetscTupIntCompRevlex_4;
6823f27d899SToby Isaac       break;
6833f27d899SToby Isaac     case 4:
6843f27d899SToby Isaac       comp = PetscTupIntCompRevlex_5;
6853f27d899SToby Isaac       break;
6863f27d899SToby Isaac     case 5:
6873f27d899SToby Isaac       comp = PetscTupIntCompRevlex_6;
6883f27d899SToby Isaac       break;
6893f27d899SToby Isaac     case 6:
6903f27d899SToby Isaac       comp = PetscTupIntCompRevlex_7;
6913f27d899SToby Isaac       break;
6923f27d899SToby Isaac     default:
6933f27d899SToby Isaac       comp = PetscTupIntCompRevlex_N;
6943f27d899SToby Isaac       break;
6953f27d899SToby Isaac     }
6963f27d899SToby Isaac     qsort(sorter, m, (nodeIdxDim + 2) * sizeof(PetscInt), comp);
6973f27d899SToby Isaac     ierr = PetscMalloc1(m, &prm);CHKERRQ(ierr);
6983f27d899SToby Isaac     for (i = 0; i < m; i++) prm[i] = sorter[(nodeIdxDim + 2) * i + 1];
6993f27d899SToby Isaac     ni->perm = prm;
7003f27d899SToby Isaac     ierr = PetscFree(sorter);
7013f27d899SToby Isaac   }
7023f27d899SToby Isaac   *perm = ni->perm;
7033f27d899SToby Isaac   PetscFunctionReturn(0);
7043f27d899SToby Isaac }
70520cf1dd8SToby Isaac 
7066f905325SMatthew G. Knepley static PetscErrorCode PetscDualSpaceDestroy_Lagrange(PetscDualSpace sp)
70720cf1dd8SToby Isaac {
70820cf1dd8SToby Isaac   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *) sp->data;
7096f905325SMatthew G. Knepley   PetscErrorCode      ierr;
71020cf1dd8SToby Isaac 
71120cf1dd8SToby Isaac   PetscFunctionBegin;
7123f27d899SToby Isaac   if (lag->symperms) {
7133f27d899SToby Isaac     PetscInt **selfSyms = lag->symperms[0];
7146f905325SMatthew G. Knepley 
7156f905325SMatthew G. Knepley     if (selfSyms) {
7166f905325SMatthew G. Knepley       PetscInt i, **allocated = &selfSyms[-lag->selfSymOff];
7176f905325SMatthew G. Knepley 
7186f905325SMatthew G. Knepley       for (i = 0; i < lag->numSelfSym; i++) {
7196f905325SMatthew G. Knepley         ierr = PetscFree(allocated[i]);CHKERRQ(ierr);
7206f905325SMatthew G. Knepley       }
7216f905325SMatthew G. Knepley       ierr = PetscFree(allocated);CHKERRQ(ierr);
7226f905325SMatthew G. Knepley     }
7233f27d899SToby Isaac     ierr = PetscFree(lag->symperms);CHKERRQ(ierr);
7246f905325SMatthew G. Knepley   }
7253f27d899SToby Isaac   if (lag->symflips) {
7263f27d899SToby Isaac     PetscScalar **selfSyms = lag->symflips[0];
7273f27d899SToby Isaac 
7283f27d899SToby Isaac     if (selfSyms) {
7293f27d899SToby Isaac       PetscInt i;
7303f27d899SToby Isaac       PetscScalar **allocated = &selfSyms[-lag->selfSymOff];
7313f27d899SToby Isaac 
7323f27d899SToby Isaac       for (i = 0; i < lag->numSelfSym; i++) {
7333f27d899SToby Isaac         ierr = PetscFree(allocated[i]);CHKERRQ(ierr);
7346f905325SMatthew G. Knepley       }
7353f27d899SToby Isaac       ierr = PetscFree(allocated);CHKERRQ(ierr);
7363f27d899SToby Isaac     }
7373f27d899SToby Isaac     ierr = PetscFree(lag->symflips);CHKERRQ(ierr);
7383f27d899SToby Isaac   }
7393f27d899SToby Isaac   ierr = Petsc1DNodeFamilyDestroy(&(lag->nodeFamily));CHKERRQ(ierr);
7403f27d899SToby Isaac   ierr = PetscLagNodeIndicesDestroy(&(lag->vertIndices));CHKERRQ(ierr);
7413f27d899SToby Isaac   ierr = PetscLagNodeIndicesDestroy(&(lag->intNodeIndices));CHKERRQ(ierr);
7423f27d899SToby Isaac   ierr = PetscLagNodeIndicesDestroy(&(lag->allNodeIndices));CHKERRQ(ierr);
7436f905325SMatthew G. Knepley   ierr = PetscFree(lag);CHKERRQ(ierr);
7446f905325SMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeGetContinuity_C", NULL);CHKERRQ(ierr);
7456f905325SMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeSetContinuity_C", NULL);CHKERRQ(ierr);
7466f905325SMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeGetTensor_C", NULL);CHKERRQ(ierr);
7476f905325SMatthew G. Knepley   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeSetTensor_C", NULL);CHKERRQ(ierr);
7483f27d899SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeGetTrimmed_C", NULL);CHKERRQ(ierr);
7493f27d899SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeSetTrimmed_C", NULL);CHKERRQ(ierr);
7503f27d899SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeGetNodeType_C", NULL);CHKERRQ(ierr);
7513f27d899SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeSetNodeType_C", NULL);CHKERRQ(ierr);
75220cf1dd8SToby Isaac   PetscFunctionReturn(0);
75320cf1dd8SToby Isaac }
75420cf1dd8SToby Isaac 
7556f905325SMatthew G. Knepley static PetscErrorCode PetscDualSpaceLagrangeView_Ascii(PetscDualSpace sp, PetscViewer viewer)
75620cf1dd8SToby Isaac {
75720cf1dd8SToby Isaac   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *) sp->data;
7586f905325SMatthew G. Knepley   PetscErrorCode      ierr;
75920cf1dd8SToby Isaac 
76020cf1dd8SToby Isaac   PetscFunctionBegin;
7613f27d899SToby Isaac   ierr = PetscViewerASCIIPrintf(viewer, "%s %s%sLagrange dual space\n", lag->continuous ? "Continuous" : "Discontinuous", lag->tensorSpace ? "tensor " : "", lag->trimmed ? "trimmed " : "");CHKERRQ(ierr);
76220cf1dd8SToby Isaac   PetscFunctionReturn(0);
76320cf1dd8SToby Isaac }
76420cf1dd8SToby Isaac 
7656f905325SMatthew G. Knepley static PetscErrorCode PetscDualSpaceView_Lagrange(PetscDualSpace sp, PetscViewer viewer)
76620cf1dd8SToby Isaac {
7676f905325SMatthew G. Knepley   PetscBool      iascii;
7686f905325SMatthew G. Knepley   PetscErrorCode ierr;
7696f905325SMatthew G. Knepley 
77020cf1dd8SToby Isaac   PetscFunctionBegin;
7716f905325SMatthew G. Knepley   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
7726f905325SMatthew G. Knepley   PetscValidHeaderSpecific(viewer, PETSC_VIEWER_CLASSID, 2);
7736f905325SMatthew G. Knepley   ierr = PetscObjectTypeCompare((PetscObject) viewer, PETSCVIEWERASCII, &iascii);CHKERRQ(ierr);
7746f905325SMatthew G. Knepley   if (iascii) {ierr = PetscDualSpaceLagrangeView_Ascii(sp, viewer);CHKERRQ(ierr);}
77520cf1dd8SToby Isaac   PetscFunctionReturn(0);
77620cf1dd8SToby Isaac }
77720cf1dd8SToby Isaac 
7786f905325SMatthew G. Knepley static PetscErrorCode PetscDualSpaceSetFromOptions_Lagrange(PetscOptionItems *PetscOptionsObject,PetscDualSpace sp)
77920cf1dd8SToby Isaac {
7803f27d899SToby Isaac   PetscBool      continuous, tensor, trimmed, flg, flg2, flg3;
7813f27d899SToby Isaac   PetscDTNodeType nodeType;
7823f27d899SToby Isaac   PetscReal      nodeExponent;
7833f27d899SToby Isaac   PetscBool      nodeEndpoints;
7846f905325SMatthew G. Knepley   PetscErrorCode ierr;
7856f905325SMatthew G. Knepley 
7866f905325SMatthew G. Knepley   PetscFunctionBegin;
7876f905325SMatthew G. Knepley   ierr = PetscDualSpaceLagrangeGetContinuity(sp, &continuous);CHKERRQ(ierr);
7886f905325SMatthew G. Knepley   ierr = PetscDualSpaceLagrangeGetTensor(sp, &tensor);CHKERRQ(ierr);
7893f27d899SToby Isaac   ierr = PetscDualSpaceLagrangeGetTrimmed(sp, &trimmed);CHKERRQ(ierr);
7903f27d899SToby Isaac   ierr = PetscDualSpaceLagrangeGetNodeType(sp, &nodeType, &nodeEndpoints, &nodeExponent);CHKERRQ(ierr);
7913f27d899SToby Isaac   if (nodeType == PETSCDTNODES_DEFAULT) nodeType = PETSCDTNODES_GAUSSJACOBI;
7926f905325SMatthew G. Knepley   ierr = PetscOptionsHead(PetscOptionsObject,"PetscDualSpace Lagrange Options");CHKERRQ(ierr);
7936f905325SMatthew G. Knepley   ierr = PetscOptionsBool("-petscdualspace_lagrange_continuity", "Flag for continuous element", "PetscDualSpaceLagrangeSetContinuity", continuous, &continuous, &flg);CHKERRQ(ierr);
7946f905325SMatthew G. Knepley   if (flg) {ierr = PetscDualSpaceLagrangeSetContinuity(sp, continuous);CHKERRQ(ierr);}
7953f27d899SToby Isaac   ierr = PetscOptionsBool("-petscdualspace_lagrange_tensor", "Flag for tensor dual space", "PetscDualSpaceLagrangeSetTensor", tensor, &tensor, &flg);CHKERRQ(ierr);
7966f905325SMatthew G. Knepley   if (flg) {ierr = PetscDualSpaceLagrangeSetTensor(sp, tensor);CHKERRQ(ierr);}
7973f27d899SToby Isaac   ierr = PetscOptionsBool("-petscdualspace_lagrange_trimmed", "Flag for trimmed dual space", "PetscDualSpaceLagrangeSetTrimmed", trimmed, &trimmed, &flg);CHKERRQ(ierr);
7983f27d899SToby Isaac   if (flg) {ierr = PetscDualSpaceLagrangeSetTrimmed(sp, trimmed);CHKERRQ(ierr);}
7993f27d899SToby Isaac   ierr = PetscOptionsEnum("-petscdualspace_lagrange_node_type", "Lagrange node location type", "PetscDualSpaceLagrangeSetNodeType", PetscDTNodeTypes, (PetscEnum)nodeType, (PetscEnum *)&nodeType, &flg);CHKERRQ(ierr);
8003f27d899SToby Isaac   ierr = PetscOptionsBool("-petscdualspace_lagrange_node_endpoints", "Flag for nodes that include endpoints", "PetscDualSpaceLagrangeSetNodeType", nodeEndpoints, &nodeEndpoints, &flg2);CHKERRQ(ierr);
8013f27d899SToby Isaac   flg3 = PETSC_FALSE;
8023f27d899SToby Isaac   if (nodeType == PETSCDTNODES_GAUSSJACOBI) {
8033f27d899SToby Isaac     ierr = PetscOptionsReal("-petscdualspace_lagrange_node_exponent", "Gauss-Jacobi weight function exponent", "PetscDualSpaceLagrangeSetNodeType", nodeExponent, &nodeExponent, &flg3);CHKERRQ(ierr);
8043f27d899SToby Isaac   }
8053f27d899SToby Isaac   if (flg || flg2 || flg3) {ierr = PetscDualSpaceLagrangeSetNodeType(sp, nodeType, nodeEndpoints, nodeExponent);CHKERRQ(ierr);}
8066f905325SMatthew G. Knepley   ierr = PetscOptionsTail();CHKERRQ(ierr);
8076f905325SMatthew G. Knepley   PetscFunctionReturn(0);
8086f905325SMatthew G. Knepley }
8096f905325SMatthew G. Knepley 
810b4457527SToby Isaac static PetscErrorCode PetscDualSpaceDuplicate_Lagrange(PetscDualSpace sp, PetscDualSpace spNew)
8116f905325SMatthew G. Knepley {
8123f27d899SToby Isaac   PetscBool           cont, tensor, trimmed, boundary;
8133f27d899SToby Isaac   PetscDTNodeType     nodeType;
8143f27d899SToby Isaac   PetscReal           exponent;
8153f27d899SToby Isaac   PetscDualSpace_Lag *lag    = (PetscDualSpace_Lag *) sp->data;
8166f905325SMatthew G. Knepley   PetscErrorCode      ierr;
8176f905325SMatthew G. Knepley 
8186f905325SMatthew G. Knepley   PetscFunctionBegin;
8196f905325SMatthew G. Knepley   ierr = PetscDualSpaceLagrangeGetContinuity(sp, &cont);CHKERRQ(ierr);
820b4457527SToby Isaac   ierr = PetscDualSpaceLagrangeSetContinuity(spNew, cont);CHKERRQ(ierr);
8216f905325SMatthew G. Knepley   ierr = PetscDualSpaceLagrangeGetTensor(sp, &tensor);CHKERRQ(ierr);
822b4457527SToby Isaac   ierr = PetscDualSpaceLagrangeSetTensor(spNew, tensor);CHKERRQ(ierr);
8233f27d899SToby Isaac   ierr = PetscDualSpaceLagrangeGetTrimmed(sp, &trimmed);CHKERRQ(ierr);
8243f27d899SToby Isaac   ierr = PetscDualSpaceLagrangeSetTrimmed(spNew, trimmed);CHKERRQ(ierr);
8253f27d899SToby Isaac   ierr = PetscDualSpaceLagrangeGetNodeType(sp, &nodeType, &boundary, &exponent);CHKERRQ(ierr);
8263f27d899SToby Isaac   ierr = PetscDualSpaceLagrangeSetNodeType(spNew, nodeType, boundary, exponent);CHKERRQ(ierr);
8273f27d899SToby Isaac   if (lag->nodeFamily) {
8283f27d899SToby Isaac     PetscDualSpace_Lag *lagnew = (PetscDualSpace_Lag *) spNew->data;
8293f27d899SToby Isaac 
8303f27d899SToby Isaac     ierr = Petsc1DNodeFamilyReference(lag->nodeFamily);CHKERRQ(ierr);
8313f27d899SToby Isaac     lagnew->nodeFamily = lag->nodeFamily;
8323f27d899SToby Isaac   }
8336f905325SMatthew G. Knepley   PetscFunctionReturn(0);
8346f905325SMatthew G. Knepley }
8356f905325SMatthew G. Knepley 
8363f27d899SToby Isaac static PetscErrorCode PetscDualSpaceCreateEdgeSubspace_Lagrange(PetscDualSpace sp, PetscInt order, PetscInt k, PetscInt Nc, PetscBool interiorOnly, PetscDualSpace *bdsp)
8376f905325SMatthew G. Knepley {
8383f27d899SToby Isaac   DM                 K;
8393f27d899SToby Isaac   PetscDualSpace_Lag *newlag;
8406f905325SMatthew G. Knepley   PetscErrorCode     ierr;
8416f905325SMatthew G. Knepley 
8426f905325SMatthew G. Knepley   PetscFunctionBegin;
8436f905325SMatthew G. Knepley   ierr = PetscDualSpaceDuplicate(sp,bdsp);CHKERRQ(ierr);
8443f27d899SToby Isaac   ierr = PetscDualSpaceSetFormDegree(*bdsp, k);CHKERRQ(ierr);
8453f27d899SToby Isaac   ierr = PetscDualSpaceCreateReferenceCell(*bdsp, 1, PETSC_TRUE, &K);CHKERRQ(ierr);
8466f905325SMatthew G. Knepley   ierr = PetscDualSpaceSetDM(*bdsp, K);CHKERRQ(ierr);
8476f905325SMatthew G. Knepley   ierr = DMDestroy(&K);CHKERRQ(ierr);
8483f27d899SToby Isaac   ierr = PetscDualSpaceSetOrder(*bdsp, order);CHKERRQ(ierr);
8493f27d899SToby Isaac   ierr = PetscDualSpaceSetNumComponents(*bdsp, Nc);CHKERRQ(ierr);
8503f27d899SToby Isaac   newlag = (PetscDualSpace_Lag *) (*bdsp)->data;
8513f27d899SToby Isaac   newlag->interiorOnly = interiorOnly;
8526f905325SMatthew G. Knepley   ierr = PetscDualSpaceSetUp(*bdsp);CHKERRQ(ierr);
8533f27d899SToby Isaac   PetscFunctionReturn(0);
8546f905325SMatthew G. Knepley }
8553f27d899SToby Isaac 
8563f27d899SToby Isaac /* just the points, weights aren't handled */
8573f27d899SToby Isaac static PetscErrorCode PetscQuadratureCreateTensor(PetscQuadrature trace, PetscQuadrature fiber, PetscQuadrature *product)
8583f27d899SToby Isaac {
8593f27d899SToby Isaac   PetscInt         dimTrace, dimFiber;
8603f27d899SToby Isaac   PetscInt         numPointsTrace, numPointsFiber;
8613f27d899SToby Isaac   PetscInt         dim, numPoints;
8623f27d899SToby Isaac   const PetscReal *pointsTrace;
8633f27d899SToby Isaac   const PetscReal *pointsFiber;
8643f27d899SToby Isaac   PetscReal       *points;
8653f27d899SToby Isaac   PetscInt         i, j, k, p;
8663f27d899SToby Isaac   PetscErrorCode   ierr;
8673f27d899SToby Isaac 
8683f27d899SToby Isaac   PetscFunctionBegin;
8693f27d899SToby Isaac   ierr = PetscQuadratureGetData(trace, &dimTrace, NULL, &numPointsTrace, &pointsTrace, NULL);CHKERRQ(ierr);
8703f27d899SToby Isaac   ierr = PetscQuadratureGetData(fiber, &dimFiber, NULL, &numPointsFiber, &pointsFiber, NULL);CHKERRQ(ierr);
8713f27d899SToby Isaac   dim = dimTrace + dimFiber;
8723f27d899SToby Isaac   numPoints = numPointsFiber * numPointsTrace;
8733f27d899SToby Isaac   ierr = PetscMalloc1(numPoints * dim, &points);CHKERRQ(ierr);
8743f27d899SToby Isaac   for (p = 0, j = 0; j < numPointsFiber; j++) {
8753f27d899SToby Isaac     for (i = 0; i < numPointsTrace; i++, p++) {
8763f27d899SToby Isaac       for (k = 0; k < dimTrace; k++) points[p * dim +            k] = pointsTrace[i * dimTrace + k];
8773f27d899SToby Isaac       for (k = 0; k < dimFiber; k++) points[p * dim + dimTrace + k] = pointsFiber[j * dimFiber + k];
8783f27d899SToby Isaac     }
8793f27d899SToby Isaac   }
8803f27d899SToby Isaac   ierr = PetscQuadratureCreate(PETSC_COMM_SELF, product);CHKERRQ(ierr);
8813f27d899SToby Isaac   ierr = PetscQuadratureSetData(*product, dim, 0, numPoints, points, NULL);CHKERRQ(ierr);
8823f27d899SToby Isaac   PetscFunctionReturn(0);
8833f27d899SToby Isaac }
8843f27d899SToby Isaac 
8853f27d899SToby Isaac static PetscErrorCode MatTensorAltV(Mat trace, Mat fiber, PetscInt dimTrace, PetscInt kTrace, PetscInt dimFiber, PetscInt kFiber, Mat *product)
8863f27d899SToby Isaac {
8873f27d899SToby Isaac   PetscInt mTrace, nTrace, mFiber, nFiber, m, n, k, i, j, l;
8883f27d899SToby Isaac   PetscInt dim, NkTrace, NkFiber, Nk;
8893f27d899SToby Isaac   PetscInt dT, dF;
8903f27d899SToby Isaac   PetscInt *nnzTrace, *nnzFiber, *nnz;
8913f27d899SToby Isaac   PetscInt iT, iF, jT, jF, il, jl;
8923f27d899SToby Isaac   PetscReal *workT, *workT2, *workF, *workF2, *work, *workstar;
8933f27d899SToby Isaac   PetscReal *projT, *projF;
8943f27d899SToby Isaac   PetscReal *projTstar, *projFstar;
8953f27d899SToby Isaac   PetscReal *wedgeMat;
8963f27d899SToby Isaac   PetscReal sign;
8973f27d899SToby Isaac   PetscScalar *workS;
8983f27d899SToby Isaac   Mat prod;
8993f27d899SToby Isaac   /* this produces dof groups that look like the identity */
9003f27d899SToby Isaac   PetscErrorCode ierr;
9013f27d899SToby Isaac 
9023f27d899SToby Isaac   PetscFunctionBegin;
9033f27d899SToby Isaac   ierr = MatGetSize(trace, &mTrace, &nTrace);CHKERRQ(ierr);
9043f27d899SToby Isaac   ierr = PetscDTBinomialInt(dimTrace, PetscAbsInt(kTrace), &NkTrace);CHKERRQ(ierr);
9053f27d899SToby Isaac   if (nTrace % NkTrace) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "point value space of trace matrix is not a multiple of k-form size");
9063f27d899SToby Isaac   ierr = MatGetSize(fiber, &mFiber, &nFiber);CHKERRQ(ierr);
9073f27d899SToby Isaac   ierr = PetscDTBinomialInt(dimFiber, PetscAbsInt(kFiber), &NkFiber);CHKERRQ(ierr);
9083f27d899SToby Isaac   if (nFiber % NkFiber) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "point value space of fiber matrix is not a multiple of k-form size");
9093f27d899SToby Isaac   ierr = PetscMalloc2(mTrace, &nnzTrace, mFiber, &nnzFiber);CHKERRQ(ierr);
9103f27d899SToby Isaac   for (i = 0; i < mTrace; i++) {
9113f27d899SToby Isaac     ierr = MatGetRow(trace, i, &(nnzTrace[i]), NULL, NULL);CHKERRQ(ierr);
9123f27d899SToby Isaac     if (nnzTrace[i] % NkTrace) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "nonzeros in trace matrix are not in k-form size blocks");
9133f27d899SToby Isaac   }
9143f27d899SToby Isaac   for (i = 0; i < mFiber; i++) {
9153f27d899SToby Isaac     ierr = MatGetRow(fiber, i, &(nnzFiber[i]), NULL, NULL);CHKERRQ(ierr);
9163f27d899SToby Isaac     if (nnzFiber[i] % NkFiber) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "nonzeros in fiber matrix are not in k-form size blocks");
9173f27d899SToby Isaac   }
9183f27d899SToby Isaac   dim = dimTrace + dimFiber;
9193f27d899SToby Isaac   k = kFiber + kTrace;
9203f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(k), &Nk);CHKERRQ(ierr);
9213f27d899SToby Isaac   m = mTrace * mFiber;
9223f27d899SToby Isaac   ierr = PetscMalloc1(m, &nnz);CHKERRQ(ierr);
9233f27d899SToby Isaac   for (l = 0, j = 0; j < mFiber; j++) for (i = 0; i < mTrace; i++, l++) nnz[l] = (nnzTrace[i] / NkTrace) * (nnzFiber[j] / NkFiber) * Nk;
9243f27d899SToby Isaac   n = (nTrace / NkTrace) * (nFiber / NkFiber) * Nk;
9253f27d899SToby Isaac   ierr = MatCreateSeqAIJ(PETSC_COMM_SELF, m, n, 0, nnz, &prod);CHKERRQ(ierr);
9263f27d899SToby Isaac   ierr = PetscFree(nnz);CHKERRQ(ierr);
9273f27d899SToby Isaac   ierr = PetscFree2(nnzTrace,nnzFiber);CHKERRQ(ierr);
9283f27d899SToby Isaac   /* reasoning about which points each dof needs depends on having zeros computed at points preserved */
9293f27d899SToby Isaac   ierr = MatSetOption(prod, MAT_IGNORE_ZERO_ENTRIES, PETSC_FALSE);CHKERRQ(ierr);
9303f27d899SToby Isaac   /* compute pullbacks */
9313f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(kTrace), &dT);CHKERRQ(ierr);
9323f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(kFiber), &dF);CHKERRQ(ierr);
9333f27d899SToby Isaac   ierr = PetscMalloc4(dimTrace * dim, &projT, dimFiber * dim, &projF, dT * NkTrace, &projTstar, dF * NkFiber, &projFstar);CHKERRQ(ierr);
9343f27d899SToby Isaac   ierr = PetscArrayzero(projT, dimTrace * dim);CHKERRQ(ierr);
9353f27d899SToby Isaac   for (i = 0; i < dimTrace; i++) projT[i * (dim + 1)] = 1.;
9363f27d899SToby Isaac   ierr = PetscArrayzero(projF, dimFiber * dim);CHKERRQ(ierr);
9373f27d899SToby Isaac   for (i = 0; i < dimFiber; i++) projF[i * (dim + 1) + dimTrace] = 1.;
9383f27d899SToby Isaac   ierr = PetscDTAltVPullbackMatrix(dim, dimTrace, projT, kTrace, projTstar);CHKERRQ(ierr);
9393f27d899SToby Isaac   ierr = PetscDTAltVPullbackMatrix(dim, dimFiber, projF, kFiber, projFstar);CHKERRQ(ierr);
9403f27d899SToby Isaac   ierr = PetscMalloc5(dT, &workT, dF, &workF, Nk, &work, Nk, &workstar, Nk, &workS);CHKERRQ(ierr);
9413f27d899SToby Isaac   ierr = PetscMalloc2(dT, &workT2, dF, &workF2);CHKERRQ(ierr);
9423f27d899SToby Isaac   ierr = PetscMalloc1(Nk * dT, &wedgeMat);CHKERRQ(ierr);
9433f27d899SToby Isaac   sign = (PetscAbsInt(kTrace * kFiber) & 1) ? -1. : 1.;
9443f27d899SToby Isaac   for (i = 0, iF = 0; iF < mFiber; iF++) {
9453f27d899SToby Isaac     PetscInt           ncolsF, nformsF;
9463f27d899SToby Isaac     const PetscInt    *colsF;
9473f27d899SToby Isaac     const PetscScalar *valsF;
9483f27d899SToby Isaac 
9493f27d899SToby Isaac     ierr = MatGetRow(fiber, iF, &ncolsF, &colsF, &valsF);CHKERRQ(ierr);
9503f27d899SToby Isaac     nformsF = ncolsF / NkFiber;
9513f27d899SToby Isaac     for (iT = 0; iT < mTrace; iT++, i++) {
9523f27d899SToby Isaac       PetscInt           ncolsT, nformsT;
9533f27d899SToby Isaac       const PetscInt    *colsT;
9543f27d899SToby Isaac       const PetscScalar *valsT;
9553f27d899SToby Isaac 
9563f27d899SToby Isaac       ierr = MatGetRow(trace, iT, &ncolsT, &colsT, &valsT);CHKERRQ(ierr);
9573f27d899SToby Isaac       nformsT = ncolsT / NkTrace;
9583f27d899SToby Isaac       for (j = 0, jF = 0; jF < nformsF; jF++) {
9593f27d899SToby Isaac         PetscInt colF = colsF[jF * NkFiber] / NkFiber;
9603f27d899SToby Isaac 
9613f27d899SToby Isaac         for (il = 0; il < dF; il++) {
9623f27d899SToby Isaac           PetscReal val = 0.;
9633f27d899SToby Isaac           for (jl = 0; jl < NkFiber; jl++) val += projFstar[il * NkFiber + jl] * PetscRealPart(valsF[jF * NkFiber + jl]);
9643f27d899SToby Isaac           workF[il] = val;
9653f27d899SToby Isaac         }
9663f27d899SToby Isaac         if (kFiber < 0) {
9673f27d899SToby Isaac           for (il = 0; il < dF; il++) workF2[il] = workF[il];
9683f27d899SToby Isaac           ierr = PetscDTAltVStar(dim, PetscAbsInt(kFiber), 1, workF2, workF);CHKERRQ(ierr);
9693f27d899SToby Isaac         }
9703f27d899SToby Isaac         ierr = PetscDTAltVWedgeMatrix(dim, PetscAbsInt(kFiber), PetscAbsInt(kTrace), workF, wedgeMat);CHKERRQ(ierr);
9713f27d899SToby Isaac         for (jT = 0; jT < nformsT; jT++, j++) {
9723f27d899SToby Isaac           PetscInt colT = colsT[jT * NkTrace] / NkTrace;
9733f27d899SToby Isaac           PetscInt col = colF * (nTrace / NkTrace) + colT;
9743f27d899SToby Isaac           const PetscScalar *vals;
9753f27d899SToby Isaac 
9763f27d899SToby Isaac           for (il = 0; il < dT; il++) {
9773f27d899SToby Isaac             PetscReal val = 0.;
9783f27d899SToby Isaac             for (jl = 0; jl < NkTrace; jl++) val += projTstar[il * NkTrace + jl] * PetscRealPart(valsT[jT * NkTrace + jl]);
9793f27d899SToby Isaac             workT[il] = val;
9803f27d899SToby Isaac           }
9813f27d899SToby Isaac           if (kTrace < 0) {
9823f27d899SToby Isaac             for (il = 0; il < dT; il++) workT2[il] = workT[il];
9833f27d899SToby Isaac             ierr = PetscDTAltVStar(dim, PetscAbsInt(kTrace), 1, workT2, workT);CHKERRQ(ierr);
9843f27d899SToby Isaac           }
9853f27d899SToby Isaac 
9863f27d899SToby Isaac           for (il = 0; il < Nk; il++) {
9873f27d899SToby Isaac             PetscReal val = 0.;
9883f27d899SToby Isaac             for (jl = 0; jl < dT; jl++) val += sign * wedgeMat[il * dT + jl] * workT[jl];
9893f27d899SToby Isaac             work[il] = val;
9903f27d899SToby Isaac           }
9913f27d899SToby Isaac           if (k < 0) {
9923f27d899SToby Isaac             ierr = PetscDTAltVStar(dim, PetscAbsInt(k), -1, work, workstar);CHKERRQ(ierr);
9933f27d899SToby Isaac #if defined(PETSC_USE_COMPLEX)
9943f27d899SToby Isaac             for (l = 0; l < Nk; l++) workS[l] = workstar[l];
9953f27d899SToby Isaac             vals = &workS[0];
9963f27d899SToby Isaac #else
9973f27d899SToby Isaac             vals = &workstar[0];
9983f27d899SToby Isaac #endif
9993f27d899SToby Isaac           } else {
10003f27d899SToby Isaac #if defined(PETSC_USE_COMPLEX)
10013f27d899SToby Isaac             for (l = 0; l < Nk; l++) workS[l] = work[l];
10023f27d899SToby Isaac             vals = &workS[0];
10033f27d899SToby Isaac #else
10043f27d899SToby Isaac             vals = &work[0];
10053f27d899SToby Isaac #endif
10063f27d899SToby Isaac           }
10073f27d899SToby Isaac           for (l = 0; l < Nk; l++) {
10083f27d899SToby Isaac             ierr = MatSetValue(prod, i, col * Nk + l, vals[l], INSERT_VALUES);CHKERRQ(ierr);
10093f27d899SToby Isaac           } /* Nk */
10103f27d899SToby Isaac         } /* jT */
10113f27d899SToby Isaac       } /* jF */
10123f27d899SToby Isaac       ierr = MatRestoreRow(trace, iT, &ncolsT, &colsT, &valsT);CHKERRQ(ierr);
10133f27d899SToby Isaac     } /* iT */
10143f27d899SToby Isaac     ierr = MatRestoreRow(fiber, iF, &ncolsF, &colsF, &valsF);CHKERRQ(ierr);
10153f27d899SToby Isaac   } /* iF */
10163f27d899SToby Isaac   ierr = PetscFree(wedgeMat);CHKERRQ(ierr);
10173f27d899SToby Isaac   ierr = PetscFree4(projT, projF, projTstar, projFstar);CHKERRQ(ierr);
10183f27d899SToby Isaac   ierr = PetscFree2(workT2, workF2);CHKERRQ(ierr);
10193f27d899SToby Isaac   ierr = PetscFree5(workT, workF, work, workstar, workS);CHKERRQ(ierr);
10203f27d899SToby Isaac   ierr = MatAssemblyBegin(prod, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10213f27d899SToby Isaac   ierr = MatAssemblyEnd(prod, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
10223f27d899SToby Isaac   *product = prod;
10233f27d899SToby Isaac   PetscFunctionReturn(0);
10243f27d899SToby Isaac }
10253f27d899SToby Isaac 
10263f27d899SToby Isaac static PetscErrorCode PetscQuadraturePointsMerge(PetscQuadrature quadA, PetscQuadrature quadB, PetscQuadrature *quadJoint, PetscInt *aToJoint[], PetscInt *bToJoint[])
10273f27d899SToby Isaac {
10283f27d899SToby Isaac   PetscInt         dimA, dimB;
10293f27d899SToby Isaac   PetscInt         nA, nB, nJoint, i, j, d;
10303f27d899SToby Isaac   const PetscReal *pointsA;
10313f27d899SToby Isaac   const PetscReal *pointsB;
10323f27d899SToby Isaac   PetscReal       *pointsJoint;
10333f27d899SToby Isaac   PetscInt        *aToJ, *bToJ;
10343f27d899SToby Isaac   PetscQuadrature  qJ;
10353f27d899SToby Isaac   PetscErrorCode   ierr;
10363f27d899SToby Isaac 
10373f27d899SToby Isaac   PetscFunctionBegin;
10383f27d899SToby Isaac   ierr = PetscQuadratureGetData(quadA, &dimA, NULL, &nA, &pointsA, NULL);CHKERRQ(ierr);
10393f27d899SToby Isaac   ierr = PetscQuadratureGetData(quadB, &dimB, NULL, &nB, &pointsB, NULL);CHKERRQ(ierr);
10403f27d899SToby Isaac   if (dimA != dimB) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Quadrature points must be in the same dimension");
10413f27d899SToby Isaac   nJoint = nA;
10423f27d899SToby Isaac   ierr = PetscMalloc1(nA, &aToJ);CHKERRQ(ierr);
10433f27d899SToby Isaac   for (i = 0; i < nA; i++) aToJ[i] = i;
10443f27d899SToby Isaac   ierr = PetscMalloc1(nB, &bToJ);CHKERRQ(ierr);
10453f27d899SToby Isaac   for (i = 0; i < nB; i++) {
10463f27d899SToby Isaac     for (j = 0; j < nA; j++) {
10473f27d899SToby Isaac       bToJ[i] = -1;
1048*6ff15688SToby Isaac       for (d = 0; d < dimA; d++) if (PetscAbsReal(pointsB[i * dimA + d] - pointsA[j * dimA + d]) > PETSC_SMALL) break;
10493f27d899SToby Isaac       if (d == dimA) {
10503f27d899SToby Isaac         bToJ[i] = j;
10513f27d899SToby Isaac         break;
10523f27d899SToby Isaac       }
10533f27d899SToby Isaac     }
10543f27d899SToby Isaac     if (bToJ[i] == -1) {
10553f27d899SToby Isaac       bToJ[i] = nJoint++;
10563f27d899SToby Isaac     }
10573f27d899SToby Isaac   }
10583f27d899SToby Isaac   *aToJoint = aToJ;
10593f27d899SToby Isaac   *bToJoint = bToJ;
10603f27d899SToby Isaac   ierr = PetscMalloc1(nJoint * dimA, &pointsJoint);CHKERRQ(ierr);
10613f27d899SToby Isaac   ierr = PetscArraycpy(pointsJoint, pointsA, nA * dimA);CHKERRQ(ierr);
10623f27d899SToby Isaac   for (i = 0; i < nB; i++) {
10633f27d899SToby Isaac     if (bToJ[i] >= nA) {
10643f27d899SToby Isaac       for (d = 0; d < dimA; d++) pointsJoint[bToJ[i] * dimA + d] = pointsB[i * dimA + d];
10653f27d899SToby Isaac     }
10663f27d899SToby Isaac   }
10673f27d899SToby Isaac   ierr = PetscQuadratureCreate(PETSC_COMM_SELF, &qJ);CHKERRQ(ierr);
10683f27d899SToby Isaac   ierr = PetscQuadratureSetData(qJ, dimA, 0, nJoint, pointsJoint, NULL);CHKERRQ(ierr);
10693f27d899SToby Isaac   *quadJoint = qJ;
10703f27d899SToby Isaac   PetscFunctionReturn(0);
10713f27d899SToby Isaac }
10723f27d899SToby Isaac 
10733f27d899SToby Isaac static PetscErrorCode MatricesMerge(Mat matA, Mat matB, PetscInt dim, PetscInt k, PetscInt numMerged, const PetscInt aToMerged[], const PetscInt bToMerged[], Mat *matMerged)
10743f27d899SToby Isaac {
10753f27d899SToby Isaac   PetscInt m, n, mA, nA, mB, nB, Nk, i, j, l;
10763f27d899SToby Isaac   Mat      M;
10773f27d899SToby Isaac   PetscInt *nnz;
10783f27d899SToby Isaac   PetscInt maxnnz;
10793f27d899SToby Isaac   PetscInt *work;
10803f27d899SToby Isaac   PetscErrorCode ierr;
10813f27d899SToby Isaac 
10823f27d899SToby Isaac   PetscFunctionBegin;
10833f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(k), &Nk);CHKERRQ(ierr);
10843f27d899SToby Isaac   ierr = MatGetSize(matA, &mA, &nA);CHKERRQ(ierr);
10853f27d899SToby Isaac   if (nA % Nk) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "matA column space not a multiple of k-form size");
10863f27d899SToby Isaac   ierr = MatGetSize(matB, &mB, &nB);CHKERRQ(ierr);
10873f27d899SToby Isaac   if (nB % Nk) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "matB column space not a multiple of k-form size");
10883f27d899SToby Isaac   m = mA + mB;
10893f27d899SToby Isaac   n = numMerged * Nk;
10903f27d899SToby Isaac   ierr = PetscMalloc1(m, &nnz);CHKERRQ(ierr);
10913f27d899SToby Isaac   maxnnz = 0;
10923f27d899SToby Isaac   for (i = 0; i < mA; i++) {
10933f27d899SToby Isaac     ierr = MatGetRow(matA, i, &(nnz[i]), NULL, NULL);CHKERRQ(ierr);
10943f27d899SToby Isaac     if (nnz[i] % Nk) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "nonzeros in matA are not in k-form size blocks");
10953f27d899SToby Isaac     maxnnz = PetscMax(maxnnz, nnz[i]);
10963f27d899SToby Isaac   }
10973f27d899SToby Isaac   for (i = 0; i < mB; i++) {
10983f27d899SToby Isaac     ierr = MatGetRow(matB, i, &(nnz[i+mA]), NULL, NULL);CHKERRQ(ierr);
10993f27d899SToby Isaac     if (nnz[i+mA] % Nk) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "nonzeros in matB are not in k-form size blocks");
11003f27d899SToby Isaac     maxnnz = PetscMax(maxnnz, nnz[i+mA]);
11013f27d899SToby Isaac   }
11023f27d899SToby Isaac   ierr = MatCreateSeqAIJ(PETSC_COMM_SELF, m, n, 0, nnz, &M);CHKERRQ(ierr);
11033f27d899SToby Isaac   ierr = PetscFree(nnz);CHKERRQ(ierr);
11043f27d899SToby Isaac   /* reasoning about which points each dof needs depends on having zeros computed at points preserved */
11053f27d899SToby Isaac   ierr = MatSetOption(M, MAT_IGNORE_ZERO_ENTRIES, PETSC_FALSE);CHKERRQ(ierr);
11063f27d899SToby Isaac   ierr = PetscMalloc1(maxnnz, &work);CHKERRQ(ierr);
11073f27d899SToby Isaac   for (i = 0; i < mA; i++) {
11083f27d899SToby Isaac     const PetscInt *cols;
11093f27d899SToby Isaac     const PetscScalar *vals;
11103f27d899SToby Isaac     PetscInt nCols;
11113f27d899SToby Isaac     ierr = MatGetRow(matA, i, &nCols, &cols, &vals);CHKERRQ(ierr);
11123f27d899SToby Isaac     for (j = 0; j < nCols / Nk; j++) {
11133f27d899SToby Isaac       PetscInt newCol = aToMerged[cols[j * Nk] / Nk];
11143f27d899SToby Isaac       for (l = 0; l < Nk; l++) work[j * Nk + l] = newCol * Nk + l;
11153f27d899SToby Isaac     }
11163f27d899SToby Isaac     ierr = MatSetValuesBlocked(M, 1, &i, nCols, work, vals, INSERT_VALUES);CHKERRQ(ierr);
11173f27d899SToby Isaac     ierr = MatRestoreRow(matA, i, &nCols, &cols, &vals);CHKERRQ(ierr);
11183f27d899SToby Isaac   }
11193f27d899SToby Isaac   for (i = 0; i < mB; i++) {
11203f27d899SToby Isaac     const PetscInt *cols;
11213f27d899SToby Isaac     const PetscScalar *vals;
11223f27d899SToby Isaac 
11233f27d899SToby Isaac     PetscInt row = i + mA;
11243f27d899SToby Isaac     PetscInt nCols;
11253f27d899SToby Isaac     ierr = MatGetRow(matB, i, &nCols, &cols, &vals);CHKERRQ(ierr);
11263f27d899SToby Isaac     for (j = 0; j < nCols / Nk; j++) {
11273f27d899SToby Isaac       PetscInt newCol = bToMerged[cols[j * Nk] / Nk];
11283f27d899SToby Isaac       for (l = 0; l < Nk; l++) work[j * Nk + l] = newCol * Nk + l;
11293f27d899SToby Isaac     }
11303f27d899SToby Isaac     ierr = MatSetValuesBlocked(M, 1, &row, nCols, work, vals, INSERT_VALUES);CHKERRQ(ierr);
11313f27d899SToby Isaac     ierr = MatRestoreRow(matB, i, &nCols, &cols, &vals);CHKERRQ(ierr);
11323f27d899SToby Isaac   }
11333f27d899SToby Isaac   ierr = PetscFree(work);CHKERRQ(ierr);
11343f27d899SToby Isaac   ierr = MatAssemblyBegin(M, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11353f27d899SToby Isaac   ierr = MatAssemblyEnd(M, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11363f27d899SToby Isaac   *matMerged = M;
11373f27d899SToby Isaac   PetscFunctionReturn(0);
11383f27d899SToby Isaac }
11393f27d899SToby Isaac 
11403f27d899SToby Isaac static PetscErrorCode PetscDualSpaceCreateFacetSubspace_Lagrange(PetscDualSpace sp, DM K, PetscInt f, PetscInt k, PetscInt Ncopies, PetscBool interiorOnly, PetscDualSpace *bdsp)
11413f27d899SToby Isaac {
11423f27d899SToby Isaac   PetscInt           Nknew, Ncnew;
11433f27d899SToby Isaac   PetscInt           dim, pointDim = -1;
11443f27d899SToby Isaac   PetscInt           depth;
11453f27d899SToby Isaac   DM                 dm;
11463f27d899SToby Isaac   PetscDualSpace_Lag *newlag;
11473f27d899SToby Isaac   PetscErrorCode     ierr;
11483f27d899SToby Isaac 
11493f27d899SToby Isaac   PetscFunctionBegin;
11503f27d899SToby Isaac   ierr = PetscDualSpaceGetDM(sp,&dm);CHKERRQ(ierr);
11513f27d899SToby Isaac   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
11523f27d899SToby Isaac   ierr = DMPlexGetDepth(dm,&depth);CHKERRQ(ierr);
11533f27d899SToby Isaac   ierr = PetscDualSpaceDuplicate(sp,bdsp);CHKERRQ(ierr);
11543f27d899SToby Isaac   ierr = PetscDualSpaceSetFormDegree(*bdsp,k);CHKERRQ(ierr);
11553f27d899SToby Isaac   if (!K) {
11563f27d899SToby Isaac     PetscBool isSimplex;
11573f27d899SToby Isaac 
11583f27d899SToby Isaac 
11593f27d899SToby Isaac     if (depth == dim) {
11603f27d899SToby Isaac       PetscInt coneSize;
11613f27d899SToby Isaac 
1162*6ff15688SToby Isaac       pointDim = dim - 1;
11633f27d899SToby Isaac       ierr = DMPlexGetConeSize(dm,f,&coneSize);CHKERRQ(ierr);
11643f27d899SToby Isaac       isSimplex = (PetscBool) (coneSize == dim);
11653f27d899SToby Isaac       ierr = PetscDualSpaceCreateReferenceCell(*bdsp, dim-1, isSimplex, &K);CHKERRQ(ierr);
11663f27d899SToby Isaac     } else if (depth == 1) {
11673f27d899SToby Isaac       pointDim = 0;
11683f27d899SToby Isaac       ierr = PetscDualSpaceCreateReferenceCell(*bdsp, 0, PETSC_TRUE, &K);CHKERRQ(ierr);
11693f27d899SToby Isaac     } else SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Unsupported interpolation state of reference element");
11703f27d899SToby Isaac   } else {
11713f27d899SToby Isaac     ierr = PetscObjectReference((PetscObject)K);CHKERRQ(ierr);
11723f27d899SToby Isaac     ierr = DMGetDimension(K, &pointDim);CHKERRQ(ierr);
11733f27d899SToby Isaac   }
11743f27d899SToby Isaac   ierr = PetscDualSpaceSetDM(*bdsp, K);CHKERRQ(ierr);
11753f27d899SToby Isaac   ierr = DMDestroy(&K);CHKERRQ(ierr);
11763f27d899SToby Isaac   ierr = PetscDTBinomialInt(pointDim, PetscAbsInt(k), &Nknew);CHKERRQ(ierr);
11773f27d899SToby Isaac   Ncnew = Nknew * Ncopies;
11783f27d899SToby Isaac   ierr = PetscDualSpaceSetNumComponents(*bdsp, Ncnew);CHKERRQ(ierr);
11793f27d899SToby Isaac   newlag = (PetscDualSpace_Lag *) (*bdsp)->data;
11803f27d899SToby Isaac   newlag->interiorOnly = interiorOnly;
11813f27d899SToby Isaac   ierr = PetscDualSpaceSetUp(*bdsp);CHKERRQ(ierr);
11823f27d899SToby Isaac   PetscFunctionReturn(0);
11833f27d899SToby Isaac }
11843f27d899SToby Isaac 
11853f27d899SToby Isaac static PetscErrorCode PetscDualSpaceLagrangeCreateSimplexNodeMat(Petsc1DNodeFamily nodeFamily, PetscInt dim, PetscInt sum, PetscInt Nk, PetscInt numNodeSkip, PetscQuadrature *iNodes, Mat *iMat, PetscLagNodeIndices *nodeIndices)
11863f27d899SToby Isaac {
11873f27d899SToby Isaac   PetscReal *extraNodeCoords, *nodeCoords;
11883f27d899SToby Isaac   PetscInt nNodes, nExtraNodes;
11893f27d899SToby Isaac   PetscInt i, j, k, extraSum = sum + numNodeSkip * (1 + dim);
11903f27d899SToby Isaac   PetscQuadrature intNodes;
11913f27d899SToby Isaac   Mat intMat;
11923f27d899SToby Isaac   PetscLagNodeIndices ni;
11933f27d899SToby Isaac   PetscErrorCode ierr;
11943f27d899SToby Isaac 
11953f27d899SToby Isaac   PetscFunctionBegin;
11963f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim + sum, dim, &nNodes);CHKERRQ(ierr);
11973f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim + extraSum, dim, &nExtraNodes);CHKERRQ(ierr);
11983f27d899SToby Isaac 
11993f27d899SToby Isaac   ierr = PetscMalloc1(dim * nExtraNodes, &extraNodeCoords);CHKERRQ(ierr);
12003f27d899SToby Isaac   ierr = PetscNew(&ni);CHKERRQ(ierr);
12013f27d899SToby Isaac   ni->nodeIdxDim = dim + 1;
12023f27d899SToby Isaac   ni->nodeVecDim = Nk;
12033f27d899SToby Isaac   ni->nNodes = nNodes * Nk;
12043f27d899SToby Isaac   ni->refct = 1;
12053f27d899SToby Isaac   ierr = PetscMalloc1(nNodes * Nk * (dim + 1), &(ni->nodeIdx));CHKERRQ(ierr);
12063f27d899SToby Isaac   ierr = PetscMalloc1(nNodes * Nk * Nk, &(ni->nodeVec));CHKERRQ(ierr);
12073f27d899SToby Isaac   for (i = 0; i < nNodes; i++) for (j = 0; j < Nk; j++) for (k = 0; k < Nk; k++) ni->nodeVec[(i * Nk + j) * Nk + k] = (j == k) ? 1. : 0.;
12083f27d899SToby Isaac   ierr = Petsc1DNodeFamilyComputeSimplexNodes(nodeFamily, dim, extraSum, extraNodeCoords);CHKERRQ(ierr);
12093f27d899SToby Isaac   if (numNodeSkip) {
12103f27d899SToby Isaac     PetscInt k;
12113f27d899SToby Isaac     PetscInt *tup;
12123f27d899SToby Isaac 
12133f27d899SToby Isaac     ierr = PetscMalloc1(dim * nNodes, &nodeCoords);CHKERRQ(ierr);
12143f27d899SToby Isaac     ierr = PetscMalloc1(dim + 1, &tup);CHKERRQ(ierr);
12153f27d899SToby Isaac     for (k = 0; k < nNodes; k++) {
12163f27d899SToby Isaac       PetscInt j, c;
12173f27d899SToby Isaac       PetscInt index;
12183f27d899SToby Isaac 
12193f27d899SToby Isaac       ierr = PetscDTIndexToBary(dim + 1, sum, k, tup);CHKERRQ(ierr);
12203f27d899SToby Isaac       for (j = 0; j < dim + 1; j++) tup[j] += numNodeSkip;
12213f27d899SToby Isaac       for (c = 0; c < Nk; c++) {
12223f27d899SToby Isaac         for (j = 0; j < dim + 1; j++) {
12233f27d899SToby Isaac           ni->nodeIdx[(k * Nk + c) * (dim + 1) + j] = tup[j] + 1;
12243f27d899SToby Isaac         }
12253f27d899SToby Isaac       }
12263f27d899SToby Isaac       ierr = PetscDTBaryToIndex(dim + 1, extraSum, tup, &index);CHKERRQ(ierr);
12273f27d899SToby Isaac       for (j = 0; j < dim; j++) nodeCoords[k * dim + j] = extraNodeCoords[index * dim + j];
12283f27d899SToby Isaac     }
12293f27d899SToby Isaac     ierr = PetscFree(tup);CHKERRQ(ierr);
12303f27d899SToby Isaac     ierr = PetscFree(extraNodeCoords);CHKERRQ(ierr);
12313f27d899SToby Isaac   } else {
12323f27d899SToby Isaac     PetscInt k;
12333f27d899SToby Isaac     PetscInt *tup;
12343f27d899SToby Isaac 
12353f27d899SToby Isaac     nodeCoords = extraNodeCoords;
12363f27d899SToby Isaac     ierr = PetscMalloc1(dim + 1, &tup);CHKERRQ(ierr);
12373f27d899SToby Isaac     for (k = 0; k < nNodes; k++) {
12383f27d899SToby Isaac       PetscInt j, c;
12393f27d899SToby Isaac 
12403f27d899SToby Isaac       ierr = PetscDTIndexToBary(dim + 1, sum, k, tup);CHKERRQ(ierr);
12413f27d899SToby Isaac       for (c = 0; c < Nk; c++) {
12423f27d899SToby Isaac         for (j = 0; j < dim + 1; j++) {
12433f27d899SToby Isaac           /* barycentric indices can have zeros, but we don't want to push forward zeros because it makes it harder to
12443f27d899SToby Isaac            * determine which nodes correspond to which under symmetries, so we increase by 1 */
12453f27d899SToby Isaac           ni->nodeIdx[(k * Nk + c) * (dim + 1) + j] = tup[j] + 1;
12463f27d899SToby Isaac         }
12473f27d899SToby Isaac       }
12483f27d899SToby Isaac     }
12493f27d899SToby Isaac     ierr = PetscFree(tup);CHKERRQ(ierr);
12503f27d899SToby Isaac   }
12513f27d899SToby Isaac   ierr = PetscQuadratureCreate(PETSC_COMM_SELF, &intNodes);CHKERRQ(ierr);
12523f27d899SToby Isaac   ierr = PetscQuadratureSetData(intNodes, dim, 0, nNodes, nodeCoords, NULL);CHKERRQ(ierr);
12533f27d899SToby Isaac   ierr = MatCreateSeqAIJ(PETSC_COMM_SELF, nNodes * Nk, nNodes * Nk, Nk, NULL, &intMat);CHKERRQ(ierr);
12543f27d899SToby Isaac   ierr = MatSetOption(intMat,MAT_IGNORE_ZERO_ENTRIES,PETSC_FALSE);CHKERRQ(ierr);
12553f27d899SToby Isaac   for (j = 0; j < nNodes * Nk; j++) {
12563f27d899SToby Isaac     PetscInt rem = j % Nk;
12573f27d899SToby Isaac     PetscInt a, aprev = j - rem;
12583f27d899SToby Isaac     PetscInt anext = aprev + Nk;
12593f27d899SToby Isaac 
12603f27d899SToby Isaac     for (a = aprev; a < anext; a++) {
12613f27d899SToby Isaac       ierr = MatSetValue(intMat, j, a, (a == j) ? 1. : 0., INSERT_VALUES);CHKERRQ(ierr);
12623f27d899SToby Isaac     }
12633f27d899SToby Isaac   }
12643f27d899SToby Isaac   ierr = MatAssemblyBegin(intMat, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
12653f27d899SToby Isaac   ierr = MatAssemblyEnd(intMat, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
12663f27d899SToby Isaac   *iNodes = intNodes;
12673f27d899SToby Isaac   *iMat = intMat;
12683f27d899SToby Isaac   *nodeIndices = ni;
12693f27d899SToby Isaac   PetscFunctionReturn(0);
12703f27d899SToby Isaac }
12713f27d899SToby Isaac 
12723f27d899SToby Isaac static PetscErrorCode PetscDualSpaceLagrangeCreateAllNodeIdx(PetscDualSpace sp)
12733f27d899SToby Isaac {
12743f27d899SToby Isaac   DM             dm;
12753f27d899SToby Isaac   PetscInt       dim, nDofs;
12763f27d899SToby Isaac   PetscSection   section;
12773f27d899SToby Isaac   PetscInt       pStart, pEnd, p;
12783f27d899SToby Isaac   PetscInt       formDegree, Nk;
12793f27d899SToby Isaac   PetscInt       nodeIdxDim, spintdim;
12803f27d899SToby Isaac   PetscDualSpace_Lag *lag;
12813f27d899SToby Isaac   PetscLagNodeIndices ni, verti;
12823f27d899SToby Isaac   PetscErrorCode ierr;
12833f27d899SToby Isaac 
12843f27d899SToby Isaac   PetscFunctionBegin;
12853f27d899SToby Isaac   lag = (PetscDualSpace_Lag *) sp->data;
12863f27d899SToby Isaac   verti = lag->vertIndices;
12873f27d899SToby Isaac   ierr = PetscDualSpaceGetDM(sp, &dm);CHKERRQ(ierr);
12883f27d899SToby Isaac   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
12893f27d899SToby Isaac   ierr = PetscDualSpaceGetFormDegree(sp, &formDegree);CHKERRQ(ierr);
12903f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(formDegree), &Nk);CHKERRQ(ierr);
12913f27d899SToby Isaac   ierr = PetscDualSpaceGetSection(sp, &section);CHKERRQ(ierr);
12923f27d899SToby Isaac   ierr = PetscSectionGetStorageSize(section, &nDofs);CHKERRQ(ierr);
12933f27d899SToby Isaac   ierr = PetscNew(&ni);CHKERRQ(ierr);
12943f27d899SToby Isaac   ni->nodeIdxDim = nodeIdxDim = verti->nodeIdxDim;
12953f27d899SToby Isaac   ni->nodeVecDim = Nk;
12963f27d899SToby Isaac   ni->nNodes = nDofs;
12973f27d899SToby Isaac   ni->refct = 1;
12983f27d899SToby Isaac   ierr = PetscMalloc1(nodeIdxDim * nDofs, &(ni->nodeIdx));CHKERRQ(ierr);
12993f27d899SToby Isaac   ierr = PetscMalloc1(Nk * nDofs, &(ni->nodeVec));CHKERRQ(ierr);
13003f27d899SToby Isaac   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
13013f27d899SToby Isaac   ierr = PetscSectionGetDof(section, 0, &spintdim);CHKERRQ(ierr);
13023f27d899SToby Isaac   if (spintdim) {
13033f27d899SToby Isaac     ierr = PetscArraycpy(ni->nodeIdx, lag->intNodeIndices->nodeIdx, spintdim * nodeIdxDim);CHKERRQ(ierr);
13043f27d899SToby Isaac     ierr = PetscArraycpy(ni->nodeVec, lag->intNodeIndices->nodeVec, spintdim * Nk);CHKERRQ(ierr);
13053f27d899SToby Isaac   }
13063f27d899SToby Isaac   for (p = pStart + 1; p < pEnd; p++) {
13073f27d899SToby Isaac     PetscDualSpace psp = sp->pointSpaces[p];
13083f27d899SToby Isaac     PetscDualSpace_Lag *plag;
13093f27d899SToby Isaac     PetscInt dof, off;
13103f27d899SToby Isaac 
13113f27d899SToby Isaac     ierr = PetscSectionGetDof(section, p, &dof);CHKERRQ(ierr);
13123f27d899SToby Isaac     if (!dof) continue;
13133f27d899SToby Isaac     plag = (PetscDualSpace_Lag *) psp->data;
13143f27d899SToby Isaac     ierr = PetscSectionGetOffset(section, p, &off);CHKERRQ(ierr);
13153f27d899SToby Isaac     ierr = PetscLagNodeIndicesPushForward(dm, verti, p, plag->vertIndices, plag->intNodeIndices, 0, formDegree, &(ni->nodeIdx[off * nodeIdxDim]), &(ni->nodeVec[off * Nk]));CHKERRQ(ierr);
13163f27d899SToby Isaac   }
13173f27d899SToby Isaac   lag->allNodeIndices = ni;
13183f27d899SToby Isaac   PetscFunctionReturn(0);
13193f27d899SToby Isaac }
13203f27d899SToby Isaac 
13213f27d899SToby Isaac static PetscErrorCode PetscDualSpaceCreateAllDataFromInteriorData(PetscDualSpace sp)
13223f27d899SToby Isaac {
13233f27d899SToby Isaac   DM               dm;
13243f27d899SToby Isaac   PetscSection     section;
13253f27d899SToby Isaac   PetscInt         pStart, pEnd, p, k, Nk, dim, Nc;
13263f27d899SToby Isaac   PetscInt         nNodes;
13273f27d899SToby Isaac   PetscInt         countNodes;
13283f27d899SToby Isaac   Mat              allMat;
13293f27d899SToby Isaac   PetscQuadrature  allNodes;
13303f27d899SToby Isaac   PetscInt         nDofs;
13313f27d899SToby Isaac   PetscInt         maxNzforms, j;
13323f27d899SToby Isaac   PetscScalar      *work;
13333f27d899SToby Isaac   PetscReal        *L, *J, *Jinv, *v0, *pv0;
13343f27d899SToby Isaac   PetscInt         *iwork;
13353f27d899SToby Isaac   PetscReal        *nodes;
13363f27d899SToby Isaac   PetscErrorCode   ierr;
13373f27d899SToby Isaac 
13383f27d899SToby Isaac   PetscFunctionBegin;
13393f27d899SToby Isaac   ierr = PetscDualSpaceGetDM(sp, &dm);CHKERRQ(ierr);
13403f27d899SToby Isaac   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
13413f27d899SToby Isaac   ierr = PetscDualSpaceGetSection(sp, &section);CHKERRQ(ierr);
13423f27d899SToby Isaac   ierr = PetscSectionGetStorageSize(section, &nDofs);CHKERRQ(ierr);
13433f27d899SToby Isaac   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
13443f27d899SToby Isaac   ierr = PetscDualSpaceGetFormDegree(sp, &k);CHKERRQ(ierr);
13453f27d899SToby Isaac   ierr = PetscDualSpaceGetNumComponents(sp, &Nc);CHKERRQ(ierr);
13463f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(k), &Nk);CHKERRQ(ierr);
13473f27d899SToby Isaac   for (p = pStart, nNodes = 0, maxNzforms = 0; p < pEnd; p++) {
13483f27d899SToby Isaac     PetscDualSpace  psp;
13493f27d899SToby Isaac     DM              pdm;
13503f27d899SToby Isaac     PetscInt        pdim, pNk;
13513f27d899SToby Isaac     PetscQuadrature intNodes;
13523f27d899SToby Isaac     Mat intMat;
13533f27d899SToby Isaac 
13543f27d899SToby Isaac     ierr = PetscDualSpaceGetPointSubspace(sp, p, &psp);CHKERRQ(ierr);
13553f27d899SToby Isaac     if (!psp) continue;
13563f27d899SToby Isaac     ierr = PetscDualSpaceGetDM(psp, &pdm);CHKERRQ(ierr);
13573f27d899SToby Isaac     ierr = DMGetDimension(pdm, &pdim);CHKERRQ(ierr);
13583f27d899SToby Isaac     if (pdim < PetscAbsInt(k)) continue;
13593f27d899SToby Isaac     ierr = PetscDTBinomialInt(pdim, PetscAbsInt(k), &pNk);CHKERRQ(ierr);
13603f27d899SToby Isaac     ierr = PetscDualSpaceGetInteriorData(psp, &intNodes, &intMat);CHKERRQ(ierr);
13613f27d899SToby Isaac     if (intNodes) {
13623f27d899SToby Isaac       PetscInt nNodesp;
13633f27d899SToby Isaac 
13643f27d899SToby Isaac       ierr = PetscQuadratureGetData(intNodes, NULL, NULL, &nNodesp, NULL, NULL);CHKERRQ(ierr);
13653f27d899SToby Isaac       nNodes += nNodesp;
13663f27d899SToby Isaac     }
13673f27d899SToby Isaac     if (intMat) {
13683f27d899SToby Isaac       PetscInt maxNzsp;
13693f27d899SToby Isaac       PetscInt maxNzformsp;
13703f27d899SToby Isaac 
13713f27d899SToby Isaac       ierr = MatSeqAIJGetMaxRowNonzeros(intMat, &maxNzsp);CHKERRQ(ierr);
13723f27d899SToby Isaac       if (maxNzsp % pNk) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "interior matrix is not laid out as blocks of k-forms");
13733f27d899SToby Isaac       maxNzformsp = maxNzsp / pNk;
13743f27d899SToby Isaac       maxNzforms = PetscMax(maxNzforms, maxNzformsp);
13753f27d899SToby Isaac     }
13763f27d899SToby Isaac   }
13773f27d899SToby Isaac   ierr = MatCreateSeqAIJ(PETSC_COMM_SELF, nDofs, nNodes * Nc, maxNzforms * Nk, NULL, &allMat);CHKERRQ(ierr);
13783f27d899SToby Isaac   ierr = MatSetOption(allMat,MAT_IGNORE_ZERO_ENTRIES,PETSC_FALSE);CHKERRQ(ierr);
13793f27d899SToby Isaac   ierr = PetscMalloc7(dim, &v0, dim, &pv0, dim * dim, &J, dim * dim, &Jinv, Nk * Nk, &L, maxNzforms * Nk, &work, maxNzforms * Nk, &iwork);CHKERRQ(ierr);
13803f27d899SToby Isaac   for (j = 0; j < dim; j++) pv0[j] = -1.;
13813f27d899SToby Isaac   ierr = PetscMalloc1(dim * nNodes, &nodes);CHKERRQ(ierr);
13823f27d899SToby Isaac   for (p = pStart, countNodes = 0; p < pEnd; p++) {
13833f27d899SToby Isaac     PetscDualSpace  psp;
13843f27d899SToby Isaac     PetscQuadrature intNodes;
13853f27d899SToby Isaac     DM pdm;
13863f27d899SToby Isaac     PetscInt pdim, pNk;
13873f27d899SToby Isaac     PetscInt countNodesIn = countNodes;
13883f27d899SToby Isaac     PetscReal detJ;
13893f27d899SToby Isaac     Mat intMat;
13903f27d899SToby Isaac 
13913f27d899SToby Isaac     ierr = PetscDualSpaceGetPointSubspace(sp, p, &psp);CHKERRQ(ierr);
13923f27d899SToby Isaac     if (!psp) continue;
13933f27d899SToby Isaac     ierr = PetscDualSpaceGetDM(psp, &pdm);CHKERRQ(ierr);
13943f27d899SToby Isaac     ierr = DMGetDimension(pdm, &pdim);CHKERRQ(ierr);
13953f27d899SToby Isaac     if (pdim < PetscAbsInt(k)) continue;
13963f27d899SToby Isaac     ierr = PetscDualSpaceGetInteriorData(psp, &intNodes, &intMat);CHKERRQ(ierr);
13973f27d899SToby Isaac     if (intNodes == NULL && intMat == NULL) continue;
13983f27d899SToby Isaac     ierr = PetscDTBinomialInt(pdim, PetscAbsInt(k), &pNk);CHKERRQ(ierr);
13993f27d899SToby Isaac     if (p) {
14003f27d899SToby Isaac       ierr = DMPlexComputeCellGeometryAffineFEM(dm, p, v0, J, Jinv, &detJ);CHKERRQ(ierr);
14013f27d899SToby Isaac     } else { /* identity */
14023f27d899SToby Isaac       PetscInt i,j;
14033f27d899SToby Isaac 
14043f27d899SToby Isaac       for (i = 0; i < dim; i++) for (j = 0; j < dim; j++) J[i * dim + j] = Jinv[i * dim + j] = 0.;
14053f27d899SToby Isaac       for (i = 0; i < dim; i++) J[i * dim + i] = Jinv[i * dim + i] = 1.;
14063f27d899SToby Isaac       for (i = 0; i < dim; i++) v0[i] = -1.;
14073f27d899SToby Isaac     }
14083f27d899SToby Isaac     if (pdim != dim) { /* compactify Jacobian */
14093f27d899SToby Isaac       PetscInt i, j;
14103f27d899SToby Isaac 
14113f27d899SToby Isaac       for (i = 0; i < dim; i++) for (j = 0; j < pdim; j++) J[i * pdim + j] = J[i * dim + j];
14123f27d899SToby Isaac     }
14133f27d899SToby Isaac     ierr = PetscDTAltVPullbackMatrix(pdim, dim, J, k, L);CHKERRQ(ierr);
14143f27d899SToby Isaac     if (intNodes) { /* "push forward" dof by pulling back a k-form to be evaluated on the point: multiply on the right by L^T */
14153f27d899SToby Isaac       PetscInt nNodesp;
14163f27d899SToby Isaac       const PetscReal *nodesp;
14173f27d899SToby Isaac       PetscInt j;
14183f27d899SToby Isaac 
14193f27d899SToby Isaac       ierr = PetscQuadratureGetData(intNodes, NULL, NULL, &nNodesp, &nodesp, NULL);CHKERRQ(ierr);
14203f27d899SToby Isaac       for (j = 0; j < nNodesp; j++, countNodes++) {
14213f27d899SToby Isaac         PetscInt d, e;
14223f27d899SToby Isaac 
14233f27d899SToby Isaac         for (d = 0; d < dim; d++) {
14243f27d899SToby Isaac           nodes[countNodes * dim + d] = v0[d];
14253f27d899SToby Isaac           for (e = 0; e < pdim; e++) {
14263f27d899SToby Isaac             nodes[countNodes * dim + d] += J[d * pdim + e] * (nodesp[j * pdim + e] - pv0[e]);
14273f27d899SToby Isaac           }
14283f27d899SToby Isaac         }
14293f27d899SToby Isaac       }
14303f27d899SToby Isaac     }
14313f27d899SToby Isaac     if (intMat) {
14323f27d899SToby Isaac       PetscInt nrows;
14333f27d899SToby Isaac       PetscInt off;
14343f27d899SToby Isaac 
14353f27d899SToby Isaac       ierr = PetscSectionGetDof(section, p, &nrows);CHKERRQ(ierr);
14363f27d899SToby Isaac       ierr = PetscSectionGetOffset(section, p, &off);CHKERRQ(ierr);
14373f27d899SToby Isaac       for (j = 0; j < nrows; j++) {
14383f27d899SToby Isaac         PetscInt ncols;
14393f27d899SToby Isaac         const PetscInt *cols;
14403f27d899SToby Isaac         const PetscScalar *vals;
14413f27d899SToby Isaac         PetscInt l, d, e;
14423f27d899SToby Isaac         PetscInt row = j + off;
14433f27d899SToby Isaac 
14443f27d899SToby Isaac         ierr = MatGetRow(intMat, j, &ncols, &cols, &vals);CHKERRQ(ierr);
14453f27d899SToby Isaac         if (ncols % pNk) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "interior matrix is not laid out as blocks of k-forms");
14463f27d899SToby Isaac         for (l = 0; l < ncols / pNk; l++) {
14473f27d899SToby Isaac           PetscInt blockcol;
14483f27d899SToby Isaac 
14493f27d899SToby Isaac           for (d = 0; d < pNk; d++) {
14503f27d899SToby Isaac             if ((cols[l * pNk + d] % pNk) != d) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "interior matrix is not laid out as blocks of k-forms");
14513f27d899SToby Isaac           }
14523f27d899SToby Isaac           blockcol = cols[l * pNk] / pNk;
14533f27d899SToby Isaac           for (d = 0; d < Nk; d++) {
14543f27d899SToby Isaac             iwork[l * Nk + d] = (blockcol + countNodesIn) * Nk + d;
14553f27d899SToby Isaac           }
14563f27d899SToby Isaac           for (d = 0; d < Nk; d++) work[l * Nk + d] = 0.;
14573f27d899SToby Isaac           for (d = 0; d < Nk; d++) {
14583f27d899SToby Isaac             for (e = 0; e < pNk; e++) {
14593f27d899SToby Isaac               /* "push forward" dof by pulling back a k-form to be evaluated on the point: multiply on the right by L */
14603f27d899SToby Isaac               work[l * Nk + d] += vals[l * pNk + e] * L[e * pNk + d];
14613f27d899SToby Isaac             }
14623f27d899SToby Isaac           }
14633f27d899SToby Isaac         }
14643f27d899SToby Isaac         ierr = MatSetValues(allMat, 1, &row, (ncols / pNk) * Nk, iwork, work, INSERT_VALUES);CHKERRQ(ierr);
14653f27d899SToby Isaac         ierr = MatRestoreRow(intMat, j, &ncols, &cols, &vals);CHKERRQ(ierr);
14663f27d899SToby Isaac       }
14673f27d899SToby Isaac     }
14683f27d899SToby Isaac   }
14693f27d899SToby Isaac   ierr = MatAssemblyBegin(allMat, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
14703f27d899SToby Isaac   ierr = MatAssemblyEnd(allMat, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
14713f27d899SToby Isaac   ierr = PetscQuadratureCreate(PETSC_COMM_SELF, &allNodes);CHKERRQ(ierr);
14723f27d899SToby Isaac   ierr = PetscQuadratureSetData(allNodes, dim, 0, nNodes, nodes, NULL);CHKERRQ(ierr);
14733f27d899SToby Isaac   ierr = PetscFree7(v0, pv0, J, Jinv, L, work, iwork);CHKERRQ(ierr);
14743f27d899SToby Isaac   ierr = MatDestroy(&(sp->allMat));CHKERRQ(ierr);
14753f27d899SToby Isaac   sp->allMat = allMat;
14763f27d899SToby Isaac   ierr = PetscQuadratureDestroy(&(sp->allNodes));CHKERRQ(ierr);
14773f27d899SToby Isaac   sp->allNodes = allNodes;
14783f27d899SToby Isaac   PetscFunctionReturn(0);
14793f27d899SToby Isaac }
14803f27d899SToby Isaac 
14813f27d899SToby Isaac static PetscErrorCode PetscDualSpaceComputeFunctionalsFromAllData(PetscDualSpace sp)
14823f27d899SToby Isaac {
14833f27d899SToby Isaac   PetscQuadrature allNodes;
14843f27d899SToby Isaac   Mat             allMat;
14853f27d899SToby Isaac   PetscInt        nDofs;
14863f27d899SToby Isaac   PetscInt        dim, k, Nk, Nc, f;
14873f27d899SToby Isaac   DM              dm;
14883f27d899SToby Isaac   PetscInt        nNodes, spdim;
14893f27d899SToby Isaac   const PetscReal *nodes = NULL;
14903f27d899SToby Isaac   PetscSection    section;
14913f27d899SToby Isaac   PetscErrorCode  ierr;
14923f27d899SToby Isaac 
14933f27d899SToby Isaac   PetscFunctionBegin;
14943f27d899SToby Isaac   ierr = PetscDualSpaceGetDM(sp, &dm);CHKERRQ(ierr);
14953f27d899SToby Isaac   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
14963f27d899SToby Isaac   ierr = PetscDualSpaceGetNumComponents(sp, &Nc);CHKERRQ(ierr);
14973f27d899SToby Isaac   ierr = PetscDualSpaceGetFormDegree(sp, &k);CHKERRQ(ierr);
14983f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim, PetscAbsInt(k), &Nk);CHKERRQ(ierr);
14993f27d899SToby Isaac   ierr = PetscDualSpaceGetAllData(sp, &allNodes, &allMat);CHKERRQ(ierr);
15003f27d899SToby Isaac   nNodes = 0;
15013f27d899SToby Isaac   if (allNodes) {
15023f27d899SToby Isaac     ierr = PetscQuadratureGetData(allNodes, NULL, NULL, &nNodes, &nodes, NULL);CHKERRQ(ierr);
15033f27d899SToby Isaac   }
15043f27d899SToby Isaac   ierr = MatGetSize(allMat, &nDofs, NULL);CHKERRQ(ierr);
15053f27d899SToby Isaac   ierr = PetscDualSpaceGetSection(sp, &section);CHKERRQ(ierr);
15063f27d899SToby Isaac   ierr = PetscSectionGetStorageSize(section, &spdim);CHKERRQ(ierr);
15073f27d899SToby Isaac   if (spdim != nDofs) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "incompatible all matrix size");
15083f27d899SToby Isaac   ierr = PetscMalloc1(nDofs, &(sp->functional));CHKERRQ(ierr);
15093f27d899SToby Isaac   for (f = 0; f < nDofs; f++) {
15103f27d899SToby Isaac     PetscInt ncols, c;
15113f27d899SToby Isaac     const PetscInt *cols;
15123f27d899SToby Isaac     const PetscScalar *vals;
15133f27d899SToby Isaac     PetscReal *nodesf;
15143f27d899SToby Isaac     PetscReal *weightsf;
15153f27d899SToby Isaac     PetscInt nNodesf;
15163f27d899SToby Isaac     PetscInt countNodes;
15173f27d899SToby Isaac 
15183f27d899SToby Isaac     ierr = MatGetRow(allMat, f, &ncols, &cols, &vals);CHKERRQ(ierr);
15193f27d899SToby Isaac     if (ncols % Nk) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "all matrix is not laid out as blocks of k-forms");
15203f27d899SToby Isaac     for (c = 1, nNodesf = 1; c < ncols; c++) {
15213f27d899SToby Isaac       if ((cols[c] / Nc) != (cols[c-1] / Nc)) nNodesf++;
15223f27d899SToby Isaac     }
15233f27d899SToby Isaac     ierr = PetscMalloc1(dim * nNodesf, &nodesf);CHKERRQ(ierr);
15243f27d899SToby Isaac     ierr = PetscMalloc1(Nc * nNodesf, &weightsf);CHKERRQ(ierr);
15253f27d899SToby Isaac     for (c = 0, countNodes = 0; c < ncols; c++) {
15263f27d899SToby Isaac       if (!c || ((cols[c] / Nc) != (cols[c-1] / Nc))) {
15273f27d899SToby Isaac         PetscInt d;
15283f27d899SToby Isaac 
15293f27d899SToby Isaac         for (d = 0; d < Nc; d++) {
15303f27d899SToby Isaac           weightsf[countNodes * Nc + d] = 0.;
15313f27d899SToby Isaac         }
15323f27d899SToby Isaac         for (d = 0; d < dim; d++) {
15333f27d899SToby Isaac           nodesf[countNodes * dim + d] = nodes[(cols[c] / Nc) * dim + d];
15343f27d899SToby Isaac         }
15353f27d899SToby Isaac         countNodes++;
15363f27d899SToby Isaac       }
15373f27d899SToby Isaac       weightsf[(countNodes - 1) * Nc + (cols[c] % Nc)] = PetscRealPart(vals[c]);
15383f27d899SToby Isaac     }
15393f27d899SToby Isaac     ierr = PetscQuadratureCreate(PETSC_COMM_SELF, &(sp->functional[f]));CHKERRQ(ierr);
15403f27d899SToby Isaac     ierr = PetscQuadratureSetData(sp->functional[f], dim, Nc, nNodesf, nodesf, weightsf);CHKERRQ(ierr);
15413f27d899SToby Isaac     ierr = MatRestoreRow(allMat, f, &ncols, &cols, &vals);CHKERRQ(ierr);
15423f27d899SToby Isaac   }
15433f27d899SToby Isaac   PetscFunctionReturn(0);
15443f27d899SToby Isaac }
15453f27d899SToby Isaac 
15463f27d899SToby Isaac /* take a matrix meant for k-forms and expand it to one for Ncopies */
15473f27d899SToby Isaac static PetscErrorCode PetscDualSpaceLagrangeMatrixCreateCopies(Mat A, PetscInt Nk, PetscInt Ncopies, Mat *Abs)
15483f27d899SToby Isaac {
15493f27d899SToby Isaac   PetscInt       m, n, i, j, k;
15503f27d899SToby Isaac   PetscInt       maxnnz, *nnz, *iwork;
15513f27d899SToby Isaac   Mat            Ac;
15523f27d899SToby Isaac   PetscErrorCode ierr;
15533f27d899SToby Isaac 
15543f27d899SToby Isaac   PetscFunctionBegin;
15553f27d899SToby Isaac   ierr = MatGetSize(A, &m, &n);CHKERRQ(ierr);
15563f27d899SToby Isaac   if (n % Nk) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Number of columns in A %D is not a multiple of Nk %D", n, Nk);
15573f27d899SToby Isaac   ierr = PetscMalloc1(m * Ncopies, &nnz);CHKERRQ(ierr);
15583f27d899SToby Isaac   for (i = 0, maxnnz = 0; i < m; i++) {
15593f27d899SToby Isaac     PetscInt innz;
15603f27d899SToby Isaac     ierr = MatGetRow(A, i, &innz, NULL, NULL);CHKERRQ(ierr);
15613f27d899SToby Isaac     if (innz % Nk) SETERRQ2(PETSC_COMM_SELF, PETSC_ERR_PLIB, "A row %D nnzs is not a multiple of Nk %D", innz, Nk);
15623f27d899SToby Isaac     for (j = 0; j < Ncopies; j++) nnz[i * Ncopies + j] = innz;
15633f27d899SToby Isaac     maxnnz = PetscMax(maxnnz, innz);
15643f27d899SToby Isaac   }
15653f27d899SToby Isaac   ierr = MatCreateSeqAIJ(PETSC_COMM_SELF, m * Ncopies, n * Ncopies, 0, nnz, &Ac);CHKERRQ(ierr);
15663f27d899SToby Isaac   ierr = MatSetOption(Ac, MAT_IGNORE_ZERO_ENTRIES, PETSC_FALSE);CHKERRQ(ierr);
15673f27d899SToby Isaac   ierr = PetscFree(nnz);CHKERRQ(ierr);
15683f27d899SToby Isaac   ierr = PetscMalloc1(maxnnz, &iwork);CHKERRQ(ierr);
15693f27d899SToby Isaac   for (i = 0; i < m; i++) {
15703f27d899SToby Isaac     PetscInt innz;
15713f27d899SToby Isaac     const PetscInt    *cols;
15723f27d899SToby Isaac     const PetscScalar *vals;
15733f27d899SToby Isaac 
15743f27d899SToby Isaac     ierr = MatGetRow(A, i, &innz, &cols, &vals);CHKERRQ(ierr);
15753f27d899SToby Isaac     for (j = 0; j < innz; j++) iwork[j] = (cols[j] / Nk) * (Nk * Ncopies) + (cols[j] % Nk);
15763f27d899SToby Isaac     for (j = 0; j < Ncopies; j++) {
15773f27d899SToby Isaac       PetscInt row = i * Ncopies + j;
15783f27d899SToby Isaac 
15793f27d899SToby Isaac       ierr = MatSetValues(Ac, 1, &row, innz, iwork, vals, INSERT_VALUES);CHKERRQ(ierr);
15803f27d899SToby Isaac       for (k = 0; k < innz; k++) iwork[k] += Nk;
15813f27d899SToby Isaac     }
15823f27d899SToby Isaac     ierr = MatRestoreRow(A, i, &innz, &cols, &vals);CHKERRQ(ierr);
15833f27d899SToby Isaac   }
15843f27d899SToby Isaac   ierr = PetscFree(iwork);CHKERRQ(ierr);
15853f27d899SToby Isaac   ierr = MatAssemblyBegin(Ac, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
15863f27d899SToby Isaac   ierr = MatAssemblyEnd(Ac, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
15873f27d899SToby Isaac   *Abs = Ac;
15883f27d899SToby Isaac   PetscFunctionReturn(0);
15893f27d899SToby Isaac }
15903f27d899SToby Isaac 
15913f27d899SToby Isaac static PetscErrorCode DMPlexPointIsTensor_Internal_Given(DM dm, PetscInt p, PetscInt f, PetscInt f2, PetscBool *isTensor)
15923f27d899SToby Isaac {
15933f27d899SToby Isaac   PetscInt        coneSize, c;
15943f27d899SToby Isaac   const PetscInt *cone;
15953f27d899SToby Isaac   const PetscInt *fCone;
15963f27d899SToby Isaac   const PetscInt *f2Cone;
15973f27d899SToby Isaac   PetscInt        fs[2];
15983f27d899SToby Isaac   PetscInt        meetSize, nmeet;
15993f27d899SToby Isaac   const PetscInt *meet;
16003f27d899SToby Isaac   PetscErrorCode  ierr;
16013f27d899SToby Isaac 
16023f27d899SToby Isaac   PetscFunctionBegin;
16033f27d899SToby Isaac   fs[0] = f;
16043f27d899SToby Isaac   fs[1] = f2;
16053f27d899SToby Isaac   ierr = DMPlexGetMeet(dm, 2, fs, &meetSize, &meet);CHKERRQ(ierr);
16063f27d899SToby Isaac   nmeet = meetSize;
16073f27d899SToby Isaac   ierr = DMPlexRestoreMeet(dm, 2, fs, &meetSize, &meet);CHKERRQ(ierr);
16083f27d899SToby Isaac   if (nmeet) {
16093f27d899SToby Isaac     *isTensor = PETSC_FALSE;
16103f27d899SToby Isaac     PetscFunctionReturn(0);
16113f27d899SToby Isaac   }
16123f27d899SToby Isaac   ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr);
16133f27d899SToby Isaac   ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr);
16143f27d899SToby Isaac   ierr = DMPlexGetCone(dm, f, &fCone);CHKERRQ(ierr);
16153f27d899SToby Isaac   ierr = DMPlexGetCone(dm, f2, &f2Cone);CHKERRQ(ierr);
16163f27d899SToby Isaac   for (c = 0; c < coneSize; c++) {
16173f27d899SToby Isaac     PetscInt e, ef;
16183f27d899SToby Isaac     PetscInt d = -1, d2 = -1;
16193f27d899SToby Isaac     PetscInt dcount, d2count;
16203f27d899SToby Isaac     PetscInt t = cone[c];
16213f27d899SToby Isaac     PetscInt tConeSize;
16223f27d899SToby Isaac     PetscBool tIsTensor;
16233f27d899SToby Isaac     const PetscInt *tCone;
16243f27d899SToby Isaac 
16253f27d899SToby Isaac     if (t == f || t == f2) continue;
16263f27d899SToby Isaac     ierr = DMPlexGetConeSize(dm, t, &tConeSize);CHKERRQ(ierr);
16273f27d899SToby Isaac     ierr = DMPlexGetCone(dm, t, &tCone);CHKERRQ(ierr);
16283f27d899SToby Isaac 
16293f27d899SToby Isaac     dcount = 0;
16303f27d899SToby Isaac     d2count = 0;
16313f27d899SToby Isaac     for (e = 0; e < tConeSize; e++) {
16323f27d899SToby Isaac       PetscInt q = tCone[e];
16333f27d899SToby Isaac       for (ef = 0; ef < coneSize - 2; ef++) {
16343f27d899SToby Isaac         if (fCone[ef] == q) {
16353f27d899SToby Isaac           if (dcount) {
16363f27d899SToby Isaac             *isTensor = PETSC_FALSE;
16373f27d899SToby Isaac             PetscFunctionReturn(0);
16383f27d899SToby Isaac           }
16393f27d899SToby Isaac           d = q;
16403f27d899SToby Isaac           dcount++;
16413f27d899SToby Isaac         } else if (f2Cone[ef] == q) {
16423f27d899SToby Isaac           if (d2count) {
16433f27d899SToby Isaac             *isTensor = PETSC_FALSE;
16443f27d899SToby Isaac             PetscFunctionReturn(0);
16453f27d899SToby Isaac           }
16463f27d899SToby Isaac           d2 = q;
16473f27d899SToby Isaac           d2count++;
16483f27d899SToby Isaac         }
16493f27d899SToby Isaac       }
16503f27d899SToby Isaac     }
16513f27d899SToby Isaac     ierr = DMPlexPointIsTensor_Internal_Given(dm, t, d, d2, &tIsTensor);CHKERRQ(ierr);
16523f27d899SToby Isaac     if (!tIsTensor) {
16533f27d899SToby Isaac       *isTensor = PETSC_FALSE;
16543f27d899SToby Isaac       PetscFunctionReturn(0);
16553f27d899SToby Isaac     }
16563f27d899SToby Isaac   }
16573f27d899SToby Isaac   *isTensor = PETSC_TRUE;
16583f27d899SToby Isaac   PetscFunctionReturn(0);
16593f27d899SToby Isaac }
16603f27d899SToby Isaac 
16613f27d899SToby Isaac static PetscErrorCode DMPlexPointIsTensor_Internal(DM dm, PetscInt p, PetscBool *isTensor, PetscInt *endA, PetscInt *endB)
16623f27d899SToby Isaac {
16633f27d899SToby Isaac   PetscInt        coneSize, c, c2;
16643f27d899SToby Isaac   const PetscInt *cone;
16653f27d899SToby Isaac   PetscErrorCode  ierr;
16663f27d899SToby Isaac 
16673f27d899SToby Isaac   PetscFunctionBegin;
16683f27d899SToby Isaac   ierr = DMPlexGetConeSize(dm, p, &coneSize);CHKERRQ(ierr);
16693f27d899SToby Isaac   if (!coneSize) {
16703f27d899SToby Isaac     if (isTensor) *isTensor = PETSC_FALSE;
16713f27d899SToby Isaac     if (endA) *endA = -1;
16723f27d899SToby Isaac     if (endB) *endB = -1;
16733f27d899SToby Isaac   }
16743f27d899SToby Isaac   ierr = DMPlexGetCone(dm, p, &cone);CHKERRQ(ierr);
16753f27d899SToby Isaac   for (c = 0; c < coneSize; c++) {
16763f27d899SToby Isaac     PetscInt f = cone[c];
16773f27d899SToby Isaac     PetscInt fConeSize;
16783f27d899SToby Isaac 
16793f27d899SToby Isaac     ierr = DMPlexGetConeSize(dm, f, &fConeSize);CHKERRQ(ierr);
16803f27d899SToby Isaac     if (fConeSize != coneSize - 2) continue;
16813f27d899SToby Isaac 
16823f27d899SToby Isaac     for (c2 = c + 1; c2 < coneSize; c2++) {
16833f27d899SToby Isaac       PetscInt  f2 = cone[c2];
16843f27d899SToby Isaac       PetscBool isTensorff2;
16853f27d899SToby Isaac       PetscInt f2ConeSize;
16863f27d899SToby Isaac 
16873f27d899SToby Isaac       ierr = DMPlexGetConeSize(dm, f2, &f2ConeSize);CHKERRQ(ierr);
16883f27d899SToby Isaac       if (f2ConeSize != coneSize - 2) continue;
16893f27d899SToby Isaac 
16903f27d899SToby Isaac       ierr = DMPlexPointIsTensor_Internal_Given(dm, p, f, f2, &isTensorff2);CHKERRQ(ierr);
16913f27d899SToby Isaac       if (isTensorff2) {
16923f27d899SToby Isaac         if (isTensor) *isTensor = PETSC_TRUE;
16933f27d899SToby Isaac         if (endA) *endA = f;
16943f27d899SToby Isaac         if (endB) *endB = f2;
16953f27d899SToby Isaac         PetscFunctionReturn(0);
16963f27d899SToby Isaac       }
16973f27d899SToby Isaac     }
16983f27d899SToby Isaac   }
16993f27d899SToby Isaac   if (isTensor) *isTensor = PETSC_FALSE;
17003f27d899SToby Isaac   if (endA) *endA = -1;
17013f27d899SToby Isaac   if (endB) *endB = -1;
17023f27d899SToby Isaac   PetscFunctionReturn(0);
17033f27d899SToby Isaac }
17043f27d899SToby Isaac 
17053f27d899SToby Isaac static PetscErrorCode DMPlexPointIsTensor(DM dm, PetscInt p, PetscBool *isTensor, PetscInt *endA, PetscInt *endB)
17063f27d899SToby Isaac {
17073f27d899SToby Isaac   DMPlexInterpolatedFlag interpolated;
17083f27d899SToby Isaac   PetscErrorCode ierr;
17093f27d899SToby Isaac 
17103f27d899SToby Isaac   PetscFunctionBegin;
17113f27d899SToby Isaac   ierr = DMPlexIsInterpolated(dm, &interpolated);CHKERRQ(ierr);
17123f27d899SToby Isaac   if (interpolated != DMPLEX_INTERPOLATED_FULL) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONGSTATE, "Only for interpolated DMPlex's");
17133f27d899SToby Isaac   ierr = DMPlexPointIsTensor_Internal(dm, p, isTensor, endA, endB);CHKERRQ(ierr);
17143f27d899SToby Isaac   PetscFunctionReturn(0);
17153f27d899SToby Isaac }
17163f27d899SToby Isaac 
17173f27d899SToby Isaac static PetscErrorCode MatPermuteByNodeIdx(Mat A, PetscLagNodeIndices ni, Mat *Aperm)
17183f27d899SToby Isaac {
17193f27d899SToby Isaac   PetscInt       m, n, i, j;
17203f27d899SToby Isaac   PetscInt       nodeIdxDim = ni->nodeIdxDim;
17213f27d899SToby Isaac   PetscInt       nodeVecDim = ni->nodeVecDim;
17223f27d899SToby Isaac   PetscInt       *perm;
17233f27d899SToby Isaac   IS             permIS;
17243f27d899SToby Isaac   IS             id;
17253f27d899SToby Isaac   PetscInt       *nIdxPerm;
17263f27d899SToby Isaac   PetscReal      *nVecPerm;
17273f27d899SToby Isaac   PetscErrorCode ierr;
17283f27d899SToby Isaac 
17293f27d899SToby Isaac   PetscFunctionBegin;
17303f27d899SToby Isaac   ierr = PetscLagNodeIndicesGetPermutation(ni, &perm);CHKERRQ(ierr);
17313f27d899SToby Isaac   ierr = MatGetSize(A, &m, &n);CHKERRQ(ierr);
17323f27d899SToby Isaac   ierr = PetscMalloc1(nodeIdxDim * m, &nIdxPerm);CHKERRQ(ierr);
17333f27d899SToby Isaac   ierr = PetscMalloc1(nodeVecDim * m, &nVecPerm);CHKERRQ(ierr);
17343f27d899SToby Isaac   for (i = 0; i < m; i++) for (j = 0; j < nodeIdxDim; j++) nIdxPerm[i * nodeIdxDim + j] = ni->nodeIdx[perm[i] * nodeIdxDim + j];
17353f27d899SToby Isaac   for (i = 0; i < m; i++) for (j = 0; j < nodeVecDim; j++) nVecPerm[i * nodeVecDim + j] = ni->nodeVec[perm[i] * nodeVecDim + j];
17363f27d899SToby Isaac   ierr = ISCreateGeneral(PETSC_COMM_SELF, m, perm, PETSC_USE_POINTER, &permIS);CHKERRQ(ierr);
17373f27d899SToby Isaac   ierr = ISSetPermutation(permIS);CHKERRQ(ierr);
17383f27d899SToby Isaac   ierr = ISCreateStride(PETSC_COMM_SELF, n, 0, 1, &id);CHKERRQ(ierr);
17393f27d899SToby Isaac   ierr = ISSetPermutation(id);CHKERRQ(ierr);
17403f27d899SToby Isaac   ierr = MatPermute(A, permIS, id, Aperm);CHKERRQ(ierr);
17413f27d899SToby Isaac   ierr = ISDestroy(&permIS);CHKERRQ(ierr);
17423f27d899SToby Isaac   ierr = ISDestroy(&id);CHKERRQ(ierr);
17433f27d899SToby Isaac   for (i = 0; i < m; i++) perm[i] = i;
17443f27d899SToby Isaac   ierr = PetscFree(ni->nodeIdx);CHKERRQ(ierr);
17453f27d899SToby Isaac   ierr = PetscFree(ni->nodeVec);CHKERRQ(ierr);
17463f27d899SToby Isaac   ni->nodeIdx = nIdxPerm;
17473f27d899SToby Isaac   ni->nodeVec = nVecPerm;
17486f905325SMatthew G. Knepley   PetscFunctionReturn(0);
17496f905325SMatthew G. Knepley }
17506f905325SMatthew G. Knepley 
17516f905325SMatthew G. Knepley static PetscErrorCode PetscDualSpaceSetUp_Lagrange(PetscDualSpace sp)
17526f905325SMatthew G. Knepley {
17536f905325SMatthew G. Knepley   PetscDualSpace_Lag *lag   = (PetscDualSpace_Lag *) sp->data;
17546f905325SMatthew G. Knepley   DM                  dm    = sp->dm;
17553f27d899SToby Isaac   DM                  dmint = NULL;
17563f27d899SToby Isaac   PetscInt            order;
17576f905325SMatthew G. Knepley   PetscInt            Nc    = sp->Nc;
17586f905325SMatthew G. Knepley   MPI_Comm            comm;
17596f905325SMatthew G. Knepley   PetscBool           continuous;
17603f27d899SToby Isaac   PetscSection        section;
17613f27d899SToby Isaac   PetscInt            depth, dim, pStart, pEnd, cStart, cEnd, p, *pStratStart, *pStratEnd, d;
17623f27d899SToby Isaac   PetscInt            formDegree, Nk, Ncopies;
17633f27d899SToby Isaac   PetscInt            tensorf = -1, tensorf2 = -1;
17643f27d899SToby Isaac   PetscBool           tensorCell, tensorSpace;
17653f27d899SToby Isaac   PetscBool           uniform, trimmed;
17663f27d899SToby Isaac   Petsc1DNodeFamily   nodeFamily;
17673f27d899SToby Isaac   PetscInt            numNodeSkip;
17683f27d899SToby Isaac   DMPlexInterpolatedFlag interpolated;
17693f27d899SToby Isaac   PetscBool           isbdm;
17706f905325SMatthew G. Knepley   PetscErrorCode      ierr;
17716f905325SMatthew G. Knepley 
17726f905325SMatthew G. Knepley   PetscFunctionBegin;
17733f27d899SToby Isaac   /* step 1: sanitize input */
17746f905325SMatthew G. Knepley   ierr = PetscObjectGetComm((PetscObject) sp, &comm);CHKERRQ(ierr);
17756f905325SMatthew G. Knepley   ierr = DMGetDimension(dm, &dim);CHKERRQ(ierr);
17763f27d899SToby Isaac   ierr = PetscObjectTypeCompare((PetscObject)sp, "bdm", &isbdm);CHKERRQ(ierr);
17773f27d899SToby Isaac   if (isbdm) {
17783f27d899SToby Isaac     sp->k = -(dim-1); /* form degree of H-div */
17793f27d899SToby Isaac     ierr = PetscObjectChangeTypeName((PetscObject)sp, PETSCDUALSPACELAGRANGE);CHKERRQ(ierr);
17803f27d899SToby Isaac   }
17813f27d899SToby Isaac   ierr = PetscDualSpaceGetFormDegree(sp, &formDegree);CHKERRQ(ierr);
17823f27d899SToby Isaac   if (PetscAbsInt(formDegree) > dim) SETERRQ(comm, PETSC_ERR_ARG_OUTOFRANGE, "Form degree must be bounded by dimension");
17833f27d899SToby Isaac   ierr = PetscDTBinomialInt(dim,PetscAbsInt(formDegree),&Nk);CHKERRQ(ierr);
17843f27d899SToby Isaac   if (sp->Nc <= 0 && lag->numCopies > 0) sp->Nc = Nk * lag->numCopies;
17853f27d899SToby Isaac   Nc = sp->Nc;
17863f27d899SToby Isaac   if (Nc % Nk) SETERRQ(comm, PETSC_ERR_ARG_INCOMP, "Number of components is not a multiple of form degree size");
17873f27d899SToby Isaac   if (lag->numCopies <= 0) lag->numCopies = Nc / Nk;
17883f27d899SToby Isaac   Ncopies = lag->numCopies;
17893f27d899SToby Isaac   if (Nc / Nk != Ncopies) SETERRQ(comm, PETSC_ERR_ARG_INCOMP, "Number of copies * (dim choose k) != Nc");
17903f27d899SToby Isaac   if (!dim) sp->order = 0;
17913f27d899SToby Isaac   order = sp->order;
17923f27d899SToby Isaac   uniform = sp->uniform;
17933f27d899SToby Isaac   if (!uniform) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_SUP, "Variable order not supported yet");
17943f27d899SToby Isaac   if (lag->trimmed && !formDegree) lag->trimmed = PETSC_FALSE; /* trimmed spaces are the same as full spaces for 0-forms */
17953f27d899SToby Isaac   if (lag->nodeType == PETSCDTNODES_DEFAULT) {
17963f27d899SToby Isaac     lag->nodeType = PETSCDTNODES_GAUSSJACOBI;
17973f27d899SToby Isaac     lag->nodeExponent = 0.;
17983f27d899SToby Isaac     /* trimmed spaces don't include corner vertices, so don't use end nodes by default */
17993f27d899SToby Isaac     lag->endNodes = lag->trimmed ? PETSC_FALSE : PETSC_TRUE;
18003f27d899SToby Isaac   }
18013f27d899SToby Isaac   /* If a trimmed space and the user did choose nodes with endpoints, skip them by default */
18023f27d899SToby Isaac   if (lag->numNodeSkip < 0) lag->numNodeSkip = (lag->trimmed && lag->endNodes) ? 1 : 0;
18033f27d899SToby Isaac   numNodeSkip = lag->numNodeSkip;
18043f27d899SToby Isaac   if (lag->trimmed && !order) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot have zeroth order trimmed elements");
18053f27d899SToby Isaac   if (lag->trimmed && PetscAbsInt(formDegree) == dim) { /* convert trimmed n-forms to untrimmed of one polynomial order less */
18063f27d899SToby Isaac     sp->order--;
18073f27d899SToby Isaac     order--;
18083f27d899SToby Isaac     lag->trimmed = PETSC_FALSE;
18093f27d899SToby Isaac   }
18103f27d899SToby Isaac   trimmed = lag->trimmed;
18113f27d899SToby Isaac   if (!order || PetscAbsInt(formDegree) == dim) lag->continuous = PETSC_FALSE;
18123f27d899SToby Isaac   continuous = lag->continuous;
18136f905325SMatthew G. Knepley   ierr = DMPlexGetDepth(dm, &depth);CHKERRQ(ierr);
18146f905325SMatthew G. Knepley   ierr = DMPlexGetChart(dm, &pStart, &pEnd);CHKERRQ(ierr);
18153f27d899SToby Isaac   ierr = DMPlexGetHeightStratum(dm, 0, &cStart, &cEnd);CHKERRQ(ierr);
18163f27d899SToby Isaac   if (pStart != 0 || cStart != 0) SETERRQ(PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_WRONGSTATE, "Expect DM with chart starting at zero and cells first");
18173f27d899SToby Isaac   if (cEnd != 1) SETERRQ(PetscObjectComm((PetscObject)sp), PETSC_ERR_ARG_WRONGSTATE, "Use PETSCDUALSPACEREFINED for multi-cell reference meshes");
18183f27d899SToby Isaac   ierr = DMPlexIsInterpolated(dm, &interpolated);CHKERRQ(ierr);
18193f27d899SToby Isaac   if (interpolated != DMPLEX_INTERPOLATED_FULL) {
18203f27d899SToby Isaac     ierr = DMPlexInterpolate(dm, &dmint);CHKERRQ(ierr);
18213f27d899SToby Isaac   } else {
18223f27d899SToby Isaac     ierr = PetscObjectReference((PetscObject)dm);CHKERRQ(ierr);
18233f27d899SToby Isaac     dmint = dm;
18243f27d899SToby Isaac   }
18253f27d899SToby Isaac   tensorCell = PETSC_FALSE;
18263f27d899SToby Isaac   if (dim > 1) {
18273f27d899SToby Isaac     ierr = DMPlexPointIsTensor(dmint, 0, &tensorCell, &tensorf, &tensorf2);CHKERRQ(ierr);
18283f27d899SToby Isaac   }
18293f27d899SToby Isaac   lag->tensorCell = tensorCell;
18303f27d899SToby Isaac   if (dim < 2 || !lag->tensorCell) lag->tensorSpace = PETSC_FALSE;
18316f905325SMatthew G. Knepley   tensorSpace = lag->tensorSpace;
18323f27d899SToby Isaac   if (!lag->nodeFamily) {
18333f27d899SToby Isaac     ierr = Petsc1DNodeFamilyCreate(lag->nodeType, lag->nodeExponent, lag->endNodes, &lag->nodeFamily);CHKERRQ(ierr);
18343f27d899SToby Isaac   }
18353f27d899SToby Isaac   nodeFamily = lag->nodeFamily;
18363f27d899SToby Isaac   if (interpolated != DMPLEX_INTERPOLATED_FULL && continuous && (PetscAbsInt(formDegree) > 0 || order > 1)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Reference element won't support all boundary nodes");
183720cf1dd8SToby Isaac 
18383f27d899SToby Isaac   /* step 2: construct the boundary spaces */
18393f27d899SToby Isaac   ierr = PetscMalloc2(depth+1,&pStratStart,depth+1,&pStratEnd);CHKERRQ(ierr);
18403f27d899SToby Isaac   ierr = PetscCalloc1(pEnd,&(sp->pointSpaces));CHKERRQ(ierr);
18413f27d899SToby Isaac   for (d = 0; d <= depth; ++d) {ierr = DMPlexGetDepthStratum(dm, d, &pStratStart[d], &pStratEnd[d]);CHKERRQ(ierr);}
18423f27d899SToby Isaac   ierr = PetscDualSpaceSectionCreate_Internal(sp, &section);CHKERRQ(ierr);
18433f27d899SToby Isaac   sp->pointSection = section;
18443f27d899SToby Isaac   if (continuous && !(lag->interiorOnly)) {
18453f27d899SToby Isaac     PetscInt h;
18466f905325SMatthew G. Knepley 
18473f27d899SToby Isaac     for (p = pStratStart[depth - 1]; p < pStratEnd[depth - 1]; p++) { /* calculate the facet dual spaces */
18483f27d899SToby Isaac       PetscReal v0[3];
18493f27d899SToby Isaac       DMPolytopeType ptype;
18503f27d899SToby Isaac       PetscReal J[9], detJ;
18516f905325SMatthew G. Knepley       PetscInt  q;
18526f905325SMatthew G. Knepley 
18533f27d899SToby Isaac       ierr = DMPlexComputeCellGeometryAffineFEM(dm, p, v0, J, NULL, &detJ);CHKERRQ(ierr);
18543f27d899SToby Isaac       ierr = DMPlexGetCellType(dm, p, &ptype);CHKERRQ(ierr);
18556f905325SMatthew G. Knepley 
18563f27d899SToby Isaac       /* compare orders to previous facets: if computed, reference that dualspace */
18573f27d899SToby Isaac       for (q = pStratStart[depth - 1]; q < p; q++) {
18583f27d899SToby Isaac         DMPolytopeType qtype;
18596f905325SMatthew G. Knepley 
18603f27d899SToby Isaac         ierr = DMPlexGetCellType(dm, q, &qtype);CHKERRQ(ierr);
18613f27d899SToby Isaac         if (qtype == ptype) break;
18626f905325SMatthew G. Knepley       }
18633f27d899SToby Isaac       if (q < p) { /* this facet has the same dual space as that one */
18643f27d899SToby Isaac         ierr = PetscObjectReference((PetscObject)sp->pointSpaces[q]);CHKERRQ(ierr);
18653f27d899SToby Isaac         sp->pointSpaces[p] = sp->pointSpaces[q];
18663f27d899SToby Isaac         continue;
18676f905325SMatthew G. Knepley       }
18683f27d899SToby Isaac       /* if not, recursively compute this dual space */
18693f27d899SToby Isaac       ierr = PetscDualSpaceCreateFacetSubspace_Lagrange(sp,NULL,p,formDegree,Ncopies,PETSC_FALSE,&sp->pointSpaces[p]);CHKERRQ(ierr);
18706f905325SMatthew G. Knepley     }
18713f27d899SToby Isaac     for (h = 2; h <= depth; h++) { /* get the higher subspaces from the facet subspaces */
18723f27d899SToby Isaac       PetscInt hd = depth - h;
18733f27d899SToby Isaac       PetscInt hdim = dim - h;
18746f905325SMatthew G. Knepley 
18753f27d899SToby Isaac       if (hdim < PetscAbsInt(formDegree)) break;
18763f27d899SToby Isaac       for (p = pStratStart[hd]; p < pStratEnd[hd]; p++) {
18773f27d899SToby Isaac         PetscInt suppSize, s;
18783f27d899SToby Isaac         const PetscInt *supp;
18796f905325SMatthew G. Knepley 
18803f27d899SToby Isaac         ierr = DMPlexGetSupportSize(dm, p, &suppSize);CHKERRQ(ierr);
18813f27d899SToby Isaac         ierr = DMPlexGetSupport(dm, p, &supp);CHKERRQ(ierr);
18823f27d899SToby Isaac         for (s = 0; s < suppSize; s++) {
18833f27d899SToby Isaac           DM             qdm;
18843f27d899SToby Isaac           PetscDualSpace qsp, psp;
18853f27d899SToby Isaac           PetscInt c, coneSize, q;
18863f27d899SToby Isaac           const PetscInt *cone;
18873f27d899SToby Isaac           const PetscInt *refCone;
18886f905325SMatthew G. Knepley 
18893f27d899SToby Isaac           q = supp[0];
18903f27d899SToby Isaac           qsp = sp->pointSpaces[q];
18913f27d899SToby Isaac           ierr = DMPlexGetConeSize(dm, q, &coneSize);CHKERRQ(ierr);
18923f27d899SToby Isaac           ierr = DMPlexGetCone(dm, q, &cone);CHKERRQ(ierr);
18933f27d899SToby Isaac           for (c = 0; c < coneSize; c++) if (cone[c] == p) break;
18943f27d899SToby Isaac           if (c == coneSize) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_PLIB, "cone/suppport mismatch");
18953f27d899SToby Isaac           ierr = PetscDualSpaceGetDM(qsp, &qdm);CHKERRQ(ierr);
18963f27d899SToby Isaac           ierr = DMPlexGetCone(qdm, 0, &refCone);CHKERRQ(ierr);
18973f27d899SToby Isaac           /* get the equivalent dual space from the support dual space */
18983f27d899SToby Isaac           ierr = PetscDualSpaceGetPointSubspace(qsp, refCone[c], &psp);CHKERRQ(ierr);
18993f27d899SToby Isaac           if (!s) {
19003f27d899SToby Isaac             ierr = PetscObjectReference((PetscObject)psp);CHKERRQ(ierr);
19013f27d899SToby Isaac             sp->pointSpaces[p] = psp;
19023f27d899SToby Isaac           }
19033f27d899SToby Isaac         }
19043f27d899SToby Isaac       }
19053f27d899SToby Isaac     }
19063f27d899SToby Isaac     for (p = 1; p < pEnd; p++) {
19073f27d899SToby Isaac       PetscInt pspdim;
19083f27d899SToby Isaac       if (!sp->pointSpaces[p]) continue;
19093f27d899SToby Isaac       ierr = PetscDualSpaceGetInteriorDimension(sp->pointSpaces[p], &pspdim);CHKERRQ(ierr);
19103f27d899SToby Isaac       ierr = PetscSectionSetDof(section, p, pspdim);CHKERRQ(ierr);
19113f27d899SToby Isaac     }
19123f27d899SToby Isaac   }
19136f905325SMatthew G. Knepley 
19143f27d899SToby Isaac   if (Ncopies > 1) {
19153f27d899SToby Isaac     Mat intMatScalar, allMatScalar;
19163f27d899SToby Isaac     PetscDualSpace scalarsp;
19173f27d899SToby Isaac     PetscDualSpace_Lag *scalarlag;
19183f27d899SToby Isaac 
19193f27d899SToby Isaac     ierr = PetscDualSpaceDuplicate(sp, &scalarsp);CHKERRQ(ierr);
19203f27d899SToby Isaac     ierr = PetscDualSpaceSetNumComponents(scalarsp, Nk);CHKERRQ(ierr);
19213f27d899SToby Isaac     ierr = PetscDualSpaceSetUp(scalarsp);CHKERRQ(ierr);
19223f27d899SToby Isaac     ierr = PetscDualSpaceGetInteriorData(scalarsp, &(sp->intNodes), &intMatScalar);CHKERRQ(ierr);
19233f27d899SToby Isaac     ierr = PetscObjectReference((PetscObject)(sp->intNodes));CHKERRQ(ierr);
19243f27d899SToby Isaac     if (intMatScalar) {ierr = PetscDualSpaceLagrangeMatrixCreateCopies(intMatScalar, Nk, Ncopies, &(sp->intMat));CHKERRQ(ierr);}
19253f27d899SToby Isaac     ierr = PetscDualSpaceGetAllData(scalarsp, &(sp->allNodes), &allMatScalar);CHKERRQ(ierr);
19263f27d899SToby Isaac     ierr = PetscObjectReference((PetscObject)(sp->allNodes));CHKERRQ(ierr);
19273f27d899SToby Isaac     ierr = PetscDualSpaceLagrangeMatrixCreateCopies(allMatScalar, Nk, Ncopies, &(sp->allMat));CHKERRQ(ierr);
19283f27d899SToby Isaac     sp->spdim = scalarsp->spdim * Ncopies;
19293f27d899SToby Isaac     sp->spintdim = scalarsp->spintdim * Ncopies;
19303f27d899SToby Isaac     scalarlag = (PetscDualSpace_Lag *) scalarsp->data;
19313f27d899SToby Isaac     ierr = PetscLagNodeIndicesReference(scalarlag->vertIndices);CHKERRQ(ierr);
19323f27d899SToby Isaac     lag->vertIndices = scalarlag->vertIndices;
19333f27d899SToby Isaac     ierr = PetscLagNodeIndicesReference(scalarlag->intNodeIndices);CHKERRQ(ierr);
19343f27d899SToby Isaac     lag->intNodeIndices = scalarlag->intNodeIndices;
19353f27d899SToby Isaac     ierr = PetscLagNodeIndicesReference(scalarlag->allNodeIndices);CHKERRQ(ierr);
19363f27d899SToby Isaac     lag->allNodeIndices = scalarlag->allNodeIndices;
19373f27d899SToby Isaac     ierr = PetscDualSpaceDestroy(&scalarsp);CHKERRQ(ierr);
19383f27d899SToby Isaac     ierr = PetscSectionSetDof(section, 0, sp->spintdim);CHKERRQ(ierr);
19393f27d899SToby Isaac     ierr = PetscDualSpaceSectionSetUp_Internal(sp, section);CHKERRQ(ierr);
19403f27d899SToby Isaac     ierr = PetscDualSpaceComputeFunctionalsFromAllData(sp);CHKERRQ(ierr);
19416f905325SMatthew G. Knepley     ierr = PetscFree2(pStratStart, pStratEnd);CHKERRQ(ierr);
19423f27d899SToby Isaac     ierr = DMDestroy(&dmint);CHKERRQ(ierr);
194320cf1dd8SToby Isaac     PetscFunctionReturn(0);
194420cf1dd8SToby Isaac   }
194520cf1dd8SToby Isaac 
19463f27d899SToby Isaac   if (trimmed && !continuous) {
19473f27d899SToby Isaac     /* the dofs of a trimmed space don't have a nice tensor/lattice structure:
19483f27d899SToby Isaac      * just construct the continuous dual space and copy all of the data over,
19493f27d899SToby Isaac      * allocating it all to the cell instead of splitting it up between the boundaries */
19503f27d899SToby Isaac     PetscDualSpace  spcont;
19513f27d899SToby Isaac     PetscInt        spdim, f;
19523f27d899SToby Isaac     PetscQuadrature allNodes;
19533f27d899SToby Isaac     PetscDualSpace_Lag *lagc;
19543f27d899SToby Isaac     Mat             allMat;
19553f27d899SToby Isaac 
19563f27d899SToby Isaac     ierr = PetscDualSpaceDuplicate(sp, &spcont);CHKERRQ(ierr);
19573f27d899SToby Isaac     ierr = PetscDualSpaceLagrangeSetContinuity(spcont, PETSC_TRUE);CHKERRQ(ierr);
19583f27d899SToby Isaac     ierr = PetscDualSpaceSetUp(spcont);CHKERRQ(ierr);
19593f27d899SToby Isaac     ierr = PetscDualSpaceGetDimension(spcont, &spdim);CHKERRQ(ierr);
19603f27d899SToby Isaac     sp->spdim = sp->spintdim = spdim;
19613f27d899SToby Isaac     ierr = PetscSectionSetDof(section, 0, spdim);CHKERRQ(ierr);
19623f27d899SToby Isaac     ierr = PetscDualSpaceSectionSetUp_Internal(sp, section);CHKERRQ(ierr);
19633f27d899SToby Isaac     ierr = PetscMalloc1(spdim, &(sp->functional));CHKERRQ(ierr);
19643f27d899SToby Isaac     for (f = 0; f < spdim; f++) {
19653f27d899SToby Isaac       PetscQuadrature fn;
19663f27d899SToby Isaac 
19673f27d899SToby Isaac       ierr = PetscDualSpaceGetFunctional(spcont, f, &fn);CHKERRQ(ierr);
19683f27d899SToby Isaac       ierr = PetscObjectReference((PetscObject)fn);CHKERRQ(ierr);
19693f27d899SToby Isaac       sp->functional[f] = fn;
19703f27d899SToby Isaac     }
19713f27d899SToby Isaac     ierr = PetscDualSpaceGetAllData(spcont, &allNodes, &allMat);CHKERRQ(ierr);
19723f27d899SToby Isaac     ierr = PetscObjectReference((PetscObject) allNodes);CHKERRQ(ierr);
19733f27d899SToby Isaac     ierr = PetscObjectReference((PetscObject) allNodes);CHKERRQ(ierr);
19743f27d899SToby Isaac     sp->allNodes = sp->intNodes = allNodes;
19753f27d899SToby Isaac     ierr = PetscObjectReference((PetscObject) allMat);CHKERRQ(ierr);
19763f27d899SToby Isaac     ierr = PetscObjectReference((PetscObject) allMat);CHKERRQ(ierr);
19773f27d899SToby Isaac     sp->allMat = sp->intMat = allMat;
19783f27d899SToby Isaac     /* TODO: copy over symmetries */
19793f27d899SToby Isaac     lagc = (PetscDualSpace_Lag *) spcont->data;
19803f27d899SToby Isaac     ierr = PetscLagNodeIndicesReference(lagc->vertIndices);CHKERRQ(ierr);
19813f27d899SToby Isaac     lag->vertIndices = lagc->vertIndices;
19823f27d899SToby Isaac     ierr = PetscLagNodeIndicesReference(lagc->allNodeIndices);CHKERRQ(ierr);
19833f27d899SToby Isaac     ierr = PetscLagNodeIndicesReference(lagc->allNodeIndices);CHKERRQ(ierr);
19843f27d899SToby Isaac     lag->intNodeIndices = lagc->allNodeIndices;
19853f27d899SToby Isaac     lag->allNodeIndices = lagc->allNodeIndices;
19863f27d899SToby Isaac     ierr = PetscDualSpaceDestroy(&spcont);CHKERRQ(ierr);
19873f27d899SToby Isaac     ierr = PetscFree2(pStratStart, pStratEnd);CHKERRQ(ierr);
19883f27d899SToby Isaac     ierr = DMDestroy(&dmint);CHKERRQ(ierr);
19893f27d899SToby Isaac     PetscFunctionReturn(0);
19903f27d899SToby Isaac   }
19913f27d899SToby Isaac 
19923f27d899SToby Isaac   /* step 3: construct intNodes, and intMat, and combine it with boundray data to make allNodes and allMat */
19933f27d899SToby Isaac   if (!tensorSpace) {
1994*6ff15688SToby Isaac     if (!tensorCell) {ierr = PetscLagNodeIndicesCreateSimplexVertices(dm, &(lag->vertIndices));CHKERRQ(ierr);}
19953f27d899SToby Isaac 
19963f27d899SToby Isaac     if (trimmed) {
19973f27d899SToby Isaac       if (order + PetscAbsInt(formDegree) > dim) {
19983f27d899SToby Isaac         PetscInt sum = order + PetscAbsInt(formDegree) - dim - 1;
19993f27d899SToby Isaac         PetscInt nDofs;
20003f27d899SToby Isaac 
20013f27d899SToby Isaac         ierr = PetscDualSpaceLagrangeCreateSimplexNodeMat(nodeFamily, dim, sum, Nk, numNodeSkip, &sp->intNodes, &sp->intMat, &(lag->intNodeIndices));CHKERRQ(ierr);
20023f27d899SToby Isaac         ierr = MatGetSize(sp->intMat, &nDofs, NULL);CHKERRQ(ierr);
20033f27d899SToby Isaac         ierr = PetscSectionSetDof(section, 0, nDofs);CHKERRQ(ierr);
20043f27d899SToby Isaac       }
20053f27d899SToby Isaac       ierr = PetscDualSpaceSectionSetUp_Internal(sp, section);CHKERRQ(ierr);
20063f27d899SToby Isaac       ierr = PetscDualSpaceCreateAllDataFromInteriorData(sp);CHKERRQ(ierr);
20073f27d899SToby Isaac       ierr = PetscDualSpaceLagrangeCreateAllNodeIdx(sp);CHKERRQ(ierr);
20083f27d899SToby Isaac     } else {
20093f27d899SToby Isaac       if (!continuous) {
20103f27d899SToby Isaac         PetscInt sum = order;
20113f27d899SToby Isaac         PetscInt nDofs;
20123f27d899SToby Isaac 
20133f27d899SToby Isaac         ierr = PetscDualSpaceLagrangeCreateSimplexNodeMat(nodeFamily, dim, sum, Nk, numNodeSkip, &sp->intNodes, &sp->intMat, &(lag->intNodeIndices));CHKERRQ(ierr);
20143f27d899SToby Isaac         ierr = MatGetSize(sp->intMat, &nDofs, NULL);CHKERRQ(ierr);
20153f27d899SToby Isaac         ierr = PetscSectionSetDof(section, 0, nDofs);CHKERRQ(ierr);
20163f27d899SToby Isaac         ierr = PetscDualSpaceSectionSetUp_Internal(sp, section);CHKERRQ(ierr);
20173f27d899SToby Isaac         ierr = PetscObjectReference((PetscObject)(sp->intNodes));CHKERRQ(ierr);
20183f27d899SToby Isaac         sp->allNodes = sp->intNodes;
20193f27d899SToby Isaac         ierr = PetscObjectReference((PetscObject)(sp->intMat));CHKERRQ(ierr);
20203f27d899SToby Isaac         sp->allMat = sp->intMat;
20213f27d899SToby Isaac         ierr = PetscLagNodeIndicesReference(lag->intNodeIndices);CHKERRQ(ierr);
20223f27d899SToby Isaac         lag->allNodeIndices = lag->intNodeIndices;
20233f27d899SToby Isaac       } else {
20243f27d899SToby Isaac         if (order + PetscAbsInt(formDegree) > dim) {
20253f27d899SToby Isaac           PetscDualSpace trimmedsp;
20263f27d899SToby Isaac           PetscDualSpace_Lag *trimmedlag;
20273f27d899SToby Isaac           PetscQuadrature intNodes;
20283f27d899SToby Isaac           PetscInt trFormDegree = formDegree >= 0 ? formDegree - dim : dim - PetscAbsInt(formDegree);
20293f27d899SToby Isaac           PetscInt nDofs;
20303f27d899SToby Isaac           Mat intMat;
20313f27d899SToby Isaac 
20323f27d899SToby Isaac           ierr = PetscDualSpaceDuplicate(sp, &trimmedsp);CHKERRQ(ierr);
20333f27d899SToby Isaac           ierr = PetscDualSpaceLagrangeSetTrimmed(trimmedsp, PETSC_TRUE);CHKERRQ(ierr);
20343f27d899SToby Isaac           ierr = PetscDualSpaceSetOrder(trimmedsp, order + PetscAbsInt(formDegree) - dim);CHKERRQ(ierr);
20353f27d899SToby Isaac           ierr = PetscDualSpaceSetFormDegree(trimmedsp, trFormDegree);CHKERRQ(ierr);
20363f27d899SToby Isaac           trimmedlag = (PetscDualSpace_Lag *) trimmedsp->data;
20373f27d899SToby Isaac           trimmedlag->numNodeSkip = numNodeSkip + 1;
20383f27d899SToby Isaac           ierr = PetscDualSpaceSetUp(trimmedsp);CHKERRQ(ierr);
20393f27d899SToby Isaac           ierr = PetscDualSpaceGetAllData(trimmedsp, &intNodes, &intMat);CHKERRQ(ierr);
20403f27d899SToby Isaac           ierr = PetscObjectReference((PetscObject)intNodes);CHKERRQ(ierr);
20413f27d899SToby Isaac           sp->intNodes = intNodes;
20423f27d899SToby Isaac           ierr = PetscObjectReference((PetscObject)intMat);CHKERRQ(ierr);
20433f27d899SToby Isaac           sp->intMat = intMat;
20443f27d899SToby Isaac           ierr = MatGetSize(sp->intMat, &nDofs, NULL);CHKERRQ(ierr);
20453f27d899SToby Isaac           ierr = PetscLagNodeIndicesReference(trimmedlag->allNodeIndices);CHKERRQ(ierr);
20463f27d899SToby Isaac           lag->intNodeIndices = trimmedlag->allNodeIndices;
20473f27d899SToby Isaac           ierr = PetscDualSpaceDestroy(&trimmedsp);CHKERRQ(ierr);
20483f27d899SToby Isaac           ierr = PetscSectionSetDof(section, 0, nDofs);CHKERRQ(ierr);
20493f27d899SToby Isaac         }
20503f27d899SToby Isaac         ierr = PetscDualSpaceSectionSetUp_Internal(sp, section);CHKERRQ(ierr);
20513f27d899SToby Isaac         ierr = PetscDualSpaceCreateAllDataFromInteriorData(sp);CHKERRQ(ierr);
20523f27d899SToby Isaac         ierr = PetscDualSpaceLagrangeCreateAllNodeIdx(sp);CHKERRQ(ierr);
20533f27d899SToby Isaac       }
20543f27d899SToby Isaac     }
20553f27d899SToby Isaac   } else {
20563f27d899SToby Isaac     /* assume the tensor element has the first facet being the cross-section, having its normal
20573f27d899SToby Isaac      * pointing in the last coordinate direction */
20583f27d899SToby Isaac     PetscQuadrature intNodesTrace = NULL;
20593f27d899SToby Isaac     PetscQuadrature intNodesFiber = NULL;
20603f27d899SToby Isaac     PetscQuadrature intNodes = NULL;
20613f27d899SToby Isaac     PetscLagNodeIndices intNodeIndices = NULL;
20623f27d899SToby Isaac     Mat             intMat = NULL;
20633f27d899SToby Isaac 
20643f27d899SToby Isaac     if (PetscAbsInt(formDegree) < dim) { /* get the trace k-forms on the first facet, and the 0-forms on the edge */
20653f27d899SToby Isaac       PetscDualSpace  trace, fiber;
20663f27d899SToby Isaac       PetscDualSpace_Lag *tracel, *fiberl;
20673f27d899SToby Isaac       Mat             intMatTrace, intMatFiber;
20683f27d899SToby Isaac 
20693f27d899SToby Isaac       if (sp->pointSpaces[tensorf]) {
20703f27d899SToby Isaac         ierr = PetscObjectReference((PetscObject)(sp->pointSpaces[tensorf]));CHKERRQ(ierr);
20713f27d899SToby Isaac         trace = sp->pointSpaces[tensorf];
20723f27d899SToby Isaac       } else {
20733f27d899SToby Isaac         ierr = PetscDualSpaceCreateFacetSubspace_Lagrange(sp,NULL,tensorf,formDegree,Ncopies,PETSC_TRUE,&trace);CHKERRQ(ierr);
20743f27d899SToby Isaac       }
20753f27d899SToby Isaac       ierr = PetscDualSpaceCreateEdgeSubspace_Lagrange(sp,order,0,1,PETSC_TRUE,&fiber);CHKERRQ(ierr);
20763f27d899SToby Isaac       tracel = (PetscDualSpace_Lag *) trace->data;
20773f27d899SToby Isaac       fiberl = (PetscDualSpace_Lag *) fiber->data;
20783f27d899SToby Isaac       ierr = PetscLagNodeIndicesCreateTensorVertices(dm, tracel->vertIndices, &(lag->vertIndices));CHKERRQ(ierr);
20793f27d899SToby Isaac       ierr = PetscDualSpaceGetInteriorData(trace, &intNodesTrace, &intMatTrace);CHKERRQ(ierr);
20803f27d899SToby Isaac       ierr = PetscDualSpaceGetInteriorData(fiber, &intNodesFiber, &intMatFiber);CHKERRQ(ierr);
20813f27d899SToby Isaac       if (intNodesTrace && intNodesFiber) {
20823f27d899SToby Isaac         ierr = PetscQuadratureCreateTensor(intNodesTrace, intNodesFiber, &intNodes);CHKERRQ(ierr);
20833f27d899SToby Isaac         ierr = MatTensorAltV(intMatTrace, intMatFiber, dim-1, formDegree, 1, 0, &intMat);CHKERRQ(ierr);
20843f27d899SToby Isaac         ierr = PetscLagNodeIndicesTensor(tracel->intNodeIndices, dim - 1, formDegree, fiberl->intNodeIndices, 1, 0, &intNodeIndices);CHKERRQ(ierr);
20853f27d899SToby Isaac       }
20863f27d899SToby Isaac       ierr = PetscObjectReference((PetscObject) intNodesTrace);CHKERRQ(ierr);
20873f27d899SToby Isaac       ierr = PetscObjectReference((PetscObject) intNodesFiber);CHKERRQ(ierr);
20883f27d899SToby Isaac       ierr = PetscDualSpaceDestroy(&fiber);CHKERRQ(ierr);
20893f27d899SToby Isaac       ierr = PetscDualSpaceDestroy(&trace);CHKERRQ(ierr);
20903f27d899SToby Isaac     }
20913f27d899SToby Isaac     if (PetscAbsInt(formDegree) > 0) { /* get the trace (k-1)-forms on the first facet, and the 1-forms on the edge */
20923f27d899SToby Isaac       PetscDualSpace  trace, fiber;
20933f27d899SToby Isaac       PetscDualSpace_Lag *tracel, *fiberl;
20943f27d899SToby Isaac       PetscQuadrature intNodesTrace2, intNodesFiber2, intNodes2;
20953f27d899SToby Isaac       PetscLagNodeIndices intNodeIndices2;
20963f27d899SToby Isaac       Mat             intMatTrace, intMatFiber, intMat2;
20973f27d899SToby Isaac       PetscInt        traceDegree = formDegree > 0 ? formDegree - 1 : formDegree + 1;
20983f27d899SToby Isaac       PetscInt        fiberDegree = formDegree > 0 ? 1 : -1;
20993f27d899SToby Isaac 
21003f27d899SToby Isaac       ierr = PetscDualSpaceCreateFacetSubspace_Lagrange(sp,NULL,tensorf,traceDegree,Ncopies,PETSC_TRUE,&trace);CHKERRQ(ierr);
21013f27d899SToby Isaac       ierr = PetscDualSpaceCreateEdgeSubspace_Lagrange(sp,order,fiberDegree,1,PETSC_TRUE,&fiber);CHKERRQ(ierr);
21023f27d899SToby Isaac       tracel = (PetscDualSpace_Lag *) trace->data;
21033f27d899SToby Isaac       fiberl = (PetscDualSpace_Lag *) fiber->data;
21043f27d899SToby Isaac       if (!lag->vertIndices) {
21053f27d899SToby Isaac         ierr = PetscLagNodeIndicesCreateTensorVertices(dm, tracel->vertIndices, &(lag->vertIndices));CHKERRQ(ierr);
21063f27d899SToby Isaac       }
21073f27d899SToby Isaac       ierr = PetscDualSpaceGetInteriorData(trace, &intNodesTrace2, &intMatTrace);CHKERRQ(ierr);
21083f27d899SToby Isaac       ierr = PetscDualSpaceGetInteriorData(fiber, &intNodesFiber2, &intMatFiber);CHKERRQ(ierr);
21093f27d899SToby Isaac       if (intNodesTrace2 && intNodesFiber2) {
21103f27d899SToby Isaac         ierr = PetscQuadratureCreateTensor(intNodesTrace2, intNodesFiber2, &intNodes2);CHKERRQ(ierr);
21113f27d899SToby Isaac         ierr = MatTensorAltV(intMatTrace, intMatFiber, dim-1, traceDegree, 1, fiberDegree, &intMat2);CHKERRQ(ierr);
21123f27d899SToby Isaac         ierr = PetscLagNodeIndicesTensor(tracel->intNodeIndices, dim - 1, traceDegree, fiberl->intNodeIndices, 1, fiberDegree, &intNodeIndices2);CHKERRQ(ierr);
21133f27d899SToby Isaac         if (!intMat) {
21143f27d899SToby Isaac           intMat = intMat2;
21153f27d899SToby Isaac           intNodes = intNodes2;
21163f27d899SToby Isaac           intNodeIndices = intNodeIndices2;
21173f27d899SToby Isaac         } else {
21183f27d899SToby Isaac           /* merge the two matrices and the two sets of points */
21193f27d899SToby Isaac           PetscInt         nM;
21203f27d899SToby Isaac           PetscInt         nDof, nDof2;
2121*6ff15688SToby Isaac           PetscInt        *toMerged = NULL, *toMerged2 = NULL;
2122*6ff15688SToby Isaac           PetscQuadrature  merged = NULL;
21233f27d899SToby Isaac           PetscLagNodeIndices intNodeIndicesMerged = NULL;
21243f27d899SToby Isaac           Mat              matMerged = NULL;
21253f27d899SToby Isaac 
21263f27d899SToby Isaac           ierr = MatGetSize(intMat, &nDof, 0);CHKERRQ(ierr);
21273f27d899SToby Isaac           ierr = MatGetSize(intMat2, &nDof2, 0);CHKERRQ(ierr);
21283f27d899SToby Isaac           ierr = PetscQuadraturePointsMerge(intNodes, intNodes2, &merged, &toMerged, &toMerged2);CHKERRQ(ierr);
21293f27d899SToby Isaac           ierr = PetscQuadratureGetData(merged, NULL, NULL, &nM, NULL, NULL);CHKERRQ(ierr);
21303f27d899SToby Isaac           ierr = MatricesMerge(intMat, intMat2, dim, formDegree, nM, toMerged, toMerged2, &matMerged);CHKERRQ(ierr);
21313f27d899SToby Isaac           ierr = PetscLagNodeIndicesMerge(intNodeIndices, intNodeIndices2, &intNodeIndicesMerged);CHKERRQ(ierr);
2132*6ff15688SToby Isaac           ierr = PetscFree(toMerged);CHKERRQ(ierr);
2133*6ff15688SToby Isaac           ierr = PetscFree(toMerged2);CHKERRQ(ierr);
21343f27d899SToby Isaac           ierr = MatDestroy(&intMat);CHKERRQ(ierr);
21353f27d899SToby Isaac           ierr = MatDestroy(&intMat2);CHKERRQ(ierr);
21363f27d899SToby Isaac           ierr = PetscQuadratureDestroy(&intNodes);CHKERRQ(ierr);
21373f27d899SToby Isaac           ierr = PetscQuadratureDestroy(&intNodes2);CHKERRQ(ierr);
21383f27d899SToby Isaac           ierr = PetscLagNodeIndicesDestroy(&intNodeIndices);CHKERRQ(ierr);
21393f27d899SToby Isaac           ierr = PetscLagNodeIndicesDestroy(&intNodeIndices2);CHKERRQ(ierr);
21403f27d899SToby Isaac           intNodes = merged;
21413f27d899SToby Isaac           intMat = matMerged;
21423f27d899SToby Isaac           intNodeIndices = intNodeIndicesMerged;
21433f27d899SToby Isaac           if (!trimmed) {
21443f27d899SToby Isaac             Mat intMatPerm;
21453f27d899SToby Isaac 
21463f27d899SToby Isaac             ierr = MatPermuteByNodeIdx(intMat, intNodeIndices, &intMatPerm);CHKERRQ(ierr);
21473f27d899SToby Isaac             ierr = MatDestroy(&intMat);CHKERRQ(ierr);
21483f27d899SToby Isaac             intMat = intMatPerm;
21493f27d899SToby Isaac           }
21503f27d899SToby Isaac         }
21513f27d899SToby Isaac       }
21523f27d899SToby Isaac       ierr = PetscDualSpaceDestroy(&fiber);CHKERRQ(ierr);
21533f27d899SToby Isaac       ierr = PetscDualSpaceDestroy(&trace);CHKERRQ(ierr);
21543f27d899SToby Isaac     }
21553f27d899SToby Isaac     ierr = PetscQuadratureDestroy(&intNodesTrace);CHKERRQ(ierr);
21563f27d899SToby Isaac     ierr = PetscQuadratureDestroy(&intNodesFiber);CHKERRQ(ierr);
21573f27d899SToby Isaac     sp->intNodes = intNodes;
21583f27d899SToby Isaac     sp->intMat = intMat;
21593f27d899SToby Isaac     lag->intNodeIndices = intNodeIndices;
21606f905325SMatthew G. Knepley     {
21613f27d899SToby Isaac       PetscInt nDofs = 0;
21623f27d899SToby Isaac 
21633f27d899SToby Isaac       if (intMat) {
21643f27d899SToby Isaac         ierr = MatGetSize(intMat, &nDofs, NULL);CHKERRQ(ierr);
21653f27d899SToby Isaac       }
21663f27d899SToby Isaac       ierr = PetscSectionSetDof(section, 0, nDofs);CHKERRQ(ierr);
21673f27d899SToby Isaac     }
21683f27d899SToby Isaac     ierr = PetscDualSpaceSectionSetUp_Internal(sp, section);CHKERRQ(ierr);
21693f27d899SToby Isaac     if (continuous) {
21703f27d899SToby Isaac       ierr = PetscDualSpaceCreateAllDataFromInteriorData(sp);CHKERRQ(ierr);
21713f27d899SToby Isaac       ierr = PetscDualSpaceLagrangeCreateAllNodeIdx(sp);CHKERRQ(ierr);
21723f27d899SToby Isaac     } else {
21733f27d899SToby Isaac       ierr = PetscObjectReference((PetscObject) intNodes);CHKERRQ(ierr);
21743f27d899SToby Isaac       sp->allNodes = intNodes;
21753f27d899SToby Isaac       ierr = PetscObjectReference((PetscObject) intMat);CHKERRQ(ierr);
21763f27d899SToby Isaac       sp->allMat = intMat;
21773f27d899SToby Isaac       ierr = PetscLagNodeIndicesReference(intNodeIndices);CHKERRQ(ierr);
21783f27d899SToby Isaac       lag->allNodeIndices = intNodeIndices;
21793f27d899SToby Isaac     }
21803f27d899SToby Isaac   }
21813f27d899SToby Isaac   ierr = PetscSectionGetStorageSize(section, &sp->spdim);CHKERRQ(ierr);
21823f27d899SToby Isaac   ierr = PetscSectionGetConstrainedStorageSize(section, &sp->spintdim);CHKERRQ(ierr);
21833f27d899SToby Isaac   ierr = PetscDualSpaceComputeFunctionalsFromAllData(sp);CHKERRQ(ierr);
21843f27d899SToby Isaac   ierr = PetscFree2(pStratStart, pStratEnd);CHKERRQ(ierr);
21853f27d899SToby Isaac   ierr = DMDestroy(&dmint);CHKERRQ(ierr);
21863f27d899SToby Isaac   PetscFunctionReturn(0);
21873f27d899SToby Isaac }
21883f27d899SToby Isaac 
21893f27d899SToby Isaac PetscErrorCode PetscDualSpaceCreateInteriorSymmetryMatrix_Lagrange(PetscDualSpace sp, PetscInt ornt, Mat *symMat)
21903f27d899SToby Isaac {
21913f27d899SToby Isaac   PetscDualSpace_Lag *lag;
21923f27d899SToby Isaac   DM dm;
21933f27d899SToby Isaac   PetscLagNodeIndices vertIndices, intNodeIndices;
21943f27d899SToby Isaac   PetscLagNodeIndices ni;
21953f27d899SToby Isaac   PetscInt nodeIdxDim, nodeVecDim, nNodes;
21963f27d899SToby Isaac   PetscInt formDegree;
21973f27d899SToby Isaac   PetscInt *perm, *permOrnt;
21983f27d899SToby Isaac   PetscInt *nnz;
21993f27d899SToby Isaac   PetscInt n;
22003f27d899SToby Isaac   PetscInt maxGroupSize;
22013f27d899SToby Isaac   PetscScalar *V, *W, *work;
22023f27d899SToby Isaac   Mat A;
22036f905325SMatthew G. Knepley   PetscErrorCode ierr;
22046f905325SMatthew G. Knepley 
22056f905325SMatthew G. Knepley   PetscFunctionBegin;
22063f27d899SToby Isaac   if (!sp->spintdim) {
22073f27d899SToby Isaac     *symMat = NULL;
22083f27d899SToby Isaac     PetscFunctionReturn(0);
22096f905325SMatthew G. Knepley   }
22103f27d899SToby Isaac   lag = (PetscDualSpace_Lag *) sp->data;
22113f27d899SToby Isaac   vertIndices = lag->vertIndices;
22123f27d899SToby Isaac   intNodeIndices = lag->intNodeIndices;
22133f27d899SToby Isaac   ierr = PetscDualSpaceGetDM(sp, &dm);CHKERRQ(ierr);
22143f27d899SToby Isaac   ierr = PetscDualSpaceGetFormDegree(sp, &formDegree);CHKERRQ(ierr);
22153f27d899SToby Isaac   ierr = PetscNew(&ni);CHKERRQ(ierr);
22163f27d899SToby Isaac   ni->refct = 1;
22173f27d899SToby Isaac   ni->nodeIdxDim = nodeIdxDim = intNodeIndices->nodeIdxDim;
22183f27d899SToby Isaac   ni->nodeVecDim = nodeVecDim = intNodeIndices->nodeVecDim;
22193f27d899SToby Isaac   ni->nNodes = nNodes = intNodeIndices->nNodes;
22203f27d899SToby Isaac   ierr = PetscMalloc1(nNodes * nodeIdxDim, &(ni->nodeIdx));CHKERRQ(ierr);
22213f27d899SToby Isaac   ierr = PetscMalloc1(nNodes * nodeVecDim, &(ni->nodeVec));CHKERRQ(ierr);
22223f27d899SToby Isaac   ierr = PetscLagNodeIndicesPushForward(dm, vertIndices, 0, vertIndices, intNodeIndices, ornt, formDegree, ni->nodeIdx, ni->nodeVec);CHKERRQ(ierr);
22233f27d899SToby Isaac   ierr = PetscLagNodeIndicesGetPermutation(intNodeIndices, &perm);CHKERRQ(ierr);
22243f27d899SToby Isaac   ierr = PetscLagNodeIndicesGetPermutation(ni, &permOrnt);CHKERRQ(ierr);
22253f27d899SToby Isaac   ierr = PetscMalloc1(nNodes, &nnz);CHKERRQ(ierr);
22263f27d899SToby Isaac   for (n = 0, maxGroupSize = 0; n < nNodes;) { /* incremented in the loop */
22273f27d899SToby Isaac     PetscInt *nind = &(ni->nodeIdx[permOrnt[n] * nodeIdxDim]);
22283f27d899SToby Isaac     PetscInt m, nEnd;
22293f27d899SToby Isaac     PetscInt groupSize;
22303f27d899SToby Isaac     for (nEnd = n + 1; nEnd < nNodes; nEnd++) {
22313f27d899SToby Isaac       PetscInt *mind = &(ni->nodeIdx[permOrnt[nEnd] * nodeIdxDim]);
22323f27d899SToby Isaac       PetscInt d;
22333f27d899SToby Isaac 
22343f27d899SToby Isaac       /* compare the oriented permutation indices */
22353f27d899SToby Isaac       for (d = 0; d < nodeIdxDim; d++) if (mind[d] != nind[d]) break;
22363f27d899SToby Isaac       if (d < nodeIdxDim) break;
22373f27d899SToby Isaac     }
22383f27d899SToby Isaac #if defined(PETSC_USE_DEBUG)
22393f27d899SToby Isaac     {
22403f27d899SToby Isaac       PetscInt m;
22413f27d899SToby Isaac       PetscInt *nind = &(intNodeIndices->nodeIdx[perm[n] * nodeIdxDim]);
22423f27d899SToby Isaac 
22433f27d899SToby Isaac       for (m = n + 1; m < nEnd; m++) {
22443f27d899SToby Isaac         PetscInt *mind = &(intNodeIndices->nodeIdx[perm[m] * nodeIdxDim]);
22453f27d899SToby Isaac         PetscInt d;
22463f27d899SToby Isaac 
22473f27d899SToby Isaac         /* compare the oriented permutation indices */
22483f27d899SToby Isaac         for (d = 0; d < nodeIdxDim; d++) if (mind[d] != nind[d]) break;
22493f27d899SToby Isaac         if (d < nodeIdxDim) break;
22503f27d899SToby Isaac       }
22513f27d899SToby Isaac       if (m < nEnd) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Dofs with same index after symmetry not same block size");
22523f27d899SToby Isaac     }
22533f27d899SToby Isaac #endif
22543f27d899SToby Isaac     groupSize = nEnd - n;
22553f27d899SToby Isaac     for (m = n; m < nEnd; m++) nnz[permOrnt[m]] = groupSize;
22563f27d899SToby Isaac 
22573f27d899SToby Isaac     maxGroupSize = PetscMax(maxGroupSize, nEnd - n);
22583f27d899SToby Isaac     /* permOrnt[[n, nEnd)] is a group of dofs that, under the symmetry are at the same location */
22593f27d899SToby Isaac     n = nEnd;
22603f27d899SToby Isaac   }
22613f27d899SToby Isaac   if (maxGroupSize > nodeVecDim) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Dofs are not in blocks that can be solved");
22623f27d899SToby Isaac   ierr = MatCreateSeqAIJ(PETSC_COMM_SELF, nNodes, nNodes, 0, nnz, &A);CHKERRQ(ierr);
22633f27d899SToby Isaac   ierr = PetscFree(nnz);CHKERRQ(ierr);
22643f27d899SToby Isaac   ierr = PetscMalloc3(maxGroupSize * nodeVecDim, &V, maxGroupSize * nodeVecDim, &W, nodeVecDim * 2, &work);CHKERRQ(ierr);
22653f27d899SToby Isaac   for (n = 0; n < nNodes;) { /* incremented in the loop */
22663f27d899SToby Isaac     PetscInt *nind = &(ni->nodeIdx[permOrnt[n] * nodeIdxDim]);
22673f27d899SToby Isaac     PetscInt nEnd;
22683f27d899SToby Isaac     PetscInt m;
22693f27d899SToby Isaac     PetscInt groupSize;
22703f27d899SToby Isaac     for (nEnd = n + 1; nEnd < nNodes; nEnd++) {
22713f27d899SToby Isaac       PetscInt *mind = &(ni->nodeIdx[permOrnt[nEnd] * nodeIdxDim]);
22723f27d899SToby Isaac       PetscInt d;
22733f27d899SToby Isaac 
22743f27d899SToby Isaac       /* compare the oriented permutation indices */
22753f27d899SToby Isaac       for (d = 0; d < nodeIdxDim; d++) if (mind[d] != nind[d]) break;
22763f27d899SToby Isaac       if (d < nodeIdxDim) break;
22773f27d899SToby Isaac     }
22783f27d899SToby Isaac     groupSize = nEnd - n;
22793f27d899SToby Isaac     /* get all of the vectors from the original */
22803f27d899SToby Isaac     for (m = n; m < nEnd; m++) {
22813f27d899SToby Isaac       PetscInt d;
22823f27d899SToby Isaac 
22833f27d899SToby Isaac       for (d = 0; d < nodeVecDim; d++) {
22843f27d899SToby Isaac         V[(m - n) * nodeVecDim + d] = intNodeIndices->nodeVec[perm[m] * nodeVecDim + d];
22853f27d899SToby Isaac         W[(m - n) * nodeVecDim + d] = ni->nodeVec[permOrnt[m] * nodeVecDim + d];
22863f27d899SToby Isaac       }
22873f27d899SToby Isaac     }
22883f27d899SToby Isaac     /* now we have to solve for W in terms of V */
22893f27d899SToby Isaac     {
22903f27d899SToby Isaac       char transpose = 'N';
22913f27d899SToby Isaac       PetscBLASInt bm = nodeVecDim;
22923f27d899SToby Isaac       PetscBLASInt bn = groupSize;
22933f27d899SToby Isaac       PetscBLASInt bnrhs = groupSize;
22943f27d899SToby Isaac       PetscBLASInt blda = bm;
22953f27d899SToby Isaac       PetscBLASInt bldb = bm;
22963f27d899SToby Isaac       PetscBLASInt blwork = 2 * nodeVecDim;
22973f27d899SToby Isaac       PetscBLASInt info;
22983f27d899SToby Isaac 
22993f27d899SToby Isaac       PetscStackCallBLAS("LAPACKgels",LAPACKgels_(&transpose,&bm,&bn,&bnrhs,V,&blda,W,&bldb,work,&blwork, &info));
23003f27d899SToby Isaac       if (info != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad argument to GELS");
23013f27d899SToby Isaac       /* repack */
23023f27d899SToby Isaac       {
23033f27d899SToby Isaac         PetscInt i, j;
23043f27d899SToby Isaac 
23053f27d899SToby Isaac         for (i = 0; i < groupSize; i++) {
23063f27d899SToby Isaac           for (j = 0; j < groupSize; j++) {
23073f27d899SToby Isaac             V[i * groupSize + j] = W[i * nodeVecDim + j];
23083f27d899SToby Isaac           }
23093f27d899SToby Isaac         }
23103f27d899SToby Isaac       }
23113f27d899SToby Isaac     }
23123f27d899SToby Isaac     ierr = MatSetValues(A, groupSize, &permOrnt[n], groupSize, &perm[n], V, INSERT_VALUES);CHKERRQ(ierr);
23133f27d899SToby Isaac     /* permOrnt[[n, nEnd)] is a group of dofs that, under the symmetry are at the same location */
23143f27d899SToby Isaac     n = nEnd;
23153f27d899SToby Isaac   }
23163f27d899SToby Isaac   ierr = MatAssemblyBegin(A, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23173f27d899SToby Isaac   ierr = MatAssemblyEnd(A, MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23183f27d899SToby Isaac   *symMat = A;
23193f27d899SToby Isaac   ierr = PetscFree3(V,W,work);CHKERRQ(ierr);
23203f27d899SToby Isaac   ierr = PetscLagNodeIndicesDestroy(&ni);CHKERRQ(ierr);
23216f905325SMatthew G. Knepley   PetscFunctionReturn(0);
23226f905325SMatthew G. Knepley }
232320cf1dd8SToby Isaac 
232420cf1dd8SToby Isaac #define BaryIndex(perEdge,a,b,c) (((b)*(2*perEdge+1-(b)))/2)+(c)
232520cf1dd8SToby Isaac 
232620cf1dd8SToby Isaac #define CartIndex(perEdge,a,b) (perEdge*(a)+b)
232720cf1dd8SToby Isaac 
232820cf1dd8SToby Isaac static PetscErrorCode PetscDualSpaceGetSymmetries_Lagrange(PetscDualSpace sp, const PetscInt ****perms, const PetscScalar ****flips)
232920cf1dd8SToby Isaac {
233020cf1dd8SToby Isaac   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *) sp->data;
23313f27d899SToby Isaac   PetscInt           dim, order, Nc;
233220cf1dd8SToby Isaac   PetscErrorCode     ierr;
233320cf1dd8SToby Isaac 
233420cf1dd8SToby Isaac   PetscFunctionBegin;
233520cf1dd8SToby Isaac   ierr = PetscDualSpaceGetOrder(sp,&order);CHKERRQ(ierr);
233620cf1dd8SToby Isaac   ierr = PetscDualSpaceGetNumComponents(sp,&Nc);CHKERRQ(ierr);
233720cf1dd8SToby Isaac   ierr = DMGetDimension(sp->dm,&dim);CHKERRQ(ierr);
23383f27d899SToby Isaac   if (!lag->symComputed) { /* store symmetries */
23393f27d899SToby Isaac     PetscInt       pStart, pEnd, p;
23403f27d899SToby Isaac     PetscInt       numPoints;
234120cf1dd8SToby Isaac     PetscInt       numFaces;
23423f27d899SToby Isaac     PetscInt       spintdim;
23433f27d899SToby Isaac     PetscInt       ***symperms;
23443f27d899SToby Isaac     PetscScalar    ***symflips;
234520cf1dd8SToby Isaac 
23463f27d899SToby Isaac     ierr = DMPlexGetChart(sp->dm, &pStart, &pEnd);CHKERRQ(ierr);
23473f27d899SToby Isaac     numPoints = pEnd - pStart;
23483f27d899SToby Isaac     ierr = DMPlexGetConeSize(sp->dm, 0, &numFaces);CHKERRQ(ierr);
23493f27d899SToby Isaac     ierr = PetscCalloc1(numPoints,&symperms);CHKERRQ(ierr);
23503f27d899SToby Isaac     ierr = PetscCalloc1(numPoints,&symflips);CHKERRQ(ierr);
23513f27d899SToby Isaac     spintdim = sp->spintdim;
23523f27d899SToby Isaac     /* The nodal symmetry behavior is not present when tensorSpace != tensorCell: someone might want this for the "S"
23533f27d899SToby Isaac      * family of FEEC spaces.  Most used in particular are discontinuous polynomial L2 spaces in tensor cells, where
23543f27d899SToby Isaac      * the symmetries are not necessary for FE assembly.  So for now we assume this is the case and don't return
23553f27d899SToby Isaac      * symmetries if tensorSpace != tensorCell */
23563f27d899SToby Isaac     if (spintdim && 0 < dim && dim < 3 && (lag->tensorSpace == lag->tensorCell)) { /* compute self symmetries */
23573f27d899SToby Isaac       PetscInt **cellSymperms;
23583f27d899SToby Isaac       PetscScalar **cellSymflips;
23593f27d899SToby Isaac       PetscInt ornt;
23603f27d899SToby Isaac       PetscInt nCopies = Nc / lag->intNodeIndices->nodeVecDim;
23613f27d899SToby Isaac       PetscInt nNodes = lag->intNodeIndices->nNodes;
236220cf1dd8SToby Isaac 
236320cf1dd8SToby Isaac       lag->numSelfSym = 2 * numFaces;
236420cf1dd8SToby Isaac       lag->selfSymOff = numFaces;
23653f27d899SToby Isaac       ierr = PetscCalloc1(2*numFaces,&cellSymperms);CHKERRQ(ierr);
23663f27d899SToby Isaac       ierr = PetscCalloc1(2*numFaces,&cellSymflips);CHKERRQ(ierr);
236720cf1dd8SToby Isaac       /* we want to be able to index symmetries directly with the orientations, which range from [-numFaces,numFaces) */
23683f27d899SToby Isaac       symperms[0] = &cellSymperms[numFaces];
23693f27d899SToby Isaac       symflips[0] = &cellSymflips[numFaces];
23703f27d899SToby Isaac       if (lag->intNodeIndices->nodeVecDim * nCopies != Nc) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Node indices incompatible with dofs");
23713f27d899SToby Isaac       if (nNodes * nCopies != spintdim) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_PLIB, "Node indices incompatible with dofs");
23723f27d899SToby Isaac       for (ornt = -numFaces; ornt < numFaces; ornt++) { /* for every symmetry, compute the symmetry matrix, and extract rows to see if it fits in the perm + flip framework */
23733f27d899SToby Isaac         Mat symMat;
23743f27d899SToby Isaac         PetscInt *perm;
23753f27d899SToby Isaac         PetscScalar *flips;
23763f27d899SToby Isaac         PetscInt i;
237720cf1dd8SToby Isaac 
23783f27d899SToby Isaac         if (!ornt) continue;
23793f27d899SToby Isaac         ierr = PetscMalloc1(spintdim, &perm);CHKERRQ(ierr);
23803f27d899SToby Isaac         ierr = PetscCalloc1(spintdim, &flips);CHKERRQ(ierr);
23813f27d899SToby Isaac         for (i = 0; i < spintdim; i++) perm[i] = -1;
23823f27d899SToby Isaac         ierr = PetscDualSpaceCreateInteriorSymmetryMatrix_Lagrange(sp, ornt, &symMat);CHKERRQ(ierr);
23833f27d899SToby Isaac         for (i = 0; i < nNodes; i++) {
23843f27d899SToby Isaac           PetscInt ncols;
23853f27d899SToby Isaac           PetscInt j, k;
23863f27d899SToby Isaac           const PetscInt *cols;
23873f27d899SToby Isaac           const PetscScalar *vals;
23883f27d899SToby Isaac           PetscBool nz_seen = PETSC_FALSE;
238920cf1dd8SToby Isaac 
23903f27d899SToby Isaac           ierr = MatGetRow(symMat, i, &ncols, &cols, &vals);CHKERRQ(ierr);
23913f27d899SToby Isaac           for (j = 0; j < ncols; j++) {
23923f27d899SToby Isaac             if (PetscAbsScalar(vals[j]) > PETSC_SMALL) {
23933f27d899SToby Isaac               if (nz_seen) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "This dual space has symmetries that can't be described as a permutation + sign flips");
23943f27d899SToby Isaac               nz_seen = PETSC_TRUE;
23953f27d899SToby Isaac               if (PetscAbsScalar(PetscAbsScalar(vals[j]) - 1.) > PETSC_SMALL) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "This dual space has symmetries that can't be described as a permutation + sign flips");
23963f27d899SToby Isaac               if (PetscAbsReal(PetscImaginaryPart(vals[j])) > PETSC_SMALL) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "This dual space has symmetries that can't be described as a permutation + sign flips");
23973f27d899SToby Isaac               if (perm[cols[j] * nCopies] >= 0) SETERRQ(PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "This dual space has symmetries that can't be described as a permutation + sign flips");
23983f27d899SToby Isaac               for (k = 0; k < nCopies; k++) {
23993f27d899SToby Isaac                 perm[cols[j] * nCopies + k] = i * nCopies + k;
240020cf1dd8SToby Isaac               }
24013f27d899SToby Isaac               if (PetscRealPart(vals[j]) < 0.) {
24023f27d899SToby Isaac                 for (k = 0; k < nCopies; k++) {
24033f27d899SToby Isaac                   flips[i * nCopies + k] = -1.;
240420cf1dd8SToby Isaac                 }
240520cf1dd8SToby Isaac               } else {
24063f27d899SToby Isaac                 for (k = 0; k < nCopies; k++) {
24073f27d899SToby Isaac                   flips[i * nCopies + k] = 1.;
24083f27d899SToby Isaac                 }
24093f27d899SToby Isaac               }
24103f27d899SToby Isaac             }
24113f27d899SToby Isaac           }
24123f27d899SToby Isaac           ierr = MatRestoreRow(symMat, i, &ncols, &cols, &vals);CHKERRQ(ierr);
24133f27d899SToby Isaac         }
24143f27d899SToby Isaac         ierr = MatDestroy(&symMat);CHKERRQ(ierr);
24153f27d899SToby Isaac         /* if there were no sign flips, keep NULL */
24163f27d899SToby Isaac         for (i = 0; i < spintdim; i++) if (flips[i] != 1.) break;
24173f27d899SToby Isaac         if (i == spintdim) {
24183f27d899SToby Isaac           ierr = PetscFree(flips);CHKERRQ(ierr);
24193f27d899SToby Isaac           flips = NULL;
24203f27d899SToby Isaac         }
24213f27d899SToby Isaac         /* if the permutation is identity, keep NULL */
24223f27d899SToby Isaac         for (i = 0; i < spintdim; i++) if (perm[i] != i) break;
24233f27d899SToby Isaac         if (i == spintdim) {
24243f27d899SToby Isaac           ierr = PetscFree(perm);CHKERRQ(ierr);
24253f27d899SToby Isaac           perm = NULL;
24263f27d899SToby Isaac         }
24273f27d899SToby Isaac         symperms[0][ornt] = perm;
24283f27d899SToby Isaac         symflips[0][ornt] = flips;
24293f27d899SToby Isaac       }
24303f27d899SToby Isaac       /* if no orientations produced non-identity permutations, keep NULL */
24313f27d899SToby Isaac       for (ornt = -numFaces; ornt < numFaces; ornt++) if (symperms[0][ornt]) break;
24323f27d899SToby Isaac       if (ornt == numFaces) {
24333f27d899SToby Isaac         ierr = PetscFree(cellSymperms);CHKERRQ(ierr);
24343f27d899SToby Isaac         symperms[0] = NULL;
24353f27d899SToby Isaac       }
24363f27d899SToby Isaac       /* if no orientations produced sign flips, keep NULL */
24373f27d899SToby Isaac       for (ornt = -numFaces; ornt < numFaces; ornt++) if (symflips[0][ornt]) break;
24383f27d899SToby Isaac       if (ornt == numFaces) {
24393f27d899SToby Isaac         ierr = PetscFree(cellSymflips);CHKERRQ(ierr);
24403f27d899SToby Isaac         symflips[0] = NULL;
24413f27d899SToby Isaac       }
24423f27d899SToby Isaac     }
24433f27d899SToby Isaac     {
24443f27d899SToby Isaac       PetscInt closureSize = 0;
24453f27d899SToby Isaac       PetscInt *closure = NULL;
24463f27d899SToby Isaac       PetscInt r;
244720cf1dd8SToby Isaac 
24483f27d899SToby Isaac       ierr = DMPlexGetTransitiveClosure(sp->dm,0,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
24493f27d899SToby Isaac       for (r = 0; r < closureSize; r++) {
24503f27d899SToby Isaac         PetscDualSpace psp;
24513f27d899SToby Isaac         PetscInt point = closure[2 * r];
24523f27d899SToby Isaac         PetscInt pspintdim;
24533f27d899SToby Isaac         const PetscInt ***psymperms = NULL;
24543f27d899SToby Isaac         const PetscScalar ***psymflips = NULL;
245520cf1dd8SToby Isaac 
24563f27d899SToby Isaac         if (!point) continue;
24573f27d899SToby Isaac         ierr = PetscDualSpaceGetPointSubspace(sp, point, &psp);CHKERRQ(ierr);
24583f27d899SToby Isaac         if (!psp) continue;
24593f27d899SToby Isaac         ierr = PetscDualSpaceGetInteriorDimension(psp, &pspintdim);CHKERRQ(ierr);
24603f27d899SToby Isaac         if (!pspintdim) continue;
24613f27d899SToby Isaac         ierr = PetscDualSpaceGetSymmetries(psp,&psymperms,&psymflips);CHKERRQ(ierr);
24623f27d899SToby Isaac         symperms[r] = (PetscInt **) (psymperms ? psymperms[0] : NULL);
24633f27d899SToby Isaac         symflips[r] = (PetscScalar **) (psymflips ? psymflips[0] : NULL);
246420cf1dd8SToby Isaac       }
24653f27d899SToby Isaac       ierr = DMPlexRestoreTransitiveClosure(sp->dm,0,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
246620cf1dd8SToby Isaac     }
24673f27d899SToby Isaac     for (p = 0; p < pEnd; p++) if (symperms[p]) break;
24683f27d899SToby Isaac     if (p == pEnd) {
24693f27d899SToby Isaac       ierr = PetscFree(symperms);CHKERRQ(ierr);
24703f27d899SToby Isaac       symperms = NULL;
247120cf1dd8SToby Isaac     }
24723f27d899SToby Isaac     for (p = 0; p < pEnd; p++) if (symflips[p]) break;
24733f27d899SToby Isaac     if (p == pEnd) {
24743f27d899SToby Isaac       ierr = PetscFree(symflips);CHKERRQ(ierr);
24753f27d899SToby Isaac       symflips = NULL;
247620cf1dd8SToby Isaac     }
24773f27d899SToby Isaac     lag->symperms = symperms;
24783f27d899SToby Isaac     lag->symflips = symflips;
24793f27d899SToby Isaac     lag->symComputed = PETSC_TRUE;
248020cf1dd8SToby Isaac   }
24813f27d899SToby Isaac   if (perms) *perms = (const PetscInt ***) lag->symperms;
24823f27d899SToby Isaac   if (flips) *flips = (const PetscScalar ***) lag->symflips;
248320cf1dd8SToby Isaac   PetscFunctionReturn(0);
248420cf1dd8SToby Isaac }
248520cf1dd8SToby Isaac 
248620cf1dd8SToby Isaac static PetscErrorCode PetscDualSpaceLagrangeGetContinuity_Lagrange(PetscDualSpace sp, PetscBool *continuous)
248720cf1dd8SToby Isaac {
248820cf1dd8SToby Isaac   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *) sp->data;
248920cf1dd8SToby Isaac 
249020cf1dd8SToby Isaac   PetscFunctionBegin;
249120cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
249220cf1dd8SToby Isaac   PetscValidPointer(continuous, 2);
249320cf1dd8SToby Isaac   *continuous = lag->continuous;
249420cf1dd8SToby Isaac   PetscFunctionReturn(0);
249520cf1dd8SToby Isaac }
249620cf1dd8SToby Isaac 
249720cf1dd8SToby Isaac static PetscErrorCode PetscDualSpaceLagrangeSetContinuity_Lagrange(PetscDualSpace sp, PetscBool continuous)
249820cf1dd8SToby Isaac {
249920cf1dd8SToby Isaac   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *) sp->data;
250020cf1dd8SToby Isaac 
250120cf1dd8SToby Isaac   PetscFunctionBegin;
250220cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
250320cf1dd8SToby Isaac   lag->continuous = continuous;
250420cf1dd8SToby Isaac   PetscFunctionReturn(0);
250520cf1dd8SToby Isaac }
250620cf1dd8SToby Isaac 
250720cf1dd8SToby Isaac /*@
250820cf1dd8SToby Isaac   PetscDualSpaceLagrangeGetContinuity - Retrieves the flag for element continuity
250920cf1dd8SToby Isaac 
251020cf1dd8SToby Isaac   Not Collective
251120cf1dd8SToby Isaac 
251220cf1dd8SToby Isaac   Input Parameter:
251320cf1dd8SToby Isaac . sp         - the PetscDualSpace
251420cf1dd8SToby Isaac 
251520cf1dd8SToby Isaac   Output Parameter:
251620cf1dd8SToby Isaac . continuous - flag for element continuity
251720cf1dd8SToby Isaac 
251820cf1dd8SToby Isaac   Level: intermediate
251920cf1dd8SToby Isaac 
252020cf1dd8SToby Isaac .seealso: PetscDualSpaceLagrangeSetContinuity()
252120cf1dd8SToby Isaac @*/
252220cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceLagrangeGetContinuity(PetscDualSpace sp, PetscBool *continuous)
252320cf1dd8SToby Isaac {
252420cf1dd8SToby Isaac   PetscErrorCode ierr;
252520cf1dd8SToby Isaac 
252620cf1dd8SToby Isaac   PetscFunctionBegin;
252720cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
252820cf1dd8SToby Isaac   PetscValidPointer(continuous, 2);
252920cf1dd8SToby Isaac   ierr = PetscTryMethod(sp, "PetscDualSpaceLagrangeGetContinuity_C", (PetscDualSpace,PetscBool*),(sp,continuous));CHKERRQ(ierr);
253020cf1dd8SToby Isaac   PetscFunctionReturn(0);
253120cf1dd8SToby Isaac }
253220cf1dd8SToby Isaac 
253320cf1dd8SToby Isaac /*@
253420cf1dd8SToby Isaac   PetscDualSpaceLagrangeSetContinuity - Indicate whether the element is continuous
253520cf1dd8SToby Isaac 
2536d083f849SBarry Smith   Logically Collective on sp
253720cf1dd8SToby Isaac 
253820cf1dd8SToby Isaac   Input Parameters:
253920cf1dd8SToby Isaac + sp         - the PetscDualSpace
254020cf1dd8SToby Isaac - continuous - flag for element continuity
254120cf1dd8SToby Isaac 
254220cf1dd8SToby Isaac   Options Database:
254320cf1dd8SToby Isaac . -petscdualspace_lagrange_continuity <bool>
254420cf1dd8SToby Isaac 
254520cf1dd8SToby Isaac   Level: intermediate
254620cf1dd8SToby Isaac 
254720cf1dd8SToby Isaac .seealso: PetscDualSpaceLagrangeGetContinuity()
254820cf1dd8SToby Isaac @*/
254920cf1dd8SToby Isaac PetscErrorCode PetscDualSpaceLagrangeSetContinuity(PetscDualSpace sp, PetscBool continuous)
255020cf1dd8SToby Isaac {
255120cf1dd8SToby Isaac   PetscErrorCode ierr;
255220cf1dd8SToby Isaac 
255320cf1dd8SToby Isaac   PetscFunctionBegin;
255420cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
255520cf1dd8SToby Isaac   PetscValidLogicalCollectiveBool(sp, continuous, 2);
255620cf1dd8SToby Isaac   ierr = PetscTryMethod(sp, "PetscDualSpaceLagrangeSetContinuity_C", (PetscDualSpace,PetscBool),(sp,continuous));CHKERRQ(ierr);
255720cf1dd8SToby Isaac   PetscFunctionReturn(0);
255820cf1dd8SToby Isaac }
255920cf1dd8SToby Isaac 
25606f905325SMatthew G. Knepley static PetscErrorCode PetscDualSpaceLagrangeGetTensor_Lagrange(PetscDualSpace sp, PetscBool *tensor)
256120cf1dd8SToby Isaac {
256220cf1dd8SToby Isaac   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *)sp->data;
25636f905325SMatthew G. Knepley 
25646f905325SMatthew G. Knepley   PetscFunctionBegin;
25656f905325SMatthew G. Knepley   *tensor = lag->tensorSpace;
25666f905325SMatthew G. Knepley   PetscFunctionReturn(0);
25676f905325SMatthew G. Knepley }
25686f905325SMatthew G. Knepley 
25696f905325SMatthew G. Knepley static PetscErrorCode PetscDualSpaceLagrangeSetTensor_Lagrange(PetscDualSpace sp, PetscBool tensor)
25706f905325SMatthew G. Knepley {
25716f905325SMatthew G. Knepley   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *)sp->data;
25726f905325SMatthew G. Knepley 
25736f905325SMatthew G. Knepley   PetscFunctionBegin;
25746f905325SMatthew G. Knepley   lag->tensorSpace = tensor;
25756f905325SMatthew G. Knepley   PetscFunctionReturn(0);
25766f905325SMatthew G. Knepley }
25776f905325SMatthew G. Knepley 
25783f27d899SToby Isaac static PetscErrorCode PetscDualSpaceLagrangeGetTrimmed_Lagrange(PetscDualSpace sp, PetscBool *trimmed)
25793f27d899SToby Isaac {
25803f27d899SToby Isaac   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *)sp->data;
25813f27d899SToby Isaac 
25823f27d899SToby Isaac   PetscFunctionBegin;
25833f27d899SToby Isaac   *trimmed = lag->trimmed;
25843f27d899SToby Isaac   PetscFunctionReturn(0);
25853f27d899SToby Isaac }
25863f27d899SToby Isaac 
25873f27d899SToby Isaac static PetscErrorCode PetscDualSpaceLagrangeSetTrimmed_Lagrange(PetscDualSpace sp, PetscBool trimmed)
25883f27d899SToby Isaac {
25893f27d899SToby Isaac   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *)sp->data;
25903f27d899SToby Isaac 
25913f27d899SToby Isaac   PetscFunctionBegin;
25923f27d899SToby Isaac   lag->trimmed = trimmed;
25933f27d899SToby Isaac   PetscFunctionReturn(0);
25943f27d899SToby Isaac }
25953f27d899SToby Isaac 
25963f27d899SToby Isaac static PetscErrorCode PetscDualSpaceLagrangeGetNodeType_Lagrange(PetscDualSpace sp, PetscDTNodeType *nodeType, PetscBool *boundary, PetscReal *exponent)
25973f27d899SToby Isaac {
25983f27d899SToby Isaac   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *)sp->data;
25993f27d899SToby Isaac 
26003f27d899SToby Isaac   PetscFunctionBegin;
26013f27d899SToby Isaac   if (nodeType) *nodeType = lag->nodeType;
26023f27d899SToby Isaac   if (boundary) *boundary = lag->endNodes;
26033f27d899SToby Isaac   if (exponent) *exponent = lag->nodeExponent;
26043f27d899SToby Isaac   PetscFunctionReturn(0);
26053f27d899SToby Isaac }
26063f27d899SToby Isaac 
26073f27d899SToby Isaac static PetscErrorCode PetscDualSpaceLagrangeSetNodeType_Lagrange(PetscDualSpace sp, PetscDTNodeType nodeType, PetscBool boundary, PetscReal exponent)
26083f27d899SToby Isaac {
26093f27d899SToby Isaac   PetscDualSpace_Lag *lag = (PetscDualSpace_Lag *)sp->data;
26103f27d899SToby Isaac 
26113f27d899SToby Isaac   PetscFunctionBegin;
26123f27d899SToby Isaac   if (nodeType == PETSCDTNODES_GAUSSJACOBI && exponent <= -1.) SETERRQ(PetscObjectComm((PetscObject) sp), PETSC_ERR_ARG_OUTOFRANGE, "Exponent must be > -1");
26133f27d899SToby Isaac   lag->nodeType = nodeType;
26143f27d899SToby Isaac   lag->endNodes = boundary;
26153f27d899SToby Isaac   lag->nodeExponent = exponent;
26163f27d899SToby Isaac   PetscFunctionReturn(0);
26173f27d899SToby Isaac }
26183f27d899SToby Isaac 
26196f905325SMatthew G. Knepley /*@
26206f905325SMatthew G. Knepley   PetscDualSpaceLagrangeGetTensor - Get the tensor nature of the dual space
26216f905325SMatthew G. Knepley 
26226f905325SMatthew G. Knepley   Not collective
26236f905325SMatthew G. Knepley 
26246f905325SMatthew G. Knepley   Input Parameter:
26256f905325SMatthew G. Knepley . sp - The PetscDualSpace
26266f905325SMatthew G. Knepley 
26276f905325SMatthew G. Knepley   Output Parameter:
26286f905325SMatthew G. Knepley . tensor - Whether the dual space has tensor layout (vs. simplicial)
26296f905325SMatthew G. Knepley 
26306f905325SMatthew G. Knepley   Level: intermediate
26316f905325SMatthew G. Knepley 
26326f905325SMatthew G. Knepley .seealso: PetscDualSpaceLagrangeSetTensor(), PetscDualSpaceCreate()
26336f905325SMatthew G. Knepley @*/
26346f905325SMatthew G. Knepley PetscErrorCode PetscDualSpaceLagrangeGetTensor(PetscDualSpace sp, PetscBool *tensor)
26356f905325SMatthew G. Knepley {
263620cf1dd8SToby Isaac   PetscErrorCode ierr;
263720cf1dd8SToby Isaac 
263820cf1dd8SToby Isaac   PetscFunctionBegin;
263920cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
26406f905325SMatthew G. Knepley   PetscValidPointer(tensor, 2);
26416f905325SMatthew G. Knepley   ierr = PetscTryMethod(sp,"PetscDualSpaceLagrangeGetTensor_C",(PetscDualSpace,PetscBool *),(sp,tensor));CHKERRQ(ierr);
264220cf1dd8SToby Isaac   PetscFunctionReturn(0);
264320cf1dd8SToby Isaac }
264420cf1dd8SToby Isaac 
26456f905325SMatthew G. Knepley /*@
26466f905325SMatthew G. Knepley   PetscDualSpaceLagrangeSetTensor - Set the tensor nature of the dual space
26476f905325SMatthew G. Knepley 
26486f905325SMatthew G. Knepley   Not collective
26496f905325SMatthew G. Knepley 
26506f905325SMatthew G. Knepley   Input Parameters:
26516f905325SMatthew G. Knepley + sp - The PetscDualSpace
26526f905325SMatthew G. Knepley - tensor - Whether the dual space has tensor layout (vs. simplicial)
26536f905325SMatthew G. Knepley 
26546f905325SMatthew G. Knepley   Level: intermediate
26556f905325SMatthew G. Knepley 
26566f905325SMatthew G. Knepley .seealso: PetscDualSpaceLagrangeGetTensor(), PetscDualSpaceCreate()
26576f905325SMatthew G. Knepley @*/
26586f905325SMatthew G. Knepley PetscErrorCode PetscDualSpaceLagrangeSetTensor(PetscDualSpace sp, PetscBool tensor)
26596f905325SMatthew G. Knepley {
26606f905325SMatthew G. Knepley   PetscErrorCode ierr;
26616f905325SMatthew G. Knepley 
26626f905325SMatthew G. Knepley   PetscFunctionBegin;
26636f905325SMatthew G. Knepley   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
26646f905325SMatthew G. Knepley   ierr = PetscTryMethod(sp,"PetscDualSpaceLagrangeSetTensor_C",(PetscDualSpace,PetscBool),(sp,tensor));CHKERRQ(ierr);
26656f905325SMatthew G. Knepley   PetscFunctionReturn(0);
26666f905325SMatthew G. Knepley }
26676f905325SMatthew G. Knepley 
26683f27d899SToby Isaac /*@
26693f27d899SToby Isaac   PetscDualSpaceLagrangeGetTrimmed - Get the trimmed nature of the dual space
26703f27d899SToby Isaac 
26713f27d899SToby Isaac   Not collective
26723f27d899SToby Isaac 
26733f27d899SToby Isaac   Input Parameter:
26743f27d899SToby Isaac . sp - The PetscDualSpace
26753f27d899SToby Isaac 
26763f27d899SToby Isaac   Output Parameter:
26773f27d899SToby Isaac . trimmed - Whether the dual space represents to dual basis of a trimmed polynomial space (e.g. Raviart-Thomas and higher order / other form degree variants)
26783f27d899SToby Isaac 
26793f27d899SToby Isaac   Level: intermediate
26803f27d899SToby Isaac 
26813f27d899SToby Isaac .seealso: PetscDualSpaceLagrangeSetTrimmed(), PetscDualSpaceCreate()
26823f27d899SToby Isaac @*/
26833f27d899SToby Isaac PetscErrorCode PetscDualSpaceLagrangeGetTrimmed(PetscDualSpace sp, PetscBool *trimmed)
26843f27d899SToby Isaac {
26853f27d899SToby Isaac   PetscErrorCode ierr;
26863f27d899SToby Isaac 
26873f27d899SToby Isaac   PetscFunctionBegin;
26883f27d899SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
26893f27d899SToby Isaac   PetscValidPointer(trimmed, 2);
26903f27d899SToby Isaac   ierr = PetscTryMethod(sp,"PetscDualSpaceLagrangeGetTrimmed_C",(PetscDualSpace,PetscBool *),(sp,trimmed));CHKERRQ(ierr);
26913f27d899SToby Isaac   PetscFunctionReturn(0);
26923f27d899SToby Isaac }
26933f27d899SToby Isaac 
26943f27d899SToby Isaac /*@
26953f27d899SToby Isaac   PetscDualSpaceLagrangeSetTrimmed - Set the trimmed nature of the dual space
26963f27d899SToby Isaac 
26973f27d899SToby Isaac   Not collective
26983f27d899SToby Isaac 
26993f27d899SToby Isaac   Input Parameters:
27003f27d899SToby Isaac + sp - The PetscDualSpace
27013f27d899SToby Isaac - trimmed - Whether the dual space represents to dual basis of a trimmed polynomial space (e.g. Raviart-Thomas and higher order / other form degree variants)
27023f27d899SToby Isaac 
27033f27d899SToby Isaac   Level: intermediate
27043f27d899SToby Isaac 
27053f27d899SToby Isaac .seealso: PetscDualSpaceLagrangeGetTrimmed(), PetscDualSpaceCreate()
27063f27d899SToby Isaac @*/
27073f27d899SToby Isaac PetscErrorCode PetscDualSpaceLagrangeSetTrimmed(PetscDualSpace sp, PetscBool trimmed)
27083f27d899SToby Isaac {
27093f27d899SToby Isaac   PetscErrorCode ierr;
27103f27d899SToby Isaac 
27113f27d899SToby Isaac   PetscFunctionBegin;
27123f27d899SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
27133f27d899SToby Isaac   ierr = PetscTryMethod(sp,"PetscDualSpaceLagrangeSetTrimmed_C",(PetscDualSpace,PetscBool),(sp,trimmed));CHKERRQ(ierr);
27143f27d899SToby Isaac   PetscFunctionReturn(0);
27153f27d899SToby Isaac }
27163f27d899SToby Isaac 
27173f27d899SToby Isaac /*@
27183f27d899SToby Isaac   PetscDualSpaceLagrangeGetNodeType - Get a description of how nodes are laid out for Lagrange polynomials in this
27193f27d899SToby Isaac   dual space
27203f27d899SToby Isaac 
27213f27d899SToby Isaac   Not collective
27223f27d899SToby Isaac 
27233f27d899SToby Isaac   Input Parameter:
27243f27d899SToby Isaac . sp - The PetscDualSpace
27253f27d899SToby Isaac 
27263f27d899SToby Isaac   Output Parameters:
27273f27d899SToby Isaac + nodeType - The type of nodes
27283f27d899SToby Isaac . boundary - Whether the node type is one that includes endpoints (if nodeType is PETSCDTNODES_GAUSSJACOBI, nodes that
27293f27d899SToby Isaac              include the boundary are Gauss-Lobatto-Jacobi nodes)
27303f27d899SToby Isaac - exponent - If nodeType is PETSCDTNODES_GAUSJACOBI, indicates the exponent used for both ends of the 1D Jacobi weight function
27313f27d899SToby Isaac              '0' is Gauss-Legendre, '-0.5' is Gauss-Chebyshev of the first type, '0.5' is Gauss-Chebyshev of the second type
27323f27d899SToby Isaac 
27333f27d899SToby Isaac   Level: advanced
27343f27d899SToby Isaac 
27353f27d899SToby Isaac .seealso: PetscDTNodeType, PetscDualSpaceLagrangeSetNodeType()
27363f27d899SToby Isaac @*/
27373f27d899SToby Isaac PetscErrorCode PetscDualSpaceLagrangeGetNodeType(PetscDualSpace sp, PetscDTNodeType *nodeType, PetscBool *boundary, PetscReal *exponent)
27383f27d899SToby Isaac {
27393f27d899SToby Isaac   PetscErrorCode ierr;
27403f27d899SToby Isaac 
27413f27d899SToby Isaac   PetscFunctionBegin;
27423f27d899SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
27433f27d899SToby Isaac   if (nodeType) PetscValidPointer(nodeType, 2);
27443f27d899SToby Isaac   if (boundary) PetscValidPointer(boundary, 3);
27453f27d899SToby Isaac   if (exponent) PetscValidPointer(exponent, 4);
27463f27d899SToby Isaac   ierr = PetscTryMethod(sp,"PetscDualSpaceLagrangeGetNodeType_C",(PetscDualSpace,PetscDTNodeType *,PetscBool *,PetscReal *),(sp,nodeType,boundary,exponent));CHKERRQ(ierr);
27473f27d899SToby Isaac   PetscFunctionReturn(0);
27483f27d899SToby Isaac }
27493f27d899SToby Isaac 
27503f27d899SToby Isaac /*@
27513f27d899SToby Isaac   PetscDualSpaceLagrangeSetNodeType - Set a description of how nodes are laid out for Lagrange polynomials in this
27523f27d899SToby Isaac   dual space
27533f27d899SToby Isaac 
27543f27d899SToby Isaac   Logically collective
27553f27d899SToby Isaac 
27563f27d899SToby Isaac   Input Parameters:
27573f27d899SToby Isaac + sp - The PetscDualSpace
27583f27d899SToby Isaac . nodeType - The type of nodes
27593f27d899SToby Isaac . boundary - Whether the node type is one that includes endpoints (if nodeType is PETSCDTNODES_GAUSSJACOBI, nodes that
27603f27d899SToby Isaac              include the boundary are Gauss-Lobatto-Jacobi nodes)
27613f27d899SToby Isaac - exponent - If nodeType is PETSCDTNODES_GAUSJACOBI, indicates the exponent used for both ends of the 1D Jacobi weight function
27623f27d899SToby Isaac              '0' is Gauss-Legendre, '-0.5' is Gauss-Chebyshev of the first type, '0.5' is Gauss-Chebyshev of the second type
27633f27d899SToby Isaac 
27643f27d899SToby Isaac   Level: advanced
27653f27d899SToby Isaac 
27663f27d899SToby Isaac .seealso: PetscDTNodeType, PetscDualSpaceLagrangeGetNodeType()
27673f27d899SToby Isaac @*/
27683f27d899SToby Isaac PetscErrorCode PetscDualSpaceLagrangeSetNodeType(PetscDualSpace sp, PetscDTNodeType nodeType, PetscBool boundary, PetscReal exponent)
27693f27d899SToby Isaac {
27703f27d899SToby Isaac   PetscErrorCode ierr;
27713f27d899SToby Isaac 
27723f27d899SToby Isaac   PetscFunctionBegin;
27733f27d899SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
27743f27d899SToby Isaac   ierr = PetscTryMethod(sp,"PetscDualSpaceLagrangeSetNodeType_C",(PetscDualSpace,PetscDTNodeType,PetscBool,PetscReal),(sp,nodeType,boundary,exponent));CHKERRQ(ierr);
27753f27d899SToby Isaac   PetscFunctionReturn(0);
27763f27d899SToby Isaac }
27773f27d899SToby Isaac 
27783f27d899SToby Isaac 
27796f905325SMatthew G. Knepley static PetscErrorCode PetscDualSpaceInitialize_Lagrange(PetscDualSpace sp)
278020cf1dd8SToby Isaac {
278120cf1dd8SToby Isaac   PetscFunctionBegin;
278220cf1dd8SToby Isaac   sp->ops->destroy              = PetscDualSpaceDestroy_Lagrange;
27836f905325SMatthew G. Knepley   sp->ops->view                 = PetscDualSpaceView_Lagrange;
27846f905325SMatthew G. Knepley   sp->ops->setfromoptions       = PetscDualSpaceSetFromOptions_Lagrange;
278520cf1dd8SToby Isaac   sp->ops->duplicate            = PetscDualSpaceDuplicate_Lagrange;
27866f905325SMatthew G. Knepley   sp->ops->setup                = PetscDualSpaceSetUp_Lagrange;
27873f27d899SToby Isaac   sp->ops->createheightsubspace = NULL;
27883f27d899SToby Isaac   sp->ops->createpointsubspace  = NULL;
278920cf1dd8SToby Isaac   sp->ops->getsymmetries        = PetscDualSpaceGetSymmetries_Lagrange;
279020cf1dd8SToby Isaac   sp->ops->apply                = PetscDualSpaceApplyDefault;
279120cf1dd8SToby Isaac   sp->ops->applyall             = PetscDualSpaceApplyAllDefault;
2792b4457527SToby Isaac   sp->ops->applyint             = PetscDualSpaceApplyInteriorDefault;
27933f27d899SToby Isaac   sp->ops->createalldata        = PetscDualSpaceCreateAllDataDefault;
2794b4457527SToby Isaac   sp->ops->createintdata        = PetscDualSpaceCreateInteriorDataDefault;
279520cf1dd8SToby Isaac   PetscFunctionReturn(0);
279620cf1dd8SToby Isaac }
279720cf1dd8SToby Isaac 
279820cf1dd8SToby Isaac /*MC
279920cf1dd8SToby Isaac   PETSCDUALSPACELAGRANGE = "lagrange" - A PetscDualSpace object that encapsulates a dual space of pointwise evaluation functionals
280020cf1dd8SToby Isaac 
280120cf1dd8SToby Isaac   Level: intermediate
280220cf1dd8SToby Isaac 
280320cf1dd8SToby Isaac .seealso: PetscDualSpaceType, PetscDualSpaceCreate(), PetscDualSpaceSetType()
280420cf1dd8SToby Isaac M*/
280520cf1dd8SToby Isaac PETSC_EXTERN PetscErrorCode PetscDualSpaceCreate_Lagrange(PetscDualSpace sp)
280620cf1dd8SToby Isaac {
280720cf1dd8SToby Isaac   PetscDualSpace_Lag *lag;
280820cf1dd8SToby Isaac   PetscErrorCode      ierr;
280920cf1dd8SToby Isaac 
281020cf1dd8SToby Isaac   PetscFunctionBegin;
281120cf1dd8SToby Isaac   PetscValidHeaderSpecific(sp, PETSCDUALSPACE_CLASSID, 1);
281220cf1dd8SToby Isaac   ierr     = PetscNewLog(sp,&lag);CHKERRQ(ierr);
281320cf1dd8SToby Isaac   sp->data = lag;
281420cf1dd8SToby Isaac 
28153f27d899SToby Isaac   lag->tensorCell  = PETSC_FALSE;
281620cf1dd8SToby Isaac   lag->tensorSpace = PETSC_FALSE;
281720cf1dd8SToby Isaac   lag->continuous  = PETSC_TRUE;
28183f27d899SToby Isaac   lag->numCopies   = PETSC_DEFAULT;
28193f27d899SToby Isaac   lag->numNodeSkip = PETSC_DEFAULT;
28203f27d899SToby Isaac   lag->nodeType    = PETSCDTNODES_DEFAULT;
282120cf1dd8SToby Isaac 
282220cf1dd8SToby Isaac   ierr = PetscDualSpaceInitialize_Lagrange(sp);CHKERRQ(ierr);
282320cf1dd8SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeGetContinuity_C", PetscDualSpaceLagrangeGetContinuity_Lagrange);CHKERRQ(ierr);
282420cf1dd8SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeSetContinuity_C", PetscDualSpaceLagrangeSetContinuity_Lagrange);CHKERRQ(ierr);
282520cf1dd8SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeGetTensor_C", PetscDualSpaceLagrangeGetTensor_Lagrange);CHKERRQ(ierr);
282620cf1dd8SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeSetTensor_C", PetscDualSpaceLagrangeSetTensor_Lagrange);CHKERRQ(ierr);
28273f27d899SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeGetTrimmed_C", PetscDualSpaceLagrangeGetTrimmed_Lagrange);CHKERRQ(ierr);
28283f27d899SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeSetTrimmed_C", PetscDualSpaceLagrangeSetTrimmed_Lagrange);CHKERRQ(ierr);
28293f27d899SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeGetNodeType_C", PetscDualSpaceLagrangeGetNodeType_Lagrange);CHKERRQ(ierr);
28303f27d899SToby Isaac   ierr = PetscObjectComposeFunction((PetscObject) sp, "PetscDualSpaceLagrangeSetNodeType_C", PetscDualSpaceLagrangeSetNodeType_Lagrange);CHKERRQ(ierr);
283120cf1dd8SToby Isaac   PetscFunctionReturn(0);
283220cf1dd8SToby Isaac }
283320cf1dd8SToby Isaac 
2834