python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-07-28 17:10:02 +03:00
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-06-10 08:52:20 +00:00
parent 6c546f851e
commit 727237c774
2 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Tests/test_file_qoi.py
View File

@ -12,6 +12,7 @@ from .helper import (
hopper, hopper,
) )
def test_sanity() -> None: def test_sanity() -> None:
with Image.open("Tests/images/hopper.qoi") as im: with Image.open("Tests/images/hopper.qoi") as im:
assert im.mode == "RGB" assert im.mode == "RGB"
@ -54,4 +55,3 @@ def test_save(tmp_path: Path) -> None:
im = hopper("P") im = hopper("P")
with pytest.raises(ValueError): with pytest.raises(ValueError):
im.save(f) im.save(f)

23
src/PIL/QoiImagePlugin.py
View File

@ -11,7 +11,9 @@ import os
from typing import IO from typing import IO
from . import Image, ImageFile from . import Image, ImageFile
from ._binary import i32be as i32, o32be as o32, o8 from ._binary import i32be as i32
from ._binary import o8
from ._binary import o32be as o32
def _accept(prefix: bytes) -> bool: def _accept(prefix: bytes) -> bool:
@ -140,10 +142,10 @@ class QoiEncoder(ImageFile.PyEncoder):
_previously_seen_pixels: dict[int, tuple[int]] = {} _previously_seen_pixels: dict[int, tuple[int]] = {}
def _write_run(self, run): def _write_run(self, run):
return o8(0xc0 | (run - 1)) # QOI_OP_RUN return o8(0xC0 | (run - 1)) # QOI_OP_RUN
def _delta(self, left, right): def _delta(self, left, right):
result = (left - right) & 0xff result = (left - right) & 0xFF
if result >= 0x80: if result >= 0x80:
result -= 0x100 result -= 0x100
return result return result
@ -178,7 +180,7 @@ class QoiEncoder(ImageFile.PyEncoder):
r, g, b, a = pixel r, g, b, a = pixel
hash_value = (r * 3 + g * 5 + b * 7 + a * 11) % 64 hash_value = (r * 3 + g * 5 + b * 7 + a * 11) % 64
if self._previously_seen_pixels.get(hash_value) == pixel: if self._previously_seen_pixels.get(hash_value) == pixel:
data += o8(hash_value) # QOI_OP_INDEX data += o8(hash_value) # QOI_OP_INDEX
else: else:
self._previously_seen_pixels[hash_value] = pixel self._previously_seen_pixels[hash_value] = pixel
@ -191,23 +193,24 @@ class QoiEncoder(ImageFile.PyEncoder):
dgb = self._delta(db, dg) dgb = self._delta(db, dg)
if -2 <= dr < 2 and -2 <= dg < 2 and -2 <= db < 2: if -2 <= dr < 2 and -2 <= dg < 2 and -2 <= db < 2:
data += o8(0x40 | (dr + 2) << 4 | data += o8(
(dg + 2) << 2 | (db + 2)) # QOI_OP_DIFF 0x40 | (dr + 2) << 4 | (dg + 2) << 2 | (db + 2)
) # QOI_OP_DIFF
elif -8 <= dgr < 8 and -32 <= dg < 32 and -8 <= dgb < 8: elif -8 <= dgr < 8 and -32 <= dg < 32 and -8 <= dgb < 8:
data += o8(0x80 | (dg + 32)) # QOI_OP_LUMA data += o8(0x80 | (dg + 32)) # QOI_OP_LUMA
data += o8((dgr + 8) << 4 | (dgb + 8)) data += o8((dgr + 8) << 4 | (dgb + 8))
else: else:
data += o8(0xfe) # QOI_OP_RGB data += o8(0xFE) # QOI_OP_RGB
data += bytes(pixel[:3]) data += bytes(pixel[:3])
else: else:
data += o8(0xff) # QOI_OP_RGBA data += o8(0xFF) # QOI_OP_RGBA
data += bytes(pixel) data += bytes(pixel)
self._previous_pixel = pixel self._previous_pixel = pixel
if run > 0: if run > 0:
data += self._write_run(run) data += self._write_run(run)
data += bytes((0,0,0,0,0,0,0,1)) # padding data += bytes((0, 0, 0, 0, 0, 0, 0, 1)) # padding
return len(data), 0, data return len(data), 0, data