Initial VTF support

Tests/test_file_vtf.py

@@ -0,0 +1,123 @@
+from __future__ import annotations
+
+import pytest
+
+from PIL import Image
+from PIL.VtfImagePlugin import (
+    VtfPF,
+    _closest_power,
+    _get_mipmap_count,
+    _get_texture_size,
+)
+
+from .helper import assert_image_equal, assert_image_similar
+
+
+@pytest.mark.parametrize(
+    ("size", "expected_size"),
+    [
+        (8, 8),
+        (7, 8),
+        (9, 8),
+        (192, 256),
+        (1, 1),
+        (2000, 2048),
+    ],
+)
+def test_closest_power(size: int, expected_size: int):
+    assert _closest_power(size) == expected_size
+
+
+@pytest.mark.parametrize(
+    ("size", "expected_count"),
+    [
+        ((1, 1), 1),
+        ((2, 2), 2),
+        ((4, 4), 3),
+        ((8, 8), 4),
+        ((128, 128), 8),
+        ((256, 256), 9),
+        ((512, 512), 10),
+        ((1024, 1024), 11),
+        ((1024, 1), 11),
+    ],
+)
+def test_get_mipmap_count(size: tuple[int, int], expected_count: int):
+    assert _get_mipmap_count(*size) == expected_count
+
+
+@pytest.mark.parametrize(
+    ("pixel_format", "size", "expected_size"),
+    [
+        (VtfPF.DXT1, (16, 16), (16 * 16) // 2),
+        (VtfPF.DXT1_ONEBITALPHA, (16, 16), (16 * 16) // 2),
+        (VtfPF.DXT3, (16, 16), 16 * 16),
+        (VtfPF.DXT5, (16, 16), 16 * 16),
+        (VtfPF.BGR888, (16, 16), 16 * 16 * 3),
+        (VtfPF.RGB888, (16, 16), 16 * 16 * 3),
+        (VtfPF.RGBA8888, (16, 16), 16 * 16 * 4),
+        (VtfPF.UV88, (16, 16), 16 * 16 * 2),
+        (VtfPF.A8, (16, 16), 16 * 16),
+        (VtfPF.I8, (16, 16), 16 * 16),
+        (VtfPF.IA88, (16, 16), 16 * 16 * 2),
+    ],
+)
+def test_get_texture_size(
+    pixel_format: VtfPF, size: tuple[int, int], expected_size: int
+):
+    assert _get_texture_size(pixel_format, *size) == expected_size
+
+
+@pytest.mark.parametrize(
+    ("etalon_path", "file_path", "expected_mode", "epsilon"),
+    [
+        ("Tests/images/vtf_i8.png", "Tests/images/vtf_i8.vtf", "L", 0.0),
+        ("Tests/images/vtf_a8.png", "Tests/images/vtf_a8.vtf", "RGBA", 0.0),
+        ("Tests/images/vtf_ia88.png", "Tests/images/vtf_ia88.vtf", "LA", 0.0),
+        ("Tests/images/vtf_uv88.png", "Tests/images/vtf_uv88.vtf", "RGB", 0.0),
+        ("Tests/images/vtf_rgb888.png", "Tests/images/vtf_rgb888.vtf", "RGB", 0.0),
+        ("Tests/images/vtf_bgr888.png", "Tests/images/vtf_bgr888.vtf", "RGB", 0.0),
+        ("Tests/images/vtf_dxt1.png", "Tests/images/vtf_dxt1.vtf", "RGBA", 3.0),
+        ("Tests/images/vtf_dxt1A.png", "Tests/images/vtf_dxt1A.vtf", "RGBA", 8.0),
+        ("Tests/images/vtf_rgba8888.png", "Tests/images/vtf_rgba8888.vtf", "RGBA", 0),
+    ],
+)
+def test_vtf_read(etalon_path: str, file_path: str, expected_mode: str, epsilon: float):
+    e = Image.open(etalon_path)
+    f = Image.open(file_path)
+    assert f.mode == expected_mode
+    e = e.convert(expected_mode)
+    if epsilon == 0:
+        assert_image_equal(e, f)
+    else:
+        assert_image_similar(e, f, epsilon)
+
+
+@pytest.mark.parametrize(
+    ("pixel_format", "file_path", "expected_mode", "epsilon"),
+    [
+        (VtfPF.I8, "Tests/images/vtf_i8.png", "L", 0.0),
+        (VtfPF.A8, "Tests/images/vtf_a8.png", "RGBA", 0.0),
+        (VtfPF.IA88, "Tests/images/vtf_ia88.png", "LA", 0.0),
+        (VtfPF.UV88, "Tests/images/vtf_uv88.png", "RGB", 0.0),
+        (VtfPF.RGB888, "Tests/images/vtf_rgb888.png", "RGB", 0.0),
+        (VtfPF.BGR888, "Tests/images/vtf_bgr888.png", "RGB", 0.0),
+        (VtfPF.DXT1, "Tests/images/vtf_dxt1.png", "RGBA", 3.0),
+        (VtfPF.DXT1_ONEBITALPHA, "Tests/images/vtf_dxt1A.png", "RGBA", 8.0),
+        (VtfPF.RGBA8888, "Tests/images/vtf_rgba8888.png", "RGBA", 0),
+    ],
+)
+def test_vtf_save(
+    pixel_format: VtfPF, file_path: str, expected_mode: str, epsilon: float, tmp_path
+):
+    f: Image.Image = Image.open(file_path)
+    out = (tmp_path / "tmp.vtf").as_posix()
+    f.save(out, pixel_format=pixel_format)
+    if pixel_format == VtfPF.DXT1:
+        f = f.convert("RGBA")
+    e = Image.open(out)
+    assert e.mode == expected_mode
+    if epsilon == 0:
+        assert_image_equal(e, f)
+    else:
+        assert_image_similar(e, f, epsilon)

setup.py

@@ -66,6 +66,7 @@ def get_version():
     "Reduce",
     "Bands",
     "BcnDecode",
+    "BcnEncode",
     "BitDecode",
     "Blend",
     "Chops",