Changeset 13458 for NEMO/trunk/src/SWE/stpctl.F90
- Timestamp:
- 2020-09-11T11:22:24+02:00 (4 years ago)
- File:
-
- 1 edited
-
NEMO/trunk/src/SWE/stpctl.F90 (modified) (3 diffs)
Legend:
- Unmodified
- Added
- Removed
-
NEMO/trunk/src/SWE/stpctl.F90
r12983 r13458 19 19 USE dom_oce ! ocean space and time domain variables 20 20 USE c1d ! 1D vertical configuration 21 USE zdf_oce , ONLY : ln_zad_Aimp ! ocean vertical physics variables 22 USE wet_dry, ONLY : ll_wd, ssh_ref ! reference depth for negative bathy 23 ! 21 24 USE diawri ! Standard run outputs (dia_wri_state routine) 22 !23 25 USE in_out_manager ! I/O manager 24 26 USE lbclnk ! ocean lateral boundary conditions (or mpp link) 25 27 USE lib_mpp ! distributed memory computing 26 USE zdf_oce , ONLY : ln_zad_Aimp ! ocean vertical physics variables 27 USE wet_dry, ONLY : ll_wd, ssh_ref ! reference depth for negative bathy 28 28 ! 29 29 USE netcdf ! NetCDF library 30 30 IMPLICIT NONE … … 33 33 PUBLIC stp_ctl ! routine called by step.F90 34 34 35 INTEGER :: idrun, idtime, idssh, idu, ids1, ids2, idt1, idt2, idc1, idw1, istatus 36 LOGICAL :: lsomeoce 37 !!stoops 38 # include "domzgr_substitute.h90" 35 INTEGER :: nrunid ! netcdf file id 36 INTEGER, DIMENSION(2) :: nvarid ! netcdf variable id 39 37 !!---------------------------------------------------------------------- 40 38 !! NEMO/OCE 4.0 , NEMO Consortium (2018) 41 !! $Id: stpctl.F90 1 2614 2020-03-26 14:59:52Z gm$39 !! $Id: stpctl.F90 13216 2020-07-02 09:25:49Z rblod $ 42 40 !! Software governed by the CeCILL license (see ./LICENSE) 43 41 !!---------------------------------------------------------------------- 44 42 CONTAINS 45 43 46 SUBROUTINE stp_ctl( kt, K bb, Kmm, kindic)44 SUBROUTINE stp_ctl( kt, Kmm ) 47 45 !!---------------------------------------------------------------------- 48 46 !! *** ROUTINE stp_ctl *** … … 51 49 !! 52 50 !! ** Method : - Save the time step in numstp 53 !! - Print it each 50 time steps 54 !! - Stop the run IF problem encountered by setting indic=-3 55 !! Problems checked: |ssh| maximum larger than 10 m 51 !! - Stop the run IF problem encountered by setting nstop > 0 52 !! Problems checked: negative sea surface height 56 53 !! |U| maximum larger than 10 m/s 57 !! negative sea surface salinity58 54 !! 59 55 !! ** Actions : "time.step" file = last ocean time-step 60 56 !! "run.stat" file = run statistics 61 !! nstop indicator sheared among all local domain (lk_mpp=T)57 !! nstop indicator sheared among all local domain 62 58 !!---------------------------------------------------------------------- 63 59 INTEGER, INTENT(in ) :: kt ! ocean time-step index 64 INTEGER, INTENT(in ) :: Kbb, Kmm ! ocean time level index 65 INTEGER, INTENT(inout) :: kindic ! error indicator 66 !! 67 INTEGER :: ji, jj, jk ! dummy loop indices 68 INTEGER, DIMENSION(2) :: ih ! min/max loc indices 69 INTEGER, DIMENSION(3) :: iu, is1, is2 ! min/max loc indices 70 REAL(wp) :: zzz ! local real 71 REAL(wp), DIMENSION(3) :: zmax 72 LOGICAL :: ll_wrtstp, ll_colruns, ll_wrtruns 73 CHARACTER(len=20) :: clname 74 !!---------------------------------------------------------------------- 75 ! 76 ll_wrtstp = ( MOD( kt, sn_cfctl%ptimincr ) == 0 ) .OR. ( kt == nitend ) 77 ll_colruns = ll_wrtstp .AND. ( sn_cfctl%l_runstat ) 78 ll_wrtruns = ll_colruns .AND. lwm 79 IF( kt == nit000 .AND. lwp ) THEN 80 WRITE(numout,*) 81 WRITE(numout,*) 'stp_ctl : time-stepping control' 82 WRITE(numout,*) '~~~~~~~' 83 ! ! open time.step file 84 IF( lwm ) CALL ctl_opn( numstp, 'time.step', 'REPLACE', 'FORMATTED', 'SEQUENTIAL', -1, numout, lwp, narea ) 85 ! ! open run.stat file(s) at start whatever 86 ! ! the value of sn_cfctl%ptimincr 87 IF( lwm .AND. ( sn_cfctl%l_runstat ) ) THEN 60 INTEGER, INTENT(in ) :: Kmm ! ocean time level index 61 !! 62 INTEGER :: ji ! dummy loop indices 63 INTEGER :: idtime, istatus 64 INTEGER , DIMENSION(3) :: iareasum, iareamin, iareamax 65 INTEGER , DIMENSION(3,2) :: iloc ! min/max loc indices 66 REAL(wp) :: zzz ! local real 67 REAL(wp), DIMENSION(3) :: zmax, zmaxlocal 68 LOGICAL :: ll_wrtstp, ll_colruns, ll_wrtruns 69 LOGICAL, DIMENSION(jpi,jpj,jpk) :: llmsk 70 CHARACTER(len=20) :: clname 71 !!---------------------------------------------------------------------- 72 IF( nstop > 0 .AND. ngrdstop > -1 ) RETURN ! stpctl was already called by a child grid 73 ! 74 ll_wrtstp = ( MOD( kt-nit000, sn_cfctl%ptimincr ) == 0 ) .OR. ( kt == nitend ) 75 ll_colruns = ll_wrtstp .AND. sn_cfctl%l_runstat .AND. jpnij > 1 76 ll_wrtruns = ( ll_colruns .OR. jpnij == 1 ) .AND. lwm 77 ! 78 IF( kt == nit000 ) THEN 79 ! 80 IF( lwp ) THEN 81 WRITE(numout,*) 82 WRITE(numout,*) 'stp_ctl : time-stepping control' 83 WRITE(numout,*) '~~~~~~~' 84 ENDIF 85 ! ! open time.step ascii file, done only by 1st subdomain 86 IF( lwm ) CALL ctl_opn( numstp, 'time.step', 'REPLACE', 'FORMATTED', 'SEQUENTIAL', -1, numout, lwp, narea ) 87 ! 88 IF( ll_wrtruns ) THEN 89 ! ! open run.stat ascii file, done only by 1st subdomain 88 90 CALL ctl_opn( numrun, 'run.stat', 'REPLACE', 'FORMATTED', 'SEQUENTIAL', -1, numout, lwp, narea ) 91 ! ! open run.stat.nc netcdf file, done only by 1st subdomain 89 92 clname = 'run.stat.nc' 90 93 IF( .NOT. Agrif_Root() ) clname = TRIM(Agrif_CFixed())//"_"//TRIM(clname) 91 istatus = NF90_CREATE( TRIM(clname), NF90_CLOBBER, idrun ) 92 istatus = NF90_DEF_DIM( idrun, 'time', NF90_UNLIMITED, idtime ) 93 istatus = NF90_DEF_VAR( idrun, 'abs_ssh_max', NF90_DOUBLE, (/ idtime /), idssh ) 94 istatus = NF90_DEF_VAR( idrun, 'abs_u_max', NF90_DOUBLE, (/ idtime /), idu ) 95 istatus = NF90_ENDDEF(idrun) 96 ENDIF 97 ENDIF 98 IF( kt == nit000 ) lsomeoce = COUNT( ssmask(:,:) == 1._wp ) > 0 99 ! 100 IF(lwm .AND. ll_wrtstp) THEN !== current time step ==! ("time.step" file) 94 istatus = NF90_CREATE( TRIM(clname), NF90_CLOBBER, nrunid ) 95 istatus = NF90_DEF_DIM( nrunid, 'time', NF90_UNLIMITED, idtime ) 96 istatus = NF90_DEF_VAR( nrunid, 'abs_ssh_max', NF90_DOUBLE, (/ idtime /), nvarid(1) ) 97 istatus = NF90_DEF_VAR( nrunid, 'abs_u_max', NF90_DOUBLE, (/ idtime /), nvarid(2) ) 98 istatus = NF90_ENDDEF(nrunid) 99 ENDIF 100 ! 101 ENDIF 102 ! 103 ! !== write current time step ==! 104 ! !== done only by 1st subdomain at writting timestep ==! 105 IF( lwm .AND. ll_wrtstp ) THEN 101 106 WRITE ( numstp, '(1x, i8)' ) kt 102 107 REWIND( numstp ) 103 108 ENDIF 104 ! 105 ! !== test of extrema ==! 106 IF( ll_wd ) THEN 107 zmax(1) = MAXVAL( ABS( ssh(:,:,Kmm) + ssh_ref*tmask(:,:,1) ) ) ! ssh max 108 ELSE 109 zmax(1) = MINVAL( e3t(:,:,1,Kmm) ) ! ssh min 110 ENDIF 111 zmax(2) = MAXVAL( ABS( uu(:,:,:,Kmm) ) ) ! velocity max (zonal only) 112 zmax(3) = REAL( nstop , wp ) ! stop indicator 113 ! 109 ! !== test of local extrema ==! 110 ! !== done by all processes at every time step ==! 111 ! 112 llmsk( 1:Nis1,:,:) = .FALSE. ! exclude halos from the checked region 113 llmsk(Nie1: jpi,:,:) = .FALSE. 114 llmsk(:, 1:Njs1,:) = .FALSE. 115 llmsk(:,Nje1: jpj,:) = .FALSE. 116 ! 117 llmsk(Nis0:Nie0,Njs0:Nje0,1) = tmask(Nis0:Nie0,Njs0:Nje0,1) == 1._wp ! define only the inner domain 118 zmax(1) = MAXVAL( -e3t(:,:,1,Kmm) ), mask = llmsk(:,:,1) ) ! ssh max 119 llmsk(Nis0:Nie0,Njs0:Nje0,:) = umask(Nis0:Nie0,Njs0:Nje0,:) == 1._wp ! define only the inner domain 120 zmax(2) = MAXVAL( ABS( uu(:,:,:,Kmm) ), mask = llmsk(:,:,:) ) ! velocity max (zonal only) 121 zmax(3) = REAL( nstop, wp ) ! stop indicator 122 ! !== get global extrema ==! 123 ! !== done by all processes if writting run.stat ==! 114 124 IF( ll_colruns ) THEN 125 zmaxlocal(:) = zmax(:) 115 126 CALL mpp_max( "stpctl", zmax ) ! max over the global domain 116 nstop = NINT( zmax(3) ) ! nstop indicator sheared among all local domains 117 ENDIF 118 ! !== run statistics ==! ("run.stat" files) 127 nstop = NINT( zmax(3) ) ! update nstop indicator (now sheared among all local domains) 128 ENDIF 129 ! !== write "run.stat" files ==! 130 ! !== done only by 1st subdomain at writting timestep ==! 119 131 IF( ll_wrtruns ) THEN 120 132 WRITE(numrun,9500) kt, zmax(1), zmax(2) 121 istatus = NF90_PUT_VAR( idrun, idssh, (/ zmax(1)/), (/kt/), (/1/) ) 122 istatus = NF90_PUT_VAR( idrun, idu, (/ zmax(2)/), (/kt/), (/1/) ) 123 IF( MOD( kt , 100 ) == 0 ) istatus = NF90_SYNC(idrun) 124 IF( kt == nitend ) istatus = NF90_CLOSE(idrun) 133 istatus = NF90_PUT_VAR( nrunid, nvarid(1), (/ -zmax(1)/), (/kt/), (/1/) ) 134 istatus = NF90_PUT_VAR( nrunid, nvarid(2), (/ zmax(2)/), (/kt/), (/1/) ) 135 IF( kt == nitend ) istatus = NF90_CLOSE(nrunid) 125 136 END IF 126 ! !== error handling ==! 127 IF( ( sn_cfctl%l_glochk .OR. lsomeoce ) .AND. ( & ! domain contains some ocean points, check for sensible ranges 128 & zmax(1) < 0._wp .OR. & ! negative sea surface height 129 & zmax(2) > 10._wp .OR. & ! too large velocity ( > 10 m/s) 130 & ISNAN( zmax(1) + zmax(2) ) ) ) THEN ! NaN encounter in the tests 131 IF( lk_mpp .AND. sn_cfctl%l_glochk ) THEN 132 ! have use mpp_max (because sn_cfctl%l_glochk=.T. and distributed) 133 CALL mpp_maxloc( 'stpctl', ABS(ssh(:,:,Kmm)) , ssmask(:,:) , zzz, ih ) 134 CALL mpp_maxloc( 'stpctl', ABS(uu(:,:,:,Kmm)) , umask (:,:,:), zzz, iu ) 137 ! !== error handling ==! 138 ! !== done by all processes at every time step ==! 139 ! 140 IF( zmax(1) > 0._wp .OR. & ! negative sea surface height 141 & zmax(2) > 10._wp .OR. & ! too large velocity ( > 10 m/s) 142 & ISNAN( zmax(1) + zmax(2) ) .OR. & ! NaN encounter in the tests 143 & ABS( zmax(1) + zmax(2) ) > HUGE(1._wp) ) THEN ! Infinity encounter in the tests 144 ! 145 iloc(:,:) = 0 146 IF( ll_colruns ) THEN ! zmax is global, so it is the same on all subdomains -> no dead lock with mpp_maxloc 147 ! first: close the netcdf file, so we can read it 148 IF( lwm .AND. kt /= nitend ) istatus = NF90_CLOSE(nrunid) 149 ! get global loc on the min/max 150 llmsk(Nis0:Nie0,Njs0:Nje0,1) = tmask(Nis0:Nie0,Njs0:Nje0,1) == 1._wp ! define only the inner domain 151 CALL mpp_maxloc( 'stpctl', -e3t(:,:,1,Kmm) , llmsk(:,:,1), zzz, iloc(1:2,1) ) ! mpp_maxloc ok if mask = F 152 llmsk(Nis0:Nie0,Njs0:Nje0,:) = umask(Nis0:Nie0,Njs0:Nje0,:) == 1._wp ! define only the inner domain 153 CALL mpp_maxloc( 'stpctl', ABS(uu(:,:,:,Kmm)), llmsk(:,:,:), zzz, iloc(1:3,2) ) 154 ! find which subdomain has the max. 155 iareamin(:) = jpnij+1 ; iareamax(:) = 0 ; iareasum(:) = 0 156 DO ji = 1, 3 157 IF( zmaxlocal(ji) == zmax(ji) ) THEN 158 iareamin(ji) = narea ; iareamax(ji) = narea ; iareasum(ji) = 1 159 ENDIF 160 END DO 161 CALL mpp_min( "stpctl", iareamin ) ! min over the global domain 162 CALL mpp_max( "stpctl", iareamax ) ! max over the global domain 163 CALL mpp_sum( "stpctl", iareasum ) ! sum over the global domain 164 ELSE ! find local min and max locations: 165 ! if we are here, this means that the subdomain contains some oce points -> no need to test the mask used in maxloc 166 llmsk(Nis0:Nie0,Njs0:Nje0,1) = tmask(Nis0:Nie0,Njs0:Nje0,1) == 1._wp ! define only the inner domain 167 iloc(1:2,1) = MAXLOC( -e3t(:,:,1,Kmm) , mask = llmsk(:,:,1) ) 168 llmsk(Nis0:Nie0,Njs0:Nje0,:) = umask(Nis0:Nie0,Njs0:Nje0,:) == 1._wp ! define only the inner domain 169 iloc(1:3,2) = MAXLOC( ABS(uu(:,:,:,Kmm)), mask = llmsk(:,:,:) ) 170 DO ji = 1, 2 ! local domain indices ==> global domain indices, excluding halos 171 iloc(1:2,ji) = (/ mig0(iloc(1,ji)), mjg0(iloc(2,ji)) /) 172 END DO 173 iareamin(:) = narea ; iareamax(:) = narea ; iareasum(:) = 1 ! this is local information 174 ENDIF 175 ! 176 WRITE(ctmp1,*) ' stp_ctl: |ssh| > 20 m or |U| > 10 m/s or S <= 0 or S >= 100 or NaN encounter in the tests' 177 CALL wrt_line( ctmp2, kt, '|e3t| min', -zmax(1), iloc(:,1), iareasum(1), iareamin(1), iareamax(1) ) 178 CALL wrt_line( ctmp3, kt, '|U| max', zmax(2), iloc(:,2), iareasum(2), iareamin(2), iareamax(2) ) 179 IF( Agrif_Root() ) THEN 180 WRITE(ctmp6,*) ' ===> output of last computed fields in output.abort* files' 135 181 ELSE 136 ! find local min and max locations 137 ih(:) = MAXLOC( ABS( ssh(:,:,Kmm) ) ) + (/ nimpp - 1, njmpp - 1 /) 138 iu(:) = MAXLOC( ABS( uu (:,:,:,Kmm) ) ) + (/ nimpp - 1, njmpp - 1, 0 /) 139 ENDIF 140 141 WRITE(ctmp1,*) ' stp_ctl: (e3t0) ssh < 0 m or |U| > 10 m/s or NaN encounter in the tests' 142 WRITE(ctmp2,9100) kt, zmax(1), ih(1) , ih(2) 143 WRITE(ctmp3,9200) kt, zmax(2), iu(1) , iu(2) , iu(3) 144 WRITE(ctmp4,*) ' ===> output of last computed fields in output.abort.nc file' 145 182 WRITE(ctmp6,*) ' ===> output of last computed fields in '//TRIM(Agrif_CFixed())//'_output.abort* files' 183 ENDIF 184 ! 146 185 CALL dia_wri_state( Kmm, 'output.abort' ) ! create an output.abort file 147 148 IF( .NOT. sn_cfctl%l_glochk ) THEN 149 WRITE(ctmp8,*) 'E R R O R message from sub-domain: ', narea 150 CALL ctl_stop( 'STOP', ctmp1, ' ', ctmp8, ' ', ctmp2, ctmp3, ctmp4 ) 151 ELSE 152 CALL ctl_stop( ctmp1, ' ', ctmp2, ctmp3, ctmp4 ) 153 ENDIF 154 155 kindic = -3 156 ! 157 ENDIF 158 ! 159 9100 FORMAT (' kt=',i8,' |ssh| min: ',1pg11.4,', at i j : ',2i5) 160 9200 FORMAT (' kt=',i8,' |U| max: ',1pg11.4,', at i j k: ',3i5) 161 9500 FORMAT(' it :', i8, ' |ssh|_max: ', D23.16, ' |U|_max: ', D23.16) 186 ! 187 IF( ll_colruns .or. jpnij == 1 ) THEN ! all processes synchronized -> use lwp to print in opened ocean.output files 188 IF(lwp) THEN ; CALL ctl_stop( ctmp1, ' ', ctmp2, ctmp3, ' ', ctmp6 ) 189 ELSE ; nstop = MAX(1, nstop) ! make sure nstop > 0 (automatically done when calling ctl_stop) 190 ENDIF 191 ELSE ! only mpi subdomains with errors are here -> STOP now 192 CALL ctl_stop( 'STOP', ctmp1, ' ', ctmp2, ctmp3, ' ', ctmp6 ) 193 ENDIF 194 ! 195 ENDIF 196 ! 197 IF( nstop > 0 ) THEN ! an error was detected and we did not abort yet... 198 ngrdstop = Agrif_Fixed() ! store which grid got this error 199 IF( .NOT. ll_colruns .AND. jpnij > 1 ) CALL ctl_stop( 'STOP' ) ! we must abort here to avoid MPI deadlock 200 ENDIF 201 ! 202 9500 FORMAT(' it :', i8, ' |ssh|_max: ', D23.16, ' |U|_max: ', D23.16,' S_min: ', D23.16,' S_max: ', D23.16) 162 203 ! 163 204 END SUBROUTINE stp_ctl 205 206 207 SUBROUTINE wrt_line( cdline, kt, cdprefix, pval, kloc, ksum, kmin, kmax ) 208 !!---------------------------------------------------------------------- 209 !! *** ROUTINE wrt_line *** 210 !! 211 !! ** Purpose : write information line 212 !! 213 !!---------------------------------------------------------------------- 214 CHARACTER(len=*), INTENT( out) :: cdline 215 CHARACTER(len=*), INTENT(in ) :: cdprefix 216 REAL(wp), INTENT(in ) :: pval 217 INTEGER, DIMENSION(3), INTENT(in ) :: kloc 218 INTEGER, INTENT(in ) :: kt, ksum, kmin, kmax 219 ! 220 CHARACTER(len=80) :: clsuff 221 CHARACTER(len=9 ) :: clkt, clsum, clmin, clmax 222 CHARACTER(len=9 ) :: cli, clj, clk 223 CHARACTER(len=1 ) :: clfmt 224 CHARACTER(len=4 ) :: cl4 ! needed to be able to compile with Agrif, I don't know why 225 INTEGER :: ifmtk 226 !!---------------------------------------------------------------------- 227 WRITE(clkt , '(i9)') kt 228 229 WRITE(clfmt, '(i1)') INT(LOG10(REAL(jpnij ,wp))) + 1 ! how many digits to we need to write ? (we decide max = 9) 230 !!! WRITE(clsum, '(i'//clfmt//')') ksum ! this is creating a compilation error with AGRIF 231 cl4 = '(i'//clfmt//')' ; WRITE(clsum, cl4) ksum 232 WRITE(clfmt, '(i1)') INT(LOG10(REAL(MAX(1,jpnij-1),wp))) + 1 ! how many digits to we need to write ? (we decide max = 9) 233 cl4 = '(i'//clfmt//')' ; WRITE(clmin, cl4) kmin-1 234 WRITE(clmax, cl4) kmax-1 235 ! 236 WRITE(clfmt, '(i1)') INT(LOG10(REAL(jpiglo,wp))) + 1 ! how many digits to we need to write jpiglo? (we decide max = 9) 237 cl4 = '(i'//clfmt//')' ; WRITE(cli, cl4) kloc(1) ! this is ok with AGRIF 238 WRITE(clfmt, '(i1)') INT(LOG10(REAL(jpjglo,wp))) + 1 ! how many digits to we need to write jpjglo? (we decide max = 9) 239 cl4 = '(i'//clfmt//')' ; WRITE(clj, cl4) kloc(2) ! this is ok with AGRIF 240 ! 241 IF( ksum == 1 ) THEN ; WRITE(clsuff,9100) TRIM(clmin) 242 ELSE ; WRITE(clsuff,9200) TRIM(clsum), TRIM(clmin), TRIM(clmax) 243 ENDIF 244 IF(kloc(3) == 0) THEN 245 ifmtk = INT(LOG10(REAL(jpk,wp))) + 1 ! how many digits to we need to write jpk? (we decide max = 9) 246 clk = REPEAT(' ', ifmtk) ! create the equivalent in blank string 247 WRITE(cdline,9300) TRIM(ADJUSTL(clkt)), TRIM(ADJUSTL(cdprefix)), pval, TRIM(cli), TRIM(clj), clk(1:ifmtk), TRIM(clsuff) 248 ELSE 249 WRITE(clfmt, '(i1)') INT(LOG10(REAL(jpk,wp))) + 1 ! how many digits to we need to write jpk? (we decide max = 9) 250 !!! WRITE(clk, '(i'//clfmt//')') kloc(3) ! this is creating a compilation error with AGRIF 251 cl4 = '(i'//clfmt//')' ; WRITE(clk, cl4) kloc(3) ! this is ok with AGRIF 252 WRITE(cdline,9400) TRIM(ADJUSTL(clkt)), TRIM(ADJUSTL(cdprefix)), pval, TRIM(cli), TRIM(clj), TRIM(clk), TRIM(clsuff) 253 ENDIF 254 ! 255 9100 FORMAT('MPI rank ', a) 256 9200 FORMAT('found in ', a, ' MPI tasks, spread out among ranks ', a, ' to ', a) 257 9300 FORMAT('kt ', a, ' ', a, ' ', 1pg11.4, ' at i j ', a, ' ', a, ' ', a, ' ', a) 258 9400 FORMAT('kt ', a, ' ', a, ' ', 1pg11.4, ' at i j k ', a, ' ', a, ' ', a, ' ', a) 259 ! 260 END SUBROUTINE wrt_line 261 164 262 165 263 !!======================================================================
Note: See TracChangeset
for help on using the changeset viewer.
