From d2a201ad873e0a356ed7e94049ef32dc00c9cc94 Mon Sep 17 00:00:00 2001
From: James Roseborough <james.roseborough@luaj.org>
Date: Mon, 14 Jul 2008 17:58:37 +0000
Subject: [PATCH] Add simple performance tests

---
 src/test/perf/fannkuch.lua   |  51 ++++++++++++++
 src/test/perf/mandelbrot.lua |  40 +++++++++++
 src/test/perf/nbody.lua      | 125 +++++++++++++++++++++++++++++++++++
 3 files changed, 216 insertions(+)
 create mode 100644 src/test/perf/fannkuch.lua
 create mode 100644 src/test/perf/mandelbrot.lua
 create mode 100644 src/test/perf/nbody.lua

diff --git a/src/test/perf/fannkuch.lua b/src/test/perf/fannkuch.lua
new file mode 100644
index 00000000..7f907a94
--- /dev/null
+++ b/src/test/perf/fannkuch.lua
@@ -0,0 +1,51 @@
+-- The Computer Language Shootout
+-- http://shootout.alioth.debian.org/
+-- contributed by Mike Pall
+-- modified for luaj to remove dependency on io
+
+local function fannkuch(n)
+  local p, q, s, odd, check, maxflips = {}, {}, {}, true, 0, 0
+  for i=1,n do p[i] = i; s[i] = i end
+  repeat
+    -- Print max. 30 permutations.
+    if check < 30 then
+      if not p[n] then return maxflips end	-- Catch n = 0, 1, 2.
+      for i=1,n do print(p[i]) end;
+      check = check + 1
+    end
+    -- Copy and flip.
+    local q1 = p[1]				-- Cache 1st element.
+    if p[n] ~= n and q1 ~= 1 then		-- Avoid useless work.
+      for i=2,n do q[i] = p[i] end		-- Work on a copy.
+      for flips=1,1000000 do			-- Flip ...
+	local j = q1 - 1
+	for i=2,q1/2 do q[i], q[j] = q[j], q[i]; j = j - 1 end
+	local qq = q[q1]
+	if qq == 1 then				-- ... until 1st element is 1.
+	  if flips > maxflips then maxflips = flips end -- New maximum?
+	  break
+	end
+	q[q1] = q1
+	q1 = qq
+      end
+    end
+    -- Permute.
+    if odd then
+      p[2], p[1] = p[1], p[2]; odd = false	-- Rotate 1<-2.
+    else
+      p[2], p[3] = p[3], p[2]; odd = true	-- Rotate 1<-2 and 1<-2<-3.
+      for i=3,n do
+	local sx = s[i]
+	if sx ~= 1 then sx = sx - 1; s[i] = sx; break end
+	if i == n then return maxflips end	-- Out of permutations.
+	s[i] = i
+	-- Rotate 1<-...<-i+1.
+	local t = p[1]; for j=1,i do p[j] = p[j+1] end; p[i+1] = t
+      end
+    end
+  until false
+end
+
+local arg = {...}
+local n = tonumber(arg and arg[1]) or 5
+print("Pfannkuchen("..n..") = "..fannkuch(n))
\ No newline at end of file
diff --git a/src/test/perf/mandelbrot.lua b/src/test/perf/mandelbrot.lua
new file mode 100644
index 00000000..ad2a0a7a
--- /dev/null
+++ b/src/test/perf/mandelbrot.lua
@@ -0,0 +1,40 @@
+-- The Computer Language Shootout
+-- http://shootout.alioth.debian.org/
+-- contributed by Mike Pall
+-- modified for luaj to remove dependency on io
+
+local arg = {...}
+local width = tonumber(arg and arg[1]) or 100
+local height, wscale = width, 2/width
+local m, limit2 = 50, 4.0
+local write, char = print, string.char
+
+write("P4\n", width, " ", height, "\n")
+
+for y=0,height-1 do
+  local Ci = 2*y / height - 1
+  for xb=0,width-1,8 do
+    local bits = 0
+    local xbb = xb+7
+    for x=xb,xbb < width and xbb or width-1 do
+      bits = bits + bits
+      local Zr, Zi, Zrq, Ziq = 0.0, 0.0, 0.0, 0.0
+      local Cr = x * wscale - 1.5
+      for i=1,m do
+        local Zri = Zr*Zi
+        Zr = Zrq - Ziq + Cr
+        Zi = Zri + Zri + Ci
+        Zrq = Zr*Zr
+        Ziq = Zi*Zi
+        if Zrq + Ziq > limit2 then
+          bits = bits + 1
+          break
+        end
+      end
+    end
+    if xbb >= width then
+      for x=width,xbb do bits = bits + bits + 1 end
+    end
+    write(char(255-bits))
+  end
+end
diff --git a/src/test/perf/nbody.lua b/src/test/perf/nbody.lua
new file mode 100644
index 00000000..d688717f
--- /dev/null
+++ b/src/test/perf/nbody.lua
@@ -0,0 +1,125 @@
+-- The Great Computer Language Shootout
+-- http://shootout.alioth.debian.org/
+-- contributed by Isaac Gouy, tuned by Mike Pall
+-- modified for luaj to remove dependency on io
+
+local sqrt = math.sqrt
+
+local PI = 3.141592653589793
+local SOLAR_MASS = 4 * PI * PI
+local DAYS_PER_YEAR = 365.24
+
+local Jupiter = {
+   x = 4.84143144246472090e+00
+  ,y = -1.16032004402742839e+00
+  ,z = -1.03622044471123109e-01
+  ,vx = 1.66007664274403694e-03 * DAYS_PER_YEAR
+  ,vy = 7.69901118419740425e-03 * DAYS_PER_YEAR
+  ,vz = -6.90460016972063023e-05 * DAYS_PER_YEAR
+  ,mass = 9.54791938424326609e-04 * SOLAR_MASS
+}
+
+local Saturn = {
+   x = 8.34336671824457987e+00
+  ,y = 4.12479856412430479e+00
+  ,z = -4.03523417114321381e-01
+  ,vx = -2.76742510726862411e-03 * DAYS_PER_YEAR
+  ,vy = 4.99852801234917238e-03 * DAYS_PER_YEAR
+  ,vz = 2.30417297573763929e-05 * DAYS_PER_YEAR
+  ,mass = 2.85885980666130812e-04 * SOLAR_MASS
+}
+
+local Uranus = {
+   x = 1.28943695621391310e+01
+  ,y = -1.51111514016986312e+01
+  ,z = -2.23307578892655734e-01
+  ,vx = 2.96460137564761618e-03 * DAYS_PER_YEAR
+  ,vy = 2.37847173959480950e-03 * DAYS_PER_YEAR
+  ,vz = -2.96589568540237556e-05 * DAYS_PER_YEAR
+  ,mass = 4.36624404335156298e-05 * SOLAR_MASS
+}
+
+local Neptune = {
+   x = 1.53796971148509165e+01
+  ,y = -2.59193146099879641e+01
+  ,z = 1.79258772950371181e-01
+  ,vx = 2.68067772490389322e-03 * DAYS_PER_YEAR
+  ,vy = 1.62824170038242295e-03 * DAYS_PER_YEAR
+  ,vz = -9.51592254519715870e-05 * DAYS_PER_YEAR
+  ,mass = 5.15138902046611451e-05 * SOLAR_MASS
+}
+
+local Sun = { x = 0, y = 0, z = 0,
+              vx = 0, vy = 0, vz = 0, mass = SOLAR_MASS }
+
+local function advance(bodies, nbody, dt)
+  for i=1,nbody do
+    local bi = bodies[i]
+    local bix, biy, biz, bimass = bi.x, bi.y, bi.z, bi.mass
+    local bivx, bivy, bivz = bi.vx, bi.vy, bi.vz
+    for j=i+1,nbody do
+      local bj = bodies[j]
+      local dx, dy, dz = bix-bj.x, biy-bj.y, biz-bj.z
+      local distance = sqrt(dx*dx + dy*dy + dz*dz)
+      local mag = dt / (distance * distance * distance)
+      local bim, bjm = bimass*mag, bj.mass*mag
+      bivx = bivx - (dx * bjm)
+      bivy = bivy - (dy * bjm)
+      bivz = bivz - (dz * bjm)
+      bj.vx = bj.vx + (dx * bim)
+      bj.vy = bj.vy + (dy * bim)
+      bj.vz = bj.vz + (dz * bim)
+    end
+    bi.vx = bivx
+    bi.vy = bivy
+    bi.vz = bivz
+  end
+  for i=1,nbody do
+    local bi = bodies[i]
+    bi.x = bi.x + (dt * bi.vx)
+    bi.y = bi.y + (dt * bi.vy)
+    bi.z = bi.z + (dt * bi.vz)
+  end
+end
+
+
+local function energy(bodies, nbody)
+  local e = 0
+  for i=1,nbody do
+    local bi = bodies[i]
+    local vx, vy, vz, bim = bi.vx, bi.vy, bi.vz, bi.mass
+    e = e + (0.5 * bim * (vx*vx + vy*vy + vz*vz))
+    for j=i+1,nbody do
+      local bj = bodies[j]
+      local dx, dy, dz = bi.x-bj.x, bi.y-bj.y, bi.z-bj.z
+      local distance = sqrt(dx*dx + dy*dy + dz*dz)
+      e = e - ((bim * bj.mass) / distance)
+    end
+  end
+  return e
+end
+
+
+local function offsetMomentum(b, nbody)
+  local px, py, pz = 0, 0, 0
+  for i=1,nbody do
+    local bi = b[i]
+    local bim = bi.mass
+    px = px + (bi.vx * bim)
+    py = py + (bi.vy * bim)
+    pz = pz + (bi.vz * bim)
+  end
+  b[1].vx = -px / SOLAR_MASS
+  b[1].vy = -py / SOLAR_MASS
+  b[1].vz = -pz / SOLAR_MASS
+end
+
+local arg = {...}
+local N = tonumber(arg and arg[1]) or 1000
+local bodies = { Sun, Jupiter, Saturn, Uranus, Neptune }
+local nbody = #bodies
+
+offsetMomentum(bodies, nbody)
+print( tostring(energy(bodies, nbody)))
+for i=1,N do advance(bodies, nbody, 0.01) end
+print( tostring(energy(bodies, nbody)))