"""
将 PNG 图片转换为 C 头文件（uint32_t 像素数组）。

用法:
    python tools/png_to_header.py input.png output.h [var_name]

像素格式: (R << 24) | (G << 16) | (B << 8) | A
与 FrameBuffer 的 uint32_t 格式一致。

如果图片没有 alpha 通道，A 默认为 0xFF。
"""

import sys
import os

from PIL import Image


def convert(input_path: str, output_path: str, var_name: str):
    img = Image.open(input_path).convert("RGBA")
    w, h = img.size
    pixels = list(img.getdata()) if not hasattr(img, "get_flattened_data") else list(img.get_flattened_data())

    packed = []
    for r, g, b, a in pixels:
        packed.append((r << 24) | (g << 16) | (b << 8) | a)

    with open(output_path, "w", encoding="utf-8") as f:
        f.write("// Auto-generated by tools/png_to_header.py\n")
        f.write(f"#pragma once\n#include <cstdint>\n\n")
        f.write(f"static const int32_t {var_name}_width = {w};\n")
        f.write(f"static const int32_t {var_name}_height = {h};\n\n")
        f.write(f"static const uint32_t {var_name}_pixels[] = {{\n")

        for i in range(0, len(packed), 16):
            chunk = packed[i : i + 16]
            line = ", ".join(f"0x{p:08X}" for p in chunk)
            f.write(f"    {line},\n")

        f.write("};\n")


def main():
    if len(sys.argv) < 3:
        print(f"用法: {sys.argv[0]} input.png output.h [var_name]")
        sys.exit(1)

    input_path = sys.argv[1]
    output_path = sys.argv[2]

    if len(sys.argv) >= 4:
        var_name = sys.argv[3]
    else:
        # 从文件名生成变量名: player_idle.png -> player_idle
        var_name = os.path.splitext(os.path.basename(input_path))[0]

    convert(input_path, output_path, var_name)
    print(f"已生成: {output_path}  (var: {var_name}, 像素数: {var_name}_width * {var_name}_height)")


if __name__ == "__main__":
    main()