JavaDescriptorResolver.java 74.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
/*
 * Copyright 2010-2012 JetBrains s.r.o.
 *
 * Licensed 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.
 */

17 18
package org.jetbrains.jet.lang.resolve.java;

19
import com.google.common.collect.Lists;
A
Andrey Breslav 已提交
20
import com.google.common.collect.Maps;
A
Andrey Breslav 已提交
21
import com.google.common.collect.Sets;
22
import com.intellij.openapi.project.Project;
23
import com.intellij.openapi.util.Pair;
24
import com.intellij.openapi.vfs.VirtualFile;
25
import com.intellij.psi.*;
26
import com.intellij.psi.impl.compiled.ClsClassImpl;
27
import com.intellij.psi.search.DelegatingGlobalSearchScope;
28
import com.intellij.psi.search.GlobalSearchScope;
29
import jet.typeinfo.TypeInfoVariance;
30 31
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
32
import org.jetbrains.jet.lang.descriptors.*;
A
Andrey Breslav 已提交
33
import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptor;
34
import org.jetbrains.jet.lang.psi.JetPsiUtil;
35
import org.jetbrains.jet.lang.resolve.BindingContext;
36
import org.jetbrains.jet.lang.resolve.DescriptorUtils;
37
import org.jetbrains.jet.lang.resolve.OverrideResolver;
38
import org.jetbrains.jet.lang.resolve.constants.*;
M
Maxim Shafirov 已提交
39
import org.jetbrains.jet.lang.resolve.java.alt.AltClassFinder;
40
import org.jetbrains.jet.lang.resolve.java.kt.JetClassAnnotation;
41
import org.jetbrains.jet.lang.types.*;
S
Stepan Koltsov 已提交
42
import org.jetbrains.jet.lang.types.lang.JetStandardClasses;
43
import org.jetbrains.jet.lang.types.lang.JetStandardLibrary;
44
import org.jetbrains.jet.plugin.JetFileType;
45
import org.jetbrains.jet.rt.signature.JetSignatureAdapter;
46 47 48
import org.jetbrains.jet.rt.signature.JetSignatureExceptionsAdapter;
import org.jetbrains.jet.rt.signature.JetSignatureReader;
import org.jetbrains.jet.rt.signature.JetSignatureVisitor;
49

50
import java.util.*;
51 52 53 54 55

/**
 * @author abreslav
 */
public class JavaDescriptorResolver {
56 57
    
    public static String JAVA_ROOT = "<java_root>";
58

59 60
    /*package*/ static final DeclarationDescriptor JAVA_METHOD_TYPE_PARAMETER_PARENT = new DeclarationDescriptorImpl(null, Collections.<AnnotationDescriptor>emptyList(), "<java_generic_method>") {

61 62
        @Override
        public DeclarationDescriptor substitute(TypeSubstitutor substitutor) {
63
            throw new UnsupportedOperationException();
64 65
        }

66 67
        @Override
        public <R, D> R accept(DeclarationDescriptorVisitor<R, D> visitor, D data) {
68
            return visitor.visitDeclarationDescriptor(this, data);
69 70 71
        }
    };

A
Andrey Breslav 已提交
72
    /*package*/ static final DeclarationDescriptor JAVA_CLASS_OBJECT = new DeclarationDescriptorImpl(null, Collections.<AnnotationDescriptor>emptyList(), "<java_class_object_emulation>") {
A
Andrey Breslav 已提交
73 74 75
        @NotNull
        @Override
        public DeclarationDescriptor substitute(TypeSubstitutor substitutor) {
76
            throw new UnsupportedOperationException();
A
Andrey Breslav 已提交
77 78 79 80 81 82 83
        }

        @Override
        public <R, D> R accept(DeclarationDescriptorVisitor<R, D> visitor, D data) {
            return visitor.visitDeclarationDescriptor(this, data);
        }
    };
84 85 86 87 88
    
    private enum TypeParameterDescriptorOrigin {
        JAVA,
        KOTLIN,
    }
89

90 91
    public static class TypeParameterDescriptorInitialization {
        @NotNull
92
        private final TypeParameterDescriptorOrigin origin;
93 94 95
        @NotNull
        final TypeParameterDescriptor descriptor;
        final PsiTypeParameter psiTypeParameter;
96 97 98 99
        @Nullable
        private final List<JetType> upperBoundsForKotlin;
        @Nullable
        private final List<JetType> lowerBoundsForKotlin;
100

101
        private TypeParameterDescriptorInitialization(@NotNull TypeParameterDescriptor descriptor, @NotNull PsiTypeParameter psiTypeParameter) {
102
            this.origin = TypeParameterDescriptorOrigin.JAVA;
103
            this.descriptor = descriptor;
104
            this.psiTypeParameter = psiTypeParameter;
105 106 107 108
            this.upperBoundsForKotlin = null;
            this.lowerBoundsForKotlin = null;
        }

109
        private TypeParameterDescriptorInitialization(@NotNull TypeParameterDescriptor descriptor, @NotNull PsiTypeParameter psiTypeParameter,
110 111 112
                List<JetType> upperBoundsForKotlin, List<JetType> lowerBoundsForKotlin) {
            this.origin = TypeParameterDescriptorOrigin.KOTLIN;
            this.descriptor = descriptor;
113
            this.psiTypeParameter = psiTypeParameter;
114 115
            this.upperBoundsForKotlin = upperBoundsForKotlin;
            this.lowerBoundsForKotlin = lowerBoundsForKotlin;
116 117
        }
    }
118 119


120
    private static abstract class ResolverScopeData {
121
        protected boolean kotlin;
122 123
        
        private Map<String, NamedMembers> namedMembersMap;
124 125 126
        
        @NotNull
        public abstract List<TypeParameterDescriptor> getTypeParameters();
127
    }
A
Andrey Breslav 已提交
128

129 130 131 132
    static abstract class ResolverClassData extends ResolverScopeData {

        @NotNull
        public abstract ClassDescriptor getClassDescriptor();
133 134 135 136 137 138

        @NotNull
        @Override
        public List<TypeParameterDescriptor> getTypeParameters() {
            return getClassDescriptor().getTypeConstructor().getParameters();
        }
139 140
    }

141
    /** Class with instance members */
142 143 144 145 146
    static class ResolverBinaryClassData extends ResolverClassData {

        ResolverBinaryClassData() {
        }

147
        private MutableClassDescriptorLite classDescriptor;
148

149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166
        List<TypeParameterDescriptorInitialization> typeParameters;

        @Override
        @NotNull
        public ClassDescriptor getClassDescriptor() {
            return classDescriptor;
        }
    }

    static class ResolverSrcClassData extends ResolverClassData {
        @NotNull
        private final ClassDescriptor classDescriptor;

        ResolverSrcClassData(@NotNull ClassDescriptor classDescriptor) {
            this.classDescriptor = classDescriptor;
        }

        @Override
167 168 169 170 171 172
        @NotNull
        public ClassDescriptor getClassDescriptor() {
            return classDescriptor;
        }
    }

173
    /** Either package or class with static members */
174
    private static class ResolverNamespaceData extends ResolverScopeData {
175 176 177 178 179 180
        private JavaNamespaceDescriptor namespaceDescriptor;

        @NotNull
        public NamespaceDescriptor getNamespaceDescriptor() {
            return namespaceDescriptor;
        }
181 182 183 184 185 186

        @NotNull
        @Override
        public List<TypeParameterDescriptor> getTypeParameters() {
            return new ArrayList<TypeParameterDescriptor>(0);
        }
187 188
    }

189
    protected final Map<String, ResolverBinaryClassData> classDescriptorCache = Maps.newHashMap();
190 191 192
    protected final Map<String, ResolverNamespaceData> namespaceDescriptorCacheByFqn = Maps.newHashMap();
    protected final Map<PsiElement, ResolverNamespaceData> namespaceDescriptorCache = Maps.newHashMap();

193 194 195
    protected final JavaPsiFacade javaFacade;
    protected final GlobalSearchScope javaSearchScope;
    protected final JavaSemanticServices semanticServices;
M
Maxim Shafirov 已提交
196
    private final AltClassFinder altClassFinder;
197 198 199

    public JavaDescriptorResolver(Project project, JavaSemanticServices semanticServices) {
        this.javaFacade = JavaPsiFacade.getInstance(project);
200 201 202 203 204 205
        this.javaSearchScope = new DelegatingGlobalSearchScope(GlobalSearchScope.allScope(project)) {
            @Override
            public boolean contains(VirtualFile file) {
                return myBaseScope.contains(file) && file.getFileType() != JetFileType.INSTANCE;
            }
        };
206
        this.semanticServices = semanticServices;
M
Maxim Shafirov 已提交
207
        altClassFinder = new AltClassFinder(project);
208
    }
209
    
210
    @Nullable
211
    ResolverClassData resolveClassData(@NotNull PsiClass psiClass) {
212
        String qualifiedName = psiClass.getQualifiedName();
213

214 215 216 217
        if (qualifiedName.endsWith(JvmAbi.TRAIT_IMPL_SUFFIX)) {
            // TODO: only if -$$TImpl class is created by Kotlin
            return null;
        }
218

219
        // First, let's check that this is a real Java class, not a Java's view on a Kotlin class:
220 221
        ClassDescriptor kotlinClassDescriptor = semanticServices.getKotlinClassDescriptor(qualifiedName);
        if (kotlinClassDescriptor != null) {
222
            // TODO: return null, no loops here
223
            return new ResolverSrcClassData(kotlinClassDescriptor);
224 225
        }

226 227 228 229 230 231
        PsiClass containingClass = psiClass.getContainingClass();
        if (containingClass != null) {
            // must resolve containing class first, because inner class must have a reference to it
            resolveClass(containingClass);
        }

232
        // Not let's take a descriptor of a Java class
233
        ResolverBinaryClassData classData = classDescriptorCache.get(qualifiedName);
234 235 236
        if (classData == null) {
            classData = createJavaClassDescriptor(psiClass);
            classDescriptorCache.put(qualifiedName, classData);
237
        }
238 239 240 241 242 243 244 245 246 247 248
        return classData;
    }

    @Nullable
    public ClassDescriptor resolveClass(@NotNull PsiClass psiClass) {
        ResolverClassData classData = resolveClassData(psiClass);
        if (classData != null) {
            return classData.getClassDescriptor();
        } else {
            return null;
        }
249 250 251 252
    }

