dict/pool changes in opt_clean

1 files changed, 9 insertions, 0 deletions

diff --git a/kernel/sigtools.h b/kernel/sigtools.h
index 5281b7a4..c38736e7 100644
--- a/kernel/sigtools.h
+++ b/kernel/sigtools.h
@@ -195,6 +195,15 @@ struct SigSet
 			}
 	}
 
+	void find(RTLIL::SigSpec sig, pool<T> &result)
+	{
+		for (auto &bit : sig)
+			if (bit.wire != NULL) {
+				auto &data = bits[bit];
+				result.insert(data.begin(), data.end());
+			}
+	}
+
 	std::set<T> find(RTLIL::SigSpec sig)
 	{
 		std::set<T> result;