ANTLR 4 - 树模式匹配

Question

我正在尝试了解 ANTLR 4 中的解析树匹配，因此我有以下 Java 代码：

package sampleCodes;

public class fruits {
  public static void main(String[] args){
    int a = 10;
    System.out.println(a);
  }
}

我正在使用 ANTLR 4 创建此代码的解析树。 现在，我想使用树模式匹配函数来查找“int a = 10;”。 GitHub 上有一个文档： https : //github.com/antlr/antlr4/blob/master/doc/tree-matching.md通过一个例子解释了这个（类似这样）：

ParseTree t = ...; // assume t is a statement
ParseTreePattern p = parser.compileParseTreePattern("<ID> = <expr>;", MyParser.RULE_statement);
ParseTreeMatch m = p.match(t);
if ( m.succeeded() ) {...}

通过阅读本文档和其他一些资源，我了解到：

ParseTreePattern p = parser.compileParseTreePattern("<ID> = <expr>;", MyParser.RULE_statement);

作为第二个参数传递的规则必须能够正确解析作为第一个参数提供的模式。 现在我使用的语法是这里给出的 java： https : //github.com/antlr/grammars-v4/tree/master/java

JavaLexer.g4、JavaParser.g4

我无法从上述 GitHub 文档中获得有关如何构建模式字符串及其相应规则的太多信息。 所以我尝试了几种组合来获得匹配，但它们似乎都不起作用。例如：

ParseTreePattern p = parser.compileParseTreePattern("<variableDeclaratorId> = <variableInitializer>", parser.RULE_variableDeclarator);
ParseTreeMatch m = p.match(tree);
System.out.println(m);

这给出：

匹配失败； 找到 0 个标签

我知道我的字符串模式肯定做错了。 任何人都可以帮我解释这个模式匹配函数，并告诉我在这种情况下应该使用什么正确的参数。 此外，提供指向一些有用资源的链接将非常有帮助，我可以在这些资源中了解更多信息并处理复杂模式。（我在 ANTLR4 参考中找不到它）

此代码的解析树的一部分

Answer 1

我认为你想要的在Combining XPath and tree pattern matching 中有描述。

可能是这样的：

import org.antlr.v4.runtime.*;
import org.antlr.v4.runtime.tree.ParseTree;
import org.antlr.v4.runtime.tree.pattern.ParseTreeMatch;
import org.antlr.v4.runtime.tree.pattern.ParseTreePattern;

import java.util.List;

public class Main {

  public static void main(String[] args) {

    String source = "package sampleCodes;\n" +
            "\n" +
            "public class fruits {\n" +
            "\n" +
            "  static { int q = 42; }\n" +
            "\n" +
            "  public static void main(String[] args){\n" +
            "    int a = 10;\n" +
            "    System.out.println(a);\n" +
            "  }\n" +
            "}\n";

    JavaLexer lexer = new JavaLexer(CharStreams.fromString(source));
    JavaParser parser = new JavaParser(new CommonTokenStream(lexer));
    ParseTree tree = parser.compilationUnit();

    ParseTreePattern p = parser.compileParseTreePattern("<IDENTIFIER> = <expression>", JavaParser.RULE_variableDeclarator);
    List<ParseTreeMatch> matches = p.findAll(tree, "//variableDeclarator");

    for (ParseTreeMatch match : matches) {
      System.out.println("\nMATCH:");
      System.out.printf(" - IDENTIFIER: %s\n", match.get("IDENTIFIER").getText());
      System.out.printf(" - expression: %s\n", match.get("expression").getText());
    }
  }
}

导致以下输出：

MATCH:
 - IDENTIFIER: q
 - expression: 42

MATCH:
 - IDENTIFIER: a
 - expression: 10

Answer 2

关于您使用的语法，您的字符串模式是正确的。

match()没有找到任何东西的原因，可能是您将整棵树传递给它（即根中带有规则compilationUnit的树），并且您可能希望它搜索整棵树，而match()只尝试将模式与给定的ParseTree对象匹配。 match()不会尝试在给定ParseTree子树中找到给定模式。 要使其工作，您首先需要找到VariableDeclaratorContext所有节点（通过覆盖BaseListener的enterVariableDeclarator()方法），然后尝试匹配每个节点上的模式。 例如

import org.antlr.v4.runtime.CharStreams;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.tree.ParseTree;
import org.antlr.v4.runtime.tree.ParseTreeWalker;
import org.antlr.v4.runtime.tree.pattern.ParseTreeMatch;
import org.antlr.v4.runtime.tree.pattern.ParseTreePattern;

public class Main {
    public static void main(String[] args) {
        String javaCode = "public class Main {\n" +
                "            public static void main() {\n" +
                "                    int i =0;\n" +
                "            }\n" +
                "}";


        JavaGLexer javaGLexer = new JavaGLexer(CharStreams.fromString(javaCode));
        CommonTokenStream tokens = new CommonTokenStream(javaGLexer);
        JavaGParser javaGParser = new JavaGParser(tokens);
        ParseTree tree = javaGParser.compilationUnit();
        ParseTreePattern p = javaGParser.compileParseTreePattern("<variableDeclaratorId> = <variableInitializer>", javaGParser.RULE_variableDeclarator);
        ParseTreeWalker walker = new ParseTreeWalker();
        IDListener idListener = new IDListener();
        walker.walk(idListener, tree);
        ParseTreeMatch match;
        for (JavaGParser.VariableDeclaratorContext ctx: idListener.getVarCTXs())
        {
            match  = p.match(ctx);
            if (match.succeeded()) {
                System.out.println("Match \n" + " - IDENTIFIER: " +
                        match.get("variableDeclaratorId").getText() +
                        "\n - INITIALIZER: " + match.get("variableInitializer").getText());
            }
        }
    }
}

IDListener扩展JavaGBaseListener并覆盖enterVariableDeclarator() ，并将variableDeclator节点放在一个列表中，该列表可由getVarCTXs()检索。

输出是：

Match
 - IDENTIFIER: i
 - INITIALIZER: 0