Merge pull request #3290 from mpasternacki/bugfix/freebsd-build
[yosys.git] / kernel / consteval.h
index 7423f950f92577ea8bfa86de8e615062b3858801..642eb42b25d0f7d0692865f9a3e49e40a8f230ac 100644 (file)
@@ -1,12 +1,12 @@
 /*
  *  yosys -- Yosys Open SYnthesis Suite
  *
- *  Copyright (C) 2012  Clifford Wolf <clifford@clifford.at>
- *  
+ *  Copyright (C) 2012  Claire Xenia Wolf <claire@yosyshq.com>
+ *
  *  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
@@ -25,6 +25,8 @@
 #include "kernel/celltypes.h"
 #include "kernel/macc.h"
 
+YOSYS_NAMESPACE_BEGIN
+
 struct ConstEval
 {
        RTLIL::Module *module;
@@ -34,8 +36,9 @@ struct ConstEval
        SigSet<RTLIL::Cell*> sig2driver;
        std::set<RTLIL::Cell*> busy;
        std::vector<SigMap> stack;
+       RTLIL::State defaultval;
 
-       ConstEval(RTLIL::Module *module) : module(module), assign_map(module)
+       ConstEval(RTLIL::Module *module, RTLIL::State defaultval = RTLIL::State::Sm) : module(module), assign_map(module), defaultval(defaultval)
        {
                CellTypes ct;
                ct.setup_internals();
@@ -72,7 +75,7 @@ struct ConstEval
                assign_map.apply(sig);
 #ifndef NDEBUG
                RTLIL::SigSpec current_val = values_map(sig);
-               for (int i = 0; i < SIZE(current_val); i++)
+               for (int i = 0; i < GetSize(current_val); i++)
                        log_assert(current_val[i].wire != NULL || current_val[i] == value.bits[i]);
 #endif
                values_map.add(sig, RTLIL::SigSpec(value));
@@ -86,31 +89,69 @@ struct ConstEval
 
        bool eval(RTLIL::Cell *cell, RTLIL::SigSpec &undef)
        {
+               if (cell->type == ID($lcu))
+               {
+                       RTLIL::SigSpec sig_p = cell->getPort(ID::P);
+                       RTLIL::SigSpec sig_g = cell->getPort(ID::G);
+                       RTLIL::SigSpec sig_ci = cell->getPort(ID::CI);
+                       RTLIL::SigSpec sig_co = values_map(assign_map(cell->getPort(ID::CO)));
+
+                       if (sig_co.is_fully_const())
+                               return true;
+
+                       if (!eval(sig_p, undef, cell))
+                               return false;
+
+                       if (!eval(sig_g, undef, cell))
+                               return false;
+
+                       if (!eval(sig_ci, undef, cell))
+                               return false;
+
+                       if (sig_p.is_fully_def() && sig_g.is_fully_def() && sig_ci.is_fully_def())
+                       {
+                               RTLIL::Const coval(RTLIL::Sx, GetSize(sig_co));
+                               bool carry = sig_ci.as_bool();
+
+                               for (int i = 0; i < GetSize(coval); i++) {
+                                       carry = (sig_g[i] == State::S1) || (sig_p[i] == RTLIL::S1 && carry);
+                                       coval.bits[i] = carry ? State::S1 : State::S0;
+                               }
+
+                               set(sig_co, coval);
+                       }
+                       else
+                               set(sig_co, RTLIL::Const(RTLIL::Sx, GetSize(sig_co)));
+
+                       return true;
+               }
+
                RTLIL::SigSpec sig_a, sig_b, sig_s, sig_y;
 
-               log_assert(cell->hasPort("\\Y"));
-               sig_y = values_map(assign_map(cell->getPort("\\Y")));
+               log_assert(cell->hasPort(ID::Y));
+               sig_y = values_map(assign_map(cell->getPort(ID::Y)));
                if (sig_y.is_fully_const())
                        return true;
 
-               if (cell->hasPort("\\S")) {
-                       sig_s = cell->getPort("\\S");
-                       if (!eval(sig_s, undef, cell))
-                               return false;
+               if (cell->hasPort(ID::S)) {
+                       sig_s = cell->getPort(ID::S);
                }
 
-               if (cell->hasPort("\\A"))
-                       sig_a = cell->getPort("\\A");
+               if (cell->hasPort(ID::A))
+                       sig_a = cell->getPort(ID::A);
 
-               if (cell->hasPort("\\B"))
-                       sig_b = cell->getPort("\\B");
+               if (cell->hasPort(ID::B))
+                       sig_b = cell->getPort(ID::B);
 
-               if (cell->type == "$mux" || cell->type == "$pmux" || cell->type == "$_MUX_")
+               if (cell->type.in(ID($mux), ID($pmux), ID($_MUX_), ID($_NMUX_)))
                {
                        std::vector<RTLIL::SigSpec> y_candidates;
                        int count_maybe_set_s_bits = 0;
                        int count_set_s_bits = 0;
 
+                       if (!eval(sig_s, undef, cell))
+                               return false;
+
                        for (int i = 0; i < sig_s.size(); i++)
                        {
                                RTLIL::State s_bit = sig_s.extract(i, 1).as_const().bits.at(0);
@@ -135,7 +176,10 @@ struct ConstEval
                        for (auto &yc : y_candidates) {
                                if (!eval(yc, undef, cell))
                                        return false;
-                               y_values.push_back(yc.as_const());
+                               if (cell->type == ID($_NMUX_))
+                                       y_values.push_back(RTLIL::const_not(yc.as_const(), Const(), false, false, GetSize(yc)));
+                               else
+                                       y_values.push_back(yc.as_const());
                        }
 
                        if (y_values.size() > 1)
@@ -155,11 +199,41 @@ struct ConstEval
                        else
                                set(sig_y, y_values.front());
                }
-               else if (cell->type == "$fa")
+               else if (cell->type == ID($bmux))
                {
-                       RTLIL::SigSpec sig_c = cell->getPort("\\C");
-                       RTLIL::SigSpec sig_x = cell->getPort("\\X");
-                       int width = SIZE(sig_c);
+                       if (!eval(sig_s, undef, cell))
+                               return false;
+
+                       if (sig_s.is_fully_def()) {
+                               int sel = sig_s.as_int();
+                               int width = GetSize(sig_y);
+                               SigSpec res = sig_a.extract(sel * width, width);
+                               if (!eval(res, undef, cell))
+                                       return false;
+                               set(sig_y, res.as_const());
+                       } else {
+                               if (!eval(sig_a, undef, cell))
+                                       return false;
+                               set(sig_y, const_bmux(sig_a.as_const(), sig_s.as_const()));
+                       }
+               }
+               else if (cell->type == ID($demux))
+               {
+                       if (!eval(sig_a, undef, cell))
+                               return false;
+                       if (sig_a.is_fully_zero()) {
+                               set(sig_y, Const(0, GetSize(sig_y)));
+                       } else {
+                               if (!eval(sig_s, undef, cell))
+                                       return false;
+                               set(sig_y, const_demux(sig_a.as_const(), sig_s.as_const()));
+                       }
+               }
+               else if (cell->type == ID($fa))
+               {
+                       RTLIL::SigSpec sig_c = cell->getPort(ID::C);
+                       RTLIL::SigSpec sig_x = cell->getPort(ID::X);
+                       int width = GetSize(sig_c);
 
                        if (!eval(sig_a, undef, cell))
                                return false;
@@ -177,16 +251,20 @@ struct ConstEval
                        RTLIL::Const t3 = const_and(sig_c.as_const(), t1, false, false, width);
                        RTLIL::Const val_x = const_or(t2, t3, false, false, width);
 
+                       for (int i = 0; i < GetSize(val_y); i++)
+                               if (val_y.bits[i] == RTLIL::Sx)
+                                       val_x.bits[i] = RTLIL::Sx;
+
                        set(sig_y, val_y);
                        set(sig_x, val_x);
                }
-               else if (cell->type == "$alu")
+               else if (cell->type == ID($alu))
                {
-                       bool signed_a = cell->parameters.count("\\A_SIGNED") > 0 && cell->parameters["\\A_SIGNED"].as_bool();
-                       bool signed_b = cell->parameters.count("\\B_SIGNED") > 0 && cell->parameters["\\B_SIGNED"].as_bool();
+                       bool signed_a = cell->parameters.count(ID::A_SIGNED) > 0 && cell->parameters[ID::A_SIGNED].as_bool();
+                       bool signed_b = cell->parameters.count(ID::B_SIGNED) > 0 && cell->parameters[ID::B_SIGNED].as_bool();
 
-                       RTLIL::SigSpec sig_ci = cell->getPort("\\CI");
-                       RTLIL::SigSpec sig_bi = cell->getPort("\\BI");
+                       RTLIL::SigSpec sig_ci = cell->getPort(ID::CI);
+                       RTLIL::SigSpec sig_bi = cell->getPort(ID::BI);
 
                        if (!eval(sig_a, undef, cell))
                                return false;
@@ -200,43 +278,43 @@ struct ConstEval
                        if (!eval(sig_bi, undef, cell))
                                return false;
 
-                       RTLIL::SigSpec sig_x = cell->getPort("\\X");
-                       RTLIL::SigSpec sig_co = cell->getPort("\\CO");
+                       RTLIL::SigSpec sig_x = cell->getPort(ID::X);
+                       RTLIL::SigSpec sig_co = cell->getPort(ID::CO);
 
                        bool any_input_undef = !(sig_a.is_fully_def() && sig_b.is_fully_def() && sig_ci.is_fully_def() && sig_bi.is_fully_def());
-                       sig_a.extend_u0(SIZE(sig_y), signed_a);
-                       sig_b.extend_u0(SIZE(sig_y), signed_b);
+                       sig_a.extend_u0(GetSize(sig_y), signed_a);
+                       sig_b.extend_u0(GetSize(sig_y), signed_b);
 
-                       bool carry = sig_ci[0] == RTLIL::S1;
-                       bool b_inv = sig_bi[0] == RTLIL::S1;
+                       bool carry = sig_ci[0] == State::S1;
+                       bool b_inv = sig_bi[0] == State::S1;
 
-                       for (int i = 0; i < SIZE(sig_y); i++)
+                       for (int i = 0; i < GetSize(sig_y); i++)
                        {
                                RTLIL::SigSpec x_inputs = { sig_a[i], sig_b[i], sig_bi[0] };
 
                                if (!x_inputs.is_fully_def()) {
                                        set(sig_x[i], RTLIL::Sx);
                                } else {
-                                       bool bit_a = sig_a[i] == RTLIL::S1;
-                                       bool bit_b = (sig_b[i] == RTLIL::S1) != b_inv;
+                                       bool bit_a = sig_a[i] == State::S1;
+                                       bool bit_b = (sig_b[i] == State::S1) != b_inv;
                                        bool bit_x = bit_a != bit_b;
-                                       set(sig_x[i], bit_x ? RTLIL::S1 : RTLIL::S0);
+                                       set(sig_x[i], bit_x ? State::S1 : State::S0);
                                }
 
                                if (any_input_undef) {
                                        set(sig_y[i], RTLIL::Sx);
                                        set(sig_co[i], RTLIL::Sx);
                                } else {
-                                       bool bit_a = sig_a[i] == RTLIL::S1;
-                                       bool bit_b = (sig_b[i] == RTLIL::S1) != b_inv;
+                                       bool bit_a = sig_a[i] == State::S1;
+                                       bool bit_b = (sig_b[i] == State::S1) != b_inv;
                                        bool bit_y = (bit_a != bit_b) != carry;
                                        carry = (bit_a && bit_b) || (bit_a && carry) || (bit_b && carry);
-                                       set(sig_y[i], bit_y ? RTLIL::S1 : RTLIL::S0);
-                                       set(sig_co[i], carry ? RTLIL::S1 : RTLIL::S0);
+                                       set(sig_y[i], bit_y ? State::S1 : State::S0);
+                                       set(sig_co[i], carry ? State::S1 : State::S0);
                                }
                        }
                }
-               else if (cell->type == "$macc")
+               else if (cell->type == ID($macc))
                {
                        Macc macc;
                        macc.from_cell(cell);
@@ -251,21 +329,21 @@ struct ConstEval
                                        return false;
                        }
 
-                       RTLIL::Const result(0, SIZE(cell->getPort("\\Y")));
+                       RTLIL::Const result(0, GetSize(cell->getPort(ID::Y)));
                        if (!macc.eval(result))
                                log_abort();
 
-                       set(cell->getPort("\\Y"), result);
+                       set(cell->getPort(ID::Y), result);
                }
                else
                {
                        RTLIL::SigSpec sig_c, sig_d;
 
-                       if (cell->type.in("$_AOI3_", "$_OAI3_", "$_AOI4_", "$_OAI4_")) {
-                               if (cell->hasPort("\\C"))
-                                       sig_c = cell->getPort("\\C");
-                               if (cell->hasPort("\\D"))
-                                       sig_d = cell->getPort("\\D");
+                       if (cell->type.in(ID($_AOI3_), ID($_OAI3_), ID($_AOI4_), ID($_OAI4_))) {
+                               if (cell->hasPort(ID::C))
+                                       sig_c = cell->getPort(ID::C);
+                               if (cell->hasPort(ID::D))
+                                       sig_d = cell->getPort(ID::D);
                        }
 
                        if (sig_a.size() > 0 && !eval(sig_a, undef, cell))
@@ -277,8 +355,13 @@ struct ConstEval
                        if (sig_d.size() > 0 && !eval(sig_d, undef, cell))
                                return false;
 
-                       set(sig_y, CellTypes::eval(cell, sig_a.as_const(), sig_b.as_const(),
-                                       sig_c.as_const(), sig_d.as_const()));
+                       bool eval_err = false;
+                       RTLIL::Const eval_ret = CellTypes::eval(cell, sig_a.as_const(), sig_b.as_const(), sig_c.as_const(), sig_d.as_const(), &eval_err);
+
+                       if (eval_err)
+                               return false;
+
+                       set(sig_y, eval_ret);
                }
 
                return true;
@@ -322,6 +405,12 @@ struct ConstEval
                if (sig.is_fully_const())
                        return true;
 
+               if (defaultval != RTLIL::State::Sm) {
+                       for (auto &bit : sig)
+                               if (bit.wire) bit = defaultval;
+                       return true;
+               }
+
                for (auto &c : sig.chunks())
                        if (c.wire != NULL)
                                undef.append(c);
@@ -335,4 +424,6 @@ struct ConstEval
        }
 };
 
+YOSYS_NAMESPACE_END
+
 #endif