xref: /petsc/src/dm/impls/da/da3.c (revision 7a8be3513cf479fb6a672bd91de7eb6883fcfd02)
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>
10e0877f53SBarry Smith static PetscErrorCode DMView_DA_3d(DM da,PetscViewer viewer)
1147c6ae99SBarry Smith {
1247c6ae99SBarry Smith   PetscErrorCode ierr;
1347c6ae99SBarry Smith   PetscMPIInt    rank;
14c9493c35SStefano Zampini   PetscBool      iascii,isdraw,isglvis,isbinary;
1547c6ae99SBarry Smith   DM_DA          *dd = (DM_DA*)da->data;
169a42bb27SBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
179a42bb27SBarry Smith   PetscBool ismatlab;
189a42bb27SBarry Smith #endif
1947c6ae99SBarry Smith 
2047c6ae99SBarry Smith   PetscFunctionBegin;
21ffc4695bSBarry Smith   ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)da),&rank);CHKERRMPI(ierr);
2247c6ae99SBarry Smith 
23251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
24251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
25c9493c35SStefano Zampini   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERGLVIS,&isglvis);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 
331575c14dSBarry Smith     ierr = PetscViewerASCIIPushSynchronized(viewer);CHKERRQ(ierr);
3447c6ae99SBarry Smith     ierr = PetscViewerGetFormat(viewer, &format);CHKERRQ(ierr);
3576a8abe0SBarry Smith     if (format == PETSC_VIEWER_LOAD_BALANCE) {
3676a8abe0SBarry Smith       PetscInt      i,nmax = 0,nmin = PETSC_MAX_INT,navg = 0,*nz,nzlocal;
3776a8abe0SBarry Smith       DMDALocalInfo info;
3876a8abe0SBarry Smith       PetscMPIInt   size;
39ffc4695bSBarry Smith       ierr = MPI_Comm_size(PetscObjectComm((PetscObject)da),&size);CHKERRMPI(ierr);
4076a8abe0SBarry Smith       ierr = DMDAGetLocalInfo(da,&info);CHKERRQ(ierr);
4176a8abe0SBarry Smith       nzlocal = info.xm*info.ym*info.zm;
4276a8abe0SBarry Smith       ierr = PetscMalloc1(size,&nz);CHKERRQ(ierr);
43ffc4695bSBarry Smith       ierr = MPI_Allgather(&nzlocal,1,MPIU_INT,nz,1,MPIU_INT,PetscObjectComm((PetscObject)da));CHKERRMPI(ierr);
4476a8abe0SBarry Smith       for (i=0; i<(PetscInt)size; i++) {
4576a8abe0SBarry Smith         nmax = PetscMax(nmax,nz[i]);
4676a8abe0SBarry Smith         nmin = PetscMin(nmin,nz[i]);
4776a8abe0SBarry Smith         navg += nz[i];
4876a8abe0SBarry Smith       }
4976a8abe0SBarry Smith       ierr = PetscFree(nz);CHKERRQ(ierr);
5076a8abe0SBarry Smith       navg = navg/size;
5176a8abe0SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"  Load Balance - Grid Points: Min %D  avg %D  max %D\n",nmin,navg,nmax);CHKERRQ(ierr);
5276a8abe0SBarry Smith       PetscFunctionReturn(0);
5376a8abe0SBarry Smith     }
548ec8862eSJed Brown     if (format != PETSC_VIEWER_ASCII_VTK_DEPRECATED && format != PETSC_VIEWER_ASCII_VTK_CELL_DEPRECATED && format != PETSC_VIEWER_ASCII_GLVIS) {
55aa219208SBarry Smith       DMDALocalInfo info;
56aa219208SBarry Smith       ierr = DMDAGetLocalInfo(da,&info);CHKERRQ(ierr);
5747c6ae99SBarry 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);
5847c6ae99SBarry Smith       ierr = PetscViewerASCIISynchronizedPrintf(viewer,"X range of indices: %D %D, Y range of indices: %D %D, Z range of indices: %D %D\n",
5947c6ae99SBarry Smith                                                 info.xs,info.xs+info.xm,info.ys,info.ys+info.ym,info.zs,info.zs+info.zm);CHKERRQ(ierr);
6047c6ae99SBarry Smith #if !defined(PETSC_USE_COMPLEX)
616636e97aSMatthew G Knepley       if (da->coordinates) {
6247c6ae99SBarry Smith         PetscInt        last;
6347c6ae99SBarry Smith         const PetscReal *coors;
646636e97aSMatthew G Knepley         ierr = VecGetArrayRead(da->coordinates,&coors);CHKERRQ(ierr);
656636e97aSMatthew G Knepley         ierr = VecGetLocalSize(da->coordinates,&last);CHKERRQ(ierr);
6647c6ae99SBarry Smith         last = last - 3;
6757622a8eSBarry 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);
686636e97aSMatthew G Knepley         ierr = VecRestoreArrayRead(da->coordinates,&coors);CHKERRQ(ierr);
6947c6ae99SBarry Smith       }
7047c6ae99SBarry Smith #endif
7147c6ae99SBarry Smith       ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
721575c14dSBarry Smith       ierr = PetscViewerASCIIPopSynchronized(viewer);CHKERRQ(ierr);
738135c375SStefano Zampini     } else if (format == PETSC_VIEWER_ASCII_GLVIS) {
748135c375SStefano Zampini       ierr = DMView_DA_GLVis(da,viewer);CHKERRQ(ierr);
75616157d6SJed Brown     } else {
76616157d6SJed Brown       ierr = DMView_DA_VTK(da,viewer);CHKERRQ(ierr);
7747c6ae99SBarry Smith     }
7847c6ae99SBarry Smith   } else if (isdraw) {
7947c6ae99SBarry Smith     PetscDraw      draw;
8047c6ae99SBarry Smith     PetscReal      ymin = -1.0,ymax = (PetscReal)dd->N;
8147c6ae99SBarry Smith     PetscReal      xmin = -1.0,xmax = (PetscReal)((dd->M+2)*dd->P),x,y,ycoord,xcoord;
828ea3bf28SBarry Smith     PetscInt       k,plane,base;
838ea3bf28SBarry Smith     const PetscInt *idx;
8447c6ae99SBarry Smith     char           node[10];
8547c6ae99SBarry Smith     PetscBool      isnull;
8647c6ae99SBarry Smith 
8747c6ae99SBarry Smith     ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
885b399a63SLisandro Dalcin     ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
895b399a63SLisandro Dalcin     if (isnull) PetscFunctionReturn(0);
9047c6ae99SBarry Smith 
915b399a63SLisandro Dalcin     ierr = PetscDrawCheckResizedWindow(draw);CHKERRQ(ierr);
925b399a63SLisandro Dalcin     ierr = PetscDrawClear(draw);CHKERRQ(ierr);
935b399a63SLisandro Dalcin     ierr = PetscDrawSetCoordinates(draw,xmin,ymin,xmax,ymax);CHKERRQ(ierr);
945b399a63SLisandro Dalcin 
955b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
9647c6ae99SBarry Smith     /* first processor draw all node lines */
97dd400576SPatrick Sanan     if (rank == 0) {
9847c6ae99SBarry Smith       for (k=0; k<dd->P; k++) {
9947c6ae99SBarry Smith         ymin = 0.0; ymax = (PetscReal)(dd->N - 1);
10047c6ae99SBarry Smith         for (xmin=(PetscReal)(k*(dd->M+1)); xmin<(PetscReal)(dd->M+(k*(dd->M+1))); xmin++) {
10147c6ae99SBarry Smith           ierr = PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_BLACK);CHKERRQ(ierr);
10247c6ae99SBarry Smith         }
10347c6ae99SBarry Smith         xmin = (PetscReal)(k*(dd->M+1)); xmax = xmin + (PetscReal)(dd->M - 1);
10447c6ae99SBarry Smith         for (ymin=0; ymin<(PetscReal)dd->N; ymin++) {
10547c6ae99SBarry Smith           ierr = PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_BLACK);CHKERRQ(ierr);
10647c6ae99SBarry Smith         }
10747c6ae99SBarry Smith       }
10847c6ae99SBarry Smith     }
1095b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
1105b399a63SLisandro Dalcin     ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
11147c6ae99SBarry Smith     ierr = PetscDrawPause(draw);CHKERRQ(ierr);
11247c6ae99SBarry Smith 
1135b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
1145b399a63SLisandro Dalcin     /*Go through and draw for each plane*/
1155b399a63SLisandro Dalcin     for (k=0; k<dd->P; k++) {
11647c6ae99SBarry Smith       if ((k >= dd->zs) && (k < dd->ze)) {
11747c6ae99SBarry Smith         /* draw my box */
11847c6ae99SBarry Smith         ymin = dd->ys;
11947c6ae99SBarry Smith         ymax = dd->ye - 1;
12047c6ae99SBarry Smith         xmin = dd->xs/dd->w    + (dd->M+1)*k;
12147c6ae99SBarry Smith         xmax =(dd->xe-1)/dd->w + (dd->M+1)*k;
12247c6ae99SBarry Smith 
12347c6ae99SBarry Smith         ierr = PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_RED);CHKERRQ(ierr);
12447c6ae99SBarry Smith         ierr = PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_RED);CHKERRQ(ierr);
12547c6ae99SBarry Smith         ierr = PetscDrawLine(draw,xmin,ymax,xmax,ymax,PETSC_DRAW_RED);CHKERRQ(ierr);
12647c6ae99SBarry Smith         ierr = PetscDrawLine(draw,xmax,ymin,xmax,ymax,PETSC_DRAW_RED);CHKERRQ(ierr);
12747c6ae99SBarry Smith 
12847c6ae99SBarry Smith         xmin = dd->xs/dd->w;
12947c6ae99SBarry Smith         xmax =(dd->xe-1)/dd->w;
13047c6ae99SBarry Smith 
131832b7cebSLisandro Dalcin         /* identify which processor owns the box */
132832b7cebSLisandro Dalcin         ierr = PetscSNPrintf(node,sizeof(node),"%d",(int)rank);CHKERRQ(ierr);
133832b7cebSLisandro Dalcin         ierr = PetscDrawString(draw,xmin+(dd->M+1)*k+.2,ymin+.3,PETSC_DRAW_RED,node);CHKERRQ(ierr);
13447c6ae99SBarry Smith         /* put in numbers*/
13547c6ae99SBarry Smith         base = (dd->base+(dd->xe-dd->xs)*(dd->ye-dd->ys)*(k-dd->zs))/dd->w;
13647c6ae99SBarry Smith         for (y=ymin; y<=ymax; y++) {
13747c6ae99SBarry Smith           for (x=xmin+(dd->M+1)*k; x<=xmax+(dd->M+1)*k; x++) {
138832b7cebSLisandro Dalcin             ierr = PetscSNPrintf(node,sizeof(node),"%d",(int)base++);CHKERRQ(ierr);
13947c6ae99SBarry Smith             ierr = PetscDrawString(draw,x,y,PETSC_DRAW_BLACK,node);CHKERRQ(ierr);
14047c6ae99SBarry Smith           }
14147c6ae99SBarry Smith         }
14247c6ae99SBarry Smith 
14347c6ae99SBarry Smith       }
14447c6ae99SBarry Smith     }
1455b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
1465b399a63SLisandro Dalcin     ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
14747c6ae99SBarry Smith     ierr = PetscDrawPause(draw);CHKERRQ(ierr);
14847c6ae99SBarry Smith 
1495b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
15047c6ae99SBarry Smith     for (k=0-dd->s; k<dd->P+dd->s; k++) {
15147c6ae99SBarry Smith       /* Go through and draw for each plane */
15247c6ae99SBarry Smith       if ((k >= dd->Zs) && (k < dd->Ze)) {
15347c6ae99SBarry Smith         /* overlay ghost numbers, useful for error checking */
154565245c5SBarry Smith         base = (dd->Xe-dd->Xs)*(dd->Ye-dd->Ys)*(k-dd->Zs)/dd->w;
15545b6f7e9SBarry Smith         ierr = ISLocalToGlobalMappingGetBlockIndices(da->ltogmap,&idx);CHKERRQ(ierr);
15647c6ae99SBarry Smith         plane=k;
157565245c5SBarry Smith         /* Keep z wrap around points on the drawing */
1588865f1eaSKarl Rupp         if (k<0) plane=dd->P+k;
1598865f1eaSKarl Rupp         if (k>=dd->P) plane=k-dd->P;
16047c6ae99SBarry Smith         ymin = dd->Ys; ymax = dd->Ye;
16147c6ae99SBarry Smith         xmin = (dd->M+1)*plane*dd->w;
16247c6ae99SBarry Smith         xmax = (dd->M+1)*plane*dd->w+dd->M*dd->w;
16347c6ae99SBarry Smith         for (y=ymin; y<ymax; y++) {
16447c6ae99SBarry Smith           for (x=xmin+dd->Xs; x<xmin+dd->Xe; x+=dd->w) {
1658ea3bf28SBarry Smith             sprintf(node,"%d",(int)(idx[base]));
16647c6ae99SBarry Smith             ycoord = y;
16747c6ae99SBarry Smith             /*Keep y wrap around points on drawing */
1688865f1eaSKarl Rupp             if (y<0) ycoord = dd->N+y;
1698865f1eaSKarl Rupp             if (y>=dd->N) ycoord = y-dd->N;
17047c6ae99SBarry Smith             xcoord = x;   /* Keep x wrap points on drawing */
1718865f1eaSKarl Rupp             if (x<xmin) xcoord = xmax - (xmin-x);
1728865f1eaSKarl Rupp             if (x>=xmax) xcoord = xmin + (x-xmax);
17347c6ae99SBarry Smith             ierr = PetscDrawString(draw,xcoord/dd->w,ycoord,PETSC_DRAW_BLUE,node);CHKERRQ(ierr);
174565245c5SBarry Smith             base++;
17547c6ae99SBarry Smith           }
17647c6ae99SBarry Smith         }
17745b6f7e9SBarry Smith         ierr = ISLocalToGlobalMappingRestoreBlockIndices(da->ltogmap,&idx);CHKERRQ(ierr);
17847c6ae99SBarry Smith       }
17947c6ae99SBarry Smith     }
1805b399a63SLisandro Dalcin     ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
1815b399a63SLisandro Dalcin     ierr = PetscDrawFlush(draw);CHKERRQ(ierr);
18247c6ae99SBarry Smith     ierr = PetscDrawPause(draw);CHKERRQ(ierr);
183832b7cebSLisandro Dalcin     ierr = PetscDrawSave(draw);CHKERRQ(ierr);
184c9493c35SStefano Zampini   } else if (isglvis) {
185c9493c35SStefano Zampini     ierr = DMView_DA_GLVis(da,viewer);CHKERRQ(ierr);
1869a42bb27SBarry Smith   } else if (isbinary) {
1879a42bb27SBarry Smith     ierr = DMView_DA_Binary(da,viewer);CHKERRQ(ierr);
1889a42bb27SBarry Smith #if defined(PETSC_HAVE_MATLAB_ENGINE)
1899a42bb27SBarry Smith   } else if (ismatlab) {
1909a42bb27SBarry Smith     ierr = DMView_DA_Matlab(da,viewer);CHKERRQ(ierr);
1919a42bb27SBarry Smith #endif
19211aeaf0aSBarry Smith   }
19347c6ae99SBarry Smith   PetscFunctionReturn(0);
19447c6ae99SBarry Smith }
19547c6ae99SBarry Smith 
1967087cfbeSBarry Smith PetscErrorCode  DMSetUp_DA_3D(DM da)
19747c6ae99SBarry Smith {
19847c6ae99SBarry Smith   DM_DA            *dd          = (DM_DA*)da->data;
19947c6ae99SBarry Smith   const PetscInt   M            = dd->M;
20047c6ae99SBarry Smith   const PetscInt   N            = dd->N;
20147c6ae99SBarry Smith   const PetscInt   P            = dd->P;
20247c6ae99SBarry Smith   PetscInt         m            = dd->m;
20347c6ae99SBarry Smith   PetscInt         n            = dd->n;
20447c6ae99SBarry Smith   PetscInt         p            = dd->p;
20547c6ae99SBarry Smith   const PetscInt   dof          = dd->w;
20647c6ae99SBarry Smith   const PetscInt   s            = dd->s;
207bff4a2f0SMatthew G. Knepley   DMBoundaryType   bx           = dd->bx;
208bff4a2f0SMatthew G. Knepley   DMBoundaryType   by           = dd->by;
209bff4a2f0SMatthew G. Knepley   DMBoundaryType   bz           = dd->bz;
21019fd82e9SBarry Smith   DMDAStencilType  stencil_type = dd->stencil_type;
21147c6ae99SBarry Smith   PetscInt         *lx          = dd->lx;
21247c6ae99SBarry Smith   PetscInt         *ly          = dd->ly;
21347c6ae99SBarry Smith   PetscInt         *lz          = dd->lz;
21447c6ae99SBarry Smith   MPI_Comm         comm;
21547c6ae99SBarry Smith   PetscMPIInt      rank,size;
216ce00eea3SSatish Balay   PetscInt         xs = 0,xe,ys = 0,ye,zs = 0,ze,x = 0,y = 0,z = 0;
217bd1fc5aeSBarry Smith   PetscInt         Xs,Xe,Ys,Ye,Zs,Ze,IXs,IXe,IYs,IYe,IZs,IZe,pm;
2188ea3bf28SBarry Smith   PetscInt         left,right,up,down,bottom,top,i,j,k,*idx,nn;
21947c6ae99SBarry Smith   PetscInt         n0,n1,n2,n3,n4,n5,n6,n7,n8,n9,n10,n11,n12,n14;
22047c6ae99SBarry Smith   PetscInt         n15,n16,n17,n18,n19,n20,n21,n22,n23,n24,n25,n26;
221db87c5ecSEthan Coon   PetscInt         *bases,*ldims,base,x_t,y_t,z_t,s_t,count,s_x,s_y,s_z;
22247c6ae99SBarry Smith   PetscInt         sn0  = 0,sn1 = 0,sn2 = 0,sn3 = 0,sn5 = 0,sn6 = 0,sn7 = 0;
22347c6ae99SBarry Smith   PetscInt         sn8  = 0,sn9 = 0,sn11 = 0,sn15 = 0,sn24 = 0,sn25 = 0,sn26 = 0;
22447c6ae99SBarry Smith   PetscInt         sn17 = 0,sn18 = 0,sn19 = 0,sn20 = 0,sn21 = 0,sn23 = 0;
22547c6ae99SBarry Smith   Vec              local,global;
226bd1fc5aeSBarry Smith   VecScatter       gtol;
22745b6f7e9SBarry Smith   IS               to,from;
2286f951b95Secoon   PetscBool        twod;
22947c6ae99SBarry Smith   PetscErrorCode   ierr;
23047c6ae99SBarry Smith 
23147c6ae99SBarry Smith   PetscFunctionBegin;
232*7a8be351SBarry Smith   PetscCheck(stencil_type != DMDA_STENCIL_BOX || (bx != DM_BOUNDARY_MIRROR && by != DM_BOUNDARY_MIRROR && bz != DM_BOUNDARY_MIRROR),PetscObjectComm((PetscObject)da),PETSC_ERR_SUP,"Mirror boundary and box stencil");
23347c6ae99SBarry Smith   ierr = PetscObjectGetComm((PetscObject) da, &comm);CHKERRQ(ierr);
2343855c12bSBarry Smith #if !defined(PETSC_USE_64BIT_INDICES)
235*7a8be351SBarry Smith   PetscCheck(((PetscInt64) M)*((PetscInt64) N)*((PetscInt64) P)*((PetscInt64) dof) <= (PetscInt64) PETSC_MPI_INT_MAX,comm,PETSC_ERR_INT_OVERFLOW,"Mesh of %D by %D by %D by %D (dof) is too large for 32 bit indices",M,N,P,dof);
2363855c12bSBarry Smith #endif
2373855c12bSBarry Smith 
238ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRMPI(ierr);
239ffc4695bSBarry Smith   ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr);
24047c6ae99SBarry Smith 
24147c6ae99SBarry Smith   if (m != PETSC_DECIDE) {
2422c71b3e2SJacob Faibussowitsch     PetscCheckFalse(m < 1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in X direction: %D",m);
2432c71b3e2SJacob Faibussowitsch     else PetscCheckFalse(m > size,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in X direction: %D %d",m,size);
24447c6ae99SBarry Smith   }
24547c6ae99SBarry Smith   if (n != PETSC_DECIDE) {
2462c71b3e2SJacob Faibussowitsch     PetscCheckFalse(n < 1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in Y direction: %D",n);
2472c71b3e2SJacob Faibussowitsch     else PetscCheckFalse(n > size,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in Y direction: %D %d",n,size);
24847c6ae99SBarry Smith   }
24947c6ae99SBarry Smith   if (p != PETSC_DECIDE) {
2502c71b3e2SJacob Faibussowitsch     PetscCheckFalse(p < 1,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in Z direction: %D",p);
2512c71b3e2SJacob Faibussowitsch     else PetscCheckFalse(p > size,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in Z direction: %D %d",p,size);
25247c6ae99SBarry Smith   }
2532c71b3e2SJacob Faibussowitsch   PetscCheckFalse((m > 0) && (n > 0) && (p > 0) && (m*n*p != size),PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"m %D * n %D * p %D != size %d",m,n,p,size);
25447c6ae99SBarry Smith 
25547c6ae99SBarry Smith   /* Partition the array among the processors */
25647c6ae99SBarry Smith   if (m == PETSC_DECIDE && n != PETSC_DECIDE && p != PETSC_DECIDE) {
25747c6ae99SBarry Smith     m = size/(n*p);
25847c6ae99SBarry Smith   } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) {
25947c6ae99SBarry Smith     n = size/(m*p);
26047c6ae99SBarry Smith   } else if (m != PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) {
26147c6ae99SBarry Smith     p = size/(m*n);
26247c6ae99SBarry Smith   } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) {
26347c6ae99SBarry Smith     /* try for squarish distribution */
264369cc0aeSBarry Smith     m = (int)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)N*p)));
26547c6ae99SBarry Smith     if (!m) m = 1;
26647c6ae99SBarry Smith     while (m > 0) {
26747c6ae99SBarry Smith       n = size/(m*p);
26847c6ae99SBarry Smith       if (m*n*p == size) break;
26947c6ae99SBarry Smith       m--;
27047c6ae99SBarry Smith     }
271*7a8be351SBarry Smith     PetscCheck(m,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad p value: p = %D",p);
27247c6ae99SBarry Smith     if (M > N && m < n) {PetscInt _m = m; m = n; n = _m;}
27347c6ae99SBarry Smith   } else if (m == PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) {
27447c6ae99SBarry Smith     /* try for squarish distribution */
275369cc0aeSBarry Smith     m = (int)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)P*n)));
27647c6ae99SBarry Smith     if (!m) m = 1;
27747c6ae99SBarry Smith     while (m > 0) {
27847c6ae99SBarry Smith       p = size/(m*n);
27947c6ae99SBarry Smith       if (m*n*p == size) break;
28047c6ae99SBarry Smith       m--;
28147c6ae99SBarry Smith     }
282*7a8be351SBarry Smith     PetscCheck(m,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad n value: n = %D",n);
28347c6ae99SBarry Smith     if (M > P && m < p) {PetscInt _m = m; m = p; p = _m;}
28447c6ae99SBarry Smith   } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) {
28547c6ae99SBarry Smith     /* try for squarish distribution */
286369cc0aeSBarry Smith     n = (int)(0.5 + PetscSqrtReal(((PetscReal)N)*((PetscReal)size)/((PetscReal)P*m)));
28747c6ae99SBarry Smith     if (!n) n = 1;
28847c6ae99SBarry Smith     while (n > 0) {
28947c6ae99SBarry Smith       p = size/(m*n);
29047c6ae99SBarry Smith       if (m*n*p == size) break;
29147c6ae99SBarry Smith       n--;
29247c6ae99SBarry Smith     }
293*7a8be351SBarry Smith     PetscCheck(n,PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad m value: m = %D",n);
29447c6ae99SBarry Smith     if (N > P && n < p) {PetscInt _n = n; n = p; p = _n;}
29547c6ae99SBarry Smith   } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) {
29647c6ae99SBarry Smith     /* try for squarish distribution */
297369cc0aeSBarry Smith     n = (PetscInt)(0.5 + PetscPowReal(((PetscReal)N*N)*((PetscReal)size)/((PetscReal)P*M),(PetscReal)(1./3.)));
29847c6ae99SBarry Smith     if (!n) n = 1;
29947c6ae99SBarry Smith     while (n > 0) {
30047c6ae99SBarry Smith       pm = size/n;
30147c6ae99SBarry Smith       if (n*pm == size) break;
30247c6ae99SBarry Smith       n--;
30347c6ae99SBarry Smith     }
30447c6ae99SBarry Smith     if (!n) n = 1;
305369cc0aeSBarry Smith     m = (PetscInt)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)P*n)));
30647c6ae99SBarry Smith     if (!m) m = 1;
30747c6ae99SBarry Smith     while (m > 0) {
30847c6ae99SBarry Smith       p = size/(m*n);
30947c6ae99SBarry Smith       if (m*n*p == size) break;
31047c6ae99SBarry Smith       m--;
31147c6ae99SBarry Smith     }
31247c6ae99SBarry Smith     if (M > P && m < p) {PetscInt _m = m; m = p; p = _m;}
3132c71b3e2SJacob Faibussowitsch   } else PetscCheckFalse(m*n*p != size,PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Given Bad partition");
31447c6ae99SBarry Smith 
3152c71b3e2SJacob Faibussowitsch   PetscCheckFalse(m*n*p != size,PetscObjectComm((PetscObject)da),PETSC_ERR_PLIB,"Could not find good partition");
3162c71b3e2SJacob Faibussowitsch   PetscCheckFalse(M < m,PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in x direction is too fine! %D %D",M,m);
3172c71b3e2SJacob Faibussowitsch   PetscCheckFalse(N < n,PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in y direction is too fine! %D %D",N,n);
3182c71b3e2SJacob Faibussowitsch   PetscCheckFalse(P < p,PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in z direction is too fine! %D %D",P,p);
31947c6ae99SBarry Smith 
32047c6ae99SBarry Smith   /*
32147c6ae99SBarry Smith      Determine locally owned region
32247c6ae99SBarry Smith      [x, y, or z]s is the first local node number, [x, y, z] is the number of local nodes
32347c6ae99SBarry Smith   */
32447c6ae99SBarry Smith 
32547c6ae99SBarry Smith   if (!lx) {
326785e854fSJed Brown     ierr = PetscMalloc1(m, &dd->lx);CHKERRQ(ierr);
32747c6ae99SBarry Smith     lx   = dd->lx;
3288865f1eaSKarl Rupp     for (i=0; i<m; i++) lx[i] = M/m + ((M % m) > (i % m));
32947c6ae99SBarry Smith   }
33047c6ae99SBarry Smith   x  = lx[rank % m];
33147c6ae99SBarry Smith   xs = 0;
3328865f1eaSKarl Rupp   for (i=0; i<(rank%m); i++) xs += lx[i];
3332c71b3e2SJacob Faibussowitsch   PetscCheckFalse((x < s) && ((m > 1) || (bx == DM_BOUNDARY_PERIODIC)),PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local x-width of domain x %D is smaller than stencil width s %D",x,s);
33447c6ae99SBarry Smith 
33547c6ae99SBarry Smith   if (!ly) {
336785e854fSJed Brown     ierr = PetscMalloc1(n, &dd->ly);CHKERRQ(ierr);
33747c6ae99SBarry Smith     ly   = dd->ly;
3388865f1eaSKarl Rupp     for (i=0; i<n; i++) ly[i] = N/n + ((N % n) > (i % n));
33947c6ae99SBarry Smith   }
34047c6ae99SBarry Smith   y = ly[(rank % (m*n))/m];
3412c71b3e2SJacob Faibussowitsch   PetscCheckFalse((y < s) && ((n > 1) || (by == DM_BOUNDARY_PERIODIC)),PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local y-width of domain y %D is smaller than stencil width s %D",y,s);
34230729d88SBarry Smith 
34347c6ae99SBarry Smith   ys = 0;
3448865f1eaSKarl Rupp   for (i=0; i<(rank % (m*n))/m; i++) ys += ly[i];
34547c6ae99SBarry Smith 
34647c6ae99SBarry Smith   if (!lz) {
347785e854fSJed Brown     ierr = PetscMalloc1(p, &dd->lz);CHKERRQ(ierr);
34847c6ae99SBarry Smith     lz = dd->lz;
3498865f1eaSKarl Rupp     for (i=0; i<p; i++) lz[i] = P/p + ((P % p) > (i % p));
35047c6ae99SBarry Smith   }
35147c6ae99SBarry Smith   z = lz[rank/(m*n)];
352bcea557cSEthan Coon 
353fdc81ce8SEthan Coon   /* note this is different than x- and y-, as we will handle as an important special
354bff4a2f0SMatthew G. Knepley    case when p=P=1 and DM_BOUNDARY_PERIODIC and s > z.  This is to deal with 2D problems
355fdc81ce8SEthan Coon    in a 3D code.  Additional code for this case is noted with "2d case" comments */
3566f951b95Secoon   twod = PETSC_FALSE;
3578865f1eaSKarl Rupp   if (P == 1) twod = PETSC_TRUE;
3582c71b3e2SJacob Faibussowitsch   else PetscCheckFalse((z < s) && ((p > 1) || (bz == DM_BOUNDARY_PERIODIC)),PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Local z-width of domain z %D is smaller than stencil width s %D",z,s);
35947c6ae99SBarry Smith   zs = 0;
3608865f1eaSKarl Rupp   for (i=0; i<(rank/(m*n)); i++) zs += lz[i];
36147c6ae99SBarry Smith   ye = ys + y;
36247c6ae99SBarry Smith   xe = xs + x;
36347c6ae99SBarry Smith   ze = zs + z;
36447c6ae99SBarry Smith 
36588661749SPeter Brune   /* determine ghost region (Xs) and region scattered into (IXs)  */
366d9c9ebe5SPeter Brune   if (xs-s > 0) {
367d9c9ebe5SPeter Brune     Xs = xs - s; IXs = xs - s;
36888661749SPeter Brune   } else {
3698865f1eaSKarl Rupp     if (bx) Xs = xs - s;
3708865f1eaSKarl Rupp     else Xs = 0;
37188661749SPeter Brune     IXs = 0;
37288661749SPeter Brune   }
373d9c9ebe5SPeter Brune   if (xe+s <= M) {
374d9c9ebe5SPeter Brune     Xe = xe + s; IXe = xe + s;
37588661749SPeter Brune   } else {
37688661749SPeter Brune     if (bx) {
377d9c9ebe5SPeter Brune       Xs = xs - s; Xe = xe + s;
3788865f1eaSKarl Rupp     } else Xe = M;
37988661749SPeter Brune     IXe = M;
38088661749SPeter Brune   }
38147c6ae99SBarry Smith 
382bff4a2f0SMatthew G. Knepley   if (bx == DM_BOUNDARY_PERIODIC || bx == DM_BOUNDARY_MIRROR) {
383d9c9ebe5SPeter Brune     IXs = xs - s;
384d9c9ebe5SPeter Brune     IXe = xe + s;
385d9c9ebe5SPeter Brune     Xs  = xs - s;
386d9c9ebe5SPeter Brune     Xe  = xe + s;
38788661749SPeter Brune   }
38888661749SPeter Brune 
389d9c9ebe5SPeter Brune   if (ys-s > 0) {
390d9c9ebe5SPeter Brune     Ys = ys - s; IYs = ys - s;
39188661749SPeter Brune   } else {
3928865f1eaSKarl Rupp     if (by) Ys = ys - s;
3938865f1eaSKarl Rupp     else Ys = 0;
39488661749SPeter Brune     IYs = 0;
39588661749SPeter Brune   }
396d9c9ebe5SPeter Brune   if (ye+s <= N) {
397d9c9ebe5SPeter Brune     Ye = ye + s; IYe = ye + s;
39888661749SPeter Brune   } else {
3998865f1eaSKarl Rupp     if (by) Ye = ye + s;
4008865f1eaSKarl Rupp     else Ye = N;
40188661749SPeter Brune     IYe = N;
40288661749SPeter Brune   }
40388661749SPeter Brune 
404bff4a2f0SMatthew G. Knepley   if (by == DM_BOUNDARY_PERIODIC || by == DM_BOUNDARY_MIRROR) {
405d9c9ebe5SPeter Brune     IYs = ys - s;
406d9c9ebe5SPeter Brune     IYe = ye + s;
407d9c9ebe5SPeter Brune     Ys  = ys - s;
408d9c9ebe5SPeter Brune     Ye  = ye + s;
40988661749SPeter Brune   }
41088661749SPeter Brune 
411d9c9ebe5SPeter Brune   if (zs-s > 0) {
412d9c9ebe5SPeter Brune     Zs = zs - s; IZs = zs - s;
41388661749SPeter Brune   } else {
4148865f1eaSKarl Rupp     if (bz) Zs = zs - s;
4158865f1eaSKarl Rupp     else Zs = 0;
41688661749SPeter Brune     IZs = 0;
41788661749SPeter Brune   }
418d9c9ebe5SPeter Brune   if (ze+s <= P) {
419d9c9ebe5SPeter Brune     Ze = ze + s; IZe = ze + s;
42088661749SPeter Brune   } else {
4218865f1eaSKarl Rupp     if (bz) Ze = ze + s;
4228865f1eaSKarl Rupp     else Ze = P;
42388661749SPeter Brune     IZe = P;
42488661749SPeter Brune   }
42588661749SPeter Brune 
426bff4a2f0SMatthew G. Knepley   if (bz == DM_BOUNDARY_PERIODIC || bz == DM_BOUNDARY_MIRROR) {
427d9c9ebe5SPeter Brune     IZs = zs - s;
428d9c9ebe5SPeter Brune     IZe = ze + s;
429d9c9ebe5SPeter Brune     Zs  = zs - s;
430d9c9ebe5SPeter Brune     Ze  = ze + s;
43188661749SPeter Brune   }
43247c6ae99SBarry Smith 
43347c6ae99SBarry Smith   /* Resize all X parameters to reflect w */
434d9c9ebe5SPeter Brune   s_x = s;
435d9c9ebe5SPeter Brune   s_y = s;
436d9c9ebe5SPeter Brune   s_z = s;
43747c6ae99SBarry Smith 
43847c6ae99SBarry Smith   /* determine starting point of each processor */
43947c6ae99SBarry Smith   nn       = x*y*z;
440dcca6d9dSJed Brown   ierr     = PetscMalloc2(size+1,&bases,size,&ldims);CHKERRQ(ierr);
44155b25c41SPierre Jolivet   ierr     = MPI_Allgather(&nn,1,MPIU_INT,ldims,1,MPIU_INT,comm);CHKERRMPI(ierr);
44247c6ae99SBarry Smith   bases[0] = 0;
4438865f1eaSKarl Rupp   for (i=1; i<=size; i++) bases[i] = ldims[i-1];
4448865f1eaSKarl Rupp   for (i=1; i<=size; i++) bases[i] += bases[i-1];
445ce00eea3SSatish Balay   base = bases[rank]*dof;
44647c6ae99SBarry Smith 
44747c6ae99SBarry Smith   /* allocate the base parallel and sequential vectors */
448ce00eea3SSatish Balay   dd->Nlocal = x*y*z*dof;
449b1fb7eb7SBarry Smith   ierr       = VecCreateMPIWithArray(comm,dof,dd->Nlocal,PETSC_DECIDE,NULL,&global);CHKERRQ(ierr);
450ce00eea3SSatish Balay   dd->nlocal = (Xe-Xs)*(Ye-Ys)*(Ze-Zs)*dof;
451b1fb7eb7SBarry Smith   ierr       = VecCreateSeqWithArray(PETSC_COMM_SELF,dof,dd->nlocal,NULL,&local);CHKERRQ(ierr);
45247c6ae99SBarry Smith 
4534104a7a0SPatrick Sanan   /* generate global to local vector scatter and local to global mapping*/
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 */
4574104a7a0SPatrick Sanan   ierr   = PetscMalloc1((IXe-IXs)*(IYe-IYs)*(IZe-IZs),&idx);CHKERRQ(ierr);
458d9c9ebe5SPeter Brune   if (stencil_type == DMDA_STENCIL_BOX) {
459ce00eea3SSatish Balay     left   = IXs - Xs; right = left + (IXe-IXs);
460ce00eea3SSatish Balay     bottom = IYs - Ys; top = bottom + (IYe-IYs);
461ce00eea3SSatish Balay     down   = IZs - Zs; up  = down + (IZe-IZs);
462ce00eea3SSatish Balay     count  = 0;
463ce00eea3SSatish Balay     for (i=down; i<up; i++) {
464ce00eea3SSatish Balay       for (j=bottom; j<top; j++) {
465ce00eea3SSatish Balay         for (k=left; k<right; k++) {
466ce00eea3SSatish Balay           idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
467ce00eea3SSatish Balay         }
468ce00eea3SSatish Balay       }
469ce00eea3SSatish Balay     }
470ce00eea3SSatish Balay     ierr = ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&to);CHKERRQ(ierr);
47147c6ae99SBarry Smith   } else {
47247c6ae99SBarry Smith     /* This is way ugly! We need to list the funny cross type region */
473ce00eea3SSatish Balay     left   = xs - Xs; right = left + x;
47447c6ae99SBarry Smith     bottom = ys - Ys; top = bottom + y;
47547c6ae99SBarry Smith     down   = zs - Zs;   up  = down + z;
47647c6ae99SBarry Smith     count  = 0;
477a5b23f4aSJose E. Roman     /* the bottom chunk */
478ce00eea3SSatish Balay     for (i=(IZs-Zs); i<down; i++) {
47947c6ae99SBarry Smith       for (j=bottom; j<top; j++) {
480ce00eea3SSatish Balay         for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
48147c6ae99SBarry Smith       }
48247c6ae99SBarry Smith     }
48347c6ae99SBarry Smith     /* the middle piece */
48447c6ae99SBarry Smith     for (i=down; i<up; i++) {
48547c6ae99SBarry Smith       /* front */
486ce00eea3SSatish Balay       for (j=(IYs-Ys); j<bottom; j++) {
487ce00eea3SSatish Balay         for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
48847c6ae99SBarry Smith       }
48947c6ae99SBarry Smith       /* middle */
49047c6ae99SBarry Smith       for (j=bottom; j<top; j++) {
491ce00eea3SSatish Balay         for (k=IXs-Xs; k<IXe-Xs; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
49247c6ae99SBarry Smith       }
49347c6ae99SBarry Smith       /* back */
494ce00eea3SSatish Balay       for (j=top; j<top+IYe-ye; j++) {
495ce00eea3SSatish Balay         for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
49647c6ae99SBarry Smith       }
49747c6ae99SBarry Smith     }
49847c6ae99SBarry Smith     /* the top piece */
499ce00eea3SSatish Balay     for (i=up; i<up+IZe-ze; i++) {
50047c6ae99SBarry Smith       for (j=bottom; j<top; j++) {
501ce00eea3SSatish Balay         for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
50247c6ae99SBarry Smith       }
50347c6ae99SBarry Smith     }
50447c6ae99SBarry Smith     ierr = ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&to);CHKERRQ(ierr);
50547c6ae99SBarry Smith   }
50647c6ae99SBarry Smith 
50747c6ae99SBarry Smith   /* determine who lies on each side of use stored in    n24 n25 n26
50847c6ae99SBarry Smith                                                          n21 n22 n23
50947c6ae99SBarry Smith                                                          n18 n19 n20
51047c6ae99SBarry Smith 
51147c6ae99SBarry Smith                                                          n15 n16 n17
51247c6ae99SBarry Smith                                                          n12     n14
51347c6ae99SBarry Smith                                                          n9  n10 n11
51447c6ae99SBarry Smith 
51547c6ae99SBarry Smith                                                          n6  n7  n8
51647c6ae99SBarry Smith                                                          n3  n4  n5
51747c6ae99SBarry Smith                                                          n0  n1  n2
51847c6ae99SBarry Smith   */
51947c6ae99SBarry Smith 
52047c6ae99SBarry Smith   /* Solve for X,Y, and Z Periodic Case First, Then Modify Solution */
52147c6ae99SBarry Smith   /* Assume Nodes are Internal to the Cube */
52247c6ae99SBarry Smith   n0 = rank - m*n - m - 1;
52347c6ae99SBarry Smith   n1 = rank - m*n - m;
52447c6ae99SBarry Smith   n2 = rank - m*n - m + 1;
52547c6ae99SBarry Smith   n3 = rank - m*n -1;
52647c6ae99SBarry Smith   n4 = rank - m*n;
52747c6ae99SBarry Smith   n5 = rank - m*n + 1;
52847c6ae99SBarry Smith   n6 = rank - m*n + m - 1;
52947c6ae99SBarry Smith   n7 = rank - m*n + m;
53047c6ae99SBarry Smith   n8 = rank - m*n + m + 1;
53147c6ae99SBarry Smith 
53247c6ae99SBarry Smith   n9  = rank - m - 1;
53347c6ae99SBarry Smith   n10 = rank - m;
53447c6ae99SBarry Smith   n11 = rank - m + 1;
53547c6ae99SBarry Smith   n12 = rank - 1;
53647c6ae99SBarry Smith   n14 = rank + 1;
53747c6ae99SBarry Smith   n15 = rank + m - 1;
53847c6ae99SBarry Smith   n16 = rank + m;
53947c6ae99SBarry Smith   n17 = rank + m + 1;
54047c6ae99SBarry Smith 
54147c6ae99SBarry Smith   n18 = rank + m*n - m - 1;
54247c6ae99SBarry Smith   n19 = rank + m*n - m;
54347c6ae99SBarry Smith   n20 = rank + m*n - m + 1;
54447c6ae99SBarry Smith   n21 = rank + m*n - 1;
54547c6ae99SBarry Smith   n22 = rank + m*n;
54647c6ae99SBarry Smith   n23 = rank + m*n + 1;
54747c6ae99SBarry Smith   n24 = rank + m*n + m - 1;
54847c6ae99SBarry Smith   n25 = rank + m*n + m;
54947c6ae99SBarry Smith   n26 = rank + m*n + m + 1;
55047c6ae99SBarry Smith 
55147c6ae99SBarry Smith   /* Assume Pieces are on Faces of Cube */
55247c6ae99SBarry Smith 
55347c6ae99SBarry Smith   if (xs == 0) { /* First assume not corner or edge */
55447c6ae99SBarry Smith     n0  = rank       -1 - (m*n);
55547c6ae99SBarry Smith     n3  = rank + m   -1 - (m*n);
55647c6ae99SBarry Smith     n6  = rank + 2*m -1 - (m*n);
55747c6ae99SBarry Smith     n9  = rank       -1;
55847c6ae99SBarry Smith     n12 = rank + m   -1;
55947c6ae99SBarry Smith     n15 = rank + 2*m -1;
56047c6ae99SBarry Smith     n18 = rank       -1 + (m*n);
56147c6ae99SBarry Smith     n21 = rank + m   -1 + (m*n);
56247c6ae99SBarry Smith     n24 = rank + 2*m -1 + (m*n);
56347c6ae99SBarry Smith   }
56447c6ae99SBarry Smith 
565ce00eea3SSatish Balay   if (xe == M) { /* First assume not corner or edge */
56647c6ae99SBarry Smith     n2  = rank -2*m +1 - (m*n);
56747c6ae99SBarry Smith     n5  = rank - m  +1 - (m*n);
56847c6ae99SBarry Smith     n8  = rank      +1 - (m*n);
56947c6ae99SBarry Smith     n11 = rank -2*m +1;
57047c6ae99SBarry Smith     n14 = rank - m  +1;
57147c6ae99SBarry Smith     n17 = rank      +1;
57247c6ae99SBarry Smith     n20 = rank -2*m +1 + (m*n);
57347c6ae99SBarry Smith     n23 = rank - m  +1 + (m*n);
57447c6ae99SBarry Smith     n26 = rank      +1 + (m*n);
57547c6ae99SBarry Smith   }
57647c6ae99SBarry Smith 
57747c6ae99SBarry Smith   if (ys==0) { /* First assume not corner or edge */
57847c6ae99SBarry Smith     n0  = rank + m * (n-1) -1 - (m*n);
57947c6ae99SBarry Smith     n1  = rank + m * (n-1)    - (m*n);
58047c6ae99SBarry Smith     n2  = rank + m * (n-1) +1 - (m*n);
58147c6ae99SBarry Smith     n9  = rank + m * (n-1) -1;
58247c6ae99SBarry Smith     n10 = rank + m * (n-1);
58347c6ae99SBarry Smith     n11 = rank + m * (n-1) +1;
58447c6ae99SBarry Smith     n18 = rank + m * (n-1) -1 + (m*n);
58547c6ae99SBarry Smith     n19 = rank + m * (n-1)    + (m*n);
58647c6ae99SBarry Smith     n20 = rank + m * (n-1) +1 + (m*n);
58747c6ae99SBarry Smith   }
58847c6ae99SBarry Smith 
58947c6ae99SBarry Smith   if (ye == N) { /* First assume not corner or edge */
59047c6ae99SBarry Smith     n6  = rank - m * (n-1) -1 - (m*n);
59147c6ae99SBarry Smith     n7  = rank - m * (n-1)    - (m*n);
59247c6ae99SBarry Smith     n8  = rank - m * (n-1) +1 - (m*n);
59347c6ae99SBarry Smith     n15 = rank - m * (n-1) -1;
59447c6ae99SBarry Smith     n16 = rank - m * (n-1);
59547c6ae99SBarry Smith     n17 = rank - m * (n-1) +1;
59647c6ae99SBarry Smith     n24 = rank - m * (n-1) -1 + (m*n);
59747c6ae99SBarry Smith     n25 = rank - m * (n-1)    + (m*n);
59847c6ae99SBarry Smith     n26 = rank - m * (n-1) +1 + (m*n);
59947c6ae99SBarry Smith   }
60047c6ae99SBarry Smith 
60147c6ae99SBarry Smith   if (zs == 0) { /* First assume not corner or edge */
60247c6ae99SBarry Smith     n0 = size - (m*n) + rank - m - 1;
60347c6ae99SBarry Smith     n1 = size - (m*n) + rank - m;
60447c6ae99SBarry Smith     n2 = size - (m*n) + rank - m + 1;
60547c6ae99SBarry Smith     n3 = size - (m*n) + rank - 1;
60647c6ae99SBarry Smith     n4 = size - (m*n) + rank;
60747c6ae99SBarry Smith     n5 = size - (m*n) + rank + 1;
60847c6ae99SBarry Smith     n6 = size - (m*n) + rank + m - 1;
60947c6ae99SBarry Smith     n7 = size - (m*n) + rank + m;
61047c6ae99SBarry Smith     n8 = size - (m*n) + rank + m + 1;
61147c6ae99SBarry Smith   }
61247c6ae99SBarry Smith 
61347c6ae99SBarry Smith   if (ze == P) { /* First assume not corner or edge */
61447c6ae99SBarry Smith     n18 = (m*n) - (size-rank) - m - 1;
61547c6ae99SBarry Smith     n19 = (m*n) - (size-rank) - m;
61647c6ae99SBarry Smith     n20 = (m*n) - (size-rank) - m + 1;
61747c6ae99SBarry Smith     n21 = (m*n) - (size-rank) - 1;
61847c6ae99SBarry Smith     n22 = (m*n) - (size-rank);
61947c6ae99SBarry Smith     n23 = (m*n) - (size-rank) + 1;
62047c6ae99SBarry Smith     n24 = (m*n) - (size-rank) + m - 1;
62147c6ae99SBarry Smith     n25 = (m*n) - (size-rank) + m;
62247c6ae99SBarry Smith     n26 = (m*n) - (size-rank) + m + 1;
62347c6ae99SBarry Smith   }
62447c6ae99SBarry Smith 
62547c6ae99SBarry Smith   if ((xs==0) && (zs==0)) { /* Assume an edge, not corner */
62647c6ae99SBarry Smith     n0 = size - m*n + rank + m-1 - m;
62747c6ae99SBarry Smith     n3 = size - m*n + rank + m-1;
62847c6ae99SBarry Smith     n6 = size - m*n + rank + m-1 + m;
62947c6ae99SBarry Smith   }
63047c6ae99SBarry Smith 
63147c6ae99SBarry Smith   if ((xs==0) && (ze==P)) { /* Assume an edge, not corner */
63247c6ae99SBarry Smith     n18 = m*n - (size - rank) + m-1 - m;
63347c6ae99SBarry Smith     n21 = m*n - (size - rank) + m-1;
63447c6ae99SBarry Smith     n24 = m*n - (size - rank) + m-1 + m;
63547c6ae99SBarry Smith   }
63647c6ae99SBarry Smith 
63747c6ae99SBarry Smith   if ((xs==0) && (ys==0)) { /* Assume an edge, not corner */
63847c6ae99SBarry Smith     n0  = rank + m*n -1 - m*n;
63947c6ae99SBarry Smith     n9  = rank + m*n -1;
64047c6ae99SBarry Smith     n18 = rank + m*n -1 + m*n;
64147c6ae99SBarry Smith   }
64247c6ae99SBarry Smith 
64347c6ae99SBarry Smith   if ((xs==0) && (ye==N)) { /* Assume an edge, not corner */
64447c6ae99SBarry Smith     n6  = rank - m*(n-1) + m-1 - m*n;
64547c6ae99SBarry Smith     n15 = rank - m*(n-1) + m-1;
64647c6ae99SBarry Smith     n24 = rank - m*(n-1) + m-1 + m*n;
64747c6ae99SBarry Smith   }
64847c6ae99SBarry Smith 
649ce00eea3SSatish Balay   if ((xe==M) && (zs==0)) { /* Assume an edge, not corner */
65047c6ae99SBarry Smith     n2 = size - (m*n-rank) - (m-1) - m;
65147c6ae99SBarry Smith     n5 = size - (m*n-rank) - (m-1);
65247c6ae99SBarry Smith     n8 = size - (m*n-rank) - (m-1) + m;
65347c6ae99SBarry Smith   }
65447c6ae99SBarry Smith 
655ce00eea3SSatish Balay   if ((xe==M) && (ze==P)) { /* Assume an edge, not corner */
65647c6ae99SBarry Smith     n20 = m*n - (size - rank) - (m-1) - m;
65747c6ae99SBarry Smith     n23 = m*n - (size - rank) - (m-1);
65847c6ae99SBarry Smith     n26 = m*n - (size - rank) - (m-1) + m;
65947c6ae99SBarry Smith   }
66047c6ae99SBarry Smith 
661ce00eea3SSatish Balay   if ((xe==M) && (ys==0)) { /* Assume an edge, not corner */
66247c6ae99SBarry Smith     n2  = rank + m*(n-1) - (m-1) - m*n;
66347c6ae99SBarry Smith     n11 = rank + m*(n-1) - (m-1);
66447c6ae99SBarry Smith     n20 = rank + m*(n-1) - (m-1) + m*n;
66547c6ae99SBarry Smith   }
66647c6ae99SBarry Smith 
667ce00eea3SSatish Balay   if ((xe==M) && (ye==N)) { /* Assume an edge, not corner */
66847c6ae99SBarry Smith     n8  = rank - m*n +1 - m*n;
66947c6ae99SBarry Smith     n17 = rank - m*n +1;
67047c6ae99SBarry Smith     n26 = rank - m*n +1 + m*n;
67147c6ae99SBarry Smith   }
67247c6ae99SBarry Smith 
67347c6ae99SBarry Smith   if ((ys==0) && (zs==0)) { /* Assume an edge, not corner */
67447c6ae99SBarry Smith     n0 = size - m + rank -1;
67547c6ae99SBarry Smith     n1 = size - m + rank;
67647c6ae99SBarry Smith     n2 = size - m + rank +1;
67747c6ae99SBarry Smith   }
67847c6ae99SBarry Smith 
67947c6ae99SBarry Smith   if ((ys==0) && (ze==P)) { /* Assume an edge, not corner */
68047c6ae99SBarry Smith     n18 = m*n - (size - rank) + m*(n-1) -1;
68147c6ae99SBarry Smith     n19 = m*n - (size - rank) + m*(n-1);
68247c6ae99SBarry Smith     n20 = m*n - (size - rank) + m*(n-1) +1;
68347c6ae99SBarry Smith   }
68447c6ae99SBarry Smith 
68547c6ae99SBarry Smith   if ((ye==N) && (zs==0)) { /* Assume an edge, not corner */
68647c6ae99SBarry Smith     n6 = size - (m*n-rank) - m * (n-1) -1;
68747c6ae99SBarry Smith     n7 = size - (m*n-rank) - m * (n-1);
68847c6ae99SBarry Smith     n8 = size - (m*n-rank) - m * (n-1) +1;
68947c6ae99SBarry Smith   }
69047c6ae99SBarry Smith 
69147c6ae99SBarry Smith   if ((ye==N) && (ze==P)) { /* Assume an edge, not corner */
69247c6ae99SBarry Smith     n24 = rank - (size-m) -1;
69347c6ae99SBarry Smith     n25 = rank - (size-m);
69447c6ae99SBarry Smith     n26 = rank - (size-m) +1;
69547c6ae99SBarry Smith   }
69647c6ae99SBarry Smith 
69747c6ae99SBarry Smith   /* Check for Corners */
6988865f1eaSKarl Rupp   if ((xs==0) && (ys==0) && (zs==0)) n0  = size -1;
6998865f1eaSKarl Rupp   if ((xs==0) && (ys==0) && (ze==P)) n18 = m*n-1;
7008865f1eaSKarl Rupp   if ((xs==0) && (ye==N) && (zs==0)) n6  = (size-1)-m*(n-1);
7018865f1eaSKarl Rupp   if ((xs==0) && (ye==N) && (ze==P)) n24 = m-1;
7028865f1eaSKarl Rupp   if ((xe==M) && (ys==0) && (zs==0)) n2  = size-m;
7038865f1eaSKarl Rupp   if ((xe==M) && (ys==0) && (ze==P)) n20 = m*n-m;
7048865f1eaSKarl Rupp   if ((xe==M) && (ye==N) && (zs==0)) n8  = size-m*n;
7058865f1eaSKarl Rupp   if ((xe==M) && (ye==N) && (ze==P)) n26 = 0;
70647c6ae99SBarry Smith 
70747c6ae99SBarry Smith   /* Check for when not X,Y, and Z Periodic */
70847c6ae99SBarry Smith 
70947c6ae99SBarry Smith   /* If not X periodic */
710bff4a2f0SMatthew G. Knepley   if (bx != DM_BOUNDARY_PERIODIC) {
7118865f1eaSKarl Rupp     if (xs==0) n0 = n3 = n6 = n9  = n12 = n15 = n18 = n21 = n24 = -2;
7128865f1eaSKarl Rupp     if (xe==M) n2 = n5 = n8 = n11 = n14 = n17 = n20 = n23 = n26 = -2;
71347c6ae99SBarry Smith   }
71447c6ae99SBarry Smith 
71547c6ae99SBarry Smith   /* If not Y periodic */
716bff4a2f0SMatthew G. Knepley   if (by != DM_BOUNDARY_PERIODIC) {
7178865f1eaSKarl Rupp     if (ys==0) n0 = n1 = n2 = n9  = n10 = n11 = n18 = n19 = n20 = -2;
7188865f1eaSKarl Rupp     if (ye==N) n6 = n7 = n8 = n15 = n16 = n17 = n24 = n25 = n26 = -2;
71947c6ae99SBarry Smith   }
72047c6ae99SBarry Smith 
72147c6ae99SBarry Smith   /* If not Z periodic */
722bff4a2f0SMatthew G. Knepley   if (bz != DM_BOUNDARY_PERIODIC) {
7238865f1eaSKarl Rupp     if (zs==0) n0  = n1  = n2  = n3  = n4  = n5  = n6  = n7  = n8  = -2;
7248865f1eaSKarl Rupp     if (ze==P) n18 = n19 = n20 = n21 = n22 = n23 = n24 = n25 = n26 = -2;
72547c6ae99SBarry Smith   }
72647c6ae99SBarry Smith 
727785e854fSJed Brown   ierr = PetscMalloc1(27,&dd->neighbors);CHKERRQ(ierr);
7288865f1eaSKarl Rupp 
72947c6ae99SBarry Smith   dd->neighbors[0]  = n0;
73047c6ae99SBarry Smith   dd->neighbors[1]  = n1;
73147c6ae99SBarry Smith   dd->neighbors[2]  = n2;
73247c6ae99SBarry Smith   dd->neighbors[3]  = n3;
73347c6ae99SBarry Smith   dd->neighbors[4]  = n4;
73447c6ae99SBarry Smith   dd->neighbors[5]  = n5;
73547c6ae99SBarry Smith   dd->neighbors[6]  = n6;
73647c6ae99SBarry Smith   dd->neighbors[7]  = n7;
73747c6ae99SBarry Smith   dd->neighbors[8]  = n8;
73847c6ae99SBarry Smith   dd->neighbors[9]  = n9;
73947c6ae99SBarry Smith   dd->neighbors[10] = n10;
74047c6ae99SBarry Smith   dd->neighbors[11] = n11;
74147c6ae99SBarry Smith   dd->neighbors[12] = n12;
74247c6ae99SBarry Smith   dd->neighbors[13] = rank;
74347c6ae99SBarry Smith   dd->neighbors[14] = n14;
74447c6ae99SBarry Smith   dd->neighbors[15] = n15;
74547c6ae99SBarry Smith   dd->neighbors[16] = n16;
74647c6ae99SBarry Smith   dd->neighbors[17] = n17;
74747c6ae99SBarry Smith   dd->neighbors[18] = n18;
74847c6ae99SBarry Smith   dd->neighbors[19] = n19;
74947c6ae99SBarry Smith   dd->neighbors[20] = n20;
75047c6ae99SBarry Smith   dd->neighbors[21] = n21;
75147c6ae99SBarry Smith   dd->neighbors[22] = n22;
75247c6ae99SBarry Smith   dd->neighbors[23] = n23;
75347c6ae99SBarry Smith   dd->neighbors[24] = n24;
75447c6ae99SBarry Smith   dd->neighbors[25] = n25;
75547c6ae99SBarry Smith   dd->neighbors[26] = n26;
75647c6ae99SBarry Smith 
75747c6ae99SBarry Smith   /* If star stencil then delete the corner neighbors */
758d9c9ebe5SPeter Brune   if (stencil_type == DMDA_STENCIL_STAR) {
75947c6ae99SBarry Smith     /* save information about corner neighbors */
76047c6ae99SBarry Smith     sn0 = n0; sn1 = n1; sn2 = n2; sn3 = n3; sn5 = n5; sn6 = n6; sn7 = n7;
76147c6ae99SBarry Smith     sn8 = n8; sn9 = n9; sn11 = n11; sn15 = n15; sn17 = n17; sn18 = n18;
76247c6ae99SBarry Smith     sn19 = n19; sn20 = n20; sn21 = n21; sn23 = n23; sn24 = n24; sn25 = n25;
76347c6ae99SBarry Smith     sn26 = n26;
7648865f1eaSKarl Rupp     n0 = n1 = n2 = n3 = n5 = n6 = n7 = n8 = n9 = n11 = n15 = n17 = n18 = n19 = n20 = n21 = n23 = n24 = n25 = n26 = -1;
76547c6ae99SBarry Smith   }
76647c6ae99SBarry Smith 
767785e854fSJed Brown   ierr = PetscMalloc1((Xe-Xs)*(Ye-Ys)*(Ze-Zs),&idx);CHKERRQ(ierr);
76847c6ae99SBarry Smith 
76947c6ae99SBarry Smith   nn = 0;
77047c6ae99SBarry Smith   /* Bottom Level */
77147c6ae99SBarry Smith   for (k=0; k<s_z; k++) {
77247c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
77347c6ae99SBarry Smith       if (n0 >= 0) { /* left below */
774ce00eea3SSatish Balay         x_t = lx[n0 % m];
77547c6ae99SBarry Smith         y_t = ly[(n0 % (m*n))/m];
77647c6ae99SBarry Smith         z_t = lz[n0 / (m*n)];
77747c6ae99SBarry 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;
7788865f1eaSKarl 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 */
7798865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
78047c6ae99SBarry Smith       }
78147c6ae99SBarry Smith       if (n1 >= 0) { /* directly below */
78247c6ae99SBarry Smith         x_t = x;
78347c6ae99SBarry Smith         y_t = ly[(n1 % (m*n))/m];
78447c6ae99SBarry Smith         z_t = lz[n1 / (m*n)];
78547c6ae99SBarry 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;
7868865f1eaSKarl Rupp         if (twod && (s_t < 0)) s_t = bases[n1] + x_t*y_t*z_t - (s_y+1-i)*x_t; /* 2D case */
7878865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
78847c6ae99SBarry Smith       }
78947c6ae99SBarry Smith       if (n2 >= 0) { /* right below */
790ce00eea3SSatish Balay         x_t = lx[n2 % m];
79147c6ae99SBarry Smith         y_t = ly[(n2 % (m*n))/m];
79247c6ae99SBarry Smith         z_t = lz[n2 / (m*n)];
79347c6ae99SBarry 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;
7948865f1eaSKarl Rupp         if (twod && (s_t < 0)) s_t = bases[n2] + x_t*y_t*z_t - (s_y+1-i)*x_t; /* 2D case */
7958865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
79647c6ae99SBarry Smith       }
79747c6ae99SBarry Smith     }
79847c6ae99SBarry Smith 
79947c6ae99SBarry Smith     for (i=0; i<y; i++) {
80047c6ae99SBarry Smith       if (n3 >= 0) { /* directly left */
801ce00eea3SSatish Balay         x_t = lx[n3 % m];
80247c6ae99SBarry Smith         y_t = y;
80347c6ae99SBarry Smith         z_t = lz[n3 / (m*n)];
80447c6ae99SBarry Smith         s_t = bases[n3] + (i+1)*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8058865f1eaSKarl 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 */
8068865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
80747c6ae99SBarry Smith       }
80847c6ae99SBarry Smith 
80947c6ae99SBarry Smith       if (n4 >= 0) { /* middle */
81047c6ae99SBarry Smith         x_t = x;
81147c6ae99SBarry Smith         y_t = y;
81247c6ae99SBarry Smith         z_t = lz[n4 / (m*n)];
81347c6ae99SBarry Smith         s_t = bases[n4] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8148865f1eaSKarl 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 */
8158865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
816bff4a2f0SMatthew G. Knepley       } else if (bz == DM_BOUNDARY_MIRROR) {
817a6fd6b0aSBarry Smith         for (j=0; j<x; j++) idx[nn++] = bases[rank] + j + x*i + (s_z - k - 1)*x*y;
81847c6ae99SBarry Smith       }
81947c6ae99SBarry Smith 
82047c6ae99SBarry Smith       if (n5 >= 0) { /* directly right */
821ce00eea3SSatish Balay         x_t = lx[n5 % m];
82247c6ae99SBarry Smith         y_t = y;
82347c6ae99SBarry Smith         z_t = lz[n5 / (m*n)];
82447c6ae99SBarry Smith         s_t = bases[n5] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8258865f1eaSKarl 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 */
8268865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
82747c6ae99SBarry Smith       }
82847c6ae99SBarry Smith     }
82947c6ae99SBarry Smith 
83047c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
83147c6ae99SBarry Smith       if (n6 >= 0) { /* left above */
832ce00eea3SSatish Balay         x_t = lx[n6 % m];
83347c6ae99SBarry Smith         y_t = ly[(n6 % (m*n))/m];
83447c6ae99SBarry Smith         z_t = lz[n6 / (m*n)];
83547c6ae99SBarry Smith         s_t = bases[n6] + i*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8368865f1eaSKarl 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 */
8378865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
83847c6ae99SBarry Smith       }
83947c6ae99SBarry Smith       if (n7 >= 0) { /* directly above */
84047c6ae99SBarry Smith         x_t = x;
84147c6ae99SBarry Smith         y_t = ly[(n7 % (m*n))/m];
84247c6ae99SBarry Smith         z_t = lz[n7 / (m*n)];
84347c6ae99SBarry Smith         s_t = bases[n7] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8448865f1eaSKarl 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 */
8458865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
84647c6ae99SBarry Smith       }
84747c6ae99SBarry Smith       if (n8 >= 0) { /* right above */
848ce00eea3SSatish Balay         x_t = lx[n8 % m];
84947c6ae99SBarry Smith         y_t = ly[(n8 % (m*n))/m];
85047c6ae99SBarry Smith         z_t = lz[n8 / (m*n)];
85147c6ae99SBarry Smith         s_t = bases[n8] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
8528865f1eaSKarl 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 */
8538865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
85447c6ae99SBarry Smith       }
85547c6ae99SBarry Smith     }
85647c6ae99SBarry Smith   }
85747c6ae99SBarry Smith 
85847c6ae99SBarry Smith   /* Middle Level */
85947c6ae99SBarry Smith   for (k=0; k<z; k++) {
86047c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
86147c6ae99SBarry Smith       if (n9 >= 0) { /* left below */
862ce00eea3SSatish Balay         x_t = lx[n9 % m];
86347c6ae99SBarry Smith         y_t = ly[(n9 % (m*n))/m];
86447c6ae99SBarry Smith         /* z_t = z; */
86547c6ae99SBarry Smith         s_t = bases[n9] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
8668865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
86747c6ae99SBarry Smith       }
86847c6ae99SBarry Smith       if (n10 >= 0) { /* directly below */
86947c6ae99SBarry Smith         x_t = x;
87047c6ae99SBarry Smith         y_t = ly[(n10 % (m*n))/m];
87147c6ae99SBarry Smith         /* z_t = z; */
87247c6ae99SBarry Smith         s_t = bases[n10] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
8738865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
874bff4a2f0SMatthew G. Knepley       }  else if (by == DM_BOUNDARY_MIRROR) {
875a6fd6b0aSBarry Smith         for (j=0; j<x; j++) idx[nn++] = bases[rank] + j + k*x*y + (s_y - i)*x;
87647c6ae99SBarry Smith       }
87747c6ae99SBarry Smith       if (n11 >= 0) { /* right below */
878ce00eea3SSatish Balay         x_t = lx[n11 % m];
87947c6ae99SBarry Smith         y_t = ly[(n11 % (m*n))/m];
88047c6ae99SBarry Smith         /* z_t = z; */
88147c6ae99SBarry Smith         s_t = bases[n11] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
8828865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
88347c6ae99SBarry Smith       }
88447c6ae99SBarry Smith     }
88547c6ae99SBarry Smith 
88647c6ae99SBarry Smith     for (i=0; i<y; i++) {
88747c6ae99SBarry Smith       if (n12 >= 0) { /* directly left */
888ce00eea3SSatish Balay         x_t = lx[n12 % m];
88947c6ae99SBarry Smith         y_t = y;
89047c6ae99SBarry Smith         /* z_t = z; */
89147c6ae99SBarry Smith         s_t = bases[n12] + (i+1)*x_t - s_x + k*x_t*y_t;
8928865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
893bff4a2f0SMatthew G. Knepley       }  else if (bx == DM_BOUNDARY_MIRROR) {
894a6fd6b0aSBarry Smith         for (j=0; j<s_x; j++) idx[nn++] = bases[rank] + s_x - j - 1 + k*x*y + i*x;
89547c6ae99SBarry Smith       }
89647c6ae99SBarry Smith 
89747c6ae99SBarry Smith       /* Interior */
89847c6ae99SBarry Smith       s_t = bases[rank] + i*x + k*x*y;
8998865f1eaSKarl Rupp       for (j=0; j<x; j++) idx[nn++] = s_t++;
90047c6ae99SBarry Smith 
90147c6ae99SBarry Smith       if (n14 >= 0) { /* directly right */
902ce00eea3SSatish Balay         x_t = lx[n14 % m];
90347c6ae99SBarry Smith         y_t = y;
90447c6ae99SBarry Smith         /* z_t = z; */
90547c6ae99SBarry Smith         s_t = bases[n14] + i*x_t + k*x_t*y_t;
9068865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
907bff4a2f0SMatthew G. Knepley       } else if (bx == DM_BOUNDARY_MIRROR) {
908a6fd6b0aSBarry Smith         for (j=0; j<s_x; j++) idx[nn++] = bases[rank] + x - j - 1 + k*x*y + i*x;
90947c6ae99SBarry Smith       }
91047c6ae99SBarry Smith     }
91147c6ae99SBarry Smith 
91247c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
91347c6ae99SBarry Smith       if (n15 >= 0) { /* left above */
914ce00eea3SSatish Balay         x_t = lx[n15 % m];
91547c6ae99SBarry Smith         y_t = ly[(n15 % (m*n))/m];
91647c6ae99SBarry Smith         /* z_t = z; */
91747c6ae99SBarry Smith         s_t = bases[n15] + i*x_t - s_x + k*x_t*y_t;
9188865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
91947c6ae99SBarry Smith       }
92047c6ae99SBarry Smith       if (n16 >= 0) { /* directly above */
92147c6ae99SBarry Smith         x_t = x;
92247c6ae99SBarry Smith         y_t = ly[(n16 % (m*n))/m];
92347c6ae99SBarry Smith         /* z_t = z; */
92447c6ae99SBarry Smith         s_t = bases[n16] + (i-1)*x_t + k*x_t*y_t;
9258865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
926bff4a2f0SMatthew G. Knepley       } else if (by == DM_BOUNDARY_MIRROR) {
927a6fd6b0aSBarry Smith         for (j=0; j<x; j++) idx[nn++] = bases[rank] + j + k*x*y + (y-i)*x;
92847c6ae99SBarry Smith       }
92947c6ae99SBarry Smith       if (n17 >= 0) { /* right above */
930ce00eea3SSatish Balay         x_t = lx[n17 % m];
93147c6ae99SBarry Smith         y_t = ly[(n17 % (m*n))/m];
93247c6ae99SBarry Smith         /* z_t = z; */
93347c6ae99SBarry Smith         s_t = bases[n17] + (i-1)*x_t + k*x_t*y_t;
9348865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
93547c6ae99SBarry Smith       }
93647c6ae99SBarry Smith     }
93747c6ae99SBarry Smith   }
93847c6ae99SBarry Smith 
93947c6ae99SBarry Smith   /* Upper Level */
94047c6ae99SBarry Smith   for (k=0; k<s_z; k++) {
94147c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
94247c6ae99SBarry Smith       if (n18 >= 0) { /* left below */
943ce00eea3SSatish Balay         x_t = lx[n18 % m];
94447c6ae99SBarry Smith         y_t = ly[(n18 % (m*n))/m];
94547c6ae99SBarry Smith         /* z_t = lz[n18 / (m*n)]; */
94647c6ae99SBarry Smith         s_t = bases[n18] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
9478865f1eaSKarl 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 */
9488865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
94947c6ae99SBarry Smith       }
95047c6ae99SBarry Smith       if (n19 >= 0) { /* directly below */
95147c6ae99SBarry Smith         x_t = x;
95247c6ae99SBarry Smith         y_t = ly[(n19 % (m*n))/m];
95347c6ae99SBarry Smith         /* z_t = lz[n19 / (m*n)]; */
95447c6ae99SBarry Smith         s_t = bases[n19] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
9558865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n19] - (s_y+1-i)*x_t + x_t*y_t; /* 2d case */
9568865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
95747c6ae99SBarry Smith       }
95847c6ae99SBarry Smith       if (n20 >= 0) { /* right below */
959ce00eea3SSatish Balay         x_t = lx[n20 % m];
96047c6ae99SBarry Smith         y_t = ly[(n20 % (m*n))/m];
96147c6ae99SBarry Smith         /* z_t = lz[n20 / (m*n)]; */
96247c6ae99SBarry Smith         s_t = bases[n20] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
9638865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n20] - (s_y+1-i)*x_t + x_t*y_t; /* 2d case */
9648865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
96547c6ae99SBarry Smith       }
96647c6ae99SBarry Smith     }
96747c6ae99SBarry Smith 
96847c6ae99SBarry Smith     for (i=0; i<y; i++) {
96947c6ae99SBarry Smith       if (n21 >= 0) { /* directly left */
970ce00eea3SSatish Balay         x_t = lx[n21 % m];
97147c6ae99SBarry Smith         y_t = y;
97247c6ae99SBarry Smith         /* z_t = lz[n21 / (m*n)]; */
97347c6ae99SBarry Smith         s_t = bases[n21] + (i+1)*x_t - s_x + k*x_t*y_t;
9748865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n21] + (i+1)*x_t - s_x;  /* 2d case */
9758865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
97647c6ae99SBarry Smith       }
97747c6ae99SBarry Smith 
97847c6ae99SBarry Smith       if (n22 >= 0) { /* middle */
97947c6ae99SBarry Smith         x_t = x;
98047c6ae99SBarry Smith         y_t = y;
98147c6ae99SBarry Smith         /* z_t = lz[n22 / (m*n)]; */
98247c6ae99SBarry Smith         s_t = bases[n22] + i*x_t + k*x_t*y_t;
9838865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n22] + i*x_t; /* 2d case */
9848865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
985bff4a2f0SMatthew G. Knepley       } else if (bz == DM_BOUNDARY_MIRROR) {
986a6fd6b0aSBarry Smith         for (j=0; j<x; j++) idx[nn++] = bases[rank] + j + (z-k-1)*x*y + i*x;
98747c6ae99SBarry Smith       }
98847c6ae99SBarry Smith 
98947c6ae99SBarry Smith       if (n23 >= 0) { /* directly right */
990ce00eea3SSatish Balay         x_t = lx[n23 % m];
99147c6ae99SBarry Smith         y_t = y;
99247c6ae99SBarry Smith         /* z_t = lz[n23 / (m*n)]; */
99347c6ae99SBarry Smith         s_t = bases[n23] + i*x_t + k*x_t*y_t;
9948865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n23] + i*x_t; /* 2d case */
9958865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
99647c6ae99SBarry Smith       }
99747c6ae99SBarry Smith     }
99847c6ae99SBarry Smith 
99947c6ae99SBarry Smith     for (i=1; i<=s_y; i++) {
100047c6ae99SBarry Smith       if (n24 >= 0) { /* left above */
1001ce00eea3SSatish Balay         x_t = lx[n24 % m];
100247c6ae99SBarry Smith         y_t = ly[(n24 % (m*n))/m];
100347c6ae99SBarry Smith         /* z_t = lz[n24 / (m*n)]; */
100447c6ae99SBarry Smith         s_t = bases[n24] + i*x_t - s_x + k*x_t*y_t;
10058865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n24] + i*x_t - s_x; /* 2d case */
10068865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
100747c6ae99SBarry Smith       }
100847c6ae99SBarry Smith       if (n25 >= 0) { /* directly above */
100947c6ae99SBarry Smith         x_t = x;
101047c6ae99SBarry Smith         y_t = ly[(n25 % (m*n))/m];
101147c6ae99SBarry Smith         /* z_t = lz[n25 / (m*n)]; */
101247c6ae99SBarry Smith         s_t = bases[n25] + (i-1)*x_t + k*x_t*y_t;
10138865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n25] + (i-1)*x_t; /* 2d case */
10148865f1eaSKarl Rupp         for (j=0; j<x_t; j++) idx[nn++] = s_t++;
101547c6ae99SBarry Smith       }
101647c6ae99SBarry Smith       if (n26 >= 0) { /* right above */
1017ce00eea3SSatish Balay         x_t = lx[n26 % m];
101847c6ae99SBarry Smith         y_t = ly[(n26 % (m*n))/m];
101947c6ae99SBarry Smith         /* z_t = lz[n26 / (m*n)]; */
102047c6ae99SBarry Smith         s_t = bases[n26] + (i-1)*x_t + k*x_t*y_t;
10218865f1eaSKarl Rupp         if (twod && (s_t >= M*N*P)) s_t = bases[n26] + (i-1)*x_t; /* 2d case */
10228865f1eaSKarl Rupp         for (j=0; j<s_x; j++) idx[nn++] = s_t++;
102347c6ae99SBarry Smith       }
102447c6ae99SBarry Smith     }
102547c6ae99SBarry Smith   }
1026ce00eea3SSatish Balay 
1027b1fb7eb7SBarry Smith   ierr = ISCreateBlock(comm,dof,nn,idx,PETSC_USE_POINTER,&from);CHKERRQ(ierr);
10289448b7f1SJunchao Zhang   ierr = VecScatterCreate(global,from,local,to,&gtol);CHKERRQ(ierr);
10293bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)da,(PetscObject)gtol);CHKERRQ(ierr);
1030fcfd50ebSBarry Smith   ierr = ISDestroy(&to);CHKERRQ(ierr);
1031fcfd50ebSBarry Smith   ierr = ISDestroy(&from);CHKERRQ(ierr);
103247c6ae99SBarry Smith 
1033d9c9ebe5SPeter Brune   if (stencil_type == DMDA_STENCIL_STAR) {
103447c6ae99SBarry Smith     n0  = sn0;  n1  = sn1;  n2  = sn2;  n3  = sn3;  n5  = sn5;  n6  = sn6; n7 = sn7;
103547c6ae99SBarry Smith     n8  = sn8;  n9  = sn9;  n11 = sn11; n15 = sn15; n17 = sn17; n18 = sn18;
103647c6ae99SBarry Smith     n19 = sn19; n20 = sn20; n21 = sn21; n23 = sn23; n24 = sn24; n25 = sn25;
103747c6ae99SBarry Smith     n26 = sn26;
1038ce00eea3SSatish Balay   }
103947c6ae99SBarry Smith 
1040288e7d53SBarry Smith   if (((stencil_type == DMDA_STENCIL_STAR) || (bx != DM_BOUNDARY_PERIODIC && bx) || (by != DM_BOUNDARY_PERIODIC && by) || (bz != DM_BOUNDARY_PERIODIC && bz))) {
1041ce00eea3SSatish Balay     /*
1042ce00eea3SSatish Balay         Recompute the local to global mappings, this time keeping the
1043ce00eea3SSatish Balay       information about the cross corner processor numbers.
1044ce00eea3SSatish Balay     */
104547c6ae99SBarry Smith     nn = 0;
104647c6ae99SBarry Smith     /* Bottom Level */
104747c6ae99SBarry Smith     for (k=0; k<s_z; k++) {
104847c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
104947c6ae99SBarry Smith         if (n0 >= 0) { /* left below */
1050ce00eea3SSatish Balay           x_t = lx[n0 % m];
105147c6ae99SBarry Smith           y_t = ly[(n0 % (m*n))/m];
105247c6ae99SBarry Smith           z_t = lz[n0 / (m*n)];
105347c6ae99SBarry 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;
10548865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1055ce00eea3SSatish Balay         } else if (Xs-xs < 0 && Ys-ys < 0 && Zs-zs < 0) {
10568865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
105747c6ae99SBarry Smith         }
105847c6ae99SBarry Smith         if (n1 >= 0) { /* directly below */
105947c6ae99SBarry Smith           x_t = x;
106047c6ae99SBarry Smith           y_t = ly[(n1 % (m*n))/m];
106147c6ae99SBarry Smith           z_t = lz[n1 / (m*n)];
106247c6ae99SBarry 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;
10638865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1064ce00eea3SSatish Balay         } else if (Ys-ys < 0 && Zs-zs < 0) {
10658865f1eaSKarl Rupp           for (j=0; j<x; j++) idx[nn++] = -1;
106647c6ae99SBarry Smith         }
106747c6ae99SBarry Smith         if (n2 >= 0) { /* right below */
1068ce00eea3SSatish Balay           x_t = lx[n2 % m];
106947c6ae99SBarry Smith           y_t = ly[(n2 % (m*n))/m];
107047c6ae99SBarry Smith           z_t = lz[n2 / (m*n)];
107147c6ae99SBarry 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;
10728865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1073ce00eea3SSatish Balay         } else if (xe-Xe < 0 && Ys-ys < 0 && Zs-zs < 0) {
10748865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
107547c6ae99SBarry Smith         }
107647c6ae99SBarry Smith       }
107747c6ae99SBarry Smith 
107847c6ae99SBarry Smith       for (i=0; i<y; i++) {
107947c6ae99SBarry Smith         if (n3 >= 0) { /* directly left */
1080ce00eea3SSatish Balay           x_t = lx[n3 % m];
108147c6ae99SBarry Smith           y_t = y;
108247c6ae99SBarry Smith           z_t = lz[n3 / (m*n)];
108347c6ae99SBarry Smith           s_t = bases[n3] + (i+1)*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
10848865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1085ce00eea3SSatish Balay         } else if (Xs-xs < 0 && Zs-zs < 0) {
10868865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
108747c6ae99SBarry Smith         }
108847c6ae99SBarry Smith 
108947c6ae99SBarry Smith         if (n4 >= 0) { /* middle */
109047c6ae99SBarry Smith           x_t = x;
109147c6ae99SBarry Smith           y_t = y;
109247c6ae99SBarry Smith           z_t = lz[n4 / (m*n)];
109347c6ae99SBarry Smith           s_t = bases[n4] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
10948865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1095ce00eea3SSatish Balay         } else if (Zs-zs < 0) {
1096bff4a2f0SMatthew G. Knepley           if (bz == DM_BOUNDARY_MIRROR) {
1097a6fd6b0aSBarry Smith             for (j=0; j<x; j++) idx[nn++] = bases[rank] + j + x*i + (s_z - k - 1)*x*y;
1098c2859e5eSBarry Smith           } else {
10998865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = -1;
110047c6ae99SBarry Smith           }
1101c2859e5eSBarry Smith         }
110247c6ae99SBarry Smith 
110347c6ae99SBarry Smith         if (n5 >= 0) { /* directly right */
1104ce00eea3SSatish Balay           x_t = lx[n5 % m];
110547c6ae99SBarry Smith           y_t = y;
110647c6ae99SBarry Smith           z_t = lz[n5 / (m*n)];
110747c6ae99SBarry Smith           s_t = bases[n5] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
11088865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1109ce00eea3SSatish Balay         } else if (xe-Xe < 0 && Zs-zs < 0) {
11108865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
111147c6ae99SBarry Smith         }
111247c6ae99SBarry Smith       }
111347c6ae99SBarry Smith 
111447c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
111547c6ae99SBarry Smith         if (n6 >= 0) { /* left above */
1116ce00eea3SSatish Balay           x_t = lx[n6 % m];
111747c6ae99SBarry Smith           y_t = ly[(n6 % (m*n))/m];
111847c6ae99SBarry Smith           z_t = lz[n6 / (m*n)];
111947c6ae99SBarry Smith           s_t = bases[n6] + i*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
11208865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1121ce00eea3SSatish Balay         } else if (Xs-xs < 0 && ye-Ye < 0 && Zs-zs < 0) {
11228865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
112347c6ae99SBarry Smith         }
112447c6ae99SBarry Smith         if (n7 >= 0) { /* directly above */
112547c6ae99SBarry Smith           x_t = x;
112647c6ae99SBarry Smith           y_t = ly[(n7 % (m*n))/m];
112747c6ae99SBarry Smith           z_t = lz[n7 / (m*n)];
112847c6ae99SBarry Smith           s_t = bases[n7] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
11298865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1130ce00eea3SSatish Balay         } else if (ye-Ye < 0 && Zs-zs < 0) {
11318865f1eaSKarl Rupp           for (j=0; j<x; j++) idx[nn++] = -1;
113247c6ae99SBarry Smith         }
113347c6ae99SBarry Smith         if (n8 >= 0) { /* right above */
1134ce00eea3SSatish Balay           x_t = lx[n8 % m];
113547c6ae99SBarry Smith           y_t = ly[(n8 % (m*n))/m];
113647c6ae99SBarry Smith           z_t = lz[n8 / (m*n)];
113747c6ae99SBarry Smith           s_t = bases[n8] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
11388865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1139ce00eea3SSatish Balay         } else if (xe-Xe < 0 && ye-Ye < 0 && Zs-zs < 0) {
11408865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
114147c6ae99SBarry Smith         }
114247c6ae99SBarry Smith       }
114347c6ae99SBarry Smith     }
114447c6ae99SBarry Smith 
114547c6ae99SBarry Smith     /* Middle Level */
114647c6ae99SBarry Smith     for (k=0; k<z; k++) {
114747c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
114847c6ae99SBarry Smith         if (n9 >= 0) { /* left below */
1149ce00eea3SSatish Balay           x_t = lx[n9 % m];
115047c6ae99SBarry Smith           y_t = ly[(n9 % (m*n))/m];
115147c6ae99SBarry Smith           /* z_t = z; */
115247c6ae99SBarry Smith           s_t = bases[n9] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
11538865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1154ce00eea3SSatish Balay         } else if (Xs-xs < 0 && Ys-ys < 0) {
11558865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
115647c6ae99SBarry Smith         }
115747c6ae99SBarry Smith         if (n10 >= 0) { /* directly below */
115847c6ae99SBarry Smith           x_t = x;
115947c6ae99SBarry Smith           y_t = ly[(n10 % (m*n))/m];
116047c6ae99SBarry Smith           /* z_t = z; */
116147c6ae99SBarry Smith           s_t = bases[n10] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
11628865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1163ce00eea3SSatish Balay         } else if (Ys-ys < 0) {
1164bff4a2f0SMatthew G. Knepley           if (by == DM_BOUNDARY_MIRROR) {
1165feb237baSPierre Jolivet             for (j=0; j<x; j++) idx[nn++] = bases[rank] + j + k*x*y + (s_y - i)*x;
1166c2859e5eSBarry Smith           } else {
11678865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = -1;
1168c2859e5eSBarry Smith           }
116947c6ae99SBarry Smith         }
117047c6ae99SBarry Smith         if (n11 >= 0) { /* right below */
1171ce00eea3SSatish Balay           x_t = lx[n11 % m];
117247c6ae99SBarry Smith           y_t = ly[(n11 % (m*n))/m];
117347c6ae99SBarry Smith           /* z_t = z; */
117447c6ae99SBarry Smith           s_t = bases[n11] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
11758865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1176ce00eea3SSatish Balay         } else if (xe-Xe < 0 && Ys-ys < 0) {
11778865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
117847c6ae99SBarry Smith         }
117947c6ae99SBarry Smith       }
118047c6ae99SBarry Smith 
118147c6ae99SBarry Smith       for (i=0; i<y; i++) {
118247c6ae99SBarry Smith         if (n12 >= 0) { /* directly left */
1183ce00eea3SSatish Balay           x_t = lx[n12 % m];
118447c6ae99SBarry Smith           y_t = y;
118547c6ae99SBarry Smith           /* z_t = z; */
118647c6ae99SBarry Smith           s_t = bases[n12] + (i+1)*x_t - s_x + k*x_t*y_t;
11878865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1188ce00eea3SSatish Balay         } else if (Xs-xs < 0) {
1189bff4a2f0SMatthew G. Knepley           if (bx == DM_BOUNDARY_MIRROR) {
1190a6fd6b0aSBarry Smith             for (j=0; j<s_x; j++) idx[nn++] = bases[rank] + s_x - j - 1 + k*x*y + i*x;
1191c2859e5eSBarry Smith           } else {
11928865f1eaSKarl Rupp             for (j=0; j<s_x; j++) idx[nn++] = -1;
119347c6ae99SBarry Smith           }
1194c2859e5eSBarry Smith         }
119547c6ae99SBarry Smith 
119647c6ae99SBarry Smith         /* Interior */
119747c6ae99SBarry Smith         s_t = bases[rank] + i*x + k*x*y;
11988865f1eaSKarl Rupp         for (j=0; j<x; j++) idx[nn++] = s_t++;
119947c6ae99SBarry Smith 
120047c6ae99SBarry Smith         if (n14 >= 0) { /* directly right */
1201ce00eea3SSatish Balay           x_t = lx[n14 % m];
120247c6ae99SBarry Smith           y_t = y;
120347c6ae99SBarry Smith           /* z_t = z; */
120447c6ae99SBarry Smith           s_t = bases[n14] + i*x_t + k*x_t*y_t;
12058865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1206ce00eea3SSatish Balay         } else if (xe-Xe < 0) {
1207bff4a2f0SMatthew G. Knepley           if (bx == DM_BOUNDARY_MIRROR) {
1208a6fd6b0aSBarry Smith             for (j=0; j<s_x; j++) idx[nn++] = bases[rank] + x - j - 1 + k*x*y + i*x;
1209c2859e5eSBarry Smith           } else {
12108865f1eaSKarl Rupp             for (j=0; j<s_x; j++) idx[nn++] = -1;
121147c6ae99SBarry Smith           }
121247c6ae99SBarry Smith         }
1213c2859e5eSBarry Smith       }
121447c6ae99SBarry Smith 
121547c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
121647c6ae99SBarry Smith         if (n15 >= 0) { /* left above */
1217ce00eea3SSatish Balay           x_t = lx[n15 % m];
121847c6ae99SBarry Smith           y_t = ly[(n15 % (m*n))/m];
121947c6ae99SBarry Smith           /* z_t = z; */
122047c6ae99SBarry Smith           s_t = bases[n15] + i*x_t - s_x + k*x_t*y_t;
12218865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1222ce00eea3SSatish Balay         } else if (Xs-xs < 0 && ye-Ye < 0) {
12238865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
122447c6ae99SBarry Smith         }
122547c6ae99SBarry Smith         if (n16 >= 0) { /* directly above */
122647c6ae99SBarry Smith           x_t = x;
122747c6ae99SBarry Smith           y_t = ly[(n16 % (m*n))/m];
122847c6ae99SBarry Smith           /* z_t = z; */
122947c6ae99SBarry Smith           s_t = bases[n16] + (i-1)*x_t + k*x_t*y_t;
12308865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1231ce00eea3SSatish Balay         } else if (ye-Ye < 0) {
1232bff4a2f0SMatthew G. Knepley           if (by == DM_BOUNDARY_MIRROR) {
1233a6fd6b0aSBarry Smith             for (j=0; j<x; j++) idx[nn++] = bases[rank] + j + k*x*y + (y-i)*x;
1234c2859e5eSBarry Smith           } else {
12358865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = -1;
123647c6ae99SBarry Smith           }
1237c2859e5eSBarry Smith         }
123847c6ae99SBarry Smith         if (n17 >= 0) { /* right above */
1239ce00eea3SSatish Balay           x_t = lx[n17 % m];
124047c6ae99SBarry Smith           y_t = ly[(n17 % (m*n))/m];
124147c6ae99SBarry Smith           /* z_t = z; */
124247c6ae99SBarry Smith           s_t = bases[n17] + (i-1)*x_t + k*x_t*y_t;
12438865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1244ce00eea3SSatish Balay         } else if (xe-Xe < 0 && ye-Ye < 0) {
12458865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
124647c6ae99SBarry Smith         }
124747c6ae99SBarry Smith       }
124847c6ae99SBarry Smith     }
124947c6ae99SBarry Smith 
125047c6ae99SBarry Smith     /* Upper Level */
125147c6ae99SBarry Smith     for (k=0; k<s_z; k++) {
125247c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
125347c6ae99SBarry Smith         if (n18 >= 0) { /* left below */
1254ce00eea3SSatish Balay           x_t = lx[n18 % m];
125547c6ae99SBarry Smith           y_t = ly[(n18 % (m*n))/m];
125647c6ae99SBarry Smith           /* z_t = lz[n18 / (m*n)]; */
125747c6ae99SBarry Smith           s_t = bases[n18] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
12588865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1259ce00eea3SSatish Balay         } else if (Xs-xs < 0 && Ys-ys < 0 && ze-Ze < 0) {
12608865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
126147c6ae99SBarry Smith         }
126247c6ae99SBarry Smith         if (n19 >= 0) { /* directly below */
126347c6ae99SBarry Smith           x_t = x;
126447c6ae99SBarry Smith           y_t = ly[(n19 % (m*n))/m];
126547c6ae99SBarry Smith           /* z_t = lz[n19 / (m*n)]; */
126647c6ae99SBarry Smith           s_t = bases[n19] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
12678865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1268ce00eea3SSatish Balay         } else if (Ys-ys < 0 && ze-Ze < 0) {
12698865f1eaSKarl Rupp           for (j=0; j<x; j++) idx[nn++] = -1;
127047c6ae99SBarry Smith         }
127147c6ae99SBarry Smith         if (n20 >= 0) { /* right below */
1272ce00eea3SSatish Balay           x_t = lx[n20 % m];
127347c6ae99SBarry Smith           y_t = ly[(n20 % (m*n))/m];
127447c6ae99SBarry Smith           /* z_t = lz[n20 / (m*n)]; */
127547c6ae99SBarry Smith           s_t = bases[n20] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
12768865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1277ce00eea3SSatish Balay         } else if (xe-Xe < 0 && Ys-ys < 0 && ze-Ze < 0) {
12788865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
127947c6ae99SBarry Smith         }
128047c6ae99SBarry Smith       }
128147c6ae99SBarry Smith 
128247c6ae99SBarry Smith       for (i=0; i<y; i++) {
128347c6ae99SBarry Smith         if (n21 >= 0) { /* directly left */
1284ce00eea3SSatish Balay           x_t = lx[n21 % m];
128547c6ae99SBarry Smith           y_t = y;
128647c6ae99SBarry Smith           /* z_t = lz[n21 / (m*n)]; */
128747c6ae99SBarry Smith           s_t = bases[n21] + (i+1)*x_t - s_x + k*x_t*y_t;
12888865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1289ce00eea3SSatish Balay         } else if (Xs-xs < 0 && ze-Ze < 0) {
12908865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
129147c6ae99SBarry Smith         }
129247c6ae99SBarry Smith 
129347c6ae99SBarry Smith         if (n22 >= 0) { /* middle */
129447c6ae99SBarry Smith           x_t = x;
129547c6ae99SBarry Smith           y_t = y;
129647c6ae99SBarry Smith           /* z_t = lz[n22 / (m*n)]; */
129747c6ae99SBarry Smith           s_t = bases[n22] + i*x_t + k*x_t*y_t;
12988865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1299ce00eea3SSatish Balay         } else if (ze-Ze < 0) {
1300bff4a2f0SMatthew G. Knepley           if (bz == DM_BOUNDARY_MIRROR) {
1301a6fd6b0aSBarry Smith             for (j=0; j<x; j++) idx[nn++] = bases[rank] + j + (z-k-1)*x*y + i*x;
1302c2859e5eSBarry Smith           } else {
13038865f1eaSKarl Rupp             for (j=0; j<x; j++) idx[nn++] = -1;
130447c6ae99SBarry Smith           }
1305c2859e5eSBarry Smith         }
130647c6ae99SBarry Smith 
130747c6ae99SBarry Smith         if (n23 >= 0) { /* directly right */
1308ce00eea3SSatish Balay           x_t = lx[n23 % m];
130947c6ae99SBarry Smith           y_t = y;
131047c6ae99SBarry Smith           /* z_t = lz[n23 / (m*n)]; */
131147c6ae99SBarry Smith           s_t = bases[n23] + i*x_t + k*x_t*y_t;
13128865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1313ce00eea3SSatish Balay         } else if (xe-Xe < 0 && ze-Ze < 0) {
13148865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
131547c6ae99SBarry Smith         }
131647c6ae99SBarry Smith       }
131747c6ae99SBarry Smith 
131847c6ae99SBarry Smith       for (i=1; i<=s_y; i++) {
131947c6ae99SBarry Smith         if (n24 >= 0) { /* left above */
1320ce00eea3SSatish Balay           x_t = lx[n24 % m];
132147c6ae99SBarry Smith           y_t = ly[(n24 % (m*n))/m];
132247c6ae99SBarry Smith           /* z_t = lz[n24 / (m*n)]; */
132347c6ae99SBarry Smith           s_t = bases[n24] + i*x_t - s_x + k*x_t*y_t;
13248865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1325ce00eea3SSatish Balay         } else if (Xs-xs < 0 && ye-Ye < 0 && ze-Ze < 0) {
13268865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
132747c6ae99SBarry Smith         }
132847c6ae99SBarry Smith         if (n25 >= 0) { /* directly above */
132947c6ae99SBarry Smith           x_t = x;
133047c6ae99SBarry Smith           y_t = ly[(n25 % (m*n))/m];
133147c6ae99SBarry Smith           /* z_t = lz[n25 / (m*n)]; */
133247c6ae99SBarry Smith           s_t = bases[n25] + (i-1)*x_t + k*x_t*y_t;
13338865f1eaSKarl Rupp           for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1334ce00eea3SSatish Balay         } else if (ye-Ye < 0 && ze-Ze < 0) {
13358865f1eaSKarl Rupp           for (j=0; j<x; j++) idx[nn++] = -1;
133647c6ae99SBarry Smith         }
133747c6ae99SBarry Smith         if (n26 >= 0) { /* right above */
1338ce00eea3SSatish Balay           x_t = lx[n26 % m];
133947c6ae99SBarry Smith           y_t = ly[(n26 % (m*n))/m];
134047c6ae99SBarry Smith           /* z_t = lz[n26 / (m*n)]; */
134147c6ae99SBarry Smith           s_t = bases[n26] + (i-1)*x_t + k*x_t*y_t;
13428865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1343ce00eea3SSatish Balay         } else if (xe-Xe < 0 && ye-Ye < 0 && ze-Ze < 0) {
13448865f1eaSKarl Rupp           for (j=0; j<s_x; j++) idx[nn++] = -1;
134547c6ae99SBarry Smith         }
134647c6ae99SBarry Smith       }
134747c6ae99SBarry Smith     }
134847c6ae99SBarry Smith   }
134947c6ae99SBarry Smith   /*
135047c6ae99SBarry Smith      Set the local to global ordering in the global vector, this allows use
135147c6ae99SBarry Smith      of VecSetValuesLocal().
135247c6ae99SBarry Smith   */
135345b6f7e9SBarry Smith   ierr = ISLocalToGlobalMappingCreate(comm,dof,nn,idx,PETSC_OWN_POINTER,&da->ltogmap);CHKERRQ(ierr);
13543bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)da,(PetscObject)da->ltogmap);CHKERRQ(ierr);
135547c6ae99SBarry Smith 
1356ce00eea3SSatish Balay   ierr  = PetscFree2(bases,ldims);CHKERRQ(ierr);
1357ce00eea3SSatish Balay   dd->m = m;  dd->n  = n;  dd->p  = p;
1358ce00eea3SSatish Balay   /* note petsc expects xs/xe/Xs/Xe to be multiplied by #dofs in many places */
1359ce00eea3SSatish Balay   dd->xs = xs*dof; dd->xe = xe*dof; dd->ys = ys; dd->ye = ye; dd->zs = zs; dd->ze = ze;
1360ce00eea3SSatish Balay   dd->Xs = Xs*dof; dd->Xe = Xe*dof; dd->Ys = Ys; dd->Ye = Ye; dd->Zs = Zs; dd->Ze = Ze;
1361ce00eea3SSatish Balay 
1362fcfd50ebSBarry Smith   ierr = VecDestroy(&local);CHKERRQ(ierr);
1363fcfd50ebSBarry Smith   ierr = VecDestroy(&global);CHKERRQ(ierr);
1364ce00eea3SSatish Balay 
1365ce00eea3SSatish Balay   dd->gtol      = gtol;
1366ce00eea3SSatish Balay   dd->base      = base;
1367ce00eea3SSatish Balay   da->ops->view = DMView_DA_3d;
13680298fd71SBarry Smith   dd->ltol      = NULL;
13690298fd71SBarry Smith   dd->ao        = NULL;
137047c6ae99SBarry Smith   PetscFunctionReturn(0);
137147c6ae99SBarry Smith }
1372564755cdSBarry Smith 
137347c6ae99SBarry Smith /*@C
1374aa219208SBarry Smith    DMDACreate3d - Creates an object that will manage the communication of three-dimensional
137547c6ae99SBarry Smith    regular array data that is distributed across some processors.
137647c6ae99SBarry Smith 
1377d083f849SBarry Smith    Collective
137847c6ae99SBarry Smith 
137947c6ae99SBarry Smith    Input Parameters:
138047c6ae99SBarry Smith +  comm - MPI communicator
13811321219cSEthan Coon .  bx,by,bz - type of ghost nodes the array have.
1382bff4a2f0SMatthew G. Knepley          Use one of DM_BOUNDARY_NONE, DM_BOUNDARY_GHOSTED, DM_BOUNDARY_PERIODIC.
1383aa219208SBarry Smith .  stencil_type - Type of stencil (DMDA_STENCIL_STAR or DMDA_STENCIL_BOX)
1384897f7067SBarry Smith .  M,N,P - global dimension in each direction of the array
138547c6ae99SBarry Smith .  m,n,p - corresponding number of processors in each dimension
138647c6ae99SBarry Smith            (or PETSC_DECIDE to have calculated)
138747c6ae99SBarry Smith .  dof - number of degrees of freedom per node
138810d7c030SMatthew G Knepley .  s - stencil width
138910d7c030SMatthew G Knepley -  lx, ly, lz - arrays containing the number of nodes in each cell along
13900298fd71SBarry Smith           the x, y, and z coordinates, or NULL. If non-null, these
139147c6ae99SBarry Smith           must be of length as m,n,p and the corresponding
139247c6ae99SBarry Smith           m,n, or p cannot be PETSC_DECIDE. Sum of the lx[] entries must be M, sum of
139347c6ae99SBarry Smith           the ly[] must N, sum of the lz[] must be P
139447c6ae99SBarry Smith 
139547c6ae99SBarry Smith    Output Parameter:
139647c6ae99SBarry Smith .  da - the resulting distributed array object
139747c6ae99SBarry Smith 
139847c6ae99SBarry Smith    Options Database Key:
1399706a11cbSBarry Smith +  -dm_view - Calls DMView() at the conclusion of DMDACreate3d()
1400e43dc8daSBarry Smith .  -da_grid_x <nx> - number of grid points in x direction
1401e43dc8daSBarry Smith .  -da_grid_y <ny> - number of grid points in y direction
1402e43dc8daSBarry Smith .  -da_grid_z <nz> - number of grid points in z direction
140347c6ae99SBarry Smith .  -da_processors_x <MX> - number of processors in x direction
140447c6ae99SBarry Smith .  -da_processors_y <MY> - number of processors in y direction
140547c6ae99SBarry Smith .  -da_processors_z <MZ> - number of processors in z direction
1406e0f5d30fSBarry Smith .  -da_refine_x <rx> - refinement ratio in x direction
1407e0f5d30fSBarry Smith .  -da_refine_y <ry> - refinement ratio in y direction
1408e0f5d30fSBarry Smith .  -da_refine_z <rz>- refinement ratio in z directio
1409e43dc8daSBarry Smith -  -da_refine <n> - refine the DMDA n times before creating it
141047c6ae99SBarry Smith 
141147c6ae99SBarry Smith    Level: beginner
141247c6ae99SBarry Smith 
141347c6ae99SBarry Smith    Notes:
1414aa219208SBarry Smith    The stencil type DMDA_STENCIL_STAR with width 1 corresponds to the
1415aa219208SBarry Smith    standard 7-pt stencil, while DMDA_STENCIL_BOX with width 1 denotes
141647c6ae99SBarry Smith    the standard 27-pt stencil.
141747c6ae99SBarry Smith 
1418aa219208SBarry Smith    The array data itself is NOT stored in the DMDA, it is stored in Vec objects;
1419564755cdSBarry Smith    The appropriate vector objects can be obtained with calls to DMCreateGlobalVector()
1420564755cdSBarry Smith    and DMCreateLocalVector() and calls to VecDuplicate() if more are needed.
142147c6ae99SBarry Smith 
1422897f7067SBarry Smith    You must call DMSetUp() after this call before using this DM.
1423897f7067SBarry Smith 
1424897f7067SBarry Smith    If you wish to use the options database to change values in the DMDA call DMSetFromOptions() after this call
1425897f7067SBarry Smith    but before DMSetUp().
1426897f7067SBarry Smith 
1427aa219208SBarry Smith .seealso: DMDestroy(), DMView(), DMDACreate1d(), DMDACreate2d(), DMGlobalToLocalBegin(), DMDAGetRefinementFactor(),
142899f0b487SRichard Tran Mills           DMGlobalToLocalEnd(), DMLocalToGlobalBegin(), DMLocalToLocalBegin(), DMLocalToLocalEnd(), DMDASetRefinementFactor(),
14293bd220d7SPatrick Sanan           DMDAGetInfo(), DMCreateGlobalVector(), DMCreateLocalVector(), DMDACreateNaturalVector(), DMLoad(), DMDAGetOwnershipRanges(),
14303bd220d7SPatrick Sanan           DMStagCreate3d()
143147c6ae99SBarry Smith 
143247c6ae99SBarry Smith @*/
1433bff4a2f0SMatthew G. Knepley PetscErrorCode  DMDACreate3d(MPI_Comm comm,DMBoundaryType bx,DMBoundaryType by,DMBoundaryType bz,DMDAStencilType stencil_type,PetscInt M,
14349a42bb27SBarry 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)
143547c6ae99SBarry Smith {
143647c6ae99SBarry Smith   PetscErrorCode ierr;
143747c6ae99SBarry Smith 
143847c6ae99SBarry Smith   PetscFunctionBegin;
1439aa219208SBarry Smith   ierr = DMDACreate(comm, da);CHKERRQ(ierr);
1440c73cfb54SMatthew G. Knepley   ierr = DMSetDimension(*da, 3);CHKERRQ(ierr);
1441aa219208SBarry Smith   ierr = DMDASetSizes(*da, M, N, P);CHKERRQ(ierr);
1442aa219208SBarry Smith   ierr = DMDASetNumProcs(*da, m, n, p);CHKERRQ(ierr);
14431321219cSEthan Coon   ierr = DMDASetBoundaryType(*da, bx, by, bz);CHKERRQ(ierr);
1444aa219208SBarry Smith   ierr = DMDASetDof(*da, dof);CHKERRQ(ierr);
1445aa219208SBarry Smith   ierr = DMDASetStencilType(*da, stencil_type);CHKERRQ(ierr);
1446aa219208SBarry Smith   ierr = DMDASetStencilWidth(*da, s);CHKERRQ(ierr);
1447aa219208SBarry Smith   ierr = DMDASetOwnershipRanges(*da, lx, ly, lz);CHKERRQ(ierr);
144847c6ae99SBarry Smith   PetscFunctionReturn(0);
144947c6ae99SBarry Smith }
1450