Changeset 2639 in lmdz_wrf

Timestamp:

Jun 25, 2019, 3:21:53 PM (5 years ago)

Author:

lfita

Message:

Adding:

• `modify_color': Function to modify a given color

File:

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

trunk/tools/generic_tools.py

@@ -174 +174 @@
 # mindist: Function to provide the minimum distance toa pair of lon,lat
 # minmax_range: Function to provide a range for a given pair of minimum and maximum
+# modify_color: Function to modify a given color
 # multi_index_mat: Function to provide the multiple coordinates of a given value inside a matrix
 # multi_index_string: Function to provide the indeces of every repetition of a group of characters within a string
@@ -16846 +16847 @@
     return plon, plat
 
+def modify_color(color, coltype, modval, action):
+    """ Function to modify a given color
+      color: color to modify
+      coltype: type of color
+        'hex': color must be ASCII '#[RR][GG][BB]' in heximal mode
+        'dec': color must be list [[R], [G], [B]] with [0,1] values
+        '256': color must be list [[R], [G], [B]] with [0,255] values
+      modval: value to use to modify in [0,255] values
+      action: action of the modification
+        'light': increase all values by modval
+        'darken': decrease (above zero) all values by modval
+    >>> modify_color('#AA0000', 'hex', 32, 'light')
+    CA2020
+    >>> modify_color([0.2,0.8,0.6], 'dec', 32, 'darken')
+    [ 0.0745098  0.6745098  0.4745098]
+    >>> modify_color([255, 255, 0], '256', 32, 'light')
+    [ 255.  255.   32.]
+    """
+    fname = 'modify_color'
+    availtype = ['hex', 'dec', '256']
+    availaction = ['light', 'darken']
+
+    if coltype == 'hex':
+        deccol = np.array(colhex_dec(color))
+    elif coltype == 'dec':
+        deccol = np.array(color)
+    elif coltype == '256':
+        deccol = np.array(color)/255.
+    else:
+        print errormsg
+        print '  ' + fname + ": color type '" + coltype + "' not ready !!"
+        print '     available ones:', availtype
+        quit(-1)
+
+    modv = modval/255.
+    if action == 'darken':
+        newcol = np.where(deccol < modv, 0., deccol - modv)
+    elif action == 'light':
+        newcol = np.where(deccol > 1.-modv, 1., deccol + modv)
+    else:
+        print errormsg
+        print '  ' + fname + ": action '" + action + "' not ready !!"
+        print '     available ones:', availaction
+        quit(-1)
+
+    if coltype == 'hex':
+        finalcol = np.array(coldec_hex(newcol))
+    elif coltype == 'dec':
+        finalcol = newcol
+    elif coltype == '256':
+        finalcol = newcol*255.
+
+    return finalcol
+
+print modify_color('#AA0000', 'hex', 32, 'light')
+
+print modify_color([0.2,0.8,0.6], 'dec', 32, 'darken')
+
+print modify_color([255, 255, 0], '256', 32, 'light')
+
 #quit()
 
 
-