ANTLR4令牌流或解析树重写

Question

我有一个案例，我不知道该如何处理，需要一些帮助。假设我有以下文字：

spec Foo A, B, C

因为上面是以下内容的简写：

spec Foo A
spec Foo B
spec Foo C

我想以某种方式将第一个版本转换为第二个版本。 因此，具有如下语法：

program: element* EOF;
element: spec;
spec: 'spec' specType=Identifier Identifier (',' Identifier)*;

我希望解析树由每个规范标识符的不同元素节点组成。

我看到ANTLR4不支持树重写，但是有什么办法可以做到吗？