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

diff --git a/frontends/ast/simplify.cc b/frontends/ast/simplify.cc
index 68b6ca7fa27ecf92b63e59764139e00a056a0ca7..95f2d18101509953f1686d726937f6f9e96212f1 100644 (file)
--- 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 408c6f5c8b8f69cd13683dff59781ede55b9c32d..87fa08f89243eb7fc73c16fe1f62740e094b587e 100644 (file)
--- 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