Spaces:
Running
Running
# | |
# The Python Imaging Library | |
# $Id$ | |
# | |
# HDF5 stub adapter | |
# | |
# Copyright (c) 2000-2003 by Fredrik Lundh | |
# | |
# See the README file for information on usage and redistribution. | |
# | |
from __future__ import annotations | |
from . import Image, ImageFile | |
_handler = None | |
def register_handler(handler): | |
""" | |
Install application-specific HDF5 image handler. | |
:param handler: Handler object. | |
""" | |
global _handler | |
_handler = handler | |
# -------------------------------------------------------------------- | |
# Image adapter | |
def _accept(prefix): | |
return prefix[:8] == b"\x89HDF\r\n\x1a\n" | |
class HDF5StubImageFile(ImageFile.StubImageFile): | |
format = "HDF5" | |
format_description = "HDF5" | |
def _open(self): | |
offset = self.fp.tell() | |
if not _accept(self.fp.read(8)): | |
msg = "Not an HDF file" | |
raise SyntaxError(msg) | |
self.fp.seek(offset) | |
# make something up | |
self._mode = "F" | |
self._size = 1, 1 | |
loader = self._load() | |
if loader: | |
loader.open(self) | |
def _load(self): | |
return _handler | |
def _save(im, fp, filename): | |
if _handler is None or not hasattr(_handler, "save"): | |
msg = "HDF5 save handler not installed" | |
raise OSError(msg) | |
_handler.save(im, fp, filename) | |
# -------------------------------------------------------------------- | |
# Registry | |
Image.register_open(HDF5StubImageFile.format, HDF5StubImageFile, _accept) | |
Image.register_save(HDF5StubImageFile.format, _save) | |
Image.register_extensions(HDF5StubImageFile.format, [".h5", ".hdf"]) | |