All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.strobel.reflection.Resolver Maven / Gradle / Ivy

There is a newer version: 0.6.0
Show newest version
/*
 * Resolver.java
 *
 * Copyright (c) 2012 Mike Strobel
 *
 * This source code is subject to terms and conditions of the Apache License, Version 2.0.
 * A copy of the license can be found in the License.html file at the root of this distribution.
 * By using this source code in any fashion, you are agreeing to be bound by the terms of the
 * Apache License, Version 2.0.
 *
 * You must not remove this notice, or any other, from this software.
 */

package com.strobel.reflection;

import com.strobel.core.ArrayUtilities;
import com.strobel.core.Comparer;
import com.strobel.core.VerifyArgument;
import com.strobel.util.ContractUtils;

import java.lang.reflect.Constructor;
import java.lang.reflect.Field;
import java.lang.reflect.GenericArrayType;
import java.lang.reflect.GenericDeclaration;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.TypeVariable;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Stack;
import java.util.concurrent.atomic.AtomicIntegerFieldUpdater;

/**
 * @author Mike Strobel
 */
@SuppressWarnings("unchecked")
final class Resolver {
    final static byte FLAG_RESOLVE_METHODS      = 0x01;
    final static byte FLAG_RESOLVE_FIELDS       = 0x02;
    final static byte FLAG_RESOLVE_CONSTRUCTORS = 0x04;
    final static byte FLAG_RESOLVE_NESTED_TYPES = 0x08;

    void resolveMembers(final ReflectedType type, final int flags) {
        final Frame frame = new Frame(type, null);

        this.visit(type, frame, flags);
    }

    public final class Frame {
        private final ReflectedType                  _type;
        private final java.lang.reflect.Type            _typeElement;
        private final Frame                             _previous;
        private final Map _elementTypeMap;
        private final Stack            _methods;

        private final ArrayList> _typeArguments = new ArrayList<>();

        public Frame(final java.lang.reflect.Type typeElement, final Frame previous) {
            _typeElement = VerifyArgument.notNull(typeElement, "typeElement");
            _previous = previous;
            _elementTypeMap = previous != null ? previous._elementTypeMap : new HashMap();
            _methods = previous != null ? previous._methods : new Stack();
            _type = new ReflectedType<>((Class) typeElement);
            _elementTypeMap.put(typeElement, _type);
        }

        Frame(final ReflectedType type, final Frame previous) {
            _type = VerifyArgument.notNull(type, "type");
            _typeElement = type;
            _previous = previous;
            _elementTypeMap = previous != null ? previous._elementTypeMap : new HashMap();
            _methods = previous != null ? previous._methods : new Stack();
            _elementTypeMap.put(_typeElement, _type);

            final Class classType = type.getErasedClass();
            final Class enclosingClass = classType.getEnclosingClass();

            if (enclosingClass != null && enclosingClass != classType) {
                final Frame ownerFrame = findFrame(enclosingClass);

                if (ownerFrame != null) {
                    if (ownerFrame._type.findNestedType(_type.getErasedClass()) == null) {
                        ownerFrame._type.addNestedType(_type);
                    }
                }
            }
        }

        public Type getResult() {
            return _type;
        }

        void pushMethod(final ReflectedMethod method) {
            _methods.push(method);
        }

        ReflectedMethod popMethod() {
            return _methods.pop();
        }

        ReflectedMethod currentMethod() {
            return _methods.peek();
        }

        ReflectedType getCurrentClass() {
            return _type;
        }

        java.lang.reflect.Type getCurrentType() {
            return _typeElement;
        }

        List> getTypeArguments() {
            return _typeArguments;
        }

        Type findType(final java.lang.reflect.Type e) {
            Frame currentFrame = this;

            while (currentFrame != null) {
                Type result = currentFrame._elementTypeMap.get(e);

                if (result != null) {
                    return result;
                }

                if (e instanceof java.lang.reflect.TypeVariable) {
                    final TypeVariable typeVariable = (TypeVariable) e;

                    result = currentFrame._type.findGenericParameter(typeVariable);

                    if (result != null) {
                        return result;
                    }

                    if (!_typeArguments.isEmpty()) {
                        for (final Type typeArgument : _typeArguments) {
                            if (!(typeArgument instanceof GenericParameter)) {
                                continue;
                            }
                            if (((GenericParameter) typeArgument).getRawTypeVariable() == typeVariable) {
                                return typeArgument;
                            }
                        }
                    }
                }

                currentFrame = currentFrame._previous;
            }

            return Type.tryFind(e);
        }

        Type resolveType(final java.lang.reflect.Type t) {
            final Type result = resolveTypeCore(t);

/*
            if (result != null && result.isGenericType()) {
                if (t instanceof Class &&
                    ArrayUtilities.isNullOrEmpty(((Class)t).getTypeParameters())) {
                    return result.getErasedType();
                }
            }
*/

            if (result != null) {
                _elementTypeMap.put(t, result);
                Type.CACHE.add(result);
            }

            return result;
        }

