Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.11/site-packages/PIL/ImagePalette.py: 52%

1# 

2# The Python Imaging Library. 

3# $Id$ 

4# 

5# image palette object 

6# 

7# History: 

8# 1996-03-11 fl Rewritten. 

9# 1997-01-03 fl Up and running. 

10# 1997-08-23 fl Added load hack 

11# 2001-04-16 fl Fixed randint shadow bug in random() 

12# 

13# Copyright (c) 1997-2001 by Secret Labs AB 

14# Copyright (c) 1996-1997 by Fredrik Lundh 

15# 

16# See the README file for information on usage and redistribution. 

17# 

18from __future__ import annotations 

19 

20import array 

21from collections.abc import Sequence 

22from typing import IO 

23 

24from . import GimpGradientFile, GimpPaletteFile, ImageColor, PaletteFile 

25 

26TYPE_CHECKING = False 

27if TYPE_CHECKING: 

28 from . import Image 

29 

30 

31class ImagePalette: 

32 """ 

33 Color palette for palette mapped images 

34 

35 :param mode: The mode to use for the palette. See: 

36 :ref:`concept-modes`. Defaults to "RGB" 

37 :param palette: An optional palette. If given, it must be a bytearray, 

38 an array or a list of ints between 0-255. The list must consist of 

39 all channels for one color followed by the next color (e.g. RGBRGBRGB). 

40 Defaults to an empty palette. 

41 """ 

42 

43 def __init__( 

44 self, 

45 mode: str = "RGB", 

46 palette: Sequence[int] | bytes | bytearray | None = None, 

47 ) -> None: 

48 self.mode = mode 

49 self.rawmode: str | None = None # if set, palette contains raw data 

50 self.palette = palette or bytearray() 

51 self.dirty: int | None = None 

52 

53 @property 

54 def palette(self) -> Sequence[int] | bytes | bytearray: 

55 return self._palette 

56 

57 @palette.setter 

58 def palette(self, palette: Sequence[int] | bytes | bytearray) -> None: 

59 self._colors: dict[tuple[int, ...], int] | None = None 

60 self._palette = palette 

61 

62 @property 

63 def colors(self) -> dict[tuple[int, ...], int]: 

64 if self._colors is None: 

65 mode_len = len(self.mode) 

66 self._colors = {} 

67 for i in range(0, len(self.palette), mode_len): 

68 color = tuple(self.palette[i : i + mode_len]) 

69 if color in self._colors: 

70 continue 

71 self._colors[color] = i // mode_len 

72 return self._colors 

73 

74 @colors.setter 

75 def colors(self, colors: dict[tuple[int, ...], int]) -> None: 

76 self._colors = colors 

77 

78 def copy(self) -> ImagePalette: 

79 new = ImagePalette() 

80 

81 new.mode = self.mode 

82 new.rawmode = self.rawmode 

83 if self.palette is not None: 

84 new.palette = self.palette[:] 

85 new.dirty = self.dirty 

86 

87 return new 

88 

89 def getdata(self) -> tuple[str, Sequence[int] | bytes | bytearray]: 

90 """ 

91 Get palette contents in format suitable for the low-level 

92 ``im.putpalette`` primitive. 

93 

94 .. warning:: This method is experimental. 

95 """ 

96 if self.rawmode: 

97 return self.rawmode, self.palette 

98 return self.mode, self.tobytes() 

99 

100 def tobytes(self) -> bytes: 

101 """Convert palette to bytes. 

102 

103 .. warning:: This method is experimental. 

104 """ 

105 if self.rawmode: 

106 msg = "palette contains raw palette data" 

107 raise ValueError(msg) 

108 if isinstance(self.palette, bytes): 

109 return self.palette 

110 arr = array.array("B", self.palette) 

111 return arr.tobytes() 

112 

113 # Declare tostring as an alias for tobytes 

114 tostring = tobytes 

115 

116 def _new_color_index( 

117 self, image: Image.Image | None = None, e: Exception | None = None 

118 ) -> int: 

119 if not isinstance(self.palette, bytearray): 

120 self._palette = bytearray(self.palette) 

121 index = len(self.palette) // len(self.mode) 

122 special_colors: tuple[int | tuple[int, ...] | None, ...] = () 

123 if image: 

124 special_colors = ( 

125 image.info.get("background"), 

126 image.info.get("transparency"), 

127 ) 

128 while index in special_colors: 

129 index += 1 

130 if index >= 256: 

131 if image: 

132 # Search for an unused index 

133 for i, count in reversed(list(enumerate(image.histogram()))): 

134 if count == 0 and i not in special_colors: 

135 index = i 

136 break 

137 if index >= 256: 

138 msg = "cannot allocate more than 256 colors" 

139 raise ValueError(msg) from e 

140 return index 

141 

142 def getcolor( 

143 self, 

144 color: tuple[int, ...], 

145 image: Image.Image | None = None, 

146 ) -> int: 

