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

Use LazyResole to find direct inheritors of class

Browse Source
This commit is contained in:
Natalia.Ukhorskaya
2012-10-09 19:17:51 +04:00
parent d257256960
commit 6f7b6e4ff3
1 changed files with 20 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/jet/plugin/search/KotlinDirectInheritorsSearcher.java
+20 -34
View File
@@ -19,20 +19,23 @@ package org.jetbrains.jet.plugin.search;
import com.intellij.openapi.application.ApplicationManager; import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.application.QueryExecutorBase; import com.intellij.openapi.application.QueryExecutorBase;
import com.intellij.psi.PsiClass; import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiMethod;
import com.intellij.psi.PsiReference;
import com.intellij.psi.search.GlobalSearchScope; import com.intellij.psi.search.GlobalSearchScope;
import com.intellij.psi.search.searches.DirectClassInheritorsSearch; import com.intellij.psi.search.searches.DirectClassInheritorsSearch;
import com.intellij.util.Processor; import com.intellij.util.Processor;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import org.jetbrains.jet.asJava.JetLightClass; import org.jetbrains.jet.asJava.JetLightClass;
import org.jetbrains.jet.lang.psi.*; import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
import org.jetbrains.jet.lang.resolve.name.FqName; import org.jetbrains.jet.lang.descriptors.ClassifierDescriptor;
import org.jetbrains.jet.lang.psi.JetClass;
import org.jetbrains.jet.lang.psi.JetClassOrObject;
import org.jetbrains.jet.lang.psi.JetFile;
import org.jetbrains.jet.lang.resolve.DescriptorUtils;
import org.jetbrains.jet.lang.resolve.lazy.ResolveSession;
import org.jetbrains.jet.lang.types.JetType;
import org.jetbrains.jet.plugin.project.WholeProjectAnalyzerFacade;
import org.jetbrains.jet.plugin.stubindex.JetSuperClassIndex; import org.jetbrains.jet.plugin.stubindex.JetSuperClassIndex;
import java.util.Collection; import java.util.Collection;
import java.util.List;
/** /**
* @author yole * @author yole
@@ -62,34 +65,17 @@ public class KotlinDirectInheritorsSearcher extends QueryExecutorBase<PsiClass,
Collection<JetClassOrObject> candidates = JetSuperClassIndex.getInstance().get(name, clazz.getProject(), scope); Collection<JetClassOrObject> candidates = JetSuperClassIndex.getInstance().get(name, clazz.getProject(), scope);
for (JetClassOrObject candidate : candidates) { for (JetClassOrObject candidate : candidates) {
if (!(candidate instanceof JetClass)) continue; if (!(candidate instanceof JetClass)) continue;
List<JetDelegationSpecifier> specifiers = candidate.getDelegationSpecifiers(); JetFile containingFile = (JetFile) candidate.getContainingFile();
for (JetDelegationSpecifier specifier : specifiers) { ResolveSession sessionForFile = WholeProjectAnalyzerFacade.getLazyResolveSessionForFile(containingFile);
JetUserType type = specifier.getTypeAsUserType(); ClassDescriptor classDescriptor = (ClassDescriptor) sessionForFile.resolveToDescriptor(candidate);
if (type != null) { for (JetType type : classDescriptor.getTypeConstructor().getSupertypes()) {
JetSimpleNameExpression referenceExpression = type.getReferenceExpression(); ClassifierDescriptor declarationDescriptor = type.getConstructor().getDeclarationDescriptor();
if (referenceExpression != null) { if (declarationDescriptor != null) {
PsiReference reference = referenceExpression.getReference(); String fqName = DescriptorUtils.getFQName(declarationDescriptor).getFqName();
PsiElement resolved = reference != null ? reference.resolve() : null; if (qualifiedName.equals(fqName)) {
JetLightClass lightClass = JetLightClass.wrapDelegate((JetClass) candidate);
String resolvedFQName = null; if (lightClass != null) {
if (resolved instanceof PsiClass) { consumer.process(lightClass);
resolvedFQName = ((PsiClass)resolved).getQualifiedName();
}
else if (resolved instanceof JetClass) {
FqName fqName = JetPsiUtil.getFQName((JetClass) resolved);
if (fqName != null) {
resolvedFQName = fqName.getFqName();
}
} else if (resolved instanceof PsiMethod && ((PsiMethod) resolved).isConstructor()) {
PsiClass containingClass = ((PsiMethod) resolved).getContainingClass();
resolvedFQName = containingClass.getQualifiedName();
}
if (qualifiedName.equals(resolvedFQName)) {
JetLightClass lightClass = JetLightClass.wrapDelegate((JetClass) candidate);
if (lightClass != null) {
consumer.process(lightClass);
}
} }
} }
} }