Changeset 2681 in lmdz_wrf for trunk

Timestamp:

Jul 14, 2019, 8:49:34 AM (5 years ago)

Author:

lfita

Message:

Adding:

• 'LONLATdx', 'LONLATdy'

File:

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

trunk/tools/diagnostics.py

@@ -103 +103 @@
 
 # Variables not to check
-NONcheckingvars = ['accum', 'cllmh', 'deaccum', 'face', 'LONLATdxdy',                \
-  'params', 'reglonlatbnds', 'TSrhs', 'TStd', 'TSwds', 'TSwss',                      \
+NONcheckingvars = ['accum', 'cllmh', 'deaccum', 'face', 'LONLATdxdy', 'LONLATdx',    \
+  'LONLATdy', 'params', 'reglonlatbnds', 'TSrhs', 'TStd', 'TSwds', 'TSwss',          \
   'UNua', 'UNva', 'UNwa',                                                            \
   'WRFbils',  'WRFbnds',                                                             \
@@ -170 +170 @@
 WRFz_compute = False
 WRFdx_compute = False
-WRFdx_compute = False
+WRFdy_compute = False
 WRFdz_compute = False
 WRFdxdy_compute = False
 WRFdxdywps_compute = False
 LONLATdxdy_compute = False
+LONLATdx_compute = False
+LONLATdy_compute = False
 UNua_compute = False
 UNva_compute = False
@@ -208 +210 @@
     if dnv == 'WRFdxdywps':WRFdxdywps_compute = True
     if dnv == 'LONLATdxdy':LONLATdxdy_compute = True
+    if dnv == 'LONLATdx':LONLATdx_compute = True
+    if dnv == 'LONLATdy':LONLATdy_compute = True
     if dnv[0:4] == 'UNua':
         UNua_compute = True
@@ -240 +244 @@
         if depvars == 'WRFdxdywps': WRFdxdywps_compute = True
         if depvars == 'LONLATdxdy': LONLATdxdy_compute = True
+        if depvars == 'LONLATdx': LONLATdx_compute = True
+        if depvars == 'LONLATdy': LONLATdy_compute = True
         if depvars[0:4] == 'UNua': 
             UNua_compute = True
@@ -266 +272 @@
         if gen.searchInlist(depvars, 'WRFdxdywps'): WRFdxdywps_compute = True
         if gen.searchInlist(depvars, 'LONLATdxdy'): LONLATdxdy_compute = True
+        if gen.searchInlist(depvars, 'LONLATdx'): LONLATdx_compute = True
+        if gen.searchInlist(depvars, 'LONLATdy'): LONLATdy_compute = True
         if gen.searchInlist_Strsec(depvars, 0, 3, 'UNua'): 
             UNua_compute = True
@@ -462 +470 @@
       'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
 
-if WRFdxdy_compute or WRFdx_compute or WRFdy_copute:
+if WRFdxdy_compute or WRFdx_compute or WRFdy_compute:
     print '  ' + main + ': Retrieving dxdy: real distance between grid points ' +    \
       'from WRF as dx=(XLONG(i+1)-XLONG(i))*DX/MAPFAC_M, dy=(XLAT(j+1)-XLAT(i))*DY/'+\
@@ -549 +557 @@
       'long_name': Vvals[4].replace('|',' '), 'units': Vvals[5]}
 
-if LONLATdxdy_compute:
+if LONLATdxdy_compute or LONLATdx_compute or LONLATdy_compute :
     print '  ' + main + ': Retrieving dxdy: real distance between grid points ' +    \
       'from a regular lonlat projection as dx=(lon[i+1]-lon[i])*raddeg*Rearth*' +    \
@@ -921 +929 @@
         var1 = ncobj.variables[depvars[1]]
         var2 = ncobj.variables[depvars[2]]
-        if depvars[3] == 'WRFdx': var3 = WRFdx
+        if depvars[3] == 'WRFdx' or depvars[3] == 'LONLATdx': var3 = WRFdx
         else: var3 = ncobj.variables[depvars[3]]
-        if depvars[4] == 'WRFdy': var4 = WRFdy
+        if depvars[4] == 'WRFdy' or depvars[4] == 'LONLATdy': var4 = WRFdy
         else: var4 = ncobj.variables[depvars[4]]
         par1 = np.float(depvars[5].split(':')[1])
@@ -997 +1005 @@
         else: var3 = ncobj.variables[depvars[3]][:]
         if depvars[4] == 'WRFp': var4 = WRFp
-        else: var4 = ncobj.variables[depvars[4]]
+        else: var4 = ncobj.variables[depvars[4]][:]
         if depvars[5] == 'WRFdx': var5 = WRFdx
-        else: var5 = ncobj.variables[depvars[5]]
+        else: var5 = ncobj.variables[depvars[5]][:]
         if depvars[6] == 'WRFdy': var6 = WRFdy
-        else: var6 = ncobj.variables[depvars[6]]
+        else: var6 = ncobj.variables[depvars[6]][:]
         if depvars[7] == 'WRFdz': var7 = WRFdz[0,0:dz,0,0]
-        else: var7 = ncobj.variables[depvars[7]]
+        else: var7 = ncobj.variables[depvars[7]][:]
         if depvars[8] == 'WRFtime': var08 = WRFtime
-        else: var08 = ncobj.variables[depvars[8]]
+        else: var08 = ncobj.variables[depvars[8]][:]
 
         # Assuming monotonic time-axis...
         var8 = var08[1] - var08[0]
 
-        var5 = var5/var5
-        var6 = var6/var6
-        var7 = var7/var7
+        if len(var4.shape) == 1:
+            print '  ' + diagn + ': rank 1 press !!'
+            print '  spreading over 4D !!'
+            var40 = var4 + 0.
+            var4 = np.zeros((dt,dz,dy,dx), dtype=np.float)
+            for it in range(dt):
+                for iy in range(dy):
+                    for ix in range(dx):
+                        var4[it,:,iy,ix] = var40
+
+        # Unifying...
+        #var5 = var5/var5
+        #var6 = var6/var6
+        #var7 = var7/var7
 
         diag1, diag2, diag3, diag4, diag5, diag6, diag7, diag8, diag9, diag10,       \