#!/usr/bin/env python

import json
import argparse
import itertools

modules = [
    'IS42S16160',
    'IS42S16320',
    'MT48LC4M16',
    'MT48LC16M16',
    'AS4C16M16',
    'AS4C32M16',
    'AS4C32M8',
    'M12L64322A',
    'M12L16161A',
    'MT46V32M16',
    'MT46H32M16',
    'MT46H32M32',
    'MT47H128M8',
    'MT47H32M16',
    'MT47H64M16',
    'P3R1GE4JGF',
    'MT41K64M16',
    'MT41J128M16',
    'MT41K128M16',
    'MT41J256M16',
    'MT41K256M16',
    'K4B1G0446F',
    'K4B2G1646F',
    'H5TC4G63CFR',
    'IS43TR16128B',
    'MT8JTF12864',
    'MT8KTF51264',
    #  'MT18KSF1G72HZ',
    #  'AS4C256M16D3A',
    #  'MT16KTF1G64HZ',
    #  'EDY4016A',
    #  'MT40A1G8',
    #  'MT40A512M16',
]
data_widths = [32]
bist_lengths = [1, 1024, 8192]
bist_randoms = [False]
access_patterns = ['access_pattern.csv']

def main():
    parser = argparse.ArgumentParser(description='Generate configuration for all possible argument combinations.',
                                     formatter_class=argparse.ArgumentDefaultsHelpFormatter)
    parser.add_argument('--sdram-modules',     nargs='+', default=modules,         help='--sdram-module options')
    parser.add_argument('--sdram-data-widths', nargs='+', default=data_widths,     help='--sdram-data-width options')
    parser.add_argument('--bist-lengths',      nargs='+', default=bist_lengths,    help='--bist-length options')
    parser.add_argument('--bist-randoms',      nargs='+', default=bist_randoms,    help='--bist-random options')
    parser.add_argument('--access-patterns',   nargs='+', default=access_patterns, help='--access-pattern options')
    parser.add_argument('--name-format',                  default='test_%d',       help='Name format for i-th test')
    args = parser.parse_args()

    bist_product = itertools.product(args.sdram_modules, args.sdram_data_widths, args.bist_lengths, args.bist_randoms)
    pattern_product = itertools.product(args.sdram_modules, args.sdram_data_widths, args.access_patterns)

    i = 0
    configurations = {}
    for module, data_width, bist_length, bist_random in bist_product:
        configurations[args.name_format % i] = {
            'sdram_module':     module,
            'sdram_data_width': data_width,
            'access_pattern': {
                'bist_length':  bist_length,
                'bist_random':  bist_random,
            }
        }
        i += 1
    for module, data_width, access_pattern in pattern_product:
        configurations[args.name_format % i] = {
            'sdram_module':     module,
            'sdram_data_width': data_width,
            'access_pattern': {
                'pattern_file': access_pattern,
            }
        }
        i += 1

    json_str = json.dumps(configurations, indent=4)
    print(json_str)

if __name__ == "__main__":
    main()