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     /*
  57     public Object walk(Tree<String> tree) {
  58         if (tree.head()!=null) {
  59             Member m = member("$"+normalize(tree.head()), 0, false);
  60             if (m!=null) {
  61                 if ((m instanceof Method) && ((Method)m).getReturnType()==Void.TYPE) {
  62                     Reflection.fuzzyInvoke(target, m, new Object[0]);
  63                 }
  64             }
  65         }
  66         return super.walk(tree);
  67     }
  68     */
  69     public Object walk(String tag, Object[] argo) {
  70         if (argo.length==0) return super.walk(tag, argo);
  71         if (argo==null) return tag;
  72         if (tag==null || "".equals(tag)) return argo;
  73         Member m = member(normalize(tag), argo.length, argo.length>0);
  74         if (m==null) return super.walk(tag, argo);
  75         //System.out.println("preparing to invoke method " + (m==null ? "null" : (m.toString())) + " for sequence " + (owner()+"."+tag));
  76         if (m != null) return Reflection.fuzzyInvoke(target, m, argo);
  77         if (argo.length==0) return null;
  78         if (argo.length==1) return argo[0];
  79         Class c = argo[0].getClass();
  80         for(Object o : argo) c = Reflection.lub(c, o.getClass());
  81         Object[] ret = Reflection.newArray(c, argo.length);
  82         System.arraycopy(argo, 0, ret, 0, argo.length);
  83         return ret;
  84     }
  85
  86     public Member member(String methodName, int nonVoid, boolean complain) {
  87         Class target = this.target.getClass();
  88         if (methodName == null || methodName.equals("")) return null;
  89         Member ret = null;
  90         for (Method m : target.getMethods()) {
  91             if (!m.getName().equals(methodName)) continue;
  92             if (m.getParameterTypes().length != nonVoid) continue;
  93             if (ret != null) {
  94                 if (!complain) return null;
  95                 throw new Error("two methods with " + nonVoid + " parameters: " + target.getName() + "." + methodName);
  96             }
  97             ret = m;
  98         }
  99         if (ret != null) return ret;
 100         Class t = target;
 101         while(t != null) {
 102             Class c = Reflection.forNameOrNull(t.getName() + "$" + methodName);
 103             if (c != null) {
 104                 for (Constructor m : c.getConstructors()) {
 105                     if (m.getParameterTypes().length != nonVoid) continue;
 106                     if (ret != null) {
 107                         if (!complain) return null;
 108                         throw new Error("two constructors with " + nonVoid + " parameters: " + c.getName() + ".<init>()");
 109                     }
 110                     ret = m;
 111                 }
 112                 if (ret != null) return ret;
 113             }
 114             t = t.getSuperclass();
 115         }
 116         if (ret != null) return ret;
 117         if (!complain) return null;
 118         throw new Error("while computing return type of " +methodName+ " could not find method with name " + methodName +
 119                         " and " + nonVoid + " arguments");
 120     }
 121 }