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

1# 

2# The Python Imaging Library. 

3# $Id$ 

4# 

5# XBM File handling 

6# 

7# History: 

8# 1995-09-08 fl Created 

9# 1996-11-01 fl Added save support 

10# 1997-07-07 fl Made header parser more tolerant 

11# 1997-07-22 fl Fixed yet another parser bug 

12# 2001-02-17 fl Use 're' instead of 'regex' (Python 2.1) (0.4) 

13# 2001-05-13 fl Added hotspot handling (based on code from Bernhard Herzog) 

14# 2004-02-24 fl Allow some whitespace before first #define 

15# 

16# Copyright (c) 1997-2004 by Secret Labs AB 

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

18# 

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

20# 

21from __future__ import annotations 

22 

23import re 

24from typing import IO 

25 

26from . import Image, ImageFile 

27 

28# XBM header 

29xbm_head = re.compile( 

30 rb"\s*#define[ \t]+.*_width[ \t]+(?P<width>[0-9]+)[\r\n]+" 

31 b"#define[ \t]+.*_height[ \t]+(?P<height>[0-9]+)[\r\n]+" 

32 b"(?P<hotspot>" 

33 b"#define[ \t]+[^_]*_x_hot[ \t]+(?P<xhot>[0-9]+)[\r\n]+" 

34 b"#define[ \t]+[^_]*_y_hot[ \t]+(?P<yhot>[0-9]+)[\r\n]+" 

35 b")?" 

36 rb"[\000-\377]*_bits\[]" 

37) 

38 

39 

40def _accept(prefix: bytes) -> bool: 

41 return prefix.lstrip().startswith(b"#define") 

42 

43 

44## 

45# Image plugin for X11 bitmaps. 

46 

47 

48class XbmImageFile(ImageFile.ImageFile): 

49 format = "XBM" 

50 format_description = "X11 Bitmap" 

51 

52 def _open(self) -> None: 

53 assert self.fp is not None 

54 

55 m = xbm_head.match(self.fp.read(512)) 

56 

57 if not m: 

58 msg = "not a XBM file" 

59 raise SyntaxError(msg) 

60 

61 xsize = int(m.group("width")) 

62 ysize = int(m.group("height")) 

63 

64 if m.group("hotspot"): 

65 self.info["hotspot"] = (int(m.group("xhot")), int(m.group("yhot"))) 

66 

67 self._mode = "1" 

68 self._size = xsize, ysize 

69 

70 self.tile = [ImageFile._Tile("xbm", (0, 0) + self.size, m.end())] 

71 

72 

73def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None: 

74 if im.mode != "1": 

75 msg = f"cannot write mode {im.mode} as XBM" 

76 raise OSError(msg) 

77 

78 fp.write(f"#define im_width {im.size[0]}\n".encode("ascii")) 

79 fp.write(f"#define im_height {im.size[1]}\n".encode("ascii")) 

80 

81 hotspot = im.encoderinfo.get("hotspot") 

82 if hotspot: 

83 fp.write(f"#define im_x_hot {hotspot[0]}\n".encode("ascii")) 

84 fp.write(f"#define im_y_hot {hotspot[1]}\n".encode("ascii")) 

85 

86 fp.write(b"static char im_bits[] = {\n") 

87 

88 ImageFile._save(im, fp, [ImageFile._Tile("xbm", (0, 0) + im.size)]) 

89 

90 fp.write(b"};\n") 

91 

92 

93Image.register_open(XbmImageFile.format, XbmImageFile, _accept) 

94Image.register_save(XbmImageFile.format, _save) 

95 

96Image.register_extension(XbmImageFile.format, ".xbm") 

97 

98Image.register_mime(XbmImageFile.format, "image/xbm")