        private Type resolveTypeCore(final java.lang.reflect.Type t) {
            final Type fromMap = findType(t);

            if (fromMap != null) {
                return fromMap;
            }

            if (t instanceof Class) {
                final Class classType = (Class) t;

                if (classType.isPrimitive()) {
                    return Type.of(classType);
                }

                if (classType.isArray()) {
                    final Type componentType = resolveType(classType.getComponentType());

                    if (componentType != null) {
                        return componentType.makeArrayType();
                    }

                    return null;
                }

                final Type cachedType = Type.CACHE.find((Class) t);

                if (cachedType != null) {
                    return cachedType;
                }

                return visit(classType, this);
            }

            if (t instanceof GenericArrayType) {
                final java.lang.reflect.Type genericComponentType = ((GenericArrayType) t).getGenericComponentType();
                final Type componentType = resolveType(genericComponentType);

                if (componentType != null) {
                    return componentType.makeArrayType();
                }

                return null;
            }

            if (t instanceof java.lang.reflect.WildcardType) {
                final java.lang.reflect.WildcardType w = (java.lang.reflect.WildcardType) t;
                final java.lang.reflect.Type[] upperBounds = w.getUpperBounds();
                final java.lang.reflect.Type[] lowerBounds = w.getLowerBounds();

                List> resolvedUpperBounds = null;
                List> resolvedLowerBounds = null;

                for (final java.lang.reflect.Type lowerBound : lowerBounds) {
                    final Type resolvedLowerBound = resolveType(lowerBound);

                    if (resolvedLowerBound == null) {
                        continue;
                    }

                    if (resolvedLowerBounds == null) {
                        resolvedLowerBounds = new ArrayList<>();
                    }

                    resolvedLowerBounds.add(resolvedLowerBound);
                }

                if (resolvedLowerBounds != null && resolvedLowerBounds.size() != lowerBounds.length) {
                    return null;
                }

                for (final java.lang.reflect.Type upperBound : upperBounds) {
                    final Type resolvedUpperBound = resolveType(upperBound);

                    if (resolvedUpperBound == null) {
                        continue;
                    }

                    if (resolvedUpperBounds == null) {
                        resolvedUpperBounds = new ArrayList<>();
                    }

                    resolvedUpperBounds.add(resolvedUpperBound);
                }

                if (resolvedUpperBounds != null && resolvedUpperBounds.size() != upperBounds.length) {
                    return null;
                }

                Type upperBound = Types.Object;
                Type lowerBound = Type.Bottom;

                if (resolvedUpperBounds != null) {
                    if (resolvedUpperBounds.size() == 1) {
                        upperBound = resolvedUpperBounds.get(0);
                    }
                    else {
                        upperBound = Type.makeCompoundType(Type.list(resolvedUpperBounds));
                    }
                }

                if (resolvedLowerBounds != null) {
                    if (resolvedLowerBounds.size() == 1) {
                        lowerBound = resolvedLowerBounds.get(0);
                    }
                    else {
                        lowerBound = Type.makeCompoundType(Type.list(resolvedLowerBounds));
                    }
                }

                return new WildcardType(upperBound, lowerBound);
            }

            // Now redundant to findType()...
            if (t instanceof java.lang.reflect.TypeVariable) {
                final java.lang.reflect.TypeVariable typeVariable = (java.lang.reflect.TypeVariable) t;

                Frame currentFrame = this;

                while (currentFrame != null) {
                    final GenericParameter genericParameter = currentFrame._type.findGenericParameter(typeVariable);

                    if (genericParameter != null) {
                        return genericParameter;
                    }

                    currentFrame = currentFrame._previous;
                }
            }

            if (t instanceof ParameterizedType) {
                final ParameterizedType type = (ParameterizedType) t;

                final Type rawType = resolveType(type.getRawType());
                final java.lang.reflect.Type[] typeArguments = type.getActualTypeArguments();

                if (ArrayUtilities.isNullOrEmpty(typeArguments) || !rawType.isGenericType()) {
                    return rawType;
                }

                final Type[] resolvedTypeArguments = new Type[typeArguments.length];

                for (int i = 0, n = typeArguments.length; i < n; i++) {
                    final java.lang.reflect.Type typeArg = typeArguments[i];

                    if (typeArg instanceof java.lang.reflect.TypeVariable) {
                        final java.lang.reflect.TypeVariable typeVariable = (java.lang.reflect.TypeVariable) typeArg;
                        final GenericDeclaration genericDeclaration = typeVariable.getGenericDeclaration();
                        final Type existingTypeArgument = findType(typeVariable);

                        if (existingTypeArgument != null) {
                            resolvedTypeArguments[i] = existingTypeArgument;
                            continue;
                        }

                        if (genericDeclaration instanceof Method) {
                            final ReflectedType declaringType = (ReflectedType) resolveType(((Method) genericDeclaration).getDeclaringClass());
                            final Method method = (Method) genericDeclaration;
                            final int position = ArrayUtilities.indexOf(method.getTypeParameters(), typeVariable);
                            final ReflectedMethod declaredMethod = declaringType.findMethod(method);

                            resolvedTypeArguments[i] = declaredMethod.getTypeArguments().get(position);
                        }
                        else {
                            final Class declaringClass = (Class) genericDeclaration;
                            final Type declaringType = resolveType(declaringClass);
                            final int position = ArrayUtilities.indexOf(declaringClass.getTypeParameters(), typeVariable);

                            if (declaringType instanceof ReflectedType) {
                                resolvedTypeArguments[i] = ((ReflectedType) declaringType).getGenericParameters().get(position);
                            }
                            else {
                                resolvedTypeArguments[i] = declaringType.getTypeArguments().get(position);
                            }
                        }
                    }
                    else {
                        if (typeArg == null) {
                            resolvedTypeArguments[i] = rawType.getGenericTypeParameters().get(i);
                        }
                        else {
                            resolvedTypeArguments[i] = resolveType(typeArg);
                        }
                    }
                }

                final Type fromCache = Type.CACHE.find(
                    Type.CACHE.key(
                        rawType,
                        Type.list(resolvedTypeArguments)
                    )
                );

                if (fromCache != null) {
                    return fromCache;
                }

                for (final Type resolvedTypeArgument : resolvedTypeArguments) {
                    if (!resolvedTypeArgument.isGenericParameter() || resolvedTypeArgument.getDeclaringType() != rawType) {
                        return Type.CACHE.getGenericType(rawType, Type.list(resolvedTypeArguments));
                    }
                }

                return rawType;
            }

            return null;
        }

