aboutsummaryrefslogtreecommitdiff
path: root/src/impl/blc2.c
blob: 3cf7ff8a58800c67bb285a16342c5f731037b5ef (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
// Copyright (c) 2024, Marvin Borner <dev@marvinborner.de>
// SPDX-License-Identifier: MIT

#include <log.h>
#include <impl.h>
#include <stdint.h>

// haskell by John Tromp
static void levenshtein_encode(size_t n, FILE *fp)
{
	if (n == 0) {
		fprintf(fp, "10");
		return;
	}

	fprintf(fp, "1");

	size_t length = (8 * sizeof(n)) - __builtin_clzll(n);
	levenshtein_encode(length - 1, fp);

	if (length == 1)
		return;

	for (size_t i = 1 << (length - 2); i != 0; i >>= 1)
		fprintf(fp, "%d", (i & n) != 0);
}

static void encode(Term *term, FILE *fp)
{
	switch (term->type) {
	case ABS:
		fprintf(fp, "00");
		encode(term->u.abs.body, fp);
		break;
	case APP:
		fprintf(fp, "01");
		encode(term->u.app.lhs, fp);
		encode(term->u.app.rhs, fp);
		break;
	case IDX:
		levenshtein_encode(term->u.index, fp);
		break;
	default:
		fatal("invalid type %d\n", term->type);
	}
}

// by John Tromp
static size_t levenshtein_decode(FILE *fp)
{
	if (getc(fp) == '0')
		return 0;
	size_t x, l = levenshtein_decode(fp);
	for (x = 1; l--;)
		x = 2 * x + (getc(fp) != '0');
	return x;
}

static Term *decode(FILE *fp)
{
	Term *res = 0;

	char a = getc(fp);

	if (a == '0') {
		char b = getc(fp);
		if (b == '0') {
			res = term_new(ABS);
			res->u.abs.body = decode(fp);
		} else if (b == '1') {
			res = term_new(APP);
			res->u.app.lhs = decode(fp);
			res->u.app.rhs = decode(fp);
		}
	} else if (a == '1') {
		res = term_new(IDX);
		res->u.index = levenshtein_decode(fp);
	}

	if (!res)
		fatal("invalid parsing state!\n");

	return res;
}

Impl impl_blc2(void)
{
	return (Impl){
		.name = "blc2",
		.encode = encode,
		.decode = decode,
	};
}