xref: /petsc/src/dm/impls/plex/plexglvis.c (revision c3c203b264e2b6b76a5cd3913d71d10bfa55cb13)
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   PetscErrorCode ierr;
198135c375SStefano Zampini 
208135c375SStefano Zampini   PetscFunctionBegin;
218135c375SStefano Zampini   for (i=0;i<ctx->nf;i++) {
228135c375SStefano Zampini     ierr = VecScatterDestroy(&ctx->scctx[i]);CHKERRQ(ierr);
238135c375SStefano Zampini   }
248135c375SStefano Zampini   ierr = PetscFree(ctx->scctx);CHKERRQ(ierr);
2551f51421SSatish Balay   ierr = PetscFree(vctx);CHKERRQ(ierr);
268135c375SStefano Zampini   PetscFunctionReturn(0);
278135c375SStefano Zampini }
288135c375SStefano Zampini 
298135c375SStefano Zampini static PetscErrorCode DMPlexSampleGLVisFields_Private(PetscObject oX, PetscInt nf, PetscObject oXfield[], void *vctx)
308135c375SStefano Zampini {
318135c375SStefano Zampini   GLVisViewerCtx *ctx = (GLVisViewerCtx*)vctx;
328135c375SStefano Zampini   PetscInt       f;
338135c375SStefano Zampini   PetscErrorCode ierr;
348135c375SStefano Zampini 
358135c375SStefano Zampini   PetscFunctionBegin;
368135c375SStefano Zampini   for (f=0;f<nf;f++) {
378135c375SStefano Zampini     ierr = VecScatterBegin(ctx->scctx[f],(Vec)oX,(Vec)oXfield[f],INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
388135c375SStefano Zampini     ierr = VecScatterEnd(ctx->scctx[f],(Vec)oX,(Vec)oXfield[f],INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr);
398135c375SStefano Zampini   }
408135c375SStefano Zampini   PetscFunctionReturn(0);
418135c375SStefano Zampini }
428135c375SStefano Zampini 
438135c375SStefano Zampini /* for FEM, it works for H1 fields only and extracts dofs at cell vertices, discarding any other dof */
448135c375SStefano Zampini PetscErrorCode DMSetUpGLVisViewer_Plex(PetscObject odm, PetscViewer viewer)
458135c375SStefano Zampini {
468135c375SStefano Zampini   DM             dm = (DM)odm;
474cac2994SStefano Zampini   Vec            xlocal,xfield,*Ufield;
488135c375SStefano Zampini   PetscDS        ds;
498135c375SStefano Zampini   IS             globalNum,isfield;
508135c375SStefano Zampini   PetscBT        vown;
518135c375SStefano Zampini   char           **fieldname = NULL,**fec_type = NULL;
528135c375SStefano Zampini   const PetscInt *gNum;
53bb77a09fSStefano Zampini   PetscInt       *nlocal,*bs,*idxs,*dims;
548135c375SStefano Zampini   PetscInt       f,maxfields,nfields,c,totc,totdofs,Nv,cum,i;
55b135d7daSStefano Zampini   PetscInt       dim,cStart,cEnd,vStart,vEnd;
568135c375SStefano Zampini   GLVisViewerCtx *ctx;
578135c375SStefano Zampini   PetscSection   s;
588135c375SStefano Zampini   PetscErrorCode ierr;
598135c375SStefano Zampini 
608135c375SStefano Zampini   PetscFunctionBegin;
618135c375SStefano Zampini   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
628135c375SStefano Zampini   ierr = DMPlexGetDepthStratum(dm,0,&vStart,&vEnd);CHKERRQ(ierr);
638135c375SStefano Zampini   ierr = DMPlexGetHeightStratum(dm,0,&cStart,&cEnd);CHKERRQ(ierr);
64b135d7daSStefano Zampini   ierr = PetscObjectQuery((PetscObject)dm,"_glvis_plex_gnum",(PetscObject*)&globalNum);CHKERRQ(ierr);
65b135d7daSStefano Zampini   if (!globalNum) {
66b135d7daSStefano Zampini     ierr = DMPlexCreateCellNumbering_Internal(dm,PETSC_TRUE,&globalNum);CHKERRQ(ierr);
67b135d7daSStefano Zampini     ierr = PetscObjectCompose((PetscObject)dm,"_glvis_plex_gnum",(PetscObject)globalNum);CHKERRQ(ierr);
68b135d7daSStefano Zampini     ierr = PetscObjectDereference((PetscObject)globalNum);CHKERRQ(ierr);
69b135d7daSStefano Zampini   }
708135c375SStefano Zampini   ierr = ISGetIndices(globalNum,&gNum);CHKERRQ(ierr);
718135c375SStefano Zampini   ierr = PetscBTCreate(vEnd-vStart,&vown);CHKERRQ(ierr);
728135c375SStefano Zampini   for (c = cStart, totc = 0; c < cEnd; c++) {
738135c375SStefano Zampini     if (gNum[c-cStart] >= 0) {
748135c375SStefano Zampini       PetscInt i,numPoints,*points = NULL;
758135c375SStefano Zampini 
768135c375SStefano Zampini       totc++;
778135c375SStefano Zampini       ierr = DMPlexGetTransitiveClosure(dm,c,PETSC_TRUE,&numPoints,&points);CHKERRQ(ierr);
788135c375SStefano Zampini       for (i=0;i<numPoints*2;i+= 2) {
798135c375SStefano Zampini         if ((points[i] >= vStart) && (points[i] < vEnd)) {
808135c375SStefano Zampini           ierr = PetscBTSet(vown,points[i]-vStart);CHKERRQ(ierr);
818135c375SStefano Zampini         }
828135c375SStefano Zampini       }
838135c375SStefano Zampini       ierr = DMPlexRestoreTransitiveClosure(dm,c,PETSC_TRUE,&numPoints,&points);CHKERRQ(ierr);
848135c375SStefano Zampini     }
858135c375SStefano Zampini   }
8677eacf09SStefano Zampini   for (f=0,Nv=0;f<vEnd-vStart;f++) if (PetscLikely(PetscBTLookup(vown,f))) Nv++;
878135c375SStefano Zampini 
888135c375SStefano Zampini   ierr = DMCreateLocalVector(dm,&xlocal);CHKERRQ(ierr);
898135c375SStefano Zampini   ierr = VecGetLocalSize(xlocal,&totdofs);CHKERRQ(ierr);
9092fd8e1eSJed Brown   ierr = DMGetLocalSection(dm,&s);CHKERRQ(ierr);
918135c375SStefano Zampini   ierr = PetscSectionGetNumFields(s,&nfields);CHKERRQ(ierr);
928135c375SStefano Zampini   for (f=0,maxfields=0;f<nfields;f++) {
938135c375SStefano Zampini     PetscInt bs;
948135c375SStefano Zampini 
958135c375SStefano Zampini     ierr = PetscSectionGetFieldComponents(s,f,&bs);CHKERRQ(ierr);
968135c375SStefano Zampini     maxfields += bs;
978135c375SStefano Zampini   }
984cac2994SStefano Zampini   ierr = PetscCalloc7(maxfields,&fieldname,maxfields,&nlocal,maxfields,&bs,maxfields,&dims,maxfields,&fec_type,totdofs,&idxs,maxfields,&Ufield);CHKERRQ(ierr);
998135c375SStefano Zampini   ierr = PetscNew(&ctx);CHKERRQ(ierr);
1008135c375SStefano Zampini   ierr = PetscCalloc1(maxfields,&ctx->scctx);CHKERRQ(ierr);
1018135c375SStefano Zampini   ierr = DMGetDS(dm,&ds);CHKERRQ(ierr);
1028135c375SStefano Zampini   if (ds) {
1038135c375SStefano Zampini     for (f=0;f<nfields;f++) {
1048135c375SStefano Zampini       const char* fname;
1058135c375SStefano Zampini       char        name[256];
1068135c375SStefano Zampini       PetscObject disc;
1078135c375SStefano Zampini       size_t      len;
1088135c375SStefano Zampini 
1098135c375SStefano Zampini       ierr = PetscSectionGetFieldName(s,f,&fname);CHKERRQ(ierr);
1108135c375SStefano Zampini       ierr = PetscStrlen(fname,&len);CHKERRQ(ierr);
1118135c375SStefano Zampini       if (len) {
1128135c375SStefano Zampini         ierr = PetscStrcpy(name,fname);CHKERRQ(ierr);
1138135c375SStefano Zampini       } else {
114bfadf1d8SStefano Zampini         ierr = PetscSNPrintf(name,256,"Field%D",f);CHKERRQ(ierr);
1158135c375SStefano Zampini       }
1168135c375SStefano Zampini       ierr = PetscDSGetDiscretization(ds,f,&disc);CHKERRQ(ierr);
1178135c375SStefano Zampini       if (disc) {
1188135c375SStefano Zampini         PetscClassId id;
1198135c375SStefano Zampini         PetscInt     Nc;
1208135c375SStefano Zampini         char         fec[64];
1218135c375SStefano Zampini 
1228135c375SStefano Zampini         ierr = PetscObjectGetClassId(disc, &id);CHKERRQ(ierr);
1238135c375SStefano Zampini         if (id == PETSCFE_CLASSID) {
1248135c375SStefano Zampini           PetscFE            fem = (PetscFE)disc;
1258135c375SStefano Zampini           PetscDualSpace     sp;
1268135c375SStefano Zampini           PetscDualSpaceType spname;
1278135c375SStefano Zampini           PetscInt           order;
1288135c375SStefano Zampini           PetscBool          islag,continuous,H1 = PETSC_TRUE;
1298135c375SStefano Zampini 
1308135c375SStefano Zampini           ierr = PetscFEGetNumComponents(fem,&Nc);CHKERRQ(ierr);
1318135c375SStefano Zampini           ierr = PetscFEGetDualSpace(fem,&sp);CHKERRQ(ierr);
1328135c375SStefano Zampini           ierr = PetscDualSpaceGetType(sp,&spname);CHKERRQ(ierr);
1338135c375SStefano Zampini           ierr = PetscStrcmp(spname,PETSCDUALSPACELAGRANGE,&islag);CHKERRQ(ierr);
1348135c375SStefano Zampini           if (!islag) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Unsupported dual space");
1358135c375SStefano Zampini           ierr = PetscDualSpaceLagrangeGetContinuity(sp,&continuous);CHKERRQ(ierr);
1368135c375SStefano Zampini           ierr = PetscDualSpaceGetOrder(sp,&order);CHKERRQ(ierr);
13728d58a37SPierre Jolivet           if (continuous && order > 0) { /* no support for high-order viz, still have to figure out the numbering */
138bfadf1d8SStefano Zampini             ierr = PetscSNPrintf(fec,64,"FiniteElementCollection: H1_%DD_P1",dim);CHKERRQ(ierr);
1398135c375SStefano Zampini           } else {
14028d58a37SPierre Jolivet             if (!continuous && order) SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Discontinuous space visualization currently unsupported for order %D",order);
1418135c375SStefano Zampini             H1   = PETSC_FALSE;
142bfadf1d8SStefano Zampini             ierr = PetscSNPrintf(fec,64,"FiniteElementCollection: L2_%DD_P%D",dim,order);CHKERRQ(ierr);
1438135c375SStefano Zampini           }
1448135c375SStefano Zampini           ierr = PetscStrallocpy(name,&fieldname[ctx->nf]);CHKERRQ(ierr);
1458135c375SStefano Zampini           bs[ctx->nf]   = Nc;
146bb77a09fSStefano Zampini           dims[ctx->nf] = dim;
1478135c375SStefano Zampini           if (H1) {
1488135c375SStefano Zampini             nlocal[ctx->nf] = Nc * Nv;
1498135c375SStefano Zampini             ierr = PetscStrallocpy(fec,&fec_type[ctx->nf]);CHKERRQ(ierr);
1508135c375SStefano Zampini             ierr = VecCreateSeq(PETSC_COMM_SELF,Nv*Nc,&xfield);CHKERRQ(ierr);
1518135c375SStefano Zampini             for (i=0,cum=0;i<vEnd-vStart;i++) {
1528135c375SStefano Zampini               PetscInt j,off;
1538135c375SStefano Zampini 
1548135c375SStefano Zampini               if (PetscUnlikely(!PetscBTLookup(vown,i))) continue;
1558135c375SStefano Zampini               ierr = PetscSectionGetFieldOffset(s,i+vStart,f,&off);CHKERRQ(ierr);
1568135c375SStefano Zampini               for (j=0;j<Nc;j++) idxs[cum++] = off + j;
1578135c375SStefano Zampini             }
1588135c375SStefano Zampini             ierr = ISCreateGeneral(PetscObjectComm((PetscObject)xlocal),Nv*Nc,idxs,PETSC_USE_POINTER,&isfield);CHKERRQ(ierr);
1598135c375SStefano Zampini           } else {
1608135c375SStefano Zampini             nlocal[ctx->nf] = Nc * totc;
1618135c375SStefano Zampini             ierr = PetscStrallocpy(fec,&fec_type[ctx->nf]);CHKERRQ(ierr);
1628135c375SStefano Zampini             ierr = VecCreateSeq(PETSC_COMM_SELF,Nc*totc,&xfield);CHKERRQ(ierr);
1638135c375SStefano Zampini             for (i=0,cum=0;i<cEnd-cStart;i++) {
1648135c375SStefano Zampini               PetscInt j,off;
1658135c375SStefano Zampini 
1668135c375SStefano Zampini               if (PetscUnlikely(gNum[i] < 0)) continue;
1678135c375SStefano Zampini               ierr = PetscSectionGetFieldOffset(s,i+cStart,f,&off);CHKERRQ(ierr);
1688135c375SStefano Zampini               for (j=0;j<Nc;j++) idxs[cum++] = off + j;
1698135c375SStefano Zampini             }
1708135c375SStefano Zampini             ierr = ISCreateGeneral(PetscObjectComm((PetscObject)xlocal),totc*Nc,idxs,PETSC_USE_POINTER,&isfield);CHKERRQ(ierr);
1718135c375SStefano Zampini           }
1729448b7f1SJunchao Zhang           ierr = VecScatterCreate(xlocal,isfield,xfield,NULL,&ctx->scctx[ctx->nf]);CHKERRQ(ierr);
1738135c375SStefano Zampini           ierr = VecDestroy(&xfield);CHKERRQ(ierr);
1748135c375SStefano Zampini           ierr = ISDestroy(&isfield);CHKERRQ(ierr);
1758135c375SStefano Zampini           ctx->nf++;
1768135c375SStefano Zampini         } else if (id == PETSCFV_CLASSID) {
1778135c375SStefano Zampini           PetscInt c;
1788135c375SStefano Zampini 
1798135c375SStefano Zampini           ierr = PetscFVGetNumComponents((PetscFV)disc,&Nc);CHKERRQ(ierr);
180bfadf1d8SStefano Zampini           ierr = PetscSNPrintf(fec,64,"FiniteElementCollection: L2_%DD_P0",dim);CHKERRQ(ierr);
1818135c375SStefano Zampini           for (c = 0; c < Nc; c++) {
1828135c375SStefano Zampini             char comp[256];
183bfadf1d8SStefano Zampini             ierr = PetscSNPrintf(comp,256,"%s-Comp%D",name,c);CHKERRQ(ierr);
1848135c375SStefano Zampini             ierr = PetscStrallocpy(comp,&fieldname[ctx->nf]);CHKERRQ(ierr);
1858135c375SStefano Zampini             bs[ctx->nf] = 1; /* Does PetscFV support components with different block size? */
1868135c375SStefano Zampini             nlocal[ctx->nf] = totc;
187bb77a09fSStefano Zampini             dims[ctx->nf] = dim;
1888135c375SStefano Zampini             ierr = PetscStrallocpy(fec,&fec_type[ctx->nf]);CHKERRQ(ierr);
1898135c375SStefano Zampini             ierr = VecCreateSeq(PETSC_COMM_SELF,totc,&xfield);CHKERRQ(ierr);
1908135c375SStefano Zampini             for (i=0,cum=0;i<cEnd-cStart;i++) {
1918135c375SStefano Zampini               PetscInt off;
1928135c375SStefano Zampini 
1938135c375SStefano Zampini               if (PetscUnlikely(gNum[i])<0) continue;
1948135c375SStefano Zampini               ierr = PetscSectionGetFieldOffset(s,i+cStart,f,&off);CHKERRQ(ierr);
1958135c375SStefano Zampini               idxs[cum++] = off + c;
1968135c375SStefano Zampini             }
1978135c375SStefano Zampini             ierr = ISCreateGeneral(PetscObjectComm((PetscObject)xlocal),totc,idxs,PETSC_USE_POINTER,&isfield);CHKERRQ(ierr);
1989448b7f1SJunchao Zhang             ierr = VecScatterCreate(xlocal,isfield,xfield,NULL,&ctx->scctx[ctx->nf]);CHKERRQ(ierr);
1998135c375SStefano Zampini             ierr = VecDestroy(&xfield);CHKERRQ(ierr);
2008135c375SStefano Zampini             ierr = ISDestroy(&isfield);CHKERRQ(ierr);
2018135c375SStefano Zampini             ctx->nf++;
2028135c375SStefano Zampini           }
2038135c375SStefano Zampini         } else SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONG,"Unknown discretization type for field %D",f);
2048135c375SStefano Zampini       } else SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Missing discretization for field %D",f);
2058135c375SStefano Zampini     }
2068135c375SStefano Zampini   } else SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Needs a DS attached to the DM");
2078135c375SStefano Zampini   ierr = PetscBTDestroy(&vown);CHKERRQ(ierr);
2088135c375SStefano Zampini   ierr = VecDestroy(&xlocal);CHKERRQ(ierr);
2098135c375SStefano Zampini   ierr = ISRestoreIndices(globalNum,&gNum);CHKERRQ(ierr);
2108135c375SStefano Zampini 
2114cac2994SStefano Zampini   /* create work vectors */
2124cac2994SStefano Zampini   for (f=0;f<ctx->nf;f++) {
2134cac2994SStefano Zampini     ierr = VecCreateMPI(PetscObjectComm((PetscObject)dm),nlocal[f],PETSC_DECIDE,&Ufield[f]);CHKERRQ(ierr);
2144cac2994SStefano Zampini     ierr = PetscObjectSetName((PetscObject)Ufield[f],fieldname[f]);CHKERRQ(ierr);
2154cac2994SStefano Zampini     ierr = VecSetBlockSize(Ufield[f],bs[f]);CHKERRQ(ierr);
2164cac2994SStefano Zampini     ierr = VecSetDM(Ufield[f],dm);CHKERRQ(ierr);
2174cac2994SStefano Zampini   }
2184cac2994SStefano Zampini 
2198135c375SStefano Zampini   /* customize the viewer */
2204cac2994SStefano Zampini   ierr = PetscViewerGLVisSetFields(viewer,ctx->nf,(const char**)fec_type,dims,DMPlexSampleGLVisFields_Private,(PetscObject*)Ufield,ctx,DestroyGLVisViewerCtx_Private);CHKERRQ(ierr);
2218135c375SStefano Zampini   for (f=0;f<ctx->nf;f++) {
2228135c375SStefano Zampini     ierr = PetscFree(fieldname[f]);CHKERRQ(ierr);
2238135c375SStefano Zampini     ierr = PetscFree(fec_type[f]);CHKERRQ(ierr);
2244cac2994SStefano Zampini     ierr = VecDestroy(&Ufield[f]);CHKERRQ(ierr);
2258135c375SStefano Zampini   }
2264cac2994SStefano Zampini   ierr = PetscFree7(fieldname,nlocal,bs,dims,fec_type,idxs,Ufield);CHKERRQ(ierr);
2278135c375SStefano Zampini   PetscFunctionReturn(0);
2288135c375SStefano Zampini }
2298135c375SStefano Zampini 
230b135d7daSStefano Zampini typedef enum {MFEM_POINT=0,MFEM_SEGMENT,MFEM_TRIANGLE,MFEM_SQUARE,MFEM_TETRAHEDRON,MFEM_CUBE,MFEM_PRISM,MFEM_UNDEF} MFEM_cid;
2318135c375SStefano Zampini 
2328135c375SStefano Zampini MFEM_cid mfem_table_cid[4][7]       = { {MFEM_POINT,MFEM_UNDEF,MFEM_UNDEF  ,MFEM_UNDEF   ,MFEM_UNDEF      ,MFEM_UNDEF,MFEM_UNDEF},
2338135c375SStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_UNDEF   ,MFEM_UNDEF      ,MFEM_UNDEF,MFEM_UNDEF},
2348135c375SStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_TRIANGLE,MFEM_SQUARE     ,MFEM_UNDEF,MFEM_UNDEF},
235b135d7daSStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_UNDEF   ,MFEM_TETRAHEDRON,MFEM_PRISM,MFEM_CUBE } };
2368135c375SStefano Zampini 
237b135d7daSStefano 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},
238b135d7daSStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_UNDEF   ,MFEM_UNDEF      ,MFEM_UNDEF,MFEM_PRISM,MFEM_UNDEF,MFEM_UNDEF},
239b135d7daSStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_TRIANGLE,MFEM_SQUARE     ,MFEM_UNDEF,MFEM_PRISM,MFEM_UNDEF,MFEM_UNDEF},
240b135d7daSStefano Zampini                                         {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_UNDEF   ,MFEM_TETRAHEDRON,MFEM_UNDEF,MFEM_PRISM,MFEM_UNDEF,MFEM_CUBE } };
241044a5661SStefano Zampini 
242f86f7544SStefano Zampini static PetscErrorCode DMPlexGetPointMFEMCellID_Internal(DM dm, DMLabel label, PetscInt minl, PetscInt p, PetscInt *mid, PetscInt *cid)
2438135c375SStefano Zampini {
2448135c375SStefano Zampini   DMLabel        dlabel;
245044a5661SStefano Zampini   PetscInt       depth,csize,pdepth,dim;
2468135c375SStefano Zampini   PetscErrorCode ierr;
2478135c375SStefano Zampini 
2488135c375SStefano Zampini   PetscFunctionBegin;
2498135c375SStefano Zampini   ierr = DMPlexGetDepthLabel(dm,&dlabel);CHKERRQ(ierr);
250044a5661SStefano Zampini   ierr = DMLabelGetValue(dlabel,p,&pdepth);CHKERRQ(ierr);
2518135c375SStefano Zampini   ierr = DMPlexGetConeSize(dm,p,&csize);CHKERRQ(ierr);
252044a5661SStefano Zampini   ierr = DMPlexGetDepth(dm,&depth);CHKERRQ(ierr);
253044a5661SStefano Zampini   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
2548135c375SStefano Zampini   if (label) {
2558135c375SStefano Zampini     ierr = DMLabelGetValue(label,p,mid);CHKERRQ(ierr);
256f86f7544SStefano Zampini     *mid = *mid - minl + 1; /* MFEM does not like negative markers */
25777eacf09SStefano Zampini   } else *mid = 1;
258044a5661SStefano Zampini   if (depth >=0 && dim != depth) { /* not interpolated, it assumes cell-vertex mesh */
259044a5661SStefano Zampini #if defined PETSC_USE_DEBUG
260044a5661SStefano Zampini     if (dim < 0 || dim > 3) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Dimension %D",dim);
261044a5661SStefano Zampini     if (csize > 8) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Found cone size %D for point %D",csize,p);
262044a5661SStefano Zampini     if (depth != 1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Found depth %D for point %D. You should interpolate the mesh first",depth,p);
263044a5661SStefano Zampini #endif
264044a5661SStefano Zampini     *cid = mfem_table_cid_unint[dim][csize];
265044a5661SStefano Zampini   } else {
266044a5661SStefano Zampini #if defined PETSC_USE_DEBUG
267044a5661SStefano Zampini     if (csize > 6) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cone size %D for point %D",csize,p);
268044a5661SStefano Zampini     if (pdepth < 0 || pdepth > 3) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Depth %D for point %D",csize,p);
269044a5661SStefano Zampini #endif
270044a5661SStefano Zampini     *cid = mfem_table_cid[pdepth][csize];
271044a5661SStefano Zampini   }
2728135c375SStefano Zampini   PetscFunctionReturn(0);
2738135c375SStefano Zampini }
2748135c375SStefano Zampini 
27596ca5757SLisandro Dalcin static PetscErrorCode DMPlexGetPointMFEMVertexIDs_Internal(DM dm, PetscInt p, PetscSection csec, PetscInt *nv, PetscInt vids[])
2768135c375SStefano Zampini {
277cc0d3ed7SStefano Zampini   PetscInt       dim,sdim,dof = 0,off = 0,i,q,vStart,vEnd,numPoints,*points = NULL;
2788135c375SStefano Zampini   PetscErrorCode ierr;
2798135c375SStefano Zampini 
2808135c375SStefano Zampini   PetscFunctionBegin;
2818135c375SStefano Zampini   ierr = DMPlexGetDepthStratum(dm,0,&vStart,&vEnd);CHKERRQ(ierr);
282cc0d3ed7SStefano Zampini   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
283cc0d3ed7SStefano Zampini   sdim = dim;
284cc0d3ed7SStefano Zampini   if (csec) {
28584f354e3SLisandro Dalcin     PetscInt sStart,sEnd;
28684f354e3SLisandro Dalcin 
287cc0d3ed7SStefano Zampini     ierr = DMGetCoordinateDim(dm,&sdim);CHKERRQ(ierr);
28884f354e3SLisandro Dalcin     ierr = PetscSectionGetChart(csec,&sStart,&sEnd);CHKERRQ(ierr);
289cc0d3ed7SStefano Zampini     ierr = PetscSectionGetOffset(csec,vStart,&off);CHKERRQ(ierr);
290cc0d3ed7SStefano Zampini     off  = off/sdim;
29184f354e3SLisandro Dalcin     if (p >= sStart && p < sEnd) {
292cc0d3ed7SStefano Zampini       ierr = PetscSectionGetDof(csec,p,&dof);CHKERRQ(ierr);
293cc0d3ed7SStefano Zampini     }
29484f354e3SLisandro Dalcin   }
295cc0d3ed7SStefano Zampini   if (!dof) {
2968135c375SStefano Zampini     ierr = DMPlexGetTransitiveClosure(dm,p,PETSC_TRUE,&numPoints,&points);CHKERRQ(ierr);
2978135c375SStefano Zampini     for (i=0,q=0;i<numPoints*2;i+= 2)
2988135c375SStefano Zampini       if ((points[i] >= vStart) && (points[i] < vEnd))
29996ca5757SLisandro Dalcin         vids[q++] = points[i]-vStart+off;
3008135c375SStefano Zampini     ierr = DMPlexRestoreTransitiveClosure(dm,p,PETSC_TRUE,&numPoints,&points);CHKERRQ(ierr);
301cc0d3ed7SStefano Zampini   } else {
302cc0d3ed7SStefano Zampini     ierr = PetscSectionGetOffset(csec,p,&off);CHKERRQ(ierr);
303cc0d3ed7SStefano Zampini     ierr = PetscSectionGetDof(csec,p,&dof);CHKERRQ(ierr);
30496ca5757SLisandro Dalcin     for (q=0;q<dof/sdim;q++) vids[q] = off/sdim + q;
305cc0d3ed7SStefano Zampini   }
3068135c375SStefano Zampini   *nv = q;
3078135c375SStefano Zampini   PetscFunctionReturn(0);
3088135c375SStefano Zampini }
3098135c375SStefano Zampini 
31077eacf09SStefano Zampini /*
31177eacf09SStefano Zampini    ASCII visualization/dump: full support for simplices and tensor product cells. It supports AMR
31277eacf09SStefano Zampini    Higher order meshes are also supported
31377eacf09SStefano Zampini */
3148135c375SStefano Zampini static PetscErrorCode DMPlexView_GLVis_ASCII(DM dm, PetscViewer viewer)
3158135c375SStefano Zampini {
3168135c375SStefano Zampini   DMLabel              label;
3178135c375SStefano Zampini   PetscSection         coordSection,parentSection;
31877eacf09SStefano Zampini   Vec                  coordinates,hovec;
3198135c375SStefano Zampini   const PetscScalar    *array;
320f86f7544SStefano Zampini   PetscInt             bf,p,sdim,dim,depth,novl,minl;
321412e9a14SMatthew G. Knepley   PetscInt             cStart,cEnd,vStart,vEnd,nvert;
3223924b612SStefano Zampini   PetscMPIInt          size;
3233e6c54aaSStefano Zampini   PetscBool            localized,isascii;
32428d58a37SPierre Jolivet   PetscBool            enable_mfem,enable_boundary,enable_ncmesh,view_ovl = PETSC_FALSE;
3253e6c54aaSStefano Zampini   PetscBT              pown,vown;
3268135c375SStefano Zampini   PetscErrorCode       ierr;
3278135c375SStefano Zampini   PetscContainer       glvis_container;
328044a5661SStefano Zampini   PetscBool            cellvertex = PETSC_FALSE, periodic, enabled = PETSC_TRUE;
329f86f7544SStefano Zampini   PetscBool            enable_emark,enable_bmark;
33077eacf09SStefano Zampini   const char           *fmt;
3317bf4dd16SStefano Zampini   char                 emark[64] = "",bmark[64] = "";
3328135c375SStefano Zampini 
3338135c375SStefano Zampini   PetscFunctionBegin;
3348135c375SStefano Zampini   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
3358135c375SStefano Zampini   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
3368135c375SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
3378135c375SStefano Zampini   if (!isascii) SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Viewer must be of type VIEWERASCII");
3383924b612SStefano Zampini   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)viewer),&size);CHKERRQ(ierr);
3393924b612SStefano Zampini   if (size > 1) SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Use single sequential viewers for parallel visualization");
3408135c375SStefano Zampini   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
3418135c375SStefano Zampini 
3428135c375SStefano Zampini   /* get container: determines if a process visualizes is portion of the data or not */
3438135c375SStefano Zampini   ierr = PetscObjectQuery((PetscObject)viewer,"_glvis_info_container",(PetscObject*)&glvis_container);CHKERRQ(ierr);
3448135c375SStefano Zampini   if (!glvis_container) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_PLIB,"Missing GLVis container");
3458135c375SStefano Zampini   {
3468135c375SStefano Zampini     PetscViewerGLVisInfo glvis_info;
3478135c375SStefano Zampini     ierr    = PetscContainerGetPointer(glvis_container,(void**)&glvis_info);CHKERRQ(ierr);
3488135c375SStefano Zampini     enabled = glvis_info->enabled;
34977eacf09SStefano Zampini     fmt     = glvis_info->fmt;
3508135c375SStefano Zampini   }
35121414b21SStefano Zampini 
35221414b21SStefano Zampini   /* Users can attach a coordinate vector to the DM in case they have a higher-order mesh
35321414b21SStefano Zampini      DMPlex does not currently support HO meshes, so there's no API for this */
35421414b21SStefano Zampini   ierr = PetscObjectQuery((PetscObject)dm,"_glvis_mesh_coords",(PetscObject*)&hovec);CHKERRQ(ierr);
35521414b21SStefano Zampini 
356*c3c203b2SStefano Zampini   ierr = DMPlexGetHeightStratum(dm,0,&cStart,&cEnd);CHKERRQ(ierr);
357*c3c203b2SStefano Zampini   ierr = DMPlexGetGhostCellStratum(dm,&p,NULL);CHKERRQ(ierr);
358*c3c203b2SStefano Zampini   if (p >= 0) cEnd = p;
3598135c375SStefano Zampini   ierr = DMPlexGetDepthStratum(dm,0,&vStart,&vEnd);CHKERRQ(ierr);
36077eacf09SStefano Zampini   ierr = DMGetPeriodicity(dm,&periodic,NULL,NULL,NULL);CHKERRQ(ierr);
361cc0d3ed7SStefano Zampini   ierr = DMGetCoordinatesLocalized(dm,&localized);CHKERRQ(ierr);
36221414b21SStefano Zampini   if (periodic && !localized && !hovec) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Coordinates need to be localized");
363cc0d3ed7SStefano Zampini   ierr = DMGetCoordinateSection(dm,&coordSection);CHKERRQ(ierr);
36477eacf09SStefano Zampini   ierr = DMGetCoordinateDim(dm,&sdim);CHKERRQ(ierr);
36577eacf09SStefano Zampini   ierr = DMGetCoordinatesLocal(dm,&coordinates);CHKERRQ(ierr);
36621414b21SStefano Zampini   if (!coordinates && !hovec) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Missing local coordinates vector");
3678135c375SStefano Zampini 
3688135c375SStefano Zampini   /*
3698135c375SStefano Zampini      a couple of sections of the mesh specification are disabled
3703e96840aSStefano 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)
37177eacf09SStefano Zampini        - vertex_parents: used for non-conforming meshes only when we want to use MFEM as a discretization package
3723e6c54aaSStefano Zampini                          and be able to derefine the mesh (MFEM does not currently have to ability to read ncmeshes in parallel)
3738135c375SStefano Zampini   */
3743e96840aSStefano Zampini   enable_boundary = PETSC_FALSE;
3758135c375SStefano Zampini   enable_ncmesh   = PETSC_FALSE;
3763e6c54aaSStefano Zampini   enable_mfem     = PETSC_FALSE;
377f86f7544SStefano Zampini   enable_emark    = PETSC_FALSE;
378f86f7544SStefano Zampini   enable_bmark    = PETSC_FALSE;
3797bf4dd16SStefano Zampini   /* I'm tired of problems with negative values in the markers, disable them */
3808135c375SStefano Zampini   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)dm),((PetscObject)dm)->prefix,"GLVis PetscViewer DMPlex Options","PetscViewer");CHKERRQ(ierr);
3813e6c54aaSStefano Zampini   ierr = PetscOptionsBool("-viewer_glvis_dm_plex_enable_boundary","Enable boundary section in mesh representation",NULL,enable_boundary,&enable_boundary,NULL);CHKERRQ(ierr);
3823e6c54aaSStefano Zampini   ierr = PetscOptionsBool("-viewer_glvis_dm_plex_enable_ncmesh","Enable vertex_parents section in mesh representation (allows derefinement)",NULL,enable_ncmesh,&enable_ncmesh,NULL);CHKERRQ(ierr);
3833e6c54aaSStefano Zampini   ierr = PetscOptionsBool("-viewer_glvis_dm_plex_enable_mfem","Dump a mesh that can be used with MFEM's FiniteElementSpaces",NULL,enable_mfem,&enable_mfem,NULL);CHKERRQ(ierr);
38428d58a37SPierre Jolivet   ierr = PetscOptionsBool("-viewer_glvis_dm_plex_overlap","Include overlap region in local meshes",NULL,view_ovl,&view_ovl,NULL);CHKERRQ(ierr);
385f86f7544SStefano Zampini   ierr = PetscOptionsString("-viewer_glvis_dm_plex_emarker","String for the material id label",NULL,emark,emark,sizeof(emark),&enable_emark);CHKERRQ(ierr);
386f86f7544SStefano Zampini   ierr = PetscOptionsString("-viewer_glvis_dm_plex_bmarker","String for the boundary id label",NULL,bmark,bmark,sizeof(bmark),&enable_bmark);CHKERRQ(ierr);
3878135c375SStefano Zampini   ierr = PetscOptionsEnd();CHKERRQ(ierr);
388f86f7544SStefano Zampini   if (enable_bmark) enable_boundary = PETSC_TRUE;
389f86f7544SStefano Zampini 
3903924b612SStefano Zampini   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)dm),&size);CHKERRQ(ierr);
3913924b612SStefano Zampini   if (enable_ncmesh && size > 1) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Not supported in parallel");
3927e1aca4eSStefano Zampini   ierr = DMPlexGetDepth(dm,&depth);CHKERRQ(ierr);
3937e1aca4eSStefano Zampini   if (enable_boundary && depth >= 0 && dim != depth) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Mesh must be interpolated. "
3947e1aca4eSStefano Zampini                                                              "Alternatively, run with -viewer_glvis_dm_plex_enable_boundary 0");
3957e1aca4eSStefano Zampini   if (enable_ncmesh && depth >= 0 && dim != depth) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Mesh must be interpolated. "
3967e1aca4eSStefano Zampini                                                            "Alternatively, run with -viewer_glvis_dm_plex_enable_ncmesh 0");
397044a5661SStefano Zampini   if (depth >=0 && dim != depth) { /* not interpolated, it assumes cell-vertex mesh */
398044a5661SStefano Zampini     if (depth != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported depth %D. You should interpolate the mesh first",depth);
399044a5661SStefano Zampini     cellvertex = PETSC_TRUE;
400044a5661SStefano Zampini   }
4018135c375SStefano Zampini 
4028135c375SStefano Zampini   /* Identify possible cells in the overlap */
4038135c375SStefano Zampini   novl = 0;
4048135c375SStefano Zampini   pown = NULL;
4053924b612SStefano Zampini   if (size > 1) {
4063e6c54aaSStefano Zampini     IS             globalNum = NULL;
4073e6c54aaSStefano Zampini     const PetscInt *gNum;
4083e6c54aaSStefano Zampini     PetscBool      ovl  = PETSC_FALSE;
4093e6c54aaSStefano Zampini 
410b135d7daSStefano Zampini     ierr = PetscObjectQuery((PetscObject)dm,"_glvis_plex_gnum",(PetscObject*)&globalNum);CHKERRQ(ierr);
411b135d7daSStefano Zampini     if (!globalNum) {
41228d58a37SPierre Jolivet       if (view_ovl) {
41328d58a37SPierre Jolivet         ierr = ISCreateStride(PetscObjectComm((PetscObject)dm),cEnd-cStart,0,1,&globalNum);CHKERRQ(ierr);
41428d58a37SPierre Jolivet       } else {
415b135d7daSStefano Zampini         ierr = DMPlexCreateCellNumbering_Internal(dm,PETSC_TRUE,&globalNum);CHKERRQ(ierr);
41628d58a37SPierre Jolivet       }
417b135d7daSStefano Zampini       ierr = PetscObjectCompose((PetscObject)dm,"_glvis_plex_gnum",(PetscObject)globalNum);CHKERRQ(ierr);
418b135d7daSStefano Zampini       ierr = PetscObjectDereference((PetscObject)globalNum);CHKERRQ(ierr);
419b135d7daSStefano Zampini     }
4208135c375SStefano Zampini     ierr = ISGetIndices(globalNum,&gNum);CHKERRQ(ierr);
4218135c375SStefano Zampini     for (p=cStart; p<cEnd; p++) {
4228135c375SStefano Zampini       if (gNum[p-cStart] < 0) {
4238135c375SStefano Zampini         ovl = PETSC_TRUE;
4248135c375SStefano Zampini         novl++;
4258135c375SStefano Zampini       }
4268135c375SStefano Zampini     }
4278135c375SStefano Zampini     if (ovl) {
4288135c375SStefano Zampini       /* it may happen that pown get not destroyed, if the user closes the window while this function is running.
4298135c375SStefano Zampini          TODO: garbage collector? attach pown to dm?  */
4303e6c54aaSStefano Zampini       ierr = PetscBTCreate(cEnd-cStart,&pown);CHKERRQ(ierr);
4313e6c54aaSStefano Zampini       for (p=cStart; p<cEnd; p++) {
4323e6c54aaSStefano Zampini         if (gNum[p-cStart] < 0) continue;
4333e6c54aaSStefano Zampini         else {
4343e6c54aaSStefano Zampini           ierr = PetscBTSet(pown,p-cStart);CHKERRQ(ierr);
4358135c375SStefano Zampini         }
4368135c375SStefano Zampini       }
4373e6c54aaSStefano Zampini     }
4383e6c54aaSStefano Zampini     ierr = ISRestoreIndices(globalNum,&gNum);CHKERRQ(ierr);
4393e6c54aaSStefano Zampini   }
4408135c375SStefano Zampini 
4413e6c54aaSStefano Zampini   /* return if this process is disabled */
4428135c375SStefano Zampini   if (!enabled) {
4438135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"MFEM mesh v1.1\n");CHKERRQ(ierr);
4448135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\ndimension\n");CHKERRQ(ierr);
4458135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",dim);CHKERRQ(ierr);
4468135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\nelements\n");CHKERRQ(ierr);
4478135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
4488135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\nboundary\n");CHKERRQ(ierr);
4498135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
4508135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\nvertices\n");CHKERRQ(ierr);
4518135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
4528135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",sdim);CHKERRQ(ierr);
4538135c375SStefano Zampini     ierr = PetscBTDestroy(&pown);CHKERRQ(ierr);
4548135c375SStefano Zampini     PetscFunctionReturn(0);
4558135c375SStefano Zampini   }
4568135c375SStefano Zampini 
4573e6c54aaSStefano Zampini   if (enable_mfem) {
4583e6c54aaSStefano Zampini     if (periodic && !hovec) { /* we need to generate a vector of L2 coordinates, as this is how MFEM handles periodic meshes */
4593e6c54aaSStefano Zampini       PetscInt    vpc = 0;
4603e6c54aaSStefano Zampini       char        fec[64];
46196ca5757SLisandro Dalcin       PetscInt    vids[8] = {0,1,2,3,4,5,6,7};
46296ca5757SLisandro Dalcin       PetscInt    hexv[8] = {0,1,3,2,4,5,7,6}, tetv[4] = {0,1,2,3};
46396ca5757SLisandro Dalcin       PetscInt    quadv[8] = {0,1,3,2}, triv[3] = {0,1,2};
46496ca5757SLisandro Dalcin       PetscInt    *dof = NULL;
4653e6c54aaSStefano Zampini       PetscScalar *array,*ptr;
4663e6c54aaSStefano Zampini 
467bfadf1d8SStefano Zampini       ierr = PetscSNPrintf(fec,sizeof(fec),"FiniteElementCollection: L2_T1_%DD_P1",dim);CHKERRQ(ierr);
4683e6c54aaSStefano Zampini       if (cEnd-cStart) {
4693e6c54aaSStefano Zampini         PetscInt fpc;
4703e6c54aaSStefano Zampini 
4713e6c54aaSStefano Zampini         ierr = DMPlexGetConeSize(dm,cStart,&fpc);CHKERRQ(ierr);
4723e6c54aaSStefano Zampini         switch(dim) {
4733e6c54aaSStefano Zampini           case 1:
4743e6c54aaSStefano Zampini             vpc = 2;
4753e6c54aaSStefano Zampini             dof = hexv;
4763e6c54aaSStefano Zampini             break;
4773e6c54aaSStefano Zampini           case 2:
4783e6c54aaSStefano Zampini             switch (fpc) {
4793e6c54aaSStefano Zampini               case 3:
4803e6c54aaSStefano Zampini                 vpc = 3;
481044a5661SStefano Zampini                 dof = triv;
4823e6c54aaSStefano Zampini                 break;
4833e6c54aaSStefano Zampini               case 4:
4843e6c54aaSStefano Zampini                 vpc = 4;
485044a5661SStefano Zampini                 dof = quadv;
4863e6c54aaSStefano Zampini                 break;
4873e6c54aaSStefano Zampini               default:
4883e6c54aaSStefano Zampini                 SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case: faces per cell %D",fpc);
4893e6c54aaSStefano Zampini                 break;
4903e6c54aaSStefano Zampini             }
4913e6c54aaSStefano Zampini             break;
4923e6c54aaSStefano Zampini           case 3:
4933e6c54aaSStefano Zampini             switch (fpc) {
494044a5661SStefano Zampini               case 4: /* TODO: still need to understand L2 ordering for tets */
4953e6c54aaSStefano Zampini                 vpc = 4;
496044a5661SStefano Zampini                 dof = tetv;
497044a5661SStefano Zampini                 SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled tethraedral case");
4983e6c54aaSStefano Zampini                 break;
4993e6c54aaSStefano Zampini               case 6:
500044a5661SStefano Zampini                 if (cellvertex) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case: vertices per cell %D",fpc);
501044a5661SStefano Zampini                 vpc = 8;
502044a5661SStefano Zampini                 dof = hexv;
503044a5661SStefano Zampini                 break;
504044a5661SStefano Zampini               case 8:
505044a5661SStefano Zampini                 if (!cellvertex) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case: faces per cell %D",fpc);
5063e6c54aaSStefano Zampini                 vpc = 8;
5073e6c54aaSStefano Zampini                 dof = hexv;
5083e6c54aaSStefano Zampini                 break;
5093e6c54aaSStefano Zampini               default:
5103e6c54aaSStefano Zampini                 SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case: faces per cell %D",fpc);
5113e6c54aaSStefano Zampini                 break;
5123e6c54aaSStefano Zampini             }
5133e6c54aaSStefano Zampini             break;
5143e6c54aaSStefano Zampini           default:
5153e6c54aaSStefano Zampini             SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Unhandled dim");
5163e6c54aaSStefano Zampini             break;
5173e6c54aaSStefano Zampini         }
51896ca5757SLisandro Dalcin         ierr = DMPlexReorderCell(dm,cStart,vids);CHKERRQ(ierr);
5193e6c54aaSStefano Zampini       }
520044a5661SStefano Zampini       if (!dof) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_PLIB,"Missing dofs");
5213e6c54aaSStefano Zampini       ierr = VecCreateSeq(PETSC_COMM_SELF,(cEnd-cStart-novl)*vpc*sdim,&hovec);CHKERRQ(ierr);
5223e6c54aaSStefano Zampini       ierr = PetscObjectCompose((PetscObject)dm,"_glvis_mesh_coords",(PetscObject)hovec);CHKERRQ(ierr);
5233e6c54aaSStefano Zampini       ierr = PetscObjectDereference((PetscObject)hovec);CHKERRQ(ierr);
5243e6c54aaSStefano Zampini       ierr = PetscObjectSetName((PetscObject)hovec,fec);CHKERRQ(ierr);
5253e6c54aaSStefano Zampini       ierr = VecGetArray(hovec,&array);CHKERRQ(ierr);
5263e6c54aaSStefano Zampini       ptr  = array;
5273e6c54aaSStefano Zampini       for (p=cStart;p<cEnd;p++) {
5283e6c54aaSStefano Zampini         PetscInt    csize,v,d;
5293e6c54aaSStefano Zampini         PetscScalar *vals = NULL;
5303e6c54aaSStefano Zampini 
5313e6c54aaSStefano Zampini         if (PetscUnlikely(pown && !PetscBTLookup(pown,p-cStart))) continue;
5323e6c54aaSStefano Zampini         ierr = DMPlexVecGetClosure(dm,coordSection,coordinates,p,&csize,&vals);CHKERRQ(ierr);
5333e6c54aaSStefano Zampini         if (csize != vpc*sdim && csize != vpc*sdim*2) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unsupported closure size %D (vpc %D, sdim %D)",csize,vpc,sdim);
5343e6c54aaSStefano Zampini         for (v=0;v<vpc;v++) {
5353e6c54aaSStefano Zampini           for (d=0;d<sdim;d++) {
5363e6c54aaSStefano Zampini             ptr[sdim*dof[v]+d] = vals[sdim*vids[v]+d];
5373e6c54aaSStefano Zampini           }
5383e6c54aaSStefano Zampini         }
5393e6c54aaSStefano Zampini         ptr += vpc*sdim;
5403e6c54aaSStefano Zampini         ierr = DMPlexVecRestoreClosure(dm,coordSection,coordinates,p,&csize,&vals);CHKERRQ(ierr);
5413e6c54aaSStefano Zampini       }
5423e6c54aaSStefano Zampini       ierr = VecRestoreArray(hovec,&array);CHKERRQ(ierr);
5433e6c54aaSStefano Zampini     }
5443e6c54aaSStefano Zampini   }
5453e96840aSStefano Zampini   /* if we have high-order coordinates in 3D, we need to specify the boundary */
5463e96840aSStefano Zampini   if (hovec && dim == 3) enable_boundary = PETSC_TRUE;
5473e6c54aaSStefano Zampini 
5488135c375SStefano Zampini   /* header */
5498135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"MFEM mesh v1.1\n");CHKERRQ(ierr);
5508135c375SStefano Zampini 
5518135c375SStefano Zampini   /* topological dimension */
5528135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"\ndimension\n");CHKERRQ(ierr);
5538135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"%D\n",dim);CHKERRQ(ierr);
5548135c375SStefano Zampini 
5558135c375SStefano Zampini   /* elements */
556f86f7544SStefano Zampini   minl = 1;
557f86f7544SStefano Zampini   label = NULL;
558f86f7544SStefano Zampini   if (enable_emark) {
559f86f7544SStefano Zampini     PetscInt lminl = PETSC_MAX_INT;
560f86f7544SStefano Zampini 
5617bf4dd16SStefano Zampini     ierr = DMGetLabel(dm,emark,&label);CHKERRQ(ierr);
562f86f7544SStefano Zampini     if (label) {
563f86f7544SStefano Zampini       IS       vals;
564f86f7544SStefano Zampini       PetscInt ldef;
565f86f7544SStefano Zampini 
566f86f7544SStefano Zampini       ierr = DMLabelGetDefaultValue(label,&ldef);CHKERRQ(ierr);
567f86f7544SStefano Zampini       ierr = DMLabelGetValueIS(label,&vals);CHKERRQ(ierr);
568f86f7544SStefano Zampini       ierr = ISGetMinMax(vals,&lminl,NULL);CHKERRQ(ierr);
569f86f7544SStefano Zampini       ierr = ISDestroy(&vals);CHKERRQ(ierr);
570f86f7544SStefano Zampini       lminl = PetscMin(ldef,lminl);
571f86f7544SStefano Zampini     }
572f86f7544SStefano Zampini     ierr = MPIU_Allreduce(&lminl,&minl,1,MPIU_INT,MPI_MIN,PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
573f86f7544SStefano Zampini     if (minl == PETSC_MAX_INT) minl = 1;
574f86f7544SStefano Zampini   }
5758135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"\nelements\n");CHKERRQ(ierr);
5768135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"%D\n",cEnd-cStart-novl);CHKERRQ(ierr);
5778135c375SStefano Zampini   for (p=cStart;p<cEnd;p++) {
57896ca5757SLisandro Dalcin     PetscInt       vids[8];
57911a4995dSStefano Zampini     PetscInt       i,nv = 0,cid = -1,mid = 1;
5808135c375SStefano Zampini 
5813e6c54aaSStefano Zampini     if (PetscUnlikely(pown && !PetscBTLookup(pown,p-cStart))) continue;
582f86f7544SStefano Zampini     ierr = DMPlexGetPointMFEMCellID_Internal(dm,label,minl,p,&mid,&cid);CHKERRQ(ierr);
58377eacf09SStefano Zampini     ierr = DMPlexGetPointMFEMVertexIDs_Internal(dm,p,(localized && !hovec) ? coordSection : NULL,&nv,vids);CHKERRQ(ierr);
58496ca5757SLisandro Dalcin     ierr = DMPlexReorderCell(dm,p,vids);CHKERRQ(ierr);
5858135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D %D",mid,cid);CHKERRQ(ierr);
5868135c375SStefano Zampini     for (i=0;i<nv;i++) {
58796ca5757SLisandro Dalcin       ierr = PetscViewerASCIIPrintf(viewer," %D",vids[i]);CHKERRQ(ierr);
5888135c375SStefano Zampini     }
5898135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
5908135c375SStefano Zampini   }
5918135c375SStefano Zampini 
592cc0d3ed7SStefano Zampini   /* boundary */
593cc0d3ed7SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"\nboundary\n");CHKERRQ(ierr);
594cc0d3ed7SStefano Zampini   if (!enable_boundary) {
595cc0d3ed7SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
596cc0d3ed7SStefano Zampini   } else {
59777eacf09SStefano Zampini     DMLabel  perLabel;
59877eacf09SStefano Zampini     PetscBT  bfaces;
599b135d7daSStefano Zampini     PetscInt fStart,fEnd,*fcells;
600cc0d3ed7SStefano Zampini 
6018135c375SStefano Zampini     ierr = DMPlexGetHeightStratum(dm,1,&fStart,&fEnd);CHKERRQ(ierr);
60277eacf09SStefano Zampini     ierr = PetscBTCreate(fEnd-fStart,&bfaces);CHKERRQ(ierr);
60377eacf09SStefano Zampini     ierr = DMPlexGetMaxSizes(dm,NULL,&p);CHKERRQ(ierr);
60477eacf09SStefano Zampini     ierr = PetscMalloc1(p,&fcells);CHKERRQ(ierr);
60577eacf09SStefano Zampini     ierr = DMGetLabel(dm,"glvis_periodic_cut",&perLabel);CHKERRQ(ierr);
60677eacf09SStefano Zampini     if (!perLabel && localized) { /* this periodic cut can be moved up to DMPlex setup */
60777eacf09SStefano Zampini       ierr = DMCreateLabel(dm,"glvis_periodic_cut");CHKERRQ(ierr);
60877eacf09SStefano Zampini       ierr = DMGetLabel(dm,"glvis_periodic_cut",&perLabel);CHKERRQ(ierr);
60977eacf09SStefano Zampini       ierr = DMLabelSetDefaultValue(perLabel,1);CHKERRQ(ierr);
61077eacf09SStefano Zampini       for (p=cStart;p<cEnd;p++) {
611*c3c203b2SStefano Zampini         DMPolytopeType cellType;
612*c3c203b2SStefano Zampini         PetscInt       dof;
613b135d7daSStefano Zampini 
614*c3c203b2SStefano Zampini         ierr = DMPlexGetCellType(dm,p,&cellType);CHKERRQ(ierr);
61577eacf09SStefano Zampini         ierr = PetscSectionGetDof(coordSection,p,&dof);CHKERRQ(ierr);
61677eacf09SStefano Zampini         if (dof) {
617*c3c203b2SStefano Zampini           PetscInt    uvpc, v,csize,cellClosureSize,*cellClosure = NULL,*vidxs = NULL;
61877eacf09SStefano Zampini           PetscScalar *vals = NULL;
619*c3c203b2SStefano Zampini 
620*c3c203b2SStefano Zampini           uvpc = DMPolytopeTypeGetNumVertices(cellType);
62177eacf09SStefano Zampini           if (dof%sdim) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Incompatible number of cell dofs %D and space dimension %D",dof,sdim);
622b135d7daSStefano Zampini           if (dof/sdim != uvpc) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_SUP,"Incompatible number of cell dofs %D, vertices %D and space dim %D",dof/sdim,uvpc,sdim);
62377eacf09SStefano Zampini           ierr = DMPlexVecGetClosure(dm,coordSection,coordinates,p,&csize,&vals);CHKERRQ(ierr);
62477eacf09SStefano Zampini           ierr = DMPlexGetTransitiveClosure(dm,p,PETSC_TRUE,&cellClosureSize,&cellClosure);CHKERRQ(ierr);
62577eacf09SStefano Zampini           for (v=0;v<cellClosureSize;v++)
62677eacf09SStefano Zampini             if (cellClosure[2*v] >= vStart && cellClosure[2*v] < vEnd) {
62777eacf09SStefano Zampini               vidxs = cellClosure + 2*v;
62877eacf09SStefano Zampini               break;
62977eacf09SStefano Zampini             }
63077eacf09SStefano Zampini           if (!vidxs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Missing vertices");
631b135d7daSStefano Zampini           for (v=0;v<uvpc;v++) {
63277eacf09SStefano Zampini             PetscInt s;
633044a5661SStefano Zampini 
63477eacf09SStefano Zampini             for (s=0;s<sdim;s++) {
635b135d7daSStefano Zampini               if (PetscAbsScalar(vals[v*sdim+s]-vals[v*sdim+s+uvpc*sdim])>PETSC_MACHINE_EPSILON) {
63677eacf09SStefano Zampini                 ierr = DMLabelSetValue(perLabel,vidxs[2*v],2);CHKERRQ(ierr);
63777eacf09SStefano Zampini               }
63877eacf09SStefano Zampini             }
63977eacf09SStefano Zampini           }
64077eacf09SStefano Zampini           ierr = DMPlexRestoreTransitiveClosure(dm,p,PETSC_TRUE,&cellClosureSize,&cellClosure);CHKERRQ(ierr);
64177eacf09SStefano Zampini           ierr = DMPlexVecRestoreClosure(dm,coordSection,coordinates,p,&csize,&vals);CHKERRQ(ierr);
64277eacf09SStefano Zampini         }
64377eacf09SStefano Zampini       }
64477eacf09SStefano Zampini       if (dim > 1) {
645b135d7daSStefano Zampini         PetscInt eEnd,eStart;
646044a5661SStefano Zampini 
64777eacf09SStefano Zampini         ierr = DMPlexGetDepthStratum(dm,1,&eStart,&eEnd);CHKERRQ(ierr);
64877eacf09SStefano Zampini         for (p=eStart;p<eEnd;p++) {
64977eacf09SStefano Zampini           const PetscInt *cone;
65077eacf09SStefano Zampini           PetscInt       coneSize,i;
65177eacf09SStefano Zampini           PetscBool      ispe = PETSC_TRUE;
65277eacf09SStefano Zampini 
65377eacf09SStefano Zampini           ierr = DMPlexGetCone(dm,p,&cone);CHKERRQ(ierr);
65477eacf09SStefano Zampini           ierr = DMPlexGetConeSize(dm,p,&coneSize);CHKERRQ(ierr);
65577eacf09SStefano Zampini           for (i=0;i<coneSize;i++) {
65677eacf09SStefano Zampini             PetscInt v;
65777eacf09SStefano Zampini 
65877eacf09SStefano Zampini             ierr = DMLabelGetValue(perLabel,cone[i],&v);CHKERRQ(ierr);
65977eacf09SStefano Zampini             ispe = (PetscBool)(ispe && (v==2));
66077eacf09SStefano Zampini           }
66177eacf09SStefano Zampini           if (ispe && coneSize) {
6623e96840aSStefano Zampini             PetscInt       ch, numChildren;
6633e96840aSStefano Zampini             const PetscInt *children;
6643e96840aSStefano Zampini 
66577eacf09SStefano Zampini             ierr = DMLabelSetValue(perLabel,p,2);CHKERRQ(ierr);
6663e96840aSStefano Zampini             ierr = DMPlexGetTreeChildren(dm,p,&numChildren,&children);CHKERRQ(ierr);
6673e96840aSStefano Zampini             for (ch = 0; ch < numChildren; ch++) {
6683e96840aSStefano Zampini               ierr = DMLabelSetValue(perLabel,children[ch],2);CHKERRQ(ierr);
6693e96840aSStefano Zampini             }
67077eacf09SStefano Zampini           }
67177eacf09SStefano Zampini         }
67277eacf09SStefano Zampini         if (dim > 2) {
67377eacf09SStefano Zampini           for (p=fStart;p<fEnd;p++) {
67477eacf09SStefano Zampini             const PetscInt *cone;
67577eacf09SStefano Zampini             PetscInt       coneSize,i;
67677eacf09SStefano Zampini             PetscBool      ispe = PETSC_TRUE;
67777eacf09SStefano Zampini 
67877eacf09SStefano Zampini             ierr = DMPlexGetCone(dm,p,&cone);CHKERRQ(ierr);
67977eacf09SStefano Zampini             ierr = DMPlexGetConeSize(dm,p,&coneSize);CHKERRQ(ierr);
68077eacf09SStefano Zampini             for (i=0;i<coneSize;i++) {
68177eacf09SStefano Zampini               PetscInt v;
68277eacf09SStefano Zampini 
68377eacf09SStefano Zampini               ierr = DMLabelGetValue(perLabel,cone[i],&v);CHKERRQ(ierr);
68477eacf09SStefano Zampini               ispe = (PetscBool)(ispe && (v==2));
68577eacf09SStefano Zampini             }
68677eacf09SStefano Zampini             if (ispe && coneSize) {
6873e96840aSStefano Zampini               PetscInt       ch, numChildren;
6883e96840aSStefano Zampini               const PetscInt *children;
6893e96840aSStefano Zampini 
69077eacf09SStefano Zampini               ierr = DMLabelSetValue(perLabel,p,2);CHKERRQ(ierr);
6913e96840aSStefano Zampini               ierr = DMPlexGetTreeChildren(dm,p,&numChildren,&children);CHKERRQ(ierr);
6923e96840aSStefano Zampini               for (ch = 0; ch < numChildren; ch++) {
6933e96840aSStefano Zampini                 ierr = DMLabelSetValue(perLabel,children[ch],2);CHKERRQ(ierr);
6943e96840aSStefano Zampini               }
69577eacf09SStefano Zampini             }
69677eacf09SStefano Zampini           }
69777eacf09SStefano Zampini         }
69877eacf09SStefano Zampini       }
69977eacf09SStefano Zampini     }
70077eacf09SStefano Zampini     for (p=fStart;p<fEnd;p++) {
70177eacf09SStefano Zampini       const PetscInt *support;
7028135c375SStefano Zampini       PetscInt       supportSize;
70377eacf09SStefano Zampini       PetscBool      isbf = PETSC_FALSE;
7048135c375SStefano Zampini 
7058135c375SStefano Zampini       ierr = DMPlexGetSupportSize(dm,p,&supportSize);CHKERRQ(ierr);
7063e6c54aaSStefano Zampini       if (pown) {
7078135c375SStefano Zampini         PetscBool has_owned = PETSC_FALSE, has_ghost = PETSC_FALSE;
70877eacf09SStefano Zampini         PetscInt  i;
70977eacf09SStefano Zampini 
71077eacf09SStefano Zampini         ierr = DMPlexGetSupport(dm,p,&support);CHKERRQ(ierr);
71177eacf09SStefano Zampini         for (i=0;i<supportSize;i++) {
71277eacf09SStefano Zampini           if (PetscLikely(PetscBTLookup(pown,support[i]-cStart))) has_owned = PETSC_TRUE;
71377eacf09SStefano Zampini           else has_ghost = PETSC_TRUE;
71477eacf09SStefano Zampini         }
71577eacf09SStefano Zampini         isbf = (PetscBool)((supportSize == 1 && has_owned) || (supportSize > 1 && has_owned && has_ghost));
71677eacf09SStefano Zampini       } else {
71777eacf09SStefano Zampini         isbf = (PetscBool)(supportSize == 1);
71877eacf09SStefano Zampini       }
71977eacf09SStefano Zampini       if (!isbf && perLabel) {
72077eacf09SStefano Zampini         const PetscInt *cone;
72177eacf09SStefano Zampini         PetscInt       coneSize,i;
72277eacf09SStefano Zampini 
72377eacf09SStefano Zampini         ierr = DMPlexGetCone(dm,p,&cone);CHKERRQ(ierr);
72477eacf09SStefano Zampini         ierr = DMPlexGetConeSize(dm,p,&coneSize);CHKERRQ(ierr);
72577eacf09SStefano Zampini         isbf = PETSC_TRUE;
72677eacf09SStefano Zampini         for (i=0;i<coneSize;i++) {
72777eacf09SStefano Zampini           PetscInt v,d;
72877eacf09SStefano Zampini 
72977eacf09SStefano Zampini           ierr = DMLabelGetValue(perLabel,cone[i],&v);CHKERRQ(ierr);
73077eacf09SStefano Zampini           ierr = DMLabelGetDefaultValue(perLabel,&d);CHKERRQ(ierr);
73177eacf09SStefano Zampini           isbf = (PetscBool)(isbf && v != d);
73277eacf09SStefano Zampini         }
73377eacf09SStefano Zampini       }
73477eacf09SStefano Zampini       if (isbf) {
73577eacf09SStefano Zampini         ierr = PetscBTSet(bfaces,p-fStart);CHKERRQ(ierr);
73677eacf09SStefano Zampini       }
73777eacf09SStefano Zampini     }
73877eacf09SStefano Zampini     /* count boundary faces */
73977eacf09SStefano Zampini     for (p=fStart,bf=0;p<fEnd;p++) {
74077eacf09SStefano Zampini       if (PetscUnlikely(PetscBTLookup(bfaces,p-fStart))) {
74177eacf09SStefano Zampini         const PetscInt *support;
74277eacf09SStefano Zampini         PetscInt       supportSize,c;
7438135c375SStefano Zampini 
7448135c375SStefano Zampini         ierr = DMPlexGetSupportSize(dm,p,&supportSize);CHKERRQ(ierr);
7458135c375SStefano Zampini         ierr = DMPlexGetSupport(dm,p,&support);CHKERRQ(ierr);
74677eacf09SStefano Zampini         for (c=0;c<supportSize;c++) {
7473e96840aSStefano Zampini           const    PetscInt *cone;
748b135d7daSStefano Zampini           PetscInt cell,cl,coneSize;
7493e96840aSStefano Zampini 
7503e96840aSStefano Zampini           cell = support[c];
7513e96840aSStefano Zampini           if (pown && PetscUnlikely(!PetscBTLookup(pown,cell-cStart))) continue;
7523e96840aSStefano Zampini           ierr = DMPlexGetCone(dm,cell,&cone);CHKERRQ(ierr);
753b135d7daSStefano Zampini           ierr = DMPlexGetConeSize(dm,cell,&coneSize);CHKERRQ(ierr);
754b135d7daSStefano Zampini           for (cl=0;cl<coneSize;cl++) {
7553e96840aSStefano Zampini             if (cone[cl] == p) {
7563e96840aSStefano Zampini               bf += 1;
7573e96840aSStefano Zampini               break;
7588135c375SStefano Zampini             }
75977eacf09SStefano Zampini           }
7603e96840aSStefano Zampini         }
7618135c375SStefano Zampini       }
7628135c375SStefano Zampini     }
763f86f7544SStefano Zampini     minl = 1;
764f86f7544SStefano Zampini     label = NULL;
765f86f7544SStefano Zampini     if (enable_bmark) {
766f86f7544SStefano Zampini       PetscInt lminl = PETSC_MAX_INT;
767f86f7544SStefano Zampini 
7687bf4dd16SStefano Zampini       ierr = DMGetLabel(dm,bmark,&label);CHKERRQ(ierr);
769f86f7544SStefano Zampini       if (label) {
770f86f7544SStefano Zampini         IS       vals;
771f86f7544SStefano Zampini         PetscInt ldef;
772f86f7544SStefano Zampini 
773f86f7544SStefano Zampini         ierr = DMLabelGetDefaultValue(label,&ldef);CHKERRQ(ierr);
774f86f7544SStefano Zampini         ierr = DMLabelGetValueIS(label,&vals);CHKERRQ(ierr);
775f86f7544SStefano Zampini         ierr = ISGetMinMax(vals,&lminl,NULL);CHKERRQ(ierr);
776f86f7544SStefano Zampini         ierr = ISDestroy(&vals);CHKERRQ(ierr);
777f86f7544SStefano Zampini         lminl = PetscMin(ldef,lminl);
778f86f7544SStefano Zampini       }
779f86f7544SStefano Zampini       ierr = MPIU_Allreduce(&lminl,&minl,1,MPIU_INT,MPI_MIN,PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
780f86f7544SStefano Zampini       if (minl == PETSC_MAX_INT) minl = 1;
781f86f7544SStefano Zampini     }
7828135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",bf);CHKERRQ(ierr);
7838135c375SStefano Zampini     for (p=fStart;p<fEnd;p++) {
78477eacf09SStefano Zampini       if (PetscUnlikely(PetscBTLookup(bfaces,p-fStart))) {
7858135c375SStefano Zampini         const PetscInt *support;
78677eacf09SStefano Zampini         PetscInt       supportSize,c,nc = 0;
7878135c375SStefano Zampini 
7888135c375SStefano Zampini         ierr = DMPlexGetSupportSize(dm,p,&supportSize);CHKERRQ(ierr);
7898135c375SStefano Zampini         ierr = DMPlexGetSupport(dm,p,&support);CHKERRQ(ierr);
7903e6c54aaSStefano Zampini         if (pown) {
79177eacf09SStefano Zampini           for (c=0;c<supportSize;c++) {
79277eacf09SStefano Zampini             if (PetscLikely(PetscBTLookup(pown,support[c]-cStart))) {
79377eacf09SStefano Zampini               fcells[nc++] = support[c];
7948135c375SStefano Zampini             }
79577eacf09SStefano Zampini           }
79677eacf09SStefano Zampini         } else for (c=0;c<supportSize;c++) fcells[nc++] = support[c];
79777eacf09SStefano Zampini         for (c=0;c<nc;c++) {
798*c3c203b2SStefano Zampini           const DMPolytopeType *faceTypes;
799*c3c203b2SStefano Zampini           DMPolytopeType       cellType;
800*c3c203b2SStefano Zampini           const PetscInt       *faceSizes,*cone;
801*c3c203b2SStefano Zampini           PetscInt             vids[8],*faces,st,i,coneSize,cell,cl,nv,cid = -1,mid = -1;
8028135c375SStefano Zampini 
80377eacf09SStefano Zampini           cell = fcells[c];
80477eacf09SStefano Zampini           ierr = DMPlexGetCone(dm,cell,&cone);CHKERRQ(ierr);
805b135d7daSStefano Zampini           ierr = DMPlexGetConeSize(dm,cell,&coneSize);CHKERRQ(ierr);
806b135d7daSStefano Zampini           for (cl=0;cl<coneSize;cl++)
80777eacf09SStefano Zampini             if (cone[cl] == p)
8088135c375SStefano Zampini               break;
809b135d7daSStefano Zampini           if (cl == coneSize) continue;
8108135c375SStefano Zampini 
81177eacf09SStefano Zampini           /* face material id and type */
812f86f7544SStefano Zampini           ierr = DMPlexGetPointMFEMCellID_Internal(dm,label,minl,p,&mid,&cid);CHKERRQ(ierr);
8138135c375SStefano Zampini           ierr = PetscViewerASCIIPrintf(viewer,"%D %D",mid,cid);CHKERRQ(ierr);
81477eacf09SStefano Zampini           /* vertex ids */
815*c3c203b2SStefano Zampini           ierr = DMPlexGetCellType(dm,cell,&cellType);CHKERRQ(ierr);
81677eacf09SStefano Zampini           ierr = DMPlexGetPointMFEMVertexIDs_Internal(dm,cell,(localized && !hovec) ? coordSection : NULL,&nv,vids);CHKERRQ(ierr);
817*c3c203b2SStefano Zampini           ierr = DMPlexGetRawFaces_Internal(dm,cellType,vids,NULL,&faceTypes,&faceSizes,(const PetscInt**)&faces);CHKERRQ(ierr);
818*c3c203b2SStefano Zampini           st = 0;
819*c3c203b2SStefano Zampini           for (i=0;i<cl;i++) st += faceSizes[i];
820*c3c203b2SStefano Zampini           ierr = DMPlexInvertCell(faceTypes[cl],faces + st);CHKERRQ(ierr);
821*c3c203b2SStefano Zampini           for (i=0;i<faceSizes[cl];i++) {
822*c3c203b2SStefano Zampini             ierr = PetscViewerASCIIPrintf(viewer," %d",faces[st+i]);CHKERRQ(ierr);
823b135d7daSStefano Zampini           }
8248135c375SStefano Zampini           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
825*c3c203b2SStefano Zampini           ierr = DMPlexRestoreRawFaces_Internal(dm,cellType,vids,NULL,&faceTypes,&faceSizes,(const PetscInt**)&faces);CHKERRQ(ierr);
8263e96840aSStefano Zampini           bf -= 1;
82777eacf09SStefano Zampini         }
8288135c375SStefano Zampini       }
8298135c375SStefano Zampini     }
83077eacf09SStefano Zampini     if (bf) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Remaining boundary faces %D",bf);
83177eacf09SStefano Zampini     ierr = PetscBTDestroy(&bfaces);CHKERRQ(ierr);
83277eacf09SStefano Zampini     ierr = PetscFree(fcells);CHKERRQ(ierr);
8338135c375SStefano Zampini   }
8348135c375SStefano Zampini 
8358135c375SStefano Zampini   /* mark owned vertices */
8363e6c54aaSStefano Zampini   vown = NULL;
8373e6c54aaSStefano Zampini   if (pown) {
8383e6c54aaSStefano Zampini     ierr = PetscBTCreate(vEnd-vStart,&vown);CHKERRQ(ierr);
8398135c375SStefano Zampini     for (p=cStart;p<cEnd;p++) {
8408135c375SStefano Zampini       PetscInt i,closureSize,*closure = NULL;
8418135c375SStefano Zampini 
8423e6c54aaSStefano Zampini       if (PetscUnlikely(!PetscBTLookup(pown,p-cStart))) continue;
8438135c375SStefano Zampini       ierr = DMPlexGetTransitiveClosure(dm,p,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
8448135c375SStefano Zampini       for (i=0;i<closureSize;i++) {
8458135c375SStefano Zampini         const PetscInt pp = closure[2*i];
8468135c375SStefano Zampini 
8478135c375SStefano Zampini         if (pp >= vStart && pp < vEnd) {
8483e6c54aaSStefano Zampini           ierr = PetscBTSet(vown,pp-vStart);CHKERRQ(ierr);
8498135c375SStefano Zampini         }
8508135c375SStefano Zampini       }
8518135c375SStefano Zampini       ierr = DMPlexRestoreTransitiveClosure(dm,p,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
8528135c375SStefano Zampini     }
8538135c375SStefano Zampini   }
8548135c375SStefano Zampini 
8558135c375SStefano Zampini   /* vertex_parents (Non-conforming meshes) */
8568135c375SStefano Zampini   parentSection  = NULL;
8578135c375SStefano Zampini   if (enable_ncmesh) {
8588135c375SStefano Zampini     ierr = DMPlexGetTree(dm,&parentSection,NULL,NULL,NULL,NULL);CHKERRQ(ierr);
8598135c375SStefano Zampini   }
8608135c375SStefano Zampini   if (parentSection) {
8618135c375SStefano Zampini     PetscInt vp,gvp;
8628135c375SStefano Zampini 
8638135c375SStefano Zampini     for (vp=0,p=vStart;p<vEnd;p++) {
8648135c375SStefano Zampini       DMLabel  dlabel;
8658135c375SStefano Zampini       PetscInt parent,depth;
8668135c375SStefano Zampini 
8673e6c54aaSStefano Zampini       if (PetscUnlikely(vown && !PetscBTLookup(vown,p-vStart))) continue;
8688135c375SStefano Zampini       ierr = DMPlexGetDepthLabel(dm,&dlabel);CHKERRQ(ierr);
8698135c375SStefano Zampini       ierr = DMLabelGetValue(dlabel,p,&depth);CHKERRQ(ierr);
8708135c375SStefano Zampini       ierr = DMPlexGetTreeParent(dm,p,&parent,NULL);CHKERRQ(ierr);
8718135c375SStefano Zampini       if (parent != p) vp++;
8728135c375SStefano Zampini     }
8738135c375SStefano Zampini     ierr = MPIU_Allreduce(&vp,&gvp,1,MPIU_INT,MPI_SUM,PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
8748135c375SStefano Zampini     if (gvp) {
8758135c375SStefano Zampini       PetscInt  maxsupp;
8768135c375SStefano Zampini       PetscBool *skip = NULL;
8778135c375SStefano Zampini 
8788135c375SStefano Zampini       ierr = PetscViewerASCIIPrintf(viewer,"\nvertex_parents\n");CHKERRQ(ierr);
8798135c375SStefano Zampini       ierr = PetscViewerASCIIPrintf(viewer,"%D\n",vp);CHKERRQ(ierr);
8808135c375SStefano Zampini       ierr = DMPlexGetMaxSizes(dm,NULL,&maxsupp);CHKERRQ(ierr);
8818135c375SStefano Zampini       ierr = PetscMalloc1(maxsupp,&skip);CHKERRQ(ierr);
8828135c375SStefano Zampini       for (p=vStart;p<vEnd;p++) {
8838135c375SStefano Zampini         DMLabel  dlabel;
8848135c375SStefano Zampini         PetscInt parent;
8858135c375SStefano Zampini 
8863e6c54aaSStefano Zampini         if (PetscUnlikely(vown && !PetscBTLookup(vown,p-vStart))) continue;
8878135c375SStefano Zampini         ierr = DMPlexGetDepthLabel(dm,&dlabel);CHKERRQ(ierr);
8888135c375SStefano Zampini         ierr = DMPlexGetTreeParent(dm,p,&parent,NULL);CHKERRQ(ierr);
8898135c375SStefano Zampini         if (parent != p) {
89096ca5757SLisandro Dalcin           PetscInt       vids[8] = { -1, -1, -1, -1, -1, -1, -1, -1 }; /* silent overzealous clang static analyzer */
8913924b612SStefano Zampini           PetscInt       i,nv,ssize,n,numChildren,depth = -1;
8928135c375SStefano Zampini           const PetscInt *children;
8933924b612SStefano Zampini 
8943924b612SStefano Zampini           ierr = DMPlexGetConeSize(dm,parent,&ssize);CHKERRQ(ierr);
8953924b612SStefano Zampini           switch (ssize) {
8968135c375SStefano Zampini             case 2: /* edge */
8978135c375SStefano Zampini               nv   = 0;
898cc0d3ed7SStefano Zampini               ierr = DMPlexGetPointMFEMVertexIDs_Internal(dm,parent,localized ? coordSection : NULL,&nv,vids);CHKERRQ(ierr);
8998135c375SStefano Zampini               ierr = PetscViewerASCIIPrintf(viewer,"%D",p-vStart);CHKERRQ(ierr);
9008135c375SStefano Zampini               for (i=0;i<nv;i++) {
90196ca5757SLisandro Dalcin                 ierr = PetscViewerASCIIPrintf(viewer," %D",vids[i]);CHKERRQ(ierr);
9028135c375SStefano Zampini               }
9038135c375SStefano Zampini               ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
9048135c375SStefano Zampini               vp--;
9058135c375SStefano Zampini               break;
9068135c375SStefano Zampini             case 4: /* face */
9078135c375SStefano Zampini               ierr = DMPlexGetTreeChildren(dm,parent,&numChildren,&children);CHKERRQ(ierr);
9088135c375SStefano Zampini               for (n=0;n<numChildren;n++) {
9098135c375SStefano Zampini                 ierr = DMLabelGetValue(dlabel,children[n],&depth);CHKERRQ(ierr);
9108135c375SStefano Zampini                 if (!depth) {
9118135c375SStefano Zampini                   const PetscInt *hvsupp,*hesupp,*cone;
9128135c375SStefano Zampini                   PetscInt       hvsuppSize,hesuppSize,coneSize;
913451a39c7SStefano Zampini                   PetscInt       hv = children[n],he = -1,f;
9148135c375SStefano Zampini 
915580bdb30SBarry Smith                   ierr = PetscArrayzero(skip,maxsupp);CHKERRQ(ierr);
9168135c375SStefano Zampini                   ierr = DMPlexGetSupportSize(dm,hv,&hvsuppSize);CHKERRQ(ierr);
9178135c375SStefano Zampini                   ierr = DMPlexGetSupport(dm,hv,&hvsupp);CHKERRQ(ierr);
9188135c375SStefano Zampini                   for (i=0;i<hvsuppSize;i++) {
9198135c375SStefano Zampini                     PetscInt ep;
9208135c375SStefano Zampini                     ierr = DMPlexGetTreeParent(dm,hvsupp[i],&ep,NULL);CHKERRQ(ierr);
9218135c375SStefano Zampini                     if (ep != hvsupp[i]) {
9228135c375SStefano Zampini                       he = hvsupp[i];
9238135c375SStefano Zampini                     } else {
9248135c375SStefano Zampini                       skip[i] = PETSC_TRUE;
9258135c375SStefano Zampini                     }
9268135c375SStefano Zampini                   }
927451a39c7SStefano Zampini                   if (he == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Vertex %D support size %D: hanging edge not found",hv,hvsuppSize);
9288135c375SStefano Zampini                   ierr    = DMPlexGetCone(dm,he,&cone);CHKERRQ(ierr);
92996ca5757SLisandro Dalcin                   vids[0] = (cone[0] == hv) ? cone[1] : cone[0];
9308135c375SStefano Zampini                   ierr    = DMPlexGetSupportSize(dm,he,&hesuppSize);CHKERRQ(ierr);
9318135c375SStefano Zampini                   ierr    = DMPlexGetSupport(dm,he,&hesupp);CHKERRQ(ierr);
9328135c375SStefano Zampini                   for (f=0;f<hesuppSize;f++) {
9338135c375SStefano Zampini                     PetscInt j;
9348135c375SStefano Zampini 
9358135c375SStefano Zampini                     ierr = DMPlexGetCone(dm,hesupp[f],&cone);CHKERRQ(ierr);
9368135c375SStefano Zampini                     ierr = DMPlexGetConeSize(dm,hesupp[f],&coneSize);CHKERRQ(ierr);
9378135c375SStefano Zampini                     for (j=0;j<coneSize;j++) {
9388135c375SStefano Zampini                       PetscInt k;
9398135c375SStefano Zampini                       for (k=0;k<hvsuppSize;k++) {
9408135c375SStefano Zampini                         if (hvsupp[k] == cone[j]) {
9418135c375SStefano Zampini                           skip[k] = PETSC_TRUE;
9428135c375SStefano Zampini                           break;
9438135c375SStefano Zampini                         }
9448135c375SStefano Zampini                       }
9458135c375SStefano Zampini                     }
9468135c375SStefano Zampini                   }
9478135c375SStefano Zampini                   for (i=0;i<hvsuppSize;i++) {
9488135c375SStefano Zampini                     if (!skip[i]) {
9498135c375SStefano Zampini                       ierr = DMPlexGetCone(dm,hvsupp[i],&cone);CHKERRQ(ierr);
95096ca5757SLisandro Dalcin                       vids[1] = (cone[0] == hv) ? cone[1] : cone[0];
9518135c375SStefano Zampini                     }
9528135c375SStefano Zampini                   }
9538135c375SStefano Zampini                   ierr = PetscViewerASCIIPrintf(viewer,"%D",hv-vStart);CHKERRQ(ierr);
9548135c375SStefano Zampini                   for (i=0;i<2;i++) {
95596ca5757SLisandro Dalcin                     ierr = PetscViewerASCIIPrintf(viewer," %D",vids[i]-vStart);CHKERRQ(ierr);
9568135c375SStefano Zampini                   }
9578135c375SStefano Zampini                   ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
9588135c375SStefano Zampini                   vp--;
9598135c375SStefano Zampini                 }
9608135c375SStefano Zampini               }
9618135c375SStefano Zampini               break;
9628135c375SStefano Zampini             default:
9633924b612SStefano Zampini               SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Don't know how to deal with support size %D",ssize);
9648135c375SStefano Zampini           }
9658135c375SStefano Zampini         }
9668135c375SStefano Zampini       }
9678135c375SStefano Zampini       ierr = PetscFree(skip);CHKERRQ(ierr);
9688135c375SStefano Zampini     }
9698135c375SStefano Zampini     if (vp) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected %D hanging vertices",vp);
9708135c375SStefano Zampini   }
9718135c375SStefano Zampini   ierr = PetscBTDestroy(&pown);CHKERRQ(ierr);
9723e6c54aaSStefano Zampini   ierr = PetscBTDestroy(&vown);CHKERRQ(ierr);
9738135c375SStefano Zampini 
9748135c375SStefano Zampini   /* vertices */
97577eacf09SStefano Zampini   if (hovec) { /* higher-order meshes */
97677eacf09SStefano Zampini     const char *fec;
9770286d493SLisandro Dalcin     PetscInt   i,n,s;
97877eacf09SStefano Zampini 
97977eacf09SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\nvertices\n");CHKERRQ(ierr);
98077eacf09SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",vEnd-vStart);CHKERRQ(ierr);
98177eacf09SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"nodes\n");CHKERRQ(ierr);
98277eacf09SStefano Zampini     ierr = PetscObjectGetName((PetscObject)hovec,&fec);CHKERRQ(ierr);
98377eacf09SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"FiniteElementSpace\n");CHKERRQ(ierr);
9843e7cab22SLisandro Dalcin     ierr = PetscViewerASCIIPrintf(viewer,"%s\n",fec);CHKERRQ(ierr);
98577eacf09SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"VDim: %D\n",sdim);CHKERRQ(ierr);
98677eacf09SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"Ordering: 1\n\n");CHKERRQ(ierr); /*Ordering::byVDIM*/
98777eacf09SStefano Zampini     ierr = VecGetArrayRead(hovec,&array);CHKERRQ(ierr);
98877eacf09SStefano Zampini     ierr = VecGetLocalSize(hovec,&n);CHKERRQ(ierr);
98977eacf09SStefano Zampini     if (n%sdim) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_USER,"Size of local coordinate vector %D incompatible with space dimension %D",n,sdim);
99077eacf09SStefano Zampini     for (i=0;i<n/sdim;i++) {
99177eacf09SStefano Zampini       for (s=0;s<sdim;s++) {
99277eacf09SStefano Zampini         ierr = PetscViewerASCIIPrintf(viewer,fmt,PetscRealPart(array[i*sdim+s]));CHKERRQ(ierr);
99377eacf09SStefano Zampini       }
99477eacf09SStefano Zampini       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
99577eacf09SStefano Zampini     }
99677eacf09SStefano Zampini     ierr = VecRestoreArrayRead(hovec,&array);CHKERRQ(ierr);
99777eacf09SStefano Zampini   } else {
998cc0d3ed7SStefano Zampini     ierr = VecGetLocalSize(coordinates,&nvert);CHKERRQ(ierr);
9998135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\nvertices\n");CHKERRQ(ierr);
1000cc0d3ed7SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",nvert/sdim);CHKERRQ(ierr);
10018135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",sdim);CHKERRQ(ierr);
10028135c375SStefano Zampini     ierr = VecGetArrayRead(coordinates,&array);CHKERRQ(ierr);
1003cc0d3ed7SStefano Zampini     for (p=0;p<nvert/sdim;p++) {
1004cc0d3ed7SStefano Zampini       PetscInt s;
1005cc0d3ed7SStefano Zampini       for (s=0;s<sdim;s++) {
10063e96840aSStefano Zampini         PetscReal v = PetscRealPart(array[p*sdim+s]);
10073e96840aSStefano Zampini 
10083e96840aSStefano Zampini         ierr = PetscViewerASCIIPrintf(viewer,fmt,PetscIsInfOrNanReal(v) ? 0.0 : v);CHKERRQ(ierr);
10098135c375SStefano Zampini       }
1010cc0d3ed7SStefano Zampini       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
10118135c375SStefano Zampini     }
10128135c375SStefano Zampini     ierr = VecRestoreArrayRead(coordinates,&array);CHKERRQ(ierr);
101377eacf09SStefano Zampini   }
10148135c375SStefano Zampini   PetscFunctionReturn(0);
10158135c375SStefano Zampini }
10168135c375SStefano Zampini 
10170286d493SLisandro Dalcin PetscErrorCode DMPlexView_GLVis(DM dm, PetscViewer viewer)
10188135c375SStefano Zampini {
10198135c375SStefano Zampini   PetscErrorCode ierr;
10208135c375SStefano Zampini   PetscFunctionBegin;
10210286d493SLisandro Dalcin   ierr = DMView_GLVis(dm,viewer,DMPlexView_GLVis_ASCII);CHKERRQ(ierr);
10228135c375SStefano Zampini   PetscFunctionReturn(0);
10238135c375SStefano Zampini }
1024