More use of IdString::in()

diff --git a/kernel/satgen.h b/kernel/satgen.h
index de480f28ee1bdfcee9aed4230e8b164df3df1c1f..aab3017c2533f8c260e44be3c53ae8196a4c233f 100644 (file)
--- a/kernel/satgen.h
+++ b/kernel/satgen.h
@@ -293,7 +293,7 @@ struct SatGen
                        int undef_any_b = ez->expression(ezSAT::OpOr, undef_b);
                        int undef_y_bit = ez->OR(undef_any_a, undef_any_b);
 
-                       if (cell->type == ID($div) || cell->type == ID($mod)) {
+                       if (cell->type.in(ID($div), ID($mod))) {
                                std::vector<int> b = importSigSpec(cell->getPort(ID(B)), timestep);
                                undef_y_bit = ez->OR(undef_y_bit, ez->NOT(ez->expression(ezSAT::OpOr, b)));
                        }
@@ -320,17 +320,17 @@ struct SatGen
 
                        std::vector<int> yy = model_undef ? ez->vec_var(y.size()) : y;
 
-                       if (cell->type == ID($and) || cell->type == ID($_AND_))
+                       if (cell->type.in(ID($and), ID($_AND_)))
                                ez->assume(ez->vec_eq(ez->vec_and(a, b), yy));
                        if (cell->type == ID($_NAND_))
                                ez->assume(ez->vec_eq(ez->vec_not(ez->vec_and(a, b)), yy));
-                       if (cell->type == ID($or) || cell->type == ID($_OR_))
+                       if (cell->type.in(ID($or), ID($_OR_)))
                                ez->assume(ez->vec_eq(ez->vec_or(a, b), yy));
                        if (cell->type == ID($_NOR_))
                                ez->assume(ez->vec_eq(ez->vec_not(ez->vec_or(a, b)), yy));
-                       if (cell->type == ID($xor) || cell->type == ID($_XOR_))
+                       if (cell->type.in(ID($xor), ID($_XOR_)))
                                ez->assume(ez->vec_eq(ez->vec_xor(a, b), yy));
-                       if (cell->type == ID($xnor) || cell->type == ID($_XNOR_))
+                       if (cell->type.in(ID($xnor), ID($_XNOR_)))
                                ez->assume(ez->vec_eq(ez->vec_not(ez->vec_xor(a, b)), yy));
                        if (cell->type == ID($_ANDNOT_))
                                ez->assume(ez->vec_eq(ez->vec_and(a, ez->vec_not(b)), yy));
@@ -456,7 +456,7 @@ struct SatGen
                        return true;
                }
 
-               if (cell->type == ID($_NOT_) || cell->type == ID($not))
+               if (cell->type.in(ID($_NOT_), ID($not)))
                {
                        std::vector<int> a = importDefSigSpec(cell->getPort(ID(A)), timestep);
                        std::vector<int> y = importDefSigSpec(cell->getPort(ID(Y)), timestep);
@@ -475,7 +475,7 @@ struct SatGen
                        return true;
                }
 
-               if (cell->type == ID($_MUX_) || cell->type == ID($mux) || cell->type == ID($_NMUX_))
+               if (cell->type.in(ID($_MUX_), ID($mux), ID($_NMUX_)))
                {
                        std::vector<int> a = importDefSigSpec(cell->getPort(ID(A)), timestep);
                        std::vector<int> b = importDefSigSpec(cell->getPort(ID(B)), timestep);
@@ -555,7 +555,7 @@ struct SatGen
                        return true;
                }
 
-               if (cell->type == ID($pos) || cell->type == ID($neg))
+               if (cell->type.in(ID($pos), ID($neg)))
                {
                        std::vector<int> a = importDefSigSpec(cell->getPort(ID(A)), timestep);
                        std::vector<int> y = importDefSigSpec(cell->getPort(ID(Y)), timestep);
@@ -589,8 +589,7 @@ struct SatGen
                        return true;
                }
 
