From: Eddie Hung <eddie@fpgeh.com>
Date: Fri, 23 Aug 2019 21:32:17 +0000 (-0700)
Subject: Add a unique argument to pmgen's nusers()
X-Git-Tag: working-ls180~1085^2~40
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;h=1d88887cfdbeedff7dce9024d8fb4ceb014cb2ef;p=yosys.git

Add a unique argument to pmgen's nusers()
---

diff --git a/passes/pmgen/pmgen.py b/passes/pmgen/pmgen.py
index 573722d68..2f2545c22 100644
--- a/passes/pmgen/pmgen.py
+++ b/passes/pmgen/pmgen.py
@@ -458,12 +458,16 @@ with open(outfile, "w") as f:
     print("  }", file=f)
     print("", file=f)
 
-    print("  int nusers(const SigSpec &sig) {", file=f)
+    print("  int nusers(const SigSpec &sig, bool unique=true) {", file=f)
+    print("    int i = 0;", file=f)
     print("    pool<Cell*> users;", file=f)
     print("    for (auto bit : sigmap(sig))", file=f)
-    print("      for (auto user : sigusers[bit])", file=f)
-    print("        users.insert(user);", file=f)
-    print("    return GetSize(users);", file=f)
+    print("      if (unique)", file=f);
+    print("        i += GetSize(sigusers[bit]);", file=f);
+    print("      else", file=f);
+    print("        for (auto user : sigusers[bit])", file=f)
+    print("          users.insert(user);", file=f)
+    print("    return unique ? GetSize(users) : i;", file=f)
     print("  }", file=f)
     print("", file=f)