Changeset 1762 in lmdz_wrf

Timestamp:

Jan 24, 2018, 3:54:22 PM (7 years ago)

Author:

lfita

Message:

Adding:

`mrso': total soil moisture
`slw': total liquid water

Location:

trunk/tools

Files:

• diag_tools.py (modified) (1 diff)
• diagnostics.inf (modified) (2 diffs)
• diagnostics.py (modified) (3 diffs)
• module_ForDiagnostics.f90 (modified) (1 diff)
• variables_values.dat (modified) (2 diffs)

trunk/tools/diag_tools.py

@@ -639 +639 @@
     return virtual
 
+def Forcompute_zint(var, zinterlev, zweights, dimns, dimvns):
+    """ Function to compute a vertical integration of volumetric quantities
+    Forcompute_mrso(smois, dsoil, dimns, dimvns)
+      [var]= values (assuming [[t],z,y,x]) [volumetric units]
+      [zinterlev]= depth of each layer (assuming [z]) [same z units as var]
+      [zweights]= weights to apply to each level (just in case...)
+      [dimns]= list of the name of the dimensions of [smois]
+      [dimvns]= list of the name of the variables with the values of the 
+        dimensions of [smois]
+    """
+    fname = 'Forcompute_zint'
+
+    vardims = dimns[:]
+    varvdims = dimvns[:]
+
+    if len(var.shape) == 4:
+        zint = np.zeros((var.shape[0],var.shape[2],var.shape[3]), dtype=np.float)
+        dx = var.shape[3]
+        dy = var.shape[2]
+        dz = var.shape[1]
+        dt = var.shape[0]
+        vardims.pop(1)
+        varvdims.pop(1)
+
+        zintvart=fdin.module_fordiagnostics.compute_zint4d(var4d=var[:].transpose(), \
+          dlev=zinterlev[:].transpose(), zweight=zweights[:].transpose(), d1=dx,     \
+          d2=dy, d3=dz, d4=dt)
+        zintvar = zintvart.transpose()
+    else:
+        print errormsg
+        print '  ' + fname + ': rank', len(var.shape), 'not ready !!'
+        print '  it only computes 4D [t,z,y,x] rank values'
+        quit(-1)
+
+    return zintvar, vardims, varvdims
 
 def var_mslp(pres, psfc, ter, tk, qv):

trunk/tools/diagnostics.inf

@@ -16 +16 @@
 hurs, TSrhs, psfc@t@q
 hurs, WRFrhs, PSFC@T2@Q2
+mrso, WRFmrso,  SMOIS@DZS
 p, WRFp, P@PB
 pr, RAINTOT, RAINC@RAINNC@WRFtime
@@ -26 +27 @@
 rvor, rvor, U@V
 rvors, WRFrvors, U10@V10@WRFpos
+slw, WRFslw,  SH2O@DZS
 ta, WRFt, T@P@PB
 td, WRFtd, WRFp@WRFt2@QVAPOR

trunk/tools/diagnostics.py

@@ -78 +78 @@
 availdiags = ['ACRAINTOT', 'accum', 'clt', 'cllmh', 'deaccum', 'LMDZrh', 'mslp',     \
   'OMEGAw', 'RAINTOT',                                                               \
-  'rvors', 'td', 'turbulence', 'WRFclivi', 'WRFclwvl', 'WRFgeop', 'WRFp',            \
-  'WRFpsl_ptarget', 'WRFrvors', 'ws', 'wds', 'wss', 'WRFheight', 'WRFheightrel',     \
-  'WRFua', 'WRFva']
+  'rvors', 'td', 'turbulence', 'WRFcape_afwa', 'WRFclivi', 'WRFclwvl', 'WRFgeop',    \
+  'WRFmrso', 'WRFp',                                                                 \
+  'WRFpsl_ptarget', 'WRFrvors', 'WRFslw', 'ws', 'wds', 'wss', 'WRFheight',           \
+  'WRFheightrel', 'WRFua', 'WRFva']
 
 methods = ['accum', 'deaccum']
@@ -489 +490 @@
 
         ncvar.insert_variable(ncobj, 'hurs', diagout, diagoutd, diagoutvd, newnc)
+
+# mrso: total soil moisture SMOIS, DZS
+    elif diagn == 'WRFmrso':
+            
+        var0 = ncobj.variables[depvars[0]][:]
+        var10 = ncobj.variables[depvars[1]][:]
+        dnamesvar = list(ncobj.variables[depvars[0]].dimensions)
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        var1 = var0.copy()*0.
+        var2 = var0.copy()*0.+1.
+        # Must be a better way....
+        for j in range(var0.shape[2]):
+          for i in range(var0.shape[3]):
+              var1[:,:,j,i] = var10
+
+        diagout, diagoutd, diagoutvd = diag.Forcompute_zint(var0, var1, var2,        \
+          dnamesvar, dvnamesvar)
+
+        # Removing the nonChecking variable-dimensions from the initial list
+        varsadd = []
+        diagoutvd = list(dvnames)
+        for nonvd in NONchkvardims:
+            if gen.searchInlist(dvnames,nonvd): diagoutvd.remove(nonvd)
+            varsadd.append(nonvd)
+        ncvar.insert_variable(ncobj, 'mrso', diagout, diagoutd, diagoutvd, newnc)
 
 # mslp: mean sea level pressure (pres, psfc, terrain, temp, qv)
@@ -601 +628 @@
         ncvar.insert_variable(ncobj, 'hurs', diagout, diagoutd, diagoutvd, newnc)
 
