Changeset 1583 in lmdz_wrf for trunk/tools

Timestamp:

Jun 9, 2017, 11:15:58 PM (8 years ago)

Author:

lfita

Message:

Adding WRFu', WRFv' to 'pinterp'

File:

• trunk/tools/nc_var_tools.py (modified) (3 diffs)

trunk/tools/nc_var_tools.py

@@ -17998 +17998 @@
         'WRFrh': for WRF relative humidity
         'WRFt': for WRF temperature
+        'WRFu': for WRF x-wind de-staggered
+        'WRFv': for WRF y-wind de-staggered
     """
     import module_ForInt as fin
@@ -18035 +18037 @@
         MODdims = ['Time', 'bottom_top', 'south_north', 'west_east']
         newMODdims = ['Time', 'pres', 'south_north', 'west_east']
-        notCHK = ['WRFght', 'WRFt', 'WRFrh']
+        notCHK = ['WRFght', 'WRFrh', 'WRFt', 'WRFu', 'WRFv']
         MODvarrequired = ['P', 'PB', 'PSFC', 'PH', 'PHB', 'HGT', 'T', 'QVAPOR',      \
           'XLONG', 'XLAT', 'Times']
@@ -18302 +18304 @@
                 isgeop = False
                 varattrs = gen.variables_values('WRFt')
+                CFvn = varattrs[0]
+                newvarattr['standard_name'] = varattrs[1]
+                newvarattr['long_name'] = varattrs[4].replace('|',' ')
+                newvarattr['units'] = varattrs[5]
+            elif vn == 'WRFu':
+                ovarin = onc.variables['U']
+                print infmsg
+                print '  ' + fname + ': De-staggering x-wind variable !!'
+                print '    from:', ovarin.shape, 'to', (dimt, dimz, dimy, dimx)
+                varin = np.zeros((dimt, dimz, dimy, dimx), dtype=np.float64)
+                varin = 0.5*(ovarin[:,:,:,0:dimx] + ovarin[:,:,:,1:dimx+1])
+                isgeop = False
+                varattrs = gen.variables_values('ua')
+                CFvn = varattrs[0]
+                newvarattr['standard_name'] = varattrs[1]
+                newvarattr['long_name'] = varattrs[4].replace('|',' ')
+                newvarattr['units'] = varattrs[5]
+            elif vn == 'WRFv':
+                ovarin = onc.variables['V']
+                print infmsg
+                print '  ' + fname + ': De-staggering y-wind variable !!'
+                print '    from:', ovarin.shape, 'to', (dimt, dimz, dimy, dimx)
+                varin = np.zeros((dimt, dimz, dimy, dimx), dtype=np.float64)
+                varin = 0.5*(ovarin[:,:,0:dimy,:] + ovarin[:,:,1:dimy+1,:])
+                isgeop = False
+                varattrs = gen.variables_values('va')
                 CFvn = varattrs[0]
                 newvarattr['standard_name'] = varattrs[1]