// File: basisu_transcoder_pybind11.cpp // pybind11 native bindings for the transcoder's pure C API basisu_wasm_transcoder_api.h #include #include #include "../encoder/basisu_wasm_transcoder_api.h" namespace py = pybind11; // wasm_bool_t is uint32_t — convert to Python bool static inline bool to_bool(wasm_bool_t v) { return v != 0; } PYBIND11_MODULE(basisu_transcoder_python, m) { m.doc() = "Native Basis Universal transcoder (pybind11 binding over basisu_wasm_transcoder_api)"; // ------------------------------------------------------------------------ // High-level functions // ------------------------------------------------------------------------ m.def("get_version", &bt_get_version, "Get BasisU transcoder version"); m.def("enable_debug_printf", [](bool flag) { bt_enable_debug_printf(flag ? 1u : 0u); }, "Enable or disable debug printf output"); m.def("init", &bt_init, "Initialize transcoder library"); m.def("alloc", &bt_alloc, "Allocate a buffer, returns uint64 offset/pointer"); m.def("free", &bt_free, "Free a buffer allocated by bt_alloc"); // ------------------------------------------------------------------------ // basis_tex_format helpers // ------------------------------------------------------------------------ m.def("basis_tex_format_is_xuastc_ldr", [](uint32_t fmt) { return to_bool(bt_basis_tex_format_is_xuastc_ldr(fmt)); }); m.def("basis_tex_format_is_astc_ldr", [](uint32_t fmt) { return to_bool(bt_basis_tex_format_is_astc_ldr(fmt)); }); m.def("basis_tex_format_get_block_width", &bt_basis_tex_format_get_block_width); m.def("basis_tex_format_get_block_height", &bt_basis_tex_format_get_block_height); m.def("basis_tex_format_is_hdr", [](uint32_t fmt) { return to_bool(bt_basis_tex_format_is_hdr(fmt)); }); m.def("basis_tex_format_is_ldr", [](uint32_t fmt) { return to_bool(bt_basis_tex_format_is_ldr(fmt)); }); // ------------------------------------------------------------------------ // transcoder_texture_format helpers // ------------------------------------------------------------------------ m.def("basis_get_bytes_per_block_or_pixel", &bt_basis_get_bytes_per_block_or_pixel); m.def("basis_transcoder_format_has_alpha", [](uint32_t tfmt) { return to_bool(bt_basis_transcoder_format_has_alpha(tfmt)); }); m.def("basis_transcoder_format_is_hdr", [](uint32_t tfmt) { return to_bool(bt_basis_transcoder_format_is_hdr(tfmt)); }); m.def("basis_transcoder_format_is_ldr", [](uint32_t tfmt) { return to_bool(bt_basis_transcoder_format_is_ldr(tfmt)); }); m.def("basis_transcoder_texture_format_is_astc", [](uint32_t tfmt) { return to_bool(bt_basis_transcoder_texture_format_is_astc(tfmt)); }); m.def("basis_transcoder_format_is_uncompressed", [](uint32_t tfmt) { return to_bool(bt_basis_transcoder_format_is_uncompressed(tfmt)); }); m.def("basis_get_uncompressed_bytes_per_pixel", &bt_basis_get_uncompressed_bytes_per_pixel); m.def("basis_get_block_width", &bt_basis_get_block_width); m.def("basis_get_block_height", &bt_basis_get_block_height); m.def("basis_get_transcoder_texture_format_from_basis_tex_format", &bt_basis_get_transcoder_texture_format_from_basis_tex_format); m.def("basis_is_format_supported", [](uint32_t tfmt, uint32_t basis_fmt) { return to_bool(bt_basis_is_format_supported(tfmt, basis_fmt)); }); m.def("basis_compute_transcoded_image_size_in_bytes", &bt_basis_compute_transcoded_image_size_in_bytes); // ------------------------------------------------------------------------ // KTX2 open/close & basic info // ------------------------------------------------------------------------ m.def("ktx2_open", &bt_ktx2_open, "Open a KTX2 image from memory; returns handle"); m.def("ktx2_close", &bt_ktx2_close, "Close a previously opened KTX2 handle"); m.def("ktx2_get_width", &bt_ktx2_get_width); m.def("ktx2_get_height", &bt_ktx2_get_height); m.def("ktx2_get_levels", &bt_ktx2_get_levels); m.def("ktx2_get_faces", &bt_ktx2_get_faces); m.def("ktx2_get_layers", &bt_ktx2_get_layers); m.def("ktx2_get_basis_tex_format", &bt_ktx2_get_basis_tex_format); m.def("ktx2_is_etc1s", [](uint64_t h) { return to_bool(bt_ktx2_is_etc1s(h)); }); m.def("ktx2_is_uastc_ldr_4x4", [](uint64_t h) { return to_bool(bt_ktx2_is_uastc_ldr_4x4(h)); }); m.def("ktx2_is_hdr", [](uint64_t h) { return to_bool(bt_ktx2_is_hdr(h)); }); m.def("ktx2_is_hdr_4x4", [](uint64_t h) { return to_bool(bt_ktx2_is_hdr_4x4(h)); }); m.def("ktx2_is_hdr_6x6", [](uint64_t h) { return to_bool(bt_ktx2_is_hdr_6x6(h)); }); m.def("ktx2_is_ldr", [](uint64_t h) { return to_bool(bt_ktx2_is_ldr(h)); }); m.def("ktx2_is_astc_ldr", [](uint64_t h) { return to_bool(bt_ktx2_is_astc_ldr(h)); }); m.def("ktx2_is_xuastc_ldr", [](uint64_t h) { return to_bool(bt_ktx2_is_xuastc_ldr(h)); }); m.def("ktx2_get_block_width", &bt_ktx2_get_block_width); m.def("ktx2_get_block_height", &bt_ktx2_get_block_height); m.def("ktx2_has_alpha", [](uint64_t h) { return to_bool(bt_ktx2_has_alpha(h)); }); m.def("ktx2_get_dfd_color_model", &bt_ktx2_get_dfd_color_model); m.def("ktx2_get_dfd_color_primaries", &bt_ktx2_get_dfd_color_primaries); m.def("ktx2_get_dfd_transfer_func", &bt_ktx2_get_dfd_transfer_func); m.def("ktx2_is_srgb", [](uint64_t h) { return to_bool(bt_ktx2_is_srgb(h)); }); m.def("ktx2_get_dfd_flags", &bt_ktx2_get_dfd_flags); m.def("ktx2_get_dfd_total_samples", &bt_ktx2_get_dfd_total_samples); m.def("ktx2_get_dfd_channel_id0", &bt_ktx2_get_dfd_channel_id0); m.def("ktx2_get_dfd_channel_id1", &bt_ktx2_get_dfd_channel_id1); m.def("ktx2_is_video", [](uint64_t h) { return to_bool(bt_ktx2_is_video(h)); }); m.def("ktx2_get_ldr_hdr_upconversion_nit_multiplier", &bt_ktx2_get_ldr_hdr_upconversion_nit_multiplier); // ------------------------------------------------------------------------ // KTX2 per-level info // ------------------------------------------------------------------------ m.def("ktx2_get_level_orig_width", &bt_ktx2_get_level_orig_width); m.def("ktx2_get_level_orig_height", &bt_ktx2_get_level_orig_height); m.def("ktx2_get_level_actual_width", &bt_ktx2_get_level_actual_width); m.def("ktx2_get_level_actual_height", &bt_ktx2_get_level_actual_height); m.def("ktx2_get_level_num_blocks_x", &bt_ktx2_get_level_num_blocks_x); m.def("ktx2_get_level_num_blocks_y", &bt_ktx2_get_level_num_blocks_y); m.def("ktx2_get_level_total_blocks", &bt_ktx2_get_level_total_blocks); m.def("ktx2_get_level_alpha_flag", [](uint64_t h, uint32_t level, uint32_t layer, uint32_t face) { return to_bool(bt_ktx2_get_level_alpha_flag(h, level, layer, face)); }); m.def("ktx2_get_level_iframe_flag", [](uint64_t h, uint32_t level, uint32_t layer, uint32_t face) { return to_bool(bt_ktx2_get_level_iframe_flag(h, level, layer, face)); }); // ------------------------------------------------------------------------ // Transcoding state and operations // ------------------------------------------------------------------------ m.def("ktx2_start_transcoding", [](uint64_t h) { return to_bool(bt_ktx2_start_transcoding(h)); }); m.def("ktx2_create_transcode_state", &bt_ktx2_create_transcode_state); m.def("ktx2_destroy_transcode_state", &bt_ktx2_destroy_transcode_state); m.def("ktx2_transcode_image_level", [](uint64_t ktx2_handle, uint32_t level_index, uint32_t layer_index, uint32_t face_index, uint64_t out_mem_ofs, uint32_t out_blocks_or_pixels, uint32_t transcoder_texture_format_u32, uint32_t decode_flags, uint32_t row_pitch_blocks_or_pixels, uint32_t rows_in_pixels, int channel0, int channel1, uint64_t state_handle) { return to_bool(bt_ktx2_transcode_image_level( ktx2_handle, level_index, layer_index, face_index, out_mem_ofs, out_blocks_or_pixels, transcoder_texture_format_u32, decode_flags, row_pitch_blocks_or_pixels, rows_in_pixels, channel0, channel1, state_handle)); }, py::arg("ktx2_handle"), py::arg("level_index"), py::arg("layer_index"), py::arg("face_index"), py::arg("output_block_mem_ofs"), py::arg("output_blocks_buf_size_in_blocks_or_pixels"), py::arg("transcoder_texture_format_u32"), py::arg("decode_flags"), py::arg("output_row_pitch_in_blocks_or_pixels") = 0, py::arg("output_rows_in_pixels") = 0, py::arg("channel0") = -1, py::arg("channel1") = -1, py::arg("state_handle") = 0); m.def("read_memory", [](uint64_t ptr, uint32_t size) { return py::bytes((const char*)ptr, size); }, "Read `size` bytes starting at native memory address `ptr`"); m.def("write_memory", [](uint64_t dest_ptr, py::buffer src) { py::buffer_info info = src.request(); memcpy((void*)dest_ptr, info.ptr, info.size * info.itemsize); }, "Write bytes/buffer-like object into native memory at address `ptr`"); }