Changeset 1777 in lmdz_wrf for trunk/tools/diagnostics.py

Timestamp:

Feb 16, 2018, 3:17:02 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `WRFz': as height above surface unstaggering WRFgeop
• Fixing zwind computation

File:

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

trunk/tools/diagnostics.py

@@ -6 +6 @@
 #      ciclad: f2py --f90flags="-fPIC" --f90exec=/usr/bin/gfortran -L/opt/canopy-1.3.0/Canopy_64bit/System/lib/ -L/usr/lib64/ -L/opt/canopy-1.3.0/Canopy_64bit/System/lib/ -m module_ForDiagnostics -c module_generic.F90 module_ForDiagnosticsVars.F90 module_ForDiagnostics.F90 >& run_f2py.log
 
-## e.g. # diagnostics.py -d 'Time@time,bottom_top@ZNU,south_north@XLAT,west_east@XLONG' -v 'clt|CLDFRA,cllmh|CLDFRA@WRFp,RAINTOT|RAINC@RAINNC@XTIME' -f WRF_LMDZ/NPv31/wrfout_d01_1980-03-01_00:00:00
+## e.g. # diagnostics.py -d 'Time@WRFtime,bottom_top@ZNU,south_north@XLAT,west_east@XLONG' -v 'clt|CLDFRA,cllmh|CLDFRA@WRFp,RAINTOT|RAINC@RAINNC@XTIME' -f WRF_LMDZ/NPv31/wrfout_d01_1980-03-01_00:00:00
 ## e.g. # diagnostics.py -f /home/lluis/PY/diagnostics.inf -d variable_combo -v WRFprc
 
@@ -90 +90 @@
   'WRFp', 'WRFtd',                                                                   \
   'WRFpos', 'WRFprc', 'WRFprls', 'WRFrh', 'LMDZrh', 'LMDZrhs', 'WRFrhs', 'WRFrvors', \
-  'WRFt', 'WRFtime', 'WRFua', 'WRFva', 'WRFwds', 'WRFwss', 'WRFheight']
+  'WRFt', 'WRFtime', 'WRFua', 'WRFva', 'WRFwds', 'WRFwss', 'WRFheight', 'WRFz']
 
 NONchkvardims = ['WRFtime']
@@ -124 +124 @@
 WRFpos_compute = False
 WRFtime_compute = False
+WRFz_compute = False
 
 # File creation
@@ -147 +148 @@
     if dnv == 'WRFpos': WRFpos_compute = True
     if dnv == 'WRFtime': WRFtime_compute = True
+    if dnv == 'WRFz':WRFz_compute = True
 
 # diagnostics to compute
@@ -163 +165 @@
         if depvars == 'WRFpos': WRFpos_compute = True
         if depvars == 'WRFtime': WRFtime_compute = True
+        if depvars == 'WRFz': WRFz_compute = True
     else:
         depvars = diags[idiag].split('|')[1].split('@')
@@ -173 +176 @@
         if gen.searchInlist(depvars, 'WRFpos'): WRFpos_compute = True
         if gen.searchInlist(depvars, 'WRFtime'): WRFtime_compute = True
+        if gen.searchInlist(depvars, 'WRFz'): WRFz_compute = True
 
 # Dictionary with the new computed variables to be able to add them
@@ -325 +329 @@
     dictcompvars['WRFtime'] = {'name': 'time', 'standard_name': 'time',              \
       'long_name': 'time', 'units': tunits, 'calendar': 'gregorian'}
+
+if WRFz_compute:
+    print '  ' + main + ': Retrieving z: height above surface value from WRF as ' +  \
+      'unstagger(PH + PHB)/9.8-hgt'
+    dimv = ncobj.variables['PH'].shape
+    WRFzg = (ncobj.variables['PH'][:] + ncobj.variables['PHB'][:])/9.8
+
+    unzgd = (dimv[0], dimv[1]-1, dimv[2], dimv[3])
+    unzg = np.zeros(unzgd, dtype=np.float)
+    unzg = 0.5*(WRFzg[:,0:dimv[1]-1,:,:] + WRFzg[:,1:dimv[1],:,:])
+
+    WRFz = np.zeros(unzgd, dtype=np.float)
+    for iz in range(dimv[1]-1):
+        WRFz[:,iz,:,:] = unzg[:,iz,:,:] - ncobj.variables['HGT'][:]
+
+    # Attributes of the variable
+    Vvals = gen.variables_values('WRFz')
+    dictcompvars['WRFz'] = {'name': Vvals[0], 'standard_name': Vvals[1],          \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
 
 ### ## #
@@ -1123 +1146 @@
         ncvar.insert_variable(ncobj, 'zmla', diagout, diagoutd, diagoutvd, newnc)
 
-# WRFzwind wind extrapolation at a given hieght computation from WRF U, V, WRFgeop, 
-#   U10, V10, HGT, SINALPHA, COSALPHA, z=[zval]
+# WRFzwind wind extrapolation at a given hieght computation from WRF U, V, WRFz, 
+#   U10, V10, SINALPHA, COSALPHA, z=[zval]
     elif diagn == 'WRFzwind':
         var0 = ncobj.variables[depvars[0]][:]
         var1 = ncobj.variables[depvars[1]][:]
-        dimz = var0.shape[1]
-        var2 = WRFgeop[:,1:dimz+1,:,:]/9.8
+        var2 = WRFz
         var3 = ncobj.variables[depvars[3]][:]
         var4 = ncobj.variables[depvars[4]][:]
         var5 = ncobj.variables[depvars[5]][0,:,:]
         var6 = ncobj.variables[depvars[6]][0,:,:]
-        var7 = ncobj.variables[depvars[7]][0,:,:]
-        var8 = np.float(depvars[8].split('=')[1])
+        var7 = np.float(depvars[7].split('=')[1])
 
         # un-staggering 3D winds
@@ -1146 +1167 @@
 
         diagout1, diagout2, diagoutd, diagoutvd = diag.Forcompute_zwind(unvar0,      \
-          unvar1, var2, var3, var4, var5, var6, var7, var8, dnames, dvnames)
+          unvar1, var2, var3, var4, var5, var6, var7, dnames, dvnames)
 
         # Removing the nonChecking variable-dimensions from the initial list