From 54b0f2e659ac0c34c69b0c251c72b2a90fe8e6b6 Mon Sep 17 00:00:00 2001
From: Clifford Wolf <clifford@clifford.at>
Date: Mon, 21 Jul 2014 12:02:55 +0200
Subject: [PATCH] Added module->remove(), module->addWire(), module->addCell(),
 cell->check()

---
 kernel/rtlil.cc | 47 +++++++++++++++++++++++++++++++++++++++--------
 kernel/rtlil.h  |  5 +++++
 2 files changed, 44 insertions(+), 8 deletions(-)

diff --git a/kernel/rtlil.cc b/kernel/rtlil.cc
index 748deae3e..b16b62ca1 100644
--- a/kernel/rtlil.cc
+++ b/kernel/rtlil.cc
@@ -316,9 +316,9 @@ namespace {
 			fputc(0, f);
 			fclose(f);
 
-			log_error("Found error in internal cell %s.%s (%s) at %s:%d:\n%s",
-					module->name.c_str(), cell->name.c_str(), cell->type.c_str(),
-					__FILE__, linenr, ptr);
+			log_error("Found error in internal cell %s%s%s (%s) at %s:%d:\n%s",
+					module ? module->name.c_str() : "", module ? "." : "",
+					cell->name.c_str(), cell->type.c_str(), __FILE__, linenr, ptr);
 		}
 
 		int param(const char *name)
@@ -395,6 +395,10 @@ namespace {
 
 		void check()
 		{
+			if (cell->type[0] != '$' || cell->type.substr(0, 3) == "$__" || cell->type.substr(0, 8) == "$paramod" ||
+					cell->type.substr(0, 9) == "$verific$" || cell->type.substr(0, 7) == "$array:")
+				return;
+
 			if (cell->type == "$not" || cell->type == "$pos" || cell->type == "$bu0" || cell->type == "$neg") {
 				param_bool("\\A_SIGNED");
 				port("\\A", param("\\A_WIDTH"));
@@ -740,11 +744,8 @@ void RTLIL::Module::check()
 		for (auto &it2 : it.second->parameters) {
 			assert(it2.first.size() > 0 && (it2.first[0] == '\\' || it2.first[0] == '$'));
 		}
-		if (it.second->type[0] == '$' && it.second->type.substr(0, 3) != "$__" && it.second->type.substr(0, 8) != "$paramod" &&
-				it.second->type.substr(0, 9) != "$verific$" && it.second->type.substr(0, 7) != "$array:") {
-			InternalCellChecker checker(this, it.second);
-			checker.check();
-		}
+		InternalCellChecker checker(this, it.second);
+		checker.check();
 	}
 
 	for (auto &it : processes) {
@@ -841,6 +842,13 @@ void RTLIL::Module::add(RTLIL::Cell *cell)
 	cells[cell->name] = cell;
 }
 
+void RTLIL::Module::remove(RTLIL::Cell *cell)
+{
+	assert(cells.count(cell->name) != 0);
+	cells.erase(cell->name);
+	delete cell;
+}
+
 static bool fixup_ports_compare(const RTLIL::Wire *a, const RTLIL::Wire *b)
 {
 	if (a->port_id && !b->port_id)
@@ -868,6 +876,23 @@ void RTLIL::Module::fixup_ports()
 		all_ports[i]->port_id = i+1;
 }
 
+RTLIL::Wire *RTLIL::Module::addWire(RTLIL::IdString name, int width)
+{
+	RTLIL::Wire *wire = new RTLIL::Wire;
+	wire->name = name;
+	wire->width = width;
+	add(wire);
+	return wire;
+}
+
+RTLIL::Cell *RTLIL::Module::addCell(RTLIL::IdString name, RTLIL::IdString type)
+{
+	RTLIL::Cell *cell = new RTLIL::Cell;
+	cell->name = name;
+	cell->type = type;
+	add(cell);
+	return cell;
+}
 
 #define DEF_METHOD(_func, _y_size, _type) \
 	RTLIL::Cell* RTLIL::Module::add ## _func(RTLIL::IdString name, RTLIL::SigSpec sig_a, RTLIL::SigSpec sig_y, bool is_signed) { \
@@ -1285,6 +1310,12 @@ void RTLIL::Cell::optimize()
 		it.second.optimize();
 }
 
+void RTLIL::Cell::check()
+{
+	InternalCellChecker checker(NULL, this);
+	checker.check();
+}
+
 RTLIL::SigChunk::SigChunk()
 {
 	wire = NULL;
diff --git a/kernel/rtlil.h b/kernel/rtlil.h
index 64136de04..19666481d 100644
--- a/kernel/rtlil.h
+++ b/kernel/rtlil.h
@@ -290,12 +290,16 @@ struct RTLIL::Module {
 	RTLIL::Wire *new_wire(int width, RTLIL::IdString name);
 	void add(RTLIL::Wire *wire);
 	void add(RTLIL::Cell *cell);
+	void remove(RTLIL::Cell *cell);
 	void fixup_ports();
 
 	template<typename T> void rewrite_sigspecs(T functor);
 	void cloneInto(RTLIL::Module *new_mod) const;
 	virtual RTLIL::Module *clone() const;
 
+	RTLIL::Wire *addWire(RTLIL::IdString name, int width = 1);
+	RTLIL::Cell *addCell(RTLIL::IdString name, RTLIL::IdString type);
+
 	// The add* methods create a cell and return the created cell. All signals must exist in advance.
 
 	RTLIL::Cell* addNot (RTLIL::IdString name, RTLIL::SigSpec sig_a, RTLIL::SigSpec sig_y, bool is_signed = false);
@@ -449,6 +453,7 @@ struct RTLIL::Cell {
 	std::map<RTLIL::IdString, RTLIL::Const> parameters;
 	RTLIL_ATTRIBUTE_MEMBERS
 	void optimize();
+	void check();
 
 	template<typename T> void rewrite_sigspecs(T functor);
 };
-- 
2.30.2