        Frame findFrame(final java.lang.reflect.Type e) {
            Frame current = this;

            while (current != null) {
                if (current._typeElement.equals(e)) {
                    return current;
                }

                current = current._previous;
            }

            return null;
        }

        void addTypeArgument(final GenericParameter genericParameter) {
            _typeArguments.add(genericParameter);
        }
    }

    public Type resolve(final java.lang.reflect.Type type) {
        return resolveExisting(null, type, true);
    }

    private Type visit(final java.lang.reflect.Type type, final Frame frame) {
        if (type instanceof ParameterizedType) {
            return visitParameterizedType((ParameterizedType) type, frame);
        }
        else if (type instanceof java.lang.reflect.WildcardType) {
            return visitWildcardType((java.lang.reflect.WildcardType) type, frame);
        }
        else if (type instanceof GenericArrayType) {
            return visitGenericArrayType((GenericArrayType) type, frame);
        }
        else if (type instanceof java.lang.reflect.TypeVariable) {
            return visitTypeVariable((java.lang.reflect.TypeVariable) type, frame);
        }
        else {
            return visitClass((Class) type, frame);
        }
    }

    @SuppressWarnings("ConstantConditions")
    private Type visitClass(final Class c, final Frame frame) {
        final Method declaringMethod;
        final Constructor declaringConstructor;
        final Class enclosingClass = c.getEnclosingClass();

        if (enclosingClass != null) {
            if (c.isLocalClass()) {
                declaringMethod = c.getEnclosingMethod();
                declaringConstructor = c.getEnclosingConstructor();
            }
            else {
                if (c.isAnonymousClass()) {
                    return null;
                }
                declaringMethod = null;
                declaringConstructor = null;
            }
        }
        else {
            declaringMethod = null;
            declaringConstructor = null;
        }

        final Frame currentFrame = new Frame(c, frame);
        final ReflectedType currentType = currentFrame.getCurrentClass();

        if (enclosingClass != null) {
            final Type declaringType = currentFrame.resolveType(enclosingClass);

            currentType.setDeclaringType(declaringType);

            if (declaringMethod != null) {
                for (final MethodInfo method : declaringType.getMethods(BindingFlags.AllDeclared)) {
                    if (declaringMethod.equals(method.getRawMethod())) {
                        currentType.setDeclaringMethod(method);
                        break;
                    }
                }
            }
            else if (declaringConstructor != null) {
                for (final ConstructorInfo constructor : declaringType.getConstructors(BindingFlags.AllDeclared)) {
                    if (declaringConstructor.equals(constructor.getRawConstructor())) {
                        currentType.setDeclaringMethod(constructor);
                        break;
                    }
                }
            }

            // Redundant: Frame constructor adds nested type to its declaring type.
            if (declaringType instanceof ReflectedType) {
                final ReflectedType owner = (ReflectedType) declaringType;
                final ReflectedType nestedType = owner.findNestedType(c);

                if (nestedType != null) {
                    return nestedType;
                }

                if (nestedType == null) {
                    owner.addNestedType(currentType);
                }
            }
        }

        final TypeVariable[] typeParameters = c.getTypeParameters();

        if (!ArrayUtilities.isNullOrEmpty(typeParameters)) {
            for (int i = 0, n = typeParameters.length; i < n; i++) {
                final TypeVariable typeVariable = typeParameters[i];

                currentType.addGenericParameter(
                    new GenericParameter(
                        typeVariable.getName(),
                        currentType,
                        null,
                        i
                    )
                );
            }
        }

        final Type baseType;

        java.lang.reflect.Type baseClass;

        if (!c.isInterface()) {
            baseClass = c.getGenericSuperclass();

            if (baseClass == null) {
                baseClass = c.getSuperclass();
            }

            if (baseClass != null && baseClass != c) {
                baseType = currentFrame.resolveType(baseClass);

                if (baseType != null) {
                    currentType.setBaseType(baseType);
                }
                else {
                    return null;
                }
            }
        }

        List> interfaceList = null;

        java.lang.reflect.Type[] interfaces = c.getGenericInterfaces();

        if (ArrayUtilities.isNullOrEmpty(interfaces)) {
            interfaces = c.getInterfaces();
        }

        if (!ArrayUtilities.isNullOrEmpty(interfaces)) {
            for (final java.lang.reflect.Type t : interfaces) {
                final Type interfaceType = currentFrame.resolveType(t);

                if (interfaceType == null) {
                    return null;
                }

                if (interfaceList == null) {
                    interfaceList = new ArrayList<>();
                }

                interfaceList.add(interfaceType);
            }
        }

        if (interfaceList != null) {
            currentType.setInterfaces(Type.list(interfaceList));
        }

/*
        for (final Class ee : c.getDeclaredClasses()) {
            this.visit(ee, currentFrame);
        }
*/

        for (final GenericParameter genericParameter : currentType.getGenericParameters()) {
            final java.lang.reflect.Type[] bounds = genericParameter.getRawTypeVariable().getBounds();

            if (ArrayUtilities.isNullOrEmpty(bounds)) {
                continue;
            }

            final Type boundType;

            if (bounds.length == 1) {
                boundType = currentFrame.resolveType(bounds[0]);
            }
            else {
                final Type[] resolvedBounds = new Type[bounds.length];

                for (int i = 0, n = bounds.length; i < n; i++) {
                    resolvedBounds[i] = currentFrame.resolveType(bounds[i]);
                }

                boundType = Type.makeCompoundType(Type.list(resolvedBounds));
            }

            genericParameter.setUpperBound(boundType);
        }

        currentType.complete();

        if (!java.lang.reflect.Modifier.isPrivate(currentType.getModifiers())) {
            Type.CACHE.add(currentType);
        }

        return currentType;
    }

