X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=src%2Fedu%2Fberkeley%2Fsbp%2Futil%2FReflection.java;h=a7c25391539bb1fd9ec569c5aaafe809bb243546;hb=c8a17fdd2e149fe5feecd96c71b7f2cab286ab96;hp=fd43a454fce23b41c5ade3192337aa65a0537ead;hpb=74e53cbf7227f5958ff7495abe31cb0462d62c4e;p=sbp.git

diff --git a/src/edu/berkeley/sbp/util/Reflection.java b/src/edu/berkeley/sbp/util/Reflection.java
index fd43a45..a7c2539 100644
--- a/src/edu/berkeley/sbp/util/Reflection.java
+++ b/src/edu/berkeley/sbp/util/Reflection.java
@@ -1,7 +1,9 @@
 package edu.berkeley.sbp.util;
 import java.io.*;
 import java.lang.reflect.*;
+import java.lang.annotation.*;
 
+// FIXME: decent error reporting
 /** Random reflection-related utilities */
 public final class Reflection {
     
@@ -51,31 +53,33 @@ public final class Reflection {
         return (Object[])Array.newInstance(c, i);
     }
 
-    public static String indent(String s, int indent) {
-        if (s.indexOf('\n')==-1) return s;
-        StringBuffer ret = new StringBuffer();
-        for(int i=0; i<s.length(); i++) {
-            char c = s.charAt(i);
-            ret.append(c);
-            if (c=='\n')
-                for(int j=0; j<indent; j++)
-                    ret.append(' ');
-        }
-        return ret.toString();
-    }
-
     public static String show(Object o) {
         if (o==null) return "null";
+        if (o instanceof Show) return show((Show)o);
         if (! (o instanceof Object[])) return o.toString() + " : " + o.getClass().getName();
         Object[] arr = (Object[])o;
         StringBuffer ret = new StringBuffer();
         ret.append(o.getClass().getComponentType().getName());
-        ret.append("[]:\n");
+        ret.append("["+arr.length+"]:");
         for(int i=0; i<arr.length; i++)
-            ret.append(indent(show(arr[i]), 4) + "\n");
+            ret.append(StringUtil.indent("\n"+show(arr[i]), 4));
         return ret.toString();
     }
 
+    public static String show(Show o) {
+        StringBuffer ret = new StringBuffer();
+        for(Field f : o.getClass().getFields()) {
+            ret.append("\n" + f.getName() + " = ");
+            try {
+                ret.append(show(f.get(o)));
+            } catch (Exception e) {
+                ret.append("**"+e+"**");
+                throw new RuntimeException(e);
+            }
+        }
+        return o.getClass().getName() + " {" + StringUtil.indent(ret.toString(), 4) + "\n}";
+    }
+
     public static Object lub(Object argo) {
         if (argo instanceof Object[]) return lub((Object[])argo);
         return argo;
@@ -197,4 +201,140 @@ public final class Reflection {
         return null;
     }
 
+    public static interface Show {
+    }
+
+    public static Object impose(Object o, Object[] fields) {
+        if (o instanceof Class) return impose((Class)o, fields);
+        if (o instanceof Method) return impose((Method)o, fields);
+        if (o instanceof Constructor) return impose((Constructor)o, fields);
+        return null;
+    }
+    public static Object impose(Class _class, Object[] fields) {
+        Object ret = null;
+        try { ret = _class.newInstance(); }
+        catch (Exception e) { rethrow(e, "while trying to instantiate a " + _class.getName()); }
+        Field[] f = _class.getFields();
+        int j = 0;
+        for(int i=0; i<f.length; i++) {
+            Object tgt = Reflection.lub(fields[i]);
+            if (f[i].getType() == String.class) tgt = stringify(tgt);
+            // FUGLY
+            tgt = coerce(tgt, f[i].getType());
+            try {
+                f[i].set(ret, tgt);
+            } catch (Exception e) {
+                rethrow(e, "while setting \n    " +
+                        f[i] +
+                        "\n     to " + show(tgt));
+            }
+        }
+        return ret;
+    }
+
+    public static Object rethrow(Exception e, String message) {
+        StackTraceElement[] st = e.getStackTrace();
+        RuntimeException re = new RuntimeException(e.getMessage() + "\n  " + message);
+        re.setStackTrace(st);
+        throw re;
+    }
+
+    public static Object impose(Method _method, Object[] fields) {
+        Object[] args = mkArgs(_method.getParameterTypes(), fields);
+        try {
+            return _method.invoke(null, args);
+        } catch (Exception e) {
+            return rethrow(e, "while trying to invoke \n    " +
+                           _method +
+                           "\n    with arguments " + show(args));
+        }
+    }
+    public static Object impose(Constructor _ctor, Object[] fields) {
+        Object[] args = mkArgs(_ctor.getParameterTypes(), fields);
+        try {
+            return _ctor.newInstance(args);
+        } catch (Exception e) {
+            return rethrow(e, "while trying to invoke \n    " +
+                           _ctor +
+                           "\n    with arguments " + show(args));
+        }
+    }
+
+    private static Object[] mkArgs(Class[] argTypes, Object[] fields) {
+        int j = 0;
+        Object[] args = new Object[argTypes.length];
+        for(int i=0; i<args.length; i++) {
+            Object tgt = Reflection.lub(fields[i]);
+            if (argTypes[i] == String.class) tgt = Reflection.stringify(tgt);
+            // FUGLY
+            tgt = Reflection.coerce(tgt, argTypes[i]);
+            args[i] = tgt;
+        }
+        return args;
+    }
+
+    public static String stringify(Object o) {
+        if (o==null) return "";
+        if (!(o instanceof Object[])) return o.toString();
+        Object[] arr = (Object[])o;
+        StringBuffer ret = new StringBuffer();
+        for(int i=0; i<arr.length; i++)
+            ret.append(arr[i]);
+        return ret.toString();
+    }
+
+    public static Object coerce(Object o, Class c) {
+        try {
+            return coerce0(o, c);
+        } catch (Exception e) {
+            return (Object[])rethrow(e, "while trying to coerce " + show(o) + " to a " + c.getName());
+        }
+    }
+    public static Object coerce0(Object o, Class c) {
+        if (o==null) return null;
+        if (c.isInstance(o)) return o;
+        if (c == char.class) {
+            return o.toString().charAt(0);
+        }
+        if (c==int.class || c==Integer.class) {
+            String s = (String)coerce(o, String.class);
+            return new Integer(Integer.parseInt(s));
+        }
+
+        if (o.getClass().isArray() &&
+            o.getClass().getComponentType().isArray() &&
+            o.getClass().getComponentType().getComponentType() == String.class &&
+            c.isArray() &&
+            c.getComponentType() == String.class) {
+            String[] ret = new String[((Object[])o).length];
+            for(int i=0; i<ret.length; i++) {
+                StringBuffer sb = new StringBuffer();
+                for(Object ob : (Object[])(((Object[])o)[i]))
+                    sb.append(ob);
+                ret[i] = sb.toString();
+            }
+            return ret;
+        }
+
+        if (c.isArray() && (c.getComponentType().isInstance(o))) {
+            Object[] ret = (Object[])Array.newInstance(c.getComponentType(), 1);
+            ret[0] = o;
+            return ret;
+        }
+
+        if (o.getClass().isArray() && c.isArray()) {
+            boolean ok = true;
+            for(int i=0; i<((Object[])o).length; i++) {
+                Object ob = (((Object[])o)[i]);
+                if (ob != null) {
+                    ok = false;
+                }
+            }
+            if (ok) {
+                return Array.newInstance(c.getComponentType(), ((Object[])o).length);
+            }
+        }
+        return o;
+    }
+
 }