xref: /petsc/src/dm/impls/plex/plexglvis.c (revision 08401ef684002a709c6d3db98a0c9f54a8bcf1ec)
18135c375SStefano Zampini #include <petsc/private/glvisviewerimpl.h>
28135c375SStefano Zampini #include <petsc/private/petscimpl.h>
38135c375SStefano Zampini #include <petsc/private/dmpleximpl.h>
48135c375SStefano Zampini #include <petscbt.h>
58135c375SStefano Zampini #include <petscdmplex.h>
68135c375SStefano Zampini #include <petscsf.h>
78135c375SStefano Zampini #include <petscds.h>
88135c375SStefano Zampini 
98135c375SStefano Zampini typedef struct {
108135c375SStefano Zampini   PetscInt   nf;
118135c375SStefano Zampini   VecScatter *scctx;
128135c375SStefano Zampini } GLVisViewerCtx;
138135c375SStefano Zampini 
148135c375SStefano Zampini static PetscErrorCode DestroyGLVisViewerCtx_Private(void *vctx)
158135c375SStefano Zampini {
168135c375SStefano Zampini   GLVisViewerCtx *ctx = (GLVisViewerCtx*)vctx;
178135c375SStefano Zampini   PetscInt       i;
188135c375SStefano Zampini 
198135c375SStefano Zampini   PetscFunctionBegin;
208135c375SStefano Zampini   for (i=0;i<ctx->nf;i++) {
219566063dSJacob Faibussowitsch     PetscCall(VecScatterDestroy(&ctx->scctx[i]));
228135c375SStefano Zampini   }
239566063dSJacob Faibussowitsch   PetscCall(PetscFree(ctx->scctx));
249566063dSJacob Faibussowitsch   PetscCall(PetscFree(vctx));
258135c375SStefano Zampini   PetscFunctionReturn(0);
268135c375SStefano Zampini }
278135c375SStefano Zampini 
288135c375SStefano Zampini static PetscErrorCode DMPlexSampleGLVisFields_Private(PetscObject oX, PetscInt nf, PetscObject oXfield[], void *vctx)
298135c375SStefano Zampini {
308135c375SStefano Zampini   GLVisViewerCtx *ctx = (GLVisViewerCtx*)vctx;
318135c375SStefano Zampini   PetscInt       f;
328135c375SStefano Zampini 
338135c375SStefano Zampini   PetscFunctionBegin;
348135c375SStefano Zampini   for (f=0;f<nf;f++) {
359566063dSJacob Faibussowitsch     PetscCall(VecScatterBegin(ctx->scctx[f],(Vec)oX,(Vec)oXfield[f],INSERT_VALUES,SCATTER_FORWARD));
369566063dSJacob Faibussowitsch     PetscCall(VecScatterEnd(ctx->scctx[f],(Vec)oX,(Vec)oXfield[f],INSERT_VALUES,SCATTER_FORWARD));
378135c375SStefano Zampini   }
388135c375SStefano Zampini   PetscFunctionReturn(0);
398135c375SStefano Zampini }
408135c375SStefano Zampini 
418135c375SStefano Zampini /* for FEM, it works for H1 fields only and extracts dofs at cell vertices, discarding any other dof */
428135c375SStefano Zampini PetscErrorCode DMSetUpGLVisViewer_Plex(PetscObject odm, PetscViewer viewer)
438135c375SStefano Zampini {
448135c375SStefano Zampini   DM             dm = (DM)odm;
454cac2994SStefano Zampini   Vec            xlocal,xfield,*Ufield;
468135c375SStefano Zampini   PetscDS        ds;
478135c375SStefano Zampini   IS             globalNum,isfield;
488135c375SStefano Zampini   PetscBT        vown;
498135c375SStefano Zampini   char           **fieldname = NULL,**fec_type = NULL;
508135c375SStefano Zampini   const PetscInt *gNum;
51bb77a09fSStefano Zampini   PetscInt       *nlocal,*bs,*idxs,*dims;
528135c375SStefano Zampini   PetscInt       f,maxfields,nfields,c,totc,totdofs,Nv,cum,i;
53b135d7daSStefano Zampini   PetscInt       dim,cStart,cEnd,vStart,vEnd;
548135c375SStefano Zampini   GLVisViewerCtx *ctx;
558135c375SStefano Zampini   PetscSection   s;
568135c375SStefano Zampini 
578135c375SStefano Zampini   PetscFunctionBegin;
589566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm,&dim));
599566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm,0,&vStart,&vEnd));
609566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm,0,&cStart,&cEnd));
619566063dSJacob Faibussowitsch   PetscCall(PetscObjectQuery((PetscObject)dm,"_glvis_plex_gnum",(PetscObject*)&globalNum));
62b135d7daSStefano Zampini   if (!globalNum) {
639566063dSJacob Faibussowitsch     PetscCall(DMPlexCreateCellNumbering_Internal(dm,PETSC_TRUE,&globalNum));
649566063dSJacob Faibussowitsch     PetscCall(PetscObjectCompose((PetscObject)dm,"_glvis_plex_gnum",(PetscObject)globalNum));
659566063dSJacob Faibussowitsch     PetscCall(PetscObjectDereference((PetscObject)globalNum));
66b135d7daSStefano Zampini   }
679566063dSJacob Faibussowitsch   PetscCall(ISGetIndices(globalNum,&gNum));
689566063dSJacob Faibussowitsch   PetscCall(PetscBTCreate(vEnd-vStart,&vown));
698135c375SStefano Zampini   for (c = cStart, totc = 0; c < cEnd; c++) {
708135c375SStefano Zampini     if (gNum[c-cStart] >= 0) {
718135c375SStefano Zampini       PetscInt i,numPoints,*points = NULL;
728135c375SStefano Zampini 
738135c375SStefano Zampini       totc++;
749566063dSJacob Faibussowitsch       PetscCall(DMPlexGetTransitiveClosure(dm,c,PETSC_TRUE,&numPoints,&points));
758135c375SStefano Zampini       for (i=0;i<numPoints*2;i+= 2) {
768135c375SStefano Zampini         if ((points[i] >= vStart) && (points[i] < vEnd)) {
779566063dSJacob Faibussowitsch           PetscCall(PetscBTSet(vown,points[i]-vStart));
788135c375SStefano Zampini         }
798135c375SStefano Zampini       }
809566063dSJacob Faibussowitsch       PetscCall(DMPlexRestoreTransitiveClosure(dm,c,PETSC_TRUE,&numPoints,&points));
818135c375SStefano Zampini     }
828135c375SStefano Zampini   }
8377eacf09SStefano Zampini   for (f=0,Nv=0;f<vEnd-vStart;f++) if (PetscLikely(PetscBTLookup(vown,f))) Nv++;
848135c375SStefano Zampini 
859566063dSJacob Faibussowitsch   PetscCall(DMCreateLocalVector(dm,&xlocal));
869566063dSJacob Faibussowitsch   PetscCall(VecGetLocalSize(xlocal,&totdofs));
879566063dSJacob Faibussowitsch   PetscCall(DMGetLocalSection(dm,&s));
889566063dSJacob Faibussowitsch   PetscCall(PetscSectionGetNumFields(s,&nfields));
898135c375SStefano Zampini   for (f=0,maxfields=0;f<nfields;f++) {
908135c375SStefano Zampini     PetscInt bs;
918135c375SStefano Zampini 
929566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetFieldComponents(s,f,&bs));
938135c375SStefano Zampini     maxfields += bs;
948135c375SStefano Zampini   }
959566063dSJacob Faibussowitsch   PetscCall(PetscCalloc7(maxfields,&fieldname,maxfields,&nlocal,maxfields,&bs,maxfields,&dims,maxfields,&fec_type,totdofs,&idxs,maxfields,&Ufield));
969566063dSJacob Faibussowitsch   PetscCall(PetscNew(&ctx));
979566063dSJacob Faibussowitsch   PetscCall(PetscCalloc1(maxfields,&ctx->scctx));
989566063dSJacob Faibussowitsch   PetscCall(DMGetDS(dm,&ds));
998135c375SStefano Zampini   if (ds) {
1008135c375SStefano Zampini     for (f=0;f<nfields;f++) {
1018135c375SStefano Zampini       const char* fname;
1028135c375SStefano Zampini       char        name[256];
1038135c375SStefano Zampini       PetscObject disc;
1048135c375SStefano Zampini       size_t      len;
1058135c375SStefano Zampini 
1069566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetFieldName(s,f,&fname));
1079566063dSJacob Faibussowitsch       PetscCall(PetscStrlen(fname,&len));
1088135c375SStefano Zampini       if (len) {
1099566063dSJacob Faibussowitsch         PetscCall(PetscStrcpy(name,fname));
1108135c375SStefano Zampini       } else {
1119566063dSJacob Faibussowitsch         PetscCall(PetscSNPrintf(name,256,"Field%D",f));
1128135c375SStefano Zampini       }
1139566063dSJacob Faibussowitsch       PetscCall(PetscDSGetDiscretization(ds,f,&disc));
1148135c375SStefano Zampini       if (disc) {
1158135c375SStefano Zampini         PetscClassId id;
1168135c375SStefano Zampini         PetscInt     Nc;
1178135c375SStefano Zampini         char         fec[64];
1188135c375SStefano Zampini 
1199566063dSJacob Faibussowitsch         PetscCall(PetscObjectGetClassId(disc, &id));
1208135c375SStefano Zampini         if (id == PETSCFE_CLASSID) {
1218135c375SStefano Zampini           PetscFE            fem = (PetscFE)disc;
1228135c375SStefano Zampini           PetscDualSpace     sp;
1238135c375SStefano Zampini           PetscDualSpaceType spname;
1248135c375SStefano Zampini           PetscInt           order;
1258135c375SStefano Zampini           PetscBool          islag,continuous,H1 = PETSC_TRUE;
1268135c375SStefano Zampini 
1279566063dSJacob Faibussowitsch           PetscCall(PetscFEGetNumComponents(fem,&Nc));
1289566063dSJacob Faibussowitsch           PetscCall(PetscFEGetDualSpace(fem,&sp));
1299566063dSJacob Faibussowitsch           PetscCall(PetscDualSpaceGetType(sp,&spname));
1309566063dSJacob Faibussowitsch           PetscCall(PetscStrcmp(spname,PETSCDUALSPACELAGRANGE,&islag));
13128b400f6SJacob Faibussowitsch           PetscCheck(islag,PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Unsupported dual space");
1329566063dSJacob Faibussowitsch           PetscCall(PetscDualSpaceLagrangeGetContinuity(sp,&continuous));
1339566063dSJacob Faibussowitsch           PetscCall(PetscDualSpaceGetOrder(sp,&order));
13428d58a37SPierre Jolivet           if (continuous && order > 0) { /* no support for high-order viz, still have to figure out the numbering */
1359566063dSJacob Faibussowitsch             PetscCall(PetscSNPrintf(fec,64,"FiniteElementCollection: H1_%DD_P1",dim));
1368135c375SStefano Zampini           } else {
137*08401ef6SPierre Jolivet             PetscCheck(continuous || !order,PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Discontinuous space visualization currently unsupported for order %D",order);
1388135c375SStefano Zampini             H1   = PETSC_FALSE;
1399566063dSJacob Faibussowitsch             PetscCall(PetscSNPrintf(fec,64,"FiniteElementCollection: L2_%DD_P%D",dim,order));
1408135c375SStefano Zampini           }
1419566063dSJacob Faibussowitsch           PetscCall(PetscStrallocpy(name,&fieldname[ctx->nf]));
1428135c375SStefano Zampini           bs[ctx->nf]   = Nc;
143bb77a09fSStefano Zampini           dims[ctx->nf] = dim;
1448135c375SStefano Zampini           if (H1) {
1458135c375SStefano Zampini             nlocal[ctx->nf] = Nc * Nv;
1469566063dSJacob Faibussowitsch             PetscCall(PetscStrallocpy(fec,&fec_type[ctx->nf]));
1479566063dSJacob Faibussowitsch             PetscCall(VecCreateSeq(PETSC_COMM_SELF,Nv*Nc,&xfield));
1488135c375SStefano Zampini             for (i=0,cum=0;i<vEnd-vStart;i++) {
1498135c375SStefano Zampini               PetscInt j,off;
1508135c375SStefano Zampini 
1518135c375SStefano Zampini               if (PetscUnlikely(!PetscBTLookup(vown,i))) continue;
1529566063dSJacob Faibussowitsch               PetscCall(PetscSectionGetFieldOffset(s,i+vStart,f,&off));
1538135c375SStefano Zampini               for (j=0;j<Nc;j++) idxs[cum++] = off + j;
1548135c375SStefano Zampini             }
1559566063dSJacob Faibussowitsch             PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)xlocal),Nv*Nc,idxs,PETSC_USE_POINTER,&isfield));
1568135c375SStefano Zampini           } else {
1578135c375SStefano Zampini             nlocal[ctx->nf] = Nc * totc;
1589566063dSJacob Faibussowitsch             PetscCall(PetscStrallocpy(fec,&fec_type[ctx->nf]));
1599566063dSJacob Faibussowitsch             PetscCall(VecCreateSeq(PETSC_COMM_SELF,Nc*totc,&xfield));
1608135c375SStefano Zampini             for (i=0,cum=0;i<cEnd-cStart;i++) {
1618135c375SStefano Zampini               PetscInt j,off;
1628135c375SStefano Zampini 
1638135c375SStefano Zampini               if (PetscUnlikely(gNum[i] < 0)) continue;
1649566063dSJacob Faibussowitsch               PetscCall(PetscSectionGetFieldOffset(s,i+cStart,f,&off));
1658135c375SStefano Zampini               for (j=0;j<Nc;j++) idxs[cum++] = off + j;
1668135c375SStefano Zampini             }
1679566063dSJacob Faibussowitsch             PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)xlocal),totc*Nc,idxs,PETSC_USE_POINTER,&isfield));
1688135c375SStefano Zampini           }
1699566063dSJacob Faibussowitsch           PetscCall(VecScatterCreate(xlocal,isfield,xfield,NULL,&ctx->scctx[ctx->nf]));
1709566063dSJacob Faibussowitsch           PetscCall(VecDestroy(&xfield));
1719566063dSJacob Faibussowitsch           PetscCall(ISDestroy(&isfield));
1728135c375SStefano Zampini           ctx->nf++;
1738135c375SStefano Zampini         } else if (id == PETSCFV_CLASSID) {
1748135c375SStefano Zampini           PetscInt c;
1758135c375SStefano Zampini 
1769566063dSJacob Faibussowitsch           PetscCall(PetscFVGetNumComponents((PetscFV)disc,&Nc));
1779566063dSJacob Faibussowitsch           PetscCall(PetscSNPrintf(fec,64,"FiniteElementCollection: L2_%DD_P0",dim));
1788135c375SStefano Zampini           for (c = 0; c < Nc; c++) {
1798135c375SStefano Zampini             char comp[256];
1809566063dSJacob Faibussowitsch             PetscCall(PetscSNPrintf(comp,256,"%s-Comp%D",name,c));
1819566063dSJacob Faibussowitsch             PetscCall(PetscStrallocpy(comp,&fieldname[ctx->nf]));
1828135c375SStefano Zampini             bs[ctx->nf] = 1; /* Does PetscFV support components with different block size? */
1838135c375SStefano Zampini             nlocal[ctx->nf] = totc;
184bb77a09fSStefano Zampini             dims[ctx->nf] = dim;
1859566063dSJacob Faibussowitsch             PetscCall(PetscStrallocpy(fec,&fec_type[ctx->nf]));
1869566063dSJacob Faibussowitsch             PetscCall(VecCreateSeq(PETSC_COMM_SELF,totc,&xfield));
1878135c375SStefano Zampini             for (i=0,cum=0;i<cEnd-cStart;i++) {
1888135c375SStefano Zampini               PetscInt off;
1898135c375SStefano Zampini 
1908135c375SStefano Zampini               if (PetscUnlikely(gNum[i])<0) continue;
1919566063dSJacob Faibussowitsch               PetscCall(PetscSectionGetFieldOffset(s,i+cStart,f,&off));
1928135c375SStefano Zampini               idxs[cum++] = off + c;
1938135c375SStefano Zampini             }
1949566063dSJacob Faibussowitsch             PetscCall(ISCreateGeneral(PetscObjectComm((PetscObject)xlocal),totc,idxs,PETSC_USE_POINTER,&isfield));
1959566063dSJacob Faibussowitsch             PetscCall(VecScatterCreate(xlocal,isfield,xfield,NULL,&ctx->scctx[ctx->nf]));
1969566063dSJacob Faibussowitsch             PetscCall(VecDestroy(&xfield));
1979566063dSJacob Faibussowitsch             PetscCall(ISDestroy(&isfield));
1988135c375SStefano Zampini             ctx->nf++;
1998135c375SStefano Zampini           }
20098921bdaSJacob Faibussowitsch         } else SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONG,"Unknown discretization type for field %D",f);
20198921bdaSJacob Faibussowitsch       } else SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Missing discretization for field %D",f);
2028135c375SStefano Zampini     }
2038135c375SStefano Zampini   } else SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Needs a DS attached to the DM");
2049566063dSJacob Faibussowitsch   PetscCall(PetscBTDestroy(&vown));
2059566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&xlocal));
2069566063dSJacob Faibussowitsch   PetscCall(ISRestoreIndices(globalNum,&gNum));
2078135c375SStefano Zampini 
2084cac2994SStefano Zampini   /* create work vectors */
2094cac2994SStefano Zampini   for (f=0;f<ctx->nf;f++) {
2109566063dSJacob Faibussowitsch     PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)dm),nlocal[f],PETSC_DECIDE,&Ufield[f]));
2119566063dSJacob Faibussowitsch     PetscCall(PetscObjectSetName((PetscObject)Ufield[f],fieldname[f]));
2129566063dSJacob Faibussowitsch     PetscCall(VecSetBlockSize(Ufield[f],bs[f]));
2139566063dSJacob Faibussowitsch     PetscCall(VecSetDM(Ufield[f],dm));
2144cac2994SStefano Zampini   }
2154cac2994SStefano Zampini 
2168135c375SStefano Zampini   /* customize the viewer */
2179566063dSJacob Faibussowitsch   PetscCall(PetscViewerGLVisSetFields(viewer,ctx->nf,(const char**)fec_type,dims,DMPlexSampleGLVisFields_Private,(PetscObject*)Ufield,ctx,DestroyGLVisViewerCtx_Private));
2188135c375SStefano Zampini   for (f=0;f<ctx->nf;f++) {
2199566063dSJacob Faibussowitsch     PetscCall(PetscFree(fieldname[f]));
2209566063dSJacob Faibussowitsch     PetscCall(PetscFree(fec_type[f]));
2219566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&Ufield[f]));
2228135c375SStefano Zampini   }
2239566063dSJacob Faibussowitsch   PetscCall(PetscFree7(fieldname,nlocal,bs,dims,fec_type,idxs,Ufield));
2248135c375SStefano Zampini   PetscFunctionReturn(0);
2258135c375SStefano Zampini }
2268135c375SStefano Zampini 
227b135d7daSStefano Zampini typedef enum {MFEM_POINT=0,MFEM_SEGMENT,MFEM_TRIANGLE,MFEM_SQUARE,MFEM_TETRAHEDRON,MFEM_CUBE,MFEM_PRISM,MFEM_UNDEF} MFEM_cid;
2288135c375SStefano Zampini 
2298135c375SStefano Zampini MFEM_cid mfem_table_cid[4][7]       = { {MFEM_POINT,MFEM_UNDEF,MFEM_UNDEF  ,MFEM_UNDEF   ,MFEM_UNDEF      ,MFEM_UNDEF,MFEM_UNDEF},
2308135c375SStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_UNDEF   ,MFEM_UNDEF      ,MFEM_UNDEF,MFEM_UNDEF},
2318135c375SStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_TRIANGLE,MFEM_SQUARE     ,MFEM_UNDEF,MFEM_UNDEF},
232b135d7daSStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_UNDEF   ,MFEM_TETRAHEDRON,MFEM_PRISM,MFEM_CUBE } };
2338135c375SStefano Zampini 
234b135d7daSStefano Zampini MFEM_cid mfem_table_cid_unint[4][9] = { {MFEM_POINT,MFEM_UNDEF,MFEM_UNDEF  ,MFEM_UNDEF   ,MFEM_UNDEF      ,MFEM_UNDEF,MFEM_PRISM,MFEM_UNDEF,MFEM_UNDEF},
235b135d7daSStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_UNDEF   ,MFEM_UNDEF      ,MFEM_UNDEF,MFEM_PRISM,MFEM_UNDEF,MFEM_UNDEF},
236b135d7daSStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_TRIANGLE,MFEM_SQUARE     ,MFEM_UNDEF,MFEM_PRISM,MFEM_UNDEF,MFEM_UNDEF},
237b135d7daSStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_UNDEF   ,MFEM_TETRAHEDRON,MFEM_UNDEF,MFEM_PRISM,MFEM_UNDEF,MFEM_CUBE } };
238044a5661SStefano Zampini 
239f86f7544SStefano Zampini static PetscErrorCode DMPlexGetPointMFEMCellID_Internal(DM dm, DMLabel label, PetscInt minl, PetscInt p, PetscInt *mid, PetscInt *cid)
2408135c375SStefano Zampini {
2418135c375SStefano Zampini   DMLabel        dlabel;
242044a5661SStefano Zampini   PetscInt       depth,csize,pdepth,dim;
2438135c375SStefano Zampini 
2448135c375SStefano Zampini   PetscFunctionBegin;
2459566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthLabel(dm,&dlabel));
2469566063dSJacob Faibussowitsch   PetscCall(DMLabelGetValue(dlabel,p,&pdepth));
2479566063dSJacob Faibussowitsch   PetscCall(DMPlexGetConeSize(dm,p,&csize));
2489566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepth(dm,&depth));
2499566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm,&dim));
2508135c375SStefano Zampini   if (label) {
2519566063dSJacob Faibussowitsch     PetscCall(DMLabelGetValue(label,p,mid));
252f86f7544SStefano Zampini     *mid = *mid - minl + 1; /* MFEM does not like negative markers */
25377eacf09SStefano Zampini   } else *mid = 1;
254044a5661SStefano Zampini   if (depth >=0 && dim != depth) { /* not interpolated, it assumes cell-vertex mesh */
2552c71b3e2SJacob Faibussowitsch     PetscCheckFalse(dim < 0 || dim > 3,PETSC_COMM_SELF,PETSC_ERR_SUP,"Dimension %D",dim);
256*08401ef6SPierre Jolivet     PetscCheck(csize <= 8,PETSC_COMM_SELF,PETSC_ERR_SUP,"Found cone size %D for point %D",csize,p);
257*08401ef6SPierre Jolivet     PetscCheck(depth == 1,PETSC_COMM_SELF,PETSC_ERR_SUP,"Found depth %D for point %D. You should interpolate the mesh first",depth,p);
258044a5661SStefano Zampini     *cid = mfem_table_cid_unint[dim][csize];
259044a5661SStefano Zampini   } else {
260*08401ef6SPierre Jolivet     PetscCheck(csize <= 6,PETSC_COMM_SELF,PETSC_ERR_SUP,"Cone size %D for point %D",csize,p);
2612c71b3e2SJacob Faibussowitsch     PetscCheckFalse(pdepth < 0 || pdepth > 3,PETSC_COMM_SELF,PETSC_ERR_SUP,"Depth %D for point %D",csize,p);
262044a5661SStefano Zampini     *cid = mfem_table_cid[pdepth][csize];
263044a5661SStefano Zampini   }
2648135c375SStefano Zampini   PetscFunctionReturn(0);
2658135c375SStefano Zampini }
2668135c375SStefano Zampini 
26796ca5757SLisandro Dalcin static PetscErrorCode DMPlexGetPointMFEMVertexIDs_Internal(DM dm, PetscInt p, PetscSection csec, PetscInt *nv, PetscInt vids[])
2688135c375SStefano Zampini {
269cc0d3ed7SStefano Zampini   PetscInt       dim,sdim,dof = 0,off = 0,i,q,vStart,vEnd,numPoints,*points = NULL;
2708135c375SStefano Zampini 
2718135c375SStefano Zampini   PetscFunctionBegin;
2729566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm,0,&vStart,&vEnd));
2739566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm,&dim));
274cc0d3ed7SStefano Zampini   sdim = dim;
275cc0d3ed7SStefano Zampini   if (csec) {
27684f354e3SLisandro Dalcin     PetscInt sStart,sEnd;
27784f354e3SLisandro Dalcin 
2789566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDim(dm,&sdim));
2799566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetChart(csec,&sStart,&sEnd));
2809566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetOffset(csec,vStart,&off));
281cc0d3ed7SStefano Zampini     off  = off/sdim;
28284f354e3SLisandro Dalcin     if (p >= sStart && p < sEnd) {
2839566063dSJacob Faibussowitsch       PetscCall(PetscSectionGetDof(csec,p,&dof));
284cc0d3ed7SStefano Zampini     }
28584f354e3SLisandro Dalcin   }
286cc0d3ed7SStefano Zampini   if (!dof) {
2879566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTransitiveClosure(dm,p,PETSC_TRUE,&numPoints,&points));
2888135c375SStefano Zampini     for (i=0,q=0;i<numPoints*2;i+= 2)
2898135c375SStefano Zampini       if ((points[i] >= vStart) && (points[i] < vEnd))
29096ca5757SLisandro Dalcin         vids[q++] = points[i]-vStart+off;
2919566063dSJacob Faibussowitsch     PetscCall(DMPlexRestoreTransitiveClosure(dm,p,PETSC_TRUE,&numPoints,&points));
292cc0d3ed7SStefano Zampini   } else {
2939566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetOffset(csec,p,&off));
2949566063dSJacob Faibussowitsch     PetscCall(PetscSectionGetDof(csec,p,&dof));
29596ca5757SLisandro Dalcin     for (q=0;q<dof/sdim;q++) vids[q] = off/sdim + q;
296cc0d3ed7SStefano Zampini   }
2978135c375SStefano Zampini   *nv = q;
2988135c375SStefano Zampini   PetscFunctionReturn(0);
2998135c375SStefano Zampini }
3008135c375SStefano Zampini 
3010c2bc6bfSStefano Zampini static PetscErrorCode GLVisCreateFE(PetscFE femIn,char name[32],PetscFE *fem,IS *perm)
302066ea43fSLisandro Dalcin {
303066ea43fSLisandro Dalcin   DM              K;
304066ea43fSLisandro Dalcin   PetscSpace      P;
305066ea43fSLisandro Dalcin   PetscDualSpace  Q;
306066ea43fSLisandro Dalcin   PetscQuadrature q,fq;
307066ea43fSLisandro Dalcin   PetscInt        dim,deg,dof;
308066ea43fSLisandro Dalcin   DMPolytopeType  ptype;
309066ea43fSLisandro Dalcin   PetscBool       isSimplex,isTensor;
310066ea43fSLisandro Dalcin   PetscBool       continuity = PETSC_FALSE;
311066ea43fSLisandro Dalcin   PetscDTNodeType nodeType   = PETSCDTNODES_GAUSSJACOBI;
312066ea43fSLisandro Dalcin   PetscBool       endpoint   = PETSC_TRUE;
313066ea43fSLisandro Dalcin   MPI_Comm        comm;
314066ea43fSLisandro Dalcin 
315066ea43fSLisandro Dalcin   PetscFunctionBegin;
3169566063dSJacob Faibussowitsch   PetscCall(PetscObjectGetComm((PetscObject)femIn, &comm));
3179566063dSJacob Faibussowitsch   PetscCall(PetscFEGetBasisSpace(femIn,&P));
3189566063dSJacob Faibussowitsch   PetscCall(PetscFEGetDualSpace(femIn,&Q));
3199566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceGetDM(Q,&K));
3209566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(K,&dim));
3219566063dSJacob Faibussowitsch   PetscCall(PetscSpaceGetDegree(P,&deg,NULL));
3229566063dSJacob Faibussowitsch   PetscCall(PetscSpaceGetNumComponents(P,&dof));
3239566063dSJacob Faibussowitsch   PetscCall(DMPlexGetCellType(K,0,&ptype));
324066ea43fSLisandro Dalcin   switch (ptype) {
325066ea43fSLisandro Dalcin   case DM_POLYTOPE_QUADRILATERAL:
326066ea43fSLisandro Dalcin   case DM_POLYTOPE_HEXAHEDRON:
327066ea43fSLisandro Dalcin     isSimplex = PETSC_FALSE; break;
328066ea43fSLisandro Dalcin   default:
329066ea43fSLisandro Dalcin     isSimplex = PETSC_TRUE; break;
330066ea43fSLisandro Dalcin   }
331066ea43fSLisandro Dalcin   isTensor = isSimplex ? PETSC_FALSE : PETSC_TRUE;
3320c2bc6bfSStefano Zampini   if (isSimplex) deg = PetscMin(deg,3); /* Permutation not coded for degree higher than 3 */
333066ea43fSLisandro Dalcin   /* Create space */
3349566063dSJacob Faibussowitsch   PetscCall(PetscSpaceCreate(comm,&P));
3359566063dSJacob Faibussowitsch   PetscCall(PetscSpaceSetType(P,PETSCSPACEPOLYNOMIAL));
3369566063dSJacob Faibussowitsch   PetscCall(PetscSpacePolynomialSetTensor(P,isTensor));
3379566063dSJacob Faibussowitsch   PetscCall(PetscSpaceSetNumComponents(P,dof));
3389566063dSJacob Faibussowitsch   PetscCall(PetscSpaceSetNumVariables(P,dim));
3399566063dSJacob Faibussowitsch   PetscCall(PetscSpaceSetDegree(P,deg,PETSC_DETERMINE));
3409566063dSJacob Faibussowitsch   PetscCall(PetscSpaceSetUp(P));
341066ea43fSLisandro Dalcin   /* Create dual space */
3429566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceCreate(comm,&Q));
3439566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceSetType(Q,PETSCDUALSPACELAGRANGE));
3449566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceLagrangeSetTensor(Q,isTensor));
3459566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceLagrangeSetContinuity(Q,continuity));
3469566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceLagrangeSetNodeType(Q,nodeType,endpoint,0));
3479566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceSetNumComponents(Q,dof));
3489566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceSetOrder(Q,deg));
3499566063dSJacob Faibussowitsch   PetscCall(DMPlexCreateReferenceCell(PETSC_COMM_SELF, DMPolytopeTypeSimpleShape(dim, isSimplex), &K));
3509566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceSetDM(Q,K));
3519566063dSJacob Faibussowitsch   PetscCall(DMDestroy(&K));
3529566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceSetUp(Q));
353066ea43fSLisandro Dalcin   /* Create quadrature */
354066ea43fSLisandro Dalcin   if (isSimplex) {
3559566063dSJacob Faibussowitsch     PetscCall(PetscDTStroudConicalQuadrature(dim,  1,deg+1,-1,+1,&q));
3569566063dSJacob Faibussowitsch     PetscCall(PetscDTStroudConicalQuadrature(dim-1,1,deg+1,-1,+1,&fq));
357066ea43fSLisandro Dalcin   } else {
3589566063dSJacob Faibussowitsch     PetscCall(PetscDTGaussTensorQuadrature(dim,  1,deg+1,-1,+1,&q));
3599566063dSJacob Faibussowitsch     PetscCall(PetscDTGaussTensorQuadrature(dim-1,1,deg+1,-1,+1,&fq));
360066ea43fSLisandro Dalcin   }
361066ea43fSLisandro Dalcin   /* Create finite element */
3629566063dSJacob Faibussowitsch   PetscCall(PetscFECreate(comm,fem));
3639566063dSJacob Faibussowitsch   PetscCall(PetscSNPrintf(name,32,"L2_T1_%DD_P%D",dim,deg));
3649566063dSJacob Faibussowitsch   PetscCall(PetscObjectSetName((PetscObject)*fem,name));
3659566063dSJacob Faibussowitsch   PetscCall(PetscFESetType(*fem,PETSCFEBASIC));
3669566063dSJacob Faibussowitsch   PetscCall(PetscFESetNumComponents(*fem,dof));
3679566063dSJacob Faibussowitsch   PetscCall(PetscFESetBasisSpace(*fem,P));
3689566063dSJacob Faibussowitsch   PetscCall(PetscFESetDualSpace(*fem,Q));
3699566063dSJacob Faibussowitsch   PetscCall(PetscFESetQuadrature(*fem,q));
3709566063dSJacob Faibussowitsch   PetscCall(PetscFESetFaceQuadrature(*fem,fq));
3719566063dSJacob Faibussowitsch   PetscCall(PetscFESetUp(*fem));
3720c2bc6bfSStefano Zampini 
3730c2bc6bfSStefano Zampini   /* Both MFEM and PETSc are lexicographic, but PLEX stores the swapped cone */
3740c2bc6bfSStefano Zampini   *perm = NULL;
3750c2bc6bfSStefano Zampini   if (isSimplex && dim == 3) {
3760c2bc6bfSStefano Zampini     PetscInt celldofs,*pidx;
3770c2bc6bfSStefano Zampini 
3780c2bc6bfSStefano Zampini     PetscCall(PetscDualSpaceGetDimension(Q,&celldofs));
3790c2bc6bfSStefano Zampini     celldofs /= dof;
3800c2bc6bfSStefano Zampini     PetscCall(PetscMalloc1(celldofs,&pidx));
3810c2bc6bfSStefano Zampini     switch (celldofs) {
3820c2bc6bfSStefano Zampini     case 4:
3830c2bc6bfSStefano Zampini       pidx[0] = 2;
3840c2bc6bfSStefano Zampini       pidx[1] = 0;
3850c2bc6bfSStefano Zampini       pidx[2] = 1;
3860c2bc6bfSStefano Zampini       pidx[3] = 3;
3870c2bc6bfSStefano Zampini     break;
3880c2bc6bfSStefano Zampini     case 10:
3890c2bc6bfSStefano Zampini       pidx[0] = 5;
3900c2bc6bfSStefano Zampini       pidx[1] = 3;
3910c2bc6bfSStefano Zampini       pidx[2] = 0;
3920c2bc6bfSStefano Zampini       pidx[3] = 4;
3930c2bc6bfSStefano Zampini       pidx[4] = 1;
3940c2bc6bfSStefano Zampini       pidx[5] = 2;
3950c2bc6bfSStefano Zampini       pidx[6] = 8;
3960c2bc6bfSStefano Zampini       pidx[7] = 6;
3970c2bc6bfSStefano Zampini       pidx[8] = 7;
3980c2bc6bfSStefano Zampini       pidx[9] = 9;
3990c2bc6bfSStefano Zampini     break;
4000c2bc6bfSStefano Zampini     case 20:
4010c2bc6bfSStefano Zampini       pidx[ 0] = 9;
4020c2bc6bfSStefano Zampini       pidx[ 1] = 7;
4030c2bc6bfSStefano Zampini       pidx[ 2] = 4;
4040c2bc6bfSStefano Zampini       pidx[ 3] = 0;
4050c2bc6bfSStefano Zampini       pidx[ 4] = 8;
4060c2bc6bfSStefano Zampini       pidx[ 5] = 5;
4070c2bc6bfSStefano Zampini       pidx[ 6] = 1;
4080c2bc6bfSStefano Zampini       pidx[ 7] = 6;
4090c2bc6bfSStefano Zampini       pidx[ 8] = 2;
4100c2bc6bfSStefano Zampini       pidx[ 9] = 3;
4110c2bc6bfSStefano Zampini       pidx[10] = 15;
4120c2bc6bfSStefano Zampini       pidx[11] = 13;
4130c2bc6bfSStefano Zampini       pidx[12] = 10;
4140c2bc6bfSStefano Zampini       pidx[13] = 14;
4150c2bc6bfSStefano Zampini       pidx[14] = 11;
4160c2bc6bfSStefano Zampini       pidx[15] = 12;
4170c2bc6bfSStefano Zampini       pidx[16] = 18;
4180c2bc6bfSStefano Zampini       pidx[17] = 16;
4190c2bc6bfSStefano Zampini       pidx[18] = 17;
4200c2bc6bfSStefano Zampini       pidx[19] = 19;
4210c2bc6bfSStefano Zampini     break;
4220c2bc6bfSStefano Zampini     default:
4230c2bc6bfSStefano Zampini       SETERRQ(comm,PETSC_ERR_SUP,"Unhandled degree,dof pair %D,%D",deg,celldofs);
4240c2bc6bfSStefano Zampini     break;
4250c2bc6bfSStefano Zampini     }
4260c2bc6bfSStefano Zampini     PetscCall(ISCreateBlock(PETSC_COMM_SELF,dof,celldofs,pidx,PETSC_OWN_POINTER,perm));
4270c2bc6bfSStefano Zampini   }
4280c2bc6bfSStefano Zampini 
429066ea43fSLisandro Dalcin   /* Cleanup */
4309566063dSJacob Faibussowitsch   PetscCall(PetscSpaceDestroy(&P));
4319566063dSJacob Faibussowitsch   PetscCall(PetscDualSpaceDestroy(&Q));
4329566063dSJacob Faibussowitsch   PetscCall(PetscQuadratureDestroy(&q));
4339566063dSJacob Faibussowitsch   PetscCall(PetscQuadratureDestroy(&fq));
434066ea43fSLisandro Dalcin   PetscFunctionReturn(0);
435066ea43fSLisandro Dalcin }
436066ea43fSLisandro Dalcin 
43777eacf09SStefano Zampini /*
43877eacf09SStefano Zampini    ASCII visualization/dump: full support for simplices and tensor product cells. It supports AMR
43977eacf09SStefano Zampini    Higher order meshes are also supported
44077eacf09SStefano Zampini */
4418135c375SStefano Zampini static PetscErrorCode DMPlexView_GLVis_ASCII(DM dm, PetscViewer viewer)
4428135c375SStefano Zampini {
4438135c375SStefano Zampini   DMLabel              label;
4440c2bc6bfSStefano Zampini   PetscSection         coordSection,parentSection,hoSection = NULL;
44577eacf09SStefano Zampini   Vec                  coordinates,hovec;
4468135c375SStefano Zampini   const PetscScalar    *array;
447f86f7544SStefano Zampini   PetscInt             bf,p,sdim,dim,depth,novl,minl;
448412e9a14SMatthew G. Knepley   PetscInt             cStart,cEnd,vStart,vEnd,nvert;
4493924b612SStefano Zampini   PetscMPIInt          size;
4503e6c54aaSStefano Zampini   PetscBool            localized,isascii;
45128d58a37SPierre Jolivet   PetscBool            enable_mfem,enable_boundary,enable_ncmesh,view_ovl = PETSC_FALSE;
4523e6c54aaSStefano Zampini   PetscBT              pown,vown;
4538135c375SStefano Zampini   PetscErrorCode       ierr;
4548135c375SStefano Zampini   PetscContainer       glvis_container;
455044a5661SStefano Zampini   PetscBool            cellvertex = PETSC_FALSE, periodic, enabled = PETSC_TRUE;
456f86f7544SStefano Zampini   PetscBool            enable_emark,enable_bmark;
45777eacf09SStefano Zampini   const char           *fmt;
4587bf4dd16SStefano Zampini   char                 emark[64] = "",bmark[64] = "";
4598135c375SStefano Zampini 
4608135c375SStefano Zampini   PetscFunctionBegin;
4618135c375SStefano Zampini   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
4628135c375SStefano Zampini   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
4639566063dSJacob Faibussowitsch   PetscCall(PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii));
46428b400f6SJacob Faibussowitsch   PetscCheck(isascii,PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Viewer must be of type VIEWERASCII");
4659566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)viewer),&size));
466*08401ef6SPierre Jolivet   PetscCheck(size <= 1,PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Use single sequential viewers for parallel visualization");
4679566063dSJacob Faibussowitsch   PetscCall(DMGetDimension(dm,&dim));
4680c2bc6bfSStefano Zampini   PetscCall(DMPlexGetDepth(dm,&depth));
4698135c375SStefano Zampini 
4708135c375SStefano Zampini   /* get container: determines if a process visualizes is portion of the data or not */
4719566063dSJacob Faibussowitsch   PetscCall(PetscObjectQuery((PetscObject)viewer,"_glvis_info_container",(PetscObject*)&glvis_container));
47228b400f6SJacob Faibussowitsch   PetscCheck(glvis_container,PetscObjectComm((PetscObject)dm),PETSC_ERR_PLIB,"Missing GLVis container");
4738135c375SStefano Zampini   {
4748135c375SStefano Zampini     PetscViewerGLVisInfo glvis_info;
4759566063dSJacob Faibussowitsch     PetscCall(PetscContainerGetPointer(glvis_container,(void**)&glvis_info));
4768135c375SStefano Zampini     enabled = glvis_info->enabled;
47777eacf09SStefano Zampini     fmt     = glvis_info->fmt;
4788135c375SStefano Zampini   }
47921414b21SStefano Zampini 
4800c2bc6bfSStefano Zampini   /* Users can attach a coordinate vector to the DM in case they have a higher-order mesh */
4819566063dSJacob Faibussowitsch   PetscCall(PetscObjectQuery((PetscObject)dm,"_glvis_mesh_coords",(PetscObject*)&hovec));
4829566063dSJacob Faibussowitsch   PetscCall(PetscObjectReference((PetscObject)hovec));
483066ea43fSLisandro Dalcin   if (!hovec) {
484066ea43fSLisandro Dalcin     DM           cdm;
485066ea43fSLisandro Dalcin     PetscFE      disc;
486066ea43fSLisandro Dalcin     PetscClassId classid;
487066ea43fSLisandro Dalcin 
4889566063dSJacob Faibussowitsch     PetscCall(DMGetCoordinateDM(dm,&cdm));
4899566063dSJacob Faibussowitsch     PetscCall(DMGetField(cdm,0,NULL,(PetscObject*)&disc));
4909566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetClassId((PetscObject)disc,&classid));
491066ea43fSLisandro Dalcin     if (classid == PETSCFE_CLASSID) {
492066ea43fSLisandro Dalcin       DM      hocdm;
493066ea43fSLisandro Dalcin       PetscFE hodisc;
494066ea43fSLisandro Dalcin       Vec     vec;
495066ea43fSLisandro Dalcin       Mat     mat;
496066ea43fSLisandro Dalcin       char    name[32],fec_type[64];
4970c2bc6bfSStefano Zampini       IS      perm = NULL;
498066ea43fSLisandro Dalcin 
4990c2bc6bfSStefano Zampini       PetscCall(GLVisCreateFE(disc,name,&hodisc,&perm));
5009566063dSJacob Faibussowitsch       PetscCall(DMClone(cdm,&hocdm));
5019566063dSJacob Faibussowitsch       PetscCall(DMSetField(hocdm,0,NULL,(PetscObject)hodisc));
5029566063dSJacob Faibussowitsch       PetscCall(PetscFEDestroy(&hodisc));
5039566063dSJacob Faibussowitsch       PetscCall(DMCreateDS(hocdm));
504066ea43fSLisandro Dalcin 
5059566063dSJacob Faibussowitsch       PetscCall(DMGetCoordinates(dm,&vec));
5069566063dSJacob Faibussowitsch       PetscCall(DMCreateGlobalVector(hocdm,&hovec));
5079566063dSJacob Faibussowitsch       PetscCall(DMCreateInterpolation(cdm,hocdm,&mat,NULL));
5089566063dSJacob Faibussowitsch       PetscCall(MatInterpolate(mat,vec,hovec));
5099566063dSJacob Faibussowitsch       PetscCall(MatDestroy(&mat));
5100c2bc6bfSStefano Zampini       PetscCall(DMGetLocalSection(hocdm,&hoSection));
5110c2bc6bfSStefano Zampini       PetscCall(PetscSectionSetClosurePermutation(hoSection, (PetscObject)hocdm, depth, perm));
5120c2bc6bfSStefano Zampini       PetscCall(ISDestroy(&perm));
5139566063dSJacob Faibussowitsch       PetscCall(DMDestroy(&hocdm));
5149566063dSJacob Faibussowitsch       PetscCall(PetscSNPrintf(fec_type,sizeof(fec_type),"FiniteElementCollection: %s", name));
5159566063dSJacob Faibussowitsch       PetscCall(PetscObjectSetName((PetscObject)hovec,fec_type));
516066ea43fSLisandro Dalcin     }
517066ea43fSLisandro Dalcin   }
51821414b21SStefano Zampini 
5199566063dSJacob Faibussowitsch   PetscCall(DMPlexGetHeightStratum(dm,0,&cStart,&cEnd));
5209566063dSJacob Faibussowitsch   PetscCall(DMPlexGetGhostCellStratum(dm,&p,NULL));
521c3c203b2SStefano Zampini   if (p >= 0) cEnd = p;
5229566063dSJacob Faibussowitsch   PetscCall(DMPlexGetDepthStratum(dm,0,&vStart,&vEnd));
5239566063dSJacob Faibussowitsch   PetscCall(DMGetPeriodicity(dm,&periodic,NULL,NULL,NULL));
5249566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocalized(dm,&localized));
5252c71b3e2SJacob Faibussowitsch   PetscCheckFalse(periodic && !localized && !hovec,PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Coordinates need to be localized");
5269566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateSection(dm,&coordSection));
5279566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinateDim(dm,&sdim));
5289566063dSJacob Faibussowitsch   PetscCall(DMGetCoordinatesLocal(dm,&coordinates));
529*08401ef6SPierre Jolivet   PetscCheck(coordinates || hovec,PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Missing local coordinates vector");
5308135c375SStefano Zampini 
5318135c375SStefano Zampini   /*
5328135c375SStefano Zampini      a couple of sections of the mesh specification are disabled
5333e96840aSStefano Zampini        - boundary: the boundary is not needed for proper mesh visualization unless we want to visualize boundary attributes or we have high-order coordinates in 3D (topologically)
53477eacf09SStefano Zampini        - vertex_parents: used for non-conforming meshes only when we want to use MFEM as a discretization package
5353e6c54aaSStefano Zampini                          and be able to derefine the mesh (MFEM does not currently have to ability to read ncmeshes in parallel)
5368135c375SStefano Zampini   */
5373e96840aSStefano Zampini   enable_boundary = PETSC_FALSE;
5388135c375SStefano Zampini   enable_ncmesh   = PETSC_FALSE;
5393e6c54aaSStefano Zampini   enable_mfem     = PETSC_FALSE;
540f86f7544SStefano Zampini   enable_emark    = PETSC_FALSE;
541f86f7544SStefano Zampini   enable_bmark    = PETSC_FALSE;
5427bf4dd16SStefano Zampini   /* I'm tired of problems with negative values in the markers, disable them */
5439566063dSJacob Faibussowitsch   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)dm),((PetscObject)dm)->prefix,"GLVis PetscViewer DMPlex Options","PetscViewer");PetscCall(ierr);
5449566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-viewer_glvis_dm_plex_enable_boundary","Enable boundary section in mesh representation",NULL,enable_boundary,&enable_boundary,NULL));
5459566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-viewer_glvis_dm_plex_enable_ncmesh","Enable vertex_parents section in mesh representation (allows derefinement)",NULL,enable_ncmesh,&enable_ncmesh,NULL));
5469566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-viewer_glvis_dm_plex_enable_mfem","Dump a mesh that can be used with MFEM's FiniteElementSpaces",NULL,enable_mfem,&enable_mfem,NULL));
5479566063dSJacob Faibussowitsch   PetscCall(PetscOptionsBool("-viewer_glvis_dm_plex_overlap","Include overlap region in local meshes",NULL,view_ovl,&view_ovl,NULL));
5489566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-viewer_glvis_dm_plex_emarker","String for the material id label",NULL,emark,emark,sizeof(emark),&enable_emark));
5499566063dSJacob Faibussowitsch   PetscCall(PetscOptionsString("-viewer_glvis_dm_plex_bmarker","String for the boundary id label",NULL,bmark,bmark,sizeof(bmark),&enable_bmark));
5509566063dSJacob Faibussowitsch   ierr = PetscOptionsEnd();PetscCall(ierr);
551f86f7544SStefano Zampini   if (enable_bmark) enable_boundary = PETSC_TRUE;
552f86f7544SStefano Zampini 
5539566063dSJacob Faibussowitsch   PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)dm),&size));
5542c71b3e2SJacob Faibussowitsch   PetscCheckFalse(enable_ncmesh && size > 1,PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Not supported in parallel");
5552c71b3e2SJacob Faibussowitsch   PetscCheckFalse(enable_boundary && depth >= 0 && dim != depth,PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Mesh must be interpolated. "
5567e1aca4eSStefano Zampini                                                              "Alternatively, run with -viewer_glvis_dm_plex_enable_boundary 0");
5572c71b3e2SJacob Faibussowitsch   PetscCheckFalse(enable_ncmesh && depth >= 0 && dim != depth,PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Mesh must be interpolated. "
5587e1aca4eSStefano Zampini                                                            "Alternatively, run with -viewer_glvis_dm_plex_enable_ncmesh 0");
559044a5661SStefano Zampini   if (depth >=0 && dim != depth) { /* not interpolated, it assumes cell-vertex mesh */
560*08401ef6SPierre Jolivet     PetscCheck(depth == 1,PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported depth %D. You should interpolate the mesh first",depth);
561044a5661SStefano Zampini     cellvertex = PETSC_TRUE;
562044a5661SStefano Zampini   }
5638135c375SStefano Zampini 
5648135c375SStefano Zampini   /* Identify possible cells in the overlap */
5658135c375SStefano Zampini   novl = 0;
5668135c375SStefano Zampini   pown = NULL;
5673924b612SStefano Zampini   if (size > 1) {
5683e6c54aaSStefano Zampini     IS             globalNum = NULL;
5693e6c54aaSStefano Zampini     const PetscInt *gNum;
5703e6c54aaSStefano Zampini     PetscBool      ovl  = PETSC_FALSE;
5713e6c54aaSStefano Zampini 
5729566063dSJacob Faibussowitsch     PetscCall(PetscObjectQuery((PetscObject)dm,"_glvis_plex_gnum",(PetscObject*)&globalNum));
573b135d7daSStefano Zampini     if (!globalNum) {
57428d58a37SPierre Jolivet       if (view_ovl) {
5759566063dSJacob Faibussowitsch         PetscCall(ISCreateStride(PetscObjectComm((PetscObject)dm),cEnd-cStart,0,1,&globalNum));
57628d58a37SPierre Jolivet       } else {
5779566063dSJacob Faibussowitsch         PetscCall(DMPlexCreateCellNumbering_Internal(dm,PETSC_TRUE,&globalNum));
57828d58a37SPierre Jolivet       }
5799566063dSJacob Faibussowitsch       PetscCall(PetscObjectCompose((PetscObject)dm,"_glvis_plex_gnum",(PetscObject)globalNum));
5809566063dSJacob Faibussowitsch       PetscCall(PetscObjectDereference((PetscObject)globalNum));
581b135d7daSStefano Zampini     }
5829566063dSJacob Faibussowitsch     PetscCall(ISGetIndices(globalNum,&gNum));
5838135c375SStefano Zampini     for (p=cStart; p<cEnd; p++) {
5848135c375SStefano Zampini       if (gNum[p-cStart] < 0) {
5858135c375SStefano Zampini         ovl = PETSC_TRUE;
5868135c375SStefano Zampini         novl++;
5878135c375SStefano Zampini       }
5888135c375SStefano Zampini     }
5898135c375SStefano Zampini     if (ovl) {
5908135c375SStefano Zampini       /* it may happen that pown get not destroyed, if the user closes the window while this function is running.
5918135c375SStefano Zampini          TODO: garbage collector? attach pown to dm?  */
5929566063dSJacob Faibussowitsch       PetscCall(PetscBTCreate(cEnd-cStart,&pown));
5933e6c54aaSStefano Zampini       for (p=cStart; p<cEnd; p++) {
5943e6c54aaSStefano Zampini         if (gNum[p-cStart] < 0) continue;
5953e6c54aaSStefano Zampini         else {
5969566063dSJacob Faibussowitsch           PetscCall(PetscBTSet(pown,p-cStart));
5978135c375SStefano Zampini         }
5988135c375SStefano Zampini       }
5993e6c54aaSStefano Zampini     }
6009566063dSJacob Faibussowitsch     PetscCall(ISRestoreIndices(globalNum,&gNum));
6013e6c54aaSStefano Zampini   }
6028135c375SStefano Zampini 
6031d4815f0SStefano Zampini   /* vertex_parents (Non-conforming meshes) */
6041d4815f0SStefano Zampini   parentSection  = NULL;
6051d4815f0SStefano Zampini   if (enable_ncmesh) {
6069566063dSJacob Faibussowitsch     PetscCall(DMPlexGetTree(dm,&parentSection,NULL,NULL,NULL,NULL));
6071d4815f0SStefano Zampini     enable_ncmesh = (PetscBool)(enable_ncmesh && parentSection);
6081d4815f0SStefano Zampini   }
6093e6c54aaSStefano Zampini   /* return if this process is disabled */
6108135c375SStefano Zampini   if (!enabled) {
6119566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"MFEM mesh %s\n",enable_ncmesh ? "v1.1" : "v1.0"));
6129566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"\ndimension\n"));
6139566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",dim));
6149566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"\nelements\n"));
6159566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",0));
6169566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"\nboundary\n"));
6179566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",0));
6189566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"\nvertices\n"));
6199566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",0));
6209566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",sdim));
6219566063dSJacob Faibussowitsch     PetscCall(PetscBTDestroy(&pown));
6229566063dSJacob Faibussowitsch     PetscCall(VecDestroy(&hovec));
6238135c375SStefano Zampini     PetscFunctionReturn(0);
6248135c375SStefano Zampini   }
6258135c375SStefano Zampini 
6263e6c54aaSStefano Zampini   if (enable_mfem) {
6273e6c54aaSStefano Zampini     if (periodic && !hovec) { /* we need to generate a vector of L2 coordinates, as this is how MFEM handles periodic meshes */
6283e6c54aaSStefano Zampini       PetscInt    vpc = 0;
6293e6c54aaSStefano Zampini       char        fec[64];
63096ca5757SLisandro Dalcin       PetscInt    vids[8] = {0,1,2,3,4,5,6,7};
63196ca5757SLisandro Dalcin       PetscInt    hexv[8] = {0,1,3,2,4,5,7,6}, tetv[4] = {0,1,2,3};
63296ca5757SLisandro Dalcin       PetscInt    quadv[8] = {0,1,3,2}, triv[3] = {0,1,2};
63396ca5757SLisandro Dalcin       PetscInt    *dof = NULL;
6343e6c54aaSStefano Zampini       PetscScalar *array,*ptr;
6353e6c54aaSStefano Zampini 
6369566063dSJacob Faibussowitsch       PetscCall(PetscSNPrintf(fec,sizeof(fec),"FiniteElementCollection: L2_T1_%DD_P1",dim));
6373e6c54aaSStefano Zampini       if (cEnd-cStart) {
6383e6c54aaSStefano Zampini         PetscInt fpc;
6393e6c54aaSStefano Zampini 
6409566063dSJacob Faibussowitsch         PetscCall(DMPlexGetConeSize(dm,cStart,&fpc));
6413e6c54aaSStefano Zampini         switch(dim) {
6423e6c54aaSStefano Zampini           case 1:
6433e6c54aaSStefano Zampini             vpc = 2;
6443e6c54aaSStefano Zampini             dof = hexv;
6453e6c54aaSStefano Zampini             break;
6463e6c54aaSStefano Zampini           case 2:
6473e6c54aaSStefano Zampini             switch (fpc) {
6483e6c54aaSStefano Zampini               case 3:
6493e6c54aaSStefano Zampini                 vpc = 3;
650044a5661SStefano Zampini                 dof = triv;
6513e6c54aaSStefano Zampini                 break;
6523e6c54aaSStefano Zampini               case 4:
6533e6c54aaSStefano Zampini                 vpc = 4;
654044a5661SStefano Zampini                 dof = quadv;
6553e6c54aaSStefano Zampini                 break;
6563e6c54aaSStefano Zampini               default:
65798921bdaSJacob Faibussowitsch                 SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case: faces per cell %D",fpc);
6583e6c54aaSStefano Zampini             }
6593e6c54aaSStefano Zampini             break;
6603e6c54aaSStefano Zampini           case 3:
6613e6c54aaSStefano Zampini             switch (fpc) {
662044a5661SStefano Zampini               case 4: /* TODO: still need to understand L2 ordering for tets */
6633e6c54aaSStefano Zampini                 vpc = 4;
664044a5661SStefano Zampini                 dof = tetv;
665044a5661SStefano Zampini                 SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled tethraedral case");
6663e6c54aaSStefano Zampini               case 6:
66728b400f6SJacob Faibussowitsch                 PetscCheck(!cellvertex,PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case: vertices per cell %D",fpc);
668044a5661SStefano Zampini                 vpc = 8;
669044a5661SStefano Zampini                 dof = hexv;
670044a5661SStefano Zampini                 break;
671044a5661SStefano Zampini               case 8:
67228b400f6SJacob Faibussowitsch                 PetscCheck(cellvertex,PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case: faces per cell %D",fpc);
6733e6c54aaSStefano Zampini                 vpc = 8;
6743e6c54aaSStefano Zampini                 dof = hexv;
6753e6c54aaSStefano Zampini                 break;
6763e6c54aaSStefano Zampini               default:
67798921bdaSJacob Faibussowitsch                 SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case: faces per cell %D",fpc);
6783e6c54aaSStefano Zampini             }
6793e6c54aaSStefano Zampini             break;
6803e6c54aaSStefano Zampini           default:
6813e6c54aaSStefano Zampini             SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Unhandled dim");
6823e6c54aaSStefano Zampini         }
6839566063dSJacob Faibussowitsch         PetscCall(DMPlexReorderCell(dm,cStart,vids));
6843e6c54aaSStefano Zampini       }
68528b400f6SJacob Faibussowitsch       PetscCheck(dof,PetscObjectComm((PetscObject)dm),PETSC_ERR_PLIB,"Missing dofs");
6869566063dSJacob Faibussowitsch       PetscCall(VecCreateSeq(PETSC_COMM_SELF,(cEnd-cStart-novl)*vpc*sdim,&hovec));
6879566063dSJacob Faibussowitsch       PetscCall(PetscObjectSetName((PetscObject)hovec,fec));
6889566063dSJacob Faibussowitsch       PetscCall(VecGetArray(hovec,&array));
6893e6c54aaSStefano Zampini       ptr  = array;
6903e6c54aaSStefano Zampini       for (p=cStart;p<cEnd;p++) {
6913e6c54aaSStefano Zampini         PetscInt    csize,v,d;
6923e6c54aaSStefano Zampini         PetscScalar *vals = NULL;
6933e6c54aaSStefano Zampini 
6943e6c54aaSStefano Zampini         if (PetscUnlikely(pown && !PetscBTLookup(pown,p-cStart))) continue;
6959566063dSJacob Faibussowitsch         PetscCall(DMPlexVecGetClosure(dm,coordSection,coordinates,p,&csize,&vals));
6962c71b3e2SJacob Faibussowitsch         PetscCheckFalse(csize != vpc*sdim && csize != vpc*sdim*2,PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported closure size %D (vpc %D, sdim %D)",csize,vpc,sdim);
6973e6c54aaSStefano Zampini         for (v=0;v<vpc;v++) {
6983e6c54aaSStefano Zampini           for (d=0;d<sdim;d++) {
6993e6c54aaSStefano Zampini             ptr[sdim*dof[v]+d] = vals[sdim*vids[v]+d];
7003e6c54aaSStefano Zampini           }
7013e6c54aaSStefano Zampini         }
7023e6c54aaSStefano Zampini         ptr += vpc*sdim;
7039566063dSJacob Faibussowitsch         PetscCall(DMPlexVecRestoreClosure(dm,coordSection,coordinates,p,&csize,&vals));
7043e6c54aaSStefano Zampini       }
7059566063dSJacob Faibussowitsch       PetscCall(VecRestoreArray(hovec,&array));
7063e6c54aaSStefano Zampini     }
7073e6c54aaSStefano Zampini   }
7083e96840aSStefano Zampini   /* if we have high-order coordinates in 3D, we need to specify the boundary */
7093e96840aSStefano Zampini   if (hovec && dim == 3) enable_boundary = PETSC_TRUE;
7103e6c54aaSStefano Zampini 
7118135c375SStefano Zampini   /* header */
7129566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"MFEM mesh %s\n",enable_ncmesh ? "v1.1" : "v1.0"));
7138135c375SStefano Zampini 
7148135c375SStefano Zampini   /* topological dimension */
7159566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"\ndimension\n"));
7169566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",dim));
7178135c375SStefano Zampini 
7188135c375SStefano Zampini   /* elements */
719f86f7544SStefano Zampini   minl = 1;
720f86f7544SStefano Zampini   label = NULL;
721f86f7544SStefano Zampini   if (enable_emark) {
722f86f7544SStefano Zampini     PetscInt lminl = PETSC_MAX_INT;
723f86f7544SStefano Zampini 
7249566063dSJacob Faibussowitsch     PetscCall(DMGetLabel(dm,emark,&label));
725f86f7544SStefano Zampini     if (label) {
726f86f7544SStefano Zampini       IS       vals;
727f86f7544SStefano Zampini       PetscInt ldef;
728f86f7544SStefano Zampini 
7299566063dSJacob Faibussowitsch       PetscCall(DMLabelGetDefaultValue(label,&ldef));
7309566063dSJacob Faibussowitsch       PetscCall(DMLabelGetValueIS(label,&vals));
7319566063dSJacob Faibussowitsch       PetscCall(ISGetMinMax(vals,&lminl,NULL));
7329566063dSJacob Faibussowitsch       PetscCall(ISDestroy(&vals));
733f86f7544SStefano Zampini       lminl = PetscMin(ldef,lminl);
734f86f7544SStefano Zampini     }
7351c2dc1cbSBarry Smith     PetscCall(MPIU_Allreduce(&lminl,&minl,1,MPIU_INT,MPI_MIN,PetscObjectComm((PetscObject)dm)));
736f86f7544SStefano Zampini     if (minl == PETSC_MAX_INT) minl = 1;
737f86f7544SStefano Zampini   }
7389566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"\nelements\n"));
7399566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",cEnd-cStart-novl));
7408135c375SStefano Zampini   for (p=cStart;p<cEnd;p++) {
74196ca5757SLisandro Dalcin     PetscInt       vids[8];
74211a4995dSStefano Zampini     PetscInt       i,nv = 0,cid = -1,mid = 1;
7438135c375SStefano Zampini 
7443e6c54aaSStefano Zampini     if (PetscUnlikely(pown && !PetscBTLookup(pown,p-cStart))) continue;
7459566063dSJacob Faibussowitsch     PetscCall(DMPlexGetPointMFEMCellID_Internal(dm,label,minl,p,&mid,&cid));
7469566063dSJacob Faibussowitsch     PetscCall(DMPlexGetPointMFEMVertexIDs_Internal(dm,p,(localized && !hovec) ? coordSection : NULL,&nv,vids));
7479566063dSJacob Faibussowitsch     PetscCall(DMPlexReorderCell(dm,p,vids));
7489566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D %D",mid,cid));
7498135c375SStefano Zampini     for (i=0;i<nv;i++) {
7509566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer," %D",vids[i]));
7518135c375SStefano Zampini     }
7529566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
7538135c375SStefano Zampini   }
7548135c375SStefano Zampini 
755cc0d3ed7SStefano Zampini   /* boundary */
7569566063dSJacob Faibussowitsch   PetscCall(PetscViewerASCIIPrintf(viewer,"\nboundary\n"));
757cc0d3ed7SStefano Zampini   if (!enable_boundary) {
7589566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",0));
759cc0d3ed7SStefano Zampini   } else {
76077eacf09SStefano Zampini     DMLabel  perLabel;
76177eacf09SStefano Zampini     PetscBT  bfaces;
762b135d7daSStefano Zampini     PetscInt fStart,fEnd,*fcells;
763cc0d3ed7SStefano Zampini 
7649566063dSJacob Faibussowitsch     PetscCall(DMPlexGetHeightStratum(dm,1,&fStart,&fEnd));
7659566063dSJacob Faibussowitsch     PetscCall(PetscBTCreate(fEnd-fStart,&bfaces));
7669566063dSJacob Faibussowitsch     PetscCall(DMPlexGetMaxSizes(dm,NULL,&p));
7679566063dSJacob Faibussowitsch     PetscCall(PetscMalloc1(p,&fcells));
7689566063dSJacob Faibussowitsch     PetscCall(DMGetLabel(dm,"glvis_periodic_cut",&perLabel));
7690c2bc6bfSStefano Zampini     if (!perLabel && periodic) { /* this periodic cut can be moved up to DMPlex setup */
7709566063dSJacob Faibussowitsch       PetscCall(DMCreateLabel(dm,"glvis_periodic_cut"));
7719566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(dm,"glvis_periodic_cut",&perLabel));
7729566063dSJacob Faibussowitsch       PetscCall(DMLabelSetDefaultValue(perLabel,1));
77377eacf09SStefano Zampini       for (p=cStart;p<cEnd;p++) {
774c3c203b2SStefano Zampini         DMPolytopeType cellType;
775c3c203b2SStefano Zampini         PetscInt       dof;
776b135d7daSStefano Zampini 
7779566063dSJacob Faibussowitsch         PetscCall(DMPlexGetCellType(dm,p,&cellType));
7789566063dSJacob Faibussowitsch         PetscCall(PetscSectionGetDof(coordSection,p,&dof));
77977eacf09SStefano Zampini         if (dof) {
780c3c203b2SStefano Zampini           PetscInt    uvpc, v,csize,cellClosureSize,*cellClosure = NULL,*vidxs = NULL;
78177eacf09SStefano Zampini           PetscScalar *vals = NULL;
782c3c203b2SStefano Zampini 
783c3c203b2SStefano Zampini           uvpc = DMPolytopeTypeGetNumVertices(cellType);
784*08401ef6SPierre Jolivet           PetscCheck(dof%sdim == 0,PETSC_COMM_SELF,PETSC_ERR_USER,"Incompatible number of cell dofs %D and space dimension %D",dof,sdim);
7859566063dSJacob Faibussowitsch           PetscCall(DMPlexVecGetClosure(dm,coordSection,coordinates,p,&csize,&vals));
7869566063dSJacob Faibussowitsch           PetscCall(DMPlexGetTransitiveClosure(dm,p,PETSC_TRUE,&cellClosureSize,&cellClosure));
78777eacf09SStefano Zampini           for (v=0;v<cellClosureSize;v++)
78877eacf09SStefano Zampini             if (cellClosure[2*v] >= vStart && cellClosure[2*v] < vEnd) {
78977eacf09SStefano Zampini               vidxs = cellClosure + 2*v;
79077eacf09SStefano Zampini               break;
79177eacf09SStefano Zampini             }
79228b400f6SJacob Faibussowitsch           PetscCheck(vidxs,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Missing vertices");
793b135d7daSStefano Zampini           for (v=0;v<uvpc;v++) {
79477eacf09SStefano Zampini             PetscInt s;
795044a5661SStefano Zampini 
79677eacf09SStefano Zampini             for (s=0;s<sdim;s++) {
797b135d7daSStefano Zampini               if (PetscAbsScalar(vals[v*sdim+s]-vals[v*sdim+s+uvpc*sdim])>PETSC_MACHINE_EPSILON) {
7989566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(perLabel,vidxs[2*v],2));
79977eacf09SStefano Zampini               }
80077eacf09SStefano Zampini             }
80177eacf09SStefano Zampini           }
8029566063dSJacob Faibussowitsch           PetscCall(DMPlexRestoreTransitiveClosure(dm,p,PETSC_TRUE,&cellClosureSize,&cellClosure));
8039566063dSJacob Faibussowitsch           PetscCall(DMPlexVecRestoreClosure(dm,coordSection,coordinates,p,&csize,&vals));
80477eacf09SStefano Zampini         }
80577eacf09SStefano Zampini       }
80677eacf09SStefano Zampini       if (dim > 1) {
807b135d7daSStefano Zampini         PetscInt eEnd,eStart;
808044a5661SStefano Zampini 
8099566063dSJacob Faibussowitsch         PetscCall(DMPlexGetDepthStratum(dm,1,&eStart,&eEnd));
81077eacf09SStefano Zampini         for (p=eStart;p<eEnd;p++) {
81177eacf09SStefano Zampini           const PetscInt *cone;
81277eacf09SStefano Zampini           PetscInt       coneSize,i;
81377eacf09SStefano Zampini           PetscBool      ispe = PETSC_TRUE;
81477eacf09SStefano Zampini 
8159566063dSJacob Faibussowitsch           PetscCall(DMPlexGetCone(dm,p,&cone));
8169566063dSJacob Faibussowitsch           PetscCall(DMPlexGetConeSize(dm,p,&coneSize));
81777eacf09SStefano Zampini           for (i=0;i<coneSize;i++) {
81877eacf09SStefano Zampini             PetscInt v;
81977eacf09SStefano Zampini 
8209566063dSJacob Faibussowitsch             PetscCall(DMLabelGetValue(perLabel,cone[i],&v));
82177eacf09SStefano Zampini             ispe = (PetscBool)(ispe && (v==2));
82277eacf09SStefano Zampini           }
82377eacf09SStefano Zampini           if (ispe && coneSize) {
8243e96840aSStefano Zampini             PetscInt       ch, numChildren;
8253e96840aSStefano Zampini             const PetscInt *children;
8263e96840aSStefano Zampini 
8279566063dSJacob Faibussowitsch             PetscCall(DMLabelSetValue(perLabel,p,2));
8289566063dSJacob Faibussowitsch             PetscCall(DMPlexGetTreeChildren(dm,p,&numChildren,&children));
8293e96840aSStefano Zampini             for (ch = 0; ch < numChildren; ch++) {
8309566063dSJacob Faibussowitsch               PetscCall(DMLabelSetValue(perLabel,children[ch],2));
8313e96840aSStefano Zampini             }
83277eacf09SStefano Zampini           }
83377eacf09SStefano Zampini         }
83477eacf09SStefano Zampini         if (dim > 2) {
83577eacf09SStefano Zampini           for (p=fStart;p<fEnd;p++) {
83677eacf09SStefano Zampini             const PetscInt *cone;
83777eacf09SStefano Zampini             PetscInt       coneSize,i;
83877eacf09SStefano Zampini             PetscBool      ispe = PETSC_TRUE;
83977eacf09SStefano Zampini 
8409566063dSJacob Faibussowitsch             PetscCall(DMPlexGetCone(dm,p,&cone));
8419566063dSJacob Faibussowitsch             PetscCall(DMPlexGetConeSize(dm,p,&coneSize));
84277eacf09SStefano Zampini             for (i=0;i<coneSize;i++) {
84377eacf09SStefano Zampini               PetscInt v;
84477eacf09SStefano Zampini 
8459566063dSJacob Faibussowitsch               PetscCall(DMLabelGetValue(perLabel,cone[i],&v));
84677eacf09SStefano Zampini               ispe = (PetscBool)(ispe && (v==2));
84777eacf09SStefano Zampini             }
84877eacf09SStefano Zampini             if (ispe && coneSize) {
8493e96840aSStefano Zampini               PetscInt       ch, numChildren;
8503e96840aSStefano Zampini               const PetscInt *children;
8513e96840aSStefano Zampini 
8529566063dSJacob Faibussowitsch               PetscCall(DMLabelSetValue(perLabel,p,2));
8539566063dSJacob Faibussowitsch               PetscCall(DMPlexGetTreeChildren(dm,p,&numChildren,&children));
8543e96840aSStefano Zampini               for (ch = 0; ch < numChildren; ch++) {
8559566063dSJacob Faibussowitsch                 PetscCall(DMLabelSetValue(perLabel,children[ch],2));
8563e96840aSStefano Zampini               }
85777eacf09SStefano Zampini             }
85877eacf09SStefano Zampini           }
85977eacf09SStefano Zampini         }
86077eacf09SStefano Zampini       }
86177eacf09SStefano Zampini     }
86277eacf09SStefano Zampini     for (p=fStart;p<fEnd;p++) {
86377eacf09SStefano Zampini       const PetscInt *support;
8648135c375SStefano Zampini       PetscInt       supportSize;
86577eacf09SStefano Zampini       PetscBool      isbf = PETSC_FALSE;
8668135c375SStefano Zampini 
8679566063dSJacob Faibussowitsch       PetscCall(DMPlexGetSupportSize(dm,p,&supportSize));
8683e6c54aaSStefano Zampini       if (pown) {
8698135c375SStefano Zampini         PetscBool has_owned = PETSC_FALSE, has_ghost = PETSC_FALSE;
87077eacf09SStefano Zampini         PetscInt  i;
87177eacf09SStefano Zampini 
8729566063dSJacob Faibussowitsch         PetscCall(DMPlexGetSupport(dm,p,&support));
87377eacf09SStefano Zampini         for (i=0;i<supportSize;i++) {
87477eacf09SStefano Zampini           if (PetscLikely(PetscBTLookup(pown,support[i]-cStart))) has_owned = PETSC_TRUE;
87577eacf09SStefano Zampini           else has_ghost = PETSC_TRUE;
87677eacf09SStefano Zampini         }
87777eacf09SStefano Zampini         isbf = (PetscBool)((supportSize == 1 && has_owned) || (supportSize > 1 && has_owned && has_ghost));
87877eacf09SStefano Zampini       } else {
87977eacf09SStefano Zampini         isbf = (PetscBool)(supportSize == 1);
88077eacf09SStefano Zampini       }
88177eacf09SStefano Zampini       if (!isbf && perLabel) {
88277eacf09SStefano Zampini         const PetscInt *cone;
88377eacf09SStefano Zampini         PetscInt       coneSize,i;
88477eacf09SStefano Zampini 
8859566063dSJacob Faibussowitsch         PetscCall(DMPlexGetCone(dm,p,&cone));
8869566063dSJacob Faibussowitsch         PetscCall(DMPlexGetConeSize(dm,p,&coneSize));
88777eacf09SStefano Zampini         isbf = PETSC_TRUE;
88877eacf09SStefano Zampini         for (i=0;i<coneSize;i++) {
88977eacf09SStefano Zampini           PetscInt v,d;
89077eacf09SStefano Zampini 
8919566063dSJacob Faibussowitsch           PetscCall(DMLabelGetValue(perLabel,cone[i],&v));
8929566063dSJacob Faibussowitsch           PetscCall(DMLabelGetDefaultValue(perLabel,&d));
89377eacf09SStefano Zampini           isbf = (PetscBool)(isbf && v != d);
89477eacf09SStefano Zampini         }
89577eacf09SStefano Zampini       }
89677eacf09SStefano Zampini       if (isbf) {
8979566063dSJacob Faibussowitsch         PetscCall(PetscBTSet(bfaces,p-fStart));
89877eacf09SStefano Zampini       }
89977eacf09SStefano Zampini     }
90077eacf09SStefano Zampini     /* count boundary faces */
90177eacf09SStefano Zampini     for (p=fStart,bf=0;p<fEnd;p++) {
90277eacf09SStefano Zampini       if (PetscUnlikely(PetscBTLookup(bfaces,p-fStart))) {
90377eacf09SStefano Zampini         const PetscInt *support;
90477eacf09SStefano Zampini         PetscInt       supportSize,c;
9058135c375SStefano Zampini 
9069566063dSJacob Faibussowitsch         PetscCall(DMPlexGetSupportSize(dm,p,&supportSize));
9079566063dSJacob Faibussowitsch         PetscCall(DMPlexGetSupport(dm,p,&support));
90877eacf09SStefano Zampini         for (c=0;c<supportSize;c++) {
9093e96840aSStefano Zampini           const    PetscInt *cone;
910b135d7daSStefano Zampini           PetscInt cell,cl,coneSize;
9113e96840aSStefano Zampini 
9123e96840aSStefano Zampini           cell = support[c];
9133e96840aSStefano Zampini           if (pown && PetscUnlikely(!PetscBTLookup(pown,cell-cStart))) continue;
9149566063dSJacob Faibussowitsch           PetscCall(DMPlexGetCone(dm,cell,&cone));
9159566063dSJacob Faibussowitsch           PetscCall(DMPlexGetConeSize(dm,cell,&coneSize));
916b135d7daSStefano Zampini           for (cl=0;cl<coneSize;cl++) {
9173e96840aSStefano Zampini             if (cone[cl] == p) {
9183e96840aSStefano Zampini               bf += 1;
9193e96840aSStefano Zampini               break;
9208135c375SStefano Zampini             }
92177eacf09SStefano Zampini           }
9223e96840aSStefano Zampini         }
9238135c375SStefano Zampini       }
9248135c375SStefano Zampini     }
925f86f7544SStefano Zampini     minl = 1;
926f86f7544SStefano Zampini     label = NULL;
927f86f7544SStefano Zampini     if (enable_bmark) {
928f86f7544SStefano Zampini       PetscInt lminl = PETSC_MAX_INT;
929f86f7544SStefano Zampini 
9309566063dSJacob Faibussowitsch       PetscCall(DMGetLabel(dm,bmark,&label));
931f86f7544SStefano Zampini       if (label) {
932f86f7544SStefano Zampini         IS       vals;
933f86f7544SStefano Zampini         PetscInt ldef;
934f86f7544SStefano Zampini 
9359566063dSJacob Faibussowitsch         PetscCall(DMLabelGetDefaultValue(label,&ldef));
9369566063dSJacob Faibussowitsch         PetscCall(DMLabelGetValueIS(label,&vals));
9379566063dSJacob Faibussowitsch         PetscCall(ISGetMinMax(vals,&lminl,NULL));
9389566063dSJacob Faibussowitsch         PetscCall(ISDestroy(&vals));
939f86f7544SStefano Zampini         lminl = PetscMin(ldef,lminl);
940f86f7544SStefano Zampini       }
9411c2dc1cbSBarry Smith       PetscCall(MPIU_Allreduce(&lminl,&minl,1,MPIU_INT,MPI_MIN,PetscObjectComm((PetscObject)dm)));
942f86f7544SStefano Zampini       if (minl == PETSC_MAX_INT) minl = 1;
943f86f7544SStefano Zampini     }
9449566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",bf));
9458135c375SStefano Zampini     for (p=fStart;p<fEnd;p++) {
94677eacf09SStefano Zampini       if (PetscUnlikely(PetscBTLookup(bfaces,p-fStart))) {
9478135c375SStefano Zampini         const PetscInt *support;
94877eacf09SStefano Zampini         PetscInt       supportSize,c,nc = 0;
9498135c375SStefano Zampini 
9509566063dSJacob Faibussowitsch         PetscCall(DMPlexGetSupportSize(dm,p,&supportSize));
9519566063dSJacob Faibussowitsch         PetscCall(DMPlexGetSupport(dm,p,&support));
9523e6c54aaSStefano Zampini         if (pown) {
95377eacf09SStefano Zampini           for (c=0;c<supportSize;c++) {
95477eacf09SStefano Zampini             if (PetscLikely(PetscBTLookup(pown,support[c]-cStart))) {
95577eacf09SStefano Zampini               fcells[nc++] = support[c];
9568135c375SStefano Zampini             }
95777eacf09SStefano Zampini           }
95877eacf09SStefano Zampini         } else for (c=0;c<supportSize;c++) fcells[nc++] = support[c];
95977eacf09SStefano Zampini         for (c=0;c<nc;c++) {
960c3c203b2SStefano Zampini           const DMPolytopeType *faceTypes;
961c3c203b2SStefano Zampini           DMPolytopeType       cellType;
962c3c203b2SStefano Zampini           const PetscInt       *faceSizes,*cone;
963c3c203b2SStefano Zampini           PetscInt             vids[8],*faces,st,i,coneSize,cell,cl,nv,cid = -1,mid = -1;
9648135c375SStefano Zampini 
96577eacf09SStefano Zampini           cell = fcells[c];
9669566063dSJacob Faibussowitsch           PetscCall(DMPlexGetCone(dm,cell,&cone));
9679566063dSJacob Faibussowitsch           PetscCall(DMPlexGetConeSize(dm,cell,&coneSize));
968b135d7daSStefano Zampini           for (cl=0;cl<coneSize;cl++)
96977eacf09SStefano Zampini             if (cone[cl] == p)
9708135c375SStefano Zampini               break;
971b135d7daSStefano Zampini           if (cl == coneSize) continue;
9728135c375SStefano Zampini 
97377eacf09SStefano Zampini           /* face material id and type */
9749566063dSJacob Faibussowitsch           PetscCall(DMPlexGetPointMFEMCellID_Internal(dm,label,minl,p,&mid,&cid));
9759566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer,"%D %D",mid,cid));
97677eacf09SStefano Zampini           /* vertex ids */
9779566063dSJacob Faibussowitsch           PetscCall(DMPlexGetCellType(dm,cell,&cellType));
9789566063dSJacob Faibussowitsch           PetscCall(DMPlexGetPointMFEMVertexIDs_Internal(dm,cell,(localized && !hovec) ? coordSection : NULL,&nv,vids));
9799566063dSJacob Faibussowitsch           PetscCall(DMPlexGetRawFaces_Internal(dm,cellType,vids,NULL,&faceTypes,&faceSizes,(const PetscInt**)&faces));
980c3c203b2SStefano Zampini           st = 0;
981c3c203b2SStefano Zampini           for (i=0;i<cl;i++) st += faceSizes[i];
9829566063dSJacob Faibussowitsch           PetscCall(DMPlexInvertCell(faceTypes[cl],faces + st));
983c3c203b2SStefano Zampini           for (i=0;i<faceSizes[cl];i++) {
9849566063dSJacob Faibussowitsch             PetscCall(PetscViewerASCIIPrintf(viewer," %d",faces[st+i]));
985b135d7daSStefano Zampini           }
9869566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
9879566063dSJacob Faibussowitsch           PetscCall(DMPlexRestoreRawFaces_Internal(dm,cellType,vids,NULL,&faceTypes,&faceSizes,(const PetscInt**)&faces));
9883e96840aSStefano Zampini           bf -= 1;
98977eacf09SStefano Zampini         }
9908135c375SStefano Zampini       }
9918135c375SStefano Zampini     }
99228b400f6SJacob Faibussowitsch     PetscCheck(!bf,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Remaining boundary faces %D",bf);
9939566063dSJacob Faibussowitsch     PetscCall(PetscBTDestroy(&bfaces));
9949566063dSJacob Faibussowitsch     PetscCall(PetscFree(fcells));
9958135c375SStefano Zampini   }
9968135c375SStefano Zampini 
9978135c375SStefano Zampini   /* mark owned vertices */
9983e6c54aaSStefano Zampini   vown = NULL;
9993e6c54aaSStefano Zampini   if (pown) {
10009566063dSJacob Faibussowitsch     PetscCall(PetscBTCreate(vEnd-vStart,&vown));
10018135c375SStefano Zampini     for (p=cStart;p<cEnd;p++) {
10028135c375SStefano Zampini       PetscInt i,closureSize,*closure = NULL;
10038135c375SStefano Zampini 
10043e6c54aaSStefano Zampini       if (PetscUnlikely(!PetscBTLookup(pown,p-cStart))) continue;
10059566063dSJacob Faibussowitsch       PetscCall(DMPlexGetTransitiveClosure(dm,p,PETSC_TRUE,&closureSize,&closure));
10068135c375SStefano Zampini       for (i=0;i<closureSize;i++) {
10078135c375SStefano Zampini         const PetscInt pp = closure[2*i];
10088135c375SStefano Zampini 
10098135c375SStefano Zampini         if (pp >= vStart && pp < vEnd) {
10109566063dSJacob Faibussowitsch           PetscCall(PetscBTSet(vown,pp-vStart));
10118135c375SStefano Zampini         }
10128135c375SStefano Zampini       }
10139566063dSJacob Faibussowitsch       PetscCall(DMPlexRestoreTransitiveClosure(dm,p,PETSC_TRUE,&closureSize,&closure));
10148135c375SStefano Zampini     }
10158135c375SStefano Zampini   }
10168135c375SStefano Zampini 
10178135c375SStefano Zampini   if (parentSection) {
10188135c375SStefano Zampini     PetscInt vp,gvp;
10198135c375SStefano Zampini 
10208135c375SStefano Zampini     for (vp=0,p=vStart;p<vEnd;p++) {
10218135c375SStefano Zampini       DMLabel  dlabel;
10228135c375SStefano Zampini       PetscInt parent,depth;
10238135c375SStefano Zampini 
10243e6c54aaSStefano Zampini       if (PetscUnlikely(vown && !PetscBTLookup(vown,p-vStart))) continue;
10259566063dSJacob Faibussowitsch       PetscCall(DMPlexGetDepthLabel(dm,&dlabel));
10269566063dSJacob Faibussowitsch       PetscCall(DMLabelGetValue(dlabel,p,&depth));
10279566063dSJacob Faibussowitsch       PetscCall(DMPlexGetTreeParent(dm,p,&parent,NULL));
10288135c375SStefano Zampini       if (parent != p) vp++;
10298135c375SStefano Zampini     }
10301c2dc1cbSBarry Smith     PetscCall(MPIU_Allreduce(&vp,&gvp,1,MPIU_INT,MPI_SUM,PetscObjectComm((PetscObject)dm)));
10318135c375SStefano Zampini     if (gvp) {
10328135c375SStefano Zampini       PetscInt  maxsupp;
10338135c375SStefano Zampini       PetscBool *skip = NULL;
10348135c375SStefano Zampini 
10359566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"\nvertex_parents\n"));
10369566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",vp));
10379566063dSJacob Faibussowitsch       PetscCall(DMPlexGetMaxSizes(dm,NULL,&maxsupp));
10389566063dSJacob Faibussowitsch       PetscCall(PetscMalloc1(maxsupp,&skip));
10398135c375SStefano Zampini       for (p=vStart;p<vEnd;p++) {
10408135c375SStefano Zampini         DMLabel  dlabel;
10418135c375SStefano Zampini         PetscInt parent;
10428135c375SStefano Zampini 
10433e6c54aaSStefano Zampini         if (PetscUnlikely(vown && !PetscBTLookup(vown,p-vStart))) continue;
10449566063dSJacob Faibussowitsch         PetscCall(DMPlexGetDepthLabel(dm,&dlabel));
10459566063dSJacob Faibussowitsch         PetscCall(DMPlexGetTreeParent(dm,p,&parent,NULL));
10468135c375SStefano Zampini         if (parent != p) {
104796ca5757SLisandro Dalcin           PetscInt       vids[8] = { -1, -1, -1, -1, -1, -1, -1, -1 }; /* silent overzealous clang static analyzer */
10483924b612SStefano Zampini           PetscInt       i,nv,ssize,n,numChildren,depth = -1;
10498135c375SStefano Zampini           const PetscInt *children;
10503924b612SStefano Zampini 
10519566063dSJacob Faibussowitsch           PetscCall(DMPlexGetConeSize(dm,parent,&ssize));
10523924b612SStefano Zampini           switch (ssize) {
10538135c375SStefano Zampini             case 2: /* edge */
10548135c375SStefano Zampini               nv   = 0;
10559566063dSJacob Faibussowitsch               PetscCall(DMPlexGetPointMFEMVertexIDs_Internal(dm,parent,localized ? coordSection : NULL,&nv,vids));
10569566063dSJacob Faibussowitsch               PetscCall(PetscViewerASCIIPrintf(viewer,"%D",p-vStart));
10578135c375SStefano Zampini               for (i=0;i<nv;i++) {
10589566063dSJacob Faibussowitsch                 PetscCall(PetscViewerASCIIPrintf(viewer," %D",vids[i]));
10598135c375SStefano Zampini               }
10609566063dSJacob Faibussowitsch               PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
10618135c375SStefano Zampini               vp--;
10628135c375SStefano Zampini               break;
10638135c375SStefano Zampini             case 4: /* face */
10649566063dSJacob Faibussowitsch               PetscCall(DMPlexGetTreeChildren(dm,parent,&numChildren,&children));
10658135c375SStefano Zampini               for (n=0;n<numChildren;n++) {
10669566063dSJacob Faibussowitsch                 PetscCall(DMLabelGetValue(dlabel,children[n],&depth));
10678135c375SStefano Zampini                 if (!depth) {
10688135c375SStefano Zampini                   const PetscInt *hvsupp,*hesupp,*cone;
10698135c375SStefano Zampini                   PetscInt       hvsuppSize,hesuppSize,coneSize;
1070451a39c7SStefano Zampini                   PetscInt       hv = children[n],he = -1,f;
10718135c375SStefano Zampini 
10729566063dSJacob Faibussowitsch                   PetscCall(PetscArrayzero(skip,maxsupp));
10739566063dSJacob Faibussowitsch                   PetscCall(DMPlexGetSupportSize(dm,hv,&hvsuppSize));
10749566063dSJacob Faibussowitsch                   PetscCall(DMPlexGetSupport(dm,hv,&hvsupp));
10758135c375SStefano Zampini                   for (i=0;i<hvsuppSize;i++) {
10768135c375SStefano Zampini                     PetscInt ep;
10779566063dSJacob Faibussowitsch                     PetscCall(DMPlexGetTreeParent(dm,hvsupp[i],&ep,NULL));
10788135c375SStefano Zampini                     if (ep != hvsupp[i]) {
10798135c375SStefano Zampini                       he = hvsupp[i];
10808135c375SStefano Zampini                     } else {
10818135c375SStefano Zampini                       skip[i] = PETSC_TRUE;
10828135c375SStefano Zampini                     }
10838135c375SStefano Zampini                   }
1084*08401ef6SPierre Jolivet                   PetscCheck(he != -1,PETSC_COMM_SELF,PETSC_ERR_SUP,"Vertex %D support size %D: hanging edge not found",hv,hvsuppSize);
10859566063dSJacob Faibussowitsch                   PetscCall(DMPlexGetCone(dm,he,&cone));
108696ca5757SLisandro Dalcin                   vids[0] = (cone[0] == hv) ? cone[1] : cone[0];
10879566063dSJacob Faibussowitsch                   PetscCall(DMPlexGetSupportSize(dm,he,&hesuppSize));
10889566063dSJacob Faibussowitsch                   PetscCall(DMPlexGetSupport(dm,he,&hesupp));
10898135c375SStefano Zampini                   for (f=0;f<hesuppSize;f++) {
10908135c375SStefano Zampini                     PetscInt j;
10918135c375SStefano Zampini 
10929566063dSJacob Faibussowitsch                     PetscCall(DMPlexGetCone(dm,hesupp[f],&cone));
10939566063dSJacob Faibussowitsch                     PetscCall(DMPlexGetConeSize(dm,hesupp[f],&coneSize));
10948135c375SStefano Zampini                     for (j=0;j<coneSize;j++) {
10958135c375SStefano Zampini                       PetscInt k;
10968135c375SStefano Zampini                       for (k=0;k<hvsuppSize;k++) {
10978135c375SStefano Zampini                         if (hvsupp[k] == cone[j]) {
10988135c375SStefano Zampini                           skip[k] = PETSC_TRUE;
10998135c375SStefano Zampini                           break;
11008135c375SStefano Zampini                         }
11018135c375SStefano Zampini                       }
11028135c375SStefano Zampini                     }
11038135c375SStefano Zampini                   }
11048135c375SStefano Zampini                   for (i=0;i<hvsuppSize;i++) {
11058135c375SStefano Zampini                     if (!skip[i]) {
11069566063dSJacob Faibussowitsch                       PetscCall(DMPlexGetCone(dm,hvsupp[i],&cone));
110796ca5757SLisandro Dalcin                       vids[1] = (cone[0] == hv) ? cone[1] : cone[0];
11088135c375SStefano Zampini                     }
11098135c375SStefano Zampini                   }
11109566063dSJacob Faibussowitsch                   PetscCall(PetscViewerASCIIPrintf(viewer,"%D",hv-vStart));
11118135c375SStefano Zampini                   for (i=0;i<2;i++) {
11129566063dSJacob Faibussowitsch                     PetscCall(PetscViewerASCIIPrintf(viewer," %D",vids[i]-vStart));
11138135c375SStefano Zampini                   }
11149566063dSJacob Faibussowitsch                   PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
11158135c375SStefano Zampini                   vp--;
11168135c375SStefano Zampini                 }
11178135c375SStefano Zampini               }
11188135c375SStefano Zampini               break;
11198135c375SStefano Zampini             default:
112098921bdaSJacob Faibussowitsch               SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Don't know how to deal with support size %D",ssize);
11218135c375SStefano Zampini           }
11228135c375SStefano Zampini         }
11238135c375SStefano Zampini       }
11249566063dSJacob Faibussowitsch       PetscCall(PetscFree(skip));
11258135c375SStefano Zampini     }
112628b400f6SJacob Faibussowitsch     PetscCheck(!vp,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected %D hanging vertices",vp);
11278135c375SStefano Zampini   }
11289566063dSJacob Faibussowitsch   PetscCall(PetscBTDestroy(&vown));
11298135c375SStefano Zampini 
11308135c375SStefano Zampini   /* vertices */
113177eacf09SStefano Zampini   if (hovec) { /* higher-order meshes */
113277eacf09SStefano Zampini     const char *fec;
11330286d493SLisandro Dalcin     PetscInt   i,n,s;
11349566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"\nvertices\n"));
11359566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",vEnd-vStart));
11369566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"nodes\n"));
11379566063dSJacob Faibussowitsch     PetscCall(PetscObjectGetName((PetscObject)hovec,&fec));
11389566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"FiniteElementSpace\n"));
11399566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%s\n",fec));
11409566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"VDim: %D\n",sdim));
11419566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"Ordering: 1\n\n")); /*Ordering::byVDIM*/
11420c2bc6bfSStefano Zampini     if (hoSection) {
11430c2bc6bfSStefano Zampini       DM cdm;
11440c2bc6bfSStefano Zampini 
11450c2bc6bfSStefano Zampini       PetscCall(VecGetDM(hovec,&cdm));
11460c2bc6bfSStefano Zampini       for (p=cStart;p<cEnd;p++) {
11470c2bc6bfSStefano Zampini         PetscScalar *vals = NULL;
11480c2bc6bfSStefano Zampini         PetscInt    csize;
11490c2bc6bfSStefano Zampini 
11500c2bc6bfSStefano Zampini         if (PetscUnlikely(pown && !PetscBTLookup(pown,p-cStart))) continue;
11510c2bc6bfSStefano Zampini         PetscCall(DMPlexVecGetClosure(cdm,hoSection,hovec,p,&csize,&vals));
11520c2bc6bfSStefano Zampini         PetscCheck(csize%sdim == 0,PETSC_COMM_SELF,PETSC_ERR_USER,"Size of closure %D incompatible with space dimension %D",csize,sdim);
11530c2bc6bfSStefano Zampini         for (i=0;i<csize/sdim;i++) {
11540c2bc6bfSStefano Zampini           for (s=0;s<sdim;s++) {
11550c2bc6bfSStefano Zampini             PetscCall(PetscViewerASCIIPrintf(viewer,fmt,(double) PetscRealPart(vals[i*sdim+s])));
11560c2bc6bfSStefano Zampini           }
11570c2bc6bfSStefano Zampini           PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
11580c2bc6bfSStefano Zampini         }
11590c2bc6bfSStefano Zampini         PetscCall(DMPlexVecRestoreClosure(cdm,hoSection,hovec,p,&csize,&vals));
11600c2bc6bfSStefano Zampini       }
11610c2bc6bfSStefano Zampini     } else {
11629566063dSJacob Faibussowitsch       PetscCall(VecGetArrayRead(hovec,&array));
11639566063dSJacob Faibussowitsch       PetscCall(VecGetLocalSize(hovec,&n));
11640c2bc6bfSStefano Zampini       PetscCheck(n%sdim == 0,PETSC_COMM_SELF,PETSC_ERR_USER,"Size of local coordinate vector %D incompatible with space dimension %D",n,sdim);
116577eacf09SStefano Zampini       for (i=0;i<n/sdim;i++) {
116677eacf09SStefano Zampini         for (s=0;s<sdim;s++) {
11679566063dSJacob Faibussowitsch           PetscCall(PetscViewerASCIIPrintf(viewer,fmt,(double) PetscRealPart(array[i*sdim+s])));
116877eacf09SStefano Zampini         }
11699566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
117077eacf09SStefano Zampini       }
11719566063dSJacob Faibussowitsch       PetscCall(VecRestoreArrayRead(hovec,&array));
11720c2bc6bfSStefano Zampini     }
117377eacf09SStefano Zampini   } else {
11749566063dSJacob Faibussowitsch     PetscCall(VecGetLocalSize(coordinates,&nvert));
11759566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"\nvertices\n"));
11769566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",nvert/sdim));
11779566063dSJacob Faibussowitsch     PetscCall(PetscViewerASCIIPrintf(viewer,"%D\n",sdim));
11789566063dSJacob Faibussowitsch     PetscCall(VecGetArrayRead(coordinates,&array));
1179cc0d3ed7SStefano Zampini     for (p=0;p<nvert/sdim;p++) {
1180cc0d3ed7SStefano Zampini       PetscInt s;
1181cc0d3ed7SStefano Zampini       for (s=0;s<sdim;s++) {
11823e96840aSStefano Zampini         PetscReal v = PetscRealPart(array[p*sdim+s]);
11833e96840aSStefano Zampini 
11849566063dSJacob Faibussowitsch         PetscCall(PetscViewerASCIIPrintf(viewer,fmt,PetscIsInfOrNanReal(v) ? 0.0 : (double) v));
11858135c375SStefano Zampini       }
11869566063dSJacob Faibussowitsch       PetscCall(PetscViewerASCIIPrintf(viewer,"\n"));
11878135c375SStefano Zampini     }
11889566063dSJacob Faibussowitsch     PetscCall(VecRestoreArrayRead(coordinates,&array));
118977eacf09SStefano Zampini   }
11900c2bc6bfSStefano Zampini   PetscCall(PetscBTDestroy(&pown));
11919566063dSJacob Faibussowitsch   PetscCall(VecDestroy(&hovec));
11928135c375SStefano Zampini   PetscFunctionReturn(0);
11938135c375SStefano Zampini }
11948135c375SStefano Zampini 
11950286d493SLisandro Dalcin PetscErrorCode DMPlexView_GLVis(DM dm, PetscViewer viewer)
11968135c375SStefano Zampini {
11978135c375SStefano Zampini   PetscFunctionBegin;
11989566063dSJacob Faibussowitsch   PetscCall(DMView_GLVis(dm,viewer,DMPlexView_GLVis_ASCII));
11998135c375SStefano Zampini   PetscFunctionReturn(0);
12008135c375SStefano Zampini }
1201