Add keyby parser rule

cda-group · Oct 27, 2018 · 02bd84e · 02bd84e
1 parent 713974f
commit 02bd84e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/antlr4/Arc.g4 b/src/main/antlr4/Arc.g4
@@ -111,6 +111,7 @@ iterator	: iter=(TScalarIter|TSimdIter|TFringeIter|TNdIter) '(' data=expr ')' #
 			| iter=(TScalarIter|TSimdIter|TFringeIter) '(' data=expr ',' start=expr ',' end=expr ',' stride=expr ')' # FourIter
 			| TNdIter '(' data=expr ',' start=expr ',' end=expr ',' stride=expr ',' shape=expr ',' strides=expr ')' # SixIter
 			| TRangeIter '(' start=expr ',' end=expr ',' stride=expr ')' # RangeIter
+      | TByKeyIter '(' data=expr ',' keyFunc=expr ')' # ByKeyIter
 			| expr # UnkownIter
 			;
 
@@ -129,4 +130,4 @@ commutativeBinop	: TPlus # SumOp
 annotations : TAt '(' entries+=annotationPair (',' entries+=annotationPair)* ')';
 annotationPair	: name=TIdentifier ':' value=TIdentifier # IdPair
 				| name=TIdentifier ':' value=literalExpr # LiteralPair
-				;
+				;