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