Changeset 1581 in lmdz_wrf

Timestamp:

Jun 9, 2017, 9:51:14 PM (8 years ago)

Author:

lfita

Message:

Adding: clivi', clwvl' and fixing 'mslp'

File:

• trunk/tools/diagnostics.py (modified) (10 diffs)

trunk/tools/diagnostics.py

@@ -304 +304 @@
                     zclear = 1.
                     zcloud = 0.
-                    ncvar.percendone(it*dx*dy + ix*dy + iy, dx*dy*dt, 5, 'diagnosted')
+                    gen.percendone(it*dx*dy + ix*dy + iy, dx*dy*dt, 5, 'diagnosted')
                     clt[it,iy,ix] = var_clt(cldfra[it,:,iy,ix])
 
@@ -318 +318 @@
                 zclear = 1.
                 zcloud = 0.
-                ncvar.percendone(ix*dy + iy, dx*dy*dt, 5, 'diagnosted')
+                gen.percendone(ix*dy + iy, dx*dy*dt, 5, 'diagnosted')
                 clt[iy,ix] = var_clt(cldfra[:,iy,ix])
 
@@ -463 +463 @@
             for ix in range(dx):
                 for iy in range(dy):
-                    ncvar.percendone(it*dx*dy + ix*dy + iy, dx*dy*dt, 5, 'diagnosted')
+                    gen.percendone(it*dx*dy + ix*dy + iy, dx*dy*dt, 5, 'diagnosted')
                     cllmh[:,it,iy,ix] = var_cllmh(cldfra[it,:,iy,ix], pres[it,:,iy,ix])
         
@@ -477 +477 @@
         for ix in range(dx):
             for iy in range(dy):
-                ncvar.percendone(ix*dy + iy,dx*dy, 5, 'diagnosted')
+                gen.percendone(ix*dy + iy,dx*dy, 5, 'diagnosted')
                 cllmh[:,iy,ix] = var_cllmh(cldfra[:,iy,ix], pres[:,iy,ix])
 
     return cllmh, cllmhdims, cllmhvdims
+
+def compute_clivi(dens, qtot, dimns, dimvns):
+    """ Function to compute cloud-ice water path (clivi)
+      [dens] = density [in kgkg-1] (assuming [t],z,y,x)
+      [qtot] = added mixing ratio of all cloud-ice species in [kgkg-1] (assuming [t],z,y,x)
+      [dimns]= list of the name of the dimensions of [q]
+      [dimvns]= list of the name of the variables with the values of the 
+        dimensions of [q]
+    """
+    fname = 'compute_clivi'
+
+    clividims = dimns[:]
+    clivivdims = dimvns[:]
+
+    if len(qtot.shape) == 4:
+        clividims.pop(1)
+        clivivdims.pop(1)
+    else:
+        clividims.pop(0)
+        clivivdims.pop(0)
+
+    data1 = dens*qtot
+    clivi = np.sum(data1, axis=1)
+
+    return clivi, clividims, clivivdims
+
+
+def compute_clwvl(dens, qtot, dimns, dimvns):
+    """ Function to compute condensed water path (clwvl)
+      [dens] = density [in kgkg-1] (assuming [t],z,y,x)
+      [qtot] = added mixing ratio of all cloud-water species in [kgkg-1] (assuming [t],z,y,x)
+      [dimns]= list of the name of the dimensions of [q]
+      [dimvns]= list of the name of the variables with the values of the 
+        dimensions of [q]
+    """
+    fname = 'compute_clwvl'
+
+    clwvldims = dimns[:]
+    clwvlvdims = dimvns[:]
+
+    if len(qtot.shape) == 4:
+        clwvldims.pop(1)
+        clwvlvdims.pop(1)
+    else:
+        clwvldims.pop(0)
+        clwvlvdims.pop(0)
+
+    data1 = dens*qtot
+    clwvl = np.sum(data1, axis=1)
+
+    return clwvl, clwvldims, clwvlvdims
 
 def var_virtualTemp (temp,rmix):
