import sys
import filecmp
import subprocess
import sys
import os


# 主函数
def main():
    rtl_dir = sys.argv[1]

    if rtl_dir != r'..':
        tb_file = r'/tb/compliance_test/tinyriscv_soc_tb.v'
    else:
        tb_file = r'/tb/tinyriscv_soc_tb.v'

    # iverilog程序
    iverilog_cmd = ['iverilog']
    # 顶层模块
    #iverilog_cmd += ['-s', r'tinyriscv_soc_tb']
    # 编译生成文件
    iverilog_cmd += ['-o', r'out.vvp']
    # 头文件(defines.v)路径
    iverilog_cmd += ['-I', rtl_dir + r'/rtl/core']
    # 宏定义，仿真输出文件
    iverilog_cmd += ['-D', r'OUTPUT="signature.output"']
    # testbench文件
    iverilog_cmd.append(rtl_dir + tb_file)
    # ../rtl/core
    iverilog_cmd.append(rtl_dir + r'/rtl/core/clint.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/csr_reg.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/defines.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/divider.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/exu.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/exu_alu_datapath.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/exu_commit.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/exu_dispatch.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/exu_mem.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/exu_muldiv.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/gpr_reg.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/idu.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/idu_exu.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/ifu.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/ifu_idu.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/pipe_ctrl.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/tinyriscv_core.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/core/rst_ctrl.v')
    # ../rtl/perips
    iverilog_cmd.append(rtl_dir + r'/rtl/perips/ram.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/perips/rom.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/perips/timer.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/perips/uart.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/perips/gpio.v')
    # ../rtl/debug
    iverilog_cmd.append(rtl_dir + r'/rtl/debug/jtag_dm.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/debug/jtag_driver.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/debug/jtag_top.v')
    # ../rtl/sys_bus
    iverilog_cmd.append(rtl_dir + r'/rtl/sys_bus/rib.v')
    # ../rtl/utils
    iverilog_cmd.append(rtl_dir + r'/rtl/utils/full_handshake_rx.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/utils/full_handshake_tx.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/utils/gen_buf.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/utils/gen_dff.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/utils/gen_ram.v')
    iverilog_cmd.append(rtl_dir + r'/rtl/utils/vld_rdy.v')
    # ../rtl/top
    iverilog_cmd.append(rtl_dir + r'/rtl/top/tinyriscv_soc_top.v')

    # 编译
    process = subprocess.Popen(iverilog_cmd)
    process.wait(timeout=5)

if __name__ == '__main__':
    sys.exit(main())