Support HEIF image file (#1042)

imageio/config/extensions.py

@@ -1855,6 +1855,9 @@ def reset(self):
         extension=".btf",
         priority=["tifffile"],
     ),
+    FileExtension(
+        name="High Efficiency Image File Format", extension=".heic", priority=["pillow"]
+    ),
 ]
 extension_list.sort(key=lambda x: x.extension)
 

imageio/plugins/pillow.py

@@ -75,6 +75,14 @@ def __init__(self, request: Request) -> None:
 
         super().__init__(request)
 
+        # Register HEIF opener for Pillow
+        try:
+            from pillow_heif import register_heif_opener
+        except ImportError:
+            pass
+        else:
+            register_heif_opener()
+
         self._image: Image = None
         self.images_to_write = []
 

setup.py

@@ -99,6 +99,7 @@
     "swf": [],
     "tifffile": ["tifffile"],
     "pyav": ["av"],
+    "heif": ["pillow-heif"],
 }
 
 cpython_only_plugins = {

tests/test_pillow.py

@@ -705,3 +705,10 @@ def test_writable_output():
 
     frame = iio.imread(buffer, writeable_output=False, plugin="pillow")
     assert not frame.flags["WRITEABLE"]
+
+
+@pytest.mark.needs_internet
+def test_heif_remote():
+    url = "http://github.com/tigranbs/test-heic-images/raw/master/image4.heic"
+    im = iio.imread(url, plugin="pillow")
+    assert im.shape == (476, 700, 4)