| 1 | 
C $Header: /u/gcmpack/MITgcm/pkg/ex3/ex3_recv_rx2.template,v 1.2 2006/02/06 21:09:54 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_RECV_RX2 | 
| 9 | 
 | 
| 10 | 
C     !INTERFACE: | 
| 11 | 
      SUBROUTINE EX3_RECV_RX2( | 
| 12 | 
     I     bufftag, sendProc, | 
| 13 | 
     I     il,ih,is,  jl,jh,js,  kl,kh,ks, | 
| 14 | 
     I     idl1,idh1, jdl1,jdh1, kdl1,kdh1, | 
| 15 | 
     I     array1, | 
| 16 | 
     I     idl2,idh2, jdl2,jdh2, kdl2,kdh2, | 
| 17 | 
     I     array2, | 
| 18 | 
     B     buff, n_buff, | 
| 19 | 
     I     commType, | 
| 20 | 
     I     myThid ) | 
| 21 | 
 | 
| 22 | 
C     !DESCRIPTION: | 
| 23 | 
C     Scalar exchange routine which simply copies values from an input | 
| 24 | 
C     array to a buffer in the specified order and then, if necessary, | 
| 25 | 
C     calls the appropriate MPI send function. | 
| 26 | 
 | 
| 27 | 
C     !USES: | 
| 28 | 
      IMPLICIT NONE | 
| 29 | 
#include "SIZE.h" | 
| 30 | 
#include "EEPARAMS.h" | 
| 31 | 
#include "EESUPPORT.h" | 
| 32 | 
 | 
| 33 | 
C     !INPUT PARAMETERS: | 
| 34 | 
      INTEGER bufftag, sendProc | 
| 35 | 
      INTEGER il,ih,is, jl,jh,js, kl,kh,ks | 
| 36 | 
      INTEGER idl1,idh1, jdl1,jdh1, kdl1,kdh1 | 
| 37 | 
      INTEGER idl2,idh2, jdl2,jdh2, kdl2,kdh2 | 
| 38 | 
      _RX array1( idl1:idh1, jdl1:jdh1, kdl1:kdh1 ) | 
| 39 | 
      _RX array2( idl2:idh2, jdl2:jdh2, kdl2:kdh2 ) | 
| 40 | 
      INTEGER n_buff | 
| 41 | 
      _RX buff( n_buff ) | 
| 42 | 
      CHARACTER commType | 
| 43 | 
      INTEGER myThid | 
| 44 | 
CEOP | 
| 45 | 
 | 
| 46 | 
C     !LOCAL VARIABLES | 
| 47 | 
      INTEGER ii,jj,kk, ntb, nbv, nbv2 | 
| 48 | 
#ifdef ALLOW_USE_MPI | 
| 49 | 
      INTEGER mpiStatus(MPI_STATUS_SIZE) | 
| 50 | 
      INTEGER mpirc | 
| 51 | 
#endif | 
| 52 | 
      character*(100) msgbuf | 
| 53 | 
 | 
| 54 | 
C     Calculate buffer sizes | 
| 55 | 
      nbv =  IABS((ih-il+1)/is)  | 
| 56 | 
     &     * IABS((jh-jl+1)/js) | 
| 57 | 
     &     * IABS((kh-kl+1)/ks) | 
| 58 | 
      nbv2 = nbv*2 | 
| 59 | 
      IF ( nbv2 .GT. n_buff ) THEN | 
| 60 | 
        WRITE(msgbuf,'(2a)') 'EX3_RECV_RX2 ERROR: buffer too small', | 
| 61 | 
     &       '--please increase EX3_BUFF_FAC in ''EX3_SIZE.h''' | 
| 62 | 
        CALL print_error(msgbuf, mythid) | 
| 63 | 
        WRITE(msgbuf,'(a,i9)') '  current buffer length   = ', n_buff | 
| 64 | 
        CALL print_error(msgbuf, mythid) | 
| 65 | 
        WRITE(msgbuf,'(a,i9)') '  requested buffer length = ', nbv | 
| 66 | 
        CALL print_error(msgbuf, mythid) | 
| 67 | 
        STOP 'ABNORMAL END: S/R EX3_RECV_RX2' | 
| 68 | 
      ENDIF | 
| 69 | 
 | 
| 70 | 
      IF ( commType .EQ. 'P' ) THEN | 
| 71 | 
C       We may eventually need (?) some sort of synchronization | 
| 72 | 
C       mechanism for multi-threaded mode | 
| 73 | 
      ELSEIF ( commType .EQ. 'M' ) THEN | 
| 74 | 
#ifdef ALLOW_USE_MPI | 
| 75 | 
        CALL MPI_Recv( | 
| 76 | 
     I       buff, nbv2, EX3_MPI_TYPE_RX, sendProc, bufftag, | 
| 77 | 
     I       MPI_COMM_MODEL, | 
| 78 | 
     O       mpiStatus, mpirc ) | 
| 79 | 
#endif | 
| 80 | 
      ELSE | 
| 81 | 
        WRITE(msgbuf,'(3a)') 'EX3_RECV_RX2 ERROR: commType ''', | 
| 82 | 
     &       commType, ''' is invalid' | 
| 83 | 
        CALL print_error(msgbuf, mythid) | 
| 84 | 
        STOP 'ABNORMAL END: S/R EX3_RECV_RX2' | 
| 85 | 
      ENDIF | 
| 86 | 
 | 
| 87 | 
C     Copy the values into the buffer in the specified order | 
| 88 | 
      ntb = 0 | 
| 89 | 
      DO kk = kl, kh, ks | 
| 90 | 
        DO jj = jl, jh, js | 
| 91 | 
          DO ii = il, ih, is | 
| 92 | 
            ntb = ntb + 1 | 
| 93 | 
            array1( ii, jj, kk ) = buff( ntb ) | 
| 94 | 
            array2( ii, jj, kk ) = buff( nbv + ntb ) | 
| 95 | 
          ENDDO | 
| 96 | 
        ENDDO | 
| 97 | 
      ENDDO | 
| 98 | 
       | 
| 99 | 
      RETURN | 
| 100 | 
      END | 
| 101 | 
 | 
| 102 | 
C---+----1----+----2----+----3----+----4----+----5----+----6----+----7-|--+----| | 
| 103 | 
 | 
| 104 | 
CEH3 ;;; Local Variables: *** | 
| 105 | 
CEH3 ;;; mode:fortran *** | 
| 106 | 
CEH3 ;;; End: *** |