Coverage for /var/srv/projects/api.amasfac.comuna18.com/tmp/venv/lib/python3.9/site-packages/PIL/Hdf5StubImagePlugin.py: 42%

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# 

11 

12from . import Image, ImageFile 

13 

14_handler = None 

15 

16 

17def register_handler(handler): 

18 """ 

19 Install application-specific HDF5 image handler. 

20 

21 :param handler: Handler object. 

22 """ 

23 global _handler 

24 _handler = handler 

25 

26 

27# -------------------------------------------------------------------- 

28# Image adapter 

29 

30 

31def _accept(prefix): 

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

33 

34 

35class HDF5StubImageFile(ImageFile.StubImageFile): 

36 

37 format = "HDF5" 

38 format_description = "HDF5" 

39 

40 def _open(self): 

41 

42 offset = self.fp.tell() 

43 

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

45 raise SyntaxError("Not an HDF file") 

46 

47 self.fp.seek(offset) 

48 

49 # make something up 

50 self.mode = "F" 

51 self._size = 1, 1 

52 

53 loader = self._load() 

54 if loader: 

55 loader.open(self) 

56 

57 def _load(self): 

58 return _handler 

59 

60 

61def _save(im, fp, filename): 

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

63 raise OSError("HDF5 save handler not installed") 

64 _handler.save(im, fp, filename) 

65 

66 

67# -------------------------------------------------------------------- 

68# Registry 

69 

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

71Image.register_save(HDF5StubImageFile.format, _save) 

72 

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