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

FIR: Optimize SupertypeComputationSession::breakLoops

Browse Source
This commit is contained in:
Denis Zharkov
2020-01-29 12:33:55 +03:00
parent a44fa8db50
commit ac2b5beb4e
3 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirSupertypesResolution.kt
+8 -4
View File
@@ -310,15 +310,18 @@ private class SupertypeComputationSession {
}
supertypeStatusMap[classLikeDeclaration] = SupertypeComputationStatus.Computed(resolvedTypesRefs)
newClassifiersForBreakingLoops.add(classLikeDeclaration)
}
private val newClassifiersForBreakingLoops = mutableListOf<FirClassLikeDeclaration<*>>()
private val breakLoopsDfsVisited = hashSetOf<FirClassLikeDeclaration<*>>()
fun breakLoops(session: FirSession) {
val visited = hashSetOf<FirClassLikeDeclaration<*>>()
val inProcess = hashSetOf<FirClassLikeDeclaration<*>>()
fun dfs(classLikeDeclaration: FirClassLikeDeclaration<*>) {
if (classLikeDeclaration in breakLoopsDfsVisited) return
val supertypeComputationStatus = supertypeStatusMap[classLikeDeclaration] ?: return
if (classLikeDeclaration in visited) return
if (classLikeDeclaration in inProcess) return
inProcess.add(classLikeDeclaration)
@@ -351,12 +354,13 @@ private class SupertypeComputationSession {
}
inProcess.remove(classLikeDeclaration)
visited.add(classLikeDeclaration)
breakLoopsDfsVisited.add(classLikeDeclaration)
}
for (classifier in supertypeStatusMap.keys) {
for (classifier in newClassifiersForBreakingLoops) {
dfs(classifier)
}
newClassifiersForBreakingLoops.clear()
}
}