    @Nullable
    public ClassDescriptor resolveClass(@NotNull String qualifiedName) {
253

254 255 256 257 258
        if (qualifiedName.endsWith(JvmAbi.TRAIT_IMPL_SUFFIX)) {
            // TODO: only if -$$TImpl class is created by Kotlin
            return null;
        }
        
259
        // First, let's check that this is a real Java class, not a Java's view on a Kotlin class:
260 261
        ClassDescriptor kotlinClassDescriptor = semanticServices.getKotlinClassDescriptor(qualifiedName);
        if (kotlinClassDescriptor != null) {
262
            // TODO: return null, no loops here
263 264 265
            return kotlinClassDescriptor;
        }

266
        // Not let's take a descriptor of a Java class
267
        ResolverBinaryClassData classData = classDescriptorCache.get(qualifiedName);
268
        if (classData == null) {
269
            PsiClass psiClass = findClass(qualifiedName);
270 271 272
            if (psiClass == null) {
                return null;
            }
273
            classData = createJavaClassDescriptor(psiClass);
274
        }
275
        return classData.getClassDescriptor();
276 277
    }

278
    private ResolverBinaryClassData createJavaClassDescriptor(@NotNull final PsiClass psiClass) {
279 280 281
        if (classDescriptorCache.containsKey(psiClass.getQualifiedName())) {
            throw new IllegalStateException(psiClass.getQualifiedName());
        }
A
Andrey Breslav 已提交
282

283 284
        checkPsiClassIsNotJet(psiClass);

285
        String name = psiClass.getName();
286
        ResolverBinaryClassData classData = new ResolverBinaryClassData();
287
        classDescriptorCache.put(psiClass.getQualifiedName(), classData);
288
        ClassKind kind = psiClass.isInterface() ? (psiClass.isAnnotationType() ? ClassKind.ANNOTATION_CLASS : ClassKind.TRAIT) : ClassKind.CLASS;
S
Stepan Koltsov 已提交
289
        ClassOrNamespaceDescriptor containingDeclaration = resolveParentDescriptor(psiClass);
290
        classData.classDescriptor = new MutableClassDescriptorLite(containingDeclaration, kind);
291
        classData.classDescriptor.setName(name);
292 293
        classData.classDescriptor.setAnnotations(resolveAnnotations(psiClass));

A
Andrey Breslav 已提交
294
        List<JetType> supertypes = new ArrayList<JetType>();
295

S
Stepan Koltsov 已提交
296 297 298
        TypeVariableResolver outerTypeVariableByNameResolver = TypeVariableResolvers.classTypeVariableResolver(
                (ClassOrNamespaceDescriptor) classData.classDescriptor.getContainingDeclaration(),
                "class " + psiClass.getQualifiedName());
299 300

        classData.typeParameters = createUninitializedClassTypeParameters(psiClass, classData, outerTypeVariableByNameResolver);
301 302 303 304 305 306
        
        List<TypeParameterDescriptor> typeParameters = new ArrayList<TypeParameterDescriptor>();
        for (TypeParameterDescriptorInitialization typeParameter : classData.typeParameters) {
            typeParameters.add(typeParameter.descriptor);
        }
        
307 308 309
        classData.classDescriptor.setTypeParameterDescriptors(typeParameters);
        classData.classDescriptor.setSupertypes(supertypes);
        classData.classDescriptor.setVisibility(resolveVisibilityFromPsiModifiers(psiClass));
310 311 312 313 314 315 316 317 318 319
        Modality modality;
        if (classData.classDescriptor.getKind() == ClassKind.ANNOTATION_CLASS) {
            modality = Modality.FINAL;
        }
        else {
            modality = Modality.convertFromFlags(
                    psiClass.hasModifierProperty(PsiModifier.ABSTRACT) || psiClass.isInterface(),
                    !psiClass.hasModifierProperty(PsiModifier.FINAL));
        }
        classData.classDescriptor.setModality(modality);
320 321
        classData.classDescriptor.createTypeConstructor();
        classData.classDescriptor.setScopeForMemberLookup(new JavaClassMembersScope(classData.classDescriptor, psiClass, semanticServices, false));
322

S
Stepan Koltsov 已提交
323
        initializeTypeParameters(classData.typeParameters, classData.classDescriptor, "class " + psiClass.getQualifiedName());
324

S
Stepan Koltsov 已提交
325 326 327
        TypeVariableResolver resolverForTypeParameters = TypeVariableResolvers.classTypeVariableResolver(
                classData.classDescriptor,
                "class " + psiClass.getQualifiedName());
328

329
        // TODO: ugly hack: tests crash if initializeTypeParameters called with class containing proper supertypes
330
        supertypes.addAll(getSupertypes(new PsiClassWrapper(psiClass), classData.classDescriptor, classData.getTypeParameters()));
331

A
Andrey Breslav 已提交
332
        PsiMethod[] psiConstructors = psiClass.getConstructors();
333

334
        boolean isStatic = psiClass.hasModifierProperty(PsiModifier.STATIC);
335
        if (psiConstructors.length == 0) {
A
Andrey Breslav 已提交
336 337 338 339
            // We need to create default constructors for classes and abstract classes.
            // Example:
            // class Kotlin() : Java() {}
            // abstract public class Java {}
340
            if (!psiClass.isInterface()) {
341
                ConstructorDescriptorImpl constructorDescriptor = new ConstructorDescriptorImpl(
342
                        classData.classDescriptor,
A
Andrey Breslav 已提交
343
                        Collections.<AnnotationDescriptor>emptyList(),
344
                        false);
345
                constructorDescriptor.initialize(typeParameters, Collections.<ValueParameterDescriptor>emptyList(), classData.classDescriptor.getVisibility(), isStatic);
346
                constructorDescriptor.setReturnType(classData.classDescriptor.getDefaultType());
347
                classData.classDescriptor.addConstructor(constructorDescriptor, null);
348
                semanticServices.getTrace().record(BindingContext.CONSTRUCTOR, psiClass, constructorDescriptor);
349
            }
350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370
            if (psiClass.isAnnotationType()) {
                // A constructor for an annotation type takes all the "methods" in the @interface as parameters
                ConstructorDescriptorImpl constructorDescriptor = new ConstructorDescriptorImpl(
                        classData.classDescriptor,
                        Collections.<AnnotationDescriptor>emptyList(),
                        false);

                List<ValueParameterDescriptor> valueParameters = Lists.newArrayList();
                PsiMethod[] methods = psiClass.getMethods();
                for (int i = 0; i < methods.length; i++) {
                    PsiMethod method = methods[i];
                    if (method instanceof PsiAnnotationMethod) {
                        PsiAnnotationMethod annotationMethod = (PsiAnnotationMethod) method;
                        assert annotationMethod.getParameterList().getParameters().length == 0;

                        PsiType returnType = annotationMethod.getReturnType();

                        // We take the following heuristical convention:
                        // if the last method of the @interface is an array, we convert it into a vararg
                        JetType varargElementType = null;
                        if (i == methods.length - 1 && (returnType instanceof PsiArrayType)) {
371
                            varargElementType = semanticServices.getTypeTransformer().transformToType(((PsiArrayType) returnType).getComponentType(), resolverForTypeParameters);
372 373 374 375 376 377 378 379
                        }

                        valueParameters.add(new ValueParameterDescriptorImpl(
                                constructorDescriptor,
                                i,
                                Collections.<AnnotationDescriptor>emptyList(),
                                method.getName(),
                                false,
380
                                semanticServices.getTypeTransformer().transformToType(returnType, resolverForTypeParameters),
381 382 383 384 385
                                annotationMethod.getDefaultValue() != null,
                                varargElementType));
                    }
                }

386
                constructorDescriptor.initialize(typeParameters, valueParameters, classData.classDescriptor.getVisibility(), isStatic);
387 388 389 390
                constructorDescriptor.setReturnType(classData.classDescriptor.getDefaultType());
                classData.classDescriptor.addConstructor(constructorDescriptor, null);
                semanticServices.getTrace().record(BindingContext.CONSTRUCTOR, psiClass, constructorDescriptor);
            }
391 392
        }
        else {
393 394 395 396 397
            for (PsiMethod psiConstructor : psiConstructors) {
                PsiMethodWrapper constructor = new PsiMethodWrapper(psiConstructor);

                if (constructor.getJetConstructor().hidden()) {
                    continue;
398 399
                }

400
                ConstructorDescriptorImpl constructorDescriptor = new ConstructorDescriptorImpl(
401
                        classData.classDescriptor,
A
Andrey Breslav 已提交
402
                        Collections.<AnnotationDescriptor>emptyList(), // TODO
403
                        false);
S
Stepan Koltsov 已提交
404
                String context = "constructor of class " + psiClass.getQualifiedName();
405
                ValueParameterDescriptors valueParameterDescriptors = resolveParameterDescriptors(constructorDescriptor,
406
                        constructor.getParameters(),
S
Stepan Koltsov 已提交
407
                        TypeVariableResolvers.classTypeVariableResolver(classData.classDescriptor, context));
408 409 410
                if (valueParameterDescriptors.receiverType != null) {
                    throw new IllegalStateException();
                }
411
                constructorDescriptor.initialize(typeParameters, valueParameterDescriptors.descriptors,
412
                        resolveVisibilityFromPsiModifiers(psiConstructor), isStatic);
413
                constructorDescriptor.setReturnType(classData.classDescriptor.getDefaultType());
414
                classData.classDescriptor.addConstructor(constructorDescriptor, null);
415
                semanticServices.getTrace().record(BindingContext.CONSTRUCTOR, psiConstructor, constructorDescriptor);
416
            }
A
Andrey Breslav 已提交
417
        }
418

419 420 421 422
        MutableClassDescriptorLite classObject = createClassObjectDescriptor(classData.classDescriptor, psiClass);
        if (classObject != null) {
            classData.classDescriptor.setClassObjectDescriptor(classObject);
        }
423

424
        semanticServices.getTrace().record(BindingContext.CLASS, psiClass, classData.classDescriptor);
425

426
        return classData;
427 428
    }

429
    static void checkPsiClassIsNotJet(PsiClass psiClass) {
430
        if (psiClass instanceof JetJavaMirrorMarker) {
S
Stepan Koltsov 已提交
431
            throw new IllegalStateException("trying to resolve fake jet PsiClass as regular PsiClass: " + psiClass);
432 433 434
        }
    }

435 436 437 438 439 440 441 442 443 444
    @Nullable
    private PsiClass getInnerClassClassObject(@NotNull PsiClass outer) {
        for (PsiClass inner : outer.getInnerClasses()) {
            if (inner.getName().equals(JvmAbi.CLASS_OBJECT_CLASS_NAME)) {
                return inner;
            }
        }
        return null;
    }

445 446 447 448 449 450
    /**
     * TODO
     * @see #createJavaNamespaceDescriptor(com.intellij.psi.PsiClass)
     */
    @Nullable
    private MutableClassDescriptorLite createClassObjectDescriptor(@NotNull ClassDescriptor containing, @NotNull PsiClass psiClass) {
451 452 453 454 455
        PsiClass classObjectPsiClass = getInnerClassClassObject(psiClass);
        if (classObjectPsiClass == null) {
            return null;
        }

456 457
        checkPsiClassIsNotJet(psiClass);

458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477
        ResolverBinaryClassData classData = new ResolverBinaryClassData();
        classData.kotlin = true;
        classData.classDescriptor = new MutableClassDescriptorLite(containing, ClassKind.OBJECT);

        classDescriptorCache.put(classObjectPsiClass.getQualifiedName(), classData);

        classData.classDescriptor.setSupertypes(getSupertypes(new PsiClassWrapper(classObjectPsiClass), classData.classDescriptor, new ArrayList<TypeParameterDescriptor>(0)));
        classData.classDescriptor.setName(JetPsiUtil.NO_NAME_PROVIDED); // TODO
        classData.classDescriptor.setModality(Modality.FINAL);
        classData.classDescriptor.setTypeParameterDescriptors(new ArrayList<TypeParameterDescriptor>(0));
        classData.classDescriptor.createTypeConstructor();
        classData.classDescriptor.setScopeForMemberLookup(new JavaClassMembersScope(classData.classDescriptor, classObjectPsiClass, semanticServices, false));

        // TODO: wrong: class objects do not need visible constructors
        ConstructorDescriptorImpl constructor = new ConstructorDescriptorImpl(classData.classDescriptor, new ArrayList<AnnotationDescriptor>(0), true);
        constructor.setReturnType(classData.classDescriptor.getDefaultType());
        constructor.initialize(new ArrayList<TypeParameterDescriptor>(0), new ArrayList<ValueParameterDescriptor>(0), Visibility.PUBLIC);

        classData.classDescriptor.addConstructor(constructor, null);
        return classData.classDescriptor;
478 479
    }

480
    private List<TypeParameterDescriptorInitialization> createUninitializedClassTypeParameters(PsiClass psiClass, ResolverBinaryClassData classData, TypeVariableResolver typeVariableResolver) {
481 482 483 484 485
        JetClassAnnotation jetClassAnnotation = JetClassAnnotation.get(psiClass);
        classData.kotlin = jetClassAnnotation.isDefined();
        
        if (jetClassAnnotation.signature().length() > 0) {
            return resolveClassTypeParametersFromJetSignature(
S
Stepan Koltsov 已提交
486
                    jetClassAnnotation.signature(), psiClass, classData.classDescriptor);
487
        }
488

489
        return makeUninitializedTypeParameters(classData.classDescriptor, psiClass.getTypeParameters());
490 491
    }

492
    @NotNull
493
    private PsiTypeParameter getPsiTypeParameterByName(PsiTypeParameterListOwner clazz, String name) {
494 495 496 497 498
        for (PsiTypeParameter typeParameter : clazz.getTypeParameters()) {
            if (typeParameter.getName().equals(name)) {
                return typeParameter; 
            }
        }
499 500 501
        throw new IllegalStateException("PsiTypeParameter '" + name + "' is not found");
    }

S
Stepan Koltsov 已提交
502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518

