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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
#include "../../include/std/buffer.h"
#include "../../include/std/alloc.h"
#include "../../include/std/mem.h"
#include "../../include/std/arena_allocator.h"
#include <assert.h>
int buffer_init(Buffer *self, struct ArenaAllocator *allocator) {
self->data = NULL;
self->size = 0;
self->capacity = 0;
self->allocator = allocator;
if(allocator) {
return arena_allocator_add_mem(allocator, &self->allocator_index);
} else {
self->allocator_index = ~(usize)0;
return 0;
}
}
void buffer_deinit(Buffer *self) {
assert(!self->allocator && "Can't deinit buffer if it has an allocator");
am_free(self->data);
self->data = NULL;
self->size = 0;
self->capacity = 0;
}
static CHECK_RESULT int buffer_ensure_capacity(Buffer *self, usize new_capacity) {
usize capacity;
void *new_mem;
if(self->capacity >= new_capacity)
return BUFFER_OK;
capacity = self->capacity;
if(capacity == 0) {
capacity = new_capacity;
} else {
double cap = capacity;
const double new_cap = new_capacity;
while(cap < new_cap) {
cap *= 1.5;
}
capacity = cap;
}
if(am_realloc(self->data, capacity, &new_mem) != ALLOC_OK)
return BUFFER_ALLOC_FAIL;
self->data = new_mem;
self->capacity = capacity;
/* Update list of buffers in the allocator with the new address of the buffer data */
if(self->allocator)
am_memcpy(self->allocator->mems.data + self->allocator_index, &self->data, sizeof(void*));
return BUFFER_OK;
}
int buffer_append(Buffer *self, const void *data, usize size) {
assert(data);
return_if_error(buffer_ensure_capacity(self, self->size + size));
am_memcpy(self->data + self->size, data, size);
self->size += size;
return BUFFER_OK;
}
int buffer_append_empty(Buffer *self, usize size) {
return_if_error(buffer_ensure_capacity(self, self->size + size));
self->size += size;
return BUFFER_OK;
}
int buffer_expand(Buffer *self, usize size) {
return buffer_ensure_capacity(self, self->size + size);
}
void* buffer_get(Buffer *self, usize index, usize type_size) {
usize real_index;
real_index = index * type_size;
assert(real_index < self->size);
return &self->data[real_index];
}
int buffer_pop(Buffer *self, void *data, usize size) {
if(size > self->size)
return -1;
am_memcpy(data, &self->data[self->size - size], size);
self->size -= size;
return 0;
}
void buffer_clear(Buffer *self) {
self->size = 0;
}
int buffer_set_capacity(Buffer *self, usize new_capacity) {
if(am_realloc(self->data, new_capacity, (void**)&self->data) != ALLOC_OK)
return BUFFER_ALLOC_FAIL;
self->capacity = new_capacity;
if(self->size < self->capacity)
self->size = self->capacity;
/* Update list of buffers in the allocator with the new address of the buffer data */
if(self->allocator)
am_memcpy(self->allocator->mems.data + self->allocator_index, &self->data, sizeof(void*));
return BUFFER_OK;
}
void* buffer_begin(Buffer *self) {
return self->data;
}
void *buffer_end(Buffer *self) {
return self->data + self->size;
}
usize __buffer_get_size(Buffer *self, usize type_size) {
assert(type_size != 0);
return self->size / type_size;
}
|
