cpython-lru_cache-weakref: Tools/pynche/ColorDB.py annotate

annotate Tools/pynche/ColorDB.py @ 78069:a8aa77458e12 3.2

Add c_size_t and c_ssize_t to the ctypes summary table.

author	Antoine Pitrou <solipsis@pitrou.net>
date	Thu, 12 Jul 2012 20:31:50 +0200
parents	63741c7521aa
children

rev	line source
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	1 """Color Database.
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	2
9225 ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	3 This file contains one class, called ColorDB, and several utility functions.
ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	4 The class must be instantiated by the get_colordb() function in this file,
ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	5 passing it a filename to read a database out of.
ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	6
ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	7 The get_colordb() function will try to examine the file to figure out what the
ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	8 format of the file is. If it can't figure out the file format, or it has
ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	9 trouble reading the file, None is returned. You can pass get_colordb() an
ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	10 optional filetype argument.
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	11
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	12 Supporte file types are:
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	13
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	14 X_RGB_TXT -- X Consortium rgb.txt format files. Three columns of numbers
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	15 from 0 .. 255 separated by whitespace. Arbitrary trailing
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	16 columns used as the color name.
9225 ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	17
ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	18 The utility functions are useful for converting between the various expected
ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	19 color formats, and for calculating other color values.
ce8f3cef6aed Updated docstring. Barry Warsaw <barry@python.org> parents: 9224 diff changeset	20
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	21 """
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	22
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	23 import sys
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	24 import re
7236 fd7a5993d06d Simplification of conversion routines Barry Warsaw <barry@python.org> parents: 7182 diff changeset	25 from types import *
7346 5a9256b8c338 Cache conversions from triplets to rrggbb's and vice versa. Wasteful Barry Warsaw <barry@python.org> parents: 7293 diff changeset	26 import operator
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	27
7182 59362ea7e0f8 BadColor: new exception class Barry Warsaw <barry@python.org> parents: 7181 diff changeset	28 class BadColor(Exception):
59362ea7e0f8 BadColor: new exception class Barry Warsaw <barry@python.org> parents: 7181 diff changeset	29 pass
59362ea7e0f8 BadColor: new exception class Barry Warsaw <barry@python.org> parents: 7181 diff changeset	30
7236 fd7a5993d06d Simplification of conversion routines Barry Warsaw <barry@python.org> parents: 7182 diff changeset	31 DEFAULT_DB = None
18456 70af4a173a95 De-string-module-ification and other Python 2.x improvements. Barry Warsaw <barry@python.org> parents: 10744 diff changeset	32 SPACE = ' '
70af4a173a95 De-string-module-ification and other Python 2.x improvements. Barry Warsaw <barry@python.org> parents: 10744 diff changeset	33 COMMASPACE = ', '
7236 fd7a5993d06d Simplification of conversion routines Barry Warsaw <barry@python.org> parents: 7182 diff changeset	34
9146 c4602236409e Fixed unit test Barry Warsaw <barry@python.org> parents: 7371 diff changeset	35
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	36
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	37 # generic class
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	38 class ColorDB:
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	39 def __init__(self, fp):
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	40 lineno = 2
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	41 self.__name = fp.name
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	42 # Maintain several dictionaries for indexing into the color database.
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	43 # Note that while Tk supports RGB intensities of 4, 8, 12, or 16 bits,
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	44 # for now we only support 8 bit intensities. At least on OpenWindows,
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	45 # all intensities in the /usr/openwin/lib/rgb.txt file are 8-bit
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	46 #
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	47 # key is (red, green, blue) tuple, value is (name, [aliases])
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	48 self.__byrgb = {}
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	49 # key is name, value is (red, green, blue)
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	50 self.__byname = {}
9221 3c9fec57b858 all_names(): Return just the unique names of rgb colors Barry Warsaw <barry@python.org> parents: 9173 diff changeset	51 # all unique names (non-aliases). built-on demand
3c9fec57b858 all_names(): Return just the unique names of rgb colors Barry Warsaw <barry@python.org> parents: 9173 diff changeset	52 self.__allnames = None
44859 feba5d5cac22 Merged revisions 60151-60159,60161-60168,60170,60172-60173,60175 via svnmerge from Georg Brandl <georg@python.org> parents: 42427 diff changeset	53 for line in fp:
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	54 # get this compiled regular expression from derived class
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	55 mo = self._re.match(line)
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	56 if not mo:
42427 46625e6122e9 Convert print statements to function calls in Tools/. Collin Winter <collinw@gmail.com> parents: 25877 diff changeset	57 print('Error in', fp.name, ' line', lineno, file=sys.stderr)
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	58 lineno += 1
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	59 continue
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	60 # extract the red, green, blue, and name
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	61 red, green, blue = self._extractrgb(mo)
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	62 name = self._extractname(mo)
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	63 keyname = name.lower()
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	64 # BAW: for now the `name' is just the first named color with the
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	65 # rgb values we find. Later, we might want to make the two word
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	66 # version the `name', or the CapitalizedVersion, etc.
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	67 key = (red, green, blue)
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	68 foundname, aliases = self.__byrgb.get(key, (name, []))
47103 63741c7521aa More 2to3 fixes in the Tools directory. Fixes #2893. Georg Brandl <georg@python.org> parents: 45049 diff changeset	69 if foundname != name and foundname not in aliases:
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	70 aliases.append(name)
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	71 self.__byrgb[key] = (foundname, aliases)
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	72 # add to byname lookup
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	73 self.__byname[keyname] = key
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	74 lineno = lineno + 1
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	75
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	76 # override in derived classes
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	77 def _extractrgb(self, mo):
18456 70af4a173a95 De-string-module-ification and other Python 2.x improvements. Barry Warsaw <barry@python.org> parents: 10744 diff changeset	78 return [int(x) for x in mo.group('red', 'green', 'blue')]
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	79
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	80 def _extractname(self, mo):
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	81 return mo.group('name')
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	82
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	83 def filename(self):
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	84 return self.__name
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	85
7239 306f737bab57 __init__(): simplify, use tuple as key to __byrgb, lowercase __byname key. Barry Warsaw <barry@python.org> parents: 7236 diff changeset	86 def find_byrgb(self, rgbtuple):
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	87 """Return name for rgbtuple"""
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	88 try:
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	89 return self.__byrgb[rgbtuple]
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	90 except KeyError:
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	91 raise BadColor(rgbtuple)
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	92
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	93 def find_byname(self, name):
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	94 """Return (red, green, blue) for name"""
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	95 name = name.lower()
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	96 try:
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	97 return self.__byname[name]
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	98 except KeyError:
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	99 raise BadColor(name)
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	100
9173 32b19491e70b Changed interface for nearest() method to take individual red, green, Barry Warsaw <barry@python.org> parents: 9165 diff changeset	101 def nearest(self, red, green, blue):
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	102 """Return the name of color nearest (red, green, blue)"""
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	103 # BAW: should we use Voronoi diagrams, Delaunay triangulation, or
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	104 # octree for speeding up the locating of nearest point? Exhaustive
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	105 # search is inefficient, but seems fast enough.
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	106 nearest = -1
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	107 nearest_name = ''
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	108 for name, aliases in self.__byrgb.values():
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	109 r, g, b = self.__byname[name.lower()]
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	110 rdelta = red - r
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	111 gdelta = green - g
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	112 bdelta = blue - b
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	113 distance = rdelta * rdelta + gdelta * gdelta + bdelta * bdelta
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	114 if nearest == -1 or distance < nearest:
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	115 nearest = distance
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	116 nearest_name = name
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	117 return nearest_name
9221 3c9fec57b858 all_names(): Return just the unique names of rgb colors Barry Warsaw <barry@python.org> parents: 9173 diff changeset	118
9224 73df24f6bccf all_names(): => unique_names() Barry Warsaw <barry@python.org> parents: 9221 diff changeset	119 def unique_names(self):
9221 3c9fec57b858 all_names(): Return just the unique names of rgb colors Barry Warsaw <barry@python.org> parents: 9173 diff changeset	120 # sorted
3c9fec57b858 all_names(): Return just the unique names of rgb colors Barry Warsaw <barry@python.org> parents: 9173 diff changeset	121 if not self.__allnames:
3c9fec57b858 all_names(): Return just the unique names of rgb colors Barry Warsaw <barry@python.org> parents: 9173 diff changeset	122 self.__allnames = []
3c9fec57b858 all_names(): Return just the unique names of rgb colors Barry Warsaw <barry@python.org> parents: 9173 diff changeset	123 for name, aliases in self.__byrgb.values():
3c9fec57b858 all_names(): Return just the unique names of rgb colors Barry Warsaw <barry@python.org> parents: 9173 diff changeset	124 self.__allnames.append(name)
47103 63741c7521aa More 2to3 fixes in the Tools directory. Fixes #2893. Georg Brandl <georg@python.org> parents: 45049 diff changeset	125 self.__allnames.sort(key=str.lower)
9221 3c9fec57b858 all_names(): Return just the unique names of rgb colors Barry Warsaw <barry@python.org> parents: 9173 diff changeset	126 return self.__allnames
9224 73df24f6bccf all_names(): => unique_names() Barry Warsaw <barry@python.org> parents: 9221 diff changeset	127
73df24f6bccf all_names(): => unique_names() Barry Warsaw <barry@python.org> parents: 9221 diff changeset	128 def aliases_of(self, red, green, blue):
73df24f6bccf all_names(): => unique_names() Barry Warsaw <barry@python.org> parents: 9221 diff changeset	129 try:
73df24f6bccf all_names(): => unique_names() Barry Warsaw <barry@python.org> parents: 9221 diff changeset	130 name, aliases = self.__byrgb[(red, green, blue)]
73df24f6bccf all_names(): => unique_names() Barry Warsaw <barry@python.org> parents: 9221 diff changeset	131 except KeyError:
73df24f6bccf all_names(): => unique_names() Barry Warsaw <barry@python.org> parents: 9221 diff changeset	132 raise BadColor((red, green, blue))
73df24f6bccf all_names(): => unique_names() Barry Warsaw <barry@python.org> parents: 9221 diff changeset	133 return [name] + aliases
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	134
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	135
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	136 class RGBColorDB(ColorDB):
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	137 _re = re.compile(
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	138 '\s(?P<red>\d+)\s+(?P<green>\d+)\s+(?P<blue>\d+)\s+(?P<name>.)')
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	139
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	140
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	141 class HTML40DB(ColorDB):
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	142 _re = re.compile('(?P<name>\S+)\s+(?P<hexrgb>#[0-9a-fA-F]{6})')
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	143
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	144 def _extractrgb(self, mo):
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	145 return rrggbb_to_triplet(mo.group('hexrgb'))
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	146
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	147 class LightlinkDB(HTML40DB):
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	148 _re = re.compile('(?P<name>(.+))\s+(?P<hexrgb>#[0-9a-fA-F]{6})')
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	149
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	150 def _extractname(self, mo):
18456 70af4a173a95 De-string-module-ification and other Python 2.x improvements. Barry Warsaw <barry@python.org> parents: 10744 diff changeset	151 return mo.group('name').strip()
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	152
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	153 class WebsafeDB(ColorDB):
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	154 _re = re.compile('(?P<hexrgb>#[0-9a-fA-F]{6})')
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	155
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	156 def _extractrgb(self, mo):
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	157 return rrggbb_to_triplet(mo.group('hexrgb'))
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	158
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	159 def _extractname(self, mo):
18456 70af4a173a95 De-string-module-ification and other Python 2.x improvements. Barry Warsaw <barry@python.org> parents: 10744 diff changeset	160 return mo.group('hexrgb').upper()
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	161
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	162
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	163
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	164 # format is a tuple (RE, SCANLINES, CLASS) where RE is a compiled regular
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	165 # expression, SCANLINES is the number of header lines to scan, and CLASS is
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	166 # the class to instantiate if a match is found
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	167
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	168 FILETYPES = [
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	169 (re.compile('Xorg'), RGBColorDB),
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	170 (re.compile('XConsortium'), RGBColorDB),
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	171 (re.compile('HTML'), HTML40DB),
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	172 (re.compile('lightlink'), LightlinkDB),
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	173 (re.compile('Websafe'), WebsafeDB),
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	174 ]
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	175
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	176 def get_colordb(file, filetype=None):
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	177 colordb = None
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	178 fp = open(file)
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	179 try:
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	180 line = fp.readline()
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	181 if not line:
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	182 return None
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	183 # try to determine the type of RGB file it is
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	184 if filetype is None:
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	185 filetypes = FILETYPES
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	186 else:
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	187 filetypes = [filetype]
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	188 for typere, class_ in filetypes:
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	189 mo = typere.search(line)
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	190 if mo:
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	191 break
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	192 else:
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	193 # no matching type
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	194 return None
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	195 # we know the type and the class to grok the type, so suck it in
4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	196 colordb = class_(fp)
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	197 finally:
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	198 fp.close()
7236 fd7a5993d06d Simplification of conversion routines Barry Warsaw <barry@python.org> parents: 7182 diff changeset	199 # save a global copy
fd7a5993d06d Simplification of conversion routines Barry Warsaw <barry@python.org> parents: 7182 diff changeset	200 global DEFAULT_DB
fd7a5993d06d Simplification of conversion routines Barry Warsaw <barry@python.org> parents: 7182 diff changeset	201 DEFAULT_DB = colordb
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	202 return colordb
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	203
7182 59362ea7e0f8 BadColor: new exception class Barry Warsaw <barry@python.org> parents: 7181 diff changeset	204
59362ea7e0f8 BadColor: new exception class Barry Warsaw <barry@python.org> parents: 7181 diff changeset	205
7346 5a9256b8c338 Cache conversions from triplets to rrggbb's and vice versa. Wasteful Barry Warsaw <barry@python.org> parents: 7293 diff changeset	206 _namedict = {}
18456 70af4a173a95 De-string-module-ification and other Python 2.x improvements. Barry Warsaw <barry@python.org> parents: 10744 diff changeset	207
70af4a173a95 De-string-module-ification and other Python 2.x improvements. Barry Warsaw <barry@python.org> parents: 10744 diff changeset	208 def rrggbb_to_triplet(color):
7182 59362ea7e0f8 BadColor: new exception class Barry Warsaw <barry@python.org> parents: 7181 diff changeset	209 """Converts a #rrggbb color to the tuple (red, green, blue)."""
7346 5a9256b8c338 Cache conversions from triplets to rrggbb's and vice versa. Wasteful Barry Warsaw <barry@python.org> parents: 7293 diff changeset	210 rgbtuple = _namedict.get(color)
5a9256b8c338 Cache conversions from triplets to rrggbb's and vice versa. Wasteful Barry Warsaw <barry@python.org> parents: 7293 diff changeset	211 if rgbtuple is None:
47103 63741c7521aa More 2to3 fixes in the Tools directory. Fixes #2893. Georg Brandl <georg@python.org> parents: 45049 diff changeset	212 if color[0] != '#':
9249 0283759407e1 rrggbb_to_triplet(): When color doesn't start with `#' raise a Barry Warsaw <barry@python.org> parents: 9225 diff changeset	213 raise BadColor(color)
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	214 red = color[1:3]
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	215 green = color[3:5]
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	216 blue = color[5:7]
18456 70af4a173a95 De-string-module-ification and other Python 2.x improvements. Barry Warsaw <barry@python.org> parents: 10744 diff changeset	217 rgbtuple = int(red, 16), int(green, 16), int(blue, 16)
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	218 _namedict[color] = rgbtuple
7346 5a9256b8c338 Cache conversions from triplets to rrggbb's and vice versa. Wasteful Barry Warsaw <barry@python.org> parents: 7293 diff changeset	219 return rgbtuple
7236 fd7a5993d06d Simplification of conversion routines Barry Warsaw <barry@python.org> parents: 7182 diff changeset	220
7182 59362ea7e0f8 BadColor: new exception class Barry Warsaw <barry@python.org> parents: 7181 diff changeset	221
7346 5a9256b8c338 Cache conversions from triplets to rrggbb's and vice versa. Wasteful Barry Warsaw <barry@python.org> parents: 7293 diff changeset	222 _tripdict = {}
7236 fd7a5993d06d Simplification of conversion routines Barry Warsaw <barry@python.org> parents: 7182 diff changeset	223 def triplet_to_rrggbb(rgbtuple):
fd7a5993d06d Simplification of conversion routines Barry Warsaw <barry@python.org> parents: 7182 diff changeset	224 """Converts a (red, green, blue) tuple to #rrggbb."""
10744 4c9de6fd267b Lots of changes to support loading alternative color name database. Barry Warsaw <barry@python.org> parents: 9367 diff changeset	225 global _tripdict
7346 5a9256b8c338 Cache conversions from triplets to rrggbb's and vice versa. Wasteful Barry Warsaw <barry@python.org> parents: 7293 diff changeset	226 hexname = _tripdict.get(rgbtuple)
5a9256b8c338 Cache conversions from triplets to rrggbb's and vice versa. Wasteful Barry Warsaw <barry@python.org> parents: 7293 diff changeset	227 if hexname is None:
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	228 hexname = '#%02x%02x%02x' % rgbtuple
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	229 _tripdict[rgbtuple] = hexname
7346 5a9256b8c338 Cache conversions from triplets to rrggbb's and vice versa. Wasteful Barry Warsaw <barry@python.org> parents: 7293 diff changeset	230 return hexname
7182 59362ea7e0f8 BadColor: new exception class Barry Warsaw <barry@python.org> parents: 7181 diff changeset	231
59362ea7e0f8 BadColor: new exception class Barry Warsaw <barry@python.org> parents: 7181 diff changeset	232
7346 5a9256b8c338 Cache conversions from triplets to rrggbb's and vice versa. Wasteful Barry Warsaw <barry@python.org> parents: 7293 diff changeset	233 _maxtuple = (256.0,) * 3
9146 c4602236409e Fixed unit test Barry Warsaw <barry@python.org> parents: 7371 diff changeset	234 def triplet_to_fractional_rgb(rgbtuple):
47103 63741c7521aa More 2to3 fixes in the Tools directory. Fixes #2893. Georg Brandl <georg@python.org> parents: 45049 diff changeset	235 return list(map(operator.__div__, rgbtuple, _maxtuple))
7293 0dbec98acfc9 triplet_to_pmwrgb(): New function Barry Warsaw <barry@python.org> parents: 7239 diff changeset	236
0dbec98acfc9 triplet_to_pmwrgb(): New function Barry Warsaw <barry@python.org> parents: 7239 diff changeset	237
9165 d3ef4d10fb95 triplet_to_brightness(): New function Barry Warsaw <barry@python.org> parents: 9146 diff changeset	238 def triplet_to_brightness(rgbtuple):
d3ef4d10fb95 triplet_to_brightness(): New function Barry Warsaw <barry@python.org> parents: 9146 diff changeset	239 # return the brightness (grey level) along the scale 0.0==black to
d3ef4d10fb95 triplet_to_brightness(): New function Barry Warsaw <barry@python.org> parents: 9146 diff changeset	240 # 1.0==white
d3ef4d10fb95 triplet_to_brightness(): New function Barry Warsaw <barry@python.org> parents: 9146 diff changeset	241 r = 0.299
d3ef4d10fb95 triplet_to_brightness(): New function Barry Warsaw <barry@python.org> parents: 9146 diff changeset	242 g = 0.587
d3ef4d10fb95 triplet_to_brightness(): New function Barry Warsaw <barry@python.org> parents: 9146 diff changeset	243 b = 0.114
d3ef4d10fb95 triplet_to_brightness(): New function Barry Warsaw <barry@python.org> parents: 9146 diff changeset	244 return rrgbtuple[0] + grgbtuple[1] + b*rgbtuple[2]
d3ef4d10fb95 triplet_to_brightness(): New function Barry Warsaw <barry@python.org> parents: 9146 diff changeset	245
d3ef4d10fb95 triplet_to_brightness(): New function Barry Warsaw <barry@python.org> parents: 9146 diff changeset	246
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	247
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	248 if __name__ == '__main__':
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	249 colordb = get_colordb('/usr/openwin/lib/rgb.txt')
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	250 if not colordb:
42427 46625e6122e9 Convert print statements to function calls in Tools/. Collin Winter <collinw@gmail.com> parents: 25877 diff changeset	251 print('No parseable color database found')
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	252 sys.exit(1)
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	253 # on my system, this color matches exactly
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	254 target = 'navy'
9146 c4602236409e Fixed unit test Barry Warsaw <barry@python.org> parents: 7371 diff changeset	255 red, green, blue = rgbtuple = colordb.find_byname(target)
42427 46625e6122e9 Convert print statements to function calls in Tools/. Collin Winter <collinw@gmail.com> parents: 25877 diff changeset	256 print(target, ':', red, green, blue, triplet_to_rrggbb(rgbtuple))
9146 c4602236409e Fixed unit test Barry Warsaw <barry@python.org> parents: 7371 diff changeset	257 name, aliases = colordb.find_byrgb(rgbtuple)
42427 46625e6122e9 Convert print statements to function calls in Tools/. Collin Winter <collinw@gmail.com> parents: 25877 diff changeset	258 print('name:', name, 'aliases:', COMMASPACE.join(aliases))
25877 15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	259 r, g, b = (1, 1, 128) # nearest to navy
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	260 r, g, b = (145, 238, 144) # nearest to lightgreen
15607ec49faf FILETYPES: Newer XFree86 rgb.txt files use the key Xorg instead of Barry Warsaw <barry@python.org> parents: 18456 diff changeset	261 r, g, b = (255, 251, 250) # snow
42427 46625e6122e9 Convert print statements to function calls in Tools/. Collin Winter <collinw@gmail.com> parents: 25877 diff changeset	262 print('finding nearest to', target, '...')
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	263 import time
3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	264 t0 = time.time()
9367 f6819b926fb2 Updated __main__ unit test Barry Warsaw <barry@python.org> parents: 9249 diff changeset	265 nearest = colordb.nearest(r, g, b)
7180 3a960b2e2667 Initial revision Barry Warsaw <barry@python.org> parents: diff changeset	266 t1 = time.time()
42427 46625e6122e9 Convert print statements to function calls in Tools/. Collin Winter <collinw@gmail.com> parents: 25877 diff changeset	267 print('found nearest color', nearest, 'in', t1-t0, 'seconds')
9367 f6819b926fb2 Updated __main__ unit test Barry Warsaw <barry@python.org> parents: 9249 diff changeset	268 # dump the database
f6819b926fb2 Updated __main__ unit test Barry Warsaw <barry@python.org> parents: 9249 diff changeset	269 for n in colordb.unique_names():
f6819b926fb2 Updated __main__ unit test Barry Warsaw <barry@python.org> parents: 9249 diff changeset	270 r, g, b = colordb.find_byname(n)
f6819b926fb2 Updated __main__ unit test Barry Warsaw <barry@python.org> parents: 9249 diff changeset	271 aliases = colordb.aliases_of(r, g, b)
42427 46625e6122e9 Convert print statements to function calls in Tools/. Collin Winter <collinw@gmail.com> parents: 25877 diff changeset	272 print('%20s: (%3d/%3d/%3d) == %s' % (n, r, g, b,
46625e6122e9 Convert print statements to function calls in Tools/. Collin Winter <collinw@gmail.com> parents: 25877 diff changeset	273 SPACE.join(aliases[1:])))

Mercurial > cpython-lru_cache-weakref

annotate Tools/pynche/ColorDB.py @ 78069:a8aa77458e12 3.2