java - 如何在 ParseTreeMatch ANTLR4 上使用 getAll(...)

Question

我有以下语法:

input
: 
formula EOF
;

formula
:
TRUE                        
| FALSE                     
| formula AND formula       
| formula OR formula        
| (quantifier)+ ST condition        
;


condition
:
atom EQUALS QUOTE? (assignment | atom) QUOTE?
;

quantifier 
:
(FOREACH | EXISTS) variable IN domain 
;
.....

它解析简单的一阶逻辑公式。因此，使用以下代码:

String formulaPatternString = "<formula>";
ParseTreePattern formulaPattern = parser.compileParseTreePattern(formulaPatternString, GraphParser.RULE_formula);
List<ParseTreeMatch> formulaMatches = formulaPattern.findAll(tree, "//formula");

我正在查找在我的输入中找到的公式数量。例如

Exists node in GraphA -> node.color='red' 

返回一个 formulaMatch 和

Exists node in GraphA -> node.color='red' AND Foreach node in GraphA Exists node1 in GraphB -> node.color=node1.color

返回两个formulaMatches。现在我想使用 formulaMatches 来结束公式中的量词数量(如您所见，我允许一个或多个)。我认为我需要的方法是 formulaMatches.get(i).getAll("quantifier") 但这会导致 0 个匹配(在我的情况下，第一个公式中的量词部分是 Exists GraphA 中的节点，第二个是 Foreach 节点 in GraphA Exists node1 in GraphB 这是 2 个量词)。知道如何实现这一目标吗？

Answer 1

formulaMatches 的每个元素将是 ParseTreeMatch可用于获取 ParseTree 的对象对应<formula>模式中的占位符。该解析树将是 FormulaContext 。您可以使用 quantifier()方法FormulaContext获取 QuantifierContext 的数量它有 child :

for (ParseTreeMatch match : formulaMatches) {
  int quantifierCount = ((FormulaContext)match.get("formula")).quantifier().size();
}

注意:如果您使用 ParserInterpreter 进行解析，您的上下文对象将是 InterpreterRuleContext 而不是FormulaContext 。在这种情况下，您需要调用以下命令:

for (ParseTreeMatch match : formulaMatches) {
  ParserRuleContext formulaContext = (FormulaContext)match.get("formula");
  int quantifierCount = 0;
  for (int i = 0; i < formulaContext.getChildCount(); i++) {
    if (formulaContext.getChild(i) instanceof RuleNode
        && ((RuleNode)formulaContext.getChild(i)).getRuleContext().getRuleIndex()
            == RULE_quantifier)
    {
      quantifierCount++;
    }
  }

  // quantifierCount is accurate here...
}