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