Various improvements in support for generate statements

4 files changed, 91 insertions, 2 deletions

diff --git a/frontends/ast/ast.cc b/frontends/ast/ast.cc
index ccadc2069..10c7fc85b 100644
--- a/frontends/ast/ast.cc
+++ b/frontends/ast/ast.cc
@@ -135,6 +135,7 @@ std::string AST::type2str(AstNodeType type)
 	X(AST_GENVAR)
 	X(AST_GENFOR)
 	X(AST_GENIF)
+	X(AST_GENCASE)
 	X(AST_GENBLOCK)
 	X(AST_POSEDGE)
 	X(AST_NEGEDGE)
@@ -700,6 +701,15 @@ RTLIL::Const AstNode::asParaConst()
 	return val;
 }
 
+bool AstNode::asBool()
+{
+	log_assert(type == AST_CONSTANT);
+	for (auto &bit : bits)
+		if (bit == RTLIL::State::S1)
+			return true;
+	return false;
+}
+
 // create a new AstModule from an AST_MODULE AST node
 static AstModule* process_module(AstNode *ast)
 {
diff --git a/frontends/ast/ast.h b/frontends/ast/ast.h
index f90fe9b7b..ab1b9bec5 100644
--- a/frontends/ast/ast.h
+++ b/frontends/ast/ast.h
@@ -116,6 +116,7 @@ namespace AST
 		AST_GENVAR,
 		AST_GENFOR,
 		AST_GENIF,
+		AST_GENCASE,
 		AST_GENBLOCK,
 
 		AST_POSEDGE,
@@ -218,6 +219,7 @@ namespace AST
 		RTLIL::Const bitsAsConst(int width = -1);
 		RTLIL::Const asAttrConst();
 		RTLIL::Const asParaConst();
+		bool asBool();
 	};
 
 	// process an AST tree (ast must point to an AST_DESIGN node) and generate RTLIL code
diff --git a/frontends/ast/genrtlil.cc b/frontends/ast/genrtlil.cc
index 7ebc4b719..269752df5 100644
--- a/frontends/ast/genrtlil.cc
+++ b/frontends/ast/genrtlil.cc
@@ -812,6 +812,7 @@ RTLIL::SigSpec AstNode::genRTLIL(int width_hint, bool sign_hint)
 	case AST_GENFOR:
 	case AST_GENBLOCK:
 	case AST_GENIF:
+	case AST_GENCASE:
 		break;
 
 	// remember the parameter, needed for example in techmap
diff --git a/frontends/ast/simplify.cc b/frontends/ast/simplify.cc
index 0a32e9506..ae2f0caae 100644
--- a/frontends/ast/simplify.cc
+++ b/frontends/ast/simplify.cc
@@ -346,7 +346,7 @@ bool AstNode::simplify(bool const_fold, bool at_zero, bool in_lvalue, int stage,
 		bool did_something_here = true;
 		if ((type == AST_GENFOR || type == AST_FOR) && i >= 3)
 			break;
-		if (type == AST_GENIF && i >= 1)
+		if ((type == AST_GENIF || type == AST_GENCASE) && i >= 1)
 			break;
 		if (type == AST_GENBLOCK)
 			break;
@@ -726,7 +726,7 @@ bool AstNode::simplify(bool const_fold, bool at_zero, bool in_lvalue, int stage,
 			// 	dumpAst(f, "verilog-ast> ");
 			log_error("Condition for generate if at %s:%d is not constant!\n", filename.c_str(), linenum);
 		}
-		if (buf->integer != 0) {
+		if (buf->asBool() != 0) {
 			delete buf;
 			buf = children[1]->clone();
 		} else {
@@ -757,6 +757,82 @@ bool AstNode::simplify(bool const_fold, bool at_zero, bool in_lvalue, int stage,
 		did_something = true;
 	}
 
+	// simplify generate-case blocks
+	if (type == AST_GENCASE && children.size() != 0)
+	{
+		AstNode *buf = children[0]->clone();
+		while (buf->simplify(true, false, false, stage, width_hint, sign_hint)) { }
+		if (buf->type != AST_CONSTANT) {
+			// for (auto f : log_files)
+			// 	dumpAst(f, "verilog-ast> ");
+			log_error("Condition for generate case at %s:%d is not constant!\n", filename.c_str(), linenum);
+		}
+
+		bool ref_signed = buf->is_signed;
+		RTLIL::Const ref_value = buf->bitsAsConst();
+		delete buf;
+
+		AstNode *selected_case = NULL;
+		for (size_t i = 1; i < children.size(); i++)
+		{
+			log_assert(children.at(i)->type == AST_COND);
+
+			AstNode *this_genblock = NULL;
+			for (auto child : children.at(i)->children) {
+				log_assert(this_genblock == NULL);
+				if (child->type == AST_GENBLOCK)
+					this_genblock = child;
+			}
+
+			for (auto child : children.at(i)->children)
+			{
+				if (child->type == AST_DEFAULT) {
+					if (selected_case == NULL)
+						selected_case = this_genblock;
+					continue;
+				}
+				if (child->type == AST_GENBLOCK)
+					continue;
+
+				buf = child->clone();
+				while (buf->simplify(true, false, false, stage, width_hint, sign_hint)) { }
+				if (buf->type != AST_CONSTANT) {
+					// for (auto f : log_files)
+					// 	dumpAst(f, "verilog-ast> ");
+					log_error("Expression in generate case at %s:%d is not constant!\n", filename.c_str(), linenum);
+				}
+
+				if (RTLIL::const_eq(ref_value, buf->bitsAsConst(), ref_signed && buf->is_signed, ref_signed && buf->is_signed, 1).as_bool()) {
+					selected_case = this_genblock;
+					i = children.size();
+					break;
+				}
+			}
+		}
+
+		if (selected_case != NULL)
+		{
+			log_assert(selected_case->type == AST_GENBLOCK);
+			buf = selected_case->clone();
+
+			if (!buf->str.empty()) {
+				std::map<std::string, std::string> name_map;
+				buf->expand_genblock(std::string(), buf->str + ".", name_map);
+			}
+
+			for (size_t i = 0; i < buf->children.size(); i++) {
+				buf->children[i]->simplify(false, false, false, stage, -1, false);
+				current_ast_mod->children.push_back(buf->children[i]);
+			}
+
+			buf->children.clear();
+			delete buf;
+		}
+
+		delete_children();
+		did_something = true;
+	}
+
 	// replace primitives with assignmens
 	if (type == AST_PRIMITIVE)
 	{