| /* |
| * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. Oracle designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Oracle in the LICENSE file that accompanied this code. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| package com.sun.tools.internal.ws.processor.modeler.annotation; |
| |
| import javax.annotation.processing.ProcessingEnvironment; |
| import javax.lang.model.element.ElementKind; |
| import javax.lang.model.element.ExecutableElement; |
| import javax.lang.model.element.TypeElement; |
| import javax.lang.model.element.VariableElement; |
| import javax.lang.model.type.DeclaredType; |
| import javax.lang.model.type.TypeKind; |
| import javax.lang.model.type.TypeMirror; |
| import javax.lang.model.util.ElementFilter; |
| import java.util.Collection; |
| |
| /** |
| * @author WS Development Team |
| */ |
| final class TypeModeler { |
| |
| private TypeModeler() { |
| } |
| |
| public static TypeElement getDeclaration(TypeMirror typeMirror) { |
| if (typeMirror != null && typeMirror.getKind().equals(TypeKind.DECLARED)) |
| return (TypeElement) ((DeclaredType) typeMirror).asElement(); |
| return null; |
| } |
| |
| public static TypeElement getDeclaringClassMethod(TypeMirror theClass, String methodName, TypeMirror[] args) { |
| return getDeclaringClassMethod(getDeclaration(theClass), methodName, args); |
| } |
| |
| public static TypeElement getDeclaringClassMethod(TypeElement theClass, String methodName, TypeMirror[] args) { |
| |
| TypeElement retClass = null; |
| if (theClass.getKind().equals(ElementKind.CLASS)) { |
| TypeMirror superClass = theClass.getSuperclass(); |
| if (!superClass.getKind().equals(TypeKind.NONE)) |
| retClass = getDeclaringClassMethod(superClass, methodName, args); |
| } |
| if (retClass == null) { |
| for (TypeMirror interfaceType : theClass.getInterfaces()) { |
| retClass = getDeclaringClassMethod(interfaceType, methodName, args); |
| } |
| } |
| if (retClass == null) { |
| Collection<? extends ExecutableElement> methods = ElementFilter.methodsIn(theClass.getEnclosedElements()); |
| for (ExecutableElement method : methods) { |
| if (method.getSimpleName().toString().equals(methodName)) { |
| retClass = theClass; |
| break; |
| } |
| } |
| } |
| return retClass; |
| } |
| |
| public static Collection<DeclaredType> collectInterfaces(TypeElement type) { |
| @SuppressWarnings({"unchecked"}) |
| Collection<DeclaredType> interfaces = (Collection<DeclaredType>) type.getInterfaces(); |
| for (TypeMirror interfaceType : type.getInterfaces()) { |
| interfaces.addAll(collectInterfaces(getDeclaration(interfaceType))); |
| } |
| return interfaces; |
| } |
| |
| public static boolean isSubclass(String subTypeName, String superTypeName, ProcessingEnvironment env) { |
| return isSubclass(env.getElementUtils().getTypeElement(subTypeName), env.getElementUtils().getTypeElement(superTypeName), env); |
| } |
| |
| public static boolean isSubclass(TypeElement subType, TypeElement superType, ProcessingEnvironment env) { |
| return !subType.equals(superType) && isSubElement(subType, superType); |
| } |
| |
| public static TypeMirror getHolderValueType(TypeMirror type, TypeElement defHolder, ProcessingEnvironment env) { |
| TypeElement typeElement = getDeclaration(type); |
| if (typeElement == null) |
| return null; |
| |
| if (isSubElement(typeElement, defHolder)) { |
| if (type.getKind().equals(TypeKind.DECLARED)) { |
| Collection<? extends TypeMirror> argTypes = ((DeclaredType) type).getTypeArguments(); |
| if (argTypes.size() == 1) { |
| return argTypes.iterator().next(); |
| } else if (argTypes.isEmpty()) { |
| VariableElement member = getValueMember(typeElement); |
| if (member != null) { |
| return member.asType(); |
| } |
| } |
| } |
| } |
| return null; |
| } |
| |
| public static VariableElement getValueMember(TypeMirror classType) { |
| return getValueMember(getDeclaration(classType)); |
| } |
| |
| public static VariableElement getValueMember(TypeElement type) { |
| VariableElement member = null; |
| for (VariableElement field : ElementFilter.fieldsIn(type.getEnclosedElements())) { |
| if ("value".equals(field.getSimpleName().toString())) { |
| member = field; |
| break; |
| } |
| } |
| if (member == null && type.getKind().equals(ElementKind.CLASS)) |
| member = getValueMember(type.getSuperclass()); |
| return member; |
| } |
| |
| public static boolean isSubElement(TypeElement d1, TypeElement d2) { |
| if (d1.equals(d2)) |
| return true; |
| TypeElement superClassDecl = null; |
| if (d1.getKind().equals(ElementKind.CLASS)) { |
| TypeMirror superClass = d1.getSuperclass(); |
| if (!superClass.getKind().equals(TypeKind.NONE)) { |
| superClassDecl = (TypeElement) ((DeclaredType) superClass).asElement(); |
| if (superClassDecl.equals(d2)) |
| return true; |
| } |
| } |
| for (TypeMirror superIntf : d1.getInterfaces()) { |
| DeclaredType declaredSuperIntf = (DeclaredType) superIntf; |
| if (declaredSuperIntf.asElement().equals(d2)) { |
| return true; |
| } |
| if (isSubElement((TypeElement) declaredSuperIntf.asElement(), d2)) { |
| return true; |
| } else if (superClassDecl != null && isSubElement(superClassDecl, d2)) { |
| return true; |
| } |
| } |
| return false; |
| } |
| |
| } |