1 files changed, 73 insertions, 3 deletions

diff --git a/src/main.c b/src/main.c
index 802bfca..75dbb6a 100644
--- a/src/main.c
+++ b/src/main.c
@@ -1,15 +1,85 @@
 // Copyright (c) 2024, Marvin Borner <dev@marvinborner.de>
 // SPDX-License-Identifier: MIT
 
+#define _DEFAULT_SOURCE
 #include <stdio.h>
+#include <dirent.h>
+#include <sys/stat.h>
 
 #include <term.h>
 #include <parse.h>
 #include <print.h>
+#include <impl.h>
+
+#define N_IMPLS 5
+
+static size_t file_size(FILE *fp)
+{
+	struct stat st;
+	fstat(fileno(fp), &st);
+	return st.st_size;
+}
+
+static size_t test_impl(Impl impl, Term *term, FILE *out)
+{
+	rewind(out);
+	impl.encode(term, out);
+	size_t size = ftell(out);
+	rewind(out);
+
+	Term *recon = impl.decode(out);
+	term_diff(term, recon);
+	term_free(recon);
+
+	return size;
+}
+
+static void test_all(Impl impls[N_IMPLS], const char *dir_path, FILE *out)
+{
+	char path[1024];
+	struct dirent *dir_entry;
+	DIR *dir = opendir(dir_path);
+	while ((dir_entry = readdir(dir))) {
+		if (dir_entry->d_type != DT_REG)
+			continue;
+
+		sprintf(path, "%s/%s", dir_path, dir_entry->d_name);
+		printf("%s\n", path);
+
+		FILE *fp = fopen(path, "rb");
+
+		Term *term = parse_blc_fp(fp);
+
+		for (Impl *impl = impls; impl != impls + N_IMPLS; impl++) {
+			size_t prev = file_size(fp);
+			size_t after = test_impl(*impl, term, out);
+			printf("%s: %lu -> %lu (%f%% reduction)\n", impl->name,
+			       prev, after,
+			       (double)((long)prev - (long)after) /
+				       (double)prev * 100);
+		}
+
+		term_free(term);
+		fclose(fp);
+
+		printf("\n");
+	}
+	closedir(dir);
+}
 
 int main(void)
 {
-	Term *term = parse_blc_fp(stdin);
-	print_bruijn(term);
-	printf("\n");
+	Impl impls[N_IMPLS] = { impl_blc(), impl_blc2(), impl_closed(),
+				impl_abs(), impl_app() };
+
+	FILE *out = tmpfile();
+
+	const char *dir_path = "test/small";
+	test_all(impls, dir_path, out);
+	dir_path = "test/medium";
+	test_all(impls, dir_path, out);
+	/* dir_path = "test/large"; */
+	/* test_all(impls, dir_path, out); */
+
+	fclose(out);
 }