test/benchmarking/ParCtrie-nums.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

import collection.parallel.mutable.ParCtrie


case class Entry(num: Double) {
  var sqrt = num
}


object Nums extends testing.Benchmark {
  val length = sys.props("length").toInt
  val par = sys.props("par").toInt
  var entries: Seq[Entry] = null
  var results: ParCtrie[Double, Entry] = null
  
  collection.parallel.ForkJoinTasks.defaultForkJoinPool.setParallelism(par)
  
  override def setUp() {
    entries = (1 until length) map { num => Entry(num.toDouble) }
    results = ParCtrie()
    for (e <- entries) results += ((e.num, e))
  }
  
  def run() = {
    while (results.nonEmpty) {
      for ((num, e) <- results) {
        val nsqrt = 0.5 * (e.sqrt + e.num / e.sqrt)
        if (math.abs(nsqrt - e.sqrt) < 0.01) {
          results.remove(num)
        } else e.sqrt = nsqrt
      }
    }
  }
}