xref: /petsc/src/dm/impls/plex/plexglvis.c (revision bb77a09f7f3252c5ccd8b7f0ad10e118587d6fe7)
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);
258135c375SStefano Zampini   ierr = PetscFree(vctx);
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;
478135c375SStefano Zampini   Vec            xlocal,xfield;
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;
53*bb77a09fSStefano Zampini   PetscInt       *nlocal,*bs,*idxs,*dims;
548135c375SStefano Zampini   PetscInt       f,maxfields,nfields,c,totc,totdofs,Nv,cum,i;
5511a4995dSStefano Zampini   PetscInt       dim,cStart,cEnd,cEndInterior,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);
648135c375SStefano Zampini   ierr = DMPlexGetHybridBounds(dm, &cEndInterior, NULL, NULL, NULL);CHKERRQ(ierr);
658135c375SStefano Zampini   cEnd = cEndInterior < 0 ? cEnd : cEndInterior;
668135c375SStefano Zampini   ierr = DMPlexGetCellNumbering(dm,&globalNum);CHKERRQ(ierr);
678135c375SStefano Zampini   ierr = ISGetIndices(globalNum,&gNum);CHKERRQ(ierr);
688135c375SStefano Zampini   ierr = PetscBTCreate(vEnd-vStart,&vown);CHKERRQ(ierr);
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++;
748135c375SStefano Zampini       ierr = DMPlexGetTransitiveClosure(dm,c,PETSC_TRUE,&numPoints,&points);CHKERRQ(ierr);
758135c375SStefano Zampini       for (i=0;i<numPoints*2;i+= 2) {
768135c375SStefano Zampini         if ((points[i] >= vStart) && (points[i] < vEnd)) {
778135c375SStefano Zampini           ierr = PetscBTSet(vown,points[i]-vStart);CHKERRQ(ierr);
788135c375SStefano Zampini         }
798135c375SStefano Zampini       }
808135c375SStefano Zampini       ierr = DMPlexRestoreTransitiveClosure(dm,c,PETSC_TRUE,&numPoints,&points);CHKERRQ(ierr);
818135c375SStefano Zampini     }
828135c375SStefano Zampini   }
838135c375SStefano Zampini   for (f=0,Nv=0;f<vEnd-vStart;f++) if (PetscBTLookup(vown,f)) Nv++;
848135c375SStefano Zampini 
858135c375SStefano Zampini   ierr = DMCreateLocalVector(dm,&xlocal);CHKERRQ(ierr);
868135c375SStefano Zampini   ierr = VecGetLocalSize(xlocal,&totdofs);CHKERRQ(ierr);
878135c375SStefano Zampini   ierr = DMGetDefaultSection(dm,&s);CHKERRQ(ierr);
888135c375SStefano Zampini   ierr = PetscSectionGetNumFields(s,&nfields);CHKERRQ(ierr);
898135c375SStefano Zampini   for (f=0,maxfields=0;f<nfields;f++) {
908135c375SStefano Zampini     PetscInt bs;
918135c375SStefano Zampini 
928135c375SStefano Zampini     ierr = PetscSectionGetFieldComponents(s,f,&bs);CHKERRQ(ierr);
938135c375SStefano Zampini     maxfields += bs;
948135c375SStefano Zampini   }
95*bb77a09fSStefano Zampini   ierr = PetscCalloc6(maxfields,&fieldname,maxfields,&nlocal,maxfields,&bs,maxfields,&dims,maxfields,&fec_type,totdofs,&idxs);CHKERRQ(ierr);
968135c375SStefano Zampini   ierr = PetscNew(&ctx);CHKERRQ(ierr);
978135c375SStefano Zampini   ierr = PetscCalloc1(maxfields,&ctx->scctx);CHKERRQ(ierr);
988135c375SStefano Zampini   ierr = DMGetDS(dm,&ds);CHKERRQ(ierr);
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 
1068135c375SStefano Zampini       ierr = PetscSectionGetFieldName(s,f,&fname);CHKERRQ(ierr);
1078135c375SStefano Zampini       ierr = PetscStrlen(fname,&len);CHKERRQ(ierr);
1088135c375SStefano Zampini       if (len) {
1098135c375SStefano Zampini         ierr = PetscStrcpy(name,fname);CHKERRQ(ierr);
1108135c375SStefano Zampini       } else {
1118135c375SStefano Zampini         ierr = PetscSNPrintf(name,256,"Field%d",f);CHKERRQ(ierr);
1128135c375SStefano Zampini       }
1138135c375SStefano Zampini       ierr = PetscDSGetDiscretization(ds,f,&disc);CHKERRQ(ierr);
1148135c375SStefano Zampini       if (disc) {
1158135c375SStefano Zampini         PetscClassId id;
1168135c375SStefano Zampini         PetscInt     Nc;
1178135c375SStefano Zampini         char         fec[64];
1188135c375SStefano Zampini 
1198135c375SStefano Zampini         ierr = PetscObjectGetClassId(disc, &id);CHKERRQ(ierr);
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 
1278135c375SStefano Zampini           ierr = PetscFEGetNumComponents(fem,&Nc);CHKERRQ(ierr);
1288135c375SStefano Zampini           ierr = PetscFEGetDualSpace(fem,&sp);CHKERRQ(ierr);
1298135c375SStefano Zampini           ierr = PetscDualSpaceGetType(sp,&spname);CHKERRQ(ierr);
1308135c375SStefano Zampini           ierr = PetscStrcmp(spname,PETSCDUALSPACELAGRANGE,&islag);CHKERRQ(ierr);
1318135c375SStefano Zampini           if (!islag) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Unsupported dual space");
1328135c375SStefano Zampini           ierr = PetscDualSpaceLagrangeGetContinuity(sp,&continuous);CHKERRQ(ierr);
1338135c375SStefano Zampini           if (!continuous) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Discontinuous space visualization currently unsupported");
1348135c375SStefano Zampini           ierr = PetscDualSpaceGetOrder(sp,&order);CHKERRQ(ierr);
1358135c375SStefano Zampini           if (continuous && order > 0) {
1368135c375SStefano Zampini             ierr = PetscSNPrintf(fec,64,"FiniteElementCollection: H1_%dD_P1",dim);CHKERRQ(ierr);
1378135c375SStefano Zampini           } else {
1388135c375SStefano Zampini             H1   = PETSC_FALSE;
1398135c375SStefano Zampini             ierr = PetscSNPrintf(fec,64,"FiniteElementCollection: L2_%dD_P%d",dim,order);CHKERRQ(ierr);
1408135c375SStefano Zampini           }
1418135c375SStefano Zampini           ierr = PetscStrallocpy(name,&fieldname[ctx->nf]);CHKERRQ(ierr);
1428135c375SStefano Zampini           bs[ctx->nf]   = Nc;
143*bb77a09fSStefano Zampini           dims[ctx->nf] = dim;
1448135c375SStefano Zampini           if (H1) {
1458135c375SStefano Zampini             nlocal[ctx->nf] = Nc * Nv;
1468135c375SStefano Zampini             ierr = PetscStrallocpy(fec,&fec_type[ctx->nf]);CHKERRQ(ierr);
1478135c375SStefano Zampini             ierr = VecCreateSeq(PETSC_COMM_SELF,Nv*Nc,&xfield);CHKERRQ(ierr);
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;
1528135c375SStefano Zampini               ierr = PetscSectionGetFieldOffset(s,i+vStart,f,&off);CHKERRQ(ierr);
1538135c375SStefano Zampini               for (j=0;j<Nc;j++) idxs[cum++] = off + j;
1548135c375SStefano Zampini             }
1558135c375SStefano Zampini             ierr = ISCreateGeneral(PetscObjectComm((PetscObject)xlocal),Nv*Nc,idxs,PETSC_USE_POINTER,&isfield);CHKERRQ(ierr);
1568135c375SStefano Zampini           } else {
1578135c375SStefano Zampini             nlocal[ctx->nf] = Nc * totc;
1588135c375SStefano Zampini             ierr = PetscStrallocpy(fec,&fec_type[ctx->nf]);CHKERRQ(ierr);
1598135c375SStefano Zampini             ierr = VecCreateSeq(PETSC_COMM_SELF,Nc*totc,&xfield);CHKERRQ(ierr);
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;
1648135c375SStefano Zampini               ierr = PetscSectionGetFieldOffset(s,i+cStart,f,&off);CHKERRQ(ierr);
1658135c375SStefano Zampini               for (j=0;j<Nc;j++) idxs[cum++] = off + j;
1668135c375SStefano Zampini             }
1678135c375SStefano Zampini             ierr = ISCreateGeneral(PetscObjectComm((PetscObject)xlocal),totc*Nc,idxs,PETSC_USE_POINTER,&isfield);CHKERRQ(ierr);
1688135c375SStefano Zampini           }
1698135c375SStefano Zampini           ierr = VecScatterCreate(xlocal,isfield,xfield,NULL,&ctx->scctx[ctx->nf]);CHKERRQ(ierr);
1708135c375SStefano Zampini           ierr = VecDestroy(&xfield);CHKERRQ(ierr);
1718135c375SStefano Zampini           ierr = ISDestroy(&isfield);CHKERRQ(ierr);
1728135c375SStefano Zampini           ctx->nf++;
1738135c375SStefano Zampini         } else if (id == PETSCFV_CLASSID) {
1748135c375SStefano Zampini           PetscInt c;
1758135c375SStefano Zampini 
1768135c375SStefano Zampini           ierr = PetscFVGetNumComponents((PetscFV)disc,&Nc);CHKERRQ(ierr);
1778135c375SStefano Zampini           ierr = PetscSNPrintf(fec,64,"FiniteElementCollection: L2_%dD_P0",dim);CHKERRQ(ierr);
1788135c375SStefano Zampini           for (c = 0; c < Nc; c++) {
1798135c375SStefano Zampini             char comp[256];
1808135c375SStefano Zampini             ierr = PetscSNPrintf(comp,256,"%s-Comp%d",name,c);CHKERRQ(ierr);
1818135c375SStefano Zampini             ierr = PetscStrallocpy(comp,&fieldname[ctx->nf]);CHKERRQ(ierr);
1828135c375SStefano Zampini             bs[ctx->nf] = 1; /* Does PetscFV support components with different block size? */
1838135c375SStefano Zampini             nlocal[ctx->nf] = totc;
184*bb77a09fSStefano Zampini             dims[ctx->nf] = dim;
1858135c375SStefano Zampini             ierr = PetscStrallocpy(fec,&fec_type[ctx->nf]);CHKERRQ(ierr);
1868135c375SStefano Zampini             ierr = VecCreateSeq(PETSC_COMM_SELF,totc,&xfield);CHKERRQ(ierr);
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;
1918135c375SStefano Zampini               ierr = PetscSectionGetFieldOffset(s,i+cStart,f,&off);CHKERRQ(ierr);
1928135c375SStefano Zampini               idxs[cum++] = off + c;
1938135c375SStefano Zampini             }
1948135c375SStefano Zampini             ierr = ISCreateGeneral(PetscObjectComm((PetscObject)xlocal),totc,idxs,PETSC_USE_POINTER,&isfield);CHKERRQ(ierr);
1958135c375SStefano Zampini             ierr = VecScatterCreate(xlocal,isfield,xfield,NULL,&ctx->scctx[ctx->nf]);CHKERRQ(ierr);
1968135c375SStefano Zampini             ierr = VecDestroy(&xfield);CHKERRQ(ierr);
1978135c375SStefano Zampini             ierr = ISDestroy(&isfield);CHKERRQ(ierr);
1988135c375SStefano Zampini             ctx->nf++;
1998135c375SStefano Zampini           }
2008135c375SStefano Zampini         } else SETERRQ1(PetscObjectComm((PetscObject)dm),PETSC_ERR_ARG_WRONG,"Unknown discretization type for field %D",f);
2018135c375SStefano Zampini       } else SETERRQ1(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");
2048135c375SStefano Zampini   ierr = PetscBTDestroy(&vown);CHKERRQ(ierr);
2058135c375SStefano Zampini   ierr = VecDestroy(&xlocal);CHKERRQ(ierr);
2068135c375SStefano Zampini   ierr = ISRestoreIndices(globalNum,&gNum);CHKERRQ(ierr);
2078135c375SStefano Zampini 
2088135c375SStefano Zampini   /* customize the viewer */
209*bb77a09fSStefano Zampini   ierr = PetscViewerGLVisSetFields(viewer,ctx->nf,(const char**)fieldname,(const char**)fec_type,nlocal,bs,dims,DMPlexSampleGLVisFields_Private,ctx,DestroyGLVisViewerCtx_Private);CHKERRQ(ierr);
2108135c375SStefano Zampini   for (f=0;f<ctx->nf;f++) {
2118135c375SStefano Zampini     ierr = PetscFree(fieldname[f]);CHKERRQ(ierr);
2128135c375SStefano Zampini     ierr = PetscFree(fec_type[f]);CHKERRQ(ierr);
2138135c375SStefano Zampini   }
214*bb77a09fSStefano Zampini   ierr = PetscFree6(fieldname,nlocal,bs,dims,fec_type,idxs);CHKERRQ(ierr);
2158135c375SStefano Zampini   PetscFunctionReturn(0);
2168135c375SStefano Zampini }
2178135c375SStefano Zampini 
2188135c375SStefano Zampini typedef enum {MFEM_POINT=0,MFEM_SEGMENT,MFEM_TRIANGLE,MFEM_SQUARE,MFEM_TETRAHEDRON,MFEM_CUBE,MFEM_UNDEF} MFEM_cid;
2198135c375SStefano Zampini 
2208135c375SStefano Zampini MFEM_cid mfem_table_cid[4][7] = { {MFEM_POINT,MFEM_UNDEF,MFEM_UNDEF  ,MFEM_UNDEF   ,MFEM_UNDEF      ,MFEM_UNDEF, MFEM_UNDEF},
2218135c375SStefano Zampini                                   {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_UNDEF   ,MFEM_UNDEF      ,MFEM_UNDEF, MFEM_UNDEF},
2228135c375SStefano Zampini                                   {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_TRIANGLE,MFEM_SQUARE     ,MFEM_UNDEF, MFEM_UNDEF},
2238135c375SStefano Zampini                                   {MFEM_POINT,MFEM_UNDEF,MFEM_SEGMENT,MFEM_UNDEF   ,MFEM_TETRAHEDRON,MFEM_UNDEF, MFEM_CUBE } };
2248135c375SStefano Zampini 
2258135c375SStefano Zampini #undef __FUNCT__
2268135c375SStefano Zampini #define __FUNCT__ "DMPlexGetPointMFEMCellID_Internal"
2278135c375SStefano Zampini PETSC_STATIC_INLINE PetscErrorCode DMPlexGetPointMFEMCellID_Internal(DM dm, DMLabel label, PetscInt p, PetscInt *mid, PetscInt *cid)
2288135c375SStefano Zampini {
2298135c375SStefano Zampini   DMLabel        dlabel;
2308135c375SStefano Zampini   PetscInt       depth,csize;
2318135c375SStefano Zampini   PetscErrorCode ierr;
2328135c375SStefano Zampini 
2338135c375SStefano Zampini   PetscFunctionBegin;
2348135c375SStefano Zampini   ierr = DMPlexGetDepthLabel(dm,&dlabel);CHKERRQ(ierr);
2358135c375SStefano Zampini   ierr = DMLabelGetValue(dlabel,p,&depth);CHKERRQ(ierr);
2368135c375SStefano Zampini   ierr = DMPlexGetConeSize(dm,p,&csize);CHKERRQ(ierr);
2378135c375SStefano Zampini   if (label) {
2388135c375SStefano Zampini     ierr = DMLabelGetValue(label,p,mid);CHKERRQ(ierr);
2398135c375SStefano Zampini   } else {
2408135c375SStefano Zampini     *mid = 1;
2418135c375SStefano Zampini   }
2428135c375SStefano Zampini   *cid = mfem_table_cid[depth][csize];
2438135c375SStefano Zampini   PetscFunctionReturn(0);
2448135c375SStefano Zampini }
2458135c375SStefano Zampini 
2468135c375SStefano Zampini #undef __FUNCT__
2478135c375SStefano Zampini #define __FUNCT__ "DMPlexGetPointMFEMVertexIDs_Internal"
24811a4995dSStefano Zampini PETSC_STATIC_INLINE PetscErrorCode DMPlexGetPointMFEMVertexIDs_Internal(DM dm, PetscInt p, PetscInt *nv, int vids[])
2498135c375SStefano Zampini {
2508135c375SStefano Zampini   PetscInt       dim,i,q,vStart,vEnd,numPoints,*points = NULL;
2518135c375SStefano Zampini   PetscErrorCode ierr;
2528135c375SStefano Zampini 
2538135c375SStefano Zampini   PetscFunctionBegin;
2548135c375SStefano Zampini   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
2558135c375SStefano Zampini   ierr = DMPlexGetDepthStratum(dm,0,&vStart,&vEnd);CHKERRQ(ierr);
2568135c375SStefano Zampini   ierr = DMPlexGetTransitiveClosure(dm,p,PETSC_TRUE,&numPoints,&points);CHKERRQ(ierr);
2578135c375SStefano Zampini   for (i=0,q=0;i<numPoints*2;i+= 2)
2588135c375SStefano Zampini     if ((points[i] >= vStart) && (points[i] < vEnd))
2598135c375SStefano Zampini       vids[q++] = points[i]-vStart;
2608135c375SStefano Zampini   ierr = DMPlexRestoreTransitiveClosure(dm,p,PETSC_TRUE,&numPoints,&points);CHKERRQ(ierr);
2618135c375SStefano Zampini   ierr = DMPlexInvertCell(dim,q,vids);CHKERRQ(ierr);
2628135c375SStefano Zampini   *nv  = q;
2638135c375SStefano Zampini   PetscFunctionReturn(0);
2648135c375SStefano Zampini }
2658135c375SStefano Zampini 
2668135c375SStefano Zampini /* ASCII visualization/dump: full support for simplices and tensor product cells. It supports AMR */
2678135c375SStefano Zampini static PetscErrorCode DMPlexView_GLVis_ASCII(DM dm, PetscViewer viewer)
2688135c375SStefano Zampini {
2698135c375SStefano Zampini   DM                   cdm;
2708135c375SStefano Zampini   DMLabel              label;
2718135c375SStefano Zampini   PetscSection         coordSection,parentSection;
2728135c375SStefano Zampini   Vec                  coordinates;
2738135c375SStefano Zampini   IS                   globalNum = NULL;
2748135c375SStefano Zampini   const PetscScalar    *array;
2758135c375SStefano Zampini   const PetscInt       *gNum;
2768135c375SStefano Zampini   PetscInt             bf,p,sdim,dim,depth,novl;
2778135c375SStefano Zampini   PetscInt             cStart,cEnd,cEndInterior,fStart,fEnd,fEndInterior,vStart,vEnd;
2788135c375SStefano Zampini   PetscMPIInt          commsize;
2798135c375SStefano Zampini   PetscBool            ovl,isascii,enable_boundary,enable_ncmesh;
2808135c375SStefano Zampini   PetscBT              pown;
2818135c375SStefano Zampini   PetscErrorCode       ierr;
28211a4995dSStefano Zampini   PetscInt             *faces = NULL,fpc = 0,vpf = 0;
2838135c375SStefano Zampini   PetscInt             fv1[]     = {0,1},
2848135c375SStefano Zampini                        fv2tri[]  = {0,1,
2858135c375SStefano Zampini                                     1,2,
2868135c375SStefano Zampini                                     2,0},
2878135c375SStefano Zampini                        fv2quad[] = {0,1,
2888135c375SStefano Zampini                                     1,2,
2898135c375SStefano Zampini                                     2,3,
2908135c375SStefano Zampini                                     3,0},
2918135c375SStefano Zampini                        fv3tet[]  = {0,1,2,
2928135c375SStefano Zampini                                     0,3,1,
2938135c375SStefano Zampini                                     0,2,3,
2948135c375SStefano Zampini                                     2,1,3},
2958135c375SStefano Zampini                        fv3hex[]  = {0,1,2,3,
2968135c375SStefano Zampini                                  4,5,6,7,
2978135c375SStefano Zampini                                  0,3,5,4,
2988135c375SStefano Zampini                                  2,1,7,6,
2998135c375SStefano Zampini                                  3,2,6,5,
3008135c375SStefano Zampini                                  0,4,7,1};
3018135c375SStefano Zampini   PetscContainer       glvis_container;
3028135c375SStefano Zampini   PetscBool            enabled = PETSC_TRUE;
3038135c375SStefano Zampini 
3048135c375SStefano Zampini   PetscFunctionBegin;
3058135c375SStefano Zampini   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
3068135c375SStefano Zampini   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
3078135c375SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
3088135c375SStefano Zampini   if (!isascii) SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Viewer must be of type VIEWERASCII");
3098135c375SStefano Zampini   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)viewer),&commsize);CHKERRQ(ierr);
3108135c375SStefano Zampini   if (commsize > 1) SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Use single sequential viewers for parallel visualization");
3118135c375SStefano Zampini   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
3128135c375SStefano Zampini   ierr = DMPlexGetDepth(dm,&depth);CHKERRQ(ierr);
3138135c375SStefano Zampini   if (depth >= 0 && dim != depth) SETERRQ(PetscObjectComm((PetscObject)dm), PETSC_ERR_ARG_WRONG, "Mesh must be interpolated");
3148135c375SStefano Zampini 
3158135c375SStefano Zampini   /* get container: determines if a process visualizes is portion of the data or not */
3168135c375SStefano Zampini   ierr = PetscObjectQuery((PetscObject)viewer,"_glvis_info_container",(PetscObject*)&glvis_container);CHKERRQ(ierr);
3178135c375SStefano Zampini   if (!glvis_container) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_PLIB,"Missing GLVis container");
3188135c375SStefano Zampini   {
3198135c375SStefano Zampini     PetscViewerGLVisInfo glvis_info;
3208135c375SStefano Zampini     ierr = PetscContainerGetPointer(glvis_container,(void**)&glvis_info);CHKERRQ(ierr);
3218135c375SStefano Zampini     enabled = glvis_info->enabled;
3228135c375SStefano Zampini   }
3238135c375SStefano Zampini   ierr = DMPlexGetHybridBounds(dm, &cEndInterior, &fEndInterior, NULL, NULL);CHKERRQ(ierr);
3248135c375SStefano Zampini   ierr = DMPlexGetHeightStratum(dm,0,&cStart,&cEnd);CHKERRQ(ierr);
3258135c375SStefano Zampini   cEnd = cEndInterior < 0 ? cEnd : cEndInterior;
3268135c375SStefano Zampini   ierr = DMPlexGetDepthStratum(dm,0,&vStart,&vEnd);CHKERRQ(ierr);
3278135c375SStefano Zampini 
3288135c375SStefano Zampini   /*
3298135c375SStefano Zampini      a couple of sections of the mesh specification are disabled
3308135c375SStefano Zampini        - boundary: unless we want to visualize boundary attributes, the boundary is not needed for proper mesh visualization
3318135c375SStefano Zampini        - vertex_parents: used for non-conforming meshes only when we want to use MFEM as a discretization package and be able to derefine the mesh
3328135c375SStefano Zampini   */
3338135c375SStefano Zampini   enable_boundary = PETSC_FALSE;
3348135c375SStefano Zampini   enable_ncmesh   = PETSC_FALSE;
3358135c375SStefano Zampini   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)dm),((PetscObject)dm)->prefix,"GLVis PetscViewer DMPlex Options","PetscViewer");CHKERRQ(ierr);
3368135c375SStefano Zampini   ierr = PetscOptionsBool("-viewer_glvis_dm_plex_enable_boundary","Enable boundary section in mesh representation; useful for debugging purposes",NULL,enable_boundary,&enable_boundary,NULL);CHKERRQ(ierr);
3378135c375SStefano Zampini   ierr = PetscOptionsBool("-viewer_glvis_dm_plex_enable_ncmesh","Enable vertex_parents section in mesh representation; useful for debugging purposes",NULL,enable_ncmesh,&enable_ncmesh,NULL);CHKERRQ(ierr);
3388135c375SStefano Zampini   ierr = PetscOptionsEnd();CHKERRQ(ierr);
3398135c375SStefano Zampini 
3408135c375SStefano Zampini   /* Identify possible cells in the overlap */
3418135c375SStefano Zampini   gNum = NULL;
3428135c375SStefano Zampini   novl = 0;
3438135c375SStefano Zampini   ovl  = PETSC_FALSE;
3448135c375SStefano Zampini   pown = NULL;
3458135c375SStefano Zampini   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)dm),&commsize);CHKERRQ(ierr);
3468135c375SStefano Zampini   if (commsize > 1) {
3478135c375SStefano Zampini     ierr = DMPlexGetCellNumbering(dm,&globalNum);CHKERRQ(ierr);
3488135c375SStefano Zampini     ierr = ISGetIndices(globalNum,&gNum);CHKERRQ(ierr);
3498135c375SStefano Zampini     for (p=cStart; p<cEnd; p++) {
3508135c375SStefano Zampini       if (gNum[p-cStart] < 0) {
3518135c375SStefano Zampini         ovl = PETSC_TRUE;
3528135c375SStefano Zampini         novl++;
3538135c375SStefano Zampini       }
3548135c375SStefano Zampini     }
3558135c375SStefano Zampini     if (ovl) {
3568135c375SStefano Zampini       /* it may happen that pown get not destroyed, if the user closes the window while this function is running.
3578135c375SStefano Zampini          TODO: garbage collector? attach pown to dm?  */
3588135c375SStefano Zampini       ierr = PetscBTCreate(PetscMax(cEnd-cStart,vEnd-vStart),&pown);CHKERRQ(ierr);
3598135c375SStefano Zampini     }
3608135c375SStefano Zampini   }
3618135c375SStefano Zampini 
3628135c375SStefano Zampini   if (!enabled) {
3638135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"MFEM mesh v1.1\n");CHKERRQ(ierr);
3648135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\ndimension\n");CHKERRQ(ierr);
3658135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",dim);CHKERRQ(ierr);
3668135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\nelements\n");CHKERRQ(ierr);
3678135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
3688135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\nboundary\n");CHKERRQ(ierr);
3698135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
3708135c375SStefano Zampini     ierr = DMGetCoordinateDim(dm,&sdim);CHKERRQ(ierr);
3718135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\nvertices\n");CHKERRQ(ierr);
3728135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
3738135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",sdim);CHKERRQ(ierr);
3748135c375SStefano Zampini     ierr = PetscBTDestroy(&pown);CHKERRQ(ierr);
3758135c375SStefano Zampini     if (globalNum) {
3768135c375SStefano Zampini       ierr = ISRestoreIndices(globalNum,&gNum);CHKERRQ(ierr);
3778135c375SStefano Zampini     }
3788135c375SStefano Zampini     PetscFunctionReturn(0);
3798135c375SStefano Zampini   }
3808135c375SStefano Zampini 
3818135c375SStefano Zampini   /* header */
3828135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"MFEM mesh v1.1\n");CHKERRQ(ierr);
3838135c375SStefano Zampini 
3848135c375SStefano Zampini   /* topological dimension */
3858135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"\ndimension\n");CHKERRQ(ierr);
3868135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"%D\n",dim);CHKERRQ(ierr);
3878135c375SStefano Zampini 
3888135c375SStefano Zampini   /* elements */
3898135c375SStefano Zampini   /* TODO: is this the label we want to use for marking material IDs?
3908135c375SStefano Zampini      We should decide to have a single marker for these stuff
3918135c375SStefano Zampini      Proposal: DMSetMaterialLabel?
3928135c375SStefano Zampini   */
3938135c375SStefano Zampini   ierr = DMGetLabel(dm,"Cell sets",&label);CHKERRQ(ierr);
3948135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"\nelements\n");CHKERRQ(ierr);
3958135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"%D\n",cEnd-cStart-novl);CHKERRQ(ierr);
3968135c375SStefano Zampini   for (p=cStart;p<cEnd;p++) {
39711a4995dSStefano Zampini     int      vids[8];
39811a4995dSStefano Zampini     PetscInt i,nv = 0,cid = -1,mid = 1;
3998135c375SStefano Zampini 
4008135c375SStefano Zampini     if (ovl) {
4018135c375SStefano Zampini       if (gNum[p-cStart] < 0) continue;
4028135c375SStefano Zampini       else {
4038135c375SStefano Zampini         ierr = PetscBTSet(pown,p-cStart);CHKERRQ(ierr);
4048135c375SStefano Zampini       }
4058135c375SStefano Zampini     }
4068135c375SStefano Zampini     ierr = DMPlexGetPointMFEMCellID_Internal(dm,label,p,&mid,&cid);CHKERRQ(ierr);
4078135c375SStefano Zampini     ierr = DMPlexGetPointMFEMVertexIDs_Internal(dm,p,&nv,vids);CHKERRQ(ierr);
4088135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D %D",mid,cid);CHKERRQ(ierr);
4098135c375SStefano Zampini     for (i=0;i<nv;i++) {
41011a4995dSStefano Zampini       ierr = PetscViewerASCIIPrintf(viewer," %d",vids[i]);CHKERRQ(ierr);
4118135c375SStefano Zampini     }
4128135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
4138135c375SStefano Zampini   }
4148135c375SStefano Zampini 
4158135c375SStefano Zampini   /* determine orientation of boundary mesh */
4168135c375SStefano Zampini   if (cEnd-cStart) {
4178135c375SStefano Zampini     ierr = DMPlexGetConeSize(dm,cStart,&fpc);CHKERRQ(ierr);
4188135c375SStefano Zampini     switch(dim) {
4198135c375SStefano Zampini       case 1:
4208135c375SStefano Zampini         if (fpc != 2) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case faces per cell %D",fpc);
4218135c375SStefano Zampini         faces = fv1;
4228135c375SStefano Zampini         vpf = 1;
4238135c375SStefano Zampini         break;
4248135c375SStefano Zampini       case 2:
4258135c375SStefano Zampini         switch (fpc) {
4268135c375SStefano Zampini           case 3:
4278135c375SStefano Zampini             faces = fv2tri;
4288135c375SStefano Zampini             vpf   = 2;
4298135c375SStefano Zampini             break;
4308135c375SStefano Zampini           case 4:
4318135c375SStefano Zampini             faces = fv2quad;
4328135c375SStefano Zampini             vpf   = 2;
4338135c375SStefano Zampini             break;
4348135c375SStefano Zampini           default:
4358135c375SStefano Zampini             SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case: faces per cell %D",fpc);
4368135c375SStefano Zampini             break;
4378135c375SStefano Zampini         }
4388135c375SStefano Zampini         break;
4398135c375SStefano Zampini       case 3:
4408135c375SStefano Zampini         switch (fpc) {
4418135c375SStefano Zampini           case 4:
4428135c375SStefano Zampini             faces = fv3tet;
4438135c375SStefano Zampini             vpf   = 3;
4448135c375SStefano Zampini             break;
4458135c375SStefano Zampini           case 6:
4468135c375SStefano Zampini             faces = fv3hex;
4478135c375SStefano Zampini             vpf   = 4;
4488135c375SStefano Zampini             break;
4498135c375SStefano Zampini           default:
4508135c375SStefano Zampini             SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Unhandled case: faces per cell %D",fpc);
4518135c375SStefano Zampini             break;
4528135c375SStefano Zampini         }
4538135c375SStefano Zampini         break;
4548135c375SStefano Zampini       default:
4558135c375SStefano Zampini         SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Unhandled dim");
4568135c375SStefano Zampini         break;
4578135c375SStefano Zampini     }
4588135c375SStefano Zampini   }
4598135c375SStefano Zampini 
4608135c375SStefano Zampini   /* boundary */
4618135c375SStefano Zampini   ierr = DMPlexGetHeightStratum(dm,1,&fStart,&fEnd);CHKERRQ(ierr);
4628135c375SStefano Zampini   fEnd = fEndInterior < 0 ? fEnd : fEndInterior;
4638135c375SStefano Zampini   if (!ovl) {
4648135c375SStefano Zampini     for (p=fStart,bf=0;p<fEnd;p++) {
4658135c375SStefano Zampini       PetscInt supportSize;
4668135c375SStefano Zampini 
4678135c375SStefano Zampini       ierr = DMPlexGetSupportSize(dm,p,&supportSize);CHKERRQ(ierr);
4688135c375SStefano Zampini       if (supportSize == 1) bf++;
4698135c375SStefano Zampini     }
4708135c375SStefano Zampini   } else {
4718135c375SStefano Zampini     for (p=fStart,bf=0;p<fEnd;p++) {
4728135c375SStefano Zampini       const PetscInt *support;
4738135c375SStefano Zampini       PetscInt       i,supportSize;
4748135c375SStefano Zampini       PetscBool      has_owned = PETSC_FALSE, has_ghost = PETSC_FALSE;
4758135c375SStefano Zampini 
4768135c375SStefano Zampini       ierr = DMPlexGetSupportSize(dm,p,&supportSize);CHKERRQ(ierr);
4778135c375SStefano Zampini       ierr = DMPlexGetSupport(dm,p,&support);CHKERRQ(ierr);
4788135c375SStefano Zampini       for (i=0;i<supportSize;i++) {
4798135c375SStefano Zampini         if (PetscBTLookup(pown,support[i]-cStart)) has_owned = PETSC_TRUE;
4808135c375SStefano Zampini         else has_ghost = PETSC_TRUE;
4818135c375SStefano Zampini       }
4828135c375SStefano Zampini       if ((supportSize == 1 && has_owned) || (supportSize > 1 && has_owned && has_ghost)) bf++;
4838135c375SStefano Zampini     }
4848135c375SStefano Zampini   }
4858135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"\nboundary\n");CHKERRQ(ierr);
4868135c375SStefano Zampini   if (!enable_boundary) {
4878135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
4888135c375SStefano Zampini   } else {
4898135c375SStefano Zampini     /* TODO: is this the label we want to use for marking boundary facets?
4908135c375SStefano Zampini        We should decide to have a single marker for these stuff
4918135c375SStefano Zampini        Proposal: DMSetBoundaryLabel?
4928135c375SStefano Zampini     */
4938135c375SStefano Zampini     ierr = DMGetLabel(dm,"marker",&label);CHKERRQ(ierr);
4948135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",bf);CHKERRQ(ierr);
4958135c375SStefano Zampini     if (!ovl) {
4968135c375SStefano Zampini       for (p=fStart;p<fEnd;p++) {
4978135c375SStefano Zampini         PetscInt supportSize;
4988135c375SStefano Zampini 
4998135c375SStefano Zampini         ierr = DMPlexGetSupportSize(dm,p,&supportSize);CHKERRQ(ierr);
5008135c375SStefano Zampini         if (supportSize == 1) {
5018135c375SStefano Zampini           const    PetscInt *support, *cone;
5028135c375SStefano Zampini           PetscInt i,c,v,cid = -1,mid = -1;
5038135c375SStefano Zampini           PetscInt cellClosureSize,*cellClosure = NULL;
5048135c375SStefano Zampini 
5058135c375SStefano Zampini           ierr = DMPlexGetSupport(dm,p,&support);CHKERRQ(ierr);
5068135c375SStefano Zampini           ierr = DMPlexGetCone(dm,support[0],&cone);CHKERRQ(ierr);
5078135c375SStefano Zampini           for (c=0;c<fpc;c++)
5088135c375SStefano Zampini             if (cone[c] == p)
5098135c375SStefano Zampini               break;
5108135c375SStefano Zampini 
5118135c375SStefano Zampini           ierr = DMPlexGetTransitiveClosure(dm,support[0],PETSC_TRUE,&cellClosureSize,&cellClosure);CHKERRQ(ierr);
5128135c375SStefano Zampini           for (v=0;v<cellClosureSize;v++)
5138135c375SStefano Zampini             if (cellClosure[2*v] >= vStart && cellClosure[2*v] < vEnd)
5148135c375SStefano Zampini               break;
5158135c375SStefano Zampini           ierr = DMPlexGetPointMFEMCellID_Internal(dm,label,p,&mid,&cid);CHKERRQ(ierr);
5168135c375SStefano Zampini           ierr = PetscViewerASCIIPrintf(viewer,"%D %D",mid,cid);CHKERRQ(ierr);
5178135c375SStefano Zampini           for (i=0;i<vpf;i++) {
5188135c375SStefano Zampini             ierr = PetscViewerASCIIPrintf(viewer," %D",cellClosure[2*(v+faces[c*vpf+i])]-vStart);CHKERRQ(ierr);
5198135c375SStefano Zampini           }
5208135c375SStefano Zampini           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
5218135c375SStefano Zampini           ierr = DMPlexRestoreTransitiveClosure(dm,support[0],PETSC_TRUE,&cellClosureSize,&cellClosure);CHKERRQ(ierr);
5228135c375SStefano Zampini         }
5238135c375SStefano Zampini       }
5248135c375SStefano Zampini     } else {
5258135c375SStefano Zampini       for (p=fStart;p<fEnd;p++) {
5268135c375SStefano Zampini         const PetscInt *support;
5278135c375SStefano Zampini         PetscInt       i,supportSize,validcell = -1;
5288135c375SStefano Zampini         PetscBool      has_owned = PETSC_FALSE, has_ghost = PETSC_FALSE;
5298135c375SStefano Zampini 
5308135c375SStefano Zampini         ierr = DMPlexGetSupportSize(dm,p,&supportSize);CHKERRQ(ierr);
5318135c375SStefano Zampini         ierr = DMPlexGetSupport(dm,p,&support);CHKERRQ(ierr);
5328135c375SStefano Zampini         for (i=0;i<supportSize;i++) {
5338135c375SStefano Zampini           if (PetscBTLookup(pown,support[i]-cStart)) {
5348135c375SStefano Zampini             has_owned = PETSC_TRUE;
5358135c375SStefano Zampini             validcell = support[i];
5368135c375SStefano Zampini           } else has_ghost = PETSC_TRUE;
5378135c375SStefano Zampini         }
5388135c375SStefano Zampini         if ((supportSize == 1 && has_owned) || (supportSize > 1 && has_owned && has_ghost)) {
5398135c375SStefano Zampini           const    PetscInt *support, *cone;
5408135c375SStefano Zampini           PetscInt i,c,v,cid = -1,mid = -1;
5418135c375SStefano Zampini           PetscInt cellClosureSize,*cellClosure = NULL;
5428135c375SStefano Zampini 
5438135c375SStefano Zampini           ierr = DMPlexGetSupport(dm,p,&support);CHKERRQ(ierr);
5448135c375SStefano Zampini           ierr = DMPlexGetCone(dm,validcell,&cone);CHKERRQ(ierr);
5458135c375SStefano Zampini           for (c=0;c<fpc;c++)
5468135c375SStefano Zampini             if (cone[c] == p)
5478135c375SStefano Zampini               break;
5488135c375SStefano Zampini 
5498135c375SStefano Zampini           ierr = DMPlexGetTransitiveClosure(dm,validcell,PETSC_TRUE,&cellClosureSize,&cellClosure);CHKERRQ(ierr);
5508135c375SStefano Zampini           for (v=0;v<cellClosureSize;v++)
5518135c375SStefano Zampini             if (cellClosure[2*v] >= vStart && cellClosure[2*v] < vEnd)
5528135c375SStefano Zampini               break;
5538135c375SStefano Zampini           ierr = DMPlexGetPointMFEMCellID_Internal(dm,label,p,&mid,&cid);CHKERRQ(ierr);
5548135c375SStefano Zampini           ierr = PetscViewerASCIIPrintf(viewer,"%D %D",mid,cid);CHKERRQ(ierr);
5558135c375SStefano Zampini           for (i=0;i<vpf;i++) {
5568135c375SStefano Zampini             ierr = PetscViewerASCIIPrintf(viewer," %D",cellClosure[2*(v+faces[c*vpf+i])]-vStart);CHKERRQ(ierr);
5578135c375SStefano Zampini           }
5588135c375SStefano Zampini           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
5598135c375SStefano Zampini           ierr = DMPlexRestoreTransitiveClosure(dm,validcell,PETSC_TRUE,&cellClosureSize,&cellClosure);CHKERRQ(ierr);
5608135c375SStefano Zampini         }
5618135c375SStefano Zampini       }
5628135c375SStefano Zampini     }
5638135c375SStefano Zampini   }
5648135c375SStefano Zampini 
5658135c375SStefano Zampini   /* mark owned vertices */
5668135c375SStefano Zampini   if (ovl) {
5678135c375SStefano Zampini     ierr = PetscBTMemzero(vEnd-vStart,pown);CHKERRQ(ierr);
5688135c375SStefano Zampini     for (p=cStart;p<cEnd;p++) {
5698135c375SStefano Zampini       PetscInt i,closureSize,*closure = NULL;
5708135c375SStefano Zampini 
5718135c375SStefano Zampini       if (gNum[p-cStart] < 0) continue;
5728135c375SStefano Zampini       ierr = DMPlexGetTransitiveClosure(dm,p,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
5738135c375SStefano Zampini       for (i=0;i<closureSize;i++) {
5748135c375SStefano Zampini         const PetscInt pp = closure[2*i];
5758135c375SStefano Zampini 
5768135c375SStefano Zampini         if (pp >= vStart && pp < vEnd) {
5778135c375SStefano Zampini           ierr = PetscBTSet(pown,pp-vStart);CHKERRQ(ierr);
5788135c375SStefano Zampini         }
5798135c375SStefano Zampini       }
5808135c375SStefano Zampini       ierr = DMPlexRestoreTransitiveClosure(dm,p,PETSC_TRUE,&closureSize,&closure);CHKERRQ(ierr);
5818135c375SStefano Zampini     }
5828135c375SStefano Zampini   }
5838135c375SStefano Zampini   if (globalNum) {
5848135c375SStefano Zampini     ierr = ISRestoreIndices(globalNum,&gNum);CHKERRQ(ierr);
5858135c375SStefano Zampini   }
5868135c375SStefano Zampini 
5878135c375SStefano Zampini   /* vertex_parents (Non-conforming meshes) */
5888135c375SStefano Zampini   parentSection  = NULL;
5898135c375SStefano Zampini   if (enable_ncmesh) {
5908135c375SStefano Zampini     ierr = DMPlexGetTree(dm,&parentSection,NULL,NULL,NULL,NULL);CHKERRQ(ierr);
5918135c375SStefano Zampini   }
5928135c375SStefano Zampini   if (parentSection) {
5938135c375SStefano Zampini     PetscInt vp,gvp;
5948135c375SStefano Zampini 
5958135c375SStefano Zampini     for (vp=0,p=vStart;p<vEnd;p++) {
5968135c375SStefano Zampini       DMLabel  dlabel;
5978135c375SStefano Zampini       PetscInt parent,depth;
5988135c375SStefano Zampini 
5998135c375SStefano Zampini       if (PetscUnlikely(ovl && !PetscBTLookup(pown,p-vStart))) continue;
6008135c375SStefano Zampini       ierr = DMPlexGetDepthLabel(dm,&dlabel);CHKERRQ(ierr);
6018135c375SStefano Zampini       ierr = DMLabelGetValue(dlabel,p,&depth);CHKERRQ(ierr);
6028135c375SStefano Zampini       ierr = DMPlexGetTreeParent(dm,p,&parent,NULL);CHKERRQ(ierr);
6038135c375SStefano Zampini       if (parent != p) vp++;
6048135c375SStefano Zampini     }
6058135c375SStefano Zampini     ierr = MPIU_Allreduce(&vp,&gvp,1,MPIU_INT,MPI_SUM,PetscObjectComm((PetscObject)dm));CHKERRQ(ierr);
6068135c375SStefano Zampini     if (gvp) {
6078135c375SStefano Zampini       PetscInt  maxsupp;
6088135c375SStefano Zampini       PetscBool *skip = NULL;
6098135c375SStefano Zampini 
6108135c375SStefano Zampini       ierr = PetscViewerASCIIPrintf(viewer,"\nvertex_parents\n");CHKERRQ(ierr);
6118135c375SStefano Zampini       ierr = PetscViewerASCIIPrintf(viewer,"%D\n",vp);CHKERRQ(ierr);
6128135c375SStefano Zampini       ierr = DMPlexGetMaxSizes(dm,NULL,&maxsupp);CHKERRQ(ierr);
6138135c375SStefano Zampini       ierr = PetscMalloc1(maxsupp,&skip);CHKERRQ(ierr);
6148135c375SStefano Zampini       for (p=vStart;p<vEnd;p++) {
6158135c375SStefano Zampini         DMLabel  dlabel;
6168135c375SStefano Zampini         PetscInt parent;
6178135c375SStefano Zampini 
6188135c375SStefano Zampini         if (PetscUnlikely(ovl && !PetscBTLookup(pown,p-vStart))) continue;
6198135c375SStefano Zampini         ierr = DMPlexGetDepthLabel(dm,&dlabel);CHKERRQ(ierr);
6208135c375SStefano Zampini         ierr = DMPlexGetTreeParent(dm,p,&parent,NULL);CHKERRQ(ierr);
6218135c375SStefano Zampini         if (parent != p) {
62211a4995dSStefano Zampini           int            vids[8];
62311a4995dSStefano Zampini           PetscInt       i,nv,size,n,numChildren,depth = -1;
6248135c375SStefano Zampini           const PetscInt *children;
6258135c375SStefano Zampini           ierr = DMPlexGetConeSize(dm,parent,&size);CHKERRQ(ierr);
6268135c375SStefano Zampini           switch (size) {
6278135c375SStefano Zampini             case 2: /* edge */
6288135c375SStefano Zampini               nv   = 0;
6298135c375SStefano Zampini               ierr = DMPlexGetPointMFEMVertexIDs_Internal(dm,parent,&nv,vids);CHKERRQ(ierr);
6308135c375SStefano Zampini               ierr = PetscViewerASCIIPrintf(viewer,"%D",p-vStart);CHKERRQ(ierr);
6318135c375SStefano Zampini               for (i=0;i<nv;i++) {
63211a4995dSStefano Zampini                 ierr = PetscViewerASCIIPrintf(viewer," %d",vids[i]);CHKERRQ(ierr);
6338135c375SStefano Zampini               }
6348135c375SStefano Zampini               ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
6358135c375SStefano Zampini               vp--;
6368135c375SStefano Zampini               break;
6378135c375SStefano Zampini             case 4: /* face */
6388135c375SStefano Zampini               ierr = PetscMemzero(skip,maxsupp*sizeof(PetscBool));CHKERRQ(ierr);
6398135c375SStefano Zampini               ierr = DMPlexGetTreeChildren(dm,parent,&numChildren,&children);CHKERRQ(ierr);
6408135c375SStefano Zampini               for (n=0;n<numChildren;n++) {
6418135c375SStefano Zampini                 ierr = DMLabelGetValue(dlabel,children[n],&depth);CHKERRQ(ierr);
6428135c375SStefano Zampini                 if (!depth) {
6438135c375SStefano Zampini                   const PetscInt *hvsupp,*hesupp,*cone;
6448135c375SStefano Zampini                   PetscInt       hvsuppSize,hesuppSize,coneSize;
6458135c375SStefano Zampini                   PetscInt       hv = children[n],he,f;
6468135c375SStefano Zampini                   PetscBool      found = PETSC_FALSE;
6478135c375SStefano Zampini 
6488135c375SStefano Zampini                   ierr = DMPlexGetSupportSize(dm,hv,&hvsuppSize);CHKERRQ(ierr);
6498135c375SStefano Zampini                   ierr = DMPlexGetSupport(dm,hv,&hvsupp);CHKERRQ(ierr);
6508135c375SStefano Zampini                   for (i=0;i<hvsuppSize;i++) {
6518135c375SStefano Zampini                     PetscInt ep;
6528135c375SStefano Zampini                     ierr = DMPlexGetTreeParent(dm,hvsupp[i],&ep,NULL);CHKERRQ(ierr);
6538135c375SStefano Zampini                     if (ep != hvsupp[i]) {
6548135c375SStefano Zampini                       he = hvsupp[i];
6558135c375SStefano Zampini                       found = PETSC_TRUE;
6568135c375SStefano Zampini                     } else {
6578135c375SStefano Zampini                       skip[i] = PETSC_TRUE;
6588135c375SStefano Zampini                     }
6598135c375SStefano Zampini                   }
6608135c375SStefano Zampini                   if (!found) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Vertex %D support size %D: hanging edge not found",hv,hvsuppSize);
6618135c375SStefano Zampini                   he      = hvsupp[i];
6628135c375SStefano Zampini                   skip[i] = PETSC_TRUE;
6638135c375SStefano Zampini                   ierr    = DMPlexGetCone(dm,he,&cone);CHKERRQ(ierr);
66411a4995dSStefano Zampini                   vids[0] = (int)((cone[0] == hv) ? cone[1] : cone[0]);
6658135c375SStefano Zampini                   ierr    = DMPlexGetSupportSize(dm,he,&hesuppSize);CHKERRQ(ierr);
6668135c375SStefano Zampini                   ierr    = DMPlexGetSupport(dm,he,&hesupp);CHKERRQ(ierr);
6678135c375SStefano Zampini                   for (f=0;f<hesuppSize;f++) {
6688135c375SStefano Zampini                     PetscInt j;
6698135c375SStefano Zampini 
6708135c375SStefano Zampini                     ierr = DMPlexGetCone(dm,hesupp[f],&cone);CHKERRQ(ierr);
6718135c375SStefano Zampini                     ierr = DMPlexGetConeSize(dm,hesupp[f],&coneSize);CHKERRQ(ierr);
6728135c375SStefano Zampini                     for (j=0;j<coneSize;j++) {
6738135c375SStefano Zampini                       PetscInt k;
6748135c375SStefano Zampini                       for (k=0;k<hvsuppSize;k++) {
6758135c375SStefano Zampini                         if (hvsupp[k] == cone[j]) {
6768135c375SStefano Zampini                           skip[k] = PETSC_TRUE;
6778135c375SStefano Zampini                           break;
6788135c375SStefano Zampini                         }
6798135c375SStefano Zampini                       }
6808135c375SStefano Zampini                     }
6818135c375SStefano Zampini                   }
6828135c375SStefano Zampini                   for (i=0;i<hvsuppSize;i++) {
6838135c375SStefano Zampini                     if (!skip[i]) {
6848135c375SStefano Zampini                       ierr = DMPlexGetCone(dm,hvsupp[i],&cone);CHKERRQ(ierr);
68511a4995dSStefano Zampini                       vids[1] = (int)((cone[0] == hv) ? cone[1] : cone[0]);
6868135c375SStefano Zampini                     }
6878135c375SStefano Zampini                   }
6888135c375SStefano Zampini                   ierr = PetscViewerASCIIPrintf(viewer,"%D",hv-vStart);CHKERRQ(ierr);
6898135c375SStefano Zampini                   for (i=0;i<2;i++) {
69011a4995dSStefano Zampini                     ierr = PetscViewerASCIIPrintf(viewer," %d",vids[i]-vStart);CHKERRQ(ierr);
6918135c375SStefano Zampini                   }
6928135c375SStefano Zampini                   ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
6938135c375SStefano Zampini                   vp--;
6948135c375SStefano Zampini                 }
6958135c375SStefano Zampini               }
6968135c375SStefano Zampini               break;
6978135c375SStefano Zampini             default:
6988135c375SStefano Zampini               SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Don't know how to deal with support size %d",size);
6998135c375SStefano Zampini           }
7008135c375SStefano Zampini         }
7018135c375SStefano Zampini       }
7028135c375SStefano Zampini       ierr = PetscFree(skip);CHKERRQ(ierr);
7038135c375SStefano Zampini     }
7048135c375SStefano Zampini     if (vp) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Unexpected %D hanging vertices",vp);
7058135c375SStefano Zampini   }
7068135c375SStefano Zampini   ierr = PetscBTDestroy(&pown);CHKERRQ(ierr);
7078135c375SStefano Zampini 
7088135c375SStefano Zampini   /* vertices */
7098135c375SStefano Zampini   ierr = DMGetCoordinateDim(dm,&sdim);CHKERRQ(ierr);
7108135c375SStefano Zampini   ierr = DMGetCoordinateDM(dm,&cdm);CHKERRQ(ierr);
7118135c375SStefano Zampini   ierr = DMGetCoordinateSection(dm,&coordSection);CHKERRQ(ierr);
7128135c375SStefano Zampini   ierr = DMGetCoordinatesLocal(dm,&coordinates);CHKERRQ(ierr);
7138135c375SStefano Zampini   ierr = DMPlexGetDepthStratum(cdm,0,&vStart,&vEnd);CHKERRQ(ierr);
7148135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"\nvertices\n");CHKERRQ(ierr);
7158135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"%D\n",vEnd-vStart);CHKERRQ(ierr);
7168135c375SStefano Zampini   ierr = PetscViewerASCIIPrintf(viewer,"%D\n",sdim);CHKERRQ(ierr);
7178135c375SStefano Zampini   ierr = VecGetArrayRead(coordinates,&array);CHKERRQ(ierr);
7188135c375SStefano Zampini   for (p=vStart;p<vEnd;p++) {
7198135c375SStefano Zampini     PetscInt i,st,dof;
7208135c375SStefano Zampini 
7218135c375SStefano Zampini     ierr = PetscSectionGetDof(coordSection,p,&dof);CHKERRQ(ierr);
7228135c375SStefano Zampini     ierr = PetscSectionGetOffset(coordSection,p,&st);CHKERRQ(ierr);
7238135c375SStefano Zampini     for (i=st;i<st+dof-1;i++) {
7248135c375SStefano Zampini       ierr = PetscViewerASCIIPrintf(viewer,"%g ",PetscRealPart(array[i]));CHKERRQ(ierr);
7258135c375SStefano Zampini     }
7268135c375SStefano Zampini     ierr = PetscViewerASCIIPrintf(viewer,"%g\n",PetscRealPart(array[i]));CHKERRQ(ierr);
7278135c375SStefano Zampini   }
7288135c375SStefano Zampini   ierr = VecRestoreArrayRead(coordinates,&array);CHKERRQ(ierr);
7298135c375SStefano Zampini   PetscFunctionReturn(0);
7308135c375SStefano Zampini }
7318135c375SStefano Zampini 
7328135c375SStefano Zampini /* dispatching, prints through the socket by prepending the mesh keyword to the usual ASCII dump */
7338135c375SStefano Zampini PETSC_INTERN PetscErrorCode DMPlexView_GLVis(DM dm, PetscViewer viewer)
7348135c375SStefano Zampini {
7358135c375SStefano Zampini   PetscErrorCode ierr;
7368135c375SStefano Zampini   PetscBool      isglvis,isascii;
7378135c375SStefano Zampini 
7388135c375SStefano Zampini   PetscFunctionBegin;
7398135c375SStefano Zampini   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
7408135c375SStefano Zampini   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
7418135c375SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERGLVIS,&isglvis);CHKERRQ(ierr);
7428135c375SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
7438135c375SStefano Zampini   if (!isglvis && !isascii) SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Viewer must be of type VIEWERGLVIS or VIEWERASCII");
7448135c375SStefano Zampini   if (isglvis) {
7458135c375SStefano Zampini     PetscViewer          view;
7468135c375SStefano Zampini     PetscViewerGLVisType type;
7478135c375SStefano Zampini 
7488135c375SStefano Zampini     ierr = PetscViewerGLVisGetType_Private(viewer,&type);CHKERRQ(ierr);
7498135c375SStefano Zampini     ierr = PetscViewerGLVisGetDMWindow_Private(viewer,&view);CHKERRQ(ierr);
7508135c375SStefano Zampini     if (view) { /* in the socket case, it may happen that the connection failed */
7518135c375SStefano Zampini       if (type == PETSC_VIEWER_GLVIS_SOCKET) {
7528135c375SStefano Zampini         PetscMPIInt size,rank;
7538135c375SStefano Zampini         ierr = MPI_Comm_size(PetscObjectComm((PetscObject)dm),&size);CHKERRQ(ierr);
7548135c375SStefano Zampini         ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)dm),&rank);CHKERRQ(ierr);
7558135c375SStefano Zampini         ierr = PetscViewerASCIIPrintf(view,"parallel %D %D\nmesh\n",size,rank);CHKERRQ(ierr);
7568135c375SStefano Zampini       }
7578135c375SStefano Zampini       ierr = DMPlexView_GLVis_ASCII(dm,view);CHKERRQ(ierr);
7588135c375SStefano Zampini       ierr = PetscViewerFlush(view);CHKERRQ(ierr);
7598135c375SStefano Zampini       if (type == PETSC_VIEWER_GLVIS_SOCKET) {
7608135c375SStefano Zampini         PetscInt    dim;
7618135c375SStefano Zampini         const char* name;
7628135c375SStefano Zampini 
7638135c375SStefano Zampini         ierr = PetscObjectGetName((PetscObject)dm,&name);CHKERRQ(ierr);
7648135c375SStefano Zampini         ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
7658135c375SStefano Zampini         ierr = PetscViewerGLVisInitWindow_Private(view,PETSC_TRUE,dim,name);CHKERRQ(ierr);
7668135c375SStefano Zampini         ierr = PetscBarrier((PetscObject)dm);CHKERRQ(ierr);
7678135c375SStefano Zampini       }
7688135c375SStefano Zampini     }
7698135c375SStefano Zampini   } else {
7708135c375SStefano Zampini     ierr = DMPlexView_GLVis_ASCII(dm,viewer);CHKERRQ(ierr);
7718135c375SStefano Zampini   }
7728135c375SStefano Zampini   PetscFunctionReturn(0);
7738135c375SStefano Zampini }
774