Mercurial Hosting > luan
view src/luan/LuaJavaFunction.java @ 21:c93d8c781853
add functions
git-svn-id: https://luan-java.googlecode.com/svn/trunk@22 21e917c8-12df-6dd8-5cb6-c86387c605b9
author | fschmidt@gmail.com <fschmidt@gmail.com@21e917c8-12df-6dd8-5cb6-c86387c605b9> |
---|---|
date | Tue, 04 Dec 2012 09:16:03 +0000 |
parents | 9cea1aea5eef |
children | 8a57ebfdfd78 |
line wrap: on
line source
package luan; import java.lang.reflect.Array; import java.lang.reflect.Method; import java.lang.reflect.InvocationTargetException; public final class LuaJavaFunction extends LuaFunction { private final Method method; private final Object obj; private final RtnConverter rtnConverter; private final boolean takesLuaState; private final ArgConverter[] argConverters; private final Class<?> varArgCls; public LuaJavaFunction(Method method,Object obj) { this.method = method; this.obj = obj; this.rtnConverter = getRtnConverter(method); this.takesLuaState = takesLuaState(method); this.argConverters = getArgConverters(takesLuaState,method); if( method.isVarArgs() ) { Class<?>[] paramTypes = method.getParameterTypes(); this.varArgCls = paramTypes[paramTypes.length-1].getComponentType(); } else { this.varArgCls = null; } } @Override public Object[] call(LuaState lua,Object... args) { args = fixArgs(lua,args); Object rtn; try { rtn = method.invoke(obj,args); } catch(IllegalAccessException e) { throw new RuntimeException(e); } catch(InvocationTargetException e) { throw new RuntimeException(e); } return rtnConverter.convert(rtn); } private Object[] fixArgs(LuaState lua,Object[] args) { int n = argConverters.length; Object[] rtn; int start = 0; if( !takesLuaState && varArgCls==null && args.length == n ) { rtn = args; } else { if( takesLuaState ) n++; rtn = new Object[n]; if( takesLuaState ) { rtn[start++] = lua; } n = argConverters.length; if( varArgCls != null ) { n--; if( args.length < argConverters.length ) { rtn[rtn.length-1] = Array.newInstance(varArgCls,0); } else { Object[] varArgs = (Object[])Array.newInstance(varArgCls,args.length-n); ArgConverter ac = argConverters[n]; for( int i=0; i<varArgs.length; i++ ) { varArgs[i] = ac.convert(args[n+i]); } rtn[rtn.length-1] = varArgs; } } System.arraycopy(args,0,rtn,start,Math.min(args.length,n)); } for( int i=0; i<n; i++ ) { rtn[start+i] = argConverters[i].convert(rtn[start+i]); } return rtn; } private interface RtnConverter { public Object[] convert(Object obj); } private static final RtnConverter RTN_EMPTY = new RtnConverter() { public Object[] convert(Object obj) { return EMPTY_RTN; } }; private static final RtnConverter RTN_ARRAY = new RtnConverter() { public Object[] convert(Object obj) { return (Object[])obj; } }; private static final RtnConverter RTN_ONE = new RtnConverter() { public Object[] convert(Object obj) { return new Object[]{obj}; } }; private static final RtnConverter RTN_NUMBER = new RtnConverter() { public Object[] convert(Object obj) { if( obj == null ) return new Object[1]; Number n = (Number)obj; LuaNumber ln = new LuaNumber(n.doubleValue()); return new Object[]{ln}; } }; private static RtnConverter getRtnConverter(Method m) { Class<?> rtnType = m.getReturnType(); if( rtnType == Void.TYPE ) return RTN_EMPTY; if( Number.class.isAssignableFrom(rtnType) || rtnType == Short.TYPE || rtnType == Integer.TYPE || rtnType == Long.TYPE || rtnType == Float.TYPE || rtnType == Double.TYPE ) return RTN_NUMBER; if( rtnType.isArray() ) return RTN_ARRAY; return RTN_ONE; } private interface ArgConverter { public Object convert(Object obj); } private static final ArgConverter ARG_SAME = new ArgConverter() { public Object convert(Object obj) { return obj; } }; private static final ArgConverter ARG_DOUBLE = new ArgConverter() { public Object convert(Object obj) { if( obj instanceof LuaNumber ) { LuaNumber ln = (LuaNumber)obj; return Double.valueOf(ln.n); } if( obj instanceof String ) { String s = (String)obj; try { return Double.valueOf(s); } catch(NumberFormatException e) {} } return obj; } }; private static final ArgConverter ARG_FLOAT = new ArgConverter() { public Object convert(Object obj) { if( obj instanceof LuaNumber ) { LuaNumber ln = (LuaNumber)obj; return Float.valueOf((float)ln.n); } if( obj instanceof String ) { String s = (String)obj; try { return Float.valueOf(s); } catch(NumberFormatException e) {} } return obj; } }; private static final ArgConverter ARG_LONG = new ArgConverter() { public Object convert(Object obj) { if( obj instanceof LuaNumber ) { LuaNumber ln = (LuaNumber)obj; long i = (long)ln.n; if( i == ln.n ) return Long.valueOf(i); } else if( obj instanceof String ) { String s = (String)obj; try { return Long.valueOf(s); } catch(NumberFormatException e) {} } return obj; } }; private static final ArgConverter ARG_INTEGER = new ArgConverter() { public Object convert(Object obj) { if( obj instanceof LuaNumber ) { LuaNumber ln = (LuaNumber)obj; int i = (int)ln.n; if( i == ln.n ) return Integer.valueOf(i); } else if( obj instanceof String ) { String s = (String)obj; try { return Integer.valueOf(s); } catch(NumberFormatException e) {} } return obj; } }; private static final ArgConverter ARG_SHORT = new ArgConverter() { public Object convert(Object obj) { if( obj instanceof LuaNumber ) { LuaNumber ln = (LuaNumber)obj; short i = (short)ln.n; if( i == ln.n ) return Short.valueOf(i); } else if( obj instanceof String ) { String s = (String)obj; try { return Short.valueOf(s); } catch(NumberFormatException e) {} } return obj; } }; private static boolean takesLuaState(Method m) { Class<?>[] paramTypes = m.getParameterTypes(); return paramTypes.length > 0 && paramTypes[0].equals(LuaState.class); } private static ArgConverter[] getArgConverters(boolean takesLuaState,Method m) { final boolean isVarArgs = m.isVarArgs(); Class<?>[] paramTypes = m.getParameterTypes(); if( takesLuaState ) { Class<?>[] t = new Class<?>[paramTypes.length-1]; System.arraycopy(paramTypes,1,t,0,t.length); paramTypes = t; } ArgConverter[] a = new ArgConverter[paramTypes.length]; for( int i=0; i<a.length; i++ ) { Class<?> paramType = paramTypes[i]; if( isVarArgs && i == a.length-1 ) paramType = paramType.getComponentType(); a[i] = getArgConverter(paramType); } return a; } private static ArgConverter getArgConverter(Class<?> cls) { if( cls == Double.TYPE || cls.equals(Double.class) ) return ARG_DOUBLE; if( cls == Float.TYPE || cls.equals(Float.class) ) return ARG_FLOAT; if( cls == Long.TYPE || cls.equals(Long.class) ) return ARG_LONG; if( cls == Integer.TYPE || cls.equals(Integer.class) ) return ARG_INTEGER; if( cls == Short.TYPE || cls.equals(Short.class) ) return ARG_SHORT; return ARG_SAME; } }