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; 21ce94432eSBarry Smith ierr = MPI_Comm_rank(PetscObjectComm((PetscObject)da),&rank);CHKERRQ(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; 3976a8abe0SBarry Smith ierr = MPI_Comm_size(PetscObjectComm((PetscObject)da),&size);CHKERRQ(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); 4376a8abe0SBarry Smith ierr = MPI_Allgather(&nzlocal,1,MPIU_INT,nz,1,MPIU_INT,PetscObjectComm((PetscObject)da));CHKERRQ(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 } 548135c375SStefano Zampini if (format != PETSC_VIEWER_ASCII_VTK && format != PETSC_VIEWER_ASCII_VTK_CELL && 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 */ 9747c6ae99SBarry Smith if (!rank) { 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 2316f951b95Secoon 23247c6ae99SBarry Smith PetscFunctionBegin; 233bff4a2f0SMatthew 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"); 23440088605SBarry Smith if ((bx == DM_BOUNDARY_MIRROR) || (by == DM_BOUNDARY_MIRROR) || (bz == DM_BOUNDARY_MIRROR)) SETERRQ(PetscObjectComm((PetscObject)da),PETSC_ERR_SUP,"Mirror boundary not supported yet in 3d"); 23547c6ae99SBarry Smith ierr = PetscObjectGetComm((PetscObject) da, &comm);CHKERRQ(ierr); 2363855c12bSBarry Smith #if !defined(PETSC_USE_64BIT_INDICES) 237bafee8b4SSatish Balay if (((PetscInt64) M)*((PetscInt64) N)*((PetscInt64) P)*((PetscInt64) dof) > (PetscInt64) 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); 2383855c12bSBarry Smith #endif 2393855c12bSBarry Smith 24047c6ae99SBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 24147c6ae99SBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 24247c6ae99SBarry Smith 24347c6ae99SBarry Smith if (m != PETSC_DECIDE) { 24447c6ae99SBarry Smith if (m < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in X direction: %D",m); 24547c6ae99SBarry Smith else if (m > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in X direction: %D %d",m,size); 24647c6ae99SBarry Smith } 24747c6ae99SBarry Smith if (n != PETSC_DECIDE) { 24847c6ae99SBarry Smith if (n < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in Y direction: %D",n); 24947c6ae99SBarry Smith else if (n > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in Y direction: %D %d",n,size); 25047c6ae99SBarry Smith } 25147c6ae99SBarry Smith if (p != PETSC_DECIDE) { 25247c6ae99SBarry Smith if (p < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in Z direction: %D",p); 25347c6ae99SBarry Smith else if (p > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in Z direction: %D %d",p,size); 25447c6ae99SBarry Smith } 2550716a85fSBarry 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); 25647c6ae99SBarry Smith 25747c6ae99SBarry Smith /* Partition the array among the processors */ 25847c6ae99SBarry Smith if (m == PETSC_DECIDE && n != PETSC_DECIDE && p != PETSC_DECIDE) { 25947c6ae99SBarry Smith m = size/(n*p); 26047c6ae99SBarry Smith } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) { 26147c6ae99SBarry Smith n = size/(m*p); 26247c6ae99SBarry Smith } else if (m != PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) { 26347c6ae99SBarry Smith p = size/(m*n); 26447c6ae99SBarry Smith } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) { 26547c6ae99SBarry Smith /* try for squarish distribution */ 266369cc0aeSBarry Smith m = (int)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)N*p))); 26747c6ae99SBarry Smith if (!m) m = 1; 26847c6ae99SBarry Smith while (m > 0) { 26947c6ae99SBarry Smith n = size/(m*p); 27047c6ae99SBarry Smith if (m*n*p == size) break; 27147c6ae99SBarry Smith m--; 27247c6ae99SBarry Smith } 27347c6ae99SBarry Smith if (!m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad p value: p = %D",p); 27447c6ae99SBarry Smith if (M > N && m < n) {PetscInt _m = m; m = n; n = _m;} 27547c6ae99SBarry Smith } else if (m == PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) { 27647c6ae99SBarry Smith /* try for squarish distribution */ 277369cc0aeSBarry Smith m = (int)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)P*n))); 27847c6ae99SBarry Smith if (!m) m = 1; 27947c6ae99SBarry Smith while (m > 0) { 28047c6ae99SBarry Smith p = size/(m*n); 28147c6ae99SBarry Smith if (m*n*p == size) break; 28247c6ae99SBarry Smith m--; 28347c6ae99SBarry Smith } 28447c6ae99SBarry Smith if (!m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad n value: n = %D",n); 28547c6ae99SBarry Smith if (M > P && m < p) {PetscInt _m = m; m = p; p = _m;} 28647c6ae99SBarry Smith } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) { 28747c6ae99SBarry Smith /* try for squarish distribution */ 288369cc0aeSBarry Smith n = (int)(0.5 + PetscSqrtReal(((PetscReal)N)*((PetscReal)size)/((PetscReal)P*m))); 28947c6ae99SBarry Smith if (!n) n = 1; 29047c6ae99SBarry Smith while (n > 0) { 29147c6ae99SBarry Smith p = size/(m*n); 29247c6ae99SBarry Smith if (m*n*p == size) break; 29347c6ae99SBarry Smith n--; 29447c6ae99SBarry Smith } 29547c6ae99SBarry Smith if (!n) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad m value: m = %D",n); 29647c6ae99SBarry Smith if (N > P && n < p) {PetscInt _n = n; n = p; p = _n;} 29747c6ae99SBarry Smith } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) { 29847c6ae99SBarry Smith /* try for squarish distribution */ 299369cc0aeSBarry Smith n = (PetscInt)(0.5 + PetscPowReal(((PetscReal)N*N)*((PetscReal)size)/((PetscReal)P*M),(PetscReal)(1./3.))); 30047c6ae99SBarry Smith if (!n) n = 1; 30147c6ae99SBarry Smith while (n > 0) { 30247c6ae99SBarry Smith pm = size/n; 30347c6ae99SBarry Smith if (n*pm == size) break; 30447c6ae99SBarry Smith n--; 30547c6ae99SBarry Smith } 30647c6ae99SBarry Smith if (!n) n = 1; 307369cc0aeSBarry Smith m = (PetscInt)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)P*n))); 30847c6ae99SBarry Smith if (!m) m = 1; 30947c6ae99SBarry Smith while (m > 0) { 31047c6ae99SBarry Smith p = size/(m*n); 31147c6ae99SBarry Smith if (m*n*p == size) break; 31247c6ae99SBarry Smith m--; 31347c6ae99SBarry Smith } 31447c6ae99SBarry Smith if (M > P && m < p) {PetscInt _m = m; m = p; p = _m;} 315ce94432eSBarry Smith } else if (m*n*p != size) SETERRQ(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Given Bad partition"); 31647c6ae99SBarry Smith 317ce94432eSBarry Smith if (m*n*p != size) SETERRQ(PetscObjectComm((PetscObject)da),PETSC_ERR_PLIB,"Could not find good partition"); 318ce94432eSBarry Smith if (M < m) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in x direction is too fine! %D %D",M,m); 319ce94432eSBarry Smith if (N < n) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in y direction is too fine! %D %D",N,n); 320ce94432eSBarry Smith if (P < p) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in z direction is too fine! %D %D",P,p); 32147c6ae99SBarry Smith 32247c6ae99SBarry Smith /* 32347c6ae99SBarry Smith Determine locally owned region 32447c6ae99SBarry Smith [x, y, or z]s is the first local node number, [x, y, z] is the number of local nodes 32547c6ae99SBarry Smith */ 32647c6ae99SBarry Smith 32747c6ae99SBarry Smith if (!lx) { 328785e854fSJed Brown ierr = PetscMalloc1(m, &dd->lx);CHKERRQ(ierr); 32947c6ae99SBarry Smith lx = dd->lx; 3308865f1eaSKarl Rupp for (i=0; i<m; i++) lx[i] = M/m + ((M % m) > (i % m)); 33147c6ae99SBarry Smith } 33247c6ae99SBarry Smith x = lx[rank % m]; 33347c6ae99SBarry Smith xs = 0; 3348865f1eaSKarl Rupp for (i=0; i<(rank%m); i++) xs += lx[i]; 335bff4a2f0SMatthew 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); 33647c6ae99SBarry Smith 33747c6ae99SBarry Smith if (!ly) { 338785e854fSJed Brown ierr = PetscMalloc1(n, &dd->ly);CHKERRQ(ierr); 33947c6ae99SBarry Smith ly = dd->ly; 3408865f1eaSKarl Rupp for (i=0; i<n; i++) ly[i] = N/n + ((N % n) > (i % n)); 34147c6ae99SBarry Smith } 34247c6ae99SBarry Smith y = ly[(rank % (m*n))/m]; 343bff4a2f0SMatthew 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); 34430729d88SBarry Smith 34547c6ae99SBarry Smith ys = 0; 3468865f1eaSKarl Rupp for (i=0; i<(rank % (m*n))/m; i++) ys += ly[i]; 34747c6ae99SBarry Smith 34847c6ae99SBarry Smith if (!lz) { 349785e854fSJed Brown ierr = PetscMalloc1(p, &dd->lz);CHKERRQ(ierr); 35047c6ae99SBarry Smith lz = dd->lz; 3518865f1eaSKarl Rupp for (i=0; i<p; i++) lz[i] = P/p + ((P % p) > (i % p)); 35247c6ae99SBarry Smith } 35347c6ae99SBarry Smith z = lz[rank/(m*n)]; 354bcea557cSEthan Coon 355fdc81ce8SEthan Coon /* note this is different than x- and y-, as we will handle as an important special 356bff4a2f0SMatthew G. Knepley case when p=P=1 and DM_BOUNDARY_PERIODIC and s > z. This is to deal with 2D problems 357fdc81ce8SEthan Coon in a 3D code. Additional code for this case is noted with "2d case" comments */ 3586f951b95Secoon twod = PETSC_FALSE; 3598865f1eaSKarl Rupp if (P == 1) twod = PETSC_TRUE; 360bff4a2f0SMatthew 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); 36147c6ae99SBarry Smith zs = 0; 3628865f1eaSKarl Rupp for (i=0; i<(rank/(m*n)); i++) zs += lz[i]; 36347c6ae99SBarry Smith ye = ys + y; 36447c6ae99SBarry Smith xe = xs + x; 36547c6ae99SBarry Smith ze = zs + z; 36647c6ae99SBarry Smith 36788661749SPeter Brune /* determine ghost region (Xs) and region scattered into (IXs) */ 368d9c9ebe5SPeter Brune if (xs-s > 0) { 369d9c9ebe5SPeter Brune Xs = xs - s; IXs = xs - s; 37088661749SPeter Brune } else { 3718865f1eaSKarl Rupp if (bx) Xs = xs - s; 3728865f1eaSKarl Rupp else Xs = 0; 37388661749SPeter Brune IXs = 0; 37488661749SPeter Brune } 375d9c9ebe5SPeter Brune if (xe+s <= M) { 376d9c9ebe5SPeter Brune Xe = xe + s; IXe = xe + s; 37788661749SPeter Brune } else { 37888661749SPeter Brune if (bx) { 379d9c9ebe5SPeter Brune Xs = xs - s; Xe = xe + s; 3808865f1eaSKarl Rupp } else Xe = M; 38188661749SPeter Brune IXe = M; 38288661749SPeter Brune } 38347c6ae99SBarry Smith 384bff4a2f0SMatthew G. Knepley if (bx == DM_BOUNDARY_PERIODIC || bx == DM_BOUNDARY_MIRROR) { 385d9c9ebe5SPeter Brune IXs = xs - s; 386d9c9ebe5SPeter Brune IXe = xe + s; 387d9c9ebe5SPeter Brune Xs = xs - s; 388d9c9ebe5SPeter Brune Xe = xe + s; 38988661749SPeter Brune } 39088661749SPeter Brune 391d9c9ebe5SPeter Brune if (ys-s > 0) { 392d9c9ebe5SPeter Brune Ys = ys - s; IYs = ys - s; 39388661749SPeter Brune } else { 3948865f1eaSKarl Rupp if (by) Ys = ys - s; 3958865f1eaSKarl Rupp else Ys = 0; 39688661749SPeter Brune IYs = 0; 39788661749SPeter Brune } 398d9c9ebe5SPeter Brune if (ye+s <= N) { 399d9c9ebe5SPeter Brune Ye = ye + s; IYe = ye + s; 40088661749SPeter Brune } else { 4018865f1eaSKarl Rupp if (by) Ye = ye + s; 4028865f1eaSKarl Rupp else Ye = N; 40388661749SPeter Brune IYe = N; 40488661749SPeter Brune } 40588661749SPeter Brune 406bff4a2f0SMatthew G. Knepley if (by == DM_BOUNDARY_PERIODIC || by == DM_BOUNDARY_MIRROR) { 407d9c9ebe5SPeter Brune IYs = ys - s; 408d9c9ebe5SPeter Brune IYe = ye + s; 409d9c9ebe5SPeter Brune Ys = ys - s; 410d9c9ebe5SPeter Brune Ye = ye + s; 41188661749SPeter Brune } 41288661749SPeter Brune 413d9c9ebe5SPeter Brune if (zs-s > 0) { 414d9c9ebe5SPeter Brune Zs = zs - s; IZs = zs - s; 41588661749SPeter Brune } else { 4168865f1eaSKarl Rupp if (bz) Zs = zs - s; 4178865f1eaSKarl Rupp else Zs = 0; 41888661749SPeter Brune IZs = 0; 41988661749SPeter Brune } 420d9c9ebe5SPeter Brune if (ze+s <= P) { 421d9c9ebe5SPeter Brune Ze = ze + s; IZe = ze + s; 42288661749SPeter Brune } else { 4238865f1eaSKarl Rupp if (bz) Ze = ze + s; 4248865f1eaSKarl Rupp else Ze = P; 42588661749SPeter Brune IZe = P; 42688661749SPeter Brune } 42788661749SPeter Brune 428bff4a2f0SMatthew G. Knepley if (bz == DM_BOUNDARY_PERIODIC || bz == DM_BOUNDARY_MIRROR) { 429d9c9ebe5SPeter Brune IZs = zs - s; 430d9c9ebe5SPeter Brune IZe = ze + s; 431d9c9ebe5SPeter Brune Zs = zs - s; 432d9c9ebe5SPeter Brune Ze = ze + s; 43388661749SPeter Brune } 43447c6ae99SBarry Smith 43547c6ae99SBarry Smith /* Resize all X parameters to reflect w */ 436d9c9ebe5SPeter Brune s_x = s; 437d9c9ebe5SPeter Brune s_y = s; 438d9c9ebe5SPeter Brune s_z = s; 43947c6ae99SBarry Smith 44047c6ae99SBarry Smith /* determine starting point of each processor */ 44147c6ae99SBarry Smith nn = x*y*z; 442dcca6d9dSJed Brown ierr = PetscMalloc2(size+1,&bases,size,&ldims);CHKERRQ(ierr); 44347c6ae99SBarry Smith ierr = MPI_Allgather(&nn,1,MPIU_INT,ldims,1,MPIU_INT,comm);CHKERRQ(ierr); 44447c6ae99SBarry Smith bases[0] = 0; 4458865f1eaSKarl Rupp for (i=1; i<=size; i++) bases[i] = ldims[i-1]; 4468865f1eaSKarl Rupp for (i=1; i<=size; i++) bases[i] += bases[i-1]; 447ce00eea3SSatish Balay base = bases[rank]*dof; 44847c6ae99SBarry Smith 44947c6ae99SBarry Smith /* allocate the base parallel and sequential vectors */ 450ce00eea3SSatish Balay dd->Nlocal = x*y*z*dof; 451b1fb7eb7SBarry Smith ierr = VecCreateMPIWithArray(comm,dof,dd->Nlocal,PETSC_DECIDE,NULL,&global);CHKERRQ(ierr); 452ce00eea3SSatish Balay dd->nlocal = (Xe-Xs)*(Ye-Ys)*(Ze-Zs)*dof; 453b1fb7eb7SBarry Smith ierr = VecCreateSeqWithArray(PETSC_COMM_SELF,dof,dd->nlocal,NULL,&local);CHKERRQ(ierr); 45447c6ae99SBarry Smith 4554104a7a0SPatrick Sanan /* generate global to local vector scatter and local to global mapping*/ 45647c6ae99SBarry Smith 457ce00eea3SSatish Balay /* global to local must include ghost points within the domain, 458ce00eea3SSatish Balay but not ghost points outside the domain that aren't periodic */ 4594104a7a0SPatrick Sanan ierr = PetscMalloc1((IXe-IXs)*(IYe-IYs)*(IZe-IZs),&idx);CHKERRQ(ierr); 460d9c9ebe5SPeter Brune if (stencil_type == DMDA_STENCIL_BOX) { 461ce00eea3SSatish Balay left = IXs - Xs; right = left + (IXe-IXs); 462ce00eea3SSatish Balay bottom = IYs - Ys; top = bottom + (IYe-IYs); 463ce00eea3SSatish Balay down = IZs - Zs; up = down + (IZe-IZs); 464ce00eea3SSatish Balay count = 0; 465ce00eea3SSatish Balay for (i=down; i<up; i++) { 466ce00eea3SSatish Balay for (j=bottom; j<top; j++) { 467ce00eea3SSatish Balay for (k=left; k<right; k++) { 468ce00eea3SSatish Balay idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 469ce00eea3SSatish Balay } 470ce00eea3SSatish Balay } 471ce00eea3SSatish Balay } 472ce00eea3SSatish Balay ierr = ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&to);CHKERRQ(ierr); 47347c6ae99SBarry Smith } else { 47447c6ae99SBarry Smith /* This is way ugly! We need to list the funny cross type region */ 475ce00eea3SSatish Balay left = xs - Xs; right = left + x; 47647c6ae99SBarry Smith bottom = ys - Ys; top = bottom + y; 47747c6ae99SBarry Smith down = zs - Zs; up = down + z; 47847c6ae99SBarry Smith count = 0; 479ce00eea3SSatish Balay /* the bottom chunck */ 480ce00eea3SSatish Balay for (i=(IZs-Zs); i<down; i++) { 48147c6ae99SBarry Smith for (j=bottom; j<top; j++) { 482ce00eea3SSatish Balay for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 48347c6ae99SBarry Smith } 48447c6ae99SBarry Smith } 48547c6ae99SBarry Smith /* the middle piece */ 48647c6ae99SBarry Smith for (i=down; i<up; i++) { 48747c6ae99SBarry Smith /* front */ 488ce00eea3SSatish Balay for (j=(IYs-Ys); j<bottom; j++) { 489ce00eea3SSatish Balay for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 49047c6ae99SBarry Smith } 49147c6ae99SBarry Smith /* middle */ 49247c6ae99SBarry Smith for (j=bottom; j<top; j++) { 493ce00eea3SSatish Balay for (k=IXs-Xs; k<IXe-Xs; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 49447c6ae99SBarry Smith } 49547c6ae99SBarry Smith /* back */ 496ce00eea3SSatish Balay for (j=top; j<top+IYe-ye; j++) { 497ce00eea3SSatish Balay for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 49847c6ae99SBarry Smith } 49947c6ae99SBarry Smith } 50047c6ae99SBarry Smith /* the top piece */ 501ce00eea3SSatish Balay for (i=up; i<up+IZe-ze; i++) { 50247c6ae99SBarry Smith for (j=bottom; j<top; j++) { 503ce00eea3SSatish Balay for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k; 50447c6ae99SBarry Smith } 50547c6ae99SBarry Smith } 50647c6ae99SBarry Smith ierr = ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&to);CHKERRQ(ierr); 50747c6ae99SBarry Smith } 50847c6ae99SBarry Smith 50947c6ae99SBarry Smith /* determine who lies on each side of use stored in n24 n25 n26 51047c6ae99SBarry Smith n21 n22 n23 51147c6ae99SBarry Smith n18 n19 n20 51247c6ae99SBarry Smith 51347c6ae99SBarry Smith n15 n16 n17 51447c6ae99SBarry Smith n12 n14 51547c6ae99SBarry Smith n9 n10 n11 51647c6ae99SBarry Smith 51747c6ae99SBarry Smith n6 n7 n8 51847c6ae99SBarry Smith n3 n4 n5 51947c6ae99SBarry Smith n0 n1 n2 52047c6ae99SBarry Smith */ 52147c6ae99SBarry Smith 52247c6ae99SBarry Smith /* Solve for X,Y, and Z Periodic Case First, Then Modify Solution */ 52347c6ae99SBarry Smith /* Assume Nodes are Internal to the Cube */ 52447c6ae99SBarry Smith n0 = rank - m*n - m - 1; 52547c6ae99SBarry Smith n1 = rank - m*n - m; 52647c6ae99SBarry Smith n2 = rank - m*n - m + 1; 52747c6ae99SBarry Smith n3 = rank - m*n -1; 52847c6ae99SBarry Smith n4 = rank - m*n; 52947c6ae99SBarry Smith n5 = rank - m*n + 1; 53047c6ae99SBarry Smith n6 = rank - m*n + m - 1; 53147c6ae99SBarry Smith n7 = rank - m*n + m; 53247c6ae99SBarry Smith n8 = rank - m*n + m + 1; 53347c6ae99SBarry Smith 53447c6ae99SBarry Smith n9 = rank - m - 1; 53547c6ae99SBarry Smith n10 = rank - m; 53647c6ae99SBarry Smith n11 = rank - m + 1; 53747c6ae99SBarry Smith n12 = rank - 1; 53847c6ae99SBarry Smith n14 = rank + 1; 53947c6ae99SBarry Smith n15 = rank + m - 1; 54047c6ae99SBarry Smith n16 = rank + m; 54147c6ae99SBarry Smith n17 = rank + m + 1; 54247c6ae99SBarry Smith 54347c6ae99SBarry Smith n18 = rank + m*n - m - 1; 54447c6ae99SBarry Smith n19 = rank + m*n - m; 54547c6ae99SBarry Smith n20 = rank + m*n - m + 1; 54647c6ae99SBarry Smith n21 = rank + m*n - 1; 54747c6ae99SBarry Smith n22 = rank + m*n; 54847c6ae99SBarry Smith n23 = rank + m*n + 1; 54947c6ae99SBarry Smith n24 = rank + m*n + m - 1; 55047c6ae99SBarry Smith n25 = rank + m*n + m; 55147c6ae99SBarry Smith n26 = rank + m*n + m + 1; 55247c6ae99SBarry Smith 55347c6ae99SBarry Smith /* Assume Pieces are on Faces of Cube */ 55447c6ae99SBarry Smith 55547c6ae99SBarry Smith if (xs == 0) { /* First assume not corner or edge */ 55647c6ae99SBarry Smith n0 = rank -1 - (m*n); 55747c6ae99SBarry Smith n3 = rank + m -1 - (m*n); 55847c6ae99SBarry Smith n6 = rank + 2*m -1 - (m*n); 55947c6ae99SBarry Smith n9 = rank -1; 56047c6ae99SBarry Smith n12 = rank + m -1; 56147c6ae99SBarry Smith n15 = rank + 2*m -1; 56247c6ae99SBarry Smith n18 = rank -1 + (m*n); 56347c6ae99SBarry Smith n21 = rank + m -1 + (m*n); 56447c6ae99SBarry Smith n24 = rank + 2*m -1 + (m*n); 56547c6ae99SBarry Smith } 56647c6ae99SBarry Smith 567ce00eea3SSatish Balay if (xe == M) { /* First assume not corner or edge */ 56847c6ae99SBarry Smith n2 = rank -2*m +1 - (m*n); 56947c6ae99SBarry Smith n5 = rank - m +1 - (m*n); 57047c6ae99SBarry Smith n8 = rank +1 - (m*n); 57147c6ae99SBarry Smith n11 = rank -2*m +1; 57247c6ae99SBarry Smith n14 = rank - m +1; 57347c6ae99SBarry Smith n17 = rank +1; 57447c6ae99SBarry Smith n20 = rank -2*m +1 + (m*n); 57547c6ae99SBarry Smith n23 = rank - m +1 + (m*n); 57647c6ae99SBarry Smith n26 = rank +1 + (m*n); 57747c6ae99SBarry Smith } 57847c6ae99SBarry Smith 57947c6ae99SBarry Smith if (ys==0) { /* First assume not corner or edge */ 58047c6ae99SBarry Smith n0 = rank + m * (n-1) -1 - (m*n); 58147c6ae99SBarry Smith n1 = rank + m * (n-1) - (m*n); 58247c6ae99SBarry Smith n2 = rank + m * (n-1) +1 - (m*n); 58347c6ae99SBarry Smith n9 = rank + m * (n-1) -1; 58447c6ae99SBarry Smith n10 = rank + m * (n-1); 58547c6ae99SBarry Smith n11 = rank + m * (n-1) +1; 58647c6ae99SBarry Smith n18 = rank + m * (n-1) -1 + (m*n); 58747c6ae99SBarry Smith n19 = rank + m * (n-1) + (m*n); 58847c6ae99SBarry Smith n20 = rank + m * (n-1) +1 + (m*n); 58947c6ae99SBarry Smith } 59047c6ae99SBarry Smith 59147c6ae99SBarry Smith if (ye == N) { /* First assume not corner or edge */ 59247c6ae99SBarry Smith n6 = rank - m * (n-1) -1 - (m*n); 59347c6ae99SBarry Smith n7 = rank - m * (n-1) - (m*n); 59447c6ae99SBarry Smith n8 = rank - m * (n-1) +1 - (m*n); 59547c6ae99SBarry Smith n15 = rank - m * (n-1) -1; 59647c6ae99SBarry Smith n16 = rank - m * (n-1); 59747c6ae99SBarry Smith n17 = rank - m * (n-1) +1; 59847c6ae99SBarry Smith n24 = rank - m * (n-1) -1 + (m*n); 59947c6ae99SBarry Smith n25 = rank - m * (n-1) + (m*n); 60047c6ae99SBarry Smith n26 = rank - m * (n-1) +1 + (m*n); 60147c6ae99SBarry Smith } 60247c6ae99SBarry Smith 60347c6ae99SBarry Smith if (zs == 0) { /* First assume not corner or edge */ 60447c6ae99SBarry Smith n0 = size - (m*n) + rank - m - 1; 60547c6ae99SBarry Smith n1 = size - (m*n) + rank - m; 60647c6ae99SBarry Smith n2 = size - (m*n) + rank - m + 1; 60747c6ae99SBarry Smith n3 = size - (m*n) + rank - 1; 60847c6ae99SBarry Smith n4 = size - (m*n) + rank; 60947c6ae99SBarry Smith n5 = size - (m*n) + rank + 1; 61047c6ae99SBarry Smith n6 = size - (m*n) + rank + m - 1; 61147c6ae99SBarry Smith n7 = size - (m*n) + rank + m; 61247c6ae99SBarry Smith n8 = size - (m*n) + rank + m + 1; 61347c6ae99SBarry Smith } 61447c6ae99SBarry Smith 61547c6ae99SBarry Smith if (ze == P) { /* First assume not corner or edge */ 61647c6ae99SBarry Smith n18 = (m*n) - (size-rank) - m - 1; 61747c6ae99SBarry Smith n19 = (m*n) - (size-rank) - m; 61847c6ae99SBarry Smith n20 = (m*n) - (size-rank) - m + 1; 61947c6ae99SBarry Smith n21 = (m*n) - (size-rank) - 1; 62047c6ae99SBarry Smith n22 = (m*n) - (size-rank); 62147c6ae99SBarry Smith n23 = (m*n) - (size-rank) + 1; 62247c6ae99SBarry Smith n24 = (m*n) - (size-rank) + m - 1; 62347c6ae99SBarry Smith n25 = (m*n) - (size-rank) + m; 62447c6ae99SBarry Smith n26 = (m*n) - (size-rank) + m + 1; 62547c6ae99SBarry Smith } 62647c6ae99SBarry Smith 62747c6ae99SBarry Smith if ((xs==0) && (zs==0)) { /* Assume an edge, not corner */ 62847c6ae99SBarry Smith n0 = size - m*n + rank + m-1 - m; 62947c6ae99SBarry Smith n3 = size - m*n + rank + m-1; 63047c6ae99SBarry Smith n6 = size - m*n + rank + m-1 + m; 63147c6ae99SBarry Smith } 63247c6ae99SBarry Smith 63347c6ae99SBarry Smith if ((xs==0) && (ze==P)) { /* Assume an edge, not corner */ 63447c6ae99SBarry Smith n18 = m*n - (size - rank) + m-1 - m; 63547c6ae99SBarry Smith n21 = m*n - (size - rank) + m-1; 63647c6ae99SBarry Smith n24 = m*n - (size - rank) + m-1 + m; 63747c6ae99SBarry Smith } 63847c6ae99SBarry Smith 63947c6ae99SBarry Smith if ((xs==0) && (ys==0)) { /* Assume an edge, not corner */ 64047c6ae99SBarry Smith n0 = rank + m*n -1 - m*n; 64147c6ae99SBarry Smith n9 = rank + m*n -1; 64247c6ae99SBarry Smith n18 = rank + m*n -1 + m*n; 64347c6ae99SBarry Smith } 64447c6ae99SBarry Smith 64547c6ae99SBarry Smith if ((xs==0) && (ye==N)) { /* Assume an edge, not corner */ 64647c6ae99SBarry Smith n6 = rank - m*(n-1) + m-1 - m*n; 64747c6ae99SBarry Smith n15 = rank - m*(n-1) + m-1; 64847c6ae99SBarry Smith n24 = rank - m*(n-1) + m-1 + m*n; 64947c6ae99SBarry Smith } 65047c6ae99SBarry Smith 651ce00eea3SSatish Balay if ((xe==M) && (zs==0)) { /* Assume an edge, not corner */ 65247c6ae99SBarry Smith n2 = size - (m*n-rank) - (m-1) - m; 65347c6ae99SBarry Smith n5 = size - (m*n-rank) - (m-1); 65447c6ae99SBarry Smith n8 = size - (m*n-rank) - (m-1) + m; 65547c6ae99SBarry Smith } 65647c6ae99SBarry Smith 657ce00eea3SSatish Balay if ((xe==M) && (ze==P)) { /* Assume an edge, not corner */ 65847c6ae99SBarry Smith n20 = m*n - (size - rank) - (m-1) - m; 65947c6ae99SBarry Smith n23 = m*n - (size - rank) - (m-1); 66047c6ae99SBarry Smith n26 = m*n - (size - rank) - (m-1) + m; 66147c6ae99SBarry Smith } 66247c6ae99SBarry Smith 663ce00eea3SSatish Balay if ((xe==M) && (ys==0)) { /* Assume an edge, not corner */ 66447c6ae99SBarry Smith n2 = rank + m*(n-1) - (m-1) - m*n; 66547c6ae99SBarry Smith n11 = rank + m*(n-1) - (m-1); 66647c6ae99SBarry Smith n20 = rank + m*(n-1) - (m-1) + m*n; 66747c6ae99SBarry Smith } 66847c6ae99SBarry Smith 669ce00eea3SSatish Balay if ((xe==M) && (ye==N)) { /* Assume an edge, not corner */ 67047c6ae99SBarry Smith n8 = rank - m*n +1 - m*n; 67147c6ae99SBarry Smith n17 = rank - m*n +1; 67247c6ae99SBarry Smith n26 = rank - m*n +1 + m*n; 67347c6ae99SBarry Smith } 67447c6ae99SBarry Smith 67547c6ae99SBarry Smith if ((ys==0) && (zs==0)) { /* Assume an edge, not corner */ 67647c6ae99SBarry Smith n0 = size - m + rank -1; 67747c6ae99SBarry Smith n1 = size - m + rank; 67847c6ae99SBarry Smith n2 = size - m + rank +1; 67947c6ae99SBarry Smith } 68047c6ae99SBarry Smith 68147c6ae99SBarry Smith if ((ys==0) && (ze==P)) { /* Assume an edge, not corner */ 68247c6ae99SBarry Smith n18 = m*n - (size - rank) + m*(n-1) -1; 68347c6ae99SBarry Smith n19 = m*n - (size - rank) + m*(n-1); 68447c6ae99SBarry Smith n20 = m*n - (size - rank) + m*(n-1) +1; 68547c6ae99SBarry Smith } 68647c6ae99SBarry Smith 68747c6ae99SBarry Smith if ((ye==N) && (zs==0)) { /* Assume an edge, not corner */ 68847c6ae99SBarry Smith n6 = size - (m*n-rank) - m * (n-1) -1; 68947c6ae99SBarry Smith n7 = size - (m*n-rank) - m * (n-1); 69047c6ae99SBarry Smith n8 = size - (m*n-rank) - m * (n-1) +1; 69147c6ae99SBarry Smith } 69247c6ae99SBarry Smith 69347c6ae99SBarry Smith if ((ye==N) && (ze==P)) { /* Assume an edge, not corner */ 69447c6ae99SBarry Smith n24 = rank - (size-m) -1; 69547c6ae99SBarry Smith n25 = rank - (size-m); 69647c6ae99SBarry Smith n26 = rank - (size-m) +1; 69747c6ae99SBarry Smith } 69847c6ae99SBarry Smith 69947c6ae99SBarry Smith /* Check for Corners */ 7008865f1eaSKarl Rupp if ((xs==0) && (ys==0) && (zs==0)) n0 = size -1; 7018865f1eaSKarl Rupp if ((xs==0) && (ys==0) && (ze==P)) n18 = m*n-1; 7028865f1eaSKarl Rupp if ((xs==0) && (ye==N) && (zs==0)) n6 = (size-1)-m*(n-1); 7038865f1eaSKarl Rupp if ((xs==0) && (ye==N) && (ze==P)) n24 = m-1; 7048865f1eaSKarl Rupp if ((xe==M) && (ys==0) && (zs==0)) n2 = size-m; 7058865f1eaSKarl Rupp if ((xe==M) && (ys==0) && (ze==P)) n20 = m*n-m; 7068865f1eaSKarl Rupp if ((xe==M) && (ye==N) && (zs==0)) n8 = size-m*n; 7078865f1eaSKarl Rupp if ((xe==M) && (ye==N) && (ze==P)) n26 = 0; 70847c6ae99SBarry Smith 70947c6ae99SBarry Smith /* Check for when not X,Y, and Z Periodic */ 71047c6ae99SBarry Smith 71147c6ae99SBarry Smith /* If not X periodic */ 712bff4a2f0SMatthew G. Knepley if (bx != DM_BOUNDARY_PERIODIC) { 7138865f1eaSKarl Rupp if (xs==0) n0 = n3 = n6 = n9 = n12 = n15 = n18 = n21 = n24 = -2; 7148865f1eaSKarl Rupp if (xe==M) n2 = n5 = n8 = n11 = n14 = n17 = n20 = n23 = n26 = -2; 71547c6ae99SBarry Smith } 71647c6ae99SBarry Smith 71747c6ae99SBarry Smith /* If not Y periodic */ 718bff4a2f0SMatthew G. Knepley if (by != DM_BOUNDARY_PERIODIC) { 7198865f1eaSKarl Rupp if (ys==0) n0 = n1 = n2 = n9 = n10 = n11 = n18 = n19 = n20 = -2; 7208865f1eaSKarl Rupp if (ye==N) n6 = n7 = n8 = n15 = n16 = n17 = n24 = n25 = n26 = -2; 72147c6ae99SBarry Smith } 72247c6ae99SBarry Smith 72347c6ae99SBarry Smith /* If not Z periodic */ 724bff4a2f0SMatthew G. Knepley if (bz != DM_BOUNDARY_PERIODIC) { 7258865f1eaSKarl Rupp if (zs==0) n0 = n1 = n2 = n3 = n4 = n5 = n6 = n7 = n8 = -2; 7268865f1eaSKarl Rupp if (ze==P) n18 = n19 = n20 = n21 = n22 = n23 = n24 = n25 = n26 = -2; 72747c6ae99SBarry Smith } 72847c6ae99SBarry Smith 729785e854fSJed Brown ierr = PetscMalloc1(27,&dd->neighbors);CHKERRQ(ierr); 7308865f1eaSKarl Rupp 73147c6ae99SBarry Smith dd->neighbors[0] = n0; 73247c6ae99SBarry Smith dd->neighbors[1] = n1; 73347c6ae99SBarry Smith dd->neighbors[2] = n2; 73447c6ae99SBarry Smith dd->neighbors[3] = n3; 73547c6ae99SBarry Smith dd->neighbors[4] = n4; 73647c6ae99SBarry Smith dd->neighbors[5] = n5; 73747c6ae99SBarry Smith dd->neighbors[6] = n6; 73847c6ae99SBarry Smith dd->neighbors[7] = n7; 73947c6ae99SBarry Smith dd->neighbors[8] = n8; 74047c6ae99SBarry Smith dd->neighbors[9] = n9; 74147c6ae99SBarry Smith dd->neighbors[10] = n10; 74247c6ae99SBarry Smith dd->neighbors[11] = n11; 74347c6ae99SBarry Smith dd->neighbors[12] = n12; 74447c6ae99SBarry Smith dd->neighbors[13] = rank; 74547c6ae99SBarry Smith dd->neighbors[14] = n14; 74647c6ae99SBarry Smith dd->neighbors[15] = n15; 74747c6ae99SBarry Smith dd->neighbors[16] = n16; 74847c6ae99SBarry Smith dd->neighbors[17] = n17; 74947c6ae99SBarry Smith dd->neighbors[18] = n18; 75047c6ae99SBarry Smith dd->neighbors[19] = n19; 75147c6ae99SBarry Smith dd->neighbors[20] = n20; 75247c6ae99SBarry Smith dd->neighbors[21] = n21; 75347c6ae99SBarry Smith dd->neighbors[22] = n22; 75447c6ae99SBarry Smith dd->neighbors[23] = n23; 75547c6ae99SBarry Smith dd->neighbors[24] = n24; 75647c6ae99SBarry Smith dd->neighbors[25] = n25; 75747c6ae99SBarry Smith dd->neighbors[26] = n26; 75847c6ae99SBarry Smith 75947c6ae99SBarry Smith /* If star stencil then delete the corner neighbors */ 760d9c9ebe5SPeter Brune if (stencil_type == DMDA_STENCIL_STAR) { 76147c6ae99SBarry Smith /* save information about corner neighbors */ 76247c6ae99SBarry Smith sn0 = n0; sn1 = n1; sn2 = n2; sn3 = n3; sn5 = n5; sn6 = n6; sn7 = n7; 76347c6ae99SBarry Smith sn8 = n8; sn9 = n9; sn11 = n11; sn15 = n15; sn17 = n17; sn18 = n18; 76447c6ae99SBarry Smith sn19 = n19; sn20 = n20; sn21 = n21; sn23 = n23; sn24 = n24; sn25 = n25; 76547c6ae99SBarry Smith sn26 = n26; 7668865f1eaSKarl Rupp n0 = n1 = n2 = n3 = n5 = n6 = n7 = n8 = n9 = n11 = n15 = n17 = n18 = n19 = n20 = n21 = n23 = n24 = n25 = n26 = -1; 76747c6ae99SBarry Smith } 76847c6ae99SBarry Smith 769785e854fSJed Brown ierr = PetscMalloc1((Xe-Xs)*(Ye-Ys)*(Ze-Zs),&idx);CHKERRQ(ierr); 77047c6ae99SBarry Smith 77147c6ae99SBarry Smith nn = 0; 77247c6ae99SBarry Smith /* Bottom Level */ 77347c6ae99SBarry Smith for (k=0; k<s_z; k++) { 77447c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 77547c6ae99SBarry Smith if (n0 >= 0) { /* left below */ 776ce00eea3SSatish Balay x_t = lx[n0 % m]; 77747c6ae99SBarry Smith y_t = ly[(n0 % (m*n))/m]; 77847c6ae99SBarry Smith z_t = lz[n0 / (m*n)]; 77947c6ae99SBarry 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; 7808865f1eaSKarl 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 */ 7818865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 78247c6ae99SBarry Smith } 78347c6ae99SBarry Smith if (n1 >= 0) { /* directly below */ 78447c6ae99SBarry Smith x_t = x; 78547c6ae99SBarry Smith y_t = ly[(n1 % (m*n))/m]; 78647c6ae99SBarry Smith z_t = lz[n1 / (m*n)]; 78747c6ae99SBarry 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; 7888865f1eaSKarl Rupp if (twod && (s_t < 0)) s_t = bases[n1] + x_t*y_t*z_t - (s_y+1-i)*x_t; /* 2D case */ 7898865f1eaSKarl Rupp for (j=0; j<x_t; j++) idx[nn++] = s_t++; 79047c6ae99SBarry Smith } 79147c6ae99SBarry Smith if (n2 >= 0) { /* right below */ 792ce00eea3SSatish Balay x_t = lx[n2 % m]; 79347c6ae99SBarry Smith y_t = ly[(n2 % (m*n))/m]; 79447c6ae99SBarry Smith z_t = lz[n2 / (m*n)]; 79547c6ae99SBarry 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; 7968865f1eaSKarl Rupp if (twod && (s_t < 0)) s_t = bases[n2] + x_t*y_t*z_t - (s_y+1-i)*x_t; /* 2D case */ 7978865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 79847c6ae99SBarry Smith } 79947c6ae99SBarry Smith } 80047c6ae99SBarry Smith 80147c6ae99SBarry Smith for (i=0; i<y; i++) { 80247c6ae99SBarry Smith if (n3 >= 0) { /* directly left */ 803ce00eea3SSatish Balay x_t = lx[n3 % m]; 80447c6ae99SBarry Smith y_t = y; 80547c6ae99SBarry Smith z_t = lz[n3 / (m*n)]; 80647c6ae99SBarry Smith s_t = bases[n3] + (i+1)*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8078865f1eaSKarl 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 */ 8088865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 80947c6ae99SBarry Smith } 81047c6ae99SBarry Smith 81147c6ae99SBarry Smith if (n4 >= 0) { /* middle */ 81247c6ae99SBarry Smith x_t = x; 81347c6ae99SBarry Smith y_t = y; 81447c6ae99SBarry Smith z_t = lz[n4 / (m*n)]; 81547c6ae99SBarry Smith s_t = bases[n4] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8168865f1eaSKarl 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 */ 8178865f1eaSKarl Rupp for (j=0; j<x_t; j++) idx[nn++] = s_t++; 818bff4a2f0SMatthew G. Knepley } else if (bz == DM_BOUNDARY_MIRROR) { 8198865f1eaSKarl Rupp for (j=0; j<x; j++) idx[nn++] = 0; 82047c6ae99SBarry Smith } 82147c6ae99SBarry Smith 82247c6ae99SBarry Smith if (n5 >= 0) { /* directly right */ 823ce00eea3SSatish Balay x_t = lx[n5 % m]; 82447c6ae99SBarry Smith y_t = y; 82547c6ae99SBarry Smith z_t = lz[n5 / (m*n)]; 82647c6ae99SBarry Smith s_t = bases[n5] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8278865f1eaSKarl 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 */ 8288865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 82947c6ae99SBarry Smith } 83047c6ae99SBarry Smith } 83147c6ae99SBarry Smith 83247c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 83347c6ae99SBarry Smith if (n6 >= 0) { /* left above */ 834ce00eea3SSatish Balay x_t = lx[n6 % m]; 83547c6ae99SBarry Smith y_t = ly[(n6 % (m*n))/m]; 83647c6ae99SBarry Smith z_t = lz[n6 / (m*n)]; 83747c6ae99SBarry Smith s_t = bases[n6] + i*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8388865f1eaSKarl 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 */ 8398865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 84047c6ae99SBarry Smith } 84147c6ae99SBarry Smith if (n7 >= 0) { /* directly above */ 84247c6ae99SBarry Smith x_t = x; 84347c6ae99SBarry Smith y_t = ly[(n7 % (m*n))/m]; 84447c6ae99SBarry Smith z_t = lz[n7 / (m*n)]; 84547c6ae99SBarry Smith s_t = bases[n7] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8468865f1eaSKarl 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 */ 8478865f1eaSKarl Rupp for (j=0; j<x_t; j++) idx[nn++] = s_t++; 84847c6ae99SBarry Smith } 84947c6ae99SBarry Smith if (n8 >= 0) { /* right above */ 850ce00eea3SSatish Balay x_t = lx[n8 % m]; 85147c6ae99SBarry Smith y_t = ly[(n8 % (m*n))/m]; 85247c6ae99SBarry Smith z_t = lz[n8 / (m*n)]; 85347c6ae99SBarry Smith s_t = bases[n8] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t; 8548865f1eaSKarl 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 */ 8558865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 85647c6ae99SBarry Smith } 85747c6ae99SBarry Smith } 85847c6ae99SBarry Smith } 85947c6ae99SBarry Smith 86047c6ae99SBarry Smith /* Middle Level */ 86147c6ae99SBarry Smith for (k=0; k<z; k++) { 86247c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 86347c6ae99SBarry Smith if (n9 >= 0) { /* left below */ 864ce00eea3SSatish Balay x_t = lx[n9 % m]; 86547c6ae99SBarry Smith y_t = ly[(n9 % (m*n))/m]; 86647c6ae99SBarry Smith /* z_t = z; */ 86747c6ae99SBarry Smith s_t = bases[n9] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t; 8688865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 86947c6ae99SBarry Smith } 87047c6ae99SBarry Smith if (n10 >= 0) { /* directly below */ 87147c6ae99SBarry Smith x_t = x; 87247c6ae99SBarry Smith y_t = ly[(n10 % (m*n))/m]; 87347c6ae99SBarry Smith /* z_t = z; */ 87447c6ae99SBarry Smith s_t = bases[n10] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 8758865f1eaSKarl Rupp for (j=0; j<x_t; j++) idx[nn++] = s_t++; 876bff4a2f0SMatthew G. Knepley } else if (by == DM_BOUNDARY_MIRROR) { 8778865f1eaSKarl Rupp for (j=0; j<x; j++) idx[nn++] = 0; 87847c6ae99SBarry Smith } 87947c6ae99SBarry Smith if (n11 >= 0) { /* right below */ 880ce00eea3SSatish Balay x_t = lx[n11 % m]; 88147c6ae99SBarry Smith y_t = ly[(n11 % (m*n))/m]; 88247c6ae99SBarry Smith /* z_t = z; */ 88347c6ae99SBarry Smith s_t = bases[n11] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 8848865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 88547c6ae99SBarry Smith } 88647c6ae99SBarry Smith } 88747c6ae99SBarry Smith 88847c6ae99SBarry Smith for (i=0; i<y; i++) { 88947c6ae99SBarry Smith if (n12 >= 0) { /* directly left */ 890ce00eea3SSatish Balay x_t = lx[n12 % m]; 89147c6ae99SBarry Smith y_t = y; 89247c6ae99SBarry Smith /* z_t = z; */ 89347c6ae99SBarry Smith s_t = bases[n12] + (i+1)*x_t - s_x + k*x_t*y_t; 8948865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 895bff4a2f0SMatthew G. Knepley } else if (bx == DM_BOUNDARY_MIRROR) { 8968865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = 0; 89747c6ae99SBarry Smith } 89847c6ae99SBarry Smith 89947c6ae99SBarry Smith /* Interior */ 90047c6ae99SBarry Smith s_t = bases[rank] + i*x + k*x*y; 9018865f1eaSKarl Rupp for (j=0; j<x; j++) idx[nn++] = s_t++; 90247c6ae99SBarry Smith 90347c6ae99SBarry Smith if (n14 >= 0) { /* directly right */ 904ce00eea3SSatish Balay x_t = lx[n14 % m]; 90547c6ae99SBarry Smith y_t = y; 90647c6ae99SBarry Smith /* z_t = z; */ 90747c6ae99SBarry Smith s_t = bases[n14] + i*x_t + k*x_t*y_t; 9088865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 909bff4a2f0SMatthew G. Knepley } else if (bx == DM_BOUNDARY_MIRROR) { 9108865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = 0; 91147c6ae99SBarry Smith } 91247c6ae99SBarry Smith } 91347c6ae99SBarry Smith 91447c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 91547c6ae99SBarry Smith if (n15 >= 0) { /* left above */ 916ce00eea3SSatish Balay x_t = lx[n15 % m]; 91747c6ae99SBarry Smith y_t = ly[(n15 % (m*n))/m]; 91847c6ae99SBarry Smith /* z_t = z; */ 91947c6ae99SBarry Smith s_t = bases[n15] + i*x_t - s_x + k*x_t*y_t; 9208865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 92147c6ae99SBarry Smith } 92247c6ae99SBarry Smith if (n16 >= 0) { /* directly above */ 92347c6ae99SBarry Smith x_t = x; 92447c6ae99SBarry Smith y_t = ly[(n16 % (m*n))/m]; 92547c6ae99SBarry Smith /* z_t = z; */ 92647c6ae99SBarry Smith s_t = bases[n16] + (i-1)*x_t + k*x_t*y_t; 9278865f1eaSKarl Rupp for (j=0; j<x_t; j++) idx[nn++] = s_t++; 928bff4a2f0SMatthew G. Knepley } else if (by == DM_BOUNDARY_MIRROR) { 9298865f1eaSKarl Rupp for (j=0; j<x; j++) idx[nn++] = 0; 93047c6ae99SBarry Smith } 93147c6ae99SBarry Smith if (n17 >= 0) { /* right above */ 932ce00eea3SSatish Balay x_t = lx[n17 % m]; 93347c6ae99SBarry Smith y_t = ly[(n17 % (m*n))/m]; 93447c6ae99SBarry Smith /* z_t = z; */ 93547c6ae99SBarry Smith s_t = bases[n17] + (i-1)*x_t + k*x_t*y_t; 9368865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 93747c6ae99SBarry Smith } 93847c6ae99SBarry Smith } 93947c6ae99SBarry Smith } 94047c6ae99SBarry Smith 94147c6ae99SBarry Smith /* Upper Level */ 94247c6ae99SBarry Smith for (k=0; k<s_z; k++) { 94347c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 94447c6ae99SBarry Smith if (n18 >= 0) { /* left below */ 945ce00eea3SSatish Balay x_t = lx[n18 % m]; 94647c6ae99SBarry Smith y_t = ly[(n18 % (m*n))/m]; 94747c6ae99SBarry Smith /* z_t = lz[n18 / (m*n)]; */ 94847c6ae99SBarry Smith s_t = bases[n18] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t; 9498865f1eaSKarl 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 */ 9508865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 95147c6ae99SBarry Smith } 95247c6ae99SBarry Smith if (n19 >= 0) { /* directly below */ 95347c6ae99SBarry Smith x_t = x; 95447c6ae99SBarry Smith y_t = ly[(n19 % (m*n))/m]; 95547c6ae99SBarry Smith /* z_t = lz[n19 / (m*n)]; */ 95647c6ae99SBarry Smith s_t = bases[n19] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 9578865f1eaSKarl Rupp if (twod && (s_t >= M*N*P)) s_t = bases[n19] - (s_y+1-i)*x_t + x_t*y_t; /* 2d case */ 9588865f1eaSKarl Rupp for (j=0; j<x_t; j++) idx[nn++] = s_t++; 95947c6ae99SBarry Smith } 96047c6ae99SBarry Smith if (n20 >= 0) { /* right below */ 961ce00eea3SSatish Balay x_t = lx[n20 % m]; 96247c6ae99SBarry Smith y_t = ly[(n20 % (m*n))/m]; 96347c6ae99SBarry Smith /* z_t = lz[n20 / (m*n)]; */ 96447c6ae99SBarry Smith s_t = bases[n20] - (s_y+1-i)*x_t + (k+1)*x_t*y_t; 9658865f1eaSKarl Rupp if (twod && (s_t >= M*N*P)) s_t = bases[n20] - (s_y+1-i)*x_t + x_t*y_t; /* 2d case */ 9668865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 96747c6ae99SBarry Smith } 96847c6ae99SBarry Smith } 96947c6ae99SBarry Smith 97047c6ae99SBarry Smith for (i=0; i<y; i++) { 97147c6ae99SBarry Smith if (n21 >= 0) { /* directly left */ 972ce00eea3SSatish Balay x_t = lx[n21 % m]; 97347c6ae99SBarry Smith y_t = y; 97447c6ae99SBarry Smith /* z_t = lz[n21 / (m*n)]; */ 97547c6ae99SBarry Smith s_t = bases[n21] + (i+1)*x_t - s_x + k*x_t*y_t; 9768865f1eaSKarl Rupp if (twod && (s_t >= M*N*P)) s_t = bases[n21] + (i+1)*x_t - s_x; /* 2d case */ 9778865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 97847c6ae99SBarry Smith } 97947c6ae99SBarry Smith 98047c6ae99SBarry Smith if (n22 >= 0) { /* middle */ 98147c6ae99SBarry Smith x_t = x; 98247c6ae99SBarry Smith y_t = y; 98347c6ae99SBarry Smith /* z_t = lz[n22 / (m*n)]; */ 98447c6ae99SBarry Smith s_t = bases[n22] + i*x_t + k*x_t*y_t; 9858865f1eaSKarl Rupp if (twod && (s_t >= M*N*P)) s_t = bases[n22] + i*x_t; /* 2d case */ 9868865f1eaSKarl Rupp for (j=0; j<x_t; j++) idx[nn++] = s_t++; 987bff4a2f0SMatthew G. Knepley } else if (bz == DM_BOUNDARY_MIRROR) { 9888865f1eaSKarl Rupp for (j=0; j<x; j++) idx[nn++] = 0; 98947c6ae99SBarry Smith } 99047c6ae99SBarry Smith 99147c6ae99SBarry Smith if (n23 >= 0) { /* directly right */ 992ce00eea3SSatish Balay x_t = lx[n23 % m]; 99347c6ae99SBarry Smith y_t = y; 99447c6ae99SBarry Smith /* z_t = lz[n23 / (m*n)]; */ 99547c6ae99SBarry Smith s_t = bases[n23] + i*x_t + k*x_t*y_t; 9968865f1eaSKarl Rupp if (twod && (s_t >= M*N*P)) s_t = bases[n23] + i*x_t; /* 2d case */ 9978865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 99847c6ae99SBarry Smith } 99947c6ae99SBarry Smith } 100047c6ae99SBarry Smith 100147c6ae99SBarry Smith for (i=1; i<=s_y; i++) { 100247c6ae99SBarry Smith if (n24 >= 0) { /* left above */ 1003ce00eea3SSatish Balay x_t = lx[n24 % m]; 100447c6ae99SBarry Smith y_t = ly[(n24 % (m*n))/m]; 100547c6ae99SBarry Smith /* z_t = lz[n24 / (m*n)]; */ 100647c6ae99SBarry Smith s_t = bases[n24] + i*x_t - s_x + k*x_t*y_t; 10078865f1eaSKarl Rupp if (twod && (s_t >= M*N*P)) s_t = bases[n24] + i*x_t - s_x; /* 2d case */ 10088865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 100947c6ae99SBarry Smith } 101047c6ae99SBarry Smith if (n25 >= 0) { /* directly above */ 101147c6ae99SBarry Smith x_t = x; 101247c6ae99SBarry Smith y_t = ly[(n25 % (m*n))/m]; 101347c6ae99SBarry Smith /* z_t = lz[n25 / (m*n)]; */ 101447c6ae99SBarry Smith s_t = bases[n25] + (i-1)*x_t + k*x_t*y_t; 10158865f1eaSKarl Rupp if (twod && (s_t >= M*N*P)) s_t = bases[n25] + (i-1)*x_t; /* 2d case */ 10168865f1eaSKarl Rupp for (j=0; j<x_t; j++) idx[nn++] = s_t++; 101747c6ae99SBarry Smith } 101847c6ae99SBarry Smith if (n26 >= 0) { /* right above */ 1019ce00eea3SSatish Balay x_t = lx[n26 % m]; 102047c6ae99SBarry Smith y_t = ly[(n26 % (m*n))/m]; 102147c6ae99SBarry Smith /* z_t = lz[n26 / (m*n)]; */ 102247c6ae99SBarry Smith s_t = bases[n26] + (i-1)*x_t + k*x_t*y_t; 10238865f1eaSKarl Rupp if (twod && (s_t >= M*N*P)) s_t = bases[n26] + (i-1)*x_t; /* 2d case */ 10248865f1eaSKarl Rupp for (j=0; j<s_x; j++) idx[nn++] = s_t++; 102547c6ae99SBarry Smith } 102647c6ae99SBarry Smith } 102747c6ae99SBarry Smith } 1028ce00eea3SSatish Balay 1029b1fb7eb7SBarry Smith ierr = ISCreateBlock(comm,dof,nn,idx,PETSC_USE_POINTER,&from);CHKERRQ(ierr); 103047c6ae99SBarry Smith ierr = VecScatterCreate(global,from,local,to,>ol);CHKERRQ(ierr); 10313bb1ff40SBarry Smith ierr = PetscLogObjectParent((PetscObject)da,(PetscObject)gtol);CHKERRQ(ierr); 1032fcfd50ebSBarry Smith ierr = ISDestroy(&to);CHKERRQ(ierr); 1033fcfd50ebSBarry Smith ierr = ISDestroy(&from);CHKERRQ(ierr); 103447c6ae99SBarry Smith 1035d9c9ebe5SPeter Brune if (stencil_type == DMDA_STENCIL_STAR) { 103647c6ae99SBarry Smith n0 = sn0; n1 = sn1; n2 = sn2; n3 = sn3; n5 = sn5; n6 = sn6; n7 = sn7; 103747c6ae99SBarry Smith n8 = sn8; n9 = sn9; n11 = sn11; n15 = sn15; n17 = sn17; n18 = sn18; 103847c6ae99SBarry Smith n19 = sn19; n20 = sn20; n21 = sn21; n23 = sn23; n24 = sn24; n25 = sn25; 103947c6ae99SBarry Smith n26 = sn26; 1040ce00eea3SSatish Balay } 104147c6ae99SBarry Smith 1042288e7d53SBarry Smith if (((stencil_type == DMDA_STENCIL_STAR) || (bx != DM_BOUNDARY_PERIODIC && bx) || (by != DM_BOUNDARY_PERIODIC && by) || (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 /*@C 1377aa219208SBarry Smith DMDACreate3d - Creates an object that will manage the communication of three-dimensional 137847c6ae99SBarry Smith regular array data that is distributed across some processors. 137947c6ae99SBarry Smith 138047c6ae99SBarry Smith Collective on MPI_Comm 138147c6ae99SBarry Smith 138247c6ae99SBarry Smith Input Parameters: 138347c6ae99SBarry Smith + comm - MPI communicator 13841321219cSEthan Coon . bx,by,bz - type of ghost nodes the array have. 1385bff4a2f0SMatthew G. Knepley Use one of DM_BOUNDARY_NONE, DM_BOUNDARY_GHOSTED, DM_BOUNDARY_PERIODIC. 1386aa219208SBarry Smith . stencil_type - Type of stencil (DMDA_STENCIL_STAR or DMDA_STENCIL_BOX) 1387897f7067SBarry Smith . M,N,P - global dimension in each direction of the array 138847c6ae99SBarry Smith . m,n,p - corresponding number of processors in each dimension 138947c6ae99SBarry Smith (or PETSC_DECIDE to have calculated) 139047c6ae99SBarry Smith . dof - number of degrees of freedom per node 139110d7c030SMatthew G Knepley . s - stencil width 139210d7c030SMatthew G Knepley - lx, ly, lz - arrays containing the number of nodes in each cell along 13930298fd71SBarry Smith the x, y, and z coordinates, or NULL. If non-null, these 139447c6ae99SBarry Smith must be of length as m,n,p and the corresponding 139547c6ae99SBarry Smith m,n, or p cannot be PETSC_DECIDE. Sum of the lx[] entries must be M, sum of 139647c6ae99SBarry Smith the ly[] must N, sum of the lz[] must be P 139747c6ae99SBarry Smith 139847c6ae99SBarry Smith Output Parameter: 139947c6ae99SBarry Smith . da - the resulting distributed array object 140047c6ae99SBarry Smith 140147c6ae99SBarry Smith Options Database Key: 1402706a11cbSBarry Smith + -dm_view - Calls DMView() at the conclusion of DMDACreate3d() 1403*e43dc8daSBarry Smith . -da_grid_x <nx> - number of grid points in x direction 1404*e43dc8daSBarry Smith . -da_grid_y <ny> - number of grid points in y direction 1405*e43dc8daSBarry Smith . -da_grid_z <nz> - number of grid points in z direction 140647c6ae99SBarry Smith . -da_processors_x <MX> - number of processors in x direction 140747c6ae99SBarry Smith . -da_processors_y <MY> - number of processors in y direction 140847c6ae99SBarry Smith . -da_processors_z <MZ> - number of processors in z direction 1409e0f5d30fSBarry Smith . -da_refine_x <rx> - refinement ratio in x direction 1410e0f5d30fSBarry Smith . -da_refine_y <ry> - refinement ratio in y direction 1411e0f5d30fSBarry Smith . -da_refine_z <rz>- refinement ratio in z directio 1412*e43dc8daSBarry Smith - -da_refine <n> - refine the DMDA n times before creating it 141347c6ae99SBarry Smith 141447c6ae99SBarry Smith Level: beginner 141547c6ae99SBarry Smith 141647c6ae99SBarry Smith Notes: 1417aa219208SBarry Smith The stencil type DMDA_STENCIL_STAR with width 1 corresponds to the 1418aa219208SBarry Smith standard 7-pt stencil, while DMDA_STENCIL_BOX with width 1 denotes 141947c6ae99SBarry Smith the standard 27-pt stencil. 142047c6ae99SBarry Smith 1421aa219208SBarry Smith The array data itself is NOT stored in the DMDA, it is stored in Vec objects; 1422564755cdSBarry Smith The appropriate vector objects can be obtained with calls to DMCreateGlobalVector() 1423564755cdSBarry Smith and DMCreateLocalVector() and calls to VecDuplicate() if more are needed. 142447c6ae99SBarry Smith 1425897f7067SBarry Smith You must call DMSetUp() after this call before using this DM. 1426897f7067SBarry Smith 1427897f7067SBarry Smith If you wish to use the options database to change values in the DMDA call DMSetFromOptions() after this call 1428897f7067SBarry Smith but before DMSetUp(). 1429897f7067SBarry Smith 143047c6ae99SBarry Smith .keywords: distributed array, create, three-dimensional 143147c6ae99SBarry Smith 1432aa219208SBarry Smith .seealso: DMDestroy(), DMView(), DMDACreate1d(), DMDACreate2d(), DMGlobalToLocalBegin(), DMDAGetRefinementFactor(), 143399f0b487SRichard Tran Mills DMGlobalToLocalEnd(), DMLocalToGlobalBegin(), DMLocalToLocalBegin(), DMLocalToLocalEnd(), DMDASetRefinementFactor(), 1434d461ba97SBarry Smith DMDAGetInfo(), DMCreateGlobalVector(), DMCreateLocalVector(), DMDACreateNaturalVector(), DMLoad(), DMDAGetOwnershipRanges() 143547c6ae99SBarry Smith 143647c6ae99SBarry Smith @*/ 1437bff4a2f0SMatthew G. Knepley PetscErrorCode DMDACreate3d(MPI_Comm comm,DMBoundaryType bx,DMBoundaryType by,DMBoundaryType bz,DMDAStencilType stencil_type,PetscInt M, 14389a42bb27SBarry 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) 143947c6ae99SBarry Smith { 144047c6ae99SBarry Smith PetscErrorCode ierr; 144147c6ae99SBarry Smith 144247c6ae99SBarry Smith PetscFunctionBegin; 1443aa219208SBarry Smith ierr = DMDACreate(comm, da);CHKERRQ(ierr); 1444c73cfb54SMatthew G. Knepley ierr = DMSetDimension(*da, 3);CHKERRQ(ierr); 1445aa219208SBarry Smith ierr = DMDASetSizes(*da, M, N, P);CHKERRQ(ierr); 1446aa219208SBarry Smith ierr = DMDASetNumProcs(*da, m, n, p);CHKERRQ(ierr); 14471321219cSEthan Coon ierr = DMDASetBoundaryType(*da, bx, by, bz);CHKERRQ(ierr); 1448aa219208SBarry Smith ierr = DMDASetDof(*da, dof);CHKERRQ(ierr); 1449aa219208SBarry Smith ierr = DMDASetStencilType(*da, stencil_type);CHKERRQ(ierr); 1450aa219208SBarry Smith ierr = DMDASetStencilWidth(*da, s);CHKERRQ(ierr); 1451aa219208SBarry Smith ierr = DMDASetOwnershipRanges(*da, lx, ly, lz);CHKERRQ(ierr); 145247c6ae99SBarry Smith PetscFunctionReturn(0); 145347c6ae99SBarry Smith } 1454