-               if (cell->type == ID($reduce_and) || cell->type == ID($reduce_or) || cell->type == ID($reduce_xor) ||
-                               cell->type == ID($reduce_xnor) || cell->type == ID($reduce_bool) || cell->type == ID($logic_not))
+               if (cell->type.in(ID($reduce_and), ID($reduce_or), ID($reduce_xor), ID($reduce_xnor), ID($reduce_bool), ID($logic_not)))
                {
                        std::vector<int> a = importDefSigSpec(cell->getPort(ID(A)), timestep);
                        std::vector<int> y = importDefSigSpec(cell->getPort(ID(Y)), timestep);
@@ -599,7 +598,7 @@ struct SatGen
 
                        if (cell->type == ID($reduce_and))
                                ez->SET(ez->expression(ez->OpAnd, a), yy.at(0));
-                       if (cell->type == ID($reduce_or) || cell->type == ID($reduce_bool))
+                       if (cell->type.in(ID($reduce_or), ID($reduce_bool)))
                                ez->SET(ez->expression(ez->OpOr, a), yy.at(0));
                        if (cell->type == ID($reduce_xor))
                                ez->SET(ez->expression(ez->OpXor, a), yy.at(0));
@@ -620,11 +619,11 @@ struct SatGen
                                        int a0 = ez->expression(ezSAT::OpOr, ez->vec_and(ez->vec_not(a), ez->vec_not(undef_a)));
                                        ez->assume(ez->IFF(ez->AND(ez->NOT(a0), aX), undef_y.at(0)));
                                }
