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