From b156fe903f75b589cb270c062a5cbd65f55e5cf6 Mon Sep 17 00:00:00 2001
From: Jannis Harder <me@jix.one>
Date: Fri, 5 Aug 2022 15:24:28 +0200
Subject: yosys-witness: Add stats command

---
 backends/smt2/witness.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'backends')

diff --git a/backends/smt2/witness.py b/backends/smt2/witness.py
index 03d72a17b..a1e65569d 100644
--- a/backends/smt2/witness.py
+++ b/backends/smt2/witness.py
@@ -62,6 +62,24 @@ def display(input):
                 yield f"  {step_prefix} {sig.pretty()} = {display_bits}"
     click.echo_via_pager([line + "\n" for line in output()])
 
+
+@cli.command(help="""
+Display statistics of a Yosys witness trace.
+""")
+@click.argument("input", type=click.File("r"))
+def stats(input):
+    click.echo(f"Reading Yosys witness trace {input.name!r}...")
+    inyw = ReadWitness(input)
+
+    total = 0
+
+    for t, values in inyw.steps():
+        click.echo(f"{t:5}: {len(values.values):8} bits")
+        total += len(values.values)
+
+    click.echo(f"total: {total:8} bits")
+
+
 @cli.command(help="""
 Transform a Yosys witness trace.
 
-- 
cgit v1.2.3