// File: basisu_transcoder_pybind11.cpp
// pybind11 native bindings for the transcoder's pure C API basisu_wasm_transcoder_api.h

#include <pybind11/pybind11.h>
#include <stdint.h>

#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_is_xubc7",
          [](uint32_t fmt) { return to_bool(bt_basis_tex_format_is_xubc7(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_is_xubc7",
          [](uint64_t h) { return to_bool(bt_ktx2_is_xubc7(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_get_deblocking_filter_index", &bt_ktx2_get_deblocking_filter_index);

    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`");
}