C $Header: /u/gcmpack/MITgcm/pkg/exch2/exch2_get_uv_bounds.F,v 1.4 2012/09/04 00:40:25 jmc Exp $
C $Name:  $

#include "CPP_EEOPTIONS.h"

CBOP
C     !ROUTINE: EXCH2_GET_UV_BOUNDS

C     !INTERFACE:
      SUBROUTINE EXCH2_GET_UV_BOUNDS(
     I                 fCode, eWdth, updateCorners,
     I                 tgTile, tgNb,
     O                 tIlo1, tIhi1, tJlo1, tJhi1,
     O                 tIlo2, tIhi2, tJlo2, tJhi2,
     O                 tiStride, tjStride,
     O                 e2_oi1, e2_oj1, e2_oi2, e2_oj2,
     I                 myThid )

C     !DESCRIPTION:
C     Return the index range & step of the part of the array (overlap-region)
C     which is going to be updated by the exchange with 1 neighbour.
C     2 components vector field (UV) version.


C     !USES:
      IMPLICIT NONE
C     == Global data ==
#include "SIZE.h"
#include "W2_EXCH2_SIZE.h"
#include "W2_EXCH2_TOPOLOGY.h"

C     !INPUT/OUTPUT PARAMETERS:
C     fCode         :: field code (position on staggered grid)
C     eWdth         :: width of data region to exchange
C     updateCorners :: flag, do update corner halo region if true
C     tgTile        :: target tile
C     tgNb          :: target Neighbour entry
C     tIlo1, tIhi1  :: index range in I that will be filled in 1rst comp. array
C     tJlo1, tJhi1  :: index range in J that will be filled in 1rst comp. array
C     tIlo2, tIhi2  :: index range in I that will be filled in 2nd  comp. array
C     tJlo2, tJhi2  :: index range in J that will be filled in 2nd  comp. array
C     tiStride      :: index step  in I that will be filled in target arrays
C     tjStride      :: index step  in J that will be filled in target arrays
C     e2_oi1        :: index offset in target to source-1 index relation
C     e2_oj1        :: index offset in target to source-1 index relation
C     e2_oi2        :: index offset in target to source-2 index relation
C     e2_oj2        :: index offset in target to source-2 index relation
C     myThid        :: my Thread Id. number

      CHARACTER*2 fCode
      INTEGER     eWdth
      LOGICAL     updateCorners
      INTEGER     tgTile, tgNb
      INTEGER     tIlo1, tIhi1, tJlo1, tJhi1
      INTEGER     tIlo2, tIhi2, tJlo2, tJhi2
      INTEGER     tiStride, tjStride
      INTEGER     e2_oi1, e2_oj1
      INTEGER     e2_oi2, e2_oj2
      INTEGER     myThid
C
C     !LOCAL VARIABLES:
C     soTile        :: source tile
C     soNb          :: source Neighbour entry
      INTEGER  soTile
      INTEGER  soNb
      INTEGER  tIlo,  tIhi,  tJlo,  tJhi
      INTEGER  i, e2_pij(4)

C---  exch2 target to source index relation:
C     is = pij(1)*it + pij(2)*jt + oi
C     js = pij(3)*it + pij(4)*jt + oj

C---  Initialise index range from Topology values:
      tIlo = exch2_iLo(tgNb,tgTile)
      tIhi = exch2_iHi(tgNb,tgTile)
      tJlo = exch2_jLo(tgNb,tgTile)
      tJhi = exch2_jHi(tgNb,tgTile)
      soNb = exch2_opposingSend(tgNb,tgTile)
      soTile = exch2_neighbourId(tgNb,tgTile)
      e2_oi1 = exch2_oi(soNb,soTile)
      e2_oj1 = exch2_oj(soNb,soTile)
      DO i=1,4
        e2_pij(i) = exch2_pij(i,soNb,soTile)
      ENDDO

C---  Expand index range according to exchange-Width "eWdth"
      IF ( tIlo.EQ.tIhi .AND. tIlo.EQ.0 ) THEN
C      Filling a west edge overlap
       tIlo1 = 1-eWdth
       tIhi1 = 0
       tiStride = 1
       IF ( tJlo.LE.tJhi ) THEN
        tjStride=1
       ELSE
        tjStride=-1
       ENDIF
       IF ( updateCorners ) THEN
        tJlo1 = tJlo-tjStride*(eWdth-1)
        tJhi1 = tJhi+tjStride*(eWdth-1)
       ELSE
        tJlo1 = tJlo+tjStride
        tJhi1 = tJhi-tjStride
       ENDIF
      ENDIF
      IF ( tIlo.EQ.tIhi .AND. tIlo.GT.1 ) THEN
C      Filling an east edge overlap
       tIlo1 = tIlo
       tIhi1 = tIhi+eWdth-1
       tiStride = 1
       IF ( tJlo.LE.tJhi ) THEN
        tjStride = 1
       ELSE
        tjStride =-1
       ENDIF
       IF ( updateCorners ) THEN
        tJlo1 = tJlo-tjStride*(eWdth-1)
        tJhi1 = tJhi+tjStride*(eWdth-1)
       ELSE
        tJlo1 = tJlo+tjStride
        tJhi1 = tJhi-tjStride
       ENDIF
      ENDIF
      IF ( tJlo.EQ.tJhi .AND. tJlo.EQ.0 ) THEN
C      Filling a south edge overlap
       tJlo1 = 1-eWdth
       tJhi1 = 0
       tjStride = 1
       IF ( tIlo .LE. tIhi ) THEN
        tiStride = 1
       ELSE
        tiStride =-1
       ENDIF
       IF ( updateCorners ) THEN
        tIlo1 = tIlo-tiStride*(eWdth-1)
        tIhi1 = tIhi+tiStride*(eWdth-1)
       ELSE
        tIlo1 = tIlo+tiStride
        tIhi1 = tIhi-tiStride
       ENDIF
      ENDIF
      IF ( tJlo.EQ.tJhi .AND. tJlo.GT.1 ) THEN
