Changeset 914 in lmdz_wrf

Timestamp:

Jun 20, 2016, 8:01:02 PM (8 years ago)

Author:

lfita

Message:

Adding and working WRFua' and WRFva' rotated 3D winds!!

Location:

trunk/tools

Files:

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

trunk/tools/diagnostics.inf

@@ -26 +26 @@
 tds, TStd, psfc@t@q
 tds, WRFtds, PSFC@T2@Q2
+ua, WRFua, U@V@SINALPHA@COSALPHA
+va, WRFva, U@V@SINALPHA@COSALPHA
 wds, TSwds, u@v
 wds, WRFwds, U10@V10

trunk/tools/diagnostics.py

@@ -878 +878 @@
   'OMEGAw', 'RAINTOT',   \
   'rvors', 'td', 'turbulence', 'WRFgeop', 'WRFp', 'WRFrvors', 'wds', 'wss',          \
-  'WRFheight']
+  'WRFheight', 'WRFua', 'WRFva']
 
 methods = ['accum', 'deaccum']
@@ -887 +887 @@
   'WRFp', 'WRFtd',                                                                   \
   'WRFpos', 'WRFprc', 'WRFprls', 'WRFrh', 'LMDZrh', 'LMDZrhs', 'WRFrhs', 'WRFrvors', \
-  'WRFt', 'WRFtime', 'WRFwds', 'WRFwss', 'WRFheight']
+  'WRFt', 'WRFtime', 'WRFua', 'WRFva', 'WRFwds', 'WRFwss', 'WRFheight']
 
 ofile = 'diagnostics.nc'
@@ -1489 +1489 @@
         ncvar.insert_variable(ncobj, 'ta', WRFt, dnames, dvnames, newnc)
 
-# WRFtime (u10, v10, WRFpos)
+# WRFua (U, V, SINALPHA, COSALPHA) to be rotated !!
+    elif diag == 'WRFua':
+        var0 = ncobj.variables[depvars[0]][:]
+        var1 = ncobj.variables[depvars[1]][:]
+        var2 = ncobj.variables[depvars[2]][:]
+        var3 = ncobj.variables[depvars[3]][:]
+
+        # un-staggering variables
+        unstgdims = [var0.shape[0], var0.shape[1], var0.shape[2], var0.shape[3]-1]
+        ua = 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],:])
+
+        for iz in range(var0.shape[1]):
+            ua[:,iz,:,:] = unstgvar0[:,iz,:,:]*var3 - unstgvar1[:,iz,:,:]*var2
+
+        dnamesvar = list(ncobj.variables[depvars[0]].dimensions)
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        ncvar.insert_variable(ncobj, 'ua', ua, dnames, dvnames, newnc)
+
+# WRFua (U, V, SINALPHA, COSALPHA) to be rotated !!
+    elif diag == 'WRFva':
+        var0 = ncobj.variables[depvars[0]][:]
+        var1 = ncobj.variables[depvars[1]][:]
+        var2 = ncobj.variables[depvars[2]][:]
+        var3 = ncobj.variables[depvars[3]][:]
+
+        # 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],:])
+        for iz in range(var0.shape[1]):
+            va[:,iz,:,:] = unstgvar0[:,iz,:,:]*var2 + unstgvar1[:,iz,:,:]*var3
+
+        dnamesvar = list(ncobj.variables[depvars[0]].dimensions)
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        ncvar.insert_variable(ncobj, 'va', va, dnames, dvnames, newnc)
+
+# WRFtime
     elif diag == 'WRFtime':