C $Header: /u/gcmpack/MITgcm/pkg/ctrl/ctrl_set_unpack_xy.F,v 1.34 2015/06/16 20:20:26 gforget Exp $
C $Name: $
#include "CTRL_OPTIONS.h"
subroutine CTRL_SET_UNPACK_XY(
& lxxadxx, cunit, ivartype, PrecondScalar,
& fname, masktype, weighttype,
& nwetglobal, mythid)
c ==================================================================
c SUBROUTINE ctrl_set_unpack_xy
c ==================================================================
c
c o Unpack the control vector such that the land points are filled
c in.
c
c ==================================================================
implicit none
c == global variables ==
#include "EEPARAMS.h"
#include "SIZE.h"
#include "PARAMS.h"
#include "GRID.h"
#include "ctrl.h"
#include "optim.h"
c == routine arguments ==
logical lxxadxx
integer cunit
integer ivartype
_RL precondScalar
character*( 80) fname
character*( 9) masktype
character*( 80) weighttype
integer nwetglobal(nr)
integer mythid
#ifndef EXCLUDE_CTRL_PACK
# ifndef ALLOW_PACKUNPACK_METHOD2
c == local variables ==
integer bi,bj
integer ip,jp
integer i,j,k
integer ii
integer il
integer irec,nrec_nl
integer itlo,ithi
integer jtlo,jthi
integer jmin,jmax
integer imin,imax
integer cbuffindex
_RL globmsk ( snx,nsx,npx,sny,nsy,npy,nr )
_RL globfld3d( snx,nsx,npx,sny,nsy,npy,nr )
#ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO
_RL weightfld2d( snx,nsx,npx,sny,nsy,npy )
#endif
real*4 cbuff ( snx*nsx*npx*sny*nsy*npy )
character*( 80) weightname
integer reclen,irectrue
integer cunit2, cunit3
character*(80) cfile2, cfile3
real*4 globfldtmp2( snx,nsx,npx,sny,nsy,npy )
real*4 globfldtmp3( snx,nsx,npx,sny,nsy,npy )
LOGICAL doPackOld
c == external ==
integer ilnblnk
external
c == end of interface ==
jtlo = 1
jthi = nsy
itlo = 1
ithi = nsx
jmin = 1
jmax = sny
imin = 1
imax = snx
nbuffGlobal = nbuffGlobal + 1
doPackOld = (.NOT.ctrlSmoothCorrel2D).AND.(.NOT.ctrlUseGen)
c Initialise temporary file
do k = 1,nr
do jp = 1,nPy
do bj = jtlo,jthi
do j = jmin,jmax
do ip = 1,nPx
do bi = itlo,ithi
do i = imin,imax
globfld3d (i,bi,ip,j,bj,jp,k) = 0. _d 0
globmsk (i,bi,ip,j,bj,jp,k) = 0. _d 0
globfldtmp2(i,bi,ip,j,bj,jp) = 0. _d 0
globfldtmp3(i,bi,ip,j,bj,jp) = 0. _d 0
enddo
enddo
enddo
enddo
enddo
enddo
enddo
c-- Only the master thread will do I/O.
_BEGIN_MASTER( mythid )
if ( doPackDiag ) then
write(cfile2(1:80),'(80a)') ' '
write(cfile3(1:80),'(80a)') ' '
if ( lxxadxx ) then
write(cfile2(1:80),'(a,I3.3,a,I4.4,a)')
& 'diag_unpack_nondim_ctrl_',
& ivartype, '_', optimcycle, '.bin'
write(cfile3(1:80),'(a,I3.3,a,I4.4,a)')
& 'diag_unpack_dimens_ctrl_',
& ivartype, '_', optimcycle, '.bin'
else
write(cfile2(1:80),'(a,I3.3,a,I4.4,a)')
& 'diag_unpack_nondim_grad_',
& ivartype, '_', optimcycle, '.bin'
write(cfile3(1:80),'(a,I3.3,a,I4.4,a)')
& 'diag_unpack_dimens_grad_',
& ivartype, '_', optimcycle, '.bin'
endif
reclen = FLOAT(snx*nsx*npx*sny*nsy*npy*4)
call MDSFINDUNIT( cunit2, mythid )
open( cunit2, file=cfile2, status='unknown',
& access='direct', recl=reclen )
call MDSFINDUNIT( cunit3, mythid )
open( cunit3, file=cfile3, status='unknown',
& access='direct', recl=reclen )
endif
#ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO
if (weighttype.NE.' ') then
il=ilnblnk( weighttype)
write(weightname(1:80),'(80a)') ' '
write(weightname(1:80),'(a)') weighttype(1:il)
call MDSREADFIELD_2D_GL(
& weightname, ctrlprec, 'RL',
& 1, weightfld2d, 1, mythid)
else
do jp = 1,nPy
do bj = jtlo,jthi
do j = jmin,jmax
do ip = 1,nPx
do bi = itlo,ithi
do i = imin,imax
weightfld2d(i,bi,ip,j,bj,jp) = 1. _d 0
enddo
enddo
enddo
enddo
enddo
enddo
endif
#endif
call MDSREADFIELD_3D_GL(
& masktype, ctrlprec, 'RL',
& Nr, globmsk, 1, mythid)
nrec_nl=int(ncvarrecs(ivartype)/Nr)
do irec = 1, nrec_nl
do k = 1,Nr
irectrue = (irec-1)*nr + k
#ifndef ALLOW_ADMTLM
read(cunit) filencvarindex(ivartype)
if (filencvarindex(ivartype) .NE. ncvarindex(ivartype))
& then
print *, 'ctrl_set_unpack_xy:WARNING: wrong ncvarindex ',
& filencvarindex(ivartype), ncvarindex(ivartype)
STOP 'in S/R ctrl_set_unpack_xy'
endif
read(cunit) filej
read(cunit) filei
#endif /* ndef ALLOW_ADMTLM */
cbuffindex = nwetglobal(1)
if ( cbuffindex .gt. 0 ) then
#ifndef ALLOW_ADMTLM
read(cunit) filencbuffindex
if (filencbuffindex .NE. cbuffindex) then
print *, 'WARNING: wrong cbuffindex ',
& filencbuffindex, cbuffindex
STOP 'in S/R ctrl_set_unpack_xy'
endif
read(cunit) filek
if (filek .NE. 1) then
print *, 'WARNING: wrong k ',
& filek, 1
STOP 'in S/R ctrl_set_unpack_xy'
endif
cph#endif /* ndef ALLOW_ADMTLM */
read(cunit) (cbuff(ii), ii=1,cbuffindex)
#endif /* ndef ALLOW_ADMTLM */
endif
c
cbuffindex = 0
do jp = 1,nPy
do bj = jtlo,jthi
do j = jmin,jmax
do ip = 1,nPx
do bi = itlo,ithi
do i = imin,imax
if ( globmsk(i,bi,ip,j,bj,jp,1) .ne. 0. ) then
cbuffindex = cbuffindex + 1
globfld3d(i,bi,ip,j,bj,jp,k) = cbuff(cbuffindex)
cph(
globfldtmp2(i,bi,ip,j,bj,jp) = cbuff(cbuffindex)
cph)
#ifdef ALLOW_ADMTLM
nveccount = nveccount + 1
globfld3d(i,bi,ip,j,bj,jp,k) =
& phtmpadmtlm(nveccount)
cph(
globfldtmp2(i,bi,ip,j,bj,jp) =
& phtmpadmtlm(nveccount)
cph)
#endif
IF ( doPackOld ) THEN
if ( lxxadxx ) then
globfld3d(i,bi,ip,j,bj,jp,k) =
& globfld3d(i,bi,ip,j,bj,jp,k)
#ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO
& / sqrt(weightfld2d(i,bi,ip,j,bj,jp))
#endif
& * precondScalar
else
globfld3d(i,bi,ip,j,bj,jp,k) =
& globfld3d(i,bi,ip,j,bj,jp,k)
#ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO
& * sqrt(weightfld2d(i,bi,ip,j,bj,jp))
#endif
& / precondScalar
endif
ELSE !IF ( doPackOld ) THEN
if ( lxxadxx ) then
globfld3d(i,bi,ip,j,bj,jp,k) =
& globfld3d(i,bi,ip,j,bj,jp,k)
& * precondScalar
else
globfld3d(i,bi,ip,j,bj,jp,k) =
& globfld3d(i,bi,ip,j,bj,jp,k)
& / precondScalar
endif
ENDIF !IF ( doPackOld ) THEN
else
globfld3d(i,bi,ip,j,bj,jp,k) = 0. _d 0
endif
cph(
globfldtmp3(i,bi,ip,j,bj,jp) =
& globfld3d(i,bi,ip,j,bj,jp,k)
cph)
enddo
enddo
enddo
enddo
enddo
enddo
cph(
if ( doPackDiag ) then
write(cunit2,rec=irectrue) globfldtmp2
write(cunit3,rec=irectrue) globfldtmp3
endif
cph)
enddo
call MDSWRITEFIELD_3D_GL( fname, ctrlprec, 'RL',
& NR, globfld3d,
& irec, optimcycle, mythid)
enddo
do irec = nrec_nl*Nr+1,ncvarrecs(ivartype)
#ifndef ALLOW_ADMTLM
read(cunit) filencvarindex(ivartype)
if (filencvarindex(ivartype) .NE. ncvarindex(ivartype))
& then
print *, 'ctrl_set_unpack_xy:WARNING: wrong ncvarindex ',
& filencvarindex(ivartype), ncvarindex(ivartype)
STOP 'in S/R ctrl_set_unpack_xy'
endif
read(cunit) filej
read(cunit) filei
#endif /* ALLOW_ADMTLM */
do k = 1,1
irectrue = irec
cbuffindex = nwetglobal(k)
if ( cbuffindex .gt. 0 ) then
#ifndef ALLOW_ADMTLM
read(cunit) filencbuffindex
if (filencbuffindex .NE. cbuffindex) then
print *, 'WARNING: wrong cbuffindex ',
& filencbuffindex, cbuffindex
STOP 'in S/R ctrl_set_unpack_xy'
endif
read(cunit) filek
if (filek .NE. k) then
print *, 'WARNING: wrong k ',
& filek, k
STOP 'in S/R ctrl_set_unpack_xy'
endif
cph#endif /* ALLOW_ADMTLM */
read(cunit) (cbuff(ii), ii=1,cbuffindex)
#endif /* ALLOW_ADMTLM */
endif
c
cbuffindex = 0
do jp = 1,nPy
do bj = jtlo,jthi
do j = jmin,jmax
do ip = 1,nPx
do bi = itlo,ithi
do i = imin,imax
if ( globmsk(i,bi,ip,j,bj,jp,k) .ne. 0. ) then
cbuffindex = cbuffindex + 1
globfld3d(i,bi,ip,j,bj,jp,k) = cbuff(cbuffindex)
cph(
globfldtmp2(i,bi,ip,j,bj,jp) = cbuff(cbuffindex)
cph)
#ifdef ALLOW_ADMTLM
nveccount = nveccount + 1
globfld3d(i,bi,ip,j,bj,jp,k) =
& phtmpadmtlm(nveccount)
cph(
globfldtmp2(i,bi,ip,j,bj,jp) =
& phtmpadmtlm(nveccount)
cph)
#endif
IF ( doPackOld ) THEN
if ( lxxadxx ) then
globfld3d(i,bi,ip,j,bj,jp,k) =
& globfld3d(i,bi,ip,j,bj,jp,k)
#ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO
& / sqrt(weightfld2d(i,bi,ip,j,bj,jp))
#endif
& * precondScalar
else
globfld3d(i,bi,ip,j,bj,jp,k) =
& globfld3d(i,bi,ip,j,bj,jp,k)
#ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO
& * sqrt(weightfld2d(i,bi,ip,j,bj,jp))
#endif
& / precondScalar
endif
ELSE !IF ( doPackOld ) THEN
if ( lxxadxx ) then
globfld3d(i,bi,ip,j,bj,jp,k) =
& globfld3d(i,bi,ip,j,bj,jp,k)
& * precondScalar
else
globfld3d(i,bi,ip,j,bj,jp,k) =
& globfld3d(i,bi,ip,j,bj,jp,k)
& / precondScalar
endif
ENDIF !IF ( doPackOld ) THEN
else
globfld3d(i,bi,ip,j,bj,jp,k) = 0. _d 0
endif
cph(
globfldtmp3(i,bi,ip,j,bj,jp) =
& globfld3d(i,bi,ip,j,bj,jp,k)
cph)
enddo
enddo
enddo
enddo
enddo
enddo
cph(
if ( doPackDiag ) then
write(cunit2,rec=irectrue) globfldtmp2
write(cunit3,rec=irectrue) globfldtmp3
endif
cph)
enddo
call MDSWRITEFIELD_2D_GL( fname, ctrlprec, 'RL',
& 1, globfld3d(1,1,1,1,1,1,1),
& irec, optimcycle, mythid)
enddo
if ( doPackDiag ) then
close ( cunit2 )
close ( cunit3 )
endif
_END_MASTER( mythid )
else
c == local variables ==
integer bi,bj
integer ip,jp
integer i,j,k
integer ii
integer il
integer irec
integer itlo,ithi
integer jtlo,jthi
integer cbuffindex
_RL msk3d(1-Olx:sNx+Olx,1-Oly:sNy+Oly,Nr,nSx,nSy)
real*8 msk2d_buf(sNx,sNy,nSx,nSy)
real*8 msk2d_buf_glo(Nx,Ny)
_RL fld2d(1-Olx:sNx+Olx,1-Oly:sNy+Oly,nSx,nSy)
real*8 fld2d_buf(sNx,sNy,nSx,nSy)
real*8 fld2d_buf_glo(Nx,Ny)
_RL fld2dDim(1-Olx:sNx+Olx,1-Oly:sNy+Oly,nSx,nSy)
_RL fld2dNodim(1-Olx:sNx+Olx,1-Oly:sNy+Oly,nSx,nSy)
_RL wei2d(1-Olx:sNx+Olx,1-Oly:sNy+Oly,nSx,nSy)
real*4 cbuff ( snx*nsx*npx*sny*nsy*npy )
character*(80) weightname
_RL delZnorm
character*(80) cfile2, cfile3
_RL dummy
LOGICAL doPackOld
c == external ==
integer ilnblnk
external
c == end of interface ==
c-- part 1: preliminary reads and definitions
doPackOld = (.NOT.ctrlSmoothCorrel2D).AND.(.NOT.ctrlUseGen)
#ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO
#ifdef ALLOW_AUTODIFF
call ACTIVE_READ_XY(weighttype, wei2d, 1,
& .FALSE., .FALSE., 0 , mythid, dummy)
#else
CALL READ_REC_XY_RL( weighttype, wei2d, 1, 1, myThid )
#endif
#endif
#ifdef ALLOW_AUTODIFF
call ACTIVE_READ_XYZ(masktype, msk3d, 1,
& .FALSE., .FALSE., 0 , mythid, dummy)
#else
CALL READ_REC_XYZ_RL( masktype, msk3d, 1, 1, myThid )
#endif
if ( doPackDiag ) then
write(cfile2(1:80),'(80a)') ' '
write(cfile3(1:80),'(80a)') ' '
il = ilnblnk( fname )
if ( lxxadxx ) then
write(cfile2(1:80),'(2a)') fname(1:il),'.unpack_ctrl_adim'
write(cfile3(1:80),'(2a)') fname(1:il),'.unpack_ctrl_dim'
else
write(cfile2(1:80),'(2a)') fname(1:il),'.unpack_grad_adim'
write(cfile3(1:80),'(2a)') fname(1:il),'.unpack_grad_dim'
endif
endif
c-- part 2: loop over records
do irec = 1, ncvarrecs(ivartype)
c-- 2.1: array <- buffer <- global buffer <- global file
#ifndef ALLOW_ADMTLM
_BEGIN_MASTER( mythid )
IF ( myProcId .eq. 0 ) THEN
read(cunit) filencvarindex(ivartype)
if (filencvarindex(ivartype) .NE. ncvarindex(ivartype))
& then
print *, 'ctrl_set_unpack_xy:WARNING: wrong ncvarindex ',
& filencvarindex(ivartype), ncvarindex(ivartype)
STOP 'in S/R ctrl_set_unpack_xy'
endif
read(cunit) filej
read(cunit) filei
ENDIF
_END_MASTER( mythid )
_BARRIER
#endif /* ALLOW_ADMTLM */
do k = 1, 1
CALL MDS_PASS_R8TORL( msk2d_buf, msk3d,
& 0, 0, 1, k, Nr, 0, 0, .FALSE., myThid )
CALL BAR2( myThid )
CALL GATHER_2D_R8( msk2d_buf_glo, msk2d_buf,
& Nx,Ny,.FALSE.,.TRUE.,myThid)
CALL BAR2( myThid )
_BEGIN_MASTER( mythid )
cbuffindex = nwetglobal(k)
IF ( myProcId .eq. 0 ) THEN
#ifndef ALLOW_ADMTLM
if ( cbuffindex .gt. 0) then
read(cunit) filencbuffindex
read(cunit) filek
if (filencbuffindex .NE. cbuffindex) then
print *, 'WARNING: wrong cbuffindex ',
& filencbuffindex, cbuffindex
STOP 'in S/R ctrl_unpack'
endif
if (filek .NE. 1) then
print *, 'WARNING: wrong k ',
& filek, 1
STOP 'in S/R ctrl_set_unpack_xy'
endif
read(cunit) (cbuff(ii), ii=1,cbuffindex)
endif
#endif
cbuffindex = 0
DO j=1,Ny
DO i=1,Nx
if (msk2d_buf_glo(i,j) .ne. 0. ) then
cbuffindex = cbuffindex + 1
fld2d_buf_glo(i,j) = cbuff(cbuffindex)
endif
ENDDO
ENDDO
ENDIF
_END_MASTER( mythid )
_BARRIER
CALL BAR2( myThid )
CALL SCATTER_2D_R8( fld2d_buf_glo, fld2d_buf,
& Nx,Ny,.FALSE.,.TRUE.,myThid)
CALL BAR2( myThid )
CALL MDS_PASS_R8TORL( fld2d_buf, fld2dNodim,
& 0, 0, 1, k, 1, 0, 0, .TRUE., myThid )
enddo !do k = 1, 1
c-- 2.2: normalize field if needed
DO bj = myByLo(myThid), myByHi(myThid)
DO bi = myBxLo(myThid), myBxHi(myThid)
DO j=1,sNy
DO i=1,sNx
if (msk3d(i,j,1,bi,bj).EQ.0. _d 0) then
fld2dDim(i,j,bi,bj)=0. _d 0
fld2dNodim(i,j,bi,bj)=0. _d 0
else
#ifdef ALLOW_ADMTLM
nveccount = nveccount + 1
fld2dNodim(i,j,bi,bj)=phtmpadmtlm(nveccount)
#endif
IF ( .NOT.doPackOld ) THEN
if (lxxadxx) then
fld2dDim(i,j,bi,bj) =
& fld2dNodim(i,j,bi,bj) * precondScalar
else
fld2dDim(i,j,bi,bj) =
& fld2dNodim(i,j,bi,bj) / precondScalar
endif
ELSE !IF ( .NOT.doPackOld ) THEN
# ifndef ALLOW_NONDIMENSIONAL_CONTROL_IO
if (lxxadxx) then
fld2dDim(i,j,bi,bj) = fld2dNodim(i,j,bi,bj)
& * precondScalar
else
fld2dDim(i,j,bi,bj) = fld2dNodim(i,j,bi,bj)
& / precondScalar
endif
else
if (lxxadxx) then
fld2dDim(i,j,bi,bj) =
& fld2dNodim(i,j,bi,bj) / sqrt(wei2d(i,j,bi,bj))
& * precondScalar
else
fld2dDim(i,j,bi,bj) =
& fld2dNodim(i,j,bi,bj) * sqrt(wei2d(i,j,bi,bj))
& / precondScalar
endif
# endif /* ALLOW_NONDIMENSIONAL_CONTROL_IO */
ENDIF !IF ( .NOT.doPackOld ) THEN
endif
ENDDO
ENDDO
ENDDO
ENDDO
c-- 2.3:
if ( doPackDiag ) then
call WRITE_REC_3D_RL( cfile2, ctrlprec,
& 1, fld2dNodim, irec, 0, mythid)
call WRITE_REC_3D_RL( cfile3, ctrlprec,
& 1, fld2dDim, irec, 0, mythid)
endif
c-- 2.4:
call WRITE_REC_3D_RL( fname, ctrlprec,
& 1, fld2dDim, irec, 0, mythid)
enddo !do irec = 1, ncvarrecs(ivartype)
# endif /* ALLOW_PACKUNPACK_METHOD2 */
# endif /* EXCLUDE_CTRL_PACK */
return
end