Kotlin中的多线程数组初始化

时间:2017-08-01 13:50:17

标签: arrays multithreading matrix multidimensional-array kotlin

说,我有以下允许我创建矩阵的函数:

inline fun <reified T> Matrix2D(w: Int, h: Int, init: (Int, Int) -> T) =
    Array(w){ x -> Array(h){ y -> init(x, y) } }

inline fun <reified T> Matrix2D(w: Int, h: Int, value: T) =
    Array(w){ Array(h){ value } }

// For example:
val m = Matrix2D(400, 400) { x, y ->
    Color(x.toDouble() / 400.0, y.toDouble() / 400.0, 0.0)
}

我也不想允许可空性,因为以后处理元素访问会很麻烦。

如何使用多个线程/协同程序初始化这样的矩阵,其中每个线程初始化其自己的矩阵块,而不允许矩阵单元的可空性?

1 个答案:

答案 0 :(得分:1)

是的,您可以并行创建Matrix2D,例如:

val threads = Matrix2D(10, 3){x, y-> Thread.currentThread() }.flatten().distinct()

//              v--- size > 1
println(threads.size)
inline fun <reified T> Matrix2D(w: Int, h: Int, crossinline init: (Int, Int) -> T)=
       //                     v--- create array in parallel
       generatorOf(w){x -> Array(h) {y -> init(x, y) } }.map{ it() }.toTypedArray()
       //                          get the final result `Matrix2D` ---^

//                     v--- create array generator
inline fun <reified T> generatorOf(size: Int, crossinline init:(Int) -> T) = 
  //v--- you can use `ForkJoinPool#commonPool`, but it will run task in work thread
   ForkJoinPool(20).let {
       Array(size) { i -> it.submit(Callable { init(i) }).let { { it.get() } } }
       //                 return the lambda generator ()->T  ---^
   }

您还可以编写扩展功能toArray以删除中间map操作&amp;提升绩效,例如:

inline fun <reified T> Matrix2D(w: Int, h: Int, crossinline init: (Int, Int) -> T)=
        //                         v--- init array elements in parallel
        generatorOf(w) { x -> Array(h) { y -> init(x, y) } }.toArray{it()}
        //           transform to the `Matrix2D` directly  ---^

//                                v--- convert Array<T> to Array<R>
inline fun <T,reified R> Array<T>.toArray(transform:(T)->R):Array<R>{
    return Array(size){i->transform(get(i))}
}

为什么在外部数组初始化中启动了Matrix2D个线程,您可以进一步看到sum two double[][] with parallel stream