-                               else if (cell->type == ID($reduce_or) || cell->type == ID($reduce_bool) || cell->type == ID($logic_not)) {
+                               else if (cell->type.in(ID($reduce_or), ID($reduce_bool), ID($logic_not))) {
                                        int a1 = ez->expression(ezSAT::OpOr, ez->vec_and(a, ez->vec_not(undef_a)));
                                        ez->assume(ez->IFF(ez->AND(ez->NOT(a1), aX), undef_y.at(0)));
                                }
-                               else if (cell->type == ID($reduce_xor) || cell->type == ID($reduce_xnor)) {
+                               else if (cell->type.in(ID($reduce_xor), ID($reduce_xnor))) {
                                        ez->assume(ez->IFF(aX, undef_y.at(0)));
                                } else
                                        log_abort();
@@ -637,7 +636,7 @@ struct SatGen
                        return true;
                }
 
-               if (cell->type == ID($logic_and) || cell->type == ID($logic_or))
+               if (cell->type.in(ID($logic_and), ID($logic_or)))
                {
                        std::vector<int> vec_a = importDefSigSpec(cell->getPort(ID(A)), timestep);
                        std::vector<int> vec_b = importDefSigSpec(cell->getPort(ID(B)), timestep);
@@ -683,7 +682,7 @@ struct SatGen
                        return true;
                }
 
-               if (cell->type == ID($lt) || cell->type == ID($le) || cell->type == ID($eq) || cell->type == ID($ne) || cell->type == ID($eqx) || cell->type == ID($nex) || cell->type == ID($ge) || cell->type == ID($gt))
+               if (cell->type.in(ID($lt), ID($le), ID($eq), ID($ne), ID($eqx), ID($nex), ID($ge), ID($gt)))
                {
                        bool is_signed = cell->parameters[ID(A_SIGNED)].as_bool() && cell->parameters[ID(B_SIGNED)].as_bool();
                        std::vector<int> a = importDefSigSpec(cell->getPort(ID(A)), timestep);
@@ -693,7 +692,7 @@ struct SatGen
 
                        std::vector<int> yy = model_undef ? ez->vec_var(y.size()) : y;
 
-                       if (model_undef && (cell->type == ID($eqx) || cell->type == ID($nex))) {
+                       if (model_undef && cell->type.in(ID($eqx), ID($nex))) {
                                std::vector<int> undef_a = importUndefSigSpec(cell->getPort(ID(A)), timestep);
                                std::vector<int> undef_b = importUndefSigSpec(cell->getPort(ID(B)), timestep);
                                extendSignalWidth(undef_a, undef_b, cell, true);
@@ -705,9 +704,9 @@ struct SatGen
                                ez->SET(is_signed ? ez->vec_lt_signed(a, b) : ez->vec_lt_unsigned(a, b), yy.at(0));
                        if (cell->type == ID($le))
                                ez->SET(is_signed ? ez->vec_le_signed(a, b) : ez->vec_le_unsigned(a, b), yy.at(0));
-                       if (cell->type == ID($eq) || cell->type == ID($eqx))
+                       if (cell->type.in(ID($eq), ID($eqx)))
                                ez->SET(ez->vec_eq(a, b), yy.at(0));
-                       if (cell->type == ID($ne) || cell->type == ID($nex))
+                       if (cell->type.in(ID($ne), ID($nex)))
                                ez->SET(ez->vec_ne(a, b), yy.at(0));
                        if (cell->type == ID($ge))
                                ez->SET(is_signed ? ez->vec_ge_signed(a, b) : ez->vec_ge_unsigned(a, b), yy.at(0));
@@ -716,7 +715,7 @@ struct SatGen
                        for (size_t i = 1; i < y.size(); i++)
                                ez->SET(ez->CONST_FALSE, yy.at(i));
 
-                       if (model_undef && (cell->type == ID($eqx) || cell->type == ID($nex)))
+                       if (model_undef && cell->type.in(ID($eqx), ID($nex)))
                        {
                                std::vector<int> undef_a = importUndefSigSpec(cell->getPort(ID(A)), timestep);
                                std::vector<int> undef_b = importUndefSigSpec(cell->getPort(ID(B)), timestep);
@@ -733,7 +732,7 @@ struct SatGen
 
                                ez->assume(ez->vec_eq(y, yy));
                        }
-                       else if (model_undef && (cell->type == ID($eq) || cell->type == ID($ne)))
+                       else if (model_undef && cell->type.in(ID($eq), ID($ne)))
                        {
                                std::vector<int> undef_a = importUndefSigSpec(cell->getPort(ID(A)), timestep);
                                std::vector<int> undef_b = importUndefSigSpec(cell->getPort(ID(B)), timestep);
@@ -767,7 +766,7 @@ struct SatGen
                        return true;
                }
 
-               if (cell->type == ID($shl) || cell->type == ID($shr) || cell->type == ID($sshl) || cell->type == ID($sshr) || cell->type == ID($shift) || cell->type == ID($shiftx))
+               if (cell->type.in(ID($shl), ID($shr), ID($sshl), ID($sshr), ID($shift), ID($shiftx)))
                {
                        std::vector<int> a = importDefSigSpec(cell->getPort(ID(A)), timestep);
                        std::vector<int> b = importDefSigSpec(cell->getPort(ID(B)), timestep);
@@ -786,7 +785,7 @@ struct SatGen
                        std::vector<int> yy = model_undef ? ez->vec_var(y.size()) : y;
                        std::vector<int> shifted_a;
 
-                       if (cell->type == ID($shl) || cell->type == ID($sshl))
+                       if (cell->type.in( ID($shl), ID($sshl)))
                                shifted_a = ez->vec_shift_left(a, b, false, ez->CONST_FALSE, ez->CONST_FALSE);
 
                        if (cell->type == ID($shr))
@@ -795,7 +794,7 @@ struct SatGen
                        if (cell->type == ID($sshr))
                                shifted_a = ez->vec_shift_right(a, b, false, cell->parameters[ID(A_SIGNED)].as_bool() ? a.back() : ez->CONST_FALSE, ez->CONST_FALSE);
 
-                       if (cell->type == ID($shift) || cell->type == ID($shiftx))
+                       if (cell->type.in(ID($shift), ID($shiftx)))
                                shifted_a = ez->vec_shift_right(a, b, cell->parameters[ID(B_SIGNED)].as_bool(), ez->CONST_FALSE, ez->CONST_FALSE);
 
                        ez->assume(ez->vec_eq(shifted_a, yy));
@@ -816,7 +815,7 @@ struct SatGen
                                while (undef_y.size() > undef_a.size())
                                        undef_a.push_back(extend_bit);
 
-                               if (cell->type == ID($shl) || cell->type == ID($sshl))
+                               if (cell->type.in(ID($shl), ID($sshl)))
                                        undef_a_shifted = ez->vec_shift_left(undef_a, b, false, ez->CONST_FALSE, ez->CONST_FALSE);
 
                                if (cell->type == ID($shr))
@@ -936,7 +935,7 @@ struct SatGen
                        return true;
                }
 
-               if (cell->type == ID($div) || cell->type == ID($mod))
+               if (cell->type.in(ID($div), ID($mod)))
                {
                        std::vector<int> a = importDefSigSpec(cell->getPort(ID(A)), timestep);
                        std::vector<int> b = importDefSigSpec(cell->getPort(ID(B)), timestep);
@@ -1356,7 +1355,7 @@ struct SatGen
                        return true;
                }
 
-               if (cell->type == ID($_BUF_) || cell->type == ID($equiv))
+               if (cell->type.in(ID($_BUF_), ID($equiv)))
                {
                        std::vector<int> a = importDefSigSpec(cell->getPort(ID(A)), timestep);
                        std::vector<int> y = importDefSigSpec(cell->getPort(ID(Y)), timestep);

diff --git a/passes/cmds/delete.cc b/passes/cmds/delete.cc
index f8d91ea484d0780a362aae961d4c287ecb320589..5822c09f8888c5b7b6c0566ed68b3dc4f40191ee 100644 (file)
--- a/passes/cmds/delete.cc
+++ b/passes/cmds/delete.cc
@@ -107,7 +107,7 @@ struct DeletePass : public Pass {
                        for (auto &it : module->cells_) {
                                if (design->selected(module, it.second))
                                        delete_cells.insert(it.second);
-                               if ((it.second->type == "$memrd" || it.second->type == "$memwr") &&
+                               if (it.second->type.in("$memrd", "$memwr") &&
                                                delete_mems.count(it.second->parameters.at("\\MEMID").decode_string()) != 0)
                                        delete_cells.insert(it.second);
                        }

diff --git a/passes/sat/expose.cc b/passes/sat/expose.cc
index 7631d87e69dbc302534a247086dd99b28943974a..29dfc7b19f01e03bd317f2d41162315c8ea15d3e 100644 (file)
--- a/passes/sat/expose.cc
+++ b/passes/sat/expose.cc
@@ -143,7 +143,7 @@ void create_dff_dq_map(std::map<RTLIL::IdString, dff_map_info_t> &map, RTLIL::De
                        continue;
                }
 
-               if (info.cell->type == "$_DFF_N_" || info.cell->type == "$_DFF_P_") {
+               if (info.cell->type.in("$_DFF_N_", "$_DFF_P_")) {
                        info.bit_clk = sigmap(info.cell->getPort("\\C")).as_bit();
                        info.clk_polarity = info.cell->type == "$_DFF_P_";
                        info.bit_d = sigmap(info.cell->getPort("\\D")).as_bit();

diff --git a/passes/techmap/abc.cc b/passes/techmap/abc.cc
index 9f963a8bf510df4f8738ee15cf12f814dc116424..1de844f2b50b43bd193a80438db7fd3d8dba747b 100644 (file)
--- a/passes/techmap/abc.cc
+++ b/passes/techmap/abc.cc
@@ -1043,7 +1043,7 @@ void abc_module(RTLIL::Design *design, RTLIL::Module *current_module, std::strin
                        if (builtin_lib)
                        {
                                cell_stats[RTLIL::unescape_id(c->type)]++;
-                               if (c->type == "\\ZERO" || c->type == "\\ONE") {
+                               if (c->type.in("\\ZERO", "\\ONE")) {
                                        RTLIL::SigSig conn;
                                        conn.first = RTLIL::SigSpec(module->wires_[remap_name(c->getPort("\\Y").as_wire()->name)]);
                                        conn.second = RTLIL::SigSpec(c->type == "\\ZERO" ? 0 : 1, 1);
@@ -1065,8 +1065,7 @@ void abc_module(RTLIL::Design *design, RTLIL::Module *current_module, std::strin
                                        design->select(module, cell);
                                        continue;
                                }
-                               if (c->type == "\\AND" || c->type == "\\OR" || c->type == "\\XOR" || c->type == "\\NAND" || c->type == "\\NOR" ||
-                                               c->type == "\\XNOR" || c->type == "\\ANDNOT" || c->type == "\\ORNOT") {
+                               if (c->type.in("\\AND", "\\OR", "\\XOR", "\\NAND", "\\NOR", "\\XNOR", "\\ANDNOT", "\\ORNOT")) {
                                        RTLIL::Cell *cell = module->addCell(remap_name(c->name), "$_" + c->type.substr(1) + "_");
                                        if (markgroups) cell->attributes["\\abcgroup"] = map_autoidx;
                                        cell->setPort("\\A", RTLIL::SigSpec(module->wires_[remap_name(c->getPort("\\A").as_wire()->name)]));
@@ -1075,7 +1074,7 @@ void abc_module(RTLIL::Design *design, RTLIL::Module *current_module, std::strin
                                        design->select(module, cell);
                                        continue;
                                }
-                               if (c->type == "\\MUX" || c->type == "\\NMUX") {
+                               if (c->type.in("\\MUX", "\\NMUX")) {
                                        RTLIL::Cell *cell = module->addCell(remap_name(c->name), "$_" + c->type.substr(1) + "_");
                                        if (markgroups) cell->attributes["\\abcgroup"] = map_autoidx;
                                        cell->setPort("\\A", RTLIL::SigSpec(module->wires_[remap_name(c->getPort("\\A").as_wire()->name)]));
@@ -1143,7 +1142,7 @@ void abc_module(RTLIL::Design *design, RTLIL::Module *current_module, std::strin
                                        design->select(module, cell);
                                        continue;
                                }
-                               if (c->type == "\\AOI3" || c->type == "\\OAI3") {
+                               if (c->type.in("\\AOI3", "\\OAI3")) {
                                        RTLIL::Cell *cell = module->addCell(remap_name(c->name), "$_" + c->type.substr(1) + "_");
                                        if (markgroups) cell->attributes["\\abcgroup"] = map_autoidx;
                                        cell->setPort("\\A", RTLIL::SigSpec(module->wires_[remap_name(c->getPort("\\A").as_wire()->name)]));
@@ -1153,7 +1152,7 @@ void abc_module(RTLIL::Design *design, RTLIL::Module *current_module, std::strin
                                        design->select(module, cell);
                                        continue;
                                }
-                               if (c->type == "\\AOI4" || c->type == "\\OAI4") {
+                               if (c->type.in("\\AOI4", "\\OAI4")) {
                                        RTLIL::Cell *cell = module->addCell(remap_name(c->name), "$_" + c->type.substr(1) + "_");
                                        if (markgroups) cell->attributes["\\abcgroup"] = map_autoidx;
                                        cell->setPort("\\A", RTLIL::SigSpec(module->wires_[remap_name(c->getPort("\\A").as_wire()->name)]));
@@ -1185,7 +1184,7 @@ void abc_module(RTLIL::Design *design, RTLIL::Module *current_module, std::strin
                        else
                                cell_stats[RTLIL::unescape_id(c->type)]++;
 
-                       if (c->type == "\\_const0_" || c->type == "\\_const1_") {
+                       if (c->type.in("\\_const0_", "\\_const1_")) {
                                RTLIL::SigSig conn;
                                conn.first = RTLIL::SigSpec(module->wires_[remap_name(c->connections().begin()->second.as_wire()->name)]);
                                conn.second = RTLIL::SigSpec(c->type == "\\_const0_" ? 0 : 1, 1);

diff --git a/passes/techmap/shregmap.cc b/passes/techmap/shregmap.cc
index 004ab1eb9a70c1b6e56655233c640b0ab6c9e69e..959db7627e85d06ad2272fcad22b32ff11e154c8 100644 (file)
--- a/passes/techmap/shregmap.cc
+++ b/passes/techmap/shregmap.cc
@@ -130,7 +130,7 @@ struct ShregmapTechXilinx7 : ShregmapTech
                if (cell) {
                        if (cell->type == "$shiftx" && port == "\\A")
                                return;
-                       if (cell->type == "$mux" && (port == "\\A" || port == "\\B"))
+                       if (cell->type == "$mux" && port.in("\\A", "\\B"))
                                return;
                }
                sigbit_to_shiftx_offset.erase(it);

diff --git a/passes/techmap/techmap.cc b/passes/techmap/techmap.cc
index 2708b08aed756d9acad5346a68eb657656ba8a72..ca231eb53b8c10a8216c5cdbbb53641f80731c6a 100644 (file)
--- a/passes/techmap/techmap.cc
+++ b/passes/techmap/techmap.cc
@@ -349,7 +349,7 @@ struct TechmapWorker
                                port_signal_map.apply(it2.second);
                        }
 
-                       if (c->type == "$memrd" || c->type == "$memwr" || c->type == "$meminit") {
+                       if (c->type.in("$memrd", "$memwr", "$meminit")) {
                                IdString memid = c->getParam("\\MEMID").decode_string();
                                log_assert(memory_renames.count(memid) != 0);
                                c->setParam("\\MEMID", Const(memory_renames[memid].str()));
@@ -520,7 +520,7 @@ struct TechmapWorker
                                                                int port_counter = 1;
                                                                for (auto &c : extmapper_cell->connections_) {
                                                                        RTLIL::Wire *w = extmapper_module->addWire(c.first, GetSize(c.second));
-                                                                       if (w->name == "\\Y" || w->name == "\\Q")
+                                                                       if (w->name.in("\\Y", "\\Q"))
                                                                                w->port_output = true;
                                                                        else
                                                                                w->port_input = true;

diff --git a/passes/tests/test_cell.cc b/passes/tests/test_cell.cc
index 31966995570d37d3eae7343c9cdbd23eb8162a17..88116eeece57fae79e189373bb180ef2619684c0 100644 (file)
--- a/passes/tests/test_cell.cc
+++ b/passes/tests/test_cell.cc
@@ -43,7 +43,7 @@ static void create_gold_module(RTLIL::Design *design, RTLIL::IdString cell_type,
        RTLIL::Cell *cell = module->addCell("\\UUT", cell_type);
        RTLIL::Wire *wire;
 
-       if (cell_type == "$mux" || cell_type == "$pmux")
+       if (cell_type.in("$mux", "$pmux"))
        {
                int width = 1 + xorshift32(8);
                int swidth = cell_type == "$mux" ? 1 : 1 + xorshift32(8);
@@ -264,7 +264,7 @@ static void create_gold_module(RTLIL::Design *design, RTLIL::IdString cell_type,
                cell->setPort("\\Y", wire);
        }
 
-       if (muxdiv && (cell_type == "$div" || cell_type == "$mod")) {
+       if (muxdiv && cell_type.in("$div", "$mod")) {
                auto b_not_zero = module->ReduceBool(NEW_ID, cell->getPort("\\B"));
                auto div_out = module->addWire(NEW_ID, GetSize(cell->getPort("\\Y")));
                module->addMux(NEW_ID, RTLIL::SigSpec(0, GetSize(div_out)), div_out, b_not_zero, cell->getPort("\\Y"));