# -*- coding: utf-8 -*-
u"""只显示指定铜层截一张 PCB 图，截完把图层可见性恢复原样。

    /c/Python314/python .claude/shot_pcb_layers.py <out.png> <层号,逗号分隔> [pcbUuid] [x0 y0 x1 y1]

例：`shot_pcb_layers.py L2-inner1.png 15,12 942ed1560d81c1e7`
    —— 15 = Inner1，12 = 多层（通孔焊盘），不带这一层的话内层看不到焊盘位置。

⚠️ `setLayerVisible(list, true)` 的第二个参数是「把其余层设为不可见」。
   截完必须调 `setLayerVisible()` 不带参数**恢复**，否则编辑器就一直只剩那几层，
   下次谁来截图会以为板子空了。
"""
import base64, io, os, sys

sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from eda import run

out = sys.argv[1]
layers = [int(v) for v in sys.argv[2].split(',')]
uuid = sys.argv[3] if len(sys.argv) > 3 else '942ed1560d81c1e7'
MIL = 39.3701
# 可选的框选区域（mm）。⚠️ zoomToRegion 的参数顺序是 (left, right, top, bottom)。
if len(sys.argv) > 7:
    x0, y0, x1, y1 = [float(v) * MIL for v in sys.argv[4:8]]
    ZOOM = u'await eda.dmt_EditorControl.zoomToRegion(%f, %f, %f, %f);' % (x0, x1, y1, y0)
else:
    ZOOM = u'await eda.dmt_EditorControl.zoomToAllPrimitives();'

JS = u"""
const sleep = ms => new Promise(r => setTimeout(r, ms));
await eda.dmt_EditorControl.openDocument('%s');
await sleep(2000);
const before = (await eda.pcb_Layer.getAllLayers() || [])
                 .filter(l => l.visible !== false)
                 .map(l => l.id !== undefined ? l.id : l.layer);
await eda.pcb_Layer.setLayerVisible(%s, true);
await sleep(1200);
%s
await sleep(1800);
const img = await eda.dmt_EditorControl.getCurrentRenderedAreaImage();
const bytes = new Uint8Array(await new Response(img).arrayBuffer());
let s = ''; for (let i = 0; i < bytes.length; i++) s += String.fromCharCode(bytes[i]);
await eda.pcb_Layer.setLayerVisible(before, true);      // 恢复
await sleep(800);
return {n: bytes.length, b64: btoa(s), restored: before.length};
""" % (uuid, layers, ZOOM)

r = run(JS, timeout=300)
assert r.get('success'), str(r)[:500]
io.open(out, 'wb').write(base64.b64decode(r['result']['b64']))
print(out, r['result']['n'], 'bytes; 恢复了', r['result']['restored'], '个可见层')