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

J2K Interner: convert and prettify

Browse Source
This commit is contained in:
Alexander Udalov
2018-03-09 17:53:40 +01:00
parent bda7514b43
commit 779290be0c
1 changed files with 16 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/util/src/org/jetbrains/kotlin/utils/Interner.kt
+16 -45
View File
@@ -14,56 +14,27 @@
* limitations under the License. * limitations under the License.
*/ */
package org.jetbrains.kotlin.utils; package org.jetbrains.kotlin.utils
import kotlin.collections.CollectionsKt; class Interner<T>(private val parent: Interner<T>? = null) {
import org.jetbrains.annotations.NotNull; private val firstIndex: Int = parent?.run { interned.size + firstIndex } ?: 0
import org.jetbrains.annotations.Nullable; private val interned = hashMapOf<T, Int>()
import java.util.HashMap; val allInternedObjects: List<T>
import java.util.List; get() = interned.keys.sortedBy(interned::get)
import java.util.Map;
public final class Interner<T> { val isEmpty: Boolean
private final Interner<T> parent; get() = interned.isEmpty() && parent?.isEmpty != false
private final int firstIndex;
private final Map<T, Integer> interned = new HashMap<>();
public Interner(Interner<T> parent) { private fun find(obj: T): Int? {
this.parent = parent; assert(parent == null || parent.interned.size + parent.firstIndex == firstIndex) {
this.firstIndex = parent != null ? parent.interned.size() + parent.firstIndex : 0; "Parent changed in parallel with child: indexes will be wrong"
}
public Interner() {
this(null);
}
@Nullable
private Integer find(@NotNull T obj) {
assert parent == null || parent.interned.size() + parent.firstIndex == firstIndex :
"Parent changed in parallel with child: indexes will be wrong";
if (parent != null) {
Integer index = parent.find(obj);
if (index != null) return index;
} }
return interned.get(obj); return parent?.find(obj) ?: interned[obj]
} }
public int intern(@NotNull T obj) { fun intern(obj: T): Int =
Integer index = find(obj); find(obj) ?: (firstIndex + interned.size).also {
if (index != null) return index; interned[obj] = it
}
index = firstIndex + interned.size();
interned.put(obj, index);
return index;
}
@NotNull
public List<T> getAllInternedObjects() {
return CollectionsKt.sortedBy(interned.keySet(), interned::get);
}
public boolean isEmpty() {
return interned.isEmpty() && (parent == null || parent.isEmpty());
}
} }