Changeset 959 in lmdz_wrf for trunk/tools

Timestamp:

Jun 27, 2016, 12:05:22 PM (8 years ago)

Author:

lfita

Message:

Adding `ws': air wind speed

Location:

trunk/tools

Files:

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

trunk/tools/diagnostics.inf

@@ -30 +30 @@
 wds, TSwds, u@v
 wds, wds, U10@V10
-#ws, SPDUV, U@V
+ws, ws, U@V
 #wss, SPDUV10, U10@V10
 #wss, TSwss, u@v
 wss, wss, U10@V10
+wssturb, turbulence, wss
 

trunk/tools/diagnostics.py

@@ -877 +877 @@
 availdiags = ['ACRAINTOT', 'accum', 'clt', 'cllmh', 'deaccum', 'LMDZrh', 'mslp',     \
   'OMEGAw', 'RAINTOT',   \
-  'rvors', 'td', 'turbulence', 'WRFgeop', 'WRFp', 'WRFrvors', 'wds', 'wss',          \
+  'rvors', 'td', 'turbulence', 'WRFgeop', 'WRFp', 'WRFrvors', 'ws', 'wds', 'wss',    \
   'WRFheight', 'WRFua', 'WRFva']
 
@@ -1382 +1382 @@
 
         diagout, diagoutd, diagoutvd = compute_turbulence(var0,dnamesvar,dvnamesvar)
-        valsvar = ncvar.variables_values(depvars)
-
-        ncvar.insert_variable(ncobj, valsvar[0] + 'turb', diagout, diagoutd, 
+        valsvar = gen.variables_values(depvars)
+
+        newvarn = depvars + 'turb'
+        print main + '; Lluis newvarn:', newvarn
+        ncvar.insert_variable(ncobj, newvarn, diagout, diagoutd, 
           diagoutvd, newnc)
-        varobj = newnc.variables[valsvar[0] + 'turb']
+        print main + '; Lluis variables:', newnc.variables.keys()
+        varobj = newnc.variables[newvarn]
         attrv = varobj.long_name
         attr = varobj.delncattr('long_name')
@@ -1507 +1510 @@
             ua[:,iz,:,:] = unstgvar0[:,iz,:,:]*var3 - unstgvar1[:,iz,:,:]*var2
 
-        dnamesvar = list(ncobj.variables[depvars[0]].dimensions)
+#        dnamesvar = list(ncobj.variables[depvars[0]].dimensions)
+        dnamesvar = ['Time','bottom_top','south_north','west_east']
         dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
 
@@ -1529 +1533 @@
             va[:,iz,:,:] = unstgvar0[:,iz,:,:]*var2 + unstgvar1[:,iz,:,:]*var3
 
-        dnamesvar = list(ncobj.variables[depvars[0]].dimensions)
+        dnamesvar = ['Time','bottom_top','south_north','west_east']
         dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
 
@@ -1544 +1548 @@
         ncvar.insert_variable(ncobj, 'time', diagout, dnamesvar, dvnamesvar, newnc)
 
+# ws (U, V)
+    elif diag == 'ws':
+            
+        var0 = ncobj.variables[depvars[0]][:]
+        var1 = ncobj.variables[depvars[1]][:]
+        # un-staggering variables
+        unstgdims = [var0.shape[0], var0.shape[1], var0.shape[2], var0.shape[3]-1]
+        va = np.zeros(tuple(unstgdims), dtype=np.float)
+        unstgvar0 = np.zeros(tuple(unstgdims), dtype=np.float)
+        unstgvar1 = np.zeros(tuple(unstgdims), dtype=np.float)
+        unstgvar0 = 0.5*(var0[:,:,:,0:var0.shape[3]-1] + var0[:,:,:,1:var0.shape[3]])
+        unstgvar1 = 0.5*(var1[:,:,0:var1.shape[2]-1,:] + var1[:,:,1:var1.shape[2],:])
+
+        dnamesvar = ['Time','bottom_top','south_north','west_east']
+        diagout = np.sqrt(unstgvar0*unstgvar0 + unstgvar1*unstgvar1)
+
+#        dnamesvar = ncobj.variables[depvars[0]].dimensions
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        ncvar.insert_variable(ncobj, 'ws', diagout, dnamesvar, dvnamesvar, newnc)
+
 # wss (u10, v10)
     elif diag == 'wss':
@@ -1554 +1579 @@
         dnamesvar = ncobj.variables[depvars[0]].dimensions
         dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
-
-        print 'dnamesvar',dnamesvar
-        print 'dnames',dnames
-        print 'dvnames',dvnames
-        print 'dvnamesvar',dvnamesvar
 
         ncvar.insert_variable(ncobj, 'wss', diagout, dnamesvar, dvnamesvar, newnc)