Add ktx

2026-06-14 19:09:18 +01:00
parent 14bd1a9271
commit 13fa90a0e9
3958 changed files with 999286 additions and 4 deletions
@@ -0,0 +1,9 @@
+# Copyright (c) 2023, Shukant Pal and Contributors
+# SPDX-License-Identifier: Apache-2.0
+
+from pathlib import Path
+import os
+
+__test_images__ = os.environ['KTX_IMAGES_DIR'] \
+    if 'KTX_IMAGES_DIR' in os.environ \
+    else str((Path(__file__) / Path('../../../../tests/testimages')).resolve())
@@ -0,0 +1,26 @@
+# Copyright (c) 2023, Shukant Pal and Contributors
+# SPDX-License-Identifier: Apache-2.0
+
+from pyktx import *
+from test_config import __test_images__
+import unittest
+
+
+class TestKtxTexture(unittest.TestCase):
+    def test_kv_data(self):
+        test_ktx_file = os.path.join(__test_images__, 'astc_ldr_4x4_FlightHelmet_baseColor.ktx2')
+        texture = KtxTexture2.create_from_named_file(test_ktx_file)
+
+        self.assertEqual(texture.kv_data.find_value('KTXorientation'), b'rd\x00')
+        self.assertEqual(texture.kv_data.copy(), {
+            'KTXorientation': b'rd\x00',
+            'KTXwriter': b'toktx v4.0.__default__ / libktx v4.0.__default__\x00',
+            'KTXwriterScParams': b'--encode astc --astc_blk_d 4x4\x00',
+        })
+
+        for key in ['KTXorientation', 'KTXwriter', 'KTXwriterScParams']:
+            texture.kv_data.delete_kv_pair(key)
+        texture.kv_data.add_kv_pair('KTXwriter', b'pyktx v4.0.__default__ / libktx v4.0.__default__\x00')
+
+        self.assertEqual(texture.kv_data.find_value('KTXwriter'),
+                         b'pyktx v4.0.__default__ / libktx v4.0.__default__\x00')
@@ -0,0 +1,60 @@
+# Copyright (c) 2023, Shukant Pal and Contributors
+# SPDX-License-Identifier: Apache-2.0
+
+from pyktx import *
+from tempfile import NamedTemporaryFile
+from test_config import __test_images__
+import os
+import unittest
+
+
+class TestKtxTexture1(unittest.TestCase):
+    def test_create_from_named_file(self):
+        test_ktx_file = os.path.join(__test_images__, 'etc1.ktx')
+        texture = KtxTexture1.create_from_named_file(test_ktx_file)
+
+        self.assertEqual(texture.gl_internalformat, GlInternalformat.ETC1_RGB8_OES)
+        self.assertFalse(texture.is_array)
+        self.assertFalse(texture.generate_mipmaps)
+        self.assertEqual(texture.num_levels, 1)
+
+    def test_write_to_named_file(self):
+        test_ktx_file = os.path.join(__test_images__, 'etc2-rgb.ktx')
+        copy_file = NamedTemporaryFile(delete=False)
+        copy_file.close()
+
+        texture = KtxTexture1.create_from_named_file(test_ktx_file, KtxTextureCreateFlagBits.LOAD_IMAGE_DATA_BIT)
+        texture.write_to_named_file(copy_file.name)
+
+        with open(test_ktx_file, 'rb') as original, open(copy_file.name, 'rb') as copy:
+            self.assertEqual(original.read(), copy.read())
+
+        os.unlink(copy_file.name)
+
+    def test_write_to_memory(self):
+        test_ktx_file = os.path.join(__test_images__, 'etc2-rgba1.ktx')
+        texture = KtxTexture1.create_from_named_file(test_ktx_file, KtxTextureCreateFlagBits.LOAD_IMAGE_DATA_BIT)
+        texture_bytes = texture.write_to_memory()
+
+        with open(test_ktx_file, 'rb') as original:
+            self.assertEqual(original.read(), texture_bytes)
+
+    def test_get_data(self):
+        test_ktx_file = os.path.join(__test_images__, 'etc2-rgba1.ktx')
+        texture = KtxTexture1.create_from_named_file(test_ktx_file, KtxTextureCreateFlagBits.LOAD_IMAGE_DATA_BIT)
+
+        data = texture.data()
+
+        with open(test_ktx_file, 'rb') as file:
+            level_0_size = texture.image_size(0)
+            self.assertEqual(file.read()[-level_0_size:], data[:])
+
+    def test_create(self):
+        info = KtxTextureCreateInfo(
+            gl_internal_format=GlInternalformat.COMPRESSED_RGBA_ASTC_4x4_KHR,
+            base_width=10,
+            base_height=10,
+            base_depth=1)
+
+        texture = KtxTexture1.create(info, KtxTextureCreateStorage.ALLOC)
+        texture.set_image_from_memory(0, 0, 0, bytes(texture.data_size))
@@ -0,0 +1,104 @@
+# Copyright (c) 2023, Shukant Pal and Contributors
+# SPDX-License-Identifier: Apache-2.0
+
+from pyktx import *
+from test_config import __test_images__
+import unittest
+
+
+class TestKtxTexture2(unittest.TestCase):
+    def test_create_from_named_file(self):
+        test_ktx_file = os.path.join(__test_images__, 'astc_ldr_4x4_FlightHelmet_baseColor.ktx2')
+        texture = KtxTexture2.create_from_named_file(test_ktx_file)
+
+        self.assertEqual(texture.num_levels, 1)
+        self.assertEqual(texture.num_faces, 1)
+        self.assertEqual(texture.vk_format, VkFormat.VK_FORMAT_ASTC_4x4_SRGB_BLOCK)
+        self.assertEqual(texture.base_width, 2048)
+        self.assertEqual(texture.base_height, 2048)
+        self.assertEqual(texture.supercompression_scheme, KtxSupercmpScheme.NONE)
+
+    def test_create_from_named_file_mipmapped(self):
+        test_ktx_file = os.path.join(__test_images__, 'astc_mipmap_ldr_4x4_posx.ktx2')
+        texture = KtxTexture2.create_from_named_file(test_ktx_file, KtxTextureCreateFlagBits.NO_FLAGS)
+
+        self.assertEqual(texture.num_levels, 12)
+        self.assertEqual(texture.base_width, 2048)
+        self.assertEqual(texture.base_height, 2048)
+
+    def test_get_image_size(self):
+        test_ktx_file = os.path.join(__test_images__, 'astc_mipmap_ldr_4x4_posx.ktx2')
+        texture = KtxTexture2.create_from_named_file(test_ktx_file)
+
+        self.assertEqual(texture.image_size(0), 4194304)
+
+    def test_get_image_offset(self):
+        test_ktx_file = os.path.join(__test_images__, 'astc_mipmap_ldr_4x4_posx.ktx2')
+        texture = KtxTexture2.create_from_named_file(test_ktx_file)
+
+        level11_offset = texture.image_offset(11, 0, 0)
+        level0_offset = texture.image_offset(0, 0, 0)
+
+        self.assertEqual(level11_offset, 0)
+        self.assertEqual(level0_offset - level11_offset, 0x155790 - 0x220)
+
+    def test_get_size(self):
+        test_ktx_file = os.path.join(__test_images__, 'astc_mipmap_ldr_4x4_posx.ktx2')
+        texture = KtxTexture2.create_from_named_file(test_ktx_file)
+
+        self.assertEqual(texture.num_levels, 12)
+
+        data_size = texture.data_size
+        total_size = 0
+
+        for i in range(0, 12):
+            total_size += texture.image_size(i)
+
+        self.assertEqual(total_size, data_size)
+
+        data = texture.data()
+        self.assertEqual(len(data), data_size)
+
+    def test_get_data(self):
+        test_ktx_file = os.path.join(__test_images__, 'astc_mipmap_ldr_4x4_posx.ktx2')
+        texture = KtxTexture2.create_from_named_file(test_ktx_file)
+
+        self.assertEqual(texture.num_levels, 12)
+
+        level0_length = texture.image_size(0)
+
+        with open(test_ktx_file, 'rb') as file:
+            data = texture.data()
+            self.assertEqual(file.read()[-level0_length - 1:], data[-level0_length - 1:])
+
+    def test_compress_basis(self):
+        test_ktx_file = os.path.join(__test_images__, 'arraytex_7_mipmap_reference_u.ktx2')
+        texture = KtxTexture2.create_from_named_file(test_ktx_file, KtxTextureCreateFlagBits.LOAD_IMAGE_DATA_BIT)
+
+        self.assertEqual(texture.is_compressed, False)
+        self.assertEqual(texture.supercompression_scheme, KtxSupercmpScheme.NONE)
+
+        texture.compress_basis(KtxBasisParams(quality_level=1))
+
+        self.assertEqual(texture.is_compressed, True)
+        self.assertEqual(texture.supercompression_scheme, KtxSupercmpScheme.BASIS_LZ)
+
+    def test_transcode_basis(self):
+        test_ktx_file = os.path.join(__test_images__, 'color_grid_basis.ktx2')
+        texture = KtxTexture2.create_from_named_file(test_ktx_file, KtxTextureCreateFlagBits.LOAD_IMAGE_DATA_BIT)
+
+        texture.transcode_basis(KtxTranscodeFmt.ASTC_4x4_RGBA, 0)
+
+        self.assertEqual(texture.vk_format, VkFormat.VK_FORMAT_ASTC_4x4_SRGB_BLOCK)
+
+    def test_create(self):
+        info = KtxTextureCreateInfo(
+            gl_internal_format=None,  # ignored
+            base_width=10,
+            base_height=10,
+            base_depth=1,
+            vk_format=VkFormat.VK_FORMAT_ASTC_4x4_SRGB_BLOCK)
+
+        texture = KtxTexture2.create(info, KtxTextureCreateStorage.ALLOC)
+        self.assertEqual(texture.vk_format, VkFormat.VK_FORMAT_ASTC_4x4_SRGB_BLOCK)
+        texture.set_image_from_memory(0, 0, 0, bytes(texture.data_size))