| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172 |
- #
- # Python Imaging Library
- # $Id$
- #
- # stuff to read GIMP palette files
- #
- # History:
- # 1997-08-23 fl Created
- # 2004-09-07 fl Support GIMP 2.0 palette files.
- #
- # Copyright (c) Secret Labs AB 1997-2004. All rights reserved.
- # Copyright (c) Fredrik Lundh 1997-2004.
- #
- # See the README file for information on usage and redistribution.
- #
- from __future__ import annotations
- import re
- from io import BytesIO
- from typing import IO
- class GimpPaletteFile:
- """File handler for GIMP's palette format."""
- rawmode = "RGB"
- def _read(self, fp: IO[bytes], limit: bool = True) -> None:
- if not fp.readline().startswith(b"GIMP Palette"):
- msg = "not a GIMP palette file"
- raise SyntaxError(msg)
- palette: list[int] = []
- i = 0
- while True:
- if limit and i == 256 + 3:
- break
- i += 1
- s = fp.readline()
- if not s:
- break
- # skip fields and comment lines
- if re.match(rb"\w+:|#", s):
- continue
- if limit and len(s) > 100:
- msg = "bad palette file"
- raise SyntaxError(msg)
- v = s.split(maxsplit=3)
- if len(v) < 3:
- msg = "bad palette entry"
- raise ValueError(msg)
- palette += (int(v[i]) for i in range(3))
- if limit and len(palette) == 768:
- break
- self.palette = bytes(palette)
- def __init__(self, fp: IO[bytes]) -> None:
- self._read(fp)
- @classmethod
- def frombytes(cls, data: bytes) -> GimpPaletteFile:
- self = cls.__new__(cls)
- self._read(BytesIO(data), False)
- return self
- def getpalette(self) -> tuple[bytes, str]:
- return self.palette, self.rawmode
|
