Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.10/site-packages/pillow-11.1.0-py3.10-linux-x86_64.egg/PIL/WmfImagePlugin.py: 27%

1# 

2# The Python Imaging Library 

3# $Id$ 

4# 

5# WMF stub codec 

6# 

7# history: 

8# 1996-12-14 fl Created 

9# 2004-02-22 fl Turned into a stub driver 

10# 2004-02-23 fl Added EMF support 

11# 

12# Copyright (c) Secret Labs AB 1997-2004. All rights reserved. 

13# Copyright (c) Fredrik Lundh 1996. 

14# 

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

16# 

17# WMF/EMF reference documentation: 

18# https://winprotocoldoc.blob.core.windows.net/productionwindowsarchives/MS-WMF/[MS-WMF].pdf 

19# http://wvware.sourceforge.net/caolan/index.html 

20# http://wvware.sourceforge.net/caolan/ora-wmf.html 

21from __future__ import annotations 

22 

23from typing import IO 

24 

25from . import Image, ImageFile 

26from ._binary import i16le as word 

27from ._binary import si16le as short 

28from ._binary import si32le as _long 

29 

30_handler = None 

31 

32 

33def register_handler(handler: ImageFile.StubHandler | None) -> None: 

34 """ 

35 Install application-specific WMF image handler. 

36 

37 :param handler: Handler object. 

38 """ 

39 global _handler 

40 _handler = handler 

41 

42 

43if hasattr(Image.core, "drawwmf"): 

44 # install default handler (windows only) 

45 

46 class WmfHandler(ImageFile.StubHandler): 

47 def open(self, im: ImageFile.StubImageFile) -> None: 

48 im._mode = "RGB" 

49 self.bbox = im.info["wmf_bbox"] 

50 

51 def load(self, im: ImageFile.StubImageFile) -> Image.Image: 

52 im.fp.seek(0) # rewind 

53 return Image.frombytes( 

54 "RGB", 

55 im.size, 

56 Image.core.drawwmf(im.fp.read(), im.size, self.bbox), 

57 "raw", 

58 "BGR", 

59 (im.size[0] * 3 + 3) & -4, 

60 -1, 

61 ) 

62 

63 register_handler(WmfHandler()) 

64 

65# 

66# -------------------------------------------------------------------- 

67# Read WMF file 

68 

69 

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

71 return ( 

72 prefix[:6] == b"\xd7\xcd\xc6\x9a\x00\x00" or prefix[:4] == b"\x01\x00\x00\x00" 

73 ) 

74 

75 

76## 

77# Image plugin for Windows metafiles. 

78 

79 

80class WmfStubImageFile(ImageFile.StubImageFile): 

81 format = "WMF" 

82 format_description = "Windows Metafile" 

83 

84 def _open(self) -> None: 

85 self._inch = None 

86 

87 # check placable header 

88 s = self.fp.read(80) 

89 

90 if s[:6] == b"\xd7\xcd\xc6\x9a\x00\x00": 

91 # placeable windows metafile 

92 

93 # get units per inch 

94 self._inch = word(s, 14) 

95 if self._inch == 0: 

96 msg = "Invalid inch" 

97 raise ValueError(msg) 

98 

99 # get bounding box 

100 x0 = short(s, 6) 

101 y0 = short(s, 8) 

102 x1 = short(s, 10) 

103 y1 = short(s, 12) 

104 

105 # normalize size to 72 dots per inch 

106 self.info["dpi"] = 72 

107 size = ( 

108 (x1 - x0) * self.info["dpi"] // self._inch, 

109 (y1 - y0) * self.info["dpi"] // self._inch, 

110 ) 

111 

112 self.info["wmf_bbox"] = x0, y0, x1, y1 

113 

114 # sanity check (standard metafile header) 

115 if s[22:26] != b"\x01\x00\t\x00": 

116 msg = "Unsupported WMF file format" 

117 raise SyntaxError(msg) 

118 

119 elif s[:4] == b"\x01\x00\x00\x00" and s[40:44] == b" EMF": 

120 # enhanced metafile 

121 

122 # get bounding box 

123 x0 = _long(s, 8) 

124 y0 = _long(s, 12) 

125 x1 = _long(s, 16) 

126 y1 = _long(s, 20) 

127 

128 # get frame (in 0.01 millimeter units) 

129 frame = _long(s, 24), _long(s, 28), _long(s, 32), _long(s, 36) 

130 

131 size = x1 - x0, y1 - y0 

132 

133 # calculate dots per inch from bbox and frame 

134 xdpi = 2540.0 * (x1 - x0) / (frame[2] - frame[0]) 

135 ydpi = 2540.0 * (y1 - y0) / (frame[3] - frame[1]) 

136 

137 self.info["wmf_bbox"] = x0, y0, x1, y1 

138 

139 if xdpi == ydpi: 

140 self.info["dpi"] = xdpi 

141 else: 

142 self.info["dpi"] = xdpi, ydpi 

143 

144 else: 

145 msg = "Unsupported file format" 

146 raise SyntaxError(msg) 

147 

148 self._mode = "RGB" 

149 self._size = size 

150 

151 loader = self._load() 

152 if loader: 

153 loader.open(self) 

154 

155 def _load(self) -> ImageFile.StubHandler | None: 

156 return _handler 

157 

158 def load(self, dpi: int | None = None) -> Image.core.PixelAccess | None: 

159 if dpi is not None and self._inch is not None: 

160 self.info["dpi"] = dpi 

161 x0, y0, x1, y1 = self.info["wmf_bbox"] 

162 self._size = ( 

163 (x1 - x0) * self.info["dpi"] // self._inch, 

164 (y1 - y0) * self.info["dpi"] // self._inch, 

165 ) 

166 return super().load() 

167 

168 

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

170 if _handler is None or not hasattr(_handler, "save"): 

171 msg = "WMF save handler not installed" 

172 raise OSError(msg) 

173 _handler.save(im, fp, filename) 

174 

175 

176# 

177# -------------------------------------------------------------------- 

178# Registry stuff 

179 

180 

181Image.register_open(WmfStubImageFile.format, WmfStubImageFile, _accept) 

182Image.register_save(WmfStubImageFile.format, _save) 

183 

184Image.register_extensions(WmfStubImageFile.format, [".wmf", ".emf"])