    private Type visitTypeVariable(final java.lang.reflect.TypeVariable type, final Frame frame) {
        return frame.findType(type);
    }

    private Type visitGenericArrayType(final GenericArrayType type, final Frame frame) {
        final Type elementType = frame.resolveType(type.getGenericComponentType());

        if (elementType != null) {
            return elementType.makeArrayType();
        }

        return null;
    }

    private Type visitWildcardType(final java.lang.reflect.WildcardType w, final Frame frame) {
        final java.lang.reflect.Type[] upperBounds = w.getUpperBounds();
        final java.lang.reflect.Type[] lowerBounds = w.getLowerBounds();

        List> resolvedUpperBounds = null;
        List> resolvedLowerBounds = null;

        for (final java.lang.reflect.Type lowerBound : lowerBounds) {
            final Type resolvedLowerBound = visit(lowerBound, frame);

            if (resolvedLowerBound == null) {
                continue;
            }

            if (resolvedLowerBounds == null) {
                resolvedLowerBounds = new ArrayList<>();
            }

            resolvedLowerBounds.add(resolvedLowerBound);
        }

        if (resolvedLowerBounds != null && resolvedLowerBounds.size() != lowerBounds.length) {
            return null;
        }

        for (final java.lang.reflect.Type upperBound : upperBounds) {
            final Type resolvedUpperBound = visit(upperBound, frame);

            if (resolvedUpperBound == null) {
                continue;
            }

            if (resolvedUpperBounds == null) {
                resolvedUpperBounds = new ArrayList<>();
            }

            resolvedUpperBounds.add(resolvedUpperBound);
        }

        if (resolvedUpperBounds != null && resolvedUpperBounds.size() != upperBounds.length) {
            return null;
        }

        Type upperBound = Types.Object;
        Type lowerBound = Type.Bottom;

        if (resolvedUpperBounds != null) {
            if (resolvedUpperBounds.size() == 1) {
                upperBound = resolvedUpperBounds.get(0);
            }
            else {
                upperBound = Type.makeCompoundType(Type.list(resolvedUpperBounds));
            }
        }

        if (resolvedLowerBounds != null) {
            if (resolvedLowerBounds.size() == 1) {
                lowerBound = resolvedLowerBounds.get(0);
            }
            else {
                lowerBound = Type.makeCompoundType(Type.list(resolvedLowerBounds));
            }
        }

        return new WildcardType(upperBound, lowerBound);
    }

    private Type visitParameterizedType(final ParameterizedType type, final Frame frame) {
        final Type rawType = frame.resolveType(type.getRawType());

        if (rawType == null) {
            return null;
        }

        final java.lang.reflect.Type[] typeArguments = type.getActualTypeArguments();
        final Type[] resolvedTypeArguments = new Type[typeArguments.length];

        for (int i = 0, n = typeArguments.length; i < n; i++) {
            resolvedTypeArguments[i] = frame.resolveType(typeArguments[i]);

            if (resolvedTypeArguments[i] == null) {
                return null;
            }
        }

        return rawType.makeGenericType(resolvedTypeArguments);
    }

