summaryrefslogtreecommitdiff
path: root/src/util/vec.c
blob: ff4aebbb407aa365ea94bd31ba2971aeed67ee48 (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
86
87
88
89
90
91
92
93
94
95
96
97
98
#include <stddef.h>
#include <string.h>

#include "vec.h"
#include "alloc.h"

static LEO_Vec_Error LEO_Vec_grow(LEO_Vec *vec)
{
	size_t new_capacity = vec->capacity
		* LEO_Vec_GROWTH_FACTOR_NUMER
		/ LEO_Vec_GROWTH_FACTOR_DENOM;

	if (new_capacity <= vec->capacity)
		return LEO_Vec_TOO_BIG;

	vec->capacity = new_capacity;
	vec->data = vec->allocator.realloc(vec->data,
					   vec->capacity * vec->item_size);

	if (vec->data == NULL)
		return LEO_Vec_ALLOCATION_FAILURE;

	return LEO_Vec_OK;
}

LEO_Vec_Error LEO_Vec_init(LEO_Vec *vec, size_t item_size,
			   LEO_Allocator allocator)
{
	if (vec == NULL)
		return LEO_Vec_NULL_VEC_POINTER;
	if (item_size == 0)
		return LEO_Vec_INVALID_ITEM_SIZE;

	vec->allocator = allocator;
	vec->item_size = item_size;
	vec->length = 0;
	vec->capacity = LEO_Vec_DEFAULT_CAPACITY;
	vec->data = vec->allocator.alloc(vec->capacity * vec->item_size);

	if (!vec->data)
		return LEO_Vec_ALLOCATION_FAILURE;

	return LEO_Vec_OK;
}

void LEO_Vec_free(LEO_Vec *vec)
{
	if (vec == NULL)
		return;

	vec->allocator.free(vec->data);
	vec->data = NULL;
}

LEO_Vec_Error LEO_Vec_push(LEO_Vec *vec, const void *item)
{
	if (vec == NULL)
		return LEO_Vec_NULL_VEC_POINTER;
	if (item == NULL)
		return LEO_Vec_NULL_ITEM_POINTER;

	if (vec->length >= vec->capacity) {
		LEO_Vec_Error result = LEO_Vec_grow(vec);
		if (result != LEO_Vec_OK)
			return result;
	}

	unsigned char *dest = (unsigned char *)vec->data + (vec->length++) * vec->item_size;
	memcpy(dest, item, vec->item_size);

	return LEO_Vec_OK;
}

LEO_Vec_Error LEO_Vec_pop(LEO_Vec *vec, void *item)
{
	if (vec == NULL)
		return LEO_Vec_NULL_VEC_POINTER;
	if (vec->length == 0)
		return LEO_Vec_OUT_OF_BOUNDS;

	void *src = vec->data + (vec->length--) * vec->item_size;
	memcpy(item, src, vec->item_size);

	return LEO_Vec_OK;
}

LEO_Vec_Error LEO_Vec_get(LEO_Vec *vec, size_t i, void *item)
{
	if (vec == NULL)
		return LEO_Vec_NULL_VEC_POINTER;
	if (i >= vec->length)
		return LEO_Vec_OUT_OF_BOUNDS;

	unsigned char *src = (unsigned char *) vec->data + i * vec->item_size;
	memcpy(item, src, vec->item_size);

	return LEO_Vec_OK;
}