X-Git-Url: http://git.megacz.com/?p=sbp.git;a=blobdiff_plain;f=src%2Fedu%2Fberkeley%2Fsbp%2FTree.java;h=38fcd32c8d2808afacd912a5b13f3b2c6d2ab924;hp=ca351d337d3fec34777962e5c93b9495c9a89ae6;hb=8abe88ea7a5d7b1fc20c45b05e6397dad81e7cdb;hpb=0bf2fced924400a496fb154be71188864296564c

diff --git a/src/edu/berkeley/sbp/Tree.java b/src/edu/berkeley/sbp/Tree.java
index ca351d3..38fcd32 100644
--- a/src/edu/berkeley/sbp/Tree.java
+++ b/src/edu/berkeley/sbp/Tree.java
@@ -42,7 +42,15 @@ public class Tree<T>
     }
 
     protected String headToString() { return head==null?null:head.toString(); }
-    protected String headToJava()   { return head==null?null:StringUtil.toJavaString(head+""); }
+    protected String headToJava()   {
+        if (head==null) return null;
+        if (head instanceof ToJava) {
+            StringBuffer sb = new StringBuffer();
+            ((ToJava)head).toJava(sb);
+            return sb.toString();
+        }
+        return head==null?"null":("\""+StringUtil.toJavaString(head.toString())+"\"");
+    }
     protected String left()   { return "{"; }
     protected String right()  { return "}"; }
     protected boolean ignoreSingleton() { return false; }