source: LMDZ6/branches/Amaury_dev/libf/dyn3d/dynredem.F90 @ 5182

SUBROUTINE dynredem0(fichnom, iday_end, phis)

  !-------------------------------------------------------------------------------
  ! Write the NetCDF restart file (initialization).
  !-------------------------------------------------------------------------------
  USE IOIPSL
  USE lmdz_strings, ONLY: maxlen
  USE lmdz_infotrac, ONLY: nqtot, tracers
  USE netcdf, ONLY: nf90_create, nf90_def_dim, nf90_inq_varid, nf90_global, &
          nf90_close, nf90_put_att, nf90_unlimited, nf90_clobber, &
          nf90_64bit_offset
  USE dynredem_mod, ONLY: cre_var, put_var1, put_var2, err, modname, fil
  USE comvert_mod, ONLY: ap, bp, presnivs, pa, preff, nivsig, nivsigs
  USE comconst_mod, ONLY: cpp, daysec, dtvr, g, kappa, omeg, rad
  USE logic_mod, ONLY: fxyhypb, ysinus
  USE serre_mod, ONLY: clon, clat, grossismx, grossismy, dzoomx, dzoomy, &
          taux, tauy
  USE temps_mod, ONLY: annee_ref, day_ref, itau_dyn, itaufin, start_time
  USE ener_mod, ONLY: etot0, ptot0, ztot0, stot0, ang0
  USE lmdz_description, ONLY: descript
  USE lmdz_iniprint, ONLY: lunout, prt_level
  USE lmdz_comgeom2

