Added sig.{replace,remove,extract} variants for std::{map,set} pattern

diff --git a/kernel/rtlil.cc b/kernel/rtlil.cc
index 96ae0f97a2fa7a607d09cae9c8fb5ccf4e90ed13..297537f001ab00a13a153f64971b2821a50cb674 100644 (file)
--- a/kernel/rtlil.cc
+++ b/kernel/rtlil.cc
@@ -2071,26 +2071,40 @@ void RTLIL::SigSpec::replace(const RTLIL::SigSpec &pattern, const RTLIL::SigSpec
 
 void RTLIL::SigSpec::replace(const RTLIL::SigSpec &pattern, const RTLIL::SigSpec &with, RTLIL::SigSpec *other) const
 {
-       cover("kernel.rtlil.sigspec.replace");
+       log_assert(pattern.width_ == with.width_);
 
-       unpack();
        pattern.unpack();
        with.unpack();
 
-       log_assert(other != NULL);
-       log_assert(width_ == other->width_);
-       other->unpack();
-
-       log_assert(pattern.width_ == with.width_);
+       std::map<RTLIL::SigBit, RTLIL::SigBit> rules;
 
-       std::map<RTLIL::SigBit, RTLIL::SigBit> pattern_map;
        for (int i = 0; i < SIZE(pattern.bits_); i++)
                if (pattern.bits_[i].wire != NULL)
-                       pattern_map[pattern.bits_[i]] = with.bits_[i];
+                       rules[pattern.bits_[i]] = with.bits_[i];
+
+       replace(rules, other);
+}
 
-       for (int i = 0; i < SIZE(bits_); i++)
-               if (pattern_map.count(bits_[i]))
-                       other->bits_[i] = pattern_map.at(bits_[i]);
+void RTLIL::SigSpec::replace(const std::map<RTLIL::SigBit, RTLIL::SigBit> &rules)
+{
+       replace(rules, this);
+}
+
+void RTLIL::SigSpec::replace(const std::map<RTLIL::SigBit, RTLIL::SigBit> &rules, RTLIL::SigSpec *other) const
+{
+       cover("kernel.rtlil.sigspec.replace");
+
+       log_assert(other != NULL);
+       log_assert(width_ == other->width_);
+
+       unpack();
+       other->unpack();
+
+       for (int i = 0; i < SIZE(bits_); i++) {
+               auto it = rules.find(bits_[i]);
+               if (it != rules.end())
+                       other->bits_[i] = it->second;
+       }
 
        other->check();
 }
@@ -2107,6 +2121,23 @@ void RTLIL::SigSpec::remove(const RTLIL::SigSpec &pattern, RTLIL::SigSpec *other
 }
 
 void RTLIL::SigSpec::remove2(const RTLIL::SigSpec &pattern, RTLIL::SigSpec *other)
+{
+       std::set<RTLIL::SigBit> pattern_bits = pattern.to_sigbit_set();
+       remove2(pattern_bits, other);
+}
+
+void RTLIL::SigSpec::remove(const std::set<RTLIL::SigBit> &pattern)
+{
+       remove2(pattern, NULL);
+}
+
+void RTLIL::SigSpec::remove(const std::set<RTLIL::SigBit> &pattern, RTLIL::SigSpec *other) const
+{
+       RTLIL::SigSpec tmp = *this;
+       tmp.remove2(pattern, other);
+}
+
+void RTLIL::SigSpec::remove2(const std::set<RTLIL::SigBit> &pattern, RTLIL::SigSpec *other)
 {
        if (other)
                cover("kernel.rtlil.sigspec.remove_other");
@@ -2120,11 +2151,10 @@ void RTLIL::SigSpec::remove2(const RTLIL::SigSpec &pattern, RTLIL::SigSpec *othe
                other->unpack();
        }
 
-       std::set<RTLIL::SigBit> pattern_bits = pattern.to_sigbit_set();
        std::vector<RTLIL::SigBit> new_bits, new_other_bits;
 
        for (int i = 0; i < SIZE(bits_); i++) {
-               if (bits_[i].wire != NULL && pattern_bits.count(bits_[i]))
+               if (bits_[i].wire != NULL && pattern.count(bits_[i]))
                        continue;
                if (other != NULL)
                        new_other_bits.push_back(other->bits_[i]);
@@ -2142,33 +2172,32 @@ void RTLIL::SigSpec::remove2(const RTLIL::SigSpec &pattern, RTLIL::SigSpec *othe
        check();
 }
 
-RTLIL::SigSpec RTLIL::SigSpec::extract(RTLIL::SigSpec pattern, const RTLIL::SigSpec *other) const
+RTLIL::SigSpec RTLIL::SigSpec::extract(const RTLIL::SigSpec &pattern, const RTLIL::SigSpec *other) const
+{
+       std::set<RTLIL::SigBit> pattern_bits = pattern.to_sigbit_set();
+       return extract(pattern_bits, other);
+}
+
+RTLIL::SigSpec RTLIL::SigSpec::extract(const std::set<RTLIL::SigBit> &pattern, const RTLIL::SigSpec *other) const
 {
        if (other)
                cover("kernel.rtlil.sigspec.extract_other");
        else
                cover("kernel.rtlil.sigspec.extract");
 
-       pack();
-       pattern.pack();
-
-       if (other != NULL)
-               other->pack();
-
        log_assert(other == NULL || width_ == other->width_);
 
-       std::set<RTLIL::SigBit> pat = pattern.to_sigbit_set();
        std::vector<RTLIL::SigBit> bits_match = to_sigbit_vector();
        RTLIL::SigSpec ret;
 
        if (other) {
                std::vector<RTLIL::SigBit> bits_other = other->to_sigbit_vector();
                for (int i = 0; i < width_; i++)
-                       if (bits_match[i].wire && pat.count(bits_match[i]))
+                       if (bits_match[i].wire && pattern.count(bits_match[i]))
                                ret.append_bit(bits_other[i]);
        } else {
                for (int i = 0; i < width_; i++)
-                       if (bits_match[i].wire && pat.count(bits_match[i]))
+                       if (bits_match[i].wire && pattern.count(bits_match[i]))
                                ret.append_bit(bits_match[i]);
        }
 

diff --git a/kernel/rtlil.h b/kernel/rtlil.h
index 0093b8a1b2933d281e7aab62c099765e8adaafb8..8f780c822ed7506ab50453af77e97627653b096d 100644 (file)
--- a/kernel/rtlil.h
+++ b/kernel/rtlil.h
@@ -968,15 +968,25 @@ public:
 
        void replace(const RTLIL::SigSpec &pattern, const RTLIL::SigSpec &with);
        void replace(const RTLIL::SigSpec &pattern, const RTLIL::SigSpec &with, RTLIL::SigSpec *other) const;
+
+       void replace(const std::map<RTLIL::SigBit, RTLIL::SigBit> &rules);
+       void replace(const std::map<RTLIL::SigBit, RTLIL::SigBit> &rules, RTLIL::SigSpec *other) const;
+
        void replace(int offset, const RTLIL::SigSpec &with);
 
        void remove(const RTLIL::SigSpec &pattern);
        void remove(const RTLIL::SigSpec &pattern, RTLIL::SigSpec *other) const;
        void remove2(const RTLIL::SigSpec &pattern, RTLIL::SigSpec *other);
+
+       void remove(const std::set<RTLIL::SigBit> &pattern);
+       void remove(const std::set<RTLIL::SigBit> &pattern, RTLIL::SigSpec *other) const;
+       void remove2(const std::set<RTLIL::SigBit> &pattern, RTLIL::SigSpec *other);
+
        void remove(int offset, int length = 1);
        void remove_const();
 
-       RTLIL::SigSpec extract(RTLIL::SigSpec pattern, const RTLIL::SigSpec *other = NULL) const;
+       RTLIL::SigSpec extract(const RTLIL::SigSpec &pattern, const RTLIL::SigSpec *other = NULL) const;
+       RTLIL::SigSpec extract(const std::set<RTLIL::SigBit> &pattern, const RTLIL::SigSpec *other = NULL) const;
        RTLIL::SigSpec extract(int offset, int length = 1) const;
 
        void append(const RTLIL::SigSpec &signal);