Check special matrix functions arguments

Author: mdbrnowski

semantic_listener.py

@@ -15,7 +15,6 @@ class SemanticListener(MyParserListener):
     """Checks break and continue statements, variable declarations,types and assignments."""
 
     def __init__(self):
-        # super().__init__()
         self.nested_loop_counter = 0
         self.variables: dict[str, Type | None] = {}
         self.expr_type: dict[
@@ -114,11 +113,24 @@ def exitMinusExpression(self, ctx: MyParser.MinusExpressionContext):
     def exitSingleExpression(self, ctx: MyParser.SingleExpressionContext):
         self.expr_type[ctx] = self.expr_type[ctx.getChild(0)]
 
-    def enterSpecialMatrixFunction(self, ctx: MyParser.SpecialMatrixFunctionContext):
-        pass
-
     def exitSpecialMatrixFunction(self, ctx: MyParser.SpecialMatrixFunctionContext):
-        pass
+        dimentions = ctx.children[2::2]
+        for dim in dimentions:
+            if self.expr_type[dim] != Type.INT:
+                ctx.parser.notifyErrorListeners(
+                    "Matrix dimentions must be integers", ctx.getChild(0).getSymbol()
+                )
+                self.expr_type[ctx] = None
+                break
+        type_dimentions = []
+        for dim in dimentions:
+            if isinstance(dim, MyParser.SingleExpressionContext) and isinstance(
+                dim.getChild(0), MyParser.IntContext
+            ):
+                type_dimentions.append(int(dim.getText()))
+            else:
+                type_dimentions.append(None)
+        self.expr_type[ctx] = (Type.INT, *type_dimentions)
 
     def exitVector(self, ctx: MyParser.VectorContext):
         elements = ctx.children[1::2]

test_main.py

@@ -78,3 +78,11 @@ def test_sem_error_transpose():
     assert "line 7" in result.stdout
     assert "transpose" in result.stdout.lower()
     assert result.stdout.count("line") == 1
+
+
+def test_sem_error_special_matrix():
+    result = runner.invoke(app, ["sem", "tests/semantic/input_special_matrix.txt"])
+    assert result.exit_code == 0
+    assert "line 1" in result.stdout
+    assert "line 11" in result.stdout
+    assert result.stdout.count("line") == 2

tests/semantic/input_special_matrix.txt

@@ -0,0 +1,11 @@
+A = ones(2.0);
+
+B = ones(1, 1, 1);
+
+n = 10;
+
+C = zeros(2, n);
+
+m = 3.14;
+
+D = eye(m);