| 1 | 
edhill | 
1.1 | 
C $Header: /u/gcmpack/MITgcm/pkg/ex3/ex3_xy_rx.template,v 1.1 2005/10/16 06:55:48 edhill Exp $ | 
| 2 | 
  | 
  | 
C $Name:  $ | 
| 3 | 
  | 
  | 
 | 
| 4 | 
  | 
  | 
#include "EX3_OPTIONS.h" | 
| 5 | 
  | 
  | 
 | 
| 6 | 
  | 
  | 
C---+----1----+----2----+----3----+----4----+----5----+----6----+----7-|--+----| | 
| 7 | 
  | 
  | 
CBOP 0 | 
| 8 | 
  | 
  | 
C     !ROUTINE: EX3_XY_RX | 
| 9 | 
  | 
  | 
 | 
| 10 | 
  | 
  | 
C     !INTERFACE: | 
| 11 | 
  | 
  | 
      SUBROUTINE EX3_XY_RX( | 
| 12 | 
  | 
  | 
     I     gtype, | 
| 13 | 
  | 
  | 
     B     phi, | 
| 14 | 
  | 
  | 
     I     myThid ) | 
| 15 | 
  | 
  | 
 | 
| 16 | 
  | 
  | 
C     !DESCRIPTION: | 
| 17 | 
  | 
  | 
C     Perform an exchange for 2D scalars located at either Arakawa mass | 
| 18 | 
  | 
  | 
C     [M|T] or vorticity [Z|V] points. | 
| 19 | 
  | 
  | 
 | 
| 20 | 
  | 
  | 
C     !USES: | 
| 21 | 
  | 
  | 
      IMPLICIT NONE | 
| 22 | 
  | 
  | 
#include "SIZE.h" | 
| 23 | 
  | 
  | 
#include "EEPARAMS.h" | 
| 24 | 
  | 
  | 
#include "EESUPPORT.h" | 
| 25 | 
  | 
  | 
#include "EX3_SIZE.h" | 
| 26 | 
  | 
  | 
#include "EX3_PARAMS.h" | 
| 27 | 
  | 
  | 
#include "EX3_TOPOLOGY.h" | 
| 28 | 
  | 
  | 
 | 
| 29 | 
  | 
  | 
C     !INPUT PARAMETERS: | 
| 30 | 
  | 
  | 
C     gtype  :: grid type: [M|T]=mass point, [Z|V]=vorticity point | 
| 31 | 
  | 
  | 
C     phi    :: Array with overlap regions to be exchanged | 
| 32 | 
  | 
  | 
C     myThid :: My thread id. | 
| 33 | 
  | 
  | 
      CHARACTER*(*) gtype | 
| 34 | 
  | 
  | 
      _RX phi(1-OLx:sNx+OLx,1-OLy:sNy+OLy,nSx,nSy) | 
| 35 | 
  | 
  | 
      INTEGER myThid | 
| 36 | 
  | 
  | 
CEOP | 
| 37 | 
  | 
  | 
 | 
| 38 | 
  | 
  | 
C     !LOCAL VARIABLES: | 
| 39 | 
  | 
  | 
      INTEGER iloc,in,nN | 
| 40 | 
  | 
  | 
      CHARACTER*(MAX_LEN_MBUF) msgbuf | 
| 41 | 
  | 
  | 
C | 
| 42 | 
  | 
  | 
      INTEGER | 
| 43 | 
  | 
  | 
     I     bufftag, sendProc, recvProc, | 
| 44 | 
  | 
  | 
     I     il,ih,is,  jl,jh,js,  kl,kh,ks, | 
| 45 | 
  | 
  | 
     I     io1,jo1,ko1, | 
| 46 | 
  | 
  | 
     I     idl1,idh1, jdl1,jdh1, kdl1,kdh1 | 
| 47 | 
  | 
  | 
      INTEGER | 
| 48 | 
  | 
  | 
     I     i_sendtile, i_recvtile | 
| 49 | 
  | 
  | 
      LOGICAL along_i | 
| 50 | 
  | 
  | 
      CHARACTER*(1) commType | 
| 51 | 
  | 
  | 
      INTEGER msgID(nSx*nSy) | 
| 52 | 
  | 
  | 
C | 
| 53 | 
  | 
  | 
#ifdef ALLOW_USE_MPI | 
| 54 | 
  | 
  | 
      INTEGER mpiStatus(MPI_STATUS_SIZE) | 
| 55 | 
  | 
  | 
      INTEGER mpiRc | 
| 56 | 
  | 
  | 
      INTEGER wHandle | 
