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
|
#include "Array.h"
#include "Types.h"
#include "GlobalArena.h"
#include <stdlib.h>
#include <string.h>
struct ArrayMeta {
BumpArena b;
usize element_size;
usize count;
usize length;
};
static inline struct ArrayMeta*
A_metadata(void* arr) {
return arr - sizeof(struct ArrayMeta);
}
typedef void* Array;
Array
A_new(BumpArena b, usize element_size, usize length) {
if (NULL == b)
return NULL;
struct ArrayMeta am = {
.b = b,
.element_size = element_size,
.count = 0,
.length = length,
};
void* start = B_alloc(b, sizeof(am) + element_size * length);
memcpy(start, &am, sizeof(am));
return start + sizeof(am);
}
usize
A_length(Array arr) {
if (NULL == arr)
return 0;
return A_metadata(arr)->length;
}
usize
A_count(Array arr) {
if (NULL == arr)
return 0;
return A_metadata(arr)->count;
}
Result
A_push(Array arr, void *value) {
if (NULL == arr)
return R_Uninitialized;
struct ArrayMeta *am = A_metadata(arr);
if (am->count >= am->length) {
if (!B_resizeable(am->b, arr))
abort();
B_resize(am->b, sizeof(*am) + am->element_size * (am->length <<= 2));
}
memcpy(arr + am->count++ * am->element_size, value, am->element_size);
return R_Ok;
}
void*
A_pop(Array arr) {
if (NULL == arr)
return NULL;
struct ArrayMeta *am = A_metadata(arr);
if (am->count == 0)
return NULL;
return arr + am->element_size * --am->count;
}
Array
AG_new(usize element_size, usize length) {
return A_new(g_arena, element_size, length);
}
|
