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

DFS Utility class used in TypeHierarchyResolver

Browse Source
This commit is contained in:
Andrey Breslav
2012-09-11 19:09:20 +04:00
parent ab18b12165
commit a632d53e58
1 changed files with 22 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/frontend/src/org/jetbrains/jet/lang/resolve/TypeHierarchyResolver.java
+22 -29
View File
@@ -21,6 +21,7 @@ import com.google.common.collect.Multimap;
import com.google.common.collect.Sets; import com.google.common.collect.Sets;
import com.intellij.psi.PsiElement; import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiNameIdentifierOwner; import com.intellij.psi.PsiNameIdentifierOwner;
import com.intellij.util.containers.ContainerUtil;
import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable; import org.jetbrains.annotations.Nullable;
import org.jetbrains.jet.lang.descriptors.*; import org.jetbrains.jet.lang.descriptors.*;
@@ -36,6 +37,7 @@ import org.jetbrains.jet.lang.types.TypeConstructor;
import org.jetbrains.jet.lang.types.TypeProjection; import org.jetbrains.jet.lang.types.TypeProjection;
import org.jetbrains.jet.lang.types.checker.JetTypeChecker; import org.jetbrains.jet.lang.types.checker.JetTypeChecker;
import org.jetbrains.jet.lexer.JetTokens; import org.jetbrains.jet.lexer.JetTokens;
import org.jetbrains.jet.utils.DFS;
import javax.inject.Inject; import javax.inject.Inject;
import java.util.*; import java.util.*;
@@ -63,7 +65,7 @@ public class TypeHierarchyResolver {
private BindingTrace trace; private BindingTrace trace;
// state // state
private LinkedList<MutableClassDescriptor> topologicalOrder; private List<MutableClassDescriptor> topologicalOrder;
@Inject @Inject
public void setContext(@NotNull TopDownAnalysisContext context) { public void setContext(@NotNull TopDownAnalysisContext context) {
@@ -395,18 +397,27 @@ public class TypeHierarchyResolver {
} }
} }
private LinkedList<MutableClassDescriptor> topologicallySortClassesAndObjects() { private List<MutableClassDescriptor> topologicallySortClassesAndObjects() {
// A topsort is needed only for better diagnostics: // A topsort is needed only for better diagnostics:
// edges that get removed to disconnect loops are more reasonable in this case // edges that get removed to disconnect loops are more reasonable in this case
LinkedList<MutableClassDescriptor> topologicalOrder = Lists.newLinkedList(); return DFS.topologicalOrder(
Set<ClassDescriptor> visited = Sets.newHashSet(); ContainerUtil.<MutableClassDescriptor>concat(context.getClasses().values(), context.getObjects().values()),
for (MutableClassDescriptor mutableClassDescriptor : context.getClasses().values()) { new DFS.Neighbors<MutableClassDescriptor>() {
topologicallySort(mutableClassDescriptor, visited, topologicalOrder); @NotNull
} @Override
for (MutableClassDescriptor mutableClassDescriptor : context.getObjects().values()) { public Iterable<MutableClassDescriptor> getNeighbors(MutableClassDescriptor current) {
topologicallySort(mutableClassDescriptor, visited, topologicalOrder); List<MutableClassDescriptor> result = Lists.newArrayList();
} for (JetType supertype : current.getSupertypes()) {
return topologicalOrder; DeclarationDescriptor declarationDescriptor = supertype.getConstructor().getDeclarationDescriptor();
if (declarationDescriptor instanceof MutableClassDescriptor) {
MutableClassDescriptor classDescriptor = (MutableClassDescriptor) declarationDescriptor;
result.add(classDescriptor);
}
}
return result;
}
});
} }
private void detectAndDisconnectLoops() { private void detectAndDisconnectLoops() {
@@ -419,24 +430,6 @@ public class TypeHierarchyResolver {
} }
} }
private static void topologicallySort(
MutableClassDescriptor mutableClassDescriptor,
Set<ClassDescriptor> visited,
LinkedList<MutableClassDescriptor> topologicalOrder
) {
if (!visited.add(mutableClassDescriptor)) {
return;
}
for (JetType supertype : mutableClassDescriptor.getSupertypes()) {
DeclarationDescriptor declarationDescriptor = supertype.getConstructor().getDeclarationDescriptor();
if (declarationDescriptor instanceof MutableClassDescriptor) {
MutableClassDescriptor classDescriptor = (MutableClassDescriptor) declarationDescriptor;
topologicallySort(classDescriptor, visited, topologicalOrder);
}
}
topologicalOrder.addFirst(mutableClassDescriptor);
}
private void traverseTypeHierarchy( private void traverseTypeHierarchy(
MutableClassDescriptor currentClass, MutableClassDescriptor currentClass,
Set<ClassDescriptor> visited, Set<ClassDescriptor> visited,