diff graal/com.oracle.truffle.api/src/com/oracle/truffle/api/nodes/Node.java @ 10590:6eb8d63cea34

Added a feature to trace rewrites of truffle nodes.
author Christian Humer <christian.humer@gmail.com>
date Wed, 26 Jun 2013 17:21:59 +0200
parents 29e9a5d18c70
children 99b58803d6d9
line wrap: on
line diff
--- a/graal/com.oracle.truffle.api/src/com/oracle/truffle/api/nodes/Node.java	Wed Jun 26 15:22:21 2013 +0200
+++ b/graal/com.oracle.truffle.api/src/com/oracle/truffle/api/nodes/Node.java	Wed Jun 26 17:21:59 2013 +0200
@@ -26,6 +26,7 @@
 import java.util.*;
 
 import com.oracle.truffle.api.*;
+import com.oracle.truffle.api.nodes.NodeInfo.Kind;
 
 /**
  * Abstract base class for all Truffle nodes.
@@ -199,6 +200,72 @@
      * @param reason the reason the replace supplied
      */
     protected void onReplace(Node newNode, String reason) {
+        if (TruffleOptions.TraceRewrites) {
+            Class<? extends Node> from = getClass();
+            Class<? extends Node> to = newNode.getClass();
+
+            if (TruffleOptions.TraceRewritesFilterFromKind != null) {
+                if (filterByKind(from, TruffleOptions.TraceRewritesFilterFromKind)) {
+                    return;
+                }
+            }
+
+            if (TruffleOptions.TraceRewritesFilterToKind != null) {
+                if (filterByKind(to, TruffleOptions.TraceRewritesFilterToKind)) {
+                    return;
+                }
+            }
+
+            String filter = TruffleOptions.TraceRewritesFilterClass;
+            if (filter != null && (filterByContainsClassName(from, filter) || filterByContainsClassName(to, filter))) {
+                return;
+            }
+
+            // CheckStyle: stop system..print check
+            System.out.printf("[truffle]   rewrite %-50s |From %-40s |To %-40s |Reason %s.%n", this.toString(), formatNodeInfo(from), formatNodeInfo(to), reason);
+            // CheckStyle: resume system..print check
+        }
+    }
+
+    private static String formatNodeInfo(Class<? extends Node> clazz) {
+        NodeInfo nodeInfo = clazz.getAnnotation(NodeInfo.class);
+        String kind = "?";
+        if (nodeInfo != null) {
+            switch (nodeInfo.kind()) {
+                case GENERIC:
+                    kind = "G";
+                    break;
+                case SPECIALIZED:
+                    kind = "S";
+                    break;
+                case UNINITIALIZED:
+                    kind = "U";
+                    break;
+                default:
+                    kind = "?";
+                    break;
+            }
+        }
+        return kind + " " + clazz.getSimpleName();
+    }
+
+    private static boolean filterByKind(Class<?> clazz, Kind kind) {
+        NodeInfo info = clazz.getAnnotation(NodeInfo.class);
+        if (info != null) {
+            return info.kind() != kind;
+        }
+        return true;
+    }
+
+    private static boolean filterByContainsClassName(Class<? extends Node> from, String filter) {
+        Class<?> currentFrom = from;
+        while (currentFrom != null) {
+            if (currentFrom.getName().contains(filter)) {
+                return false;
+            }
+            currentFrom = currentFrom.getSuperclass();
+        }
+        return true;
     }
 
     /**