Get rid of polymorphism in set TrieNode

Author: Abduqodiri Qurbonzoda

kotlinx-collections-immutable/src/main/kotlin/kotlinx/collections/immutable/implementations/immutableMap/TrieNode.kt

@@ -88,7 +88,7 @@ internal class TrieNode<K, V>(
         private set
 
     internal fun isCollision(): Boolean {
-        return dataMap == 0 && nodeMap == 0 && buffer.size > 0
+        return dataMap == 0 && nodeMap == 0 && buffer.isNotEmpty()
     }
 
     /** Returns number of entries stored in this trie node (not counting subnodes) */

kotlinx-collections-immutable/src/main/kotlin/kotlinx/collections/immutable/implementations/immutableSet/PersistentHashSet.kt

@@ -63,7 +63,7 @@ internal class PersistentHashSet<E>(internal val node: TrieNode<E>,
     }
 
     internal companion object {
-        private val EMPTY = PersistentHashSet(CompactTrieNode.EMPTY, 0)
+        private val EMPTY = PersistentHashSet(TrieNode.EMPTY, 0)
         internal fun <E> emptyOf(): PersistentSet<E> = PersistentHashSet.EMPTY
     }
 }

kotlinx-collections-immutable/src/main/kotlin/kotlinx/collections/immutable/implementations/immutableSet/PersistentHashSetBuilder.kt

@@ -57,13 +57,13 @@ internal class PersistentHashSetBuilder<E>(private var set: PersistentHashSet<E>
     override fun remove(element: E): Boolean {
         val size = this.size
         @Suppress("UNCHECKED_CAST")
-        node = node.mutableRemove(element.hashCode(), element, 0, this) ?: CompactTrieNode.EMPTY as TrieNode<E>
+        node = node.mutableRemove(element.hashCode(), element, 0, this) ?: TrieNode.EMPTY as TrieNode<E>
         return size != this.size
     }
 
     override fun clear() {
         @Suppress("UNCHECKED_CAST")
-        node = CompactTrieNode.EMPTY as TrieNode<E>
+        node = TrieNode.EMPTY as TrieNode<E>
         size = 0
     }
 

kotlinx-collections-immutable/src/main/kotlin/kotlinx/collections/immutable/implementations/immutableSet/PersistentHashSetIterator.kt

@@ -85,7 +85,7 @@ internal open class PersistentHashSetIterator<E>(node: TrieNode<E>) : Iterator<E
 }
 
 internal class TrieNodeIterator<out E> {
-    private var buffer = CompactTrieNode.EMPTY.buffer
+    private var buffer = TrieNode.EMPTY.buffer
     private var index = 0
 
     fun reset(buffer: Array<Any?>, index: Int = 0) {

kotlinx-collections-immutable/src/main/kotlin/kotlinx/collections/immutable/implementations/immutableSet/PersistentHashSetMutableIterator.kt

@@ -56,7 +56,7 @@ internal class PersistentHashSetMutableIterator<E>(private val builder: Persiste
         }
 
         val position = 1 shl ((hashCode shr (pathIndex * LOG_MAX_BRANCHING_FACTOR)) and MAX_BRANCHING_FACTOR_MINUS_ONE)
-        val index = Integer.bitCount((node as CompactTrieNode<*>).bitmap and (position - 1))
+        val index = Integer.bitCount(node.bitmap and (position - 1))
 
         path[pathIndex].reset(node.buffer, index)