xref: /petsc/src/dm/impls/plex/plexvtu.c (revision ce94432eddcd14845bc7e8083b7f8ea723b9bf7d)
1552f7358SJed Brown #include <petsc-private/pleximpl.h>
2552f7358SJed Brown #include <../src/sys/classes/viewer/impls/vtk/vtkvimpl.h>
3552f7358SJed Brown 
4552f7358SJed Brown typedef struct {
5552f7358SJed Brown   PetscInt nvertices;
6552f7358SJed Brown   PetscInt ncells;
7552f7358SJed Brown   PetscInt nconn;               /* number of entries in cell->vertex connectivity array */
8552f7358SJed Brown } PieceInfo;
9552f7358SJed Brown 
10552f7358SJed Brown #if defined(PETSC_USE_REAL_SINGLE)
11552f7358SJed Brown static const char precision[] = "Float32";
12552f7358SJed Brown #elif defined(PETSC_USE_REAL_DOUBLE)
13552f7358SJed Brown static const char precision[] = "Float64";
14552f7358SJed Brown #else
15552f7358SJed Brown static const char precision[] = "UnknownPrecision";
16552f7358SJed Brown #endif
17552f7358SJed Brown 
18552f7358SJed Brown #undef __FUNCT__
19552f7358SJed Brown #define __FUNCT__ "TransferWrite"
20552f7358SJed Brown static PetscErrorCode TransferWrite(PetscViewer viewer,FILE *fp,PetscMPIInt srank,PetscMPIInt root,const void *send,void *recv,PetscMPIInt count,PetscDataType datatype,PetscMPIInt tag)
21552f7358SJed Brown {
22552f7358SJed Brown   PetscMPIInt    rank;
23552f7358SJed Brown   PetscErrorCode ierr;
24*ce94432eSBarry Smith   MPI_Comm       comm;
25552f7358SJed Brown   MPI_Datatype   mpidatatype;
26552f7358SJed Brown 
27552f7358SJed Brown   PetscFunctionBegin;
28*ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
29552f7358SJed Brown   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
30552f7358SJed Brown   ierr = PetscDataTypeToMPIDataType(datatype,&mpidatatype);CHKERRQ(ierr);
31552f7358SJed Brown 
32552f7358SJed Brown   if (rank == srank && rank != root) {
33552f7358SJed Brown     ierr = MPI_Send((void*)send,count,mpidatatype,root,tag,comm);CHKERRQ(ierr);
34552f7358SJed Brown   } else if (rank == root) {
35552f7358SJed Brown     const void *buffer;
36552f7358SJed Brown     if (root == srank) {        /* self */
37552f7358SJed Brown       buffer = send;
38552f7358SJed Brown     } else {
39552f7358SJed Brown       MPI_Status  status;
40552f7358SJed Brown       PetscMPIInt nrecv;
41552f7358SJed Brown       ierr = MPI_Recv(recv,count,mpidatatype,srank,tag,comm,&status);CHKERRQ(ierr);
42552f7358SJed Brown       ierr = MPI_Get_count(&status,mpidatatype,&nrecv);CHKERRQ(ierr);
43552f7358SJed Brown       if (count != nrecv) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Array size mismatch");
44552f7358SJed Brown       buffer = recv;
45552f7358SJed Brown     }
46552f7358SJed Brown     ierr = PetscViewerVTKFWrite(viewer,fp,buffer,count,datatype);CHKERRQ(ierr);
47552f7358SJed Brown   }
48552f7358SJed Brown   PetscFunctionReturn(0);
49552f7358SJed Brown }
50552f7358SJed Brown 
51552f7358SJed Brown #undef __FUNCT__
52552f7358SJed Brown #define __FUNCT__ "DMPlexGetVTKConnectivity"
53552f7358SJed Brown static PetscErrorCode DMPlexGetVTKConnectivity(DM dm,PieceInfo *piece,PetscVTKInt **oconn,PetscVTKInt **ooffsets,PetscVTKType **otypes)
54552f7358SJed Brown {
55552f7358SJed Brown   PetscErrorCode ierr;
56552f7358SJed Brown   PetscVTKInt    *conn,*offsets;
57552f7358SJed Brown   PetscVTKType   *types;
58552f7358SJed Brown   PetscInt       dim,vStart,vEnd,cStart,cEnd,pStart,pEnd,cellHeight,cMax,numLabelCells,hasLabel,c,v,countcell,countconn;
59552f7358SJed Brown 
60552f7358SJed Brown   PetscFunctionBegin;
61552f7358SJed Brown   ierr = PetscMalloc3(piece->nconn,PetscVTKInt,&conn,piece->ncells,PetscVTKInt,&offsets,piece->ncells,PetscVTKType,&types);CHKERRQ(ierr);
62552f7358SJed Brown 
63552f7358SJed Brown   ierr = DMPlexGetDimension(dm,&dim);CHKERRQ(ierr);
64552f7358SJed Brown   ierr = DMPlexGetChart(dm,&pStart,&pEnd);CHKERRQ(ierr);
65552f7358SJed Brown   ierr = DMPlexGetVTKCellHeight(dm, &cellHeight);CHKERRQ(ierr);
66552f7358SJed Brown   ierr = DMPlexGetHeightStratum(dm, cellHeight, &cStart, &cEnd);CHKERRQ(ierr);
67552f7358SJed Brown   ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
680298fd71SBarry Smith   ierr = DMPlexGetHybridBounds(dm, &cMax, NULL, NULL, NULL);CHKERRQ(ierr);
698865f1eaSKarl Rupp   if (cMax >= 0) cEnd = PetscMin(cEnd, cMax);
70552f7358SJed Brown   ierr     = DMPlexGetStratumSize(dm, "vtk", 1, &numLabelCells);CHKERRQ(ierr);
71552f7358SJed Brown   hasLabel = numLabelCells > 0 ? PETSC_TRUE : PETSC_FALSE;
72552f7358SJed Brown 
73552f7358SJed Brown   countcell = 0;
74552f7358SJed Brown   countconn = 0;
75552f7358SJed Brown   for (c = cStart; c < cEnd; ++c) {
760298fd71SBarry Smith     PetscInt *closure = NULL;
77552f7358SJed Brown     PetscInt  closureSize,nverts,celltype,startoffset;
78552f7358SJed Brown 
79552f7358SJed Brown     if (hasLabel) {
80552f7358SJed Brown       PetscInt value;
81552f7358SJed Brown 
82552f7358SJed Brown       ierr = DMPlexGetLabelValue(dm, "vtk", c, &value);CHKERRQ(ierr);
83552f7358SJed Brown       if (value != 1) continue;
84552f7358SJed Brown     }
85552f7358SJed Brown     startoffset = countconn;
86552f7358SJed Brown     ierr        = DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
87552f7358SJed Brown     for (v = 0; v < closureSize*2; v += 2) {
88552f7358SJed Brown       if ((closure[v] >= vStart) && (closure[v] < vEnd)) {
89552f7358SJed Brown         conn[countconn++] = closure[v] - vStart;
90552f7358SJed Brown       }
91552f7358SJed Brown     }
92552f7358SJed Brown     ierr = DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
938865f1eaSKarl Rupp 
94552f7358SJed Brown     offsets[countcell] = countconn;
958865f1eaSKarl Rupp 
96552f7358SJed Brown     nverts = countconn - startoffset;
97552f7358SJed Brown     ierr   = DMPlexVTKGetCellType(dm,dim,nverts,&celltype);CHKERRQ(ierr);
988865f1eaSKarl Rupp 
99552f7358SJed Brown     types[countcell] = celltype;
100552f7358SJed Brown     countcell++;
101552f7358SJed Brown   }
102552f7358SJed Brown   if (countcell != piece->ncells) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Inconsistent cell count");
103552f7358SJed Brown   if (countconn != piece->nconn) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Inconsistent connectivity count");
104552f7358SJed Brown   *oconn    = conn;
105552f7358SJed Brown   *ooffsets = offsets;
106552f7358SJed Brown   *otypes   = types;
107552f7358SJed Brown   PetscFunctionReturn(0);
108552f7358SJed Brown }
109552f7358SJed Brown 
110552f7358SJed Brown #undef __FUNCT__
111552f7358SJed Brown #define __FUNCT__ "DMPlexVTKWriteAll_VTU"
112552f7358SJed Brown /*
113552f7358SJed Brown   Write all fields that have been provided to the viewer
114552f7358SJed Brown   Multi-block XML format with binary appended data.
115552f7358SJed Brown */
116552f7358SJed Brown PetscErrorCode DMPlexVTKWriteAll_VTU(DM dm,PetscViewer viewer)
117552f7358SJed Brown {
118*ce94432eSBarry Smith   MPI_Comm                 comm;
119552f7358SJed Brown   PetscViewer_VTK          *vtk = (PetscViewer_VTK*)viewer->data;
120552f7358SJed Brown   PetscViewerVTKObjectLink link;
121552f7358SJed Brown   FILE                     *fp;
122552f7358SJed Brown   PetscMPIInt              rank,size,tag;
123552f7358SJed Brown   PetscErrorCode           ierr;
124552f7358SJed Brown   PetscInt                 dim,cellHeight,cStart,cEnd,vStart,vEnd,cMax,numLabelCells,hasLabel,c,v,r,i;
1250298fd71SBarry Smith   PieceInfo                piece,*gpiece = NULL;
1260298fd71SBarry Smith   void                     *buffer = NULL;
127552f7358SJed Brown 
128552f7358SJed Brown   PetscFunctionBegin;
129*ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)dm,&comm);CHKERRQ(ierr);
130552f7358SJed Brown #if defined(PETSC_USE_COMPLEX)
131*ce94432eSBarry Smith   SETERRQ(PetscObjectComm((PetscObject)dm),PETSC_ERR_SUP,"Complex values not supported");
132552f7358SJed Brown #endif
133552f7358SJed Brown   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
134552f7358SJed Brown   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
135552f7358SJed Brown   tag  = ((PetscObject)viewer)->tag;
136552f7358SJed Brown 
137552f7358SJed Brown   ierr = PetscFOpen(comm,vtk->filename,"wb",&fp);CHKERRQ(ierr);
138552f7358SJed Brown   ierr = PetscFPrintf(comm,fp,"<?xml version=\"1.0\"?>\n");CHKERRQ(ierr);
139519f805aSKarl Rupp #if defined(PETSC_WORDS_BIGENDIAN)
140552f7358SJed Brown   ierr = PetscFPrintf(comm,fp,"<VTKFile type=\"UnstructuredGrid\" version=\"0.1\" byte_order=\"BigEndian\">\n");CHKERRQ(ierr);
141552f7358SJed Brown #else
142552f7358SJed Brown   ierr = PetscFPrintf(comm,fp,"<VTKFile type=\"UnstructuredGrid\" version=\"0.1\" byte_order=\"LittleEndian\">\n");CHKERRQ(ierr);
143552f7358SJed Brown #endif
144552f7358SJed Brown   ierr = PetscFPrintf(comm,fp,"  <UnstructuredGrid>\n");CHKERRQ(ierr);
145552f7358SJed Brown 
146552f7358SJed Brown   ierr = DMPlexGetDimension(dm, &dim);CHKERRQ(ierr);
147552f7358SJed Brown   ierr = DMPlexGetVTKCellHeight(dm, &cellHeight);CHKERRQ(ierr);
148552f7358SJed Brown   ierr = DMPlexGetHeightStratum(dm, cellHeight, &cStart, &cEnd);CHKERRQ(ierr);
149552f7358SJed Brown   ierr = DMPlexGetDepthStratum(dm, 0, &vStart, &vEnd);CHKERRQ(ierr);
1500298fd71SBarry Smith   ierr = DMPlexGetHybridBounds(dm, &cMax, NULL, NULL, NULL);CHKERRQ(ierr);
1518865f1eaSKarl Rupp   if (cMax >= 0) cEnd = PetscMin(cEnd, cMax);
152552f7358SJed Brown   ierr = DMPlexGetStratumSize(dm, "vtk", 1, &numLabelCells);CHKERRQ(ierr);
1538865f1eaSKarl Rupp 
154552f7358SJed Brown   hasLabel        = numLabelCells > 0 ? PETSC_TRUE : PETSC_FALSE;
155552f7358SJed Brown   piece.nvertices = vEnd - vStart;
156552f7358SJed Brown   piece.ncells    = 0;
157552f7358SJed Brown   piece.nconn     = 0;
158552f7358SJed Brown   for (c = cStart; c < cEnd; ++c) {
1590298fd71SBarry Smith     PetscInt *closure = NULL;
160552f7358SJed Brown     PetscInt closureSize;
161552f7358SJed Brown 
162552f7358SJed Brown     if (hasLabel) {
163552f7358SJed Brown       PetscInt value;
164552f7358SJed Brown 
165552f7358SJed Brown       ierr = DMPlexGetLabelValue(dm, "vtk", c, &value);CHKERRQ(ierr);
166552f7358SJed Brown       if (value != 1) continue;
167552f7358SJed Brown     }
168552f7358SJed Brown     ierr = DMPlexGetTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
169552f7358SJed Brown     for (v = 0; v < closureSize*2; v += 2) {
170552f7358SJed Brown       if ((closure[v] >= vStart) && (closure[v] < vEnd)) piece.nconn++;
171552f7358SJed Brown     }
172552f7358SJed Brown     ierr = DMPlexRestoreTransitiveClosure(dm, c, PETSC_TRUE, &closureSize, &closure);CHKERRQ(ierr);
173552f7358SJed Brown     piece.ncells++;
174552f7358SJed Brown   }
175552f7358SJed Brown   if (!rank) {ierr = PetscMalloc(size*sizeof(piece),&gpiece);CHKERRQ(ierr);}
176552f7358SJed Brown   ierr = MPI_Gather((PetscInt*)&piece,sizeof(piece)/sizeof(PetscInt),MPIU_INT,(PetscInt*)gpiece,sizeof(piece)/sizeof(PetscInt),MPIU_INT,0,comm);CHKERRQ(ierr);
177552f7358SJed Brown 
178552f7358SJed Brown   /*
179552f7358SJed Brown    * Write file header
180552f7358SJed Brown    */
181552f7358SJed Brown   if (!rank) {
182552f7358SJed Brown     PetscInt boffset = 0;
183552f7358SJed Brown 
184552f7358SJed Brown     for (r=0; r<size; r++) {
185552f7358SJed Brown       ierr = PetscFPrintf(PETSC_COMM_SELF,fp,"    <Piece NumberOfPoints=\"%D\" NumberOfCells=\"%D\">\n",gpiece[r].nvertices,gpiece[r].ncells);CHKERRQ(ierr);
186552f7358SJed Brown       /* Coordinate positions */
187552f7358SJed Brown       ierr     = PetscFPrintf(PETSC_COMM_SELF,fp,"      <Points>\n");CHKERRQ(ierr);
188552f7358SJed Brown       ierr     = PetscFPrintf(PETSC_COMM_SELF,fp,"        <DataArray type=\"%s\" Name=\"Position\" NumberOfComponents=\"3\" format=\"appended\" offset=\"%D\" />\n",precision,boffset);CHKERRQ(ierr);
189552f7358SJed Brown       boffset += gpiece[r].nvertices*3*sizeof(PetscScalar) + sizeof(int);
190552f7358SJed Brown       ierr     = PetscFPrintf(PETSC_COMM_SELF,fp,"      </Points>\n");CHKERRQ(ierr);
191552f7358SJed Brown       /* Cell connectivity */
192552f7358SJed Brown       ierr     = PetscFPrintf(PETSC_COMM_SELF,fp,"      <Cells>\n");CHKERRQ(ierr);
193552f7358SJed Brown       ierr     = PetscFPrintf(PETSC_COMM_SELF,fp,"        <DataArray type=\"Int32\" Name=\"connectivity\" NumberOfComponents=\"1\" format=\"appended\" offset=\"%D\" />\n",boffset);CHKERRQ(ierr);
194552f7358SJed Brown       boffset += gpiece[r].nconn*sizeof(PetscInt) + sizeof(int);
195552f7358SJed Brown       ierr     = PetscFPrintf(PETSC_COMM_SELF,fp,"        <DataArray type=\"Int32\" Name=\"offsets\"      NumberOfComponents=\"1\" format=\"appended\" offset=\"%D\" />\n",boffset);CHKERRQ(ierr);
196552f7358SJed Brown       boffset += gpiece[r].ncells*sizeof(PetscInt) + sizeof(int);
197552f7358SJed Brown       ierr     = PetscFPrintf(PETSC_COMM_SELF,fp,"        <DataArray type=\"UInt8\" Name=\"types\"        NumberOfComponents=\"1\" format=\"appended\" offset=\"%D\" />\n",boffset);CHKERRQ(ierr);
198552f7358SJed Brown       boffset += gpiece[r].ncells*sizeof(unsigned char) + sizeof(int);
199552f7358SJed Brown       ierr     = PetscFPrintf(PETSC_COMM_SELF,fp,"      </Cells>\n");CHKERRQ(ierr);
200552f7358SJed Brown 
201552f7358SJed Brown       /*
202552f7358SJed Brown        * Cell Data headers
203552f7358SJed Brown        */
204552f7358SJed Brown       ierr     = PetscFPrintf(PETSC_COMM_SELF,fp,"      <CellData>\n");CHKERRQ(ierr);
205552f7358SJed Brown       ierr     = PetscFPrintf(PETSC_COMM_SELF,fp,"        <DataArray type=\"Int32\" Name=\"Rank\" NumberOfComponents=\"1\" format=\"appended\" offset=\"%D\" />\n",boffset);CHKERRQ(ierr);
206552f7358SJed Brown       boffset += gpiece[r].ncells*sizeof(int) + sizeof(int);
207552f7358SJed Brown       /* all the vectors */
208552f7358SJed Brown       for (link=vtk->link; link; link=link->next) {
209552f7358SJed Brown         Vec        X = (Vec)link->vec;
210552f7358SJed Brown         PetscInt   bs;
211552f7358SJed Brown         const char *vecname = "";
212552f7358SJed Brown         if ((link->ft != PETSC_VTK_CELL_FIELD) && (link->ft != PETSC_VTK_CELL_VECTOR_FIELD)) continue;
213552f7358SJed Brown         if (((PetscObject)X)->name || link != vtk->link) { /* If the object is already named, use it. If it is past the first link, name it to disambiguate. */
214552f7358SJed Brown           ierr = PetscObjectGetName((PetscObject)X,&vecname);CHKERRQ(ierr);
215552f7358SJed Brown         }
216552f7358SJed Brown         ierr = PetscSectionGetDof(dm->defaultSection,cStart,&bs);CHKERRQ(ierr);
217552f7358SJed Brown         for (i=0; i<bs; i++) {
218552f7358SJed Brown           char       buf[256];
2190298fd71SBarry Smith           const char *fieldname = NULL;
220552f7358SJed Brown           /* ierr = DMDAGetFieldName(da,i,&fieldname);CHKERRQ(ierr); */
221552f7358SJed Brown           if (!fieldname) {
222552f7358SJed Brown             ierr      = PetscSNPrintf(buf,sizeof(buf),"Unnamed%D",i);CHKERRQ(ierr);
223552f7358SJed Brown             fieldname = buf;
224552f7358SJed Brown           }
225552f7358SJed Brown           ierr     = PetscFPrintf(comm,fp,"        <DataArray type=\"%s\" Name=\"%s%s\" NumberOfComponents=\"1\" format=\"appended\" offset=\"%D\" />\n",precision,vecname,fieldname,boffset);CHKERRQ(ierr);
226552f7358SJed Brown           boffset += gpiece[r].ncells*sizeof(PetscScalar) + sizeof(int);
227552f7358SJed Brown         }
228552f7358SJed Brown       }
229552f7358SJed Brown       ierr = PetscFPrintf(PETSC_COMM_SELF,fp,"      </CellData>\n");CHKERRQ(ierr);
230552f7358SJed Brown 
231552f7358SJed Brown       /*
232552f7358SJed Brown        * Point Data headers
233552f7358SJed Brown        */
234552f7358SJed Brown       ierr = PetscFPrintf(PETSC_COMM_SELF,fp,"      <PointData>\n");CHKERRQ(ierr);
235552f7358SJed Brown       for (link=vtk->link; link; link=link->next) {
236552f7358SJed Brown         Vec        X = (Vec)link->vec;
237552f7358SJed Brown         PetscInt   bs;
238552f7358SJed Brown         const char *vecname = "";
239552f7358SJed Brown         if ((link->ft != PETSC_VTK_POINT_FIELD) && (link->ft != PETSC_VTK_POINT_VECTOR_FIELD)) continue;
240552f7358SJed Brown         if (((PetscObject)X)->name || link != vtk->link) { /* If the object is already named, use it. If it is past the first link, name it to disambiguate. */
241552f7358SJed Brown           ierr = PetscObjectGetName((PetscObject)X,&vecname);CHKERRQ(ierr);
242552f7358SJed Brown         }
243552f7358SJed Brown         ierr = PetscSectionGetDof(dm->defaultSection,vStart,&bs);CHKERRQ(ierr);
244552f7358SJed Brown         for (i=0; i<bs; i++) {
245552f7358SJed Brown           char fieldname[256];
246552f7358SJed Brown           ierr     = PetscSNPrintf(fieldname,sizeof(fieldname),"Point%D",i);CHKERRQ(ierr);
247552f7358SJed Brown           ierr     = PetscFPrintf(comm,fp,"        <DataArray type=\"%s\" Name=\"%s%s\" NumberOfComponents=\"1\" format=\"appended\" offset=\"%D\" />\n",precision,vecname,fieldname,boffset);CHKERRQ(ierr);
248552f7358SJed Brown           boffset += gpiece[r].nvertices*sizeof(PetscScalar) + sizeof(int);
249552f7358SJed Brown         }
250552f7358SJed Brown       }
251552f7358SJed Brown       ierr = PetscFPrintf(PETSC_COMM_SELF,fp,"      </PointData>\n");CHKERRQ(ierr);
252552f7358SJed Brown 
253552f7358SJed Brown       ierr = PetscFPrintf(PETSC_COMM_SELF,fp,"    </Piece>\n");CHKERRQ(ierr);
254552f7358SJed Brown     }
255552f7358SJed Brown   }
256552f7358SJed Brown 
257552f7358SJed Brown   ierr = PetscFPrintf(comm,fp,"  </UnstructuredGrid>\n");CHKERRQ(ierr);
258552f7358SJed Brown   ierr = PetscFPrintf(comm,fp,"  <AppendedData encoding=\"raw\">\n");CHKERRQ(ierr);
259552f7358SJed Brown   ierr = PetscFPrintf(comm,fp,"_");CHKERRQ(ierr);
260552f7358SJed Brown 
261552f7358SJed Brown   if (!rank) {
262552f7358SJed Brown     PetscInt maxsize = 0;
263552f7358SJed Brown     for (r=0; r<size; r++) {
264552f7358SJed Brown       maxsize = PetscMax(maxsize, (PetscInt) (gpiece[r].nvertices*3*sizeof(PetscScalar)));
265552f7358SJed Brown       maxsize = PetscMax(maxsize, (PetscInt) (gpiece[r].ncells*sizeof(PetscScalar)));
266552f7358SJed Brown       maxsize = PetscMax(maxsize, (PetscInt) (gpiece[r].nconn*sizeof(PetscVTKInt)));
267552f7358SJed Brown     }
268552f7358SJed Brown     ierr = PetscMalloc(maxsize,&buffer);CHKERRQ(ierr);
269552f7358SJed Brown   }
270552f7358SJed Brown   for (r=0; r<size; r++) {
271552f7358SJed Brown     if (r == rank) {
272552f7358SJed Brown       PetscInt nsend;
273552f7358SJed Brown       {                         /* Position */
274552f7358SJed Brown         const PetscScalar *x;
2750298fd71SBarry Smith         PetscScalar       *y = NULL;
276552f7358SJed Brown         Vec               coords;
277552f7358SJed Brown         nsend = piece.nvertices*3;
278552f7358SJed Brown         ierr  = DMGetCoordinatesLocal(dm,&coords);CHKERRQ(ierr);
279552f7358SJed Brown         ierr  = VecGetArrayRead(coords,&x);CHKERRQ(ierr);
280552f7358SJed Brown         if (dim != 3) {
281552f7358SJed Brown           ierr = PetscMalloc(piece.nvertices*3*sizeof(PetscScalar),&y);CHKERRQ(ierr);
282552f7358SJed Brown           for (i=0; i<piece.nvertices; i++) {
283552f7358SJed Brown             y[i*3+0] = x[i*dim+0];
284552f7358SJed Brown             y[i*3+1] = (dim > 1) ? x[i*dim+1] : 0;
285552f7358SJed Brown             y[i*3+2] = 0;
286552f7358SJed Brown           }
287552f7358SJed Brown         }
288552f7358SJed Brown         ierr = TransferWrite(viewer,fp,r,0,y ? y : x,buffer,nsend,PETSC_SCALAR,tag);CHKERRQ(ierr);
289552f7358SJed Brown         ierr = PetscFree(y);CHKERRQ(ierr);
290552f7358SJed Brown         ierr = VecRestoreArrayRead(coords,&x);CHKERRQ(ierr);
291552f7358SJed Brown       }
292552f7358SJed Brown       {                           /* Connectivity, offsets, types */
2930298fd71SBarry Smith         PetscVTKInt  *connectivity = NULL,*offsets;
294552f7358SJed Brown         PetscVTKType *types;
295552f7358SJed Brown         ierr = DMPlexGetVTKConnectivity(dm,&piece,&connectivity,&offsets,&types);CHKERRQ(ierr);
296552f7358SJed Brown         ierr = TransferWrite(viewer,fp,r,0,connectivity,buffer,piece.nconn,PETSC_INT32,tag);CHKERRQ(ierr);
297552f7358SJed Brown         ierr = TransferWrite(viewer,fp,r,0,offsets,buffer,piece.ncells,PETSC_INT32,tag);CHKERRQ(ierr);
298552f7358SJed Brown         ierr = TransferWrite(viewer,fp,r,0,types,buffer,piece.ncells,PETSC_UINT8,tag);CHKERRQ(ierr);
299552f7358SJed Brown         ierr = PetscFree3(connectivity,offsets,types);CHKERRQ(ierr);
300552f7358SJed Brown       }
301552f7358SJed Brown       {                         /* Owners (cell data) */
302552f7358SJed Brown         PetscVTKInt *owners;
303552f7358SJed Brown         ierr = PetscMalloc(piece.ncells*sizeof(PetscVTKInt),&owners);CHKERRQ(ierr);
304552f7358SJed Brown         for (i=0; i<piece.ncells; i++) owners[i] = rank;
305552f7358SJed Brown         ierr = TransferWrite(viewer,fp,r,0,owners,buffer,piece.ncells,PETSC_INT32,tag);CHKERRQ(ierr);
306552f7358SJed Brown         ierr = PetscFree(owners);CHKERRQ(ierr);
307552f7358SJed Brown       }
308552f7358SJed Brown       /* Cell data */
309552f7358SJed Brown       for (link=vtk->link; link; link=link->next) {
310552f7358SJed Brown         Vec               X = (Vec)link->vec;
311552f7358SJed Brown         const PetscScalar *x;
312552f7358SJed Brown         PetscScalar       *y;
313552f7358SJed Brown         PetscInt          bs;
314552f7358SJed Brown         if ((link->ft != PETSC_VTK_CELL_FIELD) && (link->ft != PETSC_VTK_CELL_VECTOR_FIELD)) continue;
315552f7358SJed Brown         ierr = PetscSectionGetDof(dm->defaultSection,cStart,&bs);CHKERRQ(ierr);
316552f7358SJed Brown         ierr = VecGetArrayRead(X,&x);CHKERRQ(ierr);
317552f7358SJed Brown         ierr = PetscMalloc(piece.ncells*sizeof(PetscScalar),&y);CHKERRQ(ierr);
318552f7358SJed Brown         for (i=0; i<bs; i++) {
319552f7358SJed Brown           PetscInt cnt;
320552f7358SJed Brown           for (c=cStart,cnt=0; c<cEnd; c++) {
321552f7358SJed Brown             const PetscScalar *xpoint;
322552f7358SJed Brown             if (hasLabel) {     /* Ignore some cells */
323552f7358SJed Brown               PetscInt value;
324552f7358SJed Brown               ierr = DMPlexGetLabelValue(dm, "vtk", c, &value);CHKERRQ(ierr);
325552f7358SJed Brown               if (value != 1) continue;
326552f7358SJed Brown             }
327552f7358SJed Brown             ierr     = DMPlexPointLocalRead(dm,c,x,&xpoint);CHKERRQ(ierr);
328552f7358SJed Brown             y[cnt++] = xpoint[i];
329552f7358SJed Brown           }
330552f7358SJed Brown           if (cnt != piece.ncells) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Count does not match");
331552f7358SJed Brown           ierr = TransferWrite(viewer,fp,r,0,y,buffer,piece.ncells,PETSC_SCALAR,tag);CHKERRQ(ierr);
332552f7358SJed Brown         }
333552f7358SJed Brown         ierr = PetscFree(y);CHKERRQ(ierr);
334552f7358SJed Brown         ierr = VecRestoreArrayRead(X,&x);CHKERRQ(ierr);
335552f7358SJed Brown       }
336552f7358SJed Brown 
337552f7358SJed Brown       for (link=vtk->link; link; link=link->next) {
338552f7358SJed Brown         Vec               X = (Vec)link->vec;
339552f7358SJed Brown         const PetscScalar *x;
340552f7358SJed Brown         PetscScalar       *y;
341552f7358SJed Brown         PetscInt          bs;
342552f7358SJed Brown         if ((link->ft != PETSC_VTK_POINT_FIELD) && (link->ft != PETSC_VTK_POINT_VECTOR_FIELD)) continue;
343552f7358SJed Brown         ierr = PetscSectionGetDof(dm->defaultSection,vStart,&bs);CHKERRQ(ierr);
344552f7358SJed Brown         ierr = VecGetArrayRead(X,&x);CHKERRQ(ierr);
345552f7358SJed Brown         ierr = PetscMalloc(piece.nvertices*sizeof(PetscScalar),&y);CHKERRQ(ierr);
346552f7358SJed Brown         for (i=0; i<bs; i++) {
347552f7358SJed Brown           PetscInt cnt;
348552f7358SJed Brown           for (v=vStart,cnt=0; v<vEnd; v++) {
349552f7358SJed Brown             const PetscScalar *xpoint;
350552f7358SJed Brown             ierr     = DMPlexPointLocalRead(dm,c,x,&xpoint);CHKERRQ(ierr);
351552f7358SJed Brown             y[cnt++] = xpoint[i];
352552f7358SJed Brown           }
353552f7358SJed Brown           if (cnt != piece.nvertices) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Count does not match");
354552f7358SJed Brown           ierr = TransferWrite(viewer,fp,r,0,y,buffer,piece.nvertices,PETSC_SCALAR,tag);CHKERRQ(ierr);
355552f7358SJed Brown         }
356552f7358SJed Brown         ierr = PetscFree(y);CHKERRQ(ierr);
357552f7358SJed Brown         ierr = VecRestoreArrayRead(X,&x);CHKERRQ(ierr);
358552f7358SJed Brown       }
359552f7358SJed Brown     } else if (!rank) {
3600298fd71SBarry Smith       ierr = TransferWrite(viewer,fp,r,0,NULL,buffer,gpiece[r].nvertices*3,PETSC_SCALAR,tag);CHKERRQ(ierr); /* positions */
3610298fd71SBarry Smith       ierr = TransferWrite(viewer,fp,r,0,NULL,buffer,gpiece[r].nconn,PETSC_INT32,tag);CHKERRQ(ierr); /* connectivity */
3620298fd71SBarry Smith       ierr = TransferWrite(viewer,fp,r,0,NULL,buffer,gpiece[r].ncells,PETSC_INT32,tag);CHKERRQ(ierr); /* offsets */
3630298fd71SBarry Smith       ierr = TransferWrite(viewer,fp,r,0,NULL,buffer,gpiece[r].ncells,PETSC_UINT8,tag);CHKERRQ(ierr); /* types */
3640298fd71SBarry Smith       ierr = TransferWrite(viewer,fp,r,0,NULL,buffer,gpiece[r].ncells,PETSC_INT32,tag);CHKERRQ(ierr); /* owner rank (cells) */
365552f7358SJed Brown       /* all cell data */
366552f7358SJed Brown       for (link=vtk->link; link; link=link->next) {
367552f7358SJed Brown         PetscInt bs;
368552f7358SJed Brown         if ((link->ft != PETSC_VTK_CELL_FIELD) && (link->ft != PETSC_VTK_CELL_VECTOR_FIELD)) continue;
369552f7358SJed Brown         ierr = PetscSectionGetDof(dm->defaultSection,vStart,&bs);CHKERRQ(ierr);
370552f7358SJed Brown         for (i=0; i<bs; i++) {
3710298fd71SBarry Smith           ierr = TransferWrite(viewer,fp,r,0,NULL,buffer,gpiece[r].ncells,PETSC_SCALAR,tag);CHKERRQ(ierr);
372552f7358SJed Brown         }
373552f7358SJed Brown       }
374552f7358SJed Brown       /* all point data */
375552f7358SJed Brown       for (link=vtk->link; link; link=link->next) {
376552f7358SJed Brown         PetscInt bs;
377552f7358SJed Brown         if ((link->ft != PETSC_VTK_POINT_FIELD) && (link->ft != PETSC_VTK_POINT_VECTOR_FIELD)) continue;
378552f7358SJed Brown         ierr = PetscSectionGetDof(dm->defaultSection,vStart,&bs);CHKERRQ(ierr);
379552f7358SJed Brown         for (i=0; i<bs; i++) {
3800298fd71SBarry Smith           ierr = TransferWrite(viewer,fp,r,0,NULL,buffer,gpiece[r].nvertices,PETSC_SCALAR,tag);CHKERRQ(ierr);
381552f7358SJed Brown         }
382552f7358SJed Brown       }
383552f7358SJed Brown     }
384552f7358SJed Brown   }
385552f7358SJed Brown   ierr = PetscFree(gpiece);CHKERRQ(ierr);
386552f7358SJed Brown   ierr = PetscFree(buffer);CHKERRQ(ierr);
387552f7358SJed Brown   ierr = PetscFPrintf(comm,fp,"\n  </AppendedData>\n");CHKERRQ(ierr);
388552f7358SJed Brown   ierr = PetscFPrintf(comm,fp,"</VTKFile>\n");CHKERRQ(ierr);
389552f7358SJed Brown   PetscFunctionReturn(0);
390552f7358SJed Brown }
391