17d0a6c19SBarry Smith 247c6ae99SBarry Smith /* 347c6ae99SBarry Smith Code for manipulating distributed regular 3d arrays in parallel. 447c6ae99SBarry Smith File created by Peter Mell 7/14/95 547c6ae99SBarry Smith */ 647c6ae99SBarry Smith 7b45d2f2cSJed Brown #include <petsc-private/daimpl.h> /*I "petscdmda.h" I*/ 847c6ae99SBarry Smith 947c6ae99SBarry Smith #undef __FUNCT__ 109a42bb27SBarry Smith #define __FUNCT__ "DMView_DA_3d" 119a42bb27SBarry Smith PetscErrorCode DMView_DA_3d(DM da,PetscViewer viewer) 1247c6ae99SBarry Smith { 1347c6ae99SBarry Smith PetscErrorCode ierr; 1447c6ae99SBarry Smith PetscMPIInt rank; 159a42bb27SBarry Smith PetscBool iascii,isdraw,isbinary; 1647c6ae99SBarry Smith DM_DA *dd = (DM_DA*)da->data; 179a42bb27SBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 189a42bb27SBarry Smith PetscBool ismatlab; 199a42bb27SBarry Smith #endif 2047c6ae99SBarry Smith 2147c6ae99SBarry Smith PetscFunctionBegin; 2247c6ae99SBarry Smith ierr = MPI_Comm_rank(((PetscObject)da)->comm,&rank);CHKERRQ(ierr); 2347c6ae99SBarry Smith 24251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 25251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 26251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr); 279a42bb27SBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 28251f4c67SDmitry Karpeev ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERMATLAB,&ismatlab);CHKERRQ(ierr); 299a42bb27SBarry Smith #endif 3047c6ae99SBarry Smith if (iascii) { 3147c6ae99SBarry Smith PetscViewerFormat format; 3247c6ae99SBarry Smith 337b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 3447c6ae99SBarry Smith ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr); 3547c6ae99SBarry Smith if (format != PETSC_VIEWER_ASCII_VTK && format != PETSC_VIEWER_ASCII_VTK_CELL) { 36aa219208SBarry Smith DMDALocalInfo info; 37aa219208SBarry Smith ierr = DMDAGetLocalInfo(da,&info);CHKERRQ(ierr); 3847c6ae99SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Processor [%d] M %D N %D P %D m %D n %D p %D w %D s %D\n",rank,dd->M,dd->N,dd->P,dd->m,dd->n,dd->p,dd->w,dd->s);CHKERRQ(ierr); 3947c6ae99SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"X range of indices: %D %D, Y range of indices: %D %D, Z range of indices: %D %D\n", 4047c6ae99SBarry Smith info.xs,info.xs+info.xm,info.ys,info.ys+info.ym,info.zs,info.zs+info.zm);CHKERRQ(ierr); 4147c6ae99SBarry Smith #if !defined(PETSC_USE_COMPLEX) 426636e97aSMatthew G Knepley if (da->coordinates) { 4347c6ae99SBarry Smith PetscInt last; 4447c6ae99SBarry Smith const PetscReal *coors; 456636e97aSMatthew G Knepley ierr = VecGetArrayRead(da->coordinates,&coors);CHKERRQ(ierr); 466636e97aSMatthew G Knepley ierr = VecGetLocalSize(da->coordinates,&last);CHKERRQ(ierr); 4747c6ae99SBarry Smith last = last - 3; 4847c6ae99SBarry Smith ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Lower left corner %G %G %G : Upper right %G %G %G\n",coors[0],coors[1],coors[2],coors[last],coors[last+1],coors[last+2]);CHKERRQ(ierr); 496636e97aSMatthew G Knepley ierr = VecRestoreArrayRead(da->coordinates,&coors);CHKERRQ(ierr); 5047c6ae99SBarry Smith } 5147c6ae99SBarry Smith #endif 5247c6ae99SBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 537b23a99aSBarry Smith ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);CHKERRQ(ierr); 54616157d6SJed Brown } else { 55616157d6SJed Brown ierr = DMView_DA_VTK(da,viewer);CHKERRQ(ierr); 5647c6ae99SBarry Smith } 5747c6ae99SBarry Smith } else if (isdraw) { 5847c6ae99SBarry Smith PetscDraw draw; 5947c6ae99SBarry Smith PetscReal ymin = -1.0,ymax = (PetscReal)dd->N; 6047c6ae99SBarry Smith PetscReal xmin = -1.0,xmax = (PetscReal)((dd->M+2)*dd->P),x,y,ycoord,xcoord; 6147c6ae99SBarry Smith PetscInt k,plane,base,*idx; 6247c6ae99SBarry Smith char node[10]; 6347c6ae99SBarry Smith PetscBool isnull; 6447c6ae99SBarry Smith 6547c6ae99SBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 6647c6ae99SBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0); 6747c6ae99SBarry Smith ierr = PetscDrawSetCoordinates(draw,xmin,ymin,xmax,ymax);CHKERRQ(ierr); 6847c6ae99SBarry Smith ierr = PetscDrawSynchronizedClear(draw);CHKERRQ(ierr); 6947c6ae99SBarry Smith 7047c6ae99SBarry Smith /* first processor draw all node lines */ 7147c6ae99SBarry Smith if (!rank) { 7247c6ae99SBarry Smith for (k=0; k<dd->P; k++) { 7347c6ae99SBarry Smith ymin = 0.0; ymax = (PetscReal)(dd->N - 1); 7447c6ae99SBarry Smith for (xmin=(PetscReal)(k*(dd->M+1)); xmin<(PetscReal)(dd->M+(k*(dd->M+1))); xmin++) { 7547c6ae99SBarry Smith ierr = PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_BLACK);CHKERRQ(ierr); 7647c6ae99SBarry Smith } 7747c6ae99SBarry Smith 7847c6ae99SBarry Smith xmin = (PetscReal)(k*(dd->M+1)); xmax = xmin + (PetscReal)(dd->M - 1); 7947c6ae99SBarry Smith for (ymin=0; ymin<(PetscReal)dd->N; ymin++) { 8047c6ae99SBarry Smith ierr = PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_BLACK);CHKERRQ(ierr); 8147c6ae99SBarry Smith } 8247c6ae99SBarry Smith } 8347c6ae99SBarry Smith } 8447c6ae99SBarry Smith ierr = PetscDrawSynchronizedFlush(draw);CHKERRQ(ierr); 8547c6ae99SBarry Smith ierr = PetscDrawPause(draw);CHKERRQ(ierr); 8647c6ae99SBarry Smith 8747c6ae99SBarry Smith for (k=0; k<dd->P; k++) { /*Go through and draw for each plane*/ 8847c6ae99SBarry Smith if ((k >= dd->zs) && (k < dd->ze)) { 8947c6ae99SBarry Smith /* draw my box */ 9047c6ae99SBarry Smith ymin = dd->ys; 9147c6ae99SBarry Smith ymax = dd->ye - 1; 9247c6ae99SBarry Smith xmin = dd->xs/dd->w + (dd->M+1)*k; 9347c6ae99SBarry Smith xmax =(dd->xe-1)/dd->w + (dd->M+1)*k; 9447c6ae99SBarry Smith 9547c6ae99SBarry Smith ierr = PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_RED);CHKERRQ(ierr); 9647c6ae99SBarry Smith ierr = PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_RED);CHKERRQ(ierr); 9747c6ae99SBarry Smith ierr = PetscDrawLine(draw,xmin,ymax,xmax,ymax,PETSC_DRAW_RED);CHKERRQ(ierr); 9847c6ae99SBarry Smith ierr = PetscDrawLine(draw,xmax,ymin,xmax,ymax,PETSC_DRAW_RED);CHKERRQ(ierr); 9947c6ae99SBarry Smith 10047c6ae99SBarry Smith xmin = dd->xs/dd->w; 10147c6ae99SBarry Smith xmax =(dd->xe-1)/dd->w; 10247c6ae99SBarry Smith 10347c6ae99SBarry Smith /* put in numbers*/ 10447c6ae99SBarry Smith base = (dd->base+(dd->xe-dd->xs)*(dd->ye-dd->ys)*(k-dd->zs))/dd->w; 10547c6ae99SBarry Smith 10647c6ae99SBarry Smith /* Identify which processor owns the box */ 10747c6ae99SBarry Smith sprintf(node,"%d",rank); 10847c6ae99SBarry Smith ierr = PetscDrawString(draw,xmin+(dd->M+1)*k+.2,ymin+.3,PETSC_DRAW_RED,node);CHKERRQ(ierr); 10947c6ae99SBarry Smith 11047c6ae99SBarry Smith for (y=ymin; y<=ymax; y++) { 11147c6ae99SBarry Smith for (x=xmin+(dd->M+1)*k; x<=xmax+(dd->M+1)*k; x++) { 11247c6ae99SBarry Smith sprintf(node,"%d",(int)base++); 11347c6ae99SBarry Smith ierr = PetscDrawString(draw,x,y,PETSC_DRAW_BLACK,node);CHKERRQ(ierr); 11447c6ae99SBarry Smith } 11547c6ae99SBarry Smith } 11647c6ae99SBarry Smith 11747c6ae99SBarry Smith } 11847c6ae99SBarry Smith } 11947c6ae99SBarry Smith ierr = PetscDrawSynchronizedFlush(draw);CHKERRQ(ierr); 12047c6ae99SBarry Smith ierr = PetscDrawPause(draw);CHKERRQ(ierr); 12147c6ae99SBarry Smith 12247c6ae99SBarry Smith for (k=0-dd->s; k<dd->P+dd->s; k++) { 12347c6ae99SBarry Smith /* Go through and draw for each plane */ 12447c6ae99SBarry Smith if ((k >= dd->Zs) && (k < dd->Ze)) { 12547c6ae99SBarry Smith 12647c6ae99SBarry Smith /* overlay ghost numbers, useful for error checking */ 12747c6ae99SBarry Smith base = (dd->Xe-dd->Xs)*(dd->Ye-dd->Ys)*(k-dd->Zs); idx = dd->idx; 12847c6ae99SBarry Smith plane=k; 12947c6ae99SBarry Smith /* Keep z wrap around points on the dradrawg */ 13047c6ae99SBarry Smith if (k<0) { plane=dd->P+k; } 13147c6ae99SBarry Smith if (k>=dd->P) { plane=k-dd->P; } 13247c6ae99SBarry Smith ymin = dd->Ys; ymax = dd->Ye; 13347c6ae99SBarry Smith xmin = (dd->M+1)*plane*dd->w; 13447c6ae99SBarry Smith xmax = (dd->M+1)*plane*dd->w+dd->M*dd->w; 13547c6ae99SBarry Smith for (y=ymin; y<ymax; y++) { 13647c6ae99SBarry Smith for (x=xmin+dd->Xs; x<xmin+dd->Xe; x+=dd->w) { 13747c6ae99SBarry Smith sprintf(node,"%d",(int)(idx[base]/dd->w)); 13847c6ae99SBarry Smith ycoord = y; 13947c6ae99SBarry Smith /*Keep y wrap around points on drawing */ 14047c6ae99SBarry Smith if (y<0) { ycoord = dd->N+y; } 14147c6ae99SBarry Smith 14247c6ae99SBarry Smith if (y>=dd->N) { ycoord = y-dd->N; } 14347c6ae99SBarry Smith xcoord = x; /* Keep x wrap points on drawing */ 14447c6ae99SBarry Smith 14547c6ae99SBarry Smith if (x<xmin) { xcoord = xmax - (xmin-x); } 14647c6ae99SBarry Smith if (x>=xmax) { xcoord = xmin + (x-xmax); } 14747c6ae99SBarry Smith ierr = PetscDrawString(draw,xcoord/dd->w,ycoord,PETSC_DRAW_BLUE,node);CHKERRQ(ierr); 14847c6ae99SBarry Smith base+=dd->w; 14947c6ae99SBarry Smith } 15047c6ae99SBarry Smith } 15147c6ae99SBarry Smith } 15247c6ae99SBarry Smith } 15347c6ae99SBarry Smith ierr = PetscDrawSynchronizedFlush(draw);CHKERRQ(ierr); 15447c6ae99SBarry Smith ierr = PetscDrawPause(draw);CHKERRQ(ierr); 1559a42bb27SBarry Smith } else if (isbinary){ 1569a42bb27SBarry Smith ierr = DMView_DA_Binary(da,viewer);CHKERRQ(ierr); 1579a42bb27SBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE) 1589a42bb27SBarry Smith } else if (ismatlab) { 1599a42bb27SBarry Smith ierr = DMView_DA_Matlab(da,viewer);CHKERRQ(ierr); 1609a42bb27SBarry Smith #endif 161aa219208SBarry Smith } else SETERRQ1(((PetscObject)da)->comm,PETSC_ERR_SUP,"Viewer type %s not supported for DMDA 1d",((PetscObject)viewer)->type_name); 16247c6ae99SBarry Smith PetscFunctionReturn(0); 16347c6ae99SBarry Smith } 16447c6ae99SBarry Smith 16547c6ae99SBarry Smith #undef __FUNCT__ 1669a42bb27SBarry Smith #define __FUNCT__ "DMSetUp_DA_3D" 1677087cfbeSBarry Smith PetscErrorCode DMSetUp_DA_3D(DM da) 16847c6ae99SBarry Smith { 16947c6ae99SBarry Smith DM_DA *dd = (DM_DA*)da->data; 17047c6ae99SBarry Smith const PetscInt M = dd->M; 17147c6ae99SBarry Smith const PetscInt N = dd->N; 17247c6ae99SBarry Smith const PetscInt P = dd->P; 17347c6ae99SBarry Smith PetscInt m = dd->m; 17447c6ae99SBarry Smith PetscInt n = dd->n; 17547c6ae99SBarry Smith PetscInt p = dd->p; 17688661749SPeter Brune PetscInt o = dd->overlap; 17747c6ae99SBarry Smith const PetscInt dof = dd->w; 17847c6ae99SBarry Smith const PetscInt s = dd->s; 17919fd82e9SBarry Smith DMDABoundaryType bx = dd->bx; 18019fd82e9SBarry Smith DMDABoundaryType by = dd->by; 18119fd82e9SBarry Smith DMDABoundaryType bz = dd->bz; 18219fd82e9SBarry Smith DMDAStencilType stencil_type = dd->stencil_type; 18347c6ae99SBarry Smith PetscInt *lx = dd->lx; 18447c6ae99SBarry Smith PetscInt *ly = dd->ly; 18547c6ae99SBarry Smith PetscInt *lz = dd->lz; 18647c6ae99SBarry Smith MPI_Comm comm; 18747c6ae99SBarry Smith PetscMPIInt rank,size; 188ce00eea3SSatish Balay PetscInt xs = 0,xe,ys = 0,ye,zs = 0,ze,x = 0,y = 0,z = 0; 189ce00eea3SSatish Balay PetscInt Xs,Xe,Ys,Ye,Zs,Ze,IXs,IXe,IYs,IYe,IZs,IZe,start,end,pm; 190ce00eea3SSatish Balay PetscInt left,right,up,down,bottom,top,i,j,k,*idx,*idx_cpy,nn; 191ce00eea3SSatish Balay const PetscInt *idx_full; 19247c6ae99SBarry Smith PetscInt n0,n1,n2,n3,n4,n5,n6,n7,n8,n9,n10,n11,n12,n14; 19347c6ae99SBarry Smith PetscInt n15,n16,n17,n18,n19,n20,n21,n22,n23,n24,n25,n26; 194db87c5ecSEthan Coon PetscInt *bases,*ldims,base,x_t,y_t,z_t,s_t,count,s_x,s_y,s_z; 19547c6ae99SBarry Smith PetscInt sn0 = 0,sn1 = 0,sn2 = 0,sn3 = 0,sn5 = 0,sn6 = 0,sn7 = 0; 19647c6ae99SBarry Smith PetscInt sn8 = 0,sn9 = 0,sn11 = 0,sn15 = 0,sn24 = 0,sn25 = 0,sn26 = 0; 19747c6ae99SBarry Smith PetscInt sn17 = 0,sn18 = 0,sn19 = 0,sn20 = 0,sn21 = 0,sn23 = 0; 19847c6ae99SBarry Smith Vec local,global; 19947c6ae99SBarry Smith VecScatter ltog,gtol; 200ce00eea3SSatish Balay IS to,from,ltogis; 2016f951b95Secoon PetscBool twod; 20247c6ae99SBarry Smith PetscErrorCode ierr; 20347c6ae99SBarry Smith 2046f951b95Secoon 20547c6ae99SBarry Smith PetscFunctionBegin; 206*c2859e5eSBarry Smith if (stencil_type == DMDA_STENCIL_BOX && (bx == DMDA_BOUNDARY_MIRROR || by == DMDA_BOUNDARY_MIRROR || bz == DMDA_BOUNDARY_MIRROR)) SETERRQ(((PetscObject)da)->comm,PETSC_ERR_SUP,"Mirror boundary and box stencil"); 20747c6ae99SBarry Smith if (dof < 1) SETERRQ1(((PetscObject)da)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Must have 1 or more degrees of freedom per node: %D",dof); 20847c6ae99SBarry Smith if (s < 0) SETERRQ1(((PetscObject)da)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Stencil width cannot be negative: %D",s); 20947c6ae99SBarry Smith ierr = PetscObjectGetComm((PetscObject) da, &comm);CHKERRQ(ierr); 2103855c12bSBarry Smith #if !defined(PETSC_USE_64BIT_INDICES) 2113855c12bSBarry Smith if (((Petsc64bitInt) M)*((Petsc64bitInt) N)*((Petsc64bitInt) P)*((Petsc64bitInt) dof) > (Petsc64bitInt) PETSC_MPI_INT_MAX) SETERRQ3(comm,PETSC_ERR_INT_OVERFLOW,"Mesh of %D by %D by %D (dof) is too large for 32 bit indices",M,N,dof); 2123855c12bSBarry Smith #endif 2133855c12bSBarry Smith 21447c6ae99SBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 21547c6ae99SBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 21647c6ae99SBarry Smith 21747c6ae99SBarry Smith dd->dim = 3; 21847c6ae99SBarry Smith ierr = PetscMalloc(dof*sizeof(char*),&dd->fieldname);CHKERRQ(ierr); 21947c6ae99SBarry Smith ierr = PetscMemzero(dd->fieldname,dof*sizeof(char*));CHKERRQ(ierr); 22047c6ae99SBarry Smith 22147c6ae99SBarry Smith if (m != PETSC_DECIDE) { 22247c6ae99SBarry Smith if (m < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in X direction: %D",m); 22347c6ae99SBarry Smith else if (m > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in X direction: %D %d",m,size); 22447c6ae99SBarry Smith } 22547c6ae99SBarry Smith if (n != PETSC_DECIDE) { 22647c6ae99SBarry Smith if (n < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in Y direction: %D",n); 22747c6ae99SBarry Smith else if (n > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in Y direction: %D %d",n,size); 22847c6ae99SBarry Smith } 22947c6ae99SBarry Smith if (p != PETSC_DECIDE) { 23047c6ae99SBarry Smith if (p < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in Z direction: %D",p); 23147c6ae99SBarry Smith else if (p > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in Z direction: %D %d",p,size); 23247c6ae99SBarry Smith } 2330716a85fSBarry Smith if ((m > 0) && (n > 0) && (p > 0) && (m*n*p != size)) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"m %D * n %D * p %D != size %d",m,n,p,size); 23447c6ae99SBarry Smith 23547c6ae99SBarry Smith /* Partition the array among the processors */ 23647c6ae99SBarry Smith if (m == PETSC_DECIDE && n != PETSC_DECIDE && p != PETSC_DECIDE) { 23747c6ae99SBarry Smith m = size/(n*p); 23847c6ae99SBarry Smith } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) { 23947c6ae99SBarry Smith n = size/(m*p); 24047c6ae99SBarry Smith } else if (m != PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) { 24147c6ae99SBarry Smith p = size/(m*n); 24247c6ae99SBarry Smith } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) { 24347c6ae99SBarry Smith /* try for squarish distribution */ 2448f1a2a5eSBarry Smith m = (int)(0.5 + sqrt(((double)M)*((double)size)/((double)N*p))); 24547c6ae99SBarry Smith if (!m) m = 1; 24647c6ae99SBarry Smith while (m > 0) { 24747c6ae99SBarry Smith n = size/(m*p); 24847c6ae99SBarry Smith if (m*n*p == size) break; 24947c6ae99SBarry Smith m--; 25047c6ae99SBarry Smith } 25147c6ae99SBarry Smith if (!m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad p value: p = %D",p); 25247c6ae99SBarry Smith if (M > N && m < n) {PetscInt _m = m; m = n; n = _m;} 25347c6ae99SBarry Smith } else if (m == PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) { 25447c6ae99SBarry Smith /* try for squarish distribution */ 2558f1a2a5eSBarry Smith m = (int)(0.5 + sqrt(((double)M)*((double)size)/((double)P*n))); 25647c6ae99SBarry Smith if (!m) m = 1; 25747c6ae99SBarry Smith while (m > 0) { 25847c6ae99SBarry Smith p = size/(m*n); 25947c6ae99SBarry Smith if (m*n*p == size) break; 26047c6ae99SBarry Smith m--; 26147c6ae99SBarry Smith } 26247c6ae99SBarry Smith if (!m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad n value: n = %D",n); 26347c6ae99SBarry Smith if (M > P && m < p) {PetscInt _m = m; m = p; p = _m;} 26447c6ae99SBarry Smith } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) { 26547c6ae99SBarry Smith /* try for squarish distribution */ 2668f1a2a5eSBarry Smith n = (int)(0.5 + sqrt(((double)N)*((double)size)/((double)P*m))); 26747c6ae99SBarry Smith if (!n) n = 1; 26847c6ae99SBarry Smith while (n > 0) { 26947c6ae99SBarry Smith p = size/(m*n); 27047c6ae99SBarry Smith if (m*n*p == size) break; 27147c6ae99SBarry Smith n--; 27247c6ae99SBarry Smith } 27347c6ae99SBarry Smith if (!n) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad m value: m = %D",n); 27447c6ae99SBarry Smith if (N > P && n < p) {PetscInt _n = n; n = p; p = _n;} 27547c6ae99SBarry Smith } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) { 27647c6ae99SBarry Smith /* try for squarish distribution */ 2778f1a2a5eSBarry Smith n = (PetscInt)(0.5 + pow(((double)N*N)*((double)size)/((double)P*M),(double)(1./3.))); 27847c6ae99SBarry Smith if (!n) n = 1; 27947c6ae99SBarry Smith while (n > 0) { 28047c6ae99SBarry Smith pm = size/n; 28147c6ae99SBarry Smith if (n*pm == size) break; 28247c6ae99SBarry Smith n--; 28347c6ae99SBarry Smith } 28447c6ae99SBarry Smith if (!n) n = 1; 2858f1a2a5eSBarry Smith m = (PetscInt)(0.5 + sqrt(((double)M)*((double)size)/((double)P*n))); 28647c6ae99SBarry Smith if (!m) m = 1; 28747c6ae99SBarry Smith while (m > 0) { 28847c6ae99SBarry Smith p = size/(m*n); 28947c6ae99SBarry Smith if (m*n*p == size) break; 29047c6ae99SBarry Smith m--; 29147c6ae99SBarry Smith } 29247c6ae99SBarry Smith if (M > P && m < p) {PetscInt _m = m; m = p; p = _m;} 29347c6ae99SBarry Smith } else if (m*n*p != size) SETERRQ(((PetscObject)da)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Given Bad partition"); 29447c6ae99SBarry Smith 29547c6ae99SBarry Smith if (m*n*p != size) SETERRQ(((PetscObject)da)->comm,PETSC_ERR_PLIB,"Could not find good partition"); 29647c6ae99SBarry Smith if (M < m) SETERRQ2(((PetscObject)da)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Partition in x direction is too fine! %D %D",M,m); 29747c6ae99SBarry Smith if (N < n) SETERRQ2(((PetscObject)da)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Partition in y direction is too fine! %D %D",N,n); 29847c6ae99SBarry Smith if (P < p) SETERRQ2(((PetscObject)da)->comm,PETSC_ERR_ARG_OUTOFRANGE,"Partition in z direction is too fine! %D %D",P,p); 29947c6ae99SBarry Smith 30047c6ae99SBarry Smith /* 30147c6ae99SBarry Smith Determine locally owned region 30247c6ae99SBarry Smith [x, y, or z]s is the first local node number, [x, y, z] is the number of local nodes 30347c6ae99SBarry Smith */ 30447c6ae99SBarry Smith 30547c6ae99SBarry Smith if (!lx) { 30647c6ae99SBarry Smith ierr = PetscMalloc(m*sizeof(PetscInt), &dd->lx);CHKERRQ(ierr); 30747c6ae99SBarry Smith lx = dd->lx; 30847c6ae99SBarry Smith for (i=0; i<m; i++) { 30947c6ae99SBarry Smith lx[i] = M/m + ((M % m) > (i % m)); 31047c6ae99SBarry Smith } 31147c6ae99SBarry Smith } 31247c6ae99SBarry Smith x = lx[rank % m]; 31347c6ae99SBarry Smith xs = 0; 31447c6ae99SBarry Smith for (i=0; i<(rank%m); i++) { xs += lx[i];} 31588661749SPeter Brune if ((x < s+o) && ((m > 1) || (bx == DMDA_BOUNDARY_PERIODIC))) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local x-width of domain x %D is smaller than stencil width s %D",x,s); 31647c6ae99SBarry Smith 31747c6ae99SBarry Smith if (!ly) { 31847c6ae99SBarry Smith ierr = PetscMalloc(n*sizeof(PetscInt), &dd->ly);CHKERRQ(ierr); 31947c6ae99SBarry Smith ly = dd->ly; 32047c6ae99SBarry Smith for (i=0; i<n; i++) { 32147c6ae99SBarry Smith ly[i] = N/n + ((N % n) > (i % n)); 32247c6ae99SBarry Smith } 32347c6ae99SBarry Smith } 32447c6ae99SBarry Smith y = ly[(rank % (m*n))/m]; 32588661749SPeter Brune if ((y < s+o) && ((n > 1) || (by == DMDA_BOUNDARY_PERIODIC))) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local y-width of domain y %D is smaller than stencil width s %D",y,s); 32630729d88SBarry Smith 32747c6ae99SBarry Smith ys = 0; 32847c6ae99SBarry Smith for (i=0; i<(rank % (m*n))/m; i++) { ys += ly[i];} 32947c6ae99SBarry Smith 33047c6ae99SBarry Smith if (!lz) { 33147c6ae99SBarry Smith ierr = PetscMalloc(p*sizeof(PetscInt), &dd->lz);CHKERRQ(ierr); 33247c6ae99SBarry Smith lz = dd->lz; 33347c6ae99SBarry Smith for (i=0; i<p; i++) { 33447c6ae99SBarry Smith lz[i] = P/p + ((P % p) > (i % p)); 33547c6ae99SBarry Smith } 33647c6ae99SBarry Smith } 33747c6ae99SBarry Smith z = lz[rank/(m*n)]; 338bcea557cSEthan Coon 339fdc81ce8SEthan Coon /* note this is different than x- and y-, as we will handle as an important special 340fdc81ce8SEthan Coon case when p=P=1 and DMDA_BOUNDARY_PERIODIC and s > z. This is to deal with 2D problems 341fdc81ce8SEthan Coon in a 3D code. Additional code for this case is noted with "2d case" comments */ 3426f951b95Secoon twod = PETSC_FALSE; 3436f951b95Secoon if (P == 1) { 3446f951b95Secoon twod = PETSC_TRUE; 34588661749SPeter Brune } else if ((z < s+o) && ((p > 1) || (bz == DMDA_BOUNDARY_PERIODIC))) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local z-width of domain z %D is smaller than stencil width s %D",z,s+o); 34647c6ae99SBarry Smith zs = 0; 34747c6ae99SBarry Smith for (i=0; i<(rank/(m*n)); i++) { zs += lz[i];} 34847c6ae99SBarry Smith ye = ys + y; 34947c6ae99SBarry Smith xe = xs + x; 35047c6ae99SBarry Smith ze = zs + z; 35147c6ae99SBarry Smith 35288661749SPeter Brune /* determine ghost region (Xs) and region scattered into (IXs) */ 35388661749SPeter Brune if (xs-s-o > 0) { 35488661749SPeter Brune Xs = xs - s - o; IXs = xs - s - o; 35588661749SPeter Brune } else { 35688661749SPeter Brune if (bx) { 35788661749SPeter Brune Xs = xs - s; 35888661749SPeter Brune } else { 35988661749SPeter Brune Xs = 0; 36088661749SPeter Brune } 36188661749SPeter Brune IXs = 0; 36288661749SPeter Brune } 36388661749SPeter Brune if (xe+s+o <= M) { 36488661749SPeter Brune Xe = xe + s + o; IXe = xe + s + o; 36588661749SPeter Brune } else { 36688661749SPeter Brune if (bx) { 36788661749SPeter Brune Xs = xs - s - o; Xe = xe + s; 36888661749SPeter Brune } else { 36988661749SPeter Brune Xe = M; 37088661749SPeter Brune } 37188661749SPeter Brune IXe = M; 37288661749SPeter Brune } 37347c6ae99SBarry Smith 374*c2859e5eSBarry Smith if (bx == DMDA_BOUNDARY_PERIODIC || bx == DMDA_BOUNDARY_MIRROR) { 37588661749SPeter Brune IXs = xs - s - o; 37688661749SPeter Brune IXe = xe + s + o; 37788661749SPeter Brune Xs = xs - s - o; 37888661749SPeter Brune Xe = xe + s + o; 37988661749SPeter Brune } 38088661749SPeter Brune 38188661749SPeter Brune if (ys-s-o > 0) { 38288661749SPeter Brune Ys = ys - s - o; IYs = ys - s - o; 38388661749SPeter Brune } else { 38488661749SPeter Brune if (by) { 38588661749SPeter Brune Ys = ys - s; 38688661749SPeter Brune } else { 38788661749SPeter Brune Ys = 0; 38888661749SPeter Brune } 38988661749SPeter Brune IYs = 0; 39088661749SPeter Brune } 39188661749SPeter Brune if (ye+s+o <= N) { 39288661749SPeter Brune Ye = ye + s + o; IYe = ye + s + o; 39388661749SPeter Brune } else { 39488661749SPeter Brune if (by) { 39588661749SPeter Brune Ye = ye + s; 39688661749SPeter Brune } else { 39788661749SPeter Brune Ye = N; 39888661749SPeter Brune } 39988661749SPeter Brune IYe = N; 40088661749SPeter Brune } 40188661749SPeter Brune 402*c2859e5eSBarry Smith if (by == DMDA_BOUNDARY_PERIODIC || by == DMDA_BOUNDARY_MIRROR) { 40388661749SPeter Brune IYs = ys - s - o; 40488661749SPeter Brune IYe = ye + s + o; 40588661749SPeter Brune Ys = ys - s - o; 40688661749SPeter Brune Ye = ye + s + o; 40788661749SPeter Brune } 40888661749SPeter Brune 40988661749SPeter Brune if (zs-s-o > 0) { 41088661749SPeter Brune Zs = zs - s - o; IZs = zs - s - o; 41188661749SPeter Brune } else { 41288661749SPeter Brune if (bz) { 41388661749SPeter Brune Zs = zs - s; 41488661749SPeter Brune } else { 41588661749SPeter Brune Zs = 0; 41688661749SPeter Brune } 41788661749SPeter Brune IZs = 0; 41888661749SPeter Brune } 41988661749SPeter Brune if (ze+s+o <= P) { 42088661749SPeter Brune Ze = ze + s + o; IZe = ze + s + o; 42188661749SPeter Brune } else { 42288661749SPeter Brune if (bz) { 42388661749SPeter Brune Ze = ze + s; 42488661749SPeter Brune } else { 42588661749SPeter Brune Ze = P; 42688661749SPeter Brune } 42788661749SPeter Brune IZe = P; 42888661749SPeter Brune } 42988661749SPeter Brune 430*c2859e5eSBarry Smith if (bz == DMDA_BOUNDARY_PERIODIC || bz == DMDA_BOUNDARY_MIRROR) { 43188661749SPeter Brune IZs = zs - s - o; 43288661749SPeter Brune IZe = ze + s + o; 43388661749SPeter Brune Zs = zs - s - o; 43488661749SPeter Brune Ze = ze + s + o; 43588661749SPeter Brune } 43647c6ae99SBarry Smith 43747c6ae99SBarry Smith /* Resize all X parameters to reflect w */ 43888661749SPeter Brune s_x = s + o; 43988661749SPeter Brune s_y = s + o; 44088661749SPeter Brune s_z = s + o; 44147c6ae99SBarry Smith 44247c6ae99SBarry Smith /* determine starting point of each processor */ 44347c6ae99SBarry Smith nn = x*y*z; 44447c6ae99SBarry Smith ierr = PetscMalloc2(size+1,PetscInt,&bases,size,PetscInt,&ldims);CHKERRQ(ierr); 44547c6ae99SBarry Smith ierr = MPI_Allgather(&nn,1,MPIU_INT,ldims,1,MPIU_INT,comm);CHKERRQ(ierr); 44647c6ae99SBarry Smith bases[0] = 0; 44747c6ae99SBarry Smith for (i=1; i<=size; i++) { 44847c6ae99SBarry Smith bases[i] = ldims[i-1]; 44947c6ae99SBarry Smith } 45047c6ae99SBarry Smith for (i=1; i<=size; i++) { 45147c6ae99SBarry Smith bases[i] += bases[i-1]; 45247c6ae99SBarry Smith } 453ce00eea3SSatish Balay base = bases[rank]*dof; 45447c6ae99SBarry Smith 45547c6ae99SBarry Smith /* allocate the base parallel and sequential vectors */ 456ce00eea3SSatish Balay dd->Nlocal = x*y*z*dof; 457778a2246SBarry Smith ierr = VecCreateMPIWithArray(comm,dof,dd->Nlocal,PETSC_DECIDE,0,&global);CHKERRQ(ierr); 458ce00eea3SSatish Balay dd->nlocal = (Xe-Xs)*(Ye-Ys)*(Ze-Zs)*dof; 459778a2246SBarry Smith ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,dof,dd->nlocal,0,&local);CHKERRQ(ierr); 46047c6ae99SBarry Smith 46147c6ae99SBarry Smith /* generate appropriate vector scatters */ 46247c6ae99SBarry Smith /* local to global inserts non-ghost point region into global */ 46347c6ae99SBarry Smith ierr = VecGetOwnershipRange(global,&start,&end);CHKERRQ(ierr); 464ce00eea3SSatish Balay ierr = ISCreateStride(comm,x*y*z*dof,start,1,&to);CHKERRQ(ierr); 46547c6ae99SBarry Smith 466db87c5ecSEthan Coon count = x*y*z; 467ce00eea3SSatish Balay ierr = PetscMalloc(x*y*z*sizeof(PetscInt),&idx);CHKERRQ(ierr); 468ce00eea3SSatish Balay left = xs - Xs; right = left + x; 46947c6ae99SBarry Smith bottom = ys - Ys; top = bottom + y; 47047c6ae99SBarry Smith down = zs - Zs; up = down + z; 47147c6ae99SBarry Smith count = 0; 47247c6ae99SBarry Smith for (i=down; i<up; i++) { 47347c6ae99SBarry Smith for (j=bottom; j<top; j++) { 474ce00eea3SSatish Balay for (k=left; k<right; k++) { 475ce00eea3SSatish Balay idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 47647c6ae99SBarry Smith } 47747c6ae99SBarry Smith } 47847c6ae99SBarry Smith } 47947c6ae99SBarry Smith 48047c6ae99SBarry Smith ierr = ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&from);CHKERRQ(ierr); 48147c6ae99SBarry Smith ierr = VecScatterCreate(local,from,global,to,<og);CHKERRQ(ierr); 48247c6ae99SBarry Smith ierr = PetscLogObjectParent(da,ltog);CHKERRQ(ierr); 483fcfd50ebSBarry Smith ierr = ISDestroy(&from);CHKERRQ(ierr); 484fcfd50ebSBarry Smith ierr = ISDestroy(&to);CHKERRQ(ierr); 48547c6ae99SBarry Smith 486ce00eea3SSatish Balay /* global to local must include ghost points within the domain, 487ce00eea3SSatish Balay but not ghost points outside the domain that aren't periodic */ 48888661749SPeter Brune if (stencil_type == DMDA_STENCIL_BOX || o > 0) { 489db87c5ecSEthan Coon count = (IXe-IXs)*(IYe-IYs)*(IZe-IZs); 490db87c5ecSEthan Coon ierr = PetscMalloc(count*sizeof(PetscInt),&idx);CHKERRQ(ierr); 491ce00eea3SSatish Balay 492ce00eea3SSatish Balay left = IXs - Xs; right = left + (IXe-IXs); 493ce00eea3SSatish Balay bottom = IYs - Ys; top = bottom + (IYe-IYs); 494ce00eea3SSatish Balay down = IZs - Zs; up = down + (IZe-IZs); 495ce00eea3SSatish Balay count = 0; 496ce00eea3SSatish Balay for (i=down; i<up; i++) { 497ce00eea3SSatish Balay for (j=bottom; j<top; j++) { 498ce00eea3SSatish Balay for (k=left; k<right; k++) { 499ce00eea3SSatish Balay idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 500ce00eea3SSatish Balay } 501ce00eea3SSatish Balay } 502ce00eea3SSatish Balay } 503ce00eea3SSatish Balay ierr = ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&to);CHKERRQ(ierr); 504ce00eea3SSatish Balay 50547c6ae99SBarry Smith } else { 50647c6ae99SBarry Smith /* This is way ugly! We need to list the funny cross type region */ 507db87c5ecSEthan Coon count = ((ys-IYs) + (IYe-ye))*x*z + ((xs-IXs) + (IXe-xe))*y*z + ((zs-IZs) + (IZe-ze))*x*y + x*y*z; 508db87c5ecSEthan Coon ierr = PetscMalloc(count*sizeof(PetscInt),&idx);CHKERRQ(ierr); 509ce00eea3SSatish Balay 510ce00eea3SSatish Balay left = xs - Xs; right = left + x; 51147c6ae99SBarry Smith bottom = ys - Ys; top = bottom + y; 51247c6ae99SBarry Smith down = zs - Zs; up = down + z; 51347c6ae99SBarry Smith count = 0; 514ce00eea3SSatish Balay /* the bottom chunck */ 515ce00eea3SSatish Balay for (i=(IZs-Zs); i<down; i++) { 51647c6ae99SBarry Smith for (j=bottom; j<top; j++) { 517ce00eea3SSatish Balay for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 51847c6ae99SBarry Smith } 51947c6ae99SBarry Smith } 52047c6ae99SBarry Smith /* the middle piece */ 52147c6ae99SBarry Smith for (i=down; i<up; i++) { 52247c6ae99SBarry Smith /* front */ 523ce00eea3SSatish Balay for (j=(IYs-Ys); j<bottom; j++) { 524ce00eea3SSatish Balay for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 52547c6ae99SBarry Smith } 52647c6ae99SBarry Smith /* middle */ 52747c6ae99SBarry Smith for (j=bottom; j<top; j++) { 528ce00eea3SSatish Balay for (k=IXs-Xs; k<IXe-Xs; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 52947c6ae99SBarry Smith } 53047c6ae99SBarry Smith /* back */ 531ce00eea3SSatish Balay for (j=top; j<top+IYe-ye; j++) { 532ce00eea3SSatish Balay for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 53347c6ae99SBarry Smith } 53447c6ae99SBarry Smith } 53547c6ae99SBarry Smith /* the top piece */ 536ce00eea3SSatish Balay for (i=up; i<up+IZe-ze; i++) { 53747c6ae99SBarry Smith for (j=bottom; j<top; j++) { 538ce00eea3SSatish Balay for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 53947c6ae99SBarry Smith } 54047c6ae99SBarry Smith } 54147c6ae99SBarry Smith ierr = ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&to);CHKERRQ(ierr); 54247c6ae99SBarry Smith } 54347c6ae99SBarry Smith 54447c6ae99SBarry Smith /* determine who lies on each side of use stored in n24 n25 n26 54547c6ae99SBarry Smith n21 n22 n23 54647c6ae99SBarry Smith n18 n19 n20 54747c6ae99SBarry Smith 54847c6ae99SBarry Smith n15 n16 n17 54947c6ae99SBarry Smith n12 n14 55047c6ae99SBarry Smith n9 n10 n11 55147c6ae99SBarry Smith 55247c6ae99SBarry Smith n6 n7 n8 55347c6ae99SBarry Smith n3 n4 n5 55447c6ae99SBarry Smith n0 n1 n2 55547c6ae99SBarry Smith */ 55647c6ae99SBarry Smith 55747c6ae99SBarry Smith /* Solve for X,Y, and Z Periodic Case First, Then Modify Solution */ 55847c6ae99SBarry Smith /* Assume Nodes are Internal to the Cube */ 55947c6ae99SBarry Smith n0 = rank - m*n - m - 1; 56047c6ae99SBarry Smith n1 = rank - m*n - m; 56147c6ae99SBarry Smith n2 = rank - m*n - m + 1; 56247c6ae99SBarry Smith n3 = rank - m*n -1; 56347c6ae99SBarry Smith n4 = rank - m*n; 56447c6ae99SBarry Smith n5 = rank - m*n + 1; 56547c6ae99SBarry Smith n6 = rank - m*n + m - 1; 56647c6ae99SBarry Smith n7 = rank - m*n + m; 56747c6ae99SBarry Smith n8 = rank - m*n + m + 1; 56847c6ae99SBarry Smith 56947c6ae99SBarry Smith n9 = rank - m - 1; 57047c6ae99SBarry Smith n10 = rank - m; 57147c6ae99SBarry Smith n11 = rank - m + 1; 57247c6ae99SBarry Smith n12 = rank - 1; 57347c6ae99SBarry Smith n14 = rank + 1; 57447c6ae99SBarry Smith n15 = rank + m - 1; 57547c6ae99SBarry Smith n16 = rank + m; 57647c6ae99SBarry Smith n17 = rank + m + 1; 57747c6ae99SBarry Smith 57847c6ae99SBarry Smith n18 = rank + m*n - m - 1; 57947c6ae99SBarry Smith n19 = rank + m*n - m; 58047c6ae99SBarry Smith n20 = rank + m*n - m + 1; 58147c6ae99SBarry Smith n21 = rank + m*n - 1; 58247c6ae99SBarry Smith n22 = rank + m*n; 58347c6ae99SBarry Smith n23 = rank + m*n + 1; 58447c6ae99SBarry Smith n24 = rank + m*n + m - 1; 58547c6ae99SBarry Smith n25 = rank + m*n + m; 58647c6ae99SBarry Smith n26 = rank + m*n + m + 1; 58747c6ae99SBarry Smith 58847c6ae99SBarry Smith /* Assume Pieces are on Faces of Cube */ 58947c6ae99SBarry Smith 59047c6ae99SBarry Smith if (xs == 0) { /* First assume not corner or edge */ 59147c6ae99SBarry Smith n0 = rank -1 - (m*n); 59247c6ae99SBarry Smith n3 = rank + m -1 - (m*n); 59347c6ae99SBarry Smith n6 = rank + 2*m -1 - (m*n); 59447c6ae99SBarry Smith n9 = rank -1; 59547c6ae99SBarry Smith n12 = rank + m -1; 59647c6ae99SBarry Smith n15 = rank + 2*m -1; 59747c6ae99SBarry Smith n18 = rank -1 + (m*n); 59847c6ae99SBarry Smith n21 = rank + m -1 + (m*n); 59947c6ae99SBarry Smith n24 = rank + 2*m -1 + (m*n); 60047c6ae99SBarry Smith } 60147c6ae99SBarry Smith 602ce00eea3SSatish Balay if (xe == M) { /* First assume not corner or edge */ 60347c6ae99SBarry Smith n2 = rank -2*m +1 - (m*n); 60447c6ae99SBarry Smith n5 = rank - m +1 - (m*n); 60547c6ae99SBarry Smith n8 = rank +1 - (m*n); 60647c6ae99SBarry Smith n11 = rank -2*m +1; 60747c6ae99SBarry Smith n14 = rank - m +1; 60847c6ae99SBarry Smith n17 = rank +1; 60947c6ae99SBarry Smith n20 = rank -2*m +1 + (m*n); 61047c6ae99SBarry Smith n23 = rank - m +1 + (m*n); 61147c6ae99SBarry Smith n26 = rank +1 + (m*n); 61247c6ae99SBarry Smith } 61347c6ae99SBarry Smith 61447c6ae99SBarry Smith if (ys==0) { /* First assume not corner or edge */ 61547c6ae99SBarry Smith n0 = rank + m * (n-1) -1 - (m*n); 61647c6ae99SBarry Smith n1 = rank + m * (n-1) - (m*n); 61747c6ae99SBarry Smith n2 = rank + m * (n-1) +1 - (m*n); 61847c6ae99SBarry Smith n9 = rank + m * (n-1) -1; 61947c6ae99SBarry Smith n10 = rank + m * (n-1); 62047c6ae99SBarry Smith n11 = rank + m * (n-1) +1; 62147c6ae99SBarry Smith n18 = rank + m * (n-1) -1 + (m*n); 62247c6ae99SBarry Smith n19 = rank + m * (n-1) + (m*n); 62347c6ae99SBarry Smith n20 = rank + m * (n-1) +1 + (m*n); 62447c6ae99SBarry Smith } 62547c6ae99SBarry Smith 62647c6ae99SBarry Smith if (ye == N) { /* First assume not corner or edge */ 62747c6ae99SBarry Smith n6 = rank - m * (n-1) -1 - (m*n); 62847c6ae99SBarry Smith n7 = rank - m * (n-1) - (m*n); 62947c6ae99SBarry Smith n8 = rank - m * (n-1) +1 - (m*n); 63047c6ae99SBarry Smith n15 = rank - m * (n-1) -1; 63147c6ae99SBarry Smith n16 = rank - m * (n-1); 63247c6ae99SBarry Smith n17 = rank - m * (n-1) +1; 63347c6ae99SBarry Smith n24 = rank - m * (n-1) -1 + (m*n); 63447c6ae99SBarry Smith n25 = rank - m * (n-1) + (m*n); 63547c6ae99SBarry Smith n26 = rank - m * (n-1) +1 + (m*n); 63647c6ae99SBarry Smith } 63747c6ae99SBarry Smith 63847c6ae99SBarry Smith if (zs == 0) { /* First assume not corner or edge */ 63947c6ae99SBarry Smith n0 = size - (m*n) + rank - m - 1; 64047c6ae99SBarry Smith n1 = size - (m*n) + rank - m; 64147c6ae99SBarry Smith n2 = size - (m*n) + rank - m + 1; 64247c6ae99SBarry Smith n3 = size - (m*n) + rank - 1; 64347c6ae99SBarry Smith n4 = size - (m*n) + rank; 64447c6ae99SBarry Smith n5 = size - (m*n) + rank + 1; 64547c6ae99SBarry Smith n6 = size - (m*n) + rank + m - 1; 64647c6ae99SBarry Smith n7 = size - (m*n) + rank + m ; 64747c6ae99SBarry Smith n8 = size - (m*n) + rank + m + 1; 64847c6ae99SBarry Smith } 64947c6ae99SBarry Smith 65047c6ae99SBarry Smith if (ze == P) { /* First assume not corner or edge */ 65147c6ae99SBarry Smith n18 = (m*n) - (size-rank) - m - 1; 65247c6ae99SBarry Smith n19 = (m*n) - (size-rank) - m; 65347c6ae99SBarry Smith n20 = (m*n) - (size-rank) - m + 1; 65447c6ae99SBarry Smith n21 = (m*n) - (size-rank) - 1; 65547c6ae99SBarry Smith n22 = (m*n) - (size-rank); 65647c6ae99SBarry Smith n23 = (m*n) - (size-rank) + 1; 65747c6ae99SBarry Smith n24 = (m*n) - (size-rank) + m - 1; 65847c6ae99SBarry Smith n25 = (m*n) - (size-rank) + m; 65947c6ae99SBarry Smith n26 = (m*n) - (size-rank) + m + 1; 66047c6ae99SBarry Smith } 66147c6ae99SBarry Smith 66247c6ae99SBarry Smith if ((xs==0) && (zs==0)) { /* Assume an edge, not corner */ 66347c6ae99SBarry Smith n0 = size - m*n + rank + m-1 - m; 66447c6ae99SBarry Smith n3 = size - m*n + rank + m-1; 66547c6ae99SBarry Smith n6 = size - m*n + rank + m-1 + m; 66647c6ae99SBarry Smith } 66747c6ae99SBarry Smith 66847c6ae99SBarry Smith if ((xs==0) && (ze==P)) { /* Assume an edge, not corner */ 66947c6ae99SBarry Smith n18 = m*n - (size - rank) + m-1 - m; 67047c6ae99SBarry Smith n21 = m*n - (size - rank) + m-1; 67147c6ae99SBarry Smith n24 = m*n - (size - rank) + m-1 + m; 67247c6ae99SBarry Smith } 67347c6ae99SBarry Smith 67447c6ae99SBarry Smith if ((xs==0) && (ys==0)) { /* Assume an edge, not corner */ 67547c6ae99SBarry Smith n0 = rank + m*n -1 - m*n; 67647c6ae99SBarry Smith n9 = rank + m*n -1; 67747c6ae99SBarry Smith n18 = rank + m*n -1 + m*n; 67847c6ae99SBarry Smith } 67947c6ae99SBarry Smith 68047c6ae99SBarry Smith if ((xs==0) && (ye==N)) { /* Assume an edge, not corner */ 68147c6ae99SBarry Smith n6 = rank - m*(n-1) + m-1 - m*n; 68247c6ae99SBarry Smith n15 = rank - m*(n-1) + m-1; 68347c6ae99SBarry Smith n24 = rank - m*(n-1) + m-1 + m*n; 68447c6ae99SBarry Smith } 68547c6ae99SBarry Smith 686ce00eea3SSatish Balay if ((xe==M) && (zs==0)) { /* Assume an edge, not corner */ 68747c6ae99SBarry Smith n2 = size - (m*n-rank) - (m-1) - m; 68847c6ae99SBarry Smith n5 = size - (m*n-rank) - (m-1); 68947c6ae99SBarry Smith n8 = size - (m*n-rank) - (m-1) + m; 69047c6ae99SBarry Smith } 69147c6ae99SBarry Smith 692ce00eea3SSatish Balay if ((xe==M) && (ze==P)) { /* Assume an edge, not corner */ 69347c6ae99SBarry Smith n20 = m*n - (size - rank) - (m-1) - m; 69447c6ae99SBarry Smith n23 = m*n - (size - rank) - (m-1); 69547c6ae99SBarry Smith n26 = m*n - (size - rank) - (m-1) + m; 69647c6ae99SBarry Smith } 69747c6ae99SBarry Smith 698ce00eea3SSatish Balay if ((xe==M) && (ys==0)) { /* Assume an edge, not corner */ 69947c6ae99SBarry Smith n2 = rank + m*(n-1) - (m-1) - m*n; 70047c6ae99SBarry Smith n11 = rank + m*(n-1) - (m-1); 70147c6ae99SBarry Smith n20 = rank + m*(n-1) - (m-1) + m*n; 70247c6ae99SBarry Smith } 70347c6ae99SBarry Smith 704ce00eea3SSatish Balay if ((xe==M) && (ye==N)) { /* Assume an edge, not corner */ 70547c6ae99SBarry Smith n8 = rank - m*n +1 - m*n; 70647c6ae99SBarry Smith n17 = rank - m*n +1; 70747c6ae99SBarry Smith n26 = rank - m*n +1 + m*n; 70847c6ae99SBarry Smith } 70947c6ae99SBarry Smith 71047c6ae99SBarry Smith if ((ys==0) && (zs==0)) { /* Assume an edge, not corner */ 71147c6ae99SBarry Smith n0 = size - m + rank -1; 71247c6ae99SBarry Smith n1 = size - m + rank; 71347c6ae99SBarry Smith n2 = size - m + rank +1; 71447c6ae99SBarry Smith } 71547c6ae99SBarry Smith 71647c6ae99SBarry Smith if ((ys==0) && (ze==P)) { /* Assume an edge, not corner */ 71747c6ae99SBarry Smith n18 = m*n - (size - rank) + m*(n-1) -1; 71847c6ae99SBarry Smith n19 = m*n - (size - rank) + m*(n-1); 71947c6ae99SBarry Smith n20 = m*n - (size - rank) + m*(n-1) +1; 72047c6ae99SBarry Smith } 72147c6ae99SBarry Smith 72247c6ae99SBarry Smith if ((ye==N) && (zs==0)) { /* Assume an edge, not corner */ 72347c6ae99SBarry Smith n6 = size - (m*n-rank) - m * (n-1) -1; 72447c6ae99SBarry Smith n7 = size - (m*n-rank) - m * (n-1); 72547c6ae99SBarry Smith n8 = size - (m*n-rank) - m * (n-1) +1; 72647c6ae99SBarry Smith } 72747c6ae99SBarry Smith 72847c6ae99SBarry Smith if ((ye==N) && (ze==P)) { /* Assume an edge, not corner */ 72947c6ae99SBarry Smith n24 = rank - (size-m) -1; 73047c6ae99SBarry Smith n25 = rank - (size-m); 73147c6ae99SBarry Smith n26 = rank - (size-m) +1; 73247c6ae99SBarry Smith } 73347c6ae99SBarry Smith 73447c6ae99SBarry Smith /* Check for Corners */ 73547c6ae99SBarry Smith if ((xs==0) && (ys==0) && (zs==0)) { n0 = size -1;} 73647c6ae99SBarry Smith if ((xs==0) && (ys==0) && (ze==P)) { n18 = m*n-1;} 73747c6ae99SBarry Smith if ((xs==0) && (ye==N) && (zs==0)) { n6 = (size-1)-m*(n-1);} 73847c6ae99SBarry Smith if ((xs==0) && (ye==N) && (ze==P)) { n24 = m-1;} 739ce00eea3SSatish Balay if ((xe==M) && (ys==0) && (zs==0)) { n2 = size-m;} 740ce00eea3SSatish Balay if ((xe==M) && (ys==0) && (ze==P)) { n20 = m*n-m;} 741ce00eea3SSatish Balay if ((xe==M) && (ye==N) && (zs==0)) { n8 = size-m*n;} 742ce00eea3SSatish Balay if ((xe==M) && (ye==N) && (ze==P)) { n26 = 0;} 74347c6ae99SBarry Smith 74447c6ae99SBarry Smith /* Check for when not X,Y, and Z Periodic */ 74547c6ae99SBarry Smith 74647c6ae99SBarry Smith /* If not X periodic */ 7471321219cSEthan Coon if (bx != DMDA_BOUNDARY_PERIODIC) { 74847c6ae99SBarry Smith if (xs==0) {n0 = n3 = n6 = n9 = n12 = n15 = n18 = n21 = n24 = -2;} 749ce00eea3SSatish Balay if (xe==M) {n2 = n5 = n8 = n11 = n14 = n17 = n20 = n23 = n26 = -2;} 75047c6ae99SBarry Smith } 75147c6ae99SBarry Smith 75247c6ae99SBarry Smith /* If not Y periodic */ 7531321219cSEthan Coon if (by != DMDA_BOUNDARY_PERIODIC) { 75447c6ae99SBarry Smith if (ys==0) {n0 = n1 = n2 = n9 = n10 = n11 = n18 = n19 = n20 = -2;} 75547c6ae99SBarry Smith if (ye==N) {n6 = n7 = n8 = n15 = n16 = n17 = n24 = n25 = n26 = -2;} 75647c6ae99SBarry Smith } 75747c6ae99SBarry Smith 75847c6ae99SBarry Smith /* If not Z periodic */ 7591321219cSEthan Coon if (bz != DMDA_BOUNDARY_PERIODIC) { 76047c6ae99SBarry Smith if (zs==0) {n0 = n1 = n2 = n3 = n4 = n5 = n6 = n7 = n8 = -2;} 76147c6ae99SBarry Smith if (ze==P) {n18 = n19 = n20 = n21 = n22 = n23 = n24 = n25 = n26 = -2;} 76247c6ae99SBarry Smith } 76347c6ae99SBarry Smith 76447c6ae99SBarry Smith ierr = PetscMalloc(27*sizeof(PetscInt),&dd->neighbors);CHKERRQ(ierr); 76547c6ae99SBarry Smith dd->neighbors[0] = n0; 76647c6ae99SBarry Smith dd->neighbors[1] = n1; 76747c6ae99SBarry Smith dd->neighbors[2] = n2; 76847c6ae99SBarry Smith dd->neighbors[3] = n3; 76947c6ae99SBarry Smith dd->neighbors[4] = n4; 77047c6ae99SBarry Smith dd->neighbors[5] = n5; 77147c6ae99SBarry Smith dd->neighbors[6] = n6; 77247c6ae99SBarry Smith dd->neighbors[7] = n7; 77347c6ae99SBarry Smith dd->neighbors[8] = n8; 77447c6ae99SBarry Smith dd->neighbors[9] = n9; 77547c6ae99SBarry Smith dd->neighbors[10] = n10; 77647c6ae99SBarry Smith dd->neighbors[11] = n11; 77747c6ae99SBarry Smith dd->neighbors[12] = n12; 77847c6ae99SBarry Smith dd->neighbors[13] = rank; 77947c6ae99SBarry Smith dd->neighbors[14] = n14; 78047c6ae99SBarry Smith dd->neighbors[15] = n15; 78147c6ae99SBarry Smith dd->neighbors[16] = n16; 78247c6ae99SBarry Smith dd->neighbors[17] = n17; 78347c6ae99SBarry Smith dd->neighbors[18] = n18; 78447c6ae99SBarry Smith dd->neighbors[19] = n19; 78547c6ae99SBarry Smith dd->neighbors[20] = n20; 78647c6ae99SBarry Smith dd->neighbors[21] = n21; 78747c6ae99SBarry Smith dd->neighbors[22] = n22; 78847c6ae99SBarry Smith dd->neighbors[23] = n23; 78947c6ae99SBarry Smith dd->neighbors[24] = n24; 79047c6ae99SBarry Smith dd->neighbors[25] = n25; 79147c6ae99SBarry Smith dd->neighbors[26] = n26; 79247c6ae99SBarry Smith 79347c6ae99SBarry Smith /* If star stencil then delete the corner neighbors */ 79488661749SPeter Brune if (stencil_type == DMDA_STENCIL_STAR && o == 0) { 79547c6ae99SBarry Smith /* save information about corner neighbors */ 79647c6ae99SBarry Smith sn0 = n0; sn1 = n1; sn2 = n2; sn3 = n3; sn5 = n5; sn6 = n6; sn7 = n7; 79747c6ae99SBarry Smith sn8 = n8; sn9 = n9; sn11 = n11; sn15 = n15; sn17 = n17; sn18 = n18; 79847c6ae99SBarry Smith sn19 = n19; sn20 = n20; sn21 = n21; sn23 = n23; sn24 = n24; sn25 = n25; 79947c6ae99SBarry Smith sn26 = n26; 80047c6ae99SBarry Smith n0 = n1 = n2 = n3 = n5 = n6 = n7 = n8 = n9 = n11 = 80147c6ae99SBarry Smith n15 = n17 = n18 = n19 = n20 = n21 = n23 = n24 = n25 = n26 = -1; 80247c6ae99SBarry Smith } 80347c6ae99SBarry Smith 80447c6ae99SBarry Smith 80547c6ae99SBarry Smith ierr = PetscMalloc((Xe-Xs)*(Ye-Ys)*(Ze-Zs)*sizeof(PetscInt),&idx);CHKERRQ(ierr); 80647c6ae99SBarry Smith ierr = PetscLogObjectMemory(da,(Xe-Xs)*(Ye-Ys)*(Ze-Zs)*sizeof(PetscInt));CHKERRQ(ierr); 80747c6ae99SBarry Smith 80847c6ae99SBarry Smith nn = 0; 80947c6ae99SBarry Smith /* Bottom Level */ 81047c6ae99SBarry Smith for (k=0; k<s_z; k++) { 81147c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 81247c6ae99SBarry Smith if (n0 >= 0) { /* left below */ 813ce00eea3SSatish Balay x_t = lx[n0 % m]; 81447c6ae99SBarry Smith y_t = ly[(n0 % (m*n))/m]; 81547c6ae99SBarry Smith z_t = lz[n0 / (m*n)]; 81647c6ae99SBarry Smith s_t = bases[n0] + x_t*y_t*z_t - (s_y-i)*x_t - s_x - (s_z-k-1)*x_t*y_t; 8176f951b95Secoon if (twod && (s_t < 0)) {s_t = bases[n0] + x_t*y_t*z_t - (s_y-i)*x_t - s_x;} /* 2D case */ 81847c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 81947c6ae99SBarry Smith } 82047c6ae99SBarry Smith if (n1 >= 0) { /* directly below */ 82147c6ae99SBarry Smith x_t = x; 82247c6ae99SBarry Smith y_t = ly[(n1 % (m*n))/m]; 82347c6ae99SBarry Smith z_t = lz[n1 / (m*n)]; 82447c6ae99SBarry Smith s_t = bases[n1] + x_t*y_t*z_t - (s_y+1-i)*x_t - (s_z-k-1)*x_t*y_t; 8256f951b95Secoon if (twod && (s_t < 0)) {s_t = bases[n1] + x_t*y_t*z_t - (s_y+1-i)*x_t;} /* 2D case */ 82647c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 82747c6ae99SBarry Smith } 82847c6ae99SBarry Smith if (n2 >= 0) { /* right below */ 829ce00eea3SSatish Balay x_t = lx[n2 % m]; 83047c6ae99SBarry Smith y_t = ly[(n2 % (m*n))/m]; 83147c6ae99SBarry Smith z_t = lz[n2 / (m*n)]; 83247c6ae99SBarry Smith s_t = bases[n2] + x_t*y_t*z_t - (s_y+1-i)*x_t - (s_z-k-1)*x_t*y_t; 8336f951b95Secoon if (twod && (s_t < 0)) {s_t = bases[n2] + x_t*y_t*z_t - (s_y+1-i)*x_t;} /* 2D case */ 83447c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 83547c6ae99SBarry Smith } 83647c6ae99SBarry Smith } 83747c6ae99SBarry Smith 83847c6ae99SBarry Smith for (i=0; i<y; i++) { 83947c6ae99SBarry Smith if (n3 >= 0) { /* directly left */ 840ce00eea3SSatish Balay x_t = lx[n3 % m]; 84147c6ae99SBarry Smith y_t = y; 84247c6ae99SBarry Smith z_t = lz[n3 / (m*n)]; 84347c6ae99SBarry Smith s_t = bases[n3] + (i+1)*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8446f951b95Secoon if (twod && (s_t < 0)) {s_t = bases[n3] + (i+1)*x_t - s_x + x_t*y_t*z_t - x_t*y_t;} /* 2D case */ 84547c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 84647c6ae99SBarry Smith } 84747c6ae99SBarry Smith 84847c6ae99SBarry Smith if (n4 >= 0) { /* middle */ 84947c6ae99SBarry Smith x_t = x; 85047c6ae99SBarry Smith y_t = y; 85147c6ae99SBarry Smith z_t = lz[n4 / (m*n)]; 85247c6ae99SBarry Smith s_t = bases[n4] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8536f951b95Secoon if (twod && (s_t < 0)) {s_t = bases[n4] + i*x_t + x_t*y_t*z_t - x_t*y_t;} /* 2D case */ 85447c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 855*c2859e5eSBarry Smith } else if (bz == DMDA_BOUNDARY_MIRROR) { 856*c2859e5eSBarry Smith for (j=0; j<x; j++) { idx[nn++] = 0;} 85747c6ae99SBarry Smith } 85847c6ae99SBarry Smith 85947c6ae99SBarry Smith if (n5 >= 0) { /* directly right */ 860ce00eea3SSatish Balay x_t = lx[n5 % m]; 86147c6ae99SBarry Smith y_t = y; 86247c6ae99SBarry Smith z_t = lz[n5 / (m*n)]; 86347c6ae99SBarry Smith s_t = bases[n5] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8646f951b95Secoon if (twod && (s_t < 0)) {s_t = bases[n5] + i*x_t + x_t*y_t*z_t - x_t*y_t;} /* 2D case */ 86547c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 86647c6ae99SBarry Smith } 86747c6ae99SBarry Smith } 86847c6ae99SBarry Smith 86947c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 87047c6ae99SBarry Smith if (n6 >= 0) { /* left above */ 871ce00eea3SSatish Balay x_t = lx[n6 % m]; 87247c6ae99SBarry Smith y_t = ly[(n6 % (m*n))/m]; 87347c6ae99SBarry Smith z_t = lz[n6 / (m*n)]; 87447c6ae99SBarry Smith s_t = bases[n6] + i*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8756f951b95Secoon if (twod && (s_t < 0)) {s_t = bases[n6] + i*x_t - s_x + x_t*y_t*z_t - x_t*y_t;} /* 2D case */ 87647c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 87747c6ae99SBarry Smith } 87847c6ae99SBarry Smith if (n7 >= 0) { /* directly above */ 87947c6ae99SBarry Smith x_t = x; 88047c6ae99SBarry Smith y_t = ly[(n7 % (m*n))/m]; 88147c6ae99SBarry Smith z_t = lz[n7 / (m*n)]; 88247c6ae99SBarry Smith s_t = bases[n7] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8836f951b95Secoon if (twod && (s_t < 0)) {s_t = bases[n7] + (i-1)*x_t + x_t*y_t*z_t - x_t*y_t;} /* 2D case */ 88447c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 88547c6ae99SBarry Smith } 88647c6ae99SBarry Smith if (n8 >= 0) { /* right above */ 887ce00eea3SSatish Balay x_t = lx[n8 % m]; 88847c6ae99SBarry Smith y_t = ly[(n8 % (m*n))/m]; 88947c6ae99SBarry Smith z_t = lz[n8 / (m*n)]; 89047c6ae99SBarry Smith s_t = bases[n8] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8916f951b95Secoon if (twod && (s_t < 0)) {s_t = bases[n8] + (i-1)*x_t + x_t*y_t*z_t - x_t*y_t;} /* 2D case */ 89247c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 89347c6ae99SBarry Smith } 89447c6ae99SBarry Smith } 89547c6ae99SBarry Smith } 89647c6ae99SBarry Smith 89747c6ae99SBarry Smith /* Middle Level */ 89847c6ae99SBarry Smith for (k=0; k<z; k++) { 89947c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 90047c6ae99SBarry Smith if (n9 >= 0) { /* left below */ 901ce00eea3SSatish Balay x_t = lx[n9 % m]; 90247c6ae99SBarry Smith y_t = ly[(n9 % (m*n))/m]; 90347c6ae99SBarry Smith /* z_t = z; */ 90447c6ae99SBarry Smith s_t = bases[n9] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t; 90547c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 90647c6ae99SBarry Smith } 90747c6ae99SBarry Smith if (n10 >= 0) { /* directly below */ 90847c6ae99SBarry Smith x_t = x; 90947c6ae99SBarry Smith y_t = ly[(n10 % (m*n))/m]; 91047c6ae99SBarry Smith /* z_t = z; */ 91147c6ae99SBarry Smith s_t = bases[n10] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 91247c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 913*c2859e5eSBarry Smith } else if (by == DMDA_BOUNDARY_MIRROR) { 914*c2859e5eSBarry Smith for (j=0; j<x; j++) { idx[nn++] = 0;} 91547c6ae99SBarry Smith } 91647c6ae99SBarry Smith if (n11 >= 0) { /* right below */ 917ce00eea3SSatish Balay x_t = lx[n11 % m]; 91847c6ae99SBarry Smith y_t = ly[(n11 % (m*n))/m]; 91947c6ae99SBarry Smith /* z_t = z; */ 92047c6ae99SBarry Smith s_t = bases[n11] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 92147c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 92247c6ae99SBarry Smith } 92347c6ae99SBarry Smith } 92447c6ae99SBarry Smith 92547c6ae99SBarry Smith for (i=0; i<y; i++) { 92647c6ae99SBarry Smith if (n12 >= 0) { /* directly left */ 927ce00eea3SSatish Balay x_t = lx[n12 % m]; 92847c6ae99SBarry Smith y_t = y; 92947c6ae99SBarry Smith /* z_t = z; */ 93047c6ae99SBarry Smith s_t = bases[n12] + (i+1)*x_t - s_x + k*x_t*y_t; 93147c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 932*c2859e5eSBarry Smith } else if (bx == DMDA_BOUNDARY_MIRROR) { 933*c2859e5eSBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = 0;} 93447c6ae99SBarry Smith } 93547c6ae99SBarry Smith 93647c6ae99SBarry Smith /* Interior */ 93747c6ae99SBarry Smith s_t = bases[rank] + i*x + k*x*y; 93847c6ae99SBarry Smith for (j=0; j<x; j++) { idx[nn++] = s_t++;} 93947c6ae99SBarry Smith 94047c6ae99SBarry Smith if (n14 >= 0) { /* directly right */ 941ce00eea3SSatish Balay x_t = lx[n14 % m]; 94247c6ae99SBarry Smith y_t = y; 94347c6ae99SBarry Smith /* z_t = z; */ 94447c6ae99SBarry Smith s_t = bases[n14] + i*x_t + k*x_t*y_t; 94547c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 946*c2859e5eSBarry Smith } else if (bx == DMDA_BOUNDARY_MIRROR) { 947*c2859e5eSBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = 0;} 94847c6ae99SBarry Smith } 94947c6ae99SBarry Smith } 95047c6ae99SBarry Smith 95147c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 95247c6ae99SBarry Smith if (n15 >= 0) { /* left above */ 953ce00eea3SSatish Balay x_t = lx[n15 % m]; 95447c6ae99SBarry Smith y_t = ly[(n15 % (m*n))/m]; 95547c6ae99SBarry Smith /* z_t = z; */ 95647c6ae99SBarry Smith s_t = bases[n15] + i*x_t - s_x + k*x_t*y_t; 95747c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 95847c6ae99SBarry Smith } 95947c6ae99SBarry Smith if (n16 >= 0) { /* directly above */ 96047c6ae99SBarry Smith x_t = x; 96147c6ae99SBarry Smith y_t = ly[(n16 % (m*n))/m]; 96247c6ae99SBarry Smith /* z_t = z; */ 96347c6ae99SBarry Smith s_t = bases[n16] + (i-1)*x_t + k*x_t*y_t; 96447c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 965*c2859e5eSBarry Smith } else if (by == DMDA_BOUNDARY_MIRROR) { 966*c2859e5eSBarry Smith for (j=0; j<x; j++) { idx[nn++] = 0;} 96747c6ae99SBarry Smith } 96847c6ae99SBarry Smith if (n17 >= 0) { /* right above */ 969ce00eea3SSatish Balay x_t = lx[n17 % m]; 97047c6ae99SBarry Smith y_t = ly[(n17 % (m*n))/m]; 97147c6ae99SBarry Smith /* z_t = z; */ 97247c6ae99SBarry Smith s_t = bases[n17] + (i-1)*x_t + k*x_t*y_t; 97347c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 97447c6ae99SBarry Smith } 97547c6ae99SBarry Smith } 97647c6ae99SBarry Smith } 97747c6ae99SBarry Smith 97847c6ae99SBarry Smith /* Upper Level */ 97947c6ae99SBarry Smith for (k=0; k<s_z; k++) { 98047c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 98147c6ae99SBarry Smith if (n18 >= 0) { /* left below */ 982ce00eea3SSatish Balay x_t = lx[n18 % m]; 98347c6ae99SBarry Smith y_t = ly[(n18 % (m*n))/m]; 98447c6ae99SBarry Smith /* z_t = lz[n18 / (m*n)]; */ 98547c6ae99SBarry Smith s_t = bases[n18] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t; 9866f951b95Secoon if (twod && (s_t >= M*N*P)) {s_t = bases[n18] - (s_y-i)*x_t -s_x + x_t*y_t;} /* 2d case */ 98747c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 98847c6ae99SBarry Smith } 98947c6ae99SBarry Smith if (n19 >= 0) { /* directly below */ 99047c6ae99SBarry Smith x_t = x; 99147c6ae99SBarry Smith y_t = ly[(n19 % (m*n))/m]; 99247c6ae99SBarry Smith /* z_t = lz[n19 / (m*n)]; */ 99347c6ae99SBarry Smith s_t = bases[n19] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 9946f951b95Secoon if (twod && (s_t >= M*N*P)) {s_t = bases[n19] - (s_y+1-i)*x_t + x_t*y_t;} /* 2d case */ 99547c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 99647c6ae99SBarry Smith } 99747c6ae99SBarry Smith if (n20 >= 0) { /* right below */ 998ce00eea3SSatish Balay x_t = lx[n20 % m]; 99947c6ae99SBarry Smith y_t = ly[(n20 % (m*n))/m]; 100047c6ae99SBarry Smith /* z_t = lz[n20 / (m*n)]; */ 100147c6ae99SBarry Smith s_t = bases[n20] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 10026f951b95Secoon if (twod && (s_t >= M*N*P)) {s_t = bases[n20] - (s_y+1-i)*x_t + x_t*y_t;} /* 2d case */ 100347c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 100447c6ae99SBarry Smith } 100547c6ae99SBarry Smith } 100647c6ae99SBarry Smith 100747c6ae99SBarry Smith for (i=0; i<y; i++) { 100847c6ae99SBarry Smith if (n21 >= 0) { /* directly left */ 1009ce00eea3SSatish Balay x_t = lx[n21 % m]; 101047c6ae99SBarry Smith y_t = y; 101147c6ae99SBarry Smith /* z_t = lz[n21 / (m*n)]; */ 101247c6ae99SBarry Smith s_t = bases[n21] + (i+1)*x_t - s_x + k*x_t*y_t; 10136f951b95Secoon if (twod && (s_t >= M*N*P)) {s_t = bases[n21] + (i+1)*x_t - s_x;} /* 2d case */ 101447c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 101547c6ae99SBarry Smith } 101647c6ae99SBarry Smith 101747c6ae99SBarry Smith if (n22 >= 0) { /* middle */ 101847c6ae99SBarry Smith x_t = x; 101947c6ae99SBarry Smith y_t = y; 102047c6ae99SBarry Smith /* z_t = lz[n22 / (m*n)]; */ 102147c6ae99SBarry Smith s_t = bases[n22] + i*x_t + k*x_t*y_t; 10226f951b95Secoon if (twod && (s_t >= M*N*P)) {s_t = bases[n22] + i*x_t;} /* 2d case */ 102347c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 1024*c2859e5eSBarry Smith } else if (bz == DMDA_BOUNDARY_MIRROR) { 1025*c2859e5eSBarry Smith for (j=0; j<x; j++) { idx[nn++] = 0;} 102647c6ae99SBarry Smith } 102747c6ae99SBarry Smith 102847c6ae99SBarry Smith if (n23 >= 0) { /* directly right */ 1029ce00eea3SSatish Balay x_t = lx[n23 % m]; 103047c6ae99SBarry Smith y_t = y; 103147c6ae99SBarry Smith /* z_t = lz[n23 / (m*n)]; */ 103247c6ae99SBarry Smith s_t = bases[n23] + i*x_t + k*x_t*y_t; 10336f951b95Secoon if (twod && (s_t >= M*N*P)) {s_t = bases[n23] + i*x_t;} /* 2d case */ 103447c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 103547c6ae99SBarry Smith } 103647c6ae99SBarry Smith } 103747c6ae99SBarry Smith 103847c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 103947c6ae99SBarry Smith if (n24 >= 0) { /* left above */ 1040ce00eea3SSatish Balay x_t = lx[n24 % m]; 104147c6ae99SBarry Smith y_t = ly[(n24 % (m*n))/m]; 104247c6ae99SBarry Smith /* z_t = lz[n24 / (m*n)]; */ 104347c6ae99SBarry Smith s_t = bases[n24] + i*x_t - s_x + k*x_t*y_t; 10446f951b95Secoon if (twod && (s_t >= M*N*P)) {s_t = bases[n24] + i*x_t - s_x;} /* 2d case */ 104547c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 104647c6ae99SBarry Smith } 104747c6ae99SBarry Smith if (n25 >= 0) { /* directly above */ 104847c6ae99SBarry Smith x_t = x; 104947c6ae99SBarry Smith y_t = ly[(n25 % (m*n))/m]; 105047c6ae99SBarry Smith /* z_t = lz[n25 / (m*n)]; */ 105147c6ae99SBarry Smith s_t = bases[n25] + (i-1)*x_t + k*x_t*y_t; 10526f951b95Secoon if (twod && (s_t >= M*N*P)) {s_t = bases[n25] + (i-1)*x_t;} /* 2d case */ 105347c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 105447c6ae99SBarry Smith } 105547c6ae99SBarry Smith if (n26 >= 0) { /* right above */ 1056ce00eea3SSatish Balay x_t = lx[n26 % m]; 105747c6ae99SBarry Smith y_t = ly[(n26 % (m*n))/m]; 105847c6ae99SBarry Smith /* z_t = lz[n26 / (m*n)]; */ 105947c6ae99SBarry Smith s_t = bases[n26] + (i-1)*x_t + k*x_t*y_t; 10606f951b95Secoon if (twod && (s_t >= M*N*P)) {s_t = bases[n26] + (i-1)*x_t;} /* 2d case */ 106147c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 106247c6ae99SBarry Smith } 106347c6ae99SBarry Smith } 106447c6ae99SBarry Smith } 1065ce00eea3SSatish Balay 1066ce00eea3SSatish Balay ierr = ISCreateBlock(comm,dof,nn,idx,PETSC_COPY_VALUES,&from);CHKERRQ(ierr); 106747c6ae99SBarry Smith ierr = VecScatterCreate(global,from,local,to,>ol);CHKERRQ(ierr); 106847c6ae99SBarry Smith ierr = PetscLogObjectParent(da,gtol);CHKERRQ(ierr); 1069fcfd50ebSBarry Smith ierr = ISDestroy(&to);CHKERRQ(ierr); 1070fcfd50ebSBarry Smith ierr = ISDestroy(&from);CHKERRQ(ierr); 107147c6ae99SBarry Smith 107288661749SPeter Brune if (stencil_type == DMDA_STENCIL_STAR && o == 0) { 107347c6ae99SBarry Smith n0 = sn0; n1 = sn1; n2 = sn2; n3 = sn3; n5 = sn5; n6 = sn6; n7 = sn7; 107447c6ae99SBarry Smith n8 = sn8; n9 = sn9; n11 = sn11; n15 = sn15; n17 = sn17; n18 = sn18; 107547c6ae99SBarry Smith n19 = sn19; n20 = sn20; n21 = sn21; n23 = sn23; n24 = sn24; n25 = sn25; 107647c6ae99SBarry Smith n26 = sn26; 1077ce00eea3SSatish Balay } 107847c6ae99SBarry Smith 107988661749SPeter Brune if (((stencil_type == DMDA_STENCIL_STAR) || 10801321219cSEthan Coon (bx != DMDA_BOUNDARY_PERIODIC && bx) || 10811321219cSEthan Coon (by != DMDA_BOUNDARY_PERIODIC && by) || 108288661749SPeter Brune (bz != DMDA_BOUNDARY_PERIODIC && bz)) && o == 0) { 1083ce00eea3SSatish Balay /* 1084ce00eea3SSatish Balay Recompute the local to global mappings, this time keeping the 1085ce00eea3SSatish Balay information about the cross corner processor numbers. 1086ce00eea3SSatish Balay */ 108747c6ae99SBarry Smith nn = 0; 108847c6ae99SBarry Smith /* Bottom Level */ 108947c6ae99SBarry Smith for (k=0; k<s_z; k++) { 109047c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 109147c6ae99SBarry Smith if (n0 >= 0) { /* left below */ 1092ce00eea3SSatish Balay x_t = lx[n0 % m]; 109347c6ae99SBarry Smith y_t = ly[(n0 % (m*n))/m]; 109447c6ae99SBarry Smith z_t = lz[n0 / (m*n)]; 109547c6ae99SBarry Smith s_t = bases[n0] + x_t*y_t*z_t - (s_y-i)*x_t - s_x - (s_z-k-1)*x_t*y_t; 109647c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1097ce00eea3SSatish Balay } else if (Xs-xs < 0 && Ys-ys < 0 && Zs-zs < 0) { 1098ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 109947c6ae99SBarry Smith } 110047c6ae99SBarry Smith if (n1 >= 0) { /* directly below */ 110147c6ae99SBarry Smith x_t = x; 110247c6ae99SBarry Smith y_t = ly[(n1 % (m*n))/m]; 110347c6ae99SBarry Smith z_t = lz[n1 / (m*n)]; 110447c6ae99SBarry Smith s_t = bases[n1] + x_t*y_t*z_t - (s_y+1-i)*x_t - (s_z-k-1)*x_t*y_t; 110547c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 1106ce00eea3SSatish Balay } else if (Ys-ys < 0 && Zs-zs < 0) { 1107ce00eea3SSatish Balay for (j=0; j<x; j++) { idx[nn++] = -1;} 110847c6ae99SBarry Smith } 110947c6ae99SBarry Smith if (n2 >= 0) { /* right below */ 1110ce00eea3SSatish Balay x_t = lx[n2 % m]; 111147c6ae99SBarry Smith y_t = ly[(n2 % (m*n))/m]; 111247c6ae99SBarry Smith z_t = lz[n2 / (m*n)]; 111347c6ae99SBarry Smith s_t = bases[n2] + x_t*y_t*z_t - (s_y+1-i)*x_t - (s_z-k-1)*x_t*y_t; 111447c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1115ce00eea3SSatish Balay } else if (xe-Xe < 0 && Ys-ys < 0 && Zs-zs < 0) { 1116ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 111747c6ae99SBarry Smith } 111847c6ae99SBarry Smith } 111947c6ae99SBarry Smith 112047c6ae99SBarry Smith for (i=0; i<y; i++) { 112147c6ae99SBarry Smith if (n3 >= 0) { /* directly left */ 1122ce00eea3SSatish Balay x_t = lx[n3 % m]; 112347c6ae99SBarry Smith y_t = y; 112447c6ae99SBarry Smith z_t = lz[n3 / (m*n)]; 112547c6ae99SBarry Smith s_t = bases[n3] + (i+1)*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 112647c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1127ce00eea3SSatish Balay } else if (Xs-xs < 0 && Zs-zs < 0) { 1128ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 112947c6ae99SBarry Smith } 113047c6ae99SBarry Smith 113147c6ae99SBarry Smith if (n4 >= 0) { /* middle */ 113247c6ae99SBarry Smith x_t = x; 113347c6ae99SBarry Smith y_t = y; 113447c6ae99SBarry Smith z_t = lz[n4 / (m*n)]; 113547c6ae99SBarry Smith s_t = bases[n4] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 113647c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 1137ce00eea3SSatish Balay } else if (Zs-zs < 0) { 1138*c2859e5eSBarry Smith if (bz == DMDA_BOUNDARY_MIRROR) { 1139*c2859e5eSBarry Smith for (j=0; j<x; j++) { idx[nn++] = 0;} 1140*c2859e5eSBarry Smith } else { 1141ce00eea3SSatish Balay for (j=0; j<x; j++) { idx[nn++] = -1;} 114247c6ae99SBarry Smith } 1143*c2859e5eSBarry Smith } 114447c6ae99SBarry Smith 114547c6ae99SBarry Smith if (n5 >= 0) { /* directly right */ 1146ce00eea3SSatish Balay x_t = lx[n5 % m]; 114747c6ae99SBarry Smith y_t = y; 114847c6ae99SBarry Smith z_t = lz[n5 / (m*n)]; 114947c6ae99SBarry Smith s_t = bases[n5] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 115047c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1151ce00eea3SSatish Balay } else if (xe-Xe < 0 && Zs-zs < 0) { 1152ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 115347c6ae99SBarry Smith } 115447c6ae99SBarry Smith } 115547c6ae99SBarry Smith 115647c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 115747c6ae99SBarry Smith if (n6 >= 0) { /* left above */ 1158ce00eea3SSatish Balay x_t = lx[n6 % m]; 115947c6ae99SBarry Smith y_t = ly[(n6 % (m*n))/m]; 116047c6ae99SBarry Smith z_t = lz[n6 / (m*n)]; 116147c6ae99SBarry Smith s_t = bases[n6] + i*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 116247c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1163ce00eea3SSatish Balay } else if (Xs-xs < 0 && ye-Ye < 0 && Zs-zs < 0) { 1164ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 116547c6ae99SBarry Smith } 116647c6ae99SBarry Smith if (n7 >= 0) { /* directly above */ 116747c6ae99SBarry Smith x_t = x; 116847c6ae99SBarry Smith y_t = ly[(n7 % (m*n))/m]; 116947c6ae99SBarry Smith z_t = lz[n7 / (m*n)]; 117047c6ae99SBarry Smith s_t = bases[n7] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 117147c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 1172ce00eea3SSatish Balay } else if (ye-Ye < 0 && Zs-zs < 0) { 1173ce00eea3SSatish Balay for (j=0; j<x; j++) { idx[nn++] = -1;} 117447c6ae99SBarry Smith } 117547c6ae99SBarry Smith if (n8 >= 0) { /* right above */ 1176ce00eea3SSatish Balay x_t = lx[n8 % m]; 117747c6ae99SBarry Smith y_t = ly[(n8 % (m*n))/m]; 117847c6ae99SBarry Smith z_t = lz[n8 / (m*n)]; 117947c6ae99SBarry Smith s_t = bases[n8] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 118047c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1181ce00eea3SSatish Balay } else if (xe-Xe < 0 && ye-Ye < 0 && Zs-zs < 0) { 1182ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 118347c6ae99SBarry Smith } 118447c6ae99SBarry Smith } 118547c6ae99SBarry Smith } 118647c6ae99SBarry Smith 118747c6ae99SBarry Smith /* Middle Level */ 118847c6ae99SBarry Smith for (k=0; k<z; k++) { 118947c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 119047c6ae99SBarry Smith if (n9 >= 0) { /* left below */ 1191ce00eea3SSatish Balay x_t = lx[n9 % m]; 119247c6ae99SBarry Smith y_t = ly[(n9 % (m*n))/m]; 119347c6ae99SBarry Smith /* z_t = z; */ 119447c6ae99SBarry Smith s_t = bases[n9] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t; 119547c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1196ce00eea3SSatish Balay } else if (Xs-xs < 0 && Ys-ys < 0) { 1197ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 119847c6ae99SBarry Smith } 119947c6ae99SBarry Smith if (n10 >= 0) { /* directly below */ 120047c6ae99SBarry Smith x_t = x; 120147c6ae99SBarry Smith y_t = ly[(n10 % (m*n))/m]; 120247c6ae99SBarry Smith /* z_t = z; */ 120347c6ae99SBarry Smith s_t = bases[n10] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 120447c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 1205ce00eea3SSatish Balay } else if (Ys-ys < 0) { 1206*c2859e5eSBarry Smith if (by == DMDA_BOUNDARY_MIRROR) { 1207ce00eea3SSatish Balay for (j=0; j<x; j++) { idx[nn++] = -1;} 1208*c2859e5eSBarry Smith } else { 1209*c2859e5eSBarry Smith for (j=0; j<x; j++) { idx[nn++] = -1;} 1210*c2859e5eSBarry Smith } 121147c6ae99SBarry Smith } 121247c6ae99SBarry Smith if (n11 >= 0) { /* right below */ 1213ce00eea3SSatish Balay x_t = lx[n11 % m]; 121447c6ae99SBarry Smith y_t = ly[(n11 % (m*n))/m]; 121547c6ae99SBarry Smith /* z_t = z; */ 121647c6ae99SBarry Smith s_t = bases[n11] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 121747c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1218ce00eea3SSatish Balay } else if (xe-Xe < 0 && Ys-ys < 0) { 1219ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 122047c6ae99SBarry Smith } 122147c6ae99SBarry Smith } 122247c6ae99SBarry Smith 122347c6ae99SBarry Smith for (i=0; i<y; i++) { 122447c6ae99SBarry Smith if (n12 >= 0) { /* directly left */ 1225ce00eea3SSatish Balay x_t = lx[n12 % m]; 122647c6ae99SBarry Smith y_t = y; 122747c6ae99SBarry Smith /* z_t = z; */ 122847c6ae99SBarry Smith s_t = bases[n12] + (i+1)*x_t - s_x + k*x_t*y_t; 122947c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1230ce00eea3SSatish Balay } else if (Xs-xs < 0) { 1231*c2859e5eSBarry Smith if (bx == DMDA_BOUNDARY_MIRROR) { 1232*c2859e5eSBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = 0;} 1233*c2859e5eSBarry Smith } else { 1234ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 123547c6ae99SBarry Smith } 1236*c2859e5eSBarry Smith } 123747c6ae99SBarry Smith 123847c6ae99SBarry Smith /* Interior */ 123947c6ae99SBarry Smith s_t = bases[rank] + i*x + k*x*y; 124047c6ae99SBarry Smith for (j=0; j<x; j++) { idx[nn++] = s_t++;} 124147c6ae99SBarry Smith 124247c6ae99SBarry Smith if (n14 >= 0) { /* directly right */ 1243ce00eea3SSatish Balay x_t = lx[n14 % m]; 124447c6ae99SBarry Smith y_t = y; 124547c6ae99SBarry Smith /* z_t = z; */ 124647c6ae99SBarry Smith s_t = bases[n14] + i*x_t + k*x_t*y_t; 124747c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1248ce00eea3SSatish Balay } else if (xe-Xe < 0) { 1249*c2859e5eSBarry Smith if (bx == DMDA_BOUNDARY_MIRROR) { 1250*c2859e5eSBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = 0;} 1251*c2859e5eSBarry Smith } else { 1252ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 125347c6ae99SBarry Smith } 125447c6ae99SBarry Smith } 1255*c2859e5eSBarry Smith } 125647c6ae99SBarry Smith 125747c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 125847c6ae99SBarry Smith if (n15 >= 0) { /* left above */ 1259ce00eea3SSatish Balay x_t = lx[n15 % m]; 126047c6ae99SBarry Smith y_t = ly[(n15 % (m*n))/m]; 126147c6ae99SBarry Smith /* z_t = z; */ 126247c6ae99SBarry Smith s_t = bases[n15] + i*x_t - s_x + k*x_t*y_t; 126347c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1264ce00eea3SSatish Balay } else if (Xs-xs < 0 && ye-Ye < 0) { 1265ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 126647c6ae99SBarry Smith } 126747c6ae99SBarry Smith if (n16 >= 0) { /* directly above */ 126847c6ae99SBarry Smith x_t = x; 126947c6ae99SBarry Smith y_t = ly[(n16 % (m*n))/m]; 127047c6ae99SBarry Smith /* z_t = z; */ 127147c6ae99SBarry Smith s_t = bases[n16] + (i-1)*x_t + k*x_t*y_t; 127247c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 1273ce00eea3SSatish Balay } else if (ye-Ye < 0) { 1274*c2859e5eSBarry Smith if (by == DMDA_BOUNDARY_MIRROR) { 1275*c2859e5eSBarry Smith for (j=0; j<x; j++) { idx[nn++] = 0;} 1276*c2859e5eSBarry Smith } else { 1277ce00eea3SSatish Balay for (j=0; j<x; j++) { idx[nn++] = -1;} 127847c6ae99SBarry Smith } 1279*c2859e5eSBarry Smith } 128047c6ae99SBarry Smith if (n17 >= 0) { /* right above */ 1281ce00eea3SSatish Balay x_t = lx[n17 % m]; 128247c6ae99SBarry Smith y_t = ly[(n17 % (m*n))/m]; 128347c6ae99SBarry Smith /* z_t = z; */ 128447c6ae99SBarry Smith s_t = bases[n17] + (i-1)*x_t + k*x_t*y_t; 128547c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1286ce00eea3SSatish Balay } else if (xe-Xe < 0 && ye-Ye < 0) { 1287ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 128847c6ae99SBarry Smith } 128947c6ae99SBarry Smith } 129047c6ae99SBarry Smith } 129147c6ae99SBarry Smith 129247c6ae99SBarry Smith /* Upper Level */ 129347c6ae99SBarry Smith for (k=0; k<s_z; k++) { 129447c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 129547c6ae99SBarry Smith if (n18 >= 0) { /* left below */ 1296ce00eea3SSatish Balay x_t = lx[n18 % m]; 129747c6ae99SBarry Smith y_t = ly[(n18 % (m*n))/m]; 129847c6ae99SBarry Smith /* z_t = lz[n18 / (m*n)]; */ 129947c6ae99SBarry Smith s_t = bases[n18] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t; 130047c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1301ce00eea3SSatish Balay } else if (Xs-xs < 0 && Ys-ys < 0 && ze-Ze < 0) { 1302ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 130347c6ae99SBarry Smith } 130447c6ae99SBarry Smith if (n19 >= 0) { /* directly below */ 130547c6ae99SBarry Smith x_t = x; 130647c6ae99SBarry Smith y_t = ly[(n19 % (m*n))/m]; 130747c6ae99SBarry Smith /* z_t = lz[n19 / (m*n)]; */ 130847c6ae99SBarry Smith s_t = bases[n19] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 130947c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 1310ce00eea3SSatish Balay } else if (Ys-ys < 0 && ze-Ze < 0) { 1311ce00eea3SSatish Balay for (j=0; j<x; j++) { idx[nn++] = -1;} 131247c6ae99SBarry Smith } 131347c6ae99SBarry Smith if (n20 >= 0) { /* right below */ 1314ce00eea3SSatish Balay x_t = lx[n20 % m]; 131547c6ae99SBarry Smith y_t = ly[(n20 % (m*n))/m]; 131647c6ae99SBarry Smith /* z_t = lz[n20 / (m*n)]; */ 131747c6ae99SBarry Smith s_t = bases[n20] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 131847c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1319ce00eea3SSatish Balay } else if (xe-Xe < 0 && Ys-ys < 0 && ze-Ze < 0) { 1320ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 132147c6ae99SBarry Smith } 132247c6ae99SBarry Smith } 132347c6ae99SBarry Smith 132447c6ae99SBarry Smith for (i=0; i<y; i++) { 132547c6ae99SBarry Smith if (n21 >= 0) { /* directly left */ 1326ce00eea3SSatish Balay x_t = lx[n21 % m]; 132747c6ae99SBarry Smith y_t = y; 132847c6ae99SBarry Smith /* z_t = lz[n21 / (m*n)]; */ 132947c6ae99SBarry Smith s_t = bases[n21] + (i+1)*x_t - s_x + k*x_t*y_t; 133047c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1331ce00eea3SSatish Balay } else if (Xs-xs < 0 && ze-Ze < 0) { 1332ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 133347c6ae99SBarry Smith } 133447c6ae99SBarry Smith 133547c6ae99SBarry Smith if (n22 >= 0) { /* middle */ 133647c6ae99SBarry Smith x_t = x; 133747c6ae99SBarry Smith y_t = y; 133847c6ae99SBarry Smith /* z_t = lz[n22 / (m*n)]; */ 133947c6ae99SBarry Smith s_t = bases[n22] + i*x_t + k*x_t*y_t; 134047c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 1341ce00eea3SSatish Balay } else if (ze-Ze < 0) { 1342*c2859e5eSBarry Smith if (bz == DMDA_BOUNDARY_MIRROR) { 1343*c2859e5eSBarry Smith for (j=0; j<x; j++) { idx[nn++] = 0;} 1344*c2859e5eSBarry Smith } else { 1345ce00eea3SSatish Balay for (j=0; j<x; j++) { idx[nn++] = -1;} 134647c6ae99SBarry Smith } 1347*c2859e5eSBarry Smith } 134847c6ae99SBarry Smith 134947c6ae99SBarry Smith if (n23 >= 0) { /* directly right */ 1350ce00eea3SSatish Balay x_t = lx[n23 % m]; 135147c6ae99SBarry Smith y_t = y; 135247c6ae99SBarry Smith /* z_t = lz[n23 / (m*n)]; */ 135347c6ae99SBarry Smith s_t = bases[n23] + i*x_t + k*x_t*y_t; 135447c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1355ce00eea3SSatish Balay } else if (xe-Xe < 0 && ze-Ze < 0) { 1356ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 135747c6ae99SBarry Smith } 135847c6ae99SBarry Smith } 135947c6ae99SBarry Smith 136047c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 136147c6ae99SBarry Smith if (n24 >= 0) { /* left above */ 1362ce00eea3SSatish Balay x_t = lx[n24 % m]; 136347c6ae99SBarry Smith y_t = ly[(n24 % (m*n))/m]; 136447c6ae99SBarry Smith /* z_t = lz[n24 / (m*n)]; */ 136547c6ae99SBarry Smith s_t = bases[n24] + i*x_t - s_x + k*x_t*y_t; 136647c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1367ce00eea3SSatish Balay } else if (Xs-xs < 0 && ye-Ye < 0 && ze-Ze < 0) { 1368ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 136947c6ae99SBarry Smith } 137047c6ae99SBarry Smith if (n25 >= 0) { /* directly above */ 137147c6ae99SBarry Smith x_t = x; 137247c6ae99SBarry Smith y_t = ly[(n25 % (m*n))/m]; 137347c6ae99SBarry Smith /* z_t = lz[n25 / (m*n)]; */ 137447c6ae99SBarry Smith s_t = bases[n25] + (i-1)*x_t + k*x_t*y_t; 137547c6ae99SBarry Smith for (j=0; j<x_t; j++) { idx[nn++] = s_t++;} 1376ce00eea3SSatish Balay } else if (ye-Ye < 0 && ze-Ze < 0) { 1377ce00eea3SSatish Balay for (j=0; j<x; j++) { idx[nn++] = -1;} 137847c6ae99SBarry Smith } 137947c6ae99SBarry Smith if (n26 >= 0) { /* right above */ 1380ce00eea3SSatish Balay x_t = lx[n26 % m]; 138147c6ae99SBarry Smith y_t = ly[(n26 % (m*n))/m]; 138247c6ae99SBarry Smith /* z_t = lz[n26 / (m*n)]; */ 138347c6ae99SBarry Smith s_t = bases[n26] + (i-1)*x_t + k*x_t*y_t; 138447c6ae99SBarry Smith for (j=0; j<s_x; j++) { idx[nn++] = s_t++;} 1385ce00eea3SSatish Balay } else if (xe-Xe < 0 && ye-Ye < 0 && ze-Ze < 0) { 1386ce00eea3SSatish Balay for (j=0; j<s_x; j++) { idx[nn++] = -1;} 138747c6ae99SBarry Smith } 138847c6ae99SBarry Smith } 138947c6ae99SBarry Smith } 139047c6ae99SBarry Smith } 139147c6ae99SBarry Smith /* 139247c6ae99SBarry Smith Set the local to global ordering in the global vector, this allows use 139347c6ae99SBarry Smith of VecSetValuesLocal(). 139447c6ae99SBarry Smith */ 1395ce00eea3SSatish Balay ierr = ISCreateBlock(comm,dof,nn,idx,PETSC_OWN_POINTER,<ogis);CHKERRQ(ierr); 1396ce00eea3SSatish Balay ierr = PetscMalloc(nn*dof*sizeof(PetscInt),&idx_cpy);CHKERRQ(ierr); 1397db87c5ecSEthan Coon ierr = PetscLogObjectMemory(da,nn*dof*sizeof(PetscInt));CHKERRQ(ierr); 1398ce00eea3SSatish Balay ierr = ISGetIndices(ltogis, &idx_full); 1399ce00eea3SSatish Balay ierr = PetscMemcpy(idx_cpy,idx_full,nn*dof*sizeof(PetscInt));CHKERRQ(ierr); 1400ce00eea3SSatish Balay ierr = ISRestoreIndices(ltogis, &idx_full); 1401ce00eea3SSatish Balay ierr = ISLocalToGlobalMappingCreateIS(ltogis,&da->ltogmap);CHKERRQ(ierr); 1402ce00eea3SSatish Balay ierr = PetscLogObjectParent(da,da->ltogmap);CHKERRQ(ierr); 1403fcfd50ebSBarry Smith ierr = ISDestroy(<ogis);CHKERRQ(ierr); 14041411c6eeSJed Brown ierr = ISLocalToGlobalMappingBlock(da->ltogmap,dd->w,&da->ltogmapb);CHKERRQ(ierr); 14051411c6eeSJed Brown ierr = PetscLogObjectParent(da,da->ltogmap);CHKERRQ(ierr); 140647c6ae99SBarry Smith 1407ce00eea3SSatish Balay ierr = PetscFree2(bases,ldims);CHKERRQ(ierr); 1408ce00eea3SSatish Balay dd->m = m; dd->n = n; dd->p = p; 1409ce00eea3SSatish Balay /* note petsc expects xs/xe/Xs/Xe to be multiplied by #dofs in many places */ 1410ce00eea3SSatish Balay dd->xs = xs*dof; dd->xe = xe*dof; dd->ys = ys; dd->ye = ye; dd->zs = zs; dd->ze = ze; 1411ce00eea3SSatish Balay dd->Xs = Xs*dof; dd->Xe = Xe*dof; dd->Ys = Ys; dd->Ye = Ye; dd->Zs = Zs; dd->Ze = Ze; 1412ce00eea3SSatish Balay 1413fcfd50ebSBarry Smith ierr = VecDestroy(&local);CHKERRQ(ierr); 1414fcfd50ebSBarry Smith ierr = VecDestroy(&global);CHKERRQ(ierr); 1415ce00eea3SSatish Balay 1416ce00eea3SSatish Balay dd->gtol = gtol; 1417ce00eea3SSatish Balay dd->ltog = ltog; 1418ce00eea3SSatish Balay dd->idx = idx_cpy; 1419ce00eea3SSatish Balay dd->Nl = nn*dof; 1420ce00eea3SSatish Balay dd->base = base; 1421ce00eea3SSatish Balay da->ops->view = DMView_DA_3d; 142247c6ae99SBarry Smith dd->ltol = PETSC_NULL; 142347c6ae99SBarry Smith dd->ao = PETSC_NULL; 1424ce00eea3SSatish Balay 142547c6ae99SBarry Smith PetscFunctionReturn(0); 142647c6ae99SBarry Smith } 1427564755cdSBarry Smith 142847c6ae99SBarry Smith 142947c6ae99SBarry Smith #undef __FUNCT__ 1430aa219208SBarry Smith #define __FUNCT__ "DMDACreate3d" 143147c6ae99SBarry Smith /*@C 1432aa219208SBarry Smith DMDACreate3d - Creates an object that will manage the communication of three-dimensional 143347c6ae99SBarry Smith regular array data that is distributed across some processors. 143447c6ae99SBarry Smith 143547c6ae99SBarry Smith Collective on MPI_Comm 143647c6ae99SBarry Smith 143747c6ae99SBarry Smith Input Parameters: 143847c6ae99SBarry Smith + comm - MPI communicator 14391321219cSEthan Coon . bx,by,bz - type of ghost nodes the array have. 14401321219cSEthan Coon Use one of DMDA_BOUNDARY_NONE, DMDA_BOUNDARY_GHOSTED, DMDA_BOUNDARY_PERIODIC. 1441aa219208SBarry Smith . stencil_type - Type of stencil (DMDA_STENCIL_STAR or DMDA_STENCIL_BOX) 144247c6ae99SBarry Smith . M,N,P - global dimension in each direction of the array (use -M, -N, and or -P to indicate that it may be set to a different value 144347c6ae99SBarry Smith from the command line with -da_grid_x <M> -da_grid_y <N> -da_grid_z <P>) 144447c6ae99SBarry Smith . m,n,p - corresponding number of processors in each dimension 144547c6ae99SBarry Smith (or PETSC_DECIDE to have calculated) 144647c6ae99SBarry Smith . dof - number of degrees of freedom per node 144710d7c030SMatthew G Knepley . s - stencil width 144810d7c030SMatthew G Knepley - lx, ly, lz - arrays containing the number of nodes in each cell along 144947c6ae99SBarry Smith the x, y, and z coordinates, or PETSC_NULL. If non-null, these 145047c6ae99SBarry Smith must be of length as m,n,p and the corresponding 145147c6ae99SBarry Smith m,n, or p cannot be PETSC_DECIDE. Sum of the lx[] entries must be M, sum of 145247c6ae99SBarry Smith the ly[] must N, sum of the lz[] must be P 145347c6ae99SBarry Smith 145447c6ae99SBarry Smith Output Parameter: 145547c6ae99SBarry Smith . da - the resulting distributed array object 145647c6ae99SBarry Smith 145747c6ae99SBarry Smith Options Database Key: 1458706a11cbSBarry Smith + -dm_view - Calls DMView() at the conclusion of DMDACreate3d() 145947c6ae99SBarry Smith . -da_grid_x <nx> - number of grid points in x direction, if M < 0 146047c6ae99SBarry Smith . -da_grid_y <ny> - number of grid points in y direction, if N < 0 146147c6ae99SBarry Smith . -da_grid_z <nz> - number of grid points in z direction, if P < 0 146247c6ae99SBarry Smith . -da_processors_x <MX> - number of processors in x direction 146347c6ae99SBarry Smith . -da_processors_y <MY> - number of processors in y direction 146447c6ae99SBarry Smith . -da_processors_z <MZ> - number of processors in z direction 1465e0f5d30fSBarry Smith . -da_refine_x <rx> - refinement ratio in x direction 1466e0f5d30fSBarry Smith . -da_refine_y <ry> - refinement ratio in y direction 1467e0f5d30fSBarry Smith . -da_refine_z <rz>- refinement ratio in z directio 1468e0f5d30fSBarry Smith - -da_refine <n> - refine the DMDA n times before creating it, , if M, N, or P < 0 146947c6ae99SBarry Smith 147047c6ae99SBarry Smith Level: beginner 147147c6ae99SBarry Smith 147247c6ae99SBarry Smith Notes: 1473aa219208SBarry Smith The stencil type DMDA_STENCIL_STAR with width 1 corresponds to the 1474aa219208SBarry Smith standard 7-pt stencil, while DMDA_STENCIL_BOX with width 1 denotes 147547c6ae99SBarry Smith the standard 27-pt stencil. 147647c6ae99SBarry Smith 1477aa219208SBarry Smith The array data itself is NOT stored in the DMDA, it is stored in Vec objects; 1478564755cdSBarry Smith The appropriate vector objects can be obtained with calls to DMCreateGlobalVector() 1479564755cdSBarry Smith and DMCreateLocalVector() and calls to VecDuplicate() if more are needed. 148047c6ae99SBarry Smith 148147c6ae99SBarry Smith .keywords: distributed array, create, three-dimensional 148247c6ae99SBarry Smith 1483aa219208SBarry Smith .seealso: DMDestroy(), DMView(), DMDACreate1d(), DMDACreate2d(), DMGlobalToLocalBegin(), DMDAGetRefinementFactor(), 1484aa219208SBarry Smith DMGlobalToLocalEnd(), DMLocalToGlobalBegin(), DMDALocalToLocalBegin(), DMDALocalToLocalEnd(), DMDASetRefinementFactor(), 1485d461ba97SBarry Smith DMDAGetInfo(), DMCreateGlobalVector(), DMCreateLocalVector(), DMDACreateNaturalVector(), DMLoad(), DMDAGetOwnershipRanges() 148647c6ae99SBarry Smith 148747c6ae99SBarry Smith @*/ 14881321219cSEthan Coon PetscErrorCode DMDACreate3d(MPI_Comm comm,DMDABoundaryType bx,DMDABoundaryType by,DMDABoundaryType bz,DMDAStencilType stencil_type,PetscInt M, 14899a42bb27SBarry Smith PetscInt N,PetscInt P,PetscInt m,PetscInt n,PetscInt p,PetscInt dof,PetscInt s,const PetscInt lx[],const PetscInt ly[],const PetscInt lz[],DM *da) 149047c6ae99SBarry Smith { 149147c6ae99SBarry Smith PetscErrorCode ierr; 149247c6ae99SBarry Smith 149347c6ae99SBarry Smith PetscFunctionBegin; 1494aa219208SBarry Smith ierr = DMDACreate(comm, da);CHKERRQ(ierr); 1495aa219208SBarry Smith ierr = DMDASetDim(*da, 3);CHKERRQ(ierr); 1496aa219208SBarry Smith ierr = DMDASetSizes(*da, M, N, P);CHKERRQ(ierr); 1497aa219208SBarry Smith ierr = DMDASetNumProcs(*da, m, n, p);CHKERRQ(ierr); 14981321219cSEthan Coon ierr = DMDASetBoundaryType(*da, bx, by, bz);CHKERRQ(ierr); 1499aa219208SBarry Smith ierr = DMDASetDof(*da, dof);CHKERRQ(ierr); 1500aa219208SBarry Smith ierr = DMDASetStencilType(*da, stencil_type);CHKERRQ(ierr); 1501aa219208SBarry Smith ierr = DMDASetStencilWidth(*da, s);CHKERRQ(ierr); 1502aa219208SBarry Smith ierr = DMDASetOwnershipRanges(*da, lx, ly, lz);CHKERRQ(ierr); 150347c6ae99SBarry Smith /* This violates the behavior for other classes, but right now users expect negative dimensions to be handled this way */ 15049a42bb27SBarry Smith ierr = DMSetFromOptions(*da);CHKERRQ(ierr); 15059a42bb27SBarry Smith ierr = DMSetUp(*da);CHKERRQ(ierr); 1506ca266f36SBarry Smith ierr = DMViewFromOptions(*da,"-dm_view");CHKERRQ(ierr); 150747c6ae99SBarry Smith PetscFunctionReturn(0); 150847c6ae99SBarry Smith } 1509