| 57 | 
  | 
  | 
#endif | 
| 58 | 
  | 
  | 
 | 
| 59 | 
  | 
  | 
      idl1 = 1-OLx | 
| 60 | 
  | 
  | 
      idh1 = sNx+OLx | 
| 61 | 
  | 
  | 
      jdl1 = 1-OLy | 
| 62 | 
  | 
  | 
      jdh1 = sNy+OLy | 
| 63 | 
  | 
  | 
      kdl1 = 1 | 
| 64 | 
  | 
  | 
      kdh1 = 1 | 
| 65 | 
  | 
  | 
      kl = 1 | 
| 66 | 
  | 
  | 
      kh = 1 | 
| 67 | 
  | 
  | 
      ks = 1 | 
| 68 | 
  | 
  | 
 | 
| 69 | 
  | 
  | 
      commType(1:1) = 'P' | 
| 70 | 
  | 
  | 
#ifdef ALLOW_USE_MPI | 
| 71 | 
  | 
  | 
      commType(1:1) = 'M' | 
| 72 | 
  | 
  | 
#endif | 
| 73 | 
  | 
  | 
 | 
| 74 | 
  | 
  | 
C     As with EXCH2, tile<->tile communication is synchronized through | 
| 75 | 
  | 
  | 
C     thread 1. | 
| 76 | 
  | 
  | 
      CALL BAR2(myThid) | 
| 77 | 
  | 
  | 
 | 
| 78 | 
  | 
  | 
      IF (gtype(1:1) .EQ. 'M' .OR. gtype(1:1) .EQ. 'T') THEN | 
| 79 | 
  | 
  | 
 | 
| 80 | 
  | 
  | 
C       phi is a scalar located at Arakawa mass (cell-center) points | 
| 81 | 
  | 
  | 
 | 
| 82 | 
  | 
  | 
C       First send | 
| 83 | 
  | 
  | 
        DO iloc = myBxLo(myThid), myBxHi(myThid) | 
| 84 | 
  | 
  | 
          i_sendtile = ex3_p_itile(iloc) | 
| 85 | 
  | 
  | 
          nN = ex3_e_n(i_sendtile) | 
| 86 | 
  | 
  | 
          DO in = 1,nN | 
| 87 | 
  | 
  | 
            i_recvtile = ex3_e_iopt(in,i_sendtile) | 
| 88 | 
  | 
  | 
            CALL EX3_GET_BUFFTAG( | 
| 89 | 
  | 
  | 
     I           i_sendtile, i_recvtile, in, | 
| 90 | 
  | 
  | 
     O           bufftag, | 
| 91 | 
  | 
  | 
     I           myThid ) | 
| 92 | 
  | 
  | 
            recvProc   = ex3_t_iproc(i_recvtile) | 
| 93 | 
  | 
  | 
 | 
| 94 | 
  | 
  | 
C           =====  I direction  ===== | 
| 95 | 
  | 
  | 
            il = ex3_e_dat(2,1,in,i_sendtile) | 
| 96 | 
  | 
  | 
            IF ( ex3_e_dat(1,1,in,i_sendtile) .EQ. 0 ) THEN | 
| 97 | 
  | 
  | 
              along_i  = .FALSE. | 
| 98 | 
  | 
  | 
              ih = ex3_e_dat(3,1,in,i_sendtile) | 
| 99 | 
  | 
  | 
            ELSE | 
| 100 | 
  | 
  | 
C             Here, "along" means the i dimension is perpendicular to | 
| 101 | 
  | 
  | 
C             the "seam" between the two tiles | 
| 102 | 
  | 
  | 
              along_i  = .TRUE. | 
| 103 | 
  | 
  | 
              IF (IABS(ex3_e_dat(1,1,in,i_sendtile)) .EQ. 1) THEN | 
| 104 | 
  | 
  | 
                ih = il + ex3_e_dat(1,1,in,i_sendtile) * OLx | 
| 105 | 
  | 
  | 
              ELSE | 
| 106 | 
  | 
  | 
                ih = il + ex3_e_dat(3,1,in,i_sendtile) | 
| 107 | 
  | 
  | 
              ENDIF | 
| 108 | 
  | 
  | 
            ENDIF | 
| 109 | 
  | 
  | 
            is = 1 | 
| 110 | 
  | 
  | 
            IF (il .GT. ih) is = -1 | 
| 111 | 
  | 
  | 
 | 
| 112 | 
  | 
  | 
C           =====  J direction  ===== | 
| 113 | 
  | 
  | 
            jl = ex3_e_dat(2,2,in,i_sendtile) | 
