1 files changed, 143 insertions, 0 deletions
diff --git a/src/test/bench/shootout-mandelbrot.rs b/src/test/bench/shootout-mandelbrot.rs
new file mode 100644
index 00000000000..2fab8685c2c
--- /dev/null
+++ b/src/test/bench/shootout-mandelbrot.rs
@@ -0,0 +1,143 @@
+//  based on:
+//  http://shootout.alioth.debian.org/u64q/program.php?test=mandelbrot&lang=python3&id=2
+//
+//  takes 2 optional numeric args: square image size and yield frequency
+//  in the shootout, they use 16000 as image size
+//  yield frequency doesn't seem to have much effect
+//
+//  writes pbm image to stdout
+
+use std;
+import std::io::writer_util;
+
+type cmplx = {re: f64, im: f64};
+type line = {i: uint, b: [u8]};
+
+impl arith for cmplx {
+	fn *(x: cmplx) -> cmplx {
+		{re: self.re*x.re - self.im*x.im, im: self.re*x.im + self.im*x.re}
+	}
+
+	fn +(x: cmplx) -> cmplx {
+		{re: self.re + x.re, im: self.im + x.im}
+	}
+}
+
+pure fn cabs(x: cmplx) -> f64
+{
+	x.re*x.re + x.im*x.im
+}
+
+fn mb(x: cmplx) -> bool
+{
+	let z = {re: 0., im: 0.};
+	let i = 0;
+	let in = true;
+	while i < 50 {
+		z = z*z + x;
+		if cabs(z) >= 4. {
+			in = false;
+			break;
+		}
+		i += 1;
+	}
+	in
+}
+
+fn fillbyte(x: cmplx, incr: f64) -> u8 {
+	let rv = 0_u8;
+	let i = 0_u8;
+	while i < 8_u8 {
+		let z = {re: x.re + (i as f64)*incr, im: x.im};
+		if mb(z) {
+			rv += 1_u8 << (7_u8 - i);
+		}
+		i += 1_u8;
+	}
+	rv
+}
+
+fn chanmb(i: uint, size: uint, ch: comm::chan<line>) -> ()
+{
+	let crv = [];
+	let incr = 2./(size as f64);
+	let y = incr*(i as f64) - 1.;
+	let xincr = 8.*incr;
+	uint::range(0_u, size/8_u) {
+		|j|
+		let x = {re: xincr*(j as f64) - 1.5, im: y};
+		crv += [fillbyte(x, incr)];
+	};
+	comm::send(ch, {i:i, b:crv});
+}
+
+fn writer(writech: comm::chan<comm::chan<line>>, size: uint)
+{
+	let p: comm::port<line> = comm::port();
+	let ch = comm::chan(p);
+	comm::send(writech, ch);
+	let cout = std::io::stdout();
+	cout.write_line("P4");
+	cout.write_line(#fmt("%u %u", size, size));
+	let lines = std::map::new_uint_hash();
+	let done = 0_u;
+	let i = 0_u;
+	while i < size {
+		let aline = comm::recv(p);
+		if aline.i == done {
+			#debug("W %u", aline.i);
+			cout.write(aline.b);
+			done += 1_u;
+			let prev = done;
+			while prev <= i {
+				if lines.contains_key(prev) {
+					#debug("WS %u", prev);
+					cout.write(lines.get(prev));
+					done += 1_u;
+					lines.remove(prev);
+					prev += 1_u;
+				}
+				else {
+					break
+				}
+			};
+			
+		}
+		else {
+			#debug("S %u", aline.i);
+			lines.insert(aline.i, aline.b);
+		};
+		i += 1_u;
+	}
+}
+
+fn main(argv: [str])
+{
+	let size = if vec::len(argv) < 2_u {
+		80u
+	}
+	else {
+		uint::from_str(argv[1])
+	};
+	let yieldevery = if vec::len(argv) < 3_u {
+		10_u
+	}
+	else {
+		uint::from_str(argv[2])
+	};
+	let writep = comm::port();
+	let writech = comm::chan(writep);
+	task::spawn {
+		|| writer(writech, size);
+	};
+	let ch = comm::recv(writep);
+	uint::range(0_u, size) {
+		|j| task::spawn {
+			|| chanmb(j, size, ch);
+		};
+		if j % yieldevery == 0_u {
+			#debug("Y %u", j);
+			task::yield();
+		};
+	};
+}