36 lines
1.2 KiB (Stored with Git LFS)
Python

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\"")