Added generic RTLIL::SigSpec::parse_sel() with support for selection variables

2 files changed, 19 insertions, 0 deletions

diff --git a/kernel/rtlil.cc b/kernel/rtlil.cc
index 2ab3320b..95d62503 100644
--- a/kernel/rtlil.cc
+++ b/kernel/rtlil.cc
@@ -1649,6 +1649,24 @@ bool RTLIL::SigSpec::parse(RTLIL::SigSpec &sig, RTLIL::Module *module, std::stri
 	return true;
 }
 
+bool RTLIL::SigSpec::parse_sel(RTLIL::SigSpec &sig, RTLIL::Design *design, RTLIL::Module *module, std::string str)
+{
+	if (str.empty() || str[0] != '@')
+		return parse(sig, module, str);
+
+	str = RTLIL::escape_id(str.substr(1));
+	if (design->selection_vars.count(str) == 0)
+		return false;
+
+	sig = RTLIL::SigSpec();
+	RTLIL::Selection &sel = design->selection_vars.at(str);
+	for (auto &it : module->wires)
+		if (sel.selected_member(module->name, it.first))
+			sig.append(it.second);
+
+	return true;
+}
+
 bool RTLIL::SigSpec::parse_rhs(const RTLIL::SigSpec &lhs, RTLIL::SigSpec &sig, RTLIL::Module *module, std::string str)
 {
 	if (str == "0") {
diff --git a/kernel/rtlil.h b/kernel/rtlil.h
index df0e94dc..caadf198 100644
--- a/kernel/rtlil.h
+++ b/kernel/rtlil.h
@@ -411,6 +411,7 @@ struct RTLIL::SigSpec {
 	std::vector<RTLIL::SigBit> to_sigbit_vector() const;
 	RTLIL::SigBit to_single_sigbit() const;
 	static bool parse(RTLIL::SigSpec &sig, RTLIL::Module *module, std::string str);
+	static bool parse_sel(RTLIL::SigSpec &sig, RTLIL::Design *design, RTLIL::Module *module, std::string str);
 	static bool parse_rhs(const RTLIL::SigSpec &lhs, RTLIL::SigSpec &sig, RTLIL::Module *module, std::string str);
 };