C $Header: /u/gcmpack/MITgcm/pkg/ctrl/ctrl_set_pack_xy.F,v 1.28 2015/06/16 20:20:26 gforget Exp $ C $Name: $ #include "CTRL_OPTIONS.h" subroutine CTRL_SET_PACK_XY( & cunit, ivartype, precondScalar, & fname, masktype, weighttype, & lxxadxx, mythid) c ================================================================== c SUBROUTINE ctrl_set_pack_xy c ================================================================== c c o Compress the control vector such that only ocean points are c written to file. 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 == integer cunit integer ivartype _RL precondScalar character*( 80) fname, fnameGlobal character*( 9) masktype character*( 80) weighttype logical lxxadxx 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_pack_nonout_ctrl_', & ivartype, '_', optimcycle, '.bin' write(cfile3(1:80),'(a,I3.3,a,I4.4,a)') & 'diag_pack_dimout_ctrl_', & ivartype, '_', optimcycle, '.bin' else write(cfile2(1:80),'(a,I3.3,a,I4.4,a)') & 'diag_pack_nonout_grad_', & ivartype, '_', optimcycle, '.bin' write(cfile3(1:80),'(a,I3.3,a,I4.4,a)') & 'diag_pack_dimout_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 call MDSREADFIELD_3D_GL( fname, ctrlprec, 'RL', & Nr, globfld3d, irec, mythid) do k = 1, Nr irectrue = (irec-1)*nr + k #ifndef ALLOW_ADMTLM write(cunit) ncvarindex(ivartype) write(cunit) 1 write(cunit) 1 #endif 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 cph( globfldtmp3(i,bi,ip,j,bj,jp) = & globfld3d(i,bi,ip,j,bj,jp,k) cph) IF ( doPackOld ) THEN #ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO if (lxxadxx) then cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) * & sqrt(weightfld2d(i,bi,ip,j,bj,jp)) & / PrecondScalar else cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) / & sqrt(weightfld2d(i,bi,ip,j,bj,jp)) & * PrecondScalar endif #else if (lxxadxx) then cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) & / PrecondScalar else cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) & * PrecondScalar endif #endif ELSE ! IF ( doPackOld ) THEN if (lxxadxx) then cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) & / PrecondScalar else cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) & * PrecondScalar endif ENDIF !IF ( doPackOld ) THEN cph( globfldtmp2(i,bi,ip,j,bj,jp) = cbuff(cbuffindex) cph) #ifdef ALLOW_ADMTLM nveccount = nveccount + 1 phtmpadmtlm(nveccount) = cbuff(cbuffindex) #endif endif enddo enddo enddo enddo enddo enddo c --> check cbuffindex. if ( cbuffindex .gt. 0) then #ifndef ALLOW_ADMTLM write(cunit) cbuffindex write(cunit) 1 cph#endif write(cunit) (cbuff(ii), ii=1,cbuffindex) #endif endif c if ( doPackDiag ) then write(cunit2,rec=irectrue) globfldtmp2 write(cunit3,rec=irectrue) globfldtmp3 endif c enddo c c -- end of irec loop -- enddo do irec = nrec_nl*Nr+1, ncvarrecs(ivartype) call MDSREADFIELD_2D_GL( fname, ctrlprec, 'RL', & 1, globfld3d(1,1,1,1,1,1,1), irec, mythid) #ifndef ALLOW_ADMTLM write(cunit) ncvarindex(ivartype) write(cunit) 1 write(cunit) 1 #endif do k = 1, 1 irectrue = irec 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 cph( globfldtmp3(i,bi,ip,j,bj,jp) = & globfld3d(i,bi,ip,j,bj,jp,k) cph) IF ( doPackOld ) THEN #ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO if (lxxadxx) then cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) * & sqrt(weightfld2d(i,bi,ip,j,bj,jp)) & / PrecondScalar else cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) / & sqrt(weightfld2d(i,bi,ip,j,bj,jp)) & * PrecondScalar endif #else if (lxxadxx) then cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) & / PrecondScalar else cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) & * PrecondScalar endif #endif ELSE !IF ( doPackOld ) THEN if (lxxadxx) then cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) & / PrecondScalar else cbuff(cbuffindex) = & globfld3d(i,bi,ip,j,bj,jp,k) & * PrecondScalar endif ENDIF !IF ( doPackOld ) THEN cph( globfldtmp2(i,bi,ip,j,bj,jp) = cbuff(cbuffindex) cph) #ifdef ALLOW_ADMTLM nveccount = nveccount + 1 phtmpadmtlm(nveccount) = cbuff(cbuffindex) #endif endif enddo enddo enddo enddo enddo enddo c --> check cbuffindex. if ( cbuffindex .gt. 0) then #ifndef ALLOW_ADMTLM write(cunit) cbuffindex write(cunit) k cph#endif write(cunit) (cbuff(ii), ii=1,cbuffindex) #endif endif c if ( doPackDiag ) then write(cunit2,rec=irectrue) globfldtmp2 write(cunit3,rec=irectrue) globfldtmp3 endif c enddo c c -- end of irec loop -- 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),'.pack_ctrl_adim' write(cfile3(1:80),'(2a)') fname(1:il),'.pack_ctrl_dim' else write(cfile2(1:80),'(2a)') fname(1:il),'.pack_grad_adim' write(cfile3(1:80),'(2a)') fname(1:il),'.pack_grad_dim' endif endif c-- part 2: loop over records do irec = 1, ncvarrecs(ivartype) c-- 2.1: call READ_REC_3D_RL( fname, ctrlprec, & 1, fld2dDim, irec, 0, mythid) 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 IF ( .NOT.doPackOld ) THEN if (lxxadxx) then fld2dNodim(i,j,bi,bj) = & fld2dDim(i,j,bi,bj) / PrecondScalar else fld2dNodim(i,j,bi,bj) = & fld2dDim(i,j,bi,bj) * PrecondScalar endif ELSE !IF ( .NOT.doPackOld ) THEN # ifndef ALLOW_NONDIMENSIONAL_CONTROL_IO if (lxxadxx) then fld2dNodim(i,j,bi,bj) = fld2dDim(i,j,bi,bj) & / PrecondScalar else fld2dNodim(i,j,bi,bj) = fld2dDim(i,j,bi,bj) & * PrecondScalar endif else if (lxxadxx) then fld2dNodim(i,j,bi,bj) = & fld2dDim(i,j,bi,bj) * sqrt(wei2d(i,j,bi,bj)) & / PrecondScalar else fld2dNodim(i,j,bi,bj) = & fld2dDim(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: array -> buffer -> global buffer -> global file #ifndef ALLOW_ADMTLM _BEGIN_MASTER( mythid ) IF ( myProcId .eq. 0 ) THEN write(cunit) ncvarindex(ivartype) write(cunit) 1 write(cunit) 1 ENDIF _END_MASTER( mythid ) _BARRIER #endif do k = 1, 1 CALL MDS_PASS_R8TORL( fld2d_buf, fld2dNodim, & 0, 0, 1, k, 1, 0, 0, .FALSE., myThid ) CALL BAR2( myThid ) CALL GATHER_2D_R8( fld2d_buf_glo, fld2d_buf, & Nx,Ny,.FALSE.,.TRUE.,myThid) CALL BAR2( myThid ) 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 = 0 IF ( myProcId .eq. 0 ) THEN DO j=1,Ny DO i=1,Nx if (msk2d_buf_glo(i,j) .ne. 0. ) then cbuffindex = cbuffindex + 1 cbuff(cbuffindex) = fld2d_buf_glo(i,j) #ifdef ALLOW_ADMTLM nveccount = nveccount + 1 phtmpadmtlm(nveccount) = cbuff(cbuffindex) #endif endif ENDDO ENDDO #ifndef ALLOW_ADMTLM if ( cbuffindex .gt. 0) then write(cunit) cbuffindex write(cunit) k write(cunit) (cbuff(ii), ii=1,cbuffindex) endif #endif ENDIF _END_MASTER( mythid ) _BARRIER enddo enddo # endif /* ALLOW_PACKUNPACK_METHOD2 */ # endif /* EXCLUDE_CTRL_PACK */ return end