Added CellEdgesDatabase API

1 files changed, 63 insertions, 0 deletions

diff --git a/kernel/celledges.h b/kernel/celledges.h
new file mode 100644
index 000000000..fe863766c
--- /dev/null
+++ b/kernel/celledges.h
@@ -0,0 +1,63 @@
+/*
+ *  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.
+ *
+ */
+
+#ifndef CELLEDGES_H
+#define CELLEDGES_H
+
+#include "kernel/yosys.h"
+#include "kernel/sigtools.h"
+
+YOSYS_NAMESPACE_BEGIN
+
+struct AbstractCellEdgesDatabase
+{
+	virtual ~AbstractCellEdgesDatabase() { }
+	virtual void add_edge(RTLIL::Cell *cell, RTLIL::IdString from_port, int from_bit, RTLIL::IdString to_port, int to_bit) = 0;
+	bool add_cell(RTLIL::Cell *cell);
+};
+
+struct FwdCellEdgesDatabase : AbstractCellEdgesDatabase
+{
+	SigMap &sigmap;
+	dict<SigBit, pool<SigBit>> db;
+	FwdCellEdgesDatabase(SigMap &sigmap) : sigmap(sigmap) { }
+
+	virtual void add_edge(RTLIL::Cell *cell, RTLIL::IdString from_port, int from_bit, RTLIL::IdString to_port, int to_bit) override {
+		SigBit from_sigbit = sigmap(cell->getPort(from_port)[from_bit]);
+		SigBit to_sigbit = sigmap(cell->getPort(to_port)[to_bit]);
+		db[from_sigbit].insert(to_sigbit);
+	}
+};
+
+struct RevCellEdgesDatabase : AbstractCellEdgesDatabase
+{
+	SigMap &sigmap;
+	dict<SigBit, pool<SigBit>> db;
+	RevCellEdgesDatabase(SigMap &sigmap) : sigmap(sigmap) { }
+
+	virtual void add_edge(RTLIL::Cell *cell, RTLIL::IdString from_port, int from_bit, RTLIL::IdString to_port, int to_bit) override {
+		SigBit from_sigbit = sigmap(cell->getPort(from_port)[from_bit]);
+		SigBit to_sigbit = sigmap(cell->getPort(to_port)[to_bit]);
+		db[to_sigbit].insert(from_sigbit);
+	}
+};
+
+YOSYS_NAMESPACE_END
+
+#endif