Merge remote-tracking branch 'origin/master' into xaig

[yosys.git] / backends / aiger / xaiger.cc

diff --git a/backends/aiger/xaiger.cc b/backends/aiger/xaiger.cc
index db0395d280dda18b8fcf53abfa6c5f62ec316862..bad9322bb40d188dd7c7dcb506f43cf027eac7aa 100644 (file)
--- a/backends/aiger/xaiger.cc
+++ b/backends/aiger/xaiger.cc
@@ -158,13 +158,15 @@ struct XAigerWriter
                }
 
                for (auto bit : input_bits) {
-                       undriven_bits.erase(bit);
+                       if (!bit.wire->port_output)
+                               undriven_bits.erase(bit);
                        // Erase POs that are also PIs
                        output_bits.erase(bit);
                }
 
                for (auto bit : output_bits)
-                       unused_bits.erase(bit);
+                       if (!bit.wire->port_input)
+                               unused_bits.erase(bit);
 
                for (auto cell : module->cells())
                {
@@ -213,7 +215,10 @@ struct XAigerWriter
                                for (auto b : c.second.bits()) {
                                        Wire *w = b.wire;
                                        if (!w) continue;
-                                       if (cell->input(c.first)) {
+                                       auto is_input = cell->input(c.first);
+                                       auto is_output = cell->output(c.first);
+                                       log_assert(is_input || is_output);
+                                       if (is_input) {
                                                if (!w->port_input) {
                                                        SigBit I = sigmap(b);
                                                        if (I != b)
@@ -222,12 +227,11 @@ struct XAigerWriter
                                                                co_bits.insert(b);
                                                }
                                        }
-                                       else if (cell->output(c.first)) {
+                                       if (is_output) {
                                                SigBit O = sigmap(b);
-                                               if (!input_bits.count(O) && !output_bits.count(O))
+                                               if (!input_bits.count(O))
                                                        ci_bits.insert(O);
                                        }
-                                       else log_abort();
                                }
                                if (!type_map.count(cell->type))
                                        type_map[cell->type] = type_map.size()+1;
@@ -235,6 +239,27 @@ struct XAigerWriter
                        //log_error("Unsupported cell type: %s (%s)\n", log_id(cell->type), log_id(cell));
                }
 
+               for (auto bit : input_bits) {
+                       RTLIL::Wire *wire = bit.wire;
+                       // If encountering an inout port, then create a new wire with $inout.out
+                       // suffix, make it a CO driven by the existing inout, and inherit existing
+                       // inout's drivers
+                       if (wire->port_input && wire->port_output && !undriven_bits.count(bit)) {
+                               RTLIL::Wire *new_wire = module->wire(wire->name.str() + "$inout.out");
+                               if (!new_wire)
+                                       new_wire = module->addWire(wire->name.str() + "$inout.out", GetSize(wire));
+                               SigBit new_bit(new_wire, bit.offset);
+                               module->connect(new_bit, bit);
+                               if (not_map.count(bit))
+                                       not_map[new_bit] = not_map.at(bit);
+                               else if (and_map.count(bit))
+                                       and_map[new_bit] = and_map.at(bit);
+                               else if (alias_map.count(bit))
+                                       alias_map[new_bit] = alias_map.at(bit);
+                               co_bits.insert(new_bit);
+                       }
+               }
+
                // Do some CI/CO post-processing:
                // Erase all POs and COs that are undriven
                for (auto bit : undriven_bits) {