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

Shorten references: Group elements by file before applying transformation

Browse Source
This commit is contained in:
Alexey Sedunov
2014-02-12 20:16:02 +04:00
parent d2e6aa2aa5
commit 86619f5f28
1 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
idea/src/org/jetbrains/jet/plugin/codeInsight/ShortenReferences.kt
+9 -11
View File
@@ -27,19 +27,17 @@ public object ShortenReferences {
}
public fun process(elements: Iterable<JetElement>) {
val first = elements.firstOrNull()
if (first == null) return
val file = first.getContainingFile() as JetFile
for ((file, fileElements) in elements.groupBy { element -> element.getContainingFile() as JetFile }) {
// first resolve all qualified references - optimization
val resolveAllVisitor = ResolveAllReferencesVisitor(file)
processElements(fileElements, resolveAllVisitor)
// first resolve all qualified references - optimization
val resolveAllVisitor = ResolveAllReferencesVisitor(file)
processElements(elements, resolveAllVisitor)
val shortenTypesVisitor = ShortenTypesVisitor(file, resolveAllVisitor.result)
processElements(fileElements, shortenTypesVisitor)
shortenTypesVisitor.finish()
val shortenTypesVisitor = ShortenTypesVisitor(file, resolveAllVisitor.result)
processElements(elements, shortenTypesVisitor)
shortenTypesVisitor.finish()
processElements(elements, ShortenQualifiedExpressionsVisitor(file, resolveAllVisitor.result))
processElements(fileElements, ShortenQualifiedExpressionsVisitor(file, resolveAllVisitor.result))
}
}
private fun processElements(elements: Iterable<JetElement>, visitor: JetVisitorVoid) {