1 files changed, 61 insertions, 0 deletions
diff --git a/src/app/deflatetest/benchmark.py b/src/app/deflatetest/benchmark.py
new file mode 100755
index 0000000..c9fc968
--- /dev/null
+++ b/src/app/deflatetest/benchmark.py
@@ -0,0 +1,61 @@
+#!/usr/bin/env python3
+
+import numpy as np
+import re
+import subprocess
+import sys
+
+
+def main(arch):
+    make_args = [f"arch={arch}", "app=deflatetest"]
+    base_rom = None
+    base_ram = None
+    defl_rom = None
+    defl_ram = None
+    defl_ms = None
+    defl_size = None
+
+    status = subprocess.run(
+        ["./mp"] + make_args + ["deflate_nop=1"],
+        stdout=subprocess.PIPE,
+        universal_newlines=True,
+    )
+
+    for line in status.stdout.split("\n"):
+        match = re.search(r"ROM: (\d+) .* RAM: (\d+)", line)
+        if match:
+            base_rom = int(match.group(1))
+            base_ram = int(match.group(2))
+
+    status = subprocess.run(
+        ["./mp"] + make_args, stdout=subprocess.PIPE, universal_newlines=True
+    )
+
+    for line in status.stdout.split("\n"):
+        match = re.search(r"ROM: (\d+) .* RAM: (\d+)", line)
+        if match:
+            defl_rom = int(match.group(1))
+            defl_ram = int(match.group(2))
+
+    rom_usage = int(np.ceil((defl_rom - base_rom) / 16)) * 16
+    ram_usage = int(np.ceil((defl_ram - base_ram) / 16)) * 16
+    print(f"| {arch} | {rom_usage} B | {ram_usage} B")
+
+    status = subprocess.run(
+        ["make", "cat"] + make_args, stdout=subprocess.PIPE, universal_newlines=True
+    )
+
+    for line in status.stdout.split("\n"):
+        match = re.match(r"took ([0-9.]+) ms", line)
+        if match:
+            defl_ms = float(match.group(1))
+        match = re.match(r"inflate returned (\d+)", line)
+        if match:
+            defl_size = int(match.group(1))
+
+    defl_speed = defl_size / defl_ms
+    print(f"{defl_speed:3.0f} kB/s    ({defl_size:4d} B @ {defl_ms:6.2f} ms)")
+
+
+if __name__ == "__main__":
+    main(*sys.argv[1:])