benchmarks/sort/sort_gendata.scala

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