openmp parallelized encoding testing

1 files changed, 180 insertions, 0 deletions

diff --git a/test_encode_decode.c b/test_encode_decode.c
new file mode 100644
index 0000000..0b24e67
--- /dev/null
+++ b/test_encode_decode.c
@@ -0,0 +1,180 @@
+#include "creole.c"
+#include <string.h>
+#include <assert.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+struct seq {
+	creole_word max;
+	unsigned encode_to;
+	unsigned high_bits;
+
+	unsigned char minbuf[7];
+	unsigned char maxbuf[7];
+};
+
+void bprint(unsigned char c) {
+	int i;
+
+	for (i = 0; i < 8; i++) {
+		printf("%u", (c >> (7 - i)) & 1);
+	}
+}
+
+void bprintb(unsigned char *b, int len) {
+	while (len-- > 0) {
+		bprint(*b++);
+		printf(" ");
+	}
+}
+
+static void encode_byte_seq(struct seq *s) {
+	creole_word i = 0;
+	int j;
+	unsigned char buf[7];
+
+	for (;;) {
+
+		assert(creole_encode(i, s->encode_to, s->high_bits,
+		                        buf) == 1);
+		if (memcmp(s->minbuf, buf, s->encode_to) != 0) {
+			printf("0x%X ", i);
+			bprintb(s->minbuf, s->encode_to);
+			printf("|");
+			bprintb(buf, s->encode_to);
+			printf("\n");
+			abort();
+		}
+
+		if (i == s->max)
+			break;
+		i++;
+
+		for (j = s->encode_to - 1; j > 0; j--) {
+			if (s->minbuf[j] == 0xBF) {
+				s->minbuf[j] = 0x80;
+			} else {
+				s->minbuf[j]++;
+				break;
+			}
+		}
+
+		if (j == 0)
+			s->minbuf[0]++;
+	}
+	assert(memcmp(s->maxbuf, s->minbuf, s->encode_to) == 0);
+}
+
+static void encode_1(void) {
+	struct seq s;
+
+	s.max = 0x7F;
+	s.encode_to = 1;
+	s.high_bits = 0;
+	s.minbuf[0] = 0x00;
+	s.maxbuf[0] = 0x7F;
+
+	encode_byte_seq(&s);
+}
+
+static void encode_2(unsigned high_bits) {
+	struct seq s;
+	s.high_bits = high_bits;
+
+	memset(s.maxbuf, 0xBF, sizeof(s.maxbuf));
+	memset(s.minbuf, 0x80, sizeof(s.minbuf));
+	s.max = 0x7F;
+	s.encode_to = 2;
+	s.maxbuf[0] = s.minbuf[0] = 0xC0 | (high_bits << 1);
+	s.maxbuf[0] = 0xC1 | (high_bits << 1);
+	s.minbuf[1] = 0x80;
+	encode_byte_seq(&s);
+}
+
+static void encode_3(unsigned high_bits) {
+	struct seq s;
+	s.high_bits = high_bits;
+
+	memset(s.maxbuf, 0xBF, sizeof(s.maxbuf));
+	memset(s.minbuf, 0x80, sizeof(s.minbuf));
+	s.max = 0xFFF;
+	s.encode_to = 3;
+	s.minbuf[0] = 0xE0 | high_bits;
+	s.maxbuf[0] = 0xE0 | high_bits;
+	s.minbuf[1] = 0x80;
+	encode_byte_seq(&s);
+}
+
+static void encode_4(unsigned high_bits) {
+	struct seq s;
+	s.high_bits = high_bits;
+
+	memset(s.maxbuf, 0xBF, sizeof(s.maxbuf));
+	memset(s.minbuf, 0x80, sizeof(s.minbuf));
+	s.max = 0x1FFFF;
+	s.encode_to = 4;
+	s.maxbuf[0] = s.minbuf[0] = 0xF0 | (high_bits >> 1);
+	s.minbuf[1] = 0x80 | ((high_bits & 0x1) << 5);
+	s.maxbuf[1] = 0x9F | ((high_bits & 0x1) << 5);
+	encode_byte_seq(&s);
+}
+
+static void encode_5(unsigned high_bits) {
+	struct seq s;
+	s.high_bits = high_bits;
+
+	memset(s.maxbuf, 0xBF, sizeof(s.maxbuf));
+	memset(s.minbuf, 0x80, sizeof(s.minbuf));
+	s.max = 0x3FFFFF;
+	s.encode_to = 5;
+	s.maxbuf[0] = s.minbuf[0] = 0xF8 | (high_bits >> 2);
+	s.minbuf[1] = 0x80 | ((high_bits & 0x3) << 4);
+	s.maxbuf[1] = 0x8F | ((high_bits & 0x3) << 4);
+	encode_byte_seq(&s);
+}
+
+static void encode_6(unsigned high_bits) {
+	struct seq s;
+	s.high_bits = high_bits;
+
+	memset(s.maxbuf, 0xBF, sizeof(s.maxbuf));
+	memset(s.minbuf, 0x80, sizeof(s.minbuf));
+	s.max = 0x7FFFFFF;
+	s.encode_to = 6;
+	s.maxbuf[0] = s.minbuf[0] = 0xFC | (high_bits >> 3);
+	s.minbuf[1] = 0x80 | ((high_bits & 0x7) << 3);
+	s.maxbuf[1] = 0x87 | ((high_bits & 0x7) << 3);
+	encode_byte_seq(&s);
+}
+
+static void encode_7(unsigned high_bits) {
+	struct seq s;
+	s.high_bits = high_bits;
+
+	memset(s.maxbuf, 0xBF, sizeof(s.maxbuf));
+	memset(s.minbuf, 0x80, sizeof(s.minbuf));
+	s.max = 0xFFFFFFFF;
+	s.encode_to = 7;
+	s.maxbuf[0] = s.minbuf[0] = 0xFE;
+	s.minbuf[1] = 0x80 | (high_bits << 2);
+	s.maxbuf[1] = 0x83 | (high_bits << 2);
+	encode_byte_seq(&s);
+}
+
+static void test_encode(void) {
+	void (*tests[6])(unsigned) = {encode_2, encode_3, encode_4, encode_5, encode_6, encode_7};
+	unsigned high_bits;
+	unsigned test;
+	encode_1();
+
+# pragma omp parallel for collapse(2) num_threads(8)
+	for (high_bits = 0; high_bits < 16; high_bits++) {
+		for (test = 0; test < 6; test++)
+			tests[test](high_bits);
+	}
+}
+
+int main(void) {
+	test_encode();
+	return 0;
+}