--- MITgcm/pkg/exf/exf_interp.F 2003/12/07 07:41:45 1.4 +++ MITgcm/pkg/exf/exf_interp.F 2006/07/02 13:02:22 1.18 @@ -36,7 +36,7 @@ I infile, I filePrec, O arrayout, - I irecord, xG, yG, + I irecord, xG_in, yG, I lon_0, lon_inc, I lat_0, lat_inc, I nx_in, ny_in, method, mythid) @@ -52,151 +52,170 @@ C lon_inc = scalar x-grid increment C lat_inc = vector y-grid increments C nx_in, ny_in = input x-grid and y-grid size -C method = 1 for bilinear 2 for bicubic +C method = 1,11,21 for bilinear; 2,12,22 for bicubic +C 1,2 for tracer; 11,12 for U; 21,22 for V C mythid = thread id C #include "SIZE.h" #include "EEPARAMS.h" -#ifdef ALLOW_USE_MPI -# include "EESUPPORT.h" -# include "PARAMS.h" -#endif /* ALLOW_USE_MPI */ +#include "PARAMS.h" C subroutine variables character*(*) infile integer filePrec, irecord, nx_in, ny_in _RL arrayout(1-OLx:sNx+OLx,1-OLy:sNy+OLy,nSx,nSy) - _RS xG (1-OLx:sNx+OLx,1-OLy:sNy+OLy,nSx,nSy) + _RS xG_in (1-OLx:sNx+OLx,1-OLy:sNy+OLy,nSx,nSy) _RS yG (1-OLx:sNx+OLx,1-OLy:sNy+OLy,nSx,nSy) _RL lon_0, lon_inc _RL lat_0, lat_inc(ny_in-1) integer method, mythid C local variables - integer ierr - real*8 ne_fac,nw_fac,se_fac,sw_fac - integer e_ind(snx),w_ind(snx) - integer n_ind(sny),s_ind(sny) + integer e_ind(snx,sny),w_ind(snx,sny) + integer n_ind(snx,sny),s_ind(snx,sny) real*8 px_ind(4), py_ind(4), ew_val(4) external lagran real*8 lagran real*4 arrayin(-1:nx_in+2 , -1:ny_in+2) real*8 x_in (-1:nx_in+2), y_in(-1:ny_in+2) - integer i, j, k, l, js, bi, bj, interp_unit, sp - real*4 global(nx_in,ny_in) - - _BEGIN_MASTER( myThid ) - -C check input arguments - if ( .NOT. (filePrec .EQ. 32) ) - & stop 'stop in exf_interp.F: value of filePrec not allowed' - -C read in input data -#ifdef ALLOW_USE_MPI - if (useSingleCPUIO) then - -C master thread of process 0, only, opens a global file - IF( mpiMyId .EQ. 0 ) THEN - call mdsfindunit( interp_unit, mythid) - open(interp_unit,file=infile,status='old',access='direct', - & recl=nx_in*ny_in*4) - read(interp_unit,rec=irecord) - & ((global(i,j),i=1,nx_in),j=1,ny_in) - close(interp_unit) - ENDIF - -C broadcast to all processes - call MPI_BCAST(global,nx_in*ny_in,MPI_REAL, - & 0,MPI_COMM_MODEL,ierr) - do j=1,ny_in - do i=1,nx_in - arrayin(i,j)=global(i,j) + real*8 ninety + PARAMETER ( ninety = 90. ) + integer i, j, k, l, js, bi, bj, sp, interp_unit + _RS xG(1-OLx:sNx+OLx,1-OLy:sNy+OLy,nSx,nSy) + _RS threeSixtyRS, NorthValue + PARAMETER ( threeSixtyRS = 360. ) + +C put xG in interval [ lon_0 , lon_0+360 [ + do bj=myByLo(myThid),myByHi(myThid) + do bi=myBxLo(myThid),myBxHi(myThid) + do j=1-OLy,sNy+OLy + do i=1-OLx,sNx+OLx + xG(i,j,bi,bj) = xG_in(i,j,bi,bj)-lon_0 + & + threeSixtyRS*2. + xG(i,j,bi,bj) = lon_0+mod(xG(i,j,bi,bj),threeSixtyRS) enddo enddo + enddo + enddo + call exf_interp_read( + I infile, filePrec, + O arrayin, + I irecord, nx_in, ny_in, mythid) + _BARRIER + +C _BEGIN_MASTER( myThid ) + +C setup input longitude grid + do i=-1,nx_in+2 + x_in(i) = lon_0 + (i-1)*lon_inc + enddo + +C setup input latitude grid + y_in(0) = lat_0 - lat_inc(1) + y_in(-1)= lat_0 - 2.*lat_inc(1) + y_in(1) = lat_0 + do j=2,ny_in + y_in(j) = y_in(j-1) + lat_inc(j-1) + enddo + do j=ny_in+1,ny_in+2 + if (y_in(j-1).eq.ninety) then + y_in(j) = 2 * ninety - y_in(j-2) else -#endif /* ALLOW_USE_MPI */ - - call mdsfindunit( interp_unit, mythid) - open(interp_unit,file=infile,status='old',access='direct', - & recl=nx_in*ny_in*4) - read(interp_unit,rec=irecord) - & ((arrayin(i,j),i=1,nx_in),j=1,ny_in) - close(interp_unit) - -#ifdef ALLOW_USE_MPI + y_in(j) = min( y_in(j-1)+lat_inc(ny_in-1), ninety ) endif -#endif /* ALLOW_USE_MPI */ - -#ifdef _BYTESWAPIO - call MDS_BYTESWAPR4((nx_in+4)*(ny_in+4), arrayin ) -#endif /* _BYTESWAPIO */ - -C setup input grid - do i=-1,nx_in+2 - x_in(i) = lon_0 + (i-1.)*lon_inc - enddo - y_in(0) = lat_0 - lat_inc(1) - y_in(-1)= lat_0 - 2.*lat_inc(1) - y_in(1) = lat_0 - do j=2,ny_in - y_in(j) = y_in(j-1) + lat_inc(j-1) - enddo - y_in(ny_in+1) = y_in(ny_in) + lat_inc(ny_in-1) - y_in(ny_in+2) = y_in(ny_in) + 2.*lat_inc(ny_in-1) + enddo C enlarge boundary - do j=1,ny_in - arrayin(0,j) = arrayin(nx_in,j) - arrayin(-1,j) = arrayin(nx_in-1,j) - arrayin(nx_in+1,j) = arrayin(1,j) - arrayin(nx_in+2,j) = arrayin(2,j) - enddo - do i=-1,nx_in+2 - arrayin(i,0) = arrayin(i,1) - arrayin(i,-1) = arrayin(i,1) - arrayin(i,ny_in+1) = arrayin(i,ny_in) - arrayin(i,ny_in+2) = arrayin(i,ny_in) - enddo - - _END_MASTER( myThid ) + do j=1,ny_in + arrayin(0,j) = arrayin(nx_in,j) + arrayin(-1,j) = arrayin(nx_in-1,j) + arrayin(nx_in+1,j) = arrayin(1,j) + arrayin(nx_in+2,j) = arrayin(2,j) + enddo + do i=-1,nx_in+2 + arrayin(i,0) = arrayin(i,1) + arrayin(i,-1) = arrayin(i,1) + arrayin(i,ny_in+1) = arrayin(i,ny_in) + arrayin(i,ny_in+2) = arrayin(i,ny_in) + enddo + +C For tracer (method=1,2) set to northernmost zonal-mean value +C at 90N to avoid sharp zonal gradients near the Pole. +C For U (method=11,12) set to zero at 90N to minimize velocity +C gradient at North Pole +C For V (method=11,12) set to northernmost zonal value at 90N, +C as is already done above in order to allow cross-PoleArctic flow + do j=ny_in,ny_in+2 + if (y_in(j).eq.ninety) then + if (method.eq.1 .or. method.eq.2) then + NorthValue = 0 + do i=1,nx_in + NorthValue = NorthValue + arrayin(i,j) + enddo + NorthValue = NorthValue / nx_in + do i=-1,nx_in+2 + arrayin(i,j) = NorthValue + enddo + elseif (method.eq.11 .or. method.eq.12) then + do i=-1,nx_in+2 + arrayin(i,j) = 0 + enddo + endif + endif + enddo +C _END_MASTER( myThid ) + do bj = mybylo(mythid), mybyhi(mythid) do bi = mybxlo(mythid), mybxhi(mythid) C check validity of input/output coordinates - if ( xG(1,1 ,bi,bj) .le. x_in(0) .or. - & xG(snx,1,bi,bj) .ge. x_in(nx_in+1) .or. - & yG(1,1 ,bi,bj) .lt. y_in(1) .or. - & yG(1,sny,bi,bj) .gt. y_in(ny_in) ) then - print*,'ERROR in S/R EXF_INTERP:' - print*,' input grid must encompass output grid.' - STOP ' ABNORMAL END: S/R EXF_INTERP' +#ifdef ALLOW_DEBUG + if ( debugLevel .ge. debLevB ) then + do i=1,snx + do j=1,sny + if ( xG(i,j,bi,bj) .lt. x_in(0) .or. + & xG(i,j,bi,bj) .ge. x_in(nx_in+1) .or. + & yG(i,j,bi,bj) .lt. y_in(0) .or. + & yG(i,j,bi,bj) .ge. y_in(ny_in+1) ) then + print*,'ERROR in S/R EXF_INTERP:' + print*,' input grid must encompass output grid.' + print*,'i,j,bi,bj' ,i,j,bi,bj + print*,'xG,yG' ,xG(i,j,bi,bj),yG(i,j,bi,bj) + print*,'nx_in,ny_in' ,nx_in ,ny_in + print*,'x_in(0,nx_in+1)',x_in(0) ,x_in(nx_in+1) + print*,'y_in(0,ny_in+1)',y_in(0) ,y_in(ny_in+1) + STOP ' ABNORMAL END: S/R EXF_INTERP' + endif + enddo + enddo endif +#endif /* ALLOW_DEBUG */ C compute interpolation indices do i=1,snx - if (xG(i,1,bi,bj)-x_in(1) .ge. 0.) then - w_ind(i) = int((xG(i,1,bi,bj)-x_in(1))/lon_inc) + 1 - else - w_ind(i) = int((xG(i,1,bi,bj)-x_in(1))/lon_inc) - endif - e_ind(i) = w_ind(i) + 1 - enddo - js = ny_in/2 - do j=1,sny - do while (yG(1,j,bi,bj) .lt. y_in(js)) - js = (js + 1)/2 - enddo - do while (yG(1,j,bi,bj) .ge. y_in(js+1)) - js = js + 1 + do j=1,sny + if (xG(i,j,bi,bj)-x_in(1) .ge. 0.) then + w_ind(i,j) = int((xG(i,j,bi,bj)-x_in(1))/lon_inc) + 1 + else + w_ind(i,j) = int((xG(i,j,bi,bj)-x_in(1))/lon_inc) + endif + e_ind(i,j) = w_ind(i,j) + 1 + js = ny_in*.5 + do while (yG(i,j,bi,bj) .lt. y_in(js)) + js = (js - 1)*.5 + enddo + do while (yG(i,j,bi,bj) .ge. y_in(js+1)) + js = js + 1 + enddo + s_ind(i,j) = js + n_ind(i,j) = js + 1 enddo - s_ind(j) = js - n_ind(j) = js + 1 enddo - if (method .eq. 1) then + if (method.eq.1 .or. method.eq.11 .or. method.eq.21) then C bilinear interpolation sp = 2 @@ -204,20 +223,20 @@ do i=1,snx arrayout(i,j,bi,bj) = 0. do l=0,1 - px_ind(l+1) = x_in(w_ind(i)+l) - py_ind(l+1) = y_in(s_ind(j)+l) + px_ind(l+1) = x_in(w_ind(i,j)+l) + py_ind(l+1) = y_in(s_ind(i,j)+l) enddo do k=1,2 - ew_val(k) = arrayin(w_ind(i),s_ind(j)+k-1) - & *lagran(1,xG(i,1,bi,bj),px_ind,sp) - & +arrayin(e_ind(i),s_ind(j)+k-1) - & *lagran(2,xG(i,1,bi,bj),px_ind,sp) + ew_val(k) = arrayin(w_ind(i,j),s_ind(i,j)+k-1) + & *lagran(1,xG(i,j,bi,bj),px_ind,sp) + & +arrayin(e_ind(i,j),s_ind(i,j)+k-1) + & *lagran(2,xG(i,j,bi,bj),px_ind,sp) arrayout(i,j,bi,bj)=arrayout(i,j,bi,bj) - & +ew_val(k)*lagran(k,yG(1,j,bi,bj),py_ind,sp) + & +ew_val(k)*lagran(k,yG(i,j,bi,bj),py_ind,sp) enddo enddo enddo - elseif (method .eq. 2) then + elseif (method .eq. 2 .or. method.eq.12 .or. method.eq.22) then C bicubic interpolation sp = 4 @@ -225,21 +244,21 @@ do i=1,snx arrayout(i,j,bi,bj) = 0. do l=-1,2 - px_ind(l+2) = x_in(w_ind(i)+l) - py_ind(l+2) = y_in(s_ind(j)+l) + px_ind(l+2) = x_in(w_ind(i,j)+l) + py_ind(l+2) = y_in(s_ind(i,j)+l) enddo do k=1,4 ew_val(k) = - & arrayin(w_ind(i)-1,s_ind(j)+k-2) - & *lagran(1,xG(i,1,bi,bj),px_ind,sp) - & +arrayin(w_ind(i) ,s_ind(j)+k-2) - & *lagran(2,xG(i,1,bi,bj),px_ind,sp) - & +arrayin(e_ind(i) ,s_ind(j)+k-2) - & *lagran(3,xG(i,1,bi,bj),px_ind,sp) - & +arrayin(e_ind(i)+1,s_ind(j)+k-2) - & *lagran(4,xG(i,1,bi,bj),px_ind,sp) + & arrayin(w_ind(i,j)-1,s_ind(i,j)+k-2) + & *lagran(1,xG(i,j,bi,bj),px_ind,sp) + & +arrayin(w_ind(i,j) ,s_ind(i,j)+k-2) + & *lagran(2,xG(i,j,bi,bj),px_ind,sp) + & +arrayin(e_ind(i,j) ,s_ind(i,j)+k-2) + & *lagran(3,xG(i,j,bi,bj),px_ind,sp) + & +arrayin(e_ind(i,j)+1,s_ind(i,j)+k-2) + & *lagran(4,xG(i,j,bi,bj),px_ind,sp) arrayout(i,j,bi,bj)=arrayout(i,j,bi,bj) - & +ew_val(k)*lagran(k,yG(1,j,bi,bj),py_ind,sp) + & +ew_val(k)*lagran(k,yG(i,j,bi,bj),py_ind,sp) enddo enddo enddo