import os

src = "data"
demiter_aig_dir = "demiter_aig"
demiter_verilog_dir = "demiter_verilog"

os.makedirs(demiter_aig_dir, exist_ok=True)
for filename in os.listdir(src):

    print("processing: ", filename)

    filepath = os.path.join(src, filename)
    os.system(f"~/toolbox/abc/abc -c \"&r {filepath}; &demiter -d -v; &demiter -f\"")

    if filename.endswith('.aiger'):
        base = filename[:-6]  # 移除 .aiger 后缀
        src_1 = os.path.join(src, f"{base}_1.aig") 
        src_2 = os.path.join(src, f"{base}_2.aig")
        
        dst_dir = demiter_aig_dir  # 指定目标文件夹
        os.makedirs(dst_dir, exist_ok=True)
        
        dst_1 = os.path.join(dst_dir, f"{base}_1.aig")
        dst_2 = os.path.join(dst_dir, f"{base}_2.aig")
        
        if os.path.exists(src_1):
            os.rename(src_1, dst_1)
        if os.path.exists(src_2):
            os.rename(src_2, dst_2)


os.makedirs(demiter_verilog_dir, exist_ok=True)
for filename in os.listdir(demiter_aig_dir):
    filepath = os.path.join(demiter_aig_dir, filename)
    os.system(f"~/toolbox/abc/abc -c \"read_aiger {filepath}; strash; write_verilog {demiter_verilog_dir}/{filename[:-4]}.v\"")