    private Type resolveExisting(final Frame frame, final java.lang.reflect.Type type, final boolean resolve) {
        final Type result;

        Type fromCacheOrFrame = Type.tryFind(type);

        if (fromCacheOrFrame != null) {
            result = fromCacheOrFrame;
        }
        else {
            if (frame != null) {
                fromCacheOrFrame = frame.resolveType(type);
            }

            if (!resolve) {
                return fromCacheOrFrame;
            }

            if (fromCacheOrFrame != null) {
                result = fromCacheOrFrame;
            }
            else if (type instanceof Class) {
                result = visitClass((Class) type, frame);
            }
            else {
                result = visit(type, frame);
            }
        }

        return result;
    }

    private Type visit(final ReflectedType type, final Frame frame, final int flags) {
        final Class erasedClass = type.getErasedClass();

        if ((flags & FLAG_RESOLVE_NESTED_TYPES) == FLAG_RESOLVE_NESTED_TYPES) {
            for (final Class nestedClass : erasedClass.getDeclaredClasses()) {
                if (type.findNestedType(nestedClass) == null) {
                    visit(nestedClass, frame);
                }
            }
        }

        if ((flags & FLAG_RESOLVE_FIELDS) == FLAG_RESOLVE_FIELDS) {
            for (final Field field : erasedClass.getDeclaredFields()) {
                if (type.findField(field) == null) {
                    visitField(field, frame);
                }
            }
        }

        if ((flags & FLAG_RESOLVE_METHODS) == FLAG_RESOLVE_METHODS) {
            for (final Method method : erasedClass.getDeclaredMethods()) {
                if (type.findMethod(method) == null) {
                    visitMethod(method, frame);
                }
            }
        }

        if ((flags & FLAG_RESOLVE_CONSTRUCTORS) == FLAG_RESOLVE_CONSTRUCTORS) {
            for (final Constructor constructor : erasedClass.getDeclaredConstructors()) {
                if (type.findConstructor(constructor) == null) {
                    visitConstructor(constructor, frame);
                }
            }
        }

        return type;
    }

    public Type visitField(final Field field, final Frame frame) {
        final ReflectedType declaringType = frame.getCurrentClass();

        final ReflectedField reflectedField = new ReflectedField(
            declaringType,
            field,
            frame.resolveType(field.getType())
        );

        declaringType.addField(reflectedField);

        return declaringType;
    }

    @SuppressWarnings("ConstantConditions")
    private Type visitMethod(final Method m, final Frame frame) {
        java.lang.reflect.Type[] parameterTypes = m.getGenericParameterTypes();
        java.lang.reflect.Type[] thrownTypes = m.getGenericExceptionTypes();
        java.lang.reflect.Type returnType = m.getGenericReturnType();

        if (parameterTypes == null) {
            parameterTypes = m.getParameterTypes();
        }

        if (thrownTypes == null) {
            thrownTypes = m.getExceptionTypes();
        }

        if (returnType == null) {
            returnType = m.getReturnType();
        }

        final Type[] resolvedThrownTypes = new Type[thrownTypes.length];
        final ParameterInfo[] parameters = new ParameterInfo[parameterTypes.length];
        final TypeVariable[] typeParameters = m.getTypeParameters();
        final TypeBindings genericParameters;

        if (ArrayUtilities.isNullOrEmpty(typeParameters)) {
            genericParameters = TypeBindings.empty();
        }
        else {
            final Type[] resolvedTypeParameters = new Type[typeParameters.length];

            for (int i = 0; i < typeParameters.length; i++) {
                final TypeVariable typeVariable = typeParameters[i];
                final GenericParameter genericParameter = new GenericParameter(typeVariable.getName(), typeVariable, i);

                resolvedTypeParameters[i] = genericParameter;
                frame.addTypeArgument(genericParameter);
            }

            genericParameters = TypeBindings.createUnbound(Type.list(resolvedTypeParameters));
        }

        for (int i = 0; i < parameterTypes.length; i++) {
            final Type resolvedParameterType = frame.resolveType(parameterTypes[i]);

            if (resolvedParameterType == null) {
                return null;
            }

            parameters[i] = new ParameterInfo("p" + i, i, resolvedParameterType);
        }

        for (int i = 0; i < thrownTypes.length; i++) {
            final Type resolvedThrownType = frame.resolveType(thrownTypes[i]);

            if (resolvedThrownType == null) {
                return null;
            }

            resolvedThrownTypes[i] = resolvedThrownType;
        }

        final Type resolvedReturnType = frame.resolveType(returnType);

        if (resolvedReturnType == null) {
            return null;
        }

        final ReflectedType declaringType = frame.getCurrentClass();

        final ReflectedMethod method = new ReflectedMethod(
            null, declaringType,
            m,
            new ParameterList(parameters),
            resolvedReturnType,
            Type.list(resolvedThrownTypes),
            genericParameters
        );

        if (!genericParameters.isEmpty()) {
            for (final Type type : genericParameters.getGenericParameters()) {
                final GenericParameter genericParameter = (GenericParameter) type;

                genericParameter.setDeclaringMethod(method);

                final java.lang.reflect.Type[] bounds = genericParameter.getRawTypeVariable().getBounds();

                if (ArrayUtilities.isNullOrEmpty(bounds)) {
                    continue;
                }

                final Type boundType;

                if (bounds.length == 1) {
                    boundType = frame.resolveType(bounds[0]);
                }
                else {
                    final Type[] resolvedBounds = new Type[bounds.length];

                    for (int i = 0, n = bounds.length; i < n; i++) {
                        resolvedBounds[i] = frame.resolveType(bounds[i]);
                    }

                    boundType = Type.makeCompoundType(Type.list(resolvedBounds));
                }

                genericParameter.setUpperBound(boundType);

            }
        }

        declaringType.addMethod(method);

        return declaringType;
    }