+# slw: total soil liquid water SH2O, DZS
+    elif diagn == 'WRFslw':
+            
+        var0 = ncobj.variables[depvars[0]][:]
+        var10 = ncobj.variables[depvars[1]][:]
+        dnamesvar = list(ncobj.variables[depvars[0]].dimensions)
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        var1 = var0.copy()*0.
+        var2 = var0.copy()*0.+1.
+        # Must be a better way....
+        for j in range(var0.shape[2]):
+          for i in range(var0.shape[3]):
+              var1[:,:,j,i] = var10
+
+        diagout, diagoutd, diagoutvd = diag.Forcompute_zint(var0, var1, var2,        \
+          dnamesvar, dvnamesvar)
+
+        # Removing the nonChecking variable-dimensions from the initial list
+        varsadd = []
+        diagoutvd = list(dvnames)
+        for nonvd in NONchkvardims:
+            if gen.searchInlist(dvnames,nonvd): diagoutvd.remove(nonvd)
+            varsadd.append(nonvd)
+        ncvar.insert_variable(ncobj, 'slw', diagout, diagoutd, diagoutvd, newnc)
+
 # td (psfc, t, q) from TimeSeries files
     elif diagn == 'TStd' or diagn == 'td':

trunk/tools/module_ForDiagnostics.f90

@@ -533 +533 @@
 
   END SUBROUTINE compute_psl_ptarget4d2
+
+  SUBROUTINE compute_massvertint1D(var, mutot, dz, deta, integral)
+    ! Subroutine to vertically integrate a 1D variable in eta vertical coordinates
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: dz
+    REAL(r_k), INTENT(in)                                :: mutot
+    REAL(r_k), DIMENSION(dz), INTENT(in)                 :: var, deta
+    REAL(r_k), INTENT(out)                               :: integral
+
+! Local
+    INTEGER                                              :: k
+
+!!!!!!! Variables
+! var: vertical variable to integrate (assuming kgkg-1)
+! mutot: total dry-air mass in column
+! dz: vertical dimension
+! deta: eta-levels difference between full eta-layers
+
+    fname = 'compute_massvertint1D'
+
+!    integral=0.
+!    DO k=1,dz
+!      integral = integral + var(k)*deta(k)
+!    END DO
+     integral = SUM(var*deta)
+
+    integral=integral*mutot/g
+
+    RETURN
+
+  END SUBROUTINE compute_massvertint1D
+
+  SUBROUTINE compute_zint4D(var4D, dlev, zweight, d1, d2, d3, d4, int3D)
+    ! Subroutine to vertically integrate a 4D variable in any vertical coordinates
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: d1,d2,d3,d4
+    REAL(r_k), DIMENSION(d1,d2,d3,d4), INTENT(in)        :: var4D, dlev, zweight
+    REAL(r_k), DIMENSION(d1,d2,d4), INTENT(out)          :: int3D
+
+! Local
+    INTEGER                                              :: i,j,l
+
+!!!!!!! Variables
+! var4D: vertical variable to integrate
+! dlev: height of layers
+! zweight: weight for each level to be applied (=1. for no effect)
+
+    fname = 'compute_zint4D'
+
+    DO i=1,d1
+      DO j=1,d2
+        DO l=1,d4
+          CALL compute_vertint1D(var4D(i,j,:,l),d3, dlev(i,j,:,l), zweight(i,j,:,l), &
+            int3D(i,j,l))
+        END DO
+      END DO
+    END DO
+
+    RETURN
+
+  END SUBROUTINE compute_zint4D
+
+  SUBROUTINE compute_vertint1D(var, dz, deta, zweight, integral)
+    ! Subroutine to vertically integrate a 1D variable in any vertical coordinates
+
+    IMPLICIT NONE
+
+    INTEGER, INTENT(in)                                  :: dz
+    REAL(r_k), DIMENSION(dz), INTENT(in)                 :: var, deta, zweight
+    REAL(r_k), INTENT(out)                               :: integral
+
+! Local
+    INTEGER                                              :: k
+
+!!!!!!! Variables
+! var: vertical variable to integrate
+! dz: vertical dimension
+! deta: eta-levels difference between layers
+! zweight: weight for each level to be applied (=1. for no effect)
+
+    fname = 'compute_vertint1D'
+
+!    integral=0.
+!    DO k=1,dz
+!      integral = integral + var(k)*deta(k)
+!    END DO
+    integral = SUM(var*deta*zweight)
+
+    RETURN
+
+  END SUBROUTINE compute_vertint1D
 
   SUBROUTINE compute_tv4d(ta,qv,tv,d1,d2,d3,d4)

trunk/tools/variables_values.dat

@@ -249 +249 @@
 lon, lon, longitude, 0., 360., longitude, degrees_East, seismic
 nav_lon, lon, longitude, 0., 360., longitude, degrees_East, seismic
+mrso, mrso, soil_moisture_content, 0., 0.003, Total|soil|moisture|content, kg m-2, Blues
 mrsos, mrsos, moisture_content_of_soil_layer, 0., 0.003, Moisture|content|of|first|soil|layer, kg m-2, Blues
 SMOIS, mrsos, moisture_content_of_soil_layer, 0., 0.003, Moisture|content|of|first|soil|layer, kg m-2, Blues
@@ -397 +398 @@
 s, s, snow_mixing_ratio, 0., 0.0003, snow|mixing|ratio, kgkg-1, Purples
 QSNOW, s, snow_mixing_ratio, 0., 0.0003, snow|mixing|ratio, kgkg-1, Purples
+slw, slw, soil_liquid_water, 0., 0.003, Total|soil|liquid|water|content, kg m-2, Blues
 sftls, sftls, land_binary_mask, 0., 1., land|mask, 1, bwr
 LANDMASK, sftls, land_binary_mask, 0., 1., land|mask, 1, bwr