Changeset 2206 in lmdz_wrf for trunk/tools

Timestamp:

Oct 31, 2018, 3:22:45 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `mrsos': first layer land moisture [kg m- 2]

Location:

trunk/tools

Files:

• diagnostics.inf (modified) (1 diff)
• diagnostics.py (modified) (2 diffs)

trunk/tools/diagnostics.inf

@@ -23 +23 @@
 hurs, WRFrhs, PSFC@T2@Q2
 mrso, WRFmrso,  SMOIS@DZS
+mrsos, WRFmrsos,  SMOIS@DZS
 p, WRFp, P@PB
 evspsblpot, WRFpotevap_orPM, WRFdens@UST@U10@V10@T2@PSFC@QVAPOR

trunk/tools/diagnostics.py

@@ -91 +91 @@
   'rvors', 'td', 'timemax', 'timeoverthres', 'turbulence', 'uavaFROMwswd',           \
   'WRFcape_afwa', 'WRFclivi', 'WRFclwvi', 'WRF_denszint', 'WRFgeop',                 \
-  'WRFmrso', 'WRFpotevap_orPM', 'WRFp', 'WRFpsl_ecmwf',                              \
+  'WRFmrso', 'WRFmrsos', 'WRFpotevap_orPM', 'WRFp', 'WRFpsl_ecmwf',                  \
   'WRFpsl_ptarget', 'WRFrvors', 'WRFslw', 'ws', 'wds', 'wss', 'WRFheight',           \
   'WRFheightrel', 'WRFtda', 'WRFtdas', 'WRFua', 'WRFva', 'WRFzwind', 'WRFzwind_log', \
@@ -702 +702 @@
         ncvar.insert_variable(ncobj, 'mrso', diagout, diagoutd, diagoutvd, newnc)
 
+# mrsos: First layer soil moisture SMOIS, DZS
+    elif diagn == 'WRFmrsos':
+            
+        var0 = ncobj.variables[depvars[0]][:]
+        var1 = ncobj.variables[depvars[1]][:]
+        diagoutd = list(ncobj.variables[depvars[0]].dimensions)
+        diagoutvd = ncvar.var_dim_dimv(diagoutd,dnames,dvnames)
+
+        diagoutd.pop(1)
+        diagoutvd.pop(1)
+
+        diagout= np.zeros((var0.shape[0],var0.shape[2],var0.shape[3]), dtype=np.float)
+
+        # Must be a better way....
+        for j in range(var0.shape[2]):
+          for i in range(var0.shape[3]):
+              print var0[:,0,j,i]*var1[:,0]
+              diagout[:,j,i] = var0[:,0,j,i]*var1[:,0]
+
+        # 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, 'mrsos', diagout, diagoutd, diagoutvd, newnc)
+
 # mslp: mean sea level pressure (pres, psfc, terrain, temp, qv)
     elif diagn == 'mslp' or diagn == 'WRFmslp':