src/edu/berkeley/sbp/util/PrintableTree.java

   1 package edu.berkeley.sbp.util;
   2 import edu.berkeley.sbp.*;
   3 import edu.berkeley.sbp.util.*;
   4 import java.io.*;
   5 import java.util.*;
   6 import java.lang.reflect.*;
   7
   8 public abstract class PrintableTree<T extends PrintableTree> implements Iterable<T>, ToJava /*, ToHTML*/ {
   9
  10     protected abstract String headToString();
  11     protected abstract String headToJava();
  12     protected abstract String left();
  13     protected abstract String right();
  14     protected abstract boolean ignoreSingleton();
  15
  16     private static final int MAXCHARS=40;
  17
  18     private boolean basic() { return toString().length() < MAXCHARS; }
  19     public String toPrettyString() { return toPrettyString("\n"); }
  20     private String toPrettyString(String nl) {
  21         String str = toString();
  22         if (str.length() < MAXCHARS) return str;
  23         String head = headToString();
  24         StringBuffer ret = new StringBuffer();
  25
  26         Iterator<T> iterator = iterator();
  27         if (!iterator.hasNext()) return head==null ? (left()+right()) : head;
  28         PrintableTree t0 = iterator.next();
  29         if (!iterator.hasNext() && ignoreSingleton())
  30             return t0.toPrettyString(nl);
  31
  32         ret.append(head==null?(left()+" "):(head+":"+nl));
  33         boolean first = true;
  34         int len = 0;
  35         for(T t : this) {
  36             String s = t.basic() ? t.toString() : t.toPrettyString(nl+"  ");
  37             if (!first) {
  38                 if (!t.basic())                ret.append(nl);
  39                 if (s.length()+len>MAXCHARS) { ret.append(nl); len = 0; }
  40                 else                         { ret.append(" "); len++; }
  41             }
  42             first = false;
  43             ret.append(s);
  44             len += s.length();
  45         }
  46         if (head==null) ret.append(" "+right());
  47         return ret.toString();
  48     }
  49
  50     public String toString() {
  51         StringBuffer ret = new StringBuffer();
  52         int count=0;
  53         for(T t : this) {
  54             count++;
  55             String q = t==null ? "null" : t.toString();
  56             if (q.length() > 0) { ret.append(q); ret.append(" "); }
  57         }
  58         if (count==1 && ignoreSingleton()) return ret.toString().trim();
  59         String tail = ret.toString().trim();
  60         String head = headToString();
  61         String h = (head!=null && !head.toString().equals("")) ? (tail.length() > 0 ? head+":" : head+"") : "";
  62         if (tail.length() > 0) tail = left() + tail + right();
  63         return h + tail;
  64     }
  65
  66     /** append Java code to <tt>sb</tt> which evaluates to this instance */
  67     public void toJava(StringBuffer sb) {
  68         sb.append("new "+this.getClass().getName()+"(null, ");
  69         String head = headToJava();
  70         sb.append(head==null ? "null" : "\"" + head + "\"");
  71         sb.append(", new "+this.getClass().getName()+"[] { ");
  72         boolean first = true;
  73         for(T t : this) {
  74             if (!first) sb.append(",\n        ");
  75             if (t==null)   sb.append("null");
  76             else           t.toJava(sb);
  77             first = false;
  78         }
  79         sb.append("})");
  80     }
  81
  82 }