changeset 20132:b91ba8932cb6

Truffle: use visitor for getCallNodes.
author Christian Humer <christian.humer@gmail.com>
date Thu, 02 Apr 2015 16:30:52 +0200
parents 4b12d5355811
children d7d33c72fdc8
files graal/com.oracle.graal.truffle/src/com/oracle/graal/truffle/OptimizedCallTarget.java
diffstat 1 files changed, 10 insertions(+), 2 deletions(-) [+]
line wrap: on
line diff
--- a/graal/com.oracle.graal.truffle/src/com/oracle/graal/truffle/OptimizedCallTarget.java	Thu Apr 02 01:27:27 2015 +0200
+++ b/graal/com.oracle.graal.truffle/src/com/oracle/graal/truffle/OptimizedCallTarget.java	Thu Apr 02 16:30:52 2015 +0200
@@ -322,7 +322,6 @@
     }
 
     private void interpreterCall() {
-        CompilerAsserts.neverPartOfCompilation();
         if (isValid()) {
             // Stubs were deoptimized => reinstall.
             this.runtime.reinstallStubs();
@@ -441,7 +440,16 @@
     }
 
     public List<OptimizedDirectCallNode> getCallNodes() {
-        return NodeUtil.findAllNodeInstances(getRootNode(), OptimizedDirectCallNode.class);
+        final List<OptimizedDirectCallNode> callNodes = new ArrayList<>();
+        getRootNode().accept(new NodeVisitor() {
+            public boolean visit(Node node) {
+                if (node instanceof OptimizedDirectCallNode) {
+                    callNodes.add((OptimizedDirectCallNode) node);
+                }
+                return true;
+            }
+        });
+        return callNodes;
     }
 
     @Override