Feature: for token

Author: Rito13

nml/parser.py

@@ -29,6 +29,7 @@
     disable_item,
     error,
     font,
+    for_,
     general,
     grf,
     item,
@@ -47,6 +48,7 @@
 )
 
 
+
 class NMLParser:
     """
     @ivar lexer: Scanner providing tokens.
@@ -145,7 +147,8 @@ def p_main_block(self, t):
         | engine_override
         | sort_vehicles
         | basecost
-        | constant"""
+        | constant
+        | for"""
         t[0] = t[1]
 
     #
@@ -389,7 +392,9 @@ def p_property_assignment(self, t):
         """property_assignment : ID COLON expression SEMICOLON
         | ID COLON expression UNIT SEMICOLON
         | NUMBER COLON expression SEMICOLON
-        | NUMBER COLON expression UNIT SEMICOLON"""
+        | NUMBER COLON expression UNIT SEMICOLON
+        | ID COLON for SEMICOLON
+        | NUMBER COLON for SEMICOLON"""
         name = t[1]
         unit_value = None if len(t) == 5 else unit.get_unit(t[4])
         t[0] = item.Property(name, t[3], unit_value, t.lineno(1))
@@ -835,3 +840,11 @@ def p_tilelayout_item_prop(self, t):
     def p_constant(self, t):
         "constant : CONST expression EQ expression SEMICOLON"
         t[0] = constant.Constant(t[2], t[4])
+
+    def p_for(self, t):
+        """for : FOR ID LPAREN id_list RPAREN LBRACE layout_sprite_list RBRACE
+        | LBRACKET non_empty_expression_list FOR ID IN expression RBRACKET"""
+        if t[1] == '[' :
+            t[0] = for_.For(" ", t[3], t[5])
+        else:
+            t[0] = for_.For(t[2], t[4], t[7])

nml/tokens.py

@@ -61,7 +61,9 @@
     "recolour_sprite":     "RECOLOUR_SPRITE",
     "engine_override":     "ENGINE_OVERRIDE",
     "sort":                "SORT_VEHICLES",
-    "const":               "CONST"
+    "const":               "CONST",
+    "for":                 "FOR",
+    "in":                  "IN",
 }
 # fmt: on