    // cache
    protected ClassDescriptor javaLangObject;

    @NotNull
    private ClassDescriptor getJavaLangObject() {
        if (javaLangObject == null) {
            javaLangObject = resolveClass("java.lang.Object");
        }
        return javaLangObject;
    }

    private boolean isJavaLangObject(JetType type) {
        return type.getConstructor().getDeclarationDescriptor() == getJavaLangObject();
    }


519
    private abstract class JetSignatureTypeParameterVisitor extends JetSignatureExceptionsAdapter {
520 521

        @NotNull
522
        private final PsiTypeParameterListOwner psiOwner;
523
        @NotNull
524
        private final String name;
525
        @NotNull
526
        private final TypeVariableResolver typeVariableResolver;
527 528
        @NotNull
        private final TypeParameterDescriptor typeParameterDescriptor;
529

530 531
        protected JetSignatureTypeParameterVisitor(PsiTypeParameterListOwner psiOwner,
                String name, TypeVariableResolver typeVariableResolver, TypeParameterDescriptor typeParameterDescriptor)
532
        {
533 534 535
            if (name.isEmpty()) {
                throw new IllegalStateException();
            }
536

537 538
            this.psiOwner = psiOwner;
            this.name = name;
539
            this.typeVariableResolver = typeVariableResolver;
540
            this.typeParameterDescriptor = typeParameterDescriptor;
541 542 543 544
        }

        List<JetType> upperBounds = new ArrayList<JetType>();
        List<JetType> lowerBounds = new ArrayList<JetType>();
S
Stepan Koltsov 已提交
545
        
546 547
        @Override
        public JetSignatureVisitor visitClassBound() {
548
            return new JetTypeJetSignatureReader(semanticServices, JetStandardLibrary.getInstance(), typeVariableResolver) {
549 550
                @Override
                protected void done(@NotNull JetType jetType) {
S
Stepan Koltsov 已提交
551 552 553
                    if (isJavaLangObject(jetType)) {
                        return;
                    }
554 555 556 557 558 559 560
                    upperBounds.add(jetType);
                }
            };
        }

        @Override
        public JetSignatureVisitor visitInterfaceBound() {
561
            return new JetTypeJetSignatureReader(semanticServices, JetStandardLibrary.getInstance(), typeVariableResolver) {
562 563 564 565 566 567 568 569 570
                @Override
                protected void done(@NotNull JetType jetType) {
                    upperBounds.add(jetType);
                }
            };
        }

        @Override
        public void visitFormalTypeParameterEnd() {
571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589
            PsiTypeParameter psiTypeParameter = getPsiTypeParameterByName(psiOwner, name);
            TypeParameterDescriptorInitialization typeParameterDescriptorInitialization = new TypeParameterDescriptorInitialization(typeParameterDescriptor, psiTypeParameter, upperBounds, lowerBounds);
            done(typeParameterDescriptorInitialization);
        }
        
        protected abstract void done(@NotNull TypeParameterDescriptorInitialization typeParameterDescriptor);
    }

    private class JetSignatureTypeParametersVisitor extends JetSignatureExceptionsAdapter {
        @NotNull
        private final DeclarationDescriptor containingDeclaration;
        @NotNull
        private final PsiTypeParameterListOwner psiOwner;

        private final List<TypeParameterDescriptor> previousTypeParameters = new ArrayList<TypeParameterDescriptor>();
        // note changes state in this method
        private final TypeVariableResolver typeVariableResolver;


S
Stepan Koltsov 已提交
590
        private JetSignatureTypeParametersVisitor(@NotNull DeclarationDescriptor containingDeclaration, @NotNull PsiTypeParameterListOwner psiOwner, @NotNull String context) {
591 592 593
            this.containingDeclaration = containingDeclaration;
            this.psiOwner = psiOwner;

S
Stepan Koltsov 已提交
594 595 596 597
            this.typeVariableResolver = TypeVariableResolvers.typeVariableResolverFromTypeParameters(
                    previousTypeParameters,
                    containingDeclaration,
                    context);
598 599 600 601 602 603 604 605 606
        }

        private int formalTypeParameterIndex = 0;


        List<TypeParameterDescriptorInitialization> r = new ArrayList<TypeParameterDescriptorInitialization>();

        @Override
        public JetSignatureVisitor visitFormalTypeParameter(final String name, final TypeInfoVariance variance, boolean reified) {
607 608 609
            TypeParameterDescriptor typeParameter = TypeParameterDescriptor.createForFurtherModification(
                    containingDeclaration,
                    Collections.<AnnotationDescriptor>emptyList(), // TODO: wrong
610
                    reified,
611 612
                    JetSignatureUtils.translateVariance(variance),
                    name,
613 614 615 616 617 618 619 620 621 622 623
                    formalTypeParameterIndex++);

            previousTypeParameters.add(typeParameter);

            return new JetSignatureTypeParameterVisitor(psiOwner, name, typeVariableResolver, typeParameter) {
                @Override
                protected void done(@NotNull TypeParameterDescriptorInitialization typeParameterDescriptor) {
                    r.add(typeParameterDescriptor);
                    previousTypeParameters.add(typeParameterDescriptor.descriptor);
                }
            };
624
        }
625 626
    }

627
    /**
S
Stepan Koltsov 已提交
628
     * @see #resolveMethodTypeParametersFromJetSignature(String, com.intellij.psi.PsiMethod, org.jetbrains.jet.lang.descriptors.DeclarationDescriptor)
629
     */
S
Stepan Koltsov 已提交
630 631 632 633
    private List<TypeParameterDescriptorInitialization> resolveClassTypeParametersFromJetSignature(String jetSignature,
            final PsiClass clazz, final ClassDescriptor classDescriptor) {
        String context = "class " + clazz.getQualifiedName();
        JetSignatureTypeParametersVisitor jetSignatureTypeParametersVisitor = new JetSignatureTypeParametersVisitor(classDescriptor, clazz, context) {
634 635 636 637 638 639 640 641 642 643 644
            @Override
            public JetSignatureVisitor visitSuperclass() {
                // TODO
                return new JetSignatureAdapter();
            }

            @Override
            public JetSignatureVisitor visitInterface() {
                // TODO
                return new JetSignatureAdapter();
            }
645 646 647
        };
        new JetSignatureReader(jetSignature).accept(jetSignatureTypeParametersVisitor);
        return jetSignatureTypeParametersVisitor.r;
648 649
    }

S
Stepan Koltsov 已提交
650
    private ClassOrNamespaceDescriptor resolveParentDescriptor(PsiClass psiClass) {
651 652
        PsiClass containingClass = psiClass.getContainingClass();
        if (containingClass != null) {
653
            return resolveClass(containingClass);
654
        }
655 656 657 658 659 660 661 662 663 664 665

        return resolveNamespace(packageNameOfClass(psiClass.getQualifiedName()));
    }

