1998-01-31 00:29:41 +00:00
|
|
|
|
"""Color Database.
|
|
|
|
|
|
|
|
|
|
To create a class that contains color lookup methods, use the module global
|
|
|
|
|
function `get_colordb(file)'. This function will try to examine the file to
|
|
|
|
|
figure out what the format of the file is. If it can't figure out the file
|
|
|
|
|
format, or it has trouble reading the file, None is returned. You can pass
|
|
|
|
|
get_colordb() an optional filetype argument.
|
|
|
|
|
|
|
|
|
|
Supporte file types are:
|
|
|
|
|
|
|
|
|
|
X_RGB_TXT -- X Consortium rgb.txt format files. Three columns of numbers
|
|
|
|
|
from 0 .. 255 separated by whitespace. Arbitrary trailing
|
|
|
|
|
columns used as the color name.
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
import sys
|
1998-02-11 18:55:37 +00:00
|
|
|
|
import string
|
1998-01-31 00:29:41 +00:00
|
|
|
|
import re
|
1998-02-11 17:19:23 +00:00
|
|
|
|
from types import *
|
1998-02-18 00:02:26 +00:00
|
|
|
|
import operator
|
1998-01-31 00:29:41 +00:00
|
|
|
|
|
1998-01-31 23:38:48 +00:00
|
|
|
|
class BadColor(Exception):
|
|
|
|
|
pass
|
|
|
|
|
|
1998-02-11 17:19:23 +00:00
|
|
|
|
DEFAULT_DB = None
|
|
|
|
|
|
1998-09-25 22:51:36 +00:00
|
|
|
|
|
1998-01-31 00:29:41 +00:00
|
|
|
|
|
|
|
|
|
# generic class
|
|
|
|
|
class ColorDB:
|
|
|
|
|
def __init__(self, fp, lineno):
|
|
|
|
|
# Maintain several dictionaries for indexing into the color database.
|
|
|
|
|
# Note that while Tk supports RGB intensities of 4, 8, 12, or 16 bits,
|
|
|
|
|
# for now we only support 8 bit intensities. At least on OpenWindows,
|
|
|
|
|
# all intensities in the /usr/openwin/lib/rgb.txt file are 8-bit
|
|
|
|
|
#
|
1998-02-11 18:55:37 +00:00
|
|
|
|
# key is (red, green, blue) tuple, value is (name, [aliases])
|
|
|
|
|
self.__byrgb = {}
|
1998-01-31 00:29:41 +00:00
|
|
|
|
#
|
1998-02-11 18:55:37 +00:00
|
|
|
|
# key is name, value is (red, green, blue)
|
1998-01-31 00:29:41 +00:00
|
|
|
|
self.__byname = {}
|
|
|
|
|
#
|
|
|
|
|
while 1:
|
|
|
|
|
line = fp.readline()
|
|
|
|
|
if not line:
|
|
|
|
|
break
|
|
|
|
|
# get this compiled regular expression from derived class
|
|
|
|
|
mo = self._re.match(line)
|
|
|
|
|
if not mo:
|
|
|
|
|
sys.stderr.write('Error in %s, line %d\n' % (fp.name, lineno))
|
|
|
|
|
lineno = lineno + 1
|
|
|
|
|
continue
|
|
|
|
|
#
|
|
|
|
|
# extract the red, green, blue, and name
|
1998-02-11 18:55:37 +00:00
|
|
|
|
#
|
1998-01-31 00:29:41 +00:00
|
|
|
|
red, green, blue = map(int, mo.group('red', 'green', 'blue'))
|
|
|
|
|
name = mo.group('name')
|
1998-02-11 18:55:37 +00:00
|
|
|
|
keyname = string.lower(name)
|
1998-01-31 00:29:41 +00:00
|
|
|
|
#
|
|
|
|
|
# TBD: for now the `name' is just the first named color with the
|
|
|
|
|
# rgb values we find. Later, we might want to make the two word
|
|
|
|
|
# version the `name', or the CapitalizedVersion, etc.
|
1998-02-11 18:55:37 +00:00
|
|
|
|
#
|
|
|
|
|
key = (red, green, blue)
|
|
|
|
|
foundname, aliases = self.__byrgb.get(key, (name, []))
|
1998-01-31 00:29:41 +00:00
|
|
|
|
if foundname <> name and foundname not in aliases:
|
|
|
|
|
aliases.append(name)
|
1998-02-11 18:55:37 +00:00
|
|
|
|
self.__byrgb[key] = (foundname, aliases)
|
1998-01-31 00:29:41 +00:00
|
|
|
|
#
|
|
|
|
|
# add to byname lookup
|
1998-02-11 18:55:37 +00:00
|
|
|
|
#
|
|
|
|
|
self.__byname[keyname] = key
|
1998-01-31 00:29:41 +00:00
|
|
|
|
lineno = lineno + 1
|
|
|
|
|
|
1998-02-11 18:55:37 +00:00
|
|
|
|
def find_byrgb(self, rgbtuple):
|
1998-01-31 23:38:48 +00:00
|
|
|
|
try:
|
1998-02-11 18:55:37 +00:00
|
|
|
|
return self.__byrgb[rgbtuple]
|
1998-01-31 23:38:48 +00:00
|
|
|
|
except KeyError:
|
1998-02-11 18:55:37 +00:00
|
|
|
|
raise BadColor(rgbtuple)
|
1998-01-31 00:29:41 +00:00
|
|
|
|
|
|
|
|
|
def find_byname(self, name):
|
1998-02-11 18:55:37 +00:00
|
|
|
|
name = string.lower(name)
|
1998-01-31 23:38:48 +00:00
|
|
|
|
try:
|
|
|
|
|
return self.__byname[name]
|
|
|
|
|
except KeyError:
|
|
|
|
|
raise BadColor(name)
|
1998-01-31 00:29:41 +00:00
|
|
|
|
|
1998-09-29 20:03:15 +00:00
|
|
|
|
def nearest(self, red, green, blue):
|
1998-01-31 00:29:41 +00:00
|
|
|
|
# TBD: use Voronoi diagrams, Delaunay triangulation, or octree for
|
1998-02-11 18:55:37 +00:00
|
|
|
|
# speeding up the locating of nearest point. Exhaustive search is
|
|
|
|
|
# inefficient, but may be fast enough.
|
1998-01-31 00:29:41 +00:00
|
|
|
|
nearest = -1
|
|
|
|
|
nearest_name = ''
|
1998-02-11 18:55:37 +00:00
|
|
|
|
for name, aliases in self.__byrgb.values():
|
|
|
|
|
r, g, b = self.__byname[string.lower(name)]
|
1998-01-31 00:29:41 +00:00
|
|
|
|
rdelta = red - r
|
|
|
|
|
gdelta = green - g
|
|
|
|
|
bdelta = blue - b
|
|
|
|
|
distance = rdelta * rdelta + gdelta * gdelta + bdelta * bdelta
|
|
|
|
|
if nearest == -1 or distance < nearest:
|
|
|
|
|
nearest = distance
|
|
|
|
|
nearest_name = name
|
|
|
|
|
return nearest_name
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class RGBColorDB(ColorDB):
|
|
|
|
|
_re = re.compile(
|
|
|
|
|
'\s*(?P<red>\d+)\s+(?P<green>\d+)\s+(?P<blue>\d+)\s+(?P<name>.*)')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# format is a tuple (RE, SCANLINES, CLASS) where RE is a compiled regular
|
|
|
|
|
# expression, SCANLINES is the number of header lines to scan, and CLASS is
|
|
|
|
|
# the class to instantiate if a match is found
|
|
|
|
|
|
|
|
|
|
X_RGB_TXT = re.compile('XConsortium'), 1, RGBColorDB
|
|
|
|
|
|
|
|
|
|
def get_colordb(file, filetype=X_RGB_TXT):
|
|
|
|
|
colordb = None
|
|
|
|
|
fp = None
|
|
|
|
|
typere, scanlines, class_ = filetype
|
|
|
|
|
try:
|
|
|
|
|
try:
|
|
|
|
|
lineno = 0
|
|
|
|
|
fp = open(file)
|
|
|
|
|
while lineno < scanlines:
|
|
|
|
|
line = fp.readline()
|
|
|
|
|
if not line:
|
|
|
|
|
break
|
|
|
|
|
mo = typere.search(line)
|
|
|
|
|
if mo:
|
|
|
|
|
colordb = class_(fp, lineno)
|
|
|
|
|
break
|
|
|
|
|
lineno = lineno + 1
|
|
|
|
|
except IOError:
|
|
|
|
|
pass
|
|
|
|
|
finally:
|
|
|
|
|
if fp:
|
|
|
|
|
fp.close()
|
1998-02-11 17:19:23 +00:00
|
|
|
|
# save a global copy
|
|
|
|
|
global DEFAULT_DB
|
|
|
|
|
DEFAULT_DB = colordb
|
1998-01-31 00:29:41 +00:00
|
|
|
|
return colordb
|
|
|
|
|
|
1998-01-31 23:38:48 +00:00
|
|
|
|
|
|
|
|
|
|
1998-02-18 00:02:26 +00:00
|
|
|
|
_namedict = {}
|
1998-02-18 17:01:12 +00:00
|
|
|
|
def rrggbb_to_triplet(color, atoi=string.atoi):
|
1998-01-31 23:38:48 +00:00
|
|
|
|
"""Converts a #rrggbb color to the tuple (red, green, blue)."""
|
1998-02-18 00:02:26 +00:00
|
|
|
|
rgbtuple = _namedict.get(color)
|
|
|
|
|
if rgbtuple is None:
|
|
|
|
|
assert color[0] == '#'
|
|
|
|
|
red = color[1:3]
|
|
|
|
|
green = color[3:5]
|
|
|
|
|
blue = color[5:7]
|
1998-02-18 17:01:12 +00:00
|
|
|
|
rgbtuple = (atoi(red, 16), atoi(green, 16), atoi(blue, 16))
|
1998-02-18 00:02:26 +00:00
|
|
|
|
_namedict[color] = rgbtuple
|
|
|
|
|
return rgbtuple
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_tripdict = {}
|
1998-02-11 17:19:23 +00:00
|
|
|
|
def triplet_to_rrggbb(rgbtuple):
|
|
|
|
|
"""Converts a (red, green, blue) tuple to #rrggbb."""
|
1998-02-18 00:02:26 +00:00
|
|
|
|
hexname = _tripdict.get(rgbtuple)
|
|
|
|
|
if hexname is None:
|
1998-02-18 17:01:12 +00:00
|
|
|
|
hexname = '#%02x%02x%02x' % rgbtuple
|
1998-02-18 00:02:26 +00:00
|
|
|
|
_tripdict[rgbtuple] = hexname
|
|
|
|
|
return hexname
|
1998-01-31 23:38:48 +00:00
|
|
|
|
|
|
|
|
|
|
1998-02-18 00:02:26 +00:00
|
|
|
|
_maxtuple = (256.0,) * 3
|
1998-09-25 22:51:36 +00:00
|
|
|
|
def triplet_to_fractional_rgb(rgbtuple):
|
1998-02-18 00:02:26 +00:00
|
|
|
|
return map(operator.__div__, rgbtuple, _maxtuple)
|
1998-02-13 21:27:56 +00:00
|
|
|
|
|
|
|
|
|
|
1998-09-28 23:39:18 +00:00
|
|
|
|
def triplet_to_brightness(rgbtuple):
|
|
|
|
|
# return the brightness (grey level) along the scale 0.0==black to
|
|
|
|
|
# 1.0==white
|
|
|
|
|
r = 0.299
|
|
|
|
|
g = 0.587
|
|
|
|
|
b = 0.114
|
|
|
|
|
return r*rgbtuple[0] + g*rgbtuple[1] + b*rgbtuple[2]
|
|
|
|
|
|
|
|
|
|
|
1998-01-31 00:29:41 +00:00
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
import string
|
|
|
|
|
|
|
|
|
|
colordb = get_colordb('/usr/openwin/lib/rgb.txt')
|
|
|
|
|
if not colordb:
|
|
|
|
|
print 'No parseable color database found'
|
|
|
|
|
sys.exit(1)
|
|
|
|
|
# on my system, this color matches exactly
|
|
|
|
|
target = 'navy'
|
1998-09-25 22:51:36 +00:00
|
|
|
|
red, green, blue = rgbtuple = colordb.find_byname(target)
|
|
|
|
|
print target, ':', red, green, blue, triplet_to_rrggbb(rgbtuple)
|
|
|
|
|
name, aliases = colordb.find_byrgb(rgbtuple)
|
1998-01-31 00:29:41 +00:00
|
|
|
|
print 'name:', name, 'aliases:', string.join(aliases, ", ")
|
|
|
|
|
target = (1, 1, 128) # nearest to navy
|
|
|
|
|
target = (145, 238, 144) # nearest to lightgreen
|
|
|
|
|
target = (255, 251, 250) # snow
|
|
|
|
|
print 'finding nearest to', target, '...'
|
|
|
|
|
import time
|
|
|
|
|
t0 = time.time()
|
1998-09-25 22:51:36 +00:00
|
|
|
|
nearest = colordb.nearest(target)
|
1998-01-31 00:29:41 +00:00
|
|
|
|
t1 = time.time()
|
|
|
|
|
print 'found nearest color', nearest, 'in', t1-t0, 'seconds'
|
1998-01-31 23:38:48 +00:00
|
|
|
|
|