commonMain.BitSet.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of encoding-jvm Show documentation

Toolbox of utilities/helpers for Kotlin development

There is a newer version: 8.0.0

package com.juul.tuulbox.encoding

public interface BitSet {
    public operator fun get(index: Int): Boolean
    public operator fun set(index: Int, value: Boolean): BitSet
    public fun asPrimitive(): T
}

// Would kill for C++ templates right about now. Which is not something I can normally say.

public val Int.bits: BitSet get() = IntBitSet(this)
public val Long.bits: BitSet get() = LongBitSet(this)

public data class IntBitSet(private var buffer: Int) : BitSet {
    override operator fun get(index: Int): Boolean {
        require(index in 0 until 32)
        val mask = 1 shl index
        return (buffer and mask) == mask
    }

    override operator fun set(index: Int, value: Boolean): IntBitSet = this.apply {
        require(index in 0 until 32)
        buffer = when (value) {
            true -> buffer or (1 shl index)
            false -> buffer and (1 shl index).inv()
        }
    }

    override fun asPrimitive(): Int = buffer
}

public data class LongBitSet(private var buffer: Long) : BitSet {
    override operator fun get(index: Int): Boolean {
        require(index in 0 until 64)
        val mask = 1L shl index
        return (buffer and mask) == mask
    }

    override operator fun set(index: Int, value: Boolean): LongBitSet = this.apply {
        require(index in 0 until 64)
        buffer = when (value) {
            true -> buffer or (1L shl index)
            false -> buffer and (1L shl index).inv()
        }
    }

    override fun asPrimitive(): Long = buffer
}