From a0afa1787e71fca1fb143a96580136f4bf355251 Mon Sep 17 00:00:00 2001
From: Eddie Hung <eddie@fpgeh.com>
Date: Sat, 2 May 2020 09:55:34 -0700
Subject: [PATCH] aiger: fixes for ports that have start_offset != 0

---
 backends/aiger/aiger.cc       | 12 +++---
 backends/aiger/xaiger.cc      |  5 +--
 frontends/aiger/aigerparse.cc | 77 +++++++++++++++++++++--------------
 3 files changed, 55 insertions(+), 39 deletions(-)

diff --git a/backends/aiger/aiger.cc b/backends/aiger/aiger.cc
index cac32a8da..e5a41b5c5 100644
--- a/backends/aiger/aiger.cc
+++ b/backends/aiger/aiger.cc
@@ -629,30 +629,30 @@ struct AigerWriter
 				int a = aig_map.at(sig[i]);
 
 				if (verbose_map)
-					wire_lines[a] += stringf("wire %d %d %s\n", a, i, log_id(wire));
+					wire_lines[a] += stringf("wire %d %d %s\n", a, wire->start_offset+i, log_id(wire));
 
 				if (wire->port_input) {
 					log_assert((a & 1) == 0);
-					input_lines[a] += stringf("input %d %d %s\n", (a >> 1)-1, i, log_id(wire));
+					input_lines[a] += stringf("input %d %d %s\n", (a >> 1)-1, wire->start_offset+i, log_id(wire));
 				}
 
 				if (wire->port_output) {
 					int o = ordered_outputs.at(sig[i]);
-					output_lines[o] += stringf("output %d %d %s\n", o, i, log_id(wire));
+					output_lines[o] += stringf("output %d %d %s\n", o, wire->start_offset+i, log_id(wire));
 				}
 
 				if (init_inputs.count(sig[i])) {
 					int a = init_inputs.at(sig[i]);
 					log_assert((a & 1) == 0);
-					init_lines[a] += stringf("init %d %d %s\n", (a >> 1)-1, i, log_id(wire));
+					init_lines[a] += stringf("init %d %d %s\n", (a >> 1)-1, wire->start_offset+i, log_id(wire));
 				}
 
 				if (ordered_latches.count(sig[i])) {
 					int l = ordered_latches.at(sig[i]);
 					if (zinit_mode && (aig_latchinit.at(l) == 1))
-						latch_lines[l] += stringf("invlatch %d %d %s\n", l, i, log_id(wire));
+						latch_lines[l] += stringf("invlatch %d %d %s\n", l, wire->start_offset+i, log_id(wire));
 					else
-						latch_lines[l] += stringf("latch %d %d %s\n", l, i, log_id(wire));
+						latch_lines[l] += stringf("latch %d %d %s\n", l, wire->start_offset+i, log_id(wire));
 				}
 			}
 		}
diff --git a/backends/aiger/xaiger.cc b/backends/aiger/xaiger.cc
index 3c7c745fe..1fb7210cb 100644
--- a/backends/aiger/xaiger.cc
+++ b/backends/aiger/xaiger.cc
@@ -725,13 +725,12 @@ struct XAigerWriter
 				if (input_bits.count(b)) {
 					int a = aig_map.at(b);
 					log_assert((a & 1) == 0);
-					input_lines[a] += stringf("input %d %d %s\n", (a >> 1)-1, i, log_id(wire));
+					input_lines[a] += stringf("input %d %d %s\n", (a >> 1)-1, wire->start_offset+i, log_id(wire));
 				}
 
 				if (output_bits.count(b)) {
 					int o = ordered_outputs.at(b);
-					int init = 2;
-					output_lines[o] += stringf("output %d %d %s %d\n", o - GetSize(co_bits), i, log_id(wire), init);
+					output_lines[o] += stringf("output %d %d %s\n", o - GetSize(co_bits), wire->start_offset+i, log_id(wire));
 					continue;
 				}
 			}
diff --git a/frontends/aiger/aigerparse.cc b/frontends/aiger/aigerparse.cc
index 92cf92fa8..6fda92d73 100644
--- a/frontends/aiger/aigerparse.cc
+++ b/frontends/aiger/aigerparse.cc
@@ -784,7 +784,7 @@ void AigerReader::post_process()
 		ff->attributes[ID::abc9_mergeability] = mergeability[i];
 	}
 