    private static String packageNameOfClass(@NotNull String qualifiedName) {
        int lastDot = qualifiedName.lastIndexOf('.');
        if (lastDot < 0) {
            return "";
        } else {
            return qualifiedName.substring(0, lastDot);
        }
666 667
    }

668 669
    private List<TypeParameterDescriptorInitialization> makeUninitializedTypeParameters(@NotNull DeclarationDescriptor containingDeclaration, @NotNull PsiTypeParameter[] typeParameters) {
        List<TypeParameterDescriptorInitialization> result = Lists.newArrayList();
670
        for (PsiTypeParameter typeParameter : typeParameters) {
671
            TypeParameterDescriptorInitialization typeParameterDescriptor = makeUninitializedTypeParameter(containingDeclaration, typeParameter);
A
Andrey Breslav 已提交
672
            result.add(typeParameterDescriptor);
673 674 675 676
        }
        return result;
    }

677
    @NotNull
678
    private TypeParameterDescriptorInitialization makeUninitializedTypeParameter(@NotNull DeclarationDescriptor containingDeclaration, @NotNull PsiTypeParameter psiTypeParameter) {
679
        TypeParameterDescriptor typeParameterDescriptor = TypeParameterDescriptor.createForFurtherModification(
680
                containingDeclaration,
A
Andrey Breslav 已提交
681
                Collections.<AnnotationDescriptor>emptyList(), // TODO
A
Andrey Breslav 已提交
682
                false,
683
                Variance.INVARIANT,
684 685
                psiTypeParameter.getName(),
                psiTypeParameter.getIndex()
686
        );
687
        return new TypeParameterDescriptorInitialization(typeParameterDescriptor, psiTypeParameter);
688 689
    }

690
    private void initializeTypeParameter(TypeParameterDescriptorInitialization typeParameter, TypeVariableResolver typeVariableByPsiResolver) {
691 692 693
        TypeParameterDescriptor typeParameterDescriptor = typeParameter.descriptor;
        if (typeParameter.origin == TypeParameterDescriptorOrigin.KOTLIN) {
            List<?> upperBounds = typeParameter.upperBoundsForKotlin;
694 695 696
            if (upperBounds.size() == 0){
                typeParameterDescriptor.addUpperBound(JetStandardClasses.getNullableAnyType());
            } else {
697
                for (JetType upperBound : typeParameter.upperBoundsForKotlin) {
698 699 700 701 702
                    typeParameterDescriptor.addUpperBound(upperBound);
                }
            }

            // TODO: lower bounds
703
        } else {
704
            PsiClassType[] referencedTypes = typeParameter.psiTypeParameter.getExtendsList().getReferencedTypes();
705 706 707 708
            if (referencedTypes.length == 0){
                typeParameterDescriptor.addUpperBound(JetStandardClasses.getNullableAnyType());
            }
            else if (referencedTypes.length == 1) {
709
                typeParameterDescriptor.addUpperBound(semanticServices.getTypeTransformer().transformToType(referencedTypes[0], typeVariableByPsiResolver));
710 711 712
            }
            else {
                for (PsiClassType referencedType : referencedTypes) {
713
                    typeParameterDescriptor.addUpperBound(semanticServices.getTypeTransformer().transformToType(referencedType, typeVariableByPsiResolver));
714
                }
A
Andrey Breslav 已提交
715 716
            }
        }
717
        typeParameterDescriptor.setInitialized();
718 719
    }

S
Stepan Koltsov 已提交
720
    private void initializeTypeParameters(List<TypeParameterDescriptorInitialization> typeParametersInitialization, @NotNull DeclarationDescriptor typeParametersOwner, @NotNull String context) {
721
        List<TypeParameterDescriptor> prevTypeParameters = new ArrayList<TypeParameterDescriptor>();
722
        for (TypeParameterDescriptorInitialization psiTypeParameter : typeParametersInitialization) {
723
            prevTypeParameters.add(psiTypeParameter.descriptor);
S
Stepan Koltsov 已提交
724
            initializeTypeParameter(psiTypeParameter, TypeVariableResolvers.typeVariableResolverFromTypeParameters(prevTypeParameters, typeParametersOwner, context));
S
Stepan Koltsov 已提交
725
        }
A
Andrey Breslav 已提交
726 727
    }

728
    private Collection<JetType> getSupertypes(PsiClassWrapper psiClass, ClassDescriptor classDescriptor, List<TypeParameterDescriptor> typeParameters) {
729 730
        final List<JetType> result = new ArrayList<JetType>();

S
Stepan Koltsov 已提交
731 732
        String context = "class " + psiClass.getQualifiedName();

733
        if (psiClass.getJetClass().signature().length() > 0) {
S
Stepan Koltsov 已提交
734
            final TypeVariableResolver typeVariableResolver = TypeVariableResolvers.typeVariableResolverFromTypeParameters(typeParameters, classDescriptor, context);
735 736 737 738 739 740 741 742 743 744
            
            new JetSignatureReader(psiClass.getJetClass().signature()).accept(new JetSignatureExceptionsAdapter() {
                @Override
                public JetSignatureVisitor visitFormalTypeParameter(String name, TypeInfoVariance variance, boolean reified) {
                    // TODO: collect
                    return new JetSignatureAdapter();
                }

                @Override
                public JetSignatureVisitor visitSuperclass() {
745
                    return new JetTypeJetSignatureReader(semanticServices, JetStandardLibrary.getInstance(), typeVariableResolver) {
746 747 748 749 750 751 752 753 754 755 756
                        @Override
                        protected void done(@NotNull JetType jetType) {
                            if (!jetType.equals(JetStandardClasses.getAnyType())) {
                                result.add(jetType);
                            }
                        }
                    };
                }

                @Override
                public JetSignatureVisitor visitInterface() {
S
Stepan Koltsov 已提交
757
                    return visitSuperclass();
758 759 760
                }
            });
        } else {
S
Stepan Koltsov 已提交
761 762 763
            TypeVariableResolver typeVariableResolverForSupertypes = TypeVariableResolvers.typeVariableResolverFromTypeParameters(typeParameters, classDescriptor, context);
            transformSupertypeList(result, psiClass.getPsiClass().getExtendsListTypes(), typeVariableResolverForSupertypes, classDescriptor.getKind() == ClassKind.ANNOTATION_CLASS);
            transformSupertypeList(result, psiClass.getPsiClass().getImplementsListTypes(), typeVariableResolverForSupertypes, classDescriptor.getKind() == ClassKind.ANNOTATION_CLASS);
764
        }
765 766 767 768 769 770 771
        
        for (JetType supertype : result) {
            if (ErrorUtils.isErrorType(supertype)) {
                semanticServices.getTrace().record(BindingContext.INCOMPLETE_HIERARCHY, classDescriptor);
            }
        }
        
772 773 774
        if (result.isEmpty()) {
            result.add(JetStandardClasses.getAnyType());
        }
775 776 777
        return result;
    }

778
    private void transformSupertypeList(List<JetType> result, PsiClassType[] extendsListTypes, TypeVariableResolver typeVariableResolver, boolean annotation) {
779
        for (PsiClassType type : extendsListTypes) {
780 781 782 783
            PsiClass resolved = type.resolve();
            if (resolved != null && resolved.getQualifiedName().equals(JvmStdlibNames.JET_OBJECT.getFqName())) {
                continue;
            }
784 785 786
            if (annotation && resolved.getQualifiedName().equals("java.lang.annotation.Annotation")) {
                continue;
            }
787
            
788
            JetType transform = semanticServices.getTypeTransformer().transformToType(type, typeVariableResolver);
789 790 791 792 793

            result.add(TypeUtils.makeNotNullable(transform));
        }
    }

794 795
    @Nullable
    public NamespaceDescriptor resolveNamespace(@NotNull String qualifiedName) {
796 797 798 799 800 801
        // First, let's check that there is no Kotlin package:
        NamespaceDescriptor kotlinNamespaceDescriptor = semanticServices.getKotlinNamespaceDescriptor(qualifiedName);
        if (kotlinNamespaceDescriptor != null) {
            return kotlinNamespaceDescriptor;
        }

802
        PsiPackage psiPackage = findPackage(qualifiedName);
803
        if (psiPackage == null) {
804
            PsiClass psiClass = findClass(qualifiedName);
A
Andrey Breslav 已提交
805 806
            if (psiClass == null) return null;
            return resolveNamespace(psiClass);
807 808 809 810
        }
        return resolveNamespace(psiPackage);
    }

811 812
    @Nullable
    public PsiClass findClass(@NotNull String qualifiedName) {
813
        PsiClass original = javaFacade.findClass(qualifiedName, javaSearchScope);
M
Maxim Shafirov 已提交
814
        PsiClass altClass = altClassFinder.findClass(qualifiedName);
815
        PsiClass result = original;
816 817 818 819 820
        if (altClass != null) {
            if (altClass instanceof ClsClassImpl) {
                altClass.putUserData(ClsClassImpl.DELEGATE_KEY, original);
            }

821 822 823 824 825 826
            result = altClass;
        }

        if (result != null) {
            String actualQualifiedName = result.getQualifiedName();
            if (!actualQualifiedName.equals(qualifiedName)) {
827
//                throw new IllegalStateException("requested " + qualifiedName + ", got " + actualQualifiedName);
828
            }
829
        }
830 831

        return result;
832 833
    }

834
    /*package*/ PsiPackage findPackage(String qualifiedName) {
835 836 837
        return javaFacade.findPackage(qualifiedName);
    }

838
    private NamespaceDescriptor resolveNamespace(@NotNull PsiPackage psiPackage) {
839 840 841 842 843
        ResolverNamespaceData namespaceData = namespaceDescriptorCache.get(psiPackage);
        if (namespaceData == null) {
            namespaceData = createJavaNamespaceDescriptor(psiPackage);
            namespaceDescriptorCache.put(psiPackage, namespaceData);
            namespaceDescriptorCacheByFqn.put(psiPackage.getQualifiedName(), namespaceData);
844
        }
845
        return namespaceData.namespaceDescriptor;
846 847
    }

A
Andrey Breslav 已提交
848
    private NamespaceDescriptor resolveNamespace(@NotNull PsiClass psiClass) {
849 850 851 852 853
        ResolverNamespaceData namespaceData = namespaceDescriptorCache.get(psiClass);
        if (namespaceData == null) {
            namespaceData = createJavaNamespaceDescriptor(psiClass);
            namespaceDescriptorCache.put(psiClass, namespaceData);
            namespaceDescriptorCacheByFqn.put(psiClass.getQualifiedName(), namespaceData);
A
Andrey Breslav 已提交
854
        }
855
        return namespaceData.namespaceDescriptor;
A
Andrey Breslav 已提交
856 857
    }

858 859
    private ResolverNamespaceData createJavaNamespaceDescriptor(@NotNull PsiPackage psiPackage) {
        ResolverNamespaceData namespaceData = new ResolverNamespaceData();
860
        String name = psiPackage.getName();
861
        namespaceData.namespaceDescriptor = new JavaNamespaceDescriptor(
862
                resolveParentDescriptor(psiPackage),
A
Andrey Breslav 已提交
863
                Collections.<AnnotationDescriptor>emptyList(), // TODO
864
                name == null ? JAVA_ROOT : name,
865 866
                name == null ? JAVA_ROOT : psiPackage.getQualifiedName(),
                true
867
        );
868

869 870 871 872 873
        namespaceData.namespaceDescriptor.setMemberScope(new JavaPackageScope(psiPackage.getQualifiedName(), namespaceData.namespaceDescriptor, semanticServices));
        semanticServices.getTrace().record(BindingContext.NAMESPACE, psiPackage, namespaceData.namespaceDescriptor);
        // TODO: hack
        namespaceData.kotlin = true;
        return namespaceData;
874 875
    }

876 877 878 879 880 881 882 883
    private DeclarationDescriptor resolveParentDescriptor(@NotNull PsiPackage psiPackage) {
        PsiPackage parentPackage = psiPackage.getParentPackage();
        if (parentPackage == null) {
            return null;
        }
        return resolveNamespace(parentPackage);
    }

884 885 886 887
    /**
     * TODO
     * @see #createClassObjectDescriptor(org.jetbrains.jet.lang.descriptors.ClassDescriptor, com.intellij.psi.PsiClass)
     */
888
    private ResolverNamespaceData createJavaNamespaceDescriptor(@NotNull final PsiClass psiClass) {
889 890 891

        checkPsiClassIsNotJet(psiClass);

892 893
        ResolverNamespaceData namespaceData = new ResolverNamespaceData();
        namespaceData.namespaceDescriptor = new JavaNamespaceDescriptor(
894
                resolveParentDescriptor(psiClass),
A
Andrey Breslav 已提交
895
                Collections.<AnnotationDescriptor>emptyList(), // TODO
896
                psiClass.getName(),
897 898
                psiClass.getQualifiedName(),
                false
899
        );
900 901 902
        namespaceData.namespaceDescriptor.setMemberScope(new JavaClassMembersScope(namespaceData.namespaceDescriptor, psiClass, semanticServices, true));
        semanticServices.getTrace().record(BindingContext.NAMESPACE, psiClass, namespaceData.namespaceDescriptor);
        return namespaceData;
903
    }
904 905 906 907 908 909 910 911 912 913
    
