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