Changeset 2257 in lmdz_wrf for trunk/tools

Timestamp:

Nov 30, 2018, 8:21:12 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `WRFdxdywps': to compute real dx,dy from WPS input (geogrid)

File:

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

trunk/tools/diagnostics.py

@@ -104 +104 @@
   'TStd', 'TSwds', 'TSwss',                                                          \
   'WRFbils',                                                                         \
-  'WRFclivi', 'WRFclwvi', 'WRFdens', 'WRFdx', 'WRFdxdy', 'WRFdy', 'WRFgeop',         \
-  'WRFp', 'WRFtd',                                                                   \
+  'WRFclivi', 'WRFclwvi', 'WRFdens', 'WRFdx', 'WRFdxdy', 'WRFdxdywps', 'WRFdy',      \
+  'WRFgeop', 'WRFp', 'WRFtd',                                                        \
   'WRFpos', 'WRFprc', 'WRFprls', 'WRFrh', 'LMDZrh', 'LMDZrhs',                       \
   'WRFrhs', 'WRFrvors',                                                              \
@@ -167 +167 @@
 WRFz_compute = False
 WRFdxdy_compute = False
+WRFdxdywps_compute = False
 LONLATdxdy_compute = False
 
@@ -193 +194 @@
     if dnv == 'WRFz':WRFz_compute = True
     if dnv == 'WRFdxdy':WRFdxdy_compute = True
+    if dnv == 'WRFdxdywps':WRFdxdywps_compute = True
     if dnv == 'LONLATdxdy':LONLATdxdy_compute = True
 
@@ -223 +225 @@
         if gen.searchInlist(depvars, 'WRFz'): WRFz_compute = True
         if gen.searchInlist(depvars, 'WRFdxdy'): WRFdxdy_compute = True
+        if gen.searchInlist(depvars, 'WRFdxdywps'): WRFdxdywps_compute = True
         if gen.searchInlist(depvars, 'LONLATdxdy'): LONLATdxdy_compute = True
 
@@ -413 +416 @@
     WRFlon = ncobj.variables['XLONG'][0,:,:]
     WRFlat = ncobj.variables['XLAT'][0,:,:]
+    WRFmapfac_m = ncobj.variables['MAPFAC_M'][0,:,:]
+    DX = ncobj.DX
+    DY = ncobj.DY
+
+    dimx = dimv[2]
+    dimy = dimv[1]
+
+    WRFdx = np.zeros((dimy,dimx), dtype=np.float)
+    WRFdy = np.zeros((dimy,dimx), dtype=np.float)
+
+    WRFdx[:,0:dimx-1]=(WRFlon[:,1:dimx]-WRFlon[:,0:dimx-1])*DX/WRFmapfac_m[:,0:dimx-1]
+    WRFdy[0:dimy-1,:]=(WRFlat[1:dimy,:]-WRFlat[0:dimy-1,:])*DY/WRFmapfac_m[0:dimy-1,:]
+    WRFds = np.sqrt(WRFdx**2 + WRFdy**2)
+
+    # Attributes of the variable
+    Vvals = gen.variables_values('WRFdx')
+    dictcompvars['WRFdx'] = {'name': Vvals[0], 'standard_name': Vvals[1],             \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
+    Vvals = gen.variables_values('WRFdy')
+    dictcompvars['WRFdy'] = {'name': Vvals[0], 'standard_name': Vvals[1],             \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
+    Vvals = gen.variables_values('WRFds')
+    dictcompvars['WRFds'] = {'name': Vvals[0], 'standard_name': Vvals[1],            \
+      'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
+
+if WRFdxdywps_compute:
+    print '  ' + main + ': Retrieving dxdy: real distance between grid points ' +    \
+      'from wpsWRF as dx=(XLONG_M(i+1)-XLONG_M(i))*DX/MAPFAC_M, ' +                  \
+      'dy=(XLAT_M(j+1)-XLAT_M(i))*DY/MAPFAC_M, ds=sqrt(dx**2+dy**2)'
+    dimv = ncobj.variables['XLONG_M'].shape
+    WRFlon = ncobj.variables['XLONG_M'][0,:,:]
+    WRFlat = ncobj.variables['XLAT_M'][0,:,:]
     WRFmapfac_m = ncobj.variables['MAPFAC_M'][0,:,:]
     DX = ncobj.DX