"""实际 HUD 函数与生成资产的 ASan/UBSan 回归；不启动 LVGL/GFX 引擎。"""
from pathlib import Path
import subprocess
import tempfile
import re

ROOT = Path(__file__).resolve().parents[1]
src = (ROOT/'components/platform/src/lcd_direct.c').read_text()
def function(name):
    matches = list(re.finditer(r'static [^;{}]+\b'+name+r'\([^;{}]*\)\s*\{', src))
    assert len(matches) == 1, name
    start = matches[0].start(); p = src.index('{', start); depth = 1; end = p+1
    while depth:
        if src[end] == '{': depth += 1
        if src[end] == '}': depth -= 1
        end += 1
    return src[start:end]
prefix = '''#include <stdint.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <assert.h>
#include "lcd_status_assets.h"
#define LCD_DIRECT_CLOCK_X 12U
#define LCD_DIRECT_CLOCK_Y 14U
#define LCD_DIRECT_CLOCK_TEXT_LENGTH 5U
#define LCD_DIRECT_CLOCK_COLOR 0xffffU
#define LCD_DIRECT_CLOCK_OUTLINE_COLOR 0U
#define LCD_DIRECT_BATTERY_WIDTH 40U
#define LCD_DIRECT_BATTERY_HEIGHT 20U
#define LCD_DIRECT_BATTERY_X (LCD_DIRECT_WIDTH - 48U)
#define LCD_DIRECT_BATTERY_Y 12U
#define portENTER_CRITICAL(x) ((void)0)
#define portEXIT_CRITICAL(x) ((void)0)
typedef struct {bool low_battery_visible, charging_visible, battery_valid; uint8_t battery_percent;} platform_lcd_status_overlay_t;
static platform_lcd_status_overlay_t s_status_overlay;
#define PLATFORM_LCD_PRESENTATION_IDLE 0
typedef struct {int kind;} lcd_direct_request_t;
static lcd_direct_request_t s_pending;
static bool s_pending_valid, s_stop_requested;
'''
test = '''int main(int argc, char **argv) {
    uint16_t full[LCD_DIRECT_WIDTH*240+2], blocks[LCD_DIRECT_WIDTH*240+2];
    for (unsigned valid=0;valid<2;++valid)
    for (unsigned charge=0;charge<2;++charge)
    for (unsigned low=0;low<2;++low)
    for (unsigned percent=0;percent<=255;++percent) {
        s_status_overlay=(platform_lcd_status_overlay_t){low,charge,valid,percent};
        for (unsigned i=0;i<LCD_DIRECT_WIDTH*240+2;++i) full[i]=blocks[i]=0x1234;
        lcd_direct_overlay_battery(full+1,0,240);
        for(unsigned row=0;row<240;row+=8)
            lcd_direct_overlay_battery(blocks+1+row*LCD_DIRECT_WIDTH,row,8);
        assert(memcmp(full,blocks,sizeof(full))==0);
        assert(full[0]==0x1234 && full[LCD_DIRECT_WIDTH*240+1]==0x1234);
        unsigned level=percent<=10?0:percent<=25?1:percent<=50?2:percent<=75?3:4;
        assert(lcd_direct_battery_icon_index(percent,charge)==level+(charge?5:0));
        for(unsigned y=0;y<240;++y) for(unsigned x=0;x<LCD_DIRECT_WIDTH;++x) {
            if(!valid || x<LCD_DIRECT_BATTERY_X || x>=LCD_DIRECT_BATTERY_X+40 || y<12 || y>=32)
                assert(full[1+y*LCD_DIRECT_WIDTH+x]==0x1234);
        }
        if(valid) {
            const lcd_status_icon_t *icon=&lcd_status_battery_icons[level+(charge?5:0)];
            unsigned p=0;
            for(unsigned r=0;r<icon->run_bytes;r+=2) {
                const uint8_t *value=icon->palette+icon->runs[r+1]*3;
                for(unsigned n=0;n<icon->runs[r];++n,++p) {
                    unsigned index=1+(12+p/40)*LCD_DIRECT_WIDTH+LCD_DIRECT_BATTERY_X+p%40;
                    uint16_t color=((uint16_t)value[0]<<8)|value[1];
                    assert(full[index]==lcd_direct_swap_color(lcd_direct_blend_rgb565(lcd_direct_swap_color(0x1234),color,value[2])));
                }
            }
            assert(p==800);
        }
    }
    for(unsigned hour=0;hour<24;++hour) for(unsigned minute=0;minute<60;++minute) {
        char text[6]; snprintf(text,sizeof(text),"%02u:%02u",hour,minute);
        for(unsigned i=0;i<LCD_DIRECT_WIDTH*240+2;++i) full[i]=blocks[i]=0x1234;
        lcd_direct_overlay_clock(full+1,0,240,text);
        for(unsigned row=0;row<240;row+=8)
            lcd_direct_overlay_clock(blocks+1+row*LCD_DIRECT_WIDTH,row,8,text);
        assert(memcmp(full,blocks,sizeof(full))==0);
        assert(full[0]==0x1234 && full[LCD_DIRECT_WIDTH*240+1]==0x1234);
        for(unsigned y=0;y<240;++y) for(unsigned x=0;x<LCD_DIRECT_WIDTH;++x)
            if(x<12 || x>=64 || y<14 || y>=31) assert(full[1+y*LCD_DIRECT_WIDTH+x]==0x1234);
    }
    lcd_direct_overlay_clock(full+1,0,240,"--:--");
    lcd_direct_overlay_clock(NULL,0,240,"--:--");
    lcd_direct_overlay_clock(full+1,0,240,NULL);
    lcd_direct_overlay_battery(NULL,0,240);
    for(unsigned pending=0;pending<2;++pending)
    for(unsigned stopping=0;stopping<2;++stopping) {
        s_pending=(lcd_direct_request_t){99}; s_pending_valid=pending; s_stop_requested=stopping;
        lcd_direct_queue_idle_if_available();
        assert(s_pending.kind==((pending || stopping)?99:PLATFORM_LCD_PRESENTATION_IDLE));
        assert(s_pending_valid==(pending || !stopping));
    }
    if(argc==2) {
        memset(full,0,sizeof(full));
        lcd_direct_overlay_clock(full,0,240,"12:34");
        s_status_overlay=(platform_lcd_status_overlay_t){false,true,true,50};
        lcd_direct_overlay_battery(full,0,240);
        FILE *f=fopen(argv[1],"wb"); assert(f); fwrite(full,2,LCD_DIRECT_WIDTH*240,f); fclose(f);
    }
    puts("PASS: 2048 battery states, all 1440 times, alpha/background, byte order, 8-row equivalence, bounds and idle guards");
}'''
names = ['lcd_direct_swap_color', 'lcd_direct_rgb888_to_rgb565', 'lcd_direct_blend_rgb565', 'lcd_direct_clock_glyph_index',
         'lcd_direct_hud_blend_pixel', 'lcd_direct_draw_clock_pass', 'lcd_direct_overlay_clock',
         'lcd_direct_battery_icon_index', 'lcd_direct_overlay_battery', 'lcd_direct_queue_idle_if_available']
with tempfile.TemporaryDirectory(prefix='lucky-hud-') as tmp:
    c = Path(tmp)/'hud.c'; exe = Path(tmp)/'hud'
    c.write_text(prefix+'\n'+'\n'.join(function(n) for n in names)+'\n'+test)
    for width in (320, 240):
        for swap in (0, 1):
            subprocess.run(['clang','-std=c11','-fsanitize=address,undefined','-g',
                            f'-DLCD_DIRECT_WIDTH={width}', f'-DCONFIG_TQ_SCREEN_SWAP_COLOR_BYTES={swap}',
                            '-I'+str(ROOT/'components/platform/internal'),str(c),
                            str(ROOT/'components/platform/src/generated/lcd_status_assets.c'),
                            '-o',str(exe)],check=True)
            preview = ['/tmp/lucky-status-hud.rgb565'] if width == 320 and swap == 0 else []
            subprocess.run([str(exe), *preview],check=True)