    private Type visitConstructor(final Constructor c, final Frame frame) {
        java.lang.reflect.Type[] parameterTypes = c.getGenericParameterTypes();
        java.lang.reflect.Type[] thrownTypes = c.getGenericExceptionTypes();

        if (parameterTypes == null) {
            parameterTypes = c.getParameterTypes();
        }

        if (thrownTypes == null) {
            thrownTypes = c.getExceptionTypes();
        }

        final Type[] resolvedThrownTypes = new Type[thrownTypes.length];
        final ParameterInfo[] parameters = new ParameterInfo[parameterTypes.length];

        for (int i = 0; i < parameterTypes.length; i++) {
            final Type resolvedParameterType = frame.resolveType(parameterTypes[i]);

            if (resolvedParameterType == null) {
                return null;
            }

            parameters[i] = new ParameterInfo("p" + i, i, resolvedParameterType);
        }

        for (int i = 0; i < thrownTypes.length; i++) {
            final Type resolvedThrownType = frame.resolveType(thrownTypes[i]);

            if (resolvedThrownType == null) {
                return null;
            }

            resolvedThrownTypes[i] = resolvedThrownType;
        }

        final ReflectedType declaringType = frame.getCurrentClass();

        final ReflectedConstructor constructor = new ReflectedConstructor(
            declaringType,
            c,
            new ParameterList(parameters),
            Type.list(resolvedThrownTypes)
        );

        declaringType.addConstructor(constructor);

        return declaringType;
    }
}

@SuppressWarnings({ "unchecked", "UnusedDeclaration" })
class ReflectedType extends Type {
    private final static AtomicIntegerFieldUpdater FLAGS_UPDATER = AtomicIntegerFieldUpdater.newUpdater(
        ReflectedType.class,
        "_flags"
    );

    private final static byte FLAG_METHODS_RESOLVED      = 0x01;
    private final static byte FLAG_FIELDS_RESOLVED       = 0x02;
    private final static byte FLAG_CONSTRUCTORS_RESOLVED = 0x04;
    private final static byte FLAG_NESTED_TYPES_RESOLVED = 0x08;
    private final static byte FLAG_ALL_MEMBERS_RESOLVED  = 0x0F;
    private final static byte FLAG_RESOLVING_MEMBERS     = 0x10;

    private final    String          _name;
    private final    String          _simpleName;
    private final    Class        _rawClass;
    private          Type _baseType;
    private          TypeList        _interfaces;
    private volatile int             _flags;
    private          boolean         _completed;
    private          Type         _declaringType;
    private          MethodBase      _declaringMethod;

    private List>  _genericParameters = null;
    private List>     _nestedTypes       = null;
    private List      _methods           = null;
    private List       _fields            = null;
    private List _constructors      = null;

    private TypeBindings _typeBindings;

    ReflectedType(final Class rawClass) {
        _rawClass = VerifyArgument.notNull(rawClass, "rawClass");
        _name = rawClass.isLocalClass() ? rawClass.getName() : rawClass.getCanonicalName();
        _simpleName = rawClass.getSimpleName();
    }

    @Override
    public String getFullName() {
        return _name;
    }

    private boolean checkFlags(final int flags) {
        return (_flags & flags) == flags;
    }

    private void setFlags(final int flags) {
        while (true) {
            final int oldFlags = _flags;
            if (FLAGS_UPDATER.compareAndSet(this, oldFlags, oldFlags | flags)) {
                return;
            }
        }
    }

    private void clearFlags(final int flags) {
        while (true) {
            final int oldFlags = _flags;
            if (FLAGS_UPDATER.compareAndSet(this, oldFlags, oldFlags & ~flags)) {
                return;
            }
        }
    }

    void setBaseType(final Type baseType) {
        _baseType = (Type) baseType;
    }

    void setInterfaces(final TypeList interfaces) {
        _interfaces = VerifyArgument.notNull(interfaces, "interfaces");
    }

    List> getGenericParameters() {
        if (_genericParameters == null) {
            return Collections.emptyList();
        }
        return _genericParameters;
    }

    ReflectedField findField(final Field rawField) {
        if (_fields == null) {
            return null;
        }
        for (final ReflectedField field : _fields) {
            if (Comparer.equals(field.getRawField(), rawField)) {
                return field;
            }
        }
        return null;
    }

