# Copyright 2020 Project U-Ray Authors
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import numpy as np
import sys

N = 5000

bufgces = []
with open(sys.argv[1], "r") as tf:
	for line in tf:
		sl = line.strip().split(",")
		if len(sl) < 4:
			continue
		for site in sl[4:]:
			if "BUFGCE" in site and "HDIO" not in site:
				bufgces.append(site.split(":")[0])

print("module layer_1(input [31:0] clk, input [71:0] cen, input d, output q);")
print("    wire [%d:0] r;" % N)
print("    assign r[0] = d;")
print("    assign q = r[%d];" % N)
print()
for i in range(N):
	print("    FDCE ff_%d (.C(clk[%d]), .CLR(1'b0), .CE(cen[%d]), .D(r[%d]), .Q(r[%d]));" % (i, (i * 32) // N, np.random.randint(72), i, i+1))
	print()
print("endmodule")


M = 16
print("module top(input [15:0] clk, cen, input d, output q);")
print("    wire [511:0] clk_int;")
print("    wire [71:0] cen_int;")
print("    assign clk_int[15:0] = clk;")
print("    assign cen_int[15:0] = cen;")
print("    assign cen_int[71:64] = 8'hFF;")
for i in range(16, 512):
	a = np.random.randint(16)
	b = None
	while b is None or b == a:
		b = np.random.randint(16)
	c = None
	while c is None or c == a or c == b:
		c = np.random.randint(16)
	bg = None
	if len(bufgces) > 0:
		bg = bufgces.pop()
	if bg is not None and np.random.randint(3) > 0:
		if "DIV" in bg:
			print("    BUFGCE_DIV #(.BUFGCE_DIVIDE(3)) bufg_%d (.I(clk[%d] ^ clk[%d] ^ clk[%d]), .CLR(0), .CE(1'b1), .O(clk_int[%d]));" % (i, a, b, c, i))
		else:
			print("    BUFGCE bufg_%d (.I(clk[%d] ^ clk[%d] ^ clk[%d]), .CE(1'b1), .O(clk_int[%d]));" % (i, a, b, c, i))
	else:
		print("    assign clk_int[%d] = clk[%d] ^ clk[%d] ^ clk[%d];" % (i, a, b, c))
	if i < 64:
		print("    assign cen_int[%d] = cen[%d] ^ cen[%d];" % (i, a, b))
print()
print("    wire [%d:0] r;" % M)
print("    assign r[0] = d;")
print("    assign q = r[%d];" % M)
for i in range(M):
	print("    layer_1 submod_%d(.clk(clk_int[%d +: 32]), .cen(cen_int), .d(r[%d]), .q(r[%d]));" % (i, 32 * i, i, i+1))
print("endmodule")