153 lines
5.7 KiB
Python
153 lines
5.7 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
check_content_pages.py — 三个叶子内容页（装置信息/交流量/直流量）像素黄金参考回归。
|
||
|
||
用法：
|
||
python simulator/tests/check_content_pages.py # 与入库黄金参考逐字节比对
|
||
python simulator/tests/check_content_pages.py --update # 有意改视觉后刷新黄金参考
|
||
前置：simulator 已构建（build/Debug/lvgl_sim.exe 存在）；纯标准库，任意 python3。
|
||
|
||
原理：每页用键序注入打开（info=e / ac=dee / dc=dede），无头截图 I1(160x160)
|
||
与 ref/shot_<page>.bin 逐字节比对——任何像素变化即 FAIL（含字体、几何、颜色）。
|
||
另做开页-关页往返池水位门禁：`es` 后 used 须等于 POOL_AFTER_ROUNDTRIP。
|
||
|
||
黄金参考更新流程（有意改视觉时）：先肉眼看图确认新效果正确，再 --update 刷新，
|
||
最后 git diff 检查 ref/ 变化符合预期。误报排查见 docs/UI 设计/UI_自动化测试流程.md。
|
||
|
||
退出码：0 全 PASS；1 有页不一致或池水位漂移；2 运行/环境错误（exe 缺失、参考缺失）。
|
||
"""
|
||
import os
|
||
import re
|
||
import shutil
|
||
import subprocess
|
||
import sys
|
||
import tempfile
|
||
from pathlib import Path
|
||
|
||
# lvgl_sim.exe 相对本脚本定位（simulator/tests/ -> simulator/build/Debug/）
|
||
EXE = Path(__file__).resolve().parents[1] / "build" / "Debug" / "lvgl_sim.exe"
|
||
REF_DIR = Path(__file__).resolve().parent / "ref"
|
||
|
||
# (页名, 打开键序)：与 ui_menu.c menu_content_page_for() 的节点映射一致
|
||
PAGES = [
|
||
("info", "e"), # 装置信息（node 1）
|
||
("ac", "dee"), # 交流量（node 3，实时数据下第 1 项）
|
||
("dc", "dede"), # 直流量（node 4，实时数据下第 2 项）
|
||
]
|
||
|
||
W = H = 160
|
||
STRIDE = W // 8 # I1：20 字节/行
|
||
|
||
# 开页-关页往返后的 LVGL 池水位（x64 模拟器）= 菜单基线 + 一次性残留 80 B。
|
||
# 页面结构变更导致漂移时，先确认新值合理再更新此常量（内存回归门禁）。
|
||
POOL_AFTER_ROUNDTRIP = 4864
|
||
|
||
|
||
def run_sim(seq, out):
|
||
r = subprocess.run([str(EXE), "--keys", seq, out], capture_output=True)
|
||
if r.returncode != 0:
|
||
raise RuntimeError("lvgl_sim --keys %r rc=%d: %s"
|
||
% (seq, r.returncode, r.stderr.decode(errors="replace")[:200]))
|
||
return r.stdout
|
||
|
||
|
||
def ink_runs(data):
|
||
"""I1 位图按行统计白像素，返回 [(y0,y1,count), ...]（count>2 的连续段）。"""
|
||
px = []
|
||
for y in range(H):
|
||
row = data[y * STRIDE:(y + 1) * STRIDE]
|
||
cnt = sum((b >> (7 - x)) & 1 for b in row for x in range(8))
|
||
px.append(cnt)
|
||
runs, start = [], None
|
||
for y, c in enumerate(px):
|
||
if c > 2 and start is None:
|
||
start = y
|
||
elif c <= 2 and start is not None:
|
||
runs.append((start, y - 1))
|
||
start = None
|
||
if start is not None:
|
||
runs.append((start, H - 1))
|
||
return [(a, b, sum(px[a:b + 1])) for a, b in runs]
|
||
|
||
|
||
def diff_count(a, b):
|
||
n = 0
|
||
for y in range(H):
|
||
ra = a[y * STRIDE:(y + 1) * STRIDE]
|
||
rb = b[y * STRIDE:(y + 1) * STRIDE]
|
||
for x in range(STRIDE):
|
||
n += bin(ra[x] ^ rb[x]).count("1")
|
||
return n
|
||
|
||
|
||
def main():
|
||
update = "--update" in sys.argv[1:]
|
||
if not EXE.exists():
|
||
print("FAIL: 找不到 %s（先构建 simulator）" % EXE)
|
||
return 2
|
||
|
||
tmp = tempfile.mkdtemp(prefix="dtu_ui_pages_")
|
||
ok = True
|
||
for name, seq in PAGES:
|
||
ref = REF_DIR / ("shot_%s.bin" % name)
|
||
shot = os.path.join(tmp, "shot_%s.bin" % name)
|
||
run_sim(seq, shot)
|
||
data = open(shot, "rb").read()
|
||
|
||
if update:
|
||
REF_DIR.mkdir(exist_ok=True)
|
||
ref.write_bytes(data)
|
||
print("[%s] 黄金参考已刷新 -> %s（键序 %r）" % (name, ref.name, seq))
|
||
continue
|
||
|
||
if not ref.exists():
|
||
print("FAIL: 缺黄金参考 %s（首次运行请加 --update 生成并入库）" % ref)
|
||
shutil.rmtree(tmp, ignore_errors=True)
|
||
return 2
|
||
refdata = ref.read_bytes()
|
||
if data == refdata:
|
||
print("[%s] PASS 与黄金参考一致（键序 %r）" % (name, seq))
|
||
continue
|
||
|
||
ok = False
|
||
n = diff_count(data, refdata)
|
||
print("[%s] FAIL 与黄金参考不一致：%d px 差异（键序 %r）" % (name, n, seq))
|
||
print(" 当前截图 ink runs: %s" % ink_runs(data))
|
||
print(" 黄金参考 ink runs: %s" % ink_runs(refdata))
|
||
print(" 有意改视觉？确认新效果正确后运行 --update 刷新参考。")
|
||
|
||
# 开页-关页往返池水位门禁（info 页：e 打开、s 关闭回菜单）
|
||
out = run_sim("es", os.path.join(tmp, "shot_rt.bin"))
|
||
m = re.search(r"pool after keys: used=(\d+)", out.decode(errors="replace"))
|
||
if not m:
|
||
print("FAIL: 往返输出未找到 'pool after keys'")
|
||
shutil.rmtree(tmp, ignore_errors=True)
|
||
return 2
|
||
used = int(m.group(1))
|
||
if update:
|
||
print("[rt] 开页-关页往返池水位 %d B（--update 不校验，仅提示）" % used)
|
||
elif used != POOL_AFTER_ROUNDTRIP:
|
||
ok = False
|
||
print("[rt] FAIL 往返池水位 %d != 预期 %d（页面结构变更？确认合理后更新常量）"
|
||
% (used, POOL_AFTER_ROUNDTRIP))
|
||
else:
|
||
print("[rt] PASS 开页-关页往返池水位 %d B" % used)
|
||
|
||
if update:
|
||
print("PASS: 黄金参考已刷新（%d 页）" % len(PAGES))
|
||
shutil.rmtree(tmp, ignore_errors=True)
|
||
return 0
|
||
|
||
if ok:
|
||
print("PASS: 3 内容页 + 往返池水位全部通过")
|
||
shutil.rmtree(tmp, ignore_errors=True)
|
||
return 0
|
||
# 失败时保留临时截图便于排查
|
||
print("（现场截图保留在 %s）" % tmp)
|
||
return 1
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|