72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
#
|
|
# Copyright (c) 2012-2017 The ANTLR Project. All rights reserved.
|
|
# Use of this file is governed by the BSD 3-clause license that
|
|
# can be found in the LICENSE.txt file in the project root.
|
|
#
|
|
|
|
#
|
|
# A pattern like {@code <ID> = <expr>;} converted to a {@link ParseTree} by
|
|
# {@link ParseTreePatternMatcher#compile(String, int)}.
|
|
#
|
|
from antlr4.tree.ParseTreePatternMatcher import ParseTreePatternMatcher
|
|
from antlr4.tree.Tree import ParseTree
|
|
from antlr4.xpath.XPath import XPath
|
|
|
|
|
|
class ParseTreePattern(object):
|
|
|
|
# Construct a new instance of the {@link ParseTreePattern} class.
|
|
#
|
|
# @param matcher The {@link ParseTreePatternMatcher} which created this
|
|
# tree pattern.
|
|
# @param pattern The tree pattern in concrete syntax form.
|
|
# @param patternRuleIndex The parser rule which serves as the root of the
|
|
# tree pattern.
|
|
# @param patternTree The tree pattern in {@link ParseTree} form.
|
|
#
|
|
def __init__(self, matcher:ParseTreePatternMatcher, pattern:str, patternRuleIndex:int , patternTree:ParseTree):
|
|
self.matcher = matcher
|
|
self.patternRuleIndex = patternRuleIndex
|
|
self.pattern = pattern
|
|
self.patternTree = patternTree
|
|
|
|
#
|
|
# Match a specific parse tree against this tree pattern.
|
|
#
|
|
# @param tree The parse tree to match against this tree pattern.
|
|
# @return A {@link ParseTreeMatch} object describing the result of the
|
|
# match operation. The {@link ParseTreeMatch#succeeded()} method can be
|
|
# used to determine whether or not the match was successful.
|
|
#
|
|
def match(self, tree:ParseTree):
|
|
return self.matcher.match(tree, self)
|
|
|
|
#
|
|
# Determine whether or not a parse tree matches this tree pattern.
|
|
#
|
|
# @param tree The parse tree to match against this tree pattern.
|
|
# @return {@code true} if {@code tree} is a match for the current tree
|
|
# pattern; otherwise, {@code false}.
|
|
#
|
|
def matches(self, tree:ParseTree):
|
|
return self.matcher.match(tree, self).succeeded()
|
|
|
|
# Find all nodes using XPath and then try to match those subtrees against
|
|
# this tree pattern.
|
|
#
|
|
# @param tree The {@link ParseTree} to match against this pattern.
|
|
# @param xpath An expression matching the nodes
|
|
#
|
|
# @return A collection of {@link ParseTreeMatch} objects describing the
|
|
# successful matches. Unsuccessful matches are omitted from the result,
|
|
# regardless of the reason for the failure.
|
|
#
|
|
def findAll(self, tree:ParseTree, xpath:str):
|
|
subtrees = XPath.findAll(tree, xpath, self.matcher.parser)
|
|
matches = list()
|
|
for t in subtrees:
|
|
match = self.match(t)
|
|
if match.succeeded():
|
|
matches.append(match)
|
|
return matches
|