Changeset 1539 in lmdz_wrf

Timestamp:

Apr 25, 2017, 9:27:35 PM (8 years ago)

Author:

lfita

Message:

Adding:

`plot_bar': Plotting a bar char plot with multiple sources
`mmd-1' in 'units_lunits'
`Stringlist' in 'format_axes' for list of strings

File:

• trunk/tools/drawing_tools.py (modified) (6 diffs)

trunk/tools/drawing_tools.py

@@ -84 +84 @@
 # plot_2lines: Function to plot two lines in different axes (x/x2 or y/y2)
 # plot_2lines_time: Function to plot two time-lines in different axes (x/x2 or y/y2)
+# plot_bars: Function to plot a bar plot with multiple columns
 # plot_lines: Function to plot a collection of lines
 # plot_Taylor: Function to draw a Taylor diagram (Taylor 2001)
@@ -1896 +1897 @@
         elif u == 'mmh-3': lu='$mmh^{-3}$'
         elif u == 'mm/d': lu='$mmd^{-1}$'
+        elif u == 'mmd-1': lu='$mmd^{-1}$'
         elif u == 'ms-1': lu='$ms^{-1}$'
         elif u == 'm s-1': lu='$ms^{-1}$'
@@ -2787 +2789 @@
         xs: style of x-axis ('auto' for 'pretty')
           'Nfix', values computed at even 'Ndx'
+          'Stringlist', a given list of strings
           'Vfix', values computed at even 'Ndx' increments
           'pretty', values computed following aprox. 'Ndx' at 'pretty' intervals (2.,2.5,4,5,10)
         xf: format of the ticks labels at the x-axis ('auto' for '5g')
-        Nx: number of ticks labels at the x-axis ('auto' for 5)
+        Nx: number of ticks labels at the x-axis ('auto' for 5, '@' list of values for 'Stringlist' '!' for spaces)
         orx: angle of orientation of ticks at the x-axis ('auto' for horizontal)
         ys: style of y-axis ('auto' for 'pretty')
@@ -2822 +2825 @@
         else:
           if Sv[0] == 'pretty' or Sv[0] == 'Nfix': Nx = int(Sv[2])
+          elif Sv[0] == 'Stringlist':
+            Nx0 = Sv[2].replace('!', ' ').split('@')
+            Nx = []
+            for Nxn in Nx0: Nx.append(gen.latex_text(Nxn))
           else: Nx = np.float(Sv[2])
         if Sv[3] == 'auto': orx = 0.
@@ -2832 +2839 @@
         else: 
           if Sv[4] == 'pretty' or Sv[4] == 'Nfix': Ny = int(Sv[6])
+          elif Sv[4] == 'Stringlist':
+            Ny0 = Sv[6].replace('!', ' ').split('@')
+            Ny = []
+            for Nyn in Ny0: Ny.append(gen.latex_text(Nyn))
           else: Ny = np.float(Sv[6])
         if Sv[7] == 'auto': ory = 0.
@@ -9121 +9132 @@
     
     output_kind(kfig, figname, ifclose)
-
+   
     return
 
+def plot_bars(xvalues, yvalues, xaxv, yaxv, dimxu, dimyu, valaxis, xlabel, ylabel,   \
+  labels, cols, width, gtit, lloc, lsiz, kfig, figname, ifclose):
+    """ Function to plot a bar plot with multiple columns
+      xvalues= list with the values along the x-ais
+      yvalues= list with the values along the y-ais
+      [x/y]axv= values at the axes of x and y
+      dim[x/y]u= units at the axes of x and y
+      valaxis= which axis holds the values 'x' or 'y'
+      xlabel= label at the x-axis
+      ylabel= label at the y-axis
+      labels= list of labels for legend (None for no legend)
+      cols= list of colors for the bars (single value for the same, 'auto' for automatic)
+      width= size of the column
+      gtit= title of the figure
+      lloc= location of the legend (0, autmoatic)
+        1: 'upper right', 2: 'upper left', 3: 'lower left', 4: 'lower right',
+        5: 'right', 6: 'center left', 7: 'center right', 8: 'lower center',
+        9: 'upper center', 10: 'center'
+      lsiz= font size of the legend
+      kfig= kind of output: pdf, png, ps
+      figname= name of the figure
+      ifclose= whether figure should be closed
+    """
+    fname = 'plot_bars'
+
+    Nvalues = len(xvalues)
+
+    # Maximum number of values:
+    Lmaxvals = -10
+    xticks = xvalues[0]
+    for iv in range(Nvalues):
+        if len(xvalues[iv]) > Lmaxvals: Lmaxvals = len(xvalues[iv])
+        NOTcoinc = list(set(xvalues[iv]).difference(set(xticks)))
+        if len(NOTcoinc) != 0:
+            xticks = xticks + NOTcoinc
+            xticks.sort
+
+    dxn = np.min(xvalues)
+    dxx = np.max(xvalues)
+    dyn = np.min(yvalues)
+    dyx = np.max(yvalues)
+
+    axstyles = ['pretty', 'Nfix', 'Stringlist', 'Vfix']
+    if xaxv[0] == 'pretty':
+        dimxt = np.array(gen.pretty_int(dxn,dxx,xaxv[2]))
+        dimxl = np.array(dimxt, dtype=str)
+    elif xaxv[0] == 'Nfix':
+        dimxl = np.arange(0.,1.,1./(xaxv[2]+1))
+        dimxt = np.arange(0.,1.,1./varsv.shape[1])
+    elif xaxv[0] == 'Stringlist':
+        if len(xaxv[2]) != len(xticks):
+            print warnmsg
+            print '  ' + fname + ': number of ticks to plot=', len(xticks),         \
+              'and number of labels provided', len(xaxv[2]), 'differ !!'
+            print '  ' + fname + ': ticks to plot=', xticks, 'and provided labels', \
+                xaxv[2], 'differ !!'
+
+        dimxl = xaxv[2]
+        dimxt = xticks
+    elif xaxv[0] == 'Vfix':
+        dxnvfix = int(dxn/xaxv[2])
+        dimxt = np.arange(dxnvfix*xaxv[2],dxx,xaxv[2])
+    else:
+        print errormsg
+        print '  ' + fname + ": x-xais style '" + xaxv[0] + "' not ready !!"
+        print '    available ones: ', axstyles
+        quit(-1)
+
+    if yaxv[0] == 'pretty':
+        dimyt = np.array(gen.pretty_int(dyn,dyx,yaxv[2]))
+        dimyl = np.array(dimyt, dtype=str)
+    elif yaxv[0] == 'Nfix':
+        dimyt = np.arange(0.,1.,1./(yaxv[2]+1))
+        dimyl = np.arange(0.,1.,1./varsv.shape[0])
+        axytype='fixpixel'
+    elif yaxv[0] == 'Vfix':
+        dynvfix = int(dyn/yaxv[2])
+        dimyt = np.arange(dynvfix*xaxv[2],dyx,yaxv[2])
+        axytype='data'
+    else:
+        print errormsg
+        print '  ' + fname + ": x-xais style '" + xaxv[0] + "' not ready !!"
+        print '    available ones: ', axstyles
+        quit(-1)
+
+    if valaxis == 'x':
+        orient = 'horizontal'
+    else:
+        orient = 'vertical'
+
+    colvs = []
+    if len(cols) == 1:
+        if cols[0] == 'auto':
+            colvs = colorsauto[0:Nvalues+1]
+        else:
+            for ic in range(Nvalues): colvs.append(cols[0])
+    else:
+        colvs = cols
+    
+    for iv in range(Nvalues):
+        xvls = xvalues[iv]
+        Lvalues = len(xvls)
+        dxmean = np.mean(np.array(xvls[1:Lvalues])-np.array(xvls[0:Lvalues-1]))
+        if dxmean < width*Nvalues:
+            print warnmsg
+            print '  ' + fname + ': mean distance among x-values:', dxmean,         \
+              'is smaller than the total number of bars with the given witdh (',    \
+              width, '):', width*Nvalues
+            print '    reduce widht of the column, should not be larger than:',     \
+              dxmean/Nvalues
+        if width*Nvalues < 0.01*dxmean:
+            print warnmsg
+            print '  ' + fname + ': mean distance among x-values:', dxmean,         \
+              'is 100-times larger than the total number of bars with the given ' + \
+              'witdh (', width, '):', width*Nvalues
+            print '    increase widht of the column, should not be smaller than:',  \
+              0.01*dxmean
+
+        # Relative position of the column respect the xtick
+        if np.mod(Nvalues,2) == 0:
+            xrel = width*(iv - Nvalues/2.)
+        else:
+            xrel = width*(iv - (Nvalues-1)/2.)
+
+        xrels = xrel*np.ones((Lvalues),dtype=np.float)
+
+        if valaxis == 'x':
+            if labels is not None:
+                plt.barh(xvls+xrels, yvalues[iv], height=width, color=colvs[iv],     \
+                  label=labels[iv])
+            else:
+                plt.barh(xvls+xrels, yvalues[iv], height=width, color=colvs[iv])
+        else:
+            if labels is not None:
+                plt.bar(xvls+xrels, yvalues[iv], width=width, color=colvs[iv],       \
+                  label=labels[iv])
+            else:
+                plt.bar(xvls+xrels, yvalues[iv], width=width, color=colvs[iv])
+ 
+    if valaxis == 'x':
+        plt.xticks(dimyt, dimyl, rotation=yaxv[3])
+        plt.yticks(dimxt, dimxl, rotation=xaxv[3])
+        plt.xlabel(ylabel + ' (' + units_lunits(dimyu) + ')')
+        if dimxu is not None:
+            plt.ylabel(xlabel + ' (' + units_lunits(dimxu) + ')') 
+        else:
+            plt.ylabel(xlabel) 
+    else:
+        plt.xticks(dimxt, list(dimxl), rotation=xaxv[3])
+        plt.yticks(dimyt, list(dimyl), rotation=yaxv[3])
+        if dimxu is not None:
+            plt.xlabel(xlabel + ' (' + units_lunits(dimxu) + ')') 
+        else:
+            plt.xlabel(xlabel) 
+        plt.ylabel(ylabel + ' (' + units_lunits(dimyu) + ')')
+
+    if labels is not None:
+        plt.legend(loc=lloc, prop={'size':lsiz})
+
+    plt.title(gen.latex_text(gtit))
+    
+    output_kind(kfig, figname, ifclose)
+
+    return
+