| 114 | 
  | 
  | 
            IF ( ex3_e_dat(1,2,in,i_sendtile) .EQ. 0 ) THEN | 
| 115 | 
  | 
  | 
              jh = ex3_e_dat(3,2,in,i_sendtile) | 
| 116 | 
  | 
  | 
            ELSE | 
| 117 | 
  | 
  | 
              IF (IABS(ex3_e_dat(1,2,in,i_sendtile)) .EQ. 1) THEN | 
| 118 | 
  | 
  | 
                jh = jl + ex3_e_dat(1,2,in,i_sendtile) * OLy | 
| 119 | 
  | 
  | 
              ELSE | 
| 120 | 
  | 
  | 
                jh = jl + ex3_e_dat(3,2,in,i_sendtile) | 
| 121 | 
  | 
  | 
              ENDIF | 
| 122 | 
  | 
  | 
            ENDIF | 
| 123 | 
  | 
  | 
            js = 1 | 
| 124 | 
  | 
  | 
            IF (jl .GT. jh) js = -1 | 
| 125 | 
  | 
  | 
 | 
| 126 | 
  | 
  | 
            io1 = 0 | 
| 127 | 
  | 
  | 
            jo1 = 0 | 
| 128 | 
  | 
  | 
            ko1 = 0 | 
| 129 | 
  | 
  | 
            CALL EX3_SEND_RX1( | 
| 130 | 
  | 
  | 
     I           bufftag, recvProc, | 
| 131 | 
  | 
  | 
     I           along_i, | 
| 132 | 
  | 
  | 
     I           il,ih,is,  jl,jh,js,  kl,kh,ks, | 
| 133 | 
  | 
  | 
     I           io1,jo1,ko1, | 
| 134 | 
  | 
  | 
     I           idl1,idh1, jdl1,jdh1, kdl1,kdh1, | 
| 135 | 
  | 
  | 
     I           phi, | 
| 136 | 
  | 
  | 
C    B           buff, n_buff, msgID, | 
| 137 | 
  | 
  | 
     B           EX3_B_RX(1,in,iloc), EX3_MAX_BL, msgID(iloc), | 
| 138 | 
  | 
  | 
     I           commType, | 
| 139 | 
  | 
  | 
     I           myThid ) | 
| 140 | 
  | 
  | 
          ENDDO | 
| 141 | 
  | 
  | 
        ENDDO | 
| 142 | 
  | 
  | 
 | 
| 143 | 
  | 
  | 
C       Then receive | 
| 144 | 
  | 
  | 
        DO iloc = myBxLo(myThid), myBxHi(myThid) | 
| 145 | 
  | 
  | 
          i_recvtile = ex3_p_itile(iloc) | 
| 146 | 
  | 
  | 
          nN = ex3_e_n(i_recvtile) | 
| 147 | 
  | 
  | 
          DO in = 1,nN | 
| 148 | 
  | 
  | 
            i_sendtile = ex3_e_iopt(in,i_sendtile) | 
| 149 | 
  | 
  | 
            CALL EX3_GET_BUFFTAG( | 
| 150 | 
  | 
  | 
     I           i_sendtile, i_recvtile, in, | 
| 151 | 
  | 
  | 
     O           bufftag, | 
| 152 | 
  | 
  | 
     I           myThid ) | 
| 153 | 
  | 
  | 
            sendProc = ex3_t_iproc(i_sendtile) | 
| 154 | 
  | 
  | 
 | 
| 155 | 
  | 
  | 
C           =====  I direction  ===== | 
| 156 | 
  | 
  | 
            il = ex3_e_dat(2,1,in,i_sendtile) | 
| 157 | 
  | 
  | 
            IF ( ex3_e_dat(1,1,in,i_sendtile) .EQ. 0 ) THEN | 
| 158 | 
  | 
  | 
              along_i  = .FALSE. | 
| 159 | 
  | 
  | 
              ih = ex3_e_dat(3,1,in,i_sendtile) | 
| 160 | 
  | 
  | 
            ELSE | 
| 161 | 
  | 
  | 
C             Here, "along" means the i dimension is perpendicular to | 
| 162 | 
  | 
  | 
C             the "seam" between the two tiles | 
| 163 | 
  | 
  | 
              along_i  = .TRUE. | 
| 164 | 
  | 
  | 
              IF (IABS(ex3_e_dat(1,1,in,i_sendtile)) .EQ. 1) THEN | 
| 165 | 
  | 
  | 
                ih = il + ex3_e_dat(1,1,in,i_sendtile) * OLx | 
| 166 | 
  | 
  | 
              ELSE | 
