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