From fcd3d4ff6c58b2c9dd8b0d4d2cc038286ddfc081 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?J=C4=99drzej=20Boczar?= <jboczar@antmicro.com>
Date: Tue, 4 Feb 2020 13:45:08 +0100
Subject: [PATCH] test: helper scripts for generating benchmark
 configurations/access patterns

---
 test/gen_access_pattern.py | 30 +++++++++++++++++
 test/gen_config.py         | 66 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 96 insertions(+)
 create mode 100755 test/gen_access_pattern.py
 create mode 100755 test/gen_config.py

diff --git a/test/gen_access_pattern.py b/test/gen_access_pattern.py
new file mode 100755
index 0000000..f6cd200
--- /dev/null
+++ b/test/gen_access_pattern.py
@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+
+import random
+import argparse
+
+desc = """
+Generate random access pattern.
+Each address in range [base, base+length) will be accessed only once,
+but in random order. This ensures that no data will be overwritten.
+"""
+parser = argparse.ArgumentParser(description=desc)
+parser.add_argument('base',       help='Base address')
+parser.add_argument('length',     help='Number of (address, data) pairs')
+parser.add_argument('data_width', help='Width of data (used to determine max value)')
+parser.add_argument('--seed',     help='Use given random seed (int)')
+args = parser.parse_args()
+
+if args.seed:
+    random.seed(int(args.seed, 0))
+
+base = int(args.base, 0)
+length = int(args.length, 0)
+data_width = int(args.data_width, 0)
+
+address = list(range(length))
+random.shuffle(address)
+data = [random.randrange(0, 2**data_width) for _ in range(length)]
+
+for a, d in zip(address, data):
+    print('0x{:08x},0x{:08x}'.format(a, d))
diff --git a/test/gen_config.py b/test/gen_config.py
new file mode 100755
index 0000000..a0b9188
--- /dev/null
+++ b/test/gen_config.py
@@ -0,0 +1,66 @@
+#!/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]
+
+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('--name-format',                  default='test_%d',    help='Name format for i-th test')
+args = parser.parse_args()
+
+product = itertools.product(args.sdram_modules, args.sdram_data_widths, args.bist_lengths, args.bist_randoms)
+configurations = {}
+for i, (module, data_width, bist_length, bist_random) in enumerate(product):
+    configurations[args.name_format % i] = {
+        'sdram_module':     module,
+        'sdram_data_width': data_width,
+        'bist_length':      bist_length,
+        'bist_random':      bist_random,
+    }
+
+json_str = json.dumps(configurations, indent=4)
+print(json_str)