    private static class ValueParameterDescriptors {
        private final JetType receiverType;
        private final List<ValueParameterDescriptor> descriptors;

        private ValueParameterDescriptors(@Nullable JetType receiverType, List<ValueParameterDescriptor> descriptors) {
            this.receiverType = receiverType;
            this.descriptors = descriptors;
        }
    }
914

915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947
    private enum JvmMethodParameterKind {
        REGULAR,
        RECEIVER,
        TYPE_INFO,
    }
    
    private static class JvmMethodParameterMeaning {
        private final JvmMethodParameterKind kind;
        private final JetType receiverType;
        private final ValueParameterDescriptor valueParameterDescriptor;
        private final Object typeInfo;

        private JvmMethodParameterMeaning(JvmMethodParameterKind kind, JetType receiverType, ValueParameterDescriptor valueParameterDescriptor, Object typeInfo) {
            this.kind = kind;
            this.receiverType = receiverType;
            this.valueParameterDescriptor = valueParameterDescriptor;
            this.typeInfo = typeInfo;
        }
        
        public static JvmMethodParameterMeaning receiver(@NotNull JetType receiverType) {
            return new JvmMethodParameterMeaning(JvmMethodParameterKind.RECEIVER, receiverType, null, null);
        }
        
        public static JvmMethodParameterMeaning regular(@NotNull ValueParameterDescriptor valueParameterDescriptor) {
            return new JvmMethodParameterMeaning(JvmMethodParameterKind.REGULAR, null, valueParameterDescriptor, null);
        }
        
        public static JvmMethodParameterMeaning typeInfo(@NotNull Object typeInfo) {
            return new JvmMethodParameterMeaning(JvmMethodParameterKind.TYPE_INFO, null, null, typeInfo);
        }
    }

    @NotNull
948
    private JvmMethodParameterMeaning resolveParameterDescriptor(DeclarationDescriptor containingDeclaration, int i,
949 950 951 952 953 954 955
            PsiParameterWrapper parameter, TypeVariableResolver typeVariableResolver) {

        if (parameter.getJetTypeParameter().isDefined()) {
            return JvmMethodParameterMeaning.typeInfo(new Object());
        }

        PsiType psiType = parameter.getPsiParameter().getType();
956

957 958
        // TODO: must be very slow, make it lazy?
        String name = parameter.getPsiParameter().getName() != null ? parameter.getPsiParameter().getName() : "p" + i;
959

960 961
        if (parameter.getJetValueParameter().name().length() > 0) {
            name = parameter.getJetValueParameter().name();
962 963
        }
        
964 965 966 967
        String typeFromAnnotation = parameter.getJetValueParameter().type();
        boolean receiver = parameter.getJetValueParameter().receiver();
        boolean hasDefaultValue = parameter.getJetValueParameter().hasDefaultValue();

S
Stepan Koltsov 已提交
968
        JetType outType;
969
        if (typeFromAnnotation.length() > 0) {
970
            outType = semanticServices.getTypeTransformer().transformToType(typeFromAnnotation, typeVariableResolver);
S
Stepan Koltsov 已提交
971
        } else {
972
            outType = semanticServices.getTypeTransformer().transformToType(psiType, typeVariableResolver);
S
Stepan Koltsov 已提交
973
        }
974 975 976

        JetType varargElementType;
        if (psiType instanceof PsiEllipsisType) {
977
            varargElementType = JetStandardLibrary.getInstance().getArrayElementType(outType);
978 979 980 981
        } else {
            varargElementType = null;
        }

982 983 984
        if (receiver) {
            return JvmMethodParameterMeaning.receiver(outType);
        } else {
S
Stepan Koltsov 已提交
985 986 987 988 989 990 991 992 993

            JetType transformedType;
            if (parameter.getJetValueParameter().nullable()) {
                transformedType = TypeUtils.makeNullableAsSpecified(outType, parameter.getJetValueParameter().nullable());
            } else if (parameter.getPsiParameter().getModifierList().findAnnotation(JvmAbi.JETBRAINS_NOT_NULL_ANNOTATION.getFqName()) != null) {
                transformedType = TypeUtils.makeNullableAsSpecified(outType, false);
            } else {
                transformedType = outType;
            }
994 995 996 997 998
            return JvmMethodParameterMeaning.regular(new ValueParameterDescriptorImpl(
                    containingDeclaration,
                    i,
                    Collections.<AnnotationDescriptor>emptyList(), // TODO
                    name,
S
Stepan Koltsov 已提交
999
                    false,
S
Stepan Koltsov 已提交
1000
                    transformedType,
1001
                    hasDefaultValue,
1002 1003 1004
                    varargElementType
            ));
        }
1005 1006
    }

S
Stepan Koltsov 已提交
1007
    public Set<VariableDescriptor> resolveFieldGroupByName(@NotNull ClassOrNamespaceDescriptor owner, @NotNull PsiClass psiClass, String fieldName, boolean staticMembers) {
1008
        ResolverScopeData scopeData = getResolverScopeData(owner, new PsiClassWrapper(psiClass));
1009

1010 1011 1012
        NamedMembers namedMembers = scopeData.namedMembersMap.get(fieldName);
        if (namedMembers == null) {
            return Collections.emptySet();
1013
        }
1014
        Set<VariableDescriptor> r = new HashSet<VariableDescriptor>();
1015

S
Stepan Koltsov 已提交
1016
        resolveNamedGroupProperties(owner, scopeData, staticMembers, namedMembers, fieldName, "class or namespace " + psiClass.getQualifiedName());
1017

1018 1019 1020 1021 1022 1023 1024
        r.addAll(namedMembers.propertyDescriptors);

        for (JetType supertype : getSupertypes(scopeData)) {
            r.addAll(supertype.getMemberScope().getProperties(fieldName));
        }

        return r;
1025 1026 1027
    }
    
    @NotNull
S
Stepan Koltsov 已提交
1028
    public Set<VariableDescriptor> resolveFieldGroup(@NotNull ClassOrNamespaceDescriptor owner, @NotNull PsiClass psiClass, boolean staticMembers) {
1029

1030 1031 1032 1033 1034 1035
        ResolverScopeData scopeData = getResolverScopeData(owner, new PsiClassWrapper(psiClass));
        
        Set<VariableDescriptor> descriptors = Sets.newHashSet();
        Map<String, NamedMembers> membersForProperties = scopeData.namedMembersMap;
        for (Map.Entry<String, NamedMembers> entry : membersForProperties.entrySet()) {
            NamedMembers namedMembers = entry.getValue();
1036
            if (namedMembers.propertyAccessors == null) {
1037 1038 1039 1040
                continue;
            }
            
            String propertyName = entry.getKey();
1041

S
Stepan Koltsov 已提交
1042
            resolveNamedGroupProperties(owner, scopeData, staticMembers, namedMembers, propertyName, "class or namespace " + psiClass.getQualifiedName());
1043
            descriptors.addAll(namedMembers.propertyDescriptors);
1044
        }
1045 1046 1047 1048 1049 1050 1051 1052 1053 1054

        for (JetType supertype : getSupertypes(scopeData)) {
            for (DeclarationDescriptor descriptor : supertype.getMemberScope().getAllDescriptors()) {
                // TODO: ugly
                if (descriptor instanceof VariableDescriptor) {
                    descriptors.add((VariableDescriptor) descriptor);
                }
            }
        }
        
1055
        return descriptors;
1056
    }
1057 1058 1059 1060 1061 1062 1063
    
