Changeset 3867 for dynamico_lmdz/aquaplanet/LMDZ5/libf/phylmd/regr_lat_time_climoz_m.F90

Timestamp:

Jan 8, 2016, 9:52:14 AM (9 years ago)

Author:

ymipsl

Message:

• Switch to XIOS 2
• Append main part of forcing configuration (LMDZ stand alone)
• Create etat0 and limit from usual LMDZ input files, using XIOS 2 interpolation functionnalities
  • missing parametrization of gravity waves
  • missing aerosol

File:

• dynamico_lmdz/aquaplanet/LMDZ5/libf/phylmd/regr_lat_time_climoz_m.F90 (modified) (15 diffs)

dynamico_lmdz/aquaplanet/LMDZ5/libf/phylmd/regr_lat_time_climoz_m.F90

@@ -11 +11 @@
 contains
 
-  subroutine regr_lat_time_climoz(read_climoz)
+  subroutine regr_lat_time_climoz(read_climoz, regr_lat)
 
     ! "regr_lat_time_climoz" stands for "regrid latitude time
@@ -65 +65 @@
     ! periodicity for interpolation at the beginning and at the end of the
     ! year.
-    use mod_grid_phy_lmdz, ONLY : nbp_lat
+    use mod_grid_phy_lmdz, ONLY : nbp_lat, grid_type, unstructured
     use regr1_step_av_m, only: regr1_step_av
     use regr3_lint_m, only: regr3_lint
@@ -75 +75 @@
     use regular_lonlat_mod, only : boundslat_reg, south
     use nrtype, only: pi
+    use regular_lonlat_mod, only : lat_reg
+    implicit none
     integer, intent(in):: read_climoz ! read ozone climatology
     ! Allowed values are 1 and 2
@@ -82 +84 @@
 
     ! Variables local to the procedure:
+    LOGICAL, intent(in), OPTIONAL :: regr_lat
 
     integer n_plev ! number of pressure levels in the input data
     integer n_lat ! number of latitudes in the input data
     integer n_month ! number of months in the input data
+    integer n_lat_out ! number of latitudes in the output data
 
     real, pointer:: latitude(:)
@@ -158 +162 @@
     ! (time to middle of day, in days since January 1st 0h, in a
     ! 360-day calendar)
-
+    logical :: regr_lat_
+    REAL,ALLOCATABLE :: lat_out(:)
+    
     !---------------------------------
+    IF (PRESENT(regr_lat)) THEN
+      regr_lat_=regr_lat
+    ELSE
+      regr_lat_=.TRUE.
+    ENDIF
 
     print *, "Call sequence information: regr_lat_time_climoz"
@@ -183 +194 @@
     forall (j = 2:n_lat) lat_in_edg(j) = (latitude(j - 1) + latitude(j)) / 2
     lat_in_edg(n_lat + 1) = pi / 2
-    deallocate(latitude) ! pointer
 
     call nf95_inq_varid(ncid_in, "plev", varid)
@@ -204 +214 @@
     end if
 
+    IF (regr_lat_) THEN 
+      n_lat_out=nbp_lat
+      ALLOCATE(lat_out(n_lat_out))
+      lat_out=lat_reg
+    ELSE 
+      n_lat_out=n_lat
+      ALLOCATE(lat_out(n_lat_out))
+      lat_out=latitude
+    ENDIF
+
     ! Create the output file and get the variable IDs:
