src/edu/berkeley/sbp/bind/Bindable.java

   1 package edu.berkeley.sbp.bind;
   2
   3 import edu.berkeley.sbp.util.*;
   4 import edu.berkeley.sbp.*;
   5 import edu.berkeley.sbp.chr.*;
   6 import edu.berkeley.sbp.bind.*;
   7 import java.util.*;
   8 import java.lang.annotation.*;
   9 import java.lang.reflect.*;
  10 import java.io.*;
  11 import static edu.berkeley.sbp.util.Reflection.*;
  12
  13 public abstract class Bindable implements ToJava {
  14
  15     public abstract String getSimpleName();
  16     public abstract String toString();
  17     public abstract <A extends Annotation> A getAnnotation(Class<A> c);
  18     public abstract Object impose(Object[] fields);
  19     public boolean isAnnotationPresent(Class<? extends Annotation> c) { return getAnnotation(c) != null; }
  20
  21     public abstract Annotation[][] getArgAnnotations();
  22     public abstract String[]       getArgNames();
  23     public abstract Class[]        getArgTypes();
  24
  25     public static Bindable create(Object o) {
  26         if (o instanceof Class) return new BindableClass((Class)o);
  27         if (o instanceof Method) return new BindableMethod((Method)o);
  28         if (o instanceof Constructor) return new BindableConstructor((Constructor)o);
  29         return null;
  30     }
  31
  32     public Binding createBinding() { return new SimpleBinding(); }
  33     public Binding createBinding(final int[] map) { return new SimpleBinding(map); }
  34     public Binding createBinding(final int[] map, Object prepend) { return new SimpleBinding(map, prepend); }
  35
  36     public class SimpleBinding implements Binding, ToJava {
  37         private int[] map = null;
  38         private Object prepend = null;
  39         public SimpleBinding() { }
  40         public SimpleBinding(int[] map) { this.map = map; }
  41         public SimpleBinding(int[] map, Object prepend) { this.map = map; this.prepend = prepend; }
  42
  43         public Object invoke(Object[] o) {
  44             if (map==null) return impose(o);
  45             int max = 0;
  46             for(int i=0; i<map.length; i++) max = Math.max(map[i], max);
  47             Object[] o2 = new Object[max+1];
  48             for(int i=0; i<o.length; i++) o2[map[i]+(prepend==null?0:1)] = o[i];
  49             if (prepend != null) o2[0] = prepend;
  50             return impose(o2);
  51         }
  52
  53         public void toJava(StringBuffer sb) {
  54             Bindable.this.toJava(sb);
  55             sb.append(".createBinding(");
  56             if (map != null) {
  57                 sb.append("new int[] {");
  58                 for(int i=0; i<map.length; i++) {
  59                     sb.append(i);
  60                     if (i<map.length-1) sb.append(",");
  61                 }
  62                 sb.append("}");
  63             }
  64             sb.append(")");
  65         }
  66     }
  67
  68
  69     private static class BindableMethod extends Bindable {
  70         private final Method _method;
  71         public String toString() { return "BindableMethod["+_method+"]"; }
  72         public BindableMethod(Method _method) { this._method = _method; }
  73         public String getSimpleName() { return _method.getName(); }
  74         public <A extends Annotation> A getAnnotation(Class<A> c) { return _method.getAnnotation(c); }
  75         public Object impose(Object[] fields) { return Reflection.impose(_method, fields); }
  76         public Annotation[][] getArgAnnotations() { return _method.getParameterAnnotations(); }
  77         public String[]       getArgNames() { return new String[_method.getParameterTypes().length]; }
  78         public Class[]        getArgTypes() { return _method.getParameterTypes(); }
  79         public void toJava(StringBuffer sb) {
  80             sb.append("Bindable.create(");
  81             sb.append(_method.getDeclaringClass().getName().replace('$','.'));
  82             sb.append(".class.getMethod(\"");
  83             sb.append(_method.getName());
  84             sb.append("\", ");
  85             appendClassArray(sb, _method.getParameterTypes());
  86             sb.append("))");
  87         }
  88     }
  89
  90     private static class BindableConstructor extends Bindable {
  91         private final Constructor _constructor;
  92         public String toString() { return "BindableConstructor["+_constructor+"]"; }
  93         public BindableConstructor(Constructor _constructor) { this._constructor = _constructor; }
  94         public String getSimpleName() { return _constructor.getDeclaringClass().getSimpleName(); }
  95         public <A extends Annotation> A getAnnotation(Class<A> c) { return _constructor.getAnnotation(c); }
  96         public Object impose(Object[] fields) { return Reflection.impose(_constructor, fields); }
  97         public Annotation[][] getArgAnnotations() { return _constructor.getParameterAnnotations(); }
  98         public String[]       getArgNames() { return new String[_constructor.getParameterTypes().length]; }
  99         public Class[]        getArgTypes() { return _constructor.getParameterTypes(); }
 100         public void toJava(StringBuffer sb) {
 101             sb.append("Bindable.create(");
 102             sb.append(_constructor.getDeclaringClass().getName().replace('$','.'));
 103             sb.append(".class.getConstructor(");
 104             appendClassArray(sb, _constructor.getParameterTypes());
 105             sb.append("))");
 106         }
 107     }
 108
 109     private static class BindableClass extends Bindable {
 110         private final Class _class;
 111         public String toString() { return "BindableClass["+_class+"]"; }
 112         public BindableClass(Class _class) { this._class = _class; }
 113         public String getSimpleName() { return _class.getSimpleName(); }
 114         public <A extends Annotation> A getAnnotation(Class<A> c) { return (A)_class.getAnnotation(c); }
 115         public Object impose(Object[] fields) { return Reflection.impose(_class, fields); }
 116         public Annotation[][] getArgAnnotations() {
 117             Field[] fields = _class.getFields();
 118             Annotation[][] ret = new Annotation[fields.length][];
 119             for(int i=0; i<fields.length; i++)
 120                 ret[i] = fields[i].getAnnotations();
 121             return ret;
 122         }
 123         public String[]       getArgNames() {
 124             Field[] fields = _class.getFields();
 125             String[] ret = new String[fields.length];
 126             for(int i=0; i<fields.length; i++)
 127                 ret[i] = fields[i].getName();
 128             return ret;
 129         }
 130         public Class[]       getArgTypes() {
 131             Field[] fields = _class.getFields();
 132             Class[] ret = new Class[fields.length];
 133             for(int i=0; i<fields.length; i++)
 134                 ret[i] = fields[i].getType();
 135             return ret;
 136         }
 137         public void toJava(StringBuffer sb) {
 138             sb.append("Bindable.create(");
 139             sb.append(_class.getName().replace('$','.'));
 140             sb.append(".class)");
 141         }
 142     }
 143
 144
 145     private static void appendClassArray(StringBuffer sb, Class[] c) {
 146         sb.append("new Class[] {");
 147         for(int i=0; i<c.length; i++) {
 148             sb.append(makeClass(c[i]));
 149             sb.append(".class");
 150             if (i<c.length-1)
 151                 sb.append(",");
 152         }
 153         sb.append("}");
 154     }
 155
 156     private static String makeClass(Class c) {
 157         if (!c.isArray()) return c.getName().replace('$','.');
 158         return makeClass(c.getComponentType())+"[]";
 159     }
 160 }