    ReflectedMethod findMethod(final Method rawMethod) {
        if (_methods == null) {
            return null;
        }
        for (final ReflectedMethod method : _methods) {
            if (Comparer.equals(method.getRawMethod(), rawMethod)) {
                return method;
            }
        }
        return null;
    }

    ReflectedConstructor findConstructor(final Constructor rawConstructor) {
        if (_constructors == null) {
            return null;
        }
        for (final ReflectedConstructor constructor : _constructors) {
            if (Comparer.equals(constructor.getRawConstructor(), rawConstructor)) {
                return constructor;
            }
        }
        return null;
    }

    ReflectedType findNestedType(final Class rawClass) {
        if (_nestedTypes == null) {
            return null;
        }
        for (final ReflectedType nestedType : _nestedTypes) {
            if (Comparer.equals(nestedType.getErasedClass(), rawClass)) {
                return nestedType;
            }
        }
        return null;
    }

    GenericParameter findGenericParameter(final java.lang.reflect.TypeVariable typeVariable) {
        return GenericParameterFinder.visit(this, typeVariable);
    }

    void setDeclaringType(final Type declaringType) {
        _declaringType = VerifyArgument.notNull(declaringType, "declaringType");
    }

    void setDeclaringMethod(final MethodBase declaringMethod) {
        _declaringMethod = VerifyArgument.notNull(declaringMethod, "declaringMethod");
    }

    void addGenericParameter(final GenericParameter genericParameter) {
        VerifyArgument.notNull(genericParameter, "typeParameter");
        _completed = false;
        if (_genericParameters == null) {
            _genericParameters = new ArrayList<>();
        }
        _genericParameters.add(genericParameter);
    }

    void addNestedType(final ReflectedType nestedType) {
        VerifyArgument.notNull(nestedType, "nestedType");

        if (_nestedTypes == null) {
            _nestedTypes = new ArrayList<>();
        }

        _nestedTypes.add(nestedType);

        if (!checkFlags(FLAG_RESOLVING_MEMBERS)) {
            clearFlags(FLAG_NESTED_TYPES_RESOLVED);
        }
    }

    void addMethod(final ReflectedMethod method) {
        VerifyArgument.notNull(method, "method");

        if (_methods == null) {
            _methods = new ArrayList<>();
        }

        _methods.add(method);

        if (!checkFlags(FLAG_RESOLVING_MEMBERS)) {
            clearFlags(FLAG_METHODS_RESOLVED);
        }
    }

    void addConstructor(final ReflectedConstructor constructor) {
        VerifyArgument.notNull(constructor, "constructor");

        if (_constructors == null) {
            _constructors = new ArrayList<>();
        }

        _constructors.add(constructor);

        if (!checkFlags(FLAG_RESOLVING_MEMBERS)) {
            clearFlags(FLAG_CONSTRUCTORS_RESOLVED);
        }
    }

    void addField(final ReflectedField field) {
        VerifyArgument.notNull(field, "field");

        if (_fields == null) {
            _fields = new ArrayList<>();
        }

        _fields.add(field);

        if (!checkFlags(FLAG_RESOLVING_MEMBERS)) {
            clearFlags(FLAG_FIELDS_RESOLVED);
        }
    }

    private void completeIfNecessary() {
        if (!_completed) {
            synchronized (CACHE_LOCK) {
                if (!_completed) {
                    complete();
                }
            }
        }
    }

    private void ensureMembersResolved() {
        while (true) {
            if (checkFlags(FLAG_ALL_MEMBERS_RESOLVED)) {
                return;
            }
            synchronized (CACHE_LOCK) {
                if (!checkFlags(FLAG_ALL_MEMBERS_RESOLVED)) {
                    setFlags(FLAG_RESOLVING_MEMBERS);

                    final int oldFlags = _flags;

                    RESOLVER.resolveMembers(this, oldFlags ^ FLAG_ALL_MEMBERS_RESOLVED);

                    if (_nestedTypes != null) {
                        for (final ReflectedType nestedType : _nestedTypes) {
                            nestedType.complete();
                        }
                    }

                    if (FLAGS_UPDATER.compareAndSet(this, oldFlags, FLAG_ALL_MEMBERS_RESOLVED)) {
                        return;
                    }
                }
            }
        }
    }

    @Override
    public MethodBase getDeclaringMethod() {
        return _declaringMethod;
    }

    @Override
    protected ConstructorList getDeclaredConstructors() {
        ensureMembersResolved();
        if (_constructors == null) {
            return ConstructorList.empty();
        }
        return new ConstructorList(_constructors);
    }

    @Override
    protected MethodList getDeclaredMethods() {
        ensureMembersResolved();
        if (_methods == null) {
            return MethodList.empty();
        }
        return new MethodList(_methods);
    }

    @Override
    protected FieldList getDeclaredFields() {
        ensureMembersResolved();
        if (_fields == null) {
            return FieldList.empty();
        }
        return new FieldList(_fields);
    }

