在Kotlin中创建通用的2D数组

Question

假设我有一个泛型类,我需要一个泛型类型的2D数组T.如果我尝试以下

class Matrix<T>(width: Int, height: Int) {
    val data: Array<Array<T>> = Array(width, arrayOfNulls<T>(height))
}

编译器将抛出一个错误,说" 不能使用'T'作为reified类型参数.请改用类. "

Answer 1

仅仅因为语法已经移动了一下,这是我对它的看法:

class Array2D<T> (val xSize: Int, val ySize: Int, val array: Array<Array<T>>) {

    companion object {

        inline operator fun <reified T> invoke() = Array2D(0, 0, Array(0, { emptyArray<T>() }))

        inline operator fun <reified T> invoke(xWidth: Int, yWidth: Int) =
            Array2D(xWidth, yWidth, Array(xWidth, { arrayOfNulls<T>(yWidth) }))

        inline operator fun <reified T> invoke(xWidth: Int, yWidth: Int, operator: (Int, Int) -> (T)): Array2D<T> {
            val array = Array(xWidth, {
                val x = it
                Array(yWidth, {operator(x, it)})})
            return Array2D(xWidth, yWidth, array)
        }
    }

    operator fun get(x: Int, y: Int): T {
        return array[x][y]
    }

    operator fun set(x: Int, y: Int, t: T) {
        array[x][y] = t
    }

    inline fun forEach(operation: (T) -> Unit) {
        array.forEach { it.forEach { operation.invoke(it) } }
    }

    inline fun forEachIndexed(operation: (x: Int, y: Int, T) -> Unit) {
        array.forEachIndexed { x, p -> p.forEachIndexed { y, t -> operation.invoke(x, y, t) } }
    }
}

这也允许您以与1d数组类似的方式创建2d数组,例如

val array2D = Array2D<String>(5, 5) { x, y -> "$x $y" }

使用索引操作符访问/设置内容:

val xy = array2D[1, 2]

Answer 2

问题是arrayOfNulls<T>(height)使用非reified类型参数T 调用.但我们也无法生成Treified,编译器将抛出以下错误:" 只能内联函数的类型参数 "

这就是我们要做的.我们使用内联工厂方法代替构造函数:

class Matrix<T> private(width: Int, height: Int, arrayFactory: (Int) -> Array<T>) {

    class object {
        inline fun <reified T>invoke(width: Int, height: Int)
                = Matrix(width, height, { size -> arrayOfNulls<T>(size) })
    }

    val data: Array<Array<T>> = Array(width, { size -> arrayFactory(size) })
}

注意,构造函数现在是私有的,因此调用Matrix()将正确调用新的invoke()方法(相关问题).因为该方法是内联的,所以我们可以使用可以调用的reified泛型arrayOfNulls<T>.