xref: /petsc/src/dm/impls/da/grglvis.c (revision c4762a1b19cd2af06abeed90e8f9d34fb975dd94)
1 /* Routines to visualize DMDAs and fields through GLVis */
2 
3 #include <petsc/private/dmdaimpl.h>
4 #include <petsc/private/glvisviewerimpl.h>
5 
6 typedef struct {
7   PetscBool ll;
8 } DMDAGhostedGLVisViewerCtx;
9 
10 static PetscErrorCode DMDAGhostedDestroyGLVisViewerCtx_Private(void **vctx)
11 {
12   PetscErrorCode ierr;
13 
14   PetscFunctionBegin;
15   ierr = PetscFree(*vctx);CHKERRQ(ierr);
16   PetscFunctionReturn(0);
17 }
18 
19 typedef struct {
20   Vec xlocal;
21 } DMDAFieldGLVisViewerCtx;
22 
23 static PetscErrorCode DMDAFieldDestroyGLVisViewerCtx_Private(void *vctx)
24 {
25   DMDAFieldGLVisViewerCtx *ctx = (DMDAFieldGLVisViewerCtx*)vctx;
26   PetscErrorCode          ierr;
27 
28   PetscFunctionBegin;
29   ierr = VecDestroy(&ctx->xlocal);CHKERRQ(ierr);
30   ierr = PetscFree(vctx);CHKERRQ(ierr);
31   PetscFunctionReturn(0);
32 }
33 
34 /*
35    dactx->ll is false -> all but the last proc per dimension claim the ghosted node on the right
36    dactx->ll is true -> all but the first proc per dimension claim the ghosted node on the left
37 */
38 static PetscErrorCode DMDAGetNumElementsGhosted(DM da, PetscInt *nex, PetscInt *ney, PetscInt *nez)
39 {
40   DMDAGhostedGLVisViewerCtx *dactx;
41   PetscInt                  sx,sy,sz,ien,jen,ken;
42   PetscErrorCode            ierr;
43 
44   PetscFunctionBegin;
45   /* Appease -Wmaybe-uninitialized */
46   if (nex) *nex = -1;
47   if (ney) *ney = -1;
48   if (nez) *nez = -1;
49   ierr = DMDAGetCorners(da,&sx,&sy,&sz,&ien,&jen,&ken);CHKERRQ(ierr);
50   ierr = DMGetApplicationContext(da,(void**)&dactx);CHKERRQ(ierr);
51   if (dactx->ll) {
52     PetscInt dim;
53 
54     ierr = DMGetDimension(da,&dim);CHKERRQ(ierr);
55     if (!sx) ien--;
56     if (!sy && dim > 1) jen--;
57     if (!sz && dim > 2) ken--;
58   } else {
59     PetscInt M,N,P;
60 
61     ierr = DMDAGetInfo(da,NULL,&M,&N,&P,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL);CHKERRQ(ierr);
62     if (sx+ien == M) ien--;
63     if (sy+jen == N) jen--;
64     if (sz+ken == P) ken--;
65   }
66   if (nex) *nex = ien;
67   if (ney) *ney = jen;
68   if (nez) *nez = ken;
69   PetscFunctionReturn(0);
70 }
71 
72 /* inherits number of vertices from DMDAGetNumElementsGhosted */
73 static PetscErrorCode DMDAGetNumVerticesGhosted(DM da, PetscInt *nvx, PetscInt *nvy, PetscInt *nvz)
74 {
75   PetscInt       ien = 0,jen = 0,ken = 0,dim;
76   PetscInt       tote;
77   PetscErrorCode ierr;
78 
79   PetscFunctionBegin;
80   ierr = DMGetDimension(da,&dim);CHKERRQ(ierr);
81   ierr = DMDAGetNumElementsGhosted(da,&ien,&jen,&ken);CHKERRQ(ierr);
82   tote = ien * (dim > 1 ? jen : 1 ) * (dim > 2 ? ken : 1);
83   if (tote) {
84     ien = ien+1;
85     jen = dim > 1 ? jen+1 : 1;
86     ken = dim > 2 ? ken+1 : 1;
87   }
88   if (nvx) *nvx = ien;
89   if (nvy) *nvy = jen;
90   if (nvz) *nvz = ken;
91   PetscFunctionReturn(0);
92 }
93 
94 static PetscErrorCode DMDASampleGLVisFields_Private(PetscObject oX, PetscInt nf, PetscObject oXf[], void *vctx)
95 {
96   DM                        da;
97   DMDAFieldGLVisViewerCtx   *ctx = (DMDAFieldGLVisViewerCtx*)vctx;
98   DMDAGhostedGLVisViewerCtx *dactx;
99   const PetscScalar         *array;
100   PetscScalar               **arrayf;
101   PetscInt                  i,f,ii,ien,jen,ken,ie,je,ke,bs,*bss;
102   PetscInt                  sx,sy,sz,gsx,gsy,gsz,ist,jst,kst,gm,gn,gp;
103   PetscErrorCode            ierr;
104 
105   PetscFunctionBegin;
106   ierr = VecGetDM(ctx->xlocal,&da);CHKERRQ(ierr);
107   if (!da) SETERRQ(PetscObjectComm(oX),PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
108   ierr = DMGetApplicationContext(da,(void**)&dactx);CHKERRQ(ierr);
109   ierr = VecGetBlockSize(ctx->xlocal,&bs);CHKERRQ(ierr);
110   ierr = DMGlobalToLocalBegin(da,(Vec)oX,INSERT_VALUES,ctx->xlocal);CHKERRQ(ierr);
111   ierr = DMGlobalToLocalEnd(da,(Vec)oX,INSERT_VALUES,ctx->xlocal);CHKERRQ(ierr);
112   ierr = DMDAGetNumVerticesGhosted(da,&ien,&jen,&ken);CHKERRQ(ierr);
113   ierr = DMDAGetGhostCorners(da,&gsx,&gsy,&gsz,&gm,&gn,&gp);CHKERRQ(ierr);
114   ierr = DMDAGetCorners(da,&sx,&sy,&sz,NULL,NULL,NULL);CHKERRQ(ierr);
115   if (dactx->ll) {
116     kst = jst = ist = 0;
117   } else {
118     kst  = gsz != sz ? 1 : 0;
119     jst  = gsy != sy ? 1 : 0;
120     ist  = gsx != sx ? 1 : 0;
121   }
122   ierr = PetscMalloc2(nf,&arrayf,nf,&bss);CHKERRQ(ierr);
123   ierr = VecGetArrayRead(ctx->xlocal,&array);CHKERRQ(ierr);
124   for (f=0;f<nf;f++) {
125     ierr = VecGetBlockSize((Vec)oXf[f],&bss[f]);CHKERRQ(ierr);
126     ierr = VecGetArray((Vec)oXf[f],&arrayf[f]);CHKERRQ(ierr);
127   }
128   for (ke = kst, ii = 0; ke < kst + ken; ke++) {
129     for (je = jst; je < jst + jen; je++) {
130       for (ie = ist; ie < ist + ien; ie++) {
131         PetscInt cf,b;
132         i = ke * gm * gn + je * gm + ie;
133         for (f=0,cf=0;f<nf;f++)
134           for (b=0;b<bss[f];b++)
135             arrayf[f][bss[f]*ii+b] = array[i*bs+cf++];
136         ii++;
137       }
138     }
139   }
140   for (f=0;f<nf;f++) { ierr = VecRestoreArray((Vec)oXf[f],&arrayf[f]);CHKERRQ(ierr); }
141   ierr = VecRestoreArrayRead(ctx->xlocal,&array);CHKERRQ(ierr);
142   ierr = PetscFree2(arrayf,bss);CHKERRQ(ierr);
143   PetscFunctionReturn(0);
144 }
145 
146 PETSC_INTERN PetscErrorCode DMSetUpGLVisViewer_DMDA(PetscObject oda, PetscViewer viewer)
147 {
148   DM             da = (DM)oda,daview;
149   PetscErrorCode ierr;
150 
151   PetscFunctionBegin;
152   ierr = PetscObjectQuery(oda,"GLVisGraphicsDMDAGhosted",(PetscObject*)&daview);CHKERRQ(ierr);
153   if (!daview) {
154     DMDAGhostedGLVisViewerCtx *dactx;
155     DM                        dacoord = NULL;
156     Vec                       xcoor,xcoorl;
157     PetscBool                 hashocoord = PETSC_FALSE;
158     const PetscInt            *lx,*ly,*lz;
159     PetscInt                  dim,M,N,P,m,n,p,dof,s,i;
160 
161     ierr = PetscNew(&dactx);CHKERRQ(ierr);
162     dactx->ll = PETSC_TRUE; /* default to match elements layout obtained by DMDAGetElements */
163     ierr = PetscOptionsBegin(PetscObjectComm(oda),oda->prefix,"GLVis PetscViewer DMDA Options","PetscViewer");CHKERRQ(ierr);
164     ierr = PetscOptionsBool("-viewer_glvis_dm_da_ll","Left-looking subdomain view",NULL,dactx->ll,&dactx->ll,NULL);CHKERRQ(ierr);
165     ierr = PetscOptionsEnd();CHKERRQ(ierr);
166     /* Create a properly ghosted DMDA to visualize the mesh and the fields associated with */
167     ierr = DMDAGetInfo(da,&dim,&M,&N,&P,&m,&n,&p,&dof,&s,NULL,NULL,NULL,NULL);CHKERRQ(ierr);
168     ierr = DMDAGetOwnershipRanges(da,&lx,&ly,&lz);CHKERRQ(ierr);
169     ierr = PetscObjectQuery((PetscObject)da,"_glvis_mesh_coords",(PetscObject*)&xcoor);CHKERRQ(ierr);
170     if (!xcoor) {
171       ierr = DMGetCoordinates(da,&xcoor);CHKERRQ(ierr);
172     } else {
173       hashocoord = PETSC_TRUE;
174     }
175     ierr = PetscInfo(da,"Creating auxilary DMDA for managing GLVis graphics\n");CHKERRQ(ierr);
176     switch (dim) {
177     case 1:
178       ierr = DMDACreate1d(PetscObjectComm((PetscObject)da),DM_BOUNDARY_NONE,M,dof,1,lx,&daview);CHKERRQ(ierr);
179       if (!hashocoord) {
180         ierr = DMDACreate1d(PetscObjectComm((PetscObject)da),DM_BOUNDARY_NONE,M,1,1,lx,&dacoord);CHKERRQ(ierr);
181       }
182       break;
183     case 2:
184       ierr = DMDACreate2d(PetscObjectComm((PetscObject)da),DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DMDA_STENCIL_BOX,M,N,m,n,dof,1,lx,ly,&daview);CHKERRQ(ierr);
185       if (!hashocoord) {
186         ierr = DMDACreate2d(PetscObjectComm((PetscObject)da),DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DMDA_STENCIL_BOX,M,N,m,n,2,1,lx,ly,&dacoord);CHKERRQ(ierr);
187       }
188       break;
189     case 3:
190       ierr = DMDACreate3d(PetscObjectComm((PetscObject)da),DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DMDA_STENCIL_BOX,M,N,P,m,n,p,dof,1,lx,ly,lz,&daview);CHKERRQ(ierr);
191       if (!hashocoord) {
192         ierr = DMDACreate3d(PetscObjectComm((PetscObject)da),DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DM_BOUNDARY_NONE,DMDA_STENCIL_BOX,M,N,P,m,n,p,3,1,lx,ly,lz,&dacoord);CHKERRQ(ierr);
193       }
194       break;
195     default:
196       SETERRQ1(PetscObjectComm((PetscObject)da),PETSC_ERR_SUP,"Unsupported dimension %D",dim);
197       break;
198     }
199     ierr = DMSetApplicationContext(daview,dactx);CHKERRQ(ierr);
200     ierr = DMSetApplicationContextDestroy(daview,DMDAGhostedDestroyGLVisViewerCtx_Private);CHKERRQ(ierr);
201     ierr = DMSetUp(daview);CHKERRQ(ierr);
202     if (!xcoor) {
203       ierr = DMDASetUniformCoordinates(daview,0.0,1.0,0.0,1.0,0.0,1.0);CHKERRQ(ierr);
204       ierr = DMGetCoordinates(daview,&xcoor);CHKERRQ(ierr);
205     }
206     if (dacoord) {
207       ierr = DMSetUp(dacoord);CHKERRQ(ierr);
208       ierr = DMCreateLocalVector(dacoord,&xcoorl);CHKERRQ(ierr);
209       ierr = DMGlobalToLocalBegin(dacoord,xcoor,INSERT_VALUES,xcoorl);CHKERRQ(ierr);
210       ierr = DMGlobalToLocalEnd(dacoord,xcoor,INSERT_VALUES,xcoorl);CHKERRQ(ierr);
211       ierr = DMDestroy(&dacoord);CHKERRQ(ierr);
212     } else {
213       PetscInt   ien,jen,ken,nc,nl,cdof,deg;
214       char       fecmesh[64];
215       const char *name;
216       PetscBool  flg;
217 
218       ierr = DMDAGetNumElementsGhosted(daview,&ien,&jen,&ken);CHKERRQ(ierr);
219       nc   = ien*(jen>0 ? jen : 1)*(ken>0 ? ken : 1);
220 
221       ierr = VecGetLocalSize(xcoor,&nl);CHKERRQ(ierr);
222       if (nc && nl % nc) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Incompatible local coordinate size %D and number of cells %D",nl,nc);
223       ierr = VecDuplicate(xcoor,&xcoorl);CHKERRQ(ierr);
224       ierr = VecCopy(xcoor,xcoorl);CHKERRQ(ierr);
225       ierr = VecSetDM(xcoorl,NULL);CHKERRQ(ierr);
226       ierr = PetscObjectGetName((PetscObject)xcoor,&name);CHKERRQ(ierr);
227       ierr = PetscStrbeginswith(name,"FiniteElementCollection:",&flg);CHKERRQ(ierr);
228       if (!flg) {
229         deg = 0;
230         if (nc && nl) {
231           cdof = nl/(nc*dim);
232           deg  = 1;
233           while (1) {
234             PetscInt degd = 1;
235             for (i=0;i<dim;i++) degd *= (deg+1);
236             if (degd > cdof) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cell dofs %D",cdof);
237             if (degd == cdof) break;
238             deg++;
239           }
240         }
241         ierr = PetscSNPrintf(fecmesh,sizeof(fecmesh),"FiniteElementCollection: L2_T1_%DD_P%D",dim,deg);CHKERRQ(ierr);
242         ierr = PetscObjectSetName((PetscObject)xcoorl,fecmesh);CHKERRQ(ierr);
243       } else {
244         ierr = PetscObjectSetName((PetscObject)xcoorl,name);CHKERRQ(ierr);
245       }
246     }
247 
248     /* xcoorl is composed with the ghosted DMDA, the ghosted coordinate DMDA (if present) is only available through this vector */
249     ierr = PetscObjectCompose((PetscObject)daview,"GLVisGraphicsCoordsGhosted",(PetscObject)xcoorl);CHKERRQ(ierr);
250     ierr = PetscObjectDereference((PetscObject)xcoorl);CHKERRQ(ierr);
251 
252     /* daview is composed with the original DMDA */
253     ierr = PetscObjectCompose(oda,"GLVisGraphicsDMDAGhosted",(PetscObject)daview);CHKERRQ(ierr);
254     ierr = PetscObjectDereference((PetscObject)daview);CHKERRQ(ierr);
255   }
256 
257   /* customize the viewer if present */
258   if (viewer) {
259     DMDAFieldGLVisViewerCtx   *ctx;
260     DMDAGhostedGLVisViewerCtx *dactx;
261     char                      fec[64];
262     Vec                       xlocal,*Ufield;
263     const char                **dafieldname;
264     char                      **fec_type,**fieldname;
265     PetscInt                  *nlocal,*bss,*dims;
266     PetscInt                  dim,M,N,P,dof,s,i,nf;
267     PetscBool                 bsset;
268 
269     ierr = DMDAGetInfo(daview,&dim,NULL,NULL,NULL,NULL,NULL,NULL,&dof,NULL,NULL,NULL,NULL,NULL);CHKERRQ(ierr);
270     ierr = DMGetApplicationContext(daview,(void**)&dactx);CHKERRQ(ierr);
271     ierr = DMCreateLocalVector(daview,&xlocal);CHKERRQ(ierr);
272     ierr = DMDAGetFieldNames(da,(const char * const **)&dafieldname);CHKERRQ(ierr);
273     ierr = DMDAGetNumVerticesGhosted(daview,&M,&N,&P);CHKERRQ(ierr);
274     ierr = PetscSNPrintf(fec,sizeof(fec),"FiniteElementCollection: H1_%DD_P1",dim);CHKERRQ(ierr);
275     ierr = PetscMalloc6(dof,&fec_type,dof,&nlocal,dof,&bss,dof,&dims,dof,&fieldname,dof,&Ufield);CHKERRQ(ierr);
276     for (i=0;i<dof;i++) bss[i] = 1;
277     nf = dof;
278 
279     ierr = PetscOptionsBegin(PetscObjectComm(oda),oda->prefix,"GLVis PetscViewer DMDA Field options","PetscViewer");CHKERRQ(ierr);
280     ierr = PetscOptionsIntArray("-viewer_glvis_dm_da_bs","Block sizes for subfields; enable vector representation",NULL,bss,&nf,&bsset);CHKERRQ(ierr);
281     ierr = PetscOptionsEnd();CHKERRQ(ierr);
282     if (bsset) {
283       PetscInt t;
284       for (i=0,t=0;i<nf;i++) t += bss[i];
285       if (t != dof) SETERRQ2(PetscObjectComm(oda),PETSC_ERR_USER,"Sum of block sizes %D should equal %D",t,dof);
286     } else nf = dof;
287 
288     for (i=0,s=0;i<nf;i++) {
289       ierr = PetscStrallocpy(fec,&fec_type[i]);CHKERRQ(ierr);
290       if (bss[i] == 1) {
291         ierr = PetscStrallocpy(dafieldname[s],&fieldname[i]);CHKERRQ(ierr);
292       } else {
293         PetscInt b;
294         size_t tlen = 9; /* "Vector-" + end */
295         for (b=0;b<bss[i];b++) {
296           size_t len;
297           ierr = PetscStrlen(dafieldname[s+b],&len);CHKERRQ(ierr);
298           tlen += len + 1; /* field + "-" */
299         }
300         ierr = PetscMalloc1(tlen,&fieldname[i]);CHKERRQ(ierr);
301         ierr = PetscStrcpy(fieldname[i],"Vector-");CHKERRQ(ierr);
302         for (b=0;b<bss[i]-1;b++) {
303           ierr = PetscStrcat(fieldname[i],dafieldname[s+b]);CHKERRQ(ierr);
304           ierr = PetscStrcat(fieldname[i],"-");CHKERRQ(ierr);
305         }
306         ierr = PetscStrcat(fieldname[i],dafieldname[s+b]);CHKERRQ(ierr);
307       }
308       dims[i] = dim;
309       nlocal[i] = M*N*P*bss[i];
310       s += bss[i];
311     }
312 
313     /* the viewer context takes ownership of xlocal and destroys it in DMDAFieldDestroyGLVisViewerCtx_Private */
314     ierr = PetscNew(&ctx);CHKERRQ(ierr);
315     ctx->xlocal = xlocal;
316 
317     /* create work vectors */
318     for (i=0;i<nf;i++) {
319       ierr = VecCreateMPI(PetscObjectComm((PetscObject)da),nlocal[i],PETSC_DECIDE,&Ufield[i]);CHKERRQ(ierr);
320       ierr = PetscObjectSetName((PetscObject)Ufield[i],fieldname[i]);CHKERRQ(ierr);
321       ierr = VecSetBlockSize(Ufield[i],bss[i]);CHKERRQ(ierr);
322       ierr = VecSetDM(Ufield[i],da);CHKERRQ(ierr);
323     }
324 
325     ierr = PetscViewerGLVisSetFields(viewer,nf,(const char**)fec_type,dims,DMDASampleGLVisFields_Private,(PetscObject*)Ufield,ctx,DMDAFieldDestroyGLVisViewerCtx_Private);CHKERRQ(ierr);
326     for (i=0;i<nf;i++) {
327       ierr = PetscFree(fec_type[i]);CHKERRQ(ierr);
328       ierr = PetscFree(fieldname[i]);CHKERRQ(ierr);
329       ierr = VecDestroy(&Ufield[i]);CHKERRQ(ierr);
330     }
331     ierr = PetscFree6(fec_type,nlocal,bss,dims,fieldname,Ufield);CHKERRQ(ierr);
332   }
333   PetscFunctionReturn(0);
334 }
335 
336 static PetscErrorCode DMDAView_GLVis_ASCII(DM dm, PetscViewer viewer)
337 {
338   DM                da,cda;
339   Vec               xcoorl;
340   PetscMPIInt       size;
341   const PetscScalar *array;
342   PetscContainer    glvis_container;
343   PetscInt          dim,sdim,i,vid[8],mid,cid,cdof;
344   PetscInt          sx,sy,sz,ie,je,ke,ien,jen,ken,nel;
345   PetscInt          gsx,gsy,gsz,gm,gn,gp,kst,jst,ist;
346   PetscBool         enabled = PETSC_TRUE, isascii;
347   PetscErrorCode    ierr;
348   const char        *fmt;
349 
350   PetscFunctionBegin;
351   PetscValidHeaderSpecific(dm,DM_CLASSID,1);
352   PetscValidHeaderSpecific(viewer,PETSC_VIEWER_CLASSID,2);
353   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&isascii);CHKERRQ(ierr);
354   if (!isascii) SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Viewer must be of type VIEWERASCII");
355   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)viewer),&size);CHKERRQ(ierr);
356   if (size > 1) SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Use single sequential viewers for parallel visualization");
357   ierr = DMGetDimension(dm,&dim);CHKERRQ(ierr);
358 
359   /* get container: determines if a process visualizes is portion of the data or not */
360   ierr = PetscObjectQuery((PetscObject)viewer,"_glvis_info_container",(PetscObject*)&glvis_container);CHKERRQ(ierr);
361   if (!glvis_container) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_PLIB,"Missing GLVis container");
362   {
363     PetscViewerGLVisInfo glvis_info;
364     ierr = PetscContainerGetPointer(glvis_container,(void**)&glvis_info);CHKERRQ(ierr);
365     enabled = glvis_info->enabled;
366     fmt = glvis_info->fmt;
367   }
368   /* this can happen if we are calling DMView outside of VecView_GLVis */
369   ierr = PetscObjectQuery((PetscObject)dm,"GLVisGraphicsDMDAGhosted",(PetscObject*)&da);CHKERRQ(ierr);
370   if (!da) {ierr = DMSetUpGLVisViewer_DMDA((PetscObject)dm,NULL);CHKERRQ(ierr);}
371   ierr = PetscObjectQuery((PetscObject)dm,"GLVisGraphicsDMDAGhosted",(PetscObject*)&da);CHKERRQ(ierr);
372   if (!da) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_PLIB,"Missing GLVis ghosted DMDA");
373   ierr = DMGetCoordinateDim(da,&sdim);CHKERRQ(ierr);
374 
375   ierr = PetscViewerASCIIPrintf(viewer,"MFEM mesh v1.1\n");CHKERRQ(ierr);
376   ierr = PetscViewerASCIIPrintf(viewer,"\ndimension\n");CHKERRQ(ierr);
377   ierr = PetscViewerASCIIPrintf(viewer,"%D\n",dim);CHKERRQ(ierr);
378 
379   if (!enabled) {
380     ierr = PetscViewerASCIIPrintf(viewer,"\nelements\n");CHKERRQ(ierr);
381     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
382     ierr = PetscViewerASCIIPrintf(viewer,"\nboundary\n");CHKERRQ(ierr);
383     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
384     ierr = PetscViewerASCIIPrintf(viewer,"\nvertices\n");CHKERRQ(ierr);
385     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
386     ierr = PetscViewerASCIIPrintf(viewer,"%D\n",sdim);CHKERRQ(ierr);
387     PetscFunctionReturn(0);
388   }
389 
390   ierr = DMDAGetNumElementsGhosted(da,&ien,&jen,&ken);CHKERRQ(ierr);
391   nel  = ien;
392   if (dim > 1) nel *= jen;
393   if (dim > 2) nel *= ken;
394   ierr = PetscViewerASCIIPrintf(viewer,"\nelements\n");CHKERRQ(ierr);
395   ierr = PetscViewerASCIIPrintf(viewer,"%D\n",nel);CHKERRQ(ierr);
396   switch (dim) {
397   case 1:
398     for (ie = 0; ie < ien; ie++) {
399       vid[0] = ie;
400       vid[1] = ie+1;
401       mid    = 1; /* material id */
402       cid    = 1; /* segment */
403       ierr   = PetscViewerASCIIPrintf(viewer,"%D %D %D %D\n",mid,cid,vid[0],vid[1]);CHKERRQ(ierr);
404     }
405     break;
406   case 2:
407     for (je = 0; je < jen; je++) {
408       for (ie = 0; ie < ien; ie++) {
409         vid[0] =     je*(ien+1) + ie;
410         vid[1] =     je*(ien+1) + ie+1;
411         vid[2] = (je+1)*(ien+1) + ie+1;
412         vid[3] = (je+1)*(ien+1) + ie;
413         mid    = 1; /* material id */
414         cid    = 3; /* quad */
415         ierr   = PetscViewerASCIIPrintf(viewer,"%D %D %D %D %D %D\n",mid,cid,vid[0],vid[1],vid[2],vid[3]);CHKERRQ(ierr);
416       }
417     }
418     break;
419   case 3:
420     for (ke = 0; ke < ken; ke++) {
421       for (je = 0; je < jen; je++) {
422         for (ie = 0; ie < ien; ie++) {
423           vid[0] =     ke*(jen+1)*(ien+1) +     je*(ien+1) + ie;
424           vid[1] =     ke*(jen+1)*(ien+1) +     je*(ien+1) + ie+1;
425           vid[2] =     ke*(jen+1)*(ien+1) + (je+1)*(ien+1) + ie+1;
426           vid[3] =     ke*(jen+1)*(ien+1) + (je+1)*(ien+1) + ie;
427           vid[4] = (ke+1)*(jen+1)*(ien+1) +     je*(ien+1) + ie;
428           vid[5] = (ke+1)*(jen+1)*(ien+1) +     je*(ien+1) + ie+1;
429           vid[6] = (ke+1)*(jen+1)*(ien+1) + (je+1)*(ien+1) + ie+1;
430           vid[7] = (ke+1)*(jen+1)*(ien+1) + (je+1)*(ien+1) + ie;
431           mid    = 1; /* material id */
432           cid    = 5; /* hex */
433           ierr   = PetscViewerASCIIPrintf(viewer,"%D %D %D %D %D %D %D %D %D %D\n",mid,cid,vid[0],vid[1],vid[2],vid[3],vid[4],vid[5],vid[6],vid[7]);CHKERRQ(ierr);
434         }
435       }
436     }
437     break;
438   default:
439     SETERRQ1(PetscObjectComm((PetscObject)da),PETSC_ERR_SUP,"Unsupported dimension %D",dim);
440     break;
441   }
442   ierr = PetscViewerASCIIPrintf(viewer,"\nboundary\n");CHKERRQ(ierr);
443   ierr = PetscViewerASCIIPrintf(viewer,"%D\n",0);CHKERRQ(ierr);
444 
445   /* vertex coordinates */
446   ierr = PetscObjectQuery((PetscObject)da,"GLVisGraphicsCoordsGhosted",(PetscObject*)&xcoorl);CHKERRQ(ierr);
447   if (!xcoorl) SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_PLIB,"Missing GLVis ghosted coords");
448   ierr = DMDAGetNumVerticesGhosted(da,&ien,&jen,&ken);CHKERRQ(ierr);
449   ierr = PetscViewerASCIIPrintf(viewer,"\nvertices\n");CHKERRQ(ierr);
450   ierr = PetscViewerASCIIPrintf(viewer,"%D\n",ien*jen*ken);CHKERRQ(ierr);
451   if (nel) {
452     ierr = VecGetDM(xcoorl,&cda);CHKERRQ(ierr);
453     ierr = VecGetArrayRead(xcoorl,&array);CHKERRQ(ierr);
454     if (!cda) { /* HO viz */
455       const char *fecname;
456       PetscInt   nc,nl;
457 
458       ierr = PetscObjectGetName((PetscObject)xcoorl,&fecname);CHKERRQ(ierr);
459       ierr = PetscViewerASCIIPrintf(viewer,"nodes\n");CHKERRQ(ierr);
460       ierr = PetscViewerASCIIPrintf(viewer,"FiniteElementSpace\n");CHKERRQ(ierr);
461       ierr = PetscViewerASCIIPrintf(viewer,"%s\n",fecname);CHKERRQ(ierr);
462       ierr = PetscViewerASCIIPrintf(viewer,"VDim: %D\n",sdim);CHKERRQ(ierr);
463       ierr = PetscViewerASCIIPrintf(viewer,"Ordering: 1\n\n");CHKERRQ(ierr); /*Ordering::byVDIM*/
464       /* L2 coordinates */
465       ierr = DMDAGetNumElementsGhosted(da,&ien,&jen,&ken);CHKERRQ(ierr);
466       ierr = VecGetLocalSize(xcoorl,&nl);CHKERRQ(ierr);
467       nc   = ien*(jen>0 ? jen : 1)*(ken>0 ? ken : 1);
468       cdof = nc ? nl/nc : 0;
469       if (!ien) ien++;
470       if (!jen) jen++;
471       if (!ken) ken++;
472       ist = jst = kst = 0;
473       gm = ien;
474       gn = jen;
475       gp = ken;
476     } else {
477       DMDAGhostedGLVisViewerCtx *dactx;
478 
479       ierr = DMGetApplicationContext(da,(void**)&dactx);CHKERRQ(ierr);
480       ierr = PetscViewerASCIIPrintf(viewer,"%D\n",sdim);CHKERRQ(ierr);
481       cdof = sdim;
482       ierr = DMDAGetCorners(da,&sx,&sy,&sz,NULL,NULL,NULL);CHKERRQ(ierr);
483       ierr = DMDAGetGhostCorners(da,&gsx,&gsy,&gsz,&gm,&gn,&gp);CHKERRQ(ierr);
484       if (dactx->ll) {
485         kst = jst = ist = 0;
486       } else {
487         kst  = gsz != sz ? 1 : 0;
488         jst  = gsy != sy ? 1 : 0;
489         ist  = gsx != sx ? 1 : 0;
490       }
491     }
492     for (ke = kst; ke < kst + ken; ke++) {
493       for (je = jst; je < jst + jen; je++) {
494         for (ie = ist; ie < ist + ien; ie++) {
495           PetscInt c;
496 
497           i = ke * gm * gn + je * gm + ie;
498           for (c=0;c<cdof/sdim;c++) {
499             PetscInt d;
500             for (d=0;d<sdim;d++) {
501               ierr = PetscViewerASCIIPrintf(viewer,fmt,PetscRealPart(array[cdof*i+c*sdim+d]));CHKERRQ(ierr);
502             }
503             ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
504           }
505         }
506       }
507     }
508     ierr = VecRestoreArrayRead(xcoorl,&array);CHKERRQ(ierr);
509   }
510   PetscFunctionReturn(0);
511 }
512 
513 PetscErrorCode DMView_DA_GLVis(DM dm, PetscViewer viewer)
514 {
515   PetscErrorCode ierr;
516   PetscFunctionBegin;
517   ierr = DMView_GLVis(dm,viewer,DMDAView_GLVis_ASCII);CHKERRQ(ierr);
518   PetscFunctionReturn(0);
519 }
520