xref: /petsc/src/dm/impls/da/da3.c (revision 5b399a637f946da949d4901b9cfb4dd7404a5662)
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 
7af0996ceSBarry Smith #include <petsc/private/dmdaimpl.h>     /*I   "petscdmda.h"    I*/
847c6ae99SBarry Smith 
99804daf3SBarry Smith #include <petscdraw.h>
1047c6ae99SBarry Smith #undef __FUNCT__
119a42bb27SBarry Smith #define __FUNCT__ "DMView_DA_3d"
129a42bb27SBarry Smith PetscErrorCode DMView_DA_3d(DM da,PetscViewer viewer)
1347c6ae99SBarry Smith {
1447c6ae99SBarry Smith   PetscErrorCode ierr;
1547c6ae99SBarry Smith   PetscMPIInt    rank;
169a42bb27SBarry Smith   PetscBool      iascii,isdraw,isbinary;
1747c6ae99SBarry Smith   DM_DA          *dd = (DM_DA*)da->data;
189a42bb27SBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
199a42bb27SBarry Smith   PetscBool ismatlab;
209a42bb27SBarry Smith #endif
2147c6ae99SBarry Smith 
2247c6ae99SBarry Smith   PetscFunctionBegin;
23ce94432eSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)da),&rank);CHKERRQ(ierr);
2447c6ae99SBarry Smith 
25251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
26251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
27251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
289a42bb27SBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
29251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERMATLAB,&ismatlab);CHKERRQ(ierr);
309a42bb27SBarry Smith #endif
3147c6ae99SBarry Smith   if (iascii) {
3247c6ae99SBarry Smith     PetscViewerFormat format;
3347c6ae99SBarry Smith 
341575c14dSBarry Smith     ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr);
3547c6ae99SBarry Smith     ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
3647c6ae99SBarry Smith     if (format != PETSC_VIEWER_ASCII_VTK && format != PETSC_VIEWER_ASCII_VTK_CELL) {
37aa219208SBarry Smith       DMDALocalInfo info;
38aa219208SBarry Smith       ierr = DMDAGetLocalInfo(da,&info);CHKERRQ(ierr);
3947c6ae99SBarry 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);
4047c6ae99SBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"X range of indices: %D %D, Y range of indices: %D %D, Z range of indices: %D %D\n",
4147c6ae99SBarry Smith                                                 info.xs,info.xs+info.xm,info.ys,info.ys+info.ym,info.zs,info.zs+info.zm);CHKERRQ(ierr);
4247c6ae99SBarry Smith #if !defined(PETSC_USE_COMPLEX)
436636e97aSMatthew G Knepley       if (da->coordinates) {
4447c6ae99SBarry Smith         PetscInt        last;
4547c6ae99SBarry Smith         const PetscReal *coors;
466636e97aSMatthew G Knepley         ierr = VecGetArrayRead(da->coordinates,&coors);CHKERRQ(ierr);
476636e97aSMatthew G Knepley         ierr = VecGetLocalSize(da->coordinates,&last);CHKERRQ(ierr);
4847c6ae99SBarry Smith         last = last - 3;
4957622a8eSBarry Smith         ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Lower left corner %g %g %g : Upper right %g %g %g\n",(double)coors[0],(double)coors[1],(double)coors[2],(double)coors[last],(double)coors[last+1],(double)coors[last+2]);CHKERRQ(ierr);
506636e97aSMatthew G Knepley         ierr = VecRestoreArrayRead(da->coordinates,&coors);CHKERRQ(ierr);
5147c6ae99SBarry Smith       }
5247c6ae99SBarry Smith #endif
5347c6ae99SBarry Smith       ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
541575c14dSBarry Smith       ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr);
55616157d6SJed Brown     } else {
56616157d6SJed Brown       ierr = DMView_DA_VTK(da,viewer);CHKERRQ(ierr);
5747c6ae99SBarry Smith     }
5847c6ae99SBarry Smith   } else if (isdraw) {
5947c6ae99SBarry Smith     PetscDraw      draw;
6047c6ae99SBarry Smith     PetscReal      ymin = -1.0,ymax = (PetscReal)dd->N;
6147c6ae99SBarry Smith     PetscReal      xmin = -1.0,xmax = (PetscReal)((dd->M+2)*dd->P),x,y,ycoord,xcoord;
628ea3bf28SBarry Smith     PetscInt       k,plane,base;
638ea3bf28SBarry Smith     const PetscInt *idx;
6447c6ae99SBarry Smith     char           node[10];
6547c6ae99SBarry Smith     PetscBool      isnull;
6647c6ae99SBarry Smith 
6747c6ae99SBarry Smith     ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
68*5b399a63SLisandro Dalcin     ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
69*5b399a63SLisandro Dalcin     if (isnull) PetscFunctionReturn(0);
7047c6ae99SBarry Smith 
71*5b399a63SLisandro Dalcin     ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
72*5b399a63SLisandro Dalcin     ierr = PetscDrawClear(draw);CHKERRQ(ierr);
73*5b399a63SLisandro Dalcin 
74*5b399a63SLisandro Dalcin     ierr = PetscDrawSetCoordinates(draw,xmin,ymin,xmax,ymax);CHKERRQ(ierr);
75*5b399a63SLisandro Dalcin 
76*5b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
7747c6ae99SBarry Smith     /* first processor draw all node lines */
7847c6ae99SBarry Smith     if (!rank) {
7947c6ae99SBarry Smith       for (k=0; k<dd->P; k++) {
8047c6ae99SBarry Smith         ymin = 0.0; ymax = (PetscReal)(dd->N - 1);
8147c6ae99SBarry Smith         for (xmin=(PetscReal)(k*(dd->M+1)); xmin<(PetscReal)(dd->M+(k*(dd->M+1))); xmin++) {
8247c6ae99SBarry Smith           ierr = PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_BLACK);CHKERRQ(ierr);
8347c6ae99SBarry Smith         }
8447c6ae99SBarry Smith 
8547c6ae99SBarry Smith         xmin = (PetscReal)(k*(dd->M+1)); xmax = xmin + (PetscReal)(dd->M - 1);
8647c6ae99SBarry Smith         for (ymin=0; ymin<(PetscReal)dd->N; ymin++) {
8747c6ae99SBarry Smith           ierr = PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_BLACK);CHKERRQ(ierr);
8847c6ae99SBarry Smith         }
8947c6ae99SBarry Smith       }
9047c6ae99SBarry Smith     }
91*5b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
92*5b399a63SLisandro Dalcin 
93*5b399a63SLisandro Dalcin     ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
9447c6ae99SBarry Smith     ierr = PetscDrawPause(draw);CHKERRQ(ierr);
9547c6ae99SBarry Smith 
96*5b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
97*5b399a63SLisandro Dalcin     /*Go through and draw for each plane*/
98*5b399a63SLisandro Dalcin     for (k=0; k<dd->P; k++) {
9947c6ae99SBarry Smith       if ((k >= dd->zs) && (k < dd->ze)) {
10047c6ae99SBarry Smith         /* draw my box */
10147c6ae99SBarry Smith         ymin = dd->ys;
10247c6ae99SBarry Smith         ymax = dd->ye - 1;
10347c6ae99SBarry Smith         xmin = dd->xs/dd->w    + (dd->M+1)*k;
10447c6ae99SBarry Smith         xmax =(dd->xe-1)/dd->w + (dd->M+1)*k;
10547c6ae99SBarry Smith 
10647c6ae99SBarry Smith         ierr = PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_RED);CHKERRQ(ierr);
10747c6ae99SBarry Smith         ierr = PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_RED);CHKERRQ(ierr);
10847c6ae99SBarry Smith         ierr = PetscDrawLine(draw,xmin,ymax,xmax,ymax,PETSC_DRAW_RED);CHKERRQ(ierr);
10947c6ae99SBarry Smith         ierr = PetscDrawLine(draw,xmax,ymin,xmax,ymax,PETSC_DRAW_RED);CHKERRQ(ierr);
11047c6ae99SBarry Smith 
11147c6ae99SBarry Smith         xmin = dd->xs/dd->w;
11247c6ae99SBarry Smith         xmax =(dd->xe-1)/dd->w;
11347c6ae99SBarry Smith 
11447c6ae99SBarry Smith         /* put in numbers*/
11547c6ae99SBarry Smith         base = (dd->base+(dd->xe-dd->xs)*(dd->ye-dd->ys)*(k-dd->zs))/dd->w;
11647c6ae99SBarry Smith 
11747c6ae99SBarry Smith         /* Identify which processor owns the box */
11847c6ae99SBarry Smith         sprintf(node,"%d",rank);
11947c6ae99SBarry Smith         ierr = PetscDrawString(draw,xmin+(dd->M+1)*k+.2,ymin+.3,PETSC_DRAW_RED,node);CHKERRQ(ierr);
12047c6ae99SBarry Smith 
12147c6ae99SBarry Smith         for (y=ymin; y<=ymax; y++) {
12247c6ae99SBarry Smith           for (x=xmin+(dd->M+1)*k; x<=xmax+(dd->M+1)*k; x++) {
12347c6ae99SBarry Smith             sprintf(node,"%d",(int)base++);
12447c6ae99SBarry Smith             ierr = PetscDrawString(draw,x,y,PETSC_DRAW_BLACK,node);CHKERRQ(ierr);
12547c6ae99SBarry Smith           }
12647c6ae99SBarry Smith         }
12747c6ae99SBarry Smith 
12847c6ae99SBarry Smith       }
12947c6ae99SBarry Smith     }
130*5b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
131*5b399a63SLisandro Dalcin 
132*5b399a63SLisandro Dalcin     ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
13347c6ae99SBarry Smith     ierr = PetscDrawPause(draw);CHKERRQ(ierr);
13447c6ae99SBarry Smith 
135*5b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
13647c6ae99SBarry Smith     for (k=0-dd->s; k<dd->P+dd->s; k++) {
13747c6ae99SBarry Smith       /* Go through and draw for each plane */
13847c6ae99SBarry Smith       if ((k >= dd->Zs) && (k < dd->Ze)) {
13947c6ae99SBarry Smith         /* overlay ghost numbers, useful for error checking */
140565245c5SBarry Smith         base = (dd->Xe-dd->Xs)*(dd->Ye-dd->Ys)*(k-dd->Zs)/dd->w;
14145b6f7e9SBarry Smith         ierr = ISLocalToGlobalMappingGetBlockIndices(da->ltogmap,&idx);CHKERRQ(ierr);
14247c6ae99SBarry Smith         plane=k;
143565245c5SBarry Smith         /* Keep z wrap around points on the drawing */
1448865f1eaSKarl Rupp         if (k<0) plane=dd->P+k;
1458865f1eaSKarl Rupp         if (k>=dd->P) plane=k-dd->P;
14647c6ae99SBarry Smith         ymin = dd->Ys; ymax = dd->Ye;
14747c6ae99SBarry Smith         xmin = (dd->M+1)*plane*dd->w;
14847c6ae99SBarry Smith         xmax = (dd->M+1)*plane*dd->w+dd->M*dd->w;
14947c6ae99SBarry Smith         for (y=ymin; y<ymax; y++) {
15047c6ae99SBarry Smith           for (x=xmin+dd->Xs; x<xmin+dd->Xe; x+=dd->w) {
1518ea3bf28SBarry Smith             sprintf(node,"%d",(int)(idx[base]));
15247c6ae99SBarry Smith             ycoord = y;
15347c6ae99SBarry Smith             /*Keep y wrap around points on drawing */
1548865f1eaSKarl Rupp             if (y<0) ycoord = dd->N+y;
1558865f1eaSKarl Rupp             if (y>=dd->N) ycoord = y-dd->N;
15647c6ae99SBarry Smith             xcoord = x;   /* Keep x wrap points on drawing */
1578865f1eaSKarl Rupp             if (x<xmin) xcoord = xmax - (xmin-x);
1588865f1eaSKarl Rupp             if (x>=xmax) xcoord = xmin + (x-xmax);
15947c6ae99SBarry Smith             ierr = PetscDrawString(draw,xcoord/dd->w,ycoord,PETSC_DRAW_BLUE,node);CHKERRQ(ierr);
160565245c5SBarry Smith             base++;
16147c6ae99SBarry Smith           }
16247c6ae99SBarry Smith         }
16345b6f7e9SBarry Smith         ierr = ISLocalToGlobalMappingRestoreBlockIndices(da->ltogmap,&idx);CHKERRQ(ierr);
16447c6ae99SBarry Smith       }
16547c6ae99SBarry Smith     }
166*5b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
167*5b399a63SLisandro Dalcin 
168*5b399a63SLisandro Dalcin     ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
16947c6ae99SBarry Smith     ierr = PetscDrawPause(draw);CHKERRQ(ierr);
1709a42bb27SBarry Smith   } else if (isbinary) {
1719a42bb27SBarry Smith     ierr = DMView_DA_Binary(da,viewer);CHKERRQ(ierr);
1729a42bb27SBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
1739a42bb27SBarry Smith   } else if (ismatlab) {
1749a42bb27SBarry Smith     ierr = DMView_DA_Matlab(da,viewer);CHKERRQ(ierr);
1759a42bb27SBarry Smith #endif
17611aeaf0aSBarry Smith   }
17747c6ae99SBarry Smith   PetscFunctionReturn(0);
17847c6ae99SBarry Smith }
17947c6ae99SBarry Smith 
18047c6ae99SBarry Smith #undef __FUNCT__
1819a42bb27SBarry Smith #define __FUNCT__ "DMSetUp_DA_3D"
1827087cfbeSBarry Smith PetscErrorCode  DMSetUp_DA_3D(DM da)
18347c6ae99SBarry Smith {
18447c6ae99SBarry Smith   DM_DA            *dd          = (DM_DA*)da->data;
18547c6ae99SBarry Smith   const PetscInt   M            = dd->M;
18647c6ae99SBarry Smith   const PetscInt   N            = dd->N;
18747c6ae99SBarry Smith   const PetscInt   P            = dd->P;
18847c6ae99SBarry Smith   PetscInt         m            = dd->m;
18947c6ae99SBarry Smith   PetscInt         n            = dd->n;
19047c6ae99SBarry Smith   PetscInt         p            = dd->p;
19147c6ae99SBarry Smith   const PetscInt   dof          = dd->w;
19247c6ae99SBarry Smith   const PetscInt   s            = dd->s;
193bff4a2f0SMatthew G. Knepley   DMBoundaryType   bx           = dd->bx;
194bff4a2f0SMatthew G. Knepley   DMBoundaryType   by           = dd->by;
195bff4a2f0SMatthew G. Knepley   DMBoundaryType   bz           = dd->bz;
19619fd82e9SBarry Smith   DMDAStencilType  stencil_type = dd->stencil_type;
19747c6ae99SBarry Smith   PetscInt         *lx          = dd->lx;
19847c6ae99SBarry Smith   PetscInt         *ly          = dd->ly;
19947c6ae99SBarry Smith   PetscInt         *lz          = dd->lz;
20047c6ae99SBarry Smith   MPI_Comm         comm;
20147c6ae99SBarry Smith   PetscMPIInt      rank,size;
202ce00eea3SSatish Balay   PetscInt         xs = 0,xe,ys = 0,ye,zs = 0,ze,x = 0,y = 0,z = 0;
203bd1fc5aeSBarry Smith   PetscInt         Xs,Xe,Ys,Ye,Zs,Ze,IXs,IXe,IYs,IYe,IZs,IZe,pm;
2048ea3bf28SBarry Smith   PetscInt         left,right,up,down,bottom,top,i,j,k,*idx,nn;
20547c6ae99SBarry Smith   PetscInt         n0,n1,n2,n3,n4,n5,n6,n7,n8,n9,n10,n11,n12,n14;
20647c6ae99SBarry Smith   PetscInt         n15,n16,n17,n18,n19,n20,n21,n22,n23,n24,n25,n26;
207db87c5ecSEthan Coon   PetscInt         *bases,*ldims,base,x_t,y_t,z_t,s_t,count,s_x,s_y,s_z;
20847c6ae99SBarry Smith   PetscInt         sn0  = 0,sn1 = 0,sn2 = 0,sn3 = 0,sn5 = 0,sn6 = 0,sn7 = 0;
20947c6ae99SBarry Smith   PetscInt         sn8  = 0,sn9 = 0,sn11 = 0,sn15 = 0,sn24 = 0,sn25 = 0,sn26 = 0;
21047c6ae99SBarry Smith   PetscInt         sn17 = 0,sn18 = 0,sn19 = 0,sn20 = 0,sn21 = 0,sn23 = 0;
21147c6ae99SBarry Smith   Vec              local,global;
212bd1fc5aeSBarry Smith   VecScatter       gtol;
21345b6f7e9SBarry Smith   IS               to,from;
2146f951b95Secoon   PetscBool        twod;
21547c6ae99SBarry Smith   PetscErrorCode   ierr;
21647c6ae99SBarry Smith 
2176f951b95Secoon 
21847c6ae99SBarry Smith   PetscFunctionBegin;
219bff4a2f0SMatthew G. Knepley   if (stencil_type == DMDA_STENCIL_BOX && (bx == DM_BOUNDARY_MIRROR || by == DM_BOUNDARY_MIRROR || bz == DM_BOUNDARY_MIRROR)) SETERRQ(PetscObjectComm((PetscObject)da),PETSC_ERR_SUP,"Mirror boundary and box stencil");
22047c6ae99SBarry Smith   ierr = PetscObjectGetComm((PetscObject) da, &comm);CHKERRQ(ierr);
2213855c12bSBarry Smith #if !defined(PETSC_USE_64BIT_INDICES)
2223855c12bSBarry 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);
2233855c12bSBarry Smith #endif
2243855c12bSBarry Smith 
22547c6ae99SBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
22647c6ae99SBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
22747c6ae99SBarry Smith 
22847c6ae99SBarry Smith   if (m != PETSC_DECIDE) {
22947c6ae99SBarry Smith     if (m < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in X direction: %D",m);
23047c6ae99SBarry Smith     else if (m > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in X direction: %D %d",m,size);
23147c6ae99SBarry Smith   }
23247c6ae99SBarry Smith   if (n != PETSC_DECIDE) {
23347c6ae99SBarry Smith     if (n < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in Y direction: %D",n);
23447c6ae99SBarry Smith     else if (n > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in Y direction: %D %d",n,size);
23547c6ae99SBarry Smith   }
23647c6ae99SBarry Smith   if (p != PETSC_DECIDE) {
23747c6ae99SBarry Smith     if (p < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in Z direction: %D",p);
23847c6ae99SBarry Smith     else if (p > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in Z direction: %D %d",p,size);
23947c6ae99SBarry Smith   }
2400716a85fSBarry 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);
24147c6ae99SBarry Smith 
24247c6ae99SBarry Smith   /* Partition the array among the processors */
24347c6ae99SBarry Smith   if (m == PETSC_DECIDE && n != PETSC_DECIDE && p != PETSC_DECIDE) {
24447c6ae99SBarry Smith     m = size/(n*p);
24547c6ae99SBarry Smith   } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) {
24647c6ae99SBarry Smith     n = size/(m*p);
24747c6ae99SBarry Smith   } else if (m != PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) {
24847c6ae99SBarry Smith     p = size/(m*n);
24947c6ae99SBarry Smith   } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) {
25047c6ae99SBarry Smith     /* try for squarish distribution */
251369cc0aeSBarry Smith     m = (int)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)N*p)));
25247c6ae99SBarry Smith     if (!m) m = 1;
25347c6ae99SBarry Smith     while (m > 0) {
25447c6ae99SBarry Smith       n = size/(m*p);
25547c6ae99SBarry Smith       if (m*n*p == size) break;
25647c6ae99SBarry Smith       m--;
25747c6ae99SBarry Smith     }
25847c6ae99SBarry Smith     if (!m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad p value: p = %D",p);
25947c6ae99SBarry Smith     if (M > N && m < n) {PetscInt _m = m; m = n; n = _m;}
26047c6ae99SBarry Smith   } else if (m == PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) {
26147c6ae99SBarry Smith     /* try for squarish distribution */
262369cc0aeSBarry Smith     m = (int)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)P*n)));
26347c6ae99SBarry Smith     if (!m) m = 1;
26447c6ae99SBarry Smith     while (m > 0) {
26547c6ae99SBarry Smith       p = size/(m*n);
26647c6ae99SBarry Smith       if (m*n*p == size) break;
26747c6ae99SBarry Smith       m--;
26847c6ae99SBarry Smith     }
26947c6ae99SBarry Smith     if (!m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad n value: n = %D",n);
27047c6ae99SBarry Smith     if (M > P && m < p) {PetscInt _m = m; m = p; p = _m;}
27147c6ae99SBarry Smith   } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) {
27247c6ae99SBarry Smith     /* try for squarish distribution */
273369cc0aeSBarry Smith     n = (int)(0.5 + PetscSqrtReal(((PetscReal)N)*((PetscReal)size)/((PetscReal)P*m)));
27447c6ae99SBarry Smith     if (!n) n = 1;
27547c6ae99SBarry Smith     while (n > 0) {
27647c6ae99SBarry Smith       p = size/(m*n);
27747c6ae99SBarry Smith       if (m*n*p == size) break;
27847c6ae99SBarry Smith       n--;
27947c6ae99SBarry Smith     }
28047c6ae99SBarry Smith     if (!n) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad m value: m = %D",n);
28147c6ae99SBarry Smith     if (N > P && n < p) {PetscInt _n = n; n = p; p = _n;}
28247c6ae99SBarry Smith   } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) {
28347c6ae99SBarry Smith     /* try for squarish distribution */
284369cc0aeSBarry Smith     n = (PetscInt)(0.5 + PetscPowReal(((PetscReal)N*N)*((PetscReal)size)/((PetscReal)P*M),(PetscReal)(1./3.)));
28547c6ae99SBarry Smith     if (!n) n = 1;
28647c6ae99SBarry Smith     while (n > 0) {
28747c6ae99SBarry Smith       pm = size/n;
28847c6ae99SBarry Smith       if (n*pm == size) break;
28947c6ae99SBarry Smith       n--;
29047c6ae99SBarry Smith     }
29147c6ae99SBarry Smith     if (!n) n = 1;
292369cc0aeSBarry Smith     m = (PetscInt)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)P*n)));
29347c6ae99SBarry Smith     if (!m) m = 1;
29447c6ae99SBarry Smith     while (m > 0) {
29547c6ae99SBarry Smith       p = size/(m*n);
29647c6ae99SBarry Smith       if (m*n*p == size) break;
29747c6ae99SBarry Smith       m--;
29847c6ae99SBarry Smith     }
29947c6ae99SBarry Smith     if (M > P && m < p) {PetscInt _m = m; m = p; p = _m;}
300ce94432eSBarry Smith   } else if (m*n*p != size) SETERRQ(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Given Bad partition");
30147c6ae99SBarry Smith 
302ce94432eSBarry Smith   if (m*n*p != size) SETERRQ(PetscObjectComm((PetscObject)da),PETSC_ERR_PLIB,"Could not find good partition");
303ce94432eSBarry Smith   if (M < m) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in x direction is too fine! %D %D",M,m);
304ce94432eSBarry Smith   if (N < n) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in y direction is too fine! %D %D",N,n);
305ce94432eSBarry Smith   if (P < p) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in z direction is too fine! %D %D",P,p);
30647c6ae99SBarry Smith 
30747c6ae99SBarry Smith   /*
30847c6ae99SBarry Smith      Determine locally owned region
30947c6ae99SBarry Smith      [x, y, or z]s is the first local node number, [x, y, z] is the number of local nodes
31047c6ae99SBarry Smith   */
31147c6ae99SBarry Smith 
31247c6ae99SBarry Smith   if (!lx) {
313785e854fSJed Brown     ierr = PetscMalloc1(m, &dd->lx);CHKERRQ(ierr);
31447c6ae99SBarry Smith     lx   = dd->lx;
3158865f1eaSKarl Rupp     for (i=0; i<m; i++) lx[i] = M/m + ((M % m) > (i % m));
31647c6ae99SBarry Smith   }
31747c6ae99SBarry Smith   x  = lx[rank % m];
31847c6ae99SBarry Smith   xs = 0;
3198865f1eaSKarl Rupp   for (i=0; i<(rank%m); i++) xs += lx[i];
320bff4a2f0SMatthew G. Knepley   if ((x < s) && ((m > 1) || (bx == DM_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);
32147c6ae99SBarry Smith 
32247c6ae99SBarry Smith   if (!ly) {
323785e854fSJed Brown     ierr = PetscMalloc1(n, &dd->ly);CHKERRQ(ierr);
32447c6ae99SBarry Smith     ly   = dd->ly;
3258865f1eaSKarl Rupp     for (i=0; i<n; i++) ly[i] = N/n + ((N % n) > (i % n));
32647c6ae99SBarry Smith   }
32747c6ae99SBarry Smith   y = ly[(rank % (m*n))/m];
328bff4a2f0SMatthew G. Knepley   if ((y < s) && ((n > 1) || (by == DM_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);
32930729d88SBarry Smith 
33047c6ae99SBarry Smith   ys = 0;
3318865f1eaSKarl Rupp   for (i=0; i<(rank % (m*n))/m; i++) ys += ly[i];
33247c6ae99SBarry Smith 
33347c6ae99SBarry Smith   if (!lz) {
334785e854fSJed Brown     ierr = PetscMalloc1(p, &dd->lz);CHKERRQ(ierr);
33547c6ae99SBarry Smith     lz = dd->lz;
3368865f1eaSKarl Rupp     for (i=0; i<p; i++) lz[i] = P/p + ((P % p) > (i % p));
33747c6ae99SBarry Smith   }
33847c6ae99SBarry Smith   z = lz[rank/(m*n)];
339bcea557cSEthan Coon 
340fdc81ce8SEthan Coon   /* note this is different than x- and y-, as we will handle as an important special
341bff4a2f0SMatthew G. Knepley    case when p=P=1 and DM_BOUNDARY_PERIODIC and s > z.  This is to deal with 2D problems
342fdc81ce8SEthan Coon    in a 3D code.  Additional code for this case is noted with "2d case" comments */
3436f951b95Secoon   twod = PETSC_FALSE;
3448865f1eaSKarl Rupp   if (P == 1) twod = PETSC_TRUE;
345bff4a2f0SMatthew G. Knepley   else if ((z < s) && ((p > 1) || (bz == DM_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);
34647c6ae99SBarry Smith   zs = 0;
3478865f1eaSKarl Rupp   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)  */
353d9c9ebe5SPeter Brune   if (xs-s > 0) {
354d9c9ebe5SPeter Brune     Xs = xs - s; IXs = xs - s;
35588661749SPeter Brune   } else {
3568865f1eaSKarl Rupp     if (bx) Xs = xs - s;
3578865f1eaSKarl Rupp     else Xs = 0;
35888661749SPeter Brune     IXs = 0;
35988661749SPeter Brune   }
360d9c9ebe5SPeter Brune   if (xe+s <= M) {
361d9c9ebe5SPeter Brune     Xe = xe + s; IXe = xe + s;
36288661749SPeter Brune   } else {
36388661749SPeter Brune     if (bx) {
364d9c9ebe5SPeter Brune       Xs = xs - s; Xe = xe + s;
3658865f1eaSKarl Rupp     } else Xe = M;
36688661749SPeter Brune     IXe = M;
36788661749SPeter Brune   }
36847c6ae99SBarry Smith 
369bff4a2f0SMatthew G. Knepley   if (bx == DM_BOUNDARY_PERIODIC || bx == DM_BOUNDARY_MIRROR) {
370d9c9ebe5SPeter Brune     IXs = xs - s;
371d9c9ebe5SPeter Brune     IXe = xe + s;
372d9c9ebe5SPeter Brune     Xs  = xs - s;
373d9c9ebe5SPeter Brune     Xe  = xe + s;
37488661749SPeter Brune   }
37588661749SPeter Brune 
376d9c9ebe5SPeter Brune   if (ys-s > 0) {
377d9c9ebe5SPeter Brune     Ys = ys - s; IYs = ys - s;
37888661749SPeter Brune   } else {
3798865f1eaSKarl Rupp     if (by) Ys = ys - s;
3808865f1eaSKarl Rupp     else Ys = 0;
38188661749SPeter Brune     IYs = 0;
38288661749SPeter Brune   }
383d9c9ebe5SPeter Brune   if (ye+s <= N) {
384d9c9ebe5SPeter Brune     Ye = ye + s; IYe = ye + s;
38588661749SPeter Brune   } else {
3868865f1eaSKarl Rupp     if (by) Ye = ye + s;
3878865f1eaSKarl Rupp     else Ye = N;
38888661749SPeter Brune     IYe = N;
38988661749SPeter Brune   }
39088661749SPeter Brune 
391bff4a2f0SMatthew G. Knepley   if (by == DM_BOUNDARY_PERIODIC || by == DM_BOUNDARY_MIRROR) {
392d9c9ebe5SPeter Brune     IYs = ys - s;
393d9c9ebe5SPeter Brune     IYe = ye + s;
394d9c9ebe5SPeter Brune     Ys  = ys - s;
395d9c9ebe5SPeter Brune     Ye  = ye + s;
39688661749SPeter Brune   }
39788661749SPeter Brune 
398d9c9ebe5SPeter Brune   if (zs-s > 0) {
399d9c9ebe5SPeter Brune     Zs = zs - s; IZs = zs - s;
40088661749SPeter Brune   } else {
4018865f1eaSKarl Rupp     if (bz) Zs = zs - s;
4028865f1eaSKarl Rupp     else Zs = 0;
40388661749SPeter Brune     IZs = 0;
40488661749SPeter Brune   }
405d9c9ebe5SPeter Brune   if (ze+s <= P) {
406d9c9ebe5SPeter Brune     Ze = ze + s; IZe = ze + s;
40788661749SPeter Brune   } else {
4088865f1eaSKarl Rupp     if (bz) Ze = ze + s;
4098865f1eaSKarl Rupp     else Ze = P;
41088661749SPeter Brune     IZe = P;
41188661749SPeter Brune   }
41288661749SPeter Brune 
413bff4a2f0SMatthew G. Knepley   if (bz == DM_BOUNDARY_PERIODIC || bz == DM_BOUNDARY_MIRROR) {
414d9c9ebe5SPeter Brune     IZs = zs - s;
415d9c9ebe5SPeter Brune     IZe = ze + s;
416d9c9ebe5SPeter Brune     Zs  = zs - s;
417d9c9ebe5SPeter Brune     Ze  = ze + s;
41888661749SPeter Brune   }
41947c6ae99SBarry Smith 
42047c6ae99SBarry Smith   /* Resize all X parameters to reflect w */
421d9c9ebe5SPeter Brune   s_x = s;
422d9c9ebe5SPeter Brune   s_y = s;
423d9c9ebe5SPeter Brune   s_z = s;
42447c6ae99SBarry Smith 
42547c6ae99SBarry Smith   /* determine starting point of each processor */
42647c6ae99SBarry Smith   nn       = x*y*z;
427dcca6d9dSJed Brown   ierr     = PetscMalloc2(size+1,&bases,size,&ldims);CHKERRQ(ierr);
42847c6ae99SBarry Smith   ierr     = MPI_Allgather(&nn,1,MPIU_INT,ldims,1,MPIU_INT,comm);CHKERRQ(ierr);
42947c6ae99SBarry Smith   bases[0] = 0;
4308865f1eaSKarl Rupp   for (i=1; i<=size; i++) bases[i] = ldims[i-1];
4318865f1eaSKarl Rupp   for (i=1; i<=size; i++) bases[i] += bases[i-1];
432ce00eea3SSatish Balay   base = bases[rank]*dof;
43347c6ae99SBarry Smith 
43447c6ae99SBarry Smith   /* allocate the base parallel and sequential vectors */
435ce00eea3SSatish Balay   dd->Nlocal = x*y*z*dof;
436b1fb7eb7SBarry Smith   ierr       = VecCreateMPIWithArray(comm,dof,dd->Nlocal,PETSC_DECIDE,NULL,&global);CHKERRQ(ierr);
437ce00eea3SSatish Balay   dd->nlocal = (Xe-Xs)*(Ye-Ys)*(Ze-Zs)*dof;
438b1fb7eb7SBarry Smith   ierr       = VecCreateSeqWithArray(PETSC_COMM_SELF,dof,dd->nlocal,NULL,&local);CHKERRQ(ierr);
43947c6ae99SBarry Smith 
44047c6ae99SBarry Smith   /* generate appropriate vector scatters */
44147c6ae99SBarry Smith   /* local to global inserts non-ghost point region into global */
44202fe608eSBarry Smith   ierr   = PetscMalloc1((IXe-IXs)*(IYe-IYs)*(IZe-IZs),&idx);CHKERRQ(ierr);
443ce00eea3SSatish Balay   left   = xs - Xs; right = left + x;
44447c6ae99SBarry Smith   bottom = ys - Ys; top = bottom + y;
44547c6ae99SBarry Smith   down   = zs - Zs; up  = down + z;
44647c6ae99SBarry Smith   count  = 0;
44747c6ae99SBarry Smith   for (i=down; i<up; i++) {
44847c6ae99SBarry Smith     for (j=bottom; j<top; j++) {
449ce00eea3SSatish Balay       for (k=left; k<right; k++) {
450ce00eea3SSatish Balay         idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
45147c6ae99SBarry Smith       }
45247c6ae99SBarry Smith     }
45347c6ae99SBarry Smith   }
45447c6ae99SBarry Smith 
455ce00eea3SSatish Balay   /* global to local must include ghost points within the domain,
456ce00eea3SSatish Balay      but not ghost points outside the domain that aren't periodic */
457d9c9ebe5SPeter Brune   if (stencil_type == DMDA_STENCIL_BOX) {
458ce00eea3SSatish Balay     left   = IXs - Xs; right = left + (IXe-IXs);
459ce00eea3SSatish Balay     bottom = IYs - Ys; top = bottom + (IYe-IYs);
460ce00eea3SSatish Balay     down   = IZs - Zs; up  = down + (IZe-IZs);
461ce00eea3SSatish Balay     count  = 0;
462ce00eea3SSatish Balay     for (i=down; i<up; i++) {
463ce00eea3SSatish Balay       for (j=bottom; j<top; j++) {
464ce00eea3SSatish Balay         for (k=left; k<right; k++) {
465ce00eea3SSatish Balay           idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
466ce00eea3SSatish Balay         }
467ce00eea3SSatish Balay       }
468ce00eea3SSatish Balay     }
469ce00eea3SSatish Balay     ierr = ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&to);CHKERRQ(ierr);
47047c6ae99SBarry Smith   } else {
47147c6ae99SBarry Smith     /* This is way ugly! We need to list the funny cross type region */
472ce00eea3SSatish Balay     left   = xs - Xs; right = left + x;
47347c6ae99SBarry Smith     bottom = ys - Ys; top = bottom + y;
47447c6ae99SBarry Smith     down   = zs - Zs;   up  = down + z;
47547c6ae99SBarry Smith     count  = 0;
476ce00eea3SSatish Balay     /* the bottom chunck */
477ce00eea3SSatish Balay     for (i=(IZs-Zs); i<down; i++) {
47847c6ae99SBarry Smith       for (j=bottom; j<top; j++) {
479ce00eea3SSatish Balay         for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
48047c6ae99SBarry Smith       }
48147c6ae99SBarry Smith     }
48247c6ae99SBarry Smith     /* the middle piece */
48347c6ae99SBarry Smith     for (i=down; i<up; i++) {
48447c6ae99SBarry Smith       /* front */
485ce00eea3SSatish Balay       for (j=(IYs-Ys); j<bottom; j++) {
486ce00eea3SSatish Balay         for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
48747c6ae99SBarry Smith       }
48847c6ae99SBarry Smith       /* middle */
48947c6ae99SBarry Smith       for (j=bottom; j<top; j++) {
490ce00eea3SSatish Balay         for (k=IXs-Xs; k<IXe-Xs; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
49147c6ae99SBarry Smith       }
49247c6ae99SBarry Smith       /* back */
493ce00eea3SSatish Balay       for (j=top; j<top+IYe-ye; j++) {
494ce00eea3SSatish Balay         for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
49547c6ae99SBarry Smith       }
49647c6ae99SBarry Smith     }
49747c6ae99SBarry Smith     /* the top piece */
498ce00eea3SSatish Balay     for (i=up; i<up+IZe-ze; i++) {
49947c6ae99SBarry Smith       for (j=bottom; j<top; j++) {
500ce00eea3SSatish Balay         for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
50147c6ae99SBarry Smith       }
50247c6ae99SBarry Smith     }
50347c6ae99SBarry Smith     ierr = ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&to);CHKERRQ(ierr);
50447c6ae99SBarry Smith   }
50547c6ae99SBarry Smith 
50647c6ae99SBarry Smith   /* determine who lies on each side of use stored in    n24 n25 n26
50747c6ae99SBarry Smith                                                          n21 n22 n23
50847c6ae99SBarry Smith                                                          n18 n19 n20
50947c6ae99SBarry Smith 
51047c6ae99SBarry Smith                                                          n15 n16 n17
51147c6ae99SBarry Smith                                                          n12     n14
51247c6ae99SBarry Smith                                                          n9  n10 n11
51347c6ae99SBarry Smith 
51447c6ae99SBarry Smith                                                          n6  n7  n8
51547c6ae99SBarry Smith                                                          n3  n4  n5
51647c6ae99SBarry Smith                                                          n0  n1  n2
51747c6ae99SBarry Smith   */
51847c6ae99SBarry Smith 
51947c6ae99SBarry Smith   /* Solve for X,Y, and Z Periodic Case First, Then Modify Solution */
52047c6ae99SBarry Smith   /* Assume Nodes are Internal to the Cube */
52147c6ae99SBarry Smith   n0 = rank - m*n - m - 1;
52247c6ae99SBarry Smith   n1 = rank - m*n - m;
52347c6ae99SBarry Smith   n2 = rank - m*n - m + 1;
52447c6ae99SBarry Smith   n3 = rank - m*n -1;
52547c6ae99SBarry Smith   n4 = rank - m*n;
52647c6ae99SBarry Smith   n5 = rank - m*n + 1;
52747c6ae99SBarry Smith   n6 = rank - m*n + m - 1;
52847c6ae99SBarry Smith   n7 = rank - m*n + m;
52947c6ae99SBarry Smith   n8 = rank - m*n + m + 1;
53047c6ae99SBarry Smith 
53147c6ae99SBarry Smith   n9  = rank - m - 1;
53247c6ae99SBarry Smith   n10 = rank - m;
53347c6ae99SBarry Smith   n11 = rank - m + 1;
53447c6ae99SBarry Smith   n12 = rank - 1;
53547c6ae99SBarry Smith   n14 = rank + 1;
53647c6ae99SBarry Smith   n15 = rank + m - 1;
53747c6ae99SBarry Smith   n16 = rank + m;
53847c6ae99SBarry Smith   n17 = rank + m + 1;
53947c6ae99SBarry Smith 
54047c6ae99SBarry Smith   n18 = rank + m*n - m - 1;
54147c6ae99SBarry Smith   n19 = rank + m*n - m;
54247c6ae99SBarry Smith   n20 = rank + m*n - m + 1;
54347c6ae99SBarry Smith   n21 = rank + m*n - 1;
54447c6ae99SBarry Smith   n22 = rank + m*n;
54547c6ae99SBarry Smith   n23 = rank + m*n + 1;
54647c6ae99SBarry Smith   n24 = rank + m*n + m - 1;
54747c6ae99SBarry Smith   n25 = rank + m*n + m;
54847c6ae99SBarry Smith   n26 = rank + m*n + m + 1;
54947c6ae99SBarry Smith 
55047c6ae99SBarry Smith   /* Assume Pieces are on Faces of Cube */
55147c6ae99SBarry Smith 
55247c6ae99SBarry Smith   if (xs == 0) { /* First assume not corner or edge */
55347c6ae99SBarry Smith     n0  = rank       -1 - (m*n);
55447c6ae99SBarry Smith     n3  = rank + m   -1 - (m*n);
55547c6ae99SBarry Smith     n6  = rank + 2*m -1 - (m*n);
55647c6ae99SBarry Smith     n9  = rank       -1;
55747c6ae99SBarry Smith     n12 = rank + m   -1;
55847c6ae99SBarry Smith     n15 = rank + 2*m -1;
55947c6ae99SBarry Smith     n18 = rank       -1 + (m*n);
56047c6ae99SBarry Smith     n21 = rank + m   -1 + (m*n);
56147c6ae99SBarry Smith     n24 = rank + 2*m -1 + (m*n);
56247c6ae99SBarry Smith   }
56347c6ae99SBarry Smith 
564ce00eea3SSatish Balay   if (xe == M) { /* First assume not corner or edge */
56547c6ae99SBarry Smith     n2  = rank -2*m +1 - (m*n);
56647c6ae99SBarry Smith     n5  = rank - m  +1 - (m*n);
56747c6ae99SBarry Smith     n8  = rank      +1 - (m*n);
56847c6ae99SBarry Smith     n11 = rank -2*m +1;
56947c6ae99SBarry Smith     n14 = rank - m  +1;
57047c6ae99SBarry Smith     n17 = rank      +1;
57147c6ae99SBarry Smith     n20 = rank -2*m +1 + (m*n);
57247c6ae99SBarry Smith     n23 = rank - m  +1 + (m*n);
57347c6ae99SBarry Smith     n26 = rank      +1 + (m*n);
57447c6ae99SBarry Smith   }
57547c6ae99SBarry Smith 
57647c6ae99SBarry Smith   if (ys==0) { /* First assume not corner or edge */
57747c6ae99SBarry Smith     n0  = rank + m * (n-1) -1 - (m*n);
57847c6ae99SBarry Smith     n1  = rank + m * (n-1)    - (m*n);
57947c6ae99SBarry Smith     n2  = rank + m * (n-1) +1 - (m*n);
58047c6ae99SBarry Smith     n9  = rank + m * (n-1) -1;
58147c6ae99SBarry Smith     n10 = rank + m * (n-1);
58247c6ae99SBarry Smith     n11 = rank + m * (n-1) +1;
58347c6ae99SBarry Smith     n18 = rank + m * (n-1) -1 + (m*n);
58447c6ae99SBarry Smith     n19 = rank + m * (n-1)    + (m*n);
58547c6ae99SBarry Smith     n20 = rank + m * (n-1) +1 + (m*n);
58647c6ae99SBarry Smith   }
58747c6ae99SBarry Smith 
58847c6ae99SBarry Smith   if (ye == N) { /* First assume not corner or edge */
58947c6ae99SBarry Smith     n6  = rank - m * (n-1) -1 - (m*n);
59047c6ae99SBarry Smith     n7  = rank - m * (n-1)    - (m*n);
59147c6ae99SBarry Smith     n8  = rank - m * (n-1) +1 - (m*n);
59247c6ae99SBarry Smith     n15 = rank - m * (n-1) -1;
59347c6ae99SBarry Smith     n16 = rank - m * (n-1);
59447c6ae99SBarry Smith     n17 = rank - m * (n-1) +1;
59547c6ae99SBarry Smith     n24 = rank - m * (n-1) -1 + (m*n);
59647c6ae99SBarry Smith     n25 = rank - m * (n-1)    + (m*n);
59747c6ae99SBarry Smith     n26 = rank - m * (n-1) +1 + (m*n);
59847c6ae99SBarry Smith   }
59947c6ae99SBarry Smith 
60047c6ae99SBarry Smith   if (zs == 0) { /* First assume not corner or edge */
60147c6ae99SBarry Smith     n0 = size - (m*n) + rank - m - 1;
60247c6ae99SBarry Smith     n1 = size - (m*n) + rank - m;
60347c6ae99SBarry Smith     n2 = size - (m*n) + rank - m + 1;
60447c6ae99SBarry Smith     n3 = size - (m*n) + rank - 1;
60547c6ae99SBarry Smith     n4 = size - (m*n) + rank;
60647c6ae99SBarry Smith     n5 = size - (m*n) + rank + 1;
60747c6ae99SBarry Smith     n6 = size - (m*n) + rank + m - 1;
60847c6ae99SBarry Smith     n7 = size - (m*n) + rank + m;
60947c6ae99SBarry Smith     n8 = size - (m*n) + rank + m + 1;
61047c6ae99SBarry Smith   }
61147c6ae99SBarry Smith 
61247c6ae99SBarry Smith   if (ze == P) { /* First assume not corner or edge */
61347c6ae99SBarry Smith     n18 = (m*n) - (size-rank) - m - 1;
61447c6ae99SBarry Smith     n19 = (m*n) - (size-rank) - m;
61547c6ae99SBarry Smith     n20 = (m*n) - (size-rank) - m + 1;
61647c6ae99SBarry Smith     n21 = (m*n) - (size-rank) - 1;
61747c6ae99SBarry Smith     n22 = (m*n) - (size-rank);
61847c6ae99SBarry Smith     n23 = (m*n) - (size-rank) + 1;
61947c6ae99SBarry Smith     n24 = (m*n) - (size-rank) + m - 1;
62047c6ae99SBarry Smith     n25 = (m*n) - (size-rank) + m;
62147c6ae99SBarry Smith     n26 = (m*n) - (size-rank) + m + 1;
62247c6ae99SBarry Smith   }
62347c6ae99SBarry Smith 
62447c6ae99SBarry Smith   if ((xs==0) && (zs==0)) { /* Assume an edge, not corner */
62547c6ae99SBarry Smith     n0 = size - m*n + rank + m-1 - m;
62647c6ae99SBarry Smith     n3 = size - m*n + rank + m-1;
62747c6ae99SBarry Smith     n6 = size - m*n + rank + m-1 + m;
62847c6ae99SBarry Smith   }
62947c6ae99SBarry Smith 
63047c6ae99SBarry Smith   if ((xs==0) && (ze==P)) { /* Assume an edge, not corner */
63147c6ae99SBarry Smith     n18 = m*n - (size - rank) + m-1 - m;
63247c6ae99SBarry Smith     n21 = m*n - (size - rank) + m-1;
63347c6ae99SBarry Smith     n24 = m*n - (size - rank) + m-1 + m;
63447c6ae99SBarry Smith   }
63547c6ae99SBarry Smith 
63647c6ae99SBarry Smith   if ((xs==0) && (ys==0)) { /* Assume an edge, not corner */
63747c6ae99SBarry Smith     n0  = rank + m*n -1 - m*n;
63847c6ae99SBarry Smith     n9  = rank + m*n -1;
63947c6ae99SBarry Smith     n18 = rank + m*n -1 + m*n;
64047c6ae99SBarry Smith   }
64147c6ae99SBarry Smith 
64247c6ae99SBarry Smith   if ((xs==0) && (ye==N)) { /* Assume an edge, not corner */
64347c6ae99SBarry Smith     n6  = rank - m*(n-1) + m-1 - m*n;
64447c6ae99SBarry Smith     n15 = rank - m*(n-1) + m-1;
64547c6ae99SBarry Smith     n24 = rank - m*(n-1) + m-1 + m*n;
64647c6ae99SBarry Smith   }
64747c6ae99SBarry Smith 
648ce00eea3SSatish Balay   if ((xe==M) && (zs==0)) { /* Assume an edge, not corner */
64947c6ae99SBarry Smith     n2 = size - (m*n-rank) - (m-1) - m;
65047c6ae99SBarry Smith     n5 = size - (m*n-rank) - (m-1);
65147c6ae99SBarry Smith     n8 = size - (m*n-rank) - (m-1) + m;
65247c6ae99SBarry Smith   }
65347c6ae99SBarry Smith 
654ce00eea3SSatish Balay   if ((xe==M) && (ze==P)) { /* Assume an edge, not corner */
65547c6ae99SBarry Smith     n20 = m*n - (size - rank) - (m-1) - m;
65647c6ae99SBarry Smith     n23 = m*n - (size - rank) - (m-1);
65747c6ae99SBarry Smith     n26 = m*n - (size - rank) - (m-1) + m;
65847c6ae99SBarry Smith   }
65947c6ae99SBarry Smith 
660ce00eea3SSatish Balay   if ((xe==M) && (ys==0)) { /* Assume an edge, not corner */
66147c6ae99SBarry Smith     n2  = rank + m*(n-1) - (m-1) - m*n;
66247c6ae99SBarry Smith     n11 = rank + m*(n-1) - (m-1);
66347c6ae99SBarry Smith     n20 = rank + m*(n-1) - (m-1) + m*n;
66447c6ae99SBarry Smith   }
66547c6ae99SBarry Smith 
666ce00eea3SSatish Balay   if ((xe==M) && (ye==N)) { /* Assume an edge, not corner */
66747c6ae99SBarry Smith     n8  = rank - m*n +1 - m*n;
66847c6ae99SBarry Smith     n17 = rank - m*n +1;
66947c6ae99SBarry Smith     n26 = rank - m*n +1 + m*n;
67047c6ae99SBarry Smith   }
67147c6ae99SBarry Smith 
67247c6ae99SBarry Smith   if ((ys==0) && (zs==0)) { /* Assume an edge, not corner */
67347c6ae99SBarry Smith     n0 = size - m + rank -1;
67447c6ae99SBarry Smith     n1 = size - m + rank;
67547c6ae99SBarry Smith     n2 = size - m + rank +1;
67647c6ae99SBarry Smith   }
67747c6ae99SBarry Smith 
67847c6ae99SBarry Smith   if ((ys==0) && (ze==P)) { /* Assume an edge, not corner */
67947c6ae99SBarry Smith     n18 = m*n - (size - rank) + m*(n-1) -1;
68047c6ae99SBarry Smith     n19 = m*n - (size - rank) + m*(n-1);
68147c6ae99SBarry Smith     n20 = m*n - (size - rank) + m*(n-1) +1;
68247c6ae99SBarry Smith   }
68347c6ae99SBarry Smith 
68447c6ae99SBarry Smith   if ((ye==N) && (zs==0)) { /* Assume an edge, not corner */
68547c6ae99SBarry Smith     n6 = size - (m*n-rank) - m * (n-1) -1;
68647c6ae99SBarry Smith     n7 = size - (m*n-rank) - m * (n-1);
68747c6ae99SBarry Smith     n8 = size - (m*n-rank) - m * (n-1) +1;
68847c6ae99SBarry Smith   }
68947c6ae99SBarry Smith 
69047c6ae99SBarry Smith   if ((ye==N) && (ze==P)) { /* Assume an edge, not corner */
69147c6ae99SBarry Smith     n24 = rank - (size-m) -1;
69247c6ae99SBarry Smith     n25 = rank - (size-m);
69347c6ae99SBarry Smith     n26 = rank - (size-m) +1;
69447c6ae99SBarry Smith   }
69547c6ae99SBarry Smith 
69647c6ae99SBarry Smith   /* Check for Corners */
6978865f1eaSKarl Rupp   if ((xs==0) && (ys==0) && (zs==0)) n0  = size -1;
6988865f1eaSKarl Rupp   if ((xs==0) && (ys==0) && (ze==P)) n18 = m*n-1;
6998865f1eaSKarl Rupp   if ((xs==0) && (ye==N) && (zs==0)) n6  = (size-1)-m*(n-1);
7008865f1eaSKarl Rupp   if ((xs==0) && (ye==N) && (ze==P)) n24 = m-1;
7018865f1eaSKarl Rupp   if ((xe==M) && (ys==0) && (zs==0)) n2  = size-m;
7028865f1eaSKarl Rupp   if ((xe==M) && (ys==0) && (ze==P)) n20 = m*n-m;
7038865f1eaSKarl Rupp   if ((xe==M) && (ye==N) && (zs==0)) n8  = size-m*n;
7048865f1eaSKarl Rupp   if ((xe==M) && (ye==N) && (ze==P)) n26 = 0;
70547c6ae99SBarry Smith 
70647c6ae99SBarry Smith   /* Check for when not X,Y, and Z Periodic */
70747c6ae99SBarry Smith 
70847c6ae99SBarry Smith   /* If not X periodic */
709bff4a2f0SMatthew G. Knepley   if (bx != DM_BOUNDARY_PERIODIC) {
7108865f1eaSKarl Rupp     if (xs==0) n0 = n3 = n6 = n9  = n12 = n15 = n18 = n21 = n24 = -2;
7118865f1eaSKarl Rupp     if (xe==M) n2 = n5 = n8 = n11 = n14 = n17 = n20 = n23 = n26 = -2;
71247c6ae99SBarry Smith   }
71347c6ae99SBarry Smith 
71447c6ae99SBarry Smith   /* If not Y periodic */
715bff4a2f0SMatthew G. Knepley   if (by != DM_BOUNDARY_PERIODIC) {
7168865f1eaSKarl Rupp     if (ys==0) n0 = n1 = n2 = n9  = n10 = n11 = n18 = n19 = n20 = -2;
7178865f1eaSKarl Rupp     if (ye==N) n6 = n7 = n8 = n15 = n16 = n17 = n24 = n25 = n26 = -2;
71847c6ae99SBarry Smith   }
71947c6ae99SBarry Smith 
72047c6ae99SBarry Smith   /* If not Z periodic */
721bff4a2f0SMatthew G. Knepley   if (bz != DM_BOUNDARY_PERIODIC) {
7228865f1eaSKarl Rupp     if (zs==0) n0  = n1  = n2  = n3  = n4  = n5  = n6  = n7  = n8  = -2;
7238865f1eaSKarl Rupp     if (ze==P) n18 = n19 = n20 = n21 = n22 = n23 = n24 = n25 = n26 = -2;
72447c6ae99SBarry Smith   }
72547c6ae99SBarry Smith 
726785e854fSJed Brown   ierr = PetscMalloc1(27,&dd->neighbors);CHKERRQ(ierr);
7278865f1eaSKarl Rupp 
72847c6ae99SBarry Smith   dd->neighbors[0]  = n0;
72947c6ae99SBarry Smith   dd->neighbors[1]  = n1;
73047c6ae99SBarry Smith   dd->neighbors[2]  = n2;
73147c6ae99SBarry Smith   dd->neighbors[3]  = n3;
73247c6ae99SBarry Smith   dd->neighbors[4]  = n4;
73347c6ae99SBarry Smith   dd->neighbors[5]  = n5;
73447c6ae99SBarry Smith   dd->neighbors[6]  = n6;
73547c6ae99SBarry Smith   dd->neighbors[7]  = n7;
73647c6ae99SBarry Smith   dd->neighbors[8]  = n8;
73747c6ae99SBarry Smith   dd->neighbors[9]  = n9;
73847c6ae99SBarry Smith   dd->neighbors[10] = n10;
73947c6ae99SBarry Smith   dd->neighbors[11] = n11;
74047c6ae99SBarry Smith   dd->neighbors[12] = n12;
74147c6ae99SBarry Smith   dd->neighbors[13] = rank;
74247c6ae99SBarry Smith   dd->neighbors[14] = n14;
74347c6ae99SBarry Smith   dd->neighbors[15] = n15;
74447c6ae99SBarry Smith   dd->neighbors[16] = n16;
74547c6ae99SBarry Smith   dd->neighbors[17] = n17;
74647c6ae99SBarry Smith   dd->neighbors[18] = n18;
74747c6ae99SBarry Smith   dd->neighbors[19] = n19;
74847c6ae99SBarry Smith   dd->neighbors[20] = n20;
74947c6ae99SBarry Smith   dd->neighbors[21] = n21;
75047c6ae99SBarry Smith   dd->neighbors[22] = n22;
75147c6ae99SBarry Smith   dd->neighbors[23] = n23;
75247c6ae99SBarry Smith   dd->neighbors[24] = n24;
75347c6ae99SBarry Smith   dd->neighbors[25] = n25;
75447c6ae99SBarry Smith   dd->neighbors[26] = n26;
75547c6ae99SBarry Smith 
75647c6ae99SBarry Smith   /* If star stencil then delete the corner neighbors */
757d9c9ebe5SPeter Brune   if (stencil_type == DMDA_STENCIL_STAR) {
75847c6ae99SBarry Smith     /* save information about corner neighbors */
75947c6ae99SBarry Smith     sn0 = n0; sn1 = n1; sn2 = n2; sn3 = n3; sn5 = n5; sn6 = n6; sn7 = n7;
76047c6ae99SBarry Smith     sn8 = n8; sn9 = n9; sn11 = n11; sn15 = n15; sn17 = n17; sn18 = n18;
76147c6ae99SBarry Smith     sn19 = n19; sn20 = n20; sn21 = n21; sn23 = n23; sn24 = n24; sn25 = n25;
76247c6ae99SBarry Smith     sn26 = n26;
7638865f1eaSKarl Rupp     n0 = n1 = n2 = n3 = n5 = n6 = n7 = n8 = n9 = n11 = n15 = n17 = n18 = n19 = n20 = n21 = n23 = n24 = n25 = n26 = -1;
76447c6ae99SBarry Smith   }
76547c6ae99SBarry Smith 
766785e854fSJed Brown   ierr = PetscMalloc1((Xe-Xs)*(Ye-Ys)*(Ze-Zs),&idx);CHKERRQ(ierr);
76747c6ae99SBarry Smith 
76847c6ae99SBarry Smith   nn = 0;
76947c6ae99SBarry Smith   /* Bottom Level */
77047c6ae99SBarry Smith   for (k=0; k<s_z; k++) {
77147c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
77247c6ae99SBarry Smith       if (n0 >= 0) { /* left below */
773ce00eea3SSatish Balay         x_t = lx[n0 % m];
77447c6ae99SBarry Smith         y_t = ly[(n0 % (m*n))/m];
77547c6ae99SBarry Smith         z_t = lz[n0 / (m*n)];
77647c6ae99SBarry 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;
7778865f1eaSKarl Rupp         if (twod && (s_t < 0)) s_t = bases[n0] + x_t*y_t*z_t - (s_y-i)*x_t - s_x; /* 2D case */
7788865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
77947c6ae99SBarry Smith       }
78047c6ae99SBarry Smith       if (n1 >= 0) { /* directly below */
78147c6ae99SBarry Smith         x_t = x;
78247c6ae99SBarry Smith         y_t = ly[(n1 % (m*n))/m];
78347c6ae99SBarry Smith         z_t = lz[n1 / (m*n)];
78447c6ae99SBarry 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;
7858865f1eaSKarl Rupp         if (twod && (s_t < 0)) s_t = bases[n1] + x_t*y_t*z_t - (s_y+1-i)*x_t; /* 2D case */
7868865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
78747c6ae99SBarry Smith       }
78847c6ae99SBarry Smith       if (n2 >= 0) { /* right below */
789ce00eea3SSatish Balay         x_t = lx[n2 % m];
79047c6ae99SBarry Smith         y_t = ly[(n2 % (m*n))/m];
79147c6ae99SBarry Smith         z_t = lz[n2 / (m*n)];
79247c6ae99SBarry 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;
7938865f1eaSKarl Rupp         if (twod && (s_t < 0)) s_t = bases[n2] + x_t*y_t*z_t - (s_y+1-i)*x_t; /* 2D case */
7948865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
79547c6ae99SBarry Smith       }
79647c6ae99SBarry Smith     }
79747c6ae99SBarry Smith 
79847c6ae99SBarry Smith     for (i=0; i<y; i++) {
79947c6ae99SBarry Smith       if (n3 >= 0) { /* directly left */
800ce00eea3SSatish Balay         x_t = lx[n3 % m];
80147c6ae99SBarry Smith         y_t = y;
80247c6ae99SBarry Smith         z_t = lz[n3 / (m*n)];
80347c6ae99SBarry Smith         s_t = bases[n3] + (i+1)*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8048865f1eaSKarl Rupp         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 */
8058865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
80647c6ae99SBarry Smith       }
80747c6ae99SBarry Smith 
80847c6ae99SBarry Smith       if (n4 >= 0) { /* middle */
80947c6ae99SBarry Smith         x_t = x;
81047c6ae99SBarry Smith         y_t = y;
81147c6ae99SBarry Smith         z_t = lz[n4 / (m*n)];
81247c6ae99SBarry Smith         s_t = bases[n4] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8138865f1eaSKarl Rupp         if (twod && (s_t < 0)) s_t = bases[n4] + i*x_t + x_t*y_t*z_t - x_t*y_t; /* 2D case */
8148865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
815bff4a2f0SMatthew G. Knepley       } else if (bz == DM_BOUNDARY_MIRROR) {
8168865f1eaSKarl Rupp         for (j=0; j<x; j++) idx[nn++] = 0;
81747c6ae99SBarry Smith       }
81847c6ae99SBarry Smith 
81947c6ae99SBarry Smith       if (n5 >= 0) { /* directly right */
820ce00eea3SSatish Balay         x_t = lx[n5 % m];
82147c6ae99SBarry Smith         y_t = y;
82247c6ae99SBarry Smith         z_t = lz[n5 / (m*n)];
82347c6ae99SBarry Smith         s_t = bases[n5] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8248865f1eaSKarl Rupp         if (twod && (s_t < 0)) s_t = bases[n5] + i*x_t + x_t*y_t*z_t - x_t*y_t; /* 2D case */
8258865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
82647c6ae99SBarry Smith       }
82747c6ae99SBarry Smith     }
82847c6ae99SBarry Smith 
82947c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
83047c6ae99SBarry Smith       if (n6 >= 0) { /* left above */
831ce00eea3SSatish Balay         x_t = lx[n6 % m];
83247c6ae99SBarry Smith         y_t = ly[(n6 % (m*n))/m];
83347c6ae99SBarry Smith         z_t = lz[n6 / (m*n)];
83447c6ae99SBarry Smith         s_t = bases[n6] + i*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8358865f1eaSKarl Rupp         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 */
8368865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
83747c6ae99SBarry Smith       }
83847c6ae99SBarry Smith       if (n7 >= 0) { /* directly above */
83947c6ae99SBarry Smith         x_t = x;
84047c6ae99SBarry Smith         y_t = ly[(n7 % (m*n))/m];
84147c6ae99SBarry Smith         z_t = lz[n7 / (m*n)];
84247c6ae99SBarry Smith         s_t = bases[n7] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8438865f1eaSKarl Rupp         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 */
8448865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
84547c6ae99SBarry Smith       }
84647c6ae99SBarry Smith       if (n8 >= 0) { /* right above */
847ce00eea3SSatish Balay         x_t = lx[n8 % m];
84847c6ae99SBarry Smith         y_t = ly[(n8 % (m*n))/m];
84947c6ae99SBarry Smith         z_t = lz[n8 / (m*n)];
85047c6ae99SBarry Smith         s_t = bases[n8] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8518865f1eaSKarl Rupp         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 */
8528865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
85347c6ae99SBarry Smith       }
85447c6ae99SBarry Smith     }
85547c6ae99SBarry Smith   }
85647c6ae99SBarry Smith 
85747c6ae99SBarry Smith   /* Middle Level */
85847c6ae99SBarry Smith   for (k=0; k<z; k++) {
85947c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
86047c6ae99SBarry Smith       if (n9 >= 0) { /* left below */
861ce00eea3SSatish Balay         x_t = lx[n9 % m];
86247c6ae99SBarry Smith         y_t = ly[(n9 % (m*n))/m];
86347c6ae99SBarry Smith         /* z_t = z; */
86447c6ae99SBarry Smith         s_t = bases[n9] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
8658865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
86647c6ae99SBarry Smith       }
86747c6ae99SBarry Smith       if (n10 >= 0) { /* directly below */
86847c6ae99SBarry Smith         x_t = x;
86947c6ae99SBarry Smith         y_t = ly[(n10 % (m*n))/m];
87047c6ae99SBarry Smith         /* z_t = z; */
87147c6ae99SBarry Smith         s_t = bases[n10] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
8728865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
873bff4a2f0SMatthew G. Knepley       }  else if (by == DM_BOUNDARY_MIRROR) {
8748865f1eaSKarl Rupp         for (j=0; j<x; j++) idx[nn++] = 0;
87547c6ae99SBarry Smith       }
87647c6ae99SBarry Smith       if (n11 >= 0) { /* right below */
877ce00eea3SSatish Balay         x_t = lx[n11 % m];
87847c6ae99SBarry Smith         y_t = ly[(n11 % (m*n))/m];
87947c6ae99SBarry Smith         /* z_t = z; */
88047c6ae99SBarry Smith         s_t = bases[n11] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
8818865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
88247c6ae99SBarry Smith       }
88347c6ae99SBarry Smith     }
88447c6ae99SBarry Smith 
88547c6ae99SBarry Smith     for (i=0; i<y; i++) {
88647c6ae99SBarry Smith       if (n12 >= 0) { /* directly left */
887ce00eea3SSatish Balay         x_t = lx[n12 % m];
88847c6ae99SBarry Smith         y_t = y;
88947c6ae99SBarry Smith         /* z_t = z; */
89047c6ae99SBarry Smith         s_t = bases[n12] + (i+1)*x_t - s_x + k*x_t*y_t;
8918865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
892bff4a2f0SMatthew G. Knepley       }  else if (bx == DM_BOUNDARY_MIRROR) {
8938865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = 0;
89447c6ae99SBarry Smith       }
89547c6ae99SBarry Smith 
89647c6ae99SBarry Smith       /* Interior */
89747c6ae99SBarry Smith       s_t = bases[rank] + i*x + k*x*y;
8988865f1eaSKarl Rupp       for (j=0; j<x; j++) idx[nn++] = s_t++;
89947c6ae99SBarry Smith 
90047c6ae99SBarry Smith       if (n14 >= 0) { /* directly right */
901ce00eea3SSatish Balay         x_t = lx[n14 % m];
90247c6ae99SBarry Smith         y_t = y;
90347c6ae99SBarry Smith         /* z_t = z; */
90447c6ae99SBarry Smith         s_t = bases[n14] + i*x_t + k*x_t*y_t;
9058865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
906bff4a2f0SMatthew G. Knepley       } else if (bx == DM_BOUNDARY_MIRROR) {
9078865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = 0;
90847c6ae99SBarry Smith       }
90947c6ae99SBarry Smith     }
91047c6ae99SBarry Smith 
91147c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
91247c6ae99SBarry Smith       if (n15 >= 0) { /* left above */
913ce00eea3SSatish Balay         x_t = lx[n15 % m];
91447c6ae99SBarry Smith         y_t = ly[(n15 % (m*n))/m];
91547c6ae99SBarry Smith         /* z_t = z; */
91647c6ae99SBarry Smith         s_t = bases[n15] + i*x_t - s_x + k*x_t*y_t;
9178865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
91847c6ae99SBarry Smith       }
91947c6ae99SBarry Smith       if (n16 >= 0) { /* directly above */
92047c6ae99SBarry Smith         x_t = x;
92147c6ae99SBarry Smith         y_t = ly[(n16 % (m*n))/m];
92247c6ae99SBarry Smith         /* z_t = z; */
92347c6ae99SBarry Smith         s_t = bases[n16] + (i-1)*x_t + k*x_t*y_t;
9248865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
925bff4a2f0SMatthew G. Knepley       } else if (by == DM_BOUNDARY_MIRROR) {
9268865f1eaSKarl Rupp         for (j=0; j<x; j++) idx[nn++] = 0;
92747c6ae99SBarry Smith       }
92847c6ae99SBarry Smith       if (n17 >= 0) { /* right above */
929ce00eea3SSatish Balay         x_t = lx[n17 % m];
93047c6ae99SBarry Smith         y_t = ly[(n17 % (m*n))/m];
93147c6ae99SBarry Smith         /* z_t = z; */
93247c6ae99SBarry Smith         s_t = bases[n17] + (i-1)*x_t + k*x_t*y_t;
9338865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
93447c6ae99SBarry Smith       }
93547c6ae99SBarry Smith     }
93647c6ae99SBarry Smith   }
93747c6ae99SBarry Smith 
93847c6ae99SBarry Smith   /* Upper Level */
93947c6ae99SBarry Smith   for (k=0; k<s_z; k++) {
94047c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
94147c6ae99SBarry Smith       if (n18 >= 0) { /* left below */
942ce00eea3SSatish Balay         x_t = lx[n18 % m];
94347c6ae99SBarry Smith         y_t = ly[(n18 % (m*n))/m];
94447c6ae99SBarry Smith         /* z_t = lz[n18 / (m*n)]; */
94547c6ae99SBarry Smith         s_t = bases[n18] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
9468865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n18] - (s_y-i)*x_t -s_x + x_t*y_t; /* 2d case */
9478865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
94847c6ae99SBarry Smith       }
94947c6ae99SBarry Smith       if (n19 >= 0) { /* directly below */
95047c6ae99SBarry Smith         x_t = x;
95147c6ae99SBarry Smith         y_t = ly[(n19 % (m*n))/m];
95247c6ae99SBarry Smith         /* z_t = lz[n19 / (m*n)]; */
95347c6ae99SBarry Smith         s_t = bases[n19] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
9548865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n19] - (s_y+1-i)*x_t + x_t*y_t; /* 2d case */
9558865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
95647c6ae99SBarry Smith       }
95747c6ae99SBarry Smith       if (n20 >= 0) { /* right below */
958ce00eea3SSatish Balay         x_t = lx[n20 % m];
95947c6ae99SBarry Smith         y_t = ly[(n20 % (m*n))/m];
96047c6ae99SBarry Smith         /* z_t = lz[n20 / (m*n)]; */
96147c6ae99SBarry Smith         s_t = bases[n20] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
9628865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n20] - (s_y+1-i)*x_t + x_t*y_t; /* 2d case */
9638865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
96447c6ae99SBarry Smith       }
96547c6ae99SBarry Smith     }
96647c6ae99SBarry Smith 
96747c6ae99SBarry Smith     for (i=0; i<y; i++) {
96847c6ae99SBarry Smith       if (n21 >= 0) { /* directly left */
969ce00eea3SSatish Balay         x_t = lx[n21 % m];
97047c6ae99SBarry Smith         y_t = y;
97147c6ae99SBarry Smith         /* z_t = lz[n21 / (m*n)]; */
97247c6ae99SBarry Smith         s_t = bases[n21] + (i+1)*x_t - s_x + k*x_t*y_t;
9738865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n21] + (i+1)*x_t - s_x;  /* 2d case */
9748865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
97547c6ae99SBarry Smith       }
97647c6ae99SBarry Smith 
97747c6ae99SBarry Smith       if (n22 >= 0) { /* middle */
97847c6ae99SBarry Smith         x_t = x;
97947c6ae99SBarry Smith         y_t = y;
98047c6ae99SBarry Smith         /* z_t = lz[n22 / (m*n)]; */
98147c6ae99SBarry Smith         s_t = bases[n22] + i*x_t + k*x_t*y_t;
9828865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n22] + i*x_t; /* 2d case */
9838865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
984bff4a2f0SMatthew G. Knepley       } else if (bz == DM_BOUNDARY_MIRROR) {
9858865f1eaSKarl Rupp         for (j=0; j<x; j++) idx[nn++] = 0;
98647c6ae99SBarry Smith       }
98747c6ae99SBarry Smith 
98847c6ae99SBarry Smith       if (n23 >= 0) { /* directly right */
989ce00eea3SSatish Balay         x_t = lx[n23 % m];
99047c6ae99SBarry Smith         y_t = y;
99147c6ae99SBarry Smith         /* z_t = lz[n23 / (m*n)]; */
99247c6ae99SBarry Smith         s_t = bases[n23] + i*x_t + k*x_t*y_t;
9938865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n23] + i*x_t; /* 2d case */
9948865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
99547c6ae99SBarry Smith       }
99647c6ae99SBarry Smith     }
99747c6ae99SBarry Smith 
99847c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
99947c6ae99SBarry Smith       if (n24 >= 0) { /* left above */
1000ce00eea3SSatish Balay         x_t = lx[n24 % m];
100147c6ae99SBarry Smith         y_t = ly[(n24 % (m*n))/m];
100247c6ae99SBarry Smith         /* z_t = lz[n24 / (m*n)]; */
100347c6ae99SBarry Smith         s_t = bases[n24] + i*x_t - s_x + k*x_t*y_t;
10048865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n24] + i*x_t - s_x; /* 2d case */
10058865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
100647c6ae99SBarry Smith       }
100747c6ae99SBarry Smith       if (n25 >= 0) { /* directly above */
100847c6ae99SBarry Smith         x_t = x;
100947c6ae99SBarry Smith         y_t = ly[(n25 % (m*n))/m];
101047c6ae99SBarry Smith         /* z_t = lz[n25 / (m*n)]; */
101147c6ae99SBarry Smith         s_t = bases[n25] + (i-1)*x_t + k*x_t*y_t;
10128865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n25] + (i-1)*x_t; /* 2d case */
10138865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
101447c6ae99SBarry Smith       }
101547c6ae99SBarry Smith       if (n26 >= 0) { /* right above */
1016ce00eea3SSatish Balay         x_t = lx[n26 % m];
101747c6ae99SBarry Smith         y_t = ly[(n26 % (m*n))/m];
101847c6ae99SBarry Smith         /* z_t = lz[n26 / (m*n)]; */
101947c6ae99SBarry Smith         s_t = bases[n26] + (i-1)*x_t + k*x_t*y_t;
10208865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n26] + (i-1)*x_t; /* 2d case */
10218865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
102247c6ae99SBarry Smith       }
102347c6ae99SBarry Smith     }
102447c6ae99SBarry Smith   }
1025ce00eea3SSatish Balay 
1026b1fb7eb7SBarry Smith   ierr = ISCreateBlock(comm,dof,nn,idx,PETSC_USE_POINTER,&from);CHKERRQ(ierr);
102747c6ae99SBarry Smith   ierr = VecScatterCreate(global,from,local,to,&gtol);CHKERRQ(ierr);
10283bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)da,(PetscObject)gtol);CHKERRQ(ierr);
1029fcfd50ebSBarry Smith   ierr = ISDestroy(&to);CHKERRQ(ierr);
1030fcfd50ebSBarry Smith   ierr = ISDestroy(&from);CHKERRQ(ierr);
103147c6ae99SBarry Smith 
1032d9c9ebe5SPeter Brune   if (stencil_type == DMDA_STENCIL_STAR) {
103347c6ae99SBarry Smith     n0  = sn0;  n1  = sn1;  n2  = sn2;  n3  = sn3;  n5  = sn5;  n6  = sn6; n7 = sn7;
103447c6ae99SBarry Smith     n8  = sn8;  n9  = sn9;  n11 = sn11; n15 = sn15; n17 = sn17; n18 = sn18;
103547c6ae99SBarry Smith     n19 = sn19; n20 = sn20; n21 = sn21; n23 = sn23; n24 = sn24; n25 = sn25;
103647c6ae99SBarry Smith     n26 = sn26;
1037ce00eea3SSatish Balay   }
103847c6ae99SBarry Smith 
103988661749SPeter Brune   if (((stencil_type == DMDA_STENCIL_STAR) ||
1040bff4a2f0SMatthew G. Knepley       (bx != DM_BOUNDARY_PERIODIC && bx) ||
1041bff4a2f0SMatthew G. Knepley       (by != DM_BOUNDARY_PERIODIC && by) ||
1042bff4a2f0SMatthew G. Knepley        (bz != DM_BOUNDARY_PERIODIC && bz))) {
1043ce00eea3SSatish Balay     /*
1044ce00eea3SSatish Balay         Recompute the local to global mappings, this time keeping the
1045ce00eea3SSatish Balay       information about the cross corner processor numbers.
1046ce00eea3SSatish Balay     */
104747c6ae99SBarry Smith     nn = 0;
104847c6ae99SBarry Smith     /* Bottom Level */
104947c6ae99SBarry Smith     for (k=0; k<s_z; k++) {
105047c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
105147c6ae99SBarry Smith         if (n0 >= 0) { /* left below */
1052ce00eea3SSatish Balay           x_t = lx[n0 % m];
105347c6ae99SBarry Smith           y_t = ly[(n0 % (m*n))/m];
105447c6ae99SBarry Smith           z_t = lz[n0 / (m*n)];
105547c6ae99SBarry 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;
10568865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1057ce00eea3SSatish Balay         } else if (Xs-xs < 0 && Ys-ys < 0 && Zs-zs < 0) {
10588865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
105947c6ae99SBarry Smith         }
106047c6ae99SBarry Smith         if (n1 >= 0) { /* directly below */
106147c6ae99SBarry Smith           x_t = x;
106247c6ae99SBarry Smith           y_t = ly[(n1 % (m*n))/m];
106347c6ae99SBarry Smith           z_t = lz[n1 / (m*n)];
106447c6ae99SBarry 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;
10658865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1066ce00eea3SSatish Balay         } else if (Ys-ys < 0 && Zs-zs < 0) {
10678865f1eaSKarl Rupp           for (j=0; j<x; j++) idx[nn++] = -1;
106847c6ae99SBarry Smith         }
106947c6ae99SBarry Smith         if (n2 >= 0) { /* right below */
1070ce00eea3SSatish Balay           x_t = lx[n2 % m];
107147c6ae99SBarry Smith           y_t = ly[(n2 % (m*n))/m];
107247c6ae99SBarry Smith           z_t = lz[n2 / (m*n)];
107347c6ae99SBarry 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;
10748865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1075ce00eea3SSatish Balay         } else if (xe-Xe < 0 && Ys-ys < 0 && Zs-zs < 0) {
10768865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
107747c6ae99SBarry Smith         }
107847c6ae99SBarry Smith       }
107947c6ae99SBarry Smith 
108047c6ae99SBarry Smith       for (i=0; i<y; i++) {
108147c6ae99SBarry Smith         if (n3 >= 0) { /* directly left */
1082ce00eea3SSatish Balay           x_t = lx[n3 % m];
108347c6ae99SBarry Smith           y_t = y;
108447c6ae99SBarry Smith           z_t = lz[n3 / (m*n)];
108547c6ae99SBarry Smith           s_t = bases[n3] + (i+1)*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
10868865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1087ce00eea3SSatish Balay         } else if (Xs-xs < 0 && Zs-zs < 0) {
10888865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
108947c6ae99SBarry Smith         }
109047c6ae99SBarry Smith 
109147c6ae99SBarry Smith         if (n4 >= 0) { /* middle */
109247c6ae99SBarry Smith           x_t = x;
109347c6ae99SBarry Smith           y_t = y;
109447c6ae99SBarry Smith           z_t = lz[n4 / (m*n)];
109547c6ae99SBarry Smith           s_t = bases[n4] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
10968865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1097ce00eea3SSatish Balay         } else if (Zs-zs < 0) {
1098bff4a2f0SMatthew G. Knepley           if (bz == DM_BOUNDARY_MIRROR) {
10998865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = 0;
1100c2859e5eSBarry Smith           } else {
11018865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = -1;
110247c6ae99SBarry Smith           }
1103c2859e5eSBarry Smith         }
110447c6ae99SBarry Smith 
110547c6ae99SBarry Smith         if (n5 >= 0) { /* directly right */
1106ce00eea3SSatish Balay           x_t = lx[n5 % m];
110747c6ae99SBarry Smith           y_t = y;
110847c6ae99SBarry Smith           z_t = lz[n5 / (m*n)];
110947c6ae99SBarry Smith           s_t = bases[n5] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
11108865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1111ce00eea3SSatish Balay         } else if (xe-Xe < 0 && Zs-zs < 0) {
11128865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
111347c6ae99SBarry Smith         }
111447c6ae99SBarry Smith       }
111547c6ae99SBarry Smith 
111647c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
111747c6ae99SBarry Smith         if (n6 >= 0) { /* left above */
1118ce00eea3SSatish Balay           x_t = lx[n6 % m];
111947c6ae99SBarry Smith           y_t = ly[(n6 % (m*n))/m];
112047c6ae99SBarry Smith           z_t = lz[n6 / (m*n)];
112147c6ae99SBarry Smith           s_t = bases[n6] + i*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
11228865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1123ce00eea3SSatish Balay         } else if (Xs-xs < 0 && ye-Ye < 0 && Zs-zs < 0) {
11248865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
112547c6ae99SBarry Smith         }
112647c6ae99SBarry Smith         if (n7 >= 0) { /* directly above */
112747c6ae99SBarry Smith           x_t = x;
112847c6ae99SBarry Smith           y_t = ly[(n7 % (m*n))/m];
112947c6ae99SBarry Smith           z_t = lz[n7 / (m*n)];
113047c6ae99SBarry Smith           s_t = bases[n7] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
11318865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1132ce00eea3SSatish Balay         } else if (ye-Ye < 0 && Zs-zs < 0) {
11338865f1eaSKarl Rupp           for (j=0; j<x; j++) idx[nn++] = -1;
113447c6ae99SBarry Smith         }
113547c6ae99SBarry Smith         if (n8 >= 0) { /* right above */
1136ce00eea3SSatish Balay           x_t = lx[n8 % m];
113747c6ae99SBarry Smith           y_t = ly[(n8 % (m*n))/m];
113847c6ae99SBarry Smith           z_t = lz[n8 / (m*n)];
113947c6ae99SBarry Smith           s_t = bases[n8] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
11408865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1141ce00eea3SSatish Balay         } else if (xe-Xe < 0 && ye-Ye < 0 && Zs-zs < 0) {
11428865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
114347c6ae99SBarry Smith         }
114447c6ae99SBarry Smith       }
114547c6ae99SBarry Smith     }
114647c6ae99SBarry Smith 
114747c6ae99SBarry Smith     /* Middle Level */
114847c6ae99SBarry Smith     for (k=0; k<z; k++) {
114947c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
115047c6ae99SBarry Smith         if (n9 >= 0) { /* left below */
1151ce00eea3SSatish Balay           x_t = lx[n9 % m];
115247c6ae99SBarry Smith           y_t = ly[(n9 % (m*n))/m];
115347c6ae99SBarry Smith           /* z_t = z; */
115447c6ae99SBarry Smith           s_t = bases[n9] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
11558865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1156ce00eea3SSatish Balay         } else if (Xs-xs < 0 && Ys-ys < 0) {
11578865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
115847c6ae99SBarry Smith         }
115947c6ae99SBarry Smith         if (n10 >= 0) { /* directly below */
116047c6ae99SBarry Smith           x_t = x;
116147c6ae99SBarry Smith           y_t = ly[(n10 % (m*n))/m];
116247c6ae99SBarry Smith           /* z_t = z; */
116347c6ae99SBarry Smith           s_t = bases[n10] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
11648865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1165ce00eea3SSatish Balay         } else if (Ys-ys < 0) {
1166bff4a2f0SMatthew G. Knepley           if (by == DM_BOUNDARY_MIRROR) {
11678865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = -1;
1168c2859e5eSBarry Smith           } else {
11698865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = -1;
1170c2859e5eSBarry Smith           }
117147c6ae99SBarry Smith         }
117247c6ae99SBarry Smith         if (n11 >= 0) { /* right below */
1173ce00eea3SSatish Balay           x_t = lx[n11 % m];
117447c6ae99SBarry Smith           y_t = ly[(n11 % (m*n))/m];
117547c6ae99SBarry Smith           /* z_t = z; */
117647c6ae99SBarry Smith           s_t = bases[n11] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
11778865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1178ce00eea3SSatish Balay         } else if (xe-Xe < 0 && Ys-ys < 0) {
11798865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
118047c6ae99SBarry Smith         }
118147c6ae99SBarry Smith       }
118247c6ae99SBarry Smith 
118347c6ae99SBarry Smith       for (i=0; i<y; i++) {
118447c6ae99SBarry Smith         if (n12 >= 0) { /* directly left */
1185ce00eea3SSatish Balay           x_t = lx[n12 % m];
118647c6ae99SBarry Smith           y_t = y;
118747c6ae99SBarry Smith           /* z_t = z; */
118847c6ae99SBarry Smith           s_t = bases[n12] + (i+1)*x_t - s_x + k*x_t*y_t;
11898865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1190ce00eea3SSatish Balay         } else if (Xs-xs < 0) {
1191bff4a2f0SMatthew G. Knepley           if (bx == DM_BOUNDARY_MIRROR) {
11928865f1eaSKarl Rupp             for (j=0; j<s_x; j++) idx[nn++] = 0;
1193c2859e5eSBarry Smith           } else {
11948865f1eaSKarl Rupp             for (j=0; j<s_x; j++) idx[nn++] = -1;
119547c6ae99SBarry Smith           }
1196c2859e5eSBarry Smith         }
119747c6ae99SBarry Smith 
119847c6ae99SBarry Smith         /* Interior */
119947c6ae99SBarry Smith         s_t = bases[rank] + i*x + k*x*y;
12008865f1eaSKarl Rupp         for (j=0; j<x; j++) idx[nn++] = s_t++;
120147c6ae99SBarry Smith 
120247c6ae99SBarry Smith         if (n14 >= 0) { /* directly right */
1203ce00eea3SSatish Balay           x_t = lx[n14 % m];
120447c6ae99SBarry Smith           y_t = y;
120547c6ae99SBarry Smith           /* z_t = z; */
120647c6ae99SBarry Smith           s_t = bases[n14] + i*x_t + k*x_t*y_t;
12078865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1208ce00eea3SSatish Balay         } else if (xe-Xe < 0) {
1209bff4a2f0SMatthew G. Knepley           if (bx == DM_BOUNDARY_MIRROR) {
12108865f1eaSKarl Rupp             for (j=0; j<s_x; j++) idx[nn++] = 0;
1211c2859e5eSBarry Smith           } else {
12128865f1eaSKarl Rupp             for (j=0; j<s_x; j++) idx[nn++] = -1;
121347c6ae99SBarry Smith           }
121447c6ae99SBarry Smith         }
1215c2859e5eSBarry Smith       }
121647c6ae99SBarry Smith 
121747c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
121847c6ae99SBarry Smith         if (n15 >= 0) { /* left above */
1219ce00eea3SSatish Balay           x_t = lx[n15 % m];
122047c6ae99SBarry Smith           y_t = ly[(n15 % (m*n))/m];
122147c6ae99SBarry Smith           /* z_t = z; */
122247c6ae99SBarry Smith           s_t = bases[n15] + i*x_t - s_x + k*x_t*y_t;
12238865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1224ce00eea3SSatish Balay         } else if (Xs-xs < 0 && ye-Ye < 0) {
12258865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
122647c6ae99SBarry Smith         }
122747c6ae99SBarry Smith         if (n16 >= 0) { /* directly above */
122847c6ae99SBarry Smith           x_t = x;
122947c6ae99SBarry Smith           y_t = ly[(n16 % (m*n))/m];
123047c6ae99SBarry Smith           /* z_t = z; */
123147c6ae99SBarry Smith           s_t = bases[n16] + (i-1)*x_t + k*x_t*y_t;
12328865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1233ce00eea3SSatish Balay         } else if (ye-Ye < 0) {
1234bff4a2f0SMatthew G. Knepley           if (by == DM_BOUNDARY_MIRROR) {
12358865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = 0;
1236c2859e5eSBarry Smith           } else {
12378865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = -1;
123847c6ae99SBarry Smith           }
1239c2859e5eSBarry Smith         }
124047c6ae99SBarry Smith         if (n17 >= 0) { /* right above */
1241ce00eea3SSatish Balay           x_t = lx[n17 % m];
124247c6ae99SBarry Smith           y_t = ly[(n17 % (m*n))/m];
124347c6ae99SBarry Smith           /* z_t = z; */
124447c6ae99SBarry Smith           s_t = bases[n17] + (i-1)*x_t + k*x_t*y_t;
12458865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1246ce00eea3SSatish Balay         } else if (xe-Xe < 0 && ye-Ye < 0) {
12478865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
124847c6ae99SBarry Smith         }
124947c6ae99SBarry Smith       }
125047c6ae99SBarry Smith     }
125147c6ae99SBarry Smith 
125247c6ae99SBarry Smith     /* Upper Level */
125347c6ae99SBarry Smith     for (k=0; k<s_z; k++) {
125447c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
125547c6ae99SBarry Smith         if (n18 >= 0) { /* left below */
1256ce00eea3SSatish Balay           x_t = lx[n18 % m];
125747c6ae99SBarry Smith           y_t = ly[(n18 % (m*n))/m];
125847c6ae99SBarry Smith           /* z_t = lz[n18 / (m*n)]; */
125947c6ae99SBarry Smith           s_t = bases[n18] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
12608865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1261ce00eea3SSatish Balay         } else if (Xs-xs < 0 && Ys-ys < 0 && ze-Ze < 0) {
12628865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
126347c6ae99SBarry Smith         }
126447c6ae99SBarry Smith         if (n19 >= 0) { /* directly below */
126547c6ae99SBarry Smith           x_t = x;
126647c6ae99SBarry Smith           y_t = ly[(n19 % (m*n))/m];
126747c6ae99SBarry Smith           /* z_t = lz[n19 / (m*n)]; */
126847c6ae99SBarry Smith           s_t = bases[n19] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
12698865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1270ce00eea3SSatish Balay         } else if (Ys-ys < 0 && ze-Ze < 0) {
12718865f1eaSKarl Rupp           for (j=0; j<x; j++) idx[nn++] = -1;
127247c6ae99SBarry Smith         }
127347c6ae99SBarry Smith         if (n20 >= 0) { /* right below */
1274ce00eea3SSatish Balay           x_t = lx[n20 % m];
127547c6ae99SBarry Smith           y_t = ly[(n20 % (m*n))/m];
127647c6ae99SBarry Smith           /* z_t = lz[n20 / (m*n)]; */
127747c6ae99SBarry Smith           s_t = bases[n20] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
12788865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1279ce00eea3SSatish Balay         } else if (xe-Xe < 0 && Ys-ys < 0 && ze-Ze < 0) {
12808865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
128147c6ae99SBarry Smith         }
128247c6ae99SBarry Smith       }
128347c6ae99SBarry Smith 
128447c6ae99SBarry Smith       for (i=0; i<y; i++) {
128547c6ae99SBarry Smith         if (n21 >= 0) { /* directly left */
1286ce00eea3SSatish Balay           x_t = lx[n21 % m];
128747c6ae99SBarry Smith           y_t = y;
128847c6ae99SBarry Smith           /* z_t = lz[n21 / (m*n)]; */
128947c6ae99SBarry Smith           s_t = bases[n21] + (i+1)*x_t - s_x + k*x_t*y_t;
12908865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1291ce00eea3SSatish Balay         } else if (Xs-xs < 0 && ze-Ze < 0) {
12928865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
129347c6ae99SBarry Smith         }
129447c6ae99SBarry Smith 
129547c6ae99SBarry Smith         if (n22 >= 0) { /* middle */
129647c6ae99SBarry Smith           x_t = x;
129747c6ae99SBarry Smith           y_t = y;
129847c6ae99SBarry Smith           /* z_t = lz[n22 / (m*n)]; */
129947c6ae99SBarry Smith           s_t = bases[n22] + i*x_t + k*x_t*y_t;
13008865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1301ce00eea3SSatish Balay         } else if (ze-Ze < 0) {
1302bff4a2f0SMatthew G. Knepley           if (bz == DM_BOUNDARY_MIRROR) {
13038865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = 0;
1304c2859e5eSBarry Smith           } else {
13058865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = -1;
130647c6ae99SBarry Smith           }
1307c2859e5eSBarry Smith         }
130847c6ae99SBarry Smith 
130947c6ae99SBarry Smith         if (n23 >= 0) { /* directly right */
1310ce00eea3SSatish Balay           x_t = lx[n23 % m];
131147c6ae99SBarry Smith           y_t = y;
131247c6ae99SBarry Smith           /* z_t = lz[n23 / (m*n)]; */
131347c6ae99SBarry Smith           s_t = bases[n23] + i*x_t + k*x_t*y_t;
13148865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1315ce00eea3SSatish Balay         } else if (xe-Xe < 0 && ze-Ze < 0) {
13168865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
131747c6ae99SBarry Smith         }
131847c6ae99SBarry Smith       }
131947c6ae99SBarry Smith 
132047c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
132147c6ae99SBarry Smith         if (n24 >= 0) { /* left above */
1322ce00eea3SSatish Balay           x_t = lx[n24 % m];
132347c6ae99SBarry Smith           y_t = ly[(n24 % (m*n))/m];
132447c6ae99SBarry Smith           /* z_t = lz[n24 / (m*n)]; */
132547c6ae99SBarry Smith           s_t = bases[n24] + i*x_t - s_x + k*x_t*y_t;
13268865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1327ce00eea3SSatish Balay         } else if (Xs-xs < 0 && ye-Ye < 0 && ze-Ze < 0) {
13288865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
132947c6ae99SBarry Smith         }
133047c6ae99SBarry Smith         if (n25 >= 0) { /* directly above */
133147c6ae99SBarry Smith           x_t = x;
133247c6ae99SBarry Smith           y_t = ly[(n25 % (m*n))/m];
133347c6ae99SBarry Smith           /* z_t = lz[n25 / (m*n)]; */
133447c6ae99SBarry Smith           s_t = bases[n25] + (i-1)*x_t + k*x_t*y_t;
13358865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1336ce00eea3SSatish Balay         } else if (ye-Ye < 0 && ze-Ze < 0) {
13378865f1eaSKarl Rupp           for (j=0; j<x; j++) idx[nn++] = -1;
133847c6ae99SBarry Smith         }
133947c6ae99SBarry Smith         if (n26 >= 0) { /* right above */
1340ce00eea3SSatish Balay           x_t = lx[n26 % m];
134147c6ae99SBarry Smith           y_t = ly[(n26 % (m*n))/m];
134247c6ae99SBarry Smith           /* z_t = lz[n26 / (m*n)]; */
134347c6ae99SBarry Smith           s_t = bases[n26] + (i-1)*x_t + k*x_t*y_t;
13448865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1345ce00eea3SSatish Balay         } else if (xe-Xe < 0 && ye-Ye < 0 && ze-Ze < 0) {
13468865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
134747c6ae99SBarry Smith         }
134847c6ae99SBarry Smith       }
134947c6ae99SBarry Smith     }
135047c6ae99SBarry Smith   }
135147c6ae99SBarry Smith   /*
135247c6ae99SBarry Smith      Set the local to global ordering in the global vector, this allows use
135347c6ae99SBarry Smith      of VecSetValuesLocal().
135447c6ae99SBarry Smith   */
135545b6f7e9SBarry Smith   ierr = ISLocalToGlobalMappingCreate(comm,dof,nn,idx,PETSC_OWN_POINTER,&da->ltogmap);CHKERRQ(ierr);
13563bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)da,(PetscObject)da->ltogmap);CHKERRQ(ierr);
135747c6ae99SBarry Smith 
1358ce00eea3SSatish Balay   ierr  = PetscFree2(bases,ldims);CHKERRQ(ierr);
1359ce00eea3SSatish Balay   dd->m = m;  dd->n  = n;  dd->p  = p;
1360ce00eea3SSatish Balay   /* note petsc expects xs/xe/Xs/Xe to be multiplied by #dofs in many places */
1361ce00eea3SSatish Balay   dd->xs = xs*dof; dd->xe = xe*dof; dd->ys = ys; dd->ye = ye; dd->zs = zs; dd->ze = ze;
1362ce00eea3SSatish Balay   dd->Xs = Xs*dof; dd->Xe = Xe*dof; dd->Ys = Ys; dd->Ye = Ye; dd->Zs = Zs; dd->Ze = Ze;
1363ce00eea3SSatish Balay 
1364fcfd50ebSBarry Smith   ierr = VecDestroy(&local);CHKERRQ(ierr);
1365fcfd50ebSBarry Smith   ierr = VecDestroy(&global);CHKERRQ(ierr);
1366ce00eea3SSatish Balay 
1367ce00eea3SSatish Balay   dd->gtol      = gtol;
1368ce00eea3SSatish Balay   dd->base      = base;
1369ce00eea3SSatish Balay   da->ops->view = DMView_DA_3d;
13700298fd71SBarry Smith   dd->ltol      = NULL;
13710298fd71SBarry Smith   dd->ao        = NULL;
137247c6ae99SBarry Smith   PetscFunctionReturn(0);
137347c6ae99SBarry Smith }
1374564755cdSBarry Smith 
137547c6ae99SBarry Smith 
137647c6ae99SBarry Smith #undef __FUNCT__
1377aa219208SBarry Smith #define __FUNCT__ "DMDACreate3d"
137847c6ae99SBarry Smith /*@C
1379aa219208SBarry Smith    DMDACreate3d - Creates an object that will manage the communication of three-dimensional
138047c6ae99SBarry Smith    regular array data that is distributed across some processors.
138147c6ae99SBarry Smith 
138247c6ae99SBarry Smith    Collective on MPI_Comm
138347c6ae99SBarry Smith 
138447c6ae99SBarry Smith    Input Parameters:
138547c6ae99SBarry Smith +  comm - MPI communicator
13861321219cSEthan Coon .  bx,by,bz - type of ghost nodes the array have.
1387bff4a2f0SMatthew G. Knepley          Use one of DM_BOUNDARY_NONE, DM_BOUNDARY_GHOSTED, DM_BOUNDARY_PERIODIC.
1388aa219208SBarry Smith .  stencil_type - Type of stencil (DMDA_STENCIL_STAR or DMDA_STENCIL_BOX)
138947c6ae99SBarry 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
139047c6ae99SBarry Smith             from the command line with -da_grid_x <M> -da_grid_y <N> -da_grid_z <P>)
139147c6ae99SBarry Smith .  m,n,p - corresponding number of processors in each dimension
139247c6ae99SBarry Smith            (or PETSC_DECIDE to have calculated)
139347c6ae99SBarry Smith .  dof - number of degrees of freedom per node
139410d7c030SMatthew G Knepley .  s - stencil width
139510d7c030SMatthew G Knepley -  lx, ly, lz - arrays containing the number of nodes in each cell along
13960298fd71SBarry Smith           the x, y, and z coordinates, or NULL. If non-null, these
139747c6ae99SBarry Smith           must be of length as m,n,p and the corresponding
139847c6ae99SBarry Smith           m,n, or p cannot be PETSC_DECIDE. Sum of the lx[] entries must be M, sum of
139947c6ae99SBarry Smith           the ly[] must N, sum of the lz[] must be P
140047c6ae99SBarry Smith 
140147c6ae99SBarry Smith    Output Parameter:
140247c6ae99SBarry Smith .  da - the resulting distributed array object
140347c6ae99SBarry Smith 
140447c6ae99SBarry Smith    Options Database Key:
1405706a11cbSBarry Smith +  -dm_view - Calls DMView() at the conclusion of DMDACreate3d()
140647c6ae99SBarry Smith .  -da_grid_x <nx> - number of grid points in x direction, if M < 0
140747c6ae99SBarry Smith .  -da_grid_y <ny> - number of grid points in y direction, if N < 0
140847c6ae99SBarry Smith .  -da_grid_z <nz> - number of grid points in z direction, if P < 0
140947c6ae99SBarry Smith .  -da_processors_x <MX> - number of processors in x direction
141047c6ae99SBarry Smith .  -da_processors_y <MY> - number of processors in y direction
141147c6ae99SBarry Smith .  -da_processors_z <MZ> - number of processors in z direction
1412e0f5d30fSBarry Smith .  -da_refine_x <rx> - refinement ratio in x direction
1413e0f5d30fSBarry Smith .  -da_refine_y <ry> - refinement ratio in y direction
1414e0f5d30fSBarry Smith .  -da_refine_z <rz>- refinement ratio in z directio
1415e0f5d30fSBarry Smith -  -da_refine <n> - refine the DMDA n times before creating it, , if M, N, or P < 0
141647c6ae99SBarry Smith 
141747c6ae99SBarry Smith    Level: beginner
141847c6ae99SBarry Smith 
141947c6ae99SBarry Smith    Notes:
1420aa219208SBarry Smith    The stencil type DMDA_STENCIL_STAR with width 1 corresponds to the
1421aa219208SBarry Smith    standard 7-pt stencil, while DMDA_STENCIL_BOX with width 1 denotes
142247c6ae99SBarry Smith    the standard 27-pt stencil.
142347c6ae99SBarry Smith 
1424aa219208SBarry Smith    The array data itself is NOT stored in the DMDA, it is stored in Vec objects;
1425564755cdSBarry Smith    The appropriate vector objects can be obtained with calls to DMCreateGlobalVector()
1426564755cdSBarry Smith    and DMCreateLocalVector() and calls to VecDuplicate() if more are needed.
142747c6ae99SBarry Smith 
142847c6ae99SBarry Smith .keywords: distributed array, create, three-dimensional
142947c6ae99SBarry Smith 
1430aa219208SBarry Smith .seealso: DMDestroy(), DMView(), DMDACreate1d(), DMDACreate2d(), DMGlobalToLocalBegin(), DMDAGetRefinementFactor(),
143199f0b487SRichard Tran Mills           DMGlobalToLocalEnd(), DMLocalToGlobalBegin(), DMLocalToLocalBegin(), DMLocalToLocalEnd(), DMDASetRefinementFactor(),
1432d461ba97SBarry Smith           DMDAGetInfo(), DMCreateGlobalVector(), DMCreateLocalVector(), DMDACreateNaturalVector(), DMLoad(), DMDAGetOwnershipRanges()
143347c6ae99SBarry Smith 
143447c6ae99SBarry Smith @*/
1435bff4a2f0SMatthew G. Knepley PetscErrorCode  DMDACreate3d(MPI_Comm comm,DMBoundaryType bx,DMBoundaryType by,DMBoundaryType bz,DMDAStencilType stencil_type,PetscInt M,
14369a42bb27SBarry 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)
143747c6ae99SBarry Smith {
143847c6ae99SBarry Smith   PetscErrorCode ierr;
143947c6ae99SBarry Smith 
144047c6ae99SBarry Smith   PetscFunctionBegin;
1441aa219208SBarry Smith   ierr = DMDACreate(comm, da);CHKERRQ(ierr);
1442c73cfb54SMatthew G. Knepley   ierr = DMSetDimension(*da, 3);CHKERRQ(ierr);
1443aa219208SBarry Smith   ierr = DMDASetSizes(*da, M, N, P);CHKERRQ(ierr);
1444aa219208SBarry Smith   ierr = DMDASetNumProcs(*da, m, n, p);CHKERRQ(ierr);
14451321219cSEthan Coon   ierr = DMDASetBoundaryType(*da, bx, by, bz);CHKERRQ(ierr);
1446aa219208SBarry Smith   ierr = DMDASetDof(*da, dof);CHKERRQ(ierr);
1447aa219208SBarry Smith   ierr = DMDASetStencilType(*da, stencil_type);CHKERRQ(ierr);
1448aa219208SBarry Smith   ierr = DMDASetStencilWidth(*da, s);CHKERRQ(ierr);
1449aa219208SBarry Smith   ierr = DMDASetOwnershipRanges(*da, lx, ly, lz);CHKERRQ(ierr);
145047c6ae99SBarry Smith   /* This violates the behavior for other classes, but right now users expect negative dimensions to be handled this way */
14519a42bb27SBarry Smith   ierr = DMSetFromOptions(*da);CHKERRQ(ierr);
14529a42bb27SBarry Smith   ierr = DMSetUp(*da);CHKERRQ(ierr);
145347c6ae99SBarry Smith   PetscFunctionReturn(0);
145447c6ae99SBarry Smith }
1455