Actual source code: da3.c
petsc-3.5.4 2015-05-23
2: /*
3: Code for manipulating distributed regular 3d arrays in parallel.
4: File created by Peter Mell 7/14/95
5: */
7: #include <petsc-private/dmdaimpl.h> /*I "petscdmda.h" I*/
9: #include <petscdraw.h>
12: PetscErrorCode DMView_DA_3d(DM da,PetscViewer viewer)
13: {
15: PetscMPIInt rank;
16: PetscBool iascii,isdraw,isbinary;
17: DM_DA *dd = (DM_DA*)da->data;
18: #if defined(PETSC_HAVE_MATLAB_ENGINE)
19: PetscBool ismatlab;
20: #endif
23: MPI_Comm_rank(PetscObjectComm((PetscObject)da),&rank);
25: PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);
26: PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);
27: PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);
28: #if defined(PETSC_HAVE_MATLAB_ENGINE)
29: PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERMATLAB,&ismatlab);
30: #endif
31: if (iascii) {
32: PetscViewerFormat format;
34: PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);
35: PetscViewerGetFormat(viewer, &format);
36: if (format != PETSC_VIEWER_ASCII_VTK && format != PETSC_VIEWER_ASCII_VTK_CELL) {
37: DMDALocalInfo info;
38: DMDAGetLocalInfo(da,&info);
39: 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);
40: PetscViewerASCIISynchronizedPrintf(viewer,"X range of indices: %D %D, Y range of indices: %D %D, Z range of indices: %D %D\n",
41: info.xs,info.xs+info.xm,info.ys,info.ys+info.ym,info.zs,info.zs+info.zm);
42: #if !defined(PETSC_USE_COMPLEX)
43: if (da->coordinates) {
44: PetscInt last;
45: const PetscReal *coors;
46: VecGetArrayRead(da->coordinates,&coors);
47: VecGetLocalSize(da->coordinates,&last);
48: last = last - 3;
49: 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]);
50: VecRestoreArrayRead(da->coordinates,&coors);
51: }
52: #endif
53: PetscViewerFlush(viewer);
54: PetscViewerASCIISynchronizedAllow(viewer,PETSC_FALSE);
55: } else {
56: DMView_DA_VTK(da,viewer);
57: }
58: } else if (isdraw) {
59: PetscDraw draw;
60: PetscReal ymin = -1.0,ymax = (PetscReal)dd->N;
61: PetscReal xmin = -1.0,xmax = (PetscReal)((dd->M+2)*dd->P),x,y,ycoord,xcoord;
62: PetscInt k,plane,base;
63: const PetscInt *idx;
64: char node[10];
65: PetscBool isnull;
67: PetscViewerDrawGetDraw(viewer,0,&draw);
68: PetscDrawIsNull(draw,&isnull); if (isnull) return(0);
69: PetscDrawSetCoordinates(draw,xmin,ymin,xmax,ymax);
70: PetscDrawSynchronizedClear(draw);
72: /* first processor draw all node lines */
73: if (!rank) {
74: for (k=0; k<dd->P; k++) {
75: ymin = 0.0; ymax = (PetscReal)(dd->N - 1);
76: for (xmin=(PetscReal)(k*(dd->M+1)); xmin<(PetscReal)(dd->M+(k*(dd->M+1))); xmin++) {
77: PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_BLACK);
78: }
80: xmin = (PetscReal)(k*(dd->M+1)); xmax = xmin + (PetscReal)(dd->M - 1);
81: for (ymin=0; ymin<(PetscReal)dd->N; ymin++) {
82: PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_BLACK);
83: }
84: }
85: }
86: PetscDrawSynchronizedFlush(draw);
87: PetscDrawPause(draw);
89: for (k=0; k<dd->P; k++) { /*Go through and draw for each plane*/
90: if ((k >= dd->zs) && (k < dd->ze)) {
91: /* draw my box */
92: ymin = dd->ys;
93: ymax = dd->ye - 1;
94: xmin = dd->xs/dd->w + (dd->M+1)*k;
95: xmax =(dd->xe-1)/dd->w + (dd->M+1)*k;
97: PetscDrawLine(draw,xmin,ymin,xmax,ymin,PETSC_DRAW_RED);
98: PetscDrawLine(draw,xmin,ymin,xmin,ymax,PETSC_DRAW_RED);
99: PetscDrawLine(draw,xmin,ymax,xmax,ymax,PETSC_DRAW_RED);
100: PetscDrawLine(draw,xmax,ymin,xmax,ymax,PETSC_DRAW_RED);
102: xmin = dd->xs/dd->w;
103: xmax =(dd->xe-1)/dd->w;
105: /* put in numbers*/
106: base = (dd->base+(dd->xe-dd->xs)*(dd->ye-dd->ys)*(k-dd->zs))/dd->w;
108: /* Identify which processor owns the box */
109: sprintf(node,"%d",rank);
110: PetscDrawString(draw,xmin+(dd->M+1)*k+.2,ymin+.3,PETSC_DRAW_RED,node);
112: for (y=ymin; y<=ymax; y++) {
113: for (x=xmin+(dd->M+1)*k; x<=xmax+(dd->M+1)*k; x++) {
114: sprintf(node,"%d",(int)base++);
115: PetscDrawString(draw,x,y,PETSC_DRAW_BLACK,node);
116: }
117: }
119: }
120: }
121: PetscDrawSynchronizedFlush(draw);
122: PetscDrawPause(draw);
124: for (k=0-dd->s; k<dd->P+dd->s; k++) {
125: /* Go through and draw for each plane */
126: if ((k >= dd->Zs) && (k < dd->Ze)) {
128: /* overlay ghost numbers, useful for error checking */
129: base = (dd->Xe-dd->Xs)*(dd->Ye-dd->Ys)*(k-dd->Zs)/dd->w;
130: ISLocalToGlobalMappingGetBlockIndices(da->ltogmap,&idx);
131: plane=k;
132: /* Keep z wrap around points on the drawing */
133: if (k<0) plane=dd->P+k;
134: if (k>=dd->P) plane=k-dd->P;
135: ymin = dd->Ys; ymax = dd->Ye;
136: xmin = (dd->M+1)*plane*dd->w;
137: xmax = (dd->M+1)*plane*dd->w+dd->M*dd->w;
138: for (y=ymin; y<ymax; y++) {
139: for (x=xmin+dd->Xs; x<xmin+dd->Xe; x+=dd->w) {
140: sprintf(node,"%d",(int)(idx[base]));
141: ycoord = y;
142: /*Keep y wrap around points on drawing */
143: if (y<0) ycoord = dd->N+y;
145: if (y>=dd->N) ycoord = y-dd->N;
146: xcoord = x; /* Keep x wrap points on drawing */
148: if (x<xmin) xcoord = xmax - (xmin-x);
149: if (x>=xmax) xcoord = xmin + (x-xmax);
150: PetscDrawString(draw,xcoord/dd->w,ycoord,PETSC_DRAW_BLUE,node);
151: base++;
152: }
153: }
154: ISLocalToGlobalMappingRestoreBlockIndices(da->ltogmap,&idx);
155: }
156: }
157: PetscDrawSynchronizedFlush(draw);
158: PetscDrawPause(draw);
159: } else if (isbinary) {
160: DMView_DA_Binary(da,viewer);
161: #if defined(PETSC_HAVE_MATLAB_ENGINE)
162: } else if (ismatlab) {
163: DMView_DA_Matlab(da,viewer);
164: #endif
165: }
166: return(0);
167: }
171: PetscErrorCode DMSetUp_DA_3D(DM da)
172: {
173: DM_DA *dd = (DM_DA*)da->data;
174: const PetscInt M = dd->M;
175: const PetscInt N = dd->N;
176: const PetscInt P = dd->P;
177: PetscInt m = dd->m;
178: PetscInt n = dd->n;
179: PetscInt p = dd->p;
180: const PetscInt dof = dd->w;
181: const PetscInt s = dd->s;
182: DMBoundaryType bx = dd->bx;
183: DMBoundaryType by = dd->by;
184: DMBoundaryType bz = dd->bz;
185: DMDAStencilType stencil_type = dd->stencil_type;
186: PetscInt *lx = dd->lx;
187: PetscInt *ly = dd->ly;
188: PetscInt *lz = dd->lz;
189: MPI_Comm comm;
190: PetscMPIInt rank,size;
191: PetscInt xs = 0,xe,ys = 0,ye,zs = 0,ze,x = 0,y = 0,z = 0;
192: PetscInt Xs,Xe,Ys,Ye,Zs,Ze,IXs,IXe,IYs,IYe,IZs,IZe,pm;
193: PetscInt left,right,up,down,bottom,top,i,j,k,*idx,nn;
194: PetscInt n0,n1,n2,n3,n4,n5,n6,n7,n8,n9,n10,n11,n12,n14;
195: PetscInt n15,n16,n17,n18,n19,n20,n21,n22,n23,n24,n25,n26;
196: PetscInt *bases,*ldims,base,x_t,y_t,z_t,s_t,count,s_x,s_y,s_z;
197: PetscInt sn0 = 0,sn1 = 0,sn2 = 0,sn3 = 0,sn5 = 0,sn6 = 0,sn7 = 0;
198: PetscInt sn8 = 0,sn9 = 0,sn11 = 0,sn15 = 0,sn24 = 0,sn25 = 0,sn26 = 0;
199: PetscInt sn17 = 0,sn18 = 0,sn19 = 0,sn20 = 0,sn21 = 0,sn23 = 0;
200: Vec local,global;
201: VecScatter gtol;
202: IS to,from;
203: PetscBool twod;
204: PetscErrorCode ierr;
208: 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");
209: PetscObjectGetComm((PetscObject) da, &comm);
210: #if !defined(PETSC_USE_64BIT_INDICES)
211: if (((Petsc64bitInt) M)*((Petsc64bitInt) N)*((Petsc64bitInt) P)*((Petsc64bitInt) dof) > (Petsc64bitInt) PETSC_MPI_INT_MAX) SETERRQ3(comm,PETSC_ERR_INT_OVERFLOW,"Mesh of %D by %D by %D (dof) is too large for 32 bit indices",M,N,dof);
212: #endif
214: MPI_Comm_size(comm,&size);
215: MPI_Comm_rank(comm,&rank);
217: if (m != PETSC_DECIDE) {
218: if (m < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in X direction: %D",m);
219: else if (m > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in X direction: %D %d",m,size);
220: }
221: if (n != PETSC_DECIDE) {
222: if (n < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in Y direction: %D",n);
223: else if (n > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in Y direction: %D %d",n,size);
224: }
225: if (p != PETSC_DECIDE) {
226: if (p < 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Non-positive number of processors in Z direction: %D",p);
227: else if (p > size) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Too many processors in Z direction: %D %d",p,size);
228: }
229: 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);
231: /* Partition the array among the processors */
232: if (m == PETSC_DECIDE && n != PETSC_DECIDE && p != PETSC_DECIDE) {
233: m = size/(n*p);
234: } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) {
235: n = size/(m*p);
236: } else if (m != PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) {
237: p = size/(m*n);
238: } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p != PETSC_DECIDE) {
239: /* try for squarish distribution */
240: m = (int)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)N*p)));
241: if (!m) m = 1;
242: while (m > 0) {
243: n = size/(m*p);
244: if (m*n*p == size) break;
245: m--;
246: }
247: if (!m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad p value: p = %D",p);
248: if (M > N && m < n) {PetscInt _m = m; m = n; n = _m;}
249: } else if (m == PETSC_DECIDE && n != PETSC_DECIDE && p == PETSC_DECIDE) {
250: /* try for squarish distribution */
251: m = (int)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)P*n)));
252: if (!m) m = 1;
253: while (m > 0) {
254: p = size/(m*n);
255: if (m*n*p == size) break;
256: m--;
257: }
258: if (!m) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad n value: n = %D",n);
259: if (M > P && m < p) {PetscInt _m = m; m = p; p = _m;}
260: } else if (m != PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) {
261: /* try for squarish distribution */
262: n = (int)(0.5 + PetscSqrtReal(((PetscReal)N)*((PetscReal)size)/((PetscReal)P*m)));
263: if (!n) n = 1;
264: while (n > 0) {
265: p = size/(m*n);
266: if (m*n*p == size) break;
267: n--;
268: }
269: if (!n) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"bad m value: m = %D",n);
270: if (N > P && n < p) {PetscInt _n = n; n = p; p = _n;}
271: } else if (m == PETSC_DECIDE && n == PETSC_DECIDE && p == PETSC_DECIDE) {
272: /* try for squarish distribution */
273: n = (PetscInt)(0.5 + PetscPowReal(((PetscReal)N*N)*((PetscReal)size)/((PetscReal)P*M),(PetscReal)(1./3.)));
274: if (!n) n = 1;
275: while (n > 0) {
276: pm = size/n;
277: if (n*pm == size) break;
278: n--;
279: }
280: if (!n) n = 1;
281: m = (PetscInt)(0.5 + PetscSqrtReal(((PetscReal)M)*((PetscReal)size)/((PetscReal)P*n)));
282: if (!m) m = 1;
283: while (m > 0) {
284: p = size/(m*n);
285: if (m*n*p == size) break;
286: m--;
287: }
288: if (M > P && m < p) {PetscInt _m = m; m = p; p = _m;}
289: } else if (m*n*p != size) SETERRQ(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Given Bad partition");
291: if (m*n*p != size) SETERRQ(PetscObjectComm((PetscObject)da),PETSC_ERR_PLIB,"Could not find good partition");
292: if (M < m) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in x direction is too fine! %D %D",M,m);
293: if (N < n) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in y direction is too fine! %D %D",N,n);
294: if (P < p) SETERRQ2(PetscObjectComm((PetscObject)da),PETSC_ERR_ARG_OUTOFRANGE,"Partition in z direction is too fine! %D %D",P,p);
296: /*
297: Determine locally owned region
298: [x, y, or z]s is the first local node number, [x, y, z] is the number of local nodes
299: */
301: if (!lx) {
302: PetscMalloc1(m, &dd->lx);
303: lx = dd->lx;
304: for (i=0; i<m; i++) lx[i] = M/m + ((M % m) > (i % m));
305: }
306: x = lx[rank % m];
307: xs = 0;
308: for (i=0; i<(rank%m); i++) xs += lx[i];
309: 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);
311: if (!ly) {
312: PetscMalloc1(n, &dd->ly);
313: ly = dd->ly;
314: for (i=0; i<n; i++) ly[i] = N/n + ((N % n) > (i % n));
315: }
316: y = ly[(rank % (m*n))/m];
317: 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);
319: ys = 0;
320: for (i=0; i<(rank % (m*n))/m; i++) ys += ly[i];
322: if (!lz) {
323: PetscMalloc1(p, &dd->lz);
324: lz = dd->lz;
325: for (i=0; i<p; i++) lz[i] = P/p + ((P % p) > (i % p));
326: }
327: z = lz[rank/(m*n)];
329: /* note this is different than x- and y-, as we will handle as an important special
330: case when p=P=1 and DM_BOUNDARY_PERIODIC and s > z. This is to deal with 2D problems
331: in a 3D code. Additional code for this case is noted with "2d case" comments */
332: twod = PETSC_FALSE;
333: if (P == 1) twod = PETSC_TRUE;
334: 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);
335: zs = 0;
336: for (i=0; i<(rank/(m*n)); i++) zs += lz[i];
337: ye = ys + y;
338: xe = xs + x;
339: ze = zs + z;
341: /* determine ghost region (Xs) and region scattered into (IXs) */
342: if (xs-s > 0) {
343: Xs = xs - s; IXs = xs - s;
344: } else {
345: if (bx) Xs = xs - s;
346: else Xs = 0;
347: IXs = 0;
348: }
349: if (xe+s <= M) {
350: Xe = xe + s; IXe = xe + s;
351: } else {
352: if (bx) {
353: Xs = xs - s; Xe = xe + s;
354: } else Xe = M;
355: IXe = M;
356: }
358: if (bx == DM_BOUNDARY_PERIODIC || bx == DM_BOUNDARY_MIRROR) {
359: IXs = xs - s;
360: IXe = xe + s;
361: Xs = xs - s;
362: Xe = xe + s;
363: }
365: if (ys-s > 0) {
366: Ys = ys - s; IYs = ys - s;
367: } else {
368: if (by) Ys = ys - s;
369: else Ys = 0;
370: IYs = 0;
371: }
372: if (ye+s <= N) {
373: Ye = ye + s; IYe = ye + s;
374: } else {
375: if (by) Ye = ye + s;
376: else Ye = N;
377: IYe = N;
378: }
380: if (by == DM_BOUNDARY_PERIODIC || by == DM_BOUNDARY_MIRROR) {
381: IYs = ys - s;
382: IYe = ye + s;
383: Ys = ys - s;
384: Ye = ye + s;
385: }
387: if (zs-s > 0) {
388: Zs = zs - s; IZs = zs - s;
389: } else {
390: if (bz) Zs = zs - s;
391: else Zs = 0;
392: IZs = 0;
393: }
394: if (ze+s <= P) {
395: Ze = ze + s; IZe = ze + s;
396: } else {
397: if (bz) Ze = ze + s;
398: else Ze = P;
399: IZe = P;
400: }
402: if (bz == DM_BOUNDARY_PERIODIC || bz == DM_BOUNDARY_MIRROR) {
403: IZs = zs - s;
404: IZe = ze + s;
405: Zs = zs - s;
406: Ze = ze + s;
407: }
409: /* Resize all X parameters to reflect w */
410: s_x = s;
411: s_y = s;
412: s_z = s;
414: /* determine starting point of each processor */
415: nn = x*y*z;
416: PetscMalloc2(size+1,&bases,size,&ldims);
417: MPI_Allgather(&nn,1,MPIU_INT,ldims,1,MPIU_INT,comm);
418: bases[0] = 0;
419: for (i=1; i<=size; i++) bases[i] = ldims[i-1];
420: for (i=1; i<=size; i++) bases[i] += bases[i-1];
421: base = bases[rank]*dof;
423: /* allocate the base parallel and sequential vectors */
424: dd->Nlocal = x*y*z*dof;
425: VecCreateMPIWithArray(comm,dof,dd->Nlocal,PETSC_DECIDE,NULL,&global);
426: dd->nlocal = (Xe-Xs)*(Ye-Ys)*(Ze-Zs)*dof;
427: VecCreateSeqWithArray(PETSC_COMM_SELF,dof,dd->nlocal,NULL,&local);
429: /* generate appropriate vector scatters */
430: /* local to global inserts non-ghost point region into global */
431: PetscMalloc1((IXe-IXs)*(IYe-IYs)*(IZe-IZs),&idx);
432: left = xs - Xs; right = left + x;
433: bottom = ys - Ys; top = bottom + y;
434: down = zs - Zs; up = down + z;
435: count = 0;
436: for (i=down; i<up; i++) {
437: for (j=bottom; j<top; j++) {
438: for (k=left; k<right; k++) {
439: idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
440: }
441: }
442: }
444: /* global to local must include ghost points within the domain,
445: but not ghost points outside the domain that aren't periodic */
446: if (stencil_type == DMDA_STENCIL_BOX) {
447: left = IXs - Xs; right = left + (IXe-IXs);
448: bottom = IYs - Ys; top = bottom + (IYe-IYs);
449: down = IZs - Zs; up = down + (IZe-IZs);
450: count = 0;
451: for (i=down; i<up; i++) {
452: for (j=bottom; j<top; j++) {
453: for (k=left; k<right; k++) {
454: idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
455: }
456: }
457: }
458: ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&to);
459: } else {
460: /* This is way ugly! We need to list the funny cross type region */
461: left = xs - Xs; right = left + x;
462: bottom = ys - Ys; top = bottom + y;
463: down = zs - Zs; up = down + z;
464: count = 0;
465: /* the bottom chunck */
466: for (i=(IZs-Zs); i<down; i++) {
467: for (j=bottom; j<top; j++) {
468: for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
469: }
470: }
471: /* the middle piece */
472: for (i=down; i<up; i++) {
473: /* front */
474: for (j=(IYs-Ys); j<bottom; j++) {
475: for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
476: }
477: /* middle */
478: for (j=bottom; j<top; j++) {
479: for (k=IXs-Xs; k<IXe-Xs; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
480: }
481: /* back */
482: for (j=top; j<top+IYe-ye; j++) {
483: for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
484: }
485: }
486: /* the top piece */
487: for (i=up; i<up+IZe-ze; i++) {
488: for (j=bottom; j<top; j++) {
489: for (k=left; k<right; k++) idx[count++] = (i*(Ye-Ys) + j)*(Xe-Xs) + k;
490: }
491: }
492: ISCreateBlock(comm,dof,count,idx,PETSC_OWN_POINTER,&to);
493: }
495: /* determine who lies on each side of use stored in n24 n25 n26
496: n21 n22 n23
497: n18 n19 n20
499: n15 n16 n17
500: n12 n14
501: n9 n10 n11
503: n6 n7 n8
504: n3 n4 n5
505: n0 n1 n2
506: */
508: /* Solve for X,Y, and Z Periodic Case First, Then Modify Solution */
509: /* Assume Nodes are Internal to the Cube */
510: n0 = rank - m*n - m - 1;
511: n1 = rank - m*n - m;
512: n2 = rank - m*n - m + 1;
513: n3 = rank - m*n -1;
514: n4 = rank - m*n;
515: n5 = rank - m*n + 1;
516: n6 = rank - m*n + m - 1;
517: n7 = rank - m*n + m;
518: n8 = rank - m*n + m + 1;
520: n9 = rank - m - 1;
521: n10 = rank - m;
522: n11 = rank - m + 1;
523: n12 = rank - 1;
524: n14 = rank + 1;
525: n15 = rank + m - 1;
526: n16 = rank + m;
527: n17 = rank + m + 1;
529: n18 = rank + m*n - m - 1;
530: n19 = rank + m*n - m;
531: n20 = rank + m*n - m + 1;
532: n21 = rank + m*n - 1;
533: n22 = rank + m*n;
534: n23 = rank + m*n + 1;
535: n24 = rank + m*n + m - 1;
536: n25 = rank + m*n + m;
537: n26 = rank + m*n + m + 1;
539: /* Assume Pieces are on Faces of Cube */
541: if (xs == 0) { /* First assume not corner or edge */
542: n0 = rank -1 - (m*n);
543: n3 = rank + m -1 - (m*n);
544: n6 = rank + 2*m -1 - (m*n);
545: n9 = rank -1;
546: n12 = rank + m -1;
547: n15 = rank + 2*m -1;
548: n18 = rank -1 + (m*n);
549: n21 = rank + m -1 + (m*n);
550: n24 = rank + 2*m -1 + (m*n);
551: }
553: if (xe == M) { /* First assume not corner or edge */
554: n2 = rank -2*m +1 - (m*n);
555: n5 = rank - m +1 - (m*n);
556: n8 = rank +1 - (m*n);
557: n11 = rank -2*m +1;
558: n14 = rank - m +1;
559: n17 = rank +1;
560: n20 = rank -2*m +1 + (m*n);
561: n23 = rank - m +1 + (m*n);
562: n26 = rank +1 + (m*n);
563: }
565: if (ys==0) { /* First assume not corner or edge */
566: n0 = rank + m * (n-1) -1 - (m*n);
567: n1 = rank + m * (n-1) - (m*n);
568: n2 = rank + m * (n-1) +1 - (m*n);
569: n9 = rank + m * (n-1) -1;
570: n10 = rank + m * (n-1);
571: n11 = rank + m * (n-1) +1;
572: n18 = rank + m * (n-1) -1 + (m*n);
573: n19 = rank + m * (n-1) + (m*n);
574: n20 = rank + m * (n-1) +1 + (m*n);
575: }
577: if (ye == N) { /* First assume not corner or edge */
578: n6 = rank - m * (n-1) -1 - (m*n);
579: n7 = rank - m * (n-1) - (m*n);
580: n8 = rank - m * (n-1) +1 - (m*n);
581: n15 = rank - m * (n-1) -1;
582: n16 = rank - m * (n-1);
583: n17 = rank - m * (n-1) +1;
584: n24 = rank - m * (n-1) -1 + (m*n);
585: n25 = rank - m * (n-1) + (m*n);
586: n26 = rank - m * (n-1) +1 + (m*n);
587: }
589: if (zs == 0) { /* First assume not corner or edge */
590: n0 = size - (m*n) + rank - m - 1;
591: n1 = size - (m*n) + rank - m;
592: n2 = size - (m*n) + rank - m + 1;
593: n3 = size - (m*n) + rank - 1;
594: n4 = size - (m*n) + rank;
595: n5 = size - (m*n) + rank + 1;
596: n6 = size - (m*n) + rank + m - 1;
597: n7 = size - (m*n) + rank + m;
598: n8 = size - (m*n) + rank + m + 1;
599: }
601: if (ze == P) { /* First assume not corner or edge */
602: n18 = (m*n) - (size-rank) - m - 1;
603: n19 = (m*n) - (size-rank) - m;
604: n20 = (m*n) - (size-rank) - m + 1;
605: n21 = (m*n) - (size-rank) - 1;
606: n22 = (m*n) - (size-rank);
607: n23 = (m*n) - (size-rank) + 1;
608: n24 = (m*n) - (size-rank) + m - 1;
609: n25 = (m*n) - (size-rank) + m;
610: n26 = (m*n) - (size-rank) + m + 1;
611: }
613: if ((xs==0) && (zs==0)) { /* Assume an edge, not corner */
614: n0 = size - m*n + rank + m-1 - m;
615: n3 = size - m*n + rank + m-1;
616: n6 = size - m*n + rank + m-1 + m;
617: }
619: if ((xs==0) && (ze==P)) { /* Assume an edge, not corner */
620: n18 = m*n - (size - rank) + m-1 - m;
621: n21 = m*n - (size - rank) + m-1;
622: n24 = m*n - (size - rank) + m-1 + m;
623: }
625: if ((xs==0) && (ys==0)) { /* Assume an edge, not corner */
626: n0 = rank + m*n -1 - m*n;
627: n9 = rank + m*n -1;
628: n18 = rank + m*n -1 + m*n;
629: }
631: if ((xs==0) && (ye==N)) { /* Assume an edge, not corner */
632: n6 = rank - m*(n-1) + m-1 - m*n;
633: n15 = rank - m*(n-1) + m-1;
634: n24 = rank - m*(n-1) + m-1 + m*n;
635: }
637: if ((xe==M) && (zs==0)) { /* Assume an edge, not corner */
638: n2 = size - (m*n-rank) - (m-1) - m;
639: n5 = size - (m*n-rank) - (m-1);
640: n8 = size - (m*n-rank) - (m-1) + m;
641: }
643: if ((xe==M) && (ze==P)) { /* Assume an edge, not corner */
644: n20 = m*n - (size - rank) - (m-1) - m;
645: n23 = m*n - (size - rank) - (m-1);
646: n26 = m*n - (size - rank) - (m-1) + m;
647: }
649: if ((xe==M) && (ys==0)) { /* Assume an edge, not corner */
650: n2 = rank + m*(n-1) - (m-1) - m*n;
651: n11 = rank + m*(n-1) - (m-1);
652: n20 = rank + m*(n-1) - (m-1) + m*n;
653: }
655: if ((xe==M) && (ye==N)) { /* Assume an edge, not corner */
656: n8 = rank - m*n +1 - m*n;
657: n17 = rank - m*n +1;
658: n26 = rank - m*n +1 + m*n;
659: }
661: if ((ys==0) && (zs==0)) { /* Assume an edge, not corner */
662: n0 = size - m + rank -1;
663: n1 = size - m + rank;
664: n2 = size - m + rank +1;
665: }
667: if ((ys==0) && (ze==P)) { /* Assume an edge, not corner */
668: n18 = m*n - (size - rank) + m*(n-1) -1;
669: n19 = m*n - (size - rank) + m*(n-1);
670: n20 = m*n - (size - rank) + m*(n-1) +1;
671: }
673: if ((ye==N) && (zs==0)) { /* Assume an edge, not corner */
674: n6 = size - (m*n-rank) - m * (n-1) -1;
675: n7 = size - (m*n-rank) - m * (n-1);
676: n8 = size - (m*n-rank) - m * (n-1) +1;
677: }
679: if ((ye==N) && (ze==P)) { /* Assume an edge, not corner */
680: n24 = rank - (size-m) -1;
681: n25 = rank - (size-m);
682: n26 = rank - (size-m) +1;
683: }
685: /* Check for Corners */
686: if ((xs==0) && (ys==0) && (zs==0)) n0 = size -1;
687: if ((xs==0) && (ys==0) && (ze==P)) n18 = m*n-1;
688: if ((xs==0) && (ye==N) && (zs==0)) n6 = (size-1)-m*(n-1);
689: if ((xs==0) && (ye==N) && (ze==P)) n24 = m-1;
690: if ((xe==M) && (ys==0) && (zs==0)) n2 = size-m;
691: if ((xe==M) && (ys==0) && (ze==P)) n20 = m*n-m;
692: if ((xe==M) && (ye==N) && (zs==0)) n8 = size-m*n;
693: if ((xe==M) && (ye==N) && (ze==P)) n26 = 0;
695: /* Check for when not X,Y, and Z Periodic */
697: /* If not X periodic */
698: if (bx != DM_BOUNDARY_PERIODIC) {
699: if (xs==0) n0 = n3 = n6 = n9 = n12 = n15 = n18 = n21 = n24 = -2;
700: if (xe==M) n2 = n5 = n8 = n11 = n14 = n17 = n20 = n23 = n26 = -2;
701: }
703: /* If not Y periodic */
704: if (by != DM_BOUNDARY_PERIODIC) {
705: if (ys==0) n0 = n1 = n2 = n9 = n10 = n11 = n18 = n19 = n20 = -2;
706: if (ye==N) n6 = n7 = n8 = n15 = n16 = n17 = n24 = n25 = n26 = -2;
707: }
709: /* If not Z periodic */
710: if (bz != DM_BOUNDARY_PERIODIC) {
711: if (zs==0) n0 = n1 = n2 = n3 = n4 = n5 = n6 = n7 = n8 = -2;
712: if (ze==P) n18 = n19 = n20 = n21 = n22 = n23 = n24 = n25 = n26 = -2;
713: }
715: PetscMalloc1(27,&dd->neighbors);
717: dd->neighbors[0] = n0;
718: dd->neighbors[1] = n1;
719: dd->neighbors[2] = n2;
720: dd->neighbors[3] = n3;
721: dd->neighbors[4] = n4;
722: dd->neighbors[5] = n5;
723: dd->neighbors[6] = n6;
724: dd->neighbors[7] = n7;
725: dd->neighbors[8] = n8;
726: dd->neighbors[9] = n9;
727: dd->neighbors[10] = n10;
728: dd->neighbors[11] = n11;
729: dd->neighbors[12] = n12;
730: dd->neighbors[13] = rank;
731: dd->neighbors[14] = n14;
732: dd->neighbors[15] = n15;
733: dd->neighbors[16] = n16;
734: dd->neighbors[17] = n17;
735: dd->neighbors[18] = n18;
736: dd->neighbors[19] = n19;
737: dd->neighbors[20] = n20;
738: dd->neighbors[21] = n21;
739: dd->neighbors[22] = n22;
740: dd->neighbors[23] = n23;
741: dd->neighbors[24] = n24;
742: dd->neighbors[25] = n25;
743: dd->neighbors[26] = n26;
745: /* If star stencil then delete the corner neighbors */
746: if (stencil_type == DMDA_STENCIL_STAR) {
747: /* save information about corner neighbors */
748: sn0 = n0; sn1 = n1; sn2 = n2; sn3 = n3; sn5 = n5; sn6 = n6; sn7 = n7;
749: sn8 = n8; sn9 = n9; sn11 = n11; sn15 = n15; sn17 = n17; sn18 = n18;
750: sn19 = n19; sn20 = n20; sn21 = n21; sn23 = n23; sn24 = n24; sn25 = n25;
751: sn26 = n26;
752: n0 = n1 = n2 = n3 = n5 = n6 = n7 = n8 = n9 = n11 = n15 = n17 = n18 = n19 = n20 = n21 = n23 = n24 = n25 = n26 = -1;
753: }
755: PetscMalloc1((Xe-Xs)*(Ye-Ys)*(Ze-Zs),&idx);
757: nn = 0;
758: /* Bottom Level */
759: for (k=0; k<s_z; k++) {
760: for (i=1; i<=s_y; i++) {
761: if (n0 >= 0) { /* left below */
762: x_t = lx[n0 % m];
763: y_t = ly[(n0 % (m*n))/m];
764: z_t = lz[n0 / (m*n)];
765: 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;
766: if (twod && (s_t < 0)) s_t = bases[n0] + x_t*y_t*z_t - (s_y-i)*x_t - s_x; /* 2D case */
767: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
768: }
769: if (n1 >= 0) { /* directly below */
770: x_t = x;
771: y_t = ly[(n1 % (m*n))/m];
772: z_t = lz[n1 / (m*n)];
773: s_t = bases[n1] + x_t*y_t*z_t - (s_y+1-i)*x_t - (s_z-k-1)*x_t*y_t;
774: if (twod && (s_t < 0)) s_t = bases[n1] + x_t*y_t*z_t - (s_y+1-i)*x_t; /* 2D case */
775: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
776: }
777: if (n2 >= 0) { /* right below */
778: x_t = lx[n2 % m];
779: y_t = ly[(n2 % (m*n))/m];
780: z_t = lz[n2 / (m*n)];
781: s_t = bases[n2] + x_t*y_t*z_t - (s_y+1-i)*x_t - (s_z-k-1)*x_t*y_t;
782: if (twod && (s_t < 0)) s_t = bases[n2] + x_t*y_t*z_t - (s_y+1-i)*x_t; /* 2D case */
783: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
784: }
785: }
787: for (i=0; i<y; i++) {
788: if (n3 >= 0) { /* directly left */
789: x_t = lx[n3 % m];
790: y_t = y;
791: z_t = lz[n3 / (m*n)];
792: s_t = bases[n3] + (i+1)*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
793: 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 */
794: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
795: }
797: if (n4 >= 0) { /* middle */
798: x_t = x;
799: y_t = y;
800: z_t = lz[n4 / (m*n)];
801: s_t = bases[n4] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
802: if (twod && (s_t < 0)) s_t = bases[n4] + i*x_t + x_t*y_t*z_t - x_t*y_t; /* 2D case */
803: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
804: } else if (bz == DM_BOUNDARY_MIRROR) {
805: for (j=0; j<x; j++) idx[nn++] = 0;
806: }
808: if (n5 >= 0) { /* directly right */
809: x_t = lx[n5 % m];
810: y_t = y;
811: z_t = lz[n5 / (m*n)];
812: s_t = bases[n5] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
813: if (twod && (s_t < 0)) s_t = bases[n5] + i*x_t + x_t*y_t*z_t - x_t*y_t; /* 2D case */
814: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
815: }
816: }
818: for (i=1; i<=s_y; i++) {
819: if (n6 >= 0) { /* left above */
820: x_t = lx[n6 % m];
821: y_t = ly[(n6 % (m*n))/m];
822: z_t = lz[n6 / (m*n)];
823: s_t = bases[n6] + i*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
824: 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 */
825: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
826: }
827: if (n7 >= 0) { /* directly above */
828: x_t = x;
829: y_t = ly[(n7 % (m*n))/m];
830: z_t = lz[n7 / (m*n)];
831: s_t = bases[n7] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
832: 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 */
833: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
834: }
835: if (n8 >= 0) { /* right above */
836: x_t = lx[n8 % m];
837: y_t = ly[(n8 % (m*n))/m];
838: z_t = lz[n8 / (m*n)];
839: s_t = bases[n8] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
840: 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 */
841: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
842: }
843: }
844: }
846: /* Middle Level */
847: for (k=0; k<z; k++) {
848: for (i=1; i<=s_y; i++) {
849: if (n9 >= 0) { /* left below */
850: x_t = lx[n9 % m];
851: y_t = ly[(n9 % (m*n))/m];
852: /* z_t = z; */
853: s_t = bases[n9] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
854: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
855: }
856: if (n10 >= 0) { /* directly below */
857: x_t = x;
858: y_t = ly[(n10 % (m*n))/m];
859: /* z_t = z; */
860: s_t = bases[n10] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
861: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
862: } else if (by == DM_BOUNDARY_MIRROR) {
863: for (j=0; j<x; j++) idx[nn++] = 0;
864: }
865: if (n11 >= 0) { /* right below */
866: x_t = lx[n11 % m];
867: y_t = ly[(n11 % (m*n))/m];
868: /* z_t = z; */
869: s_t = bases[n11] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
870: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
871: }
872: }
874: for (i=0; i<y; i++) {
875: if (n12 >= 0) { /* directly left */
876: x_t = lx[n12 % m];
877: y_t = y;
878: /* z_t = z; */
879: s_t = bases[n12] + (i+1)*x_t - s_x + k*x_t*y_t;
880: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
881: } else if (bx == DM_BOUNDARY_MIRROR) {
882: for (j=0; j<s_x; j++) idx[nn++] = 0;
883: }
885: /* Interior */
886: s_t = bases[rank] + i*x + k*x*y;
887: for (j=0; j<x; j++) idx[nn++] = s_t++;
889: if (n14 >= 0) { /* directly right */
890: x_t = lx[n14 % m];
891: y_t = y;
892: /* z_t = z; */
893: s_t = bases[n14] + i*x_t + k*x_t*y_t;
894: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
895: } else if (bx == DM_BOUNDARY_MIRROR) {
896: for (j=0; j<s_x; j++) idx[nn++] = 0;
897: }
898: }
900: for (i=1; i<=s_y; i++) {
901: if (n15 >= 0) { /* left above */
902: x_t = lx[n15 % m];
903: y_t = ly[(n15 % (m*n))/m];
904: /* z_t = z; */
905: s_t = bases[n15] + i*x_t - s_x + k*x_t*y_t;
906: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
907: }
908: if (n16 >= 0) { /* directly above */
909: x_t = x;
910: y_t = ly[(n16 % (m*n))/m];
911: /* z_t = z; */
912: s_t = bases[n16] + (i-1)*x_t + k*x_t*y_t;
913: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
914: } else if (by == DM_BOUNDARY_MIRROR) {
915: for (j=0; j<x; j++) idx[nn++] = 0;
916: }
917: if (n17 >= 0) { /* right above */
918: x_t = lx[n17 % m];
919: y_t = ly[(n17 % (m*n))/m];
920: /* z_t = z; */
921: s_t = bases[n17] + (i-1)*x_t + k*x_t*y_t;
922: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
923: }
924: }
925: }
927: /* Upper Level */
928: for (k=0; k<s_z; k++) {
929: for (i=1; i<=s_y; i++) {
930: if (n18 >= 0) { /* left below */
931: x_t = lx[n18 % m];
932: y_t = ly[(n18 % (m*n))/m];
933: /* z_t = lz[n18 / (m*n)]; */
934: s_t = bases[n18] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
935: if (twod && (s_t >= M*N*P)) s_t = bases[n18] - (s_y-i)*x_t -s_x + x_t*y_t; /* 2d case */
936: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
937: }
938: if (n19 >= 0) { /* directly below */
939: x_t = x;
940: y_t = ly[(n19 % (m*n))/m];
941: /* z_t = lz[n19 / (m*n)]; */
942: s_t = bases[n19] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
943: if (twod && (s_t >= M*N*P)) s_t = bases[n19] - (s_y+1-i)*x_t + x_t*y_t; /* 2d case */
944: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
945: }
946: if (n20 >= 0) { /* right below */
947: x_t = lx[n20 % m];
948: y_t = ly[(n20 % (m*n))/m];
949: /* z_t = lz[n20 / (m*n)]; */
950: s_t = bases[n20] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
951: if (twod && (s_t >= M*N*P)) s_t = bases[n20] - (s_y+1-i)*x_t + x_t*y_t; /* 2d case */
952: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
953: }
954: }
956: for (i=0; i<y; i++) {
957: if (n21 >= 0) { /* directly left */
958: x_t = lx[n21 % m];
959: y_t = y;
960: /* z_t = lz[n21 / (m*n)]; */
961: s_t = bases[n21] + (i+1)*x_t - s_x + k*x_t*y_t;
962: if (twod && (s_t >= M*N*P)) s_t = bases[n21] + (i+1)*x_t - s_x; /* 2d case */
963: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
964: }
966: if (n22 >= 0) { /* middle */
967: x_t = x;
968: y_t = y;
969: /* z_t = lz[n22 / (m*n)]; */
970: s_t = bases[n22] + i*x_t + k*x_t*y_t;
971: if (twod && (s_t >= M*N*P)) s_t = bases[n22] + i*x_t; /* 2d case */
972: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
973: } else if (bz == DM_BOUNDARY_MIRROR) {
974: for (j=0; j<x; j++) idx[nn++] = 0;
975: }
977: if (n23 >= 0) { /* directly right */
978: x_t = lx[n23 % m];
979: y_t = y;
980: /* z_t = lz[n23 / (m*n)]; */
981: s_t = bases[n23] + i*x_t + k*x_t*y_t;
982: if (twod && (s_t >= M*N*P)) s_t = bases[n23] + i*x_t; /* 2d case */
983: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
984: }
985: }
987: for (i=1; i<=s_y; i++) {
988: if (n24 >= 0) { /* left above */
989: x_t = lx[n24 % m];
990: y_t = ly[(n24 % (m*n))/m];
991: /* z_t = lz[n24 / (m*n)]; */
992: s_t = bases[n24] + i*x_t - s_x + k*x_t*y_t;
993: if (twod && (s_t >= M*N*P)) s_t = bases[n24] + i*x_t - s_x; /* 2d case */
994: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
995: }
996: if (n25 >= 0) { /* directly above */
997: x_t = x;
998: y_t = ly[(n25 % (m*n))/m];
999: /* z_t = lz[n25 / (m*n)]; */
1000: s_t = bases[n25] + (i-1)*x_t + k*x_t*y_t;
1001: if (twod && (s_t >= M*N*P)) s_t = bases[n25] + (i-1)*x_t; /* 2d case */
1002: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1003: }
1004: if (n26 >= 0) { /* right above */
1005: x_t = lx[n26 % m];
1006: y_t = ly[(n26 % (m*n))/m];
1007: /* z_t = lz[n26 / (m*n)]; */
1008: s_t = bases[n26] + (i-1)*x_t + k*x_t*y_t;
1009: if (twod && (s_t >= M*N*P)) s_t = bases[n26] + (i-1)*x_t; /* 2d case */
1010: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1011: }
1012: }
1013: }
1015: ISCreateBlock(comm,dof,nn,idx,PETSC_USE_POINTER,&from);
1016: VecScatterCreate(global,from,local,to,>ol);
1017: PetscLogObjectParent((PetscObject)da,(PetscObject)gtol);
1018: ISDestroy(&to);
1019: ISDestroy(&from);
1021: if (stencil_type == DMDA_STENCIL_STAR) {
1022: n0 = sn0; n1 = sn1; n2 = sn2; n3 = sn3; n5 = sn5; n6 = sn6; n7 = sn7;
1023: n8 = sn8; n9 = sn9; n11 = sn11; n15 = sn15; n17 = sn17; n18 = sn18;
1024: n19 = sn19; n20 = sn20; n21 = sn21; n23 = sn23; n24 = sn24; n25 = sn25;
1025: n26 = sn26;
1026: }
1028: if (((stencil_type == DMDA_STENCIL_STAR) ||
1029: (bx != DM_BOUNDARY_PERIODIC && bx) ||
1030: (by != DM_BOUNDARY_PERIODIC && by) ||
1031: (bz != DM_BOUNDARY_PERIODIC && bz))) {
1032: /*
1033: Recompute the local to global mappings, this time keeping the
1034: information about the cross corner processor numbers.
1035: */
1036: nn = 0;
1037: /* Bottom Level */
1038: for (k=0; k<s_z; k++) {
1039: for (i=1; i<=s_y; i++) {
1040: if (n0 >= 0) { /* left below */
1041: x_t = lx[n0 % m];
1042: y_t = ly[(n0 % (m*n))/m];
1043: z_t = lz[n0 / (m*n)];
1044: 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;
1045: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1046: } else if (Xs-xs < 0 && Ys-ys < 0 && Zs-zs < 0) {
1047: for (j=0; j<s_x; j++) idx[nn++] = -1;
1048: }
1049: if (n1 >= 0) { /* directly below */
1050: x_t = x;
1051: y_t = ly[(n1 % (m*n))/m];
1052: z_t = lz[n1 / (m*n)];
1053: s_t = bases[n1] + x_t*y_t*z_t - (s_y+1-i)*x_t - (s_z-k-1)*x_t*y_t;
1054: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1055: } else if (Ys-ys < 0 && Zs-zs < 0) {
1056: for (j=0; j<x; j++) idx[nn++] = -1;
1057: }
1058: if (n2 >= 0) { /* right below */
1059: x_t = lx[n2 % m];
1060: y_t = ly[(n2 % (m*n))/m];
1061: z_t = lz[n2 / (m*n)];
1062: s_t = bases[n2] + x_t*y_t*z_t - (s_y+1-i)*x_t - (s_z-k-1)*x_t*y_t;
1063: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1064: } else if (xe-Xe < 0 && Ys-ys < 0 && Zs-zs < 0) {
1065: for (j=0; j<s_x; j++) idx[nn++] = -1;
1066: }
1067: }
1069: for (i=0; i<y; i++) {
1070: if (n3 >= 0) { /* directly left */
1071: x_t = lx[n3 % m];
1072: y_t = y;
1073: z_t = lz[n3 / (m*n)];
1074: s_t = bases[n3] + (i+1)*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
1075: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1076: } else if (Xs-xs < 0 && Zs-zs < 0) {
1077: for (j=0; j<s_x; j++) idx[nn++] = -1;
1078: }
1080: if (n4 >= 0) { /* middle */
1081: x_t = x;
1082: y_t = y;
1083: z_t = lz[n4 / (m*n)];
1084: s_t = bases[n4] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
1085: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1086: } else if (Zs-zs < 0) {
1087: if (bz == DM_BOUNDARY_MIRROR) {
1088: for (j=0; j<x; j++) idx[nn++] = 0;
1089: } else {
1090: for (j=0; j<x; j++) idx[nn++] = -1;
1091: }
1092: }
1094: if (n5 >= 0) { /* directly right */
1095: x_t = lx[n5 % m];
1096: y_t = y;
1097: z_t = lz[n5 / (m*n)];
1098: s_t = bases[n5] + i*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
1099: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1100: } else if (xe-Xe < 0 && Zs-zs < 0) {
1101: for (j=0; j<s_x; j++) idx[nn++] = -1;
1102: }
1103: }
1105: for (i=1; i<=s_y; i++) {
1106: if (n6 >= 0) { /* left above */
1107: x_t = lx[n6 % m];
1108: y_t = ly[(n6 % (m*n))/m];
1109: z_t = lz[n6 / (m*n)];
1110: s_t = bases[n6] + i*x_t - s_x + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
1111: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1112: } else if (Xs-xs < 0 && ye-Ye < 0 && Zs-zs < 0) {
1113: for (j=0; j<s_x; j++) idx[nn++] = -1;
1114: }
1115: if (n7 >= 0) { /* directly above */
1116: x_t = x;
1117: y_t = ly[(n7 % (m*n))/m];
1118: z_t = lz[n7 / (m*n)];
1119: s_t = bases[n7] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
1120: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1121: } else if (ye-Ye < 0 && Zs-zs < 0) {
1122: for (j=0; j<x; j++) idx[nn++] = -1;
1123: }
1124: if (n8 >= 0) { /* right above */
1125: x_t = lx[n8 % m];
1126: y_t = ly[(n8 % (m*n))/m];
1127: z_t = lz[n8 / (m*n)];
1128: s_t = bases[n8] + (i-1)*x_t + x_t*y_t*z_t - (s_z-k)*x_t*y_t;
1129: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1130: } else if (xe-Xe < 0 && ye-Ye < 0 && Zs-zs < 0) {
1131: for (j=0; j<s_x; j++) idx[nn++] = -1;
1132: }
1133: }
1134: }
1136: /* Middle Level */
1137: for (k=0; k<z; k++) {
1138: for (i=1; i<=s_y; i++) {
1139: if (n9 >= 0) { /* left below */
1140: x_t = lx[n9 % m];
1141: y_t = ly[(n9 % (m*n))/m];
1142: /* z_t = z; */
1143: s_t = bases[n9] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
1144: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1145: } else if (Xs-xs < 0 && Ys-ys < 0) {
1146: for (j=0; j<s_x; j++) idx[nn++] = -1;
1147: }
1148: if (n10 >= 0) { /* directly below */
1149: x_t = x;
1150: y_t = ly[(n10 % (m*n))/m];
1151: /* z_t = z; */
1152: s_t = bases[n10] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
1153: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1154: } else if (Ys-ys < 0) {
1155: if (by == DM_BOUNDARY_MIRROR) {
1156: for (j=0; j<x; j++) idx[nn++] = -1;
1157: } else {
1158: for (j=0; j<x; j++) idx[nn++] = -1;
1159: }
1160: }
1161: if (n11 >= 0) { /* right below */
1162: x_t = lx[n11 % m];
1163: y_t = ly[(n11 % (m*n))/m];
1164: /* z_t = z; */
1165: s_t = bases[n11] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
1166: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1167: } else if (xe-Xe < 0 && Ys-ys < 0) {
1168: for (j=0; j<s_x; j++) idx[nn++] = -1;
1169: }
1170: }
1172: for (i=0; i<y; i++) {
1173: if (n12 >= 0) { /* directly left */
1174: x_t = lx[n12 % m];
1175: y_t = y;
1176: /* z_t = z; */
1177: s_t = bases[n12] + (i+1)*x_t - s_x + k*x_t*y_t;
1178: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1179: } else if (Xs-xs < 0) {
1180: if (bx == DM_BOUNDARY_MIRROR) {
1181: for (j=0; j<s_x; j++) idx[nn++] = 0;
1182: } else {
1183: for (j=0; j<s_x; j++) idx[nn++] = -1;
1184: }
1185: }
1187: /* Interior */
1188: s_t = bases[rank] + i*x + k*x*y;
1189: for (j=0; j<x; j++) idx[nn++] = s_t++;
1191: if (n14 >= 0) { /* directly right */
1192: x_t = lx[n14 % m];
1193: y_t = y;
1194: /* z_t = z; */
1195: s_t = bases[n14] + i*x_t + k*x_t*y_t;
1196: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1197: } else if (xe-Xe < 0) {
1198: if (bx == DM_BOUNDARY_MIRROR) {
1199: for (j=0; j<s_x; j++) idx[nn++] = 0;
1200: } else {
1201: for (j=0; j<s_x; j++) idx[nn++] = -1;
1202: }
1203: }
1204: }
1206: for (i=1; i<=s_y; i++) {
1207: if (n15 >= 0) { /* left above */
1208: x_t = lx[n15 % m];
1209: y_t = ly[(n15 % (m*n))/m];
1210: /* z_t = z; */
1211: s_t = bases[n15] + i*x_t - s_x + k*x_t*y_t;
1212: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1213: } else if (Xs-xs < 0 && ye-Ye < 0) {
1214: for (j=0; j<s_x; j++) idx[nn++] = -1;
1215: }
1216: if (n16 >= 0) { /* directly above */
1217: x_t = x;
1218: y_t = ly[(n16 % (m*n))/m];
1219: /* z_t = z; */
1220: s_t = bases[n16] + (i-1)*x_t + k*x_t*y_t;
1221: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1222: } else if (ye-Ye < 0) {
1223: if (by == DM_BOUNDARY_MIRROR) {
1224: for (j=0; j<x; j++) idx[nn++] = 0;
1225: } else {
1226: for (j=0; j<x; j++) idx[nn++] = -1;
1227: }
1228: }
1229: if (n17 >= 0) { /* right above */
1230: x_t = lx[n17 % m];
1231: y_t = ly[(n17 % (m*n))/m];
1232: /* z_t = z; */
1233: s_t = bases[n17] + (i-1)*x_t + k*x_t*y_t;
1234: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1235: } else if (xe-Xe < 0 && ye-Ye < 0) {
1236: for (j=0; j<s_x; j++) idx[nn++] = -1;
1237: }
1238: }
1239: }
1241: /* Upper Level */
1242: for (k=0; k<s_z; k++) {
1243: for (i=1; i<=s_y; i++) {
1244: if (n18 >= 0) { /* left below */
1245: x_t = lx[n18 % m];
1246: y_t = ly[(n18 % (m*n))/m];
1247: /* z_t = lz[n18 / (m*n)]; */
1248: s_t = bases[n18] - (s_y-i)*x_t -s_x + (k+1)*x_t*y_t;
1249: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1250: } else if (Xs-xs < 0 && Ys-ys < 0 && ze-Ze < 0) {
1251: for (j=0; j<s_x; j++) idx[nn++] = -1;
1252: }
1253: if (n19 >= 0) { /* directly below */
1254: x_t = x;
1255: y_t = ly[(n19 % (m*n))/m];
1256: /* z_t = lz[n19 / (m*n)]; */
1257: s_t = bases[n19] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
1258: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1259: } else if (Ys-ys < 0 && ze-Ze < 0) {
1260: for (j=0; j<x; j++) idx[nn++] = -1;
1261: }
1262: if (n20 >= 0) { /* right below */
1263: x_t = lx[n20 % m];
1264: y_t = ly[(n20 % (m*n))/m];
1265: /* z_t = lz[n20 / (m*n)]; */
1266: s_t = bases[n20] - (s_y+1-i)*x_t + (k+1)*x_t*y_t;
1267: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1268: } else if (xe-Xe < 0 && Ys-ys < 0 && ze-Ze < 0) {
1269: for (j=0; j<s_x; j++) idx[nn++] = -1;
1270: }
1271: }
1273: for (i=0; i<y; i++) {
1274: if (n21 >= 0) { /* directly left */
1275: x_t = lx[n21 % m];
1276: y_t = y;
1277: /* z_t = lz[n21 / (m*n)]; */
1278: s_t = bases[n21] + (i+1)*x_t - s_x + k*x_t*y_t;
1279: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1280: } else if (Xs-xs < 0 && ze-Ze < 0) {
1281: for (j=0; j<s_x; j++) idx[nn++] = -1;
1282: }
1284: if (n22 >= 0) { /* middle */
1285: x_t = x;
1286: y_t = y;
1287: /* z_t = lz[n22 / (m*n)]; */
1288: s_t = bases[n22] + i*x_t + k*x_t*y_t;
1289: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1290: } else if (ze-Ze < 0) {
1291: if (bz == DM_BOUNDARY_MIRROR) {
1292: for (j=0; j<x; j++) idx[nn++] = 0;
1293: } else {
1294: for (j=0; j<x; j++) idx[nn++] = -1;
1295: }
1296: }
1298: if (n23 >= 0) { /* directly right */
1299: x_t = lx[n23 % m];
1300: y_t = y;
1301: /* z_t = lz[n23 / (m*n)]; */
1302: s_t = bases[n23] + i*x_t + k*x_t*y_t;
1303: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1304: } else if (xe-Xe < 0 && ze-Ze < 0) {
1305: for (j=0; j<s_x; j++) idx[nn++] = -1;
1306: }
1307: }
1309: for (i=1; i<=s_y; i++) {
1310: if (n24 >= 0) { /* left above */
1311: x_t = lx[n24 % m];
1312: y_t = ly[(n24 % (m*n))/m];
1313: /* z_t = lz[n24 / (m*n)]; */
1314: s_t = bases[n24] + i*x_t - s_x + k*x_t*y_t;
1315: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1316: } else if (Xs-xs < 0 && ye-Ye < 0 && ze-Ze < 0) {
1317: for (j=0; j<s_x; j++) idx[nn++] = -1;
1318: }
1319: if (n25 >= 0) { /* directly above */
1320: x_t = x;
1321: y_t = ly[(n25 % (m*n))/m];
1322: /* z_t = lz[n25 / (m*n)]; */
1323: s_t = bases[n25] + (i-1)*x_t + k*x_t*y_t;
1324: for (j=0; j<x_t; j++) idx[nn++] = s_t++;
1325: } else if (ye-Ye < 0 && ze-Ze < 0) {
1326: for (j=0; j<x; j++) idx[nn++] = -1;
1327: }
1328: if (n26 >= 0) { /* right above */
1329: x_t = lx[n26 % m];
1330: y_t = ly[(n26 % (m*n))/m];
1331: /* z_t = lz[n26 / (m*n)]; */
1332: s_t = bases[n26] + (i-1)*x_t + k*x_t*y_t;
1333: for (j=0; j<s_x; j++) idx[nn++] = s_t++;
1334: } else if (xe-Xe < 0 && ye-Ye < 0 && ze-Ze < 0) {
1335: for (j=0; j<s_x; j++) idx[nn++] = -1;
1336: }
1337: }
1338: }
1339: }
1340: /*
1341: Set the local to global ordering in the global vector, this allows use
1342: of VecSetValuesLocal().
1343: */
1344: ISLocalToGlobalMappingCreate(comm,dof,nn,idx,PETSC_OWN_POINTER,&da->ltogmap);
1345: PetscLogObjectParent((PetscObject)da,(PetscObject)da->ltogmap);
1347: PetscFree2(bases,ldims);
1348: dd->m = m; dd->n = n; dd->p = p;
1349: /* note petsc expects xs/xe/Xs/Xe to be multiplied by #dofs in many places */
1350: dd->xs = xs*dof; dd->xe = xe*dof; dd->ys = ys; dd->ye = ye; dd->zs = zs; dd->ze = ze;
1351: dd->Xs = Xs*dof; dd->Xe = Xe*dof; dd->Ys = Ys; dd->Ye = Ye; dd->Zs = Zs; dd->Ze = Ze;
1353: VecDestroy(&local);
1354: VecDestroy(&global);
1356: dd->gtol = gtol;
1357: dd->base = base;
1358: da->ops->view = DMView_DA_3d;
1359: dd->ltol = NULL;
1360: dd->ao = NULL;
1361: return(0);
1362: }
1367: /*@C
1368: DMDACreate3d - Creates an object that will manage the communication of three-dimensional
1369: regular array data that is distributed across some processors.
1371: Collective on MPI_Comm
1373: Input Parameters:
1374: + comm - MPI communicator
1375: . bx,by,bz - type of ghost nodes the array have.
1376: Use one of DM_BOUNDARY_NONE, DM_BOUNDARY_GHOSTED, DM_BOUNDARY_PERIODIC.
1377: . stencil_type - Type of stencil (DMDA_STENCIL_STAR or DMDA_STENCIL_BOX)
1378: . M,N,P - global dimension in each direction of the array (use -M, -N, and or -P to indicate that it may be set to a different value
1379: from the command line with -da_grid_x <M> -da_grid_y <N> -da_grid_z <P>)
1380: . m,n,p - corresponding number of processors in each dimension
1381: (or PETSC_DECIDE to have calculated)
1382: . dof - number of degrees of freedom per node
1383: . s - stencil width
1384: - lx, ly, lz - arrays containing the number of nodes in each cell along
1385: the x, y, and z coordinates, or NULL. If non-null, these
1386: must be of length as m,n,p and the corresponding
1387: m,n, or p cannot be PETSC_DECIDE. Sum of the lx[] entries must be M, sum of
1388: the ly[] must N, sum of the lz[] must be P
1390: Output Parameter:
1391: . da - the resulting distributed array object
1393: Options Database Key:
1394: + -dm_view - Calls DMView() at the conclusion of DMDACreate3d()
1395: . -da_grid_x <nx> - number of grid points in x direction, if M < 0
1396: . -da_grid_y <ny> - number of grid points in y direction, if N < 0
1397: . -da_grid_z <nz> - number of grid points in z direction, if P < 0
1398: . -da_processors_x <MX> - number of processors in x direction
1399: . -da_processors_y <MY> - number of processors in y direction
1400: . -da_processors_z <MZ> - number of processors in z direction
1401: . -da_refine_x <rx> - refinement ratio in x direction
1402: . -da_refine_y <ry> - refinement ratio in y direction
1403: . -da_refine_z <rz>- refinement ratio in z directio
1404: - -da_refine <n> - refine the DMDA n times before creating it, , if M, N, or P < 0
1406: Level: beginner
1408: Notes:
1409: The stencil type DMDA_STENCIL_STAR with width 1 corresponds to the
1410: standard 7-pt stencil, while DMDA_STENCIL_BOX with width 1 denotes
1411: the standard 27-pt stencil.
1413: The array data itself is NOT stored in the DMDA, it is stored in Vec objects;
1414: The appropriate vector objects can be obtained with calls to DMCreateGlobalVector()
1415: and DMCreateLocalVector() and calls to VecDuplicate() if more are needed.
1417: .keywords: distributed array, create, three-dimensional
1419: .seealso: DMDestroy(), DMView(), DMDACreate1d(), DMDACreate2d(), DMGlobalToLocalBegin(), DMDAGetRefinementFactor(),
1420: DMGlobalToLocalEnd(), DMLocalToGlobalBegin(), DMLocalToLocalBegin(), DMLocalToLocalEnd(), DMDASetRefinementFactor(),
1421: DMDAGetInfo(), DMCreateGlobalVector(), DMCreateLocalVector(), DMDACreateNaturalVector(), DMLoad(), DMDAGetOwnershipRanges()
1423: @*/
1424: PetscErrorCode DMDACreate3d(MPI_Comm comm,DMBoundaryType bx,DMBoundaryType by,DMBoundaryType bz,DMDAStencilType stencil_type,PetscInt M,
1425: 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)
1426: {
1430: DMDACreate(comm, da);
1431: DMDASetDim(*da, 3);
1432: DMDASetSizes(*da, M, N, P);
1433: DMDASetNumProcs(*da, m, n, p);
1434: DMDASetBoundaryType(*da, bx, by, bz);
1435: DMDASetDof(*da, dof);
1436: DMDASetStencilType(*da, stencil_type);
1437: DMDASetStencilWidth(*da, s);
1438: DMDASetOwnershipRanges(*da, lx, ly, lz);
1439: /* This violates the behavior for other classes, but right now users expect negative dimensions to be handled this way */
1440: DMSetFromOptions(*da);
1441: DMSetUp(*da);
1442: return(0);
1443: }