a

1 files changed, 28 insertions, 0 deletions

diff --git a/tests/programs/enhance.zc b/tests/programs/enhance.zc
new file mode 100644
index 0000000..327b4bd
--- /dev/null
+++ b/tests/programs/enhance.zc
@@ -0,0 +1,28 @@
+// custom opcodes: recipes, lifted vm code, several outputs, and a fallback
+enhance clampadd(a, b) micro "a = add a b; a = min a 100; a = max a 0";
+enhance cube(x) vm { 0 $ 0 $ * * };
+enhance 2 divmod(a, b) micro "r2 = sdiv a b" "\n r1 = srem a b\n r0 = r2";
+enhance div7(a) micro "a = sdiv a 7";
+enhance urem10(a) micro "a = urem a 10";
+enhance hi(a, b) micro "a = mulhu a b";
+enhance noisy(x) vm { 0 $ . } else soft_noisy;	// i/o can't be lifted, so calls use the fallback
+fn soft_noisy(x) { return x + 1; }
+let qq, rr, seed = 7;
+fn rnd() { seed = seed * 1103515245 + 12345; return seed ^ (seed >> 11); }
+fn main() {
+	let i = 0, a, bad = 0;
+	put_n(clampadd(70, 50)); put_c(' '); put_n(clampadd(-50, 2)); put_c(' '); put_n(cube(7)); put_c(' ');
+	asm { 17 5 divmod rr ! qq ! }
+	put_n(qq); put_c(','); put_n(rr); put_c(' ');
+	put_n(noisy(4)); put_c(' ');
+	put_n(hi(-1, -1)); put_c(' ');
+	while (i < 5000) {
+		a = rnd();
+		if (div7(a) != zdiv(a, 7)) bad++;
+		if (urem10(a) < 0 || urem10(a) > 9) bad++;
+		if (a / (i + 1) != zdiv(a, i + 1)) bad++;
+		i++;
+	}
+	put_n(bad); put_c(10);
+	return 0;
+}