    private Object key(TypeSource typeSource) {
        if (typeSource == null) {
            return "";
        } else if (typeSource.getTypeString().length() > 0) {
            return typeSource.getTypeString();
        } else {
1064
            return psiTypeToKey(typeSource.getPsiType());
1065 1066
        }
    }
1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079

    private Object psiTypeToKey(PsiType psiType) {
        if (psiType instanceof PsiClassType) {
            return ((PsiClassType) psiType).getClassName();
        } else if (psiType instanceof PsiPrimitiveType) {
            return psiType.getPresentableText();
        } else if (psiType instanceof PsiArrayType) {
            return Pair.create("[", psiTypeToKey(((PsiArrayType) psiType).getComponentType()));
        } else {
            throw new IllegalStateException("" + psiType.getClass());
        }
    }

1080 1081 1082 1083 1084
    private Object propertyKeyForGrouping(PropertyAccessorData propertyAccessor) {
        Object type = key(propertyAccessor.getType());
        Object receiverType = key(propertyAccessor.getReceiverType());
        return Pair.create(type, receiverType);
    }
1085

1086 1087 1088
    private void resolveNamedGroupProperties(
            @NotNull ClassOrNamespaceDescriptor owner,
            @NotNull ResolverScopeData scopeData,
S
Stepan Koltsov 已提交
1089 1090 1091
            boolean staticMembers, @NotNull NamedMembers namedMembers, @NotNull String propertyName,
            @NotNull String context
            ) {
1092 1093 1094 1095
        if (namedMembers.propertyDescriptors != null) {
            return;
        }
        
1096
        if (namedMembers.propertyAccessors == null) {
1097 1098 1099
            namedMembers.propertyDescriptors = Collections.emptySet();
            return;
        }
1100

S
Stepan Koltsov 已提交
1101
        TypeVariableResolver typeVariableResolver = TypeVariableResolvers.classTypeVariableResolver(owner, context);
1102

1103 1104 1105 1106
        class GroupingValue {
            PropertyAccessorData getter;
            PropertyAccessorData setter;
            PropertyAccessorData field;
1107
            boolean ext;
1108
        }
1109 1110
        
        Map<Object, GroupingValue> map = new HashMap<Object, GroupingValue>();
1111

1112
        for (PropertyAccessorData propertyAccessor : namedMembers.propertyAccessors) {
1113

1114 1115 1116 1117 1118
            Object key = propertyKeyForGrouping(propertyAccessor);
            
            GroupingValue value = map.get(key);
            if (value == null) {
                value = new GroupingValue();
1119
                value.ext = propertyAccessor.getReceiverType() != null;
1120 1121
                map.put(key, value);
            }
1122

1123 1124 1125 1126
            if (value.ext != (propertyAccessor.getReceiverType() != null)) {
                throw new IllegalStateException("internal error, incorrect key");
            }

1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144
            if (propertyAccessor.isGetter()) {
                if (value.getter != null) {
                    throw new IllegalStateException("oops, duplicate key");
                }
                value.getter = propertyAccessor;
            } else if (propertyAccessor.isSetter()) {
                if (value.setter != null) {
                    throw new IllegalStateException("oops, duplicate key");
                }
                value.setter = propertyAccessor;
            } else if (propertyAccessor.isField()) {
                if (value.field != null) {
                    throw new IllegalStateException("oops, duplicate key");
                }
                value.field = propertyAccessor;
            } else {
                throw new IllegalStateException();
            }
1145
        }
1146

1147
        
1148 1149
        Set<VariableDescriptor> r = new HashSet<VariableDescriptor>(1);

1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163
        int regularProperitesCount = 0;
        for (GroupingValue members : map.values()) {
            if (!members.ext) {
                ++regularProperitesCount;
            }
        }

        for (GroupingValue members : map.values()) {

            // we cannot have more then one property with given name even if java code
            // has several fields, getters and setter of different types
            if (!members.ext && regularProperitesCount > 1) {
                continue;
            }
1164

1165
            boolean isFinal;
1166 1167 1168
            if (!scopeData.kotlin) {
                isFinal = true;
            } else if (members.setter == null && members.getter == null) {
1169 1170 1171 1172 1173 1174 1175 1176
                isFinal = false;
            } else if (members.getter != null) {
                isFinal = members.getter.getMember().isFinal();
            } else if (members.setter != null) {
                isFinal = members.setter.getMember().isFinal();
            } else {
                isFinal = false;
            }
1177

1178 1179 1180 1181 1182 1183 1184 1185 1186 1187
            PropertyAccessorData anyMember;
            if (members.getter != null) {
                anyMember = members.getter;
            } else if (members.field != null) {
                anyMember = members.field;
            } else if (members.setter != null) {
                anyMember = members.setter;
            } else {
                throw new IllegalStateException();
            }
1188

1189 1190 1191 1192 1193 1194
            boolean isVar;
            if (members.getter == null && members.setter == null) {
                isVar = !members.field.getMember().isFinal();
            } else {
                isVar = members.setter != null;
            }
1195 1196 1197 1198 1199 1200 1201 1202 1203 1204

            Modality modality;
            if (isFinal) {
                modality = Modality.FINAL;
            }
            else {
                modality = anyMember.getMember().isAbstract() ? Modality.ABSTRACT : Modality.OPEN;
            }


1205 1206 1207
            PropertyDescriptor propertyDescriptor = new PropertyDescriptor(
                    owner,
                    Collections.<AnnotationDescriptor>emptyList(),
1208
                    modality,
1209 1210 1211
                    resolveVisibilityFromPsiModifiers(anyMember.getMember().psiMember),
                    isVar,
                    false,
1212 1213
                    propertyName,
                    CallableMemberDescriptor.Kind.DECLARATION);
1214

1215 1216 1217
            PropertyGetterDescriptor getterDescriptor = null;
            PropertySetterDescriptor setterDescriptor = null;
            if (members.getter != null) {
1218
                getterDescriptor = new PropertyGetterDescriptor(propertyDescriptor, Collections.<AnnotationDescriptor>emptyList(), Modality.OPEN, Visibility.PUBLIC, true, false, CallableMemberDescriptor.Kind.DECLARATION);
1219 1220
            }
            if (members.setter != null) {
1221
                setterDescriptor = new PropertySetterDescriptor(propertyDescriptor, Collections.<AnnotationDescriptor>emptyList(), Modality.OPEN, Visibility.PUBLIC, true, false, CallableMemberDescriptor.Kind.DECLARATION);
1222
            }
1223

1224
            propertyDescriptor.initialize(getterDescriptor, setterDescriptor);
1225

S
Stepan Koltsov 已提交
1226
            List<TypeParameterDescriptor> typeParameters = new ArrayList<TypeParameterDescriptor>(0);
1227 1228 1229

            if (members.setter != null) {
                PsiMethodWrapper method = (PsiMethodWrapper) members.setter.getMember();
1230

1231
                if (anyMember == members.setter) {
S
Stepan Koltsov 已提交
1232
                    typeParameters = resolveMethodTypeParameters(method, propertyDescriptor, typeVariableResolver);
1233 1234 1235 1236 1237 1238
                }
            }
            if (members.getter != null) {
                PsiMethodWrapper method = (PsiMethodWrapper) members.getter.getMember();

                if (anyMember == members.getter) {
S
Stepan Koltsov 已提交
1239
                    typeParameters = resolveMethodTypeParameters(method, propertyDescriptor, typeVariableResolver);
1240 1241 1242
                }
            }

S
Stepan Koltsov 已提交
1243
            TypeVariableResolver typeVariableResolverForPropertyInternals = TypeVariableResolvers.typeVariableResolverFromTypeParameters(typeParameters, propertyDescriptor, "property " + propertyName + " in " + context);
1244

1245 1246 1247 1248
            JetType propertyType;
            if (anyMember.getType().getTypeString().length() > 0) {
                propertyType = semanticServices.getTypeTransformer().transformToType(anyMember.getType().getTypeString(), typeVariableResolverForPropertyInternals);
            } else {
1249
                propertyType = semanticServices.getTypeTransformer().transformToType(anyMember.getType().getPsiType(), typeVariableResolverForPropertyInternals);
S
Stepan Koltsov 已提交
1250 1251 1252
                if (anyMember.getType().getPsiNotNullOwner().getModifierList().findAnnotation(JvmAbi.JETBRAINS_NOT_NULL_ANNOTATION.getFqName()) != null) {
                    propertyType = TypeUtils.makeNullableAsSpecified(propertyType, false);
                }
1253 1254 1255 1256 1257 1258 1259 1260
            }
            
            JetType receiverType;
            if (anyMember.getReceiverType() == null) {
                receiverType = null;
            } else if (anyMember.getReceiverType().getTypeString().length() > 0) {
                receiverType = semanticServices.getTypeTransformer().transformToType(anyMember.getReceiverType().getTypeString(), typeVariableResolverForPropertyInternals);
            } else {
1261
                receiverType = semanticServices.getTypeTransformer().transformToType(anyMember.getReceiverType().getPsiType(), typeVariableResolverForPropertyInternals);
1262 1263 1264 1265 1266 1267 1268
            }

            propertyDescriptor.setType(
                    propertyType,
                    typeParameters,
                    DescriptorUtils.getExpectedThisObjectIfNeeded(owner),
                    receiverType
1269
            );
1270 1271 1272 1273
            if (getterDescriptor != null) {
                getterDescriptor.initialize(propertyType);
            }
            if (setterDescriptor != null) {
1274
                setterDescriptor.initialize(new ValueParameterDescriptorImpl(setterDescriptor, 0, Collections.<AnnotationDescriptor>emptyList(), "p0"/*TODO*/, false, propertyDescriptor.getType(), false, null));
1275 1276 1277 1278 1279
            }

            semanticServices.getTrace().record(BindingContext.VARIABLE, anyMember.getMember().psiMember, propertyDescriptor);
            
            r.add(propertyDescriptor);
1280
        }
1281

1282
        namedMembers.propertyDescriptors = r;
1283
    }
1284

1285
    private void resolveNamedGroupFunctions(ClassOrNamespaceDescriptor owner, PsiClass psiClass, TypeSubstitutor typeSubstitutorForGenericSuperclasses, NamedMembers namedMembers, String methodName, ResolverScopeData scopeData) {
1286 1287 1288
        if (namedMembers.functionDescriptors != null) {
            return;
        }
1289 1290
        
        final Set<FunctionDescriptor> functions = new HashSet<FunctionDescriptor>();
1291

1292
        Set<SimpleFunctionDescriptor> functionsFromCurrent = Sets.newHashSet();
1293
        for (PsiMethodWrapper method : namedMembers.methods) {
1294 1295
            FunctionDescriptorImpl function = resolveMethodToFunctionDescriptor(owner, psiClass,
                    method);
1296
            if (function != null) {
1297
                functionsFromCurrent.add((SimpleFunctionDescriptor) function);
1298 1299 1300 1301 1302 1303
            }
        }

        if (owner instanceof ClassDescriptor) {
            ClassDescriptor classDescriptor = (ClassDescriptor) owner;

1304
            Set<SimpleFunctionDescriptor> functionsFromSupertypes = getFunctionsFromSupertypes(scopeData, methodName);
1305

S
Stepan Koltsov 已提交
1306
            OverrideResolver.generateOverridesInFunctionGroup(methodName, functionsFromSupertypes, functionsFromCurrent, classDescriptor, new OverrideResolver.DescriptorSink() {
1307
                @Override
S
Stepan Koltsov 已提交
1308 1309
                public void addToScope(@NotNull CallableMemberDescriptor fakeOverride) {
                    functions.add((FunctionDescriptor) fakeOverride);
1310
                }
1311 1312

                @Override
S
Stepan Koltsov 已提交
1313
                public void conflict(@NotNull CallableMemberDescriptor fromSuper, @NotNull CallableMemberDescriptor fromCurrent) {
1314 1315
                    // nop
                }
1316 1317 1318 1319 1320 1321 1322 1323 1324
            });

        }

        functions.addAll(functionsFromCurrent);

        namedMembers.functionDescriptors = functions;
    }
    
1325 1326
    private Set<SimpleFunctionDescriptor> getFunctionsFromSupertypes(ResolverScopeData scopeData, String methodName) {
        Set<SimpleFunctionDescriptor> r = new HashSet<SimpleFunctionDescriptor>();
1327 1328
        for (JetType supertype : getSupertypes(scopeData)) {
            for (FunctionDescriptor function : supertype.getMemberScope().getFunctions(methodName)) {
1329
                r.add((SimpleFunctionDescriptor) function);
1330
            }
1331
        }
1332
        return r;
1333 1334
    }

1335
    private ResolverScopeData getResolverScopeData(@NotNull ClassOrNamespaceDescriptor owner, PsiClassWrapper psiClass) {
1336
        // TODO: store scopeData in Java*Scope
1337
        ResolverScopeData scopeData;
1338
        boolean staticMembers;
1339 1340
        if (owner instanceof JavaNamespaceDescriptor) {
            scopeData = namespaceDescriptorCacheByFqn.get(((JavaNamespaceDescriptor) owner).getQualifiedName());
1341
            staticMembers = true;
1342
        } else if (owner instanceof ClassDescriptor) {
1343
            scopeData = classDescriptorCache.get(psiClass.getQualifiedName());
1344
            staticMembers = false;
1345
        } else {
1346
            throw new IllegalStateException("unknown owner: " + owner.getClass().getName());
1347 1348 1349 1350 1351
        }
        if (scopeData == null) {
            throw new IllegalStateException();
        }
        
1352 1353
        if (scopeData.namedMembersMap == null) {
            scopeData.namedMembersMap = JavaDescriptorResolverHelper.getNamedMembers(psiClass, staticMembers, scopeData.kotlin);
1354 1355
        }
        
1356
        return scopeData;
1357
    }
A
Andrey Breslav 已提交
1358

A
Andrey Breslav 已提交
1359
    @NotNull
1360
    public Set<FunctionDescriptor> resolveFunctionGroup(@NotNull ClassOrNamespaceDescriptor descriptor, @NotNull PsiClass psiClass, @NotNull String methodName, boolean staticMembers) {
A
Andrey Breslav 已提交
1361

1362
        ResolverScopeData resolverScopeData = getResolverScopeData(descriptor, new PsiClassWrapper(psiClass));
A
Andrey Breslav 已提交
1363

1364 1365 1366
        Map<String, NamedMembers> namedMembersMap = resolverScopeData.namedMembersMap;

        NamedMembers namedMembers = namedMembersMap.get(methodName);
1367 1368
        if (namedMembers != null && namedMembers.methods != null) {
            TypeSubstitutor typeSubstitutor = typeSubstitutorForGenericSupertypes(resolverScopeData);
1369

1370 1371 1372
            resolveNamedGroupFunctions(descriptor, psiClass, typeSubstitutor, namedMembers, methodName, resolverScopeData);

            return namedMembers.functionDescriptors;
1373
        } else {
1374
            return Collections.emptySet();
1375
        }
1376 1377
    }

1378
    private TypeSubstitutor createSubstitutorForGenericSupertypes(@Nullable ClassDescriptor classDescriptor) {
A
Andrey Breslav 已提交
1379 1380 1381 1382 1383 1384 1385 1386 1387
        TypeSubstitutor typeSubstitutor;
        if (classDescriptor != null) {
            typeSubstitutor = TypeUtils.buildDeepSubstitutor(classDescriptor.getDefaultType());
        }
        else {
            typeSubstitutor = TypeSubstitutor.EMPTY;
        }
        return typeSubstitutor;
    }
1388

1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412
    private ValueParameterDescriptors resolveParameterDescriptors(DeclarationDescriptor containingDeclaration,
            List<PsiParameterWrapper> parameters, TypeVariableResolver typeVariableResolver) {
        List<ValueParameterDescriptor> result = new ArrayList<ValueParameterDescriptor>();
        JetType receiverType = null;
        int indexDelta = 0;
        for (int i = 0, parametersLength = parameters.size(); i < parametersLength; i++) {
            PsiParameterWrapper parameter = parameters.get(i);
            JvmMethodParameterMeaning meaning = resolveParameterDescriptor(containingDeclaration, i + indexDelta, parameter, typeVariableResolver);
            if (meaning.kind == JvmMethodParameterKind.TYPE_INFO) {
                // TODO
                --indexDelta;
            } else if (meaning.kind == JvmMethodParameterKind.REGULAR) {
                result.add(meaning.valueParameterDescriptor);
            } else if (meaning.kind == JvmMethodParameterKind.RECEIVER) {
                if (receiverType != null) {
                    throw new IllegalStateException("more then one receiver");
                }
                --indexDelta;
                receiverType = meaning.receiverType;
            }
        }
        return new ValueParameterDescriptors(receiverType, result);
    }

A
Andrey Breslav 已提交
1413
    @Nullable
S
Stepan Koltsov 已提交
1414
    private FunctionDescriptorImpl resolveMethodToFunctionDescriptor(ClassOrNamespaceDescriptor owner, final PsiClass psiClass, final PsiMethodWrapper method) {
1415

1416
        PsiType returnType = method.getReturnType();
A
Andrey Breslav 已提交
1417 1418 1419
        if (returnType == null) {
            return null;
        }
1420 1421
        ResolverScopeData scopeData = getResolverScopeData(owner, new PsiClassWrapper(psiClass));

1422 1423 1424 1425 1426 1427 1428 1429 1430
        boolean kotlin;
        if (owner instanceof JavaNamespaceDescriptor) {
            JavaNamespaceDescriptor javaNamespaceDescriptor = (JavaNamespaceDescriptor) owner;
            ResolverNamespaceData namespaceData = namespaceDescriptorCacheByFqn.get(javaNamespaceDescriptor.getQualifiedName());
            if (namespaceData == null) {
                throw new IllegalStateException("namespaceData not found by name " + javaNamespaceDescriptor.getQualifiedName());
            }
            kotlin = namespaceData.kotlin;
        } else {
1431
            ResolverBinaryClassData classData = classDescriptorCache.get(psiClass.getQualifiedName());
1432 1433 1434 1435 1436 1437
            if (classData == null) {
                throw new IllegalStateException("classData not found by name " + psiClass.getQualifiedName());
            }
            kotlin = classData.kotlin;
        }

1438
        // TODO: ugly
1439
        if (method.getJetMethod().kind() == JvmStdlibNames.JET_METHOD_KIND_PROPERTY) {
1440
            return null;
1441
        }
1442

S
Stepan Koltsov 已提交
1443 1444 1445 1446 1447 1448 1449 1450
        if (kotlin) {
            // TODO: unless maybe class explicitly extends Object
            String ownerClassName = method.getPsiMethod().getContainingClass().getQualifiedName();
            if (ownerClassName.equals("java.lang.Object")) {
                return null;
            }
        }

1451
        ClassOrNamespaceDescriptor classDescriptor;
1452
        if (scopeData instanceof ResolverBinaryClassData) {
1453
            ClassDescriptor classClassDescriptor = resolveClass(method.getPsiMethod().getContainingClass());
1454
            classDescriptor = classClassDescriptor;
1455 1456 1457
        }
        else {
            classDescriptor = resolveNamespace(method.getPsiMethod().getContainingClass());
1458
        }
1459 1460 1461
        if (classDescriptor == null) {
            return null;
        }
1462

1463
        SimpleFunctionDescriptorImpl functionDescriptorImpl = new SimpleFunctionDescriptorImpl(
A
Andrey Breslav 已提交
1464
                owner,
1465
                resolveAnnotations(method.getPsiMethod()),
1466 1467
                method.getName(),
                CallableMemberDescriptor.Kind.DECLARATION
A
Andrey Breslav 已提交
1468
        );
1469

S
Stepan Koltsov 已提交
1470 1471 1472
        String context = "method " + method.getName() + " in class " + psiClass.getQualifiedName();

        final TypeVariableResolver typeVariableResolverForParameters = TypeVariableResolvers.classTypeVariableResolver(classDescriptor, context);
1473

1474
        final List<TypeParameterDescriptor> methodTypeParameters = resolveMethodTypeParameters(method, functionDescriptorImpl, typeVariableResolverForParameters);
1475

S
Stepan Koltsov 已提交
1476
        TypeVariableResolver methodTypeVariableResolver = TypeVariableResolvers.typeVariableResolverFromTypeParameters(methodTypeParameters, functionDescriptorImpl, context);
1477 1478


1479
        ValueParameterDescriptors valueParameterDescriptors = resolveParameterDescriptors(functionDescriptorImpl, method.getParameters(), methodTypeVariableResolver);
A
Andrey Breslav 已提交
1480
        functionDescriptorImpl.initialize(
1481
                valueParameterDescriptors.receiverType,
A
Andrey Breslav 已提交
1482
                DescriptorUtils.getExpectedThisObjectIfNeeded(classDescriptor),
1483
                methodTypeParameters,
1484
                valueParameterDescriptors.descriptors,
1485
                makeReturnType(returnType, method, methodTypeVariableResolver),
1486 1487
                Modality.convertFromFlags(method.getPsiMethod().hasModifierProperty(PsiModifier.ABSTRACT), !method.isFinal()),
                resolveVisibilityFromPsiModifiers(method.getPsiMethod())
A
Andrey Breslav 已提交
1488
        );
1489
        semanticServices.getTrace().record(BindingContext.FUNCTION, method.getPsiMethod(), functionDescriptorImpl);
A
Andrey Breslav 已提交
1490
        FunctionDescriptor substitutedFunctionDescriptor = functionDescriptorImpl;
S
Stepan Koltsov 已提交
1491 1492
        if (method.getPsiMethod().getContainingClass() != psiClass && !method.isStatic()) {
            throw new IllegalStateException("non-static method in subclass");
A
Andrey Breslav 已提交
1493
        }
1494
        return (FunctionDescriptorImpl) substitutedFunctionDescriptor;
A
Andrey Breslav 已提交
1495
    }
1496

1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523
    private List<AnnotationDescriptor> resolveAnnotations(PsiModifierListOwner owner) {
        PsiAnnotation[] psiAnnotations = owner.getModifierList().getAnnotations();
        List<AnnotationDescriptor> r = Lists.newArrayListWithCapacity(psiAnnotations.length);
        for (PsiAnnotation psiAnnotation : psiAnnotations) {
            AnnotationDescriptor annotation = resolveAnnotation(psiAnnotation);
            if (annotation != null) {
                r.add(annotation);
            }
        }
        return r;
    }

