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) 
    ;