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


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

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

编译器会抛出错误说“不能使用'T'作为reified类型参数。请改用一个班级。”。


6866
2018-02-16 19:11


起源



答案:


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

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]

8
2017-12-29 11:51





问题在于呼唤 arrayOfNulls<T>(height) 使用非具体类型参数T.但我们也无法制作 T reified,编译器将抛出以下错误:“只能实现内联函数的类型参数

这就是我们要做的。我们使用内联工厂方法而不是构造函数:

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()方法(相关问题)。因为该方法是内联的,我们可以使用 具体化的仿制药 这使得打电话成为可能 arrayOfNulls<T>


7
2018-02-16 19:11