import os
import random
random.seed(int(os.getenv("SEED"), 16))
from prjxray import util
from prjxray.db import Database


def gen_sites():
    db = Database(util.get_db_root())
    grid = db.grid()
    for tile_name in sorted(grid.tiles()):
        loc = grid.loc_of_tilename(tile_name)
        gridinfo = grid.gridinfo_at_loc(loc)
        if gridinfo.tile_type in ['DSP_L', 'DSP_R']:
            for site in sorted(gridinfo.sites.keys()):
                if gridinfo.sites[site] == 'DSP48E1':
                    yield tile_name, site


def write_params(lines):
    pinstr = 'tile,site,a_input,b_input,autoreset_patdet,mask,pattern\n'
    for tile, site, a_input, b_input, autoreset_patdet, mask, pattern in lines:
        pinstr += '%s,' % (tile)
        pinstr += '%s,' % (site)
        pinstr += '%s,' % (a_input)
        pinstr += '%s,' % (b_input)
        pinstr += '%s,' % (autoreset_patdet)
        pinstr += '%s,' % (mask)
        pinstr += '%s\n' % (pattern)

    open('params.csv', 'w').write(pinstr)


def run():
    print('''
module top();
    ''')

    lines = []

    sites = list(gen_sites())
    for (tile_name, site_name) in sites:
        mask = random.randint(0, 2**48 - 1)
        pattern = random.randint(0, 2**48 - 1)
        a_input = random.choice(('DIRECT', 'CASCADE'))
        b_input = random.choice(('DIRECT', 'CASCADE'))
        autoreset_patdet = random.choice(
            ('NO_RESET', 'RESET_MATCH', 'RESET_NOT_MATCH'))
        lines.append(
            (
                tile_name, site_name, a_input, b_input, autoreset_patdet, mask,
                pattern))

        print(
            '''
            (* KEEP, DONT_TOUCH, LOC = "{0}" *)
            DSP48E1 #(
                .A_INPUT("{1}"),
                .B_INPUT("{2}"),
                .AUTORESET_PATDET("{3}"),
                .MASK(48'h{4:x}),
                .PATTERN(48'h{5:x})
            ) dsp_{0} (
            );
'''.format(site_name, a_input, b_input, autoreset_patdet, mask, pattern))

    print("endmodule")
    write_params(lines)


if __name__ == '__main__':
    run()