speed up NodeFinder

6bf66ffd · Stefan Behnel · 0526d1b8 · 6bf66ffd
Commit 6bf66ffd authored Jan 01, 2013 by Stefan Behnel
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

Cython/Compiler/Visitor.py Cython/Compiler/Visitor.py +4 -2

No files found.
--- a/Cython/Compiler/Visitor.py
+++ b/Cython/Compiler/Visitor.py
@@ -550,10 +550,12 @@ class NodeFinder(TreeVisitor):
        self.found = False

    def visit_Node(self, node):
-        if node is self.node:
+        if self.found:
+            pass  # short-circuit
+        elif node is self.node:
            self.found = True
        else:
-            self.visitchildren(node)
+            self._visitchildren(node, None)

 def tree_contains(tree, node):
    finder = NodeFinder(node)