xref: /petsc/src/dm/impls/da/gr2.c (revision 03193ff8a6fdc51e7e617ca95e470ca543268d0d)
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;
78f7923d8aSBarry Smith   PetscInt           N,s,M,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;
9020d0051dSBarry Smith   PetscInt           *displayfields;
9167dd0837SBarry Smith   PetscInt           ndisplayfields,i,nbounds;
9220d0051dSBarry Smith   PetscBool          flg;
9367dd0837SBarry Smith   const PetscReal    *bounds;
9447c6ae99SBarry Smith 
9547c6ae99SBarry Smith   PetscFunctionBegin;
9647c6ae99SBarry Smith   zctx.showgrid = PETSC_FALSE;
9747c6ae99SBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
9847c6ae99SBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
99*03193ff8SBarry Smith   ierr = PetscViewerDrawGetBounds(viewer,&nbounds,&bounds);CHKERRQ(ierr);
10047c6ae99SBarry Smith 
1013c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
102aa219208SBarry Smith   if (!da) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
10347c6ae99SBarry Smith 
10447c6ae99SBarry Smith   ierr = PetscObjectGetComm((PetscObject)xin,&comm);CHKERRQ(ierr);
10547c6ae99SBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
10647c6ae99SBarry Smith 
1071321219cSEthan Coon   ierr = DMDAGetInfo(da,0,&M,&N,0,&zctx.m,&zctx.n,0,&w,&s,&bx,&by,0,&st);CHKERRQ(ierr);
108aa219208SBarry Smith   ierr = DMDAGetOwnershipRanges(da,&lx,&ly,PETSC_NULL);CHKERRQ(ierr);
10947c6ae99SBarry Smith 
11047c6ae99SBarry Smith   /*
11147c6ae99SBarry Smith         Obtain a sequential vector that is going to contain the local values plus ONE layer of
112aa219208SBarry Smith      ghosted values to draw the graphics from. We also need its corresponding DMDA (dac) that will
11347c6ae99SBarry Smith      update the local values pluse ONE layer of ghost values.
11447c6ae99SBarry Smith   */
11547c6ae99SBarry Smith   ierr = PetscObjectQuery((PetscObject)da,"GraphicsGhosted",(PetscObject*)&xlocal);CHKERRQ(ierr);
11647c6ae99SBarry Smith   if (!xlocal) {
117f7923d8aSBarry Smith     if (bx !=  DMDA_BOUNDARY_NONE || by !=  DMDA_BOUNDARY_NONE || s != 1 || st != DMDA_STENCIL_BOX) {
11847c6ae99SBarry Smith       /*
11947c6ae99SBarry Smith          if original da is not of stencil width one, or periodic or not a box stencil then
120aa219208SBarry Smith          create a special DMDA to handle one level of ghost points for graphics
12147c6ae99SBarry Smith       */
1221321219cSEthan 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);
123aa219208SBarry Smith       ierr = PetscInfo(da,"Creating auxilary DMDA for managing graphics ghost points\n");CHKERRQ(ierr);
12447c6ae99SBarry Smith     } else {
12547c6ae99SBarry Smith       /* otherwise we can use the da we already have */
12647c6ae99SBarry Smith       dac = da;
12747c6ae99SBarry Smith     }
12847c6ae99SBarry Smith     /* create local vector for holding ghosted values used in graphics */
129564755cdSBarry Smith     ierr = DMCreateLocalVector(dac,&xlocal);CHKERRQ(ierr);
13047c6ae99SBarry Smith     if (dac != da) {
131aa219208SBarry Smith       /* don't keep any public reference of this DMDA, is is only available through xlocal */
132f7923d8aSBarry Smith       ierr = PetscObjectDereference((PetscObject)dac);CHKERRQ(ierr);
13347c6ae99SBarry Smith     } else {
13447c6ae99SBarry Smith       /* remove association between xlocal and da, because below we compose in the opposite
13547c6ae99SBarry Smith          direction and if we left this connect we'd get a loop, so the objects could
13647c6ae99SBarry Smith          never be destroyed */
137bfec8eecSBarry Smith       ierr = PetscObjectRemoveReference((PetscObject)xlocal,"DM");CHKERRQ(ierr);
13847c6ae99SBarry Smith     }
13947c6ae99SBarry Smith     ierr = PetscObjectCompose((PetscObject)da,"GraphicsGhosted",(PetscObject)xlocal);CHKERRQ(ierr);
14047c6ae99SBarry Smith     ierr = PetscObjectDereference((PetscObject)xlocal);CHKERRQ(ierr);
14147c6ae99SBarry Smith   } else {
142f7923d8aSBarry Smith     if (bx !=  DMDA_BOUNDARY_NONE || by !=  DMDA_BOUNDARY_NONE || s != 1 || st != DMDA_STENCIL_BOX) {
1433c0c59f3SBarry Smith       ierr = PetscObjectQuery((PetscObject)xlocal,"DM",(PetscObject*)&dac);CHKERRQ(ierr);
144f7923d8aSBarry Smith     } else {
145f7923d8aSBarry Smith       dac = da;
14647c6ae99SBarry Smith     }
14747c6ae99SBarry Smith   }
14847c6ae99SBarry Smith 
14947c6ae99SBarry Smith   /*
15047c6ae99SBarry Smith       Get local (ghosted) values of vector
15147c6ae99SBarry Smith   */
1529a42bb27SBarry Smith   ierr = DMGlobalToLocalBegin(dac,xin,INSERT_VALUES,xlocal);CHKERRQ(ierr);
1539a42bb27SBarry Smith   ierr = DMGlobalToLocalEnd(dac,xin,INSERT_VALUES,xlocal);CHKERRQ(ierr);
15447c6ae99SBarry Smith   ierr = VecGetArray(xlocal,&zctx.v);CHKERRQ(ierr);
15547c6ae99SBarry Smith 
15647c6ae99SBarry Smith   /* get coordinates of nodes */
157aa219208SBarry Smith   ierr = DMDAGetCoordinates(da,&xcoor);CHKERRQ(ierr);
15847c6ae99SBarry Smith   if (!xcoor) {
159aa219208SBarry Smith     ierr = DMDASetUniformCoordinates(da,0.0,1.0,0.0,1.0,0.0,0.0);CHKERRQ(ierr);
160aa219208SBarry Smith     ierr = DMDAGetCoordinates(da,&xcoor);CHKERRQ(ierr);
16147c6ae99SBarry Smith   }
16247c6ae99SBarry Smith 
16347c6ae99SBarry Smith   /*
16447c6ae99SBarry Smith       Determine the min and max  coordinates in plot
16547c6ae99SBarry Smith   */
16647c6ae99SBarry Smith   ierr = VecStrideMin(xcoor,0,PETSC_NULL,&xmin);CHKERRQ(ierr);
16747c6ae99SBarry Smith   ierr = VecStrideMax(xcoor,0,PETSC_NULL,&xmax);CHKERRQ(ierr);
16847c6ae99SBarry Smith   ierr = VecStrideMin(xcoor,1,PETSC_NULL,&ymin);CHKERRQ(ierr);
16947c6ae99SBarry Smith   ierr = VecStrideMax(xcoor,1,PETSC_NULL,&ymax);CHKERRQ(ierr);
17047c6ae99SBarry Smith   coors[0] = xmin - .05*(xmax- xmin); coors[2] = xmax + .05*(xmax - xmin);
17147c6ae99SBarry Smith   coors[1] = ymin - .05*(ymax- ymin); coors[3] = ymax + .05*(ymax - ymin);
172aa219208SBarry 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);
17347c6ae99SBarry Smith 
17447c6ae99SBarry Smith   /*
17547c6ae99SBarry Smith        get local ghosted version of coordinates
17647c6ae99SBarry Smith   */
17747c6ae99SBarry Smith   ierr = PetscObjectQuery((PetscObject)da,"GraphicsCoordinateGhosted",(PetscObject*)&xcoorl);CHKERRQ(ierr);
17847c6ae99SBarry Smith   if (!xcoorl) {
179aa219208SBarry Smith     /* create DMDA to get local version of graphics */
1801321219cSEthan 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);
181aa219208SBarry Smith     ierr = PetscInfo(dag,"Creating auxilary DMDA for managing graphics coordinates ghost points\n");CHKERRQ(ierr);
182564755cdSBarry Smith     ierr = DMCreateLocalVector(dag,&xcoorl);CHKERRQ(ierr);
18347c6ae99SBarry Smith     ierr = PetscObjectCompose((PetscObject)da,"GraphicsCoordinateGhosted",(PetscObject)xcoorl);CHKERRQ(ierr);
184f7923d8aSBarry Smith     ierr = PetscObjectDereference((PetscObject)dag);CHKERRQ(ierr);
18547c6ae99SBarry Smith     ierr = PetscObjectDereference((PetscObject)xcoorl);CHKERRQ(ierr);
18647c6ae99SBarry Smith   } else {
1873c0c59f3SBarry Smith     ierr = PetscObjectQuery((PetscObject)xcoorl,"DM",(PetscObject*)&dag);CHKERRQ(ierr);
18847c6ae99SBarry Smith   }
1899a42bb27SBarry Smith   ierr = DMGlobalToLocalBegin(dag,xcoor,INSERT_VALUES,xcoorl);CHKERRQ(ierr);
1909a42bb27SBarry Smith   ierr = DMGlobalToLocalEnd(dag,xcoor,INSERT_VALUES,xcoorl);CHKERRQ(ierr);
19147c6ae99SBarry Smith   ierr = VecGetArray(xcoorl,&zctx.xy);CHKERRQ(ierr);
19247c6ae99SBarry Smith 
19347c6ae99SBarry Smith   /*
19447c6ae99SBarry Smith         Get information about size of area each processor must do graphics for
19547c6ae99SBarry Smith   */
1961321219cSEthan Coon   ierr = DMDAGetInfo(dac,0,&M,&N,0,0,0,0,&zctx.step,0,&bx,&by,0,0);CHKERRQ(ierr);
197f7923d8aSBarry Smith   ierr = DMDAGetGhostCorners(dac,0,0,0,&zctx.m,&zctx.n,0);CHKERRQ(ierr);
19847c6ae99SBarry Smith 
199671f6225SBarry Smith   ierr = PetscOptionsGetBool(PETSC_NULL,"-draw_contour_grid",&zctx.showgrid,PETSC_NULL);CHKERRQ(ierr);
20020d0051dSBarry Smith   ierr = PetscMalloc(zctx.step*sizeof(PetscInt),&displayfields);CHKERRQ(ierr);
20120d0051dSBarry Smith   for (i=0; i<zctx.step; i++) displayfields[i] = i;
20220d0051dSBarry Smith   ndisplayfields = zctx.step;
20320d0051dSBarry Smith   ierr = PetscOptionsGetIntArray(PETSC_NULL,"-draw_fields",displayfields,&ndisplayfields,&flg);CHKERRQ(ierr);
20420d0051dSBarry Smith   if (!flg) ndisplayfields = zctx.step;
20547c6ae99SBarry Smith 
20647c6ae99SBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
207671f6225SBarry Smith   ierr = PetscOptionsGetBool(PETSC_NULL,"-draw_ports",&useports,PETSC_NULL);CHKERRQ(ierr);
20847c6ae99SBarry Smith   if (useports || format == PETSC_VIEWER_DRAW_PORTS){
20947c6ae99SBarry Smith     ierr = PetscDrawSynchronizedClear(draw);CHKERRQ(ierr);
21020d0051dSBarry Smith     ierr = PetscDrawViewPortsCreate(draw,ndisplayfields,&ports);CHKERRQ(ierr);
21147c6ae99SBarry Smith   }
21220d0051dSBarry Smith 
21347c6ae99SBarry Smith   /*
21447c6ae99SBarry Smith      Loop over each field; drawing each in a different window
21547c6ae99SBarry Smith   */
21620d0051dSBarry Smith   for (i=0; i<ndisplayfields; i++) {
21720d0051dSBarry Smith     zctx.k = displayfields[i];
21847c6ae99SBarry Smith     if (useports) {
21920d0051dSBarry Smith       ierr = PetscDrawViewPortsSet(ports,i);CHKERRQ(ierr);
22047c6ae99SBarry Smith     } else {
22120d0051dSBarry Smith       ierr = PetscViewerDrawGetDraw(viewer,i,&draw);CHKERRQ(ierr);
22247c6ae99SBarry Smith       ierr = PetscDrawSynchronizedClear(draw);CHKERRQ(ierr);
22347c6ae99SBarry Smith     }
22447c6ae99SBarry Smith 
22547c6ae99SBarry Smith     /*
22647c6ae99SBarry Smith         Determine the min and max color in plot
22747c6ae99SBarry Smith     */
22847c6ae99SBarry Smith     ierr = VecStrideMin(xin,zctx.k,PETSC_NULL,&zctx.min);CHKERRQ(ierr);
22947c6ae99SBarry Smith     ierr = VecStrideMax(xin,zctx.k,PETSC_NULL,&zctx.max);CHKERRQ(ierr);
23067dd0837SBarry Smith     if (zctx.k < nbounds) {
23167dd0837SBarry Smith       zctx.min = PetscMin(zctx.min,bounds[2*zctx.k]);
23267dd0837SBarry Smith       zctx.max = PetscMax(zctx.max,bounds[2*zctx.k+1]);
23367dd0837SBarry Smith     }
23447c6ae99SBarry Smith     if (zctx.min == zctx.max) {
23547c6ae99SBarry Smith       zctx.min -= 1.e-12;
23647c6ae99SBarry Smith       zctx.max += 1.e-12;
23747c6ae99SBarry Smith     }
23847c6ae99SBarry Smith 
23947c6ae99SBarry Smith     if (!rank) {
24047c6ae99SBarry Smith       const char *title;
24147c6ae99SBarry Smith 
242aa219208SBarry Smith       ierr = DMDAGetFieldName(da,zctx.k,&title);CHKERRQ(ierr);
24347c6ae99SBarry Smith       if (title) {
24447c6ae99SBarry Smith         ierr = PetscDrawSetTitle(draw,title);CHKERRQ(ierr);
24547c6ae99SBarry Smith       }
24647c6ae99SBarry Smith     }
24747c6ae99SBarry Smith     ierr = PetscDrawSetCoordinates(draw,coors[0],coors[1],coors[2],coors[3]);CHKERRQ(ierr);
248aa219208SBarry Smith     ierr = PetscInfo2(da,"DMDA 2d contour plot min %G max %G\n",zctx.min,zctx.max);CHKERRQ(ierr);
24947c6ae99SBarry Smith 
25047c6ae99SBarry Smith     ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr);
25147c6ae99SBarry Smith     if (popup) {ierr = PetscDrawScalePopup(popup,zctx.min,zctx.max);CHKERRQ(ierr);}
25247c6ae99SBarry Smith 
25347c6ae99SBarry Smith     zctx.scale = (245.0 - PETSC_DRAW_BASIC_COLORS)/(zctx.max - zctx.min);
25447c6ae99SBarry Smith 
25547c6ae99SBarry Smith     ierr = PetscDrawZoom(draw,VecView_MPI_Draw_DA2d_Zoom,&zctx);CHKERRQ(ierr);
25647c6ae99SBarry Smith   }
25720d0051dSBarry Smith   ierr = PetscFree(displayfields);CHKERRQ(ierr);
2586bf464f9SBarry Smith   ierr = PetscDrawViewPortsDestroy(ports);CHKERRQ(ierr);
25947c6ae99SBarry Smith 
26047c6ae99SBarry Smith   ierr = VecRestoreArray(xcoorl,&zctx.xy);CHKERRQ(ierr);
26147c6ae99SBarry Smith   ierr = VecRestoreArray(xlocal,&zctx.v);CHKERRQ(ierr);
26247c6ae99SBarry Smith   PetscFunctionReturn(0);
26347c6ae99SBarry Smith }
26447c6ae99SBarry Smith 
26547c6ae99SBarry Smith 
26647c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
26747c6ae99SBarry Smith #undef __FUNCT__
26847c6ae99SBarry Smith #define __FUNCT__ "VecView_MPI_HDF5_DA"
26947c6ae99SBarry Smith PetscErrorCode VecView_MPI_HDF5_DA(Vec xin,PetscViewer viewer)
27047c6ae99SBarry Smith {
27147c6ae99SBarry Smith   PetscErrorCode ierr;
2729b2a5a86SJed Brown   DM             dm;
2739b2a5a86SJed Brown   DM_DA          *da;
27447c6ae99SBarry Smith   hsize_t        dim,dims[5];
27547c6ae99SBarry Smith   hsize_t        count[5];
27647c6ae99SBarry Smith   hsize_t        offset[5];
27747c6ae99SBarry Smith   PetscInt       cnt = 0;
27847c6ae99SBarry Smith   PetscScalar    *x;
27947c6ae99SBarry Smith   const char     *vecname;
28047c6ae99SBarry Smith   hid_t          filespace; /* file dataspace identifier */
28147c6ae99SBarry Smith   hid_t	         plist_id;  /* property list identifier */
28247c6ae99SBarry Smith   hid_t          dset_id;   /* dataset identifier */
28347c6ae99SBarry Smith   hid_t          memspace;  /* memory dataspace identifier */
28447c6ae99SBarry Smith   hid_t          file_id;
28547c6ae99SBarry Smith   herr_t         status;
28647c6ae99SBarry Smith 
28747c6ae99SBarry Smith   PetscFunctionBegin;
28847c6ae99SBarry Smith   ierr = PetscViewerHDF5GetFileId(viewer, &file_id);CHKERRQ(ierr);
2893c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&dm);CHKERRQ(ierr);
2909b2a5a86SJed Brown   if (!dm) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
2919b2a5a86SJed Brown   da = (DM_DA*)dm->data;
29247c6ae99SBarry Smith 
29347c6ae99SBarry Smith   /* Create the dataspace for the dataset */
29447c6ae99SBarry Smith   dim       = PetscHDF5IntCast(da->dim + ((da->w == 1) ? 0 : 1));
29547c6ae99SBarry Smith   if (da->dim == 3) dims[cnt++]   = PetscHDF5IntCast(da->P);
29647c6ae99SBarry Smith   if (da->dim > 1)  dims[cnt++]   = PetscHDF5IntCast(da->N);
29747c6ae99SBarry Smith   dims[cnt++]   = PetscHDF5IntCast(da->M);
29847c6ae99SBarry Smith   if (da->w > 1) dims[cnt++] = PetscHDF5IntCast(da->w);
29947c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
30047c6ae99SBarry Smith   dim++;
30147c6ae99SBarry Smith   dims[cnt++] = 2;
30247c6ae99SBarry Smith #endif
30347c6ae99SBarry Smith   filespace = H5Screate_simple(dim, dims, NULL);
30447c6ae99SBarry Smith   if (filespace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Screate_simple()");
30547c6ae99SBarry Smith 
30647c6ae99SBarry Smith   /* Create the dataset with default properties and close filespace */
30747c6ae99SBarry Smith   ierr = PetscObjectGetName((PetscObject)xin,&vecname);CHKERRQ(ierr);
30847c6ae99SBarry Smith #if (H5_VERS_MAJOR * 10000 + H5_VERS_MINOR * 100 + H5_VERS_RELEASE >= 10800)
30947c6ae99SBarry Smith   dset_id = H5Dcreate2(file_id, vecname, H5T_NATIVE_DOUBLE, filespace, H5P_DEFAULT, H5P_DEFAULT, H5P_DEFAULT);
31047c6ae99SBarry Smith #else
31147c6ae99SBarry Smith   dset_id = H5Dcreate(file_id, vecname, H5T_NATIVE_DOUBLE, filespace, H5P_DEFAULT);
31247c6ae99SBarry Smith #endif
31347c6ae99SBarry Smith   if (dset_id == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Dcreate2()");
31447c6ae99SBarry Smith   status = H5Sclose(filespace);CHKERRQ(status);
31547c6ae99SBarry Smith 
31647c6ae99SBarry Smith   /* Each process defines a dataset and writes it to the hyperslab in the file */
31747c6ae99SBarry Smith   cnt = 0;
31847c6ae99SBarry Smith   if (da->dim == 3) offset[cnt++] = PetscHDF5IntCast(da->zs);
31947c6ae99SBarry Smith   if (da->dim > 1)  offset[cnt++] = PetscHDF5IntCast(da->ys);
32047c6ae99SBarry Smith   offset[cnt++] = PetscHDF5IntCast(da->xs/da->w);
32147c6ae99SBarry Smith   if (da->w > 1) offset[cnt++] = 0;
32247c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
32347c6ae99SBarry Smith   offset[cnt++] = 0;
32447c6ae99SBarry Smith #endif
32547c6ae99SBarry Smith   cnt = 0;
32647c6ae99SBarry Smith   if (da->dim == 3) count[cnt++] = PetscHDF5IntCast(da->ze - da->zs);
32747c6ae99SBarry Smith   if (da->dim > 1)  count[cnt++] = PetscHDF5IntCast(da->ye - da->ys);
32847c6ae99SBarry Smith   count[cnt++] = PetscHDF5IntCast((da->xe - da->xs)/da->w);
32947c6ae99SBarry Smith   if (da->w > 1) count[cnt++] = PetscHDF5IntCast(da->w);
33047c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
33147c6ae99SBarry Smith   count[cnt++] = 2;
33247c6ae99SBarry Smith #endif
33347c6ae99SBarry Smith   memspace = H5Screate_simple(dim, count, NULL);
33447c6ae99SBarry Smith   if (memspace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Screate_simple()");
33547c6ae99SBarry Smith 
33647c6ae99SBarry Smith 
33747c6ae99SBarry Smith   filespace = H5Dget_space(dset_id);
33847c6ae99SBarry Smith   if (filespace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Dget_space()");
33947c6ae99SBarry Smith   status = H5Sselect_hyperslab(filespace, H5S_SELECT_SET, offset, NULL, count, NULL);CHKERRQ(status);
34047c6ae99SBarry Smith 
34147c6ae99SBarry Smith   /* Create property list for collective dataset write */
34247c6ae99SBarry Smith   plist_id = H5Pcreate(H5P_DATASET_XFER);
34347c6ae99SBarry Smith   if (plist_id == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Pcreate()");
34447c6ae99SBarry Smith #if defined(PETSC_HAVE_H5PSET_FAPL_MPIO)
34547c6ae99SBarry Smith   status = H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_COLLECTIVE);CHKERRQ(status);
34647c6ae99SBarry Smith #endif
34747c6ae99SBarry Smith   /* To write dataset independently use H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_INDEPENDENT) */
34847c6ae99SBarry Smith 
34947c6ae99SBarry Smith   ierr = VecGetArray(xin, &x);CHKERRQ(ierr);
35047c6ae99SBarry Smith   status = H5Dwrite(dset_id, H5T_NATIVE_DOUBLE, memspace, filespace, plist_id, x);CHKERRQ(status);
35147c6ae99SBarry Smith   status = H5Fflush(file_id, H5F_SCOPE_GLOBAL);CHKERRQ(status);
35247c6ae99SBarry Smith   ierr = VecRestoreArray(xin, &x);CHKERRQ(ierr);
35347c6ae99SBarry Smith 
35447c6ae99SBarry Smith   /* Close/release resources */
35547c6ae99SBarry Smith   status = H5Pclose(plist_id);CHKERRQ(status);
35647c6ae99SBarry Smith   status = H5Sclose(filespace);CHKERRQ(status);
35747c6ae99SBarry Smith   status = H5Sclose(memspace);CHKERRQ(status);
35847c6ae99SBarry Smith   status = H5Dclose(dset_id);CHKERRQ(status);
35947c6ae99SBarry Smith   ierr = PetscInfo1(xin,"Wrote Vec object with name %s\n",vecname);CHKERRQ(ierr);
36047c6ae99SBarry Smith   PetscFunctionReturn(0);
36147c6ae99SBarry Smith }
36247c6ae99SBarry Smith #endif
36347c6ae99SBarry Smith 
36409573ac7SBarry Smith extern PetscErrorCode VecView_MPI_Draw_DA1d(Vec,PetscViewer);
36547c6ae99SBarry Smith 
36647c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
36747c6ae99SBarry Smith #undef __FUNCT__
368aa219208SBarry Smith #define __FUNCT__ "DMDAArrayMPIIO"
369aa219208SBarry Smith static PetscErrorCode DMDAArrayMPIIO(DM da,PetscViewer viewer,Vec xin,PetscBool  write)
37047c6ae99SBarry Smith {
37147c6ae99SBarry Smith   PetscErrorCode    ierr;
37247c6ae99SBarry Smith   MPI_File          mfdes;
37347c6ae99SBarry Smith   PetscMPIInt       gsizes[4],lsizes[4],lstarts[4],asiz,dof;
37447c6ae99SBarry Smith   MPI_Datatype      view;
37547c6ae99SBarry Smith   const PetscScalar *array;
37647c6ae99SBarry Smith   MPI_Offset        off;
37747c6ae99SBarry Smith   MPI_Aint          ub,ul;
37847c6ae99SBarry Smith   PetscInt          type,rows,vecrows,tr[2];
37947c6ae99SBarry Smith   DM_DA             *dd = (DM_DA*)da->data;
38047c6ae99SBarry Smith 
38147c6ae99SBarry Smith   PetscFunctionBegin;
38247c6ae99SBarry Smith   ierr = VecGetSize(xin,&vecrows);CHKERRQ(ierr);
38347c6ae99SBarry Smith   if (!write) {
38447c6ae99SBarry Smith     /* Read vector header. */
38547c6ae99SBarry Smith     ierr = PetscViewerBinaryRead(viewer,tr,2,PETSC_INT);CHKERRQ(ierr);
38647c6ae99SBarry Smith     type = tr[0];
38747c6ae99SBarry Smith     rows = tr[1];
38847c6ae99SBarry Smith     if (type != VEC_FILE_CLASSID) SETERRQ(((PetscObject)da)->comm,PETSC_ERR_ARG_WRONG,"Not vector next in file");
389aa219208SBarry Smith     if (rows != vecrows) SETERRQ(((PetscObject)da)->comm,PETSC_ERR_ARG_SIZ,"Vector in file not same size as DMDA vector");
39047c6ae99SBarry Smith   } else {
39147c6ae99SBarry Smith     tr[0] = VEC_FILE_CLASSID;
39247c6ae99SBarry Smith     tr[1] = vecrows;
39347c6ae99SBarry Smith     ierr = PetscViewerBinaryWrite(viewer,tr,2,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
39447c6ae99SBarry Smith   }
39547c6ae99SBarry Smith 
39647c6ae99SBarry Smith   dof = PetscMPIIntCast(dd->w);
39747c6ae99SBarry Smith   gsizes[0]  = dof; gsizes[1] = PetscMPIIntCast(dd->M); gsizes[2] = PetscMPIIntCast(dd->N); gsizes[3] = PetscMPIIntCast(dd->P);
39847c6ae99SBarry 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);
39947c6ae99SBarry Smith   lstarts[0] = 0;  lstarts[1] = PetscMPIIntCast(dd->xs/dof); lstarts[2] = PetscMPIIntCast(dd->ys); lstarts[3] = PetscMPIIntCast(dd->zs);
40047c6ae99SBarry Smith   ierr = MPI_Type_create_subarray(dd->dim+1,gsizes,lsizes,lstarts,MPI_ORDER_FORTRAN,MPIU_SCALAR,&view);CHKERRQ(ierr);
40147c6ae99SBarry Smith   ierr = MPI_Type_commit(&view);CHKERRQ(ierr);
40247c6ae99SBarry Smith 
40347c6ae99SBarry Smith   ierr = PetscViewerBinaryGetMPIIODescriptor(viewer,&mfdes);CHKERRQ(ierr);
40447c6ae99SBarry Smith   ierr = PetscViewerBinaryGetMPIIOOffset(viewer,&off);CHKERRQ(ierr);
40547c6ae99SBarry Smith   ierr = MPI_File_set_view(mfdes,off,MPIU_SCALAR,view,(char *)"native",MPI_INFO_NULL);CHKERRQ(ierr);
40647c6ae99SBarry Smith   ierr = VecGetArrayRead(xin,&array);CHKERRQ(ierr);
40747c6ae99SBarry Smith   asiz = lsizes[1]*(lsizes[2] > 0 ? lsizes[2] : 1)*(lsizes[3] > 0 ? lsizes[3] : 1)*dof;
40847c6ae99SBarry Smith   if (write) {
40947c6ae99SBarry Smith     ierr = MPIU_File_write_all(mfdes,(PetscScalar*)array,asiz,MPIU_SCALAR,MPI_STATUS_IGNORE);CHKERRQ(ierr);
41047c6ae99SBarry Smith   } else {
41147c6ae99SBarry Smith     ierr = MPIU_File_read_all(mfdes,(PetscScalar*)array,asiz,MPIU_SCALAR,MPI_STATUS_IGNORE);CHKERRQ(ierr);
41247c6ae99SBarry Smith   }
41347c6ae99SBarry Smith   ierr = MPI_Type_get_extent(view,&ul,&ub);CHKERRQ(ierr);
41447c6ae99SBarry Smith   ierr = PetscViewerBinaryAddMPIIOOffset(viewer,ub);CHKERRQ(ierr);
41547c6ae99SBarry Smith   ierr = VecRestoreArrayRead(xin,&array);CHKERRQ(ierr);
41647c6ae99SBarry Smith   ierr = MPI_Type_free(&view);CHKERRQ(ierr);
41747c6ae99SBarry Smith   PetscFunctionReturn(0);
41847c6ae99SBarry Smith }
41947c6ae99SBarry Smith #endif
42047c6ae99SBarry Smith 
42147c6ae99SBarry Smith EXTERN_C_BEGIN
42247c6ae99SBarry Smith #undef __FUNCT__
42347c6ae99SBarry Smith #define __FUNCT__ "VecView_MPI_DA"
4247087cfbeSBarry Smith PetscErrorCode  VecView_MPI_DA(Vec xin,PetscViewer viewer)
42547c6ae99SBarry Smith {
4269a42bb27SBarry Smith   DM             da;
42747c6ae99SBarry Smith   PetscErrorCode ierr;
42847c6ae99SBarry Smith   PetscInt       dim;
42947c6ae99SBarry Smith   Vec            natural;
43047c6ae99SBarry Smith   PetscBool      isdraw;
43147c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
43247c6ae99SBarry Smith   PetscBool      ishdf5;
43347c6ae99SBarry Smith #endif
4343f3fd955SJed Brown   const char     *prefix,*name;
43547c6ae99SBarry Smith 
43647c6ae99SBarry Smith   PetscFunctionBegin;
4373c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
438aa219208SBarry Smith   if (!da) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
43947c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
44047c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
44147c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERHDF5,&ishdf5);CHKERRQ(ierr);
44247c6ae99SBarry Smith #endif
44347c6ae99SBarry Smith   if (isdraw) {
4441321219cSEthan Coon     ierr = DMDAGetInfo(da,&dim,0,0,0,0,0,0,0,0,0,0,0,0);CHKERRQ(ierr);
44547c6ae99SBarry Smith     if (dim == 1) {
44647c6ae99SBarry Smith       ierr = VecView_MPI_Draw_DA1d(xin,viewer);CHKERRQ(ierr);
44747c6ae99SBarry Smith     } else if (dim == 2) {
44847c6ae99SBarry Smith       ierr = VecView_MPI_Draw_DA2d(xin,viewer);CHKERRQ(ierr);
44947c6ae99SBarry Smith     } else {
450aa219208SBarry Smith       SETERRQ1(((PetscObject)da)->comm,PETSC_ERR_SUP,"Cannot graphically view vector associated with this dimensional DMDA %D",dim);
45147c6ae99SBarry Smith     }
45247c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
45347c6ae99SBarry Smith   } else if (ishdf5) {
45447c6ae99SBarry Smith     ierr = VecView_MPI_HDF5_DA(xin,viewer);CHKERRQ(ierr);
45547c6ae99SBarry Smith #endif
45647c6ae99SBarry Smith   } else {
45747c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
45847c6ae99SBarry Smith     PetscBool  isbinary,isMPIIO;
45947c6ae99SBarry Smith 
46047c6ae99SBarry Smith     ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
46147c6ae99SBarry Smith     if (isbinary) {
46247c6ae99SBarry Smith       ierr = PetscViewerBinaryGetMPIIO(viewer,&isMPIIO);CHKERRQ(ierr);
46347c6ae99SBarry Smith       if (isMPIIO) {
464aa219208SBarry Smith        ierr = DMDAArrayMPIIO(da,viewer,xin,PETSC_TRUE);CHKERRQ(ierr);
46547c6ae99SBarry Smith        PetscFunctionReturn(0);
46647c6ae99SBarry Smith       }
46747c6ae99SBarry Smith     }
46847c6ae99SBarry Smith #endif
46947c6ae99SBarry Smith 
47047c6ae99SBarry Smith     /* call viewer on natural ordering */
47147c6ae99SBarry Smith     ierr = PetscObjectGetOptionsPrefix((PetscObject)xin,&prefix);CHKERRQ(ierr);
472aa219208SBarry Smith     ierr = DMDACreateNaturalVector(da,&natural);CHKERRQ(ierr);
47347c6ae99SBarry Smith     ierr = PetscObjectSetOptionsPrefix((PetscObject)natural,prefix);CHKERRQ(ierr);
474aa219208SBarry Smith     ierr = DMDAGlobalToNaturalBegin(da,xin,INSERT_VALUES,natural);CHKERRQ(ierr);
475aa219208SBarry Smith     ierr = DMDAGlobalToNaturalEnd(da,xin,INSERT_VALUES,natural);CHKERRQ(ierr);
4763f3fd955SJed Brown     ierr = PetscObjectGetName((PetscObject)xin,&name);CHKERRQ(ierr);
4773f3fd955SJed Brown     ierr = PetscObjectSetName((PetscObject)natural,name);CHKERRQ(ierr);
47847c6ae99SBarry Smith     ierr = VecView(natural,viewer);CHKERRQ(ierr);
479fcfd50ebSBarry Smith     ierr = VecDestroy(&natural);CHKERRQ(ierr);
48047c6ae99SBarry Smith   }
48147c6ae99SBarry Smith   PetscFunctionReturn(0);
48247c6ae99SBarry Smith }
48347c6ae99SBarry Smith EXTERN_C_END
48447c6ae99SBarry Smith 
48547c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
48647c6ae99SBarry Smith #undef __FUNCT__
48747c6ae99SBarry Smith #define __FUNCT__ "VecLoad_HDF5_DA"
48847c6ae99SBarry Smith PetscErrorCode VecLoad_HDF5_DA(Vec xin, PetscViewer viewer)
48947c6ae99SBarry Smith {
4909a42bb27SBarry Smith   DM             da;
49147c6ae99SBarry Smith   PetscErrorCode ierr;
49225578ef6SJed Brown   hsize_t        dim;
49347c6ae99SBarry Smith   hsize_t        count[5];
49447c6ae99SBarry Smith   hsize_t        offset[5];
49547c6ae99SBarry Smith   PetscInt       cnt = 0;
49647c6ae99SBarry Smith   PetscScalar    *x;
49747c6ae99SBarry Smith   const char     *vecname;
49847c6ae99SBarry Smith   hid_t          filespace; /* file dataspace identifier */
49947c6ae99SBarry Smith   hid_t	         plist_id;  /* property list identifier */
50047c6ae99SBarry Smith   hid_t          dset_id;   /* dataset identifier */
50147c6ae99SBarry Smith   hid_t          memspace;  /* memory dataspace identifier */
50247c6ae99SBarry Smith   hid_t          file_id;
50347c6ae99SBarry Smith   herr_t         status;
5049c7c4993SBarry Smith   DM_DA          *dd;
50547c6ae99SBarry Smith 
50647c6ae99SBarry Smith   PetscFunctionBegin;
50747c6ae99SBarry Smith   ierr = PetscViewerHDF5GetFileId(viewer, &file_id);CHKERRQ(ierr);
5083c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
5099c7c4993SBarry Smith   dd = (DM_DA*)da->data;
51047c6ae99SBarry Smith 
51147c6ae99SBarry Smith   /* Create the dataspace for the dataset */
51247c6ae99SBarry Smith   dim       = PetscHDF5IntCast(dd->dim + ((dd->w == 1) ? 0 : 1));
51347c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
51447c6ae99SBarry Smith   dim++;
51547c6ae99SBarry Smith #endif
51647c6ae99SBarry Smith 
51747c6ae99SBarry Smith   /* Create the dataset with default properties and close filespace */
51847c6ae99SBarry Smith   ierr = PetscObjectGetName((PetscObject)xin,&vecname);CHKERRQ(ierr);
51947c6ae99SBarry Smith #if (H5_VERS_MAJOR * 10000 + H5_VERS_MINOR * 100 + H5_VERS_RELEASE >= 10800)
52047c6ae99SBarry Smith   dset_id = H5Dopen2(file_id, vecname, H5P_DEFAULT);
52147c6ae99SBarry Smith #else
52247c6ae99SBarry Smith   dset_id = H5Dopen(file_id, vecname);
52347c6ae99SBarry Smith #endif
52447c6ae99SBarry Smith   if (dset_id == -1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Dopen2() with Vec named %s",vecname);
52547c6ae99SBarry Smith   filespace = H5Dget_space(dset_id);
52647c6ae99SBarry Smith 
52747c6ae99SBarry Smith   /* Each process defines a dataset and reads it from the hyperslab in the file */
52847c6ae99SBarry Smith   cnt = 0;
52947c6ae99SBarry Smith   if (dd->dim == 3) offset[cnt++] = PetscHDF5IntCast(dd->zs);
53047c6ae99SBarry Smith   if (dd->dim > 1)  offset[cnt++] = PetscHDF5IntCast(dd->ys);
53147c6ae99SBarry Smith   offset[cnt++] = PetscHDF5IntCast(dd->xs/dd->w);
53247c6ae99SBarry Smith   if (dd->w > 1) offset[cnt++] = 0;
53347c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
53447c6ae99SBarry Smith   offset[cnt++] = 0;
53547c6ae99SBarry Smith #endif
53647c6ae99SBarry Smith   cnt = 0;
53747c6ae99SBarry Smith   if (dd->dim == 3) count[cnt++] = PetscHDF5IntCast(dd->ze - dd->zs);
53847c6ae99SBarry Smith   if (dd->dim > 1)  count[cnt++] = PetscHDF5IntCast(dd->ye - dd->ys);
53947c6ae99SBarry Smith   count[cnt++] = PetscHDF5IntCast((dd->xe - dd->xs)/dd->w);
54047c6ae99SBarry Smith   if (dd->w > 1) count[cnt++] = PetscHDF5IntCast(dd->w);
54147c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
54247c6ae99SBarry Smith   count[cnt++] = 2;
54347c6ae99SBarry Smith #endif
54447c6ae99SBarry Smith   memspace = H5Screate_simple(dim, count, NULL);
54547c6ae99SBarry Smith   if (memspace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Screate_simple()");
54647c6ae99SBarry Smith 
54747c6ae99SBarry Smith   status = H5Sselect_hyperslab(filespace, H5S_SELECT_SET, offset, NULL, count, NULL);CHKERRQ(status);
54847c6ae99SBarry Smith 
54947c6ae99SBarry Smith   /* Create property list for collective dataset write */
55047c6ae99SBarry Smith   plist_id = H5Pcreate(H5P_DATASET_XFER);
55147c6ae99SBarry Smith   if (plist_id == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Pcreate()");
55247c6ae99SBarry Smith #if defined(PETSC_HAVE_H5PSET_FAPL_MPIO)
55347c6ae99SBarry Smith   status = H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_COLLECTIVE);CHKERRQ(status);
55447c6ae99SBarry Smith #endif
55547c6ae99SBarry Smith   /* To write dataset independently use H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_INDEPENDENT) */
55647c6ae99SBarry Smith 
55747c6ae99SBarry Smith   ierr = VecGetArray(xin, &x);CHKERRQ(ierr);
55847c6ae99SBarry Smith   status = H5Dread(dset_id, H5T_NATIVE_DOUBLE, memspace, filespace, plist_id, x);CHKERRQ(status);
55947c6ae99SBarry Smith   ierr = VecRestoreArray(xin, &x);CHKERRQ(ierr);
56047c6ae99SBarry Smith 
56147c6ae99SBarry Smith   /* Close/release resources */
56247c6ae99SBarry Smith   status = H5Pclose(plist_id);CHKERRQ(status);
56347c6ae99SBarry Smith   status = H5Sclose(filespace);CHKERRQ(status);
56447c6ae99SBarry Smith   status = H5Sclose(memspace);CHKERRQ(status);
56547c6ae99SBarry Smith   status = H5Dclose(dset_id);CHKERRQ(status);
56647c6ae99SBarry Smith   PetscFunctionReturn(0);
56747c6ae99SBarry Smith }
56847c6ae99SBarry Smith #endif
56947c6ae99SBarry Smith 
57047c6ae99SBarry Smith #undef __FUNCT__
57147c6ae99SBarry Smith #define __FUNCT__ "VecLoad_Binary_DA"
57247c6ae99SBarry Smith PetscErrorCode VecLoad_Binary_DA(Vec xin, PetscViewer viewer)
57347c6ae99SBarry Smith {
5749a42bb27SBarry Smith   DM             da;
57547c6ae99SBarry Smith   PetscErrorCode ierr;
57647c6ae99SBarry Smith   Vec            natural;
57747c6ae99SBarry Smith   const char     *prefix;
57847c6ae99SBarry Smith   PetscInt       bs;
57947c6ae99SBarry Smith   PetscBool      flag;
58047c6ae99SBarry Smith   DM_DA          *dd;
58147c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
58247c6ae99SBarry Smith   PetscBool      isMPIIO;
58347c6ae99SBarry Smith #endif
58447c6ae99SBarry Smith 
58547c6ae99SBarry Smith   PetscFunctionBegin;
5863c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
58747c6ae99SBarry Smith   dd   = (DM_DA*)da->data;
58847c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
58947c6ae99SBarry Smith   ierr = PetscViewerBinaryGetMPIIO(viewer,&isMPIIO);CHKERRQ(ierr);
59047c6ae99SBarry Smith   if (isMPIIO) {
591aa219208SBarry Smith     ierr = DMDAArrayMPIIO(da,viewer,xin,PETSC_FALSE);CHKERRQ(ierr);
59247c6ae99SBarry Smith     PetscFunctionReturn(0);
59347c6ae99SBarry Smith   }
59447c6ae99SBarry Smith #endif
59547c6ae99SBarry Smith 
59647c6ae99SBarry Smith   ierr = PetscObjectGetOptionsPrefix((PetscObject)xin,&prefix);CHKERRQ(ierr);
597aa219208SBarry Smith   ierr = DMDACreateNaturalVector(da,&natural);CHKERRQ(ierr);
59847c6ae99SBarry Smith   ierr = PetscObjectSetName((PetscObject)natural,((PetscObject)xin)->name);CHKERRQ(ierr);
59947c6ae99SBarry Smith   ierr = PetscObjectSetOptionsPrefix((PetscObject)natural,prefix);CHKERRQ(ierr);
60047c6ae99SBarry Smith   ierr = VecLoad_Binary(natural,viewer);CHKERRQ(ierr);
601aa219208SBarry Smith   ierr = DMDANaturalToGlobalBegin(da,natural,INSERT_VALUES,xin);CHKERRQ(ierr);
602aa219208SBarry Smith   ierr = DMDANaturalToGlobalEnd(da,natural,INSERT_VALUES,xin);CHKERRQ(ierr);
603fcfd50ebSBarry Smith   ierr = VecDestroy(&natural);CHKERRQ(ierr);
604aa219208SBarry Smith   ierr = PetscInfo(xin,"Loading vector from natural ordering into DMDA\n");CHKERRQ(ierr);
60547c6ae99SBarry Smith   ierr = PetscOptionsGetInt(((PetscObject)xin)->prefix,"-vecload_block_size",&bs,&flag);CHKERRQ(ierr);
60647c6ae99SBarry Smith   if (flag && bs != dd->w) {
607aa219208SBarry Smith     ierr = PetscInfo2(xin,"Block size in file %D not equal to DMDA's dof %D\n",bs,dd->w);CHKERRQ(ierr);
60847c6ae99SBarry Smith   }
60947c6ae99SBarry Smith   PetscFunctionReturn(0);
61047c6ae99SBarry Smith }
61147c6ae99SBarry Smith 
61247c6ae99SBarry Smith EXTERN_C_BEGIN
61347c6ae99SBarry Smith #undef __FUNCT__
61447c6ae99SBarry Smith #define __FUNCT__ "VecLoad_Default_DA"
6157087cfbeSBarry Smith PetscErrorCode  VecLoad_Default_DA(Vec xin, PetscViewer viewer)
61647c6ae99SBarry Smith {
61747c6ae99SBarry Smith   PetscErrorCode ierr;
6189a42bb27SBarry Smith   DM             da;
61947c6ae99SBarry Smith   PetscBool      isbinary;
62047c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
62147c6ae99SBarry Smith   PetscBool      ishdf5;
62247c6ae99SBarry Smith #endif
62347c6ae99SBarry Smith 
62447c6ae99SBarry Smith   PetscFunctionBegin;
6253c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
626aa219208SBarry Smith   if (!da) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
62747c6ae99SBarry Smith 
62847c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
62947c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERHDF5,&ishdf5);CHKERRQ(ierr);
63047c6ae99SBarry Smith #endif
63147c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
63247c6ae99SBarry Smith 
63347c6ae99SBarry Smith   if (isbinary) {
63447c6ae99SBarry Smith     ierr = VecLoad_Binary_DA(xin,viewer);CHKERRQ(ierr);
63547c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
63647c6ae99SBarry Smith   } else if (ishdf5) {
63747c6ae99SBarry Smith     ierr = VecLoad_HDF5_DA(xin,viewer);CHKERRQ(ierr);
63847c6ae99SBarry Smith #endif
639d34fcf5fSBarry Smith   } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for vector loading", ((PetscObject)viewer)->type_name);
64047c6ae99SBarry Smith   PetscFunctionReturn(0);
64147c6ae99SBarry Smith }
64247c6ae99SBarry Smith EXTERN_C_END
643