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>) { public fun process(elements: Iterable<JetElement>) {
val first = elements.firstOrNull() for ((file, fileElements) in elements.groupBy { element -> element.getContainingFile() as JetFile }) {
if (first == null) return // first resolve all qualified references - optimization
val file = first.getContainingFile() as JetFile val resolveAllVisitor = ResolveAllReferencesVisitor(file)
processElements(fileElements, resolveAllVisitor)
// first resolve all qualified references - optimization val shortenTypesVisitor = ShortenTypesVisitor(file, resolveAllVisitor.result)
val resolveAllVisitor = ResolveAllReferencesVisitor(file) processElements(fileElements, shortenTypesVisitor)
processElements(elements, resolveAllVisitor) shortenTypesVisitor.finish()
val shortenTypesVisitor = ShortenTypesVisitor(file, resolveAllVisitor.result) processElements(fileElements, ShortenQualifiedExpressionsVisitor(file, resolveAllVisitor.result))
processElements(elements, shortenTypesVisitor) }
shortenTypesVisitor.finish()
processElements(elements, ShortenQualifiedExpressionsVisitor(file, resolveAllVisitor.result))
} }
private fun processElements(elements: Iterable<JetElement>, visitor: JetVisitorVoid) { private fun processElements(elements: Iterable<JetElement>, visitor: JetVisitorVoid) {