Changeset 1980 in lmdz_wrf

Timestamp:

Jul 25, 2018, 8:50:16 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `WRFwd': WRF-derived wind direction

Location:

trunk/tools

Files:

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

trunk/tools/diag_tools.py

@@ -17 +17 @@
 # compute_turbulence: Function to compute the rubulence term of the Taylor's decomposition ...'
 # C_diagnostic: Class to compute generic variables
+# compute_wd: Function to compute the wind direction 3D
 # compute_wds: Function to compute the wind direction
 # compute_wss: Function to compute the wind speed
@@ -1453 +1454 @@
 
     return turb, turbdims, turbvdims
+
+def compute_wd(u, v, dimns, dimvns):
+    """ Function to compute the wind direction
+      [u]= W-E wind direction [ms-1, knot, ...]
+      [v]= N-S wind direction [ms-1, knot, ...]
+      [dimns]= list of the name of the dimensions of [u]
+      [dimvns]= list of the name of the variables with the values of the 
+        dimensions of [u]
+    """
+    fname = 'compute_wds'
+
+#    print '    ' + fname + ': computing wind direction as ATAN2(v,u) ...'
+    theta = np.arctan2(v,u)
+    theta = np.where(theta < 0., theta + 2.*np.pi, theta)
+
+    var = 360.*theta/(2.*np.pi)
+
+    vardims = dimns[:]
+    varvdims = dimvns[:]
+
+    return var, vardims, varvdims
 
 def compute_wds(u, v, dimns, dimvns):

trunk/tools/diagnostics.inf

@@ -47 +47 @@
 wa, OMEGAw, vitw@pres@temp
 wds, TSwds, u@v
+wd, WRFwd, U@V@SINALPHA@COSALPHA
 wds, wds, U10@V10 # WRF
 ws, ws, U@V

trunk/tools/diagnostics.py

@@ -1312 +1312 @@
         ncvar.insert_variable(ncobj, 'va', va, dnames, diagoutvd, newnc)
 
+
+# WRFwd (U, V, SINALPHA, COSALPHA) to be rotated !!
+    elif diagn == 'WRFwd':
+        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)
+        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]):
+            ua[:,iz,:,:] = unstgvar0[:,iz,:,:]*var3 - unstgvar1[:,iz,:,:]*var2
+            va[:,iz,:,:] = unstgvar0[:,iz,:,:]*var2 + unstgvar1[:,iz,:,:]*var3
+
+        dnamesvar = ['Time','bottom_top','south_north','west_east']
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        wd, dnames, dvnames = diag.compute_wd(ua, va, 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, 'wd', wd, dnames, diagoutvd, newnc)
+
 # WRFtime
     elif diagn == 'WRFtime':