diff --git a/passes/cmds/wrapcell.cc b/passes/cmds/wrapcell.cc index 812de9ebb..6d8e87b43 100644 --- a/passes/cmds/wrapcell.cc +++ b/passes/cmds/wrapcell.cc @@ -18,12 +18,37 @@ */ #include "kernel/yosys.h" #include "kernel/celltypes.h" +#include "kernel/sigtools.h" #include "backends/rtlil/rtlil_backend.h" USING_YOSYS_NAMESPACE PRIVATE_NAMESPACE_BEGIN -std::optional format(std::string fmt, const dict ¶meters) +bool has_fmt_field(std::string fmt, std::string field_name) +{ + auto it = fmt.begin(); + while (it != fmt.end()) { + if (*it == '{') { + it++; + auto beg = it; + while (it != fmt.end() && *it != '}') it++; + if (it == fmt.end()) + return false; + + if (std::string(beg, it) == field_name) + return true; + } + it++; + } + return false; +} + +struct ContextData { + std::string unused_outputs; +}; + +std::optional format(std::string fmt, const dict ¶meters, + ContextData &context) { std::stringstream result; @@ -38,13 +63,19 @@ std::optional format(std::string fmt, const dict & return {}; } - auto id = RTLIL::escape_id(std::string(beg, it)); - if (!parameters.count(id)) { - log("Parameter %s referenced in format string '%s' not found\n", log_id(id), fmt.c_str()); - return {}; - } + std::string param_name = {beg, it}; - RTLIL_BACKEND::dump_const(result, parameters.at(id)); + if (param_name == "%unused") { + result << context.unused_outputs; + } else { + auto id = RTLIL::escape_id(std::string(beg, it)); + if (!parameters.count(id)) { + log("Parameter %s referenced in format string '%s' not found\n", log_id(id), fmt.c_str()); + return {}; + } + + RTLIL_BACKEND::dump_const(result, parameters.at(id)); + } } else { result << *it; } @@ -54,6 +85,44 @@ std::optional format(std::string fmt, const dict & return {result.str()}; } +struct Chunk { + IdString port; + int base, len; + + Chunk(IdString id, int base, int len) + : port(id), base(base), len(len) {} + + IdString format(Cell *cell) + { + if (len == cell->getPort(port).size()) + return port; + else if (len == 1) + return stringf("%s[%d]", port.c_str(), base); + else + return stringf("%s[%d:%d]", port.c_str(), base + len - 1, base); + } + + SigSpec sample(Cell *cell) + { + return cell->getPort(port).extract(base, len); + } +}; + +std::vector collect_chunks(std::vector> bits) +{ + std::vector ret; + std::sort(bits.begin(), bits.end()); + for (auto it = bits.begin(); it != bits.end();) { + auto sep = it + 1; + for (; sep != bits.end() && + sep->first == it->first && + sep->second == (sep - 1)->second + 1; sep++); + ret.emplace_back(it->first, it->second, sep - it); + it = sep; + } + return ret; +} + struct WrapcellPass : Pass { WrapcellPass() : Pass("wrapcell", "wrap individual cells into new modules") {} @@ -68,6 +137,10 @@ struct WrapcellPass : Pass { log("parameter values as specified in curly brackets. If the named module already\n"); log("exists, it is reused.\n"); log("\n"); + log("If the template contains the special string '{%unused}', the command tracks\n"); + log("unused output ports -- specialized wrapper modules will be generated per every\n"); + log("set of unused ports as appearing on a selected cell.\n"); + log("\n"); log(" -setattr \n"); log(" set the given boolean attribute on each created wrapper module\n"); log("\n"); @@ -114,35 +187,80 @@ struct WrapcellPass : Pass { CellTypes ct; ct.setup(); + bool tracking_unused = has_fmt_field(name_fmt, "%unused"); + for (auto module : d->selected_modules()) { - for (auto cell : module->selected_cells()) { - std::optional unescaped_name = format(name_fmt, cell->parameters); - if (!unescaped_name) - log_error("Formatting error when processing cell '%s' in module '%s'\n", - log_id(cell), log_id(module)); + SigPool unused; - IdString name = RTLIL::escape_id(unescaped_name.value()); - - if (d->module(name)) { - cell->type = name; - cell->parameters.clear(); - continue; + for (auto wire : module->wires()) + if (wire->has_attribute(ID::unused_bits)) { + std::string str = wire->get_string_attribute(ID::unused_bits); + for (auto it = str.begin(); it != str.end();) { + auto sep = it; + for (; sep != str.end() && *sep != ' '; sep++); + unused.add(SigBit(wire, std::stoi(std::string(it, sep)))); + for (it = sep; it != str.end() && *it == ' '; it++); } + } + + for (auto cell : module->selected_cells()) { + Module *subm; + Cell *subcell; if (!ct.cell_known(cell->type)) log_error("Non-internal cell type '%s' on cell '%s' in module '%s' unsupported\n", log_id(cell->type), log_id(cell), log_id(module)); - Module *subm = d->addModule(name); - Cell *subcell = subm->addCell("$1", cell->type); + std::vector> unused_outputs, used_outputs; for (auto conn : cell->connections()) { - Wire *w = subm->addWire(conn.first, conn.second.size()); - if (ct.cell_output(cell->type, w->name)) - w->port_output = true; - else - w->port_input = true; - subcell->setPort(conn.first, w); + if (ct.cell_output(cell->type, conn.first)) + for (int i = 0; i < conn.second.size(); i++) { + if (tracking_unused && unused.check(conn.second[i])) + unused_outputs.emplace_back(conn.first, i); + else + used_outputs.emplace_back(conn.first, i); + } } + + ContextData context; + if (!unused_outputs.empty()) { + context.unused_outputs += "_unused"; + for (auto chunk : collect_chunks(unused_outputs)) + context.unused_outputs += "_" + RTLIL::unescape_id(chunk.format(cell)); + } + + std::optional unescaped_name = format(name_fmt, cell->parameters, context); + if (!unescaped_name) + log_error("Formatting error when processing cell '%s' in module '%s'\n", + log_id(cell), log_id(module)); + + IdString name = RTLIL::escape_id(unescaped_name.value()); + if (d->module(name)) + goto replace_cell; + + subm = d->addModule(name); + subcell = subm->addCell("$1", cell->type); + for (auto conn : cell->connections()) { + if (ct.cell_output(cell->type, conn.first)) { + subcell->setPort(conn.first, SigSpec(RTLIL::Sm, conn.second.size())); + } else { + Wire *w = subm->addWire(conn.first, conn.second.size()); + w->port_input = true; + subcell->setPort(conn.first, w); + } + } + + for (auto chunk : collect_chunks(used_outputs)) { + Wire *w = subm->addWire(chunk.format(cell), chunk.len); + w->port_output = true; + subcell->connections_[chunk.port].replace(chunk.base, w); + } + + for (auto chunk : collect_chunks(unused_outputs)) { + Wire *w = subm->addWire(chunk.format(cell), chunk.len); + subcell->connections_[chunk.port].replace(chunk.base, w); + } + subcell->parameters = cell->parameters; subm->fixup_ports(); @@ -150,7 +268,7 @@ struct WrapcellPass : Pass { if (rule.value_fmt.empty()) { subm->set_bool_attribute(rule.name); } else { - std::optional value = format(rule.value_fmt, cell->parameters); + std::optional value = format(rule.value_fmt, cell->parameters, context); if (!value) log_error("Formatting error when processing cell '%s' in module '%s'\n", @@ -160,8 +278,20 @@ struct WrapcellPass : Pass { } } - cell->type = name; + replace_cell: cell->parameters.clear(); + + dict new_connections; + + for (auto conn : cell->connections()) + if (!ct.cell_output(cell->type, conn.first)) + new_connections[conn.first] = conn.second; + + for (auto chunk : collect_chunks(used_outputs)) + new_connections[chunk.format(cell)] = chunk.sample(cell); + + cell->type = name; + cell->connections_ = new_connections; } } } diff --git a/tests/various/wrapcell.ys b/tests/various/wrapcell.ys index 1c8f62ec7..6e0a457ca 100644 --- a/tests/various/wrapcell.ys +++ b/tests/various/wrapcell.ys @@ -23,3 +23,33 @@ select -assert-count 1 top/t:OR_3_4_4 select -assert-none top/t:OR_2_3_3 top/t:OR_3_4_4 %% top/t:* %D select -assert-mod-count 2 OR_2_3_3 OR_3_4_4 select -assert-mod-count 2 A:bar=w3 A:bar=w4 + +design -reset +read_verilog <