USE lmdz_dimensions, ONLY: iim, jjm, llm, ndm
  USE lmdz_paramet
  IMPLICIT NONE


  !===============================================================================
  ! Arguments:
  CHARACTER(LEN = *), INTENT(IN) :: fichnom          !--- FILE NAME
  INTEGER, INTENT(IN) :: iday_end         !---
  REAL, INTENT(IN) :: phis(iip1, jjp1) !--- GROUND GEOPOTENTIAL
  !===============================================================================
  ! Local variables:
  INTEGER :: iq
  INTEGER, PARAMETER :: length = 100
  REAL :: tab_cntrl(length)                     !--- RUN PARAMETERS TABLE
  !   For NetCDF:
  CHARACTER(LEN = maxlen) :: unites
  INTEGER :: indexID
  INTEGER :: rlonuID, rlonvID, rlatuID, rlatvID
  INTEGER :: sID, sigID, nID, timID
  INTEGER :: yyears0, jjour0, mmois0
  REAL :: zjulian, hours
  !===============================================================================
  modname = 'dynredem0'; fil = fichnom
  CALL ymds2ju(annee_ref, 1, iday_end, 0.0, zjulian)
  CALL ju2ymds(zjulian, yyears0, mmois0, jjour0, hours)

  tab_cntrl(:) = 0.
  tab_cntrl(1) = REAL(iim)
  tab_cntrl(2) = REAL(jjm)
  tab_cntrl(3) = REAL(llm)
  tab_cntrl(4) = REAL(day_ref)
  tab_cntrl(5) = REAL(annee_ref)
  tab_cntrl(6) = rad
  tab_cntrl(7) = omeg
  tab_cntrl(8) = g
  tab_cntrl(9) = cpp
  tab_cntrl(10) = kappa
  tab_cntrl(11) = daysec
  tab_cntrl(12) = dtvr
  tab_cntrl(13) = etot0
  tab_cntrl(14) = ptot0
  tab_cntrl(15) = ztot0
  tab_cntrl(16) = stot0
  tab_cntrl(17) = ang0
  tab_cntrl(18) = pa
  tab_cntrl(19) = preff

  !    .....    parameters for zoom    ......
  tab_cntrl(20) = clon
  tab_cntrl(21) = clat
  tab_cntrl(22) = grossismx
  tab_cntrl(23) = grossismy

  IF (fxyhypb)   THEN
    tab_cntrl(24) = 1.
    tab_cntrl(25) = dzoomx
    tab_cntrl(26) = dzoomy
    tab_cntrl(27) = 0.
    tab_cntrl(28) = taux
    tab_cntrl(29) = tauy
  ELSE
    tab_cntrl(24) = 0.
    tab_cntrl(25) = dzoomx
    tab_cntrl(26) = dzoomy
    tab_cntrl(27) = 0.
    tab_cntrl(28) = 0.
    tab_cntrl(29) = 0.
    IF(ysinus)  tab_cntrl(27) = 1.
  END IF
  tab_cntrl(30) = REAL(iday_end)
  tab_cntrl(31) = REAL(itau_dyn + itaufin)
  ! start_time: start_time of simulation (not necessarily 0.)
  tab_cntrl(32) = start_time

  !--- File creation
  CALL err(nf90_create(fichnom, IOR(nf90_clobber, nf90_64bit_offset), nid))

  !--- Some global attributes
  CALL err(nf90_put_att(nid, nf90_global, "title", "Fichier demarrage dynamique"))

  !--- Dimensions
  CALL err(nf90_def_dim(nid, "index", length, indexID))
  CALL err(nf90_def_dim(nid, "rlonu", iip1, rlonuID))
  CALL err(nf90_def_dim(nid, "rlatu", jjp1, rlatuID))
  CALL err(nf90_def_dim(nid, "rlonv", iip1, rlonvID))
  CALL err(nf90_def_dim(nid, "rlatv", jjm, rlatvID))
  CALL err(nf90_def_dim(nid, "sigs", llm, sID))
  CALL err(nf90_def_dim(nid, "sig", llmp1, sigID))
  CALL err(nf90_def_dim(nid, "temps", nf90_unlimited, timID))

  !--- Define and save invariant fields
  CALL put_var1(nid, "controle", "Parametres de controle", [indexID], tab_cntrl)
  CALL put_var1(nid, "rlonu", "Longitudes des points U", [rlonuID], rlonu)
  CALL put_var1(nid, "rlatu", "Latitudes des points U", [rlatuID], rlatu)
  CALL put_var1(nid, "rlonv", "Longitudes des points V", [rlonvID], rlonv)
  CALL put_var1(nid, "rlatv", "Latitudes des points V", [rlatvID], rlatv)
  CALL put_var1(nid, "nivsigs", "Numero naturel des couches s", [sID], nivsigs)
  CALL put_var1(nid, "nivsig", "Numero naturel des couches sigma", [sigID], nivsig)
  CALL put_var1(nid, "ap", "Coefficient A pour hybride", [sigID], ap)
  CALL put_var1(nid, "bp", "Coefficient B pour hybride", [sigID], bp)
  CALL put_var1(nid, "presnivs", "", [sID], presnivs)
  ! covariant <-> contravariant <-> natural conversion coefficients
  CALL put_var2(nid, "cu", "Coefficient de passage pour U", [rlonuID, rlatuID], cu)
  CALL put_var2(nid, "cv", "Coefficient de passage pour V", [rlonvID, rlatvID], cv)
  CALL put_var2(nid, "aire", "Aires de chaque maille", [rlonvID, rlatuID], aire)
  CALL put_var2(nid, "phisinit", "Geopotentiel au sol", [rlonvID, rlatuID], phis)

  !--- Define fields saved later
  WRITE(unites, "('days since ',i4,'-',i2.2,'-',i2.2,' 00:00:00')") &
          yyears0, mmois0, jjour0
  CALL cre_var(nid, "temps", "Temps de simulation", [timID], unites)
  CALL cre_var(nid, "ucov", "Vitesse U", [rlonuID, rlatuID, sID, timID])
  CALL cre_var(nid, "vcov", "Vitesse V", [rlonvID, rlatvID, sID, timID])
  CALL cre_var(nid, "teta", "Temperature", [rlonvID, rlatuID, sID, timID])
  DO iq = 1, nqtot
    CALL cre_var(nid, tracers(iq)%name, tracers(iq)%longName, [rlonvID, rlatuID, sID, timID])
  END DO
  CALL cre_var(nid, "masse", "Masse d air", [rlonvID, rlatuID, sID, timID])
  CALL cre_var(nid, "ps", "Pression au sol", [rlonvID, rlatuID, timID])
  CALL err(nf90_close (nid))

  WRITE(lunout, *)TRIM(modname) // ': iim,jjm,llm,iday_end', iim, jjm, llm, iday_end
  WRITE(lunout, *)TRIM(modname) // ': rad,omeg,g,cpp,kappa', rad, omeg, g, cpp, kappa

END SUBROUTINE dynredem0

!-------------------------------------------------------------------------------


!-------------------------------------------------------------------------------

