Changeset 1382 in lmdz_wrf for trunk/tools/diagnostics.py

Timestamp:

Dec 14, 2016, 10:17:49 AM (8 years ago)

Author:

lfita

Message:

Adding removal of non-checking variables from WRFt' and WRFgeop'
Fixing `WRFgeop' adding also de-staggering of the variable

File:

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

trunk/tools/diagnostics.py

@@ -1473 +1473 @@
 # WRFgeop geopotential from WRF as PH + PHB
     elif diag == 'WRFgeop':
-            
-        diagout = WRFgeop
+        var0 = ncobj.variables[depvars[0]][:]
+        var1 = ncobj.variables[depvars[1]][:]
+
+        # de-staggering geopotential
+        diagout0 = var0 + var1
+        dt = diagout0.shape[0]
+        dz = diagout0.shape[1]
+        dy = diagout0.shape[2]
+        dx = diagout0.shape[3]
+
+        diagout = np.zeros((dt,dz-1,dy,dx), dtype=np.float)
+        diagout = 0.5*(diagout0[:,1:dz,:,:]+diagout0[:,0:dz-1,:,:])
+
+        # Removing the nonChecking variable-dimensions from the initial list
+        varsadd = []
+        for nonvd in NONchkvardims:
+            if gen.searchInlist(dvnames,nonvd): dvnames.remove(nonvd)
+            varsadd.append(nonvd)
 
         ncvar.insert_variable(ncobj, 'zg', diagout, dnames, dvnames, newnc)
@@ -1553 +1569 @@
         dnamesvar = list(ncobj.variables[depvars[0]].dimensions)
         dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        # Removing the nonChecking variable-dimensions from the initial list
+        varsadd = []
+        for nonvd in NONchkvardims:
+            if gen.searchInlist(dvnames,nonvd): dvnames.remove(nonvd)
+            varsadd.append(nonvd)
 
         ncvar.insert_variable(ncobj, 'ta', WRFt, dnames, dvnames, newnc)