Changeset 2140 in lmdz_wrf for trunk

Timestamp:

Sep 14, 2018, 5:57:13 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `timemax': When a given variable [varname] got its maximum

Location:

trunk/tools

Files:

• diag_tools.py (modified) (2 diffs)
• diagnostics.py (modified) (1 diff)
• variables_values.dat (modified) (1 diff)

trunk/tools/diag_tools.py

@@ -65 +65 @@
 # var_td_Uhus: Function to compute dew-point air temperature from temperature and 
 #   pressure values using hus
+# var_timemax: This function returns the time at which variable reaches its maximum in time
+#   units
 # var_timeoverthres: This function returns the time at which (varv(t) > thres) in time units
 # var_virtualTemp: This function returns virtual temperature in K, 
@@ -696 +698 @@
 
     return convini, vardims, varvdims
+
+def var_timemax(varv, time, dimns, dimvns):
+    """ This function returns the time at which variable reaches its maximum in time
+        units
+      varv: values of the variable to use
+      time: time in CF coordinates
+    """
+    fname = 'var_timemax'
+
+    dt = varv.shape[0]
+    dy = varv.shape[1]
+    dx = varv.shape[2]
+
+    vardims = dimns[:]
+    varvdims = dimvns[:]
+
+    vardims.pop(0)
+    varvdims.pop(0)
+
+    timemax = np.ones((dy, dx), dtype=np.float)*gen.fillValueF
+    varmax = np.max(varv, axis=0)
+    for j in range(dy):
+        for i in range(dx):
+            it = gen.index_vec(varv[:,j,i], varmax[j,i])
+            timemax[j,i] = time[it]
+
+    return timemax, vardims, varvdims
 
 def var_timeoverthres(varv, time, thres, dimns, dimvns):

trunk/tools/diagnostics.py

@@ -799 +799 @@
         ncvar.insert_variable(ncobj, 'pr', diagout/dtime, diagoutd, diagoutvd, newnc)
 
+# timemax ([varname], time). When a given variable [varname] got its maximum
+    elif diagn == 'timemax':
+            
+        var0 = ncobj.variables[depvars[0]][:]
+        var1 = ncobj.variables[depvars[1]][:]
+
+        otime = ncobj.variables[depvars[1]]
+
+        dnamesvar = ncobj.variables[depvars[0]].dimensions
+        dvnamesvar = ncvar.var_dim_dimv(dnamesvar,dnames,dvnames)
+
+        diagout, diagoutd, diagoutvd  = diag.var_timemax(var0, var1, dnames,         \
+          dvnames)
+
+        ncvar.insert_variable(ncobj, 'timemax', diagout, diagoutd, diagoutvd, newnc, \
+          fill=gen.fillValueF)
+        # Getting the right units
+        ovar = newnc.variables['timemax']
+        if gen.searchInlist(otime.ncattrs(), 'units'): 
+            tunits = otime.getncattr('units')
+            ncvar.set_attribute(ovar, 'units', tunits)
+            newnc.sync()
+        ncvar.set_attribute(ovar, 'variable', depvars[0])
+
 # timeoverthres ([varname], time, [value], [CFvarn]). When a given variable [varname]    
 #   overpass a given [value]. Being [CFvarn] the name of the diagnostics in 

trunk/tools/variables_values.dat

@@ -501 +501 @@
 th105, th105, thickness_1000_500, 0., 2500., atmosphere|thickness|1000|hPa|to|500|hPa, m, Reds, $m$, m
 1000 hPa to 500 hPa thickness, th105, thickness_1000_500, 0., 2500., atmosphere|thickness|1000|hPa|to|500|hPa, m, Reds, $m$, m
+timemax, timemax, time_maximum, 0., 310000000., time|of|maximum, time, rainbow, $timemax$, timemax
 tke, tke, turbulent_kinetic_energy, 0., 0.003, turbulent|kinetic|energy, m2/s2, Reds, $tke$, tke
 TKE, tke, turbulent_kinetic_energy, 0., 0.003, turbulent|kinetic|energy, m2/s2, Reds, $tke$, tke