#!/usr/bin/env python3
"""生成 HUD 字形子集和无损调色板游程数据；无需运行时分配/解压缓存。

--import-zip 只导入明确列出的 40x20 PNG，不执行压缩包中的任何内容。
原 PNG 保留；优化 PNG 与生成 C 是可再生文件。--check 只校验不写入。
"""
import argparse
import io
import re
from pathlib import Path
from zipfile import ZipFile
from PIL import Image

ROOT = Path(__file__).resolve().parents[1]
ASSETS = ROOT / 'components/platform/assets/status_hud'
OUTPUT = ROOT / 'components/platform/src/generated/lcd_status_assets.c'
NAMES = [f'battery_{n}.png' for n in (1, 25, 50, 75, 100)] + [
    f'battery_charging_{n}.png' for n in (1, 25, 50, 75, 100)]

def array(name, data):
    rows = ['    ' + ', '.join(f'0x{n:02x}' for n in data[i:i+24]) + ','
            for i in range(0, len(data), 24)]
    return f'static const uint8_t {name}[] = {{\n' + '\n'.join(rows) + '\n};\n'

def generate(check=False):
    font = (ROOT / 'managed_components/lvgl__lvgl/src/font/lv_font_montserrat_14.c').read_text()
    assert '.bpp = 4,' in font and '.bitmap_format = 0,' in font
    assert re.search(r'\.line_height\s*=\s*16\b', font)
    assert re.search(r'\.base_line\s*=\s*3\b', font)
    assert '.range_start = 32, .range_length = 95, .glyph_id_start = 1,' in font
    block = re.search(r'glyph_bitmap\[\]\s*=\s*\{(.*?)\n};', font, re.S)[1]
    block = re.sub(r'/\*.*?\*/', '', block, flags=re.S)
    bitmap = bytes(int(n, 16) for n in re.findall(r'0x[0-9a-fA-F]+', block))
    block = re.search(r'glyph_dsc\[\]\s*=\s*\{(.*?)\n};', font, re.S)[1]
    glyphs = [dict((key, int(value)) for key, value in re.findall(r'\.(\w+)\s*=\s*(-?\d+)', row))
              for row in re.findall(r'\{([^{}]+)\}', block)]
    subset = bytearray(); descriptions = []
    for c in '0123456789:-':
        g = glyphs[ord(c)-32+1]
        count = (g['box_w'] * g['box_h'] + 1) // 2
        raw = bitmap[g['bitmap_index']:g['bitmap_index']+count]
        assert len(raw) == count
        descriptions.append(f"    {{{len(subset)}, {(g['adv_w']+8)//16}, {g['box_w']}, {g['box_h']}, {g['ofs_x']}, {g['ofs_y']}}}, /* {c} */")
        subset.extend(raw)
    text = ('/* Generated by tools/prepare_status_hud_assets.py; do not edit.\n'
            ' * Font subset derived from LVGL Montserrat 14, Copyright (c) 2025 LVGL Kft.\n'
            ' * See components/platform/assets/status_hud/LVGL-LICENSE.txt. */\n'
            '#include "lcd_status_assets.h"\n\n')
    text += array('lcd_status_font_bitmap', subset).replace('static const', 'const')
    text += 'const lcd_status_glyph_t lcd_status_font_glyphs[12] = {\n' + '\n'.join(descriptions) + '\n};\n'
    payload = 0
    for i, name in enumerate(NAMES):
        source = (ASSETS / 'battery_source' / name).read_bytes()
        image = Image.open(io.BytesIO(source)).convert('RGBA')
        assert image.size == (40, 20), name
        out = io.BytesIO(); image.save(out, format='PNG', optimize=True, compress_level=9)
        optimized = min((source, out.getvalue()), key=len)
        assert Image.open(io.BytesIO(optimized)).convert('RGBA').tobytes() == image.tobytes()
        destination = ASSETS / 'battery_optimized' / name
        if check:
            assert destination.read_bytes() == optimized, name
        else:
            destination.parent.mkdir(parents=True, exist_ok=True); destination.write_bytes(optimized)
        rgba = image.tobytes()
        pixels = [(((rgba[j] >> 3) << 11) | ((rgba[j+1] >> 2) << 5) | (rgba[j+2] >> 3), rgba[j+3])
                  for j in range(0, len(rgba), 4)]
        palette = list(dict.fromkeys(pixels)); assert len(palette) <= 256
        indices = {p: n for n, p in enumerate(palette)}; runs = bytearray()
        for p in pixels:
            index = indices[p]
            if runs and runs[-1] == index and runs[-2] < 255: runs[-2] += 1
            else: runs.extend((1, index))
        decoded = [palette[runs[j+1]] for j in range(0, len(runs), 2) for _ in range(runs[j])]
        assert decoded == pixels, name
        pal = bytes(v for color, alpha in palette for v in (color >> 8, color & 255, alpha))
        text += array(f'battery_palette_{i}', pal) + array(f'battery_runs_{i}', runs)
        payload += len(pal)+len(runs)
        print(f'{name}: PNG {len(source)} -> {len(optimized)} B; indexed RLE {len(pal)+len(runs)} B')
    text += 'const lcd_status_icon_t lcd_status_battery_icons[10] = {\n'
    text += ''.join(f'    {{battery_palette_{i}, battery_runs_{i}, sizeof(battery_runs_{i})}},\n' for i in range(10)) + '};\n'
    if check: assert OUTPUT.read_text() == text
    else: OUTPUT.parent.mkdir(parents=True, exist_ok=True); OUTPUT.write_text(text)
    print(f'Font bitmap {len(subset)} B; battery payload {payload} B; no runtime pixel allocations')

if __name__ == '__main__':
    parser = argparse.ArgumentParser(); parser.add_argument('--import-zip', type=Path); parser.add_argument('--check', action='store_true')
    args = parser.parse_args()
    if args.import_zip:
        assert not args.check
        with ZipFile(args.import_zip) as archive:
            for name in NAMES:
                data = archive.read('out/40x20/' + name)
                image = Image.open(io.BytesIO(data)); assert image.size == (40, 20)
                path = ASSETS / 'battery_source' / name
                path.parent.mkdir(parents=True, exist_ok=True); path.write_bytes(data)
    generate(args.check)