Changeset 1819 in lmdz_wrf

Timestamp:

Mar 19, 2018, 8:50:34 PM (7 years ago)

Author:

lfita

Message:

Adding WRFuer', WRFver' on 'pinterp':

• 'WRFuer': for WRF x-wind de-staggered Earth-rotated
• 'WRFver': for WRF y-wind de-staggered Earth-rotated

File:

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

trunk/tools/nc_var_tools.py ¶

@@ -18071 +18071 @@
         'WRFt': for WRF temperature
         'WRFu': for WRF x-wind de-staggered
+        'WRFuer': for WRF x-wind de-staggered Earth-rotated
         'WRFv': for WRF y-wind de-staggered
+        'WRFver': for WRF y-wind de-staggered Earth-rotated
     """
     import module_ForInt as fin
@@ -18471 +18473 @@
                 newvarattr['long_name'] = varattrs[4].replace('|',' ')
                 newvarattr['units'] = varattrs[5]
+            elif vn == 'WRFuer':
+                ovarin = onc.variables['U']
+                ovarin2 = onc.variables['V']
+                # Earth-rotating
+                osina =  onc.variables['SINALPHA']
+                ocosa =  onc.variables['COSALPHA']
+                print infmsg
+                print '  ' + fname + ': De-staggering x-wind variable & Earth rotating!!'
+                print '    from:', ovarin.shape, 'to', (dimt, dimz, dimy, dimx)
+                if isFR64:
+                    varin0 = np.zeros((dimt, dimz, dimy, dimx), dtype=np.float64)
+                else:
+                    varin0 = np.zeros((dimt, dimz, dimy, dimx), dtype=np.float)
+                # Not pro, but less memory problems!
+                for it in range(dimt):
+                    varin0[it,:,::] = 0.5*(ovarin[it,:,:,0:dimx] + ovarin[it,:,:,1:dimx+1])
+                    varin02[it,:,::] = 0.5*(ovarin2[it,:,0:dimy,:] + ovarin2[it,:,1:dimy+1,:])
+                for iz in range(dimz):
+                    varin[:,iz,:,:] = varin0[:,iz,:,:]*ocosa[:] - varin02[:,iz,:,:]*osina[:]
+
+                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']
@@ -18483 +18511 @@
                 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 == 'WRFver':
+                ovarin = onc.variables['U']
+                ovarin2 = onc.variables['V']
+                # Earth-rotating
+                osina =  onc.variables['SINALPHA']
+                ocosa =  onc.variables['COSALPHA']
+                print infmsg
+                print '  ' + fname + ': De-staggering y-wind variable & Earth rotating!!'
+                print '    from:', ovarin.shape, 'to', (dimt, dimz, dimy, dimx)
+                if isFR64:
+                    varin0 = np.zeros((dimt, dimz, dimy, dimx), dtype=np.float64)
+                else:
+                    varin0 = np.zeros((dimt, dimz, dimy, dimx), dtype=np.float)
+                # Not pro, but less memory problems!
+                for it in range(dimt):
+                    varin0[it,:,::] = 0.5*(ovarin[it,:,:,0:dimx] + ovarin[it,:,:,1:dimx+1])
+                    varin02[it,:,::] = 0.5*(ovarin2[it,:,0:dimy,:] + ovarin2[it,:,1:dimy+1,:])
+                for iz in range(dimz):
+                    varin[:,iz,:,:] = varin0[:,iz,:,:]*osina[:] + varin02[:,iz,:,:]*ocosa[:]
+
+                isgeop = False
+                varattrs = gen.variables_values('ua')
                 CFvn = varattrs[0]
                 newvarattr['standard_name'] = varattrs[1]