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