Differentiate between simple and compound assignments

Author: mdbrnowski

MyLexer.g4

@@ -1,4 +1,4 @@
-// $antlr-format columnLimit 100, useTab false, minEmptyLines 0
+// $antlr-format columnLimit 120, useTab false, minEmptyLines 0
 // $antlr-format alignSemicolons none, alignColons trailing
 // $antlr-format alignLexerCommands true, alignLabels true, alignTrailers true
 

MyParser.g4

@@ -1,4 +1,4 @@
-// $antlr-format columnLimit 100, useTab false, minEmptyLines 1
+// $antlr-format columnLimit 120, useTab false, minEmptyLines 1
 // $antlr-format allowShortRulesOnASingleLine false, allowShortBlocksOnASingleLine true
 // $antlr-format alignSemicolons hanging, alignColons hanging
 
@@ -57,13 +57,13 @@ comparison
     ;
 
 assignment
-    : (id | elementReference) (
-        ASSIGN
-        | ASSIGN_PLUS
+    : (id | elementReference) ASSIGN expression SEMICOLON # simpleAssignment
+    | (id | elementReference) (
+        ASSIGN_PLUS
         | ASSIGN_MINUS
         | ASSIGN_MULTIPLY
         | ASSIGN_DIVIDE
-    ) expression SEMICOLON
+    ) expression SEMICOLON # compoundAssignment
     ;
 
 print

ast_listener.py

@@ -50,19 +50,24 @@ def exitWhileLoop(self, ctx: MyParser.WhileLoopContext):
         self.level -= 1
 
     def enterComparison(self, ctx: MyParser.ComparisonContext):
-        operator = ctx.getChild(1)
-        print(IND * self.level + operator.getText())
+        print(IND * self.level + ctx.getChild(1).getText())
         self.level += 1
 
     def exitComparison(self, ctx: MyParser.ComparisonContext):
         self.level -= 1
 
-    def enterAssignment(self, ctx: MyParser.AssignmentContext):
-        operator = ctx.getChild(1)
-        print(IND * self.level + operator.getText())
+    def enterSimpleAssignment(self, ctx: MyParser.SimpleAssignmentContext):
+        print(IND * self.level + ctx.getChild(1).getText())
+        self.level += 1
+
+    def exitSimpleAssignment(self, ctx: MyParser.SimpleAssignmentContext):
+        self.level -= 1
+
+    def enterCompoundAssignment(self, ctx: MyParser.CompoundAssignmentContext):
+        print(IND * self.level + ctx.getChild(1).getText())
         self.level += 1
 
-    def exitAssignment(self, ctx: MyParser.AssignmentContext):
+    def exitCompoundAssignment(self, ctx: MyParser.CompoundAssignmentContext):
         self.level -= 1
 
     def enterPrint(self, ctx: MyParser.PrintContext):