ev111ba,ev111a,seton,Ba

ev111bb,ev111a,seton,Bb:Ba

import os
import re
import json
from PIL import Image

EXPORT_DIR = "export"

def draw_layer(canvas, img_dir, layer):
    if not layer.get("visible", 1):
        return
    png = os.path.join(img_dir, f"{layer['layer_id']}.png")
    if not os.path.exists(png):
        return
    img = Image.open(png).convert("RGBA")
    opacity = layer.get("opacity", 255)
    if opacity < 255:
        alpha = img.split()[-1].point(lambda p: p * opacity // 255)
        img.putalpha(alpha)
    canvas.alpha_composite(img, (layer["left"], layer["top"]))

def find_layer(layers, name):
    for l in layers:
        if l.get("name") == name:
            return l
    raise KeyError(f"Layer not found: {name}")

def process_rule(line):
    line = line.strip()
    if not line or line.startswith("#"):
        return

    out_name, base_name, mode, rule = [x.strip() for x in line.split(",")]

    if mode != "seton" and mode != "diff":
        print(f"[SKIP] unsupported mode: {mode}")
        return

    json_file = f"{base_name}.json"
    img_dir = base_name

    if not os.path.exists(json_file):
        print(f"[MISS] {json_file}")
        return
    if not os.path.isdir(img_dir):
        print(f"[MISS] image dir {img_dir}")
        return

    data = json.load(open(json_file, "r", encoding="utf-8"))
    W, H = data["width"], data["height"]
    layers = data["layers"]

    # ---- 解析规则 (沿用 test.py 的逻辑) ----
    if ":" in rule:
        top_name, base_layer_name = rule.split(":")
        order = [base_layer_name, top_name]
    else:
        order = [rule]
    
    # diff模式特殊处理
    if mode == "diff":
        # 对于 diff 模式，需要找到母图
        base_candidate = rule[0]  # Bb -> B
        # 把母图加在前面
        if find_layer(layers, base_candidate):
            order = [base_candidate] + order

    print(f"[DO] {out_name}: {' + '.join(order)}")

    canvas = Image.new("RGBA", (W, H), (0, 0, 0, 0))

    for lname in order:
        layer = find_layer(layers, lname)
        draw_layer(canvas, img_dir, layer)

    os.makedirs(EXPORT_DIR, exist_ok=True)
    out_png = os.path.join(EXPORT_DIR, f"{out_name}.png")
    canvas.save(out_png)

def main():
    csv_file = "imagediffmap.csv"
    
    if not os.path.exists(csv_file):
        print(f"[ERROR] {csv_file} not found!")
        return
    
    # 用正则匹配 ev 开头的规则行
    pattern = re.compile(r'^(ev\w+),(ev\w+),(seton|diff),([^,]+)$')
    
    rules = []
    with open(csv_file, "r", encoding="utf-16") as f:
        for line in f:
            line = line.strip()
            if not line:
                continue
            match = pattern.match(line)
            if match:
                rules.append(line)
    
    if not rules:
        print("[WARN] No ev rules found!")
        return
    
    print(f"Found {len(rules)} rules.")
    
    for line in rules:
        try:
            process_rule(line)
        except Exception as e:
            print(f"[ERR] {line}: {e}")
    
    print("\n[DONE]")

if __name__ == "__main__":
    main()