aboutsummaryrefslogtreecommitdiff
path: root/set/set.ha
blob: 504d71682b37758d1f51ed488543588c673b1c24 (plain)
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
export type set = struct {
	items: []size,
};

export type nosuchitem = !void;

export fn add(s: *set, item: size) bool = {
	for (let i = 0z; i < len(s.items); i += 1) {
		if (s.items[i] == item) {
			return false;
		};
	};
	append(s.items, item);
	return true;
};

export fn del(s: *set, item: size) bool = {
	match (contains(*s, item)) {
	case let i: size =>
		delete(s.items[i]);
		return true;
	case nosuchitem =>
		return false;
	};
};

export fn contains(s: set, item: size) (size | nosuchitem) = {
	for (let i = 0z; i < len(s.items); i += 1) {
		if (s.items[i] == item) {
			return i;
		};
	};
	return nosuchitem;
};

// Clears all items.
export fn clear(s: *set) void = {
	delete(s.items[..]);
};

// Free the underlying slice.
export fn finish(s: *set) void = {
	free(s.items);
};

@test fn add() void = {
	let s = set {...};
	assert(add(&s, 1z));
	assert(len(s.items) == 1);
	assert(!add(&s, 1z));
	assert(len(s.items) == 1);
};

@test fn del() void = {
	let s = set {...};
	assert(add(&s, 1z));
	assert(len(s.items) == 1);
	assert(del(&s, 1z));
	assert(len(s.items) == 0);
	assert(!del(&s, 1z));
	assert(len(s.items) == 0);
};

@test fn contains() void = {
	let s = set {...};
	assert(add(&s, 1z));
	assert(!add(&s, 1z));
	const c = contains(s, 1z);
	assert(c is size);
	assert(c as size == 0);
	assert(add(&s, 2z));
	const c = contains(s, 2z);
	assert(c is size);
	assert(c as size == 1);
};

@test fn clear() void = {
	let s = set {...};
	assert(add(&s, 1z));
	assert(len(s.items) == 1);
	assert(!add(&s, 1z));
	assert(len(s.items) == 1);
	clear(&s);
	assert(len(s.items) == 0);
};