Changeset 2767 in lmdz_wrf for trunk/tools

Timestamp:

Nov 18, 2019, 2:48:10 AM (5 years ago)

Author:

lfita

Message:

Adding:

• `gradient2Dh': 1st order Horizontal 2D-gradient of any variable

Location:

trunk/tools

Files:

• diagnostics.inf (modified) (1 diff)
• diagnostics.py (modified) (3 diffs)

trunk/tools/diagnostics.inf

@@ -19 +19 @@
 front, front_R04, T2@U10@V10@WRFdx@WRFdy@params:1.:0.0001
 frontogenesis, frontogenesis, WRFt@UNua:U@UNva:V@UNwa:W@WRFp@WRFdx@WRFdy@WRFdz@WRFtime
+gradient2dh, gradient2Dh, W@WRFdx@WRFdy
 hur, LMDZrh, pres@t@r
 hur, WRFrh, WRFrh@T@P@PB

trunk/tools/diagnostics.py

@@ -738 +738 @@
             varsadd.append(nonvd)
 
-        CFvarn = ncvar.variables_values(depvars[0])[0]
+        CFvarn = gen.variables_values(depvars[0])[0]
 
 # Removing the flux
@@ -1186 +1186 @@
 
         var0 = ncobj.variables[depvars[0]]
-        var1 = ncobj.variables[depvars[1]][0,:,:]
-        var2 = ncobj.variables[depvars[2]][0,:,:]
+        if depvars[1] == 'WRFdx':
+            var1 = WRFdx
+        else:
+            var1 = ncobj.variables[depvars[1]][0,:,:]
+        if depvars[2] == 'WRFdy':
+            var2 = WRFdy
+        else:
+            var2 = ncobj.variables[depvars[2]][0,:,:]
 
         dnamesvar = var0.dimensions
@@ -1201 +1207 @@
             varsadd.append(nonvd)
 
-        CFvarn = ncvar.variables_values(depvars[0])[0]
+        CFvarn = gen.variables_values(depvars[0])[0]
         ncvar.insert_variable(ncobj, depvars[0], diagout, diagoutd, diagoutvd,       \
           newnc, gen.fillValueF)
         newnc.sync()
-        ovar = newnc.variables[depvars[0]]
+        CFvar = gen.variables_values(depvars[0])
+        newnc.renameVariable(CFvar[0], CFvar[0]+'grad2dh')
+        newnc.sync()
+        ovar = newnc.variables[CFvar[0]+'grad2dh'
         varu = ovar.units
-        newnc.renameVariable(depvars[0], depvars[0]+'2Dhgrad')
         ovar.setncattr('units', varu+'ds-1')
+        newnc.sync()
 
 # LMDZrh (pres, t, r)