python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-07-04 11:53:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

Added type hints

Browse Source
This commit is contained in:
Andrew Murray 2025-05-05 06:44:37 +10:00
parent 64f992e7fc
commit 2ec01c9c58
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/PIL/VtfImagePlugin.py
View File

@ -129,12 +129,12 @@ HEADER_V72 = "<I2HI2H4x3f4xfIbI2bH"
HEADER_V73 = "<I2HI2H4x3f4xfIbI2bH3xI8x" HEADER_V73 = "<I2HI2H4x3f4xfIbI2bH3xI8x"
def _get_texture_size(pixel_format: VtfPF, width, height): def _get_texture_size(pixel_format: VtfPF, width: int, height: int) -> int:
if pixel_format in (VtfPF.DXT1, VtfPF.DXT1_ONEBITALPHA): if pixel_format in (VtfPF.DXT1, VtfPF.DXT1_ONEBITALPHA):
return width * height // 2 return width * height // 2
elif pixel_format in (VtfPF.DXT3, VtfPF.DXT5):
return width * height
elif pixel_format in ( elif pixel_format in (
VtfPF.DXT3,
VtfPF.DXT5,
VtfPF.A8, VtfPF.A8,
VtfPF.I8, VtfPF.I8,
): ):
@ -203,7 +203,7 @@ def _write_image(fp: IO[bytes], im: Image.Image, pixel_format: VtfPF) -> None:
ImageFile._save(im, fp, tile, _get_texture_size(pixel_format, *im.size)) ImageFile._save(im, fp, tile, _get_texture_size(pixel_format, *im.size))
def _closest_power(x): def _closest_power(x: int) -> int:
possible_results = round(log(x, 2)), ceil(log(x, 2)) possible_results = round(log(x, 2)), ceil(log(x, 2))
return 2 ** min(possible_results, key=lambda z: abs(x - 2**z)) return 2 ** min(possible_results, key=lambda z: abs(x - 2**z))