benchmarks/sort/sort_gendata.scala

   1 #!/usr/bin/env scala
   2
   3 import scala.util.Sorting
   4
   5 val size = args(0).toInt
   6
   7 def rand_array(size: Int) = {
   8   var r = new scala.util.Random
   9   Array.fill(size) { r.nextFloat() }
  10 }
  11
  12 def print_array(name: String, size: Int, arr: Array[Float]) {
  13   println("float "+name+"["+size+"] = {")
  14   for(i <- 0 to size-2)
  15     println("  "+arr(i)+",")
  16   println("  "+arr(size-1)+"\n};\n")
  17 }
  18
  19 println("#define DATA_SIZE_SORT " + size)
  20
  21 val a = rand_array(size)
  22 val sorted = a.clone()
  23 Sorting.quickSort(sorted)
  24
  25 print_array("input_data_sort", size, a)
  26 print_array("verify_data_sort", size, sorted)