Added simplemap pass

2 files changed, 518 insertions, 0 deletions

diff --git a/passes/techmap/Makefile.inc b/passes/techmap/Makefile.inc
index abe2ac5e..0a56932b 100644
--- a/passes/techmap/Makefile.inc
+++ b/passes/techmap/Makefile.inc
@@ -1,5 +1,6 @@
 
 OBJS += passes/techmap/techmap.o
+OBJS += passes/techmap/simplemap.o
 OBJS += passes/techmap/dfflibmap.o
 OBJS += passes/techmap/iopadmap.o
 OBJS += passes/techmap/libparse.o
diff --git a/passes/techmap/simplemap.cc b/passes/techmap/simplemap.cc
new file mode 100644
index 00000000..5f49fdce
--- /dev/null
+++ b/passes/techmap/simplemap.cc
@@ -0,0 +1,517 @@
+/*
+ *  yosys -- Yosys Open SYnthesis Suite
+ *
+ *  Copyright (C) 2012  Clifford Wolf <clifford@clifford.at>
+ *  
+ *  Permission to use, copy, modify, and/or distribute this software for any
+ *  purpose with or without fee is hereby granted, provided that the above
+ *  copyright notice and this permission notice appear in all copies.
+ *  
+ *  THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ *  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ *  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ *  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ *  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ *  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ *  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ *
+ */
+
+#include "kernel/register.h"
+#include "kernel/sigtools.h"
+#include "kernel/log.h"
+#include <stdlib.h>
+#include <assert.h>
+#include <stdio.h>
+#include <string.h>
+
+static void simplemap_not(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	int width = cell->parameters.at("\\Y_WIDTH").as_int();
+
+	RTLIL::SigSpec sig_a = cell->connections.at("\\A");
+	sig_a.extend(width, cell->parameters.at("\\A_SIGNED").as_bool());
+	sig_a.expand();
+
+	RTLIL::SigSpec sig_y = cell->connections.at("\\Y");
+	sig_y.expand();
+
+	for (int i = 0; i < width; i++) {
+		RTLIL::Cell *gate = new RTLIL::Cell;
+		gate->name = NEW_ID;
+		gate->type = "$_INV_";
+		gate->connections["\\A"] = sig_a.chunks.at(i);
+		gate->connections["\\Y"] = sig_y.chunks.at(i);
+		module->add(gate);
+	}
+}
+
+static void simplemap_pos(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	int width = cell->parameters.at("\\Y_WIDTH").as_int();
+
+	RTLIL::SigSpec sig_a = cell->connections.at("\\A");
+	sig_a.extend(width, cell->parameters.at("\\A_SIGNED").as_bool());
+
+	RTLIL::SigSpec sig_y = cell->connections.at("\\Y");
+
+	module->connections.push_back(RTLIL::SigSig(sig_y, sig_a));
+}
+
+static void simplemap_bitop(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	int width = cell->parameters.at("\\Y_WIDTH").as_int();
+
+	RTLIL::SigSpec sig_a = cell->connections.at("\\A");
+	sig_a.extend(width, cell->parameters.at("\\A_SIGNED").as_bool());
+	sig_a.expand();
+
+	RTLIL::SigSpec sig_b = cell->connections.at("\\B");
+	sig_b.extend(width, cell->parameters.at("\\B_SIGNED").as_bool());
+	sig_b.expand();
+
+	RTLIL::SigSpec sig_y = cell->connections.at("\\Y");
+	sig_y.expand();
+
+	if (cell->type == "$xnor")
+	{
+		RTLIL::SigSpec sig_t = module->new_wire(width, NEW_ID);
+		sig_t.expand();
+
+		for (int i = 0; i < width; i++) {
+			RTLIL::Cell *gate = new RTLIL::Cell;
+			gate->name = NEW_ID;
+			gate->type = "$_INV_";
+			gate->connections["\\A"] = sig_t.chunks.at(i);
+			gate->connections["\\Y"] = sig_y.chunks.at(i);
+			module->add(gate);
+		}
+
+		sig_y = sig_t;
+	}
+
+	std::string gate_type;
+	if (cell->type == "$and")  gate_type = "$_AND_";
+	if (cell->type == "$or")   gate_type = "$_OR_";
+	if (cell->type == "$xor")  gate_type = "$_XOR_";
+	if (cell->type == "$xnor") gate_type = "$_XOR_";
+	log_assert(!gate_type.empty());
+
+	for (int i = 0; i < width; i++) {
+		RTLIL::Cell *gate = new RTLIL::Cell;
+		gate->name = NEW_ID;
+		gate->type = gate_type;
+		gate->connections["\\A"] = sig_a.chunks.at(i);
+		gate->connections["\\B"] = sig_b.chunks.at(i);
+		gate->connections["\\Y"] = sig_y.chunks.at(i);
+		module->add(gate);
+	}
+}
+
+static void simplemap_reduce(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	RTLIL::SigSpec sig_a = cell->connections.at("\\A");
+	sig_a.expand();
+
+	RTLIL::SigSpec sig_y = cell->connections.at("\\Y");
+
+	if (sig_y.width == 0)
+		return;
+	
+	if (sig_a.width == 0) {
+		if (cell->type == "$reduce_and")  module->connections.push_back(RTLIL::SigSig(sig_y, RTLIL::SigSpec(1, sig_y.width)));
+		if (cell->type == "$reduce_or")   module->connections.push_back(RTLIL::SigSig(sig_y, RTLIL::SigSpec(0, sig_y.width)));
+		if (cell->type == "$reduce_xor")  module->connections.push_back(RTLIL::SigSig(sig_y, RTLIL::SigSpec(0, sig_y.width)));
+		if (cell->type == "$reduce_xnor") module->connections.push_back(RTLIL::SigSig(sig_y, RTLIL::SigSpec(1, sig_y.width)));
+		if (cell->type == "$reduce_bool") module->connections.push_back(RTLIL::SigSig(sig_y, RTLIL::SigSpec(0, sig_y.width)));
+		return;
+	}
+
+	if (sig_y.width > 1) {
+		module->connections.push_back(RTLIL::SigSig(sig_y.extract(1, sig_y.width-1), RTLIL::SigSpec(0, sig_y.width-1)));
+		sig_y = sig_y.extract(0, 1);
+	}
+
+	std::string gate_type;
+	if (cell->type == "$reduce_and")  gate_type = "$_AND_";
+	if (cell->type == "$reduce_or")   gate_type = "$_OR_";
+	if (cell->type == "$reduce_xor")  gate_type = "$_XOR_";
+	if (cell->type == "$reduce_xnor") gate_type = "$_XOR_";
+	if (cell->type == "$reduce_bool") gate_type = "$_OR_";
+	log_assert(!gate_type.empty());
+
+	RTLIL::SigSpec *last_output = NULL;
+
+	while (sig_a.width > 1)
+	{
+		RTLIL::SigSpec sig_t = module->new_wire(sig_a.width / 2, NEW_ID);
+		sig_t.expand();
+
+		for (int i = 0; i < sig_a.width; i += 2)
+		{
+			if (i+1 == sig_a.width) {
+				sig_t.append(sig_a.chunks.at(i));
+				continue;
+			}
+
+			RTLIL::Cell *gate = new RTLIL::Cell;
+			gate->name = NEW_ID;
+			gate->type = gate_type;
+			gate->connections["\\A"] = sig_a.chunks.at(i);
+			gate->connections["\\B"] = sig_a.chunks.at(i+1);
+			gate->connections["\\Y"] = sig_t.chunks.at(i/2);
+			last_output = &gate->connections["\\Y"];
+			module->add(gate);
+		}
+
+		sig_a = sig_t;
+	}
+
+	if (cell->type == "$reduce_xnor") {
+		RTLIL::SigSpec sig_t = module->new_wire(1, NEW_ID);
+		RTLIL::Cell *gate = new RTLIL::Cell;
+		gate->name = NEW_ID;
+		gate->type = "$_INV_";
+		gate->connections["\\A"] = sig_a;
+		gate->connections["\\Y"] = sig_t;
+		last_output = &gate->connections["\\Y"];
+		module->add(gate);
+		sig_a = sig_t;
+	}
+
+	if (last_output == NULL) {
+		module->connections.push_back(RTLIL::SigSig(sig_y, sig_a));
+	} else {
+		*last_output = sig_y;
+	}
+}
+
+static void logic_reduce(RTLIL::Module *module, RTLIL::SigSpec &sig)
+{
+	sig.expand();
+
+	while (sig.width > 1)
+	{
+		RTLIL::SigSpec sig_t = module->new_wire(sig.width / 2, NEW_ID);
+		sig_t.expand();
+
+		for (int i = 0; i < sig.width; i += 2)
+		{
+			if (i+1 == sig.width) {
+				sig_t.append(sig.chunks.at(i));
+				continue;
+			}
+
+			RTLIL::Cell *gate = new RTLIL::Cell;
+			gate->name = NEW_ID;
+			gate->type = "$_OR_";
+			gate->connections["\\A"] = sig.chunks.at(i);
+			gate->connections["\\B"] = sig.chunks.at(i+1);
+			gate->connections["\\Y"] = sig_t.chunks.at(i/2);
+			module->add(gate);
+		}
+
+		sig = sig_t;
+	}
+
+	if (sig.width == 0)
+		sig = RTLIL::SigSpec(0, 1);
+}
+
+static void simplemap_lognot(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	RTLIL::SigSpec sig_a = cell->connections.at("\\A");
+	logic_reduce(module, sig_a);
+
+	RTLIL::SigSpec sig_y = cell->connections.at("\\Y");
+
+	if (sig_y.width == 0)
+		return;
+	
+	if (sig_y.width > 1) {
+		module->connections.push_back(RTLIL::SigSig(sig_y.extract(1, sig_y.width-1), RTLIL::SigSpec(0, sig_y.width-1)));
+		sig_y = sig_y.extract(0, 1);
+	}
+
+	RTLIL::Cell *gate = new RTLIL::Cell;
+	gate->name = NEW_ID;
+	gate->type = "$_INV_";
+	gate->connections["\\A"] = sig_a;
+	gate->connections["\\Y"] = sig_y;
+	module->add(gate);
+}
+
+static void simplemap_logbin(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	RTLIL::SigSpec sig_a = cell->connections.at("\\A");
+	logic_reduce(module, sig_a);
+
+	RTLIL::SigSpec sig_b = cell->connections.at("\\B");
+	logic_reduce(module, sig_b);
+
+	RTLIL::SigSpec sig_y = cell->connections.at("\\Y");
+
+	if (sig_y.width == 0)
+		return;
+	
+	if (sig_y.width > 1) {
+		module->connections.push_back(RTLIL::SigSig(sig_y.extract(1, sig_y.width-1), RTLIL::SigSpec(0, sig_y.width-1)));
+		sig_y = sig_y.extract(0, 1);
+	}
+
+	std::string gate_type;
+	if (cell->type == "$logic_and") gate_type = "$_AND_";
+	if (cell->type == "$logic_or")  gate_type = "$_OR_";
+	log_assert(!gate_type.empty());
+
+	RTLIL::Cell *gate = new RTLIL::Cell;
+	gate->name = NEW_ID;
+	gate->type = gate_type;
+	gate->connections["\\A"] = sig_a;
+	gate->connections["\\B"] = sig_b;
+	gate->connections["\\Y"] = sig_y;
+	module->add(gate);
+}
+
+static void simplemap_mux(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	int width = cell->parameters.at("\\WIDTH").as_int();
+
+	RTLIL::SigSpec sig_a = cell->connections.at("\\A");
+	sig_a.expand();
+
+	RTLIL::SigSpec sig_b = cell->connections.at("\\B");
+	sig_b.expand();
+
+	RTLIL::SigSpec sig_y = cell->connections.at("\\Y");
+	sig_y.expand();
+
+	for (int i = 0; i < width; i++) {
+		RTLIL::Cell *gate = new RTLIL::Cell;
+		gate->name = NEW_ID;
+		gate->type = "$_MUX_";
+		gate->connections["\\A"] = sig_a.chunks.at(i);
+		gate->connections["\\B"] = sig_b.chunks.at(i);
+		gate->connections["\\S"] = cell->connections.at("\\S");
+		gate->connections["\\Y"] = sig_y.chunks.at(i);
+		module->add(gate);
+	}
+}
+
+static void simplemap_sr(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	int width = cell->parameters.at("\\WIDTH").as_int();
+	char set_pol = cell->parameters.at("\\SET_POLARITY").as_bool() ? 'P' : 'N';
+	char clr_pol = cell->parameters.at("\\CLR_POLARITY").as_bool() ? 'P' : 'N';
+
+	RTLIL::SigSpec sig_s = cell->connections.at("\\SET");
+	sig_s.expand();
+
+	RTLIL::SigSpec sig_r = cell->connections.at("\\CLR");
+	sig_r.expand();
+
+	RTLIL::SigSpec sig_q = cell->connections.at("\\Q");
+	sig_q.expand();
+
+	std::string gate_type = stringf("$_SR_%c%c_", set_pol, clr_pol);
+
+	for (int i = 0; i < width; i++) {
+		RTLIL::Cell *gate = new RTLIL::Cell;
+		gate->name = NEW_ID;
+		gate->type = gate_type;
+		gate->connections["\\S"] = sig_s.chunks.at(i);
+		gate->connections["\\R"] = sig_r.chunks.at(i);
+		gate->connections["\\Q"] = sig_q.chunks.at(i);
+		module->add(gate);
+	}
+}
+
+static void simplemap_dff(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	int width = cell->parameters.at("\\WIDTH").as_int();
+	char clk_pol = cell->parameters.at("\\CLK_POLARITY").as_bool() ? 'P' : 'N';
+
+	RTLIL::SigSpec sig_clk = cell->connections.at("\\CLK");
+
+	RTLIL::SigSpec sig_d = cell->connections.at("\\D");
+	sig_d.expand();
+
+	RTLIL::SigSpec sig_q = cell->connections.at("\\Q");
+	sig_q.expand();
+
+	std::string gate_type = stringf("$_DFF_%c_", clk_pol);
+
+	for (int i = 0; i < width; i++) {
+		RTLIL::Cell *gate = new RTLIL::Cell;
+		gate->name = NEW_ID;
+		gate->type = gate_type;
+		gate->connections["\\C"] = sig_clk;
+		gate->connections["\\D"] = sig_d.chunks.at(i);
+		gate->connections["\\Q"] = sig_q.chunks.at(i);
+		module->add(gate);
+	}
+}
+
+static void simplemap_dffsr(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	int width = cell->parameters.at("\\WIDTH").as_int();
+	char clk_pol = cell->parameters.at("\\CLK_POLARITY").as_bool() ? 'P' : 'N';
+	char set_pol = cell->parameters.at("\\SET_POLARITY").as_bool() ? 'P' : 'N';
+	char clr_pol = cell->parameters.at("\\CLR_POLARITY").as_bool() ? 'P' : 'N';
+
+	RTLIL::SigSpec sig_clk = cell->connections.at("\\CLK");
+
+	RTLIL::SigSpec sig_s = cell->connections.at("\\SET");
+	sig_s.expand();
+
+	RTLIL::SigSpec sig_r = cell->connections.at("\\CLR");
+	sig_r.expand();
+
+	RTLIL::SigSpec sig_d = cell->connections.at("\\D");
+	sig_d.expand();
+
+	RTLIL::SigSpec sig_q = cell->connections.at("\\Q");
+	sig_q.expand();
+
+	std::string gate_type = stringf("$_DFFSR_%c%c%c_", clk_pol, set_pol, clr_pol);
+
+	for (int i = 0; i < width; i++) {
+		RTLIL::Cell *gate = new RTLIL::Cell;
+		gate->name = NEW_ID;
+		gate->type = gate_type;
+		gate->connections["\\C"] = sig_clk;
+		gate->connections["\\S"] = sig_s.chunks.at(i);
+		gate->connections["\\R"] = sig_r.chunks.at(i);
+		gate->connections["\\D"] = sig_d.chunks.at(i);
+		gate->connections["\\Q"] = sig_q.chunks.at(i);
+		module->add(gate);
+	}
+}
+
+static void simplemap_adff(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	int width = cell->parameters.at("\\WIDTH").as_int();
+	char clk_pol = cell->parameters.at("\\CLK_POLARITY").as_bool() ? 'P' : 'N';
+	char rst_pol = cell->parameters.at("\\ARST_POLARITY").as_bool() ? 'P' : 'N';
+
+	std::vector<RTLIL::State> rst_val = cell->parameters.at("\\ARST_VALUE").bits;
+	while (int(rst_val.size()) < width)
+		rst_val.push_back(RTLIL::State::S0);
+
+	RTLIL::SigSpec sig_clk = cell->connections.at("\\CLK");
+	RTLIL::SigSpec sig_rst = cell->connections.at("\\ARST");
+
+	RTLIL::SigSpec sig_d = cell->connections.at("\\D");
+	sig_d.expand();
+
+	RTLIL::SigSpec sig_q = cell->connections.at("\\Q");
+	sig_q.expand();
+
+	std::string gate_type_0 = stringf("$_DFF_%c%c0_", clk_pol, rst_pol);
+	std::string gate_type_1 = stringf("$_DFF_%c%c1_", clk_pol, rst_pol);
+
+	for (int i = 0; i < width; i++) {
+		RTLIL::Cell *gate = new RTLIL::Cell;
+		gate->name = NEW_ID;
+		gate->type = rst_val.at(i) == RTLIL::State::S1 ? gate_type_1 : gate_type_0;
+		gate->connections["\\C"] = sig_clk;
+		gate->connections["\\R"] = sig_rst;
+		gate->connections["\\D"] = sig_d.chunks.at(i);
+		gate->connections["\\Q"] = sig_q.chunks.at(i);
+		module->add(gate);
+	}
+}
+
+static void simplemap_dlatch(RTLIL::Module *module, RTLIL::Cell *cell)
+{
+	int width = cell->parameters.at("\\WIDTH").as_int();
+	char en_pol = cell->parameters.at("\\EN_POLARITY").as_bool() ? 'P' : 'N';
+
+	RTLIL::SigSpec sig_en = cell->connections.at("\\EN");
+
+	RTLIL::SigSpec sig_d = cell->connections.at("\\D");
+	sig_d.expand();
+
+	RTLIL::SigSpec sig_q = cell->connections.at("\\Q");
+	sig_q.expand();
+
+	std::string gate_type = stringf("$_DLATCH_%c_", en_pol);
+
+	for (int i = 0; i < width; i++) {
+		RTLIL::Cell *gate = new RTLIL::Cell;
+		gate->name = NEW_ID;
+		gate->type = gate_type;
+		gate->connections["\\E"] = sig_en;
+		gate->connections["\\D"] = sig_d.chunks.at(i);
+		gate->connections["\\Q"] = sig_q.chunks.at(i);
+		module->add(gate);
+	}
+}
+
+struct SimplemapPass : public Pass {
+	SimplemapPass() : Pass("simplemap", "mapping simple coarse-grain cells") { }
+	virtual void help()
+	{
+		//   |---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|
+		log("\n");
+		log("    simplemap [selection]\n");
+		log("\n");
+		log("This pass maps a small selection of simple coarse-grain cells to yosys gate\n");
+		log("primitives. The following internal cell types are mapped by this pass:\n");
+		log("\n");
+		log("  $not, $pos, $and, $or, $xor, $xnor\n");
+		log("  $reduce_and, $reduce_or, $reduce_xor, $reduce_xnor, $reduce_bool\n");
+		log("  $logic_not, $logic_and, $logic_or, $mux\n");
+		log("  $sr, $dff, $dffsr, $adff, $dlatch\n");
+		log("\n");
+	}
+	virtual void execute(std::vector<std::string> args, RTLIL::Design *design)
+	{
+		log_header("Executing SIMPLEMAP pass (map simple cells to gate primitives).\n");
+		extra_args(args, 1, design);
+
+		std::map<std::string, void(*)(RTLIL::Module*, RTLIL::Cell*)> supported_cells;
+
+		supported_cells["$not"]         = simplemap_not;
+		supported_cells["$pos"]         = simplemap_pos;
+		supported_cells["$and"]         = simplemap_bitop;
+		supported_cells["$or"]          = simplemap_bitop;
+		supported_cells["$xor"]         = simplemap_bitop;
+		supported_cells["$xnor"]        = simplemap_bitop;
+		supported_cells["$reduce_and"]  = simplemap_reduce;
+		supported_cells["$reduce_or"]   = simplemap_reduce;
+		supported_cells["$reduce_xor"]  = simplemap_reduce;
+		supported_cells["$reduce_xnor"] = simplemap_reduce;
+		supported_cells["$reduce_bool"] = simplemap_reduce;
+		supported_cells["$logic_not"]   = simplemap_lognot;
+		supported_cells["$logic_and"]   = simplemap_logbin;
+		supported_cells["$logic_or"]    = simplemap_logbin;
+		supported_cells["$mux"]         = simplemap_mux;
+		supported_cells["$sr"]          = simplemap_sr;
+		supported_cells["$dff"]         = simplemap_dff;
+		supported_cells["$dffsr"]       = simplemap_dffsr;
+		supported_cells["$adff"]        = simplemap_adff;
+		supported_cells["$dlatch"]      = simplemap_dlatch;
+
+		for (auto &mod_it : design->modules) {
+			if (!design->selected(mod_it.second))
+				continue;
+			std::vector<RTLIL::Cell*> delete_cells;
+			for (auto &cell_it : mod_it.second->cells) {
+				auto mapper = supported_cells[cell_it.second->type];
+				if (mapper == NULL)
+					continue;
+				if (!design->selected(mod_it.second, cell_it.second))
+					continue;
+				log("Mapping %s.%s (%s).\n", RTLIL::id2cstr(mod_it.first), RTLIL::id2cstr(cell_it.first), RTLIL::id2cstr(cell_it.second->type));
+				mapper(mod_it.second, cell_it.second);
+				delete_cells.push_back(cell_it.second);
+			}
+			for (auto &it : delete_cells) {
+				mod_it.second->cells.erase(it->name);
+				delete it;
+			}
+		}
+	}
+} SimplemapPass;
+