pillow-avif-plugin 1.4.3__tar.gz → 1.4.4__tar.gz

{pillow-avif-plugin-1.4.3 → pillow-avif-plugin-1.4.4}/PKG-INFO

@@ -1,13 +1,12 @@
 Metadata-Version: 2.1
 Name: pillow-avif-plugin
-Version: 1.4.3
+Version: 1.4.4
 Summary: A pillow plugin that adds avif support via libavif
 Home-page: https://github.com/fdintino/pillow-avif-plugin/
+Download-URL: https://github.com/fdintino/pillow-avif-plugin/releases
 Author: Frankie Dintino
 Author-email: fdintino@theatlantic.com
 License: MIT License
-Download-URL: https://github.com/fdintino/pillow-avif-plugin/releases
-Platform: UNKNOWN
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Environment :: Web Environment
 Classifier: Intended Audience :: Developers
@@ -35,5 +34,3 @@ License-File: LICENSE
 This is a plugin that adds support for AVIF files until official support has been added (see [this pull request](https://github.com/python-pillow/Pillow/pull/5201)).
 
 To register this plugin with pillow you will need to add `import pillow_avif` somewhere in your application.
-
-

pillow-avif-plugin-1.4.4/src/pillow_avif/AvifImagePlugin.py

@@ -0,0 +1,282 @@
+from __future__ import division
+
+from io import BytesIO
+import sys
+
+from PIL import ExifTags, Image, ImageFile
+
+try:
+    from pillow_avif import _avif
+
+    SUPPORTED = True
+except ImportError:
+    SUPPORTED = False
+
+# Decoder options as module globals, until there is a way to pass parameters
+# to Image.open (see https://github.com/python-pillow/Pillow/issues/569)
+DECODE_CODEC_CHOICE = "auto"
+CHROMA_UPSAMPLING = "auto"
+DEFAULT_MAX_THREADS = 0
+
+_VALID_AVIF_MODES = {"RGB", "RGBA"}
+
+
+if sys.version_info[0] == 2:
+    text_type = unicode  # noqa
+else:
+    text_type = str
+
+
+def _accept(prefix):
+    if prefix[4:8] != b"ftyp":
+        return
+    coding_brands = (b"avif", b"avis")
+    container_brands = (b"mif1", b"msf1")
+    major_brand = prefix[8:12]
+    if major_brand in coding_brands:
+        if not SUPPORTED:
+            return (
+                "image file could not be identified because AVIF "
+                "support not installed"
+            )
+        return True
+    if major_brand in container_brands:
+        # We accept files with AVIF container brands; we can't yet know if
+        # the ftyp box has the correct compatible brands, but if it doesn't
+        # then the plugin will raise a SyntaxError which Pillow will catch
+        # before moving on to the next plugin that accepts the file.
+        #
+        # Also, because this file might not actually be an AVIF file, we
+        # don't raise an error if AVIF support isn't properly compiled.
+        return True
+
+
+class AvifImageFile(ImageFile.ImageFile):
+    format = "AVIF"
+    format_description = "AVIF image"
+    __loaded = -1
+    __frame = 0
+
+    def load_seek(self, pos):
+        pass
+
+    def _open(self):
+        self._decoder = _avif.AvifDecoder(
+            self.fp.read(), DECODE_CODEC_CHOICE, CHROMA_UPSAMPLING, DEFAULT_MAX_THREADS
+        )
+
+        # Get info from decoder
+        width, height, n_frames, mode, icc, exif, xmp = self._decoder.get_info()
+        self._size = width, height
+        self.n_frames = n_frames
+        self.is_animated = self.n_frames > 1
+        try:
+            self.mode = self.rawmode = mode
+        except AttributeError:
+            self._mode = self.rawmode = mode
+        self.tile = []
+
+        if icc:
+            self.info["icc_profile"] = icc
+        if exif:
+            self.info["exif"] = exif
+        if xmp:
+            self.info["xmp"] = xmp
+
+    def seek(self, frame):
+        if not self._seek_check(frame):
+            return
+
+        self.__frame = frame
+
+    def load(self):
+        if self.__loaded != self.__frame:
+            # We need to load the image data for this frame
+            data, timescale, tsp_in_ts, dur_in_ts = self._decoder.get_frame(
+                self.__frame
+            )
+            timestamp = round(1000 * (tsp_in_ts / timescale))
+            duration = round(1000 * (dur_in_ts / timescale))
+            self.info["timestamp"] = timestamp
+            self.info["duration"] = duration
+            self.__loaded = self.__frame
+
+            # Set tile
+            if self.fp and self._exclusive_fp:
+                self.fp.close()
+            self.fp = BytesIO(data)
+            self.tile = [("raw", (0, 0) + self.size, 0, self.rawmode)]
+
+        return super(AvifImageFile, self).load()
+
+    def tell(self):
+        return self.__frame
+
+
+def _save_all(im, fp, filename):
+    _save(im, fp, filename, save_all=True)
+
+
+def _save(im, fp, filename, save_all=False):
+    info = im.encoderinfo.copy()
+    if save_all:
+        append_images = list(info.get("append_images", []))
+    else:
+        append_images = []
+
+    total = 0
+    for ims in [im] + append_images:
+        total += getattr(ims, "n_frames", 1)
+
+    is_single_frame = total == 1
+
+    qmin = info.get("qmin", -1)
+    qmax = info.get("qmax", -1)
+    quality = info.get("quality", 75)
+    if not isinstance(quality, int) or quality < 0 or quality > 100:
+        raise ValueError("Invalid quality setting")
+
+    duration = info.get("duration", 0)
+    subsampling = info.get("subsampling", "4:2:0")
+    speed = info.get("speed", 6)
+    max_threads = info.get("max_threads", DEFAULT_MAX_THREADS)
+    codec = info.get("codec", "auto")
+    range_ = info.get("range", "full")
+    tile_rows_log2 = info.get("tile_rows", 0)
+    tile_cols_log2 = info.get("tile_cols", 0)
+    alpha_premultiplied = bool(info.get("alpha_premultiplied", False))
+    autotiling = bool(info.get("autotiling", tile_rows_log2 == tile_cols_log2 == 0))
+
+    icc_profile = info.get("icc_profile", im.info.get("icc_profile"))
+    exif = info.get("exif", im.info.get("exif"))
+    if isinstance(exif, Image.Exif):
+        exif = exif.tobytes()
+
+    exif_orientation = 0
+    if exif:
+        exif_data = Image.Exif()
+        try:
+            exif_data.load(exif)
+        except SyntaxError:
+            pass
+        else:
+            orientation_tag = next(
+                k for k, v in ExifTags.TAGS.items() if v == "Orientation"
+            )
+            exif_orientation = exif_data.get(orientation_tag) or 0
+
+    xmp = info.get("xmp", im.info.get("xmp") or im.info.get("XML:com.adobe.xmp"))
+
+    if isinstance(xmp, text_type):
+        xmp = xmp.encode("utf-8")
+
+    advanced = info.get("advanced")
+    if isinstance(advanced, dict):
+        advanced = tuple([k, v] for (k, v) in advanced.items())
+    if advanced is not None:
+        try:
+            advanced = tuple(advanced)
+        except TypeError:
+            invalid = True
+        else:
+            invalid = all(isinstance(v, tuple) and len(v) == 2 for v in advanced)
+        if invalid:
+            raise ValueError(
+                "advanced codec options must be a dict of key-value string "
+                "pairs or a series of key-value two-tuples"
+            )
+        advanced = tuple(
+            [(str(k).encode("utf-8"), str(v).encode("utf-8")) for k, v in advanced]
+        )
+
+    # Setup the AVIF encoder
+    enc = _avif.AvifEncoder(
+        im.size[0],
+        im.size[1],
+        subsampling,
+        qmin,
+        qmax,
+        quality,
+        speed,
+        max_threads,
+        codec,
+        range_,
+        tile_rows_log2,
+        tile_cols_log2,
+        alpha_premultiplied,
+        autotiling,
+        icc_profile or b"",
+        exif or b"",
+        exif_orientation,
+        xmp or b"",
+        advanced,
+    )
+
+    # Add each frame
+    frame_idx = 0
+    frame_dur = 0
+    cur_idx = im.tell()
+    try:
+        for ims in [im] + append_images:
+            # Get # of frames in this image
+            nfr = getattr(ims, "n_frames", 1)
+
+            for idx in range(nfr):
+                ims.seek(idx)
+                ims.load()
+
+                # Make sure image mode is supported
+                frame = ims
+                rawmode = ims.mode
+                if ims.mode not in _VALID_AVIF_MODES:
+                    alpha = (
+                        "A" in ims.mode
+                        or "a" in ims.mode
+                        or (ims.mode == "P" and "A" in ims.im.getpalettemode())
+                        or (
+                            ims.mode == "P"
+                            and ims.info.get("transparency", None) is not None
+                        )
+                    )
+                    rawmode = "RGBA" if alpha else "RGB"
+                    frame = ims.convert(rawmode)
+
+                # Update frame duration
+                if isinstance(duration, (list, tuple)):
+                    frame_dur = duration[frame_idx]
+                else:
+                    frame_dur = duration
+
+                # Append the frame to the animation encoder
+                enc.add(
+                    frame.tobytes("raw", rawmode),
+                    frame_dur,
+                    frame.size[0],
+                    frame.size[1],
+                    rawmode,
+                    is_single_frame,
+                )
+
+                # Update frame index
+                frame_idx += 1
+
+                if not save_all:
+                    break
+
+    finally:
+        im.seek(cur_idx)
+
+    # Get the final output from the encoder
+    data = enc.finish()
+    if data is None:
+        raise OSError("cannot write file as AVIF (encoder returned None)")
+
+    fp.write(data)
+
+
+Image.register_open(AvifImageFile.format, AvifImageFile, _accept)
+if SUPPORTED:
+    Image.register_save(AvifImageFile.format, _save)
+    Image.register_save_all(AvifImageFile.format, _save_all)
+    Image.register_extensions(AvifImageFile.format, [".avif", ".avifs"])
+    Image.register_mime(AvifImageFile.format, "image/avif")

pillow-avif-plugin-1.4.3/src/pillow_avif/AvifImagePlugin.py → pillow-avif-plugin-1.4.4/src/pillow_avif/AvifImagePlugin.py.orig

@@ -16,6 +16,11 @@ except ImportError:
 # to Image.open (see https://github.com/python-pillow/Pillow/issues/569)
 DECODE_CODEC_CHOICE = "auto"
 CHROMA_UPSAMPLING = "auto"
+<<<<<<< HEAD
+DEFAULT_MAX_THREADS = 0
+=======
+DECODE_MAX_THREADS = 0
+>>>>>>> 1e2fa65 (Let users pass max_threads manually as an argument)
 
 _VALID_AVIF_MODES = {"RGB", "RGBA"}
 
@@ -61,7 +66,7 @@ class AvifImageFile(ImageFile.ImageFile):
 
     def _open(self):
         self._decoder = _avif.AvifDecoder(
-            self.fp.read(), DECODE_CODEC_CHOICE, CHROMA_UPSAMPLING
+            self.fp.read(), DECODE_CODEC_CHOICE, CHROMA_UPSAMPLING, DECODE_MAX_THREADS
         )
 
         # Get info from decoder
@@ -138,6 +143,7 @@ def _save(im, fp, filename, save_all=False):
     duration = info.get("duration", 0)
     subsampling = info.get("subsampling", "4:2:0")
     speed = info.get("speed", 6)
+    max_threads = info.get("max_threads", DEFAULT_MAX_THREADS)
     codec = info.get("codec", "auto")
     range_ = info.get("range", "full")
     tile_rows_log2 = info.get("tile_rows", 0)
@@ -196,6 +202,7 @@ def _save(im, fp, filename, save_all=False):
         qmax,
         quality,
         speed,
+        max_threads,
         codec,
         range_,
         tile_rows_log2,

{pillow-avif-plugin-1.4.3 → pillow-avif-plugin-1.4.4}/src/pillow_avif/__init__.py

@@ -2,4 +2,4 @@ from . import AvifImagePlugin
 
 
 __all__ = ["AvifImagePlugin"]
-__version__ = "1.4.3"
+__version__ = "1.4.4"

{pillow-avif-plugin-1.4.3 → pillow-avif-plugin-1.4.4}/src/pillow_avif/_avif.c

@@ -46,7 +46,7 @@ typedef struct {
 
 static PyTypeObject AvifDecoder_Type;
 
-static int max_threads = 0;
+static int default_max_threads = 0;
 
 static void
 init_max_threads(void) {
@@ -85,7 +85,7 @@ init_max_threads(void) {
         goto error;
     }
 
-    max_threads = (int)num_cpus;
+    default_max_threads = (int)num_cpus;
 
 done:
     Py_XDECREF(os);
@@ -321,6 +321,7 @@ AvifEncoderNew(PyObject *self_, PyObject *args) {
     int quality = 75;
     int speed = 8;
     int exif_orientation = 0;
+    int max_threads = default_max_threads;
     PyObject *icc_bytes;
     PyObject *exif_bytes;
     PyObject *xmp_bytes;
@@ -336,7 +337,7 @@ AvifEncoderNew(PyObject *self_, PyObject *args) {
 
     if (!PyArg_ParseTuple(
             args,
-            "IIsiiiissiiOOSSiSO",
+            "IIsiiiiissiiOOSSiSO",
             &width,
             &height,
             &subsampling,
@@ -344,6 +345,7 @@ AvifEncoderNew(PyObject *self_, PyObject *args) {
             &qmax,
             &quality,
             &speed,
+            &max_threads,
             &codec,
             &range,
             &tile_rows_log2,
@@ -445,7 +447,10 @@ AvifEncoderNew(PyObject *self_, PyObject *args) {
         encoder = avifEncoderCreate();
 
         if (max_threads == 0) {
-            init_max_threads();
+            if (default_max_threads == 0) {
+                init_max_threads();
+            }
+            max_threads = default_max_threads;
         }
 
         int is_aom_encode = strcmp(codec, "aom") == 0 ||
@@ -726,10 +731,11 @@ AvifDecoderNew(PyObject *self_, PyObject *args) {
     char *codec_str;
     avifCodecChoice codec;
     avifChromaUpsampling upsampling;
+    int max_threads = 0;
 
     avifResult result;
 
-    if (!PyArg_ParseTuple(args, "Sss", &avif_bytes, &codec_str, &upsampling_str)) {
+    if (!PyArg_ParseTuple(args, "Sssi", &avif_bytes, &codec_str, &upsampling_str, &max_threads)) {
         return NULL;
     }
 
@@ -778,9 +784,20 @@ AvifDecoderNew(PyObject *self_, PyObject *args) {
     self->decoder = avifDecoderCreate();
 #if AVIF_VERSION >= 80400
     if (max_threads == 0) {
-        init_max_threads();
+        if (default_max_threads == 0) {
+            init_max_threads();
+        }
+        max_threads = default_max_threads;
     }
     self->decoder->maxThreads = max_threads;
+#endif
+#if AVIF_VERSION >= 90200
+    // Turn off libavif's 'clap' (clean aperture) property validation.
+    self->decoder->strictFlags &= ~AVIF_STRICT_CLAP_VALID;
+    // Allow the PixelInformationProperty ('pixi') to be missing in AV1 image
+    // items. libheif v1.11.0 and older does not add the 'pixi' item property to
+    // AV1 image items.
+    self->decoder->strictFlags &= ~AVIF_STRICT_PIXI_REQUIRED;
 #endif
     self->decoder->codecChoice = codec;
 

{pillow-avif-plugin-1.4.3 → pillow-avif-plugin-1.4.4}/src/pillow_avif_plugin.egg-info/PKG-INFO

@@ -1,13 +1,12 @@
 Metadata-Version: 2.1
 Name: pillow-avif-plugin
-Version: 1.4.3
+Version: 1.4.4
 Summary: A pillow plugin that adds avif support via libavif
 Home-page: https://github.com/fdintino/pillow-avif-plugin/
+Download-URL: https://github.com/fdintino/pillow-avif-plugin/releases
 Author: Frankie Dintino
 Author-email: fdintino@theatlantic.com
 License: MIT License
-Download-URL: https://github.com/fdintino/pillow-avif-plugin/releases
-Platform: UNKNOWN
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Environment :: Web Environment
 Classifier: Intended Audience :: Developers
@@ -35,5 +34,3 @@ License-File: LICENSE
 This is a plugin that adds support for AVIF files until official support has been added (see [this pull request](https://github.com/python-pillow/Pillow/pull/5201)).
 
 To register this plugin with pillow you will need to add `import pillow_avif` somewhere in your application.
-
-

{pillow-avif-plugin-1.4.3 → pillow-avif-plugin-1.4.4}/src/pillow_avif_plugin.egg-info/SOURCES.txt

@@ -6,10 +6,12 @@ setup.cfg
 setup.py
 tox.ini
 src/pillow_avif/AvifImagePlugin.py
+src/pillow_avif/AvifImagePlugin.py.orig
 src/pillow_avif/__init__.py
 src/pillow_avif/_avif.c
 src/pillow_avif_plugin.egg-info/PKG-INFO
 src/pillow_avif_plugin.egg-info/SOURCES.txt
 src/pillow_avif_plugin.egg-info/dependency_links.txt
-src/pillow_avif_plugin.egg-info/not-zip-safe
-src/pillow_avif_plugin.egg-info/top_level.txt
+src/pillow_avif_plugin.egg-info/top_level.txt
+src/pillow_avif_plugin.egg-info/zip-safe
+tests/test_file_avif.py

pillow-avif-plugin-1.4.4/tests/test_file_avif.py

@@ -0,0 +1,783 @@
+import os
+import xml.etree.ElementTree
+from contextlib import contextmanager
+from io import BytesIO
+import warnings
+
+try:
+    from os import cpu_count
+except ImportError:
+    from multiprocessing import cpu_count
+
+import pytest
+
+from PIL import Image, ImageDraw
+from pillow_avif import AvifImagePlugin
+
+from .helper import (
+    PillowLeakTestCase,
+    assert_image,
+    assert_image_similar,
+    assert_image_similar_tofile,
+    hopper,
+    has_alpha_premultiplied,
+)
+
+from pillow_avif import _avif
+
+try:
+    from PIL import UnidentifiedImageError
+except ImportError:
+    UnidentifiedImageError = None
+
+
+CURR_DIR = os.path.dirname(os.path.dirname(__file__))
+TEST_AVIF_FILE = "%s/tests/images/hopper.avif" % CURR_DIR
+
+
+def assert_xmp_orientation(xmp, expected):
+    assert isinstance(xmp, bytes)
+    root = xml.etree.ElementTree.fromstring(xmp)
+    orientation = None
+    for elem in root.iter():
+        if elem.tag.endswith("}Description"):
+            orientation = elem.attrib.get("{http://ns.adobe.com/tiff/1.0/}Orientation")
+            if orientation:
+                orientation = int(orientation)
+                break
+    assert orientation == expected
+
+
+def roundtrip(im, **options):
+    out = BytesIO()
+    im.save(out, "AVIF", **options)
+    out.seek(0)
+    return Image.open(out)
+
+
+def skip_unless_avif_decoder(codec_name):
+    reason = "%s decode not available" % codec_name
+    return pytest.mark.skipif(
+        not _avif or not _avif.decoder_codec_available(codec_name), reason=reason
+    )
+
+
+def skip_unless_avif_encoder(codec_name):
+    reason = "%s encode not available" % codec_name
+    return pytest.mark.skipif(
+        not _avif or not _avif.encoder_codec_available(codec_name), reason=reason
+    )
+
+
+def is_docker_qemu():
+    try:
+        init_proc_exe = os.readlink("/proc/1/exe")
+    except:  # noqa: E722
+        return False
+    else:
+        return "qemu" in init_proc_exe
+
+
+def skip_unless_avif_version_gte(version):
+    if not _avif:
+        reason = "AVIF unavailable"
+        should_skip = True
+    else:
+        version_str = ".".join([str(v) for v in version])
+        reason = "%s < %s" % (_avif.libavif_version, version_str)
+        should_skip = _avif.VERSION < version
+    return pytest.mark.skipif(should_skip, reason=reason)
+
+
+def skip_unless_avif_version_lt(version):
+    if not _avif:
+        reason = "AVIF unavailable"
+        should_skip = True
+    else:
+        version_str = ".".join([str(v) for v in version])
+        reason = "%s > %s" % (_avif.libavif_version, version_str)
+        should_skip = _avif.VERSION >= version
+    return pytest.mark.skipif(should_skip, reason=reason)
+
+
+class TestUnsupportedAvif:
+    def test_unsupported(self):
+        AvifImagePlugin.SUPPORTED = False
+
+        try:
+            file_path = "%s/tests/images/hopper.avif" % CURR_DIR
+            if UnidentifiedImageError:
+                pytest.warns(
+                    UserWarning,
+                    lambda: pytest.raises(
+                        UnidentifiedImageError, Image.open, file_path
+                    ),
+                )
+            else:
+                with pytest.raises(IOError):
+                    Image.open(file_path)
+        finally:
+            AvifImagePlugin.SUPPORTED = True
+
+
+class TestFileAvif:
+    def test_version(self):
+        _avif.AvifCodecVersions()
+
+    def test_read(self):
+        """
+        Can we read an AVIF file without error?
+        Does it have the bits we expect?
+        """
+
+        with Image.open("%s/tests/images/hopper.avif" % CURR_DIR) as image:
+            assert image.mode == "RGB"
+            assert image.size == (128, 128)
+            assert image.format == "AVIF"
+            assert image.get_format_mimetype() == "image/avif"
+            image.load()
+            image.getdata()
+
+            # generated with:
+            # avifdec hopper.avif hopper_avif_write.png
+            assert_image_similar_tofile(
+                image, "%s/tests/images/hopper_avif_write.png" % CURR_DIR, 12.0
+            )
+
+    def _roundtrip(self, tmp_path, mode, epsilon, args={}):
+        temp_file = str(tmp_path / "temp.avif")
+
+        hopper(mode).save(temp_file, **args)
+        with Image.open(temp_file) as image:
+            assert image.mode == "RGB"
+            assert image.size == (128, 128)
+            assert image.format == "AVIF"
+            image.load()
+            image.getdata()
+
+            if mode == "RGB":
+                # avifdec hopper.avif avif/hopper_avif_write.png
+                assert_image_similar_tofile(
+                    image, "%s/tests/images/hopper_avif_write.png" % CURR_DIR, 12.0
+                )
+
+            # This test asserts that the images are similar. If the average pixel
+            # difference between the two images is less than the epsilon value,
+            # then we're going to accept that it's a reasonable lossy version of
+            # the image.
+            target = hopper(mode)
+            if mode != "RGB":
+                target = target.convert("RGB")
+            assert_image_similar(image, target, epsilon)
+
+    def test_write_rgb(self, tmp_path):
+        """
+        Can we write a RGB mode file to avif without error?
+        Does it have the bits we expect?
+        """
+
+        self._roundtrip(tmp_path, "RGB", 12.5)
+
+    def test_AvifEncoder_with_invalid_args(self):
+        """
+        Calling encoder functions with no arguments should result in an error.
+        """
+        with pytest.raises(TypeError):
+            _avif.AvifEncoder()
+
+    def test_AvifDecoder_with_invalid_args(self):
+        """
+        Calling decoder functions with no arguments should result in an error.
+        """
+        with pytest.raises(TypeError):
+            _avif.AvifDecoder()
+
+    @pytest.mark.parametrize("major_brand", [b"avif", b"avis", b"mif1", b"msf1"])
+    def test_accept_ftyp_brands(self, major_brand):
+        data = b"\x00\x00\x00\x1cftyp%s\x00\x00\x00\x00" % major_brand
+        assert AvifImagePlugin._accept(data) is True
+
+    def test_no_resource_warning(self, tmp_path):
+        with Image.open(TEST_AVIF_FILE) as image:
+            temp_file = str(tmp_path / "temp.avif")
+            with warnings.catch_warnings():
+                warnings.simplefilter("error")
+                image.save(temp_file)
+
+    def test_file_pointer_could_be_reused(self):
+        with open(TEST_AVIF_FILE, "rb") as blob:
+            Image.open(blob).load()
+            Image.open(blob).load()
+
+    def test_background_from_gif(self, tmp_path):
+        with Image.open("%s/tests/images/chi.gif" % CURR_DIR) as im:
+            original_value = im.convert("RGB").getpixel((1, 1))
+
+            # Save as AVIF
+            out_avif = str(tmp_path / "temp.avif")
+            im.save(out_avif, save_all=True)
+
+        # Save as GIF
+        out_gif = str(tmp_path / "temp.gif")
+        Image.open(out_avif).save(out_gif)
+
+        with Image.open(out_gif) as reread:
+            reread_value = reread.convert("RGB").getpixel((1, 1))
+        difference = sum(
+            [abs(original_value[i] - reread_value[i]) for i in range(0, 3)]
+        )
+        assert difference < 5
+
+    def test_save_single_frame(self, tmp_path):
+        temp_file = str(tmp_path / "temp.avif")
+        with Image.open("%s/tests/images/chi.gif" % CURR_DIR) as im:
+            # Save as AVIF
+            im.save(temp_file)
+        with Image.open(temp_file) as im:
+            assert im.n_frames == 1
+
+    def test_invalid_file(self):
+        invalid_file = "tests/images/flower.jpg"
+
+        with pytest.raises(SyntaxError):
+            AvifImagePlugin.AvifImageFile(invalid_file)
+
+    def test_load_transparent_rgb(self):
+        test_file = "tests/images/transparency.avif"
+        with Image.open(test_file) as im:
+            assert_image(im, "RGBA", (64, 64))
+
+            # image has 876 transparent pixels
+            assert im.getchannel("A").getcolors()[0][0] == 876
+
+    def test_save_transparent(self, tmp_path):
+        im = Image.new("RGBA", (10, 10), (0, 0, 0, 0))
+        assert im.getcolors() == [(100, (0, 0, 0, 0))]
+
+        test_file = str(tmp_path / "temp.avif")
+        im.save(test_file)
+
+        # check if saved image contains same transparency
+        with Image.open(test_file) as im:
+            assert_image(im, "RGBA", (10, 10))
+            assert im.getcolors() == [(100, (0, 0, 0, 0))]
+
+    def test_save_icc_profile(self):
+        with Image.open("tests/images/icc_profile_none.avif") as im:
+            assert im.info.get("icc_profile") is None
+
+            with Image.open("tests/images/icc_profile.avif") as with_icc:
+                expected_icc = with_icc.info.get("icc_profile")
+                assert expected_icc is not None
+
+                im = roundtrip(im, icc_profile=expected_icc)
+                assert im.info["icc_profile"] == expected_icc
+
+    def test_discard_icc_profile(self):
+        with Image.open("tests/images/icc_profile.avif") as im:
+            im = roundtrip(im, icc_profile=None)
+        assert "icc_profile" not in im.info
+
+    def test_roundtrip_icc_profile(self):
+        with Image.open("tests/images/icc_profile.avif") as im:
+            expected_icc = im.info["icc_profile"]
+
+            im = roundtrip(im)
+        assert im.info["icc_profile"] == expected_icc
+
+    def test_roundtrip_no_icc_profile(self):
+        with Image.open("tests/images/icc_profile_none.avif") as im:
+            assert im.info.get("icc_profile") is None
+
+            im = roundtrip(im)
+        assert "icc_profile" not in im.info
+
+    def test_exif(self):
+        # With an EXIF chunk
+        with Image.open("tests/images/exif.avif") as im:
+            exif = im.getexif()
+        assert exif[274] == 1
+
+    def test_exif_save(self, tmp_path):
+        with Image.open("tests/images/exif.avif") as im:
+            test_file = str(tmp_path / "temp.avif")
+            im.save(test_file)
+
+        with Image.open(test_file) as reloaded:
+            exif = reloaded.getexif()
+        assert exif[274] == 1
+
+    def test_exif_obj_argument(self, tmp_path):
+        exif = Image.Exif()
+        exif[274] = 1
+        exif_data = exif.tobytes()
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            im.save(test_file, exif=exif)
+
+        with Image.open(test_file) as reloaded:
+            assert reloaded.info["exif"] == exif_data
+
+    def test_exif_bytes_argument(self, tmp_path):
+        exif = Image.Exif()
+        exif[274] = 1
+        exif_data = exif.tobytes()
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            im.save(test_file, exif=exif_data)
+
+        with Image.open(test_file) as reloaded:
+            assert reloaded.info["exif"] == exif_data
+
+    def test_exif_invalid(self, tmp_path):
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            with pytest.raises(ValueError):
+                im.save(test_file, exif=b"invalid")
+
+    def test_xmp(self):
+        with Image.open("tests/images/xmp_tags_orientation.avif") as im:
+            xmp = im.info.get("xmp")
+        assert_xmp_orientation(xmp, 3)
+
+    def test_xmp_save(self, tmp_path):
+        with Image.open("tests/images/xmp_tags_orientation.avif") as im:
+            test_file = str(tmp_path / "temp.avif")
+            im.save(test_file)
+
+        with Image.open(test_file) as reloaded:
+            xmp = reloaded.info.get("xmp")
+        assert_xmp_orientation(xmp, 3)
+
+    def test_xmp_save_from_png(self, tmp_path):
+        with Image.open("tests/images/xmp_tags_orientation.png") as im:
+            test_file = str(tmp_path / "temp.avif")
+            im.save(test_file)
+
+        with Image.open(test_file) as reloaded:
+            xmp = reloaded.info.get("xmp")
+        assert_xmp_orientation(xmp, 3)
+
+    def test_xmp_save_argument(self, tmp_path):
+        xmp_arg = "\n".join(
+            [
+                '<?xpacket begin="" id="W5M0MpCehiHzreSzNTczkc9d"?>',
+                '<x:xmpmeta xmlns:x="adobe:ns:meta/">',
+                ' <rdf:RDF xmlns:rdf="http://www.w3.org/1999/02/22-rdf-syntax-ns#">',
+                '  <rdf:Description rdf:about=""',
+                '    xmlns:tiff="http://ns.adobe.com/tiff/1.0/"',
+                '   tiff:Orientation="1"/>',
+                " </rdf:RDF>",
+                "</x:xmpmeta>",
+                '<?xpacket end="r"?>',
+            ]
+        )
+        with Image.open("tests/images/hopper.avif") as im:
+            test_file = str(tmp_path / "temp.avif")
+            im.save(test_file, xmp=xmp_arg)
+
+        with Image.open(test_file) as reloaded:
+            xmp = reloaded.info.get("xmp")
+        assert_xmp_orientation(xmp, 1)
+
+    def test_tell(self):
+        with Image.open(TEST_AVIF_FILE) as im:
+            assert im.tell() == 0
+
+    def test_seek(self):
+        with Image.open(TEST_AVIF_FILE) as im:
+            im.seek(0)
+
+            with pytest.raises(EOFError):
+                im.seek(1)
+
+    @pytest.mark.parametrize("subsampling", ["4:4:4", "4:2:2", "4:0:0"])
+    def test_encoder_subsampling(self, tmp_path, subsampling):
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            im.save(test_file, subsampling=subsampling)
+
+    def test_encoder_subsampling_invalid(self, tmp_path):
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            with pytest.raises(ValueError):
+                im.save(test_file, subsampling="foo")
+
+    def test_encoder_range(self, tmp_path):
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            im.save(test_file, range="limited")
+
+    def test_encoder_range_invalid(self, tmp_path):
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            with pytest.raises(ValueError):
+                im.save(test_file, range="foo")
+
+    @skip_unless_avif_encoder("aom")
+    def test_encoder_codec_param(self, tmp_path):
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            im.save(test_file, codec="aom")
+
+    def test_encoder_codec_invalid(self, tmp_path):
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            with pytest.raises(ValueError):
+                im.save(test_file, codec="foo")
+
+    @skip_unless_avif_decoder("dav1d")
+    def test_encoder_codec_cannot_encode(self, tmp_path):
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            with pytest.raises(ValueError):
+                im.save(test_file, codec="dav1d")
+
+    @skip_unless_avif_encoder("aom")
+    @skip_unless_avif_version_gte((0, 8, 2))
+    def test_encoder_advanced_codec_options(self):
+        with Image.open(TEST_AVIF_FILE) as im:
+            ctrl_buf = BytesIO()
+            im.save(ctrl_buf, "AVIF", codec="aom")
+            test_buf = BytesIO()
+            im.save(
+                test_buf,
+                "AVIF",
+                codec="aom",
+                advanced={
+                    "aq-mode": "1",
+                    "enable-chroma-deltaq": "1",
+                },
+            )
+            assert ctrl_buf.getvalue() != test_buf.getvalue()
+
+    @skip_unless_avif_encoder("aom")
+    @skip_unless_avif_version_gte((0, 8, 2))
+    @pytest.mark.parametrize("val", [{"foo": "bar"}, 1234])
+    def test_encoder_advanced_codec_options_invalid(self, tmp_path, val):
+        with Image.open(TEST_AVIF_FILE) as im:
+            test_file = str(tmp_path / "temp.avif")
+            with pytest.raises(ValueError):
+                im.save(test_file, codec="aom", advanced=val)
+
+    @skip_unless_avif_decoder("aom")
+    def test_decoder_codec_param(self):
+        AvifImagePlugin.DECODE_CODEC_CHOICE = "aom"
+        try:
+            with Image.open(TEST_AVIF_FILE) as im:
+                assert im.size == (128, 128)
+        finally:
+            AvifImagePlugin.DECODE_CODEC_CHOICE = "auto"
+
+    @skip_unless_avif_encoder("rav1e")
+    def test_decoder_codec_cannot_decode(self, tmp_path):
+        AvifImagePlugin.DECODE_CODEC_CHOICE = "rav1e"
+        try:
+            with pytest.raises(ValueError):
+                with Image.open(TEST_AVIF_FILE):
+                    pass
+        finally:
+            AvifImagePlugin.DECODE_CODEC_CHOICE = "auto"
+
+    def test_decoder_codec_invalid(self):
+        AvifImagePlugin.DECODE_CODEC_CHOICE = "foo"
+        try:
+            with pytest.raises(ValueError):
+                with Image.open(TEST_AVIF_FILE):
+                    pass
+        finally:
+            AvifImagePlugin.DECODE_CODEC_CHOICE = "auto"
+
+    @skip_unless_avif_encoder("aom")
+    def test_encoder_codec_available(self):
+        assert _avif.encoder_codec_available("aom") is True
+
+    def test_encoder_codec_available_bad_params(self):
+        with pytest.raises(TypeError):
+            _avif.encoder_codec_available()
+
+    @skip_unless_avif_encoder("dav1d")
+    def test_encoder_codec_available_cannot_decode(self):
+        assert _avif.encoder_codec_available("dav1d") is False
+
+    def test_encoder_codec_available_invalid(self):
+        assert _avif.encoder_codec_available("foo") is False
+
+    @skip_unless_avif_version_lt((1, 0, 0))
+    @pytest.mark.parametrize(
+        "quality,expected_qminmax",
+        [
+            [0, (63, 63)],
+            [100, (0, 0)],
+            [90, (0, 10)],
+            [None, (0, 25)],  # default
+            [50, (14, 50)],
+        ],
+    )
+    def test_encoder_quality_qmin_qmax_map(self, tmp_path, quality, expected_qminmax):
+        qmin, qmax = expected_qminmax
+        with Image.open("tests/images/hopper.avif") as im:
+            out_quality = BytesIO()
+            out_qminmax = BytesIO()
+            im.save(out_qminmax, "AVIF", qmin=qmin, qmax=qmax)
+            if quality is None:
+                im.save(out_quality, "AVIF")
+            else:
+                im.save(out_quality, "AVIF", quality=quality)
+        assert len(out_quality.getvalue()) == len(out_qminmax.getvalue())
+
+    def test_encoder_quality_valueerror(self, tmp_path):
+        with Image.open("tests/images/hopper.avif") as im:
+            test_file = str(tmp_path / "temp.avif")
+            with pytest.raises(ValueError):
+                im.save(test_file, quality="invalid")
+
+    @skip_unless_avif_decoder("aom")
+    def test_decoder_codec_available(self):
+        assert _avif.decoder_codec_available("aom") is True
+
+    def test_decoder_codec_available_bad_params(self):
+        with pytest.raises(TypeError):
+            _avif.decoder_codec_available()
+
+    @skip_unless_avif_encoder("rav1e")
+    def test_decoder_codec_available_cannot_decode(self):
+        assert _avif.decoder_codec_available("rav1e") is False
+
+    def test_decoder_codec_available_invalid(self):
+        assert _avif.decoder_codec_available("foo") is False
+
+    @pytest.mark.parametrize("upsampling", ["fastest", "best", "nearest", "bilinear"])
+    def test_decoder_upsampling(self, upsampling):
+        AvifImagePlugin.CHROMA_UPSAMPLING = upsampling
+        try:
+            with Image.open(TEST_AVIF_FILE):
+                pass
+        finally:
+            AvifImagePlugin.CHROMA_UPSAMPLING = "auto"
+
+    def test_decoder_upsampling_invalid(self):
+        AvifImagePlugin.CHROMA_UPSAMPLING = "foo"
+        try:
+            with pytest.raises(ValueError):
+                with Image.open(TEST_AVIF_FILE):
+                    pass
+        finally:
+            AvifImagePlugin.CHROMA_UPSAMPLING = "auto"
+
+    def test_p_mode_transparency(self):
+        im = Image.new("P", size=(64, 64))
+        draw = ImageDraw.Draw(im)
+        draw.rectangle(xy=[(0, 0), (32, 32)], fill=255)
+        draw.rectangle(xy=[(32, 32), (64, 64)], fill=255)
+
+        buf_png = BytesIO()
+        im.save(buf_png, format="PNG", transparency=0)
+        im_png = Image.open(buf_png)
+        buf_out = BytesIO()
+        im_png.save(buf_out, format="AVIF", quality=100)
+
+        assert_image_similar(im_png.convert("RGBA"), Image.open(buf_out), 1)
+
+    def test_decoder_strict_flags(self):
+        # This would fail if full avif strictFlags were enabled
+        with Image.open("%s/tests/images/chimera-missing-pixi.avif" % CURR_DIR) as im:
+            assert im.size == (480, 270)
+
+
+class TestAvifAnimation:
+    @contextmanager
+    def star_frames(self):
+        with Image.open("%s/tests/images/star.png" % CURR_DIR) as f1:
+            with Image.open("%s/tests/images/star90.png" % CURR_DIR) as f2:
+                with Image.open("%s/tests/images/star180.png" % CURR_DIR) as f3:
+                    with Image.open("%s/tests/images/star270.png" % CURR_DIR) as f4:
+                        yield [f1, f2, f3, f4]
+
+    def test_n_frames(self):
+        """
+        Ensure that AVIF format sets n_frames and is_animated attributes
+        correctly.
+        """
+
+        with Image.open("tests/images/hopper.avif") as im:
+            assert im.n_frames == 1
+            assert not im.is_animated
+
+        with Image.open("tests/images/star.avifs") as im:
+            assert im.n_frames == 5
+            assert im.is_animated
+
+    def test_write_animation_L(self, tmp_path):
+        """
+        Convert an animated GIF to animated AVIF, then compare the frame
+        count, and first and last frames to ensure they're visually similar.
+        """
+
+        with Image.open("tests/images/star.gif") as orig:
+            assert orig.n_frames > 1
+
+            temp_file = str(tmp_path / "temp.avif")
+            orig.save(temp_file, save_all=True)
+            with Image.open(temp_file) as im:
+                assert im.n_frames == orig.n_frames
+
+                # Compare first and second-to-last frames to the original animated GIF
+                orig.load()
+                im.load()
+                assert_image_similar(im.convert("RGB"), orig.convert("RGB"), 25.0)
+                orig.seek(orig.n_frames - 2)
+                im.seek(im.n_frames - 2)
+                orig.load()
+                im.load()
+                assert_image_similar(im.convert("RGB"), orig.convert("RGB"), 25.0)
+
+    def test_write_animation_RGB(self, tmp_path):
+        """
+        Write an animated AVIF from RGB frames, and ensure the frames
+        are visually similar to the originals.
+        """
+
+        def check(temp_file):
+            with Image.open(temp_file) as im:
+                assert im.n_frames == 4
+
+                # Compare first frame to original
+                im.load()
+                assert_image_similar(im, frame1.convert("RGBA"), 25.0)
+
+                # Compare second frame to original
+                im.seek(1)
+                im.load()
+                assert_image_similar(im, frame2.convert("RGBA"), 25.0)
+
+        with self.star_frames() as frames:
+            frame1 = frames[0]
+            frame2 = frames[1]
+            temp_file1 = str(tmp_path / "temp.avif")
+            frames[0].copy().save(temp_file1, save_all=True, append_images=frames[1:])
+            check(temp_file1)
+
+            # Tests appending using a generator
+            def imGenerator(ims):
+                for im in ims:
+                    yield im
+
+            temp_file2 = str(tmp_path / "temp_generator.avif")
+            frames[0].copy().save(
+                temp_file2,
+                save_all=True,
+                append_images=imGenerator(frames[1:]),
+            )
+            check(temp_file2)
+
+    def test_sequence_dimension_mismatch_check(self, tmp_path):
+        temp_file = str(tmp_path / "temp.avif")
+        frame1 = Image.new("RGB", (100, 100))
+        frame2 = Image.new("RGB", (150, 150))
+        with pytest.raises(ValueError):
+            frame1.save(temp_file, save_all=True, append_images=[frame2], duration=100)
+
+    def test_heif_raises_unidentified_image_error(self):
+        with pytest.raises(UnidentifiedImageError or IOError):
+            with Image.open("tests/images/rgba10.heif"):
+                pass
+
+    @skip_unless_avif_version_gte((0, 9, 0))
+    @pytest.mark.parametrize("alpha_premultipled", [False, True])
+    def test_alpha_premultiplied_true(self, alpha_premultipled):
+        im = Image.new("RGBA", (10, 10), (0, 0, 0, 0))
+        im_buf = BytesIO()
+        im.save(im_buf, "AVIF", alpha_premultiplied=alpha_premultipled)
+        im_bytes = im_buf.getvalue()
+        assert has_alpha_premultiplied(im_bytes) is alpha_premultipled
+
+    def test_timestamp_and_duration(self, tmp_path):
+        """
+        Try passing a list of durations, and make sure the encoded
+        timestamps and durations are correct.
+        """
+
+        durations = [1, 10, 20, 30, 40]
+        temp_file = str(tmp_path / "temp.avif")
+        with self.star_frames() as frames:
+            frames[0].save(
+                temp_file,
+                save_all=True,
+                append_images=(frames[1:] + [frames[0]]),
+                duration=durations,
+            )
+
+        with Image.open(temp_file) as im:
+            assert im.n_frames == 5
+            assert im.is_animated
+
+            # Check that timestamps and durations match original values specified
+            ts = 0
+            for frame in range(im.n_frames):
+                im.seek(frame)
+                im.load()
+                assert im.info["duration"] == durations[frame]
+                assert im.info["timestamp"] == ts
+                ts += durations[frame]
+
+    def test_seeking(self, tmp_path):
+        """
+        Create an animated AVIF file, and then try seeking through frames in
+        reverse-order, verifying the timestamps and durations are correct.
+        """
+
+        dur = 33
+        temp_file = str(tmp_path / "temp.avif")
+        with self.star_frames() as frames:
+            frames[0].save(
+                temp_file,
+                save_all=True,
+                append_images=(frames[1:] + [frames[0]]),
+                duration=dur,
+            )
+
+        with Image.open(temp_file) as im:
+            assert im.n_frames == 5
+            assert im.is_animated
+
+            # Traverse frames in reverse, checking timestamps and durations
+            ts = dur * (im.n_frames - 1)
+            for frame in reversed(range(im.n_frames)):
+                im.seek(frame)
+                im.load()
+                assert im.info["duration"] == dur
+                assert im.info["timestamp"] == ts
+                ts -= dur
+
+    def test_seek_errors(self):
+        with Image.open("tests/images/star.avifs") as im:
+            with pytest.raises(EOFError):
+                im.seek(-1)
+
+            with pytest.raises(EOFError):
+                im.seek(42)
+
+
+if hasattr(os, "sched_getaffinity"):
+    MAX_THREADS = len(os.sched_getaffinity(0))
+else:
+    MAX_THREADS = cpu_count()
+
+
+class TestAvifLeaks(PillowLeakTestCase):
+    mem_limit = MAX_THREADS * 3 * 1024
+    iterations = 100
+
+    @pytest.mark.skipif(
+        is_docker_qemu(), reason="Skipping on cross-architecture containers"
+    )
+    def test_leak_load(self):
+        with open(TEST_AVIF_FILE, "rb") as f:
+            im_data = f.read()
+
+        def core():
+            with Image.open(BytesIO(im_data)) as im:
+                im.load()
+
+        self._test_leak(core)