Changeset 2390 in lmdz_wrf for trunk/tools/diagnostics.py

Timestamp:

Mar 11, 2019, 2:42:10 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `rhs_tas_tds': Computation of relative humidity from tas and tds

File:

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

trunk/tools/diagnostics.py

@@ -90 +90 @@
     #######
 availdiags = ['ACRAINTOT', 'accum', 'clt', 'cllmh', 'convini', 'deaccum', 'fog_K84', \
-  'fog_RUC', 'LMDZrh', 'mslp', 'OMEGAw', 'RAINTOT', 'range_faces',                   \
-  'rvors', 'td', 'timemax', 'timeoverthres', 'turbulence', 'uavaFROMwswd',           \
+  'fog_RUC', 'rhs_tas_tds', 'LMDZrh', 'mslp', 'OMEGAw', 'RAINTOT', 'range_faces',    \
+  'rvors', 'td', 'timemax', 'timeoverthres', 'turbulence', 'tws', 'uavaFROMwswd',    \
   'WRFbnds', 'WRFcape_afwa', 'WRFclivi', 'WRFclwvi', 'WRF_denszint', 'WRFgeop',      \
   'WRFmrso', 'WRFmrsos', 'WRFpotevap_orPM', 'WRFp', 'WRFpsl_ecmwf',                  \
   'WRFpsl_ptarget', 'WRFrvors', 'WRFslw', 'ws', 'wds', 'wss', 'WRFheight',           \
-  'WRFheightrel', 'WRFtda', 'WRFtdas', 'WRFua', 'WRFva', 'WRFzwind', 'WRFzwind_log', \
-  'WRFzwindMO']
+  'WRFheightrel', 'WRFtda', 'WRFtdas', 'WRFtws', 'WRFua', 'WRFva', 'WRFzwind',       \
+  'WRFzwind_log', 'WRFzwindMO']
 
 methods = ['accum', 'deaccum']
@@ -516 +516 @@
 varsadd = []
 
+Varns = ncobj.variables.keys()
+Varns.sort()
+
 for idiag in range(Ndiags):
     print '    diagnostic:',diags[idiag]
@@ -535 +538 @@
                     print '  ' + main + ": file '" + opts.ncfile +                   \
                       "' does not have variable '" + depv + "' !!"
+                    print '    available ones:', Varns
                     quit(-1)
         else:
@@ -544 +548 @@
                 print '  ' + main + ": file '" + opts.ncfile +                       \
                   "' does not have variable '" + depvars + "' !!"
+                print '    available ones:', Varns
                 quit(-1)
 
@@ -928 +933 @@
         ncvar.insert_variable(ncobj, 'lat_bnds', cellbndsy, diagoutd, diagoutvd, newnc)
 
-# tws: Bet Wulb temperature following Stull 2011 (tas, hur)
-    elif diagn == 'rws':
-            
-        var0 = ncobj.variables[depvars[0]][:]
-        var1 = ncobj.variables[depvars[1]][:]
-
-        diagout, diagoutd, diagoutvd = diag.var_tws_S11(var0,var1,dnames,dvnames)
+# tws: Bet Wulb temperature following Stull 2011 (tas, hurs)
+    elif diagn == 'tws':
+            
+        var0 = ncobj.variables[depvars[0]][:]
+        var1 = ncobj.variables[depvars[1]][:]
+
+        dnamesvar = list(ncobj.variables[depvars[0]].dimensions)
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+        diagoutd = dnames
+        diagoutvd = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        diagout = diag.var_tws_S11(var0,var1)
         ncvar.insert_variable(ncobj, 'tws', diagout, diagoutd, diagoutvd, newnc)
 
@@ -1224 +1234 @@
         ncvar.insert_variable(ncobj, 'hurs', diagout, diagoutd, diagoutvd, newnc)
 
+# rhs (psfc, t, q) from tas, tds
+    elif diagn == 'rhs_tas_tds':
+            
+        var0 = ncobj.variables[depvars[0]][:]
+        var1 = ncobj.variables[depvars[1]][:]
+
+        dnamesvar = ncobj.variables[depvars[0]].dimensions
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        diagout, diagoutd, diagoutvd = diag.var_hur_tas_tds(var0,var1,dnamesvar,     \
+          dvnamesvar)
+
+        ncvar.insert_variable(ncobj, 'hurs', diagout, diagoutd, diagoutvd, newnc)
+
 # slw: total soil liquid water SH2O, DZS
     elif diagn == 'WRFslw':