-    call prepare_out(ncid_in, n_plev, ncid_out, varid_out, varid_plev, &
+    call prepare_out(ncid_in, n_lat_out, lat_out, n_plev, ncid_out, varid_out, varid_plev, &
          varid_time)
+
+    deallocate(latitude) ! pointer
 
     ! Write remaining coordinate variables:
@@ -281 +303 @@
     call nf95_close(ncid_in)
 
-    allocate(o3_regr_lat(nbp_lat, n_plev, 0:13, read_climoz))
-    allocate(o3_out(nbp_lat, n_plev, 360, read_climoz))
-
-    ! Regrid in latitude:
-    ! We average with respect to sine of latitude, which is
-    ! equivalent to weighting by cosine of latitude:
-    if (n_month == 12) then
-       print *, &
-            "Found 12 months in ozone climatologies, assuming periodicity..."
-       o3_regr_lat(nbp_lat:1:-1, :, 1:12, :) = regr1_step_av(o3_in, &
-            xs=sin(lat_in_edg), xt=sin((/- pi / 2, boundslat_reg(nbp_lat-1:1:-1,south), pi / 2/)))
-       ! (invert order of indices in "o3_regr_lat" because "rlatu" is
-       ! in descending order)
-
-       ! Duplicate January and December values, in preparation of time
-       ! interpolation:
-       o3_regr_lat(:, :, 0, :) = o3_regr_lat(:, :, 12, :)
-       o3_regr_lat(:, :, 13, :) = o3_regr_lat(:, :, 1, :)
-    else
-       print *, "Using 14 months in ozone climatologies..."
-       o3_regr_lat(nbp_lat:1:-1, :, :, :) = regr1_step_av(o3_in, &
-            xs=sin(lat_in_edg), xt=sin((/- pi / 2, boundslat_reg(nbp_lat-1:1:-1,south), pi / 2/)))
-       ! (invert order of indices in "o3_regr_lat" because "rlatu" is
-       ! in descending order)
-    end if
-
-    ! Regrid in time by linear interpolation:
-    o3_out = regr3_lint(o3_regr_lat, tmidmonth, tmidday)
+    allocate(o3_out(n_lat_out, n_plev, 360, read_climoz))
+    IF (regr_lat) THEN
+      allocate(o3_regr_lat(nbp_lat, n_plev, 0:13, read_climoz))
+
+      ! Regrid in latitude:
+      ! We average with respect to sine of latitude, which is
+      ! equivalent to weighting by cosine of latitude:
+      if (n_month == 12) then
+         print *, &
+              "Found 12 months in ozone climatologies, assuming periodicity..."
+         o3_regr_lat(nbp_lat:1:-1, :, 1:12, :) = regr1_step_av(o3_in, &
+              xs=sin(lat_in_edg), xt=sin((/- pi / 2, boundslat_reg(nbp_lat-1:1:-1,south), pi / 2/)))
+         ! (invert order of indices in "o3_regr_lat" because "rlatu" is
+         ! in descending order)
+
+         ! Duplicate January and December values, in preparation of time
+         ! interpolation:
+         o3_regr_lat(:, :, 0, :) = o3_regr_lat(:, :, 12, :)
+         o3_regr_lat(:, :, 13, :) = o3_regr_lat(:, :, 1, :)
+      else
+         print *, "Using 14 months in ozone climatologies..."
+         o3_regr_lat(nbp_lat:1:-1, :, :, :) = regr1_step_av(o3_in, &
+              xs=sin(lat_in_edg), xt=sin((/- pi / 2, boundslat_reg(nbp_lat-1:1:-1,south), pi / 2/)))
+         ! (invert order of indices in "o3_regr_lat" because "rlatu" is
+         ! in descending order)
+      end if
+      ! Regrid in time by linear interpolation:
+      o3_out = regr3_lint(o3_in, tmidmonth, tmidday)
+    ELSE
+      ! Regrid in time by linear interpolation:
+      o3_out = regr3_lint(o3_in, tmidmonth, tmidday)
+    ENDIF
 
     ! Write to file:
+
     do m = 1, read_climoz
-       call nf95_put_var(ncid_out, varid_out(m), o3_out(nbp_lat:1:-1, :, :, m))
+       IF (grid_type==unstructured) THEN
+         ! Doing spatial interpolation from XIOS need to have some point in longitude
+         ! waiting zonal mean operation from XIOS
+         call nf95_put_var(ncid_out, varid_out(m), SPREAD(o3_out(n_lat_out:1:-1, :, :, m),1,4))
+       ELSE
+         call nf95_put_var(ncid_out, varid_out(m), o3_out(n_lat_out:1:-1, :, :, m))
+       ENDIF
+       
        ! (The order of "rlatu" is inverted in the output file)
     end do
@@ -322 +356 @@
   !********************************************
 
-  subroutine prepare_out(ncid_in, n_plev, ncid_out, varid_out, varid_plev, &
+  subroutine prepare_out(ncid_in, n_lat, lat, n_plev, ncid_out, varid_out, varid_plev, &
        varid_time)
 
@@ -328 +362 @@
     ! dimensions and variables, and writes one of the coordinate variables.
 
-    use mod_grid_phy_lmdz, ONLY : nbp_lat
     use netcdf95, only: nf95_create, nf95_def_dim, nf95_def_var, &
          nf95_put_att, nf95_enddef, nf95_copy_att, nf95_put_var
-    use netcdf, only: nf90_clobber, nf90_float, nf90_global
+    use netcdf, only: nf90_clobber,nf90_netcdf4, nf90_float, nf90_global, nf90_unlimited
     use nrtype, only: pi
-    use regular_lonlat_mod, only : lat_reg
-    integer, intent(in):: ncid_in, n_plev
+    use mod_grid_phy_lmdz, ONLY :  grid_type, unstructured
+
+    integer, intent(in):: ncid_in, n_lat, n_plev
+    real, intent(in):: lat(:)
     integer, intent(out):: ncid_out, varid_plev, varid_time
 
@@ -344 +379 @@
 
     integer ncerr
-    integer dimid_rlatu, dimid_plev, dimid_time
-    integer varid_rlatu
+    integer dimid_rlatu, dimid_rlonv, dimid_plev, dimid_time
+    integer varid_rlonv, varid_rlatu
 
     !---------------------------
@@ -351 +386 @@
     print *, "Call sequence information: prepare_out"
 
-    call nf95_create("climoz_LMDZ.nc", nf90_clobber, ncid_out)
+    call nf95_create("climoz_LMDZ.nc", NF90_CLOBBER + NF90_NETCDF4, ncid_out)
 
     ! Dimensions:
-    call nf95_def_dim(ncid_out, "time", 360, dimid_time)
+    call nf95_def_dim(ncid_out, "time",  nf90_unlimited, dimid_time)
     call nf95_def_dim(ncid_out, "plev", n_plev, dimid_plev)
-    call nf95_def_dim(ncid_out, "rlatu", nbp_lat, dimid_rlatu)
+    call nf95_def_dim(ncid_out, "rlatu", n_lat, dimid_rlatu)
+    if (grid_type==unstructured) call nf95_def_dim(ncid_out, "rlonv", 4, dimid_rlonv)
 
     ! Define coordinate variables:
@@ -374 +410 @@
     call nf95_put_att(ncid_out, varid_rlatu, "standard_name", "latitude")
 
+    if (grid_type==unstructured) then
+      call nf95_def_var(ncid_out, "rlonv", nf90_float, dimid_rlonv, varid_rlonv)
+      call nf95_put_att(ncid_out, varid_rlonv, "units", "degrees_east")
+      call nf95_put_att(ncid_out, varid_rlonv, "standard_name", "longitude")
+    endif
+    
     ! Define the primary variables:
 
-    call nf95_def_var(ncid_out, "tro3", nf90_float, &
+    if (grid_type==unstructured) then
+        call nf95_def_var(ncid_out, "tro3", nf90_float, &
+         (/dimid_rlonv, dimid_rlatu, dimid_plev, dimid_time/), varid_out(1))
+    else
+        call nf95_def_var(ncid_out, "tro3", nf90_float, &
          (/dimid_rlatu, dimid_plev, dimid_time/), varid_out(1))
+    endif  
+    
     call nf95_put_att(ncid_out, varid_out(1), "long_name", &
          "ozone mole fraction")
@@ -384 +432 @@
 
     if (size(varid_out) == 2) then
-       call nf95_def_var(ncid_out, "tro3_daylight", nf90_float, &
-            (/dimid_rlatu, dimid_plev, dimid_time/), varid_out(2))
+      if (grid_type==unstructured) then
+         call nf95_def_var(ncid_out, "tro3_daylight", nf90_float, &
+             (/dimid_rlonv, dimid_rlatu, dimid_plev, dimid_time/), varid_out(2))
+      else
+         call nf95_def_var(ncid_out, "tro3_daylight", nf90_float, &
+             (/ dimid_rlatu, dimid_plev, dimid_time/), varid_out(2))
+      endif
+       
        call nf95_put_att(ncid_out, varid_out(2), "long_name", &
             "ozone mole fraction in daylight")
@@ -417 +471 @@
 
     ! Write one of the coordinate variables:
-    call nf95_put_var(ncid_out, varid_rlatu, lat_reg(nbp_lat:1:-1) / pi * 180.)
+    call nf95_put_var(ncid_out, varid_rlatu, lat(n_lat:1:-1) / pi * 180.)
     ! (convert from rad to degrees and sort in ascending order)
+    if (grid_type==unstructured)  call nf95_put_var(ncid_out, varid_rlonv, (/ 0., 90., 180.,270. /))
 
   contains