src/edu/berkeley/sbp/misc/ReflectiveWalker.java

   1 package edu.berkeley.sbp.misc;
   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 class ReflectiveWalker extends StringWalker {
   9     public ReflectiveWalker()              { this.target = this; }
  10     public ReflectiveWalker(Object target) { this.target = target; }
  11     private final Object target;
  12     private String normalize(String s) {
  13         StringBuffer ret = new StringBuffer();
  14         for(int i=0; i<s.length(); i++) {
  15             char c = s.charAt(i);
  16             if (i==0 && c >= '0' && c <= '9') { ret.append('_'); ret.append(c); continue; }
  17             switch(c) {
  18                 case ' ':  ret.append("_space_"); break;
  19                 case '~':  ret.append("_tilde_"); break;
  20                 case ';':  ret.append("_semicolon_"); break;
  21                 case '=':  ret.append("_equals_"); break;
  22                 case '-':  ret.append("_minus_"); break;
  23                 case ')':  ret.append("_rightparen_"); break;
  24                 case '(':  ret.append("_leftparen_"); break;
  25                 case '!':  ret.append("_bang_"); break;
  26                 case '`':  ret.append("_backtick_"); break;
  27                 case '\"': ret.append("_quotes_"); break;
  28                 case '\'': ret.append("_quote_"); break;
  29                 case ',':  ret.append("_comma_"); break;
  30                 case '.':  ret.append("_dot_"); break;
  31                 case '>':  ret.append("_greater_"); break;
  32                 case '<':  ret.append("_less_"); break;
  33                 case '&':  ret.append("_amp_"); break;
  34                 case '@':  ret.append("_at_"); break;
  35                 case '#':  ret.append("_hash_"); break;
  36                 case '|':  ret.append("_pipe_"); break;
  37                 case '^':  ret.append("_caret_"); break;
  38                 case '%':  ret.append("_percent_"); break;
  39                 case '$':  ret.append("_dollar_"); break;
  40                 case ':':  ret.append("_colon_"); break;
  41                 case '?':  ret.append("_question_"); break;
  42                 case '/':  ret.append("_slash_"); break;
  43                 case '\\': ret.append("_backslash_"); break;
  44                 case '*':  ret.append("_star_"); break;
  45                 case '+':  ret.append("_plus_"); break;
  46                 case '_':  ret.append("_underscore_"); break;
  47                 case '[':  ret.append("_leftbracket_"); break;
  48                 case ']':  ret.append("_rightbracket_"); break;
  49                 case '{':  ret.append("_leftbrace_"); break;
  50                 case '}':  ret.append("_rightbrace_"); break;
  51                 default:   ret.append(c); break;
  52             }
  53         }
  54         return ret.toString();
  55     }
  56     public Object walk(Tree<String> tree) {
  57         if (tree.head()!=null) {
  58             Member m = member("$"+normalize(tree.head()), 0, false);
  59         }
  60         return super.walk(tree);
  61     }
  62     public Object walk(String tag, Object[] argo) {
  63         if (argo.length==0) return super.walk(tag, argo);
  64         if (argo==null) return tag;
  65         if (tag==null || "".equals(tag)) return argo;
  66         Member m = member(normalize(tag), argo.length, argo.length>0);
  67         if (m==null) return super.walk(tag, argo);
  68         //System.out.println("preparing to invoke method " + (m==null ? "null" : (m.toString())) + " for sequence " + (owner()+"."+tag));
  69         if (m != null) return Reflection.fuzzyInvoke(target, m, argo);
  70         if (argo.length==0) return null;
  71         if (argo.length==1) return argo[0];
  72         Class c = argo[0].getClass();
  73         for(Object o : argo) c = Reflection.lub(c, o.getClass());
  74         Object[] ret = Reflection.newArray(c, argo.length);
  75         System.arraycopy(argo, 0, ret, 0, argo.length);
  76         return ret;
  77     }
  78
  79     public Member member(String methodName, int nonVoid, boolean complain) {
  80         Class target = this.target.getClass();
  81         if (methodName == null || methodName.equals("")) return null;
  82         Member ret = null;
  83         for (Method m : target.getMethods()) {
  84             if (!m.getName().equals(methodName)) continue;
  85             if (m.getParameterTypes().length != nonVoid) continue;
  86             if (ret != null) {
  87                 if (!complain) return null;
  88                 throw new Error("two methods with " + nonVoid + " parameters: " + target.getName() + "." + methodName);
  89             }
  90             ret = m;
  91         }
  92         if (ret != null) return ret;
  93         Class t = target;
  94         while(t != null) {
  95             Class c = Reflection.forNameOrNull(t.getName() + "$" + methodName);
  96             if (c != null) {
  97                 for (Constructor m : c.getConstructors()) {
  98                     if (m.getParameterTypes().length != nonVoid) continue;
  99                     if (ret != null) {
 100                         if (!complain) return null;
 101                         throw new Error("two constructors with " + nonVoid + " parameters: " + c.getName() + ".<init>()");
 102                     }
 103                     ret = m;
 104                 }
 105                 if (ret != null) return ret;
 106             }
 107             t = t.getSuperclass();
 108         }
 109         if (ret != null) return ret;
 110         if (!complain) return null;
 111         throw new Error("while computing return type of " +methodName+ " could not find method with name " + methodName +
 112                         " and " + nonVoid + " arguments");
 113     }
 114 }