Pillow/PIL/PaletteFile.py

54 lines
1.0 KiB
Python
Raw Normal View History

2010-07-31 06:52:47 +04:00
#
# Python Imaging Library
# $Id$
#
# stuff to read simple, teragon-style palette files
#
# History:
# 97-08-23 fl Created
#
# Copyright (c) Secret Labs AB 1997.
# Copyright (c) Fredrik Lundh 1997.
#
# See the README file for information on usage and redistribution.
#
##
# File handler for Teragon-style palette files.
class PaletteFile:
rawmode = "RGB"
def __init__(self, fp):
self.palette = [(i, i, i) for i in range(256)]
2010-07-31 06:52:47 +04:00
while 1:
s = fp.readline()
if not s:
break
if s[0] == "#":
continue
if len(s) > 100:
raise SyntaxError("bad palette file")
2010-07-31 06:52:47 +04:00
v = [int(x) for x in s.split()]
2010-07-31 06:52:47 +04:00
try:
[i, r, g, b] = v
except ValueError:
[i, r] = v
g = b = r
if 0 <= i <= 255:
self.palette[i] = chr(r) + chr(g) + chr(b)
self.palette = "".join(self.palette)
2010-07-31 06:52:47 +04:00
def getpalette(self):
return self.palette, self.rawmode