C $Header: /u/gcmpack/MITgcm/pkg/ctrl/ctrl_set_unpack_yz.F,v 1.19 2014/10/09 00:49:27 gforget Exp $ C $Name: $ #include "CTRL_OPTIONS.h" subroutine CTRL_SET_UNPACK_YZ( & cunit, ivartype, fname, masktype, weighttype, & weightfld, nwetglobal, mythid) c ================================================================== c SUBROUTINE ctrl_set_unpack_yz c ================================================================== c c o Unpack the control vector such that land points are filled in. c c o Open boundary packing added : c gebbie@mit.edu, 18-Mar-2003 c c changed: heimbach@mit.edu 17-Jun-2003 c merged changes from Armin to replace write of c nr * globfld2d by 1 * globfld3d c (ad hoc fix to speed up global I/O) c c ================================================================== implicit none c == global variables == #include "EEPARAMS.h" #include "SIZE.h" #include "PARAMS.h" #include "GRID.h" #include "ctrl.h" #include "CTRL_OBCS.h" #include "optim.h" c == routine arguments == integer cunit integer ivartype character*( 80) fname character* (9) masktype character*( 80) weighttype _RL weightfld( nr,nobcs ) integer nwetglobal(nr,nobcs) integer mythid #ifndef EXCLUDE_CTRL_PACK c == local variables == logical lxxadxx integer bi,bj integer ip,jp integer i,j,k integer ii,jj,kk integer irec,iobcs,nrec_nl integer itlo,ithi integer jtlo,jthi integer jmin,jmax integer imin,imax integer cbuffindex real*4 cbuff ( nsx*npx*sny*nsy*npy ) real*4 globfldtmp2( nsx,npx,sny,nsy,npy ) real*4 globfldtmp3( nsx,npx,sny,nsy,npy ) _RL globfldyz( nsx,npx,sny,nsy,npy,nr ) _RL globfld3d( snx,nsx,npx,sny,nsy,npy,nr ) _RL globmskyz( nsx,npx,sny,nsy,npy,nr,nobcs ) integer reclen, irectrue integer cunit2, cunit3 character*(80) cfile2, cfile3 #ifdef CTRL_UNPACK_PRECISE integer il character*(80) weightname _RL weightfldyz( nsx,npx,sny,nsy,npy,nr,nobcs ) c == external == integer ilnblnk external #endif cc == end of interface == jtlo = 1 jthi = nsy itlo = 1 ithi = nsx jmin = 1 jmax = sny imin = 1 imax = snx lxxadxx = .TRUE. 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 globfldyz (bi,ip,j,bj,jp,k) = 0. _d 0 globfldtmp2(bi,ip,j,bj,jp) = 0. globfldtmp3(bi,ip,j,bj,jp) = 0. do iobcs=1,nobcs globmskyz(bi,ip,j,bj,jp,k,iobcs) = 0. _d 0 enddo enddo enddo enddo enddo enddo enddo 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 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,I2.2,a,I4.4,a)') & 'diag_unpack_nondim_ctrl_', & ivartype, '_', optimcycle, '.bin' write(cfile3(1:80),'(a,I2.2,a,I4.4,a)') & 'diag_unpack_dimens_ctrl_', & ivartype, '_', optimcycle, '.bin' else write(cfile2(1:80),'(a,I2.2,a,I4.4,a)') & 'diag_unpack_nondim_grad_', & ivartype, '_', optimcycle, '.bin' write(cfile3(1:80),'(a,I2.2,a,I4.4,a)') & 'diag_unpack_dimens_grad_', & ivartype, '_', optimcycle, '.bin' endif reclen = 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 do iobcs=1,nobcs call MDSREADFIELD_YZ_GL( & masktype, ctrlprec, 'RL', & Nr, globmskyz(1,1,1,1,1,1,iobcs), iobcs, mythid) #ifdef CTRL_UNPACK_PRECISE il=ilnblnk( weighttype) write(weightname(1:80),'(80a)') ' ' write(weightname(1:80),'(a)') weighttype(1:il) call MDSREADFIELD_YZ_GL( & weightname, ctrlprec, 'RL', & Nr, weightfldyz(1,1,1,1,1,1,iobcs), iobcs, mythid) #endif /* CTRL_UNPACK_PRECISE */ enddo if ( useSingleCPUio ) then C MDSWRITEFIELD_YZ_GL does not know about useSingleCPUio, so the faster C method that works for .not.useSingleCPUio cannot be used nrec_nl = 0 else nrec_nl = int(ncvarrecs(ivartype)/Nx) endif do irec = 1, nrec_nl c And now back-calculate what iobcs should be. do i=1,snx iobcs= mod((irec-1)*snx+i-1,nobcs)+1 read(cunit) filencvarindex(ivartype) if (filencvarindex(ivartype) .NE. ncvarindex(ivartype)) & then print *, 'ctrl_set_unpack_yz:WARNING: wrong ncvarindex ', & filencvarindex(ivartype), ncvarindex(ivartype) STOP 'in S/R ctrl_set_unpack_yz' endif read(cunit) filej read(cunit) filei do k = 1, Nr irectrue = (irec-1)*nobcs*nr + (iobcs-1)*nr + k cbuffindex = nwetglobal(k,iobcs) if ( cbuffindex .gt. 0 ) then read(cunit) filencbuffindex if (filencbuffindex .NE. cbuffindex) then print *, 'WARNING: wrong cbuffindex ', & filencbuffindex, cbuffindex STOP 'in S/R ctrl_set_unpack_yz' endif read(cunit) filek if (filek .NE. k) then print *, 'WARNING: wrong k ', & filek, k STOP 'in S/R ctrl_set_unpack_yz' endif read(cunit) (cbuff(ii), ii=1,cbuffindex) endif cbuffindex = 0 do jp = 1,nPy do bj = jtlo,jthi do j = jmin,jmax do ip = 1,nPx do bi = itlo,ithi ii=mod((i-1)*nr*sny+(k-1)*sny+j-1,snx)+1 jj=mod(((i-1)*nr*sny+(k-1)*sny+j-1)/snx,sny)+1 kk=int((i-1)*nr*sny+(k-1)*sny+j-1)/(snx*sny)+1 if ( globmskyz(bi,ip,j,bj,jp,k,iobcs) .ne. 0. ) then cbuffindex = cbuffindex + 1 globfld3d(ii,bi,ip,jj,bj,jp,kk) = & cbuff(cbuffindex) cph( globfldtmp2(bi,ip,jj,bj,jp) = & cbuff(cbuffindex) cph) #ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO globfld3d(ii,bi,ip,jj,bj,jp,kk) = & globfld3d(ii,bi,ip,jj,bj,jp,kk)/ # ifdef CTRL_UNPACK_PRECISE & sqrt(weightfldyz(bi,ip,j,bj,jp,k,iobcs)) else & sqrt(weightfld(k,iobcs)) # endif #endif /* ALLOW_NONDIMENSIONAL_CONTROL_IO */ else globfld3d(ii,bi,ip,jj,bj,jp,kk) = 0. _d 0 endif cph( globfldtmp3(bi,ip,jj,bj,jp) = & globfld3d(ii,bi,ip,jj,bj,jp,kk) cph) enddo enddo enddo enddo enddo c if ( doPackDiag ) then write(cunit2,rec=irectrue) globfldtmp2 write(cunit3,rec=irectrue) globfldtmp3 endif c c -- end of k loop -- enddo c -- end of i loop -- enddo call MDSWRITEFIELD_3D_GL( fname, ctrlprec, 'RL', & Nr, globfld3d, irec, & optimcycle, mythid) c -- end of irec loop -- enddo do irec = nrec_nl*nx+1,ncvarrecs(ivartype) c And now back-calculate what iobcs should be. iobcs= mod(irec-1,nobcs)+1 read(cunit) filencvarindex(ivartype) if (filencvarindex(ivartype) .NE. ncvarindex(ivartype)) & then print *, 'ctrl_set_unpack_yz:WARNING: wrong ncvarindex ', & filencvarindex(ivartype), ncvarindex(ivartype) STOP 'in S/R ctrl_set_unpack_yz' endif read(cunit) filej read(cunit) filei do k = 1, Nr irectrue = (irec-1)*nobcs*nr + (iobcs-1)*nr + k cbuffindex = nwetglobal(k,iobcs) if ( cbuffindex .gt. 0 ) then read(cunit) filencbuffindex if (filencbuffindex .NE. cbuffindex) then print *, 'WARNING: wrong cbuffindex ', & filencbuffindex, cbuffindex STOP 'in S/R ctrl_set_unpack_yz' endif read(cunit) filek if (filek .NE. k) then print *, 'WARNING: wrong k ', & filek, k STOP 'in S/R ctrl_set_unpack_yz' endif read(cunit) (cbuff(ii), ii=1,cbuffindex) endif cbuffindex = 0 do jp = 1,nPy do bj = jtlo,jthi do j = jmin,jmax do ip = 1,nPx do bi = itlo,ithi if ( globmskyz(bi,ip,j,bj,jp,k,iobcs) .ne. 0. ) then cbuffindex = cbuffindex + 1 globfldyz(bi,ip,j,bj,jp,k) = cbuff(cbuffindex) cph( globfldtmp2(bi,ip,j,bj,jp) = cbuff(cbuffindex) cph) #ifdef ALLOW_NONDIMENSIONAL_CONTROL_IO globfldyz(bi,ip,j,bj,jp,k) = & globfldyz(bi,ip,j,bj,jp,k)/ # ifdef CTRL_UNPACK_PRECISE & sqrt(weightfldyz(bi,ip,j,bj,jp,k,iobcs)) else & sqrt(weightfld(k,iobcs)) # endif #endif /* ALLOW_NONDIMENSIONAL_CONTROL_IO */ else globfldyz(bi,ip,j,bj,jp,k) = 0. _d 0 endif cph( globfldtmp3(bi,ip,j,bj,jp) = & globfldyz(bi,ip,j,bj,jp,k) cph) enddo enddo enddo enddo enddo c c -- end of k loop enddo call MDSWRITEFIELD_YZ_GL( fname, ctrlprec, 'RL', & Nr, globfldyz, irec, & optimcycle, mythid) c -- end of irec loop -- enddo _END_MASTER( mythid ) #endif return end