Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/pillow-10.4.0-py3.8-linux-x86_64.egg/PIL/Hdf5StubImagePlugin.py: 50%

Shortcuts on this page

r m x   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

32 statements  

1# 

2# The Python Imaging Library 

3# $Id$ 

4# 

5# HDF5 stub adapter 

6# 

7# Copyright (c) 2000-2003 by Fredrik Lundh 

8# 

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

10# 

11from __future__ import annotations 

12 

13from typing import IO 

14 

15from . import Image, ImageFile 

16 

17_handler = None 

18 

19 

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

21 """ 

22 Install application-specific HDF5 image handler. 

23 

24 :param handler: Handler object. 

25 """ 

26 global _handler 

27 _handler = handler 

28 

29 

30# -------------------------------------------------------------------- 

31# Image adapter 

32 

33 

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

35 return prefix[:8] == b"\x89HDF\r\n\x1a\n" 

36 

37 

38class HDF5StubImageFile(ImageFile.StubImageFile): 

39 format = "HDF5" 

40 format_description = "HDF5" 

41 

42 def _open(self) -> None: 

43 offset = self.fp.tell() 

44 

45 if not _accept(self.fp.read(8)): 

46 msg = "Not an HDF file" 

47 raise SyntaxError(msg) 

48 

49 self.fp.seek(offset) 

50 

51 # make something up 

52 self._mode = "F" 

53 self._size = 1, 1 

54 

55 loader = self._load() 

56 if loader: 

57 loader.open(self) 

58 

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

60 return _handler 

61 

62 

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

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

65 msg = "HDF5 save handler not installed" 

66 raise OSError(msg) 

67 _handler.save(im, fp, filename) 

68 

69 

70# -------------------------------------------------------------------- 

71# Registry 

72 

73Image.register_open(HDF5StubImageFile.format, HDF5StubImageFile, _accept) 

74Image.register_save(HDF5StubImageFile.format, _save) 

75 

76Image.register_extensions(HDF5StubImageFile.format, [".h5", ".hdf"])