Changeset 642 in lmdz_wrf for trunk/tools

Timestamp:

Sep 19, 2015, 12:47:01 PM (10 years ago)

Author:

lfita

Message:

Adding OMEGA to w transformation

File:

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

trunk/tools/diagnostics.py

@@ -532 +532 @@
     return mslpv, mslpdims, mslpvdims
 
+def compute_OMEGAw(omega, p, t, dimns, dimvns):
+    """ Function to transform OMEGA [Pas-1] to velocities [ms-1]
+      tacking: https://www.ncl.ucar.edu/Document/Functions/Contributed/omega_to_w.shtml
+      [omega] = vertical velocity [in ms-1] (assuming [t],z,y,x)
+      [p] = pressure in [Pa] (assuming [t],z,y,x)
+      [t] = temperature in [K] (assuming [t],z,y,x)
+      [dimns]= list of the name of the dimensions of [q]
+      [dimvns]= list of the name of the variables with the values of the 
+        dimensions of [q]
+    """
+    fname = 'compute_OMEGAw'
+
+    rgas = 287.058            # J/(kg-K) => m2/(s2 K)
+    g    = 9.80665            # m/s2
+
+    wdims = dimns[:]
+    wvdims = dimvns[:]
+
+    rho  = p/(rgas*t)         # density => kg/m3
+    w    = -omega/(rho*g)     
+
+    return w, wdims, wvdims
+
 def compute_prw(dens, q, dimns, dimvns):
     """ Function to compute water vapour path (prw)
@@ -1032 +1055 @@
 
         ncvar.insert_variable(ncobj, 'psl', diagout, diagoutd, diagoutvd, newnc)
+
+# OMEGAw (omega, p, t) from NCL formulation (https://www.ncl.ucar.edu/Document/Functions/Contributed/omega_to_w.shtml)
+    elif diag == 'OMEGAw':
+            
+        var0 = ncobj.variables[depvars[0]][:]
+        var1 = ncobj.variables[depvars[1]][:]
+        var1 = ncobj.variables[depvars[2]][:]
+
+        dnamesvar = ncobj.variables[depvars[0]].dimensions
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        diagout, diagoutd, diagoutvd = compute_OMEGAw(var0,var1,var2,dnamesvar,dvnamesvar)
+
+        ncvar.insert_variable(ncobj, 'wa', diagout, diagoutd, diagoutvd, newnc)
 
 # raintot: instantaneous total precipitation from WRF as (RAINC + RAINC) / dTime