|
1
19 package org.netbeans.api.java.source;
20
21 import java.util.ArrayList  ;
22 import java.util.Collections ;
23 import java.util.List ;
24 import javax.lang.model.element.Element;
25 import javax.lang.model.element.TypeElement;
26 import javax.lang.model.element.TypeParameterElement;
27 import javax.lang.model.type.ArrayType;
28 import javax.lang.model.type.DeclaredType;
29 import javax.lang.model.type.ErrorType;
30 import javax.lang.model.type.TypeKind;
31 import javax.lang.model.type.TypeMirror;
32 import javax.lang.model.type.TypeVariable;
33 import javax.lang.model.type.WildcardType;
34 import org.netbeans.api.java.source.CompilationInfo;
35
36
73 public final class TypeMirrorHandle<T extends TypeMirror> {
74
75 private TypeKind kind;
76 private ElementHandle<? extends Element> element;
77 private List <TypeMirrorHandle<? extends TypeMirror>> typeMirrors;
78
79 private TypeMirrorHandle(TypeKind kind, ElementHandle<? extends Element> element, List <TypeMirrorHandle<? extends TypeMirror>> typeArguments) {
80 this.kind = kind;
81 this.element = element;
82 this.typeMirrors = typeArguments;
83 }
84
85
95 public static <T extends TypeMirror> TypeMirrorHandle<T> create(T tm) {
96 TypeKind kind = tm.getKind();
97 ElementHandle<? extends Element> element = null;
98 List <TypeMirrorHandle<? extends TypeMirror>> typeMirrors = null;
99 switch (kind) {
100 case BOOLEAN:
101 case BYTE:
102 case CHAR:
103 case DOUBLE:
104 case FLOAT:
105 case INT:
106 case LONG:
107 case NONE:
108 case NULL:
109 case SHORT:
110 case VOID:
111 break;
112 case DECLARED:
113 DeclaredType dt = (DeclaredType)tm;
114 TypeElement te = (TypeElement)dt.asElement();
115 element = ElementHandle.create(te);
116 if (te.getTypeParameters().isEmpty())
117 break;
118 List <? extends TypeMirror> targs = dt.getTypeArguments();
119 if (!targs.isEmpty()) {
120 TypeMirror targ = targs.get(0);
121 if (targ.getKind() == TypeKind.TYPEVAR) {
122 TypeParameterElement tpe = (TypeParameterElement)((TypeVariable)targ).asElement();
123 if (tpe.getGenericElement() == dt.asElement())
124 break;
125 }
126 }
127 typeMirrors = new ArrayList <TypeMirrorHandle<? extends TypeMirror>>(targs.size());
128 for (TypeMirror ta : targs)
129 typeMirrors.add(create(ta));
130 break;
131 case ARRAY:
132 typeMirrors = Collections.<TypeMirrorHandle<? extends TypeMirror>>singletonList(TypeMirrorHandle.create(((ArrayType)tm).getComponentType()));
133 break;
134 case TYPEVAR:
135 TypeVariable tv = (TypeVariable)tm;
136 element = ElementHandle.create(tv.asElement());
137 typeMirrors = new ArrayList <TypeMirrorHandle<? extends TypeMirror>>();
138 typeMirrors.add(tv.getLowerBound() != null ? TypeMirrorHandle.create(tv.getLowerBound()) : null);
139 typeMirrors.add(tv.getUpperBound() != null ? TypeMirrorHandle.create(tv.getUpperBound()) : null);
140 break;
141 case WILDCARD:
142 WildcardType wt = (WildcardType)tm;
143 typeMirrors = new ArrayList <TypeMirrorHandle<? extends TypeMirror>>();
144 typeMirrors.add(wt.getExtendsBound() != null ? TypeMirrorHandle.create(wt.getExtendsBound()) : null);
145 typeMirrors.add(wt.getSuperBound() != null ? TypeMirrorHandle.create(wt.getSuperBound()) : null);
146 break;
147 case ERROR:
148 ErrorType et = (ErrorType)tm;
149 element = ElementHandle.create(et.asElement());
150 break;
151 default:
152 throw new IllegalArgumentException ("Currently unsupported TypeKind: " + tm.getKind());
153 }
154 return new TypeMirrorHandle(kind, element, typeMirrors);
155 }
156
157
164 public T resolve(CompilationInfo info) {
165 switch (kind) {
166 case BOOLEAN:
167 case BYTE:
168 case CHAR:
169 case DOUBLE:
170 case FLOAT:
171 case INT:
172 case LONG:
173 case SHORT:
174 return (T)info.getTypes().getPrimitiveType(kind);
175 case NONE:
176 case VOID:
177 return (T)info.getTypes().getNoType(kind);
178 case NULL:
179 return (T)info.getTypes().getNullType();
180 case DECLARED:
181 TypeElement te = (TypeElement)element.resolve(info);
182 if (te == null)
183 return null;
184 if (typeMirrors == null)
185 return (T)te.asType();
186 List <TypeMirror> resolvedTypeArguments = new ArrayList <TypeMirror>();
187 for (TypeMirrorHandle t : typeMirrors) {
188 TypeMirror resolved = t.resolve(info);
189 if (resolved == null)
190 return null;
191 resolvedTypeArguments.add(resolved);
192 }
193 return (T)info.getTypes().getDeclaredType(te, resolvedTypeArguments.toArray(new TypeMirror[resolvedTypeArguments.size()]));
194 case ARRAY:
195 TypeMirror resolved = typeMirrors.get(0).resolve(info);
196 if (resolved == null)
197 return null;
198 return (T)info.getTypes().getArrayType(resolved);
199 case TYPEVAR:
200 Element e = element.resolve(info);
201 if (!(e instanceof com.sun.tools.javac.code.Symbol.TypeSymbol))
202 return null;
203 TypeMirrorHandle<? extends TypeMirror> lBound = typeMirrors.get(0);
204 TypeMirror lowerBound = lBound != null ? lBound.resolve(info) : null;
205 TypeMirrorHandle<? extends TypeMirror> uBound = typeMirrors.get(1);
206 TypeMirror upperBound = uBound != null ? uBound.resolve(info) : null;
207 return (T)new com.sun.tools.javac.code.Type.TypeVar((com.sun.tools.javac.code.Symbol.TypeSymbol)e,
208 (com.sun.tools.javac.code.Type)lowerBound, (com.sun.tools.javac.code.Type)upperBound);
209 case WILDCARD:
210 TypeMirrorHandle<? extends TypeMirror> eBound = typeMirrors.get(0);
211 TypeMirror extendsBound = eBound != null ? eBound.resolve(info) : null;
212 TypeMirrorHandle<? extends TypeMirror> sBound = typeMirrors.get(1);
213 TypeMirror superBound = sBound != null ? sBound.resolve(info) : null;
214 return (T)info.getTypes().getWildcardType(extendsBound, superBound);
215 case ERROR:
216 e = element.resolve(info);
217 if (!(e instanceof com.sun.tools.javac.code.Symbol.ClassSymbol))
218 return null;
219 return (T)new com.sun.tools.javac.code.Type.ErrorType((com.sun.tools.javac.code.Symbol.ClassSymbol)e);
220 default:
221 throw new IllegalStateException ("Internal error: unknown TypeHandle kind: " + kind);
222 }
223 }
224
225
231 public TypeKind getKind () {
232 return this.kind;
233 }
234 }
235 |
Popular Tags |
Java API By Example, From Geeks To Geeks.