Changeset 2746 in lmdz_wrf for trunk

Timestamp:

Nov 7, 2019, 6:09:59 PM (6 years ago)

Author:

lfita

Message:

Adding:

• `add_legend_topofix': Function to add the legend on a topofix plot

File:

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

trunk/tools/drawing_tools.py

@@ -163 +163 @@
 
 ####### Functions
+# add_legend_topofix: Function to add the legend on a topofix plot
 # check_colorBar: Check if the given colorbar exists in matplotlib
 # colorbar_vals: Function to provide the colorbar values for a figure
@@ -14843 +14844 @@
 
     return kfigS
+
+def add_legend_topofix(fig, legcolv=[0.8,'vertical',10.,10.,0.,0.8,0.85,'#6464AA']):
+    """ Function to add the legend on a topofix plot
+      fig= figure to which add the legend 
+      legcolv = [ijstart, orientation, labsize, sqsize, labrotation, totboxlength, 
+        ibox, seacolor] values for the legend of colors of the topography ([0.8, 'vertical', 
+        10., 10., 0., 0.8, 0.85, '#CCCCFF'], default)
+          ijstart: initial position as fraction of figure on the oposite axis of 
+            orientation
+          orientation: orientation of the legend
+          labsize: font size of the labels
+          sqsize: size of the squares of the legend
+          labrotation: rotation of the labels of the legend
+          totboxlength: total length of the color legend as percentage of the figure 
+             size along the given orientation
+          ibox: starting at the given orientation as percentage of the figure
+          seacolor: color for the sea
+    """
+    fname = 'add_legend_topofix'
+
+    topobnds = fixtopobnds
+
+    Ncolors = len(RGBfixtopocolors)
+
+    # FROM: https://matplotlib.org/examples/color/named_colors.html
+    # Get height and width of the figure
+    ncols = 1
+    nrows = len(topobnds)
+    X, Y = fig.get_dpi() * fig.get_size_inches()
+
+    # Dimensions of the color-legend
+    totboxlength = legcolv[5]
+    ibox = legcolv[6]
+
+    cbox = unichr(9632)#+unichr(9632)+unichr(9632)
+    bbox = unichr(9633)
+
+    ccs = np.arange(0.,totboxlength,totboxlength/(nrows+1))
+
+    iix = legcolv[0]
+    Ns = []
+    Cs = []
+    Xs = []
+    Ys = []
+    if legcolv[1] == 'vertical':
+        bw = totboxlength / (nrows + 1)
+        bh = bw * X / Y
+        Ns.append('> ' + str(topobnds[Ncolors-2]))
+        Cs.append(Ncolors-2)
+        Xs.append([iix,iix-bh])
+        Ys.append(ibox - ccs[0])
+        for i in range(1,Ncolors-2):
+            Ns.append(str(topobnds[Ncolors-2-i]))
+            Cs.append(Ncolors-2-i)
+            Xs.append([iix,iix-bh])
+            Ys.append(ibox - ccs[i])
+        Ns.append('< 0')
+        Cs.append(0)
+        Xs.append([iix,iix-bh])
+        Ys.append(ibox - ccs[Ncolors-2])
+
+    elif legcolv[1] == 'horizontal':
+        bw = totboxlength / (nrows + 1)
+        bh = bw * Y / X
+
+        Ns.append('< 0')
+        Cs.append(0)
+        Xs.append(ibox+ccs[0])
+        Ys.append([iix-bh,iix])
+        for i in range(1,Ncolors-2):
+            Ns.append(str(topobnds[i]))
+            Cs.append(i)
+            Xs.append(ibox + ccs[i])
+            Ys.append([iix-bh,iix])
+        Ns.append('> ' + str(topobnds[Ncolors-2]))
+        Cs.append(Ncolors-2)
+        Xs.append(ibox + ccs[Ncolors-2])
+        Ys.append([iix-bh,iix])
+    else:
+        print errormsg
+        print '  ' + fname + ": oritnation of legend of colors of topography " +     \
+          "scale '" + legcolv[1] + "' not ready !!"
+        print '    available ones:', availor
+        quit(-1)
+
+    for ib in range(0,Ncolors-1):
+        Nb = Ns[ib]
+        if legcolv[1] == 'vertical':
+            ix = Xs[ib][0]
+            iy = Ys[ib]
+        elif legcolv[1] == 'horizontal':
+            ix = Xs[ib]
+            iy = Ys[ib][0]
+        plt.annotate(Nb, xy=(ix,iy), xycoords='figure fraction', fontsize=legcolv[2],\
+          rotation=legcolv[4])
+        if legcolv[1] == 'vertical':
+            ix = Xs[ib][1]
+        elif legcolv[1] == 'horizontal':
+            iy = Ys[ib][1]
+        plt.annotate(cbox, xy=(ix,iy), xycoords='figure fraction',                   \
+          fontsize=legcolv[3], color=HEXfixtopocolors[Cs[ib]])
+        plt.annotate(bbox, xy=(ix,iy), xycoords='figure fraction',                   \
+          fontsize=legcolv[3], color='k')
+
+    return