-	dict<RTLIL::IdString, int> wideports_cache;
+	dict<RTLIL::IdString, std::pair<int,int>> wideports_cache;
 
 	if (!map_filename.empty()) {
 		std::ifstream mf(map_filename);
@@ -799,11 +799,12 @@ void AigerReader::post_process()
 				log_assert(wire->port_input);
 				log_debug("Renaming input %s", log_id(wire));
 
+				RTLIL::Wire *existing = nullptr;
 				if (index == 0) {
 					// Cope with the fact that a CI might be identical
 					// to a PI (necessary due to ABC); in those cases
 					// simply connect the latter to the former
-					RTLIL::Wire* existing = module->wire(escaped_s);
+					existing = module->wire(escaped_s);
 					if (!existing)
 						module->rename(wire, escaped_s);
 					else {
@@ -812,20 +813,29 @@ void AigerReader::post_process()
 					}
 					log_debug(" -> %s\n", log_id(escaped_s));
 				}
-				else if (index > 0) {
-					std::string indexed_name = stringf("%s[%d]", escaped_s.c_str(), index);
-					RTLIL::Wire* existing = module->wire(indexed_name);
-					if (!existing) {
+				else {
+					RTLIL::IdString indexed_name = stringf("%s[%d]", escaped_s.c_str(), index);
+					existing = module->wire(indexed_name);
+					if (!existing)
 						module->rename(wire, indexed_name);
-						if (wideports)
-							wideports_cache[escaped_s] = std::max(wideports_cache[escaped_s], index);
-					}
 					else {
 						module->connect(wire, existing);
 						wire->port_input = false;
 					}
 					log_debug(" -> %s\n", log_id(indexed_name));
 				}
+
+				if (wideports && !existing) {
+					auto r = wideports_cache.insert(escaped_s);
+					if (r.second) {
+						r.first->second.first = index;
+						r.first->second.second = index;
+					}
+					else {
+						r.first->second.first = std::min(r.first->second.first, index);
+						r.first->second.second = std::max(r.first->second.second, index);
+					}
+				}
 			}
 			else if (type == "output") {
 				log_assert(static_cast<unsigned>(variable + co_count) < outputs.size());
@@ -834,14 +844,14 @@ void AigerReader::post_process()
 				log_assert(wire->port_output);
 				log_debug("Renaming output %s", log_id(wire));
 
+				RTLIL::Wire *existing;
 				if (index == 0) {
 					// Cope with the fact that a CO might be identical
 					// to a PO (necessary due to ABC); in those cases
 					// simply connect the latter to the former
-					RTLIL::Wire* existing = module->wire(escaped_s);
-					if (!existing) {
+					existing = module->wire(escaped_s);
+					if (!existing)
 						module->rename(wire, escaped_s);
-					}
 					else {
 						wire->port_output = false;
 						existing->port_output = true;
@@ -850,14 +860,11 @@ void AigerReader::post_process()
 					}
 					log_debug(" -> %s\n", log_id(escaped_s));
 				}
-				else if (index > 0) {
-					std::string indexed_name = stringf("%s[%d]", escaped_s.c_str(), index);
-					RTLIL::Wire* existing = module->wire(indexed_name);
-					if (!existing) {
+				else {
+					RTLIL::IdString indexed_name = stringf("%s[%d]", escaped_s.c_str(), index);
+					existing = module->wire(indexed_name);
+					if (!existing)
 						module->rename(wire, indexed_name);
-						if (wideports)
-							wideports_cache[escaped_s] = std::max(wideports_cache[escaped_s], index);
-					}
 					else {
 						wire->port_output = false;
 						existing->port_output = true;
@@ -865,10 +872,18 @@ void AigerReader::post_process()
 					}
 					log_debug(" -> %s\n", log_id(indexed_name));
 				}
-				int init;
-				mf >> init;
-				if (init < 2)
-					wire->attributes[ID::init] = init;
+
+				if (wideports && !existing) {
+					auto r = wideports_cache.insert(escaped_s);
+					if (r.second) {
+						r.first->second.first = index;
+						r.first->second.second = index;
+					}
+					else {
+						r.first->second.first = std::min(r.first->second.first, index);
+						r.first->second.second = std::max(r.first->second.second, index);
+					}
+				}
 			}
 			else if (type == "box") {
 				RTLIL::Cell* cell = module->cell(stringf("$box%d", variable));
@@ -882,7 +897,8 @@ void AigerReader::post_process()
 
 	for (auto &wp : wideports_cache) {
 		auto name = wp.first;
-		int width = wp.second + 1;
+		int min = wp.second.first;
+		int max = wp.second.second;
 
 		RTLIL::Wire *wire = module->wire(name);
 		if (wire)
@@ -891,7 +907,7 @@ void AigerReader::post_process()
 		// Do not make ports with a mix of input/output into
 		// wide ports
 		bool port_input = false, port_output = false;
-		for (int i = 0; i < width; i++) {
+		for (int i = min; i <= max; i++) {
 			RTLIL::IdString other_name = name.str() + stringf("[%d]", i);
 			RTLIL::Wire *other_wire = module->wire(other_name);
 			if (other_wire) {
@@ -900,20 +916,21 @@ void AigerReader::post_process()
 			}
 		}
 
-		wire = module->addWire(name, width);
+		wire = module->addWire(name, max-min+1);
+		wire->start_offset = min;
 		wire->port_input = port_input;
 		wire->port_output = port_output;
 
-		for (int i = 0; i < width; i++) {
-			RTLIL::IdString other_name = name.str() + stringf("[%d]", i);
+		for (int i = min; i <= max; i++) {
+			RTLIL::IdString other_name = stringf("%s[%d]", name.c_str(), i);
 			RTLIL::Wire *other_wire = module->wire(other_name);
 			if (other_wire) {
 				other_wire->port_input = false;
 				other_wire->port_output = false;
 				if (wire->port_input)
-					module->connect(other_wire, SigSpec(wire, i));
+					module->connect(other_wire, SigSpec(wire, i-min));
 				else
-					module->connect(SigSpec(wire, i), other_wire);
+					module->connect(SigSpec(wire, i-min), other_wire);
 			}
 		}
 	}