147 """Given an rgb tuple, allocate palette entry. 

148 

149 .. warning:: This method is experimental. 

150 """ 

151 if self.rawmode: 

152 msg = "palette contains raw palette data" 

153 raise ValueError(msg) 

154 if isinstance(color, tuple): 

155 if self.mode == "RGB": 

156 if len(color) == 4: 

157 if color[3] != 255: 

158 msg = "cannot add non-opaque RGBA color to RGB palette" 

159 raise ValueError(msg) 

160 color = color[:3] 

161 elif self.mode == "RGBA": 

162 if len(color) == 3: 

163 color += (255,) 

164 try: 

165 return self.colors[color] 

166 except KeyError as e: 

167 # allocate new color slot 

168 index = self._new_color_index(image, e) 

169 assert isinstance(self._palette, bytearray) 

170 self.colors[color] = index 

171 mode_len = len(self.mode) 

172 if index * mode_len < len(self.palette): 

173 self._palette = ( 

174 self._palette[: index * mode_len] 

175 + bytes(color) 

176 + self._palette[index * mode_len + mode_len :] 

177 ) 

178 else: 

179 self._palette += bytes(color) 

180 self.dirty = 1 

181 return index 

182 else: 

183 msg = f"unknown color specifier: {repr(color)}" # type: ignore[unreachable] 

184 raise ValueError(msg) 

185 

186 def save(self, fp: str | IO[str]) -> None: 

187 """Save palette to text file. 

188 

189 .. warning:: This method is experimental. 

190 """ 

191 if self.rawmode: 

192 msg = "palette contains raw palette data" 

193 raise ValueError(msg) 

194 if isinstance(fp, str): 

195 fp = open(fp, "w") 

196 fp.write("# Palette\n") 

197 fp.write(f"# Mode: {self.mode}\n") 

198 for i in range(256): 

199 fp.write(f"{i}") 

200 for j in range(i * len(self.mode), (i + 1) * len(self.mode)): 

201 try: 

202 fp.write(f" {self.palette[j]}") 

203 except IndexError: 

204 fp.write(" 0") 

205 fp.write("\n") 

206 fp.close() 

207 

208 

209# -------------------------------------------------------------------- 

210# Internal 

211 

212 

213def raw(rawmode: str, data: Sequence[int] | bytes | bytearray) -> ImagePalette: 

214 palette = ImagePalette() 

215 palette.rawmode = rawmode 

216 palette.palette = data 

217 palette.dirty = 1 

218 return palette 

219 

220 

221# -------------------------------------------------------------------- 

222# Factories 

223 

224 

225def make_linear_lut(black: int, white: float) -> list[int]: 

226 if black == 0: 

227 return [int(white * i // 255) for i in range(256)] 

228 

229 msg = "unavailable when black is non-zero" 

230 raise NotImplementedError(msg) # FIXME 

231 

232 

233def make_gamma_lut(exp: float) -> list[int]: 

234 return [int(((i / 255.0) ** exp) * 255.0 + 0.5) for i in range(256)] 

235 

236 

237def negative(mode: str = "RGB") -> ImagePalette: 

238 palette = list(range(256 * len(mode))) 

239 palette.reverse() 

240 return ImagePalette(mode, [i // len(mode) for i in palette]) 

241 

242 

243def random(mode: str = "RGB") -> ImagePalette: 

244 from random import randint 

245 

246 palette = [randint(0, 255) for _ in range(256 * len(mode))] 

247 return ImagePalette(mode, palette) 

248 

249 

250def sepia(white: str = "#fff0c0") -> ImagePalette: 

251 bands = [make_linear_lut(0, band) for band in ImageColor.getrgb(white)] 

252 return ImagePalette("RGB", [bands[i % 3][i // 3] for i in range(256 * 3)]) 

253 

254 

255def wedge(mode: str = "RGB") -> ImagePalette: 

256 palette = list(range(256 * len(mode))) 

257 return ImagePalette(mode, [i // len(mode) for i in palette]) 

258 

259 

260def load(filename: str) -> tuple[bytes, str]: 

261 # FIXME: supports GIMP gradients only 

262 

263 with open(filename, "rb") as fp: 

264 paletteHandlers: list[ 

265 type[ 

266 GimpPaletteFile.GimpPaletteFile 

267 | GimpGradientFile.GimpGradientFile 

268 | PaletteFile.PaletteFile 

269 ] 

270 ] = [ 

271 GimpPaletteFile.GimpPaletteFile, 

272 GimpGradientFile.GimpGradientFile, 

273 PaletteFile.PaletteFile, 

274 ] 

275 for paletteHandler in paletteHandlers: 

276 try: 

277 fp.seek(0) 

278 lut = paletteHandler(fp).getpalette() 

279 if lut: 

280 break 

281 except (SyntaxError, ValueError): 

282 pass 

283 else: 

284 msg = "cannot load palette" 

285 raise OSError(msg) 

286 

287 return lut # data, rawmode