| 167 | 
  | 
  | 
                ih = il + ex3_e_dat(3,1,in,i_sendtile) | 
| 168 | 
  | 
  | 
              ENDIF | 
| 169 | 
  | 
  | 
            ENDIF | 
| 170 | 
  | 
  | 
            is = 1 | 
| 171 | 
  | 
  | 
            IF (il .GT. ih) is = -1 | 
| 172 | 
  | 
  | 
 | 
| 173 | 
  | 
  | 
C           =====  J direction  ===== | 
| 174 | 
  | 
  | 
            jl = ex3_e_dat(2,2,in,i_sendtile) | 
| 175 | 
  | 
  | 
            IF ( ex3_e_dat(1,2,in,i_sendtile) .EQ. 0 ) THEN | 
| 176 | 
  | 
  | 
              jh = ex3_e_dat(3,2,in,i_sendtile) | 
| 177 | 
  | 
  | 
            ELSE | 
| 178 | 
  | 
  | 
              IF (IABS(ex3_e_dat(1,2,in,i_sendtile)) .EQ. 1) THEN | 
| 179 | 
  | 
  | 
                jh = jl + ex3_e_dat(1,2,in,i_sendtile) * OLy | 
| 180 | 
  | 
  | 
              ELSE | 
| 181 | 
  | 
  | 
                jh = jl + ex3_e_dat(3,2,in,i_sendtile) | 
| 182 | 
  | 
  | 
              ENDIF | 
| 183 | 
  | 
  | 
            ENDIF | 
| 184 | 
  | 
  | 
            js = 1 | 
| 185 | 
  | 
  | 
            IF (jl .GT. jh) js = -1 | 
| 186 | 
  | 
  | 
 | 
| 187 | 
  | 
  | 
            CALL EX3_RECV_RX1( | 
| 188 | 
  | 
  | 
     I           bufftag, sendProc, | 
| 189 | 
  | 
  | 
     I           along_i, | 
| 190 | 
  | 
  | 
     I           il,ih,is,  jl,jh,js,  kl,kh,ks, | 
| 191 | 
  | 
  | 
     I           idl1,idh1, jdl1,jdh1, kdl1,kdh1, | 
| 192 | 
  | 
  | 
     I           phi, | 
| 193 | 
  | 
  | 
C    B           buff, n_buff, | 
| 194 | 
  | 
  | 
     B           EX3_B_RX(1,in,iloc), EX3_MAX_BL, | 
| 195 | 
  | 
  | 
     I           commType, | 
| 196 | 
  | 
  | 
     I           myThid ) | 
| 197 | 
  | 
  | 
          ENDDO | 
| 198 | 
  | 
  | 
        ENDDO | 
| 199 | 
  | 
  | 
 | 
| 200 | 
  | 
  | 
      ELSEIF (gtype(1:1) .EQ. 'Z' .OR. gtype(1:1) .EQ. 'V') THEN | 
| 201 | 
  | 
  | 
 | 
| 202 | 
  | 
  | 
C       phi is a scalar located at Arakawa vorticity (cell-corner) | 
| 203 | 
  | 
  | 
C       points | 
| 204 | 
  | 
  | 
 | 
| 205 | 
  | 
  | 
      ELSE | 
| 206 | 
  | 
  | 
        WRITE(msgbuf,'(3a)') | 
| 207 | 
  | 
  | 
     &       'EX3_XY_RX ERROR: grid type ''', gtype(1:1),  | 
| 208 | 
  | 
  | 
     &       ''' is invalid -- please use one of [MTZV]' | 
| 209 | 
  | 
  | 
        CALL print_error(msgbuf, mythid) | 
| 210 | 
  | 
  | 
        STOP 'ABNORMAL END: S/R EX3_XY_RX' | 
| 211 | 
  | 
  | 
      ENDIF | 
| 212 | 
  | 
  | 
 | 
| 213 | 
  | 
  | 
      CALL BAR2(myThid) | 
| 214 | 
  | 
  | 
 | 
| 215 | 
  | 
  | 
      RETURN | 
| 216 | 
  | 
  | 
      END | 
| 217 | 
  | 
  | 
 | 
| 218 | 
  | 
  | 
C---+----1----+----2----+----3----+----4----+----5----+----6----+----7-|--+----| | 
| 219 | 
  | 
  | 
 | 
| 220 | 
  | 
  | 
CEH3 ;;; Local Variables: *** | 
| 221 | 
  | 
  | 
CEH3 ;;; mode:fortran *** | 
| 222 | 
  | 
  | 
CEH3 ;;; End: *** |