xref: /petsc/src/dm/impls/da/gr2.c (revision 3c0c59f39ee3bd561a1b4dcbd0e58366bfa60aba)
147c6ae99SBarry Smith 
247c6ae99SBarry Smith /*
3aa219208SBarry Smith    Plots vectors obtained with DMDACreate2d()
447c6ae99SBarry Smith */
547c6ae99SBarry Smith 
6c6db04a5SJed Brown #include <private/daimpl.h>      /*I  "petscdmda.h"   I*/
7c6db04a5SJed Brown #include <private/vecimpl.h>
847c6ae99SBarry Smith 
947c6ae99SBarry Smith /*
1047c6ae99SBarry Smith         The data that is passed into the graphics callback
1147c6ae99SBarry Smith */
1247c6ae99SBarry Smith typedef struct {
1347c6ae99SBarry Smith   PetscInt     m,n,step,k;
1447c6ae99SBarry Smith   PetscReal    min,max,scale;
1547c6ae99SBarry Smith   PetscScalar  *xy,*v;
1647c6ae99SBarry Smith   PetscBool    showgrid;
1747c6ae99SBarry Smith } ZoomCtx;
1847c6ae99SBarry Smith 
1947c6ae99SBarry Smith /*
2047c6ae99SBarry Smith        This does the drawing for one particular field
2147c6ae99SBarry Smith     in one particular set of coordinates. It is a callback
2247c6ae99SBarry Smith     called from PetscDrawZoom()
2347c6ae99SBarry Smith */
2447c6ae99SBarry Smith #undef __FUNCT__
2547c6ae99SBarry Smith #define __FUNCT__ "VecView_MPI_Draw_DA2d_Zoom"
2647c6ae99SBarry Smith PetscErrorCode VecView_MPI_Draw_DA2d_Zoom(PetscDraw draw,void *ctx)
2747c6ae99SBarry Smith {
2847c6ae99SBarry Smith   ZoomCtx        *zctx = (ZoomCtx*)ctx;
2947c6ae99SBarry Smith   PetscErrorCode ierr;
3047c6ae99SBarry Smith   PetscInt       m,n,i,j,k,step,id,c1,c2,c3,c4;
3147c6ae99SBarry Smith   PetscReal      s,min,x1,x2,x3,x4,y_1,y2,y3,y4;
3247c6ae99SBarry Smith   PetscScalar   *v,*xy;
3347c6ae99SBarry Smith 
3447c6ae99SBarry Smith   PetscFunctionBegin;
3547c6ae99SBarry Smith   m    = zctx->m;
3647c6ae99SBarry Smith   n    = zctx->n;
3747c6ae99SBarry Smith   step = zctx->step;
3847c6ae99SBarry Smith   k    = zctx->k;
3947c6ae99SBarry Smith   v    = zctx->v;
4047c6ae99SBarry Smith   xy   = zctx->xy;
4147c6ae99SBarry Smith   s    = zctx->scale;
4247c6ae99SBarry Smith   min  = zctx->min;
4347c6ae99SBarry Smith 
4447c6ae99SBarry Smith   /* PetscDraw the contour plot patch */
4547c6ae99SBarry Smith   for (j=0; j<n-1; j++) {
4647c6ae99SBarry Smith     for (i=0; i<m-1; i++) {
4747c6ae99SBarry Smith #if !defined(PETSC_USE_COMPLEX)
4847c6ae99SBarry Smith       id = i+j*m;    x1 = xy[2*id];y_1 = xy[2*id+1];c1 = (int)(PETSC_DRAW_BASIC_COLORS+s*(v[k+step*id]-min));
4947c6ae99SBarry Smith       id = i+j*m+1;  x2 = xy[2*id];y2  = y_1;       c2 = (int)(PETSC_DRAW_BASIC_COLORS+s*(v[k+step*id]-min));
5047c6ae99SBarry Smith       id = i+j*m+1+m;x3 = x2;      y3  = xy[2*id+1];c3 = (int)(PETSC_DRAW_BASIC_COLORS+s*(v[k+step*id]-min));
5147c6ae99SBarry Smith       id = i+j*m+m;  x4 = x1;      y4  = y3;        c4 = (int)(PETSC_DRAW_BASIC_COLORS+s*(v[k+step*id]-min));
5247c6ae99SBarry Smith #else
5347c6ae99SBarry Smith       id = i+j*m;    x1 = PetscRealPart(xy[2*id]);y_1 = PetscRealPart(xy[2*id+1]);c1 = (int)(PETSC_DRAW_BASIC_COLORS+s*(PetscRealPart(v[k+step*id])-min));
5447c6ae99SBarry Smith       id = i+j*m+1;  x2 = PetscRealPart(xy[2*id]);y2  = y_1;       c2 = (int)(PETSC_DRAW_BASIC_COLORS+s*(PetscRealPart(v[k+step*id])-min));
5547c6ae99SBarry Smith       id = i+j*m+1+m;x3 = x2;      y3  = PetscRealPart(xy[2*id+1]);c3 = (int)(PETSC_DRAW_BASIC_COLORS+s*(PetscRealPart(v[k+step*id])-min));
5647c6ae99SBarry Smith       id = i+j*m+m;  x4 = x1;      y4  = y3;        c4 = (int)(PETSC_DRAW_BASIC_COLORS+s*(PetscRealPart(v[k+step*id])-min));
5747c6ae99SBarry Smith #endif
5847c6ae99SBarry Smith       ierr = PetscDrawTriangle(draw,x1,y_1,x2,y2,x3,y3,c1,c2,c3);CHKERRQ(ierr);
5947c6ae99SBarry Smith       ierr = PetscDrawTriangle(draw,x1,y_1,x3,y3,x4,y4,c1,c3,c4);CHKERRQ(ierr);
6047c6ae99SBarry Smith       if (zctx->showgrid) {
6147c6ae99SBarry Smith         ierr = PetscDrawLine(draw,x1,y_1,x2,y2,PETSC_DRAW_BLACK);CHKERRQ(ierr);
6247c6ae99SBarry Smith         ierr = PetscDrawLine(draw,x2,y2,x3,y3,PETSC_DRAW_BLACK);CHKERRQ(ierr);
6347c6ae99SBarry Smith         ierr = PetscDrawLine(draw,x3,y3,x4,y4,PETSC_DRAW_BLACK);CHKERRQ(ierr);
6447c6ae99SBarry Smith         ierr = PetscDrawLine(draw,x4,y4,x1,y_1,PETSC_DRAW_BLACK);CHKERRQ(ierr);
6547c6ae99SBarry Smith       }
6647c6ae99SBarry Smith     }
6747c6ae99SBarry Smith   }
6847c6ae99SBarry Smith   PetscFunctionReturn(0);
6947c6ae99SBarry Smith }
7047c6ae99SBarry Smith 
7147c6ae99SBarry Smith #undef __FUNCT__
7247c6ae99SBarry Smith #define __FUNCT__ "VecView_MPI_Draw_DA2d"
7347c6ae99SBarry Smith PetscErrorCode VecView_MPI_Draw_DA2d(Vec xin,PetscViewer viewer)
7447c6ae99SBarry Smith {
759a42bb27SBarry Smith   DM                 da,dac,dag;
7647c6ae99SBarry Smith   PetscErrorCode     ierr;
7747c6ae99SBarry Smith   PetscMPIInt        rank;
7847c6ae99SBarry Smith   PetscInt           igstart,N,s,M,istart,isize,jgstart,w;
7947c6ae99SBarry Smith   const PetscInt     *lx,*ly;
8047c6ae99SBarry Smith   PetscReal          coors[4],ymin,ymax,xmin,xmax;
8147c6ae99SBarry Smith   PetscDraw          draw,popup;
8247c6ae99SBarry Smith   PetscBool          isnull,useports = PETSC_FALSE;
8347c6ae99SBarry Smith   MPI_Comm           comm;
8447c6ae99SBarry Smith   Vec                xlocal,xcoor,xcoorl;
851321219cSEthan Coon   DMDABoundaryType   bx,by;
86aa219208SBarry Smith   DMDAStencilType    st;
8747c6ae99SBarry Smith   ZoomCtx            zctx;
88f8631f9bSBarry Smith   PetscDrawViewPorts *ports = PETSC_NULL;
8947c6ae99SBarry Smith   PetscViewerFormat  format;
9047c6ae99SBarry Smith 
9147c6ae99SBarry Smith   PetscFunctionBegin;
9247c6ae99SBarry Smith   zctx.showgrid = PETSC_FALSE;
9347c6ae99SBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
9447c6ae99SBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
9547c6ae99SBarry Smith 
96*3c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
97aa219208SBarry Smith   if (!da) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
9847c6ae99SBarry Smith 
9947c6ae99SBarry Smith   ierr = PetscObjectGetComm((PetscObject)xin,&comm);CHKERRQ(ierr);
10047c6ae99SBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
10147c6ae99SBarry Smith 
1021321219cSEthan Coon   ierr = DMDAGetInfo(da,0,&M,&N,0,&zctx.m,&zctx.n,0,&w,&s,&bx,&by,0,&st);CHKERRQ(ierr);
103aa219208SBarry Smith   ierr = DMDAGetOwnershipRanges(da,&lx,&ly,PETSC_NULL);CHKERRQ(ierr);
10447c6ae99SBarry Smith 
10547c6ae99SBarry Smith   /*
10647c6ae99SBarry Smith         Obtain a sequential vector that is going to contain the local values plus ONE layer of
107aa219208SBarry Smith      ghosted values to draw the graphics from. We also need its corresponding DMDA (dac) that will
10847c6ae99SBarry Smith      update the local values pluse ONE layer of ghost values.
10947c6ae99SBarry Smith   */
11047c6ae99SBarry Smith   ierr = PetscObjectQuery((PetscObject)da,"GraphicsGhosted",(PetscObject*)&xlocal);CHKERRQ(ierr);
11147c6ae99SBarry Smith   if (!xlocal) {
1121321219cSEthan Coon     if (!bx || !by || s != 1 || st != DMDA_STENCIL_BOX) {
11347c6ae99SBarry Smith       /*
11447c6ae99SBarry Smith          if original da is not of stencil width one, or periodic or not a box stencil then
115aa219208SBarry Smith          create a special DMDA to handle one level of ghost points for graphics
11647c6ae99SBarry Smith       */
1171321219cSEthan Coon       ierr = DMDACreate2d(comm,DMDA_BOUNDARY_NONE,DMDA_BOUNDARY_NONE,DMDA_STENCIL_BOX,M,N,zctx.m,zctx.n,w,1,lx,ly,&dac);CHKERRQ(ierr);
118aa219208SBarry Smith       ierr = PetscInfo(da,"Creating auxilary DMDA for managing graphics ghost points\n");CHKERRQ(ierr);
11947c6ae99SBarry Smith     } else {
12047c6ae99SBarry Smith       /* otherwise we can use the da we already have */
12147c6ae99SBarry Smith       dac = da;
12247c6ae99SBarry Smith     }
12347c6ae99SBarry Smith     /* create local vector for holding ghosted values used in graphics */
124564755cdSBarry Smith     ierr = DMCreateLocalVector(dac,&xlocal);CHKERRQ(ierr);
12547c6ae99SBarry Smith     if (dac != da) {
126aa219208SBarry Smith       /* don't keep any public reference of this DMDA, is is only available through xlocal */
127f8631f9bSBarry Smith       DM dacd = dac;
128f8631f9bSBarry Smith       ierr = DMDestroy(&dacd);CHKERRQ(ierr);
12947c6ae99SBarry Smith     } else {
13047c6ae99SBarry Smith       /* remove association between xlocal and da, because below we compose in the opposite
13147c6ae99SBarry Smith          direction and if we left this connect we'd get a loop, so the objects could
13247c6ae99SBarry Smith          never be destroyed */
133*3c0c59f3SBarry Smith       ierr = PetscObjectCompose((PetscObject)xlocal,"DM",0);CHKERRQ(ierr);
13447c6ae99SBarry Smith     }
13547c6ae99SBarry Smith     ierr = PetscObjectCompose((PetscObject)da,"GraphicsGhosted",(PetscObject)xlocal);CHKERRQ(ierr);
13647c6ae99SBarry Smith     ierr = PetscObjectDereference((PetscObject)xlocal);CHKERRQ(ierr);
13747c6ae99SBarry Smith   } else {
1381321219cSEthan Coon     if (!bx && !by && s == 1 && st == DMDA_STENCIL_BOX) {
13947c6ae99SBarry Smith       dac = da;
14047c6ae99SBarry Smith     } else {
141*3c0c59f3SBarry Smith       ierr = PetscObjectQuery((PetscObject)xlocal,"DM",(PetscObject*)&dac);CHKERRQ(ierr);
14247c6ae99SBarry Smith     }
14347c6ae99SBarry Smith   }
14447c6ae99SBarry Smith 
14547c6ae99SBarry Smith   /*
14647c6ae99SBarry Smith       Get local (ghosted) values of vector
14747c6ae99SBarry Smith   */
1489a42bb27SBarry Smith   ierr = DMGlobalToLocalBegin(dac,xin,INSERT_VALUES,xlocal);CHKERRQ(ierr);
1499a42bb27SBarry Smith   ierr = DMGlobalToLocalEnd(dac,xin,INSERT_VALUES,xlocal);CHKERRQ(ierr);
15047c6ae99SBarry Smith   ierr = VecGetArray(xlocal,&zctx.v);CHKERRQ(ierr);
15147c6ae99SBarry Smith 
15247c6ae99SBarry Smith   /* get coordinates of nodes */
153aa219208SBarry Smith   ierr = DMDAGetCoordinates(da,&xcoor);CHKERRQ(ierr);
15447c6ae99SBarry Smith   if (!xcoor) {
155aa219208SBarry Smith     ierr = DMDASetUniformCoordinates(da,0.0,1.0,0.0,1.0,0.0,0.0);CHKERRQ(ierr);
156aa219208SBarry Smith     ierr = DMDAGetCoordinates(da,&xcoor);CHKERRQ(ierr);
15747c6ae99SBarry Smith   }
15847c6ae99SBarry Smith 
15947c6ae99SBarry Smith   /*
16047c6ae99SBarry Smith       Determine the min and max  coordinates in plot
16147c6ae99SBarry Smith   */
16247c6ae99SBarry Smith   ierr = VecStrideMin(xcoor,0,PETSC_NULL,&xmin);CHKERRQ(ierr);
16347c6ae99SBarry Smith   ierr = VecStrideMax(xcoor,0,PETSC_NULL,&xmax);CHKERRQ(ierr);
16447c6ae99SBarry Smith   ierr = VecStrideMin(xcoor,1,PETSC_NULL,&ymin);CHKERRQ(ierr);
16547c6ae99SBarry Smith   ierr = VecStrideMax(xcoor,1,PETSC_NULL,&ymax);CHKERRQ(ierr);
16647c6ae99SBarry Smith   coors[0] = xmin - .05*(xmax- xmin); coors[2] = xmax + .05*(xmax - xmin);
16747c6ae99SBarry Smith   coors[1] = ymin - .05*(ymax- ymin); coors[3] = ymax + .05*(ymax - ymin);
168aa219208SBarry Smith   ierr = PetscInfo4(da,"Preparing DMDA 2d contour plot coordinates %G %G %G %G\n",coors[0],coors[1],coors[2],coors[3]);CHKERRQ(ierr);
16947c6ae99SBarry Smith 
17047c6ae99SBarry Smith   /*
17147c6ae99SBarry Smith        get local ghosted version of coordinates
17247c6ae99SBarry Smith   */
17347c6ae99SBarry Smith   ierr = PetscObjectQuery((PetscObject)da,"GraphicsCoordinateGhosted",(PetscObject*)&xcoorl);CHKERRQ(ierr);
17447c6ae99SBarry Smith   if (!xcoorl) {
175f8631f9bSBarry Smith     DM dagd;
176aa219208SBarry Smith     /* create DMDA to get local version of graphics */
1771321219cSEthan Coon     ierr = DMDACreate2d(comm,DMDA_BOUNDARY_NONE,DMDA_BOUNDARY_NONE,DMDA_STENCIL_BOX,M,N,zctx.m,zctx.n,2,1,lx,ly,&dag);CHKERRQ(ierr);
178aa219208SBarry Smith     ierr = PetscInfo(dag,"Creating auxilary DMDA for managing graphics coordinates ghost points\n");CHKERRQ(ierr);
179564755cdSBarry Smith     ierr = DMCreateLocalVector(dag,&xcoorl);CHKERRQ(ierr);
18047c6ae99SBarry Smith     ierr = PetscObjectCompose((PetscObject)da,"GraphicsCoordinateGhosted",(PetscObject)xcoorl);CHKERRQ(ierr);
181f8631f9bSBarry Smith     dagd = dag;
182f8631f9bSBarry Smith     ierr = DMDestroy(&dagd);CHKERRQ(ierr);/* dereference dag */
18347c6ae99SBarry Smith     ierr = PetscObjectDereference((PetscObject)xcoorl);CHKERRQ(ierr);
18447c6ae99SBarry Smith   } else {
185*3c0c59f3SBarry Smith     ierr = PetscObjectQuery((PetscObject)xcoorl,"DM",(PetscObject*)&dag);CHKERRQ(ierr);
18647c6ae99SBarry Smith   }
1879a42bb27SBarry Smith   ierr = DMGlobalToLocalBegin(dag,xcoor,INSERT_VALUES,xcoorl);CHKERRQ(ierr);
1889a42bb27SBarry Smith   ierr = DMGlobalToLocalEnd(dag,xcoor,INSERT_VALUES,xcoorl);CHKERRQ(ierr);
18947c6ae99SBarry Smith   ierr = VecGetArray(xcoorl,&zctx.xy);CHKERRQ(ierr);
19047c6ae99SBarry Smith 
19147c6ae99SBarry Smith   /*
19247c6ae99SBarry Smith         Get information about size of area each processor must do graphics for
19347c6ae99SBarry Smith   */
1941321219cSEthan Coon   ierr = DMDAGetInfo(dac,0,&M,&N,0,0,0,0,&zctx.step,0,&bx,&by,0,0);CHKERRQ(ierr);
195aa219208SBarry Smith   ierr = DMDAGetGhostCorners(dac,&igstart,&jgstart,0,&zctx.m,&zctx.n,0);CHKERRQ(ierr);
196aa219208SBarry Smith   ierr = DMDAGetCorners(dac,&istart,0,0,&isize,0,0);CHKERRQ(ierr);
19747c6ae99SBarry Smith 
198671f6225SBarry Smith   ierr = PetscOptionsGetBool(PETSC_NULL,"-draw_contour_grid",&zctx.showgrid,PETSC_NULL);CHKERRQ(ierr);
19947c6ae99SBarry Smith 
20047c6ae99SBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
201671f6225SBarry Smith   ierr = PetscOptionsGetBool(PETSC_NULL,"-draw_ports",&useports,PETSC_NULL);CHKERRQ(ierr);
20247c6ae99SBarry Smith   if (useports || format == PETSC_VIEWER_DRAW_PORTS){
20347c6ae99SBarry Smith     ierr = PetscDrawSynchronizedClear(draw);CHKERRQ(ierr);
20447c6ae99SBarry Smith     ierr = PetscDrawViewPortsCreate(draw,zctx.step,&ports);CHKERRQ(ierr);
20547c6ae99SBarry Smith   }
20647c6ae99SBarry Smith   /*
20747c6ae99SBarry Smith      Loop over each field; drawing each in a different window
20847c6ae99SBarry Smith   */
20947c6ae99SBarry Smith   for (zctx.k=0; zctx.k<zctx.step; zctx.k++) {
21047c6ae99SBarry Smith     if (useports) {
21147c6ae99SBarry Smith       ierr = PetscDrawViewPortsSet(ports,zctx.k);CHKERRQ(ierr);
21247c6ae99SBarry Smith     } else {
21347c6ae99SBarry Smith       ierr = PetscViewerDrawGetDraw(viewer,zctx.k,&draw);CHKERRQ(ierr);
21447c6ae99SBarry Smith       ierr = PetscDrawSynchronizedClear(draw);CHKERRQ(ierr);
21547c6ae99SBarry Smith     }
21647c6ae99SBarry Smith 
21747c6ae99SBarry Smith     /*
21847c6ae99SBarry Smith         Determine the min and max color in plot
21947c6ae99SBarry Smith     */
22047c6ae99SBarry Smith     ierr = VecStrideMin(xin,zctx.k,PETSC_NULL,&zctx.min);CHKERRQ(ierr);
22147c6ae99SBarry Smith     ierr = VecStrideMax(xin,zctx.k,PETSC_NULL,&zctx.max);CHKERRQ(ierr);
22247c6ae99SBarry Smith     if (zctx.min == zctx.max) {
22347c6ae99SBarry Smith       zctx.min -= 1.e-12;
22447c6ae99SBarry Smith       zctx.max += 1.e-12;
22547c6ae99SBarry Smith     }
22647c6ae99SBarry Smith 
22747c6ae99SBarry Smith     if (!rank) {
22847c6ae99SBarry Smith       const char *title;
22947c6ae99SBarry Smith 
230aa219208SBarry Smith       ierr = DMDAGetFieldName(da,zctx.k,&title);CHKERRQ(ierr);
23147c6ae99SBarry Smith       if (title) {
23247c6ae99SBarry Smith         ierr = PetscDrawSetTitle(draw,title);CHKERRQ(ierr);
23347c6ae99SBarry Smith       }
23447c6ae99SBarry Smith     }
23547c6ae99SBarry Smith     ierr = PetscDrawSetCoordinates(draw,coors[0],coors[1],coors[2],coors[3]);CHKERRQ(ierr);
236aa219208SBarry Smith     ierr = PetscInfo2(da,"DMDA 2d contour plot min %G max %G\n",zctx.min,zctx.max);CHKERRQ(ierr);
23747c6ae99SBarry Smith 
23847c6ae99SBarry Smith     ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr);
23947c6ae99SBarry Smith     if (popup) {ierr = PetscDrawScalePopup(popup,zctx.min,zctx.max);CHKERRQ(ierr);}
24047c6ae99SBarry Smith 
24147c6ae99SBarry Smith     zctx.scale = (245.0 - PETSC_DRAW_BASIC_COLORS)/(zctx.max - zctx.min);
24247c6ae99SBarry Smith 
24347c6ae99SBarry Smith     ierr = PetscDrawZoom(draw,VecView_MPI_Draw_DA2d_Zoom,&zctx);CHKERRQ(ierr);
24447c6ae99SBarry Smith   }
2456bf464f9SBarry Smith   ierr = PetscDrawViewPortsDestroy(ports);CHKERRQ(ierr);
24647c6ae99SBarry Smith 
24747c6ae99SBarry Smith   ierr = VecRestoreArray(xcoorl,&zctx.xy);CHKERRQ(ierr);
24847c6ae99SBarry Smith   ierr = VecRestoreArray(xlocal,&zctx.v);CHKERRQ(ierr);
24947c6ae99SBarry Smith   PetscFunctionReturn(0);
25047c6ae99SBarry Smith }
25147c6ae99SBarry Smith 
25247c6ae99SBarry Smith 
25347c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
25447c6ae99SBarry Smith #undef __FUNCT__
25547c6ae99SBarry Smith #define __FUNCT__ "VecView_MPI_HDF5_DA"
25647c6ae99SBarry Smith PetscErrorCode VecView_MPI_HDF5_DA(Vec xin,PetscViewer viewer)
25747c6ae99SBarry Smith {
25847c6ae99SBarry Smith   PetscErrorCode ierr;
2599b2a5a86SJed Brown   DM             dm;
2609b2a5a86SJed Brown   DM_DA          *da;
26147c6ae99SBarry Smith   hsize_t        dim,dims[5];
26247c6ae99SBarry Smith   hsize_t        count[5];
26347c6ae99SBarry Smith   hsize_t        offset[5];
26447c6ae99SBarry Smith   PetscInt       cnt = 0;
26547c6ae99SBarry Smith   PetscScalar    *x;
26647c6ae99SBarry Smith   const char     *vecname;
26747c6ae99SBarry Smith   hid_t          filespace; /* file dataspace identifier */
26847c6ae99SBarry Smith   hid_t	         plist_id;  /* property list identifier */
26947c6ae99SBarry Smith   hid_t          dset_id;   /* dataset identifier */
27047c6ae99SBarry Smith   hid_t          memspace;  /* memory dataspace identifier */
27147c6ae99SBarry Smith   hid_t          file_id;
27247c6ae99SBarry Smith   herr_t         status;
27347c6ae99SBarry Smith 
27447c6ae99SBarry Smith   PetscFunctionBegin;
27547c6ae99SBarry Smith   ierr = PetscViewerHDF5GetFileId(viewer, &file_id);CHKERRQ(ierr);
276*3c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&dm);CHKERRQ(ierr);
2779b2a5a86SJed Brown   if (!dm) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
2789b2a5a86SJed Brown   da = (DM_DA*)dm->data;
27947c6ae99SBarry Smith 
28047c6ae99SBarry Smith   /* Create the dataspace for the dataset */
28147c6ae99SBarry Smith   dim       = PetscHDF5IntCast(da->dim + ((da->w == 1) ? 0 : 1));
28247c6ae99SBarry Smith   if (da->dim == 3) dims[cnt++]   = PetscHDF5IntCast(da->P);
28347c6ae99SBarry Smith   if (da->dim > 1)  dims[cnt++]   = PetscHDF5IntCast(da->N);
28447c6ae99SBarry Smith   dims[cnt++]   = PetscHDF5IntCast(da->M);
28547c6ae99SBarry Smith   if (da->w > 1) dims[cnt++] = PetscHDF5IntCast(da->w);
28647c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
28747c6ae99SBarry Smith   dim++;
28847c6ae99SBarry Smith   dims[cnt++] = 2;
28947c6ae99SBarry Smith #endif
29047c6ae99SBarry Smith   filespace = H5Screate_simple(dim, dims, NULL);
29147c6ae99SBarry Smith   if (filespace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Screate_simple()");
29247c6ae99SBarry Smith 
29347c6ae99SBarry Smith   /* Create the dataset with default properties and close filespace */
29447c6ae99SBarry Smith   ierr = PetscObjectGetName((PetscObject)xin,&vecname);CHKERRQ(ierr);
29547c6ae99SBarry Smith #if (H5_VERS_MAJOR * 10000 + H5_VERS_MINOR * 100 + H5_VERS_RELEASE >= 10800)
29647c6ae99SBarry Smith   dset_id = H5Dcreate2(file_id, vecname, H5T_NATIVE_DOUBLE, filespace, H5P_DEFAULT, H5P_DEFAULT, H5P_DEFAULT);
29747c6ae99SBarry Smith #else
29847c6ae99SBarry Smith   dset_id = H5Dcreate(file_id, vecname, H5T_NATIVE_DOUBLE, filespace, H5P_DEFAULT);
29947c6ae99SBarry Smith #endif
30047c6ae99SBarry Smith   if (dset_id == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Dcreate2()");
30147c6ae99SBarry Smith   status = H5Sclose(filespace);CHKERRQ(status);
30247c6ae99SBarry Smith 
30347c6ae99SBarry Smith   /* Each process defines a dataset and writes it to the hyperslab in the file */
30447c6ae99SBarry Smith   cnt = 0;
30547c6ae99SBarry Smith   if (da->dim == 3) offset[cnt++] = PetscHDF5IntCast(da->zs);
30647c6ae99SBarry Smith   if (da->dim > 1)  offset[cnt++] = PetscHDF5IntCast(da->ys);
30747c6ae99SBarry Smith   offset[cnt++] = PetscHDF5IntCast(da->xs/da->w);
30847c6ae99SBarry Smith   if (da->w > 1) offset[cnt++] = 0;
30947c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
31047c6ae99SBarry Smith   offset[cnt++] = 0;
31147c6ae99SBarry Smith #endif
31247c6ae99SBarry Smith   cnt = 0;
31347c6ae99SBarry Smith   if (da->dim == 3) count[cnt++] = PetscHDF5IntCast(da->ze - da->zs);
31447c6ae99SBarry Smith   if (da->dim > 1)  count[cnt++] = PetscHDF5IntCast(da->ye - da->ys);
31547c6ae99SBarry Smith   count[cnt++] = PetscHDF5IntCast((da->xe - da->xs)/da->w);
31647c6ae99SBarry Smith   if (da->w > 1) count[cnt++] = PetscHDF5IntCast(da->w);
31747c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
31847c6ae99SBarry Smith   count[cnt++] = 2;
31947c6ae99SBarry Smith #endif
32047c6ae99SBarry Smith   memspace = H5Screate_simple(dim, count, NULL);
32147c6ae99SBarry Smith   if (memspace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Screate_simple()");
32247c6ae99SBarry Smith 
32347c6ae99SBarry Smith 
32447c6ae99SBarry Smith   filespace = H5Dget_space(dset_id);
32547c6ae99SBarry Smith   if (filespace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Dget_space()");
32647c6ae99SBarry Smith   status = H5Sselect_hyperslab(filespace, H5S_SELECT_SET, offset, NULL, count, NULL);CHKERRQ(status);
32747c6ae99SBarry Smith 
32847c6ae99SBarry Smith   /* Create property list for collective dataset write */
32947c6ae99SBarry Smith   plist_id = H5Pcreate(H5P_DATASET_XFER);
33047c6ae99SBarry Smith   if (plist_id == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Pcreate()");
33147c6ae99SBarry Smith #if defined(PETSC_HAVE_H5PSET_FAPL_MPIO)
33247c6ae99SBarry Smith   status = H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_COLLECTIVE);CHKERRQ(status);
33347c6ae99SBarry Smith #endif
33447c6ae99SBarry Smith   /* To write dataset independently use H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_INDEPENDENT) */
33547c6ae99SBarry Smith 
33647c6ae99SBarry Smith   ierr = VecGetArray(xin, &x);CHKERRQ(ierr);
33747c6ae99SBarry Smith   status = H5Dwrite(dset_id, H5T_NATIVE_DOUBLE, memspace, filespace, plist_id, x);CHKERRQ(status);
33847c6ae99SBarry Smith   status = H5Fflush(file_id, H5F_SCOPE_GLOBAL);CHKERRQ(status);
33947c6ae99SBarry Smith   ierr = VecRestoreArray(xin, &x);CHKERRQ(ierr);
34047c6ae99SBarry Smith 
34147c6ae99SBarry Smith   /* Close/release resources */
34247c6ae99SBarry Smith   status = H5Pclose(plist_id);CHKERRQ(status);
34347c6ae99SBarry Smith   status = H5Sclose(filespace);CHKERRQ(status);
34447c6ae99SBarry Smith   status = H5Sclose(memspace);CHKERRQ(status);
34547c6ae99SBarry Smith   status = H5Dclose(dset_id);CHKERRQ(status);
34647c6ae99SBarry Smith   ierr = PetscInfo1(xin,"Wrote Vec object with name %s\n",vecname);CHKERRQ(ierr);
34747c6ae99SBarry Smith   PetscFunctionReturn(0);
34847c6ae99SBarry Smith }
34947c6ae99SBarry Smith #endif
35047c6ae99SBarry Smith 
35109573ac7SBarry Smith extern PetscErrorCode VecView_MPI_Draw_DA1d(Vec,PetscViewer);
35247c6ae99SBarry Smith 
35347c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
35447c6ae99SBarry Smith #undef __FUNCT__
355aa219208SBarry Smith #define __FUNCT__ "DMDAArrayMPIIO"
356aa219208SBarry Smith static PetscErrorCode DMDAArrayMPIIO(DM da,PetscViewer viewer,Vec xin,PetscBool  write)
35747c6ae99SBarry Smith {
35847c6ae99SBarry Smith   PetscErrorCode    ierr;
35947c6ae99SBarry Smith   MPI_File          mfdes;
36047c6ae99SBarry Smith   PetscMPIInt       gsizes[4],lsizes[4],lstarts[4],asiz,dof;
36147c6ae99SBarry Smith   MPI_Datatype      view;
36247c6ae99SBarry Smith   const PetscScalar *array;
36347c6ae99SBarry Smith   MPI_Offset        off;
36447c6ae99SBarry Smith   MPI_Aint          ub,ul;
36547c6ae99SBarry Smith   PetscInt          type,rows,vecrows,tr[2];
36647c6ae99SBarry Smith   DM_DA             *dd = (DM_DA*)da->data;
36747c6ae99SBarry Smith 
36847c6ae99SBarry Smith   PetscFunctionBegin;
36947c6ae99SBarry Smith   ierr = VecGetSize(xin,&vecrows);CHKERRQ(ierr);
37047c6ae99SBarry Smith   if (!write) {
37147c6ae99SBarry Smith     /* Read vector header. */
37247c6ae99SBarry Smith     ierr = PetscViewerBinaryRead(viewer,tr,2,PETSC_INT);CHKERRQ(ierr);
37347c6ae99SBarry Smith     type = tr[0];
37447c6ae99SBarry Smith     rows = tr[1];
37547c6ae99SBarry Smith     if (type != VEC_FILE_CLASSID) SETERRQ(((PetscObject)da)->comm,PETSC_ERR_ARG_WRONG,"Not vector next in file");
376aa219208SBarry Smith     if (rows != vecrows) SETERRQ(((PetscObject)da)->comm,PETSC_ERR_ARG_SIZ,"Vector in file not same size as DMDA vector");
37747c6ae99SBarry Smith   } else {
37847c6ae99SBarry Smith     tr[0] = VEC_FILE_CLASSID;
37947c6ae99SBarry Smith     tr[1] = vecrows;
38047c6ae99SBarry Smith     ierr = PetscViewerBinaryWrite(viewer,tr,2,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
38147c6ae99SBarry Smith   }
38247c6ae99SBarry Smith 
38347c6ae99SBarry Smith   dof = PetscMPIIntCast(dd->w);
38447c6ae99SBarry Smith   gsizes[0]  = dof; gsizes[1] = PetscMPIIntCast(dd->M); gsizes[2] = PetscMPIIntCast(dd->N); gsizes[3] = PetscMPIIntCast(dd->P);
38547c6ae99SBarry Smith   lsizes[0]  = dof;lsizes[1] = PetscMPIIntCast((dd->xe-dd->xs)/dof); lsizes[2] = PetscMPIIntCast(dd->ye-dd->ys); lsizes[3] = PetscMPIIntCast(dd->ze-dd->zs);
38647c6ae99SBarry Smith   lstarts[0] = 0;  lstarts[1] = PetscMPIIntCast(dd->xs/dof); lstarts[2] = PetscMPIIntCast(dd->ys); lstarts[3] = PetscMPIIntCast(dd->zs);
38747c6ae99SBarry Smith   ierr = MPI_Type_create_subarray(dd->dim+1,gsizes,lsizes,lstarts,MPI_ORDER_FORTRAN,MPIU_SCALAR,&view);CHKERRQ(ierr);
38847c6ae99SBarry Smith   ierr = MPI_Type_commit(&view);CHKERRQ(ierr);
38947c6ae99SBarry Smith 
39047c6ae99SBarry Smith   ierr = PetscViewerBinaryGetMPIIODescriptor(viewer,&mfdes);CHKERRQ(ierr);
39147c6ae99SBarry Smith   ierr = PetscViewerBinaryGetMPIIOOffset(viewer,&off);CHKERRQ(ierr);
39247c6ae99SBarry Smith   ierr = MPI_File_set_view(mfdes,off,MPIU_SCALAR,view,(char *)"native",MPI_INFO_NULL);CHKERRQ(ierr);
39347c6ae99SBarry Smith   ierr = VecGetArrayRead(xin,&array);CHKERRQ(ierr);
39447c6ae99SBarry Smith   asiz = lsizes[1]*(lsizes[2] > 0 ? lsizes[2] : 1)*(lsizes[3] > 0 ? lsizes[3] : 1)*dof;
39547c6ae99SBarry Smith   if (write) {
39647c6ae99SBarry Smith     ierr = MPIU_File_write_all(mfdes,(PetscScalar*)array,asiz,MPIU_SCALAR,MPI_STATUS_IGNORE);CHKERRQ(ierr);
39747c6ae99SBarry Smith   } else {
39847c6ae99SBarry Smith     ierr = MPIU_File_read_all(mfdes,(PetscScalar*)array,asiz,MPIU_SCALAR,MPI_STATUS_IGNORE);CHKERRQ(ierr);
39947c6ae99SBarry Smith   }
40047c6ae99SBarry Smith   ierr = MPI_Type_get_extent(view,&ul,&ub);CHKERRQ(ierr);
40147c6ae99SBarry Smith   ierr = PetscViewerBinaryAddMPIIOOffset(viewer,ub);CHKERRQ(ierr);
40247c6ae99SBarry Smith   ierr = VecRestoreArrayRead(xin,&array);CHKERRQ(ierr);
40347c6ae99SBarry Smith   ierr = MPI_Type_free(&view);CHKERRQ(ierr);
40447c6ae99SBarry Smith   PetscFunctionReturn(0);
40547c6ae99SBarry Smith }
40647c6ae99SBarry Smith #endif
40747c6ae99SBarry Smith 
40847c6ae99SBarry Smith EXTERN_C_BEGIN
40947c6ae99SBarry Smith #undef __FUNCT__
41047c6ae99SBarry Smith #define __FUNCT__ "VecView_MPI_DA"
4117087cfbeSBarry Smith PetscErrorCode  VecView_MPI_DA(Vec xin,PetscViewer viewer)
41247c6ae99SBarry Smith {
4139a42bb27SBarry Smith   DM             da;
41447c6ae99SBarry Smith   PetscErrorCode ierr;
41547c6ae99SBarry Smith   PetscInt       dim;
41647c6ae99SBarry Smith   Vec            natural;
41747c6ae99SBarry Smith   PetscBool      isdraw;
41847c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
41947c6ae99SBarry Smith   PetscBool      ishdf5;
42047c6ae99SBarry Smith #endif
4213f3fd955SJed Brown   const char     *prefix,*name;
42247c6ae99SBarry Smith 
42347c6ae99SBarry Smith   PetscFunctionBegin;
424*3c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
425aa219208SBarry Smith   if (!da) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
42647c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
42747c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
42847c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERHDF5,&ishdf5);CHKERRQ(ierr);
42947c6ae99SBarry Smith #endif
43047c6ae99SBarry Smith   if (isdraw) {
4311321219cSEthan Coon     ierr = DMDAGetInfo(da,&dim,0,0,0,0,0,0,0,0,0,0,0,0);CHKERRQ(ierr);
43247c6ae99SBarry Smith     if (dim == 1) {
43347c6ae99SBarry Smith       ierr = VecView_MPI_Draw_DA1d(xin,viewer);CHKERRQ(ierr);
43447c6ae99SBarry Smith     } else if (dim == 2) {
43547c6ae99SBarry Smith       ierr = VecView_MPI_Draw_DA2d(xin,viewer);CHKERRQ(ierr);
43647c6ae99SBarry Smith     } else {
437aa219208SBarry Smith       SETERRQ1(((PetscObject)da)->comm,PETSC_ERR_SUP,"Cannot graphically view vector associated with this dimensional DMDA %D",dim);
43847c6ae99SBarry Smith     }
43947c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
44047c6ae99SBarry Smith   } else if (ishdf5) {
44147c6ae99SBarry Smith     ierr = VecView_MPI_HDF5_DA(xin,viewer);CHKERRQ(ierr);
44247c6ae99SBarry Smith #endif
44347c6ae99SBarry Smith   } else {
44447c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
44547c6ae99SBarry Smith     PetscBool  isbinary,isMPIIO;
44647c6ae99SBarry Smith 
44747c6ae99SBarry Smith     ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
44847c6ae99SBarry Smith     if (isbinary) {
44947c6ae99SBarry Smith       ierr = PetscViewerBinaryGetMPIIO(viewer,&isMPIIO);CHKERRQ(ierr);
45047c6ae99SBarry Smith       if (isMPIIO) {
451aa219208SBarry Smith        ierr = DMDAArrayMPIIO(da,viewer,xin,PETSC_TRUE);CHKERRQ(ierr);
45247c6ae99SBarry Smith        PetscFunctionReturn(0);
45347c6ae99SBarry Smith       }
45447c6ae99SBarry Smith     }
45547c6ae99SBarry Smith #endif
45647c6ae99SBarry Smith 
45747c6ae99SBarry Smith     /* call viewer on natural ordering */
45847c6ae99SBarry Smith     ierr = PetscObjectGetOptionsPrefix((PetscObject)xin,&prefix);CHKERRQ(ierr);
459aa219208SBarry Smith     ierr = DMDACreateNaturalVector(da,&natural);CHKERRQ(ierr);
46047c6ae99SBarry Smith     ierr = PetscObjectSetOptionsPrefix((PetscObject)natural,prefix);CHKERRQ(ierr);
461aa219208SBarry Smith     ierr = DMDAGlobalToNaturalBegin(da,xin,INSERT_VALUES,natural);CHKERRQ(ierr);
462aa219208SBarry Smith     ierr = DMDAGlobalToNaturalEnd(da,xin,INSERT_VALUES,natural);CHKERRQ(ierr);
4633f3fd955SJed Brown     ierr = PetscObjectGetName((PetscObject)xin,&name);CHKERRQ(ierr);
4643f3fd955SJed Brown     ierr = PetscObjectSetName((PetscObject)natural,name);CHKERRQ(ierr);
46547c6ae99SBarry Smith     ierr = VecView(natural,viewer);CHKERRQ(ierr);
466fcfd50ebSBarry Smith     ierr = VecDestroy(&natural);CHKERRQ(ierr);
46747c6ae99SBarry Smith   }
46847c6ae99SBarry Smith   PetscFunctionReturn(0);
46947c6ae99SBarry Smith }
47047c6ae99SBarry Smith EXTERN_C_END
47147c6ae99SBarry Smith 
47247c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
47347c6ae99SBarry Smith #undef __FUNCT__
47447c6ae99SBarry Smith #define __FUNCT__ "VecLoad_HDF5_DA"
47547c6ae99SBarry Smith PetscErrorCode VecLoad_HDF5_DA(Vec xin, PetscViewer viewer)
47647c6ae99SBarry Smith {
4779a42bb27SBarry Smith   DM             da;
47847c6ae99SBarry Smith   PetscErrorCode ierr;
47947c6ae99SBarry Smith   hsize_t        dim,dims[5];
48047c6ae99SBarry Smith   hsize_t        count[5];
48147c6ae99SBarry Smith   hsize_t        offset[5];
48247c6ae99SBarry Smith   PetscInt       cnt = 0;
48347c6ae99SBarry Smith   PetscScalar    *x;
48447c6ae99SBarry Smith   const char     *vecname;
48547c6ae99SBarry Smith   hid_t          filespace; /* file dataspace identifier */
48647c6ae99SBarry Smith   hid_t	         plist_id;  /* property list identifier */
48747c6ae99SBarry Smith   hid_t          dset_id;   /* dataset identifier */
48847c6ae99SBarry Smith   hid_t          memspace;  /* memory dataspace identifier */
48947c6ae99SBarry Smith   hid_t          file_id;
49047c6ae99SBarry Smith   herr_t         status;
4919c7c4993SBarry Smith   DM_DA          *dd;
49247c6ae99SBarry Smith 
49347c6ae99SBarry Smith   PetscFunctionBegin;
49447c6ae99SBarry Smith   ierr = PetscViewerHDF5GetFileId(viewer, &file_id);CHKERRQ(ierr);
495*3c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
4969c7c4993SBarry Smith   dd = (DM_DA*)da->data;
49747c6ae99SBarry Smith 
49847c6ae99SBarry Smith   /* Create the dataspace for the dataset */
49947c6ae99SBarry Smith   dim       = PetscHDF5IntCast(dd->dim + ((dd->w == 1) ? 0 : 1));
50047c6ae99SBarry Smith   if (dd->dim == 3) dims[cnt++]   = PetscHDF5IntCast(dd->P);
50147c6ae99SBarry Smith   if (dd->dim > 1)  dims[cnt++]   = PetscHDF5IntCast(dd->N);
50247c6ae99SBarry Smith   dims[cnt++]     = PetscHDF5IntCast(dd->M);
50347c6ae99SBarry Smith   if (dd->w > 1) PetscHDF5IntCast(dims[cnt++] = dd->w);
50447c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
50547c6ae99SBarry Smith   dim++;
50647c6ae99SBarry Smith   dims[cnt++] = 2;
50747c6ae99SBarry Smith #endif
50847c6ae99SBarry Smith 
50947c6ae99SBarry Smith   /* Create the dataset with default properties and close filespace */
51047c6ae99SBarry Smith   ierr = PetscObjectGetName((PetscObject)xin,&vecname);CHKERRQ(ierr);
51147c6ae99SBarry Smith #if (H5_VERS_MAJOR * 10000 + H5_VERS_MINOR * 100 + H5_VERS_RELEASE >= 10800)
51247c6ae99SBarry Smith   dset_id = H5Dopen2(file_id, vecname, H5P_DEFAULT);
51347c6ae99SBarry Smith #else
51447c6ae99SBarry Smith   dset_id = H5Dopen(file_id, vecname);
51547c6ae99SBarry Smith #endif
51647c6ae99SBarry Smith   if (dset_id == -1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Dopen2() with Vec named %s",vecname);
51747c6ae99SBarry Smith   filespace = H5Dget_space(dset_id);
51847c6ae99SBarry Smith 
51947c6ae99SBarry Smith   /* Each process defines a dataset and reads it from the hyperslab in the file */
52047c6ae99SBarry Smith   cnt = 0;
52147c6ae99SBarry Smith   if (dd->dim == 3) offset[cnt++] = PetscHDF5IntCast(dd->zs);
52247c6ae99SBarry Smith   if (dd->dim > 1)  offset[cnt++] = PetscHDF5IntCast(dd->ys);
52347c6ae99SBarry Smith   offset[cnt++] = PetscHDF5IntCast(dd->xs/dd->w);
52447c6ae99SBarry Smith   if (dd->w > 1) offset[cnt++] = 0;
52547c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
52647c6ae99SBarry Smith   offset[cnt++] = 0;
52747c6ae99SBarry Smith #endif
52847c6ae99SBarry Smith   cnt = 0;
52947c6ae99SBarry Smith   if (dd->dim == 3) count[cnt++] = PetscHDF5IntCast(dd->ze - dd->zs);
53047c6ae99SBarry Smith   if (dd->dim > 1)  count[cnt++] = PetscHDF5IntCast(dd->ye - dd->ys);
53147c6ae99SBarry Smith   count[cnt++] = PetscHDF5IntCast((dd->xe - dd->xs)/dd->w);
53247c6ae99SBarry Smith   if (dd->w > 1) count[cnt++] = PetscHDF5IntCast(dd->w);
53347c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
53447c6ae99SBarry Smith   count[cnt++] = 2;
53547c6ae99SBarry Smith #endif
53647c6ae99SBarry Smith   memspace = H5Screate_simple(dim, count, NULL);
53747c6ae99SBarry Smith   if (memspace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Screate_simple()");
53847c6ae99SBarry Smith 
53947c6ae99SBarry Smith   status = H5Sselect_hyperslab(filespace, H5S_SELECT_SET, offset, NULL, count, NULL);CHKERRQ(status);
54047c6ae99SBarry Smith 
54147c6ae99SBarry Smith   /* Create property list for collective dataset write */
54247c6ae99SBarry Smith   plist_id = H5Pcreate(H5P_DATASET_XFER);
54347c6ae99SBarry Smith   if (plist_id == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Pcreate()");
54447c6ae99SBarry Smith #if defined(PETSC_HAVE_H5PSET_FAPL_MPIO)
54547c6ae99SBarry Smith   status = H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_COLLECTIVE);CHKERRQ(status);
54647c6ae99SBarry Smith #endif
54747c6ae99SBarry Smith   /* To write dataset independently use H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_INDEPENDENT) */
54847c6ae99SBarry Smith 
54947c6ae99SBarry Smith   ierr = VecGetArray(xin, &x);CHKERRQ(ierr);
55047c6ae99SBarry Smith   status = H5Dread(dset_id, H5T_NATIVE_DOUBLE, memspace, filespace, plist_id, x);CHKERRQ(status);
55147c6ae99SBarry Smith   ierr = VecRestoreArray(xin, &x);CHKERRQ(ierr);
55247c6ae99SBarry Smith 
55347c6ae99SBarry Smith   /* Close/release resources */
55447c6ae99SBarry Smith   status = H5Pclose(plist_id);CHKERRQ(status);
55547c6ae99SBarry Smith   status = H5Sclose(filespace);CHKERRQ(status);
55647c6ae99SBarry Smith   status = H5Sclose(memspace);CHKERRQ(status);
55747c6ae99SBarry Smith   status = H5Dclose(dset_id);CHKERRQ(status);
55847c6ae99SBarry Smith   PetscFunctionReturn(0);
55947c6ae99SBarry Smith }
56047c6ae99SBarry Smith #endif
56147c6ae99SBarry Smith 
56247c6ae99SBarry Smith #undef __FUNCT__
56347c6ae99SBarry Smith #define __FUNCT__ "VecLoad_Binary_DA"
56447c6ae99SBarry Smith PetscErrorCode VecLoad_Binary_DA(Vec xin, PetscViewer viewer)
56547c6ae99SBarry Smith {
5669a42bb27SBarry Smith   DM             da;
56747c6ae99SBarry Smith   PetscErrorCode ierr;
56847c6ae99SBarry Smith   Vec            natural;
56947c6ae99SBarry Smith   const char     *prefix;
57047c6ae99SBarry Smith   PetscInt       bs;
57147c6ae99SBarry Smith   PetscBool      flag;
57247c6ae99SBarry Smith   DM_DA          *dd;
57347c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
57447c6ae99SBarry Smith   PetscBool      isMPIIO;
57547c6ae99SBarry Smith #endif
57647c6ae99SBarry Smith 
57747c6ae99SBarry Smith   PetscFunctionBegin;
578*3c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
57947c6ae99SBarry Smith   dd   = (DM_DA*)da->data;
58047c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
58147c6ae99SBarry Smith   ierr = PetscViewerBinaryGetMPIIO(viewer,&isMPIIO);CHKERRQ(ierr);
58247c6ae99SBarry Smith   if (isMPIIO) {
583aa219208SBarry Smith     ierr = DMDAArrayMPIIO(da,viewer,xin,PETSC_FALSE);CHKERRQ(ierr);
58447c6ae99SBarry Smith     PetscFunctionReturn(0);
58547c6ae99SBarry Smith   }
58647c6ae99SBarry Smith #endif
58747c6ae99SBarry Smith 
58847c6ae99SBarry Smith   ierr = PetscObjectGetOptionsPrefix((PetscObject)xin,&prefix);CHKERRQ(ierr);
589aa219208SBarry Smith   ierr = DMDACreateNaturalVector(da,&natural);CHKERRQ(ierr);
59047c6ae99SBarry Smith   ierr = PetscObjectSetName((PetscObject)natural,((PetscObject)xin)->name);CHKERRQ(ierr);
59147c6ae99SBarry Smith   ierr = PetscObjectSetOptionsPrefix((PetscObject)natural,prefix);CHKERRQ(ierr);
59247c6ae99SBarry Smith   ierr = VecLoad_Binary(natural,viewer);CHKERRQ(ierr);
593aa219208SBarry Smith   ierr = DMDANaturalToGlobalBegin(da,natural,INSERT_VALUES,xin);CHKERRQ(ierr);
594aa219208SBarry Smith   ierr = DMDANaturalToGlobalEnd(da,natural,INSERT_VALUES,xin);CHKERRQ(ierr);
595fcfd50ebSBarry Smith   ierr = VecDestroy(&natural);CHKERRQ(ierr);
596aa219208SBarry Smith   ierr = PetscInfo(xin,"Loading vector from natural ordering into DMDA\n");CHKERRQ(ierr);
59747c6ae99SBarry Smith   ierr = PetscOptionsGetInt(((PetscObject)xin)->prefix,"-vecload_block_size",&bs,&flag);CHKERRQ(ierr);
59847c6ae99SBarry Smith   if (flag && bs != dd->w) {
599aa219208SBarry Smith     ierr = PetscInfo2(xin,"Block size in file %D not equal to DMDA's dof %D\n",bs,dd->w);CHKERRQ(ierr);
60047c6ae99SBarry Smith   }
60147c6ae99SBarry Smith   PetscFunctionReturn(0);
60247c6ae99SBarry Smith }
60347c6ae99SBarry Smith 
60447c6ae99SBarry Smith EXTERN_C_BEGIN
60547c6ae99SBarry Smith #undef __FUNCT__
60647c6ae99SBarry Smith #define __FUNCT__ "VecLoad_Default_DA"
6077087cfbeSBarry Smith PetscErrorCode  VecLoad_Default_DA(Vec xin, PetscViewer viewer)
60847c6ae99SBarry Smith {
60947c6ae99SBarry Smith   PetscErrorCode ierr;
6109a42bb27SBarry Smith   DM             da;
61147c6ae99SBarry Smith   PetscBool      isbinary;
61247c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
61347c6ae99SBarry Smith   PetscBool      ishdf5;
61447c6ae99SBarry Smith #endif
61547c6ae99SBarry Smith 
61647c6ae99SBarry Smith   PetscFunctionBegin;
617*3c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
618aa219208SBarry Smith   if (!da) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
61947c6ae99SBarry Smith 
62047c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
62147c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERHDF5,&ishdf5);CHKERRQ(ierr);
62247c6ae99SBarry Smith #endif
62347c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
62447c6ae99SBarry Smith 
62547c6ae99SBarry Smith   if (isbinary) {
62647c6ae99SBarry Smith     ierr = VecLoad_Binary_DA(xin,viewer);CHKERRQ(ierr);
62747c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
62847c6ae99SBarry Smith   } else if (ishdf5) {
62947c6ae99SBarry Smith     ierr = VecLoad_HDF5_DA(xin,viewer);CHKERRQ(ierr);
63047c6ae99SBarry Smith #endif
631d34fcf5fSBarry Smith   } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for vector loading", ((PetscObject)viewer)->type_name);
63247c6ae99SBarry Smith   PetscFunctionReturn(0);
63347c6ae99SBarry Smith }
63447c6ae99SBarry Smith EXTERN_C_END
635