#!/usr/bin/env python3

from fuzzconfig import *
import numpy as np
import os

os.system("rm -rf work_prim")
os.mkdir("work_prim")

for idx in range(num):
    with open("work_prim/prim_%02d.v" % idx, "w") as f:
        clkedge = np.random.choice(["pos", "neg"])
        print("module top(input clk, input [23:0] a, b, output reg x, output reg [23:0] y);", file=f)
        print("  reg [23:0] aa, bb;", file=f)
        print("  always @(%sedge clk) aa <= a;" % clkedge, file=f)
        print("  always @(%sedge clk) bb <= b;" % clkedge, file=f)
        if np.random.choice([True, False]):
            print("  always @(%sedge clk) x <= %s%s;" % (clkedge, np.random.choice(["^", "&", "|", "!"]), np.random.choice(["a", "b", "y"])), file=f)
        else:
            print("  always @(%sedge clk) x <= a%sb;" % (clkedge, np.random.choice(["&&", "||"])), file=f)
        if np.random.choice([True, False]):
            print("  always @(%sedge clk) y <= a%sb;" % (clkedge, np.random.choice(["+", "-", "&", "|"])), file=f)
        else:
            print("  always @(%sedge clk) y <= %s%s;" % (clkedge, np.random.choice(["~", "-", ""]), np.random.choice(["a", "b"])), file=f)
        print("endmodule", file=f)
    with open("work_prim/prim_%02d.pcf" % idx, "w") as f:
        p = np.random.permutation(pins)
        if np.random.choice([True, False]):
            for i in range(24):
                print("set_io a[%d] %s" % (i, p[i]), file=f)
        if np.random.choice([True, False]):
            for i in range(24):
                print("set_io b[%d] %s" % (i, p[24+i]), file=f)
        if np.random.choice([True, False]):
            for i in range(24):
                print("set_io y[%d] %s" % (i, p[2*24+i]), file=f)
        if np.random.choice([True, False]):
            print("set_io x %s" % p[3*24], file=f)
        if np.random.choice([True, False]):
            print("set_io y %s" % p[3*24+1], file=f)
        if np.random.choice([True, False]):
            print("set_io clk %s" % p[3*24+2], file=f)

with open("work_prim/Makefile", "w") as f:
    print("all: %s" % " ".join(["prim_%02d.bin" % i for i in range(num)]), file=f)
    for i in range(num):
        print("prim_%02d.bin:" % i, file=f)
        print("\t-bash ../icecube.sh prim_%02d > prim_%02d.log 2>&1 && rm -rf prim_%02d.tmp || tail prim_%02d.log" % (i, i, i, i), file=f)