xtdb.arrow.VectorIndirection.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of xtdb-core Show documentation

An open source document database with bitemporal graph queries

The newest version!

package xtdb.arrow

interface VectorIndirection {
    fun valueCount(): Int

    fun getIndex(idx: Int): Int

    operator fun get(idx: Int): Int = getIndex(idx)

    companion object {
        @JvmStatic
        fun selection(idxs: IntArray): VectorIndirection = Selection(idxs)

        internal data class Selection(val idxs: IntArray) : VectorIndirection {
            override fun valueCount(): Int {
                return idxs.size
            }

            override fun getIndex(idx: Int): Int {
                return idxs[idx]
            }

            override fun toString(): String {
                val idxs = idxs.contentToString()
                return "(Selection {idxs=$idxs})"
            }

            override fun equals(other: Any?) = when {
                this === other -> true
                other !is Selection -> false
                else -> idxs.contentEquals(other.idxs)
            }

            override fun hashCode() = idxs.contentHashCode()
        }

        @JvmStatic
        fun slice(startIdx: Int, len: Int) : VectorIndirection = Slice(startIdx, len)

        internal data class Slice(val startIdx: Int, val len: Int) : VectorIndirection {
            override fun valueCount(): Int {
                return len
            }

            override fun getIndex(idx: Int): Int {
                return startIdx + idx
            }
        }
    }
}