Changeset 1123 in lmdz_wrf

Timestamp:

Sep 30, 2016, 4:47:55 PM (9 years ago)

Author:

lfita

Message:

Adding [orientation]' for the colorbar which becomes a new variable [colorbarvals]' in draw_2D_shad

• Rotating tick-labels in colorbar if this is orientated 'horixzontally'

Location:

trunk/tools

Files:

• drawing.py (modified) (5 diffs)
• drawing_tools.py (modified) (3 diffs)

trunk/tools/drawing.py

@@ -56 +56 @@
     draw_2D_shad(ncfile, values, varn)
       ncfile= file to use
-      values=[vnamefs]:[dimvals]:[dimxvn]:[dimyvn]:[colorbar]:[fmtcolorbar]:[sminv],[smaxv]:[figt]:
+      values=[vnamefs]:[dimvals]:[dimxvn]:[dimyvn]:[colorbarvals]:[sminv],[smaxv]:[figt]:
        [kindfig]:[reverse]:[mapv]:[close]
         [vnamefs]: Name in the figure of the variable to be shaded
@@ -62 +62 @@
           variable a given value is required (-1, all the length)
         [dimx/yvn]: name of the variables with the values of the final dimensions (x,y)
-        [colbarn]: name of the color bar
-        [fmtcolorbar]: format of the numbers in the color bar 'C'-like ('auto' for %6g)
+        [colorbarvals]=[colbarn],[fmtcolorbar],[orientation]
+          [colorbarn]: name of the color bar
+          [fmtcolorbar]: format of the numbers in the color bar 'C'-like ('auto' for %6g)
+          [orientation]: orientation of the colorbar ('vertical' (default, by 'auto'), 'horizontal')
         [smin/axv]: minimum and maximum value for the shading or:
           'Srange': for full range
@@ -101 +103 @@
         quit()
 
-    expectargs = '[vnamefs]:[dimvals]:[dimxvn]:[dimyvn]:[colbarn]:[fmtcolorbar]:' +  \
+    expectargs = '[vnamefs]:[dimvals]:[dimxvn]:[dimyvn]:[colbarvals]:' +             \
       '[sminv],[smaxv]:[figt]:[kindfig]:[reverse]:[mapv]:[close]'
  
@@ -110 +112 @@
     vdimxn = values.split(':')[2]
     vdimyn = values.split(':')[3]
-    colbarn = values.split(':')[4]
-    fmtcolbar0 = values.split(':')[5]
-    shadminmax = values.split(':')[6]
-    figtitle = values.split(':')[7].replace('|',' ')
-    figkind = values.split(':')[8]
-    revals = values.split(':')[9]
-    mapvalue = values.split(':')[10]
-    close = gen.Str_Bool(values.split(':')[11])
+    colorbarvals = values.split(':')[4]
+    shadminmax = values.split(':')[5]
+    figtitle = values.split(':')[6].replace('|',' ')
+    figkind = values.split(':')[7]
+    revals = values.split(':')[8]
+    mapvalue = values.split(':')[9]
+    close = gen.Str_Bool(values.split(':')[10])
 
     ncfiles = ncfile
@@ -224 +225 @@
     if mapvalue == 'None': mapvalue = None
 
+    colbarn = colorbarvals.split(',')[0]
+    fmtcolbar0 = colorbarvals.split(',')[1]
+    colbaror0 = colorbarvals.split(',')[2]
+
     if fmtcolbar0 == 'auto':
-        fmtcolorbar = '%6g'
-    else:
-        fmtcolorbar = fmtcolbar0
+        fmtcolbar = '%6g'
+    else:
+        fmtcolbar = fmtcolbar0
+    if colbaror0 == 'auto':
+        colbaror = 'vertical'
+    else:
+        colbaror = colbaror0
+
 
     drw.plot_2D_shadow(valshad, vnamesfig, odimxv, odimyv, odimxu, odimyu, dimnamesv,\
-      colbarn, fmtcolorbar, shading_nx, varunits, figtitle, figkind, revals,         \
+      colbarn, fmtcolbar, colbaror, shading_nx, varunits, figtitle, figkind, revals, \
       mapvalue, close)
 

trunk/tools/drawing_tools.py

@@ -3777 +3777 @@
 
 def plot_2D_shadow(varsv,vnames,dimxv,dimyv,dimxu,dimyu,dimn,          \
-  colorbar,cbarfmt,vs,uts,vtit,kfig,reva,mapv,ifclose):
+  colorbar,cbarfmt,cbaror,vs,uts,vtit,kfig,reva,mapv,ifclose):
     """ Adding labels and other staff to the graph
       varsv= 2D values to plot with shading
@@ -3786 +3786 @@
       colorbar= name of the color bar to use
       cbarfmt= format of the numbers in the colorbar
+      cbaror= orientation of the colorbar
       vs= minmum and maximum values to plot in shadow or:
         'Srange': for full range
@@ -4001 +4002 @@
 
     plt.pcolormesh(x, y, varsv, cmap=plt.get_cmap(colorbar), vmin=vsend[0], vmax=vsend[1])
-    cbar = plt.colorbar(format=cbarfmt)
+    if cbaror == 'horizontal':
+        cbar = plt.colorbar(format=cbarfmt,orientation=cbaror)
+        # From: http://stackoverflow.com/questions/32050030/rotation-of-colorbar-tick-labels-in-matplotlib
+        ticklabels= cbar.ax.get_xticklabels()
+        Nticks = len(ticklabels)
+        ticklabs = []
+        for itick in range(Nticks): ticklabs.append(ticklabels[itick].get_text())
+        cbar.ax.set_xticklabels(ticklabs,rotation=90)
+    else:
+        cbar = plt.colorbar(format=cbarfmt,orientation=cbaror)
+    
 
     if not mapv is None: