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