    @Nullable
    private AnnotationDescriptor resolveAnnotation(PsiAnnotation psiAnnotation) {
        AnnotationDescriptor annotation = new AnnotationDescriptor();

        String qname = psiAnnotation.getQualifiedName();
        if (qname.startsWith("java.lang.annotation.") || qname.startsWith("jet.runtime.typeinfo.") || qname.equals(JvmAbi.JETBRAINS_NOT_NULL_ANNOTATION.getFqName())) {
            // TODO
            return null;
        }

        ClassDescriptor clazz = resolveClass(psiAnnotation.getQualifiedName());
        if (clazz == null) {
            return null;
        }
        annotation.setAnnotationType(clazz.getDefaultType());
1524 1525 1526 1527 1528 1529 1530 1531
        ArrayList<CompileTimeConstant<?>> valueArguments = new ArrayList<CompileTimeConstant<?>>();
        if("jet.runtime.Intrinsic".equals(psiAnnotation.getQualifiedName())) {
            // temporary hack
            valueArguments.add(new StringValue(psiAnnotation.findAttributeValue("value").getText()));
            annotation.setValueArguments(valueArguments); // TODO
        }
        else
            annotation.setValueArguments(valueArguments); // TODO
1532 1533 1534
        return annotation;
    }

1535
    public List<FunctionDescriptor> resolveMethods(@NotNull PsiClass psiClass, @NotNull ClassOrNamespaceDescriptor containingDeclaration) {
1536 1537
        ResolverScopeData scopeData = getResolverScopeData(containingDeclaration, new PsiClassWrapper(psiClass));

1538
        TypeSubstitutor substitutorForGenericSupertypes = typeSubstitutorForGenericSupertypes(scopeData);
1539

1540
        List<FunctionDescriptor> functions = new ArrayList<FunctionDescriptor>();
1541 1542 1543 1544 1545

        for (Map.Entry<String, NamedMembers> entry : scopeData.namedMembersMap.entrySet()) {
            String methodName = entry.getKey();
            NamedMembers namedMembers = entry.getValue();
            resolveNamedGroupFunctions(containingDeclaration, psiClass, substitutorForGenericSupertypes, namedMembers, methodName, scopeData);
1546
            functions.addAll(namedMembers.functionDescriptors);
1547
        }
1548

1549 1550
        return functions;
    }
1551

1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569
    private Collection<JetType> getSupertypes(ResolverScopeData scope) {
        if (scope instanceof ResolverBinaryClassData) {
            return ((ResolverBinaryClassData) scope).classDescriptor.getSupertypes();
        } else if (scope instanceof ResolverNamespaceData) {
            return Collections.emptyList();
        } else {
            throw new IllegalStateException();
        }
    }

