001 /*
002 * Copyright 2010-2013 JetBrains s.r.o.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017 package org.jetbrains.jet.lang.resolve.java.scope;
018
019 import org.jetbrains.annotations.NotNull;
020 import org.jetbrains.annotations.Nullable;
021 import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
022 import org.jetbrains.jet.lang.descriptors.ConstructorDescriptor;
023 import org.jetbrains.jet.lang.resolve.java.resolver.JavaMemberResolver;
024 import org.jetbrains.jet.lang.resolve.java.structure.JavaClass;
025 import org.jetbrains.jet.lang.resolve.name.FqName;
026
027 import java.util.ArrayList;
028 import java.util.Collection;
029 import java.util.Collections;
030 import java.util.List;
031
032 import static org.jetbrains.jet.lang.resolve.java.DescriptorSearchRule.IGNORE_KOTLIN_SOURCES;
033
034 public final class JavaClassNonStaticMembersScope extends JavaClassMembersScope {
035 private Collection<ConstructorDescriptor> constructors = null;
036 private ConstructorDescriptor primaryConstructor = null;
037 @NotNull
038 private final ClassDescriptor descriptor;
039 @NotNull
040 private final JavaClass javaClass;
041 private final boolean staticMembersOfPsiClass;
042
043 public JavaClassNonStaticMembersScope(
044 @NotNull ClassDescriptor descriptor,
045 @NotNull JavaClass javaClass,
046 boolean staticMembersOfClass,
047 @NotNull JavaMemberResolver memberResolver
048 ) {
049 super(descriptor, MembersProvider.forClass(javaClass, staticMembersOfClass), memberResolver);
050 this.descriptor = descriptor;
051 this.javaClass = javaClass;
052 this.staticMembersOfPsiClass = staticMembersOfClass;
053 }
054
055
056 @NotNull
057 public Collection<ConstructorDescriptor> getConstructors() {
058 initConstructorsIfNeeded();
059 return constructors;
060 }
061
062 @Nullable
063 public ConstructorDescriptor getPrimaryConstructor() {
064 initConstructorsIfNeeded();
065 return primaryConstructor;
066 }
067
068 private void initConstructorsIfNeeded() {
069 if (constructors == null) {
070 constructors = memberResolver.resolveConstructors(javaClass, descriptor);
071
072 for (ConstructorDescriptor constructor : constructors) {
073 if (constructor.isPrimary()) {
074 if (primaryConstructor != null) {
075 throw new IllegalStateException(
076 "Class has more than one primary constructor: " + primaryConstructor + "\n" + constructor);
077 }
078 primaryConstructor = constructor;
079 }
080 }
081 }
082 }
083
084 @NotNull
085 @Override
086 protected Collection<ClassDescriptor> computeInnerClasses() {
087 if (staticMembersOfPsiClass) {
088 return Collections.emptyList();
089 }
090
091 Collection<JavaClass> innerClasses = javaClass.getInnerClasses();
092 List<ClassDescriptor> result = new ArrayList<ClassDescriptor>(innerClasses.size());
093 for (JavaClass innerClass : innerClasses) {
094 result.add(resolveInnerClass(innerClass));
095 }
096 return result;
097 }
098
099 @NotNull
100 private ClassDescriptor resolveInnerClass(@NotNull JavaClass innerClass) {
101 FqName fqName = innerClass.getFqName();
102 assert fqName != null : "Inner class has no qualified name: " + innerClass;
103 ClassDescriptor classDescriptor = memberResolver.resolveClass(fqName, IGNORE_KOTLIN_SOURCES);
104 assert classDescriptor != null : "Couldn't resolve inner class " + fqName;
105 return classDescriptor;
106 }
107 }