Changeset 2222 in lmdz_wrf

Timestamp:

Nov 8, 2018, 10:59:59 PM (6 years ago)

Author:

lfita

Message:

Adding calculation of `range_faces' for lonlat regular

File:

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

trunk/tools/diagnostics.py

@@ -101 +101 @@
 
 # Variables not to check
-NONcheckingvars = ['accum', 'cllmh', 'deaccum', 'face', 'TSrhs',                     \
+NONcheckingvars = ['accum', 'cllmh', 'deaccum', 'face', 'LONLATdxdy', 'TSrhs',       \
   'TStd', 'TSwds', 'TSwss',                                                          \
   'WRFbils',                                                                         \
@@ -167 +167 @@
 WRFz_compute = False
 WRFdxdy_compute = False
+LONLATdxdy_compute = False
 
 # File creation
@@ -192 +193 @@
     if dnv == 'WRFz':WRFz_compute = True
     if dnv == 'WRFdxdy':WRFdxdy_compute = True
+    if dnv == 'LONLATdxdy':LONLATdxdy_compute = True
 
 # diagnostics to compute
@@ -221 +223 @@
         if gen.searchInlist(depvars, 'WRFz'): WRFz_compute = True
         if gen.searchInlist(depvars, 'WRFdxdy'): WRFdxdy_compute = True
+        if gen.searchInlist(depvars, 'LONLATdxdy'): LONLATdxdy_compute = True
 
 # Dictionary with the new computed variables to be able to add them
@@ -422 +425 @@
     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 LONLATdxdy_compute:
+    print '  ' + main + ': Retrieving dxdy: real distance between grid points ' +    \
+      'from a regular lonlat projection as dx=(lon[i+1]-lon[i])*raddeg*Rearth*' +    \
+      'cos(abs(lat[i])); dy=(lat[j+1]-lat[i])*raddeg*Rearth; ds=sqrt(dx**2+dy**2); '+\
+      'raddeg = pi/180; Rearth=6370.0e03'
+    dimv = ncobj.variables['lon'].shape
+    lon = ncobj.variables['lon'][:]
+    lat = ncobj.variables['lat'][:]
+
+    WRFlon, WRFlat = gen.lonlat2D(lon,lat)
+
+    dimx = WRFlon.shape[1]
+    dimy = WRFlon.shape[0]
+
+    WRFdx = np.zeros((dimy,dimx), dtype=np.float)
+    WRFdy = np.zeros((dimy,dimx), dtype=np.float)
+
+    raddeg = np.pi/180.
+
+    Rearth = fdef.module_definitions.earthradii
+
+    WRFdx[:,0:dimx-1]=(WRFlon[:,1:dimx]-WRFlon[:,0:dimx-1])*raddeg*Rearth*           \
+      np.cos(np.abs(WRFlat[:,0:dimx-1]*raddeg))
+    WRFdy[0:dimy-1,:]=(WRFlat[1:dimy,:]-WRFlat[0:dimy-1,:])*raddeg*Rearth
     WRFds = np.sqrt(WRFdx**2 + WRFdy**2)
 
@@ -728 +768 @@
         hvalleyrange = np.float(depvars[4].split(':')[4])
 
-        dnamesvar = list(ncobj.variables[depvars[0]].dimensions)
+        dnamesvar = list(ncobj.variables[depvars[2]].dimensions)
         dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
         lon, lat = gen.lonlat2D(var0, var1)
@@ -760 +800 @@
 
         ncvar.insert_variable(ncobj, 'orogmax', rngorogmax, diagoutd, diagoutvd,     \
-          newnc)
+          newnc, fill=gen.fillValueF)
         newnc.renameVariable('orogmax', 'rangeorogmax')
         ovar = newnc.variables['rangeorogmax']