113 lines
4.0 KiB
Python
Executable File
113 lines
4.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import json
|
|
import argparse
|
|
import itertools
|
|
|
|
default_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',
|
|
]
|
|
default_bist_alternatings = [True, False]
|
|
default_data_widths = [32]
|
|
default_bist_lengths = [1, 1024, 8192]
|
|
default_bist_randoms = [True, False]
|
|
default_access_patterns = ['access_pattern.csv']
|
|
|
|
|
|
def convert_string_arg(args, arg, type):
|
|
map_func = {
|
|
bool: lambda s: {'false': False, 'true': True}[s.lower()],
|
|
int: lambda s: int(s, 0),
|
|
}
|
|
setattr(args, arg, [map_func[type](val) if not isinstance(val, type) else val for val in getattr(args, arg)])
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Generate configuration for all possible argument combinations.',
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
|
parser.add_argument('--sdram-modules', nargs='+', default=default_modules, help='--sdram-module options')
|
|
parser.add_argument('--sdram-data-widths', nargs='+', default=default_data_widths, help='--sdram-data-width options')
|
|
parser.add_argument('--bist-alternatings', nargs='+', default=default_bist_alternatings, help='--bist-alternating options')
|
|
parser.add_argument('--bist-lengths', nargs='+', default=default_bist_lengths, help='--bist-length options')
|
|
parser.add_argument('--bist-randoms', nargs='+', default=default_bist_randoms, help='--bist-random options')
|
|
parser.add_argument('--access-patterns', nargs='+', default=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()
|
|
|
|
# make sure not to write those as strings
|
|
convert_string_arg(args, 'sdram_data_widths', int)
|
|
convert_string_arg(args, 'bist_alternatings', bool)
|
|
convert_string_arg(args, 'bist_lengths', int)
|
|
convert_string_arg(args, 'bist_randoms', bool)
|
|
|
|
bist_product = itertools.product(args.sdram_modules, args.sdram_data_widths, args.bist_alternatings,
|
|
args.bist_lengths, args.bist_randoms)
|
|
pattern_product = itertools.product(args.sdram_modules, args.sdram_data_widths, args.bist_alternatings,
|
|
args.access_patterns)
|
|
|
|
i = 0
|
|
configurations = {}
|
|
for module, data_width, bist_alternating, bist_length, bist_random in bist_product:
|
|
if bist_random and not bist_alternating:
|
|
continue
|
|
configurations[args.name_format % i] = {
|
|
'sdram_module': module,
|
|
'sdram_data_width': data_width,
|
|
'bist_alternating': bist_alternating,
|
|
'access_pattern': {
|
|
'bist_length': bist_length,
|
|
'bist_random': bist_random,
|
|
}
|
|
}
|
|
i += 1
|
|
for module, data_width, bist_alternating, access_pattern in pattern_product:
|
|
if bist_random and not bist_alternating:
|
|
continue
|
|
configurations[args.name_format % i] = {
|
|
'sdram_module': module,
|
|
'sdram_data_width': data_width,
|
|
'bist_alternating': bist_alternating,
|
|
'access_pattern': {
|
|
'pattern_file': access_pattern,
|
|
}
|
|
}
|
|
i += 1
|
|
|
|
json_str = json.dumps(configurations, indent=4)
|
|
print(json_str)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|