C      Filling a north edge overlap
       tJlo1 = tJlo
       tJhi1 = tJhi+eWdth-1
       tjStride = 1
       IF ( tIlo.LE.tIhi ) THEN
        tiStride = 1
       ELSE
        tiStride =-1
       ENDIF
       IF ( updateCorners ) THEN
        tIlo1 = tIlo-tiStride*(eWdth-1)
        tIhi1 = tIhi+tiStride*(eWdth-1)
       ELSE
        tIlo1 = tIlo+tiStride
        tIhi1 = tIhi-tiStride
       ENDIF
      ENDIF

C---  copy to 2nd set of indices
      tIlo2 = tIlo1
      tIhi2 = tIhi1
      tJlo2 = tJlo1
      tJhi2 = tJhi1
      e2_oi2 = e2_oi1
      e2_oj2 = e2_oj1

      IF ( fCode.EQ.'Cg' ) THEN
C---  UV C-Grid specific code: start here

C---  half grid-cell location with inverse index relation
C     => increase the offset by 1 (relative to tracer cell-centered offset)
C     if pij(1) is -1 then +i in source aligns with -i in target
C     if pij(3) is -1 then +j in source aligns with -i in target
        IF ( e2_pij(1) .EQ. -1 ) THEN
         e2_oi1 = e2_oi1 + 1
        ENDIF
        IF ( e2_pij(3) .EQ. -1 ) THEN
         e2_oj1 = e2_oj1 + 1
        ENDIF
C     if pij(2) is -1 then +i in source aligns with -j in target
C     if pij(4) is -1 then +j in source aligns with -j in target
        IF ( e2_pij(2) .EQ. -1 ) THEN
         e2_oi2 = e2_oi2 + 1
        ENDIF
        IF ( e2_pij(4) .EQ. -1 ) THEN
         e2_oj2 = e2_oj2 + 1
        ENDIF

C---  adjust index lower and upper bounds (fct of updateCorners):
       IF ( updateCorners ) THEN

C--   as a consequence, need also to increase the index lower bound
C     (avoid "out-of bounds" problems ; formerly itlreduce,jtlreduce)
        IF ( e2_pij(1).EQ.-1 .OR. e2_pij(3).EQ.-1 ) tIlo1 = tIlo1+1
        IF ( e2_pij(2).EQ.-1 .OR. e2_pij(4).EQ.-1 ) tJlo2 = tJlo2+1

C---  Avoid updating (some) tile-corner halo region if across faces
c       IF ( tIlo.EQ.tIhi .AND. tIlo.EQ.0 ) THEN
c         IF ( exch2_isSedge(tgTile).EQ.1 ) THEN
C-      West edge is touching the face S edge
c           tJlo1 = tJlo+1
c           tJlo2 = tJlo+1
c         ENDIF
c         IF ( exch2_isNedge(tgTile).EQ.1 ) THEN
C-      West edge is touching the face N edge
c           tJhi1 = tJhi-1
c           tJhi2 = tJhi
c         ENDIF
c       ENDIF
        IF ( tIlo.EQ.tIhi .AND. tIlo.GT.1 ) THEN
          IF ( exch2_isSedge(tgTile).EQ.1 ) THEN
C-      East edge is touching the face S edge
            tJlo1 = tJlo+1
            tJlo2 = tJlo+1
          ENDIF
          IF ( exch2_isNedge(tgTile).EQ.1 ) THEN
C-      East edge is touching the face N edge
            tJhi1 = tJhi-1
            tJhi2 = tJhi
          ENDIF
        ENDIF
c       IF ( tJlo.EQ.tJhi .AND. tJlo.EQ.0 ) THEN
c         IF ( exch2_isWedge(tgTile).EQ.1 ) THEN
C-      South edge is touching the face W edge
c           tIlo1 = tIlo+1
c           tIlo2 = tIlo+1
c         ENDIF
c         IF ( exch2_isEedge(tgTile).EQ.1 ) THEN
C-      South Edge is touching the face E edge
c           tIhi1 = tIhi
c           tIhi2 = tIhi-1
c         ENDIF
c       ENDIF
        IF ( tJlo.EQ.tJhi .AND. tJlo.GT.1 ) THEN
          IF ( exch2_isWedge(tgTile).EQ.1 ) THEN
C-      North edge is touching the face W edge
            tIlo1 = tIlo+1
            tIlo2 = tIlo+1
          ENDIF
          IF ( exch2_isEedge(tgTile).EQ.1 ) THEN
C-      North Edge is touching the face E edge
            tIhi1 = tIhi
            tIhi2 = tIhi-1
          ENDIF
        ENDIF

       ELSE
C---  adjust index lower and upper bounds (updateCorners = F case):
        IF ( e2_pij(1).EQ.-1 .OR. e2_pij(3).EQ.-1 ) THEN
          tIlo1 = tIlo1+1
          tIhi1 = tIhi1+1
        ENDIF
        IF ( e2_pij(2).EQ.-1 .OR. e2_pij(4).EQ.-1 ) THEN
          tJlo2 = tJlo2+1
          tJhi2 = tJhi2+1
        ENDIF
       ENDIF

C---  UV C-Grid specific code: end

      ELSEIF ( fCode.NE.'Ag' ) THEN
        STOP 'ABNORMAL END: S/R EXCH2_GET_UV_BOUNDS (wrong fCode)'
      ENDIF

      RETURN
      END