Add 2 benchmarks for Ctries.

1 files changed, 39 insertions, 0 deletions

diff --git a/test/benchmarking/ParCtrie-nums.scala b/test/benchmarking/ParCtrie-nums.scala
new file mode 100644
index 0000000000..76d1966d1f
--- /dev/null
+++ b/test/benchmarking/ParCtrie-nums.scala
@@ -0,0 +1,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
+      }
+    }
+  }
+}
+