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