    private TypeSubstitutor typeSubstitutorForGenericSupertypes(ResolverScopeData scopeData) {
        if (scopeData instanceof ResolverClassData) {
            return createSubstitutorForGenericSupertypes(((ResolverClassData) scopeData).getClassDescriptor());
        } else {
            return TypeSubstitutor.EMPTY;
        }
    }

1570
    private List<TypeParameterDescriptor> resolveMethodTypeParameters(
1571
            @NotNull PsiMethodWrapper method,
1572
            @NotNull DeclarationDescriptor functionDescriptor,
1573 1574
            @NotNull TypeVariableResolver classTypeVariableResolver) {

1575
        List<TypeParameterDescriptorInitialization> typeParametersIntialization;
1576
        if (method.getJetMethod().typeParameters().length() > 0) {
1577
            typeParametersIntialization = resolveMethodTypeParametersFromJetSignature(
S
Stepan Koltsov 已提交
1578
                    method.getJetMethod().typeParameters(), method.getPsiMethod(), functionDescriptor);
1579
        } else {
1580
            typeParametersIntialization = makeUninitializedTypeParameters(functionDescriptor, method.getPsiMethod().getTypeParameters());
1581
        }
1582

S
Stepan Koltsov 已提交
1583 1584
        String context = "method " + method.getName() + " in class " + method.getPsiMethod().getContainingClass().getQualifiedName();
        initializeTypeParameters(typeParametersIntialization, functionDescriptor, context);
1585 1586 1587 1588 1589 1590 1591
        
        List<TypeParameterDescriptor> typeParameters = Lists.newArrayListWithCapacity(typeParametersIntialization.size());
        
        for (TypeParameterDescriptorInitialization tpdi : typeParametersIntialization) {
            typeParameters.add(tpdi.descriptor);
        }
        
1592 1593
        return typeParameters;
    }
1594 1595

    /**
S
Stepan Koltsov 已提交
1596
     * @see #resolveClassTypeParametersFromJetSignature(String, com.intellij.psi.PsiClass, org.jetbrains.jet.lang.descriptors.ClassDescriptor)
1597
     */
S
Stepan Koltsov 已提交
1598 1599
    private List<TypeParameterDescriptorInitialization> resolveMethodTypeParametersFromJetSignature(String jetSignature,
            final PsiMethod method, final DeclarationDescriptor functionDescriptor)
1600
    {
S
Stepan Koltsov 已提交
1601 1602
        String context = "method " + method.getName() + " in class " + method.getContainingClass().getQualifiedName();
        JetSignatureTypeParametersVisitor jetSignatureTypeParametersVisitor = new JetSignatureTypeParametersVisitor(functionDescriptor, method, context);
1603 1604
        new JetSignatureReader(jetSignature).acceptFormalTypeParametersOnly(jetSignatureTypeParametersVisitor);
        return jetSignatureTypeParametersVisitor.r;
1605 1606
    }

1607 1608
    private JetType makeReturnType(PsiType returnType, PsiMethodWrapper method,
            @NotNull TypeVariableResolver typeVariableResolver) {
1609 1610 1611

        String returnTypeFromAnnotation = method.getJetMethod().returnType();

S
Stepan Koltsov 已提交
1612
        JetType transformedType;
1613
        if (returnTypeFromAnnotation.length() > 0) {
1614
            transformedType = semanticServices.getTypeTransformer().transformToType(returnTypeFromAnnotation, typeVariableResolver);
S
Stepan Koltsov 已提交
1615
        } else {
1616
            transformedType = semanticServices.getTypeTransformer().transformToType(returnType, typeVariableResolver);
S
Stepan Koltsov 已提交
1617
        }
1618 1619
        if (method.getJetMethod().returnTypeNullable()) {
            return TypeUtils.makeNullableAsSpecified(transformedType, true);
S
Stepan Koltsov 已提交
1620 1621
        } else if (method.getPsiMethod().getModifierList().findAnnotation(JvmAbi.JETBRAINS_NOT_NULL_ANNOTATION.getFqName()) != null) {
            return TypeUtils.makeNullableAsSpecified(transformedType, false);
1622 1623 1624 1625 1626
        } else {
            return transformedType;
        }
    }

1627
    private static Visibility resolveVisibilityFromPsiModifiers(PsiModifierListOwner modifierListOwner) {
1628 1629 1630 1631 1632 1633
        //TODO report error
        return modifierListOwner.hasModifierProperty(PsiModifier.PUBLIC) ? Visibility.PUBLIC :
                                        (modifierListOwner.hasModifierProperty(PsiModifier.PRIVATE) ? Visibility.PRIVATE :
                                        (modifierListOwner.hasModifierProperty(PsiModifier.PROTECTED) ? Visibility.PROTECTED : Visibility.INTERNAL));
    }

1634
    public List<ClassDescriptor> resolveInnerClasses(DeclarationDescriptor owner, PsiClass psiClass, boolean staticMembers) {
1635 1636 1637 1638
        if (staticMembers) {
            return new ArrayList<ClassDescriptor>(0);
        }

1639 1640 1641
        PsiClass[] innerPsiClasses = psiClass.getInnerClasses();
        List<ClassDescriptor> r = new ArrayList<ClassDescriptor>(innerPsiClasses.length);
        for (PsiClass innerPsiClass : innerPsiClasses) {
1642 1643 1644 1645
            if (innerPsiClass.hasModifierProperty(PsiModifier.PRIVATE)) {
                // TODO: hack against inner classes
                continue;
            }
1646 1647 1648
            if (innerPsiClass.getName().equals(JvmAbi.CLASS_OBJECT_CLASS_NAME)) {
                continue;
            }
1649 1650 1651 1652
            r.add(resolveClass(innerPsiClass));
        }
        return r;
    }
1653
}