| /* |
| // Licensed to Julian Hyde under one or more contributor license |
| // agreements. See the NOTICE file distributed with this work for |
| // additional information regarding copyright ownership. |
| // |
| // Julian Hyde licenses this file to you under the Apache License, |
| // Version 2.0 (the "License"); you may not use this file except in |
| // compliance with the License. You may obtain a copy of the License at: |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| */ |
| package net.hydromatic.linq4j.expressions; |
| |
| import net.hydromatic.linq4j.Enumerator; |
| |
| import java.lang.reflect.*; |
| import java.util.*; |
| |
| /** |
| * Utilities for converting between {@link Expression}, {@link Type} and |
| * {@link Class}. |
| * |
| * @see Primitive |
| */ |
| public class Types { |
| /** |
| * Creates a type with generic parameters. |
| */ |
| public static Type of(Type type, Type... typeArguments) { |
| if (typeArguments.length == 0) { |
| return type; |
| } |
| return new ParameterizedTypeImpl(type, toList(typeArguments), null); |
| } |
| |
| /** |
| * Returns the element type of a {@link Collection}, {@link Iterable} |
| * (including {@link net.hydromatic.linq4j.Queryable Queryable} and |
| * {@link net.hydromatic.linq4j.Enumerable Enumerable}), {@link Iterator}, |
| * {@link Enumerator}, or an array. |
| * |
| * <p>Returns null if the type is not one of these.</p> |
| */ |
| public static Type getElementType(Type type) { |
| if (type instanceof ArrayType) { |
| return ((ArrayType) type).getComponentType(); |
| } |
| if (type instanceof GenericArrayType) { |
| return ((GenericArrayType) type).getGenericComponentType(); |
| } |
| Class clazz = toClass(type); |
| if (clazz.isArray()) { |
| return clazz.getComponentType(); |
| } |
| if (Collection.class.isAssignableFrom(clazz) |
| || Iterable.class.isAssignableFrom(clazz) |
| || Iterator.class.isAssignableFrom(clazz) |
| || Enumerator.class.isAssignableFrom(clazz)) { |
| if (type instanceof ParameterizedType) { |
| return ((ParameterizedType) type).getActualTypeArguments()[0]; |
| } |
| return Object.class; |
| } |
| return null; |
| } |
| |
| /** |
| * Returns a list backed by a copy of an array. The contents of the list |
| * will not change even if the contents of the array are subsequently |
| * modified. |
| */ |
| private static <T> List<T> toList(T[] ts) { |
| switch (ts.length) { |
| case 0: |
| return Collections.emptyList(); |
| case 1: |
| return Collections.singletonList(ts[0]); |
| default: |
| return Arrays.asList(ts.clone()); |
| } |
| } |
| |
| static Field getField(String fieldName, Class clazz) { |
| try { |
| return clazz.getField(fieldName); |
| } catch (NoSuchFieldException e) { |
| throw new RuntimeException( |
| "Unknown field '" + fieldName + "' in class " + clazz, e); |
| } |
| } |
| |
| static PseudoField getField(String fieldName, Type type) { |
| if (type instanceof RecordType) { |
| return getRecordField(fieldName, (RecordType) type); |
| } else if (type instanceof Class && ((Class) type).isArray()) { |
| return getSystemField(fieldName, (Class) type); |
| } else { |
| return field(getField(fieldName, toClass(type))); |
| } |
| } |
| |
| private static RecordField getRecordField(String fieldName, RecordType type) { |
| for (RecordField field : type.getRecordFields()) { |
| if (field.getName().equals(fieldName)) { |
| return field; |
| } |
| } |
| throw new RuntimeException( |
| "Unknown field '" + fieldName + "' in type " + type); |
| } |
| |
| private static RecordField getSystemField(final String fieldName, |
| final Class clazz) { |
| // The "length" field of an array does not appear in Class.getFields(). |
| return new RecordField() { |
| public boolean nullable() { |
| return false; |
| } |
| |
| public String getName() { |
| return fieldName; |
| } |
| |
| public Type getType() { |
| return int.class; |
| } |
| |
| public int getModifiers() { |
| return 0; |
| } |
| |
| public Object get(Object o) throws IllegalAccessException { |
| return Array.getLength(o); |
| } |
| |
| public Type getDeclaringClass() { |
| return clazz; |
| } |
| }; |
| } |
| |
| public static Class toClass(Type type) { |
| if (type instanceof Class) { |
| return (Class) type; |
| } |
| if (type instanceof ParameterizedType) { |
| return toClass(((ParameterizedType) type).getRawType()); |
| } |
| if (type instanceof TypeVariable) { |
| TypeVariable typeVariable = (TypeVariable) type; |
| return toClass(typeVariable.getBounds()[0]); |
| } |
| throw new RuntimeException("unsupported type " + type); // TODO: |
| } |
| |
| static Class[] toClassArray(Collection<Type> types) { |
| List<Class> classes = new ArrayList<Class>(); |
| for (Type type : types) { |
| classes.add(toClass(type)); |
| } |
| return classes.toArray(new Class[classes.size()]); |
| } |
| |
| static Class[] toClassArray(Iterable<? extends Expression> arguments) { |
| List<Class> classes = new ArrayList<Class>(); |
| for (Expression argument : arguments) { |
| classes.add(toClass(argument.getType())); |
| } |
| return classes.toArray(new Class[classes.size()]); |
| } |
| |
| /** |
| * Returns the component type of an array. |
| */ |
| public static Type getComponentType(Type type) { |
| if (type instanceof Class) { |
| return ((Class) type).getComponentType(); |
| } |
| if (type instanceof ArrayType) { |
| return ((ArrayType) type).getComponentType(); |
| } |
| if (type instanceof GenericArrayType) { |
| return ((GenericArrayType) type).getGenericComponentType(); |
| } |
| if (type instanceof ParameterizedType) { |
| return getComponentType(((ParameterizedType) type).getRawType()); |
| } |
| if (type instanceof TypeVariable) { |
| TypeVariable typeVariable = (TypeVariable) type; |
| return getComponentType(typeVariable.getBounds()[0]); |
| } |
| return null; // not an array type |
| } |
| |
| /** |
| * Boxes a type, if it is primitive, and returns the type name. |
| * The type is abbreviated if it is in the "java.lang" package. |
| * |
| * <p>For example, |
| * boxClassName(int) returns "Integer"; |
| * boxClassName(List<String>) returns "List<String>"</p> |
| * |
| * @param type Type |
| * |
| * @return Class name |
| */ |
| static String boxClassName(Type type) { |
| if (!(type instanceof Class)) { |
| return type.toString(); |
| } |
| Primitive primitive = Primitive.of(type); |
| if (primitive != null) { |
| return primitive.boxClass.getSimpleName(); |
| } else { |
| return className(type); |
| } |
| } |
| |
| public static Type box(Type type) { |
| Primitive primitive = Primitive.of(type); |
| if (primitive != null) { |
| return primitive.boxClass; |
| } else { |
| return type; |
| } |
| } |
| |
| public static Type unbox(Type type) { |
| Primitive primitive = Primitive.ofBox(type); |
| if (primitive != null) { |
| return primitive.primitiveClass; |
| } else { |
| return type; |
| } |
| } |
| |
| static String className(Type type) { |
| if (type instanceof ArrayType) { |
| return className(((ArrayType) type).getComponentType()) + "[]"; |
| } |
| if (!(type instanceof Class)) { |
| return type.toString(); |
| } |
| Class clazz = (Class) type; |
| if (clazz.isArray()) { |
| return className(clazz.getComponentType()) + "[]"; |
| } |
| String className = clazz.getName(); |
| if (clazz.getPackage() == Package.getPackage("java.lang") |
| && !clazz.isPrimitive()) { |
| return className.substring("java.lang.".length()); |
| } |
| return className.replace('$', '.'); |
| } |
| |
| public static boolean isAssignableFrom(Type type0, Type type) { |
| return toClass(type0).isAssignableFrom(toClass(type)); |
| } |
| |
| public static boolean isArray(Type type) { |
| return toClass(type).isArray(); |
| } |
| |
| public static Field nthField(int ordinal, Class clazz) { |
| return clazz.getFields()[ordinal]; |
| } |
| |
| public static PseudoField nthField(int ordinal, Type clazz) { |
| if (clazz instanceof RecordType) { |
| RecordType recordType = (RecordType) clazz; |
| return recordType.getRecordFields().get(ordinal); |
| } |
| return field(toClass(clazz).getFields()[ordinal]); |
| } |
| |
| static boolean allAssignable(boolean varArgs, Class[] parameterTypes, |
| Class[] argumentTypes) { |
| if (varArgs) { |
| if (argumentTypes.length < parameterTypes.length - 1) { |
| return false; |
| } |
| } else { |
| if (parameterTypes.length != argumentTypes.length) { |
| return false; |
| } |
| } |
| for (int i = 0; i < argumentTypes.length; i++) { |
| Class |
| parameterType = |
| !varArgs || i < parameterTypes.length - 1 |
| ? parameterTypes[i] |
| : Object.class; |
| if (!assignableFrom(parameterType, argumentTypes[i])) { |
| return false; |
| } |
| } |
| return true; |
| } |
| |
| /** |
| * Returns whether a parameter is assignable from an argument by virtue |
| * of (a) sub-classing (e.g. Writer is assignable from PrintWriter) and (b) |
| * up-casting (e.g. int is assignable from short). |
| * |
| * @param parameter Parameter type |
| * @param argument Argument type |
| * |
| * @return Whether parameter can be assigned from argument |
| */ |
| private static boolean assignableFrom(Class parameter, Class argument) { |
| return parameter.isAssignableFrom(argument) |
| || parameter.isPrimitive() |
| && argument.isPrimitive() |
| && Primitive.of(parameter).assignableFrom(Primitive.of(argument)); |
| } |
| |
| /** |
| * Finds a method of a given name that accepts a given set of arguments. |
| * Includes in its search inherited methods and methods with wider argument |
| * types. |
| * |
| * @param clazz Class against which method is invoked |
| * @param methodName Name of method |
| * @param argumentTypes Types of arguments |
| * |
| * @return A method with the given name that matches the arguments given |
| * @throws RuntimeException if method not found |
| */ |
| public static Method lookupMethod(Class clazz, String methodName, |
| Class... argumentTypes) { |
| try { |
| return clazz.getMethod(methodName, argumentTypes); |
| } catch (NoSuchMethodException e) { |
| for (Method method : clazz.getMethods()) { |
| if (method.getName().equals(methodName) && allAssignable( |
| method.isVarArgs(), method.getParameterTypes(), argumentTypes)) { |
| return method; |
| } |
| } |
| throw new RuntimeException( |
| "while resolving method '" + methodName + "' in class " + clazz, e); |
| } |
| } |
| |
| /** |
| * Finds a constructor of a given class that accepts a given set of |
| * arguments. Includes in its search methods with wider argument types. |
| * |
| * @param type Class against which method is invoked |
| * @param argumentTypes Types of arguments |
| * |
| * @return A method with the given name that matches the arguments given |
| * @throws RuntimeException if method not found |
| */ |
| public static Constructor lookupConstructor(Type type, |
| Class... argumentTypes) { |
| final Class clazz = toClass(type); |
| Constructor[] constructors = clazz.getDeclaredConstructors(); |
| for (Constructor constructor : constructors) { |
| if (allAssignable(constructor.isVarArgs(), |
| constructor.getParameterTypes(), argumentTypes)) { |
| return constructor; |
| } |
| } |
| if (constructors.length == 0 && argumentTypes.length == 0) { |
| Constructor[] constructors1 = clazz.getConstructors(); |
| try { |
| return clazz.getConstructor(); |
| } catch (NoSuchMethodException e) { |
| // ignore |
| } |
| } |
| throw new RuntimeException( |
| "while resolving constructor in class " + type + " with types " + Arrays |
| .toString(argumentTypes)); |
| } |
| |
| public static void discard(Object o) { |
| if (false) { |
| discard(o); |
| } |
| } |
| |
| /** |
| * Returns the most restrictive type that is assignable from all given |
| * types. |
| */ |
| static Type gcd(Type... types) { |
| // TODO: improve this |
| if (types.length == 0) { |
| return Object.class; |
| } |
| Type best = types[0]; |
| Primitive bestPrimitive = Primitive.of(best); |
| if (bestPrimitive != null) { |
| for (int i = 1; i < types.length; i++) { |
| final Primitive primitive = Primitive.of(types[i]); |
| if (primitive == null) { |
| return Object.class; |
| } |
| if (primitive.assignableFrom(bestPrimitive)) { |
| bestPrimitive = primitive; |
| } else if (bestPrimitive.assignableFrom(primitive)) { |
| // ok |
| } else if (bestPrimitive == Primitive.CHAR |
| || bestPrimitive == Primitive.BYTE) { |
| // 'char' and 'byte' are problematic, because they don't |
| // assign to each other. 'char' can't even assign to |
| // 'short'. Before we give up, try one last time with 'int'. |
| bestPrimitive = Primitive.INT; |
| --i; |
| } else { |
| return Object.class; |
| } |
| } |
| return bestPrimitive.primitiveClass; |
| } else { |
| for (int i = 1; i < types.length; i++) { |
| if (types[i] != types[0]) { |
| return Object.class; |
| } |
| } |
| } |
| return types[0]; |
| } |
| |
| /** |
| * Wraps an expression in a cast if it is not already of the desired type, |
| * or cannot be implicitly converted to it. |
| * |
| * @param returnType Desired type |
| * @param expression Expression |
| * |
| * @return Expression of desired type |
| */ |
| public static Expression castIfNecessary(Type returnType, |
| Expression expression) { |
| final Type type = expression.getType(); |
| if (Types.isAssignableFrom(returnType, type)) { |
| return expression; |
| } |
| if (returnType instanceof Class |
| && Number.class.isAssignableFrom((Class) returnType) |
| && type instanceof Class |
| && Number.class.isAssignableFrom((Class) type)) { |
| // E.g. |
| // Integer foo(BigDecimal o) { |
| // return o.intValue(); |
| // } |
| return Expressions.unbox(expression, Primitive.ofBox(returnType)); |
| } |
| if (Primitive.is(returnType) && !Primitive.is(type)) { |
| // E.g. |
| // int foo(Object o) { |
| // return ((Integer) o).intValue(); |
| // } |
| return Expressions.unbox( |
| Expressions.convert_(expression, Types.box(returnType)), |
| Primitive.of(returnType)); |
| } |
| if (!Primitive.is(returnType) && Primitive.is(type)) { |
| // E.g. |
| // Short foo(Object o) { |
| // return (short) (int) o; |
| // } |
| return Expressions.convert_(expression, |
| Types.unbox(returnType)); |
| } |
| return Expressions.convert_(expression, returnType); |
| } |
| |
| public static PseudoField field(final Field field) { |
| return new PseudoField() { |
| public String getName() { |
| return field.getName(); |
| } |
| |
| public Type getType() { |
| return field.getType(); |
| } |
| |
| public int getModifiers() { |
| return field.getModifiers(); |
| } |
| |
| public Object get(Object o) throws IllegalAccessException { |
| return field.get(o); |
| } |
| |
| public Class<?> getDeclaringClass() { |
| return field.getDeclaringClass(); |
| } |
| }; |
| } |
| |
| static Class arrayClass(Type type) { |
| // REVIEW: Is there a way to do this without creating an instance? We |
| // just need the inverse of Class.getComponentType(). |
| return Array.newInstance(toClass(type), 0).getClass(); |
| } |
| |
| static Type arrayType(Type type) { |
| if (type instanceof Class) { |
| Class clazz = (Class) type; |
| |
| // REVIEW: Is there a way to do this without creating an instance? |
| // We just need the inverse of Class.getComponentType(). |
| return Array.newInstance(clazz, 0).getClass(); |
| } |
| return new ArrayType(type); |
| } |
| |
| public static Type stripGenerics(Type type) { |
| if (type instanceof GenericArrayType) { |
| final Type componentType = |
| ((GenericArrayType) type).getGenericComponentType(); |
| return new ArrayType(stripGenerics(componentType)); |
| } else if (type instanceof ParameterizedType) { |
| return ((ParameterizedType) type).getRawType(); |
| } else { |
| return type; |
| } |
| } |
| |
| static class ParameterizedTypeImpl implements ParameterizedType { |
| private final Type rawType; |
| private final List<Type> typeArguments; |
| private final Type ownerType; |
| |
| ParameterizedTypeImpl(Type rawType, List<Type> typeArguments, |
| Type ownerType) { |
| super(); |
| this.rawType = rawType; |
| this.typeArguments = typeArguments; |
| this.ownerType = ownerType; |
| assert rawType != null; |
| for (Type typeArgument : typeArguments) { |
| assert typeArgument != null; |
| } |
| } |
| |
| @Override |
| public String toString() { |
| final StringBuilder buf = new StringBuilder(); |
| buf.append(className(rawType)); |
| buf.append("<"); |
| int i = 0; |
| for (Type typeArgument : typeArguments) { |
| if (i++ > 0) { |
| buf.append(", "); |
| } |
| buf.append(className(typeArgument)); |
| } |
| buf.append(">"); |
| return buf.toString(); |
| } |
| |
| public Type[] getActualTypeArguments() { |
| return typeArguments.toArray(new Type[typeArguments.size()]); |
| } |
| |
| public Type getRawType() { |
| return rawType; |
| } |
| |
| public Type getOwnerType() { |
| return ownerType; |
| } |
| } |
| |
| /** |
| * Base class for record-like types that do not mapped to (currently |
| * loaded) Java {@link Class} objects. Gives the opportunity to generate |
| * code that references temporary types, then generate classes for those |
| * types along with the code that uses them. |
| */ |
| public interface RecordType extends Type { |
| List<RecordField> getRecordFields(); |
| |
| String getName(); |
| } |
| |
| /** |
| * Field that belongs to a record. |
| */ |
| public interface RecordField extends PseudoField { |
| boolean nullable(); |
| } |
| |
| /** |
| * Array type. |
| */ |
| public static class ArrayType implements Type { |
| private final Type componentType; |
| |
| public ArrayType(Type componentType) { |
| this.componentType = componentType; |
| } |
| |
| public Type getComponentType() { |
| return componentType; |
| } |
| } |
| } |
| |
| // End Types.java |