modules/ts/misc/color.py

   1 #!/usr/bin/env python
   2
   3 import math, os, sys
   4
   5 webcolors = {
   6 "indianred": "#cd5c5c",
   7 "lightcoral": "#f08080",
   8 "salmon": "#fa8072",
   9 "darksalmon": "#e9967a",
  10 "lightsalmon": "#ffa07a",
  11 "red": "#ff0000",
  12 "crimson": "#dc143c",
  13 "firebrick": "#b22222",
  14 "darkred": "#8b0000",
  15 "pink": "#ffc0cb",
  16 "lightpink": "#ffb6c1",
  17 "hotpink": "#ff69b4",
  18 "deeppink": "#ff1493",
  19 "mediumvioletred": "#c71585",
  20 "palevioletred": "#db7093",
  21 "lightsalmon": "#ffa07a",
  22 "coral": "#ff7f50",
  23 "tomato": "#ff6347",
  24 "orangered": "#ff4500",
  25 "darkorange": "#ff8c00",
  26 "orange": "#ffa500",
  27 "gold": "#ffd700",
  28 "yellow": "#ffff00",
  29 "lightyellow": "#ffffe0",
  30 "lemonchiffon": "#fffacd",
  31 "lightgoldenrodyellow": "#fafad2",
  32 "papayawhip": "#ffefd5",
  33 "moccasin": "#ffe4b5",
  34 "peachpuff": "#ffdab9",
  35 "palegoldenrod": "#eee8aa",
  36 "khaki": "#f0e68c",
  37 "darkkhaki": "#bdb76b",
  38 "lavender": "#e6e6fa",
  39 "thistle": "#d8bfd8",
  40 "plum": "#dda0dd",
  41 "violet": "#ee82ee",
  42 "orchid": "#da70d6",
  43 "fuchsia": "#ff00ff",
  44 "magenta": "#ff00ff",
  45 "mediumorchid": "#ba55d3",
  46 "mediumpurple": "#9370db",
  47 "blueviolet": "#8a2be2",
  48 "darkviolet": "#9400d3",
  49 "darkorchid": "#9932cc",
  50 "darkmagenta": "#8b008b",
  51 "purple": "#800080",
  52 "indigo": "#4b0082",
  53 "darkslateblue": "#483d8b",
  54 "slateblue": "#6a5acd",
  55 "mediumslateblue": "#7b68ee",
  56 "greenyellow": "#adff2f",
  57 "chartreuse": "#7fff00",
  58 "lawngreen": "#7cfc00",
  59 "lime": "#00ff00",
  60 "limegreen": "#32cd32",
  61 "palegreen": "#98fb98",
  62 "lightgreen": "#90ee90",
  63 "mediumspringgreen": "#00fa9a",
  64 "springgreen": "#00ff7f",
  65 "mediumseagreen": "#3cb371",
  66 "seagreen": "#2e8b57",
  67 "forestgreen": "#228b22",
  68 "green": "#008000",
  69 "darkgreen": "#006400",
  70 "yellowgreen": "#9acd32",
  71 "olivedrab": "#6b8e23",
  72 "olive": "#808000",
  73 "darkolivegreen": "#556b2f",
  74 "mediumaquamarine": "#66cdaa",
  75 "darkseagreen": "#8fbc8f",
  76 "lightseagreen": "#20b2aa",
  77 "darkcyan": "#008b8b",
  78 "teal": "#008080",
  79 "aqua": "#00ffff",
  80 "cyan": "#00ffff",
  81 "lightcyan": "#e0ffff",
  82 "paleturquoise": "#afeeee",
  83 "aquamarine": "#7fffd4",
  84 "turquoise": "#40e0d0",
  85 "mediumturquoise": "#48d1cc",
  86 "darkturquoise": "#00ced1",
  87 "cadetblue": "#5f9ea0",
  88 "steelblue": "#4682b4",
  89 "lightsteelblue": "#b0c4de",
  90 "powderblue": "#b0e0e6",
  91 "lightblue": "#add8e6",
  92 "skyblue": "#87ceeb",
  93 "lightskyblue": "#87cefa",
  94 "deepskyblue": "#00bfff",
  95 "dodgerblue": "#1e90ff",
  96 "cornflowerblue": "#6495ed",
  97 "royalblue": "#4169e1",
  98 "blue": "#0000ff",
  99 "mediumblue": "#0000cd",
 100 "darkblue": "#00008b",
 101 "navy": "#000080",
 102 "midnightblue": "#191970",
 103 "cornsilk": "#fff8dc",
 104 "blanchedalmond": "#ffebcd",
 105 "bisque": "#ffe4c4",
 106 "navajowhite": "#ffdead",
 107 "wheat": "#f5deb3",
 108 "burlywood": "#deb887",
 109 "tan": "#d2b48c",
 110 "rosybrown": "#bc8f8f",
 111 "sandybrown": "#f4a460",
 112 "goldenrod": "#daa520",
 113 "darkgoldenrod": "#b8860b",
 114 "peru": "#cd853f",
 115 "chocolate": "#d2691e",
 116 "saddlebrown": "#8b4513",
 117 "sienna": "#a0522d",
 118 "brown": "#a52a2a",
 119 "maroon": "#800000",
 120 "white": "#ffffff",
 121 "snow": "#fffafa",
 122 "honeydew": "#f0fff0",
 123 "mintcream": "#f5fffa",
 124 "azure": "#f0ffff",
 125 "aliceblue": "#f0f8ff",
 126 "ghostwhite": "#f8f8ff",
 127 "whitesmoke": "#f5f5f5",
 128 "seashell": "#fff5ee",
 129 "beige": "#f5f5dc",
 130 "oldlace": "#fdf5e6",
 131 "floralwhite": "#fffaf0",
 132 "ivory": "#fffff0",
 133 "antiquewhite": "#faebd7",
 134 "linen": "#faf0e6",
 135 "lavenderblush": "#fff0f5",
 136 "mistyrose": "#ffe4e1",
 137 "gainsboro": "#dcdcdc",
 138 "lightgrey": "#d3d3d3",
 139 "silver": "#c0c0c0",
 140 "darkgray": "#a9a9a9",
 141 "gray": "#808080",
 142 "dimgray": "#696969",
 143 "lightslategray": "#778899",
 144 "slategray": "#708090",
 145 "darkslategray": "#2f4f4f",
 146 "black": "#000000",
 147 }
 148
 149 if os.name == "nt":
 150     consoleColors = [
 151     "#000000",  #{   0,   0,   0 },//0 - black
 152     "#000080",  #{   0,   0, 128 },//1 - navy
 153     "#008000",  #{   0, 128,   0 },//2 - green
 154     "#008080",  #{   0, 128, 128 },//3 - teal
 155     "#800000",  #{ 128,   0,   0 },//4 - maroon
 156     "#800080",  #{ 128,   0, 128 },//5 - purple
 157     "#808000",  #{ 128, 128,   0 },//6 - olive
 158     "#C0C0C0",  #{ 192, 192, 192 },//7 - silver
 159     "#808080",  #{ 128, 128, 128 },//8 - gray
 160     "#0000FF",  #{   0,   0, 255 },//9 - blue
 161     "#00FF00",  #{   0, 255,   0 },//a - lime
 162     "#00FFFF",  #{   0, 255, 255 },//b - cyan
 163     "#FF0000",  #{ 255,   0,   0 },//c - red
 164     "#FF00FF",  #{ 255,   0, 255 },//d - magenta
 165     "#FFFF00",  #{ 255, 255,   0 },//e - yellow
 166     "#FFFFFF",  #{ 255, 255, 255 } //f - white
 167     ]
 168 else:
 169     consoleColors = [
 170     "#2e3436",
 171     "#cc0000",
 172     "#4e9a06",
 173     "#c4a000",
 174     "#3465a4",
 175     "#75507b",
 176     "#06989a",
 177     "#d3d7cf",
 178     "#ffffff",
 179
 180     "#555753",
 181     "#ef2929",
 182     "#8ae234",
 183     "#fce94f",
 184     "#729fcf",
 185     "#ad7fa8",
 186     "#34e2e2",
 187     "#eeeeec",
 188     ]
 189
 190 def RGB2LAB(r,g,b):
 191     if max(r,g,b):
 192         r /= 255.
 193         g /= 255.
 194         b /= 255.
 195
 196     X = (0.412453 * r + 0.357580 * g + 0.180423 * b) / 0.950456
 197     Y = (0.212671 * r + 0.715160 * g + 0.072169 * b)
 198     Z = (0.019334 * r + 0.119193 * g + 0.950227 * b) / 1.088754
 199
 200     #[X * 0.950456]   [0.412453 0.357580 0.180423]   [R]
 201     #[Y           ] = [0.212671 0.715160 0.072169] * [G]
 202     #[Z * 1.088754]   [0.019334 0.119193 0.950227]   [B]
 203
 204     T = 0.008856 #threshold
 205
 206     if X > T:
 207         fX = math.pow(X, 1./3.)
 208     else:
 209         fX = 7.787 * X + 16./116.
 210
 211     # Compute L
 212     if Y > T:
 213         Y3 = math.pow(Y, 1./3.)
 214         fY = Y3
 215         L  = 116. * Y3 - 16.0
 216     else:
 217         fY = 7.787 * Y + 16./116.
 218         L  = 903.3 * Y
 219
 220     if Z > T:
 221         fZ = math.pow(Z, 1./3.)
 222     else:
 223         fZ = 7.787 * Z + 16./116.
 224
 225     # Compute a and b
 226     a = 500. * (fX - fY)
 227     b = 200. * (fY - fZ)
 228
 229     return (L,a,b)
 230
 231 def colorDistance(r1,g1,b1 = None, r2 = None, g2 = None,b2 = None):
 232     if type(r1) == tuple and type(g1) == tuple and b1 is None and r2 is None and g2 is None and b2 is None:
 233         (l1,a1,b1) = RGB2LAB(*r1)
 234         (l2,a2,b2) = RGB2LAB(*g1)
 235     else:
 236         (l1,a1,b1) = RGB2LAB(r1,g1,b1)
 237         (l2,a2,b2) = RGB2LAB(r2,g2,b2)
 238     #CIE94
 239     dl = l1-l2
 240     C1 = math.sqrt(a1*a1 + b1*b1)
 241     C2 = math.sqrt(a2*a2 + b2*b2)
 242     dC = C1 - C2
 243     da = a1-a2
 244     db = b1-b2
 245     dH = math.sqrt(max(0, da*da + db*db - dC*dC))
 246     Kl = 1
 247     K1 = 0.045
 248     K2 = 0.015
 249
 250     s1 = dl/Kl
 251     s2 = dC/(1. + K1 * C1)
 252     s3 = dH/(1. + K2 * C1)
 253     return math.sqrt(s1*s1 + s2*s2 + s3*s3)
 254
 255 def parseHexColor(col):
 256     if len(col) != 4 and len(col) != 7 and not col.startswith("#"):
 257         return (0,0,0)
 258     if len(col) == 4:
 259         r = col[1]*2
 260         g = col[2]*2
 261         b = col[3]*2
 262     else:
 263         r = col[1:3]
 264         g = col[3:5]
 265         b = col[5:7]
 266     return (int(r,16), int(g,16), int(b,16))
 267
 268 def getColor(col):
 269     if isinstance(col, str):
 270         if col.lower() in webcolors:
 271             return parseHexColor(webcolors[col.lower()])
 272         else:
 273             return parseHexColor(col)
 274     else:
 275         return col
 276
 277 def getNearestConsoleColor(col):
 278     color = getColor(col)
 279     minidx = 0
 280     mindist = colorDistance(color, getColor(consoleColors[0]))
 281     for i in range(len(consoleColors)):
 282         dist = colorDistance(color, getColor(consoleColors[i]))
 283         if dist < mindist:
 284             mindist = dist
 285             minidx = i
 286     return minidx
 287
 288 if os.name == 'nt':
 289     import msvcrt
 290     from ctypes import windll, Structure, c_short, c_ushort, byref
 291     SHORT = c_short
 292     WORD = c_ushort
 293
 294     class COORD(Structure):
 295         _fields_ = [
 296             ("X", SHORT),
 297             ("Y", SHORT)]
 298
 299     class SMALL_RECT(Structure):
 300         _fields_ = [
 301             ("Left", SHORT),
 302             ("Top", SHORT),
 303             ("Right", SHORT),
 304             ("Bottom", SHORT)]
 305
 306     class CONSOLE_SCREEN_BUFFER_INFO(Structure):
 307         _fields_ = [
 308             ("dwSize", COORD),
 309             ("dwCursorPosition", COORD),
 310             ("wAttributes", WORD),
 311             ("srWindow", SMALL_RECT),
 312             ("dwMaximumWindowSize", COORD)]
 313
 314     class winConsoleColorizer(object):
 315         def __init__(self, stream):
 316             self.handle = msvcrt.get_osfhandle(stream.fileno())
 317             self.default_attrs = 7#self.get_text_attr()
 318             self.stream = stream
 319
 320         def get_text_attr(self):
 321             csbi = CONSOLE_SCREEN_BUFFER_INFO()
 322             windll.kernel32.GetConsoleScreenBufferInfo(self.handle, byref(csbi))
 323             return csbi.wAttributes
 324
 325         def set_text_attr(self, color):
 326             windll.kernel32.SetConsoleTextAttribute(self.handle, color)
 327
 328         def write(self, *text, **attrs):
 329             if not text:
 330                 return
 331             color = attrs.get("color", None)
 332             if color:
 333                 col = getNearestConsoleColor(color)
 334                 self.stream.flush()
 335                 self.set_text_attr(col)
 336             self.stream.write(" ".join([str(t) for t in text]))
 337             if color:
 338                 self.stream.flush()
 339                 self.set_text_attr(self.default_attrs)
 340
 341 class dummyColorizer(object):
 342     def __init__(self, stream):
 343         self.stream = stream
 344
 345     def write(self, *text, **attrs):
 346         if text:
 347             self.stream.write(" ".join([str(t) for t in text]))
 348
 349 class asciiSeqColorizer(object):
 350     RESET_SEQ = "\033[0m"
 351     #BOLD_SEQ = "\033[1m"
 352     ITALIC_SEQ = "\033[3m"
 353     UNDERLINE_SEQ = "\033[4m"
 354     STRIKEOUT_SEQ = "\033[9m"
 355     COLOR_SEQ0 = "\033[00;%dm" #dark
 356     COLOR_SEQ1 = "\033[01;%dm" #bold and light
 357
 358     def __init__(self, stream):
 359         self.stream = stream
 360
 361     def get_seq(self, code):
 362         if code > 8:
 363             return self.__class__.COLOR_SEQ1 % (30 + code - 9)
 364         else:
 365             return self.__class__.COLOR_SEQ0 % (30 + code)
 366
 367     def write(self, *text, **attrs):
 368         if not text:
 369             return
 370         color = attrs.get("color", None)
 371         if color:
 372             col = getNearestConsoleColor(color)
 373             self.stream.write(self.get_seq(col))
 374         self.stream.write(" ".join([str(t) for t in text]))
 375         if color:
 376             self.stream.write(self.__class__.RESET_SEQ)
 377
 378
 379 def getColorizer(stream):
 380     if stream.isatty():
 381         if os.name == "nt":
 382             return winConsoleColorizer(stream)
 383         else:
 384             return asciiSeqColorizer(stream)
 385     else:
 386         return dummyColorizer(stream)