aoqi@0: /*
aoqi@0:  * Copyright (c) 1997, 2012, Oracle and/or its affiliates. All rights reserved.
aoqi@0:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0:  *
aoqi@0:  * This code is free software; you can redistribute it and/or modify it
aoqi@0:  * under the terms of the GNU General Public License version 2 only, as
aoqi@0:  * published by the Free Software Foundation.  Oracle designates this
aoqi@0:  * particular file as subject to the "Classpath" exception as provided
aoqi@0:  * by Oracle in the LICENSE file that accompanied this code.
aoqi@0:  *
aoqi@0:  * This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
aoqi@0:  * version 2 for more details (a copy is included in the LICENSE file that
aoqi@0:  * accompanied this code).
aoqi@0:  *
aoqi@0:  * You should have received a copy of the GNU General Public License version
aoqi@0:  * 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0:  *
aoqi@0:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0:  * or visit www.oracle.com if you need additional information or have any
aoqi@0:  * questions.
aoqi@0:  */
aoqi@0: 
aoqi@0: package com.sun.tools.internal.ws.processor.modeler.annotation;
aoqi@0: 
aoqi@0: import javax.annotation.processing.ProcessingEnvironment;
aoqi@0: import javax.lang.model.element.Name;
aoqi@0: import javax.lang.model.element.TypeElement;
aoqi@0: import javax.lang.model.type.ArrayType;
aoqi@0: import javax.lang.model.type.DeclaredType;
aoqi@0: import javax.lang.model.type.PrimitiveType;
aoqi@0: import javax.lang.model.type.TypeKind;
aoqi@0: import javax.lang.model.type.TypeMirror;
aoqi@0: import java.util.ArrayList;
aoqi@0: import java.util.Collection;
aoqi@0: 
aoqi@0: /**
aoqi@0:  * @author dkohlert
aoqi@0:  */
aoqi@0: public class TypeMonikerFactory {
aoqi@0: 
aoqi@0:     public static TypeMoniker getTypeMoniker(TypeMirror typeMirror) {
aoqi@0:         if (typeMirror == null)
aoqi@0:             throw new NullPointerException();
aoqi@0: 
aoqi@0:         if (typeMirror.getKind().isPrimitive())
aoqi@0:             return new PrimitiveTypeMoniker((PrimitiveType) typeMirror);
aoqi@0:         else if (typeMirror.getKind().equals(TypeKind.ARRAY))
aoqi@0:             return new ArrayTypeMoniker((ArrayType) typeMirror);
aoqi@0:         else if (typeMirror.getKind().equals(TypeKind.DECLARED))
aoqi@0:             return new DeclaredTypeMoniker((DeclaredType) typeMirror);
aoqi@0:         return getTypeMoniker(typeMirror.toString());
aoqi@0:     }
aoqi@0: 
aoqi@0:     public static TypeMoniker getTypeMoniker(String typeName) {
aoqi@0:         return new StringMoniker(typeName);
aoqi@0:     }
aoqi@0: 
aoqi@0:     static class ArrayTypeMoniker implements TypeMoniker {
aoqi@0:         private TypeMoniker arrayType;
aoqi@0: 
aoqi@0:         public ArrayTypeMoniker(ArrayType type) {
aoqi@0:             arrayType = TypeMonikerFactory.getTypeMoniker(type.getComponentType());
aoqi@0:         }
aoqi@0: 
aoqi@0:         public TypeMirror create(ProcessingEnvironment apEnv) {
aoqi@0:             return apEnv.getTypeUtils().getArrayType(arrayType.create(apEnv));
aoqi@0:         }
aoqi@0:     }
aoqi@0:     static class DeclaredTypeMoniker implements TypeMoniker {
aoqi@0:         private Name typeDeclName;
aoqi@0:         private Collection<TypeMoniker> typeArgs = new ArrayList<TypeMoniker>();
aoqi@0: 
aoqi@0:         public DeclaredTypeMoniker(DeclaredType type) {
aoqi@0:             typeDeclName = ((TypeElement) type.asElement()).getQualifiedName();
aoqi@0:             for (TypeMirror arg : type.getTypeArguments())
aoqi@0:                 typeArgs.add(TypeMonikerFactory.getTypeMoniker(arg));
aoqi@0:         }
aoqi@0: 
aoqi@0:         public TypeMirror create(ProcessingEnvironment apEnv) {
aoqi@0:             TypeElement typeDecl = apEnv.getElementUtils().getTypeElement(typeDeclName);
aoqi@0:             TypeMirror[] tmpArgs = new TypeMirror[typeArgs.size()];
aoqi@0:             int idx = 0;
aoqi@0:             for (TypeMoniker moniker : typeArgs)
aoqi@0:                 tmpArgs[idx++] = moniker.create(apEnv);
aoqi@0: 
aoqi@0:             return apEnv.getTypeUtils().getDeclaredType(typeDecl, tmpArgs);
aoqi@0:         }
aoqi@0:     }
aoqi@0:     static class PrimitiveTypeMoniker implements TypeMoniker {
aoqi@0:         private TypeKind kind;
aoqi@0: 
aoqi@0:         public PrimitiveTypeMoniker(PrimitiveType type) {
aoqi@0:             kind = type.getKind();
aoqi@0:         }
aoqi@0: 
aoqi@0:         public TypeMirror create(ProcessingEnvironment apEnv) {
aoqi@0:             return apEnv.getTypeUtils().getPrimitiveType(kind);
aoqi@0:         }
aoqi@0:     }
aoqi@0:     static class StringMoniker implements TypeMoniker {
aoqi@0:         private String typeName;
aoqi@0: 
aoqi@0:         public StringMoniker(String typeName) {
aoqi@0:             this.typeName = typeName;
aoqi@0:         }
aoqi@0: 
aoqi@0:         public TypeMirror create(ProcessingEnvironment apEnv) {
aoqi@0:             return apEnv.getTypeUtils().getDeclaredType(apEnv.getElementUtils().getTypeElement(typeName));
aoqi@0:         }
aoqi@0:     }
aoqi@0: }