SUBROUTINE dynredem1(fichnom, time, vcov, ucov, teta, q, masse, ps)

  !-------------------------------------------------------------------------------
  ! Purpose: Write the NetCDF restart file (append).
  !-------------------------------------------------------------------------------
  USE lmdz_strings, ONLY: maxlen
  USE lmdz_infotrac, ONLY: nqtot, tracers, type_trac
  USE control_mod
  USE netcdf, ONLY: nf90_open, nf90_nowrite, nf90_get_var, nf90_inq_varid, &
          nf90_close, nf90_write, nf90_put_var, nf90_noerr
  USE dynredem_mod, ONLY: dynredem_write_u, dynredem_write_v, dynredem_read_u, &
          err, modname, fil, msg
  USE temps_mod, ONLY: itau_dyn, itaufin
  USE lmdz_description, ONLY: descript
  USE lmdz_iniprint, ONLY: lunout, prt_level
  USE lmdz_comgeom

USE lmdz_dimensions, ONLY: iim, jjm, llm, ndm
  USE lmdz_paramet
  IMPLICIT NONE


  !===============================================================================
  ! Arguments:
  CHARACTER(LEN = *), INTENT(IN) :: fichnom              !-- FILE NAME
  REAL, INTENT(IN) :: time                         !-- TIME
  REAL, INTENT(IN) :: vcov(iip1, jjm, llm)          !-- V COVARIANT WIND
  REAL, INTENT(IN) :: ucov(iip1, jjp1, llm)          !-- U COVARIANT WIND
  REAL, INTENT(IN) :: teta(iip1, jjp1, llm)          !-- POTENTIAL TEMPERATURE
  REAL, INTENT(INOUT) :: q(iip1, jjp1, llm, nqtot)    !-- TRACERS
  REAL, INTENT(IN) :: masse(iip1, jjp1, llm)          !-- MASS PER CELL
  REAL, INTENT(IN) :: ps(iip1, jjp1)              !-- GROUND PRESSURE
  !===============================================================================
  ! Local variables:
  INTEGER :: iq, nid, vID, ierr, nid_trac, vID_trac
  INTEGER, SAVE :: nb = 0
  INTEGER, PARAMETER :: length = 100
  REAL :: tab_cntrl(length) ! tableau des parametres du run
  CHARACTER(LEN = maxlen) :: var, dum
  LOGICAL :: lread_inca
  !===============================================================================

  modname = 'dynredem1'; fil = fichnom
  CALL err(nf90_open(fil, nf90_write, nid), "open", fil)

  !--- Write/extend time coordinate
  nb = nb + 1
  var = "temps"
  CALL err(nf90_inq_varid(nid, var, vID), "inq", var)
  CALL err(nf90_put_var(nid, vID, [time]), "put", var)
  WRITE(lunout, *)TRIM(modname) // ": Saving for ", nb, time

  !--- Rewrite control table (itaufin undefined in dynredem0)
  var = "controle"
  CALL err(nf90_inq_varid(nid, var, vID), "inq", var)
  CALL err(nf90_get_var(nid, vID, tab_cntrl), "get", var)
  tab_cntrl(31) = DBLE(itau_dyn + itaufin)
  CALL err(nf90_inq_varid(nid, var, vID), "inq", var)
  CALL err(nf90_put_var(nid, vID, tab_cntrl), "put", var)

  !--- Save fields
  CALL dynredem_write_u(nid, "ucov", ucov, llm)
  CALL dynredem_write_v(nid, "vcov", vcov, llm)
  CALL dynredem_write_u(nid, "teta", teta, llm)
  CALL dynredem_write_u(nid, "masse", masse, llm)
  CALL dynredem_write_u(nid, "ps", ps, 1)

  !--- Tracers in file "start_trac.nc" (added by Anne)
  lread_inca = .FALSE.; fil = "start_trac.nc"
  IF(ANY(type_trac == ['inca', 'inco'])) INQUIRE(FILE = fil, EXIST = lread_inca)
  IF(lread_inca) CALL err(nf90_open(fil, nf90_nowrite, nid_trac), "open")

  !--- Save tracers
  DO iq = 1, nqtot; var = TRIM(tracers(iq)%name); ierr = -1
  IF(lread_inca) THEN                  !--- Possibly read from "start_trac.nc"
    fil = "start_trac.nc"
    ierr = nf90_inq_varid(nid_trac, var, vID_trac)
    dum = 'inq'; IF(ierr==nf90_noerr) dum = 'fnd'
    WRITE(lunout, *)msg(dum, var)

    IF(ierr==nf90_noerr) CALL dynredem_read_u(nid_trac, var, q(:, :, :, iq), llm)
  END IF
  fil = fichnom
  CALL dynredem_write_u(nid, var, q(:, :, :, iq), llm)
  END DO
  CALL err(nf90_close(nid), "close")
  fil = "start_trac.nc"
  IF(lread_inca) CALL err(nf90_close(nid_trac), "close")

END SUBROUTINE dynredem1