    @Override
    protected TypeList getDeclaredTypes() {
        ensureMembersResolved();
        if (_nestedTypes == null) {
            return TypeList.empty();
        }
        return new TypeList(_nestedTypes);
    }

    void complete() {
        if (_completed) {
            return;
        }

        _completed = true;

        if (_genericParameters == null || _genericParameters.isEmpty()) {
            _typeBindings = TypeBindings.empty();
        }
        else {
            _typeBindings = TypeBindings.createUnbound(list(_genericParameters));
        }
    }

    @Override
    public Type getBaseType() {
        return _baseType;
    }

    @Override
    public TypeList getExplicitInterfaces() {
        if (_interfaces == null) {
            return TypeList.empty();
        }
        return _interfaces;
    }

    @Override
    protected TypeBindings getTypeBindings() {
        completeIfNecessary();
        return _typeBindings;
    }

    @Override
    public Type getGenericTypeDefinition() {
        if (!isGenericType()) {
            throw Error.notGenericType(this);
        }
        if (!getTypeBindings().hasBoundParameters()) {
            return this;
        }
        throw ContractUtils.unreachable();
    }

    @Override
    protected Type makeGenericTypeCore(final TypeList typeArguments) {
        synchronized (CACHE_LOCK) {
            return CACHE.getGenericType(getGenericTypeDefinition(), typeArguments);
        }
    }

    @Override
    public Class getErasedClass() {
        completeIfNecessary();
        return _rawClass;
    }

    @Override
    public MemberType getMemberType() {
        return MemberType.TypeInfo;
    }

    @Override
    public Type getDeclaringType() {
        return _declaringType;
    }

    @Override
    public int getModifiers() {
        return _rawClass.getModifiers();
    }

    @Override
    public  R accept(final TypeVisitor visitor, final P parameter) {
        return visitor.visitClassType(this, parameter);
    }

    @Override
    protected StringBuilder _appendClassName(final StringBuilder sb, final boolean fullName, final boolean dottedName) {
        if (!fullName) {
            return sb.append(_simpleName);
        }
        if (dottedName) {
            return sb.append(_name);
        }
        return super._appendClassName(sb, true, false);
    }

    private final static SimpleVisitor GenericParameterFinder =
        new SimpleVisitor() {
            public GenericParameter visit(final TypeList types, final java.lang.reflect.Type s) {
                for (final Type type : types) {
                    final GenericParameter result = visit(type, s);
                    if (result != null) {
                        return result;
                    }
                }
                return null;
            }

            @Override
            public GenericParameter visitCapturedType(final Type t, final java.lang.reflect.Type s) {
                return null;
            }

            @Override
            public GenericParameter visitClassType(final Type type, final java.lang.reflect.Type parameter) {
                GenericParameter result;

                if (type.isGenericType()) {
                    result = visit(type.getGenericTypeParameters(), parameter);

                    if (result != null) {
                        return result;
                    }
                }

                if (type instanceof ReflectedType) {
                    final ReflectedType reflectedType = (ReflectedType) type;
                    if (reflectedType._methods != null) {
                        for (final Object o : reflectedType._methods) {
                            final ReflectedMethod method = (ReflectedMethod) o;
                            if (method.isGenericMethod()) {
                                result = visit(method.getGenericMethodParameters(), parameter);

                                if (result != null) {
                                    return result;
                                }
                            }
                        }
                    }
                }

                final Type declaringType = type.getDeclaringType();

                if (declaringType != null && declaringType != NullType) {
                    return visitClassType(declaringType, parameter);
                }

                return null;
            }

            @Override
            public GenericParameter visitPrimitiveType(final Type type, final java.lang.reflect.Type parameter) {
                return super.visitPrimitiveType(type, parameter);
            }

            @Override
            public GenericParameter visitTypeParameter(final Type type, final java.lang.reflect.Type parameter) {
                if (parameter instanceof TypeVariable) {
                    return visitTypeParameter(type, (TypeVariable) parameter);
                }
                return null;
            }

            public GenericParameter visitTypeParameter(final Type type, final TypeVariable typeVariable) {
                if (!(type instanceof GenericParameter)) {
                    return null;
                }

                final GenericParameter genericParameter = (GenericParameter) type;

                final int position = ArrayUtilities.indexOf(
                    typeVariable.getGenericDeclaration().getTypeParameters(),
                    typeVariable
                );

                if (position != genericParameter.getGenericParameterPosition()) {
                    return null;
                }

                final Type declaringType = genericParameter.getDeclaringType();

                if (declaringType != null && declaringType.getErasedClass() == typeVariable.getGenericDeclaration()) {
                    return genericParameter;
                }

                final MethodInfo method = genericParameter.getDeclaringMethod();

                if (method != null && method.getRawMethod() == typeVariable.getGenericDeclaration()) {
                    return genericParameter;
                }

                return null;
            }

            @Override
            public GenericParameter visitWildcardType(final Type type, final java.lang.reflect.Type parameter) {
                return null;
            }

            @Override
            public GenericParameter visitArrayType(final Type type, final java.lang.reflect.Type parameter) {
                return null;
            }
        };
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy