Changeset 2714 in lmdz_wrf for trunk

Timestamp:

Sep 27, 2019, 5:27:16 PM (5 years ago)

Author:

lfita

Message:

Adding:

• `fixedvalues_cbar': Function to provide a color bar atr fixed values and colors

File:

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

trunk/tools/drawing_tools.py

@@ -170 +170 @@
 # figureMap_size: Function to determine the real size of a figure adjusted to the 
 #   ratio of the maps
+# fixedvalues_cbar: Function to provide a color bar atr fixed values and colors
 # format_axes: Function to provide the format for the ticks of the axes in a figure
 # graphic_range: Function to provide the ranges of a figure following different options
@@ -14665 +14666 @@
     return
 
+def fixedvalues_cbar(fixvalbnds, RGBfixvalcolors):
+    """ Function to provide a color bar atr fixed values and colors
+      fixvalues: list values at a given color
+      RGBfixvalcolors: list of colors
+    >>> vals=[-140., 0., 100., 200., 300., 400., 500., 600., 700., 800., 900., 1000., \
+      1500., 2000., 3000., 4000., 7000.]
+    >>> colors = [[119,152,93], [149,181,112], [189,209,137], [233,238,154],         \
+      [248,242,168], [251,231,163], [252,225,144], [248,213,133], [251,189,141],     \
+      [231,175,139], [202,149,132], [206,156,149], [228,182,192], [250,214,231],     \
+      [224,224,224], [255,255,255], [255,255,255]]
+    >>> fixedvalues_cbar(vals, colors)
+    """
+    fname = 'fixedvalues_cbar'
+
+    if len(fixvalbnds) != len(RGBfixvalcolors):
+        Nbnds = len(fixvalbnds)
+        Ncls = len(RGBfixvalcolors) 
+        print errormsg
+        print '  ' + fname + ': number of values:', Nbnds, 'and number of colors',   \
+          Ncls, 'differ !!'
+        print '      passed values & colors _______'
+        if Nbnds > Ncls:
+            for i in range(Ncls):
+                print '      ', i, fixvalbnds[i], RGBfixvalcolors[i]
+            for i in range(Ncls,Nbnds):
+                print '      ', i, fixvalbnds[i]
+        else:
+            for i in range(Nbnds):
+                print '      ', i, fixvalbnds[i], RGBfixvalcolors[i]
+            for i in range(Nbnds,Ncls):
+                print '      ', i, '--', RGBfixvalcolors[i]
+        quit(-1)
+
+    # Srting values
+    bndsS = []
+    for b in fixvalbnds: bndsS.append(str(b))
+
+    HEXfixvalcolors = []
+    Nfixvalcolors = len(RGBfixvalcolors)
+    for ic in range(Nfixvalcolors):
+        rgbv = np.array(RGBfixvalcolors[ic], dtype=np.float)/255.
+        rgbv = np.where(rgbv > 0.1, rgbv - 0.1, 0.)
+        if np.all(rgbv == 0.9): rgbv = rgbv/0.9
+
+        hexcol = gen.coldec_hex(rgbv)
+        HEXfixvalcolors.append(hexcol)
+
+    fixvalcmap = mpl.colors.ListedColormap(HEXfixvalcolors)
+    fixvalnormc = mpl.colors.BoundaryNorm(fixvalbnds, fixvalcmap.N)
+
+    return bndsS, fixvalcmap, fixvalnormc
+