AST重写规则表达式

Question

我有以下ANTLR规则：

procedure 
: ('int' | 'char') IDENT '(' args ')' body -> ^(PROCEDURE IDENT (args)* body) 
; 

我想捕捉的AST的('int' | 'char')部分。您可以看到，在右侧没有出现，但我不确定如何在AST中显示所选的'int'或'char'。我希望'int'或'char'部分位于PROCEDURE根目录下的树中，但在,IDENT之前。

Answer 1

您可以标记对，然后引用标签在改写：

procedure 
    : primType=('int' | 'char') IDENT '(' args ')' body 
     -> ^(PROCEDURE $primType IDENT args body) 
    ;