@@ -619 +670 @@
                           qvapor[it,:,iy,ix])
 
-                        ncvar.percendone(it*dx*dy + ix*dy + iy, dx*dy*dt, 5, 'diagnosted')
+                        gen.percendone(it*dx*dy + ix*dy + iy, dx*dy*dt, 5, 'diagnosted')
                 else:
                     mslpv[:,iy,ix] = psurface[:,iy,ix]
@@ -639 +690 @@
         for ix in range(dx):
             for iy in range(dy):
-                ncvar.percendone(ix*dy + iy,dx*dy, 5, 'diagnosted')
+                gen.percendone(ix*dy + iy,dx*dy, 5, 'diagnosted')
                 if terval[iy,ix] > 0.:
                     mslpv[iy,ix] = var_mslp(pressure[:,iy,ix], psurface[iy,ix],          \
@@ -879 +930 @@
 availdiags = ['ACRAINTOT', 'accum', 'clt', 'cllmh', 'deaccum', 'LMDZrh', 'mslp',     \
   'OMEGAw', 'RAINTOT',                                                               \
-  'rvors', 'td', 'turbulence', 'WRFgeop', 'WRFp', 'WRFrvors', 'ws', 'wds', 'wss',    \
-  'WRFheight', 'WRFheightrel', 'WRFua', 'WRFva']
+  'rvors', 'td', 'turbulence', 'WRFclivi', 'WRFclwvl', 'WRFgeop', 'WRFp',            \
+  'WRFrvors', 'ws', 'wds', 'wss', 'WRFheight', 'WRFheightrel', 'WRFua', 'WRFva']
 
 methods = ['accum', 'deaccum']
@@ -886 +937 @@
 # Variables not to check
 NONcheckingvars = ['cllmh', 'deaccum', 'TSrhs', 'TStd', 'TSwds', 'TSwss', 'WRFbils', \
-  'WRFdens', 'WRFgeop',                                                              \
+  'WRFclivi', 'WRFclwvl', 'WRFdens', 'WRFgeop',                                      \
   'WRFp', 'WRFtd',                                                                   \
   'WRFpos', 'WRFprc', 'WRFprls', 'WRFrh', 'LMDZrh', 'LMDZrhs', 'WRFrhs', 'WRFrvors', \
@@ -1305 +1356 @@
           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, 'psl', diagout, diagoutd, diagoutvd, newnc)
 
@@ -1472 +1529 @@
         ncvar.insert_variable(ncobj, 'bils', diagout, dnamesvar, dvnamesvar, newnc)
 
+# WRFclivi WRF water vapour path WRFdens, QICE, QGRAUPEL, QHAIL
+    elif diag == 'WRFclivi':
+            
+        var0 = WRFdens
+        qtot = ncobj.variables[depvars[1]]
+        qtotv = qtot[:]
+        Nspecies = len(depvars) - 2
+        for iv in range(Nspecies):
+            var1 = ncobj.variables[depvars[iv+2]][:]
+            qtotv = qtotv + var1
+
+        dnamesvar = list(qtot.dimensions)
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        diagout, diagoutd, diagoutvd = compute_clivi(var0, qtotv, 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, 'clivi', diagout, diagoutd, diagoutvd, newnc)
+
+# WRFclwvl WRF water cloud-condensed path WRFdens, QCLOUD, QICE, QGRAUPEL, QHAIL
+    elif diag == 'WRFclwvl':
+            
+        var0 = WRFdens
+        qtot = ncobj.variables[depvars[1]]
+        qtotv = ncobj.variables[depvars[1]]
+        Nspecies = len(depvars) - 2
+        for iv in range(Nspecies):
+            var1 = ncobj.variables[depvars[iv+2]]
+            qtotv = qtotv + var1[:]
+
+        dnamesvar = list(qtot.dimensions)
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        diagout, diagoutd, diagoutvd = compute_clwvl(var0, qtotv, 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, 'clwvl', diagout, diagoutd, diagoutvd, newnc)
+
 # WRFgeop geopotential from WRF as PH + PHB
     elif diag == 'WRFgeop':