We can now handle array slices (e.g. $size(x[1]) etc. )

2 files changed, 20 insertions, 9 deletions

diff --git a/frontends/ast/simplify.cc b/frontends/ast/simplify.cc
index 68b6ca7fa..95f2d1810 100644
--- a/frontends/ast/simplify.cc
+++ b/frontends/ast/simplify.cc
@@ -1523,6 +1523,7 @@ bool AstNode::simplify(bool const_fold, bool at_zero, bool in_lvalue, int stage,
 	{
 		AstNode *index_expr = nullptr;
 
+		integer = children[0]->children.size(); // save original number of dimensions for $size() etc.
 		for (int i = 0; 2*i < GetSize(id2ast->multirange_dimensions); i++)
 		{
 			if (GetSize(children[0]->children) < i)
@@ -1721,6 +1722,7 @@ bool AstNode::simplify(bool const_fold, bool at_zero, bool in_lvalue, int stage,
 
 		newNode = new AstNode(AST_IDENTIFIER, children[1]->clone());
 		newNode->str = wire_id;
+		newNode->integer = integer; // save original number of dimensions for $size() etc.
 		newNode->id2ast = wire;
 		goto apply_newNode;
 	}
@@ -2869,15 +2871,13 @@ skip_dynamic_range_lvalue_expansion:;
 						log_file_error(filename, location.first_line, "Failed to resolve identifier %s for width detection!\n", buf->str.c_str());
 					// a slice of our identifier means we advance to the next dimension, e.g. $size(a[3])
 					if (buf->children.size() > 0) {
-						// we give up here because when we try to support thing such as $size(a[1][1]) the AST at this point doesn't contain
-						// the information how many indexes were given. The array is already flattened and a composite index is given in the AST instead.
-						log_file_error(filename, location.first_line, "%s() only supported for pure identifiers (no further indexing)!\n", str.c_str());
 						// something is hanging below this identifier
-						if (buf->children[0]->type == AST_RANGE)
+						if (buf->children[0]->type == AST_RANGE && buf->integer == 0)
+							// if integer == 0, this node was originally created as AST_RANGE so it's dimension is 1
 							dim++;
 						// more than one range, e.g. $size(a[3][2])
-						else if (buf->children[0]->type == AST_MULTIRANGE)
-							dim += buf->children[0]->children.size(); // increment by multirange size
+						else // created an AST_MULTIRANGE, converted to AST_RANGE, but original dimension saved in 'integer' field
+							dim += buf->integer; // increment by multirange size
 					}
 					// We have 4 cases:
 					// wire x;                ==> AST_WIRE, no AST_RANGE children
@@ -2957,7 +2957,6 @@ skip_dynamic_range_lvalue_expansion:;
 				else {
 					result = width * mem_depth;
 				}
-
 				newNode = mkconst_int(result, false);
 				goto apply_newNode;
 			}
diff --git a/tests/sat/sizebits.sv b/tests/sat/sizebits.sv
index 408c6f5c8..87fa08f89 100644
--- a/tests/sat/sizebits.sv
+++ b/tests/sat/sizebits.sv
@@ -25,8 +25,8 @@ assert property ($size(z, 1) == 6);
 assert property ($size(z, 2) == 8);
 assert property ($size(z, 3) == 4);
 // This is unsupported at the moment
-//assert property ($size(z[3], 1) == 8);
-//assert property ($size(z[3][3], 1) == 4);
+assert property ($size(z[3], 1) == 8);
+assert property ($size(z[3][3], 1) == 4);
 //assert property ($size(z[3][3][3], 1) == 1);
 // This should trigger an error if enabled (it does).
 //assert property ($size(z, 4) == 4);
@@ -48,6 +48,9 @@ assert property ($high(z) == 7);
 assert property ($high(z, 1) == 7);
 assert property ($high(z, 2) == 9);
 assert property ($high(z, 3) == 3);
+assert property ($high(z[3]) == 9);
+assert property ($high(z[3][3]) == 3);
+assert property ($high(z[3], 2) == 3);
 
 assert property ($low(x) == 2);
 assert property ($low(y) == 2);
@@ -58,6 +61,9 @@ assert property ($low(z) == 2);
 assert property ($low(z, 1) == 2);
 assert property ($low(z, 2) == 2);
 assert property ($low(z, 3) == 0);
+assert property ($low(z[3]) == 2);
+assert property ($low(z[3][3]) == 0);
+assert property ($low(z[3], 2) == 0);
 
 assert property ($left(x) == 5);
 assert property ($left(y) == 2);
@@ -68,6 +74,9 @@ assert property ($left(z) == 7);
 assert property ($left(z, 1) == 7);
 assert property ($left(z, 2) == 2);
 assert property ($left(z, 3) == 3);
+assert property ($left(z[3]) == 2);
+assert property ($left(z[3][3]) == 3);
+assert property ($left(z[3], 2) == 3);
 
 assert property ($right(x) == 2);
 assert property ($right(y) == 7);
@@ -78,4 +87,7 @@ assert property ($right(z) == 2);
 assert property ($right(z, 1) == 2);
 assert property ($right(z, 2) == 9);
 assert property ($right(z, 3) == 0);
+assert property ($right(z[3]) == 9);
+assert property ($right(z[3][3]) == 0);
+assert property ($right(z[3], 2) == 0);
 endmodule