Changeset 1827 in lmdz_wrf

Timestamp:

Mar 20, 2018, 4:07:36 PM (7 years ago)

Author:

lfita

Message:

Adding:

• `ẀRFw' into 'pinterp'

File:

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

trunk/tools/nc_var_tools.py

@@ -18075 +18075 @@
         'WRFv': for WRF y-wind de-staggered
         'WRFver': for WRF y-wind de-staggered Earth-rotated
+        'WRFw': for WRF z-wind de-staggered
     """
     import module_ForInt as fin
@@ -18115 +18116 @@
         MODdims = ['Time', 'bottom_top', 'south_north', 'west_east']
         newMODdims = ['Time', 'pres', 'south_north', 'west_east']
-        notCHK = ['WRFght', 'WRFrh', 'WRFt', 'WRFu', 'WRFuer', 'WRFv', 'WRFver']
+        notCHK = ['WRFght', 'WRFrh', 'WRFt', 'WRFu', 'WRFuer', 'WRFv', 'WRFver',     \
+          'WRFw']
         MODvarrequired = ['P', 'PB', 'PSFC', 'PH', 'PHB', 'HGT', 'T', 'QVAPOR',      \
           'XLONG', 'XLAT', 'Times']
@@ -18548 +18550 @@
                 isgeop = False
                 varattrs = gen.variables_values('va')
+                CFvn = varattrs[0]
+                newvarattr['standard_name'] = varattrs[1]
+                newvarattr['long_name'] = varattrs[4].replace('|',' ')
+                newvarattr['units'] = varattrs[5]
+            elif vn == 'WRFw':
+                ovarin = onc.variables['W']
+                print infmsg
+                print '  ' + fname + ': De-staggering z-wind variable !!'
+                print '    from:', ovarin.shape, 'to', (dimt, dimz, dimy, dimx)
+                if isFR64:
+                    varin = np.zeros((dimt, dimz, dimy, dimx), dtype=np.float64)
+                else:
+                    varin = np.zeros((dimt, dimz, dimy, dimx), dtype=np.float)
+                # Not pro, but less memory problems!
+                for it in range(dimt):
+                    varin[it,:,:,:] = 0.5*(ovarin[it,0:dimz,:,:] + ovarin[it,1:dimz+1,:,:])
+                isgeop = False
+                varattrs = gen.variables_values('wa')
                 CFvn = varattrs[0]
                 newvarattr['standard_name'] = varattrs[1]