source: soft/giet_vm/applications/ocean/slave2.C @ 776

Last change on this file since 776 was 598, checked in by guerin, 9 years ago

ocean: fix app broken by r589
File size: 40.1 KB
RevLine 
[598]1/*************************************************************************/
2/*                                                                       */
3/*  Copyright (c) 1994 Stanford University                               */
4/*                                                                       */
5/*  All rights reserved.                                                 */
6/*                                                                       */
7/*  Permission is given to use, copy, and modify this software for any   */
8/*  non-commercial purpose as long as this copyright notice is not       */
9/*  removed.  All other uses, including redistribution in whole or in    */
10/*  part, are forbidden without prior written permission.                */
11/*                                                                       */
12/*  This software is provided with absolutely no warranty and no         */
13/*  support.                                                             */
14/*                                                                       */
15/*************************************************************************/
[581]16
[598]17/*    ****************
18      subroutine slave2
19      ****************  */
20
21EXTERN_ENV
22
23#include <stdio.h>
24#include <math.h>
25#include <stdlib.h>
26
27#include "decs.h"
28
29void slave2(long procid, long firstrow, long lastrow, long numrows, long firstcol, long lastcol, long numcols)
30{
31    long i;
32    long j;
33    long iindex;
34    double hh1;
35    double hh3;
36    double hinv;
37    double h1inv;
38    long istart;
39    long iend;
40    long jstart;
41    long jend;
42    long ist;
43    long ien;
44    long jst;
45    long jen;
46    double ressqr;
47    double psiaipriv;
48    double f4;
49    double timst;
50    long psiindex;
51    long i_off;
52    long j_off;
53    long multi_start;
54    long multi_end;
55    double **t2a;
56    double **t2b;
57    double **t2c;
58    double **t2d;
59    double **t2e;
60    double **t2f;
61    double **t2g;
62    double **t2h;
63    double *t1a;
64    double *t1b;
65    double *t1c;
66    double *t1d;
67    double *t1e;
68    double *t1f;
69    double *t1g;
70    double *t1h;
71
72    ressqr = lev_res[numlev - 1] * lev_res[numlev - 1];
73    i_off = (*gp[procid].rownum) * numrows;
74    j_off = (*gp[procid].colnum) * numcols;
75
76    START_PHASE(procid, 1);
77
78/*   ***************************************************************
79
80          f i r s t     p h a s e   (of timestep calculation)
81
82     ***************************************************************/
83
84    t2a = (double **) ga[procid];
85    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
86        t2a[0][0] = 0.0;
87    }
88    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
89        t2a[im - 1][0] = 0.0;
90    }
91    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
92        t2a[0][jm - 1] = 0.0;
93    }
94    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
95        t2a[im - 1][jm - 1] = 0.0;
96    }
97    if (gp[procid].neighbors[UP] == -1) {
98        t1a = (double *) t2a[0];
99        for (j = firstcol; j <= lastcol; j++) {
100            t1a[j] = 0.0;
101        }
102    }
103    if (gp[procid].neighbors[DOWN] == -1) {
104        t1a = (double *) t2a[im - 1];
105        for (j = firstcol; j <= lastcol; j++) {
106            t1a[j] = 0.0;
107        }
108    }
109    if (gp[procid].neighbors[LEFT] == -1) {
110        for (j = firstrow; j <= lastrow; j++) {
111            t2a[j][0] = 0.0;
112        }
113    }
114    if (gp[procid].neighbors[RIGHT] == -1) {
115        for (j = firstrow; j <= lastrow; j++) {
116            t2a[j][jm - 1] = 0.0;
117        }
118    }
119    for (i = firstrow; i <= lastrow; i++) {
120        t1a = (double *) t2a[i];
121        for (iindex = firstcol; iindex <= lastcol; iindex++) {
122            t1a[iindex] = 0.0;
123        }
124    }
125
126    t2a = (double **) gb[procid];
127    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
128        t2a[0][0] = 0.0;
129    }
130    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
131        t2a[im - 1][0] = 0.0;
132    }
133    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
134        t2a[0][jm - 1] = 0.0;
135    }
136    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
137        t2a[im - 1][jm - 1] = 0.0;
138    }
139    if (gp[procid].neighbors[UP] == -1) {
140        t1a = (double *) t2a[0];
141        for (j = firstcol; j <= lastcol; j++) {
142            t1a[j] = 0.0;
143        }
144    }
145    if (gp[procid].neighbors[DOWN] == -1) {
146        t1a = (double *) t2a[im - 1];
147        for (j = firstcol; j <= lastcol; j++) {
148            t1a[j] = 0.0;
149        }
150    }
151    if (gp[procid].neighbors[LEFT] == -1) {
152        for (j = firstrow; j <= lastrow; j++) {
153            t2a[j][0] = 0.0;
154        }
155    }
156    if (gp[procid].neighbors[RIGHT] == -1) {
157        for (j = firstrow; j <= lastrow; j++) {
158            t2a[j][jm - 1] = 0.0;
159        }
160    }
161    for (i = firstrow; i <= lastrow; i++) {
162        t1a = (double *) t2a[i];
163        for (iindex = firstcol; iindex <= lastcol; iindex++) {
164            t1a[iindex] = 0.0;
165        }
166    }
167
168/* put the laplacian of psi{1,3} in work1{1,2}
169   note that psi(i,j,2) represents the psi3 array in
170   the original equations  */
171
172    for (psiindex = 0; psiindex <= 1; psiindex++) {
173        t2a = (double **) work1[procid][psiindex];
174        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
175            t2a[0][0] = 0;
176        }
177        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
178            t2a[im - 1][0] = 0;
179        }
180        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
181            t2a[0][jm - 1] = 0;
182        }
183        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
184            t2a[im - 1][jm - 1] = 0;
185        }
186        laplacalc(procid, psi, work1, psiindex, firstrow, lastrow, firstcol, lastcol);
187    }
188
189/* set values of work2 array to psi1 - psi3   */
190
191    t2a = (double **) work2[procid];
192    t2b = (double **) psi[procid][0];
193    t2c = (double **) psi[procid][1];
194    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
195        t2a[0][0] = t2b[0][0] - t2c[0][0];
196    }
197    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
198        t2a[im - 1][0] = t2b[im - 1][0] - t2c[im - 1][0];
199    }
200    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
201        t2a[0][jm - 1] = t2b[0][jm - 1] - t2c[0][jm - 1];
202    }
203    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
204        t2a[im - 1][jm - 1] = t2b[im - 1][jm - 1] - t2c[im - 1][jm - 1];
205    }
206    if (gp[procid].neighbors[UP] == -1) {
207        t1a = (double *) t2a[0];
208        t1b = (double *) t2b[0];
209        t1c = (double *) t2c[0];
210        for (j = firstcol; j <= lastcol; j++) {
211            t1a[j] = t1b[j] - t1c[j];
212        }
213    }
214    if (gp[procid].neighbors[DOWN] == -1) {
215        t1a = (double *) t2a[im - 1];
216        t1b = (double *) t2b[im - 1];
217        t1c = (double *) t2c[im - 1];
218        for (j = firstcol; j <= lastcol; j++) {
219            t1a[j] = t1b[j] - t1c[j];
220        }
221    }
222    if (gp[procid].neighbors[LEFT] == -1) {
223        for (j = firstrow; j <= lastrow; j++) {
224            t2a[j][0] = t2b[j][0] - t2c[j][0];
225        }
226    }
227    if (gp[procid].neighbors[RIGHT] == -1) {
228        for (j = firstrow; j <= lastrow; j++) {
229            t2a[j][jm - 1] = t2b[j][jm - 1] - t2c[j][jm - 1];
230        }
231    }
232    for (i = firstrow; i <= lastrow; i++) {
233        t1a = (double *) t2a[i];
234        t1b = (double *) t2b[i];
235        t1c = (double *) t2c[i];
236        for (iindex = firstcol; iindex <= lastcol; iindex++) {
237            t1a[iindex] = t1b[iindex] - t1c[iindex];
238        }
239    }
240
241/* set values of work3 array to h3/h * psi1 + h1/h * psi3  */
242
243    t2a = (double **) work3[procid];
244    hh3 = h3 / h;
245    hh1 = h1 / h;
246    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
247        t2a[0][0] = hh3 * t2a[0][0] + hh1 * t2c[0][0];
248    }
249    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
250        t2a[im - 1][0] = hh3 * t2a[im - 1][0] + hh1 * t2c[im - 1][0];
251    }
252    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
253        t2a[0][jm - 1] = hh3 * t2a[0][jm - 1] + hh1 * t2c[0][jm - 1];
254    }
255    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
256        t2a[im - 1][jm - 1] = hh3 * t2a[im - 1][jm - 1] + hh1 * t2c[im - 1][jm - 1];
257    }
258    if (gp[procid].neighbors[UP] == -1) {
259        for (j = firstcol; j <= lastcol; j++) {
260            t2a[0][j] = hh3 * t2a[0][j] + hh1 * t2c[0][j];
261        }
262    }
263    if (gp[procid].neighbors[DOWN] == -1) {
264        for (j = firstcol; j <= lastcol; j++) {
265            t2a[im - 1][j] = hh3 * t2a[im - 1][j] + hh1 * t2c[im - 1][j];
266        }
267    }
268    if (gp[procid].neighbors[LEFT] == -1) {
269        for (j = firstrow; j <= lastrow; j++) {
270            t2a[j][0] = hh3 * t2a[j][0] + hh1 * t2c[j][0];
271        }
272    }
273    if (gp[procid].neighbors[RIGHT] == -1) {
274        for (j = firstrow; j <= lastrow; j++) {
275            t2a[j][jm - 1] = hh3 * t2a[j][jm - 1] + hh1 * t2c[j][jm - 1];
276        }
277    }
278    for (i = firstrow; i <= lastrow; i++) {
279        t1a = (double *) t2a[i];
280        t1c = (double *) t2c[i];
281        for (iindex = firstcol; iindex <= lastcol; iindex++) {
282            t1a[iindex] = hh3 * t1a[iindex] + hh1 * t1c[iindex];
283        }
284    }
285
286/* set values of temparray{1,3} to psi{1,3}  */
287
288    for (psiindex = 0; psiindex <= 1; psiindex++) {
289        t2a = (double **) temparray[procid][psiindex];
290        t2b = (double **) psi[procid][psiindex];
291        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
292            t2a[0][0] = t2b[0][0];
293        }
294        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
295            t2a[im - 1][0] = t2b[im - 1][0];
296        }
297        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
298            t2a[0][jm - 1] = t2b[0][jm - 1];
299        }
300        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
301            t2a[im - 1][jm - 1] = t2b[im - 1][jm - 1];
302        }
303        if (gp[procid].neighbors[UP] == -1) {
304            for (j = firstcol; j <= lastcol; j++) {
305                t2a[0][j] = t2b[0][j];
306            }
307        }
308        if (gp[procid].neighbors[DOWN] == -1) {
309            for (j = firstcol; j <= lastcol; j++) {
310                t2a[im - 1][j] = t2b[im - 1][j];
311            }
312        }
313        if (gp[procid].neighbors[LEFT] == -1) {
314            for (j = firstrow; j <= lastrow; j++) {
315                t2a[j][0] = t2b[j][0];
316            }
317        }
318        if (gp[procid].neighbors[RIGHT] == -1) {
319            for (j = firstrow; j <= lastrow; j++) {
320                t2a[j][jm - 1] = t2b[j][jm - 1];
321            }
322        }
323
324        for (i = firstrow; i <= lastrow; i++) {
325            t1a = (double *) t2a[i];
326            t1b = (double *) t2b[i];
327            for (iindex = firstcol; iindex <= lastcol; iindex++) {
328                t1a[iindex] = t1b[iindex];
329            }
330        }
331    }
332
333    END_PHASE(procid, 1);
334
335#if defined(MULTIPLE_BARRIERS)
336    BARRIER(bars->sl_phase_1, nprocs)
337#else
338    BARRIER(bars->barrier, nprocs)
339#endif
340   
341/*     *******************************************************
342
343              s e c o n d   p h a s e
344
345       *******************************************************
346
347   set values of psi{1,3} to psim{1,3}   */
348
349    START_PHASE(procid, 2);
350
351    for (psiindex = 0; psiindex <= 1; psiindex++) {
352        t2a = (double **) psi[procid][psiindex];
353        t2b = (double **) psim[procid][psiindex];
354        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
355            t2a[0][0] = t2b[0][0];
356        }
357        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
358            t2a[0][jm - 1] = t2b[0][jm - 1];
359        }
360        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
361            t2a[im - 1][0] = t2b[im - 1][0];
362        }
363        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
364            t2a[im - 1][jm - 1] = t2b[im - 1][jm - 1];
365        }
366        if (gp[procid].neighbors[UP] == -1) {
367            for (j = firstcol; j <= lastcol; j++) {
368                t2a[0][j] = t2b[0][j];
369            }
370        }
371        if (gp[procid].neighbors[DOWN] == -1) {
372            for (j = firstcol; j <= lastcol; j++) {
373                t2a[im - 1][j] = t2b[im - 1][j];
374            }
375        }
376        if (gp[procid].neighbors[LEFT] == -1) {
377            for (j = firstrow; j <= lastrow; j++) {
378                t2a[j][0] = t2b[j][0];
379            }
380        }
381        if (gp[procid].neighbors[RIGHT] == -1) {
382            for (j = firstrow; j <= lastrow; j++) {
383                t2a[j][jm - 1] = t2b[j][jm - 1];
384            }
385        }
386
387        for (i = firstrow; i <= lastrow; i++) {
388            t1a = (double *) t2a[i];
389            t1b = (double *) t2b[i];
390            for (iindex = firstcol; iindex <= lastcol; iindex++) {
391                t1a[iindex] = t1b[iindex];
392            }
393        }
394    }
395
396/* put the laplacian of the psim array
397   into the work7 array; first part of a three-laplacian
398   calculation to compute the friction terms  */
399
400    for (psiindex = 0; psiindex <= 1; psiindex++) {
401        t2a = (double **) work7[procid][psiindex];
402        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
403            t2a[0][0] = 0;
404        }
405        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
406            t2a[im - 1][0] = 0;
407        }
408        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
409            t2a[0][jm - 1] = 0;
410        }
411        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
412            t2a[im - 1][jm - 1] = 0;
413        }
414        laplacalc(procid, psim, work7, psiindex, firstrow, lastrow, firstcol, lastcol);
415    }
416
417/* to the values of the work1{1,2} arrays obtained from the
418   laplacians of psi{1,2} in the previous phase, add to the
419   elements of every column the corresponding value in the
420   one-dimenional f array  */
421
422    for (psiindex = 0; psiindex <= 1; psiindex++) {
423        t2a = (double **) work1[procid][psiindex];
424        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
425            t2a[0][0] = t2a[0][0] + f[0];
426        }
427        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
428            t2a[im - 1][0] = t2a[im - 1][0] + f[0];
429        }
430        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
431            t2a[0][jm - 1] = t2a[0][jm - 1] + f[jmx[numlev - 1] - 1];
432        }
433        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
434            t2a[im - 1][jm - 1] = t2a[im - 1][jm - 1] + f[jmx[numlev - 1] - 1];
435        }
436        if (gp[procid].neighbors[UP] == -1) {
437            for (j = firstcol; j <= lastcol; j++) {
438                t2a[0][j] = t2a[0][j] + f[j + j_off];
439            }
440        }
441        if (gp[procid].neighbors[DOWN] == -1) {
442            for (j = firstcol; j <= lastcol; j++) {
443                t2a[im - 1][j] = t2a[im - 1][j] + f[j + j_off];
444            }
445        }
446        if (gp[procid].neighbors[LEFT] == -1) {
447            for (j = firstrow; j <= lastrow; j++) {
448                t2a[j][0] = t2a[j][0] + f[j + i_off];
449            }
450        }
451        if (gp[procid].neighbors[RIGHT] == -1) {
452            for (j = firstrow; j <= lastrow; j++) {
453                t2a[j][jm - 1] = t2a[j][jm - 1] + f[j + i_off];
454            }
455        }
456        for (i = firstrow; i <= lastrow; i++) {
457            t1a = (double *) t2a[i];
458            for (iindex = firstcol; iindex <= lastcol; iindex++) {
459                t1a[iindex] = t1a[iindex] + f[iindex + j_off];
460            }
461        }
462    }
463
464    END_PHASE(procid, 2);
465   
466#if defined(MULTIPLE_BARRIERS)
467    BARRIER(bars->sl_phase_2, nprocs)
468#else
469    BARRIER(bars->barrier, nprocs)
470#endif
471/*      *******************************************************
472
473                 t h i r d   p h a s e
474
475        *******************************************************
476
477   put the jacobian of the work1{1,2} and psi{1,3} arrays
478   (the latter currently in temparray) in the work5{1,2} arrays  */
479
480    START_PHASE(procid, 3);
481
482    for (psiindex = 0; psiindex <= 1; psiindex++) {
483        jacobcalc2(work1, temparray, work5, psiindex, procid, firstrow, lastrow, firstcol, lastcol);
484    }
485
486/* set values of psim{1,3} to temparray{1,3}  */
487
488    for (psiindex = 0; psiindex <= 1; psiindex++) {
489        t2a = (double **) psim[procid][psiindex];
490        t2b = (double **) temparray[procid][psiindex];
491        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
492            t2a[0][0] = t2b[0][0];
493        }
494        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
495            t2a[im - 1][0] = t2b[im - 1][0];
496        }
497        if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
498            t2a[0][jm - 1] = t2b[0][jm - 1];
499        }
500        if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
501            t2a[im - 1][jm - 1] = t2b[im - 1][jm - 1];
502        }
503        if (gp[procid].neighbors[UP] == -1) {
504            t1a = (double *) t2a[0];
505            t1b = (double *) t2b[0];
506            for (j = firstcol; j <= lastcol; j++) {
507                t1a[j] = t1b[j];
508            }
509        }
510        if (gp[procid].neighbors[DOWN] == -1) {
511            t1a = (double *) t2a[im - 1];
512            t1b = (double *) t2b[im - 1];
513            for (j = firstcol; j <= lastcol; j++) {
514                t1a[j] = t1b[j];
515            }
516        }
517        if (gp[procid].neighbors[LEFT] == -1) {
518            for (j = firstrow; j <= lastrow; j++) {
519                t2a[j][0] = t2b[j][0];
520            }
521        }
522        if (gp[procid].neighbors[RIGHT] == -1) {
523            for (j = firstrow; j <= lastrow; j++) {
524                t2a[j][jm - 1] = t2b[j][jm - 1];
525            }
526        }
527        for (i = firstrow; i <= lastrow; i++) {
528            t1a = (double *) t2a[i];
529            t1b = (double *) t2b[i];
530            for (iindex = firstcol; iindex <= lastcol; iindex++) {
531                t1a[iindex] = t1b[iindex];
532            }
533        }
534    }
535
536/* put the laplacian of the work7{1,2} arrays in the work4{1,2}
537   arrays; second step in the three-laplacian friction calculation  */
538
539    for (psiindex = 0; psiindex <= 1; psiindex++) {
540        laplacalc(procid, work7, work4, psiindex, firstrow, lastrow, firstcol, lastcol);
541    }
542
543    END_PHASE(procid, 3);
544
545#if defined(MULTIPLE_BARRIERS)
546    BARRIER(bars->sl_phase_3, nprocs)
547#else
548    BARRIER(bars->barrier, nprocs)
549#endif
550
551/*     *******************************************************
552
553                f o u r t h   p h a s e
554
555       *******************************************************
556
557   put the jacobian of the work2 and work3 arrays in the work6
558   array  */
559
560    START_PHASE(procid, 4);
561
562    jacobcalc(work2, work3, work6, procid, firstrow, lastrow, firstcol, lastcol);
563
564/* put the laplacian of the work4{1,2} arrays in the work7{1,2}
565   arrays; third step in the three-laplacian friction calculation  */
566
567    for (psiindex = 0; psiindex <= 1; psiindex++) {
568        laplacalc(procid, work4, work7, psiindex, firstrow, lastrow, firstcol, lastcol);
569    }
570   
571    END_PHASE(procid, 4);
572
573#if defined(MULTIPLE_BARRIERS)
574    BARRIER(bars->sl_phase_4, nprocs)
575#else
576    BARRIER(bars->barrier, nprocs)
577#endif
578
579/*     *******************************************************
580
581                f i f t h   p h a s e
582
583       *******************************************************
584
585   use the values of the work5, work6 and work7 arrays
586   computed in the previous time-steps to compute the
587   ga and gb arrays   */
588
589    START_PHASE(procid, 5);
590
591    hinv = 1.0 / h;
592    h1inv = 1.0 / h1;
593
594    t2a = (double **) ga[procid];
595    t2b = (double **) gb[procid];
596    t2c = (double **) work5[procid][0];
597    t2d = (double **) work5[procid][1];
598    t2e = (double **) work7[procid][0];
599    t2f = (double **) work7[procid][1];
600    t2g = (double **) work6[procid];
601    t2h = (double **) tauz[procid];
602    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
603        t2a[0][0] = t2c[0][0] - t2d[0][0] + eig2 * t2g[0][0] + h1inv * t2h[0][0] + lf * t2e[0][0] - lf * t2f[0][0];
604        t2b[0][0] = hh1 * t2c[0][0] + hh3 * t2d[0][0] + hinv * t2h[0][0] + lf * hh1 * t2e[0][0] + lf * hh3 * t2f[0][0];
605    }
606    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
607        t2a[im - 1][0] = t2c[im - 1][0] - t2d[im - 1][0] + eig2 * t2g[im - 1][0] + h1inv * t2h[im - 1][0] + lf * t2e[im - 1][0] - lf * t2f[im - 1][0];
608        t2b[im - 1][0] = hh1 * t2c[im - 1][0] + hh3 * t2d[im - 1][0] + hinv * t2h[im - 1][0] + lf * hh1 * t2e[im - 1][0] + lf * hh3 * t2f[im - 1][0];
609    }
610    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
611        t2a[0][jm - 1] = t2c[0][jm - 1] - t2d[0][jm - 1] + eig2 * t2g[0][jm - 1] + h1inv * t2h[0][jm - 1] + lf * t2e[0][jm - 1] - lf * t2f[0][jm - 1];
612        t2b[0][jm - 1] = hh1 * t2c[0][jm - 1] + hh3 * t2d[0][jm - 1] + hinv * t2h[0][jm - 1] + lf * hh1 * t2e[0][jm - 1] + lf * hh3 * t2f[0][jm - 1];
613    }
614    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
615        t2a[im - 1][jm - 1] = t2c[im - 1][jm - 1] - t2d[im - 1][jm - 1] + eig2 * t2g[im - 1][jm - 1] + h1inv * t2h[im - 1][jm - 1] + lf * t2e[im - 1][jm - 1] - lf * t2f[im - 1][jm - 1];
616        t2b[im - 1][jm - 1] = hh1 * t2c[im - 1][jm - 1] + hh3 * t2d[im - 1][jm - 1] + hinv * t2h[im - 1][jm - 1] + lf * hh1 * t2e[im - 1][jm - 1] + lf * hh3 * t2f[im - 1][jm - 1];
617    }
618    if (gp[procid].neighbors[UP] == -1) {
619        t1a = (double *) t2a[0];
620        t1b = (double *) t2b[0];
621        t1c = (double *) t2c[0];
622        t1d = (double *) t2d[0];
623        t1e = (double *) t2e[0];
624        t1f = (double *) t2f[0];
625        t1g = (double *) t2g[0];
626        t1h = (double *) t2h[0];
627        for (j = firstcol; j <= lastcol; j++) {
628            t1a[j] = t1c[j] - t1d[j] + eig2 * t1g[j] + h1inv * t1h[j] + lf * t1e[j] - lf * t1f[j];
629            t1b[j] = hh1 * t1c[j] + hh3 * t1d[j] + hinv * t1h[j] + lf * hh1 * t1e[j] + lf * hh3 * t1f[j];
630        }
631    }
632    if (gp[procid].neighbors[DOWN] == -1) {
633        t1a = (double *) t2a[im - 1];
634        t1b = (double *) t2b[im - 1];
635        t1c = (double *) t2c[im - 1];
636        t1d = (double *) t2d[im - 1];
637        t1e = (double *) t2e[im - 1];
638        t1f = (double *) t2f[im - 1];
639        t1g = (double *) t2g[im - 1];
640        t1h = (double *) t2h[im - 1];
641        for (j = firstcol; j <= lastcol; j++) {
642            t1a[j] = t1c[j] - t1d[j] + eig2 * t1g[j] + h1inv * t1h[j] + lf * t1e[j] - lf * t1f[j];
643            t1b[j] = hh1 * t1c[j] + hh3 * t1d[j] + hinv * t1h[j] + lf * hh1 * t1e[j] + lf * hh3 * t1f[j];
644        }
645    }
646    if (gp[procid].neighbors[LEFT] == -1) {
647        for (j = firstrow; j <= lastrow; j++) {
648            t2a[j][0] = t2c[j][0] - t2d[j][0] + eig2 * t2g[j][0] + h1inv * t2h[j][0] + lf * t2e[j][0] - lf * t2f[j][0];
649            t2b[j][0] = hh1 * t2c[j][0] + hh3 * t2d[j][0] + hinv * t2h[j][0] + lf * hh1 * t2e[j][0] + lf * hh3 * t2f[j][0];
650        }
651    }
652    if (gp[procid].neighbors[RIGHT] == -1) {
653        for (j = firstrow; j <= lastrow; j++) {
654            t2a[j][jm - 1] = t2c[j][jm - 1] - t2d[j][jm - 1] + eig2 * t2g[j][jm - 1] + h1inv * t2h[j][jm - 1] + lf * t2e[j][jm - 1] - lf * t2f[j][jm - 1];
655            t2b[j][jm - 1] = hh1 * t2c[j][jm - 1] + hh3 * t2d[j][jm - 1] + hinv * t2h[j][jm - 1] + lf * hh1 * t2e[j][jm - 1] + lf * hh3 * t2f[j][jm - 1];
656        }
657    }
658
659    for (i = firstrow; i <= lastrow; i++) {
660        t1a = (double *) t2a[i];
661        t1b = (double *) t2b[i];
662        t1c = (double *) t2c[i];
663        t1d = (double *) t2d[i];
664        t1e = (double *) t2e[i];
665        t1f = (double *) t2f[i];
666        t1g = (double *) t2g[i];
667        t1h = (double *) t2h[i];
668        for (iindex = firstcol; iindex <= lastcol; iindex++) {
669            t1a[iindex] = t1c[iindex] - t1d[iindex] + eig2 * t1g[iindex] + h1inv * t1h[iindex] + lf * t1e[iindex] - lf * t1f[iindex];
670            t1b[iindex] = hh1 * t1c[iindex] + hh3 * t1d[iindex] + hinv * t1h[iindex] + lf * hh1 * t1e[iindex] + lf * hh3 * t1f[iindex];
671        }
672    }
673
674    END_PHASE(procid, 5);
675
676#if defined(MULTIPLE_BARRIERS)
677    BARRIER(bars->sl_phase_5, nprocs)
678#else
679    BARRIER(bars->barrier, nprocs)
680#endif
681
682/*     *******************************************************
683
684               s i x t h   p h a s e
685
686       *******************************************************  */
687
688    START_PHASE(procid, 6);
689
690    istart = 1;
691    iend = istart + gp[procid].rel_num_y[numlev - 1] - 1;
692    jstart = 1;
693    jend = jstart + gp[procid].rel_num_x[numlev - 1] - 1;
694    ist = istart;
695    ien = iend;
696    jst = jstart;
697    jen = jend;
698
699    if (gp[procid].neighbors[UP] == -1) {
700        istart = 0;
701    }
702    if (gp[procid].neighbors[LEFT] == -1) {
703        jstart = 0;
704    }
705    if (gp[procid].neighbors[DOWN] == -1) {
706        iend = im - 1;
707    }
708    if (gp[procid].neighbors[RIGHT] == -1) {
709        jend = jm - 1;
710    }
711    t2a = (double **) rhs_multi[procid][numlev - 1];
712    t2b = (double **) ga[procid];
713    t2c = (double **) oldga[procid];
714    t2d = (double **) q_multi[procid][numlev - 1];
715    for (i = istart; i <= iend; i++) {
716        t1a = (double *) t2a[i];
717        t1b = (double *) t2b[i];
718        for (j = jstart; j <= jend; j++) {
719            t1a[j] = t1b[j] * ressqr;
720        }
721    }
722
723    if (gp[procid].neighbors[UP] == -1) {
724        t1d = (double *) t2d[0];
725        t1b = (double *) t2b[0];
726        for (j = jstart; j <= jend; j++) {
727            t1d[j] = t1b[j];
728        }
729    }
730    if (gp[procid].neighbors[DOWN] == -1) {
731        t1d = (double *) t2d[im - 1];
732        t1b = (double *) t2b[im - 1];
733        for (j = jstart; j <= jend; j++) {
734            t1d[j] = t1b[j];
735        }
736    }
737    if (gp[procid].neighbors[LEFT] == -1) {
738        for (i = istart; i <= iend; i++) {
739            t2d[i][0] = t2b[i][0];
740        }
741    }
742    if (gp[procid].neighbors[RIGHT] == -1) {
743        for (i = istart; i <= iend; i++) {
744            t2d[i][jm - 1] = t2b[i][jm - 1];
745        }
746    }
747    //fac = 1.0 / (4.0 - ressqr*eig2);
748    for (i = ist; i <= ien; i++) {
749        t1d = (double *) t2d[i];
750        t1c = (double *) t2c[i];
751        for (j = jst; j <= jen; j++) {
752            t1d[j] = t1c[j];
753        }
754    }
755
756    if ((procid == MASTER) || (do_stats)) {
757        CLOCK(multi_start);
758    }
759
760    multig(procid);
761
762    if ((procid == MASTER) || (do_stats)) {
763        CLOCK(multi_end);
764        (*gp[procid].multi_time) += (multi_end - multi_start);
765    }
766
767/* the shared sum variable psiai is initialized to 0 at
768   every time-step  */
769
770    if (procid == MASTER) {
771        global->psiai = 0.0;
772    }
773
774/*  copy the solution for use as initial guess in next time-step  */
775
776    for (i = istart; i <= iend; i++) {
777        t1b = (double *) t2b[i];
778        t1c = (double *) t2c[i];
779        t1d = (double *) t2d[i];
780        for (j = jstart; j <= jend; j++) {
781            t1b[j] = t1d[j];
782            t1c[j] = t1d[j];
783        }
784    }
785
786    END_PHASE(procid, 6);
787
788#if defined(MULTIPLE_BARRIERS)
789    BARRIER(bars->sl_phase_6, nprocs)
790#else
791    BARRIER(bars->barrier, nprocs)
792#endif
793
794/*     *******************************************************
795
796                s e v e n t h   p h a s e
797
798       *******************************************************
799
800   every process computes the running sum for its assigned portion
801   in a private variable psiaipriv   */
802
803    START_PHASE(procid, 7);
804
805    psiaipriv = 0.0;
806    t2a = (double **) ga[procid];
807    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
808        psiaipriv = psiaipriv + 0.25 * (t2a[0][0]);
809    }
810    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
811        psiaipriv = psiaipriv + 0.25 * (t2a[0][jm - 1]);
812    }
813    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
814        psiaipriv = psiaipriv + 0.25 * (t2a[im - 1][0]);
815    }
816    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
817        psiaipriv = psiaipriv + 0.25 * (t2a[im - 1][jm - 1]);
818    }
819    if (gp[procid].neighbors[UP] == -1) {
820        t1a = (double *) t2a[0];
821        for (j = firstcol; j <= lastcol; j++) {
822            psiaipriv = psiaipriv + 0.5 * t1a[j];
823        }
824    }
825    if (gp[procid].neighbors[DOWN] == -1) {
826        t1a = (double *) t2a[im - 1];
827        for (j = firstcol; j <= lastcol; j++) {
828            psiaipriv = psiaipriv + 0.5 * t1a[j];
829        }
830    }
831    if (gp[procid].neighbors[LEFT] == -1) {
832        for (j = firstrow; j <= lastrow; j++) {
833            psiaipriv = psiaipriv + 0.5 * t2a[j][0];
834        }
835    }
836    if (gp[procid].neighbors[RIGHT] == -1) {
837        for (j = firstrow; j <= lastrow; j++) {
838            psiaipriv = psiaipriv + 0.5 * t2a[j][jm - 1];
839        }
840    }
841    for (i = firstrow; i <= lastrow; i++) {
842        t1a = (double *) t2a[i];
843        for (iindex = firstcol; iindex <= lastcol; iindex++) {
844            psiaipriv = psiaipriv + t1a[iindex];
845        }
846    }
847
848/* after computing its private sum, every process adds that to the
849   shared running sum psiai  */
850
851    LOCK(locks->psiailock)
852    global->psiai = global->psiai + psiaipriv;
853    UNLOCK(locks->psiailock)
854
855    END_PHASE(procid, 7);
856
857#if defined(MULTIPLE_BARRIERS)
858    BARRIER(bars->sl_phase_7, nprocs)
859#else
860    BARRIER(bars->barrier, nprocs)
861#endif
862   
863/*      *******************************************************
864
865                e i g h t h   p h a s e
866
867        *******************************************************
868
869   augment ga(i,j) with [-psiai/psibi]*psib(i,j) */
870
871    START_PHASE(procid, 8);
872
873    f4 = (-global->psiai) /(global->psibi);
874
875    t2a = (double **) ga[procid];
876    t2b = (double **) psib[procid];
877    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
878        t2a[0][0] = t2a[0][0] + f4 * t2b[0][0];
879    }
880    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
881        t2a[im - 1][0] = t2a[im - 1][0] + f4 * t2b[im - 1][0];
882    }
883    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
884        t2a[0][jm - 1] = t2a[0][jm - 1] + f4 * t2b[0][jm - 1];
885    }
886    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
887        t2a[im - 1][jm - 1] = t2a[im - 1][jm - 1] + f4 * t2b[im - 1][jm - 1];
888    }
889    if (gp[procid].neighbors[UP] == -1) {
890        t1a = (double *) t2a[0];
891        t1b = (double *) t2b[0];
892        for (j = firstcol; j <= lastcol; j++) {
893            t1a[j] = t1a[j] + f4 * t1b[j];
894        }
895    }
896    if (gp[procid].neighbors[DOWN] == -1) {
897        t1a = (double *) t2a[im - 1];
898        t1b = (double *) t2b[im - 1];
899        for (j = firstcol; j <= lastcol; j++) {
900            t1a[j] = t1a[j] + f4 * t1b[j];
901        }
902    }
903    if (gp[procid].neighbors[LEFT] == -1) {
904        for (j = firstrow; j <= lastrow; j++) {
905            t2a[j][0] = t2a[j][0] + f4 * t2b[j][0];
906        }
907    }
908    if (gp[procid].neighbors[RIGHT] == -1) {
909        for (j = firstrow; j <= lastrow; j++) {
910            t2a[j][jm - 1] = t2a[j][jm - 1] + f4 * t2b[j][jm - 1];
911        }
912    }
913    for (i = firstrow; i <= lastrow; i++) {
914        t1a = (double *) t2a[i];
915        t1b = (double *) t2b[i];
916        for (iindex = firstcol; iindex <= lastcol; iindex++) {
917            t1a[iindex] = t1a[iindex] + f4 * t1b[iindex];
918        }
919    }
920
921    t2a = (double **) rhs_multi[procid][numlev - 1];
922    t2b = (double **) gb[procid];
923    t2c = (double **) oldgb[procid];
924    t2d = (double **) q_multi[procid][numlev - 1];
925    for (i = istart; i <= iend; i++) {
926        t1a = (double *) t2a[i];
927        t1b = (double *) t2b[i];
928        for (j = jstart; j <= jend; j++) {
929            t1a[j] = t1b[j] * ressqr;
930        }
931    }
932    if (gp[procid].neighbors[UP] == -1) {
933        t1d = (double *) t2d[0];
934        t1b = (double *) t2b[0];
935        for (j = jstart; j <= jend; j++) {
936            t1d[j] = t1b[j];
937        }
938    }
939    if (gp[procid].neighbors[DOWN] == -1) {
940        t1d = (double *) t2d[im - 1];
941        t1b = (double *) t2b[im - 1];
942        for (j = jstart; j <= jend; j++) {
943            t1d[j] = t1b[j];
944        }
945    }
946    if (gp[procid].neighbors[LEFT] == -1) {
947        for (i = istart; i <= iend; i++) {
948            t2d[i][0] = t2b[i][0];
949        }
950    }
951    if (gp[procid].neighbors[RIGHT] == -1) {
952        for (i = istart; i <= iend; i++) {
953            t2d[i][jm - 1] = t2b[i][jm - 1];
954        }
955    }
956    //fac = 1.0 / (4.0 - ressqr*eig2);
957    for (i = ist; i <= ien; i++) {
958        t1d = (double *) t2d[i];
959        t1c = (double *) t2c[i];
960        for (j = jst; j <= jen; j++) {
961            t1d[j] = t1c[j];
962        }
963    }
964
965    if ((procid == MASTER) || (do_stats)) {
966        CLOCK(multi_start);
967    }
968
969    multig(procid);
970
971    if ((procid == MASTER) || (do_stats)) {
972        CLOCK(multi_end);
973        (*gp[procid].multi_time) += (multi_end - multi_start);
974    }
975
976    for (i = istart; i <= iend; i++) {
977        t1b = (double *) t2b[i];
978        t1c = (double *) t2c[i];
979        t1d = (double *) t2d[i];
980        for (j = jstart; j <= jend; j++) {
981            t1b[j] = t1d[j];
982            t1c[j] = t1d[j];
983        }
984    }
985
986    END_PHASE(procid, 8);
987
988#if defined(MULTIPLE_BARRIERS)
989    BARRIER(bars->sl_phase_8, nprocs)
990#else
991    BARRIER(bars->barrier, nprocs)
992#endif
993
994/*      *******************************************************
995
996                n i n t h   p h a s e
997
998        *******************************************************
999
1000   put appropriate linear combinations of ga and gb in work2 and work3;
1001   note that here (as in most cases) the constant multipliers are made
1002   private variables; the specific order in which things are done is
1003   chosen in order to hopefully reuse things brought into the cache
1004
1005   note that here again we choose to have all processes share the work
1006   on both matrices despite the fact that the work done per element
1007   is the same, because the operand matrices are the same in both cases */
1008
1009    START_PHASE(procid, 9);
1010
1011    t2a = (double **) ga[procid];
1012    t2b = (double **) gb[procid];
1013    t2c = (double **) work2[procid];
1014    t2d = (double **) work3[procid];
1015    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
1016        t2c[0][0] = t2b[0][0] - hh1 * t2a[0][0];
1017        t2d[0][0] = t2b[0][0] + hh3 * t2a[0][0];
1018    }
1019    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
1020        t2c[im - 1][0] = t2b[im - 1][0] - hh1 * t2a[im - 1][0];
1021        t2d[im - 1][0] = t2b[im - 1][0] + hh3 * t2a[im - 1][0];
1022    }
1023    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
1024        t2c[0][jm - 1] = t2b[0][jm - 1] - hh1 * t2a[0][jm - 1];
1025        t2d[0][jm - 1] = t2b[0][jm - 1] + hh3 * t2a[0][jm - 1];
1026    }
1027    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
1028        t2c[im - 1][jm - 1] = t2b[im - 1][jm - 1] - hh1 * t2a[im - 1][jm - 1];
1029        t2d[im - 1][jm - 1] = t2b[im - 1][jm - 1] + hh3 * t2a[im - 1][jm - 1];
1030    }
1031    if (gp[procid].neighbors[UP] == -1) {
1032        t1a = (double *) t2a[0];
1033        t1b = (double *) t2b[0];
1034        t1c = (double *) t2c[0];
1035        t1d = (double *) t2d[0];
1036        for (j = firstcol; j <= lastcol; j++) {
1037            t1d[j] = t1b[j] + hh3 * t1a[j];
1038            t1c[j] = t1b[j] - hh1 * t1a[j];
1039        }
1040    }
1041    if (gp[procid].neighbors[DOWN] == -1) {
1042        t1a = (double *) t2a[im - 1];
1043        t1b = (double *) t2b[im - 1];
1044        t1c = (double *) t2c[im - 1];
1045        t1d = (double *) t2d[im - 1];
1046        for (j = firstcol; j <= lastcol; j++) {
1047            t1d[j] = t1b[j] + hh3 * t1a[j];
1048            t1c[j] = t1b[j] - hh1 * t1a[j];
1049        }
1050    }
1051    if (gp[procid].neighbors[LEFT] == -1) {
1052        for (j = firstrow; j <= lastrow; j++) {
1053            t2d[j][0] = t2b[j][0] + hh3 * t2a[j][0];
1054            t2c[j][0] = t2b[j][0] - hh1 * t2a[j][0];
1055        }
1056    }
1057    if (gp[procid].neighbors[RIGHT] == -1) {
1058        for (j = firstrow; j <= lastrow; j++) {
1059            t2d[j][jm - 1] = t2b[j][jm - 1] + hh3 * t2a[j][jm - 1];
1060            t2c[j][jm - 1] = t2b[j][jm - 1] - hh1 * t2a[j][jm - 1];
1061        }
1062    }
1063
1064    for (i = firstrow; i <= lastrow; i++) {
1065        t1a = (double *) t2a[i];
1066        t1b = (double *) t2b[i];
1067        t1c = (double *) t2c[i];
1068        t1d = (double *) t2d[i];
1069        for (iindex = firstcol; iindex <= lastcol; iindex++) {
1070            t1d[iindex] = t1b[iindex] + hh3 * t1a[iindex];
1071            t1c[iindex] = t1b[iindex] - hh1 * t1a[iindex];
1072        }
1073    }
1074
1075    END_PHASE(procid, 9);
1076
1077#if defined(MULTIPLE_BARRIERS)
1078    BARRIER(bars->sl_phase_9, nprocs)
1079#else
1080    BARRIER(bars->barrier, nprocs)
1081#endif
1082
1083/*      *******************************************************
1084
1085                t e n t h    p h a s e
1086
1087        *******************************************************/
1088
1089    START_PHASE(procid, 10);
1090    timst = 2 * dtau;
1091
1092/* update the psi{1,3} matrices by adding 2*dtau*work3 to each */
1093
1094    t2a = (double **) psi[procid][0];
1095    t2b = (double **) work3[procid];
1096    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
1097        t2a[0][0] = t2a[0][0] + timst * t2b[0][0];
1098    }
1099    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
1100        t2a[im - 1][0] = t2a[im - 1][0] + timst * t2b[im - 1][0];
1101    }
1102    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
1103        t2a[0][jm - 1] = t2a[0][jm - 1] + timst * t2b[0][jm - 1];
1104    }
1105    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
1106        t2a[im - 1][jm - 1] = t2a[im - 1][jm - 1] + timst * t2b[im - 1][jm - 1];
1107    }
1108    if (gp[procid].neighbors[UP] == -1) {
1109        t1a = (double *) t2a[0];
1110        t1b = (double *) t2b[0];
1111        for (j = firstcol; j <= lastcol; j++) {
1112            t1a[j] = t1a[j] + timst * t1b[j];
1113        }
1114    }
1115    if (gp[procid].neighbors[DOWN] == -1) {
1116        t1a = (double *) t2a[im - 1];
1117        t1b = (double *) t2b[im - 1];
1118        for (j = firstcol; j <= lastcol; j++) {
1119            t1a[j] = t1a[j] + timst * t1b[j];
1120        }
1121    }
1122    if (gp[procid].neighbors[LEFT] == -1) {
1123        for (j = firstrow; j <= lastrow; j++) {
1124            t2a[j][0] = t2a[j][0] + timst * t2b[j][0];
1125        }
1126    }
1127    if (gp[procid].neighbors[RIGHT] == -1) {
1128        for (j = firstrow; j <= lastrow; j++) {
1129            t2a[j][jm - 1] = t2a[j][jm - 1] + timst * t2b[j][jm - 1];
1130        }
1131    }
1132    for (i = firstrow; i <= lastrow; i++) {
1133        t1a = (double *) t2a[i];
1134        t1b = (double *) t2b[i];
1135        for (iindex = firstcol; iindex <= lastcol; iindex++) {
1136            t1a[iindex] = t1a[iindex] + timst * t1b[iindex];
1137        }
1138    }
1139
1140    t2a = (double **) psi[procid][1];
1141    t2b = (double **) work2[procid];
1142    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
1143        t2a[0][0] = t2a[0][0] + timst * t2b[0][0];
1144    }
1145    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[LEFT] == -1)) {
1146        t2a[im - 1][0] = t2a[im - 1][0] + timst * t2b[im - 1][0];
1147    }
1148    if ((gp[procid].neighbors[UP] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
1149        t2a[0][jm - 1] = t2a[0][jm - 1] + timst * t2b[0][jm - 1];
1150    }
1151    if ((gp[procid].neighbors[DOWN] == -1) && (gp[procid].neighbors[RIGHT] == -1)) {
1152        t2a[im - 1][jm - 1] = t2a[im - 1][jm - 1] + timst * t2b[im - 1][jm - 1];
1153    }
1154    if (gp[procid].neighbors[UP] == -1) {
1155        t1a = (double *) t2a[0];
1156        t1b = (double *) t2b[0];
1157        for (j = firstcol; j <= lastcol; j++) {
1158            t1a[j] = t1a[j] + timst * t1b[j];
1159        }
1160    }
1161    if (gp[procid].neighbors[DOWN] == -1) {
1162        t1a = (double *) t2a[im - 1];
1163        t1b = (double *) t2b[im - 1];
1164        for (j = firstcol; j <= lastcol; j++) {
1165            t1a[j] = t1a[j] + timst * t1b[j];
1166        }
1167    }
1168    if (gp[procid].neighbors[LEFT] == -1) {
1169        for (j = firstrow; j <= lastrow; j++) {
1170            t2a[j][0] = t2a[j][0] + timst * t2b[j][0];
1171        }
1172    }
1173    if (gp[procid].neighbors[RIGHT] == -1) {
1174        for (j = firstrow; j <= lastrow; j++) {
1175            t2a[j][jm - 1] = t2a[j][jm - 1] + timst * t2b[j][jm - 1];
1176        }
1177    }
1178
1179    for (i = firstrow; i <= lastrow; i++) {
1180        t1a = (double *) t2a[i];
1181        t1b = (double *) t2b[i];
1182        for (iindex = firstcol; iindex <= lastcol; iindex++) {
1183            t1a[iindex] = t1a[iindex] + timst * t1b[iindex];
1184        }
1185    }
1186
1187    END_PHASE(procid, 10);
1188
1189#if defined(MULTIPLE_BARRIERS)
1190    BARRIER(bars->sl_phase_10, nprocs)
1191#else
1192    BARRIER(bars->barrier, nprocs)
1193#endif
1194}
Note: See TracBrowser for help on using the repository browser.