xaiger: cleanup
[yosys.git] / backends / aiger / xaiger.cc
index 6f0ed6e89f01743851f6bb7db0068fadcc70f93c..cd63236e9a9add170182e4760e1ec541692e168d 100644 (file)
@@ -76,6 +76,7 @@ void aiger_encode(std::ostream &f, int x)
 
 struct XAigerWriter
 {
+       Design *design;
        Module *module;
        SigMap sigmap;
 
@@ -84,7 +85,7 @@ struct XAigerWriter
        dict<SigBit, SigBit> not_map, alias_map;
        dict<SigBit, pair<SigBit, SigBit>> and_map;
        vector<SigBit> ci_bits, co_bits;
-       dict<SigBit, Cell*> ff_bits;
+       vector<Cell*> ff_list;
        dict<SigBit, float> arrival_times;
 
        vector<pair<int, int>> aig_gates;
@@ -138,7 +139,7 @@ struct XAigerWriter
                return a;
        }
 
-       XAigerWriter(Module *module, bool dff_mode) : module(module), sigmap(module)
+       XAigerWriter(Module *module, bool dff_mode) : design(module->design), module(module), sigmap(module)
        {
                pool<SigBit> undriven_bits;
                pool<SigBit> unused_bits;
@@ -176,12 +177,12 @@ struct XAigerWriter
                                undriven_bits.insert(bit);
                                unused_bits.insert(bit);
 
-                               bool scc = wire->attributes.count(ID::abc9_scc);
-                               if (wire->port_input || scc)
+                               bool keep = wire->get_bool_attribute(ID::abc9_keep);
+                               if (wire->port_input || keep)
                                        input_bits.insert(bit);
 
-                               bool keep = wire->get_bool_attribute(ID::keep);
-                               if (wire->port_output || keep || scc) {
+                               keep = keep || wire->get_bool_attribute(ID::keep);
+                               if (wire->port_output || keep) {
                                        if (bit != wirebit)
                                                alias_map[wirebit] = bit;
                                        output_bits.insert(wirebit);
@@ -224,15 +225,14 @@ struct XAigerWriter
                                        continue;
                                }
 
-                               if (dff_mode && cell->type.in(ID($_DFF_N_), ID($_DFF_P_)))
+                               if (dff_mode && cell->type.in(ID($_DFF_N_), ID($_DFF_P_)) && !cell->get_bool_attribute(ID::abc9_keep))
                                {
                                        SigBit D = sigmap(cell->getPort(ID::D).as_bit());
                                        SigBit Q = sigmap(cell->getPort(ID::Q).as_bit());
                                        unused_bits.erase(D);
                                        undriven_bits.erase(Q);
                                        alias_map[Q] = D;
-                                       auto r YS_ATTRIBUTE(unused) = ff_bits.insert(std::make_pair(D, cell));
-                                       log_assert(r.second);
+                                       ff_list.emplace_back(cell);
                                        continue;
                                }
 
@@ -240,35 +240,29 @@ struct XAigerWriter
                                        continue;
                        }
 
-                       RTLIL::Module* inst_module = module->design->module(cell->type);
-                       if (inst_module) {
-                               IdString derived_type;
-                               if (cell->parameters.empty())
-                                       derived_type = cell->type;
-                               else
-                                       derived_type = inst_module->derive(module->design, cell->parameters);
-                               inst_module = module->design->module(derived_type);
-                               log_assert(inst_module);
-
+                       RTLIL::Module* inst_module = design->module(cell->type);
+                       if (inst_module && inst_module->get_blackbox_attribute()) {
                                bool abc9_flop = false;
-                               if (!cell->has_keep_attr()) {
-                                       auto it = cell->attributes.find(ID::abc9_box_seq);
-                                       if (it != cell->attributes.end()) {
-                                               int abc9_box_seq = it->second.as_int();
-                                               if (GetSize(box_list) <= abc9_box_seq)
-                                                       box_list.resize(abc9_box_seq+1);
-                                               box_list[abc9_box_seq] = cell;
-                                               // Only flop boxes may have arrival times
-                                               //   (all others are combinatorial)
-                                               abc9_flop = inst_module->get_bool_attribute(ID::abc9_flop);
-                                               if (!abc9_flop)
-                                                       continue;
-                                       }
+
+                               auto it = cell->attributes.find(ID::abc9_box_seq);
+                               if (it != cell->attributes.end()) {
+                                       log_assert(!cell->has_keep_attr());
+                                       log_assert(cell->parameters.empty());
+                                       int abc9_box_seq = it->second.as_int();
+                                       if (GetSize(box_list) <= abc9_box_seq)
+                                               box_list.resize(abc9_box_seq+1);
+                                       box_list[abc9_box_seq] = cell;
+                                       // Only flop boxes may have arrival times
+                                       //   (all others are combinatorial)
+                                       log_assert(cell->parameters.empty());
+                                       abc9_flop = inst_module->get_bool_attribute(ID::abc9_flop);
+                                       if (!abc9_flop)
+                                               continue;
                                }
 
-                               if (!timing.count(derived_type))
+                               if (!timing.count(inst_module->name))
                                        timing.setup_module(inst_module);
-                               auto &t = timing.at(derived_type).arrival;
+                               auto &t = timing.at(inst_module->name).arrival;
                                for (const auto &conn : cell->connections()) {
                                        auto port_wire = inst_module->wire(conn.first);
                                        if (!port_wire->port_output)
@@ -282,7 +276,7 @@ struct XAigerWriter
 #ifndef NDEBUG
                                                if (ys_debug(1)) {
                                                        static std::set<std::tuple<IdString,IdString,int>> seen;
-                                                       if (seen.emplace(derived_type, conn.first, i).second) log("%s.%s[%d] abc9_arrival = %d\n",
+                                                       if (seen.emplace(inst_module->name, conn.first, i).second) log("%s.%s[%d] abc9_arrival = %d\n",
                                                                        log_id(cell->type), log_id(conn.first), i, d);
                                                }
 #endif
@@ -293,10 +287,6 @@ struct XAigerWriter
                                if (abc9_flop)
                                        continue;
                        }
-                       else {
-                               if (cell->type == ID($__ABC9_DELAY))
-                                       log_error("Cell type '%s' not recognised. Check that '+/abc9_model.v' has been read.\n", cell->type.c_str());
-                       }
 
                        bool cell_known = inst_module || cell->known();
                        for (const auto &c : cell->connections()) {
@@ -330,9 +320,9 @@ struct XAigerWriter
                for (auto cell : box_list) {
                        log_assert(cell);
 
-                       RTLIL::Module* box_module = module->design->module(cell->type);
+                       RTLIL::Module* box_module = design->module(cell->type);
                        log_assert(box_module);
-                       log_assert(box_module->attributes.count(ID::abc9_box_id));
+                       log_assert(box_module->has_attribute(ID::abc9_box_id));
 
                        auto r = box_ports.insert(cell->type);
                        if (r.second) {
@@ -429,8 +419,7 @@ struct XAigerWriter
                        aig_map[bit] = 2*aig_m;
                }
 
-               for (const auto &i : ff_bits) {
-                       const Cell *cell = i.second;
+               for (auto cell : ff_list) {
                        const SigBit &q = sigmap(cell->getPort(ID::Q));
                        aig_m++, aig_i++;
                        log_assert(!aig_map.count(q));
@@ -477,8 +466,8 @@ struct XAigerWriter
                        aig_outputs.push_back(aig);
                }
 
-               for (auto &i : ff_bits) {
-                       const SigBit &d = i.first;
+               for (auto cell : ff_list) {
+                       const SigBit &d = sigmap(cell->getPort(ID::D));
                        aig_o++;
                        aig_outputs.push_back(aig_map.at(d));
                }
@@ -550,16 +539,16 @@ struct XAigerWriter
                std::stringstream h_buffer;
                auto write_h_buffer = std::bind(write_buffer, std::ref(h_buffer), std::placeholders::_1);
                write_h_buffer(1);
-               log_debug("ciNum = %d\n", GetSize(input_bits) + GetSize(ff_bits) + GetSize(ci_bits));
-               write_h_buffer(input_bits.size() + ff_bits.size() + ci_bits.size());
-               log_debug("coNum = %d\n", GetSize(output_bits) + GetSize(ff_bits) + GetSize(co_bits));
-               write_h_buffer(output_bits.size() + GetSize(ff_bits) + GetSize(co_bits));
-               log_debug("piNum = %d\n", GetSize(input_bits) + GetSize(ff_bits));
-               write_h_buffer(input_bits.size() + ff_bits.size());
-               log_debug("poNum = %d\n", GetSize(output_bits) + GetSize(ff_bits));
-               write_h_buffer(output_bits.size() + ff_bits.size());
+               log_debug("ciNum = %d\n", GetSize(input_bits) + GetSize(ff_list) + GetSize(ci_bits));
+               write_h_buffer(GetSize(input_bits) + GetSize(ff_list) + GetSize(ci_bits));
+               log_debug("coNum = %d\n", GetSize(output_bits) + GetSize(ff_list) + GetSize(co_bits));
+               write_h_buffer(GetSize(output_bits) + GetSize(ff_list) + GetSize(co_bits));
+               log_debug("piNum = %d\n", GetSize(input_bits) + GetSize(ff_list));
+               write_h_buffer(GetSize(input_bits) + GetSize(ff_list));
+               log_debug("poNum = %d\n", GetSize(output_bits) + GetSize(ff_list));
+               write_h_buffer(GetSize(output_bits) + GetSize(ff_list));
                log_debug("boxNum = %d\n", GetSize(box_list));
-               write_h_buffer(box_list.size());
+               write_h_buffer(GetSize(box_list));
 
                auto write_buffer_float = [](std::stringstream &buffer, float f32) {
                        buffer.write(reinterpret_cast<const char*>(&f32), sizeof(f32));
@@ -573,27 +562,20 @@ struct XAigerWriter
                //for (auto bit : output_bits)
                //      write_o_buffer(0);
 
-               if (!box_list.empty() || !ff_bits.empty()) {
+               if (!box_list.empty() || !ff_list.empty()) {
                        dict<IdString, std::tuple<int,int,int>> cell_cache;
 
                        int box_count = 0;
                        for (auto cell : box_list) {
                                log_assert(cell);
+                               log_assert(cell->parameters.empty());
 
-                               RTLIL::Module* box_module = module->design->module(cell->type);
-                               log_assert(box_module);
-
-                               IdString derived_type;
-                               if (cell->parameters.empty())
-                                       derived_type = cell->type;
-                               else
-                                       derived_type = box_module->derive(module->design, cell->parameters);
-                               box_module = box_module->design->module(derived_type);
-                               log_assert(box_module);
-
-                               auto r = cell_cache.insert(derived_type);
+                               auto r = cell_cache.insert(cell->type);
                                auto &v = r.first->second;
                                if (r.second) {
+                                       RTLIL::Module* box_module = design->module(cell->type);
+                                       log_assert(box_module);
+
                                        int box_inputs = 0, box_outputs = 0;
                                        for (auto port_name : box_module->ports) {
                                                RTLIL::Wire *w = box_module->wire(port_name);
@@ -617,37 +599,26 @@ struct XAigerWriter
 
                        std::stringstream r_buffer;
                        auto write_r_buffer = std::bind(write_buffer, std::ref(r_buffer), std::placeholders::_1);
-                       log_debug("flopNum = %d\n", GetSize(ff_bits));
-                       write_r_buffer(ff_bits.size());
+                       log_debug("flopNum = %d\n", GetSize(ff_list));
+                       write_r_buffer(ff_list.size());
 
                        std::stringstream s_buffer;
                        auto write_s_buffer = std::bind(write_buffer, std::ref(s_buffer), std::placeholders::_1);
-                       write_s_buffer(ff_bits.size());
-
-                       dict<SigBit, int> clk_to_mergeability;
-                       for (const auto &i : ff_bits) {
-                               const Cell *cell = i.second;
-                               log_assert(cell->type.in(ID($_DFF_N_), ID($_DFF_P_)));
-
-                               SigBit clock = sigmap(cell->getPort(ID::C));
-                               clk_to_mergeability.insert(std::make_pair(clock, clk_to_mergeability.size()*2+1));
-                       }
+                       write_s_buffer(ff_list.size());
 
-                       for (const auto &i : ff_bits) {
-                               const SigBit &d = i.first;
-                               const Cell *cell = i.second;
+                       dict<SigSpec, int> clk_to_mergeability;
+                       for (const auto cell : ff_list) {
+                               const SigBit &d = sigmap(cell->getPort(ID::D));
+                               const SigBit &q = sigmap(cell->getPort(ID::Q));
 
-                               SigBit clock = sigmap(cell->getPort(ID::C));
-                               int mergeability = clk_to_mergeability.at(clock);
+                               SigSpec clk_and_pol{sigmap(cell->getPort(ID::C)), cell->type[6] == 'P' ? State::S1 : State::S0};
+                               auto r = clk_to_mergeability.insert(std::make_pair(clk_and_pol, clk_to_mergeability.size()+1));
+                               int mergeability = r.first->second;
                                log_assert(mergeability > 0);
-                               if (cell->type == ID($_DFF_N_))
-                                       write_r_buffer(mergeability);
-                               else if (cell->type == ID($_DFF_P_))
-                                       write_r_buffer(mergeability+1);
-                               else log_abort();
-
-                               SigBit Q = sigmap(cell->getPort(ID::Q));
-                               State init = init_map.at(Q, State::Sx);
+                               write_r_buffer(mergeability);
+
+                               State init = init_map.at(q, State::Sx);
+                               log_debug("Cell '%s' (type %s) has (* init *) value '%s'.\n", log_id(cell), log_id(cell->type), log_signal(init));
                                if (init == State::S1)
                                        write_s_buffer(1);
                                else if (init == State::S0)
@@ -713,10 +684,10 @@ struct XAigerWriter
 
                f << stringf("Generated by %s\n", yosys_version_str);
 
-               module->design->scratchpad_set_int("write_xaiger.num_ands", and_map.size());
-               module->design->scratchpad_set_int("write_xaiger.num_wires", aig_map.size());
-               module->design->scratchpad_set_int("write_xaiger.num_inputs", input_bits.size());
-               module->design->scratchpad_set_int("write_xaiger.num_outputs", output_bits.size());
+               design->scratchpad_set_int("write_xaiger.num_ands", and_map.size());
+               design->scratchpad_set_int("write_xaiger.num_wires", aig_map.size());
+               design->scratchpad_set_int("write_xaiger.num_inputs", input_bits.size());
+               design->scratchpad_set_int("write_xaiger.num_outputs", output_bits.size());
        }
 
        void write_map(std::ostream &f)
@@ -726,8 +697,6 @@ struct XAigerWriter
 
                for (auto wire : module->wires())
                {
-                       SigSpec sig = sigmap(wire);
-
                        for (int i = 0; i < GetSize(wire); i++)
                        {
                                RTLIL::SigBit b(wire, i);
@@ -740,7 +709,6 @@ struct XAigerWriter
                                if (output_bits.count(b)) {
                                        int o = ordered_outputs.at(b);
                                        output_lines[o] += stringf("output %d %d %s\n", o - GetSize(co_bits), wire->start_offset+i, log_id(wire));
-                                       continue;
                                }
                        }
                }