bench/yarv/bm_so_mandelbrot.rb

   1 #  The Computer Language Benchmarks Game\r
   2 #  http://shootout.alioth.debian.org/\r
   3 #\r
   4 #  contributed by Karl von Laudermann\r
   5 #  modified by Jeremy Echols\r
   6 \r
   7 size = 600 # ARGV[0].to_i\r
   8 \r
   9 puts "P4\n#{size} #{size}"\r
  10 \r
  11 ITER = 49                           # Iterations - 1 for easy for..in looping\r
  12 LIMIT_SQUARED = 4.0                 # Presquared limit\r
  13 \r
  14 byte_acc = 0\r
  15 bit_num = 0\r
  16 \r
  17 count_size = size - 1               # Precomputed size for easy for..in looping\r
  18 \r
  19 # For..in loops are faster than .upto, .downto, .times, etc.\r
  20 for y in 0..count_size\r
  21   for x in 0..count_size\r
  22     zr = 0.0\r
  23     zi = 0.0\r
  24     cr = (2.0*x/size)-1.5\r
  25     ci = (2.0*y/size)-1.0\r
  26     escape = false\r
  27 \r
  28     # To make use of the for..in code, we use a dummy variable,\r
  29     # like one would in C\r
  30     for dummy in 0..ITER\r
  31       tr = zr*zr - zi*zi + cr\r
  32       ti = 2*zr*zi + ci\r
  33       zr, zi = tr, ti\r
  34 \r
  35       if (zr*zr+zi*zi) > LIMIT_SQUARED\r
  36         escape = true\r
  37         break\r
  38       end\r
  39     end\r
  40 \r
  41     byte_acc = (byte_acc << 1) | (escape ? 0b0 : 0b1)\r
  42     bit_num += 1\r
  43 \r
  44     # Code is very similar for these cases, but using separate blocks\r
  45     # ensures we skip the shifting when it's unnecessary, which is most cases.\r
  46     if (bit_num == 8)\r
  47       print byte_acc.chr\r
  48       byte_acc = 0\r
  49       bit_num = 0\r
  50     elsif (x == count_size)\r
  51       byte_acc <<= (8 - bit_num)\r
  52       print byte_acc.chr\r
  53       byte_acc = 0\r
  54       bit_num = 0\r
  55     end\r
  56   end\r
  57 end\r