如何在ANTLRs java运行时获得解析器规则的所有事件？

Question

我正在为自己的ANTLR语法编写Java代码格式化程序，现在我想对一个特殊规则进行格式化。

我想要的是以某种方式迭代一个解析器规则的所有出现（具体来说是if_stmt)。我正在使用ANTLR4和Java运行时。

stmt_block
:
    EOF
    | stmt_list EOF
;

stmt_list
:
    stmt
    | stmt_list stmt
;

stmt
:

    if_stmt
    |  RETURN SEMICOLON NEWLINE?
;

if_stmt
:
    IF BRACKET_OPEN expr BRACKET_CLOSED CBRACKET_OPEN stmt_list CBRACKET_CLOSED
;

expr
:
    TRUE
;

private String formatCode(String input) 
{
    // Parse the input
    MyLexer lex = new MyLexer(new CaseChangingCharStream(
        CharStreams.fromString(input), true));

    CommonTokenStream tokens = new CommonTokenStream(lex);
    tokens.fill();
    MyParser pars = new MyParser(tokens);
    Stmt_blockContext stmt_blockContext = pars.stmt_block();

    stmt_blockContext.children.forEach(parseTree->{
        //Iterate over the parseTree to get all if_stmtContext's and modify them
    });


    // Then return the modified contexts text
    return stmt_blockContext.getText();
}

任何帮助都很感激！

Answer 1

不能只使用解析器进行任何格式化任务，除非您希望它能够只处理有效的输入。格式化主要是重写空白空间，lexer更适合于此。有关如何实现代码格式化程序的示例，请参见我的用于Visual Studio代码的ANTLR4插件。这是针对ANTLR4语法的，但同样的原则也适用于任何其他编程语言。