MODULE tl_def_ini_mod ! !git $Id$ !svn $Id: tl_def_ini.F 1151 2023-02-09 03:08:53Z arango $ !================================================== Hernan G. Arango === ! Copyright (c) 2002-2023 The ROMS/TOMS Group ! ! Licensed under a MIT/X style license ! ! See License_ROMS.md ! !======================================================================= ! ! ! This module updates tangent linear initial file used for 4DVAR ! ! data assimilation using either the standard NetCDF library or ! ! the Parallel-IO (PIO) library. ! ! ! !======================================================================= ! USE mod_param USE mod_parallel USE mod_fourdvar USE mod_iounits USE mod_ncparam USE mod_scalars ! USE def_dim_mod, ONLY : def_dim USE def_info_mod, ONLY : def_info USE def_var_mod, ONLY : def_var USE strings_mod, ONLY : FoundError USE wrt_info_mod, ONLY : wrt_info ! implicit none ! PUBLIC :: tl_def_ini PRIVATE :: tl_def_ini_nf90 ! CONTAINS ! !*********************************************************************** SUBROUTINE tl_def_ini (ng) !*********************************************************************** ! ! Imported variable declarations. ! integer, intent(in) :: ng ! ! Local variable declarations. ! character (len=*), parameter :: MyFile = & & "ROMS/Tangent/tl_def_ini.F" ! !----------------------------------------------------------------------- ! Create a new history file according to IO type. !----------------------------------------------------------------------- ! SELECT CASE (ITL(ng)%IOtype) CASE (io_nf90) CALL tl_def_ini_nf90 (ng) CASE DEFAULT IF (Master) WRITE (stdout,10) ITL(ng)%IOtype exit_flag=3 END SELECT IF (FoundError(exit_flag, NoError, 71, MyFile)) RETURN ! 10 FORMAT (' TL_DEF_INI - Illegal output file type, io_type = ',i0, & & /,14x,'Check KeyWord ''OUT_LIB'' in ''roms.in''.') ! RETURN END SUBROUTINE tl_def_ini ! !*********************************************************************** SUBROUTINE tl_def_ini_nf90 (ng) !*********************************************************************** ! USE mod_netcdf ! ! Imported variable declarations. ! integer, intent(in) :: ng ! ! Local variable declarations. ! logical :: got_var(NV) ! integer, parameter :: Natt = 25 integer :: i, j, ifield, itrc, nvd3, nvd4 integer :: recdim, status, varid integer :: Fcount integer :: DimIDs(nDimID) integer :: t2dgrd(3), u2dgrd(3), v2dgrd(3) integer :: t3dgrd(4), u3dgrd(4), v3dgrd(4), w3dgrd(4) ! real(r8) :: Aval(6) ! character (len=256) :: ncname character (len=MaxLen) :: Vinfo(Natt) character (len=*), parameter :: MyFile = & & "ROMS/Tangent/tl_def_ini.F"//", tl_def_ini_nf90" ! SourceFile=MyFile ! !----------------------------------------------------------------------- ! Set and report file name. !----------------------------------------------------------------------- ! IF (FoundError(exit_flag, NoError, 137, MyFile)) RETURN ncname=ITL(ng)%name ! IF (Master) THEN IF (LdefITL(ng)) THEN WRITE (stdout,10) ng, TRIM(ncname) ELSE WRITE (stdout,20) ng, TRIM(ncname) END IF END IF ! !======================================================================= ! Create a new tangent linear model initialization file. !======================================================================= ! DEFINE : IF (LdefITL(ng)) THEN CALL netcdf_create (ng, iTLM, TRIM(ncname), ITL(ng)%ncid) IF (FoundError(exit_flag, NoError, 154, MyFile)) THEN IF (Master) WRITE (stdout,30) TRIM(ncname) RETURN END IF ! !----------------------------------------------------------------------- ! Define the dimensions of staggered fields. !----------------------------------------------------------------------- ! DimIDs=0 ! status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'xi_rho', & & IOBOUNDS(ng)%xi_rho, DimIDs( 1)) IF (FoundError(exit_flag, NoError, 167, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'xi_u', & & IOBOUNDS(ng)%xi_u, DimIDs( 2)) IF (FoundError(exit_flag, NoError, 171, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'xi_v', & & IOBOUNDS(ng)%xi_v, DimIDs( 3)) IF (FoundError(exit_flag, NoError, 175, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'xi_psi', & & IOBOUNDS(ng)%xi_psi, DimIDs( 4)) IF (FoundError(exit_flag, NoError, 179, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'eta_rho', & & IOBOUNDS(ng)%eta_rho, DimIDs( 5)) IF (FoundError(exit_flag, NoError, 183, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'eta_u', & & IOBOUNDS(ng)%eta_u, DimIDs( 6)) IF (FoundError(exit_flag, NoError, 187, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'eta_v', & & IOBOUNDS(ng)%eta_v, DimIDs( 7)) IF (FoundError(exit_flag, NoError, 191, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'eta_psi', & & IOBOUNDS(ng)%eta_psi, DimIDs( 8)) IF (FoundError(exit_flag, NoError, 195, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'N', & & N(ng), DimIDs( 9)) IF (FoundError(exit_flag, NoError, 238, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 's_rho', & & N(ng), DimIDs( 9)) IF (FoundError(exit_flag, NoError, 242, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 's_w', & & N(ng)+1, DimIDs(10)) IF (FoundError(exit_flag, NoError, 246, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'tracer', & & NT(ng), DimIDs(11)) IF (FoundError(exit_flag, NoError, 250, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'boundary', & & 4, DimIDs(14)) IF (FoundError(exit_flag, NoError, 271, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, 'Nstate', & & NstateVar(ng), DimIDs(29)) IF (FoundError(exit_flag, NoError, 276, MyFile)) RETURN status=def_dim(ng, iTLM, ITL(ng)%ncid, ncname, & & TRIM(ADJUSTL(Vname(5,idtime))), & & nf90_unlimited, DimIDs(12)) IF (FoundError(exit_flag, NoError, 294, MyFile)) RETURN recdim=DimIDs(12) ! ! Set number of dimensions for output variables. ! nvd3=3 nvd4=4 ! ! Define dimension vectors for staggered tracer type variables. ! t2dgrd(1)=DimIDs( 1) t2dgrd(2)=DimIDs( 5) t2dgrd(3)=DimIDs(12) t3dgrd(1)=DimIDs( 1) t3dgrd(2)=DimIDs( 5) t3dgrd(3)=DimIDs( 9) t3dgrd(4)=DimIDs(12) ! ! Define dimension vectors for staggered u-momentum type variables. ! u2dgrd(1)=DimIDs( 2) u2dgrd(2)=DimIDs( 6) u2dgrd(3)=DimIDs(12) u3dgrd(1)=DimIDs( 2) u3dgrd(2)=DimIDs( 6) u3dgrd(3)=DimIDs( 9) u3dgrd(4)=DimIDs(12) ! ! Define dimension vectors for staggered v-momentum type variables. ! v2dgrd(1)=DimIDs( 3) v2dgrd(2)=DimIDs( 7) v2dgrd(3)=DimIDs(12) v3dgrd(1)=DimIDs( 3) v3dgrd(2)=DimIDs( 7) v3dgrd(3)=DimIDs( 9) v3dgrd(4)=DimIDs(12) ! ! Define dimension vector for staggered w-momentum type variables. ! w3dgrd(1)=DimIDs( 1) w3dgrd(2)=DimIDs( 5) w3dgrd(3)=DimIDs(10) w3dgrd(4)=DimIDs(12) ! ! Initialize local information variable arrays. ! DO i=1,Natt DO j=1,LEN(Vinfo(1)) Vinfo(i)(j:j)=' ' END DO END DO DO i=1,6 Aval(i)=0.0_r8 END DO ! !----------------------------------------------------------------------- ! Define time-recordless information variables. !----------------------------------------------------------------------- ! CALL def_info (ng, iTLM, ITL(ng)%ncid, ncname, DimIDs) IF (FoundError(exit_flag, NoError, 438, MyFile)) RETURN ! !----------------------------------------------------------------------- ! Define time-varying variables. !----------------------------------------------------------------------- ! ! Define model time. ! Vinfo( 1)=Vname(1,idtime) Vinfo( 2)=Vname(2,idtime) WRITE (Vinfo( 3),'(a,a)') 'seconds since ', TRIM(Rclock%string) Vinfo( 4)=TRIM(Rclock%calendar) Vinfo(14)=Vname(4,idtime) Vinfo(21)=Vname(6,idtime) status=def_var(ng, iTLM, ITL(ng)%ncid, ITL(ng)%Vid(idtime), & & NF_TOUT, 1, (/recdim/), Aval, Vinfo, ncname, & & SetParAccess = .FALSE.) IF (FoundError(exit_flag, NoError, 455, MyFile)) RETURN ! ! Define free-surface. ! Vinfo( 1)=Vname(1,idFsur) WRITE (Vinfo( 2),40) TRIM(Vname(2,idFsur)) Vinfo( 3)=Vname(3,idFsur) Vinfo(14)=Vname(4,idFsur) Vinfo(16)=Vname(1,idtime) Vinfo(21)=Vname(6,idFsur) Vinfo(22)='coordinates' Aval(5)=REAL(Iinfo(1,idFsur,ng),r8) status=def_var(ng, iTLM, ITL(ng)%ncid, ITL(ng)%Vid(idFsur), & & NF_FOUT, nvd3, t2dgrd, Aval, Vinfo, ncname) IF (FoundError(exit_flag, NoError, 477, MyFile)) RETURN ! ! Define 2D U-momentum component. ! Vinfo( 1)=Vname(1,idUbar) WRITE (Vinfo( 2),40) TRIM(Vname(2,idUbar)) Vinfo( 3)=Vname(3,idUbar) Vinfo(14)=Vname(4,idUbar) Vinfo(16)=Vname(1,idtime) Vinfo(21)=Vname(6,idUbar) Vinfo(22)='coordinates' Aval(5)=REAL(Iinfo(1,idUbar,ng),r8) status=def_var(ng, iTLM, ITL(ng)%ncid, ITL(ng)%Vid(idUbar), & & NF_FOUT, nvd3, u2dgrd, Aval, Vinfo, ncname) IF (FoundError(exit_flag, NoError, 514, MyFile)) RETURN ! ! Define 2D V-momentum component. ! Vinfo( 1)=Vname(1,idVbar) WRITE (Vinfo( 2),40) TRIM(Vname(2,idVbar)) Vinfo( 3)=Vname(3,idVbar) Vinfo(14)=Vname(4,idVbar) Vinfo(16)=Vname(1,idtime) Vinfo(21)=Vname(6,idVbar) Vinfo(22)='coordinates' Aval(5)=REAL(Iinfo(1,idVbar,ng),r8) status=def_var(ng, iTLM, ITL(ng)%ncid, ITL(ng)%Vid(idVbar), & & NF_FOUT, nvd3, v2dgrd, Aval, Vinfo, ncname) IF (FoundError(exit_flag, NoError, 551, MyFile)) RETURN ! ! Define 3D U-momentum component. ! Vinfo( 1)=Vname(1,idUvel) WRITE (Vinfo( 2),40) TRIM(Vname(2,idUvel)) Vinfo( 3)=Vname(3,idUvel) Vinfo(14)=Vname(4,idUvel) Vinfo(16)=Vname(1,idtime) Vinfo(21)=Vname(6,idUvel) Vinfo(22)='coordinates' Aval(5)=REAL(Iinfo(1,idUvel,ng),r8) status=def_var(ng, iTLM, ITL(ng)%ncid, ITL(ng)%Vid(idUvel), & & NF_FOUT, nvd4, u3dgrd, Aval, Vinfo, ncname) IF (FoundError(exit_flag, NoError, 626, MyFile)) RETURN ! ! Define 3D V-momentum component. ! Vinfo( 1)=Vname(1,idVvel) WRITE (Vinfo( 2),40) TRIM(Vname(2,idVvel)) Vinfo( 3)=Vname(3,idVvel) Vinfo(14)=Vname(4,idVvel) Vinfo(16)=Vname(1,idtime) Vinfo(21)=Vname(6,idVvel) Vinfo(22)='coordinates' Aval(5)=REAL(Iinfo(1,idVvel,ng),r8) status=def_var(ng, iTLM, ITL(ng)%ncid, ITL(ng)%Vid(idVvel), & & NF_FOUT, nvd4, v3dgrd, Aval, Vinfo, ncname) IF (FoundError(exit_flag, NoError, 663, MyFile)) RETURN ! ! Define tracer type variables. ! DO itrc=1,NT(ng) Vinfo( 1)=Vname(1,idTvar(itrc)) WRITE (Vinfo( 2),40) TRIM(Vname(2,idTvar(itrc))) Vinfo( 3)=Vname(3,idTvar(itrc)) Vinfo(14)=Vname(4,idTvar(itrc)) Vinfo(16)=Vname(1,idtime) Vinfo(21)=Vname(6,idTvar(itrc)) Vinfo(22)='coordinates' Aval(5)=REAL(r3dvar,r8) status=def_var(ng, iTLM, ITL(ng)%ncid, ITL(ng)%Tid(itrc), & & NF_FOUT, nvd4, t3dgrd, Aval, Vinfo, ncname) IF (FoundError(exit_flag, NoError, 708, MyFile)) RETURN END DO ! !----------------------------------------------------------------------- ! Leave definition mode. !----------------------------------------------------------------------- ! CALL netcdf_enddef (ng, iTLM, ncname, ITL(ng)%ncid) IF (FoundError(exit_flag, NoError, 777, MyFile)) RETURN ! !----------------------------------------------------------------------- ! Write out time-recordless, information variables. !----------------------------------------------------------------------- ! CALL wrt_info (ng, iTLM, ITL(ng)%ncid, ncname) IF (FoundError(exit_flag, NoError, 784, MyFile)) RETURN END IF DEFINE ! !======================================================================= ! Open an existing initialization file, check its contents, and ! prepare for appending data. !======================================================================= ! QUERY : IF (.not.LdefITL(ng)) THEN ncname=ITL(ng)%name ! ! Open tangent linear initialization file for read/write. ! CALL netcdf_open (ng, iTLM, ncname, 1, ITL(ng)%ncid) IF (FoundError(exit_flag, NoError, 798, MyFile)) THEN WRITE (stdout,60) TRIM(ncname) RETURN END IF ! ! Inquire about the dimensions and check for consistency. ! CALL netcdf_check_dim (ng, iTLM, ncname, & & ncid = ITL(ng)%ncid) IF (FoundError(exit_flag, NoError, 807, MyFile)) RETURN ! ! Inquire about the variables. ! CALL netcdf_inq_var (ng, iTLM, ncname, & & ncid = ITL(ng)%ncid) IF (FoundError(exit_flag, NoError, 813, MyFile)) RETURN ! ! Initialize logical switches. ! DO i=1,NV got_var(i)=.FALSE. END DO ! ! Scan variable list from input NetCDF and activate switches for ! initialization variables. Get variable IDs. ! DO i=1,n_var IF (TRIM(var_name(i)).eq.TRIM(Vname(1,idtime))) THEN got_var(idtime)=.TRUE. ITL(ng)%Vid(idtime)=var_id(i) ELSE IF (TRIM(var_name(i)).eq.TRIM(Vname(1,idFsur))) THEN got_var(idFsur)=.TRUE. ITL(ng)%Vid(idFsur)=var_id(i) ELSE IF (TRIM(var_name(i)).eq.TRIM(Vname(1,idUbar))) THEN got_var(idUbar)=.TRUE. ITL(ng)%Vid(idUbar)=var_id(i) ELSE IF (TRIM(var_name(i)).eq.TRIM(Vname(1,idVbar))) THEN got_var(idVbar)=.TRUE. ITL(ng)%Vid(idVbar)=var_id(i) ELSE IF (TRIM(var_name(i)).eq.TRIM(Vname(1,idUvel))) THEN got_var(idUvel)=.TRUE. ITL(ng)%Vid(idUvel)=var_id(i) ELSE IF (TRIM(var_name(i)).eq.TRIM(Vname(1,idVvel))) THEN got_var(idVvel)=.TRUE. ITL(ng)%Vid(idVvel)=var_id(i) END IF DO itrc=1,NT(ng) IF (TRIM(var_name(i)).eq.TRIM(Vname(1,idTvar(itrc)))) THEN got_var(idTvar(itrc))=.TRUE. ITL(ng)%Tid(itrc)=var_id(i) END IF END DO END DO ! ! Check if tangent linear mode initialization variables are available ! in input NetCDF file. ! IF (.not.got_var(idtime)) THEN IF (Master) WRITE (stdout,70) TRIM(Vname(1,idtime)), & & TRIM(ncname) exit_flag=3 RETURN END IF IF (.not.got_var(idFsur)) THEN IF (Master) WRITE (stdout,70) TRIM(Vname(1,idFsur)), & & TRIM(ncname) exit_flag=3 RETURN END IF IF (.not.got_var(idUbar)) THEN IF (Master) WRITE (stdout,70) TRIM(Vname(1,idUbar)), & & TRIM(ncname) exit_flag=3 RETURN END IF IF (.not.got_var(idVbar)) THEN IF (Master) WRITE (stdout,70) TRIM(Vname(1,idVbar)), & & TRIM(ncname) exit_flag=3 RETURN END IF IF (.not.got_var(idUvel)) THEN IF (Master) WRITE (stdout,70) TRIM(Vname(1,idUvel)), & & TRIM(ncname) exit_flag=3 RETURN END IF IF (.not.got_var(idVvel)) THEN IF (Master) WRITE (stdout,70) TRIM(Vname(1,idVvel)), & & TRIM(ncname) exit_flag=3 RETURN END IF DO itrc=1,NT(ng) IF (.not.got_var(idTvar(itrc))) THEN IF (Master) WRITE (stdout,70) TRIM(Vname(1,idTvar(itrc))), & & TRIM(ncname) exit_flag=3 RETURN END IF END DO ! ! Set unlimited time record dimension to the appropriate value. ! ITL(ng)%Rindex=rec_size Fcount=ITL(ng)%Fcount ITL(ng)%Nrec(Fcount)=rec_size END IF QUERY ! 10 FORMAT (2x,'TL_DEF_INI_NF90 - creating initial file,',t56, & & 'Grid ',i2.2,': ',a) 20 FORMAT (2x,'TL_DEF_INI_NF90 - inquiring initial file,',t56, & & 'Grid ',i2.2,': ',a) 30 FORMAT (/,' TL_DEF_INI_NF90 - unable to create initial NetCDF', & & ' file:',1x,a) 40 FORMAT ('tangent linear',1x,a) 50 FORMAT (1pe11.4,1x,'millimeter') 60 FORMAT (/,' TL_DEF_INI_NF90 - unable to open initial NetCDF', & & ' file: ',a) 70 FORMAT (/,' TL_DEF_INI_NF90 - unable to find variable: ',a,2x, & & ' in file: ',a) ! RETURN END SUBROUTINE tl_def_ini_nf90 END MODULE tl_def_ini_mod