azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

smarter JavaClassOrPackageScope#getAllDescriptors

Browse Source
This commit is contained in:
Stepan Koltsov
2012-06-16 06:44:16 +04:00
parent a87f3e2a45
commit 010db0495b
3 changed files with 41 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/frontend.java/src/org/jetbrains/jet/lang/resolve/java/JavaClassOrPackageScope.java
+27 -8
View File
@@ -30,7 +30,6 @@ import org.jetbrains.jet.lang.resolve.scopes.DescriptorPredicateUtils;
import org.jetbrains.jet.lang.resolve.scopes.JetScopeImpl;
import java.util.Collection;
import java.util.Set;
/**
* @author Stepan Koltsov
@@ -74,13 +73,28 @@ public abstract class JavaClassOrPackageScope extends JetScopeImpl {
@Override
public Collection<DeclarationDescriptor> getAllDescriptors(@NotNull DescriptorPredicate predicate) {
if (allDescriptors == null) {
allDescriptors = Sets.newHashSet();
if (!predicate.includeAll()) {
return computeAllDescriptors(predicate);
}
if (resolverScopeData.psiClass != null) {
allDescriptors.addAll(semanticServices.getDescriptorResolver().resolveMethods(resolverScopeData, NamePredicate.all()));
allDescriptors = computeAllDescriptors(DescriptorPredicate.all());
}
allDescriptors.addAll(semanticServices.getDescriptorResolver().resolveFieldGroup(resolverScopeData, NamePredicate.all()));
return DescriptorPredicateUtils.filter(allDescriptors, predicate);
}
private Collection<DeclarationDescriptor> computeAllDescriptors(@NotNull DescriptorPredicate predicate) {
Collection<DeclarationDescriptor> allDescriptors = Sets.newHashSet();
if (resolverScopeData.psiClass != null) {
if (predicate.includeKind(DescriptorPredicate.DescriptorKind.CALLABLE_MEMBER)) {
// TODO: filter by extension here
allDescriptors.addAll(semanticServices.getDescriptorResolver().resolveMethods(resolverScopeData, predicate.asNamePredicate()));
allDescriptors.addAll(semanticServices.getDescriptorResolver().resolveFieldGroup(resolverScopeData, predicate.asNamePredicate()));
}
if (predicate.includeKind(DescriptorPredicate.DescriptorKind.CLASS)) {
// TODO: Trying to hack the situation when we produce namespace descriptor for java class and still want to see inner classes
if (getContainingDeclaration() instanceof JavaNamespaceDescriptor) {
allDescriptors.addAll(semanticServices.getDescriptorResolver().resolveInnerClasses(
@@ -92,11 +106,13 @@ public abstract class JavaClassOrPackageScope extends JetScopeImpl {
resolverScopeData.staticMembers));
}
}
}
if (resolverScopeData.psiPackage != null) {
boolean isKotlinNamespace = semanticServices.getKotlinNamespaceDescriptor(resolverScopeData.fqName) != null;
final JavaDescriptorResolver descriptorResolver = semanticServices.getDescriptorResolver();
if (resolverScopeData.psiPackage != null) {
boolean isKotlinNamespace = semanticServices.getKotlinNamespaceDescriptor(resolverScopeData.fqName) != null;
final JavaDescriptorResolver descriptorResolver = semanticServices.getDescriptorResolver();
if (predicate.includeKind(DescriptorPredicate.DescriptorKind.NAMESPACE)) {
for (PsiPackage psiSubPackage : resolverScopeData.psiPackage.getSubPackages()) {
NamespaceDescriptor childNs = descriptorResolver.resolveNamespace(
new FqName(psiSubPackage.getQualifiedName()), DescriptorSearchRule.IGNORE_IF_FOUND_IN_KOTLIN);
@@ -104,7 +120,9 @@ public abstract class JavaClassOrPackageScope extends JetScopeImpl {
allDescriptors.add(childNs);
}
}
}
if (predicate.includeKind(DescriptorPredicate.DescriptorKind.CLASS)) {
for (PsiClass psiClass : resolverScopeData.psiPackage.getClasses()) {
if (isKotlinNamespace && JvmAbi.PACKAGE_CLASS.equals(psiClass.getName())) {
continue;
@@ -130,6 +148,7 @@ public abstract class JavaClassOrPackageScope extends JetScopeImpl {
}
}
// filter again, because previously filter wasn't accurate
return DescriptorPredicateUtils.filter(allDescriptors, predicate);
}
}
compiler/frontend.java/src/org/jetbrains/jet/lang/resolve/java/JavaDescriptorResolver.java
+4 -4
View File
@@ -1424,7 +1424,7 @@ public class JavaDescriptorResolver implements DependencyClassByQualifiedNameRes
}
private void resolveNamedGroupFunctions(@NotNull ClassOrNamespaceDescriptor owner, PsiClass psiClass,
TypeSubstitutor typeSubstitutorForGenericSuperclasses, NamedMembers namedMembers, Name methodName, ResolverScopeData scopeData) {
TypeSubstitutor typeSubstitutorForGenericSuperclasses, NamedMembers namedMembers, @NotNull Name methodName, ResolverScopeData scopeData) {
if (namedMembers.functionDescriptors != null) {
return;
}
@@ -1464,7 +1464,7 @@ public class JavaDescriptorResolver implements DependencyClassByQualifiedNameRes
namedMembers.functionDescriptors = functions;
}
private Set<SimpleFunctionDescriptor> getFunctionsFromSupertypes(ResolverScopeData scopeData, Name methodName) {
private Set<SimpleFunctionDescriptor> getFunctionsFromSupertypes(ResolverScopeData scopeData, @NotNull Name methodName) {
Set<SimpleFunctionDescriptor> r = new HashSet<SimpleFunctionDescriptor>();
for (JetType supertype : getSupertypes(scopeData)) {
for (FunctionDescriptor function : supertype.getMemberScope().getFunctions(methodName)) {
@@ -1705,7 +1705,7 @@ public class JavaDescriptorResolver implements DependencyClassByQualifiedNameRes
new AbstractMap.SimpleEntry<Name, NamedMembers>(exact, namedMembers));
}
else {
return null;
return Collections.emptyList();
}
}
else {
@@ -1714,7 +1714,7 @@ public class JavaDescriptorResolver implements DependencyClassByQualifiedNameRes
Name methodName = e.getKey();
NamedMembers namedMembers = e.getValue();
if (predicate.matches(methodName)) {
r.add(new AbstractMap.SimpleEntry<Name, NamedMembers>(exact, namedMembers));
r.add(new AbstractMap.SimpleEntry<Name, NamedMembers>(methodName, namedMembers));
}
}
return r;