Changeset 1417 in lmdz_wrf for trunk/tools

Timestamp:

Jan 18, 2017, 11:52:29 PM (8 years ago)

Author:

lfita

Message:

Adding range' values option for draw_vector'

Location:

trunk/tools

Files:

• drawing.py (modified) (4 diffs)
• drawing_tools.py (modified) (4 diffs)

trunk/tools/drawing.py

@@ -3907 +3907 @@
           [color]: color of the vectors
             'singlecol'@[colorn]: all the vectors same color ('auto': for 'red')
-            'wind'@[colorbar]: color of the vectors according to wind speed sqrt(u^2+v^2) and given [colorbar]
+            'wind'@[colorbar]@[sminv],[smaxv]: color of the vectors according to wind speed sqrt(u^2+v^2) and given [colorbar]
               all vectors the same length
-            '3rdvar'@[colorbar]@[varn]@[units]: color of the vectors according to a 3rd variable (to be added at -v) and given [colorbar]
+              [smin/axv]: minimum and maximum value for the shading or:
+               'Srange': for full range
+               'Saroundmean@val': for mean-xtrm,mean+xtrm where xtrm = np.min(mean-min@val,max@val-mean)
+               'Saroundminmax@val': for min*val,max*val
+               'Saroundpercentile@val': for median-xtrm,median+xtrm where xtrm = np.min(median-percentile_(val),
+                 percentile_(100-val)-median)
+               'Smean@val': for -xtrm,xtrm where xtrm = np.min(mean-min*@val,max*@val-mean)
+               'Smedian@val': for -xtrm,xtrm where xtrm = np.min(median-min@val,max@val-median)
+               'Spercentile@val': for -xtrm,xtrm where xtrm = np.min(median-percentile_(val),
+                 percentile_(100-val)-median)
+ 
+            '3rdvar'@[colorbar]@[varn]@[units]@[sminv],[smaxv]: color of the vectors according to a 3rd variable (to be added at -v) and given [colorbar]
               all vectors the same length
           [length]: length of the wind vectors ('auto', for 9)
@@ -4041 +4052 @@
     if coln == 'singlecol':
         colorv = colv
+        shading_nx = None
     elif coln == 'wind':
         colorv = np.sqrt(uwvals**2 + vwvals**2)
+        # Min/Max colorbar
+        shadminmax = colorvals.split('@')[2]
+        shading_nx = shadminmax.split(',')        
     elif coln == '3rdvar':
         if len(varn.split(',')) != 3:
@@ -4053 +4068 @@
         stdvn, lonvn, unitsvn = drw.var_3desc(ocolvec)
         colorvals = colorvals + '@' + stdvn + '@' + unitsvn
+        # Min/Max colorbar
+        shadminmax = colorvals.split('@')[4]
+        shading_nx = shadminmax.split(',')        
     else:
         print errormsg
@@ -4065 +4083 @@
 
     drw.plot_vector(lonvals, latvals, uwvals, vwvals, freqv, colorvals, colorv,      \
-      lengthv, windname, windunits, mapvalues, gtit, kindfig, figuren)
+      lengthv, shading_nx, windname, windunits, mapvalues, gtit, kindfig, figuren)
 
     of.close()

trunk/tools/drawing_tools.py

@@ -6790 +6790 @@
     return
 
-def plot_vector(xvals,yvals,uvals,vvals,vecfreq,vecoln,veccolor,veclength,windn,wuts,\
-  mapv,graphtit,kfig,figname):
+def plot_vector(xvals, yvals, uvals, vvals, vecfreq, vecoln, veccolor, veclength, vs,\
+  windn, wuts, mapv, graphtit, kfig, figname):
     """ Function to plot vectors
       xvals= values for the x-axis
@@ -6807 +6807 @@
         'singlecol': 'auto', for 9
         'wind' and '3rdvar': 'auto' length as wind speed, otherwise fix length
+      vs= minmum and maximum values to plot in shadow or:
+        'Srange': for full range
+        'Saroundmean@val': for mean-xtrm,mean+xtrm where xtrm = np.min(mean-min@val,max@val-mean)
+        'Saroundminmax@val': for min*val,max*val
+        'Saroundpercentile@val': for median-xtrm,median+xtrm where xtrm = np.min(median-percentile_(val),
+          percentile_(100-val)-median)
+        'Smean@val': for -xtrm,xtrm where xtrm = np.min(mean-min*@val,max*@val-mean)
+        'Smedian@val': for -xtrm,xtrm where xtrm = np.min(median-min@val,max@val-median)
+        'Spercentile@val': for -xtrm,xtrm where xtrm = np.min(median-percentile_(val),
+           percentile_(100-val)-median)
       windn= name of the wind variable in the graph
       wuts= units of the wind variable in the graph
@@ -6855 +6865 @@
     elif VecN == 'wind' or VecN == '3rdvar':
         vcolor = vecoln.split('@')[1]
+        # Changing limits of the colors
+        vsend = graphic_range(vs,veccolor)
 
     plt.rc('text', usetex=True)
@@ -6913 +6925 @@
 
         plt.quiver(xvals[::yfreq,::xfreq], yvals[::yfreq,::xfreq],                   \
-          uvals[::yfreq,::xfreq], vvals[::yfreq,::xfreq], 
-          veccolor[::yfreq,::xfreq], cmap=plt.get_cmap(vcolor), pivot='middle')
+          uvals[::yfreq,::xfreq], vvals[::yfreq,::xfreq],                            \
+          veccolor[::yfreq,::xfreq], cmap=plt.get_cmap(vcolor), pivot='middle',      \
+          vmin=vsend[0], vmax=vsend[1])
         cbar = plt.colorbar()