xref: /petsc/src/dm/impls/da/gr2.c (revision bfec8eec0a0d4f04d4f33e2b82a3ad5fb0818d23)
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;
9020d0051dSBarry Smith   PetscInt           *displayfields;
9120d0051dSBarry Smith   PetscInt           ndisplayfields,i;
9220d0051dSBarry Smith   PetscBool          flg;
9347c6ae99SBarry Smith 
9447c6ae99SBarry Smith   PetscFunctionBegin;
9547c6ae99SBarry Smith   zctx.showgrid = PETSC_FALSE;
9647c6ae99SBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
9747c6ae99SBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
9847c6ae99SBarry Smith 
993c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
100aa219208SBarry Smith   if (!da) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
10147c6ae99SBarry Smith 
10247c6ae99SBarry Smith   ierr = PetscObjectGetComm((PetscObject)xin,&comm);CHKERRQ(ierr);
10347c6ae99SBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
10447c6ae99SBarry Smith 
1051321219cSEthan Coon   ierr = DMDAGetInfo(da,0,&M,&N,0,&zctx.m,&zctx.n,0,&w,&s,&bx,&by,0,&st);CHKERRQ(ierr);
106aa219208SBarry Smith   ierr = DMDAGetOwnershipRanges(da,&lx,&ly,PETSC_NULL);CHKERRQ(ierr);
10747c6ae99SBarry Smith 
10847c6ae99SBarry Smith   /*
10947c6ae99SBarry Smith         Obtain a sequential vector that is going to contain the local values plus ONE layer of
110aa219208SBarry Smith      ghosted values to draw the graphics from. We also need its corresponding DMDA (dac) that will
11147c6ae99SBarry Smith      update the local values pluse ONE layer of ghost values.
11247c6ae99SBarry Smith   */
11347c6ae99SBarry Smith   ierr = PetscObjectQuery((PetscObject)da,"GraphicsGhosted",(PetscObject*)&xlocal);CHKERRQ(ierr);
11447c6ae99SBarry Smith   if (!xlocal) {
1151321219cSEthan Coon     if (!bx || !by || s != 1 || st != DMDA_STENCIL_BOX) {
11647c6ae99SBarry Smith       /*
11747c6ae99SBarry Smith          if original da is not of stencil width one, or periodic or not a box stencil then
118aa219208SBarry Smith          create a special DMDA to handle one level of ghost points for graphics
11947c6ae99SBarry Smith       */
1201321219cSEthan 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);
121aa219208SBarry Smith       ierr = PetscInfo(da,"Creating auxilary DMDA for managing graphics ghost points\n");CHKERRQ(ierr);
12247c6ae99SBarry Smith     } else {
12347c6ae99SBarry Smith       /* otherwise we can use the da we already have */
12447c6ae99SBarry Smith       dac = da;
12547c6ae99SBarry Smith     }
12647c6ae99SBarry Smith     /* create local vector for holding ghosted values used in graphics */
127564755cdSBarry Smith     ierr = DMCreateLocalVector(dac,&xlocal);CHKERRQ(ierr);
12847c6ae99SBarry Smith     if (dac != da) {
129aa219208SBarry Smith       /* don't keep any public reference of this DMDA, is is only available through xlocal */
130f8631f9bSBarry Smith       DM dacd = dac;
131f8631f9bSBarry Smith       ierr = DMDestroy(&dacd);CHKERRQ(ierr);
13247c6ae99SBarry Smith     } else {
13347c6ae99SBarry Smith       /* remove association between xlocal and da, because below we compose in the opposite
13447c6ae99SBarry Smith          direction and if we left this connect we'd get a loop, so the objects could
13547c6ae99SBarry Smith          never be destroyed */
136*bfec8eecSBarry Smith       ierr = PetscObjectRemoveReference((PetscObject)xlocal,"DM");CHKERRQ(ierr);
13747c6ae99SBarry Smith     }
13847c6ae99SBarry Smith     ierr = PetscObjectCompose((PetscObject)da,"GraphicsGhosted",(PetscObject)xlocal);CHKERRQ(ierr);
13947c6ae99SBarry Smith     ierr = PetscObjectDereference((PetscObject)xlocal);CHKERRQ(ierr);
14047c6ae99SBarry Smith   } else {
1411321219cSEthan Coon     if (!bx && !by && s == 1 && st == DMDA_STENCIL_BOX) {
14247c6ae99SBarry Smith       dac = da;
14347c6ae99SBarry Smith     } else {
1443c0c59f3SBarry Smith       ierr = PetscObjectQuery((PetscObject)xlocal,"DM",(PetscObject*)&dac);CHKERRQ(ierr);
14547c6ae99SBarry Smith     }
14647c6ae99SBarry Smith   }
14747c6ae99SBarry Smith 
14847c6ae99SBarry Smith   /*
14947c6ae99SBarry Smith       Get local (ghosted) values of vector
15047c6ae99SBarry Smith   */
1519a42bb27SBarry Smith   ierr = DMGlobalToLocalBegin(dac,xin,INSERT_VALUES,xlocal);CHKERRQ(ierr);
1529a42bb27SBarry Smith   ierr = DMGlobalToLocalEnd(dac,xin,INSERT_VALUES,xlocal);CHKERRQ(ierr);
15347c6ae99SBarry Smith   ierr = VecGetArray(xlocal,&zctx.v);CHKERRQ(ierr);
15447c6ae99SBarry Smith 
15547c6ae99SBarry Smith   /* get coordinates of nodes */
156aa219208SBarry Smith   ierr = DMDAGetCoordinates(da,&xcoor);CHKERRQ(ierr);
15747c6ae99SBarry Smith   if (!xcoor) {
158aa219208SBarry Smith     ierr = DMDASetUniformCoordinates(da,0.0,1.0,0.0,1.0,0.0,0.0);CHKERRQ(ierr);
159aa219208SBarry Smith     ierr = DMDAGetCoordinates(da,&xcoor);CHKERRQ(ierr);
16047c6ae99SBarry Smith   }
16147c6ae99SBarry Smith 
16247c6ae99SBarry Smith   /*
16347c6ae99SBarry Smith       Determine the min and max  coordinates in plot
16447c6ae99SBarry Smith   */
16547c6ae99SBarry Smith   ierr = VecStrideMin(xcoor,0,PETSC_NULL,&xmin);CHKERRQ(ierr);
16647c6ae99SBarry Smith   ierr = VecStrideMax(xcoor,0,PETSC_NULL,&xmax);CHKERRQ(ierr);
16747c6ae99SBarry Smith   ierr = VecStrideMin(xcoor,1,PETSC_NULL,&ymin);CHKERRQ(ierr);
16847c6ae99SBarry Smith   ierr = VecStrideMax(xcoor,1,PETSC_NULL,&ymax);CHKERRQ(ierr);
16947c6ae99SBarry Smith   coors[0] = xmin - .05*(xmax- xmin); coors[2] = xmax + .05*(xmax - xmin);
17047c6ae99SBarry Smith   coors[1] = ymin - .05*(ymax- ymin); coors[3] = ymax + .05*(ymax - ymin);
171aa219208SBarry 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);
17247c6ae99SBarry Smith 
17347c6ae99SBarry Smith   /*
17447c6ae99SBarry Smith        get local ghosted version of coordinates
17547c6ae99SBarry Smith   */
17647c6ae99SBarry Smith   ierr = PetscObjectQuery((PetscObject)da,"GraphicsCoordinateGhosted",(PetscObject*)&xcoorl);CHKERRQ(ierr);
17747c6ae99SBarry Smith   if (!xcoorl) {
178f8631f9bSBarry Smith     DM dagd;
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);
184f8631f9bSBarry Smith     dagd = dag;
185f8631f9bSBarry Smith     ierr = DMDestroy(&dagd);CHKERRQ(ierr);/* dereference dag */
18647c6ae99SBarry Smith     ierr = PetscObjectDereference((PetscObject)xcoorl);CHKERRQ(ierr);
18747c6ae99SBarry Smith   } else {
1883c0c59f3SBarry Smith     ierr = PetscObjectQuery((PetscObject)xcoorl,"DM",(PetscObject*)&dag);CHKERRQ(ierr);
18947c6ae99SBarry Smith   }
1909a42bb27SBarry Smith   ierr = DMGlobalToLocalBegin(dag,xcoor,INSERT_VALUES,xcoorl);CHKERRQ(ierr);
1919a42bb27SBarry Smith   ierr = DMGlobalToLocalEnd(dag,xcoor,INSERT_VALUES,xcoorl);CHKERRQ(ierr);
19247c6ae99SBarry Smith   ierr = VecGetArray(xcoorl,&zctx.xy);CHKERRQ(ierr);
19347c6ae99SBarry Smith 
19447c6ae99SBarry Smith   /*
19547c6ae99SBarry Smith         Get information about size of area each processor must do graphics for
19647c6ae99SBarry Smith   */
1971321219cSEthan Coon   ierr = DMDAGetInfo(dac,0,&M,&N,0,0,0,0,&zctx.step,0,&bx,&by,0,0);CHKERRQ(ierr);
198aa219208SBarry Smith   ierr = DMDAGetGhostCorners(dac,&igstart,&jgstart,0,&zctx.m,&zctx.n,0);CHKERRQ(ierr);
199aa219208SBarry Smith   ierr = DMDAGetCorners(dac,&istart,0,0,&isize,0,0);CHKERRQ(ierr);
20047c6ae99SBarry Smith 
201671f6225SBarry Smith   ierr = PetscOptionsGetBool(PETSC_NULL,"-draw_contour_grid",&zctx.showgrid,PETSC_NULL);CHKERRQ(ierr);
20220d0051dSBarry Smith   ierr = PetscMalloc(zctx.step*sizeof(PetscInt),&displayfields);CHKERRQ(ierr);
20320d0051dSBarry Smith   for (i=0; i<zctx.step; i++) displayfields[i] = i;
20420d0051dSBarry Smith   ndisplayfields = zctx.step;
20520d0051dSBarry Smith   ierr = PetscOptionsGetIntArray(PETSC_NULL,"-draw_fields",displayfields,&ndisplayfields,&flg);CHKERRQ(ierr);
20620d0051dSBarry Smith   if (!flg) ndisplayfields = zctx.step;
20747c6ae99SBarry Smith 
20847c6ae99SBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
209671f6225SBarry Smith   ierr = PetscOptionsGetBool(PETSC_NULL,"-draw_ports",&useports,PETSC_NULL);CHKERRQ(ierr);
21047c6ae99SBarry Smith   if (useports || format == PETSC_VIEWER_DRAW_PORTS){
21147c6ae99SBarry Smith     ierr = PetscDrawSynchronizedClear(draw);CHKERRQ(ierr);
21220d0051dSBarry Smith     ierr = PetscDrawViewPortsCreate(draw,ndisplayfields,&ports);CHKERRQ(ierr);
21347c6ae99SBarry Smith   }
21420d0051dSBarry Smith 
21547c6ae99SBarry Smith   /*
21647c6ae99SBarry Smith      Loop over each field; drawing each in a different window
21747c6ae99SBarry Smith   */
21820d0051dSBarry Smith   for (i=0; i<ndisplayfields; i++) {
21920d0051dSBarry Smith     zctx.k = displayfields[i];
22047c6ae99SBarry Smith     if (useports) {
22120d0051dSBarry Smith       ierr = PetscDrawViewPortsSet(ports,i);CHKERRQ(ierr);
22247c6ae99SBarry Smith     } else {
22320d0051dSBarry Smith       ierr = PetscViewerDrawGetDraw(viewer,i,&draw);CHKERRQ(ierr);
22447c6ae99SBarry Smith       ierr = PetscDrawSynchronizedClear(draw);CHKERRQ(ierr);
22547c6ae99SBarry Smith     }
22647c6ae99SBarry Smith 
22747c6ae99SBarry Smith     /*
22847c6ae99SBarry Smith         Determine the min and max color in plot
22947c6ae99SBarry Smith     */
23047c6ae99SBarry Smith     ierr = VecStrideMin(xin,zctx.k,PETSC_NULL,&zctx.min);CHKERRQ(ierr);
23147c6ae99SBarry Smith     ierr = VecStrideMax(xin,zctx.k,PETSC_NULL,&zctx.max);CHKERRQ(ierr);
23247c6ae99SBarry Smith     if (zctx.min == zctx.max) {
23347c6ae99SBarry Smith       zctx.min -= 1.e-12;
23447c6ae99SBarry Smith       zctx.max += 1.e-12;
23547c6ae99SBarry Smith     }
23647c6ae99SBarry Smith 
23747c6ae99SBarry Smith     if (!rank) {
23847c6ae99SBarry Smith       const char *title;
23947c6ae99SBarry Smith 
240aa219208SBarry Smith       ierr = DMDAGetFieldName(da,zctx.k,&title);CHKERRQ(ierr);
24147c6ae99SBarry Smith       if (title) {
24247c6ae99SBarry Smith         ierr = PetscDrawSetTitle(draw,title);CHKERRQ(ierr);
24347c6ae99SBarry Smith       }
24447c6ae99SBarry Smith     }
24547c6ae99SBarry Smith     ierr = PetscDrawSetCoordinates(draw,coors[0],coors[1],coors[2],coors[3]);CHKERRQ(ierr);
246aa219208SBarry Smith     ierr = PetscInfo2(da,"DMDA 2d contour plot min %G max %G\n",zctx.min,zctx.max);CHKERRQ(ierr);
24747c6ae99SBarry Smith 
24847c6ae99SBarry Smith     ierr = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr);
24947c6ae99SBarry Smith     if (popup) {ierr = PetscDrawScalePopup(popup,zctx.min,zctx.max);CHKERRQ(ierr);}
25047c6ae99SBarry Smith 
25147c6ae99SBarry Smith     zctx.scale = (245.0 - PETSC_DRAW_BASIC_COLORS)/(zctx.max - zctx.min);
25247c6ae99SBarry Smith 
25347c6ae99SBarry Smith     ierr = PetscDrawZoom(draw,VecView_MPI_Draw_DA2d_Zoom,&zctx);CHKERRQ(ierr);
25447c6ae99SBarry Smith   }
25520d0051dSBarry Smith   ierr = PetscFree(displayfields);CHKERRQ(ierr);
2566bf464f9SBarry Smith   ierr = PetscDrawViewPortsDestroy(ports);CHKERRQ(ierr);
25747c6ae99SBarry Smith 
25847c6ae99SBarry Smith   ierr = VecRestoreArray(xcoorl,&zctx.xy);CHKERRQ(ierr);
25947c6ae99SBarry Smith   ierr = VecRestoreArray(xlocal,&zctx.v);CHKERRQ(ierr);
26047c6ae99SBarry Smith   PetscFunctionReturn(0);
26147c6ae99SBarry Smith }
26247c6ae99SBarry Smith 
26347c6ae99SBarry Smith 
26447c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
26547c6ae99SBarry Smith #undef __FUNCT__
26647c6ae99SBarry Smith #define __FUNCT__ "VecView_MPI_HDF5_DA"
26747c6ae99SBarry Smith PetscErrorCode VecView_MPI_HDF5_DA(Vec xin,PetscViewer viewer)
26847c6ae99SBarry Smith {
26947c6ae99SBarry Smith   PetscErrorCode ierr;
2709b2a5a86SJed Brown   DM             dm;
2719b2a5a86SJed Brown   DM_DA          *da;
27247c6ae99SBarry Smith   hsize_t        dim,dims[5];
27347c6ae99SBarry Smith   hsize_t        count[5];
27447c6ae99SBarry Smith   hsize_t        offset[5];
27547c6ae99SBarry Smith   PetscInt       cnt = 0;
27647c6ae99SBarry Smith   PetscScalar    *x;
27747c6ae99SBarry Smith   const char     *vecname;
27847c6ae99SBarry Smith   hid_t          filespace; /* file dataspace identifier */
27947c6ae99SBarry Smith   hid_t	         plist_id;  /* property list identifier */
28047c6ae99SBarry Smith   hid_t          dset_id;   /* dataset identifier */
28147c6ae99SBarry Smith   hid_t          memspace;  /* memory dataspace identifier */
28247c6ae99SBarry Smith   hid_t          file_id;
28347c6ae99SBarry Smith   herr_t         status;
28447c6ae99SBarry Smith 
28547c6ae99SBarry Smith   PetscFunctionBegin;
28647c6ae99SBarry Smith   ierr = PetscViewerHDF5GetFileId(viewer, &file_id);CHKERRQ(ierr);
2873c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&dm);CHKERRQ(ierr);
2889b2a5a86SJed Brown   if (!dm) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
2899b2a5a86SJed Brown   da = (DM_DA*)dm->data;
29047c6ae99SBarry Smith 
29147c6ae99SBarry Smith   /* Create the dataspace for the dataset */
29247c6ae99SBarry Smith   dim       = PetscHDF5IntCast(da->dim + ((da->w == 1) ? 0 : 1));
29347c6ae99SBarry Smith   if (da->dim == 3) dims[cnt++]   = PetscHDF5IntCast(da->P);
29447c6ae99SBarry Smith   if (da->dim > 1)  dims[cnt++]   = PetscHDF5IntCast(da->N);
29547c6ae99SBarry Smith   dims[cnt++]   = PetscHDF5IntCast(da->M);
29647c6ae99SBarry Smith   if (da->w > 1) dims[cnt++] = PetscHDF5IntCast(da->w);
29747c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
29847c6ae99SBarry Smith   dim++;
29947c6ae99SBarry Smith   dims[cnt++] = 2;
30047c6ae99SBarry Smith #endif
30147c6ae99SBarry Smith   filespace = H5Screate_simple(dim, dims, NULL);
30247c6ae99SBarry Smith   if (filespace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Screate_simple()");
30347c6ae99SBarry Smith 
30447c6ae99SBarry Smith   /* Create the dataset with default properties and close filespace */
30547c6ae99SBarry Smith   ierr = PetscObjectGetName((PetscObject)xin,&vecname);CHKERRQ(ierr);
30647c6ae99SBarry Smith #if (H5_VERS_MAJOR * 10000 + H5_VERS_MINOR * 100 + H5_VERS_RELEASE >= 10800)
30747c6ae99SBarry Smith   dset_id = H5Dcreate2(file_id, vecname, H5T_NATIVE_DOUBLE, filespace, H5P_DEFAULT, H5P_DEFAULT, H5P_DEFAULT);
30847c6ae99SBarry Smith #else
30947c6ae99SBarry Smith   dset_id = H5Dcreate(file_id, vecname, H5T_NATIVE_DOUBLE, filespace, H5P_DEFAULT);
31047c6ae99SBarry Smith #endif
31147c6ae99SBarry Smith   if (dset_id == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Dcreate2()");
31247c6ae99SBarry Smith   status = H5Sclose(filespace);CHKERRQ(status);
31347c6ae99SBarry Smith 
31447c6ae99SBarry Smith   /* Each process defines a dataset and writes it to the hyperslab in the file */
31547c6ae99SBarry Smith   cnt = 0;
31647c6ae99SBarry Smith   if (da->dim == 3) offset[cnt++] = PetscHDF5IntCast(da->zs);
31747c6ae99SBarry Smith   if (da->dim > 1)  offset[cnt++] = PetscHDF5IntCast(da->ys);
31847c6ae99SBarry Smith   offset[cnt++] = PetscHDF5IntCast(da->xs/da->w);
31947c6ae99SBarry Smith   if (da->w > 1) offset[cnt++] = 0;
32047c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
32147c6ae99SBarry Smith   offset[cnt++] = 0;
32247c6ae99SBarry Smith #endif
32347c6ae99SBarry Smith   cnt = 0;
32447c6ae99SBarry Smith   if (da->dim == 3) count[cnt++] = PetscHDF5IntCast(da->ze - da->zs);
32547c6ae99SBarry Smith   if (da->dim > 1)  count[cnt++] = PetscHDF5IntCast(da->ye - da->ys);
32647c6ae99SBarry Smith   count[cnt++] = PetscHDF5IntCast((da->xe - da->xs)/da->w);
32747c6ae99SBarry Smith   if (da->w > 1) count[cnt++] = PetscHDF5IntCast(da->w);
32847c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
32947c6ae99SBarry Smith   count[cnt++] = 2;
33047c6ae99SBarry Smith #endif
33147c6ae99SBarry Smith   memspace = H5Screate_simple(dim, count, NULL);
33247c6ae99SBarry Smith   if (memspace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Screate_simple()");
33347c6ae99SBarry Smith 
33447c6ae99SBarry Smith 
33547c6ae99SBarry Smith   filespace = H5Dget_space(dset_id);
33647c6ae99SBarry Smith   if (filespace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Dget_space()");
33747c6ae99SBarry Smith   status = H5Sselect_hyperslab(filespace, H5S_SELECT_SET, offset, NULL, count, NULL);CHKERRQ(status);
33847c6ae99SBarry Smith 
33947c6ae99SBarry Smith   /* Create property list for collective dataset write */
34047c6ae99SBarry Smith   plist_id = H5Pcreate(H5P_DATASET_XFER);
34147c6ae99SBarry Smith   if (plist_id == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Pcreate()");
34247c6ae99SBarry Smith #if defined(PETSC_HAVE_H5PSET_FAPL_MPIO)
34347c6ae99SBarry Smith   status = H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_COLLECTIVE);CHKERRQ(status);
34447c6ae99SBarry Smith #endif
34547c6ae99SBarry Smith   /* To write dataset independently use H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_INDEPENDENT) */
34647c6ae99SBarry Smith 
34747c6ae99SBarry Smith   ierr = VecGetArray(xin, &x);CHKERRQ(ierr);
34847c6ae99SBarry Smith   status = H5Dwrite(dset_id, H5T_NATIVE_DOUBLE, memspace, filespace, plist_id, x);CHKERRQ(status);
34947c6ae99SBarry Smith   status = H5Fflush(file_id, H5F_SCOPE_GLOBAL);CHKERRQ(status);
35047c6ae99SBarry Smith   ierr = VecRestoreArray(xin, &x);CHKERRQ(ierr);
35147c6ae99SBarry Smith 
35247c6ae99SBarry Smith   /* Close/release resources */
35347c6ae99SBarry Smith   status = H5Pclose(plist_id);CHKERRQ(status);
35447c6ae99SBarry Smith   status = H5Sclose(filespace);CHKERRQ(status);
35547c6ae99SBarry Smith   status = H5Sclose(memspace);CHKERRQ(status);
35647c6ae99SBarry Smith   status = H5Dclose(dset_id);CHKERRQ(status);
35747c6ae99SBarry Smith   ierr = PetscInfo1(xin,"Wrote Vec object with name %s\n",vecname);CHKERRQ(ierr);
35847c6ae99SBarry Smith   PetscFunctionReturn(0);
35947c6ae99SBarry Smith }
36047c6ae99SBarry Smith #endif
36147c6ae99SBarry Smith 
36209573ac7SBarry Smith extern PetscErrorCode VecView_MPI_Draw_DA1d(Vec,PetscViewer);
36347c6ae99SBarry Smith 
36447c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
36547c6ae99SBarry Smith #undef __FUNCT__
366aa219208SBarry Smith #define __FUNCT__ "DMDAArrayMPIIO"
367aa219208SBarry Smith static PetscErrorCode DMDAArrayMPIIO(DM da,PetscViewer viewer,Vec xin,PetscBool  write)
36847c6ae99SBarry Smith {
36947c6ae99SBarry Smith   PetscErrorCode    ierr;
37047c6ae99SBarry Smith   MPI_File          mfdes;
37147c6ae99SBarry Smith   PetscMPIInt       gsizes[4],lsizes[4],lstarts[4],asiz,dof;
37247c6ae99SBarry Smith   MPI_Datatype      view;
37347c6ae99SBarry Smith   const PetscScalar *array;
37447c6ae99SBarry Smith   MPI_Offset        off;
37547c6ae99SBarry Smith   MPI_Aint          ub,ul;
37647c6ae99SBarry Smith   PetscInt          type,rows,vecrows,tr[2];
37747c6ae99SBarry Smith   DM_DA             *dd = (DM_DA*)da->data;
37847c6ae99SBarry Smith 
37947c6ae99SBarry Smith   PetscFunctionBegin;
38047c6ae99SBarry Smith   ierr = VecGetSize(xin,&vecrows);CHKERRQ(ierr);
38147c6ae99SBarry Smith   if (!write) {
38247c6ae99SBarry Smith     /* Read vector header. */
38347c6ae99SBarry Smith     ierr = PetscViewerBinaryRead(viewer,tr,2,PETSC_INT);CHKERRQ(ierr);
38447c6ae99SBarry Smith     type = tr[0];
38547c6ae99SBarry Smith     rows = tr[1];
38647c6ae99SBarry Smith     if (type != VEC_FILE_CLASSID) SETERRQ(((PetscObject)da)->comm,PETSC_ERR_ARG_WRONG,"Not vector next in file");
387aa219208SBarry Smith     if (rows != vecrows) SETERRQ(((PetscObject)da)->comm,PETSC_ERR_ARG_SIZ,"Vector in file not same size as DMDA vector");
38847c6ae99SBarry Smith   } else {
38947c6ae99SBarry Smith     tr[0] = VEC_FILE_CLASSID;
39047c6ae99SBarry Smith     tr[1] = vecrows;
39147c6ae99SBarry Smith     ierr = PetscViewerBinaryWrite(viewer,tr,2,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
39247c6ae99SBarry Smith   }
39347c6ae99SBarry Smith 
39447c6ae99SBarry Smith   dof = PetscMPIIntCast(dd->w);
39547c6ae99SBarry Smith   gsizes[0]  = dof; gsizes[1] = PetscMPIIntCast(dd->M); gsizes[2] = PetscMPIIntCast(dd->N); gsizes[3] = PetscMPIIntCast(dd->P);
39647c6ae99SBarry 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);
39747c6ae99SBarry Smith   lstarts[0] = 0;  lstarts[1] = PetscMPIIntCast(dd->xs/dof); lstarts[2] = PetscMPIIntCast(dd->ys); lstarts[3] = PetscMPIIntCast(dd->zs);
39847c6ae99SBarry Smith   ierr = MPI_Type_create_subarray(dd->dim+1,gsizes,lsizes,lstarts,MPI_ORDER_FORTRAN,MPIU_SCALAR,&view);CHKERRQ(ierr);
39947c6ae99SBarry Smith   ierr = MPI_Type_commit(&view);CHKERRQ(ierr);
40047c6ae99SBarry Smith 
40147c6ae99SBarry Smith   ierr = PetscViewerBinaryGetMPIIODescriptor(viewer,&mfdes);CHKERRQ(ierr);
40247c6ae99SBarry Smith   ierr = PetscViewerBinaryGetMPIIOOffset(viewer,&off);CHKERRQ(ierr);
40347c6ae99SBarry Smith   ierr = MPI_File_set_view(mfdes,off,MPIU_SCALAR,view,(char *)"native",MPI_INFO_NULL);CHKERRQ(ierr);
40447c6ae99SBarry Smith   ierr = VecGetArrayRead(xin,&array);CHKERRQ(ierr);
40547c6ae99SBarry Smith   asiz = lsizes[1]*(lsizes[2] > 0 ? lsizes[2] : 1)*(lsizes[3] > 0 ? lsizes[3] : 1)*dof;
40647c6ae99SBarry Smith   if (write) {
40747c6ae99SBarry Smith     ierr = MPIU_File_write_all(mfdes,(PetscScalar*)array,asiz,MPIU_SCALAR,MPI_STATUS_IGNORE);CHKERRQ(ierr);
40847c6ae99SBarry Smith   } else {
40947c6ae99SBarry Smith     ierr = MPIU_File_read_all(mfdes,(PetscScalar*)array,asiz,MPIU_SCALAR,MPI_STATUS_IGNORE);CHKERRQ(ierr);
41047c6ae99SBarry Smith   }
41147c6ae99SBarry Smith   ierr = MPI_Type_get_extent(view,&ul,&ub);CHKERRQ(ierr);
41247c6ae99SBarry Smith   ierr = PetscViewerBinaryAddMPIIOOffset(viewer,ub);CHKERRQ(ierr);
41347c6ae99SBarry Smith   ierr = VecRestoreArrayRead(xin,&array);CHKERRQ(ierr);
41447c6ae99SBarry Smith   ierr = MPI_Type_free(&view);CHKERRQ(ierr);
41547c6ae99SBarry Smith   PetscFunctionReturn(0);
41647c6ae99SBarry Smith }
41747c6ae99SBarry Smith #endif
41847c6ae99SBarry Smith 
41947c6ae99SBarry Smith EXTERN_C_BEGIN
42047c6ae99SBarry Smith #undef __FUNCT__
42147c6ae99SBarry Smith #define __FUNCT__ "VecView_MPI_DA"
4227087cfbeSBarry Smith PetscErrorCode  VecView_MPI_DA(Vec xin,PetscViewer viewer)
42347c6ae99SBarry Smith {
4249a42bb27SBarry Smith   DM             da;
42547c6ae99SBarry Smith   PetscErrorCode ierr;
42647c6ae99SBarry Smith   PetscInt       dim;
42747c6ae99SBarry Smith   Vec            natural;
42847c6ae99SBarry Smith   PetscBool      isdraw;
42947c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
43047c6ae99SBarry Smith   PetscBool      ishdf5;
43147c6ae99SBarry Smith #endif
4323f3fd955SJed Brown   const char     *prefix,*name;
43347c6ae99SBarry Smith 
43447c6ae99SBarry Smith   PetscFunctionBegin;
4353c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
436aa219208SBarry Smith   if (!da) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
43747c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
43847c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
43947c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERHDF5,&ishdf5);CHKERRQ(ierr);
44047c6ae99SBarry Smith #endif
44147c6ae99SBarry Smith   if (isdraw) {
4421321219cSEthan Coon     ierr = DMDAGetInfo(da,&dim,0,0,0,0,0,0,0,0,0,0,0,0);CHKERRQ(ierr);
44347c6ae99SBarry Smith     if (dim == 1) {
44447c6ae99SBarry Smith       ierr = VecView_MPI_Draw_DA1d(xin,viewer);CHKERRQ(ierr);
44547c6ae99SBarry Smith     } else if (dim == 2) {
44647c6ae99SBarry Smith       ierr = VecView_MPI_Draw_DA2d(xin,viewer);CHKERRQ(ierr);
44747c6ae99SBarry Smith     } else {
448aa219208SBarry Smith       SETERRQ1(((PetscObject)da)->comm,PETSC_ERR_SUP,"Cannot graphically view vector associated with this dimensional DMDA %D",dim);
44947c6ae99SBarry Smith     }
45047c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
45147c6ae99SBarry Smith   } else if (ishdf5) {
45247c6ae99SBarry Smith     ierr = VecView_MPI_HDF5_DA(xin,viewer);CHKERRQ(ierr);
45347c6ae99SBarry Smith #endif
45447c6ae99SBarry Smith   } else {
45547c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
45647c6ae99SBarry Smith     PetscBool  isbinary,isMPIIO;
45747c6ae99SBarry Smith 
45847c6ae99SBarry Smith     ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
45947c6ae99SBarry Smith     if (isbinary) {
46047c6ae99SBarry Smith       ierr = PetscViewerBinaryGetMPIIO(viewer,&isMPIIO);CHKERRQ(ierr);
46147c6ae99SBarry Smith       if (isMPIIO) {
462aa219208SBarry Smith        ierr = DMDAArrayMPIIO(da,viewer,xin,PETSC_TRUE);CHKERRQ(ierr);
46347c6ae99SBarry Smith        PetscFunctionReturn(0);
46447c6ae99SBarry Smith       }
46547c6ae99SBarry Smith     }
46647c6ae99SBarry Smith #endif
46747c6ae99SBarry Smith 
46847c6ae99SBarry Smith     /* call viewer on natural ordering */
46947c6ae99SBarry Smith     ierr = PetscObjectGetOptionsPrefix((PetscObject)xin,&prefix);CHKERRQ(ierr);
470aa219208SBarry Smith     ierr = DMDACreateNaturalVector(da,&natural);CHKERRQ(ierr);
47147c6ae99SBarry Smith     ierr = PetscObjectSetOptionsPrefix((PetscObject)natural,prefix);CHKERRQ(ierr);
472aa219208SBarry Smith     ierr = DMDAGlobalToNaturalBegin(da,xin,INSERT_VALUES,natural);CHKERRQ(ierr);
473aa219208SBarry Smith     ierr = DMDAGlobalToNaturalEnd(da,xin,INSERT_VALUES,natural);CHKERRQ(ierr);
4743f3fd955SJed Brown     ierr = PetscObjectGetName((PetscObject)xin,&name);CHKERRQ(ierr);
4753f3fd955SJed Brown     ierr = PetscObjectSetName((PetscObject)natural,name);CHKERRQ(ierr);
47647c6ae99SBarry Smith     ierr = VecView(natural,viewer);CHKERRQ(ierr);
477fcfd50ebSBarry Smith     ierr = VecDestroy(&natural);CHKERRQ(ierr);
47847c6ae99SBarry Smith   }
47947c6ae99SBarry Smith   PetscFunctionReturn(0);
48047c6ae99SBarry Smith }
48147c6ae99SBarry Smith EXTERN_C_END
48247c6ae99SBarry Smith 
48347c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
48447c6ae99SBarry Smith #undef __FUNCT__
48547c6ae99SBarry Smith #define __FUNCT__ "VecLoad_HDF5_DA"
48647c6ae99SBarry Smith PetscErrorCode VecLoad_HDF5_DA(Vec xin, PetscViewer viewer)
48747c6ae99SBarry Smith {
4889a42bb27SBarry Smith   DM             da;
48947c6ae99SBarry Smith   PetscErrorCode ierr;
49025578ef6SJed Brown   hsize_t        dim;
49147c6ae99SBarry Smith   hsize_t        count[5];
49247c6ae99SBarry Smith   hsize_t        offset[5];
49347c6ae99SBarry Smith   PetscInt       cnt = 0;
49447c6ae99SBarry Smith   PetscScalar    *x;
49547c6ae99SBarry Smith   const char     *vecname;
49647c6ae99SBarry Smith   hid_t          filespace; /* file dataspace identifier */
49747c6ae99SBarry Smith   hid_t	         plist_id;  /* property list identifier */
49847c6ae99SBarry Smith   hid_t          dset_id;   /* dataset identifier */
49947c6ae99SBarry Smith   hid_t          memspace;  /* memory dataspace identifier */
50047c6ae99SBarry Smith   hid_t          file_id;
50147c6ae99SBarry Smith   herr_t         status;
5029c7c4993SBarry Smith   DM_DA          *dd;
50347c6ae99SBarry Smith 
50447c6ae99SBarry Smith   PetscFunctionBegin;
50547c6ae99SBarry Smith   ierr = PetscViewerHDF5GetFileId(viewer, &file_id);CHKERRQ(ierr);
5063c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
5079c7c4993SBarry Smith   dd = (DM_DA*)da->data;
50847c6ae99SBarry Smith 
50947c6ae99SBarry Smith   /* Create the dataspace for the dataset */
51047c6ae99SBarry Smith   dim       = PetscHDF5IntCast(dd->dim + ((dd->w == 1) ? 0 : 1));
51147c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
51247c6ae99SBarry Smith   dim++;
51347c6ae99SBarry Smith #endif
51447c6ae99SBarry Smith 
51547c6ae99SBarry Smith   /* Create the dataset with default properties and close filespace */
51647c6ae99SBarry Smith   ierr = PetscObjectGetName((PetscObject)xin,&vecname);CHKERRQ(ierr);
51747c6ae99SBarry Smith #if (H5_VERS_MAJOR * 10000 + H5_VERS_MINOR * 100 + H5_VERS_RELEASE >= 10800)
51847c6ae99SBarry Smith   dset_id = H5Dopen2(file_id, vecname, H5P_DEFAULT);
51947c6ae99SBarry Smith #else
52047c6ae99SBarry Smith   dset_id = H5Dopen(file_id, vecname);
52147c6ae99SBarry Smith #endif
52247c6ae99SBarry Smith   if (dset_id == -1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Dopen2() with Vec named %s",vecname);
52347c6ae99SBarry Smith   filespace = H5Dget_space(dset_id);
52447c6ae99SBarry Smith 
52547c6ae99SBarry Smith   /* Each process defines a dataset and reads it from the hyperslab in the file */
52647c6ae99SBarry Smith   cnt = 0;
52747c6ae99SBarry Smith   if (dd->dim == 3) offset[cnt++] = PetscHDF5IntCast(dd->zs);
52847c6ae99SBarry Smith   if (dd->dim > 1)  offset[cnt++] = PetscHDF5IntCast(dd->ys);
52947c6ae99SBarry Smith   offset[cnt++] = PetscHDF5IntCast(dd->xs/dd->w);
53047c6ae99SBarry Smith   if (dd->w > 1) offset[cnt++] = 0;
53147c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
53247c6ae99SBarry Smith   offset[cnt++] = 0;
53347c6ae99SBarry Smith #endif
53447c6ae99SBarry Smith   cnt = 0;
53547c6ae99SBarry Smith   if (dd->dim == 3) count[cnt++] = PetscHDF5IntCast(dd->ze - dd->zs);
53647c6ae99SBarry Smith   if (dd->dim > 1)  count[cnt++] = PetscHDF5IntCast(dd->ye - dd->ys);
53747c6ae99SBarry Smith   count[cnt++] = PetscHDF5IntCast((dd->xe - dd->xs)/dd->w);
53847c6ae99SBarry Smith   if (dd->w > 1) count[cnt++] = PetscHDF5IntCast(dd->w);
53947c6ae99SBarry Smith #if defined(PETSC_USE_COMPLEX)
54047c6ae99SBarry Smith   count[cnt++] = 2;
54147c6ae99SBarry Smith #endif
54247c6ae99SBarry Smith   memspace = H5Screate_simple(dim, count, NULL);
54347c6ae99SBarry Smith   if (memspace == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Screate_simple()");
54447c6ae99SBarry Smith 
54547c6ae99SBarry Smith   status = H5Sselect_hyperslab(filespace, H5S_SELECT_SET, offset, NULL, count, NULL);CHKERRQ(status);
54647c6ae99SBarry Smith 
54747c6ae99SBarry Smith   /* Create property list for collective dataset write */
54847c6ae99SBarry Smith   plist_id = H5Pcreate(H5P_DATASET_XFER);
54947c6ae99SBarry Smith   if (plist_id == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Cannot H5Pcreate()");
55047c6ae99SBarry Smith #if defined(PETSC_HAVE_H5PSET_FAPL_MPIO)
55147c6ae99SBarry Smith   status = H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_COLLECTIVE);CHKERRQ(status);
55247c6ae99SBarry Smith #endif
55347c6ae99SBarry Smith   /* To write dataset independently use H5Pset_dxpl_mpio(plist_id, H5FD_MPIO_INDEPENDENT) */
55447c6ae99SBarry Smith 
55547c6ae99SBarry Smith   ierr = VecGetArray(xin, &x);CHKERRQ(ierr);
55647c6ae99SBarry Smith   status = H5Dread(dset_id, H5T_NATIVE_DOUBLE, memspace, filespace, plist_id, x);CHKERRQ(status);
55747c6ae99SBarry Smith   ierr = VecRestoreArray(xin, &x);CHKERRQ(ierr);
55847c6ae99SBarry Smith 
55947c6ae99SBarry Smith   /* Close/release resources */
56047c6ae99SBarry Smith   status = H5Pclose(plist_id);CHKERRQ(status);
56147c6ae99SBarry Smith   status = H5Sclose(filespace);CHKERRQ(status);
56247c6ae99SBarry Smith   status = H5Sclose(memspace);CHKERRQ(status);
56347c6ae99SBarry Smith   status = H5Dclose(dset_id);CHKERRQ(status);
56447c6ae99SBarry Smith   PetscFunctionReturn(0);
56547c6ae99SBarry Smith }
56647c6ae99SBarry Smith #endif
56747c6ae99SBarry Smith 
56847c6ae99SBarry Smith #undef __FUNCT__
56947c6ae99SBarry Smith #define __FUNCT__ "VecLoad_Binary_DA"
57047c6ae99SBarry Smith PetscErrorCode VecLoad_Binary_DA(Vec xin, PetscViewer viewer)
57147c6ae99SBarry Smith {
5729a42bb27SBarry Smith   DM             da;
57347c6ae99SBarry Smith   PetscErrorCode ierr;
57447c6ae99SBarry Smith   Vec            natural;
57547c6ae99SBarry Smith   const char     *prefix;
57647c6ae99SBarry Smith   PetscInt       bs;
57747c6ae99SBarry Smith   PetscBool      flag;
57847c6ae99SBarry Smith   DM_DA          *dd;
57947c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
58047c6ae99SBarry Smith   PetscBool      isMPIIO;
58147c6ae99SBarry Smith #endif
58247c6ae99SBarry Smith 
58347c6ae99SBarry Smith   PetscFunctionBegin;
5843c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
58547c6ae99SBarry Smith   dd   = (DM_DA*)da->data;
58647c6ae99SBarry Smith #if defined(PETSC_HAVE_MPIIO)
58747c6ae99SBarry Smith   ierr = PetscViewerBinaryGetMPIIO(viewer,&isMPIIO);CHKERRQ(ierr);
58847c6ae99SBarry Smith   if (isMPIIO) {
589aa219208SBarry Smith     ierr = DMDAArrayMPIIO(da,viewer,xin,PETSC_FALSE);CHKERRQ(ierr);
59047c6ae99SBarry Smith     PetscFunctionReturn(0);
59147c6ae99SBarry Smith   }
59247c6ae99SBarry Smith #endif
59347c6ae99SBarry Smith 
59447c6ae99SBarry Smith   ierr = PetscObjectGetOptionsPrefix((PetscObject)xin,&prefix);CHKERRQ(ierr);
595aa219208SBarry Smith   ierr = DMDACreateNaturalVector(da,&natural);CHKERRQ(ierr);
59647c6ae99SBarry Smith   ierr = PetscObjectSetName((PetscObject)natural,((PetscObject)xin)->name);CHKERRQ(ierr);
59747c6ae99SBarry Smith   ierr = PetscObjectSetOptionsPrefix((PetscObject)natural,prefix);CHKERRQ(ierr);
59847c6ae99SBarry Smith   ierr = VecLoad_Binary(natural,viewer);CHKERRQ(ierr);
599aa219208SBarry Smith   ierr = DMDANaturalToGlobalBegin(da,natural,INSERT_VALUES,xin);CHKERRQ(ierr);
600aa219208SBarry Smith   ierr = DMDANaturalToGlobalEnd(da,natural,INSERT_VALUES,xin);CHKERRQ(ierr);
601fcfd50ebSBarry Smith   ierr = VecDestroy(&natural);CHKERRQ(ierr);
602aa219208SBarry Smith   ierr = PetscInfo(xin,"Loading vector from natural ordering into DMDA\n");CHKERRQ(ierr);
60347c6ae99SBarry Smith   ierr = PetscOptionsGetInt(((PetscObject)xin)->prefix,"-vecload_block_size",&bs,&flag);CHKERRQ(ierr);
60447c6ae99SBarry Smith   if (flag && bs != dd->w) {
605aa219208SBarry Smith     ierr = PetscInfo2(xin,"Block size in file %D not equal to DMDA's dof %D\n",bs,dd->w);CHKERRQ(ierr);
60647c6ae99SBarry Smith   }
60747c6ae99SBarry Smith   PetscFunctionReturn(0);
60847c6ae99SBarry Smith }
60947c6ae99SBarry Smith 
61047c6ae99SBarry Smith EXTERN_C_BEGIN
61147c6ae99SBarry Smith #undef __FUNCT__
61247c6ae99SBarry Smith #define __FUNCT__ "VecLoad_Default_DA"
6137087cfbeSBarry Smith PetscErrorCode  VecLoad_Default_DA(Vec xin, PetscViewer viewer)
61447c6ae99SBarry Smith {
61547c6ae99SBarry Smith   PetscErrorCode ierr;
6169a42bb27SBarry Smith   DM             da;
61747c6ae99SBarry Smith   PetscBool      isbinary;
61847c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
61947c6ae99SBarry Smith   PetscBool      ishdf5;
62047c6ae99SBarry Smith #endif
62147c6ae99SBarry Smith 
62247c6ae99SBarry Smith   PetscFunctionBegin;
6233c0c59f3SBarry Smith   ierr = PetscObjectQuery((PetscObject)xin,"DM",(PetscObject*)&da);CHKERRQ(ierr);
624aa219208SBarry Smith   if (!da) SETERRQ(((PetscObject)xin)->comm,PETSC_ERR_ARG_WRONG,"Vector not generated from a DMDA");
62547c6ae99SBarry Smith 
62647c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
62747c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERHDF5,&ishdf5);CHKERRQ(ierr);
62847c6ae99SBarry Smith #endif
62947c6ae99SBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
63047c6ae99SBarry Smith 
63147c6ae99SBarry Smith   if (isbinary) {
63247c6ae99SBarry Smith     ierr = VecLoad_Binary_DA(xin,viewer);CHKERRQ(ierr);
63347c6ae99SBarry Smith #if defined(PETSC_HAVE_HDF5)
63447c6ae99SBarry Smith   } else if (ishdf5) {
63547c6ae99SBarry Smith     ierr = VecLoad_HDF5_DA(xin,viewer);CHKERRQ(ierr);
63647c6ae99SBarry Smith #endif
637d34fcf5fSBarry Smith   } else SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported for vector loading", ((PetscObject)viewer)->type_name);
63847c6ae99SBarry Smith   PetscFunctionReturn(0);
63947c6ae99SBarry Smith }
64047c6ae99SBarry Smith EXTERN_C_END
641